Update ConstructGroum

newrui · newrui · commit 35a916d8e85e · 2019-04-02T09:44:53.000+08:00
diff --git a/src/ConstructGroum.java b/src/ConstructGroum.java
@@ -41,8 +41,10 @@ public Groum constructGroum(String groumPath,Map<String,String> apiMap){
 //                       int startLine = Integer.parseInt(startLineString);
 //                       int endLine = Integer.parseInt(endLineString);
 //                       GroumNode node = new GroumNode(id,api,originApi,startLine,endLine);
-                       GroumNode node = new GroumNode(id,api,originApi,-1,-1);
-                       groum.addNode(node);
+                       if(api.equals("-1")) {
+                           GroumNode node = new GroumNode(id, api, originApi, -1, -1);
+                           groum.addNode(node);
+                       }
                    }
 
 
diff --git a/src/WritePathResult.java b/src/WritePathResult.java
@@ -31,13 +31,15 @@ public static void ergodicDir(File dir, PrintWriter targetFilePW, Map<String, St
     private static void writePath(File groumFile, PrintWriter targetFilePW, Map<String, String> API2IndexMap) throws IOException {
         ConstructGroum CG = new ConstructGroum();
         Groum groum = CG.constructGroum(groumFile.getAbsolutePath(), API2IndexMap);
-        Map<String, GroumNode> nodeMap = groum.getNodeMap();
-        List<String> startList = null;
-        for (String id : nodeMap.keySet()) {
-            startList = new ArrayList<>();
-            startList.add(id);
-            List<List<String>> outList = GetPath.getAllPath(groum, startList, 4);
-            writeFile(outList, targetFilePW);
+        if(groum != null) {
+            Map<String, GroumNode> nodeMap = groum.getNodeMap();
+            List<String> startList = null;
+            for (String id : nodeMap.keySet()) {
+                startList = new ArrayList<>();
+                startList.add(id);
+                List<List<String>> outList = GetPath.getAllPath(groum, startList, 4);
+                writeFile(outList, targetFilePW);
+            }
         }
     }
 
