X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=Robust%2Fsrc%2FIR%2FTree%2FReturnNode.java;h=6697d1d5a78a87fb726170884bd4ce8deaacff44;hb=ac6191b514c0e54b468623bf868134e1ce809df5;hp=60ae5762d90eff54c0f3ca331b45f5ad9c1245ef;hpb=320d9967eb9b0064b9a9bbb8718690884768889a;p=IRC.git

diff --git a/Robust/src/IR/Tree/ReturnNode.java b/Robust/src/IR/Tree/ReturnNode.java
index 60ae5762..6697d1d5 100644
--- a/Robust/src/IR/Tree/ReturnNode.java
+++ b/Robust/src/IR/Tree/ReturnNode.java
@@ -1,21 +1,27 @@
 package IR.Tree;
 
-class ReturnNode extends BlockStatementNode {
-    ExpressionNode en;
+public class ReturnNode extends BlockStatementNode {
+  ExpressionNode en;
 
-    public ReturnNode() {
-	en=null;
-    }
+  public ReturnNode() {
+    en=null;
+  }
 
-    public ReturnNode(ExpressionNode en) {
-	this.en=en;
-    }
+  public ReturnNode(ExpressionNode en) {
+    this.en=en;
+  }
 
-    public String printNode(int indent) {
-	if (en==null)
-	    return "return";
-	else
-	    return "return "+en.printNode(indent);
-    }
+  public ExpressionNode getReturnExpression() {
+    return en;
+  }
 
+  public String printNode(int indent) {
+    if (en==null)
+      return "return";
+    else
+      return "return "+en.printNode(indent);
+  }
+  public int kind() {
+    return Kind.ReturnNode;
+  }
 }