001 /* 002 * Sonar, open source software quality management tool. 003 * Copyright (C) 2008-2011 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 */ 020 021 package org.sonar.squid.api; 022 023 import org.sonar.squid.measures.MetricDef; 024 025 public class SourceCodeTreeDecorator { 026 027 private final SourceProject project; 028 029 public SourceCodeTreeDecorator(SourceProject project) { 030 this.project = project; 031 } 032 033 public void decorateWith(MetricDef... metrics) { 034 decorateWith(project, metrics); 035 } 036 037 private void decorateWith(SourceCode sourceCode, MetricDef... metrics) { 038 if (sourceCode.hasChildren()) { 039 for (SourceCode child : sourceCode.getChildren()) { 040 decorateWith(child, metrics); 041 } 042 } 043 for (MetricDef metric : metrics) { 044 if ( !metric.aggregateIfThereIsAlreadyAValue() && sourceCode.getDouble(metric) != 0) { 045 continue; 046 } 047 if (sourceCode.hasChildren()) { 048 for (SourceCode child : sourceCode.getChildren()) { 049 if ( !metric.isCalculatedMetric() && metric.isThereAggregationFormula()) { 050 sourceCode.add(metric, child); 051 } 052 } 053 } 054 } 055 } 056 }