Merge branch 'saledouble-master_add_entry_point'

Author: khatchad

edu.cuny.hunter.streamrefactoring.core/src/edu/cuny/hunter/streamrefactoring/core/analysis/StreamAnalyzer.java

@@ -55,6 +55,8 @@ private static void addImplicitEntryPoints(Collection<Entrypoint> target, Iterab
 	 */
 	private Map<EclipseProjectAnalysisEngine<InstanceKey>, Collection<Entrypoint>> enginesWithBuiltCallGraphsToEntrypointsUsed = new HashMap<>();
 
+	private boolean findImplicitBenchmarkEntryPoints;
+
 	private boolean findImplicitEntryPoints = true;
 
 	private boolean findImplicitTestEntryPoints;
@@ -74,9 +76,11 @@ public StreamAnalyzer(boolean visitDocTags, boolean findImplicitEntryPoints) {
 		this.findImplicitEntryPoints = findImplicitEntryPoints;
 	}
 
-	public StreamAnalyzer(boolean visitDocTags, boolean findImplicitEntryPoints, boolean findImplicitTestEntryPoints) {
+	public StreamAnalyzer(boolean visitDocTags, boolean findImplicitEntryPoints, boolean findImplicitTestEntryPoints,
+			boolean findImplicitBenchmarkEntryPoints) {
 		this(visitDocTags, findImplicitEntryPoints);
 		this.findImplicitTestEntryPoints = findImplicitTestEntryPoints;
+		this.findImplicitBenchmarkEntryPoints = findImplicitBenchmarkEntryPoints;
 	}
 
 	/**
@@ -202,6 +206,14 @@ protected Collection<Entrypoint> buildCallGraph(EclipseProjectAnalysisEngine<Ins
 				addImplicitEntryPoints(entryPoints, jUnitEntryPoints);
 			}
 
+			if (this.findImplicitBenchmarkEntryPoints) {
+				// try to find benchmark entry points.
+				Set<Entrypoint> benchmarkEntryPoints = Util.findBenchmarkEntryPoints(engine.getClassHierarchy());
+
+				// add them as well.
+				addImplicitEntryPoints(entryPoints, benchmarkEntryPoints);
+			}
+
 			if (entryPoints.isEmpty()) {
 				LOGGER.warning(() -> "Project: " + engine.getProject().getElementName() + " has no entry points.");
 				return entryPoints;