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

package genius.gui.progress.session;

import java.awt.BorderLayout;
import java.awt.Color;
import java.awt.Dimension;

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

import org.jfree.chart.ChartPanel;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.axis.NumberAxis;
import org.jfree.chart.plot.XYPlot;
import org.jfree.chart.renderer.xy.XYDotRenderer;
import org.jfree.chart.renderer.xy.XYLineAndShapeRenderer;
import org.jfree.data.xy.XYDataItem;
import org.jfree.data.xy.XYDataset;
import org.jfree.data.xy.XYSeries;
import org.jfree.data.xy.XYSeriesCollection;

import genius.core.AgentID;
import genius.core.Bid;
import genius.core.BidIterator;
import genius.core.analysis.BidPoint;
import genius.core.analysis.MultilateralAnalysis;
import genius.core.utility.UtilitySpace;

/**
 * * Bilateral version of ProgressChart. Shows utility of party 1 on the X axis,
 * and of party 2 on the Y (vertical) axis.
 */
@SuppressWarnings("serial")
public class ProgressChartBi extends JPanel {

        private static final String UTILITY = " utility";
        final XYSeriesCollection dataset = new XYSeriesCollection();
        private OutcomesListModel model;
        private XYSeries bidsA, bidsB, agreements;
        private AgentID party1;
        private AgentID party2;

        /**
         * @param model
         *            all outcomes so far. Assumes that only items are being added
         *            to the OutcomesModel, one at a time. Assumes that model has 2
         *            parties.
         * @param showAllBids
         *            all bids in the bidspace are shown iff this is true.
         */
        public ProgressChartBi(OutcomesListModel model, boolean showAllBids) {
                this.model = model;
                setLayout(new BorderLayout());
                setMinimumSize(new Dimension(300, 300));
                party1 = model.getParties().get(0).getID();
                party2 = model.getParties().get(1).getID();
                NumberAxis domainAxis = new NumberAxis(party1.toString() + UTILITY);
                NumberAxis rangeAxis = new NumberAxis(party2.toString() + UTILITY);

                bidsA = new XYSeries(party1.toString(), false);
                bidsB = new XYSeries(party2.toString(), false);
                agreements = new XYSeries("agreements");
                XYPlot plot = new XYPlot(dataset, domainAxis, rangeAxis, new XYLineAndShapeRenderer());

                dataset.addSeries(agreements); // first one has priority on plot.
                dataset.addSeries(bidsA);
                dataset.addSeries(bidsB);

                // add some background info
                MultilateralAnalysis analysis = new MultilateralAnalysis(model.getParties(), null, null);
                dataset.addSeries(getNash(analysis));
                dataset.addSeries(getKalai(analysis));
                dataset.addSeries(getPareto(analysis));

                if (showAllBids) {
                        XYDotRenderer dotRenderer = new XYDotRenderer();
                        dotRenderer.setDotHeight(2);
                        dotRenderer.setDotWidth(2);
                        dotRenderer.setSeriesPaint(0, Color.PINK);
                        plot.setDataset(2, getAllBids());
                        plot.setRenderer(2, dotRenderer);
                }

                // fix the color of renderer 3: its color is invisible yellow ...
                plot.getRenderer().setSeriesPaint(3, Color.BLACK);

                JFreeChart freechart = new JFreeChart("", JFreeChart.DEFAULT_TITLE_FONT, plot, true);
                ChartPanel chartPanel = new ChartPanel(freechart);

                /**
                 * BAD: JFreeChart doesn't have a model so we must connect it directly.
                 * actually, there IS a model: {@link XYDataset}. Maybe we can fix this.
                 */
                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 XYSeries getKalai(MultilateralAnalysis analysis) {
                XYSeries kalaiSeries = new XYSeries("Kalai");
                BidPoint point = analysis.getKalaiPoint();
                kalaiSeries.add(point.getUtilityA(), point.getUtilityB());
                return kalaiSeries;
        }

        private XYSeries getNash(MultilateralAnalysis analysis) {
                XYSeries nashSeries = new XYSeries("Nash");
                BidPoint point = analysis.getNashPoint();
                nashSeries.add(point.getUtilityA(), point.getUtilityB());
                return nashSeries;
        }

        /**
         * Get all bids in a dataset. Just iterates over domain of first party
         * 
         * @return {@link XYDataSet} containing all bids.
         */
        private XYDataset getAllBids() {

                UtilitySpace utils1 = model.getParties().get(0).getUtilitySpace();
                UtilitySpace utils2 = model.getParties().get(1).getUtilitySpace();

                BidIterator bids = new BidIterator(utils1.getDomain());
                XYSeries series = new XYSeries("All bids");
                while (bids.hasNext()) {
                        Bid bid = bids.next();
                        series.add(utils1.getUtility(bid), utils2.getUtility(bid));
                }
                return new XYSeriesCollection(series);
        }

        private XYSeries getPareto(MultilateralAnalysis analysis) {
                XYSeries series = new XYSeries("pareto");
                for (BidPoint point : analysis.getParetoFrontier()) {
                        series.add(point.getUtilityA(), point.getUtilityB());
                }

                return series;
        }

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

                        @Override
                        public void run() {
                                // invariant n is current number of data points in graph.
                                int n = bidsA.getItemCount() + bidsB.getItemCount();

                                // continue till all model points are in graph
                                while (n < model.size()) {
                                        Outcome outcome = model.get(n);
                                        XYDataItem item = new XYDataItem((double) outcome.getDiscountedUtilities().get(0),
                                                        (double) outcome.getDiscountedUtilities().get(1));
                                        if (party1.equals(outcome.getAgentID())) {
                                                bidsA.add(item);
                                        } else {
                                                bidsB.add(item);
                                        }
                                        if (outcome.isAgreement()) {
                                                agreements.add(item);
                                        }
                                        n++;
                                }

                        }
                });
        }

}