package net.sf.statcvs.charts;
   
   import java.awt.Color;
   import java.awt.Dimension;
   import java.text.DecimalFormat;
   import java.util.ArrayList;
   import java.util.Collection;
   import java.util.Collections;
   import java.util.Iterator;
  import java.util.Set;
  
  import net.sf.statcvs.Messages;
  import net.sf.statcvs.model.Author;
  import net.sf.statcvs.model.Commit;
  import net.sf.statcvs.model.Repository;
  import net.sf.statcvs.model.Revision;
  import net.sf.statcvs.output.ReportConfig;
  
  import org.jfree.chart.ChartFactory;
  import org.jfree.chart.JFreeChart;
  import org.jfree.chart.axis.NumberAxis;
  import org.jfree.chart.axis.NumberTickUnit;
  import org.jfree.chart.plot.CategoryPlot;
  import org.jfree.chart.plot.PlotOrientation;
  import org.jfree.chart.renderer.category.CategoryItemRenderer;
  import org.jfree.data.category.DefaultCategoryDataset;
  import org.jfree.ui.RectangleEdge;
  
  /**
   * Class for producing the "Author Activity: Modifying/Adding" chart
   * @author jentzsch
   * @author Richard Cyganiak (richard@cyganiak.de)
   * @version $Id: ModifyAddChartMaker.java,v 1.6 2008/04/02 11:22:15 benoitx Exp $
   */
  public class ModifyAddChartMaker {
      private static final int MODIFYING = 0;
      private static final int ADDING = 1;
  
      private final ReportConfig config;
      private final Repository repository;
      private final int width;
      private double[][] categories;
      private final ArrayList categoryNames = new ArrayList();
  
      /**
       * Creates a new ModifyAddChartMaker
       * @param content Repository
       * @param width width of the chart in pixels
       */
      public ModifyAddChartMaker(final ReportConfig config, final int width) {
          this.config = config;
          this.repository = config.getRepository();
          this.width = width;
      }
  
      public ChartImage toFile() {
          final Collection authors = this.repository.getAuthors();
          final Iterator it = authors.iterator();
          while (it.hasNext()) {
              final Author author = (Author) it.next();
              if (!config.isDeveloper(author)) {
                  continue;
              }
              categoryNames.add(author.getRealName());
          }
          Collections.sort(categoryNames);
  
          categories = new double[2][categoryNames.size()];
          for (int j = 0; j < categoryNames.size(); j++) {
              categories[MODIFYING][j] = 0;
              categories[ADDING][j] = 0;
          }
  
          final Iterator commitIt = this.repository.getCommits().iterator();
          while (commitIt.hasNext()) {
              final Commit commit = (Commit) commitIt.next();
              final Set commitRevList = commit.getRevisions();
              final Iterator commitRevIt = commitRevList.iterator();
              final String authorName = commit.getAuthor().getRealName();
              if (authorName == null) {
                  continue;
              }
              final int author = categoryNames.indexOf(authorName);
              if (author == -1) {
                  continue;
              }
              int linesAdded = 0;
              int linesRemoved = 0;
              while (commitRevIt.hasNext()) {
                  final Revision revision = (Revision) commitRevIt.next();
                  if (revision.getLinesDelta() > 0) {
                      linesAdded += revision.getLinesDelta() + revision.getReplacedLines();
                      linesRemoved += revision.getReplacedLines();
                  } else {
                      linesAdded += revision.getReplacedLines();
                      linesRemoved += -revision.getLinesDelta() + revision.getReplacedLines();
                  }
              }
              if (linesAdded == linesRemoved) {
                 categories[MODIFYING][author] += linesAdded;
             }
             if (linesAdded < linesRemoved) {
                 categories[MODIFYING][author] += linesRemoved;
             }
             if (linesAdded > linesRemoved) {
                 categories[ADDING][author] += linesAdded - linesRemoved;
                 categories[MODIFYING][author] += linesRemoved;
             }
         }
 
         for (int i = 0; i < categoryNames.size(); i++) {
             final double maxLines = categories[MODIFYING][i] + categories[ADDING][i];
             for (int k = 0; k < 2; k++) {
                 categories[k][i] *= (100 / maxLines);
             }
         }
 
         final DefaultCategoryDataset data = new DefaultCategoryDataset();
         for (int i = 0; i < categories[MODIFYING].length; i++) {
             data.addValue(categories[MODIFYING][i], "modifying", (Comparable) categoryNames.get(i));
         }
         for (int j = 0; j < categories[ADDING].length; j++) {
             data.addValue(categories[ADDING][j], "adding", (Comparable) categoryNames.get(j));
         }
         //data.setSeriesName(MODIFYING, "modifying");
         //data.setSeriesName(ADDING, "adding");
         //data.setCategories(categoryNames.toArray());
 
         final JFreeChart chart = ChartFactory.createStackedBarChart(this.config.getProjectName() + ": " + Messages.getString("AUTHOR_ACTIVITY_TITLE"), "", "%",
                 data, PlotOrientation.HORIZONTAL, true, false, false);
 
         final CategoryPlot plot = chart.getCategoryPlot();
         //plot.setSeriesPaint(new Paint[] { Color.yellow, Color.green });
         final CategoryItemRenderer renderer = plot.getRenderer();
         renderer.setSeriesPaint(0, Color.yellow);
         renderer.setSeriesPaint(1, Color.green);
 
         final NumberAxis rangeAxis = (NumberAxis) plot.getRangeAxis();
         rangeAxis.setTickUnit(new NumberTickUnit(20.0, new DecimalFormat("0")));
         rangeAxis.setUpperBound(100.0);
 
         chart.getLegend().setPosition(RectangleEdge.TOP);
 
         return this.config.createChartImage("activity.png", Messages.getString("AUTHOR_ACTIVITY_TITLE"), chart, new Dimension(width, 19 * this.categoryNames
                 .size() + 110));
     }
 }