Create Line Chart - Java 2D Graphics

Description

Create Line Chart

Demo Code

import org.jfree.chart.ChartFactory;
import org.jfree.chart.JFreeChart;
import org.jfree.data.category.DefaultCategoryDataset;
import org.jfree.chart.plot.PlotOrientation;
import java.io.File;
import javax.swing.JOptionPane;

/**/*w w  w .j a va2  s .  c om*/
 * @author   D. Sherratt
 * @file   LineChart.java
 * @date  29.03.2013
 * @version     1.0
 * @see         Wicked Cool Java  By: Brian D. Eubanks, chapter 6 -creating jfreecharts
 * 
 * \brief LineChart class that creates a Line Chart
 */
public class LineChart extends Visualization {

    /**
     *  Constructor used requiring a dataset
     *
     *  @param the dataset that the line chart will represent
     */
    public LineChart(DataSet inputdataset) {
        super.setDataset(inputdataset);
    }

    /**
     *  Method that creates the line chart
     *
     *  @return the chart for display
     */
    public JFreeChart MakeChart() {

        DefaultCategoryDataset lineDataset = new DefaultCategoryDataset();

        Object[] column1Data = super.getDataset().GetColumnData(
                super.getAttribute1());
        Object[] column2Data = super.getDataset().GetColumnData(
                super.getAttribute2());

        String column2Header = super.getDataset().GetAttributeName(
                super.getAttribute2());

        boolean addDataset = true;
        String errors = "";
        int errorCounter = 0;

        for (int i = 0; i < super.getDataset().GetNoOfEntrys(); i++) {

            Comparable<Object> nextValue1;
            nextValue1 = (Comparable<Object>) column1Data[i];

            Double nextValue2 = null;

            boolean addThis = true;

            try {
                int intNextValue2 = Integer.parseInt(column2Data[i]
                        .toString());
                nextValue2 = (Double) ((double) intNextValue2);
            } catch (NumberFormatException nfe) {
                try {
                    double doubleNextValue2 = Double
                            .parseDouble(column2Data[i].toString());
                    nextValue2 = (Double) doubleNextValue2;
                } catch (NumberFormatException nfe2) {
                    String strNextValue2 = column2Data[i].toString();
                    if (strNextValue2.equalsIgnoreCase("True")) {
                        nextValue2 = TRUE;
                    } else if (strNextValue2.equalsIgnoreCase("False")) {
                        nextValue2 = FALSE;
                    } else {
                        addThis = false;
                    }
                }
            } catch (Exception e) {
                addThis = false;
            }

            if (addThis == true) {
                lineDataset.addValue(nextValue2, column2Header, nextValue1);
            } else {
                addDataset = false;
                if (errorCounter < MAX_ERROR_LENGTH) {
                    errors = errors + "\n" + column2Data[i].toString();
                    errorCounter++;
                }
            }
        }

        if (addDataset == false) {
            lineDataset = new DefaultCategoryDataset(); //Reset
            JOptionPane
                    .showMessageDialog(
                            null,
                            "Your selected y-axis has data in the wrong format"
                                    + "\n"
                                    + "The following data needs to be a number in order to be"
                                    + " represented." + errors);
        }

        JFreeChart chart = ChartFactory.createLineChart(super.getHeader(),
                super.getxAxis(), super.getyAxis(), lineDataset,
                PlotOrientation.VERTICAL, true, //include legend
                true, false);
        return chart;
    }

    private final Double TRUE = 1.0;
    private final Double FALSE = 0.0;
    private final int MAX_ERROR_LENGTH = 5;

    /**
     * Unit test used to test LineChart class.
     */
    public static void main(String args[]) {
        DataSet testdataset = new DataSet();
        File inputfile = new File("test.csv");
        testdataset.BuildDataSet(inputfile);

        System.out.println("LineChart:: LineChart()");
        LineChart testLineChart = new LineChart(testdataset);
        System.out.println("LineChart::LineChart()- Test Passed");

        System.out.println("LineChart:: MakeChart()");
        testLineChart.MakeChart();
        System.out.println("LineChart::MakeChart()- Test Passed");
    }
}

Visualization.java

Demo Code

import java.io.File;

/**/*  w  w  w .j  a v  a2s. c  o m*/
 * @file    Visualization.java
 * @author  Joshua Reynolds
 * @date    20 Feb 2013
 * @see     
 *
 *  \brief Visualization that stores selected attirbutes and data
 */

public class Visualization {

    /*
     * Method that gets the first attribute of the visualization
     *
     * @return the first attribute
     */
    public int getAttribute1() {
        return m_attribute1;
    }

    /*
     * Method that gets the second attribute of the visualization
     *
     * @return the second attribute
     */
    public int getAttribute2() {
        return m_attribute2;
    }

    /*
     * Method that gets the dataset of the visualization
     *
     * @return the dataset
     */
    public DataSet getDataset() {
        return m_dataset;
    }

    /*
     * Method that gets the header of the visualization
     *
     * @return the header of the visualization
     */
    public String getHeader() {
        return m_header;
    }

    /*
     * Method that gets the x axis of the visualization
     *
     * @eturn the x axis label
     */
    public String getxAxis() {
        return m_xAxis;
    }

    /*
     * Method that gets the y axis of the visualization
     *
     * @return the y axis label
     */
    public String getyAxis() {
        return m_yAxis;
    }

    /*
     * Method that allows you to set the first attribute of the visualization
     *
     * @param attribute1-the index of the attribute
     *
     * @return true if first attribute has been set
     */
    public boolean setAttribute1(int attribute1) {
        m_attribute1 = attribute1;
        return true;
    }

    /*
     * Method that allows you to set the second attribute of the visualization
     *
     * @param attribute2-the index of the attribute
     *
     * @return true if second attribute has been set
     */
    public boolean setAttribute2(int attribute2) {
        m_attribute2 = attribute2;
        return true;
    }

    /*
     * Method that allows you to set the dataset of the visualization
     *
     * @param dataset
     *
     * @return true if dataset has been set
     */
    public boolean setDataset(DataSet dataset) {
        m_dataset = dataset;
        return true;
    }

    /*
     * Method that allows you to set the header of the visualization
     *
     * @param header-the header of the visualization
     *
     * @return true if the header has been set
     */
    public boolean setHeader(String header) {
        m_header = header;
        return true;
    }

    /*
     * Method that allows you to set the x axis label of the visualization
     *
     * @param xAxis-the label for the x axis
     *
     * @return true if the x axis has been set
     */
    public boolean setxAxis(String xAxis) {
        m_xAxis = xAxis;
        return true;
    }

    /*
     * Method that allows you to set the y axis label of the visualization
     *
     * @param yAxis-the label for the y axis
     *
     * @return true if the y axis has been set
     */
    public boolean setyAxis(String yAxis) {
        m_yAxis = yAxis;
        return true;
    }

    private DataSet m_dataset;
    private int m_attribute1;
    private int m_attribute2;
    private String m_header;
    private String m_xAxis;
    private String m_yAxis;

    public static void main(String args[]) {

        final int TEST_DATA = 100;

        //test create visualization class
        System.out.println("Visualization:: Visualization()");
        Visualization testVisualization = new Visualization();
        System.out.println("Visualization:: Visualization() - Test Passed");

        //test DataSet class
        DataSet testdataset = new DataSet();
        File inputfile = new File("test.csv");
        testdataset.BuildDataSet(inputfile);

        //test setDataset method
        System.out.println("Visualization:: setDataset()");
        testVisualization.setDataset(testdataset);
        System.out.println("Visualization:: setDataset() - Test Passed");

        //test setAttribute1 method
        System.out.println("Visualization:: setAttribute1()");
        boolean testSetAttribute1 = testVisualization
                .setAttribute1(TEST_DATA);
        System.out.println("Visualization:: setAttribute1() - Test Passed");

        //test setAttribute2 method
        System.out.println("Visualization:: setAttribute2()");
        boolean testSetAttribute2 = testVisualization
                .setAttribute2(TEST_DATA);
        System.out.println("Visualization:: setAttribute2() - Test Passed");

        //test setHeader method
        System.out.println("Visualization:: setHeader()");
        boolean testSetHeader = testVisualization.setHeader("Visulization");
        System.out.println("Visualization:: setHeader() - Test Passed");

        //test setxAxis method
        System.out.println("Visualization:: setxAxis()");
        boolean testSetxAxis = testVisualization.setxAxis("xAxis");
        System.out.println("Visualization:: setxAxis() - Test Passed");

        //test setyAxis method
        System.out.println("Visualization:: setyAxis()");
        boolean testSetyAxis = testVisualization.setyAxis("yAxis");
        System.out.println("Visualization:: setyAxis() - Test Passed");

        //test getDataset method
        System.out.println("Visualization:: getDataset()");
        testVisualization.getDataset();
        System.out.println("Visualization:: getDataset() - Test Passed");

        //test getAttribute1 method
        System.out.println("Visualization:: getAttribute1()");
        int testGetAttribute1 = testVisualization.getAttribute1();
        System.out.println("Visualization:: getAttribute1() - Test Passed");

        //test getAttribute2 method
        System.out.println("Visualization:: getAttribute2()");
        int testGetAttribute2 = testVisualization.getAttribute2();
        System.out.println("Visualization:: getAttribute2() - Test Passed");

        //test getHeader method
        System.out.println("Visualization:: getHeader()");
        String testGetHeader = testVisualization.getHeader();
        System.out.println("Visualization:: getHeader() - Test Passed");

        //test getxAxis method
        System.out.println("Visualization:: getxAxis()");
        String testGetxAxis = testVisualization.getxAxis();
        System.out.println("Visualization:: getxAxis() - Test Passed");

        //test getyAxis method
        System.out.println("Visualization:: getyAxis()");
        String testGetyAxis = testVisualization.getyAxis();
        System.out.println("Visualization:: getyAxis() - Test Passed");
    }

} /* End of Visualization Class */

Create Line Chart - Java 2D Graphics

Description

Demo Code

Demo Code

Related Tutorials