e3fraud.gui.GraphingTool.java Source code

Introduction

Here is the source code for e3fraud.gui.GraphingTool.java

Source

/*
 * Copyright (C) 2015 Dan Ionita 
 *
 * 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 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package e3fraud.gui;

import com.hp.hpl.jena.rdf.model.Resource;
import java.io.File;
import java.io.IOException;
import java.util.Set;
import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartUtilities;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.data.xy.XYSeriesCollection;
import e3fraud.model.E3Model;
import e3fraud.vocabulary.E3value;
import java.text.DecimalFormat;
import java.util.Map;
import org.jfree.data.xy.XYSeries;

/**
 *
 * @author Dan
 */
public class GraphingTool {

    public static JFreeChart generateGraph(E3Model model, Resource need, int startValue, int endValue,
            boolean expected) {
        //Get list of actors
        Set<Resource> actors = model.getActors();
        //generate a series
        Map<Resource, XYSeries> actorSeriesMap = model.getTotalForActors(need, startValue, endValue, expected);

        //for each actor
        XYSeriesCollection line_chart_dataset = new XYSeriesCollection();

        for (Resource actor : actors) {
            //add it's series to the chart
            XYSeries series = actorSeriesMap.get(actor);
            line_chart_dataset.addSeries(series);
            double slope;
            if (series.getItemCount() > 1) {
                slope = (series.getY(0).doubleValue() - series.getY(1).doubleValue())
                        / (series.getX(0).doubleValue() - series.getX(1).doubleValue());
            } else {
                slope = 0;
            }
            DecimalFormat df = new DecimalFormat("#.##");
            series.setKey(series.getKey() + "\nAvg.\t = \t" + df.format(model.getLastKnownAverages().get(actor))
                    + "\nSlope\t = \t" + df.format(slope));
        }

        /* Step -2:Define the JFreeChart object to create line chart */
        JFreeChart lineChartObject;
        if (expected) {
            lineChartObject = ChartFactory.createScatterPlot(
                    "(Ideal) Profit Vs Occurences of \"" + need.getProperty(E3value.e3_has_name).getString()
                            + " \"",
                    "Occurences of \"" + need.getProperty(E3value.e3_has_name).getString() + " \"",
                    "Profit (in Euro)", line_chart_dataset, PlotOrientation.VERTICAL, true, true, false);
        } else {
            lineChartObject = ChartFactory.createScatterPlot(
                    "(Non-ideal) Profit Vs Occurences of \"" + need.getProperty(E3value.e3_has_name).getString()
                            + " \"",
                    "Occurences of \"" + need.getProperty(E3value.e3_has_name).getString() + " \"",
                    "Profit (in Euro)", line_chart_dataset, PlotOrientation.VERTICAL, true, true, false);
        }
        return lineChartObject;
    }

    public static void saveToFile(File file, JFreeChart lineChartObject) throws IOException {
        int width = 1024; /* Width of the image */

        int height = 720; /* Height of the image */

        ChartUtilities.saveChartAsPNG(file, lineChartObject, width, height);
    }
}