plugins/org.eclipse.dltk.python.core/src/org/eclipse/dltk/python/parser/ast/expressions/PythonForListExpression.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.expressions;

import java.util.Iterator;
import java.util.List;

import org.eclipse.dltk.ast.ASTVisitor;
import org.eclipse.dltk.ast.expressions.Expression;
import org.eclipse.dltk.ast.expressions.ExpressionList;
import org.eclipse.dltk.python.parser.ast.PythonConstants;
import org.eclipse.dltk.utils.CorePrinter;


/*
 * for a in v expression
 */
public class PythonForListExpression extends Expression
{

	// for a in [1,2,3] if a > 2
	private Expression fVars; // this is a expression.

	private Expression fFrom; // this is [1,2,3] expression

	private ExpressionList fIfList; // this a > 2 expression.

	public PythonForListExpression( Expression vars, Expression from ) {

		super( );
		this.fVars = vars;
		this.fFrom = from;
		this.setStart(vars.sourceStart());
		this.setEnd(from.sourceEnd());
	}


	@Override
	public int getKind( ) {

		return PythonConstants.E_FORIFLIST;
	}


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

		if( pVisitor.visit( this ) ) {
			if( this.fVars != null ) {
				this.fVars.traverse( pVisitor );
			}
			if( this.fFrom != null ) {
				this.fFrom.traverse( pVisitor );
			}
			if( this.fIfList != null ) {
				this.fIfList.traverse( pVisitor );
			}
			pVisitor.endvisit( this );
		}

	}


	@Override
	public void printNode( CorePrinter output ) {

		output.formatPrintLn( "for " );
		if( this.fVars != null ) {
			this.fVars.printNode( output );
		}

		output.formatPrintLn( " in " );
		if( this.fFrom != null ) {
			this.fFrom.printNode( output );
		}
		if( this.fIfList != null ) {
			List/*< Expression >*/ ifs = this.fIfList.getExpressions( );
			Iterator i = ifs.iterator();
			while( i.hasNext() ) {
				Expression ife = (Expression)i.next();
				output.formatPrintLn( "if " );
				ife.printNode( output );
			}
		}
	}

	public void setIfListExpression( ExpressionList ifBlock ) {

		this.fIfList = ifBlock;
		this.setEnd(ifBlock.sourceEnd());
	}

}