GenericScatterGraph.java

/* This file is part of Wattzap Community Edition.
 *
 * Wattzap Community Edtion 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.
 *
 * Wattzap Community Edition 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 Wattzap.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.wattzap.view.graphs;

import java.awt.BorderLayout;
import java.awt.Color;
import java.awt.Paint;
import java.awt.Shape;

import javax.swing.JPanel;

import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartPanel;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.axis.ValueAxis;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.chart.plot.ValueMarker;
import org.jfree.chart.plot.XYPlot;
import org.jfree.chart.renderer.xy.StandardXYItemRenderer;
import org.jfree.chart.renderer.xy.XYItemRenderer;
import org.jfree.data.xy.IntervalXYDataset;
import org.jfree.data.xy.XYDataset;
import org.jfree.data.xy.XYSeries;
import org.jfree.data.xy.XYSeriesCollection;
import org.jfree.util.ShapeUtilities;

/* 
 * Generic Scatter Graph
 * 
 * @author David George (c) Copyright 2014
 * @date 18 April 2014
 */
public class GenericScatterGraph extends JPanel {
	ValueMarker marker = null;
	XYPlot plot;
	private ChartPanel chartPanel = null;
	int line = 1;

	public GenericScatterGraph(XYSeries series, String xAxis, String yAxis) {
		super();

		XYDataset xyDataset = new XYSeriesCollection(series);

		JFreeChart chart = ChartFactory.createScatterPlot("", // chart title
				xAxis, // x axis label
				yAxis, // y axis label
				null, PlotOrientation.VERTICAL, false, // include legend
				true, // tooltips
				false // urls
				);

		chart.setBackgroundPaint(Color.darkGray);
		plot = chart.getXYPlot();
		plot.setDataset(0, xyDataset);
		plot.setBackgroundPaint(Color.white);
		plot.setDomainGridlinePaint(Color.lightGray);
		plot.setRangeGridlinePaint(Color.lightGray);

		// Shape cross = ShapeUtilities.createDiamond(0.5f);
		Shape cross = ShapeUtilities.createDiagonalCross(0.5f, 0.5f);
		XYItemRenderer renderer = plot.getRenderer();
		renderer.setSeriesPaint(0, new Color(252, 141, 89));
		renderer.setSeriesShape(0, cross);

		ValueAxis domainAxis = plot.getDomainAxis();
		domainAxis.setTickLabelPaint(Color.white);
		domainAxis.setLabelPaint(Color.white);

		ValueAxis rangeAxis = plot.getRangeAxis();
		rangeAxis.setTickLabelPaint(Color.white);
		rangeAxis.setLabelPaint(Color.white);

		chartPanel = new ChartPanel(chart);
		chartPanel.setSize(100, 800);

		setLayout(new BorderLayout());
		add(chartPanel, BorderLayout.CENTER);
		setBackground(Color.black);

		chartPanel.revalidate();
		setVisible(true);
	}

	public void addLine(XYSeries series) {
		// add a second dataset and renderer...
		final XYItemRenderer renderer = new StandardXYItemRenderer(){
			//Stroke regularStroke = new BasicStroke(0.7f);
			public Color color;


			//@Override
			//public Stroke getItemStroke(int row, int column) {
				//return regularStroke;
			//}

			@Override
			public Paint getItemPaint(int row, int column) {
				return color;
			}

			//public Shape lookupLegendShape(int series) {
			//	return new Rectangle(15, 15);
			//}
		};

		renderer.setBasePaint(Color.LIGHT_GRAY);
		plot.setRenderer(line, renderer);
		final IntervalXYDataset cadenceData = new XYSeriesCollection(series);
		plot.setDataset(line, cadenceData);
		line++;
	}

	private static final long serialVersionUID = 1L;
}