Skip to content
GitLab
Select Git revision
  • master default
Find file BlameHistoryPermalink
TimeSerieChart.java 5.67 KiB
1 
/*
2 
 * Copyright (C) 2016 Irstea
3 
 *
4 
 * This program is free software: you can redistribute it and/or modify
5 
 * it under the terms of the GNU General Public License as published by
6 
 * the Free Software Foundation, either version 3 of the License, or
7 
 * (at your option) any later version.
8 
 *
9 
 * This program is distributed in the hope that it will be useful,
10 
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 
 * GNU General Public License for more details.
13 
 *
14 
 * You should have received a copy of the GNU General Public License
15 
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
16 
 */
17 
package fr.irstea.associatione.model.gui;
18
19 
import fr.irstea.associatione.model.Model;
20 
import fr.irstea.associatione.model.ProcessingException;
21 
import fr.irstea.associatione.model.ReflectUtils;
22 
import java.awt.GridLayout;
23 
import java.util.ArrayList;
24 
import java.util.List;
25 
import java.util.logging.Level;
26 
import java.util.logging.Logger;
27 
import javax.swing.JPanel;
28 
import org.jfree.chart.ChartFactory;
29 
import org.jfree.chart.ChartPanel;
30 
import org.jfree.chart.JFreeChart;
31 
import org.jfree.chart.axis.NumberAxis;
32 
import org.jfree.chart.plot.PlotOrientation;
33 
import org.jfree.chart.plot.XYPlot;
34 
import org.jfree.chart.renderer.xy.StandardXYBarPainter;
35 
import org.jfree.chart.renderer.xy.XYBarRenderer;
36 
import org.jfree.data.statistics.HistogramDataset;
37 
import org.jfree.data.statistics.HistogramType;
38 
import org.jfree.data.xy.XYSeries;
39 
import org.jfree.data.xy.XYSeriesCollection;
40
41 
/**
42 
 *
43 
 * @author Nicolas Dumoulin <nicolas.dumoulin@irstea.fr>
44 
 */
45 
public class TimeSerieChart {
46
47 
    private JPanel display;
48 
    private List<Timeserie> timeseries;
49
50 
    public static class Timeserie {
51
52 
        private final String name;
53 
        private final boolean histogram, series;
54 
        private final ReflectUtils.MethodOnInstance methodOnInstance;
55 
        private final List<double[]> timeserie;
56 
        private ChartPanel histogramPanel;
57 
        private final XYSeriesCollection xySeries;
58
59 
        public Timeserie(Model model, String fetcherPath, String name, boolean histogram, boolean series) throws ProcessingException {
60 
            this.name = name;
61 
            this.histogram = histogram;
62 
            this.series = series;
63 
            try {
64 
                this.methodOnInstance = ReflectUtils.getMethodOnInstance(model, fetcherPath);
65 
                this.timeserie = new ArrayList<>();
66 
                xySeries = new XYSeriesCollection();
67 
                histogramPanel = new ChartPanel(createHistogram(name, 0, (double[]) methodOnInstance.invoke()));
68 
            } catch (NoSuchFieldException | IllegalAccessException ex) {
69 
                throw new ProcessingException("Error during method retrieval", ex);
70 
            }
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
} private JPanel getHistogram() { return histogramPanel; } private JPanel getSeries() { return new ChartPanel(ChartFactory.createXYLineChart(name, "timesteps", name, xySeries, PlotOrientation.VERTICAL, false, true, false)); } public void add(double[] data) { timeserie.add(data); } public void update(int timestep) throws IllegalAccessException { final double[] data = (double[]) methodOnInstance.invoke(); histogramPanel.setChart(createHistogram(name, timestep, data)); timeserie.add(data); if (xySeries.getSeriesCount() == 0) { for (int i = 0; i < data.length; i++) { xySeries.addSeries(new XYSeries(i)); } } for (int i = 0; i < data.length; i++) { xySeries.getSeries(i).add(timeserie.size() - 1, data[i]); } } } public TimeSerieChart() { display = new JPanel(new GridLayout(0, 2)); timeseries = new ArrayList<>(); } public JPanel getDisplay() { return display; } public void addTimeSerie(Model model, String fetcherPath, String name, boolean histogram, boolean series) throws ProcessingException { final Timeserie timeserie = new Timeserie(model, fetcherPath, name, histogram, series); timeseries.add(timeserie); if (histogram) { display.add(timeserie.getHistogram()); } if (series) { display.add(timeserie.getSeries()); } } public void updateCharts(int timestep) { for (Timeserie timeserie : timeseries) { try { timeserie.update(timestep); } catch (IllegalAccessException ex) { Logger.getLogger(TimeSerieChart.class.getName()).log(Level.SEVERE, null, ex); } } } public static JFreeChart createHistogram(String title, int timestep, double[] data) { HistogramDataset dataset = new HistogramDataset(); dataset.setType(HistogramType.RELATIVE_FREQUENCY); dataset.addSeries(title, data, 30); String plotTitle = title + " " + timestep; String xaxis = "opinions"; String yaxis = "amount"; PlotOrientation orientation = PlotOrientation.VERTICAL; boolean show = false;
141142143144145146147148149150151152153154155156
boolean toolTips = true; boolean urls = false; JFreeChart chart = ChartFactory.createHistogram(plotTitle, xaxis, yaxis, dataset, orientation, show, toolTips, urls); XYPlot plot = (XYPlot) chart.getPlot(); NumberAxis numberaxis = (NumberAxis)plot.getRangeAxis(); numberaxis.setUpperBound(1.0); // raw style XYBarRenderer renderer = (XYBarRenderer) plot.getRenderer(); renderer.setBarPainter(new StandardXYBarPainter()); renderer.setShadowVisible(false); return chart; } }