X-Git-Url: http://plrg.eecs.uci.edu/git/?a=blobdiff_plain;f=Robust%2Fsrc%2FAnalysis%2FSSJava%2FLocationSummary.java;h=e0f953a48cf36443b065becc1d60f7a0b4cda37f;hb=30c0c69af4e45824aae96fb417663da1567d1b64;hp=4682b2312a07f0ef24fbf587849fe936c20a3552;hpb=9cdbca1a3dcc949bee74ee42ddd8febff6048b9f;p=IRC.git

diff --git a/Robust/src/Analysis/SSJava/LocationSummary.java b/Robust/src/Analysis/SSJava/LocationSummary.java
index 4682b231..e0f953a4 100644
--- a/Robust/src/Analysis/SSJava/LocationSummary.java
+++ b/Robust/src/Analysis/SSJava/LocationSummary.java
@@ -1,19 +1,32 @@
 package Analysis.SSJava;
 
 import java.util.HashMap;
+import java.util.HashSet;
 import java.util.Map;
+import java.util.Set;
 
 public abstract class LocationSummary {
 
   Map<String, String> mapHNodeNameToLocationName;
+  Map<String, Set<String>> mapLocationNameToHNodeNameSet;
 
   public LocationSummary() {
     mapHNodeNameToLocationName = new HashMap<String, String>();
+    mapLocationNameToHNodeNameSet = new HashMap<String, Set<String>>();
   }
 
   public void addMapHNodeNameToLocationName(String nodeName, String locName) {
-    System.out.println("nodeName="+nodeName+"  locName="+locName);
+    // System.out.println("nodeName="+nodeName+"  locName="+locName);
     mapHNodeNameToLocationName.put(nodeName, locName);
+
+    if (!mapLocationNameToHNodeNameSet.containsKey(locName)) {
+      mapLocationNameToHNodeNameSet.put(locName, new HashSet<String>());
+    }
+    mapLocationNameToHNodeNameSet.get(locName).add(nodeName);
+  }
+
+  public Set<String> getHNodeNameSetByLatticeLoationName(String locName) {
+    return mapLocationNameToHNodeNameSet.get(locName);
   }
 
   public String getLocationName(String nodeName) {