parallelline.java

EclipseTrader is a stock exchange analysis system, featuring shares pricing watch, intraday and hi

/*
 * Copyright (c) 2004-2006 Marco Maccaferri and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Marco Maccaferri - initial API and implementation
 *     Robert Kallal - Port of single line to parallel line
 *     
 */

package net.sourceforge.eclipsetrader.charts.objects;

import java.util.Date;

import net.sourceforge.eclipsetrader.charts.ObjectPlugin;
import net.sourceforge.eclipsetrader.charts.Settings;
import net.sourceforge.eclipsetrader.charts.events.PlotMouseEvent;
import net.sourceforge.eclipsetrader.charts.internal.PixelTools;

import org.eclipse.swt.SWT;
import org.eclipse.swt.graphics.Color;
import org.eclipse.swt.graphics.GC;
import org.eclipse.swt.graphics.Point;
import org.eclipse.swt.graphics.Rectangle;

/**
 * Parallel Line
 */
public class ParallelLine extends ObjectPlugin
{
    private Date date1, date2;
    private double value1 = 0, value2 = 0;
    private double valueOffset = 0;
    private Color color = new Color(null, 0, 0, 224);
    private boolean extend1 = false, extend2 = false;
    private Point p1, p2, p3, p4, selected;
    private Point o1, o2;
    private Point d1, d2, d3, d4;
    private int lastX = -1, lastY = -1;

    public ParallelLine()
    {
    }

    /* (non-Javadoc)
     * @see net.sourceforge.eclipsetrader.charts.ObjectPlugin#dispose()
     */
    public void dispose()
    {
        color.dispose();
        super.dispose();
    }

    /* (non-Javadoc)
     * @see net.sourceforge.eclipsetrader.charts.ObjectPlugin#isOverLine(int, int)
     */
    public boolean isOverLine(int x, int y)
    {
        if (d1 == null || d2 == null || d3 == null || d4 == null)
            return false;
        if (isOverHandle(x, y))
            return true;
        return PixelTools.isPointOnLine(x, y, d1.x, d1.y, d2.x, d2.y) ||
               PixelTools.isPointOnLine(x, y, d3.x, d3.y, d4.x, d4.y);
    }

    /* (non-Javadoc)
     * @see net.sourceforge.eclipsetrader.charts.ObjectPlugin#isOverHandle(int, int)
     */
    public boolean isOverHandle(int x, int y)
    {
    	if (p1 != null && Math.abs(x - p1.x) <= 2 && Math.abs(y - p1.y) <= 2)
    		return true;
    	else if (p2 != null && Math.abs(x - p2.x) <= 2 && Math.abs(y - p2.y) <= 2)
    		return true;
    	else if (p3 != null && Math.abs(x - p3.x) <= 2 && Math.abs(y - p3.y) <= 2)
    		return true;
    	else if (p4 != null && Math.abs(x - p4.x) <= 2 && Math.abs(y - p4.y) <= 2)
    		return true;
    	else
    		return false;
    }

    /* (non-Javadoc)
     * @see net.sourceforge.eclipsetrader.charts.ObjectPlugin#mouseDown(net.sourceforge.eclipsetrader.charts.events.PlotMouseEvent)
     */
    public void mouseDown(PlotMouseEvent e)
    {
        if (p2 == null)
        {
            date1 = e.date;
            date2 = e.date;
            value1 = e.value;
            value2 = e.value;
            
            valueOffset = getPlot().getScaler().convertToValue(
            		getPlot().getScaler().convertToY(e.value) - 10) - 
            		e.value;
            p2 = new Point(e.x, e.y);
            selected = p2;
        }
        else
        {
            selected = null;
            if (Math.abs(e.x - p1.x) <= 2 && Math.abs(e.y - p1.y) <= 2)
                selected = p1;
            else if (Math.abs(e.x - p2.x) <= 2 && Math.abs(e.y - p2.y) <= 2)
                selected = p2;
            else if (Math.abs(e.x - p3.x) <= 2 && Math.abs(e.y - p3.y) <= 2)
                selected = p3;
            else if (Math.abs(e.x - p4.x) <= 2 && Math.abs(e.y - p4.y) <= 2)
                selected = p4;
            else
            {
                o1 = new Point(p1.x, p1.y);
                o2 = new Point(p2.x, p2.y);
                lastX = e.x;
                lastY = e.y;
            }
        }
    }

    /* (non-Javadoc)
     * @see net.sourceforge.eclipsetrader.charts.ObjectPlugin#mouseUp(net.sourceforge.eclipsetrader.charts.events.PlotMouseEvent)
     */
    public void mouseUp(PlotMouseEvent e)
    {
        selected = null;
        lastX = lastY = -1;

        getSettings().set("date1", date1);
        getSettings().set("value1", value1);
        getSettings().set("date2", date2);
        getSettings().set("value2", value2);
        getSettings().set("valueOffset", valueOffset);
    }

    /* (non-Javadoc)
     * @see net.sourceforge.eclipsetrader.charts.ObjectPlugin#mouseMove(net.sourceforge.eclipsetrader.charts.events.PlotMouseEvent)
     */
    public void mouseMove(PlotMouseEvent e)
    {
        if (selected != null)
        {
            Point point = new Point(getPlot().getDatePlot().mapToScreen(e.date), getPlot().getScaler().convertToY(e.roundedValue));
            if (!point.equals(selected))
            {
                if (selected == p1)
                {
                    date1 = e.date;
                    value1 = e.roundedValue;
                }
                else if (selected == p2)
                {
                    date2 = e.date;
                    value2 = e.roundedValue;
                }
                else if (selected == p3) 
                	valueOffset = e.roundedValue - value1;
                else if (selected == p4)
                	valueOffset = e.roundedValue - value2;

                
                invalidateRectangle();
            }
        }
        else
        {
            Date date1 = getPlot().getDatePlot().mapToDate(o1.x + (e.x - lastX));
            Date date2 = getPlot().getDatePlot().mapToDate(o2.x + (e.x - lastX));
            if (date1 != null && date2 != null)
            {
                this.date1 = date1;
                this.date2 = date2;
            }
            value1 = getPlot().getScaler().convertToValue(o1.y + (e.y - lastY));
            value2 = getPlot().getScaler().convertToValue(o2.y + (e.y - lastY));

            invalidateRectangle();
        }
    }
    
    private void invalidateRectangle()
    {
        getPlot().getIndicatorPlot().redraw();
        getPlot().getIndicatorPlot().update();
    }

    /* (non-Javadoc)
     * @see net.sourceforge.eclipsetrader.charts.ObjectPlugin#drawObject(org.eclipse.swt.graphics.GC, boolean)
     */
    public void drawObject(GC gc, boolean selected)
    {
        gc.setLineStyle(SWT.LINE_SOLID);
        gc.setForeground(color);
        
        if (date1 != null && date2 != null)
        {
            Point location = getPlot().getIndicatorPlot().getPlotLocation();
            Rectangle rect = getPlot().getIndicatorPlot().getPlotBounds();
            if (p1 == null)
                p1 = new Point(0, 0);
            p1.x = getPlot().getDatePlot().mapToScreen(date1);
            p1.y = getPlot().getScaler().convertToY(value1);
            if (p2 == null)
                p2 = new Point(0, 0);
            p2.x = getPlot().getDatePlot().mapToScreen(date2);
            p2.y = getPlot().getScaler().convertToY(value2);
            if (p3 == null)
            	p3 = new Point(0,0);
            p3.x = getPlot().getDatePlot().mapToScreen(date1);
            p3.y = getPlot().getScaler().convertToY(value1+valueOffset);           
            if (p4 == null)
            	p4 = new Point(0,0);
            p4.x = getPlot().getDatePlot().mapToScreen(date2);
            p4.y = getPlot().getScaler().convertToY(value2+valueOffset);
           
            gc.drawLine (p1.x + location.x, p1.y, p2.x + location.x, p2.y);
            gc.drawLine (p3.x + location.x, p3.y, p4.x + location.x, p4.y);
            
            if (d1 == null)
            	d1 = new Point(0,0);
            d1.x = p1.x;
            d1.y = p1.y;
            if (d2 == null)
            	d2 = new Point(0,0);
            d2.x = p2.x;
            d2.y = p2.y;
            if (d3 == null)
            	d3 = new Point(0,0);
            d3.x = p3.x;
            d3.y = p3.y;
            if (d4 == null)
            	d4 = new Point(0,0);
            d4.x = p4.x;
            d4.y = p4.y;

            if (extend1)
            {
                int ydiff = p2.y - p1.y;
                int xdiff = p1.x - p2.x;
                if (xdiff != 0 || ydiff != 0)
                {
                    while(d1.x > 0 && d1.x < rect.width && d1.y > 0 && d1.y < rect.height)
                    {
                        d1.x += xdiff;
                        d1.y -= ydiff;
                     }
                    gc.drawLine (d1.x + location.x, d1.y, p1.x + location.x, p1.y);
                }
                ydiff = p4.y - p3.y;
                xdiff = p3.x - p4.x;
                if (xdiff != 0 || ydiff != 0)
                {
                    while(d3.x > 0 && d3.x < rect.width && d3.y > 0 && d3.y < rect.height)
                    {
                        d3.x += xdiff;
                        d3.y -= ydiff;
                     }
                    gc.drawLine (d3.x + location.x, d3.y, p3.x + location.x, p3.y);
                }
                
                
            }
            
            if (extend2)
            {
                int ydiff = p1.y- p2.y;
                int xdiff = p2.x - p1.x;
                if (xdiff != 0 || ydiff != 0)
                {
                    while(d2.x > 0 && d2.x < rect.width && d2.y > 0 && d2.y < rect.height)
                    {
                        d2.x += xdiff;
                        d2.y -= ydiff;
                     }
                    gc.drawLine (p2.x + location.x, p2.y, d2.x + location.x, d2.y);
                }
                
                ydiff = p3.y - p4.y;
                xdiff = p4.x - p3.x;
                if (xdiff != 0 || ydiff != 0)
                {
                    while(d4.x > 0 && d4.x < rect.width && d4.y > 0 && d4.y < rect.height)
                    {
                        d4.x += xdiff;
                        d4.y -= ydiff;
                     }
                    gc.drawLine (p4.x + location.x, p4.y, d4.x + location.x, d4.y);
                }

            }

            if (selected)
            {
                gc.setBackground(color);
                gc.fillRectangle(p1.x + location.x - 2, p1.y - 2, 5, 5);
                gc.fillRectangle(p2.x + location.x - 2, p2.y - 2, 5, 5);
                gc.fillRectangle(p3.x + location.x - 2, p3.y - 2, 5, 5);
                gc.fillRectangle(p4.x + location.x - 2, p4.y - 2, 5, 5);
            }
        }
    }

    /* (non-Javadoc)
     * @see net.sourceforge.eclipsetrader.charts.ObjectPlugin#setParameters(net.sourceforge.eclipsetrader.charts.Settings)
     */
    public void setSettings(Settings settings)
    {
        super.setSettings(settings);

        date1 = settings.getDate("date1", null);
        date2 = settings.getDate("date2", null);
        value1 = settings.getDouble("value1", 0).doubleValue();
        value2 = settings.getDouble("value2", 0).doubleValue();
        valueOffset = settings.getDouble("valueOffset", 0).doubleValue();
        color = settings.getColor("color", color.getRGB());
        extend1 = settings.getBoolean("extend1", false);
        extend2 = settings.getBoolean("extend2", false);
        p1 = p2 = d1 = d2 = null;
    }
}