Merge pull request #21 from Haiyang-Sun/sourcemapping-improve

make the iidToLocation string cache on-demand to save memory
Haiyang-Sun · Jan 21, 2019 · 33d35c4 · 33d35c4
2 parents b1b9f88 + 5288419
commit 33d35c4
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/src/ch.usi.inf.nodeprof/src/ch/usi/inf/nodeprof/utils/SourceMapping.java b/src/ch.usi.inf.nodeprof/src/ch/usi/inf/nodeprof/utils/SourceMapping.java
@@ -55,17 +55,22 @@ public static int getIIDForSourceSection(SourceSection sourceSection) {
         }
         int newIId = ++iidGen;
         assert (newIId < Integer.MAX_VALUE);
-        StringBuilder b = makeLocationString(sourceSection);
-
-        iidMap.put(newIId, b.toString());
         sourceSet.put(sourceSection, newIId);
         idToSource.put(newIId, sourceSection);
         return newIId;
     }
 
     @TruffleBoundary
     public static String getLocationForIID(int iid) {
-        return iidMap.get(iid);
+        if (iidMap.containsKey(iid)) {
+            return iidMap.get(iid);
+        } else if (idToSource.containsKey(iid)) {
+            String res = makeLocationString(idToSource.get(iid)).toString();
+            iidMap.put(iid, res);
+            return res;
+        } else {
+            return null;
+        }
     }
 
     @TruffleBoundary