source: src/main/java/genius/gui/progress/session/ProgressChart.java

package genius.gui.progress.session;

import java.awt.BorderLayout;
import java.awt.Dimension;
import java.util.ArrayList;
import java.util.List;

import javax.swing.JPanel;
import javax.swing.SwingUtilities;
import javax.swing.event.ListDataEvent;
import javax.swing.event.ListDataListener;

import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartPanel;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.chart.plot.XYPlot;
import org.jfree.chart.renderer.xy.XYLineAndShapeRenderer;
import org.jfree.data.xy.XYDataItem;
import org.jfree.data.xy.XYSeries;
import org.jfree.data.xy.XYSeriesCollection;

import genius.core.parties.PartyWithUtility;

/**
 * Shows progress as a plot where each party's utility is set vertically against
 * rounds horizontally. This assumes that only items are being added to the
 * OutcomesModel, one at a time.
 */
@SuppressWarnings("serial")
public class ProgressChart extends JPanel {

        final XYSeriesCollection dataset = new XYSeriesCollection();
        private OutcomesListModel model;
        private List<XYSeries> datasets = new ArrayList<XYSeries>();

        public ProgressChart(OutcomesListModel model) {
                this.model = model;
                setLayout(new BorderLayout());
                setMinimumSize(new Dimension(300, 300));

                for (PartyWithUtility party : model.getParties()) {
                        XYSeries series = new XYSeries(party.getID().toString());
                        datasets.add(series);
                        dataset.addSeries(series);
                }

                JFreeChart xylineChart = ChartFactory.createXYLineChart("", "Round", "Utility", dataset,
                                PlotOrientation.VERTICAL, true, true, false);

                ChartPanel chartPanel = new ChartPanel(xylineChart);
                final XYPlot plot = xylineChart.getXYPlot();
                XYLineAndShapeRenderer renderer = new XYLineAndShapeRenderer();
                plot.setRenderer(renderer);

                // BAD: JFreeChart doesn't have a model so we must connect it directly
                model.addListDataListener(new ListDataListener() {

                        @Override
                        public void intervalRemoved(ListDataEvent e) {
                        }

                        @Override
                        public void intervalAdded(ListDataEvent e) {
                                addNewOutcome();
                        }

                        @Override
                        public void contentsChanged(ListDataEvent e) {
                        }

                });

                add(chartPanel, BorderLayout.CENTER);
        }

        private void addNewOutcome() {
                SwingUtilities.invokeLater(new Runnable() {

                        @Override
                        public void run() {
                                Outcome outcome = model.get(model.getSize() - 1);// last
                                for (int n = 0; n < datasets.size(); n++) {
                                        // FIXME use datasets.get(n).getMaximumItemCount() or so
                                        datasets.get(n).add(new XYDataItem((double) outcome.getRound(),
                                                        (double) outcome.getDiscountedUtilities().get(n)));
                                }
                        }
                });
        }

}