/*
 *  RapidMiner
 *
 *  Copyright (C) 2001-2014 by RapidMiner and the contributors
 *
 *  Complete list of developers available at our web site:
 *
 *       http://rapidminer.com
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU Affero 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 Affero General Public License for more details.
 *
 *  You should have received a copy of the GNU Affero General Public License
 *  along with this program.  If not, see http://www.gnu.org/licenses/.
 */
package com.rapidminer.gui.plotter;

import java.awt.Color;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.geom.Rectangle2D;
import java.util.Iterator;

import com.rapidminer.datatable.DataTable;
import com.rapidminer.gui.plotter.conditions.ColumnsPlotterCondition;
import com.rapidminer.gui.plotter.conditions.PlotterCondition;


/**
 * <code>GridViz</code> is a simple extension of <code>RadViz</code> that
 * places the dimensional anchors on a rectangular grid instead of using the
 * perimeter of a circle. The number of dimensions that can be displayed
 * increases significantly.
 * 
 * @author Daniel Hakenjos, Ingo Mierswa
 */
public class GridVizPlotter extends RadVizPlotter {

	private static final long serialVersionUID = 9178351977037267613L;
	
    private static final int MAX_NUMBER_OF_COLUMNS = 10000;
    
	/** Constructs a new GridVizPlotter. */
	public GridVizPlotter(PlotterConfigurationModel settings) {
		super(settings);
	}
	
	/** Constructs a new GridVizPlotter. */
	public GridVizPlotter(PlotterConfigurationModel settings, DataTable dataTable) {
		super(settings);
		setDataTable(dataTable);
	}
	
    @Override
	public PlotterCondition getPlotterCondition() {
        return new ColumnsPlotterCondition(MAX_NUMBER_OF_COLUMNS);
    }
    
	protected void calculateAttributeVectors(int totalSize, int gridSize, int gridDelta) {
		int dim = this.dataTable.getNumberOfColumns();
		anchorVectorX = new double[dim];
		anchorVectorY = new double[dim];

		int gridX = 0;
		int gridY = totalSize;
		int counter = 0;
		for (int i = 0; i < this.dataTable.getNumberOfColumns(); i++) {
			if ((i == colorColumn) || (shouldIgnoreColumn(i)))
				continue;
			if (counter % gridSize == 0) {
				gridX = 0;
				gridY -= gridDelta;
			}
			anchorVectorX[i] = gridX;
			anchorVectorY[i] = gridY;

			gridX += gridDelta;
			counter++;
		}
	}

	@Override
	protected void paintPlotter(Graphics graphics) {
        Graphics2D g = (Graphics2D)graphics.create();
		int width = getWidth();
		int height = getHeight();

		g.setColor(Color.BLACK);
		int totalSize = Math.min(width, height) - 2 * MARGIN;
		int numberOfColumns = this.dataTable.getNumberOfColumns();
		if (colorColumn >= 0)
			numberOfColumns--;
		numberOfColumns -= ignoreList.getSelectedIndices().length;
		if (numberOfColumns == 0) return;
		
		int gridSize = (int)Math.ceil(Math.sqrt(numberOfColumns));
		int gridDelta = totalSize / gridSize;
		
		calculateAttributeVectors(totalSize, gridSize, gridDelta);
        
		// draw the axis and calculate the attribute vectors
		g.setFont(LABEL_FONT);
		for (int i = 0; i < this.dataTable.getNumberOfColumns(); i++) {
			if ((i == colorColumn) || (shouldIgnoreColumn(i))) continue;
			int x = (int)(MARGIN + anchorVectorX[i]);
			int y = (int)(MARGIN + totalSize - anchorVectorY[i]);
            // draw weights
            if (this.dataTable.isSupportingColumnWeights()) {
                g.setColor(getWeightColor(this.dataTable.getColumnWeight(columnMapping[i]), this.maxWeight));
                Rectangle2D weightRect = new Rectangle2D.Double(x, y - gridDelta, gridDelta, gridDelta);
                g.fill(weightRect);
            }
            // draw lines
            g.setColor(GRID_COLOR);
			g.drawLine(x, y, x + gridDelta, y);
			g.drawLine(x, y, x, y - gridDelta);
			// draw axis-labels
			g.drawString(this.dataTable.getColumnName(columnMapping[i]), x + 5, y - 5);
		}

		// draw points
		calculateSamplePoints();
		int centerPoint = totalSize / 2 + MARGIN;
		Iterator<PlotterPoint> i = plotterPoints.iterator();
		ColorProvider colorProvider = getColorProvider();
		while (i.hasNext()) {
			drawPoint(g, i.next(), colorProvider, centerPoint, centerPoint, 1.0d);
		}
		
		// legend
		if ((colorColumn != -1) && (plotterPoints.size() > 0)) {
			drawLegend(g, dataTable, colorColumn);
		}
	}
}