com.od.jtimeseries.ui.visualizer.chart.creator.AbstractXYChartCreator.java Source code

Introduction

Here is the source code for com.od.jtimeseries.ui.visualizer.chart.creator.AbstractXYChartCreator.java

Source

/**
 * Copyright (C) 2011 (nick @ objectdefinitions.com)
 *
 * This file is part of JTimeseries.
 *
 * JTimeseries is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * JTimeseries is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with JTimeseries.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.od.jtimeseries.ui.visualizer.chart.creator;

import com.od.jtimeseries.chart.MovingWindowXYDataset;
import com.od.jtimeseries.ui.config.ChartRangeMode;
import com.od.jtimeseries.ui.config.DomainTimeSelection;
import com.od.jtimeseries.ui.timeseries.ChartingTimeSeries;
import com.od.jtimeseries.util.time.Time;
import com.od.jtimeseries.util.time.TimeSource;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.axis.AxisLocation;
import org.jfree.chart.axis.DateAxis;
import org.jfree.chart.axis.NumberAxis;
import org.jfree.chart.plot.XYPlot;
import org.jfree.chart.renderer.xy.XYItemRenderer;
import org.jfree.data.xy.XYDataset;
import org.jfree.ui.RectangleInsets;

import java.awt.*;
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

/**
 * Created by IntelliJ IDEA.
 * User: nick
 * Date: 27-Feb-2010
 * Time: 11:26:46
 */
public abstract class AbstractXYChartCreator {

    private XYPlot plot;
    private ChartRangeMode chartRangeMode;
    private DomainTimeSelection domainSelection;
    private boolean showLegend;
    private String title;
    private Color chartBackgroundColor;
    private DateFormat dateFormat = new SimpleDateFormat("MMMdd HH:mm");
    private java.util.List<ChartingTimeSeries> timeSeriesList = Collections.EMPTY_LIST;
    private ChartDataFilter chartDataFilter;

    public AbstractXYChartCreator(ChartCreatorParameters p) {
        this.chartBackgroundColor = p.getChartBackgroundColor();
        this.timeSeriesList = p.getTimeSeriesList();
        this.chartRangeMode = p.getChartRangeMode();
        this.domainSelection = p.getDomainSelection();
        this.showLegend = p.isShowLegend();
        this.title = p.getTitle();
        this.chartDataFilter = p.getChartDataFilter();
    }

    public JFreeChart createNewChart() {
        JFreeChart chart = createChart();
        plot = (XYPlot) chart.getPlot();
        plot.setBackgroundPaint(chartBackgroundColor);
        plot.setAxisOffset(new RectangleInsets(5, 5, 5, 5));
        addSeries();
        return chart;
    }

    private JFreeChart createChart() {
        final JFreeChart chart = buildChart();
        chart.getXYPlot().setDomainAxis(new DateAxis());
        DateAxis dateAxis = (DateAxis) chart.getXYPlot().getDomainAxis();
        dateAxis.setDateFormatOverride(dateFormat);
        return chart;
    }

    protected abstract JFreeChart buildChart();

    private void addSeries() {
        Map<String, MovingWindowXYDataset<ChartingTimeSeries>> dataSets = new HashMap<String, MovingWindowXYDataset<ChartingTimeSeries>>();
        for (ChartingTimeSeries series : timeSeriesList) {
            addToDataSet(dataSets, series);
        }

        int dataSetIndex = 0;
        for (MovingWindowXYDataset<ChartingTimeSeries> d : dataSets.values()) {
            plot.setDataset(dataSetIndex, d);

            XYItemRenderer renderer = createXYItemRenderer(d);
            plot.setRenderer(dataSetIndex, renderer);

            for (int series = 0; series < d.getSeriesCount(); series++) {
                Color c = d.getTimeSeries(series).getColor();
                renderer.setSeriesPaint(series, c);
            }

            String axisLabel;
            switch (chartRangeMode) {
            case RangePerSeries:
                axisLabel = d.getTimeSeries(0).getDisplayName();
                break;
            case RangePerId:
                axisLabel = d.getTimeSeries(0).getId();
                break;
            case SingleRange:
            default:
                axisLabel = "values";
            }

            plot.setRangeAxis(dataSetIndex, new NumberAxis(axisLabel));
            plot.setRangeAxisLocation(dataSetIndex,
                    dataSetIndex % 2 == 0 ? AxisLocation.BOTTOM_OR_LEFT : AxisLocation.BOTTOM_OR_RIGHT);
            plot.mapDatasetToRangeAxis(dataSetIndex, dataSetIndex);

            if (chartRangeMode == ChartRangeMode.RangePerSeries) {
                plot.getRangeAxis(dataSetIndex).setLabelPaint(d.getTimeSeries(0).getColor());
                plot.getRangeAxis(dataSetIndex).setTickLabelPaint(d.getTimeSeries(0).getColor());
            }

            d.startMovingWindow(Time.seconds(1));
            dataSetIndex++;
        }
    }

    private void addToDataSet(Map<String, MovingWindowXYDataset<ChartingTimeSeries>> dataSets,
            ChartingTimeSeries series) {
        String key;
        switch (chartRangeMode) {
        case RangePerSeries:
            key = series.getPath();
            break;
        case RangePerId:
            key = series.getId();
            break;
        case SingleRange:
        default:
            key = "values";
        }
        MovingWindowXYDataset<ChartingTimeSeries> dataSet = getOrCreateXYDataSet(key, dataSets);
        dataSet.addTimeSeries(series.getDisplayName(), series);
    }

    private MovingWindowXYDataset getOrCreateXYDataSet(String key,
            Map<String, MovingWindowXYDataset<ChartingTimeSeries>> dataSets) {
        MovingWindowXYDataset<ChartingTimeSeries> s = dataSets.get(key);
        if (s == null) {
            s = new MovingWindowXYDataset<ChartingTimeSeries>(domainSelection, TimeSource.OPEN_END_TIME);
            dataSets.put(key, s);
        }
        return s;
    }

    protected abstract XYItemRenderer createXYItemRenderer(XYDataset dataSet);

    protected boolean isShowLegend() {
        return showLegend;
    }

    public String getTitle() {
        return title;
    }

}