import IR.Descriptor;
import IR.FieldDescriptor;
import IR.MethodDescriptor;
-import IR.NameDescriptor;
import IR.VarDescriptor;
public class FlowGraph {
MethodDescriptor md;
- Set<FlowNode> nodeSet;
-
Set<FlowNode> returnNodeSet;
FlowNode thisVarNode;
this.md = md;
this.mapFlowNodeToLocTuple = new HashMap<FlowNode, NTuple<Location>>();
this.mapLocTupleToFlowNode = new HashMap<NTuple<Location>, FlowNode>();
- this.nodeSet = new HashSet<FlowNode>();
this.mapDescTupleToInferNode = new HashMap<NTuple<Descriptor>, FlowNode>();
this.mapParamDescToIdx = new HashMap<Descriptor, Integer>();
this.mapParamDescToIdx.putAll(mapParamDescToIdx);
this.mapLocTupleToFlowNode.putAll(in);
}
- public void setNodeSet(Set<FlowNode> in) {
- this.nodeSet.addAll(in);
- }
-
public void setReturnNodeSet(Set<FlowNode> in) {
this.returnNodeSet.addAll(in);
}
newNode.setIntermediate(true);
mapDescTupleToInferNode.put(tuple, newNode);
- nodeSet.add(newNode);
+ // nodeSet.add(newNode);
System.out.println("create new intermediate node= " + newNode);
return mapIdxToFlowNode.get(idx);
}
+ public Set<FlowEdge> getEdgeSet() {
+ Set<FlowEdge> edgeSet = new HashSet<FlowEdge>();
+
+ Set<FlowNode> nodeSet = getNodeSet();
+ for (Iterator iterator = nodeSet.iterator(); iterator.hasNext();) {
+ FlowNode flowNode = (FlowNode) iterator.next();
+ edgeSet.addAll(getOutEdgeSet(flowNode));
+ }
+
+ return edgeSet;
+ }
+
public Set<FlowNode> getNodeSet() {
- return nodeSet;
+ Set<FlowNode> set = new HashSet<FlowNode>();
+ set.addAll(mapDescTupleToInferNode.values());
+ return set;
}
public MethodDescriptor getMethodDescriptor() {
return false;
}
+ public Set<FlowEdge> getOutEdgeSetStartingFrom(FlowNode startNode) {
+
+ Descriptor prefixDesc = startNode.getCurrentDescTuple().get(0);
+
+ // returns the set of edges that share the same prefix of startNode
+ Set<FlowEdge> edgeSet = new HashSet<FlowEdge>();
+
+ for (Iterator<Set<FlowEdge>> iter = mapFlowNodeToOutEdgeSet.values().iterator(); iter.hasNext();) {
+ Set<FlowEdge> nodeEdgeSet = iter.next();
+ for (Iterator<FlowEdge> iter2 = nodeEdgeSet.iterator(); iter2.hasNext();) {
+ FlowEdge edge = iter2.next();
+ if (edge.getInitTuple().get(0).equals(prefixDesc)) {
+ edgeSet.add(edge);
+ }
+ }
+ }
+
+ return edgeSet;
+ }
+
public Set<FlowEdge> getOutEdgeSet(FlowNode node) {
if (!mapFlowNodeToOutEdgeSet.containsKey(node)) {
mapFlowNodeToOutEdgeSet.put(node, new HashSet<FlowEdge>());
FlowNode fromNode = getFlowNode(fromDescTuple);
FlowNode toNode = getFlowNode(toDescTuple);
- System.out.println("create an edge from " + fromNode + " to " + toNode);
+ // System.out.println("create an edge from " + fromNode + " to " + toNode);
int fromTupleSize = fromDescTuple.size();
NTuple<Descriptor> curFromTuple = new NTuple<Descriptor>();
if (!mapDescTupleToInferNode.containsKey(tuple)) {
FlowNode node = new FlowNode(tuple);
mapDescTupleToInferNode.put(tuple, node);
- nodeSet.add(node);
+ // nodeSet.add(node);
mapLocTupleToFlowNode.put(getLocationTuple(node), node);
return set;
}
+ public Set<FlowNode> getReachableSetFrom(NTuple<Descriptor> prefix) {
+ Set<FlowNode> reachableSet = new HashSet<FlowNode>();
+
+ Set<FlowNode> nodeSet = getNodeSet();
+ for (Iterator iterator = nodeSet.iterator(); iterator.hasNext();) {
+ FlowNode flowNode = (FlowNode) iterator.next();
+ if (flowNode.getCurrentDescTuple().startsWith(prefix)) {
+ recurReachableSetFrom(flowNode, reachableSet);
+ }
+ }
+
+ return reachableSet;
+ }
+
// private void getReachFlowNodeSetFrom(FlowNode fn, Set<FlowNode> visited) {
//
// for (Iterator iterator = fn.getOutEdgeSet().iterator();
//
// }
+ private void recurReachableSetFrom(FlowNode curNode, Set<FlowNode> reachableSet) {
+
+ Set<FlowEdge> edgeSet = getOutEdgeSet(curNode);
+ for (Iterator iterator = edgeSet.iterator(); iterator.hasNext();) {
+ FlowEdge edge = (FlowEdge) iterator.next();
+ FlowNode dstNode = getFlowNode(edge.getEndTuple());
+ if (!reachableSet.contains(dstNode)) {
+ reachableSet.add(dstNode);
+ recurReachableSetFrom(dstNode, reachableSet);
+ }
+ }
+
+ }
+
public Set<NTuple<Location>> getReachableFlowTupleSet(Set<NTuple<Location>> visited, FlowNode fn) {
Set<FlowEdge> outEdgeSet = getOutEdgeSet(fn);
public void getIncomingFlowNodeSet(FlowNode node, Set<FlowNode> visited) {
- for (Iterator iterator = nodeSet.iterator(); iterator.hasNext();) {
+ for (Iterator iterator = getNodeSet().iterator(); iterator.hasNext();) {
FlowNode curNode = (FlowNode) iterator.next();
Set<FlowEdge> edgeSet = getOutEdgeSet(curNode);
ClassDescriptor cd = null;
- for (Iterator iterator = nodeSet.iterator(); iterator.hasNext();) {
+ for (Iterator iterator = getNodeSet().iterator(); iterator.hasNext();) {
FlowNode node = (FlowNode) iterator.next();
Set<FlowEdge> edgeSet = getOutEdgeSet(node);
return mapParamDescToIdx.containsKey(firstIdxDesc);
}
+ public int getParamIdx(NTuple<Descriptor> tuple) {
+ Descriptor firstIdxDesc = tuple.get(0);
+ return mapParamDescToIdx.get(firstIdxDesc);
+ }
+
public FlowGraph clone() {
FlowGraph clone = new FlowGraph(md, mapParamDescToIdx);
- clone.setNodeSet(getNodeSet());
+ // clone.setNodeSet(getNodeSet());
clone.setMapLocTupleToFlowNode(getMapLocTupleToFlowNode());
clone.setMapFlowNodeToLocTuple(getMapFlowNodeToLocTuple());
clone.setMapDescTupleToInferNode(getMapDescTupleToInferNode());
// then visit every flow node
- Iterator<FlowNode> iter = nodeSet.iterator();
+ // Iterator<FlowNode> iter = nodeSet.iterator();
+ Iterator<FlowNode> iter = getNodeSet().iterator();
Set<FlowEdge> addedEdgeSet = new HashSet<FlowEdge>();
Set<FlowNode> addedNodeSet = new HashSet<FlowNode>();
}
public boolean constainsNode(FlowNode node) {
- return nodeSet.contains(node);
+ return getNodeSet().contains(node);
}
private void drawSubgraph(FlowNode node, BufferedWriter bw, Set<FlowEdge> addedSet)
bw.write("}\n");
}
+
+
}
\ No newline at end of file