com.itemanalysis.jmetrik.graph.scatterplot.ScatterplotPanel.java Source code

Introduction

Here is the source code for com.itemanalysis.jmetrik.graph.scatterplot.ScatterplotPanel.java

Source

/*
 * Copyright (c) 2012 Patrick Meyer
 *
 * 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 com.itemanalysis.jmetrik.graph.scatterplot;

import java.awt.*;

import com.itemanalysis.jmetrik.swing.GraphPanel;
import org.apache.log4j.Logger;
import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartPanel;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.chart.plot.XYPlot;
import org.jfree.chart.title.TextTitle;
import org.jfree.data.xy.DefaultXYDataset;
import org.jfree.data.xy.XYSeriesCollection;
import org.jfree.ui.RectangleInsets;

public class ScatterplotPanel extends GraphPanel {

    private ScatterplotCommand command = null;
    private JFreeChart chart = null;
    private String title = "";
    private String subtitle = "";
    private String xlabel = "";
    private String ylabel = "";
    static Logger logger = Logger.getLogger("jmetrik-logger");

    public ScatterplotPanel(ScatterplotCommand command) {
        this.command = command;
        processCommand();
    }

    public void updateDataset(XYSeriesCollection dataset) {
        if (dataset.getSeriesCount() == 1 || !showLegend) {
            chart.removeLegend();
        }
        XYPlot plot = (XYPlot) chart.getPlot();
        plot.setDataset(dataset);
        setXYPlotRenderer(plot);
    }

    public void updateDataset(DefaultXYDataset dataset) {
        if (dataset.getSeriesCount() == 1 || !showLegend) {
            chart.removeLegend();
        }
        XYPlot plot = (XYPlot) chart.getPlot();
        plot.setDataset(dataset);
        setXYPlotRenderer(plot);
    }

    private void processCommand() {
        try {
            title = command.getFreeOption("title").getString();
            subtitle = command.getFreeOption("subtitle").getString();
            xlabel = command.getFreeOption("xvar").getString();
            ylabel = command.getFreeOption("yvar").getString();
            showLegend = command.getSelectAllOption("options").isArgumentSelected("legend");
            showMarkers = true;//must be true otherwise chart will show not points
            width = command.getPairedOptionList("dimensions").getIntegerAt("width");
            height = command.getPairedOptionList("dimensions").getIntegerAt("height");
        } catch (IllegalArgumentException ex) {
            logger.fatal(ex.getMessage(), ex);
            this.firePropertyChange("error", "", "Error - Check log for details.");
        }
    }

    public void setGraph() {
        DefaultXYDataset dataset = new DefaultXYDataset();
        PlotOrientation orientation = PlotOrientation.VERTICAL;

        try {
            chart = ChartFactory.createScatterPlot(title, // chart title
                    xlabel, // x axis label
                    ylabel, // y axis label
                    dataset, // data
                    orientation, showLegend, // include legend
                    true, // tooltips
                    false // urls
            );

            if (subtitle != null && !"".equals(subtitle)) {
                TextTitle subtitle1 = new TextTitle(subtitle);
                chart.addSubtitle(subtitle1);
            }

            XYPlot plot = (XYPlot) chart.getPlot();
            plot.setNoDataMessage("NO DATA");
            plot.setDomainZeroBaselineVisible(false);
            plot.setRangeZeroBaselineVisible(false);
            plot.setBackgroundPaint(Color.WHITE);
            plot.setRangeGridlinePaint(Color.LIGHT_GRAY);
            plot.setDomainGridlinePaint(Color.LIGHT_GRAY);

            ChartPanel panel = new ChartPanel(chart);
            panel.getPopupMenu().addSeparator();
            this.addJpgMenuItem(this, panel.getPopupMenu());
            panel.setPreferredSize(new Dimension(width, height));

            chart.setPadding(new RectangleInsets(20.0, 5.0, 20.0, 5.0));
            this.setBackground(Color.WHITE);
            this.add(panel);

        } catch (IllegalArgumentException ex) {
            logger.fatal(ex.getMessage(), ex);
            this.firePropertyChange("error", "", "Error - Check log for details.");
        }
    }

}