sernet.gs.ui.rcp.main.bsi.views.chart.RealisierungLineChart.java Source code

Introduction

Here is the source code for sernet.gs.ui.rcp.main.bsi.views.chart.RealisierungLineChart.java

Source

/*******************************************************************************
 * Copyright (c) 2009 Alexander Koderman <ak[at]sernet[dot]de>.
 * This program is free software: you can redistribute it and/or 
 * modify it under the terms of the GNU Lesser General Public License 
 * as published by the Free Software Foundation, either version 3 
 * of the License, or (at your option) any later version.
 *     This program is distributed in the hope that it will be useful,    
 * but WITHOUT ANY WARRANTY; without even the implied warranty 
 * of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  
 * See the GNU Lesser General Public License for more details.
 *     You should have received a copy of the GNU Lesser General Public 
 * License along with this program. 
 * If not, see <http://www.gnu.org/licenses/>.
 * 
 * Contributors:
 *     Alexander Koderman <ak[at]sernet[dot]de> - initial API and implementation
 ******************************************************************************/
package sernet.gs.ui.rcp.main.bsi.views.chart;

import java.awt.Color;
import java.awt.Font;
import java.io.Serializable;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

import org.jfree.chart.JFreeChart;
import org.jfree.chart.axis.AxisLocation;
import org.jfree.chart.axis.DateAxis;
import org.jfree.chart.axis.NumberAxis;
import org.jfree.chart.plot.CombinedDomainXYPlot;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.chart.plot.ValueMarker;
import org.jfree.chart.plot.XYPlot;
import org.jfree.chart.renderer.xy.StandardXYItemRenderer;
import org.jfree.chart.renderer.xy.XYItemRenderer;
import org.jfree.data.time.Day;
import org.jfree.data.time.TimeSeries;
import org.jfree.data.time.TimeSeriesCollection;
import org.jfree.data.xy.XYDataset;
import org.jfree.ui.Layer;
import org.jfree.ui.RectangleAnchor;
import org.jfree.ui.TextAnchor;

import sernet.gs.ui.rcp.main.ExceptionUtil;
import sernet.gs.ui.rcp.main.service.ServiceFactory;
import sernet.gs.ui.rcp.main.service.statscommands.CountMassnahmen;
import sernet.gs.ui.rcp.main.service.statscommands.RealisierungSummary;
import sernet.verinice.interfaces.CommandException;

@SuppressWarnings("serial")
public class RealisierungLineChart implements IChartGenerator, Serializable {

    public JFreeChart createChart() {
        try {
            return createProgressChart(createProgressDataset());
        } catch (CommandException e) {
            ExceptionUtil.log(e, Messages.RealisierungLineChart_0);
        }
        return null;
    }

    private JFreeChart createProgressChart(Object dataset) {
        final double plotGap = 10.0;
        final int axisUpperBoundPadding = 50;
        final int labelFontSize = 10;
        XYDataset data1 = (XYDataset) dataset;
        XYItemRenderer renderer1 = new StandardXYItemRenderer();
        NumberAxis rangeAxis1 = new NumberAxis(Messages.RealisierungLineChart_1);
        XYPlot subplot1 = new XYPlot(data1, null, rangeAxis1, renderer1);
        subplot1.setRangeAxisLocation(AxisLocation.BOTTOM_OR_LEFT);

        CombinedDomainXYPlot plot = new CombinedDomainXYPlot(new DateAxis(Messages.RealisierungLineChart_2));
        plot.setGap(plotGap);

        plot.add(subplot1, 1);
        plot.setOrientation(PlotOrientation.VERTICAL);

        CountMassnahmen command = new CountMassnahmen();
        try {
            command = ServiceFactory.lookupCommandService().executeCommand(command);
        } catch (CommandException e) {
            ExceptionUtil.log(e, Messages.RealisierungLineChart_3);
        }
        int totalNum = command.getTotalCount();

        NumberAxis axis = (NumberAxis) subplot1.getRangeAxis();
        axis.setUpperBound(totalNum + axisUpperBoundPadding);

        ValueMarker bst = new ValueMarker(totalNum);
        bst.setPaint(Color.GREEN);
        bst.setLabel(Messages.RealisierungLineChart_4);
        bst.setLabelAnchor(RectangleAnchor.LEFT);
        bst.setLabelFont(new Font("SansSerif", Font.ITALIC + Font.BOLD, labelFontSize)); //$NON-NLS-1$
        bst.setLabelTextAnchor(TextAnchor.CENTER_LEFT);
        subplot1.addRangeMarker(bst, Layer.BACKGROUND);

        // return a new chart containing the overlaid plot...
        JFreeChart chart = new JFreeChart(Messages.RealisierungLineChart_6, JFreeChart.DEFAULT_TITLE_FONT, plot,
                true);
        chart.setBackgroundPaint(Color.white);
        return chart;
    }

    private Object createProgressDataset() throws CommandException {
        TimeSeries ts1 = new TimeSeries(Messages.RealisierungLineChart_7, Day.class);
        TimeSeries ts2 = new TimeSeries(Messages.RealisierungLineChart_8, Day.class);

        RealisierungSummary command = new RealisierungSummary();
        command = ServiceFactory.lookupCommandService().executeCommand(command);

        DateValues dateTotal1 = command.getTotal1();
        DateValues dateTotal2 = command.getTotal2();

        Map<Day, Integer> totals1 = dateTotal1.getDateTotals();
        Set<Entry<Day, Integer>> entrySet1 = totals1.entrySet();
        for (Entry<Day, Integer> entry : entrySet1) {
            ts1.add(entry.getKey(), entry.getValue());
        }

        Map<Day, Integer> totals2 = dateTotal2.getDateTotals();
        Set<Entry<Day, Integer>> entrySet2 = totals2.entrySet();
        for (Entry<Day, Integer> entry : entrySet2) {
            ts2.add(entry.getKey(), entry.getValue());
        }

        TimeSeriesCollection tsc = new TimeSeriesCollection();
        tsc.addSeries(ts2);
        tsc.addSeries(ts1);
        return tsc;
    }
}