From 5c18cc9c389a1ce1fd5304d55224dbc19d57a78a Mon Sep 17 00:00:00 2001
From: bdemsky <bdemsky>
Date: Fri, 3 Apr 2009 09:15:15 +0000
Subject: [PATCH] changes

---
 Robust/src/IR/Flat/BuildCode.java           |  4 ++
 Robust/src/IR/Flat/BuildFlat.java           | 24 ++++++---
 Robust/src/IR/Flat/FKind.java               |  1 +
 Robust/src/IR/Flat/FlatAtomicEnterNode.java |  5 ++
 Robust/src/IR/Flat/FlatAtomicExitNode.java  |  5 +-
 Robust/src/IR/Flat/FlatBackEdge.java        |  5 +-
 Robust/src/IR/Flat/FlatCall.java            | 16 ++++++
 Robust/src/IR/Flat/FlatCastNode.java        | 10 ++++
 Robust/src/IR/Flat/FlatCondBranch.java      | 12 +++++
 Robust/src/IR/Flat/FlatElementNode.java     | 11 +++++
 Robust/src/IR/Flat/FlatExit.java            | 22 +++++++++
 Robust/src/IR/Flat/FlatFieldNode.java       | 10 ++++
 Robust/src/IR/Flat/FlatInstanceOfNode.java  |  9 ++++
 Robust/src/IR/Flat/FlatLiteralNode.java     |  8 +++
 Robust/src/IR/Flat/FlatMethod.java          | 10 +++-
 Robust/src/IR/Flat/FlatNew.java             | 11 +++++
 Robust/src/IR/Flat/FlatNode.java            | 54 ++++++++++++++++++---
 Robust/src/IR/Flat/FlatNop.java             |  5 ++
 Robust/src/IR/Flat/FlatOffsetNode.java      |  8 +++
 Robust/src/IR/Flat/FlatOpNode.java          | 11 +++++
 Robust/src/IR/Flat/FlatReturnNode.java      |  8 +++
 Robust/src/IR/Flat/FlatSESEEnterNode.java   |  5 +-
 Robust/src/IR/Flat/FlatSESEExitNode.java    |  5 +-
 Robust/src/IR/Flat/FlatSetElementNode.java  | 10 ++++
 Robust/src/IR/Flat/FlatSetFieldNode.java    |  9 ++++
 25 files changed, 260 insertions(+), 18 deletions(-)
 create mode 100644 Robust/src/IR/Flat/FlatExit.java

diff --git a/Robust/src/IR/Flat/BuildCode.java b/Robust/src/IR/Flat/BuildCode.java
index 4b449885..576b3872 100644
--- a/Robust/src/IR/Flat/BuildCode.java
+++ b/Robust/src/IR/Flat/BuildCode.java
@@ -1581,6 +1581,10 @@ public class BuildCode {
       output.println("/* nop */");
       return;
 
+    case FKind.FlatExit:
+      output.println("/* exit */");
+      return;
+
     case FKind.FlatBackEdge:
       if ((state.THREAD||state.DSM)&&GENERATEPRECISEGC) {
 	if(state.DSM&&locality.getAtomic(lb).get(fn).intValue()>0) {
diff --git a/Robust/src/IR/Flat/BuildFlat.java b/Robust/src/IR/Flat/BuildFlat.java
index ff9553f4..681afda2 100644
--- a/Robust/src/IR/Flat/BuildFlat.java
+++ b/Robust/src/IR/Flat/BuildFlat.java
@@ -9,8 +9,9 @@ public class BuildFlat {
   MethodDescriptor currmd;
   TypeUtil typeutil;
 
-    HashSet breakset;
-    HashSet continueset;
+  HashSet breakset;
+  HashSet continueset;
+  FlatExit fe;
 
   public BuildFlat(State st, TypeUtil typeutil) {
     state=st;
@@ -42,9 +43,11 @@ public class BuildFlat {
     BlockNode bn=state.getMethodBody(td);
     NodePair np=flattenBlockNode(bn);
     FlatNode fn=np.getBegin();
+    fe=new FlatExit();
     if (np.getEnd().kind()!=FKind.FlatReturnNode) {
       FlatReturnNode rnflat=new FlatReturnNode(null);
       np.getEnd().addNext(rnflat);
+      rnflat.addNext(fe);
     }
 
     FlatFlagActionNode ffan=new FlatFlagActionNode(FlatFlagActionNode.PRE);
@@ -112,6 +115,7 @@ public class BuildFlat {
   private void flattenClass(ClassDescriptor cn) {
     Iterator methodit=cn.getMethods();
     while(methodit.hasNext()) {
+      fe=new FlatExit();
       currmd=(MethodDescriptor)methodit.next();
       BlockNode bn=state.getMethodBody(currmd);
 
@@ -133,6 +137,7 @@ public class BuildFlat {
 	  np.getEnd().addNext(fcunlock);
 	  FlatReturnNode rnflat=new FlatReturnNode(null);
 	  fcunlock.addNext(rnflat);
+	  rnflat.addNext(fe);
 	}
       } else if (state.DSM&&currmd.getModifiers().isAtomic()) {
 	curran.addNext(fn);
@@ -142,10 +147,12 @@ public class BuildFlat {
 	  np.getEnd().addNext(aen);
 	  FlatReturnNode rnflat=new FlatReturnNode(null);
 	  aen.addNext(rnflat);
+	  rnflat.addNext(fe);
 	}
-      } else if (np.getEnd().kind()!=FKind.FlatReturnNode) {
+      } else if (np.getEnd()!=null&&np.getEnd().kind()!=FKind.FlatReturnNode) {
 	FlatReturnNode rnflat=new FlatReturnNode(null);
 	np.getEnd().addNext(rnflat);
+	rnflat.addNext(fe);
       }
 
       FlatMethod fm=new FlatMethod(currmd);
@@ -961,7 +968,8 @@ public class BuildFlat {
       FlatNop nopend=new FlatNop();
       FlatBackEdge backedge=new FlatBackEdge();
 
-      body.getEnd().addNext(condition.getBegin());
+      if (body.getEnd()!=null)
+	body.getEnd().addNext(condition.getBegin());
       condition.getEnd().addNext(fcb);
       fcb.addFalseNext(nopend);
       fcb.addTrueNext(backedge);
@@ -992,6 +1000,7 @@ public class BuildFlat {
     }
 
     FlatReturnNode rnflat=new FlatReturnNode(retval);
+    rnflat.addNext(fe);
     FlatNode ln=rnflat;
     if (state.THREAD&&currmd.getModifiers().isSynchronized()) {
       MethodDescriptor memd=(MethodDescriptor)typeutil.getClass("Object").getMethodTable().get("MonitorExit");
@@ -1008,9 +1017,9 @@ public class BuildFlat {
 
     if (cond!=null) {
       cond.getEnd().addNext(ln);
-      return new NodePair(cond.getBegin(),rnflat);
+      return new NodePair(cond.getBegin(),null);
     } else
-      return new NodePair(ln,rnflat);
+      return new NodePair(ln,null);
   }
 
   private NodePair flattenTaskExitNode(TaskExitNode ten) {
@@ -1020,8 +1029,9 @@ public class BuildFlat {
     NodePair fcn=flattenConstraintCheck(ten.getChecks());
     ffan.addNext(fcn.getBegin());
     FlatReturnNode rnflat=new FlatReturnNode(null);
+    rnflat.addNext(fe);
     fcn.getEnd().addNext(rnflat);
-    return new NodePair(ffan, rnflat);
+    return new NodePair(ffan, null);
   }
 
   private NodePair flattenConstraintCheck(Vector ccs) {
diff --git a/Robust/src/IR/Flat/FKind.java b/Robust/src/IR/Flat/FKind.java
index 467337a5..58f81158 100644
--- a/Robust/src/IR/Flat/FKind.java
+++ b/Robust/src/IR/Flat/FKind.java
@@ -26,4 +26,5 @@ public class FKind {
   public static final int FlatSESEEnterNode=23;
   public static final int FlatSESEExitNode=24;
   public static final int FlatInstanceOfNode=25;
+  public static final int FlatExit=26;
 }
diff --git a/Robust/src/IR/Flat/FlatAtomicEnterNode.java b/Robust/src/IR/Flat/FlatAtomicEnterNode.java
index b1426437..a58cfcd1 100644
--- a/Robust/src/IR/Flat/FlatAtomicEnterNode.java
+++ b/Robust/src/IR/Flat/FlatAtomicEnterNode.java
@@ -10,6 +10,11 @@ public class FlatAtomicEnterNode extends FlatNode {
     this.id=identifier++;
   }
 
+  public void rewriteUse() {
+  }
+  public void rewriteDef() {
+  }
+
   /* Returns an unique identifier for this atomic enter node */
 
   public int getIdentifier() {
diff --git a/Robust/src/IR/Flat/FlatAtomicExitNode.java b/Robust/src/IR/Flat/FlatAtomicExitNode.java
index d8c10952..edcb93b4 100644
--- a/Robust/src/IR/Flat/FlatAtomicExitNode.java
+++ b/Robust/src/IR/Flat/FlatAtomicExitNode.java
@@ -6,10 +6,13 @@ public class FlatAtomicExitNode extends FlatNode {
   public FlatAtomicExitNode(FlatAtomicEnterNode faen) {
     this.faen=faen;
   }
-
   public FlatAtomicEnterNode getAtomicEnter() {
     return faen;
   }
+  public void rewriteUse() {
+  }
+  public void rewriteDef() {
+  }
 
   public String toString() {
     return "atomicexit";
diff --git a/Robust/src/IR/Flat/FlatBackEdge.java b/Robust/src/IR/Flat/FlatBackEdge.java
index cca91b29..e6905cfe 100644
--- a/Robust/src/IR/Flat/FlatBackEdge.java
+++ b/Robust/src/IR/Flat/FlatBackEdge.java
@@ -4,7 +4,10 @@ import java.util.Vector;
 public class FlatBackEdge extends FlatNode {
   public FlatBackEdge() {
   }
-
+  public void rewriteUse() {
+  }
+  public void rewriteDst() {
+  }
   public String toString() {
     return "backedge";
   }
diff --git a/Robust/src/IR/Flat/FlatCall.java b/Robust/src/IR/Flat/FlatCall.java
index 8b18abe6..f8a5d39e 100644
--- a/Robust/src/IR/Flat/FlatCall.java
+++ b/Robust/src/IR/Flat/FlatCall.java
@@ -13,6 +13,22 @@ public class FlatCall extends FlatNode {
     this.this_temp=this_temp;
     this.args=args;
   }
+  public void rewriteUse(TempMap t) {
+    for(int i=0;i<args.length;i++)
+      args[i]=t.tempMap(args[i]);
+    this_temp=t.tempMap(this_temp);
+  }
+  public void rewriteDef(TempMap t) {
+    dst=t.tempMap(dst);
+  }
+  public FlatNode clone(TempMap t) {
+    TempDescriptor ndst=t.tempMap(dst);
+    TempDescriptor nthis=t.tempMap(this_temp);
+    TempDescriptor[] nargs=new TempDescriptor[args.length];
+    for(int i=0;i<nargs.length;i++)
+      nargs[i]=t.tempMap(args[i]);
+    return new FlatCall(method, ndst, nthis, nargs);
+  }
 
   public MethodDescriptor getMethod() {
     return method;
diff --git a/Robust/src/IR/Flat/FlatCastNode.java b/Robust/src/IR/Flat/FlatCastNode.java
index 73f5a6eb..ee10d4d3 100644
--- a/Robust/src/IR/Flat/FlatCastNode.java
+++ b/Robust/src/IR/Flat/FlatCastNode.java
@@ -12,6 +12,16 @@ public class FlatCastNode extends FlatNode {
     this.dst=dst;
   }
 
+  public FlatNode clone(TempMap t) {
+    return new FlatCastNode(type, t.tempMap(src), t.tempMap(dst));
+  }
+  public void rewriteUse(TempMap t) {
+    src=t.tempMap(src);
+  }
+  public void rewriteDef(TempMap t) {
+    dst=t.tempMap(dst);
+  }
+
   public String toString() {
     return "FlatCastNode_"+dst.toString()+"=("+type.toString()+")"+src.toString();
   }
diff --git a/Robust/src/IR/Flat/FlatCondBranch.java b/Robust/src/IR/Flat/FlatCondBranch.java
index b481a5e2..34b0d903 100644
--- a/Robust/src/IR/Flat/FlatCondBranch.java
+++ b/Robust/src/IR/Flat/FlatCondBranch.java
@@ -9,6 +9,18 @@ public class FlatCondBranch extends FlatNode {
   public FlatCondBranch(TempDescriptor td) {
     test_cond=td;
   }
+  public void rewriteDef(TempMap t) {
+  }
+  public void rewriteUse(TempMap t) {
+    test_cond=t.tempMap(test_cond);
+  }
+
+  public FlatNode clone(TempMap t) {
+    FlatCondBranch fcb=new FlatCondBranch(t.tempMap(test_cond));
+    fcb.trueprob=trueprob;
+    fcb.loop=loop;
+    return fcb;
+  }
 
   public void setLoop() {
     loop=true;
diff --git a/Robust/src/IR/Flat/FlatElementNode.java b/Robust/src/IR/Flat/FlatElementNode.java
index 2e42d744..9eee5eea 100644
--- a/Robust/src/IR/Flat/FlatElementNode.java
+++ b/Robust/src/IR/Flat/FlatElementNode.java
@@ -16,6 +16,17 @@ public class FlatElementNode extends FlatNode {
     return true;
   }
 
+  public FlatNode clone(TempMap t) {
+    return new FlatElementNode(t.tempMap(src), t.tempMap(index), t.tempMap(dst));
+  }
+  public void rewriteDef(TempMap t) {
+    dst=t.tempMap(dst);
+  }
+  public void rewriteUse(TempMap t) {
+    index=t.tempMap(index);
+    src=t.tempMap(src);
+  }
+
   public TempDescriptor getIndex() {
     return index;
   }
diff --git a/Robust/src/IR/Flat/FlatExit.java b/Robust/src/IR/Flat/FlatExit.java
new file mode 100644
index 00000000..1102c5b0
--- /dev/null
+++ b/Robust/src/IR/Flat/FlatExit.java
@@ -0,0 +1,22 @@
+package IR.Flat;
+import java.util.Vector;
+
+public class FlatExit extends FlatNode {
+  public FlatExit() {
+  }
+
+  public String toString() {
+    return "exit";
+  }
+
+  public int kind() {
+    return FKind.FlatExit;
+  }
+  public FlatNode clone() {
+    return new FlatExit();
+  }
+  public void rewriteUse(TempMap t) {
+  }
+  public void rewriteDst(TempMap t) {
+  }
+}
diff --git a/Robust/src/IR/Flat/FlatFieldNode.java b/Robust/src/IR/Flat/FlatFieldNode.java
index 60462df9..d74d8582 100644
--- a/Robust/src/IR/Flat/FlatFieldNode.java
+++ b/Robust/src/IR/Flat/FlatFieldNode.java
@@ -12,6 +12,16 @@ public class FlatFieldNode extends FlatNode {
     this.dst=dst;
   }
 
+  public FlatNode clone(TempMap t) {
+    return new FlatFieldNode(field, t.tempMap(src), t.tempMap(dst));
+  }
+  public void rewriteUse(TempMap t) {
+    src=t.tempMap(src);
+  }
+  public void rewriteDef(TempMap t) {
+    dst=t.tempMap(dst);
+  }
+
   public FieldDescriptor getField() {
     return field;
   }
diff --git a/Robust/src/IR/Flat/FlatInstanceOfNode.java b/Robust/src/IR/Flat/FlatInstanceOfNode.java
index 8389bea6..0cc70f73 100644
--- a/Robust/src/IR/Flat/FlatInstanceOfNode.java
+++ b/Robust/src/IR/Flat/FlatInstanceOfNode.java
@@ -12,6 +12,15 @@ public class FlatInstanceOfNode extends FlatNode {
     this.dst=dst;
   }
 
+  public FlatNode clone(TempMap m) {
+    return new FlatInstanceOfNode(t, m.tempMap(src), m.tempMap(dst));
+  }
+  public void rewriteUse(TempMap t) {
+    src=t.tempMap(src);
+  }
+  public void rewriteDef(TempMap t) {
+    dst=t.tempMap(dst);
+  }
   public TypeDescriptor getType() {
     return t;
   }
diff --git a/Robust/src/IR/Flat/FlatLiteralNode.java b/Robust/src/IR/Flat/FlatLiteralNode.java
index fba83bfb..a141ebdf 100644
--- a/Robust/src/IR/Flat/FlatLiteralNode.java
+++ b/Robust/src/IR/Flat/FlatLiteralNode.java
@@ -12,6 +12,14 @@ public class FlatLiteralNode extends FlatNode {
     this.dst=dst;
   }
 
+  public FlatNode clone(TempMap t) {
+    return new FlatLiteralNode(type, value, t.tempMap(dst));
+  }
+  public void rewriteUse(TempMap t) {
+  }
+  public void rewriteDst(TempMap t) {
+    dst=t.tempMap(dst);
+  }
   public TypeDescriptor getType() {
     return type;
   }
diff --git a/Robust/src/IR/Flat/FlatMethod.java b/Robust/src/IR/Flat/FlatMethod.java
index 4955f28a..0cd5e7b2 100644
--- a/Robust/src/IR/Flat/FlatMethod.java
+++ b/Robust/src/IR/Flat/FlatMethod.java
@@ -141,10 +141,18 @@ public class FlatMethod extends FlatNode {
       if (current_node==null) {
 	current_node=(FlatNode)tovisit.iterator().next();
 	tovisit.remove(current_node);
+      } else {
+	if (tovisit.contains(current_node))
+	  tovisit.remove(current_node);
       }
       visited.add(current_node);
-      if (nodetolabel.containsKey(current_node))
+      if (nodetolabel.containsKey(current_node)) {
 	st+="L"+nodetolabel.get(current_node)+":\n";
+	for(int i=0;i<current_node.numPrev();i++) {
+	  st+="i="+i+" "+current_node.getPrev(i);
+	}
+	st+="\n";
+      }
       if (current_node.numNext()==0) {
 	if (map==null)
 	  st+="   "+current_node.toString()+"\n";
diff --git a/Robust/src/IR/Flat/FlatNew.java b/Robust/src/IR/Flat/FlatNew.java
index 50616075..27e9a935 100644
--- a/Robust/src/IR/Flat/FlatNew.java
+++ b/Robust/src/IR/Flat/FlatNew.java
@@ -16,6 +16,17 @@ public class FlatNew extends FlatNode {
     this.disjointId=null;
   }
 
+  public void rewriteUse(TempMap t) {
+    size=t.tempMap(size);
+  }
+  public void rewriteDef(TempMap t) {
+    dst=t.tempMap(dst);
+  }
+
+  public FlatNode clone(TempMap t) {
+    return new FlatNew(type, t.tempMap(dst), t.tempMap(size), isglobal, disjointId);
+  }
+
   public FlatNew(TypeDescriptor type, TempDescriptor dst, boolean isglobal, String disjointId) {
     this.type=type;
     this.dst=dst;
diff --git a/Robust/src/IR/Flat/FlatNode.java b/Robust/src/IR/Flat/FlatNode.java
index 57dcb736..7f82c2d7 100644
--- a/Robust/src/IR/Flat/FlatNode.java
+++ b/Robust/src/IR/Flat/FlatNode.java
@@ -26,12 +26,13 @@ public class FlatNode {
   public FlatNode getPrev(int i) {
     return (FlatNode) prev.get(i);
   }
-
   public void addNext(FlatNode n) {
     next.add(n);
     n.addPrev(this);
   }
-
+  public void removePrev(FlatNode n) {
+    prev.remove(n);
+  }
   /** This function modifies the graph */
   public void setNext(int i, FlatNode n) {
     FlatNode old=getNext(i);
@@ -39,19 +40,60 @@ public class FlatNode {
     old.prev.remove(this);
     n.addPrev(this);
   }
-
-  protected void addPrev(FlatNode p) {
+  /** This function modifies the graph */
+  public void setprev(int i, FlatNode n) {
+    prev.set(i, n);
+  }
+  /** This function modifies the graph */
+  public void setnext(int i, FlatNode n) {
+    next.set(i, n);
+  }
+  public void addPrev(FlatNode p) {
     prev.add(p);
   }
   public int kind() {
     throw new Error();
   }
-
   public TempDescriptor [] readsTemps() {
     return new TempDescriptor[0];
   }
-
   public TempDescriptor [] writesTemps() {
     return new TempDescriptor[0];
   }
+  public FlatNode clone(TempMap t) {
+    throw new Error();
+  }
+
+  public void rewriteUse(TempMap t) {
+    System.out.println(toString());
+    throw new Error();
+  }
+
+  public void rewriteDef(TempMap t) {
+    System.out.println(toString());
+    throw new Error();
+  }
+
+  public void replace(FlatNode fnnew) {
+    fnnew.prev.setSize(prev.size());
+    fnnew.next.setSize(next.size());
+    for(int i=0;i<prev.size();i++) {
+      FlatNode nprev=(FlatNode)prev.get(i);
+      fnnew.prev.set(i,nprev);
+      for(int j=0;j<nprev.numNext();j++) {
+	FlatNode n=nprev.getNext(j);
+	if (n==this)
+	  nprev.next.set(j, fnnew);
+      }
+    }
+    for(int i=0;i<next.size();i++) {
+      FlatNode nnext=(FlatNode)next.get(i);
+      fnnew.next.set(i,nnext);;
+      for(int j=0;j<nnext.numPrev();j++) {
+	FlatNode n=nnext.getPrev(j);
+	if (n==this)
+	  nnext.prev.set(j, fnnew);
+      }
+    }
+  }
 }
diff --git a/Robust/src/IR/Flat/FlatNop.java b/Robust/src/IR/Flat/FlatNop.java
index fdcab4ed..7b79dac3 100644
--- a/Robust/src/IR/Flat/FlatNop.java
+++ b/Robust/src/IR/Flat/FlatNop.java
@@ -12,4 +12,9 @@ public class FlatNop extends FlatNode {
   public int kind() {
     return FKind.FlatNop;
   }
+  public FlatNode clone() {
+    return new FlatNop();
+  }
+  public void rewriteUse(TempMap t) {
+  }
 }
diff --git a/Robust/src/IR/Flat/FlatOffsetNode.java b/Robust/src/IR/Flat/FlatOffsetNode.java
index e9c61db2..7fa88996 100644
--- a/Robust/src/IR/Flat/FlatOffsetNode.java
+++ b/Robust/src/IR/Flat/FlatOffsetNode.java
@@ -13,6 +13,14 @@ public class FlatOffsetNode extends FlatNode {
     this.dst = dst;
   }
 
+  public FlatNode clone(TempMap t) {
+    return new FlatOffsetNode(baseclass, field, t.tempMap(dst));
+  }
+  public void rewriteUse(TempMap t) {
+  }
+  public void rewriteDef(TempMap t) {
+    dst=t.tempMap(dst);
+  }
   public TypeDescriptor getClassType() {
     return baseclass;
   }
diff --git a/Robust/src/IR/Flat/FlatOpNode.java b/Robust/src/IR/Flat/FlatOpNode.java
index ade59315..87d13521 100644
--- a/Robust/src/IR/Flat/FlatOpNode.java
+++ b/Robust/src/IR/Flat/FlatOpNode.java
@@ -15,6 +15,17 @@ public class FlatOpNode extends FlatNode {
     this.op=op;
   }
 
+  public FlatNode clone(TempMap t) {
+    return new FlatOpNode(t.tempMap(dest), t.tempMap(left), t.tempMap(right), op);
+  }
+  public void rewriteDef(TempMap t) {
+    dest=t.tempMap(dest);
+  }
+  public void rewriteUse(TempMap t) {
+    left=t.tempMap(left);
+    right=t.tempMap(right);
+  }
+
   public TempDescriptor getDest() {
     return dest;
   }
diff --git a/Robust/src/IR/Flat/FlatReturnNode.java b/Robust/src/IR/Flat/FlatReturnNode.java
index f90fd751..bc98ba55 100644
--- a/Robust/src/IR/Flat/FlatReturnNode.java
+++ b/Robust/src/IR/Flat/FlatReturnNode.java
@@ -7,6 +7,14 @@ public class FlatReturnNode extends FlatNode {
     this.tempdesc=td;
   }
 
+  public FlatNode clone(TempMap t) {
+    return new FlatReturnNode(t.tempMap(tempdesc));
+  }
+  public void rewriteUse(TempMap t) {
+    tempdesc=t.tempMap(tempdesc);
+  }
+  public void rewriteDef(TempMap t) {
+  }
   public String toString() {
     return "FlatReturnNode_return "+tempdesc;
   }
diff --git a/Robust/src/IR/Flat/FlatSESEEnterNode.java b/Robust/src/IR/Flat/FlatSESEEnterNode.java
index 15a1606f..7c521a3d 100644
--- a/Robust/src/IR/Flat/FlatSESEEnterNode.java
+++ b/Robust/src/IR/Flat/FlatSESEEnterNode.java
@@ -12,7 +12,10 @@ public class FlatSESEEnterNode extends FlatNode {
     this.id=identifier++;
     treeNode = sn;    
   }
-
+  public void rewriteUse() {
+  }
+  public void rewriteDef() {
+  }
   public SESENode getTreeNode() {
     return treeNode;
   }
diff --git a/Robust/src/IR/Flat/FlatSESEExitNode.java b/Robust/src/IR/Flat/FlatSESEExitNode.java
index bcfd0451..eb77cbce 100644
--- a/Robust/src/IR/Flat/FlatSESEExitNode.java
+++ b/Robust/src/IR/Flat/FlatSESEExitNode.java
@@ -9,7 +9,10 @@ public class FlatSESEExitNode extends FlatNode {
   public FlatSESEExitNode( SESENode sn ) {
     treeNode = sn;
   }
-
+  public void rewriteUse() {
+  }
+  public void rewriteDef() {
+  }
   public SESENode getTreeNode() {
     return treeNode;
   }
diff --git a/Robust/src/IR/Flat/FlatSetElementNode.java b/Robust/src/IR/Flat/FlatSetElementNode.java
index 6dd4a624..3fdfb085 100644
--- a/Robust/src/IR/Flat/FlatSetElementNode.java
+++ b/Robust/src/IR/Flat/FlatSetElementNode.java
@@ -12,6 +12,16 @@ public class FlatSetElementNode extends FlatNode {
     this.dst=dst;
   }
 
+  public FlatNode clone(TempMap t) {
+    return new FlatSetElementNode(t.tempMap(dst), t.tempMap(index), t.tempMap(src));
+  }
+  public void rewriteUse(TempMap t) {
+    src=t.tempMap(src);
+    dst=t.tempMap(dst);
+    index=t.tempMap(index);
+  }
+  public void rewriteDef(TempMap t) {
+  }
   public boolean needsBoundsCheck() {
     return true;
   }
diff --git a/Robust/src/IR/Flat/FlatSetFieldNode.java b/Robust/src/IR/Flat/FlatSetFieldNode.java
index 50e834ed..f57d69e1 100644
--- a/Robust/src/IR/Flat/FlatSetFieldNode.java
+++ b/Robust/src/IR/Flat/FlatSetFieldNode.java
@@ -12,6 +12,15 @@ public class FlatSetFieldNode extends FlatNode {
     this.dst=dst;
   }
 
+  public FlatNode clone(TempMap t) {
+    return new FlatSetFieldNode(t.tempMap(dst), field, t.tempMap(src));
+  }
+  public void rewriteUse(TempMap t) {
+    dst=t.tempMap(dst);
+    src=t.tempMap(src);
+  }
+  public void rewriteDef(TempMap t) {
+  }
   public TempDescriptor getSrc() {
     return src;
   }
-- 
2.34.1