import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
+import java.util.LinkedList;
import java.util.Map;
import java.util.Set;
+import java.util.Stack;
import IR.ClassDescriptor;
import IR.Descriptor;
private Map<Descriptor, Map<TripleItem, String>> mapDescToIntermediateLocMap;
+ private Map<Descriptor, Map<InterLocItem, String>> mapDescToInterLocMap;
+
private Map<Pair<HNode, HNode>, Integer> mapItemToHighestIndex;
+ private Map<SSJavaLattice<String>, Set<String>> mapLatticeToLocalLocSet;
+
+ private Map<Descriptor, Map<LineIdentifier, LinkedList<LocPair>>> mapDescToLineListMap;
+
public BuildLattice(LocationInference infer) {
this.infer = infer;
this.mapSharedNodeToTripleItem = new HashMap<HNode, TripleItem>();
this.mapHNodeToHighestIndex = new HashMap<HNode, Integer>();
this.mapItemToHighestIndex = new HashMap<Pair<HNode, HNode>, Integer>();
this.mapDescToIntermediateLocMap = new HashMap<Descriptor, Map<TripleItem, String>>();
+ this.mapLatticeToLocalLocSet = new HashMap<SSJavaLattice<String>, Set<String>>();
+ this.mapDescToInterLocMap = new HashMap<Descriptor, Map<InterLocItem, String>>();
+ this.mapDescToLineListMap = new HashMap<Descriptor, Map<LineIdentifier, LinkedList<LocPair>>>();
}
public SSJavaLattice<String> buildLattice(Descriptor desc) {
HierarchyGraph inputGraph = infer.getSkeletonCombinationHierarchyGraph(desc);
LocationSummary locSummary = infer.getLocationSummary(desc);
- HierarchyGraph naiveGraph = infer.getSimpleHierarchyGraph(desc);
+ HierarchyGraph simpleGraph = infer.getSimpleHierarchyGraph(desc);
+ HierarchyGraph naiveGraph = simpleGraph.clone();
// I don't think we need to keep the below if statement anymore
// because hierarchy graph does not have any composite location
SSJavaLattice<String> naive_lattice =
buildLattice(desc, naiveBasisSet, naiveGraph, null, naive_mapImSucc);
- LocationInference.numLocationsNaive += naive_lattice.getKeySet().size();
+ int numLocs = naive_lattice.getKeySet().size();
+ LocationInference.numLocationsNaive += numLocs;
+ infer.mapNumLocsMapNaive.put(desc, new Integer(numLocs));
+
+ int numPaths = naive_lattice.countPaths();
+ infer.mapNumPathsMapNaive.put(desc, new Integer(numPaths));
+
+ System.out.println(desc + " numPaths=" + numPaths + " numLocs="
+ + naive_lattice.getKeySet().size());
+
infer.addNaiveLattice(desc, naive_lattice);
}
public SSJavaLattice<String> insertIntermediateNodesToStraightLine(Descriptor desc,
SSJavaLattice<String> skeletonLattice) {
- // copy nodes/edges from the parent method/class if possible
+
SSJavaLattice<String> lattice = skeletonLattice.clone();
+ LocationSummary locSummary = infer.getLocationSummary(desc);
Descriptor parentDesc = getParent(desc);
if (parentDesc != null) {
}
}
- // ////
+ HierarchyGraph hierarchyGraph = infer.getSimpleHierarchyGraph(desc);
+ HierarchyGraph scGraph = infer.getSkeletonCombinationHierarchyGraph(desc);
- // perform DFS that starts from each skeleton/combination node and ends by another
- // skeleton/combination node
+ Set<HNode> hierarchyGraphNodeSet = hierarchyGraph.getNodeSet();
+ for (Iterator iterator = hierarchyGraphNodeSet.iterator(); iterator.hasNext();) {
+ HNode hNode = (HNode) iterator.next();
+ if (!hNode.isSkeleton()) {
+ // here we need to insert an intermediate node for the hNode
+ System.out.println("\n#local node=" + hNode);
+
+ // 1) find the lowest node m in the lattice that is above hnode in the lattice
+ // 2) count the number of non-shared nodes d between the hnode and the node m
+ // int numNonSharedNodes;
+ int dist = 0;
+
+ HNode SCNode;
+ Set<HNode> combineSkeletonNodeSet = null;
+ Stack<String> trace = new Stack<String>();
+ if (hNode.isDirectCombinationNode()) {
+ // this node itself is the lowest node m. it is the first node of the chain
+ Set<HNode> combineSet = hierarchyGraph.getCombineSetByCombinationNode(hNode);
+
+ System.out.println(" # direct combine node::combineSkeletonNodeSet=" + combineSet);
+
+ SCNode = scGraph.getCombinationNode(combineSet);
+ // numNonSharedNodes = -1;
+ dist = 0;
+ if (hNode.isSharedNode()) {
+ trace.add("S");
+ } else {
+ trace.add("N");
+ }
+ } else {
- mapSharedNodeToTripleItem.clear();
+ Set<HNode> aboveSet = new HashSet<HNode>();
+ if (hNode.isCombinationNode()) {
+ // the current node is a combination node
+ combineSkeletonNodeSet = hierarchyGraph.getCombineSetByCombinationNode(hNode);
+ System.out.println(" combineSkeletonNodeSet=" + combineSkeletonNodeSet
+ + " combinationNode=" + scGraph.getCombinationNode(combineSkeletonNodeSet));
- HierarchyGraph simpleGraph = infer.getSimpleHierarchyGraph(desc);
- HierarchyGraph scGraph = infer.getSkeletonCombinationHierarchyGraph(desc);
- LocationSummary locSummary = infer.getLocationSummary(desc);
+ scGraph.getCombinationNode(combineSkeletonNodeSet);
- Set<HNode> visited = new HashSet<HNode>();
+ System.out.println(" firstnodeOfSimpleGraph="
+ + hierarchyGraph.getFirstNodeOfCombinationNodeChainSet(combineSkeletonNodeSet));
+ aboveSet.addAll(hierarchyGraph
+ .getFirstNodeOfCombinationNodeChainSet(combineSkeletonNodeSet));
- Set<HNode> nodeSet = simpleGraph.getNodeSet();
+ SCNode = scGraph.getCombinationNode(combineSkeletonNodeSet);
- Map<TripleItem, String> mapIntermediateLoc = getIntermediateLocMap(desc);
- // Map<TripleItem, String> mapIntermediateLoc = new HashMap<TripleItem, String>();
+ } else {
+ // the current node is not a combination node
+ // there is only one parent node which should be skeleton node.
- // System.out.println("*insert=" + desc);
- // System.out.println("***nodeSet=" + nodeSet);
- for (Iterator iterator = nodeSet.iterator(); iterator.hasNext();) {
- HNode node = (HNode) iterator.next();
- System.out.println("node=" + node);
+ // System.out.println(" hierarchyGraph.getSkeleteNodeSetReachTo(" + hNode + ")="
+ // + hierarchyGraph.getSkeleteNodeSetReachTo(hNode));
- if (node.isSkeleton() && (!visited.contains(node))) {
- visited.add(node);
+ Set<HNode> reachToSet = hierarchyGraph.getSkeleteNodeSetReachTo(hNode);
+ for (Iterator iterator2 = reachToSet.iterator(); iterator2.hasNext();) {
+ HNode reachToNode = (HNode) iterator2.next();
+ aboveSet.add(scGraph.getCurrentHNode(reachToNode));
+ }
- Set<HNode> outSet = simpleGraph.getOutgoingNodeSet(node);
- for (Iterator iterator2 = outSet.iterator(); iterator2.hasNext();) {
- HNode outNode = (HNode) iterator2.next();
+ SCNode = aboveSet.iterator().next();
+ }
- if (!outNode.isSkeleton()) {
- if (outNode.isCombinationNode()) {
- if (visited.containsAll(simpleGraph.getIncomingNodeSet(outNode))) {
- // if (needToExpandCombinationNode(desc, outNode)) {
- expandCombinationNode(desc, lattice, visited, mapIntermediateLoc, locSummary,
- outNode);
- // }
- }
- } else {
- // we have a node that is neither combination or skeleton node
- // System.out.println("%%%skeleton node=" + node + " outNode=" + outNode);
- HNode startNode = scGraph.getCurrentHNode(node);
-
- // if (node.getDescriptor() != null) {
- // // node is a skeleton node and it might be merged into another node in the SC
- // graph.
- // startNode = scGraph.getHNode(node.getDescriptor());
- // } else {
- // // this node has already been merged before the SC graph.
- // startNode = node;
- // }
-
- // TODO
- // Set<HNode> endNodeSetFromSimpleGraph =
- // simpleGraph.getDirectlyReachableSkeletonCombinationNodeFrom(outNode, null);
- // Set<HNode> endCombNodeSet = new HashSet<HNode>();
- // for (Iterator iterator3 = endNodeSetFromSimpleGraph.iterator();
- // iterator3.hasNext();) {
- // HNode endNode = (HNode) iterator3.next();
- // endCombNodeSet.add(getCombinationNodeInSCGraph(desc, endNode));
- // }
-
- Set<HNode> endCombNodeSet = scGraph.getOutgoingNodeSet(startNode);
-
- // System.out.println("endCombNodeSet=" + endCombNodeSet);
- visited.add(outNode);
- if (endCombNodeSet.size() > 0) {
- // follows the straight line up to another skeleton/combination node
- endCombNodeSet = removeTransitivelyReachToNode(desc, startNode, endCombNodeSet);
- } else if (endCombNodeSet.size() == 0) {
- // the outNode is (directly/transitively) connected to the bottom node
- // therefore, we just add a dummy bottom HNode to the endCombNodeSet.
- endCombNodeSet.add(LocationInference.BOTTOMHNODE);
- }
-
- recurDFSNormalNode(desc, lattice, startNode, endCombNodeSet, visited,
- mapIntermediateLoc, 1, locSummary, outNode);
- }
+ // update above set w.r.t the hierarchy graph with SC nodes
+ // because the skeleton nodes in the original hierarchy graph may be merged to a new node
+ Set<HNode> endSet = new HashSet<HNode>();
+ for (Iterator iterator2 = aboveSet.iterator(); iterator2.hasNext();) {
+ HNode aboveNode = (HNode) iterator2.next();
+ endSet.add(hierarchyGraph.getCurrentHNode(aboveNode));
+ }
+
+ trace = hierarchyGraph.computeDistance(hNode, endSet, combineSkeletonNodeSet);
+
+ System.out.println(" COUNT-RESULT::start=" + hNode + " end=" + endSet + " trace="
+ + trace);
+ }
+ // 3) convert the node m into a chain of nodes with the last node in the chain having m's
+ // outgoing edges.
+ Set<HNode> outgoingSCNodeSet = scGraph.getOutgoingNodeSet(SCNode);
+ System.out.println(" outgoing scnode set from " + SCNode + "=" + outgoingSCNodeSet);
+
+ // convert hnodes to location names
+ String startLocName = locSummary.getLocationName(SCNode.getName());
+ Set<String> outgoingLocNameSet = new HashSet<String>();
+ for (Iterator iterator2 = outgoingSCNodeSet.iterator(); iterator2.hasNext();) {
+ HNode outSCNode = (HNode) iterator2.next();
+ String locName = locSummary.getLocationName(outSCNode.getName());
+ if (!locName.equals(outSCNode.getName())) {
+ System.out.println(" outSCNode=" + outSCNode + " -> locName="
+ + locName);
}
+ outgoingLocNameSet.add(locName);
+ }
+ if (outgoingLocNameSet.isEmpty()) {
+ outgoingLocNameSet.add(lattice.getBottomItem());
}
- } else if (!node.isSkeleton() && !node.isCombinationNode() && !node.isMergeNode()
- && !visited.contains(node)) {
- System.out.println("n=" + node);
+ if (hNode.isCombinationNode()) {
+ System.out.println("make sure that the first node corresponds to the COMB node="
+ + startLocName);
+ LineIdentifier lineId = new LineIdentifier(startLocName, outgoingLocNameSet);
+ LinkedList<LocPair> list = getLineList(desc, lineId);
+ // make sure that the first node corresponds to the COMB node
+ if (list.size() <= 0) {
+ list.add(new LocPair());
+ }
+ LocPair firstPair = list.get(0);
+ firstPair.nonShared = startLocName;
+ }
- // an intermediate node 'node' may be located between "TOP" location and a skeleton node
- if (simpleGraph.getIncomingNodeSet(node).size() == 0) {
+ // 4) If hnode is not a shared location, check if there already exists a local variable
+ // node that has distance d below m along this chain. If such a node
+ // does not exist, insert it.
+ String locName =
+ getNewLocation(desc, startLocName, outgoingLocNameSet, trace, hNode.isSharedNode());
- // this node will be directly connected to the TOP location
- // start adding the following nodes from this node
+ System.out.println(" ###hNode=" + hNode + "---->locName=" + locName);
+ locSummary.addMapHNodeNameToLocationName(hNode.getName(), locName);
- Set<HNode> endNodeSetFromSimpleGraph =
- simpleGraph.getDirectlyReachableSkeletonCombinationNodeFrom(node, null);
+ }
+ }
- Set<HNode> endCombNodeSet = new HashSet<HNode>();
- for (Iterator iterator3 = endNodeSetFromSimpleGraph.iterator(); iterator3.hasNext();) {
- HNode endNode = (HNode) iterator3.next();
- endCombNodeSet.add(getCombinationNodeInSCGraph(desc, endNode));
- }
+ insertLocalLocations(desc, lattice);
+
+ return lattice;
+ }
+
+ private void insertLocalLocations(Descriptor desc, SSJavaLattice<String> lattice) {
+
+ Map<LineIdentifier, LinkedList<LocPair>> map = getLineListMap(desc);
+ System.out.println("####MAP=" + map);
+
+ Set<LineIdentifier> lineIdSet = map.keySet();
+ for (Iterator iterator = lineIdSet.iterator(); iterator.hasNext();) {
+ LineIdentifier lineId = (LineIdentifier) iterator.next();
+
+ LinkedList<LocPair> list = map.get(lineId);
- System.out.println("endCombNodeSet=" + endCombNodeSet);
- HNode startNode = LocationInference.TOPHNODE;
- visited.add(startNode);
- if (endCombNodeSet.size() > 0) {
- // follows the straight line up to another skeleton/combination node
- // endCombNodeSet = removeTransitivelyReachToNode(desc, node, endCombNodeSet);
- recurDFSNormalNode(desc, lattice, startNode, endCombNodeSet, visited,
- mapIntermediateLoc, 1, locSummary, node);
+ String higherLocName = lineId.startLoc;
+ Set<String> endLocNameSet = lineId.lowerLocSet;
+
+ for (int i = 0; i < list.size(); i++) {
+ LocPair pair = list.get(i);
+
+ if (pair.nonShared != null) {
+
+ if (!lattice.getKeySet().contains(pair.nonShared)) {
+ lattice.insertNewLocationBetween(higherLocName, endLocNameSet, pair.nonShared);
}
+ higherLocName = pair.nonShared;
+ }
+ if (pair.shared != null) {
+ if (!lattice.getKeySet().contains(pair.shared)) {
+ lattice.insertNewLocationBetween(higherLocName, endLocNameSet, pair.shared);
+ lattice.addSharedLoc(pair.shared);
+ }
+ higherLocName = pair.shared;
}
}
+
}
- // add shared locations
- Set<HNode> sharedNodeSet = mapSharedNodeToTripleItem.keySet();
- for (Iterator iterator = sharedNodeSet.iterator(); iterator.hasNext();) {
- HNode sharedNode = (HNode) iterator.next();
- TripleItem item = mapSharedNodeToTripleItem.get(sharedNode);
- String nonSharedLocName = mapIntermediateLoc.get(item);
+ }
+
+ private void addLocalLocation(SSJavaLattice<String> lattice, String localLoc) {
+ if (!mapLatticeToLocalLocSet.containsKey(lattice)) {
+ mapLatticeToLocalLocSet.put(lattice, new HashSet<String>());
+ }
+ mapLatticeToLocalLocSet.get(lattice).add(localLoc);
+ }
- System.out.println("sharedNode=" + sharedNode + " locName=" + nonSharedLocName);
+ private boolean isLocalLocation(SSJavaLattice<String> lattice, String localLoc) {
+ if (mapLatticeToLocalLocSet.containsKey(lattice)) {
+ return mapLatticeToLocalLocSet.get(lattice).contains(localLoc);
+ }
+ return false;
+ }
- String newLocName;
- if (locSummary.getHNodeNameSetByLatticeLoationName(nonSharedLocName) != null
- && !lattice.isSharedLoc(nonSharedLocName)) {
- // need to generate a new shared location in the lattice, which is one level lower than the
- // 'locName' location
- newLocName = "ILOC" + (LocationInference.locSeed++);
+ public Map<InterLocItem, String> getInterLocMap(Descriptor desc) {
+ if (!mapDescToInterLocMap.containsKey(desc)) {
+ mapDescToInterLocMap.put(desc, new HashMap<InterLocItem, String>());
+ }
+ return mapDescToInterLocMap.get(desc);
+ }
- // Set<String> aboveElementSet = getAboveElementSet(lattice, locName);
- Set<String> belowElementSet = new HashSet<String>();
- belowElementSet.addAll(lattice.get(nonSharedLocName));
+ public Map<LineIdentifier, LinkedList<LocPair>> getLineListMap(Descriptor desc) {
+ if (!mapDescToLineListMap.containsKey(desc)) {
+ mapDescToLineListMap.put(desc, new HashMap<LineIdentifier, LinkedList<LocPair>>());
+ }
+ return mapDescToLineListMap.get(desc);
+ }
- System.out.println("nonSharedLocName=" + nonSharedLocName + " belowElementSet="
- + belowElementSet + " newLocName=" + newLocName);
+ public LinkedList<LocPair> getLineList(Descriptor desc, LineIdentifier lineId) {
+ Map<LineIdentifier, LinkedList<LocPair>> map = getLineListMap(desc);
+ if (!map.containsKey(lineId)) {
+ map.put(lineId, new LinkedList<LocPair>());
+ }
+ return map.get(lineId);
+ }
- lattice.insertNewLocationBetween(nonSharedLocName, belowElementSet, newLocName);
+ private String generateNewLocName() {
+ String newLocName = "ILOC" + (LocationInference.locSeed++);
+ System.out.println("newLocName=" + newLocName);
+ return newLocName;
+ }
+
+ public String getNewLocation(Descriptor desc, String start, Set<String> endSet,
+ Stack<String> trace, boolean isShared) {
+
+ System.out.println(" #GetNewLocation start=" + start + " endSet=" + endSet + " trace="
+ + trace);
+
+ LineIdentifier lineId = new LineIdentifier(start, endSet);
+ LinkedList<LocPair> list = getLineList(desc, lineId);
+
+ int locPairIdx = trace.size() - 1;
+
+ LocPair pair;
+ if (list.size() > locPairIdx) {
+ // there already exsits a list of nodes up to the current one
+ pair = list.get(locPairIdx);
+ // check if we need to add a new shared or non-shred node to the pair
+ if (isShared) {
+ if (pair.shared == null) {
+ pair.shared = generateNewLocName();
+ }
+ return pair.shared;
} else {
- newLocName = nonSharedLocName;
+ if (pair.nonShared == null) {
+ pair.nonShared = generateNewLocName();
+ }
+ return pair.nonShared;
}
- lattice.addSharedLoc(newLocName);
- HierarchyGraph graph = infer.getSimpleHierarchyGraph(desc);
- Set<Descriptor> descSet = graph.getDescSetOfNode(sharedNode);
- for (Iterator iterator2 = descSet.iterator(); iterator2.hasNext();) {
- Descriptor d = (Descriptor) iterator2.next();
- locSummary.addMapHNodeNameToLocationName(d.getSymbol(), newLocName);
- }
- locSummary.addMapHNodeNameToLocationName(sharedNode.getName(), newLocName);
+ } else {
+ // we need to set up a chain of intermediate nodes to the current one.
+ return recur_getNewLocation(0, list, trace);
+ }
+ }
+
+ private String recur_getNewLocation(int idx, LinkedList<LocPair> list, Stack<String> trace) {
+
+ String curType = trace.pop();
+ boolean isCurShared = false;
+ if (curType.equals("S")) {
+ isCurShared = true;
}
- return lattice;
+ if (list.size() <= idx) {
+ // need to insert a new pair for the idx
+ list.add(new LocPair());
+ }
+
+ // here, the list already has a pair for the idx
+ LocPair pair = list.get(idx);
+ if (isCurShared && pair.shared == null) {
+ pair.shared = generateNewLocName();
+ } else if (!isCurShared && pair.nonShared == null) {
+ pair.nonShared = generateNewLocName();
+ }
+
+ if (trace.isEmpty()) {
+ if (isCurShared) {
+ return pair.shared;
+ } else {
+ return pair.nonShared;
+ }
+ }
+ idx++;
+ return recur_getNewLocation(idx, list, trace);
}
private Set<String> getAboveElementSet(SSJavaLattice<String> lattice, String loc) {
}
+class LocPair {
+ public String nonShared;
+ public String shared;
+
+ public int hashCode() {
+ int h = 0;
+ if (nonShared != null) {
+ h = nonShared.hashCode();
+ }
+ if (shared != null) {
+ h = shared.hashCode();
+ }
+ return h;
+ }
+
+ public boolean equals(Object obj) {
+
+ if (obj instanceof LocPair) {
+ LocPair in = (LocPair) obj;
+
+ if ((nonShared == null && in.nonShared == null)
+ || (nonShared != null && nonShared.equals(in.nonShared))) {
+
+ if ((shared == null && in.shared == null) || (shared != null && shared.equals(in.shared))) {
+ return true;
+ }
+
+ }
+
+ }
+
+ return false;
+ }
+
+ public String toString() {
+ String rtr = "<" + nonShared + "," + shared + ">";
+ return rtr;
+ }
+}
+
+class LineIdentifier {
+ public String startLoc;
+ public Set<String> lowerLocSet;
+
+ public LineIdentifier(String s, Set<String> lSet) {
+ startLoc = s;
+ lowerLocSet = lSet;
+ }
+
+ public int hashCode() {
+ int h = 0;
+ h = startLoc.hashCode();
+ return h + lowerLocSet.hashCode();
+ }
+
+ public boolean equals(Object obj) {
+
+ if (obj instanceof LineIdentifier) {
+ LineIdentifier in = (LineIdentifier) obj;
+ if (startLoc.equals(in.startLoc) && lowerLocSet.equals(in.lowerLocSet)) {
+ return true;
+ }
+ }
+
+ return false;
+ }
+
+ public String toString() {
+ String rtr = startLoc + "->" + lowerLocSet;
+ return rtr;
+ }
+
+}
+
+class InterLocItem {
+ public String startLoc;
+ public Set<String> lowerLocSet;
+ public int idx;
+
+ public InterLocItem(String h, Set<String> l, int i) {
+ startLoc = h;
+ lowerLocSet = l;
+ idx = i;
+ }
+
+ public int hashCode() {
+
+ int h = 0;
+ if (startLoc != null) {
+ h = startLoc.hashCode();
+ }
+
+ return h + lowerLocSet.hashCode() + idx;
+ }
+
+ public boolean equals(Object obj) {
+
+ if (obj instanceof InterLocItem) {
+ InterLocItem in = (InterLocItem) obj;
+ if ((startLoc == null || (startLoc != null && startLoc.equals(in.startLoc)))
+ && lowerLocSet.equals(in.lowerLocSet) && idx == in.idx) {
+ return true;
+ }
+ }
+
+ return false;
+ }
+
+ public String toString() {
+ String rtr = startLoc + "-" + idx + "->" + lowerLocSet;
+ if (idx % 2 != 0) {
+ rtr += " S";
+ }
+ return rtr;
+ }
+}
+
class TripleItem {
public HNode higherNode;
public Set<HNode> lowerNodeSet;
}
return rtr;
}
+
}
projects / IRC.git / blobdiff