/* ===========================================================
 * JFreeChart : a free chart library for the Java(tm) platform
 * ===========================================================
 *
 * (C) Copyright 2000-2013, by Object Refinery Limited and Contributors.
 *
 * Project Info:  http://www.jfree.org/jfreechart/index.html
 *
 * This library is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation; either version 2.1 of the License, or
 * (at your option) any later version.
 *
 * This library 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 Lesser General Public
 * License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301,
 * USA.
 *
 * [Java is a trademark or registered trademark of Sun Microsystems, Inc.
 * in the United States and other countries.]
 *
 * --------------------------------------
 * RectangularRegionSelectionHandler.java
 * --------------------------------------
 * (C) Copyright 2009-2013, by Object Refinery Limited and Contributors.
 *
 * Original Author:  David Gilbert (for Object Refinery Limited);
 * Contributor(s):   Michael Zinsmaier;
 *
 * Changes:
 * --------
 * 19-Jun-2009 : Version 1 (DG);
 *
 */

package org.jfree.chart.panel.selectionhandler;

import org.jfree.chart.ChartPanel;
import org.jfree.chart.util.ShapeUtilities;

import java.awt.*;
import java.awt.event.MouseEvent;
import java.awt.geom.Point2D;
import java.awt.geom.Rectangle2D;

/**
 * A mouse handler that allows data items to be selected. The selection shape
 * is a rectangle that can be expanded by dragging the mouse away from the starting
 * point. 
 * 
 * Will only work together with a ChartPanel as event source
 * @author zinsmaie
 */
public class RectangularHeightRegionSelectionHandler extends RegionSelectionHandler {

    /** a generated serial id. */
    private static final long serialVersionUID = -8496935828054326324L;

    /**
     * The selection rectangle (in Java2D space).
     */
    private Rectangle selectionRect;

    public Point2D getStartPoint() {
        return startPoint;
    }

    /**
     * The last mouse point.
     */
    private Point2D startPoint;

    /**
     * Creates a new default instance.
     */
    public RectangularHeightRegionSelectionHandler() {
        super();
        this.selectionRect = null;
        this.startPoint = null;
    }

    /**
     * Creates a new instance with a modifier restriction
     * @param modifier e.g. shift has to be pressed InputEvents.SHIFT_MASK
     */
    public RectangularHeightRegionSelectionHandler(int modifier) {
        super(modifier);
        this.selectionRect = null;
        this.startPoint = null;
    }

    /**
     * Creates a new selection handler with the specified attributes.
     *
     * @param outlineStroke  the outline stroke.
     * @param outlinePaint  the outline paint.
     * @param fillPaint  the fill paint.
     */
    public RectangularHeightRegionSelectionHandler(Stroke outlineStroke,
                                                   Paint outlinePaint, Paint fillPaint) {
        super(outlineStroke, outlinePaint, fillPaint);
        this.selectionRect = null;
        this.startPoint = null;
    }

    /**
     * starts the rectangle selection by fixing the left upper corner of the rectangle
     *
     * @param e the event.
     */
    public void mousePressed(MouseEvent e) {
        if (!(e.getSource() instanceof ChartPanel)) {
            return;
        }
        ChartPanel panel = (ChartPanel) e.getSource();
        Rectangle2D dataArea = panel.getScreenDataArea();
        if (dataArea.contains(e.getPoint())) {
            SelectionManager selectionManager = panel.getSelectionManager();
            if (selectionManager != null) {
                if (!e.isShiftDown()) {
                    selectionManager.clearSelection();
                }
                Point pt = e.getPoint();
                this.startPoint = new Point(pt);
                this.selectionRect = new Rectangle(pt.x, pt.y, 1, 1);
            }
        }
    }

    /**
     * adjusts with and height of the rectangle by expanding it to the actual
     * position
     *
     * @param e the event.
     */
    public void mouseDragged(MouseEvent e) {
        ChartPanel panel = (ChartPanel) e.getSource();
        if (this.startPoint == null) {
            panel.clearLiveMouseHandler();
            return; // we never started a selection
        }
        Point pt = e.getPoint();
        Point2D pt2 = ShapeUtilities.getPointInRectangle(pt.x, pt.y,
                panel.getScreenDataArea());

        selectionRect = getRect(startPoint, pt2, panel.getScreenDataArea());
        panel.setSelectionShape(selectionRect);
        panel.setSelectionFillPaint(this.fillPaint);
        panel.setSelectionOutlinePaint(this.outlinePaint);
        panel.repaint();
    }

    /**
     * finishes the selection and calls the {@link org.jfree.chart.panel.selectionhandler.SelectionManager} of
     * the event source. The SelectionManager is then responsible for the processing
     * of the geometric selection.
     */
    public void mouseReleased(MouseEvent e) {
        ChartPanel panel = (ChartPanel) e.getSource();
        if (this.startPoint == null) {
            panel.clearLiveMouseHandler();
            return; // we never started a selection
        }
        SelectionManager selectionManager = panel.getSelectionManager();

        // do something with the selection shape
        if (selectionManager != null) {
            selectionManager.select(selectionRect);
        }

        //panel.setSelectionShape(null);
        this.selectionRect = null;
        //this.startPoint = null;
        panel.repaint();
        panel.clearLiveMouseHandler();
    }

    public Rectangle2D getRectangle2D(){
        return this.selectionRect;
    }
    
    /**
     * creates a rectangle from two points
     *
     * @param p1 one corner point
     * @param p2 the other corner point
     * @param plotarea the data area of the chart panel
     * @return a new rectangle that has both points in opposite corners
     */
    public Rectangle getRect(Point2D p1, Point2D p2, Rectangle2D plotarea ) {
        int minX, minY;
        int w, h;

        // process x and w
        if (p1.getX() < p2.getX()) {
            minX = (int) p1.getX();
            w = (int) p2.getX() - minX;
        } else {
            minX = (int) p2.getX();
            w = (int) p1.getX() - minX;
            if (w <= 0) {
                w = 1;
            }
        }

        // process y and h
        minY = (int) plotarea.getMinY();
        h = (int) plotarea.getHeight();

        return new Rectangle(minX, minY, w, h);
    }
}
