plugins/org.eclipse.sirius.diagram.ui/src-diag/org/eclipse/sirius/diagram/ui/graphical/edit/policies/SetConnectionBendpointsAccordingToDraw2DCommand.java

/*******************************************************************************
 * Copyright (c) 2012, 2017 THALES GLOBAL SERVICES.
 * This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License 2.0
 * which accompanies this distribution, and is available at
 * https://www.eclipse.org/legal/epl-2.0/
 *
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 *    Obeo - initial API and implementation
 *******************************************************************************/
package org.eclipse.sirius.diagram.ui.graphical.edit.policies;

import org.eclipse.draw2d.Connection;
import org.eclipse.draw2d.IFigure;
import org.eclipse.draw2d.geometry.Point;
import org.eclipse.draw2d.geometry.PointList;
import org.eclipse.draw2d.geometry.PrecisionPoint;
import org.eclipse.draw2d.geometry.PrecisionRectangle;
import org.eclipse.draw2d.geometry.Rectangle;
import org.eclipse.emf.transaction.TransactionalEditingDomain;
import org.eclipse.gmf.runtime.diagram.ui.editparts.ConnectionEditPart;
import org.eclipse.sirius.ext.gmf.runtime.editparts.GraphicalHelper;

// CHECKSTYLE:OFF
/**
 * @author <a href="mailto:laurent.redor@obeo.fr">Laurent Redor</a>
 *
 */
public class SetConnectionBendpointsAccordingToDraw2DCommand extends SetConnectionBendpointsAndLabelCommmand {

    private boolean sourceMove;

    private PrecisionPoint moveDelta;

    /**
     * @param editingDomain
     *            the editing domain through which model changes are made
     */
    public SetConnectionBendpointsAccordingToDraw2DCommand(TransactionalEditingDomain editingDomain) {
        super(editingDomain);
    }

    /**
     * Compute the new PointList, source reference point and target reference
     * point according to given <code>connectionEditPart</code>,
     * <code>sourceMove</code> value and the <code>modeDelta</code>.
     * 
     * @param connectionEditPart
     *            The connection from which to compute the new PointList
     */
    public void setNewPointList(ConnectionEditPart connectionEditPart) {
        // Apply inverse zoom on moveDelta, because moveDelta is only
        // element in relative value
        GraphicalHelper.applyInverseZoomOnPoint(connectionEditPart, moveDelta);
        Connection connection = connectionEditPart.getConnectionFigure();

        Point tempSourceRefPoint = connection.getSourceAnchor().getReferencePoint();
        connection.translateToRelative(tempSourceRefPoint);

        Point tempTargetRefPoint = connection.getTargetAnchor().getReferencePoint();
        connection.translateToRelative(tempTargetRefPoint);

        PointList connectionPointList = connection.getPoints().getCopy();
        if (sourceMove) {
            tempSourceRefPoint.performTranslate(moveDelta.x, moveDelta.y);

            // Get the bounds of the source before and after move (to apply
            // specific move on the first point of the edge)
            IFigure sourceAnchorOnwer = connection.getSourceAnchor().getOwner();
            Rectangle rBox = (sourceAnchorOnwer instanceof Connection) ? ((Connection) sourceAnchorOnwer).getPoints().getBounds() : sourceAnchorOnwer.getBounds();
            PrecisionRectangle boxBeforeMove = new PrecisionRectangle(rBox);
            sourceAnchorOnwer.translateToAbsolute(boxBeforeMove);
            PrecisionRectangle boxAfterMove = new PrecisionRectangle(boxBeforeMove.getCopy());
            boxAfterMove.translate(moveDelta);
            boolean changeXDelta = boxBeforeMove.x != boxAfterMove.x;

            if (connectionPointList.getPoint(1).y < boxAfterMove.getCenter().y) {
                if (connectionPointList.getPoint(1).y < boxBeforeMove.getCenter().y) {
                    // The source is below the horizontal trunk segment
                    // before and after the move
                    connectionPointList.setPoint(connectionPointList.getFirstPoint().translate(moveDelta), 0);
                    if (changeXDelta) {
                        connectionPointList.setPoint(connectionPointList.getPoint(1).translate(moveDelta.x, 0), 1);
                    }
                } else {
                    // The source is above the horizontal trunk segment
                    // before the move and below after the move
                    connectionPointList.setPoint(connectionPointList.getFirstPoint().translate(moveDelta.x, moveDelta.y - boxBeforeMove.height), 0);
                    if (changeXDelta) {
                        connectionPointList.setPoint(connectionPointList.getPoint(1).translate(moveDelta.x, 0), 1);
                    }
                }
            } else {
                if (connectionPointList.getPoint(1).y < boxBeforeMove.getCenter().y) {
                    // The source is below the horizontal trunk segment
                    // before and above after the move
                    connectionPointList.setPoint(connectionPointList.getFirstPoint().translate(moveDelta.x, moveDelta.y + boxBeforeMove.height), 0);
                    if (changeXDelta) {
                        connectionPointList.setPoint(connectionPointList.getPoint(1).translate(moveDelta.x, 0), 1);
                    }
                } else {
                    // The source is above the horizontal trunk segment
                    // before and after the move
                    connectionPointList.setPoint(connectionPointList.getFirstPoint().translate(moveDelta.x, moveDelta.y), 0);
                    if (changeXDelta) {
                        connectionPointList.setPoint(connectionPointList.getPoint(1).translate(moveDelta.x, 0), 1);
                    }
                }
            }
        } else {
            tempTargetRefPoint.performTranslate(moveDelta.x, moveDelta.y);

            // Get the bounds of the target before and after move (to apply
            // specific move on the last point of the edge)
            IFigure targetAnchorOnwer = connection.getTargetAnchor().getOwner();
            Rectangle rBox = (targetAnchorOnwer instanceof Connection) ? ((Connection) targetAnchorOnwer).getPoints().getBounds() : targetAnchorOnwer.getBounds();
            PrecisionRectangle boxBeforeMove = new PrecisionRectangle(rBox.getCopy());
            targetAnchorOnwer.translateToAbsolute(boxBeforeMove.getCopy());
            PrecisionRectangle boxAfterMove = new PrecisionRectangle(boxBeforeMove.getCopy());
            boxAfterMove.translate(moveDelta);
            boolean changeXDelta = boxBeforeMove.x != boxAfterMove.x;
            if (connectionPointList.getPoint(1).y < boxAfterMove.getCenter().y) {
                if (connectionPointList.getPoint(1).y < boxBeforeMove.getCenter().y) {
                    // The source is below the horizontal trunk segment
                    // before and after the move
                    connectionPointList.setPoint(connectionPointList.getLastPoint().translate(moveDelta), 3);
                    if (changeXDelta) {
                        connectionPointList.setPoint(connectionPointList.getPoint(2).translate(moveDelta.x, 0), 2);
                    }
                } else {
                    // The source is above the horizontal trunk segment
                    // before the move and below after the move
                    connectionPointList.setPoint(connectionPointList.getLastPoint().translate(moveDelta.x, moveDelta.y - boxBeforeMove.height), 3);
                    if (changeXDelta) {
                        connectionPointList.setPoint(connectionPointList.getPoint(2).translate(moveDelta.x, 0), 2);
                    }
                }
            } else {
                if (connectionPointList.getPoint(1).y < boxBeforeMove.getCenter().y) {
                    // The source is below the horizontal trunk segment
                    // before and above after the move
                    connectionPointList.setPoint(connectionPointList.getLastPoint().translate(moveDelta.x, moveDelta.y + boxBeforeMove.height), 3);
                    if (changeXDelta) {
                        connectionPointList.setPoint(connectionPointList.getPoint(2).translate(moveDelta.x, 0), 2);
                    }
                } else {
                    // The source is above the horizontal trunk segment
                    // before and after the move
                    connectionPointList.setPoint(connectionPointList.getLastPoint().translate(moveDelta.x, moveDelta.y), 3);
                    if (changeXDelta) {
                        connectionPointList.setPoint(connectionPointList.getPoint(2).translate(moveDelta.x, 0), 2);
                    }
                }
            }
        }
        setNewPointList(connectionPointList, tempSourceRefPoint, tempTargetRefPoint);
    }

    /**
     * @param b
     */
    public void setSourceMove(boolean sourceMove) {
        this.sourceMove = sourceMove;

    }

    /**
     * @param moveDelta
     */
    public void setMoveDelta(PrecisionPoint moveDelta) {
        this.moveDelta = moveDelta;

    }
    // CHECKSTYLE:OFF
}