changes: now Inference engine works fine with the EyeTracking benchmark.
[IRC.git] / Robust / src / Analysis / SSJava / LocationSummary.java
1 package Analysis.SSJava;
2
3 import java.util.HashMap;
4 import java.util.HashSet;
5 import java.util.Map;
6 import java.util.Set;
7
8 public abstract class LocationSummary {
9
10   Map<String, String> mapHNodeNameToLocationName;
11   Map<String, Set<String>> mapLocationNameToHNodeNameSet;
12
13   public LocationSummary() {
14     mapHNodeNameToLocationName = new HashMap<String, String>();
15     mapLocationNameToHNodeNameSet = new HashMap<String, Set<String>>();
16   }
17
18   public void addMapHNodeNameToLocationName(String nodeName, String locName) {
19     // System.out.println("nodeName="+nodeName+"  locName="+locName);
20     mapHNodeNameToLocationName.put(nodeName, locName);
21
22     if (!mapLocationNameToHNodeNameSet.containsKey(locName)) {
23       mapLocationNameToHNodeNameSet.put(locName, new HashSet<String>());
24     }
25     mapLocationNameToHNodeNameSet.get(locName).add(nodeName);
26   }
27
28   public Set<String> getHNodeNameSetByLatticeLoationName(String locName) {
29     return mapLocationNameToHNodeNameSet.get(locName);
30   }
31
32   public String getLocationName(String nodeName) {
33     if (!mapHNodeNameToLocationName.containsKey(nodeName)) {
34       mapHNodeNameToLocationName.put(nodeName, nodeName);
35     }
36     return mapHNodeNameToLocationName.get(nodeName);
37   }
38
39   public Map<String, String> getMapHNodeNameToLocationName() {
40     return mapHNodeNameToLocationName;
41   }
42
43 }