ActivityTransitionEdge.java

/*
 Violet - A program for editing UML diagrams.

 Copyright (C) 2007 Cay S. Horstmann (http://horstmann.com)
 Alexandre de Pellegrin (http://alexdp.free.fr);

 This program is free software; you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation; either version 2 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 General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program; if not, write to the Free Software
 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

package com.horstmann.violet.product.diagram.activity.edge;

import java.awt.geom.Point2D;

import com.horstmann.violet.product.diagram.abstracts.edge.bentstyle.BentStyle;
import com.horstmann.violet.product.diagram.activity.ActivityDiagramConstant;
import com.horstmann.violet.product.diagram.property.ArrowheadChoiceList;
import com.horstmann.violet.product.diagram.property.BentStyleChoiceList;
import com.horstmann.violet.product.diagram.property.LineStyleChoiceList;
import com.horstmann.violet.product.diagram.abstracts.Direction;
import com.horstmann.violet.product.diagram.common.edge.LabeledLineEdge;
import com.horstmann.violet.product.diagram.abstracts.node.INode;
import com.horstmann.violet.product.diagram.activity.node.SynchronizationBarNode;

/**
 * An edge that is shaped like a line with up to three segments with an arrowhead
 */
public class ActivityTransitionEdge extends LabeledLineEdge
{
    public ActivityTransitionEdge()
    {
        super();
        setStartArrowhead(ArrowheadChoiceList.NONE);
        setEndArrowhead(ArrowheadChoiceList.V);
        setLineStyle(LineStyleChoiceList.SOLID);
        setBentStyle(BentStyle.AUTO);
    }

    protected ActivityTransitionEdge(ActivityTransitionEdge cloned)
    {
        super(cloned);
    }

    @Override
    protected void beforeReconstruction()
    {
        super.beforeReconstruction();
        setStartArrowhead(ArrowheadChoiceList.NONE);
        setEndArrowhead(ArrowheadChoiceList.V);
        setLineStyle(LineStyleChoiceList.SOLID);
    }

    @Override
    protected ActivityTransitionEdge copy() throws CloneNotSupportedException
    {
        return new ActivityTransitionEdge(this);
    }

    @Override
    public String getToolTip()
    {
        return ActivityDiagramConstant.ACTIVITY_DIAGRAM_RESOURCE.getString("tooltip.transition_edge");
    }

    @Override
    public Direction getDirection(INode node)
    {
        BentStyleChoiceList bStyle = ((BentStyleChoiceList)getBentStyleChoiceList());
        Direction straightDirection = super.getDirection(node);
        double x = straightDirection.getX();
        double y = straightDirection.getY();


        if (node.equals(getStartNode()))
        {
            if (BentStyleChoiceList.HV.equals(bStyle) || BentStyleChoiceList.HVH.equals(bStyle))
            {
                return (x >= 0) ? Direction.EAST : Direction.WEST;
            }
            if (BentStyleChoiceList.VH.equals(bStyle) || BentStyleChoiceList.VHV.equals(bStyle))
            {
                return (y >= 0) ? Direction.SOUTH : Direction.NORTH;
            }
        }
        if (node.equals(getEndNode()))
        {
            if (BentStyleChoiceList.HV.equals(bStyle) || BentStyleChoiceList.VHV.equals(bStyle))
            {
                return (y >= 0) ? Direction.SOUTH : Direction.NORTH;
            }
            if (BentStyleChoiceList.VH.equals(bStyle) || BentStyleChoiceList.HVH.equals(bStyle))
            {
                return (x >= 0) ? Direction.EAST : Direction.WEST;
            }
        }
        if (SynchronizationBarNode.class.isInstance(getStartNode()) ||
            SynchronizationBarNode.class.isInstance(getEndNode()))
        {
            SynchronizationBarNode synchronizationBarNode;

            if (SynchronizationBarNode.class.isInstance(getStartNode()))
            {
                synchronizationBarNode = (SynchronizationBarNode)getStartNode();
            }
            else
            {
                synchronizationBarNode = (SynchronizationBarNode)getEndNode();
            }

            Point2D p1 = node.getLocationOnGraph();
            Point2D p2;

            if (node.equals(getStartNode()))
            {
                p2 = getEndNode().getLocationOnGraph();
            }
            else
            {
                p2 = getStartNode().getLocationOnGraph();
            }

            if(SynchronizationBarNode.HORIZONTAL == synchronizationBarNode.getStretch())
            {
                if (p1.getY() > p2.getY())
                {
                    return Direction.SOUTH;
                }
                else
                {
                    return Direction.NORTH;
                }
            }
            else
            {
                if (p1.getX() > p2.getX())
                {
                    return Direction.EAST;
                }
                else
                {
                    return Direction.WEST;
                }
            }
        }
        return straightDirection;
    }
}