playground.dgrether.analysis.charts.DgAvgDeltaMoneyQuantilesChart.java Source code

Introduction

Here is the source code for playground.dgrether.analysis.charts.DgAvgDeltaMoneyQuantilesChart.java

Source

/* *********************************************************************** *
 * project: org.matsim.*
 * DgAvgDeltaUtilsQuantilesChart
 *                                                                         *
 * *********************************************************************** *
 *                                                                         *
 * copyright       : (C) 2009 by the members listed in the COPYING,        *
 *                   LICENSE and WARRANTY file.                            *
 * email           : info at matsim dot org                                *
 *                                                                         *
 * *********************************************************************** *
 *                                                                         *
 *   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 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *   See also COPYING, LICENSE and WARRANTY file                           *
 *                                                                         *
 * *********************************************************************** */
package playground.dgrether.analysis.charts;

import java.awt.BasicStroke;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.jfree.chart.ChartColor;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.axis.ValueAxis;
import org.jfree.chart.labels.AbstractXYItemLabelGenerator;
import org.jfree.chart.labels.XYItemLabelGenerator;
import org.jfree.chart.plot.XYPlot;
import org.jfree.chart.renderer.xy.XYItemRenderer;
import org.jfree.chart.renderer.xy.XYLineAndShapeRenderer;
import org.jfree.data.xy.XYDataset;
import org.jfree.data.xy.XYSeries;
import org.jfree.data.xy.XYSeriesCollection;
import org.matsim.core.utils.collections.Tuple;

import playground.dgrether.analysis.charts.utils.DgColorScheme;
import playground.dgrether.analysis.population.DgAnalysisPopulation;
import playground.dgrether.analysis.population.DgPersonDataIncomeComparator;

/**
 * @author dgrether
 *
 */
public class DgAvgDeltaMoneyQuantilesChart {

    protected DgAnalysisPopulation ana;

    protected int nQuantiles = 10;

    protected int groupThreshold = 3;

    protected XYSeriesCollection dataset;

    protected DgAxisBuilder axisBuilder = new DgDefaultAxisBuilder();

    private LabelGenerator labelGenerator;

    public DgAvgDeltaMoneyQuantilesChart(DgAnalysisPopulation ana, String runId1, String runId2) {
        this.ana = ana;
        this.labelGenerator = new LabelGenerator();
        this.ana.calculateMinMaxIncome();
        this.dataset = this.createDatasets(runId1, runId2);
    }

    protected Tuple<XYSeries, List<String>> createXYSeries(String title, DgAnalysisPopulation pop, String runId1,
            String runId2) {
        List<DgAnalysisPopulation> quantiles = this.ana.getQuantiles(this.nQuantiles,
                new DgPersonDataIncomeComparator());
        XYSeries series = new XYSeries(title, false, true);
        List<String> labels = new ArrayList<String>();
        for (DgAnalysisPopulation p : quantiles) {
            Double avgScore = IncomeChartUtils.calcAverageMoneyDifference(p, 0, runId1, runId2);
            p.calculateMinMaxIncome();
            if (avgScore != null) {
                double incomeLocation = p.getMinIncome() + ((p.getMaxIncome() - p.getMinIncome()) / 2.0);
                series.add(incomeLocation, avgScore);
                labels.add("Groupsize: " + p.getPersonData().size());
            }
        }
        return new Tuple<XYSeries, List<String>>(series, labels);
    }

    protected XYSeriesCollection createDatasets(String runId1, String runId2) {
        XYSeriesCollection ds = new XYSeriesCollection();
        Tuple<XYSeries, List<String>> seriesLabels = this.createXYSeries("Mean " + '\u0394' + " Chf", this.ana,
                runId1, runId2);
        ds.addSeries(seriesLabels.getFirst());
        this.labelGenerator.setLabels(0, seriesLabels.getSecond());
        return ds;
    }

    public JFreeChart createChart() {
        XYPlot plot = new XYPlot();
        ValueAxis xAxis = this.axisBuilder.createValueAxis("Income [Chf / Year]");
        ValueAxis yAxis = this.axisBuilder.createValueAxis("Delta Money [Chf]");
        plot.setDomainAxis(xAxis);
        plot.setRangeAxis(yAxis);

        DgColorScheme colorScheme = new DgColorScheme();

        XYItemRenderer renderer2;
        renderer2 = new XYLineAndShapeRenderer(true, true);
        renderer2.setSeriesItemLabelsVisible(0, true);
        renderer2.setSeriesItemLabelGenerator(0, this.labelGenerator);
        plot.setDataset(0, this.dataset);
        renderer2.setSeriesStroke(0, new BasicStroke(2.0f));
        renderer2.setSeriesOutlineStroke(0, new BasicStroke(3.0f));
        renderer2.setSeriesPaint(0, colorScheme.getColor(1, "a"));
        plot.setRenderer(0, renderer2);

        JFreeChart chart = new JFreeChart("", plot);
        chart.setBackgroundPaint(ChartColor.WHITE);
        chart.getLegend().setItemFont(this.axisBuilder.getAxisFont());
        chart.setTextAntiAlias(true);
        return chart;
    }

    public XYSeriesCollection getDataset() {
        return dataset;
    }

    public int getNumberOfClasses() {
        return nQuantiles;
    }

    public void setNumberOfClasses(int numberOfClasses) {
        this.nQuantiles = numberOfClasses;
    }

    /**
      * A custom label generator.
      */
    public static class LabelGenerator extends AbstractXYItemLabelGenerator implements XYItemLabelGenerator {

        private Map<Integer, List<String>> labels = new HashMap<Integer, List<String>>();

        public LabelGenerator() {

        }

        public void setLabels(int series, List<String> labels) {
            this.labels.put(series, labels);
        }

        @Override
        public String generateLabel(XYDataset dataset, int series, int item) {
            return this.labels.get(series).get(item);
        }
    }
}