package org.apache.maven.plugin.checkstyle;
   
   /*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you under the Apache License, Version 2.0 (the
    * "License"); you may not use this file except in compliance
   * with the License.  You may obtain a copy of the License at
   *
   *   http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing,
   * software distributed under the License is distributed on an
   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
   * KIND, either express or implied.  See the License for the
   * specific language governing permissions and limitations
   * under the License.
   */
  
  import com.puppycrawl.tools.checkstyle.api.AuditEvent;
  import com.puppycrawl.tools.checkstyle.api.Configuration;
  import com.puppycrawl.tools.checkstyle.api.SeverityLevel;
  
  import java.util.HashMap;
  import java.util.LinkedList;
  import java.util.List;
  import java.util.Map;
  
  /**
   * Object holding the references to the CheckstyleResults.
   *
   * @author <a href="mailto:joakim@erdfelt.net">Joakim Erdfelt</a>
   * @version $Id: CheckstyleResults.html 816663 2012-05-08 14:00:13Z hboutemy $
   * @todo provide fallback to disk based storage if too many results.
   */
  public class CheckstyleResults
  {
      private Map<String, List<AuditEvent>> files;
  
      private Configuration configuration;
  
      public CheckstyleResults()
      {
          files = new HashMap<String, List<AuditEvent>>();
      }
  
      public List<AuditEvent> getFileViolations( String file )
      {
          List<AuditEvent> violations;
  
          if ( this.files.containsKey( file ) )
          {
              violations = this.files.get( file );
          }
          else
          {
              violations = new LinkedList<AuditEvent>();
              this.files.put( file, violations );
          }
  
          return violations;
      }
  
      public void setFileViolations( String file, List<AuditEvent> violations )
      {
          this.files.put( file, violations );
      }
  
      public Map<String, List<AuditEvent>> getFiles()
      {
          return files;
      }
  
      public void setFiles( Map<String, List<AuditEvent>> files )
      {
          this.files = files;
      }
  
      public int getFileCount()
      {
          return this.files.size();
      }
  
      public long getSeverityCount( SeverityLevel level )
      {
          long count = 0;
  
          for ( List<AuditEvent> errors : this.files.values() )
          {
              count = count + getSeverityCount( errors, level );
          }
  
          return count;
      }
  
      public long getSeverityCount( String file, SeverityLevel level )
      {
         long count = 0;
 
         if ( !this.files.containsKey( file ) )
         {
             return count;
         }
 
         List<AuditEvent> violations = this.files.get( file );
 
         count = getSeverityCount( violations, level );
 
         return count;
     }
 
     public long getSeverityCount( List<AuditEvent> violations, SeverityLevel level )
     {
         long count = 0;
 
         for ( AuditEvent event : violations )
         {
             if ( event.getSeverityLevel().equals( level ) )
             {
                 count++;
             }
         }
 
         return count;
     }
 
     public Configuration getConfiguration()
     {
         return configuration;
     }
 
     public void setConfiguration( Configuration configuration )
     {
         this.configuration = configuration;
     }
 }