com.kryshchuk.maven.plugins.jfreechart.LineChartDataset Maven / Gradle / Ivy

Go to download
/**
 * 
 */
package com.kryshchuk.maven.plugins.jfreechart;

import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.apache.maven.plugins.annotations.Parameter;
import org.jfree.data.xy.XYSeriesCollection;

/**
 * LineChartDataset class.
 * 
 * @author yura
 */
public class LineChartDataset {

  /**
   * The regular expression for data pattern.
   */
  @Parameter(required = true)
  private String regexp;

  private Pattern pattern;

  @Parameter(required = true)
  private LineChartAxis axisDomain;

  @Parameter(required = true)
  private LineChartAxis axisRange;

  @Parameter(required = true)
  private List series;

  private final XYSeriesCollection seriesCollection = new XYSeriesCollection();

  private int sequence;

  /**
   * @param regexp
   *          the regexp to set
   */
  public void setRegexp(final String regexp) {
    this.regexp = regexp;
    pattern = Pattern.compile(regexp);
  }

  public Matcher getMatcher(final String line) {
    return pattern.matcher(line);
  }

  public Number getDomainValue(final LineChartAxis defaultDomainAxis, final Matcher m) {
    final LineChartAxis domainAxis = getDomainAxis() == null ? defaultDomainAxis : getDomainAxis();
    if (domainAxis.isSequence()) {
      return Integer.valueOf(++sequence);
    } else {
      return domainAxis.getValue(m);
    }
  }

  /**
   * @return the axisX
   */
  public LineChartAxis getDomainAxis() {
    return axisDomain;
  }

  /**
   * @return the axisY
   */
  public LineChartAxis getRangeAxis() {
    return axisRange;
  }

  /**
   * @param series
   *          the series to set
   */
  public void setSeries(final List series) {
    this.series = series;
    for (final LineChartSerie serie : series) {
      seriesCollection.addSeries(serie.getSerie());
    }
  }

  /**
   * @return the series
   */
  public List getSeries() {
    return series;
  }

  /**
   * @return the seriesCollection
   */
  public XYSeriesCollection getSeriesCollection() {
    return seriesCollection;
  }

  public void clear() {
    for (final LineChartSerie serie : getSeries()) {
      serie.clear();
    }
    sequence = 0;
  }

  @Override
  public String toString() {
    final StringBuilder str = new StringBuilder("Line Chart Dataset: ");
    str.append("regexp={").append(regexp).append("} ");
    return str.toString();
  }

}