org.sonar.javascript.metrics.LineVisitor Maven / Gradle / Ivy
/*
* SonarQube JavaScript Plugin
* Copyright (C) 2011-2018 SonarSource SA
* mailto:info AT sonarsource DOT com
*
* This program 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 3 of the License, or (at your option) any later version.
*
* This program 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 program; if not, write to the Free Software Foundation,
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*/
package org.sonar.javascript.metrics;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import org.sonar.javascript.tree.impl.lexical.InternalSyntaxToken;
import org.sonar.plugins.javascript.api.tree.Tree;
import org.sonar.plugins.javascript.api.tree.lexical.SyntaxToken;
import org.sonar.plugins.javascript.api.visitors.DoubleDispatchVisitor;
/**
* Visitor that computes the number of lines of code of a file.
*/
public class LineVisitor extends DoubleDispatchVisitor {
private Set lines = new HashSet<>();
public LineVisitor(Tree tree) {
scan(tree);
}
public LineVisitor(List treeList) {
scan(treeList);
}
@Override
public void visitToken(SyntaxToken token) {
if (!((InternalSyntaxToken) token).isEOF()) {
lines.add(token.line());
}
}
public int getLinesOfCodeNumber() {
return lines.size();
}
public Set getLinesOfCode() {
return lines;
}
}