/*
 * Contributions to FindBugs
 * Copyright (C) 2007, Institut for Software
 * An Institut of the University of Applied Sciences Rapperswil
 *
 * Author: Thierry Wyss, Marco Busarello
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */
package edu.umd.cs.findbugs.quickfix.util;

import static edu.umd.cs.findbugs.quickfix.util.ConditionCheck.checkForNull;

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

/**
 * @author <a href="mailto:twyss@hsr.ch">Thierry Wyss</a>
 * @author <a href="mailto:mbusarel@hsr.ch">Marco Busarello</a>
 * @version 1.0
 */
public class SourceLineVisitor extends ASTVisitor {

    private final CompilationUnit compilationUnit;

    private final int startLine;

    private final int endLine;

    private ASTNode node = null;

    public SourceLineVisitor(CompilationUnit compilationUnit, int startLine, int endLine) {
        super();
        checkForNull(compilationUnit, "compilation unit");
        this.compilationUnit = compilationUnit;
        this.startLine = startLine;
        this.endLine = endLine;
    }

    @Override
    public void preVisit(ASTNode node) {
        if (this.node == null) {
            int lineNumber = compilationUnit.getLineNumber(node.getStartPosition());
            if (lineNumber >= startLine && lineNumber <= endLine) {
                this.node = node;
            }
        }
    }

    public ASTNode getASTNode() {
        return node;
    }

}