Case Study: adding data and statistics

[ifi-stolz-refaktor.git] / case-study / refaktor-before / src / no / uio / ifi / refaktor / analyze / checkers / LastStatementFinder.java

package no.uio.ifi.refaktor.analyze.checkers;

import org.eclipse.jdt.core.dom.ASTNode;
import org.eclipse.jdt.core.dom.ASTVisitor;
import org.eclipse.jdt.core.dom.Statement;

class LastStatementFinder extends ASTVisitor {
	private final ASTNode parent;
	private Statement lastStatement;

	public LastStatementFinder(ASTNode parent) {
		this.parent = parent;
	}

	@Override
	public boolean preVisit2(ASTNode node) {
		if (node == parent)
			return true;
		
		if (node instanceof Statement && isPlacedAfterLastStatement(node)) {
			lastStatement = (Statement) node;
		}

		return false;
	}

	private boolean isPlacedAfterLastStatement(ASTNode node) {
		if (lastStatement == null)
			return true;
		
		int nodeEndPosition = node.getStartPosition() + node.getLength();
		int lastStatementEndPosition = lastStatement.getStartPosition() + lastStatement.getLength();
		return nodeEndPosition > lastStatementEndPosition;
	}

	public Statement getLastStatement() {
		return lastStatement;
	}
}