diff --git a/src/main/java/edu/lu/uni/serval/fixminer/jobs/EnhancedASTDiff.java b/src/main/java/edu/lu/uni/serval/fixminer/jobs/EnhancedASTDiff.java index 4bb7da3..524f09b 100755 --- a/src/main/java/edu/lu/uni/serval/fixminer/jobs/EnhancedASTDiff.java +++ b/src/main/java/edu/lu/uni/serval/fixminer/jobs/EnhancedASTDiff.java @@ -41,26 +41,39 @@ public class EnhancedASTDiff { JedisPool innerPool = new JedisPool(PoolBuilder.getPoolConfig(), "127.0.0.1",Integer.valueOf(portInner),20000000); - List> allPredicates = new ArrayList>(); - for (String s : projectList) { - Predicate predicate = x->x.getName().endsWith(s); - allPredicates.add(predicate); - } - boolean isJava = false; if (projectType.equals("java")){ isJava =true; } - File folder = new File(inputPath); File[] listOfFiles = folder.listFiles(); - Stream stream = Arrays.stream(listOfFiles); - List folders = stream - .filter(x -> !x.getName().startsWith(".")) - .filter(x -> !x.getName().startsWith("cocci")) - .filter(x -> !x.getName().endsWith(".index")) - .filter(allPredicates.stream().reduce(x->false, Predicate::or)) - .collect(Collectors.toList()); + Stream stream = Arrays.stream(listOfFiles); + List folders; + if (projectList.length == 1 && projectList[0].equals("")){ + folders = stream + .filter(x -> !x.getName().startsWith(".")) + .filter(x -> !x.getName().startsWith("cocci")) + .filter(x -> !x.getName().endsWith(".index")) + .collect(Collectors.toList()); + } + else { + List> allPredicates = new ArrayList>(); + for (String s : projectList) { + Predicate predicate = x -> x.getName().endsWith(s); + allPredicates.add(predicate); + } + folders = stream + .filter(x -> !x.getName().startsWith(".")) + .filter(x -> !x.getName().startsWith("cocci")) + .filter(x -> !x.getName().endsWith(".index")) + .filter(allPredicates.stream().reduce(x->false, Predicate::or)) + .collect(Collectors.toList()); + } + + + + + String project = folder.getName(); List allMessageFiles = new ArrayList<>();