first commit

hrmsEjb/net/sf/jasperreports/charts/fill/JRFillTimePeriodDataset.java

@@ -0,0 +1,140 @@
+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.JRTimePeriodDataset;
+import net.sf.jasperreports.charts.JRTimePeriodSeries;
+import net.sf.jasperreports.charts.util.TimePeriodDatasetLabelGenerator;
+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.SimpleTimePeriod;
+import org.jfree.data.time.TimePeriod;
+import org.jfree.data.time.TimePeriodValues;
+import org.jfree.data.time.TimePeriodValuesCollection;
+
+public class JRFillTimePeriodDataset extends JRFillChartDataset implements JRTimePeriodDataset {
+  protected JRFillTimePeriodSeries[] timePeriodSeries = null;
+  
+  private List seriesNames = null;
+  
+  private Map seriesMap = null;
+  
+  private Map labelsMap = null;
+  
+  private Map itemHyperlinks;
+  
+  public JRFillTimePeriodDataset(JRTimePeriodDataset timePeriodDataset, JRFillObjectFactory factory) {
+    super((JRChartDataset)timePeriodDataset, factory);
+    JRTimePeriodSeries[] srcTimePeriodSeries = timePeriodDataset.getSeries();
+    if (srcTimePeriodSeries != null && srcTimePeriodSeries.length > 0) {
+      this.timePeriodSeries = new JRFillTimePeriodSeries[srcTimePeriodSeries.length];
+      for (int i = 0; i < this.timePeriodSeries.length; i++)
+        this.timePeriodSeries[i] = (JRFillTimePeriodSeries)factory.getTimePeriodSeries(srcTimePeriodSeries[i]); 
+    } 
+  }
+  
+  public JRTimePeriodSeries[] getSeries() {
+    return (JRTimePeriodSeries[])this.timePeriodSeries;
+  }
+  
+  protected void customInitialize() {
+    this.seriesNames = null;
+    this.seriesMap = null;
+    this.labelsMap = null;
+    this.itemHyperlinks = null;
+  }
+  
+  protected void customEvaluate(JRCalculator calculator) throws JRExpressionEvalException {
+    if (this.timePeriodSeries != null && this.timePeriodSeries.length > 0)
+      for (int i = 0; i < this.timePeriodSeries.length; i++)
+        this.timePeriodSeries[i].evaluate(calculator);  
+  }
+  
+  protected void customIncrement() {
+    if (this.timePeriodSeries != null && this.timePeriodSeries.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.timePeriodSeries.length; i++) {
+        JRFillTimePeriodSeries crtTimePeriodSeries = this.timePeriodSeries[i];
+        Comparable seriesName = crtTimePeriodSeries.getSeries();
+        if (seriesName == null)
+          throw new JRRuntimeException("Time period series name is null."); 
+        TimePeriodValues timePeriodValues = (TimePeriodValues)this.seriesMap.get(seriesName);
+        if (timePeriodValues == null) {
+          timePeriodValues = new TimePeriodValues(seriesName.toString());
+          this.seriesNames.add(seriesName);
+          this.seriesMap.put(seriesName, timePeriodValues);
+        } 
+        SimpleTimePeriod stp = new SimpleTimePeriod(crtTimePeriodSeries.getStartDate(), crtTimePeriodSeries.getEndDate());
+        timePeriodValues.add((TimePeriod)stp, crtTimePeriodSeries.getValue());
+        if (crtTimePeriodSeries.getLabelExpression() != null) {
+          Map seriesLabels = (Map)this.labelsMap.get(seriesName);
+          if (seriesLabels == null) {
+            seriesLabels = new HashMap();
+            this.labelsMap.put(seriesName, seriesLabels);
+          } 
+          seriesLabels.put(stp, crtTimePeriodSeries.getLabel());
+        } 
+        if (crtTimePeriodSeries.hasItemHyperlink()) {
+          Map seriesLinks = (Map)this.itemHyperlinks.get(seriesName);
+          if (seriesLinks == null) {
+            seriesLinks = new HashMap();
+            this.itemHyperlinks.put(seriesName, seriesLinks);
+          } 
+          seriesLinks.put(stp, crtTimePeriodSeries.getPrintItemHyperlink());
+        } 
+      } 
+    } 
+  }
+  
+  public Dataset getCustomDataset() {
+    TimePeriodValuesCollection dataset = new TimePeriodValuesCollection();
+    if (this.seriesNames != null)
+      for (int i = 0; i < this.seriesNames.size(); i++) {
+        Comparable seriesName = this.seriesNames.get(i);
+        dataset.addSeries((TimePeriodValues)this.seriesMap.get(seriesName));
+      }  
+    return (Dataset)dataset;
+  }
+  
+  public byte getDatasetType() {
+    return 5;
+  }
+  
+  public TimePeriodDatasetLabelGenerator getLabelGenerator() {
+    return new TimePeriodDatasetLabelGenerator(this.labelsMap);
+  }
+  
+  public void collectExpressions(JRExpressionCollector collector) {
+    collector.collect(this);
+  }
+  
+  public boolean hasItemHyperlinks() {
+    boolean foundLinks = false;
+    if (this.timePeriodSeries != null && this.timePeriodSeries.length > 0)
+      for (int i = 0; i < this.timePeriodSeries.length && !foundLinks; i++)
+        foundLinks = this.timePeriodSeries[i].hasItemHyperlink();  
+    return foundLinks;
+  }
+  
+  public Map getItemHyperlinks() {
+    return this.itemHyperlinks;
+  }
+  
+  public void validate(JRVerifier verifier) {
+    verifier.verify(this);
+  }
+}