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