/*

[The "BSD licence"]
Copyright (c) 2005 Jean Bovet
All rights reserved.

Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions
are met:

1. Redistributions of source code must retain the above copyright
notice, this list of conditions and the following disclaimer.
2. Redistributions in binary form must reproduce the above copyright
notice, this list of conditions and the following disclaimer in the
documentation and/or other materials provided with the distribution.
3. The name of the author may not be used to endorse or promote products
derived from this software without specific prior written permission.

THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

*/

package edu.usfca.xj.appkit.gview.object;

import edu.usfca.xj.appkit.gview.base.Rect;
import edu.usfca.xj.appkit.gview.base.Vector2D;
import edu.usfca.xj.appkit.gview.shape.SLink;
import edu.usfca.xj.appkit.gview.shape.SLinkArc;
import edu.usfca.xj.appkit.gview.shape.SLinkBezier;
import edu.usfca.xj.appkit.gview.shape.SLinkElbow;
import edu.usfca.xj.foundation.XJXMLSerializable;

import java.awt.*;


public class GLink extends GElement implements XJXMLSerializable {

    public static final int SHAPE_ARC = 0;
    public static final int SHAPE_ELBOW = 1;
    public static final int SHAPE_BEZIER = 2;

    public GElement source = null;
    public GElement target = null;
    public String sourceAnchorKey = null;
    public String targetAnchorKey = null;
    public String pattern = null;

    protected SLink link = null;

    protected int shape = SHAPE_ARC;

    public GLink() {
        super();
    }

    public GLink(GElement source, String sourceAnchorKey, GElement target, String targetAnchorKey, int shape, String pattern, Point mouse, double flateness) {
        this.source = source;
        this.target = target;
        this.sourceAnchorKey = sourceAnchorKey;
        this.targetAnchorKey = targetAnchorKey;
        this.shape = shape;
        this.pattern = pattern;
        initializeLink(flateness);
        setSourceTangentOffset(source.getDefaultAnchorOffset(sourceAnchorKey));
        setTargetTangentOffset(target.getDefaultAnchorOffset(targetAnchorKey));
        link.setDirection(Vector2D.vector(mouse).sub(target.getPosition()));
    }

    public GLink(GElement source, String sourceAnchorKey, GElement target, String targetAnchorKey, int shape, String pattern, double flateness) {
        this.source = source;
        this.target = target;
        this.sourceAnchorKey = sourceAnchorKey;
        this.targetAnchorKey = targetAnchorKey;
        this.shape = shape;
        this.pattern = pattern;
        initializeLink(flateness);
        if(source == target)
            link.setDirection(new Vector2D(0, 1));
        else
            link.setDirection(source.getPosition().sub(target.getPosition()));
        setSourceTangentOffset(source.getDefaultAnchorOffset(sourceAnchorKey));
        setTargetTangentOffset(target.getDefaultAnchorOffset(targetAnchorKey));
    }

    public void setBezierControlPoints(Vector2D points[]) {
        if(link instanceof SLinkBezier) {
            SLinkBezier lb = (SLinkBezier)link;
            lb.setControlPoints(points);
        }
    }

    public void setBezierLabelPosition(Vector2D position) {
        if(link instanceof SLinkBezier) {
            SLinkBezier lb = (SLinkBezier)link;
            lb.setLabelPosition(position);
        }
    }

    protected SLink createLinkInstance() {
        switch(shape) {
            case SHAPE_ARC: return new SLinkArc();
            case SHAPE_ELBOW: return new SLinkElbow();
            case SHAPE_BEZIER: return new SLinkBezier();
        }
        return null;
    }

    protected void initializeLink(double flateness) {
        if(link == null) {
            link = createLinkInstance();
            link.setFlateness(flateness);
        }
    }

    public void setSource(GElement source) {
        this.source = source;
    }

    public GElement getSource() {
        return source;
    }

    public void setTarget(GElement target) {
        this.target = target;
    }

    public GElement getTarget() {
        return target;
    }

    public void setSourceAnchorKey(String key) {
        this.sourceAnchorKey = key;
    }

    public String getSourceAnchorKey() {
        return sourceAnchorKey;
    }

    public void setTargetAnchorKey(String key) {
        this.targetAnchorKey = key;
    }

    public String getTargetAnchorKey() {
        return targetAnchorKey;
    }

    public void setSourceTangentOffset(double offset) {
        link.setStartTangentOffset(offset);
    }

    public void setTargetTangentOffset(double offset) {
        link.setEndTangentOffset(offset);
    }

    public void setSourceOffset(double x, double y) {
        setSourceOffset(new Vector2D(x, y));
    }

    public void setSourceOffset(Vector2D offset) {
        link.setStartOffset(offset);
    }

    public void setTargetOffset(double x, double y) {
        setTargetOffset(new Vector2D(x, y));
    }

    public void setTargetOffset(Vector2D offset) {
        link.setEndOffset(offset);
    }

    public void setLink(SLink link) {
        this.link = link;
    }

    public SLink getLink() {
        return link;
    }

    public void setPattern(String pattern) {
        this.pattern = pattern;
    }

    public String getPattern() {
        return pattern;
    }

    public void setLabel(String label) {
        this.pattern = label;
    }

    public void setLabelColor(Color color) {
        link.setLabelColor(color);
    }

    public void setLabelVisible(boolean flag) {
        if(link != null)
            link.setLabelVisible(flag);
    }

    public boolean isLabelVisible() {
        if(link == null)
            return false;
        else
            return link.isLabelVisible();
    }

    public void setShape(int type) {
        this.shape = type;
    }

    public int getShape() {
        return shape;
    }

    public void toggleShape() {
        switch(shape) {
            case SHAPE_ARC:
                shape = SHAPE_ELBOW;
                break;
            case SHAPE_ELBOW:
                shape = SHAPE_ARC;
                break;
            case SHAPE_BEZIER:
                // Cannot toggle a bezier link
                return;
        }
        double flateness = link.getFlateness();
        Vector2D direction = link.getDirection();

        link = createLinkInstance();
        link.setFlateness(flateness);
        link.setDirection(direction);
    }

    public void setMousePosition(Point mouse) {
        link.setDirection(Vector2D.vector(mouse).sub(target.getPosition()));
        link.setMousePosition(Vector2D.vector(mouse));
    }

    public Rect getFrame() {
        update();
        return link.getFrame();
    }

    public boolean isInside(Point p) {
        if(link == null)
            return false;
        else
            return link.contains(p.x, p.y);
    }

    public void update() {
        initializeLink(0);

        source.updateAnchors();
        target.updateAnchors();

        link.setStartAnchor(source.getAnchor(sourceAnchorKey));
        link.setEndAnchor(target.getAnchor(targetAnchorKey));
        link.setLabel(pattern);
        link.setSelfLoop(source == target);

        link.update();
    }

    public void draw(Graphics2D g) {
        update();

        g.setStroke(new BasicStroke(penSize));

        if(color != null)
            link.setColor(color);
        else
            link.setColor(Color.black);

        link.draw(g);
    }

    public void drawShape(Graphics2D g) {
        link.drawShape(g);
    }

}
