changes: now Inference engine works fine with the EyeTracking benchmark.

[IRC.git] / Robust / src / Analysis / SSJava / FlowGraph.java

diff --git a/Robust/src/Analysis/SSJava/FlowGraph.java b/Robust/src/Analysis/SSJava/FlowGraph.java
index b9a39f7..4054f70 100644 (file)
--- a/Robust/src/Analysis/SSJava/FlowGraph.java
+++ b/Robust/src/Analysis/SSJava/FlowGraph.java
@@ -318,7 +318,7 @@ public class FlowGraph {
     addOutEdge(fromNode, edge);
     addInEdge(toNode, edge);
 
     addOutEdge(fromNode, edge);
     addInEdge(toNode, edge);
 

-    System.out.println("add a new edge=" + edge);
+    // System.out.println("add a new edge=" + edge);

   }
 
   private void addInEdge(FlowNode toNode, FlowEdge edge) {
   }
 
   private void addInEdge(FlowNode toNode, FlowEdge edge) {


@@ -859,4 +859,20 @@ public class FlowGraph {
     bw.write("}\n");
   }
 
     bw.write("}\n");
   }
 

+  public void removeEdge(NTuple<Descriptor> from, NTuple<Descriptor> to) {
+
+    Set<FlowEdge> toberemoved = new HashSet<FlowEdge>();
+    Set<FlowEdge> edgeSet = getOutEdgeSet(getFlowNode(from));
+
+    for (Iterator iterator = edgeSet.iterator(); iterator.hasNext();) {
+      FlowEdge flowEdge = (FlowEdge) iterator.next();
+      if (flowEdge.getInitTuple().equals(from) && flowEdge.getEndTuple().equals(to)) {
+        toberemoved.add(flowEdge);
+      }
+    }
+
+    edgeSet.removeAll(toberemoved);
+
+  }
+

 }
\ No newline at end of file
 }
\ No newline at end of file