add more comments

[IRC.git] / Robust / src / IR / Tree / ReturnNode.java

diff --git a/Robust/src/IR/Tree/ReturnNode.java b/Robust/src/IR/Tree/ReturnNode.java
index 60ae5762d90eff54c0f3ca331b45f5ad9c1245ef..780a4cbda58c82119b519306ae42c8a5c58ff947 100644 (file)
--- a/Robust/src/IR/Tree/ReturnNode.java
+++ b/Robust/src/IR/Tree/ReturnNode.java
@@ -1,6 +1,6 @@
 package IR.Tree;
 
-class ReturnNode extends BlockStatementNode {
+public class ReturnNode extends BlockStatementNode {
     ExpressionNode en;
 
     public ReturnNode() {
@@ -11,11 +11,17 @@ class ReturnNode extends BlockStatementNode {
        this.en=en;
     }
 
+    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;
+    }
 }