Chart2D.java

package wf.util;

import java.awt.Color;
import java.awt.Paint;

import org.jfree.chart.JFreeChart;
import org.jfree.chart.axis.AxisLocation;
import org.jfree.chart.axis.NumberAxis;
import org.jfree.chart.axis.NumberTickUnit;
import org.jfree.chart.plot.XYPlot;
import org.jfree.chart.renderer.PaintScale;
import org.jfree.chart.renderer.xy.XYBlockRenderer;
import org.jfree.chart.title.PaintScaleLegend;
import org.jfree.data.xy.XYZDataset;
import org.jfree.ui.RectangleEdge;
import org.jfree.ui.RectangleInsets;

public class Chart2D {

	double xAxisMin;
	double xAxisMax;
	double yAxisMin;
	double yAxisMax;

	private static final Chart2D INSTANCE = new Chart2D();

	private NumberAxis xAxis;
	private NumberAxis yAxis;
	private JFreeChart chart2D;
	private XYPlot plot;

	private XYBlockRenderer r;
	private NumberAxis scaleAxis;
	private PaintScaleLegend legend;

	private SpectrumPaintScale ps;

	public Chart2D() {


		xAxis = new NumberAxis("x Axis");
		xAxis.setRange(0, 100);

		//xAxis.setTickUnit(new NumberTickUnit(0.004));
		xAxis.setTickUnit(new NumberTickUnit(10));

		//xAxis.setUpperBound(0.1);

		yAxis = new NumberAxis("y Axis");

		//the first null is the dataset. Later plot.setData()
		plot = new XYPlot(null, xAxis, yAxis, null);


		//SpectrumPaintScale ps = new SpectrumPaintScale(0, N * N);
		//SpectrumPaintScale ps = new SpectrumPaintScale(0, 12 * 100);
		ps = new SpectrumPaintScale(0, 10);
		r = new XYBlockRenderer();
		r.setPaintScale(ps);
		r.setBlockHeight(.01f);
		r.setBlockWidth(.01f);
		plot.setRenderer(r);
		//chart2D = new JFreeChart("Title",
			//	 JFreeChart.DEFAULT_TITLE_FONT, plot, false);
		chart2D = new JFreeChart(null,
			 JFreeChart.DEFAULT_TITLE_FONT, plot, false);
		chart2D.setPadding(new RectangleInsets(10, 10, 10, 10));

		scaleAxis = new NumberAxis("Scale");
		scaleAxis.setAxisLinePaint(Color.white);
		scaleAxis.setTickMarkPaint(Color.white);
		legend = new PaintScaleLegend(ps, scaleAxis);
		legend.setSubdivisionCount(128);
		legend.setAxisLocation(AxisLocation.TOP_OR_RIGHT);
		legend.setPadding(new RectangleInsets(10, 10, 10, 10));
		legend.setStripWidth(20);
		legend.setPosition(RectangleEdge.RIGHT);
		legend.setBackgroundPaint(Color.WHITE);

		chart2D.addSubtitle(legend);
		chart2D.setBackgroundPaint(Color.white);
	}

	private static class SpectrumPaintScale implements PaintScale {

		private static final float H1 = -0.5f;

		private static final float H2 = (float) (Math.PI / 18);
		// private static final float H2 = 1f;
		//private final double lowerBound;
		//private final double upperBound;
		private double lowerBound;
		private double upperBound;

	        public SpectrumPaintScale(double lowerBound, double upperBound) {
	            this.lowerBound = lowerBound;
	            this.upperBound = upperBound;
	        }

	        @Override
	        public double getLowerBound() {
	            return lowerBound;
	        }

	        @Override
	        public double getUpperBound() {
	            return upperBound;
	        }

	        @Override
	        public Paint getPaint(double value) {
	            float scaledValue = (float) (value / (getUpperBound() - getLowerBound()));
	            float scaledH = H1 + scaledValue * (H2 - H1);
	            return Color.getHSBColor(scaledH, 1f, 1f);
	        }
	 }

	public void setXaxisMin(double value) {

		xAxis.setLowerBound(value);
	}

	public void setXaxisMax(double value) {

		xAxis.setUpperBound(value);
	}

	public void setXaxisTickUnit(double value) {

		xAxis.setTickUnit(new NumberTickUnit(value));
	}

	public void setYaxisMin(double value) {

		yAxis.setLowerBound(value);
	}

	public void setYaxisMax(double value) {

		yAxis.setUpperBound(value);
	}

	public void setYaxisTickUnit(double value) {

		yAxis.setTickUnit(new NumberTickUnit(value));
	}

	public void setXaxisTitle(String title) {

		xAxis.setLabel(title);
	}

	public void setYaxisTitle(String title) {

		yAxis.setLabel(title);
	}

	public JFreeChart getChart2D() {

		 return chart2D;
	}

	public void clearPlotData() {

		plot.setDataset(null);


	}

	public void updateLegend(double value, double stepX, double stepY) {

		//ps.setUpperBound(value);

		//r = new XYBlockRenderer();

		chart2D.removeSubtitle(legend);
		chart2D.removeLegend();


		ps = new SpectrumPaintScale(0, value);

		r.setPaintScale(ps);
		r.setBlockHeight(stepY);
		r.setBlockWidth(stepX);
		plot.setRenderer(r);
		//chart2D = new JFreeChart("Title",
					//	 JFreeChart.DEFAULT_TITLE_FONT, plot, false);
		//		chart2D = new JFreeChart(null,
			//		 JFreeChart.DEFAULT_TITLE_FONT, plot, false);
		//		chart2D.setPadding(new RectangleInsets(10, 10, 10, 10));

		//		scaleAxis = new NumberAxis("Scale");
			//	scaleAxis.setAxisLinePaint(Color.white);
				//scaleAxis.setTickMarkPaint(Color.white);



		legend = new PaintScaleLegend(ps, scaleAxis);
		legend.setSubdivisionCount(128);
		legend.setAxisLocation(AxisLocation.TOP_OR_RIGHT);
		legend.setPadding(new RectangleInsets(10, 10, 10, 10));
		legend.setStripWidth(20);
		legend.setPosition(RectangleEdge.RIGHT);
		legend.setBackgroundPaint(Color.WHITE);

		//		chart2D.removeLegend();
		chart2D.addSubtitle(legend);
		//chart2D.setBackgroundPaint(Color.white);


	}

	public void setAxisTickUnit(double valueX, double valueY) {

		xAxis.setTickUnit(new NumberTickUnit(valueX));
		yAxis.setTickUnit(new NumberTickUnit(valueY));
	}

	public void setXYZdataset(XYZDataset dataset) {

		plot.setDataset(dataset);
	}

	public static Chart2D getInstance() {

		return INSTANCE;
	}
}