ch.zhaw.init.walj.projectmanagement.util.chart.PieChart.java Source code

Introduction

Here is the source code for ch.zhaw.init.walj.projectmanagement.util.chart.PieChart.java

Source

/*
Copyright 2016-2017 Zuercher Hochschule fuer Angewandte Wissenschaften
All Rights Reserved.
    
   Licensed under the Apache License, Version 2.0 (the "License"); you may
   not use this file except in compliance with the License. You may obtain
   a copy of the License at
    
   http://www.apache.org/licenses/LICENSE-2.0
    
   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
   WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
   License for the specific language governing permissions and limitations
   under the License.
 */

package ch.zhaw.init.walj.projectmanagement.util.chart;

import java.awt.Color;
import java.io.File;
import java.io.IOException;

import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartUtilities;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.plot.PiePlot;
import org.jfree.data.general.DefaultPieDataset;

import ch.zhaw.init.walj.projectmanagement.util.DBConnection;
import ch.zhaw.init.walj.projectmanagement.util.dbclasses.Project;

/**
 * creates a pie chart with data from a given project
 * 
 * @author Janine Walther, ZHAW
 *
 */
public class PieChart {

    private final Project project;
    private final DBConnection con;
    private final String path;

    /**
     * Constructor of the PieChart class
     * @param project the project for which the chart should be created
     * @param path path where PMSaaS is located
     */
    public PieChart(Project project, String path) {
        this.project = project;
        this.path = path;
        con = new DBConnection(path);
    }

    /**
     * creates the pie chart
     * @throws NumberFormatException
     * @throws IOException
     */
    public void createChart() throws NumberFormatException, IOException {

        // create dataset with used and remaining budget
        DefaultPieDataset dataset = new DefaultPieDataset();
        dataset.setValue("spent", con.getUsedBudget(project));
        dataset.setValue("remaining", con.getRemainingBudget(project));

        // create chart
        JFreeChart chart = ChartFactory.createPieChart("", dataset, false, false, false);

        // set color and style
        PiePlot plot = (PiePlot) chart.getPlot();
        plot.setSectionPaint("remaining", new Color(0, 101, 166));
        plot.setSectionPaint("spent", new Color(0, 62, 102));
        plot.setLabelGenerator(null);
        plot.setBackgroundPaint(Color.WHITE);
        plot.setShadowXOffset(0);
        plot.setShadowYOffset(0);
        plot.setOutlineVisible(false);
        plot.setInteriorGap(0);

        // set size and save it as JPEG
        int width = 350; /* Width of the image */
        int height = 350; /* Height of the image */
        File pieChart = new File(path + "/Charts/BudgetProject" + project.getID() + ".jpg");
        ChartUtilities.saveChartAsJPEG(pieChart, chart, width, height);
    }
}