/**
 *    AbstractValueAxis.java
 *    Copyright (C) 2010 New Zealand Digital Library, http://expeditee.org
 *
 *    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 org.expeditee.items.widgets.charts;

import java.awt.BasicStroke;
import java.awt.Stroke;

import org.expeditee.core.Colour;
import org.expeditee.gio.swing.SwingConversions;
import org.expeditee.gui.Frame;
import org.expeditee.items.Text;
import org.jfree.chart.axis.Axis;
import org.jfree.chart.plot.Plot;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.chart.plot.ValueAxisPlot2;

public abstract class AbstractValueAxis extends AbstractChart {

	public AbstractValueAxis(Text source, String[] args) {
		super(source, args);
	}

	/**
	 * @param dataFrame
	 */
	protected void refreshPlot(Frame dataFrame, Plot simplePlot) {
		ValueAxisPlot2 plot = (ValueAxisPlot2) simplePlot;
		if (dataFrame.hasAnnotation("xaxis")) {
			plot.getDomainAxis()
					.setLabel(dataFrame.getAnnotationValue("xaxis"));
		}
		if (dataFrame.hasAnnotation("yaxis")) {
			plot.getRangeAxis().setLabel(dataFrame.getAnnotationValue("yaxis"));
		}
		if (dataFrame.hasAnnotation("horizontal")) {
			plot.setOrientation(PlotOrientation.HORIZONTAL);
		} else {
			plot.setOrientation(PlotOrientation.VERTICAL);
		}
	}

	@Override
	public void setSourceColor(Colour c) {
		super.setSourceColor(c);
		if (getChart() == null)
			return;

		ValueAxisPlot2 plot = ((ValueAxisPlot2) getChart().getPlot());
		if (c == null) {
			plot.getDomainAxis().setLabelPaint(Axis.DEFAULT_AXIS_LABEL_PAINT);
			plot.getRangeAxis().setLabelPaint(Axis.DEFAULT_AXIS_LABEL_PAINT);
			plot.getDomainAxis().setTickLabelPaint(
					Axis.DEFAULT_TICK_LABEL_PAINT);
			plot.getRangeAxis()
					.setTickLabelPaint(Axis.DEFAULT_TICK_LABEL_PAINT);
		} else {
			plot.getDomainAxis().setLabelPaint(SwingConversions.toSwingColor(c));
			plot.getRangeAxis().setLabelPaint(SwingConversions.toSwingColor(c));
			plot.getDomainAxis().setTickLabelPaint(SwingConversions.toSwingColor(c));
			plot.getRangeAxis().setTickLabelPaint(SwingConversions.toSwingColor(c));
		}
	}

	@Override
	public void setSourceBorderColor(Colour c) {
		super.setSourceBorderColor(c);
		if (getChart() == null)
			return;
		ValueAxisPlot2 plot = ((ValueAxisPlot2) getChart().getPlot());
		if (c == null) {
			plot.setDomainGridlinePaint(Plot.DEFAULT_OUTLINE_PAINT);
			plot.setRangeGridlinePaint(Plot.DEFAULT_OUTLINE_PAINT);
			plot.getDomainAxis().setTickMarkPaint(Plot.DEFAULT_OUTLINE_PAINT);
			plot.getRangeAxis().setTickMarkPaint(Plot.DEFAULT_OUTLINE_PAINT);
		} else {
			Colour brighter = c.brighter();
			plot.setDomainGridlinePaint(SwingConversions.toSwingColor(brighter));
			plot.setRangeGridlinePaint(SwingConversions.toSwingColor(brighter));
			plot.getDomainAxis().setTickMarkPaint(SwingConversions.toSwingColor(c));
			plot.getRangeAxis().setTickMarkPaint(SwingConversions.toSwingColor(c));
		}
	}

	@Override
	public void setSourceThickness(float newThickness, boolean setConnected) {
		super.setSourceThickness(newThickness, setConnected);
		if (getChart() == null)
			return;
		ValueAxisPlot2 plot = ((ValueAxisPlot2) getChart().getPlot());
		Stroke solid = new BasicStroke(newThickness);
		Stroke gridline = new BasicStroke(newThickness / 2,
				BasicStroke.CAP_BUTT, BasicStroke.JOIN_BEVEL, 0.0f,
				new float[] { 2 * newThickness, 2 * newThickness }, 0.0f);
		
		updateAxis(plot.getDomainAxis(), solid, newThickness);
		updateAxis(plot.getRangeAxis(), solid, newThickness);

		plot.setDomainGridlineStroke(gridline);
		plot.setRangeGridlineStroke(gridline);
		plot.getRenderer().setBaseStroke(solid);

	}

	private void updateAxis(Axis axis, Stroke solid, float thickness) {
		axis.setAxisLineStroke(solid);
		axis.setTickMarkStroke(solid);
		axis.setLabelFont(axis.getLabelFont().deriveFont(getFontSize(thickness, Axis.DEFAULT_AXIS_LABEL_FONT.getSize2D())));
		axis.setTickLabelFont(axis.getTickLabelFont().deriveFont(getFontSize(thickness, Axis.DEFAULT_TICK_LABEL_FONT.getSize2D())));
	}
}