Laboratoire ICAR » Plateforme TXM

package org.txm.ca.core.chartsengine.jfreechart.themes.highcharts.chartcreators;

import org.jfree.chart.JFreeChart;

import org.jfree.chart.axis.NumberAxis;

import org.jfree.data.category.DefaultCategoryDataset;

import org.jfree.data.general.DatasetUtilities;

import org.txm.ca.core.functions.CA;

import org.txm.ca.core.messages.CACoreMessages;

import org.txm.chartsengine.jfreechart.core.JFCChartCreator;

import org.txm.chartsengine.jfreechart.core.renderers.interfaces.IItemSelectionRenderer;

import org.txm.chartsengine.jfreechart.core.themes.base.ExtendedNumberAxis;

import org.txm.stat.StatException;

import org.txm.utils.logger.Log;

/**

 * JFC progression chart creator.

 * 

 * @author sjacquot

 *

 */

public class JFCSingularValueChartCreator extends JFCChartCreator {

	public JFCSingularValueChartCreator() {

		// TODO Auto-generated constructor stub

	}

	@Override

	public Object createChart(Object resultData, String preferencesNode) {

		CA ca = (CA) resultData;

		JFreeChart chart = null;

		try {

			// Creating the data set from the CA result

			DefaultCategoryDataset dataset = new DefaultCategoryDataset();

			double[] singularValues = ca.getValeursPropres();

			for(int i = 0; i < singularValues.length; i++) {

				dataset.setValue(singularValues[i], CACoreMessages.ChartsEngine_CA_SINGULAR_VALUES_TOOLTIP_Y_LABEL, CACoreMessages.ChartsEngine_CA_SINGULAR_VALUES_X_LABELS_PREFIX + String.valueOf(i + 1));

			}

			// Create the chart

			chart = this.getChartsEngine().createCategoryBarChart(dataset, CACoreMessages.ChartsEngine_CA_SINGULAR_VALUES_TITLE, CACoreMessages.ChartsEngine_CA_SINGULAR_VALUES_X_AXIS_LABEL, CACoreMessages.ChartsEngine_CA_SINGULAR_VALUES_Y_AXIS_LABEL, false, false, false);

			// Custom renderer

			chart.getCategoryPlot().setRenderer(this.getChartsEngine().getTheme().createCASingularValuesRenderer());

	        ((IItemSelectionRenderer) chart.getCategoryPlot().getRenderer()).getItemsSelector().setResultData(ca);

	        // Custom range axis for ticks drawing options

	        chart.getCategoryPlot().setRangeAxis(new ExtendedNumberAxis((NumberAxis) chart.getCategoryPlot().getRangeAxis(), false, true, 0, DatasetUtilities.findMaximumRangeValue(chart.getCategoryPlot().getDataset()).doubleValue()));

		}

		catch(StatException e) {

			Log.severe(CACoreMessages.ChartsEngine_CA_SINGULAR_VALUES_CANT_CREATE_CHART + e);

		}

		return chart;

	}

	@Override

	public Class getResultDataClass() {

		return CA.class;

	}

}

package org.txm.ca.core.chartsengine.jfreechart.themes.highcharts.chartcreators;

import java.awt.BasicStroke;

import java.awt.Color;

import java.text.DecimalFormat;

import java.util.ArrayList;

import org.jfree.chart.ChartFactory;

import org.jfree.chart.JFreeChart;

import org.jfree.chart.annotations.XYAnnotation;

import org.jfree.chart.annotations.XYLineAnnotation;

import org.jfree.chart.labels.ItemLabelAnchor;

import org.jfree.chart.labels.ItemLabelPosition;

import org.jfree.chart.plot.PlotOrientation;

import org.jfree.chart.plot.XYPlot;

import org.jfree.chart.renderer.xy.XYItemRenderer;

import org.jfree.chart.title.LegendTitle;

import org.jfree.ui.TextAnchor;

import org.txm.ca.core.chartsengine.base.CAChartCreator;

import org.txm.ca.core.chartsengine.base.Utils;

import org.txm.ca.core.chartsengine.jfreechart.datasets.FCAXYDataset;

import org.txm.ca.core.chartsengine.jfreechart.themes.highcharts.renderers.FCAItemSelectionRenderer;

import org.txm.ca.core.functions.CA;

import org.txm.ca.core.messages.CACoreMessages;

import org.txm.ca.core.preferences.CAPreferences;

import org.txm.chartsengine.core.preferences.ChartsEnginePreferences;

import org.txm.chartsengine.jfreechart.core.JFCChartCreator;

import org.txm.chartsengine.jfreechart.core.renderers.MultipleItemsSelector;

import org.txm.chartsengine.jfreechart.core.themes.base.BlockRoundBorder;

import org.txm.core.preferences.TXMPreferences;

import org.txm.stat.StatException;

import org.txm.utils.logger.Log;

/**

 * JFC progression chart creator.

 * 

 * @author sjacquot

 *

 */

public class JFCCAChartCreator extends JFCChartCreator implements CAChartCreator {

	public JFCCAChartCreator() {

		// TODO Auto-generated constructor stub

	}

	/**

	 * Add some borders to the limits of the specified XY plot.

	 * @param plot

	 */

	public void createCAFactorialMapChartLimitsBorder(JFreeChart chart)	{

		XYPlot plot = chart.getXYPlot();

		// Remove all existent annotations

		for(int i = plot.getAnnotations().size() - 1; i >= 0; i--) {

			plot.removeAnnotation((XYAnnotation) plot.getAnnotations().get(i));

		}

        // Gets the extreme coordinates values

		double minX = 0, maxX = 0, minY = 0, maxY = 0;

		// FIXME: new version

//		minX = plot.getDataRange(plot.getDomainAxis()).getLowerBound();

//		maxX = plot.getDataRange(plot.getDomainAxis()).getUpperBound();

//

//		minY = plot.getDataRange(plot.getRangeAxis()).getLowerBound();

//		maxY = plot.getDataRange(plot.getRangeAxis()).getUpperBound();

		// FIXME: old version, to remove when new version will be validated

		// Rows

		if(plot.getRenderer().isSeriesVisible(0))	{

			minX = ((FCAXYDataset)plot.getDataset()).getMinX(0);

		}

		// Cols

		if(plot.getRenderer().isSeriesVisible(1))	{

			double tmpMinX = ((FCAXYDataset)plot.getDataset()).getMinX(1);

			if(tmpMinX < minX)	{

				minX = tmpMinX;

			}

		}

		// Rows

		if(plot.getRenderer().isSeriesVisible(0))	{

			maxX = ((FCAXYDataset)plot.getDataset()).getMaxX(0);

		}

		// Cols

		if(plot.getRenderer().isSeriesVisible(1))	{

			double tmpMaxX = ((FCAXYDataset)plot.getDataset()).getMaxX(1);

			if(tmpMaxX > maxX)	{

				maxX = tmpMaxX;

			}

		}

		// Rows

		if(plot.getRenderer().isSeriesVisible(0))	{

			minY = ((FCAXYDataset)plot.getDataset()).getMinY(0);

		}

		// Cols

		if(plot.getRenderer().isSeriesVisible(1))	{

			double tmpMinY = ((FCAXYDataset)plot.getDataset()).getMinY(1);

			if(tmpMinY < minY)	{

				minY = tmpMinY;

			}

		}

		// Rows

		if(plot.getRenderer().isSeriesVisible(0))	{

			maxY = ((FCAXYDataset)plot.getDataset()).getMaxY(0);

		}

		// Cols

		if(plot.getRenderer().isSeriesVisible(1))	{

			double tmpMaxY = ((FCAXYDataset)plot.getDataset()).getMaxY(1);

			if(tmpMaxY > maxY)	{

				maxY = tmpMaxY;

			}

		}

		// Add some margins to the border

		// FIXME : to compute according to the real series shape dimensions

		// The code below doesn't work

//		double shapeHalfWidth = chart.getXYPlot().getRenderer().getSeriesShape(0).getBounds2D().getWidth() / 2;

		double shapeHalfWidth = 0.02;

		minX -= shapeHalfWidth;

		maxX += shapeHalfWidth;

		minY -= shapeHalfWidth;

		maxY += shapeHalfWidth;

		// Create the border

		BasicStroke dashedStroke = new BasicStroke(0.5f, BasicStroke.CAP_ROUND, BasicStroke.JOIN_ROUND, 1f, new float[] {2f}, 0f);

		XYLineAnnotation annotation;

		Color borderColor = Color.GRAY;

		// Left border

		annotation = new XYLineAnnotation(minX, minY, minX, maxY, dashedStroke, borderColor);

        plot.addAnnotation(annotation);

        // Right border

		annotation = new XYLineAnnotation(maxX, minY, maxX, maxY, dashedStroke, borderColor);

        plot.addAnnotation(annotation);

		// Bottom border

		annotation = new XYLineAnnotation(minX, minY, maxX, minY, dashedStroke, borderColor);

        plot.addAnnotation(annotation);

        // Top border

		annotation = new XYLineAnnotation(minX, maxY, maxX, maxY, dashedStroke, borderColor);

        plot.addAnnotation(annotation);

	}

	@Override

	public void updateChartCAFactorialMapHighlightPoints(Object chart, boolean rows, String[] labels) {

		MultipleItemsSelector selector = (MultipleItemsSelector) ((FCAItemSelectionRenderer)((JFreeChart) chart).getXYPlot().getRenderer()).getItemsSelector();

		// Rows

		int series = 0;

		// Columns

		if(!rows)	{

			series = 1;

		}

		selector.removeAllSelectedItems(series);

		int[] items = ((FCAXYDataset)((JFreeChart) chart).getXYPlot().getDataset()).getLabelIndices(series, labels);

		for(int i = 0; i < items.length; i++) {

			selector.addSelectedItem(series, items[i]);

		}

		//((JFreeChart) chart).setNotify(true);

	}

	@Override

	public void updateChartCAFactorialMapSetDimensions(Object chart, CA ca, int dimension1, int dimension2)	{

		 // Modify data set

		 ((FCAXYDataset) ((JFreeChart) chart).getXYPlot().getDataset()).setAxis1(dimension1);

		 ((FCAXYDataset) ((JFreeChart) chart).getXYPlot().getDataset()).setAxis2(dimension2);

		 // Update axis labels

		 // FIXME : create a method in CA to directly get a singular value as percent ?

		 try {

			 double sinuglarValuesSum = ca.getValeursPropresSum();

			 DecimalFormat f = new DecimalFormat("###.00"); //$NON-NLS-1$

			 ((JFreeChart) chart).getXYPlot().getDomainAxis().setLabel(CACoreMessages.ChartsEngine_CA_FACTORIAL_MAP_AXIS_LABEL_PREFIX + " " + dimension1 + " (" + f.format(100 * ca.getValeursPropres()[dimension1 - 1] / sinuglarValuesSum) + "%)");

			 ((JFreeChart) chart).getXYPlot().getRangeAxis().setLabel(CACoreMessages.ChartsEngine_CA_FACTORIAL_MAP_AXIS_LABEL_PREFIX + " " + dimension2 + " (" + f.format(100 * ca.getValeursPropres()[dimension2 - 1] / sinuglarValuesSum) + "%)");

			 // Refresh data set

			 // FIXME : any way to fire a data set event rather than reassign the same data set?

			 ((JFreeChart) chart).getXYPlot().setDataset(((JFreeChart) chart).getXYPlot().getDataset());

			// the code below doesn't neither center the view nor update the axes ticks of the new chart, continue tests

			//((JFreeChart) chart).setNotify(true);

			// Update the limits border

			this.createCAFactorialMapChartLimitsBorder((JFreeChart) chart);

		}

		catch(StatException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		}

	 }

	@Override

	public void updateChart(Object chart, Object resultData, String preferencesNode) {

		FCAItemSelectionRenderer renderer = (FCAItemSelectionRenderer) ((JFreeChart)chart).getXYPlot().getRenderer(); 

   	 	XYPlot plot = ((JFreeChart)chart).getXYPlot();

		renderer.setSeriesVisible(0, TXMPreferences.getBoolean(preferencesNode, resultData, CAPreferences.SHOW_VARIABLES));

		renderer.setSeriesVisible(1, TXMPreferences.getBoolean(preferencesNode, resultData, CAPreferences.SHOW_INDIVIDUALS));

		// Create the limits border

		this.createCAFactorialMapChartLimitsBorder((JFreeChart) chart);

		this.updateChartCAFactorialMapSetDimensions(chart, (CA) resultData,

				TXMPreferences.getInt(preferencesNode, resultData, CAPreferences.FIRST_DIMENSION),

				TXMPreferences.getInt(preferencesNode, resultData, CAPreferences.SECOND_DIMENSION));

		super.updateChart(chart, resultData, preferencesNode);

    	((FCAItemSelectionRenderer)renderer).setBaseLinesVisible(false);

    	renderer.setBaseOutlinePaint(Color.decode("#666666"));

    	// Labels position

    	ItemLabelPosition position = new ItemLabelPosition(ItemLabelAnchor.OUTSIDE1, TextAnchor.BOTTOM_CENTER, TextAnchor.CENTER, 0.0);

    	renderer.setBasePositiveItemLabelPosition(position);

    	renderer.setBaseNegativeItemLabelPosition(position);

    	// Zero base lines

		plot.setDomainZeroBaselineVisible(true);

		plot.setRangeZeroBaselineVisible(true);

    	// Items colors (Highcharts color codes)

        Color blue = new Color(47, 126, 216, 90);

        renderer.setSeriesPaint(0, blue);

        Color red = new Color(255, 0, 0, 90);

        renderer.setSeriesPaint(1, red);

        // Legends

        LegendTitle legendTitle = ((JFreeChart) chart).getLegend(0);

    	if(legendTitle != null)	{

    		legendTitle.setFrame(new BlockRoundBorder(Color.GRAY));

    	}

	}

	@Override

	public JFreeChart createChart(Object resultData, String preferencesNode) {

		CA ca = (CA) resultData;

		// parameters

//		boolean monochrome = (TXMPreferences.getInt(preferencesNode, resultData, ChartsEnginePreferences.RENDERING_COLORS_MODE, ChartsEnginePreferences.PREFERENCES_NODE)

//				== ChartsEngine.RENDERING_MONOCHROME_MODE);

//		boolean monostyle = TXMPreferences.getBoolean(preferencesNode, resultData, CAPreferences.CHART_MONO_STYLE);

//		boolean cumulative = TXMPreferences.getBoolean(preferencesNode, resultData, CAPreferences.CHART_CUMULATIVE);

		JFreeChart chart  = null;

		try {

			FCAXYDataset dataset = new FCAXYDataset(ca);

//			DecimalFormat f = new DecimalFormat("###.00"); //$NON-NLS-1$

//			double singularValuesSum = ca.getValeursPropresSum();

//

//			String percent1 = f.format(100 * ca.getValeursPropres()[ca.getFirstDimension() - 1] / singularValuesSum);

//			String xAxisLabel = CACoreMessages.ChartsEngine_CA_FACTORIAL_MAP_AXIS_LABEL_PREFIX + " " + ca.getFirstDimension() + " (" + percent1 + "%)";

//			String percent2 = f.format(100 * ca.getValeursPropres()[ca.getSecondDimension() - 1] / singularValuesSum);

//			String yAxisLabel = CACoreMessages.ChartsEngine_CA_FACTORIAL_MAP_AXIS_LABEL_PREFIX  + " " + ca.getSecondDimension() + " (" + percent2 + "%)";

			chart = ChartFactory.createScatterPlot(Utils.createCAFactorialMapChartTitle(ca), "", "", dataset, PlotOrientation.VERTICAL, TXMPreferences.getBoolean(CAPreferences.PREFERENCES_NODE, ChartsEnginePreferences.SHOW_LEGEND), false, false);

			// Custom renderer

			chart.getXYPlot().setRenderer(this.getChartsEngine().getTheme().createFCARenderer());

		}

		catch(StatException e) {

			Log.severe("Can't create CA factorial map scatter plot" + e);

		}

		return chart;		

	}

	@Override

	public ArrayList<String> getCAFactorialMapChartSelectedPoints(Object chart, int series) {

		MultipleItemsSelector selector = (MultipleItemsSelector) ((FCAItemSelectionRenderer)((JFreeChart) chart).getXYPlot().getRenderer()).getItemsSelector();

		ArrayList<String> pointLabels = new ArrayList<String>(selector.getSelectedItemsCount(series));

		if(selector.getSelectedItemsCount(series) > 0)	{

			FCAXYDataset dataset = (FCAXYDataset) ((JFreeChart) chart).getXYPlot().getDataset();

			int[] selectedItems = selector.getSelectedItems(series);

			for(int i = 0; i < selectedItems.length; i++) {

				pointLabels.add(dataset.getLabel(series, selectedItems[i]));

			}

		}

		return pointLabels;

	}

	@Override

	public void updateChartCAFactorialMapSetLabelItemsSelectionOrder(Object chart, String[] rowLabels, String[] colLabels) {

		MultipleItemsSelector selector = (MultipleItemsSelector) ((FCAItemSelectionRenderer)((JFreeChart) chart).getXYPlot().getRenderer()).getItemsSelector();

		selector.setCyclicItemsOrder(selector.getItemsAndSeriesOrderedByLabels((FCAXYDataset)((JFreeChart) chart).getXYPlot().getDataset(), rowLabels, colLabels)); 

	}

	@Override

	public Class getResultDataClass() {

		return CA.class;

	}

}