changes.

[IRC.git] / Robust / src / Analysis / SSJava / LocationSummary.java

diff --git a/Robust/src/Analysis/SSJava/LocationSummary.java b/Robust/src/Analysis/SSJava/LocationSummary.java
index f6aefddc367dbdb3bc8ef55ca321583a9d7c1eaa..e0f953a48cf36443b065becc1d60f7a0b4cda37f 100644 (file)
--- 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);
     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) {