plugins/org.eclipse.dltk.python.core/src/org/eclipse/dltk/python/parser/ast/statements/IfStatement.java

/*******************************************************************************
 * Copyright (c) 2005, 2016 IBM Corporation and others.
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 * 
 * SPDX-License-Identifier: EPL-2.0
 *
 *******************************************************************************/
package org.eclipse.dltk.python.parser.ast.statements;

import org.eclipse.dltk.ast.ASTVisitor;
import org.eclipse.dltk.ast.DLTKToken;
import org.eclipse.dltk.ast.statements.Block;
import org.eclipse.dltk.ast.statements.Statement;
import org.eclipse.dltk.utils.CorePrinter;

/**
 * If statement.
 */
public class IfStatement extends Statement {
	/**
	 * Condition expression.
	 */
	private Statement fCondition;

	/**
	 * Then statement of if.
	 */
	private Statement fThenStatement;

	/**
	 * Else statement of if. Can be null, or possible EmptyStatement..
	 */
	private Statement fElseStatement;

	public IfStatement(DLTKToken ifToken, Statement condition,
			Statement thenStatement) {

		super(ifToken);
		this.fCondition = condition;
		this.fThenStatement = thenStatement;
	}

	public IfStatement(Statement condition, Statement thenStatement, Statement elseStatement) {
		this.fCondition = condition;
		this.fThenStatement = thenStatement;
		this.fElseStatement = elseStatement;
	}

	@Override
	public void traverse(ASTVisitor pVisitor) throws Exception {

		if (pVisitor.visit(this)) {
			if (fCondition != null) {
				fCondition.traverse(pVisitor);
			}
			if (fThenStatement != null) {
				fThenStatement.traverse(pVisitor);
			}
			if (fElseStatement != null) {
				fElseStatement.traverse(pVisitor);
			}
			pVisitor.endvisit(this);
		}
	}

	@Override
	public int getKind() {
		return S_IF;
	}

	// TODO: Replace to acceptElse for similarity.
	/**
	 * @deprecated
	 */
	@Deprecated
	public void setElse(Statement elses) {
		this.acceptElse(elses);
	}

	/**
	 * Acccept Else statement.
	 *
	 * @param elseStatement
	 */
	public void acceptElse(Statement elseStatement) {
		this.fElseStatement = elseStatement;
		if (this.fElseStatement != null) {
			this.setEnd(this.fElseStatement.sourceEnd());
		}
	}

	/**
	 * Return else statement.
	 *
	 * @return - else statement. Be aware can be null.
	 */
	public Statement getElse() {
		return fElseStatement;
	}

	public Statement getThen() {
		return fThenStatement;
	}

	public Statement getCondition() {
		return this.fCondition;
	}

	@Override
	public void printNode(CorePrinter output) {
		output.formatPrintLn("if: ");
		if (this.fCondition != null) {
			this.fCondition.printNode(output);
		}
		if (this.fThenStatement != null) {
			if (!(this.fThenStatement instanceof Block)) {
				output.indent();
			}
			this.fThenStatement.printNode(output);
			if (!(this.fThenStatement instanceof Block)) {
				output.dedent();
			}
		}
		if (this.fElseStatement != null) {
			output.formatPrintLn("else:");
			if (!(this.fElseStatement instanceof Block)) {
				output.indent();
			}
			this.fElseStatement.printNode(output);
			if (!(this.fElseStatement instanceof Block)) {
				output.dedent();
			}
		}

	}
}