plugins/org.eclipse.gmf.xpand/src/org/eclipse/gmf/internal/xpand/ast/AstIterator.java

/**
 * Copyright (c) 2009 Borland Software Corp.
 * 
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *    Alexander Shatalin (Borland) - initial API and implementation
 */
package org.eclipse.gmf.internal.xpand.ast;

import org.eclipse.gmf.internal.xpand.model.XpandAdvice;
import org.eclipse.gmf.internal.xpand.model.XpandDefinition;

public class AstIterator {

	private AstVisitor myVisitor;

	AstIterator(AstVisitor visitor) {
		myVisitor = visitor;
	}

	public void iterate(Template template) {
		if (!myVisitor.visit(template)) {
			return;
		}
		for (XpandDefinition definition : template.getDefinitions()) {
			iterate(definition);
		}

		for (XpandAdvice advice : template.getAdvices()) {
			iterate(advice);
		}
	}

	public void iterate(XpandDefinition definition) {
		if (!myVisitor.visit(definition)) {
			return;
		}
		if (definition instanceof AbstractDefinition) {
			iterate(((AbstractDefinition) definition).getBody());
		}
	}

	public void iterate(XpandAdvice advice) {
		if (!myVisitor.visit(advice)) {
			return;
		}
		if (advice instanceof AbstractDefinition) {
			iterate(((AbstractDefinition) advice).getBody());
		}
	}

	public void iterate(Statement[] body) {
		for (Statement statement : body) {
			if (statement instanceof ErrorStatement) {
				iterate((ErrorStatement) statement);
			} else if (statement instanceof ExpandStatement) {
				iterate((ExpandStatement) statement);
			} else if (statement instanceof ExpressionStatement) {
				iterate((ExpressionStatement) statement);
			} else if (statement instanceof FileStatement) {
				iterate((FileStatement) statement);
			} else if (statement instanceof ForEachStatement) {
				iterate((ForEachStatement) statement);
			} else if (statement instanceof IfStatement) {
				iterate((IfStatement) statement);
			} else if (statement instanceof LetStatement) {
				iterate((LetStatement) statement);
			} else if (statement instanceof ProtectStatement) {
				iterate((ProtectStatement) statement);
			} else if (statement instanceof TextStatement) {
				iterate((TextStatement) statement);
			} else {
				iterate(statement);
			}
		}
	}

	public void iterate(Statement statement) {
		myVisitor.visit(statement);
	}

	public void iterate(TextStatement statement) {
		myVisitor.visit(statement);
	}

	public void iterate(ProtectStatement statement) {
		if (!myVisitor.visit(statement)) {
			return;
		}
		iterate(statement.getBody());
	}

	public void iterate(LetStatement statement) {
		if (!myVisitor.visit(statement)) {
			return;
		}
		iterate(statement.getBody());
	}

	public void iterate(IfStatement statement) {
		if (!myVisitor.visit(statement)) {
			return;
		}
		iterate(statement.getThenPart());
		if (statement.getElseIf() != null) {
			iterate(statement.getElseIf());
		}
	}

	public void iterate(ForEachStatement statement) {
		if (!myVisitor.visit(statement)) {
			return;
		}
		iterate(statement.getBody());
	}

	public void iterate(FileStatement statement) {
		if (!myVisitor.visit(statement)) {
			return;
		}
		iterate(statement.getBody());
	}

	public void iterate(ExpressionStatement statement) {
		myVisitor.visit(statement);
	}

	public void iterate(ExpandStatement statement) {
		myVisitor.visit(statement);
	}

	public void iterate(ErrorStatement statement) {
		myVisitor.visit(statement);
	}

}