HRMS/hrmsEjb/net/sf/jasperreports/charts/fill/JRFillTimeSeriesDataset.java

package net.sf.jasperreports.charts.fill;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import net.sf.jasperreports.charts.JRTimeSeries;
import net.sf.jasperreports.charts.JRTimeSeriesDataset;
import net.sf.jasperreports.charts.util.TimeSeriesLabelGenerator;
import net.sf.jasperreports.engine.JRChartDataset;
import net.sf.jasperreports.engine.JRExpressionCollector;
import net.sf.jasperreports.engine.JRRuntimeException;
import net.sf.jasperreports.engine.design.JRVerifier;
import net.sf.jasperreports.engine.fill.JRCalculator;
import net.sf.jasperreports.engine.fill.JRExpressionEvalException;
import net.sf.jasperreports.engine.fill.JRFillChartDataset;
import net.sf.jasperreports.engine.fill.JRFillObjectFactory;
import org.jfree.data.general.Dataset;
import org.jfree.data.time.RegularTimePeriod;
import org.jfree.data.time.TimeSeries;
import org.jfree.data.time.TimeSeriesCollection;

public class JRFillTimeSeriesDataset extends JRFillChartDataset implements JRTimeSeriesDataset {
  protected JRFillTimeSeries[] timeSeries = null;
  
  private List seriesNames = null;
  
  private Map seriesMap = null;
  
  private Map labelsMap = null;
  
  private Map itemHyperlinks;
  
  public JRFillTimeSeriesDataset(JRTimeSeriesDataset timeSeriesDataset, JRFillObjectFactory factory) {
    super((JRChartDataset)timeSeriesDataset, factory);
    JRTimeSeries[] srcTimeSeries = timeSeriesDataset.getSeries();
    if (srcTimeSeries != null && srcTimeSeries.length > 0) {
      this.timeSeries = new JRFillTimeSeries[srcTimeSeries.length];
      for (int i = 0; i < this.timeSeries.length; i++)
        this.timeSeries[i] = (JRFillTimeSeries)factory.getTimeSeries(srcTimeSeries[i]); 
    } 
  }
  
  public JRTimeSeries[] getSeries() {
    return (JRTimeSeries[])this.timeSeries;
  }
  
  protected void customInitialize() {
    this.seriesNames = null;
    this.seriesMap = null;
    this.labelsMap = null;
    this.itemHyperlinks = null;
  }
  
  protected void customEvaluate(JRCalculator calculator) throws JRExpressionEvalException {
    if (this.timeSeries != null && this.timeSeries.length > 0)
      for (int i = 0; i < this.timeSeries.length; i++)
        this.timeSeries[i].evaluate(calculator);  
  }
  
  protected void customIncrement() {
    if (this.timeSeries != null && this.timeSeries.length > 0) {
      if (this.seriesNames == null) {
        this.seriesNames = new ArrayList();
        this.seriesMap = new HashMap();
        this.labelsMap = new HashMap();
        this.itemHyperlinks = new HashMap();
      } 
      for (int i = 0; i < this.timeSeries.length; i++) {
        JRFillTimeSeries crtTimeSeries = this.timeSeries[i];
        Comparable seriesName = crtTimeSeries.getSeries();
        if (seriesName == null)
          throw new JRRuntimeException("Time series name is null."); 
        TimeSeries series = (TimeSeries)this.seriesMap.get(seriesName);
        if (series == null) {
          series = new TimeSeries(seriesName.toString(), getTimePeriod());
          this.seriesNames.add(seriesName);
          this.seriesMap.put(seriesName, series);
        } 
        RegularTimePeriod tp = RegularTimePeriod.createInstance(getTimePeriod(), crtTimeSeries.getTimePeriod(), getTimeZone());
        series.addOrUpdate(tp, crtTimeSeries.getValue());
        if (crtTimeSeries.getLabelExpression() != null) {
          Map seriesLabels = (Map)this.labelsMap.get(seriesName);
          if (seriesLabels == null) {
            seriesLabels = new HashMap();
            this.labelsMap.put(seriesName, seriesLabels);
          } 
          seriesLabels.put(tp, crtTimeSeries.getLabel());
        } 
        if (crtTimeSeries.hasItemHyperlink()) {
          Map seriesLinks = (Map)this.itemHyperlinks.get(seriesName);
          if (seriesLinks == null) {
            seriesLinks = new HashMap();
            this.itemHyperlinks.put(seriesName, seriesLinks);
          } 
          seriesLinks.put(tp, crtTimeSeries.getPrintItemHyperlink());
        } 
      } 
    } 
  }
  
  public Dataset getCustomDataset() {
    TimeSeriesCollection dataset = new TimeSeriesCollection();
    if (this.seriesNames != null)
      for (int i = 0; i < this.seriesNames.size(); i++) {
        Comparable seriesName = this.seriesNames.get(i);
        dataset.addSeries((TimeSeries)this.seriesMap.get(seriesName));
      }  
    return (Dataset)dataset;
  }
  
  public Class getTimePeriod() {
    return ((JRTimeSeriesDataset)this.parent).getTimePeriod();
  }
  
  public void setTimePeriod(Class timePeriod) {}
  
  public byte getDatasetType() {
    return 6;
  }
  
  public TimeSeriesLabelGenerator getLabelGenerator() {
    return new TimeSeriesLabelGenerator(this.labelsMap);
  }
  
  public void collectExpressions(JRExpressionCollector collector) {
    collector.collect(this);
  }
  
  public Map getItemHyperlinks() {
    return this.itemHyperlinks;
  }
  
  public boolean hasItemHyperlinks() {
    boolean foundLinks = false;
    if (this.timeSeries != null && this.timeSeries.length > 0)
      for (int i = 0; i < this.timeSeries.length && !foundLinks; i++)
        foundLinks = this.timeSeries[i].hasItemHyperlink();  
    return foundLinks;
  }
  
  public void validate(JRVerifier verifier) {
    verifier.verify(this);
  }
}