ASTReturnStatement.java

package edu.udel.cis.vsl.tass.front.minimp.ast.statement;

import edu.udel.cis.vsl.tass.front.minimp.ast.expression.ASTExpressionIF;

public class ASTReturnStatement extends ASTStatement {
  private ASTExpressionIF returnBody;

  public ASTReturnStatement(ASTExpressionIF body) {
    returnBody = body;
  }

  public ASTExpressionIF getReturnBody() {
    return returnBody;
  }

  public String toString() {
    if (this.returnBody == null) {
      return "return;";
    } else {
      return "return " + this.returnBody.toString() + ";";
    }
  }
}