changes on global composite assignment translation and pcloc generation
[IRC.git] / Robust / src / Analysis / SSJava / BuildLattice.java
1 package Analysis.SSJava;
2
3 import java.util.HashMap;
4 import java.util.HashSet;
5 import java.util.Iterator;
6 import java.util.Map;
7 import java.util.Set;
8
9 import IR.Descriptor;
10 import IR.MethodDescriptor;
11 import Util.Pair;
12
13 public class BuildLattice {
14
15   private LocationInference infer;
16
17   public BuildLattice(LocationInference infer) {
18     this.infer = infer;
19   }
20
21   public SSJavaLattice<String> buildLattice(Descriptor desc) {
22
23     HierarchyGraph inputGraph = infer.getSkeletonCombinationHierarchyGraph(desc);
24     LocationSummary locSummary = infer.getLocationSummary(desc);
25
26     Set<HNode> nodeSetWithCompositeLocation = new HashSet<HNode>();
27     if (desc instanceof MethodDescriptor) {
28       FlowGraph flowGraph = infer.getFlowGraph((MethodDescriptor) desc);
29
30       for (Iterator iterator = inputGraph.getNodeSet().iterator(); iterator.hasNext();) {
31         HNode hnode = (HNode) iterator.next();
32         Descriptor hnodeDesc = hnode.getDescriptor();
33         if (hnodeDesc != null) {
34           NTuple<Descriptor> descTuple = new NTuple<Descriptor>();
35           descTuple.add(hnodeDesc);
36
37           if (flowGraph.contains(descTuple)) {
38             FlowNode flowNode = flowGraph.getFlowNode(descTuple);
39             if (flowNode.getCompositeLocation() != null) {
40               nodeSetWithCompositeLocation.add(hnode);
41             }
42           }
43
44         }
45       }
46
47     }
48
49     BasisSet basisSet = inputGraph.computeBasisSet(nodeSetWithCompositeLocation);
50     debug_print(inputGraph);
51
52     Family family = generateFamily(basisSet);
53     Map<Set<Integer>, Set<Set<Integer>>> mapImSucc = coveringGraph(basisSet, family);
54
55     SSJavaLattice<String> lattice = buildLattice(basisSet, inputGraph, locSummary, mapImSucc);
56     return lattice;
57
58   }
59
60   private SSJavaLattice<String> buildLattice(BasisSet basisSet, HierarchyGraph inputGraph,
61       LocationSummary locSummary, Map<Set<Integer>, Set<Set<Integer>>> mapImSucc) {
62
63     SSJavaLattice<String> lattice =
64         new SSJavaLattice<String>(SSJavaAnalysis.TOP, SSJavaAnalysis.BOTTOM);
65
66     Map<Set<Integer>, String> mapFToLocName = new HashMap<Set<Integer>, String>();
67
68     Set<Set<Integer>> keySet = mapImSucc.keySet();
69     for (Iterator iterator = keySet.iterator(); iterator.hasNext();) {
70       Set<Integer> higher = (Set<Integer>) iterator.next();
71
72       String higherName = generateElementName(basisSet, inputGraph, mapFToLocName, higher);
73
74       HNode higherNode = inputGraph.getHNode(higherName);
75       if (higherNode != null && higherNode.isSharedNode()) {
76         lattice.addSharedLoc(higherName);
77       }
78       Set<Descriptor> descSet = inputGraph.getDescSetOfNode(higherNode);
79       // System.out.println("higherName=" + higherName + "  higherNode=" + higherNode + "  descSet="
80       // + descSet);
81       for (Iterator iterator2 = descSet.iterator(); iterator2.hasNext();) {
82         Descriptor d = (Descriptor) iterator2.next();
83         locSummary.addMapHNodeNameToLocationName(d.getSymbol(), higherName);
84       }
85       // locSummary.addMapHNodeNameToLocationName(higherName, higherName);
86
87       Set<Set<Integer>> lowerSet = mapImSucc.get(higher);
88       for (Iterator iterator2 = lowerSet.iterator(); iterator2.hasNext();) {
89         Set<Integer> lower = (Set<Integer>) iterator2.next();
90
91         String lowerName = generateElementName(basisSet, inputGraph, mapFToLocName, lower);
92         HNode lowerNode = inputGraph.getHNode(lowerName);
93         if (lowerNode != null && lowerNode.isSharedNode()) {
94           lattice.addSharedLoc(lowerName);
95         }
96
97         Set<Descriptor> lowerDescSet = inputGraph.getDescSetOfNode(lowerNode);
98         // System.out.println("lowerName=" + lowerName + "  lowerNode=" + lowerNode + "  descSet="
99         // + lowerDescSet);
100         for (Iterator iterator3 = lowerDescSet.iterator(); iterator3.hasNext();) {
101           Descriptor d = (Descriptor) iterator3.next();
102           locSummary.addMapHNodeNameToLocationName(d.getSymbol(), lowerName);
103         }
104         // locSummary.addMapHNodeNameToLocationName(lowerName, lowerName);
105
106         if (higher.size() == 0) {
107           // empty case
108           lattice.put(lowerName);
109         } else {
110           lattice.addRelationHigherToLower(higherName, lowerName);
111         }
112
113       }
114
115     }
116
117     return lattice;
118   }
119
120   public HNode getCombinationNodeInSCGraph(Descriptor desc, HNode nodeFromSimpleGraph) {
121
122     HierarchyGraph scGraph = infer.getSkeletonCombinationHierarchyGraph(desc);
123
124     if (nodeFromSimpleGraph.isSkeleton()) {
125       return scGraph.getCurrentHNode(nodeFromSimpleGraph);
126     }
127
128     Set<HNode> combineSkeletonNodeSet =
129         infer.getSimpleHierarchyGraph(desc).getCombineSetByCombinationNode(nodeFromSimpleGraph);
130     HNode combinationNodeInSCGraph =
131         infer.getSkeletonCombinationHierarchyGraph(desc).getMapCombineNodeSetToCombinationNode()
132             .get(combineSkeletonNodeSet);
133
134     // Set<HNode> combineSkeletonNodeSet =
135     // infer.getSimpleHierarchyGraph(desc).getCombineSetByCombinationNode(simpleGraphNode);
136     // HNode combinationNodeInSCGraph =
137     // infer.getSkeletonCombinationHierarchyGraph(desc).getCombinationNode(combineSkeletonNodeSet);
138     return combinationNodeInSCGraph;
139   }
140
141   public SSJavaLattice<String> insertIntermediateNodesToStraightLine(Descriptor desc,
142       SSJavaLattice<String> skeletonLattice) {
143
144     // perform DFS that starts from each skeleton/combination node and ends by another
145     // skeleton/combination node
146
147     HierarchyGraph simpleGraph = infer.getSimpleHierarchyGraph(desc);
148     HierarchyGraph scGraph = infer.getSkeletonCombinationHierarchyGraph(desc);
149     LocationSummary locSummary = infer.getLocationSummary(desc);
150
151     SSJavaLattice<String> lattice = skeletonLattice.clone();
152
153     Set<HNode> visited = new HashSet<HNode>();
154
155     Set<HNode> nodeSet = simpleGraph.getNodeSet();
156
157     Map<TripleItem, String> mapIntermediateLoc = new HashMap<TripleItem, String>();
158
159     System.out.println("*insert=" + desc);
160     System.out.println("***nodeSet=" + nodeSet);
161     for (Iterator iterator = nodeSet.iterator(); iterator.hasNext();) {
162       HNode node = (HNode) iterator.next();
163       System.out.println("node=" + node);
164
165       if (node.isSkeleton() && (!visited.contains(node))) {
166         visited.add(node);
167
168         Set<HNode> outSet = simpleGraph.getOutgoingNodeSet(node);
169         for (Iterator iterator2 = outSet.iterator(); iterator2.hasNext();) {
170           HNode outNode = (HNode) iterator2.next();
171
172           if (!outNode.isSkeleton()) {
173             if (outNode.isCombinationNode()) {
174               expandCombinationNode(desc, lattice, visited, mapIntermediateLoc, locSummary, outNode);
175             } else {
176               // we have a node that is neither combination or skeleton node
177               // System.out.println("%%%skeleton node=" + node + "  outNode=" + outNode);
178               HNode startNode = scGraph.getCurrentHNode(node);
179
180               // if (node.getDescriptor() != null) {
181               // // node is a skeleton node and it might be merged into another node in the SC
182               // graph.
183               // startNode = scGraph.getHNode(node.getDescriptor());
184               // } else {
185               // // this node has already been merged before the SC graph.
186               // startNode = node;
187               // }
188
189               Set<HNode> endNodeSetFromSimpleGraph =
190                   simpleGraph.getDirectlyReachableSkeletonCombinationNodeFrom(outNode, null);
191
192               // System.out.println("endNodeSetFromSimpleGraph=" + endNodeSetFromSimpleGraph
193               // + "   from=" + outNode);
194               Set<HNode> endCombNodeSet = new HashSet<HNode>();
195               for (Iterator iterator3 = endNodeSetFromSimpleGraph.iterator(); iterator3.hasNext();) {
196                 HNode endNode = (HNode) iterator3.next();
197                 endCombNodeSet.add(getCombinationNodeInSCGraph(desc, endNode));
198               }
199               // System.out.println("endCombNodeSet=" + endCombNodeSet);
200               visited.add(outNode);
201               if (endCombNodeSet.size() > 0) {
202                 // follows the straight line up to another skeleton/combination node
203                 endCombNodeSet = removeTransitivelyReachToNode(desc, startNode, endCombNodeSet);
204               } else if (endCombNodeSet.size() == 0) {
205                 // the outNode is (directly/transitively) connected to the bottom node
206                 // therefore, we just add a dummy bottom HNode to the endCombNodeSet.
207                 endCombNodeSet.add(LocationInference.BOTTOMHNODE);
208               }
209
210               recurDFSNormalNode(desc, lattice, startNode, endCombNodeSet, visited,
211                   mapIntermediateLoc, 1, locSummary, outNode);
212             }
213
214           }
215
216         }
217       } else if (!node.isSkeleton() && !node.isCombinationNode() && !node.isMergeNode()
218           && !visited.contains(node)) {
219
220         System.out.println("n=" + node);
221
222         // an intermediate node 'node' may be located between "TOP" location and a skeleton node
223         if (simpleGraph.getIncomingNodeSet(node).size() == 0) {
224
225           // this node will be directly connected to the TOP location
226           // start adding the following nodes from this node
227
228           Set<HNode> endNodeSetFromSimpleGraph =
229               simpleGraph.getDirectlyReachableSkeletonCombinationNodeFrom(node, null);
230
231           Set<HNode> endCombNodeSet = new HashSet<HNode>();
232           for (Iterator iterator3 = endNodeSetFromSimpleGraph.iterator(); iterator3.hasNext();) {
233             HNode endNode = (HNode) iterator3.next();
234             endCombNodeSet.add(getCombinationNodeInSCGraph(desc, endNode));
235           }
236
237           System.out.println("endCombNodeSet=" + endCombNodeSet);
238           HNode startNode = LocationInference.TOPHNODE;
239           visited.add(startNode);
240           if (endCombNodeSet.size() > 0) {
241             // follows the straight line up to another skeleton/combination node
242             // endCombNodeSet = removeTransitivelyReachToNode(desc, node, endCombNodeSet);
243             recurDFSNormalNode(desc, lattice, startNode, endCombNodeSet, visited,
244                 mapIntermediateLoc, 1, locSummary, node);
245           }
246
247         }
248
249       }
250     }
251
252     return lattice;
253
254   }
255
256   private void expandCombinationNode(Descriptor desc, SSJavaLattice<String> lattice,
257       Set<HNode> visited, Map<TripleItem, String> mapIntermediateLoc, LocationSummary locSummary,
258       HNode cnode) {
259
260     System.out.println("expandCombinationNode=" + cnode);
261     // expand the combination node 'outNode'
262     // here we need to expand the corresponding combination location in the lattice
263     HNode combinationNodeInSCGraph = getCombinationNodeInSCGraph(desc, cnode);
264
265     HierarchyGraph simpleGraph = infer.getSimpleHierarchyGraph(desc);
266
267     Set<HNode> combineSkeletonNodeSet = simpleGraph.getCombineSetByCombinationNode(cnode);
268
269     // System.out.println("combineSkeletonNodeSet=" + combineSkeletonNodeSet);
270
271     Set<HNode> combinationNodeSet =
272         simpleGraph.getCombinationNodeSetByCombineNodeSet(combineSkeletonNodeSet);
273
274     // System.out.println("combinationNodeSet=" + combinationNodeSet);
275
276     Set<HNode> endNodeSetFromSimpleGraph =
277         simpleGraph.getDirectlyReachableSkeletonCombinationNodeFrom(cnode, combinationNodeSet);
278     // System.out.println("-endNodeSetFromSimpleGraph=" + endNodeSetFromSimpleGraph);
279     Set<HNode> endCombNodeSet = new HashSet<HNode>();
280     for (Iterator iterator3 = endNodeSetFromSimpleGraph.iterator(); iterator3.hasNext();) {
281       HNode endNode = (HNode) iterator3.next();
282       endCombNodeSet.add(getCombinationNodeInSCGraph(desc, endNode));
283     }
284     visited.add(cnode);
285
286     // follows the straight line up to another skeleton/combination node
287     if (endCombNodeSet.size() > 0) {
288       System.out.println("---endCombNodeSet=" + endCombNodeSet);
289       endCombNodeSet =
290           removeTransitivelyReachToNode(desc, combinationNodeInSCGraph, endCombNodeSet);
291
292       recurDFS(desc, lattice, combinationNodeInSCGraph, endCombNodeSet, visited,
293           mapIntermediateLoc, 1, locSummary, cnode);
294     } else {
295       endCombNodeSet.add(LocationInference.BOTTOMHNODE);
296       // System.out.println("---endCombNodeSet is zero");
297       // System.out.println("---endNodeSetFromSimpleGraph=" + endNodeSetFromSimpleGraph);
298       // System.out.println("---incoming=" + simpleGraph.getIncomingNodeSet(cnode));
299       recurDFS(desc, lattice, combinationNodeInSCGraph, endCombNodeSet, visited,
300           mapIntermediateLoc, 1, locSummary, cnode);
301
302     }
303
304   }
305
306   private Set<HNode> removeTransitivelyReachToNode(Descriptor desc, HNode startNode,
307       Set<HNode> endNodeSet) {
308
309     // if an end node is not directly connected to the start node in the SC graph
310     // replace it with a directly connected one which transitively reaches to it.
311
312     HierarchyGraph scGraph = infer.getSkeletonCombinationHierarchyGraph(desc);
313
314     Set<HNode> newEndNodeSet = new HashSet<HNode>();
315     for (Iterator iterator = endNodeSet.iterator(); iterator.hasNext();) {
316       HNode endNode = (HNode) iterator.next();
317       if (scGraph.isDirectlyConnectedTo(startNode, endNode)) {
318         newEndNodeSet.add(endNode);
319       } else {
320         HNode newEndNode =
321             getDirectlyReachableNodeFromStartNodeReachToEndNode(scGraph, startNode, endNode);
322         // System.out.println("#### old END NODE=" + endNode + " --->" + newEndNode);
323         newEndNodeSet.add(newEndNode);
324       }
325     }
326
327     // System.out.println("removeTransitivelyReachToNode=" + endNodeSet + "  newSet=" +
328     // newEndNodeSet);
329
330     return newEndNodeSet;
331
332   }
333
334   private HNode getDirectlyReachableSCNodeFromEndNode(HierarchyGraph scGraph, HNode startNode,
335       Set<HNode> endNodeSet) {
336
337     System.out.println("getDirectlyReachableSCNodeFromEndNode start=" + startNode + " endNodeSet="
338         + endNodeSet);
339     Set<HNode> newStartNodeSet = new HashSet<HNode>();
340
341     for (Iterator iterator = endNodeSet.iterator(); iterator.hasNext();) {
342       HNode endNode = (HNode) iterator.next();
343       Set<HNode> connectedToEndNodeSet = scGraph.getIncomingNodeSet(endNode);
344
345       for (Iterator iterator2 = connectedToEndNodeSet.iterator(); iterator2.hasNext();) {
346         HNode curNode = (HNode) iterator2.next();
347         if (recurConnectedFromStartNode(scGraph, startNode, curNode, new HashSet<HNode>())) {
348           newStartNodeSet.add(curNode);
349         }
350       }
351     }
352
353     System.out.println("newStartNodeSet=" + newStartNodeSet);
354
355     if (newStartNodeSet.size() == 0) {
356       newStartNodeSet.add(startNode);
357     }
358
359     return newStartNodeSet.iterator().next();
360   }
361
362   private boolean recurConnectedFromStartNode(HierarchyGraph scGraph, HNode startNode,
363       HNode curNode, Set<HNode> visited) {
364     // return true if curNode is transitively connected from the startNode
365
366     boolean isConnected = false;
367     Set<HNode> inNodeSet = scGraph.getIncomingNodeSet(curNode);
368     for (Iterator iterator = inNodeSet.iterator(); iterator.hasNext();) {
369       HNode in = (HNode) iterator.next();
370       if (in.equals(startNode)) {
371         return true;
372       } else {
373         visited.add(in);
374         isConnected |= recurConnectedFromStartNode(scGraph, startNode, in, visited);
375       }
376     }
377
378     return isConnected;
379   }
380
381   private HNode getDirectlyReachableNodeFromStartNodeReachToEndNode(HierarchyGraph scGraph,
382       HNode startNode, HNode endNode) {
383     System.out.println("getDirectlyReachableNodeFromStartNodeReachToEndNode start=" + startNode
384         + " end=" + endNode);
385     Set<HNode> connected = new HashSet<HNode>();
386     recurDirectlyReachableNodeFromStartNodeReachToEndNode(scGraph, startNode, endNode, connected);
387     if (connected.size() == 0) {
388       connected.add(endNode);
389     }
390     System.out.println("connected=" + connected);
391
392     return connected.iterator().next();
393   }
394
395   private void recurDirectlyReachableNodeFromStartNodeReachToEndNode(HierarchyGraph scGraph,
396       HNode startNode, HNode curNode, Set<HNode> connected) {
397
398     Set<HNode> inNodeSet = scGraph.getIncomingNodeSet(curNode);
399     for (Iterator iterator = inNodeSet.iterator(); iterator.hasNext();) {
400       HNode inNode = (HNode) iterator.next();
401       if (inNode.equals(startNode)) {
402         connected.add(curNode);
403       } else {
404         recurDirectlyReachableNodeFromStartNodeReachToEndNode(scGraph, startNode, inNode, connected);
405       }
406     }
407
408   }
409
410   private void recurDFSNormalNode(Descriptor desc, SSJavaLattice<String> lattice, HNode startNode,
411       Set<HNode> endNodeSet, Set<HNode> visited, Map<TripleItem, String> mapIntermediateLoc,
412       int idx, LocationSummary locSummary, HNode curNode) {
413
414     TripleItem item = new TripleItem(startNode, endNodeSet, idx);
415     // System.out.println("item=" + item);
416     if (!mapIntermediateLoc.containsKey(item)) {
417       // need to create a new intermediate location in the lattice
418       String newLocName = "ILOC" + (LocationInference.locSeed++);
419       String above;
420       if (idx == 1) {
421         above = startNode.getName();
422       } else {
423         int prevIdx = idx - 1;
424         TripleItem prevItem = new TripleItem(startNode, endNodeSet, prevIdx);
425         above = mapIntermediateLoc.get(prevItem);
426       }
427
428       Set<String> belowSet = new HashSet<String>();
429       for (Iterator iterator = endNodeSet.iterator(); iterator.hasNext();) {
430         HNode endNode = (HNode) iterator.next();
431         String locName;
432         if (locSummary.getMapHNodeNameToLocationName().containsKey(endNode.getName())) {
433           locName = locSummary.getLocationName(endNode.getName());
434         } else {
435           locName = endNode.getName();
436         }
437         belowSet.add(locName);
438       }
439       lattice.insertNewLocationBetween(above, belowSet, newLocName);
440
441       mapIntermediateLoc.put(item, newLocName);
442     }
443
444     String locName = mapIntermediateLoc.get(item);
445
446     HierarchyGraph graph = infer.getSimpleHierarchyGraph(desc);
447
448     Set<Descriptor> descSet = graph.getDescSetOfNode(curNode);
449     for (Iterator iterator = descSet.iterator(); iterator.hasNext();) {
450       Descriptor d = (Descriptor) iterator.next();
451       locSummary.addMapHNodeNameToLocationName(d.getSymbol(), locName);
452     }
453     locSummary.addMapHNodeNameToLocationName(curNode.getName(), locName);
454
455     if (curNode.isSharedNode()) {
456       lattice.addSharedLoc(locName);
457     }
458
459     System.out.println("-TripleItem=" + item);
460     System.out.println("-curNode=" + curNode.getName() + " locName=" + locName);
461
462     Set<HNode> outSet = graph.getOutgoingNodeSet(curNode);
463     for (Iterator iterator2 = outSet.iterator(); iterator2.hasNext();) {
464       HNode outNode = (HNode) iterator2.next();
465       if (!outNode.isSkeleton() && !outNode.isCombinationNode() && !visited.contains(outNode)) {
466         visited.add(outNode);
467         recurDFSNormalNode(desc, lattice, startNode, endNodeSet, visited, mapIntermediateLoc,
468             idx + 1, locSummary, outNode);
469       } else if (!outNode.isSkeleton() && outNode.isCombinationNode() && !visited.contains(outNode)) {
470         expandCombinationNode(desc, lattice, visited, mapIntermediateLoc, locSummary, outNode);
471       }
472     }
473
474   }
475
476   private void recurDFS(Descriptor desc, SSJavaLattice<String> lattice,
477       HNode combinationNodeInSCGraph, Set<HNode> endNodeSet, Set<HNode> visited,
478       Map<TripleItem, String> mapIntermediateLoc, int idx, LocationSummary locSummary, HNode curNode) {
479
480     TripleItem item = new TripleItem(combinationNodeInSCGraph, endNodeSet, idx);
481
482     if (!mapIntermediateLoc.containsKey(item)) {
483       // need to create a new intermediate location in the lattice
484       String above;
485       if (idx == 1) {
486         String newLocName = combinationNodeInSCGraph.getName();
487         mapIntermediateLoc.put(item, newLocName);
488       } else {
489         String newLocName = "ILOC" + (LocationInference.locSeed++);
490         int prevIdx = idx - 1;
491         TripleItem prevItem = new TripleItem(combinationNodeInSCGraph, endNodeSet, prevIdx);
492         above = mapIntermediateLoc.get(prevItem);
493
494         Set<String> belowSet = new HashSet<String>();
495         for (Iterator iterator = endNodeSet.iterator(); iterator.hasNext();) {
496           HNode endNode = (HNode) iterator.next();
497           belowSet.add(endNode.getName());
498         }
499         lattice.insertNewLocationBetween(above, belowSet, newLocName);
500         mapIntermediateLoc.put(item, newLocName);
501
502       }
503
504     }
505
506     HierarchyGraph graph = infer.getSimpleHierarchyGraph(desc);
507     String locName = mapIntermediateLoc.get(item);
508     Set<Descriptor> descSet = graph.getDescSetOfNode(curNode);
509     for (Iterator iterator = descSet.iterator(); iterator.hasNext();) {
510       Descriptor d = (Descriptor) iterator.next();
511       locSummary.addMapHNodeNameToLocationName(d.getSymbol(), locName);
512     }
513     locSummary.addMapHNodeNameToLocationName(curNode.getName(), locName);
514
515     if (curNode.isSharedNode()) {
516       lattice.addSharedLoc(locName);
517     }
518     System.out.println("-TripleItem=" + item);
519     System.out.println("-curNode=" + curNode.getName() + " S=" + curNode.isSharedNode()
520         + " locName=" + locName);
521
522     Set<HNode> outSet = graph.getOutgoingNodeSet(curNode);
523     for (Iterator iterator2 = outSet.iterator(); iterator2.hasNext();) {
524       HNode outNode = (HNode) iterator2.next();
525       // System.out.println("recurDFS outNode=" + outNode);
526       // System.out.println("---cur combinationNodeInSCGraph=" + combinationNodeInSCGraph);
527       // System.out.println("---outNode combinationNodeInSCGraph="
528       // + getCombinationNodeInSCGraph(desc, outNode));
529       if (!outNode.isSkeleton() && !visited.contains(outNode)) {
530         if (outNode.isCombinationNode()) {
531           // check whether the next combination node is different from the current node
532           if (combinationNodeInSCGraph.equals(getCombinationNodeInSCGraph(desc, outNode))) {
533             visited.add(outNode);
534             recurDFS(desc, lattice, combinationNodeInSCGraph, endNodeSet, visited,
535                 mapIntermediateLoc, idx + 1, locSummary, outNode);
536           } else {
537             expandCombinationNode(desc, lattice, visited, mapIntermediateLoc, locSummary, outNode);
538           }
539         }
540
541       }
542     }
543
544   }
545
546   private String generateElementName(BasisSet basisSet, HierarchyGraph inputGraph,
547       Map<Set<Integer>, String> mapF2LocName, Set<Integer> F) {
548
549     if (mapF2LocName.containsKey(F)) {
550       return mapF2LocName.get(F);
551     }
552
553     HNode node = basisSet.getHNode(F);
554     if (node != null) {
555       mapF2LocName.put(F, node.getName());
556       return node.getName();
557     } else {
558       if (inputGraph.BASISTOPELEMENT.equals(F)) {
559         return SSJavaAnalysis.BOTTOM;
560       } else {
561         String str = "LOC" + (LocationInference.locSeed++);
562         mapF2LocName.put(F, str);
563         return str;
564       }
565     }
566   }
567
568   private void resetCount(Map<Set<Integer>, Integer> mapFtoCount, Family family) {
569     for (Iterator<Set<Integer>> iter = family.FIterator(); iter.hasNext();) {
570       Set<Integer> F = iter.next();
571       mapFtoCount.put(F, 0);
572     }
573   }
574
575   private Map<Set<Integer>, Set<Set<Integer>>> coveringGraph(BasisSet basisSet, Family family) {
576
577     Map<Set<Integer>, Integer> mapFtoCount = new HashMap<Set<Integer>, Integer>();
578     Map<Set<Integer>, Set<Set<Integer>>> mapImSucc = new HashMap<Set<Integer>, Set<Set<Integer>>>();
579
580     // initialize COUNT(F) to 0 for all elements of the family
581     resetCount(mapFtoCount, family);
582
583     for (Iterator<Set<Integer>> iter = family.FIterator(); iter.hasNext();) {
584       Set<Integer> F = iter.next();
585       Set<HNode> gammaF = family.getGamma(F);
586
587       Set<HNode> curHNodeSet = basisSet.getHNodeSet();
588       curHNodeSet.removeAll(gammaF);
589       Set<Set<Integer>> Bset = basisSet.getBasisSetByHNodeSet(curHNodeSet);
590
591       for (Iterator iterator = Bset.iterator(); iterator.hasNext();) {
592         Set<Integer> B = (Set<Integer>) iterator.next();
593
594         Set<Integer> Fprime = new HashSet<Integer>();
595         Fprime.addAll(F);
596         Fprime.addAll(B);
597
598         // COUNT(F')++;
599         mapFtoCount.put(Fprime, mapFtoCount.get(Fprime) + 1);
600
601         // if |gamma(F')|==COUNT(F') + |gamma(F)|
602         int numGammaFprime = family.getGamma(Fprime).size();
603         int countFprime = mapFtoCount.get(Fprime);
604         int numGammaF = family.getGamma(F).size();
605         if (numGammaFprime == (countFprime + numGammaF)) {
606           // ImSucc(F)=IMSucc(F) union F'
607           addImSucc(mapImSucc, F, Fprime);
608         }
609
610       }
611       resetCount(mapFtoCount, family);
612     }
613
614     // System.out.println("mapImSucc=" + mapImSucc);
615
616     return mapImSucc;
617   }
618
619   private Set<Set<Integer>> getImSucc(Map<Set<Integer>, Set<Set<Integer>>> mapImSucc, Set<Integer> F) {
620     if (!mapImSucc.containsKey(F)) {
621       mapImSucc.put(F, new HashSet<Set<Integer>>());
622     }
623     return mapImSucc.get(F);
624   }
625
626   private void addImSucc(Map<Set<Integer>, Set<Set<Integer>>> mapImSucc, Set<Integer> F,
627       Set<Integer> Fprime) {
628
629     if (!mapImSucc.containsKey(F)) {
630       mapImSucc.put(F, new HashSet<Set<Integer>>());
631     }
632
633     mapImSucc.get(F).add(Fprime);
634
635   }
636
637   private Family generateFamily(BasisSet basisSet) {
638
639     Family family = new Family();
640
641     for (Iterator<Set<Integer>> iterator = basisSet.basisIterator(); iterator.hasNext();) {
642       Set<Integer> B = iterator.next();
643
644       Set<Pair<Set<Integer>, Set<HNode>>> tobeadded = new HashSet<Pair<Set<Integer>, Set<HNode>>>();
645
646       for (Iterator<Set<Integer>> iterator2 = family.FIterator(); iterator2.hasNext();) {
647         Set<Integer> F = iterator2.next();
648
649         Set<Integer> Fprime = new HashSet<Integer>();
650         Fprime.addAll(F);
651         Fprime.addAll(B);
652
653         Set<HNode> gammaFPrimeSet = new HashSet<HNode>();
654         gammaFPrimeSet.addAll(family.getGamma(F));
655         gammaFPrimeSet.add(basisSet.getHNode(B));
656
657         if (!family.containsF(Fprime)) {
658           Pair<Set<Integer>, Set<HNode>> pair =
659               new Pair<Set<Integer>, Set<HNode>>(Fprime, gammaFPrimeSet);
660           tobeadded.add(pair);
661         } else {
662           family.updateGammaF(Fprime, gammaFPrimeSet);
663         }
664       }
665
666       for (Iterator<Pair<Set<Integer>, Set<HNode>>> iterator2 = tobeadded.iterator(); iterator2
667           .hasNext();) {
668         Pair<Set<Integer>, Set<HNode>> pair = iterator2.next();
669         family.addFElement(pair.getFirst());
670         family.updateGammaF(pair.getFirst(), pair.getSecond());
671       }
672
673     }
674     return family;
675   }
676
677   private void debug_print(HierarchyGraph inputGraph) {
678     System.out.println("\nBuild Lattice:" + inputGraph.getName());
679     // System.out.println("Node2Index:\n" + inputGraph.getMapHNodeToUniqueIndex());
680     // System.out.println("Node2Basis:\n" + inputGraph.getMapHNodeToBasis());
681   }
682
683 }
684
685 class Identifier {
686   public HNode node;
687   public int idx;
688
689   public Identifier(HNode n, int i) {
690     node = n;
691     idx = i;
692   }
693
694   public int hashCode() {
695     return node.hashCode() + idx;
696   }
697
698   public boolean equals(Object obj) {
699
700     if (obj instanceof Identifier) {
701       Identifier in = (Identifier) obj;
702       if (node.equals(in.node) && idx == in.idx) {
703         return true;
704       }
705     }
706
707     return false;
708   }
709
710 }
711
712 class TripleItem {
713   public HNode higherNode;
714   public Set<HNode> lowerNodeSet;
715   public int idx;
716
717   public TripleItem(HNode h, Set<HNode> l, int i) {
718     higherNode = h;
719     lowerNodeSet = l;
720     idx = i;
721   }
722
723   public int hashCode() {
724
725     int h = 0;
726     if (higherNode != null) {
727       h = higherNode.hashCode();
728     }
729
730     return h + lowerNodeSet.hashCode() + idx;
731   }
732
733   public boolean equals(Object obj) {
734
735     if (obj instanceof TripleItem) {
736       TripleItem in = (TripleItem) obj;
737       if ((higherNode == null || (higherNode != null && higherNode.equals(in.higherNode)))
738           && lowerNodeSet.equals(in.lowerNodeSet) && idx == in.idx) {
739         return true;
740       }
741     }
742
743     return false;
744   }
745
746   public String toString() {
747     return higherNode + "-" + idx + "->" + lowerNodeSet;
748   }
749 }