StackedBarChart.java

Go to the documentation of this file.

/* *********************************************************************** *
 * project: org.matsim.*
 * BarChart.java
 *                                                                         *
 * *********************************************************************** *
 *                                                                         *
 * copyright       : (C) 2007 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 org.matsim.core.utils.charts;

import org.jfree.chart.ChartFactory;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.axis.CategoryAxis;
import org.jfree.chart.plot.CategoryPlot;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.chart.renderer.category.BarRenderer;
import org.jfree.data.category.CategoryDataset;
import org.jfree.data.category.DefaultCategoryDataset;

public class StackedBarChart extends ChartUtil {

    private final String[] categories;
    private final DefaultCategoryDataset dataset;
    private final CategoryPlot plot;

    public StackedBarChart(final String title, final String xAxisLabel, final String yAxisLabel) {
        this(title, xAxisLabel, yAxisLabel, new String[]{});
    }

    public StackedBarChart(final String title, final String xAxisLabel, final String yAxisLabel, final String[] categories) {
        super(title, xAxisLabel, yAxisLabel);
        this.dataset = new DefaultCategoryDataset();
        this.chart = createChart(title, xAxisLabel, yAxisLabel, this.dataset);
        this.plot = this.chart.getCategoryPlot();
        this.categories = categories.clone();

        addDefaultFormatting();

        // leave a gap of 20% between categories (groups of bars)
        final CategoryAxis axis1 = this.plot.getDomainAxis();
        axis1.setCategoryMargin(0.20);

        // leave a gap of 10% between individual bars within one category
        BarRenderer renderer = (BarRenderer)this.plot.getRenderer();
        renderer.setItemMargin(0.10);
    }

    @Override
    public JFreeChart getChart() {
        return this.chart;
    }

    private JFreeChart createChart(final String title, final String categoryAxisLabel,
            final String valueAxisLabel, final CategoryDataset dataset) {
        return ChartFactory.createStackedBarChart(title, categoryAxisLabel, valueAxisLabel,
                dataset, PlotOrientation.VERTICAL, true, // legend?
                false, // tooltips?
                false // URLs?
                );
    }

    public void addSeries(final String title, final double[] values) {
        int cnt = 1;
        for (double value : values) {
            String category = (cnt > this.categories.length ? Integer.toString(cnt) : this.categories[cnt-1]);
            this.dataset.addValue(value, title, category);
            cnt++;
        }
    }

}