ingo@2233: package de.intevation.flys.exports;
ingo@2233: 
ingo@2238: import java.awt.Color;
ingo@2586: import java.awt.Font;
ingo@2238: 
ingo@2238: import java.util.ArrayList;
ingo@2238: import java.util.HashMap;
ingo@2238: import java.util.List;
ingo@2238: import java.util.Map;
ingo@2233: 
ingo@2233: import org.apache.log4j.Logger;
ingo@2233: 
ingo@2238: import org.jfree.chart.ChartFactory;
ingo@2233: import org.jfree.chart.JFreeChart;
ingo@2400: import org.jfree.chart.axis.ValueAxis;
ingo@2238: import org.jfree.chart.plot.XYPlot;
ingo@2238: 
ingo@2238: import org.jfree.data.Range;
ingo@2238: import org.jfree.data.time.TimeSeriesCollection;
ingo@2242: import org.jfree.data.general.Series;
ingo@2238: import org.jfree.data.xy.XYDataset;
ingo@2233: 
ingo@2400: import de.intevation.flys.jfree.Bounds;
ingo@2400: import de.intevation.flys.jfree.DoubleBounds;
ingo@2400: import de.intevation.flys.jfree.TimeBounds;
ingo@2400: 
ingo@2233: 
ingo@2233: /**
ingo@2233:  * @author <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a>
ingo@2233:  */
ingo@2233: public abstract class TimeseriesChartGenerator extends ChartGenerator {
ingo@2233: 
ingo@2238: 
ingo@2238:     /**
ingo@2238:      * Inner class TimeseriesAxisDataset stores TimeSeriesCollection.
ingo@2238:      */
ingo@2238:     public class TimeseriesAxisDataset implements AxisDataset {
ingo@2238: 
ingo@2238:         protected int axisSymbol;
ingo@2238: 
ingo@2238:         protected List<TimeSeriesCollection> datasets;
ingo@2238: 
ingo@2238:         protected Range range;
ingo@2238: 
ingo@2238:         protected int plotAxisIndex;
ingo@2238: 
ingo@2238: 
ingo@2238:         public TimeseriesAxisDataset(int axisSymbol) {
ingo@2238:             this.axisSymbol = axisSymbol;
ingo@2238:             this.datasets   = new ArrayList<TimeSeriesCollection>();
ingo@2238:         }
ingo@2238: 
ingo@2238: 
ingo@2238:         @Override
ingo@2238:         public void addDataset(XYDataset dataset) {
ingo@2238:             if (!(dataset instanceof TimeSeriesCollection)) {
ingo@2238:                 logger.warn("Skip non TimeSeriesCollection dataset.");
ingo@2238:                 return;
ingo@2238:             }
ingo@2238: 
ingo@2238:             TimeSeriesCollection tsc = (TimeSeriesCollection) dataset;
ingo@2238: 
ingo@2238:             datasets.add(tsc);
ingo@2238:             mergeRanges(tsc);
ingo@2238:         }
ingo@2238: 
ingo@2238: 
ingo@2238:         @Override
ingo@2242:         public XYDataset[] getDatasets() {
ingo@2242:             return (XYDataset[])
ingo@2242:                 datasets.toArray(new XYDataset[datasets.size()]);
ingo@2242:         }
ingo@2242: 
ingo@2242: 
ingo@2242:         @Override
ingo@2238:         public boolean isEmpty() {
ingo@2238:             return datasets.isEmpty();
ingo@2238:         }
ingo@2238: 
ingo@2238: 
ingo@2242:         @Override
ingo@2242:         public void setRange(Range range) {
ingo@2242:             this.range = range;
ingo@2242:         }
ingo@2242: 
ingo@2242: 
ingo@2242:         @Override
ingo@2242:         public Range getRange() {
ingo@2242:             return range;
ingo@2242:         }
ingo@2242: 
ingo@2242: 
ingo@2242:         @Override
ingo@2242:         public void setPlotAxisIndex(int plotAxisIndex) {
ingo@2242:             this.plotAxisIndex = plotAxisIndex;
ingo@2242:         }
ingo@2242: 
ingo@2242: 
ingo@2242:         @Override
ingo@2242:         public int getPlotAxisIndex() {
ingo@2242:             return plotAxisIndex;
ingo@2242:         }
ingo@2242: 
ingo@2242: 
ingo@2242:         @Override
ingo@2242:         public boolean isArea(XYDataset dataset) {
ingo@2242:             logger.warn("This AxisDataset doesn't support Areas yet!");
ingo@2242:             return false;
ingo@2242:         }
ingo@2242: 
ingo@2242: 
ingo@2238:         protected void mergeRanges(TimeSeriesCollection dataset) {
ingo@2238:             logger.debug("Range after merging: " + range);
ingo@2238: 
ingo@2400:             Bounds[] xyRanges = ChartHelper.getBounds(dataset);
ingo@2400: 
ingo@2400:             // TODO COMBINE BOUNDS!
ingo@2238: 
ingo@2238:             logger.debug("Range after merging: " + range);
ingo@2238:         }
ingo@2238: 
ingo@2238:     } // end of TimeseriesAxisDataset class
ingo@2238: 
ingo@2238: 
ingo@2238: 
ingo@2233:     private static final Logger logger =
ingo@2233:         Logger.getLogger(TimeseriesChartGenerator.class);
ingo@2233: 
ingo@2233: 
ingo@2400:     public static final int AXIS_SPACE = 5;
ingo@2238: 
ingo@2400: 
ingo@2587:     protected Map<Integer, Bounds> xBounds;
ingo@2400: 
ingo@2587:     protected Map<Integer, Bounds> yBounds;
ingo@2242: 
ingo@2238: 
ingo@2238: 
ingo@2238:     /**
ingo@2238:      * The default constructor that initializes internal datastructures.
ingo@2238:      */
ingo@2238:     public TimeseriesChartGenerator() {
ingo@2238:         super();
ingo@2238: 
ingo@2587:         xBounds = new HashMap<Integer, Bounds>();
ingo@2587:         yBounds = new HashMap<Integer, Bounds>();
ingo@2238:     }
ingo@2238: 
ingo@2233: 
ingo@2233: 
ingo@2233:     @Override
ingo@2234:     public JFreeChart generateChart() {
ingo@2233:         logger.info("Generate Timeseries Chart.");
ingo@2233: 
ingo@2238:         JFreeChart chart = ChartFactory.createTimeSeriesChart(
ingo@2238:             getChartTitle(),
ingo@2238:             getXAxisLabel(),
ingo@2238:             getYAxisLabel(0),
ingo@2238:             null,
ingo@2554:             isLegendVisible(),
ingo@2238:             false,
ingo@2238:             false);
ingo@2238: 
ingo@2238:         XYPlot plot = (XYPlot) chart.getPlot();
ingo@2238: 
ingo@2238:         chart.setBackgroundPaint(Color.WHITE);
ingo@2238:         plot.setBackgroundPaint(Color.WHITE);
ingo@2238: 
ingo@2238:         addSubtitles(chart);
ingo@2553:         adjustPlot(plot);
ingo@2238:         addDatasets(plot);
ingo@2586:         adjustAxes(plot);
ingo@2238: 
ingo@2400:         adaptZoom(plot);
ingo@2400: 
ingo@2238:         return chart;
ingo@2238:     }
ingo@2238: 
ingo@2238: 
ingo@2242:     @Override
ingo@2242:     protected Series getSeriesOf(XYDataset dataset, int idx) {
ingo@2242:         return ((TimeSeriesCollection) dataset).getSeries(idx);
ingo@2242:     }
ingo@2242: 
ingo@2242: 
ingo@2238:     /**
ingo@2238:      * This method creates new instances of TimeseriesAxisDataset.
ingo@2238:      *
ingo@2238:      * @param idx The symbol for the new TimeseriesAxisDataset.
ingo@2238:      */
ingo@2238:     @Override
ingo@2238:     protected AxisDataset createAxisDataset(int idx) {
ingo@2238:         logger.debug("Create a new AxisDataset for index: " + idx);
ingo@2238:         return new TimeseriesAxisDataset(idx);
ingo@2238:     }
ingo@2238: 
ingo@2238: 
ingo@2400:     @Override
ingo@2587:     protected void combineXBounds(Bounds bounds, int index) {
ingo@2400:         if (bounds != null) {
ingo@2587:             Bounds old = getXBounds(index);
ingo@2238: 
ingo@2238:             if (old != null) {
ingo@2400:                 bounds = bounds.combine(old);
ingo@2238:             }
ingo@2238: 
ingo@2400:             setXBounds(index, bounds);
ingo@2238:         }
ingo@2238:     }
ingo@2261: 
ingo@2261: 
ingo@2587:     @Override
ingo@2587:     protected void combineYBounds(Bounds bounds, int index) {
ingo@2400:         if (bounds != null) {
ingo@2400:             Bounds old = getYBounds(index);
ingo@2400: 
ingo@2400:             if (old != null) {
ingo@2400:                 bounds = bounds.combine(old);
ingo@2400:             }
ingo@2400: 
ingo@2400:             setYBounds(index, bounds);
ingo@2400:         }
ingo@2400:     }
ingo@2400: 
ingo@2400: 
ingo@2400:     // TODO REPLACE THIS METHOD WITH getBoundsForAxis(index)
ingo@2400:     @Override
ingo@2400:     public Range[] getRangesForAxis(int index) {
ingo@2400:         // TODO
ingo@2400:         Bounds[] bounds = getBoundsForAxis(index);
ingo@2400: 
ingo@2400:         return new Range[] {
ingo@2400:             new Range(
ingo@2400:                 bounds[0].getLower().doubleValue(),
ingo@2400:                 bounds[0].getUpper().doubleValue()),
ingo@2400:             new Range(
ingo@2400:                 bounds[1].getLower().doubleValue(),
ingo@2400:                 bounds[1].getUpper().doubleValue())
ingo@2400:         };
ingo@2400:     }
ingo@2400: 
ingo@2400: 
ingo@2261:     @Override
ingo@2400:     public Bounds getXBounds(int axis) {
ingo@2587:         return xBounds.get(axis);
ingo@2400:     }
ingo@2261: 
ingo@2400: 
ingo@2400:     @Override
ingo@2400:     protected void setXBounds(int axis, Bounds bounds) {
ingo@2587:         xBounds.put(axis, bounds);
ingo@2400:     }
ingo@2400: 
ingo@2400: 
ingo@2400:     @Override
ingo@2400:     public Bounds getYBounds(int axis) {
ingo@2587:         return yBounds.get(axis);
ingo@2400:     }
ingo@2400: 
ingo@2400: 
ingo@2400:     @Override
ingo@2400:     protected void setYBounds(int axis, Bounds bounds) {
ingo@2587:         if (bounds != null) {
ingo@2587:             yBounds.put(axis, bounds);
ingo@2587:         }
ingo@2400:     }
ingo@2400: 
ingo@2400: 
ingo@2400:     public Bounds[] getBoundsForAxis(int index) {
ingo@2400:         logger.debug("Return x and y bounds for axis at: " + index);
ingo@2400: 
ingo@2400:         Bounds rx = getXBounds(Integer.valueOf(index));
ingo@2400:         Bounds ry = getYBounds(Integer.valueOf(index));
ingo@2261: 
ingo@2261:         if (rx == null) {
ingo@2261:             logger.warn("Range for x axis not set." +
ingo@2261:                         " Using default values: 0 - 1.");
ingo@2400:             rx = new TimeBounds(0l, 1l);
ingo@2261:         }
ingo@2261: 
ingo@2400:         if (ry == null) {
ingo@2400:             logger.warn("Range for y axis not set." +
ingo@2400:                         " Using default values: 0 - 1.");
ingo@2400:             ry = new DoubleBounds(0l, 1l);
ingo@2400:         }
ingo@2261: 
ingo@2400:         logger.debug("X Bounds at index " + index + " is: " + rx);
ingo@2400:         logger.debug("Y Bounds at index " + index + " is: " + ry);
ingo@2400: 
ingo@2400:         return new Bounds[] {rx, ry};
ingo@2400:     }
ingo@2400: 
ingo@2400: 
ingo@2400:     public Bounds getDomainAxisRange() {
ingo@2400:         String[] ranges = getDomainAxisRangeFromRequest();
ingo@2400: 
ingo@2400:         if (ranges == null || ranges.length < 2) {
ingo@2400:             logger.debug("No zoom range for domain axis specified.");
ingo@2400:             return null;
ingo@2400:         }
ingo@2400: 
ingo@2400:         if (ranges[0] == null || ranges[1] == null) {
ingo@2400:             logger.warn("Invalid ranges for domain axis specified!");
ingo@2400:             return null;
ingo@2400:         }
ingo@2400: 
ingo@2400:         try {
ingo@2400:             double lower = Double.parseDouble(ranges[0]);
ingo@2400:             double upper = Double.parseDouble(ranges[1]);
ingo@2400: 
ingo@2400:             return new DoubleBounds(lower, upper);
ingo@2400:         }
ingo@2400:         catch (NumberFormatException nfe) {
ingo@2400:             logger.warn("Invalid ranges for domain axis specified: " + nfe);
ingo@2400:         }
ingo@2400: 
ingo@2400:         return null;
ingo@2400:     }
ingo@2400: 
ingo@2400: 
ingo@2400:     public Bounds getValueAxisRange() {
ingo@2400:         String[] ranges = getValueAxisRangeFromRequest();
ingo@2400: 
ingo@2400:         if (ranges == null || ranges.length < 2) {
ingo@2400:             logger.debug("No zoom range for domain axis specified.");
ingo@2400:             return null;
ingo@2400:         }
ingo@2400: 
ingo@2400:         if (ranges[0] == null || ranges[1] == null) {
ingo@2400:             logger.warn("Invalid ranges for domain axis specified!");
ingo@2400:             return null;
ingo@2400:         }
ingo@2400: 
ingo@2400:         try {
ingo@2400:             double lower = Double.parseDouble(ranges[0]);
ingo@2400:             double upper = Double.parseDouble(ranges[1]);
ingo@2400: 
ingo@2400:             return new DoubleBounds(lower, upper);
ingo@2400:         }
ingo@2400:         catch (NumberFormatException nfe) {
ingo@2400:             logger.warn("Invalid ranges for domain axis specified: " + nfe);
ingo@2400:         }
ingo@2400: 
ingo@2400:         return null;
ingo@2400:     }
ingo@2400: 
ingo@2400: 
ingo@2400:     protected void adaptZoom(XYPlot plot) {
ingo@2400:         logger.debug("Adapt zoom of Timeseries chart.");
ingo@2400: 
ingo@2587:         zoomX(plot, plot.getDomainAxis(), getXBounds(0), getDomainAxisRange());
ingo@2400: 
ingo@2400:         Bounds valueAxisBounds = getValueAxisRange();
ingo@2400: 
ingo@2400:         for (int j = 0, n = plot.getRangeAxisCount(); j < n; j++) {
ingo@2400:             zoomY(
ingo@2400:                 plot,
ingo@2400:                 plot.getRangeAxis(j),
ingo@2400:                 getYBounds(j),
ingo@2400:                 valueAxisBounds);
ingo@2400:         }
ingo@2400:     }
ingo@2400: 
ingo@2400: 
ingo@2400:     protected void zoomX(
ingo@2400:         XYPlot    plot,
ingo@2400:         ValueAxis axis,
ingo@2400:         Bounds    total,
ingo@2400:         Bounds    user
ingo@2400:     ) {
ingo@2400:         if (logger.isDebugEnabled()) {
ingo@2400:             logger.debug("== Zoom X axis ==");
ingo@2400:             logger.debug("    Total axis range  : " + total);
ingo@2400:             logger.debug("    User defined range: " + user);
ingo@2400:         }
ingo@2400: 
ingo@2400:         if (user != null) {
ingo@2400:             long min  = total.getLower().longValue();
ingo@2400:             long max  = total.getUpper().longValue();
ingo@2400:             long diff = max > min ? max - min : min - max;
ingo@2400: 
ingo@2400:             long newMin = (long) Math.round(min + user.getLower().doubleValue() * diff);
ingo@2400:             long newMax = (long) Math.round(min + user.getUpper().doubleValue() * diff);
ingo@2400: 
ingo@2400:             TimeBounds newBounds = new TimeBounds(newMin, newMax);
ingo@2400: 
ingo@2400:             logger.debug("    Zoom axis to: " + newBounds);
ingo@2400: 
ingo@2400:             newBounds.applyBounds(axis, AXIS_SPACE);
ingo@2400:         }
ingo@2400:         else {
ingo@2400:             logger.debug("No user specified zoom values found!");
ingo@2400:             total.applyBounds(axis, AXIS_SPACE);
ingo@2400:         }
ingo@2400:     }
ingo@2400: 
ingo@2400: 
ingo@2400:     protected void zoomY(
ingo@2400:         XYPlot    plot,
ingo@2400:         ValueAxis axis,
ingo@2400:         Bounds    total,
ingo@2400:         Bounds    user
ingo@2400:     ) {
ingo@2400:         if (logger.isDebugEnabled()) {
ingo@2400:             logger.debug("== Zoom Y axis ==");
ingo@2400:             logger.debug("    Total axis range  : " + total);
ingo@2400:             logger.debug("    User defined range: " + user);
ingo@2400:         }
ingo@2400: 
ingo@2400:         if (user != null) {
ingo@2400:             double min  = total.getLower().doubleValue();
ingo@2400:             double max  = total.getUpper().doubleValue();
ingo@2400:             double diff = max > min ? max - min : min - max;
ingo@2400: 
ingo@2400:             double newMin = min + user.getLower().doubleValue() * diff;
ingo@2400:             double newMax = min + user.getUpper().doubleValue() * diff;
ingo@2400: 
ingo@2400:             DoubleBounds newBounds = new DoubleBounds(newMin, newMax);
ingo@2400: 
ingo@2400:             logger.debug("    Zoom axis to: " + newBounds);
ingo@2400: 
ingo@2400:             newBounds.applyBounds(axis, AXIS_SPACE);
ingo@2400:         }
ingo@2400:         else {
ingo@2400:             logger.debug("No user specified zoom values found!");
ingo@2400:             total.applyBounds(axis, AXIS_SPACE);
ingo@2400:         }
ingo@2261:     }
ingo@2586: 
ingo@2586: 
ingo@2586:     /**
ingo@2586:      * Adjusts the axes of a plot. This method sets the <i>labelFont</i> of the
ingo@2586:      * X axis.
ingo@2586:      *
ingo@2586:      * @param plot The XYPlot of the chart.
ingo@2586:      */
ingo@2586:     protected void adjustAxes(XYPlot plot) {
ingo@2586:         ValueAxis xaxis = plot.getDomainAxis();
ingo@2586: 
ingo@2586:         ChartSettings chartSettings = getChartSettings();
ingo@2586:         if (chartSettings == null) {
ingo@2586:             return;
ingo@2586:         }
ingo@2586: 
ingo@2586:         Font labelFont = new Font(
ingo@2586:             DEFAULT_FONT_NAME,
ingo@2586:             Font.BOLD,
ingo@2586:             getXAxisLabelFontSize());
ingo@2586: 
ingo@2586:         xaxis.setLabelFont(labelFont);
ingo@2590:         xaxis.setTickLabelFont(labelFont);
ingo@2586:     }
ingo@2233: }
ingo@2233: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :