gavisualizer.PieChart.java Source code

Introduction

Here is the source code for gavisualizer.PieChart.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package gavisualizer;

import java.awt.Color;
import java.awt.Rectangle;
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.labels.StandardPieSectionLabelGenerator;
import org.jfree.chart.plot.PiePlot;
import org.jfree.data.general.PieDataset;
import org.jfree.util.Rotation;
import org.jfree.graphics2d.svg.SVGGraphics2D;
import org.jfree.graphics2d.svg.SVGUtils;

/**
 *
 * @author Jake
 */
public class PieChart implements IChart {

    // Global variables for chart Pie Chart
    private final PieDataset _dataset;
    private JFreeChart _chart;
    private final String _title;

    // Constructor for Line Chart that sets global variables from arguments
    PieChart(String title, PieDataset dataset) {
        _dataset = dataset;
        _title = title;
    }

    @Override
    public void generate() {
        // Creates Pie Chart from variables
        JFreeChart chart = ChartFactory.createPieChart(_title, // chart title
                _dataset, // data
                false, // include legend
                false, false);

        // Set Plot canvas color and direction
        PiePlot plot = (PiePlot) chart.getPlot();
        plot.setDirection(Rotation.CLOCKWISE);
        plot.setForegroundAlpha(0.5f);
        plot.setBackgroundPaint(Color.WHITE);

        /*
        *  Use these to edit the decimal percentage: 
        *       NumberFormat nf = NumberFormat.getPercentInstance();
        *       nf.setRoundingMode(RoundingMode.);
        *       nf.setMinimumFractionDigits(2);
        *       plot.setLabelGenerator(new StandardPieSectionLabelGenerator("{0} {2}", NumberFormat.getNumberInstance(), nf));
        *  
        *  And remove line 63
        */

        plot.setLabelGenerator(new StandardPieSectionLabelGenerator("{0} {2}"));
        plot.setOutlinePaint(Color.white);

        _chart = chart;
    }

    @Override
    public void saveAsImage(String path, int width, int height) throws IOException {
        // Save image as PNG
        ChartUtilities.saveChartAsPNG(new File(path), _chart, width, height);
    }

    @Override
    public void saveAsSVG(String path, int width, int height) throws IOException {
        // Save image as SVG
        SVGGraphics2D g2 = new SVGGraphics2D(width, height);
        Rectangle r = new Rectangle(0, 0, width, height);
        _chart.draw(g2, r);
        File f = new File(path);
        SVGUtils.writeToSVG(f, g2.getSVGElement());
    }
}