/* * Copyright (C) 2016 Irstea * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * This program 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 General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see <http://www.gnu.org/licenses/>. */ package fr.irstea.associatione.model.gui; import fr.irstea.associatione.model.Model; import fr.irstea.associatione.model.ProcessingException; import fr.irstea.associatione.model.ReflectUtils; import java.awt.GridLayout; import java.util.ArrayList; import java.util.List; import java.util.logging.Level; import java.util.logging.Logger; import javax.swing.JPanel; import org.jfree.chart.ChartFactory; import org.jfree.chart.ChartPanel; import org.jfree.chart.JFreeChart; import org.jfree.chart.axis.NumberAxis; import org.jfree.chart.plot.PlotOrientation; import org.jfree.chart.plot.XYPlot; import org.jfree.chart.renderer.xy.StandardXYBarPainter; import org.jfree.chart.renderer.xy.XYBarRenderer; import org.jfree.data.statistics.HistogramDataset; import org.jfree.data.statistics.HistogramType; import org.jfree.data.xy.XYSeries; import org.jfree.data.xy.XYSeriesCollection; /** * * @author Nicolas Dumoulin <nicolas.dumoulin@irstea.fr> */ public class TimeSerieChart { private JPanel display; private List<Timeserie> timeseries; public static class Timeserie { private final String name; private final boolean histogram, series; private final ReflectUtils.MethodOnInstance methodOnInstance; private final List<double[]> timeserie; private ChartPanel histogramPanel; private final XYSeriesCollection xySeries; public Timeserie(Model model, String fetcherPath, String name, boolean histogram, boolean series) throws ProcessingException { this.name = name; this.histogram = histogram; this.series = series; try { this.methodOnInstance = ReflectUtils.getMethodOnInstance(model, fetcherPath); this.timeserie = new ArrayList<>(); xySeries = new XYSeriesCollection(); histogramPanel = new ChartPanel(createHistogram(name, 0, (double[]) methodOnInstance.invoke())); } catch (NoSuchFieldException | IllegalAccessException ex) { throw new ProcessingException("Error during method retrieval", ex); } } 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; 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; } }