001/* 002 * Sonar, open source software quality management tool. 003 * Copyright (C) 2008-2012 SonarSource 004 * mailto:contact AT sonarsource DOT com 005 * 006 * Sonar is free software; you can redistribute it and/or 007 * modify it under the terms of the GNU Lesser General Public 008 * License as published by the Free Software Foundation; either 009 * version 3 of the License, or (at your option) any later version. 010 * 011 * Sonar is distributed in the hope that it will be useful, 012 * but WITHOUT ANY WARRANTY; without even the implied warranty of 013 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 014 * Lesser General Public License for more details. 015 * 016 * You should have received a copy of the GNU Lesser General Public 017 * License along with Sonar; if not, write to the Free Software 018 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02 019 */ 020package org.sonar.java.ast.visitor; 021 022import java.util.Arrays; 023import java.util.List; 024 025import org.sonar.squid.measures.Metric; 026 027import com.puppycrawl.tools.checkstyle.api.DetailAST; 028import com.puppycrawl.tools.checkstyle.api.TokenTypes; 029 030public class BranchVisitor extends JavaAstVisitor { 031 032 private static final List<Integer> TOKENS = Arrays.asList(TokenTypes.LITERAL_WHILE, TokenTypes.LITERAL_DO, TokenTypes.LITERAL_FOR, 033 TokenTypes.LITERAL_IF, TokenTypes.LITERAL_CASE, TokenTypes.LITERAL_CATCH, 034 TokenTypes.QUESTION, TokenTypes.LAND, TokenTypes.LOR, TokenTypes.LITERAL_THROW, 035 TokenTypes.LITERAL_RETURN); 036 037 @Override 038 public List<Integer> getWantedTokens() { 039 return TOKENS; 040 } 041 042 @Override 043 public void visitToken(DetailAST ast) { 044 if (isLastMethodReturnStatement(ast)) { 045 return; 046 } 047 peekSourceCode().add(Metric.BRANCHES, 1); 048 } 049 050 private boolean isLastMethodReturnStatement(DetailAST ast) { 051 if (ast.getType() == TokenTypes.LITERAL_RETURN) { 052 DetailAST parent = ast.getParent(); 053 if (parent.getType() == TokenTypes.SLIST) { 054 DetailAST method = AstUtils.findParent(ast, TokenTypes.METHOD_DEF); 055 if (method != null && method.findFirstToken(TokenTypes.SLIST) == parent) { 056 return true; 057 } 058 } 059 } 060 return false; 061 } 062}