playground.benjamin.scenarios.zurich.analysis.charts.BkDeltaUtilsQuantilesChart.java Source code

Introduction

Here is the source code for playground.benjamin.scenarios.zurich.analysis.charts.BkDeltaUtilsQuantilesChart.java

Source

/* *********************************************************************** *
 * project: org.matsim.*
 * ChartWriter
 *                                                                         *
 * *********************************************************************** *
 *                                                                         *
 * 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.benjamin.scenarios.zurich.analysis.charts;

import java.util.SortedMap;
import java.util.TreeMap;
import java.util.Map.Entry;

import org.jfree.chart.JFreeChart;
import org.jfree.chart.axis.NumberAxis;
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.XYSeries;
import org.jfree.data.xy.XYSeriesCollection;
import org.matsim.api.core.v01.Id;

import playground.benjamin.scenarios.zurich.analysis.WinnerLoserAnalysisRow;

public class BkDeltaUtilsQuantilesChart {

    private XYSeriesCollection dataset;
    private SortedMap<Id, WinnerLoserAnalysisRow> populationInformation;

    //constructor for row approach
    public BkDeltaUtilsQuantilesChart(SortedMap<Id, WinnerLoserAnalysisRow> populationInformation) {
        this.populationInformation = populationInformation;
        this.dataset = createNeededDataset();
    }

    //============================================================================================================   

    private XYSeriesCollection createNeededDataset() {
        //instancing the dataset 
        XYSeriesCollection ds = new XYSeriesCollection();

        ds.addSeries(this.createSeries("delta utils over income quantiles",
                personalIncomeInQuantiles2Scores(populationInformation)));
        return ds;
    }

    private XYSeries createSeries(final String title, SortedMap<Double, Double> result) {
        XYSeries series = new XYSeries(title, false, true);
        for (Entry<Double, Double> entry : result.entrySet()) {
            series.add(entry.getKey(), entry.getValue());
        }
        return series;
    }

    /**
     * Dependent on what to plot this method has to be adapted
     * @param populationInformation (Map from Id to Row (all desired information))
     * @return Map from income to the chosen information (e.g. scoreDiff)
     * 
     */
    private SortedMap<Double, Double> personalIncomeInQuantiles2Scores(
            SortedMap<Id, WinnerLoserAnalysisRow> populationInformation) {
        SortedMap<Double, Double> result = new TreeMap<Double, Double>();

        //iterating over a map and getting the desired values out of Row (personal income and score difference)
        for (Entry<Id, WinnerLoserAnalysisRow> entry : populationInformation.entrySet()) {
            WinnerLoserAnalysisRow winnerLoserAnalysisRow = entry.getValue();
            Double personalIncome = winnerLoserAnalysisRow.getPersonalIncome();
            Double scoreDiff = winnerLoserAnalysisRow.getScoreDiff();
            result.put(personalIncome, scoreDiff);
        }
        return result;
    }

    //============================================================================================================      

    public JFreeChart createChart() {
        XYPlot plot = new XYPlot(this.dataset, new NumberAxis("population deciles sorted by income"),
                new NumberAxis("delta utils"), null);
        XYItemRenderer renderer = new XYLineAndShapeRenderer(false, true);
        plot.setRenderer(renderer);

        JFreeChart jchart = new JFreeChart("", plot);
        return jchart;
    }
}