1 files changed, 179 insertions, 0 deletions
diff --git a/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.sequence/custom-src/org/eclipse/papyrus/uml/diagram/sequence/edit/policies/OccurenceSemanticEditPolicy.java b/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.sequence/custom-src/org/eclipse/papyrus/uml/diagram/sequence/edit/policies/OccurenceSemanticEditPolicy.java
new file mode 100644
index 00000000000..d66ae19ab8e
--- /dev/null
+++ b/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.sequence/custom-src/org/eclipse/papyrus/uml/diagram/sequence/edit/policies/OccurenceSemanticEditPolicy.java
@@ -0,0 +1,179 @@
+/*****************************************************************************
+ * Copyright (c) 2018 CEA LIST, EclipseSource 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:
+ *   EclipseSource - Initial API and implementation
+ *
+ *****************************************************************************/
+
+package org.eclipse.papyrus.uml.diagram.sequence.edit.policies;
+
+import org.eclipse.draw2d.IFigure;
+import org.eclipse.emf.ecore.EObject;
+import org.eclipse.emf.transaction.TransactionalEditingDomain;
+import org.eclipse.gef.EditPart;
+import org.eclipse.gef.Request;
+import org.eclipse.gef.commands.Command;
+import org.eclipse.gef.requests.ReconnectRequest;
+import org.eclipse.gmf.runtime.diagram.core.util.ViewUtil;
+import org.eclipse.gmf.runtime.diagram.ui.editparts.IGraphicalEditPart;
+import org.eclipse.gmf.runtime.diagram.ui.editpolicies.SemanticEditPolicy;
+import org.eclipse.gmf.runtime.emf.type.core.requests.ReorientRelationshipRequest;
+import org.eclipse.gmf.runtime.notation.Anchor;
+import org.eclipse.gmf.runtime.notation.Edge;
+import org.eclipse.gmf.runtime.notation.IdentityAnchor;
+import org.eclipse.gmf.runtime.notation.View;
+import org.eclipse.papyrus.infra.emf.utils.EMFHelper;
+import org.eclipse.papyrus.infra.gmfdiag.common.editpolicies.DefaultSemanticEditPolicy;
+import org.eclipse.papyrus.uml.diagram.sequence.anchors.AnchorConstants;
+import org.eclipse.papyrus.uml.diagram.sequence.util.GeneralOrderingUtil;
+import org.eclipse.papyrus.uml.diagram.sequence.util.OccurrenceSpecificationUtil;
+import org.eclipse.uml2.uml.ExecutionSpecification;
+import org.eclipse.uml2.uml.Message;
+import org.eclipse.uml2.uml.OccurrenceSpecification;
+
+/**
+ * A {@link SemanticEditPolicy} that is able to target specific {@link OccurrenceSpecification}s
+ * from a request (Typically for {@link Message#getSendEvent()},{@link Message#getReceiveEvent()},
+ * {@link ExecutionSpecification#getStart()} and {@link ExecutionSpecification#getFinish()})
+ */
+public class OccurenceSemanticEditPolicy extends DefaultSemanticEditPolicy {
+
+	/**
+	 * {@inheritDoc}
+	 *
+	 * <p>
+	 * Overridden to support {@link ReconnectRequest}, when only the anchor changes, referencing
+	 * a different {@link OccurrenceSpecification} on the same edit part.
+	 * </p>
+	 *
+	 * @param request
+	 * @return
+	 */
+	@Override
+	public Command getCommand(Request request) {
+		if (REQ_RECONNECT_SOURCE.equals(request.getType())
+				&& relationshipSourceHasChanged((ReconnectRequest) request)) {
+			EditPart connectionEP = ((ReconnectRequest) request)
+					.getConnectionEditPart();
+			if (ViewUtil.resolveSemanticElement((View) connectionEP.getModel()) == null) {
+				return getReorientRefRelationshipSourceCommand((ReconnectRequest) request);
+			} else {
+				return getReorientRelationshipSourceCommand((ReconnectRequest) request);
+			}
+		} else if (REQ_RECONNECT_TARGET.equals(request.getType())
+				&& relationshipTargetHasChanged((ReconnectRequest) request)) {
+			EditPart connectionEP = ((ReconnectRequest) request)
+					.getConnectionEditPart();
+			if (ViewUtil.resolveSemanticElement((View) connectionEP.getModel()) == null) {
+				return getReorientRefRelationshipTargetCommand((ReconnectRequest) request);
+			} else {
+				return getReorientRelationshipTargetCommand((ReconnectRequest) request);
+			}
+		}
+		return super.getCommand(request);
+	}
+
+	protected boolean relationshipSourceHasChanged(ReconnectRequest request) {
+		return getSourceOccurrence((Edge) request.getConnectionEditPart().getModel()) != getOccurrence(request);
+	}
+
+	protected boolean relationshipTargetHasChanged(ReconnectRequest request) {
+		return getTargetOccurrence((Edge) request.getConnectionEditPart().getModel()) != getOccurrence(request);
+	}
+
+	@Override
+	protected Command getReorientRelationshipSourceCommand(ReconnectRequest request) {
+		if (GeneralOrderingUtil.isGeneralOrderingLink(request)) {
+			EObject connectionSemElement = ViewUtil.resolveSemanticElement(((View) request.getConnectionEditPart()
+					.getModel()));
+			EObject targetSemElement = getOccurrence(request);
+			EObject oldSemElement = getSourceOccurrence((Edge) request.getConnectionEditPart().getModel());
+
+			TransactionalEditingDomain editingDomain = ((IGraphicalEditPart) getHost())
+					.getEditingDomain();
+			ReorientRelationshipRequest semRequest = new ReorientRelationshipRequest(
+					editingDomain, connectionSemElement, targetSemElement,
+					oldSemElement, ReorientRelationshipRequest.REORIENT_SOURCE);
+
+			semRequest.addParameters(request.getExtendedData());
+
+			return getSemanticCommand(semRequest);
+		}
+
+		return super.getReorientRefRelationshipSourceCommand(request);
+	}
+
+	@Override
+	protected Command getReorientRelationshipTargetCommand(ReconnectRequest request) {
+		if (GeneralOrderingUtil.isGeneralOrderingLink(request)) {
+			EObject connectionSemElement = ViewUtil.resolveSemanticElement((View) request.getConnectionEditPart().getModel());
+			EObject targetSemElement = getOccurrence(request);
+			EObject oldSemElement = getTargetOccurrence((Edge) request.getConnectionEditPart().getModel());
+
+			TransactionalEditingDomain editingDomain = ((IGraphicalEditPart) getHost())
+					.getEditingDomain();
+			ReorientRelationshipRequest semRequest = new ReorientRelationshipRequest(
+					editingDomain, connectionSemElement, targetSemElement,
+					oldSemElement, ReorientRelationshipRequest.REORIENT_TARGET);
+
+			semRequest.addParameters(request.getExtendedData());
+
+			return getSemanticCommand(semRequest);
+		}
+
+		return super.getReorientRelationshipTargetCommand(request);
+	}
+
+	private EObject getOccurrence(ReconnectRequest request) {
+		EObject element = EMFHelper.getEObject(request.getTarget());
+		IFigure targetFigure = ((IGraphicalEditPart) request.getTarget()).getFigure();
+		if (element instanceof Message) {
+			if (OccurrenceSpecificationUtil.isSource(targetFigure, request.getLocation())) {
+				return ((Message) element).getSendEvent();
+			} else {
+				return ((Message) element).getReceiveEvent();
+			}
+		} else if (element instanceof ExecutionSpecification) {
+			if (OccurrenceSpecificationUtil.isStart(targetFigure, request.getLocation())) {
+				return ((ExecutionSpecification) element).getStart();
+			} else {
+				return ((ExecutionSpecification) element).getFinish();
+			}
+		}
+		return element;
+	}
+
+	private EObject getTargetOccurrence(Edge connection) {
+		return getOccurrence(connection.getTargetAnchor(), connection.getTarget());
+	}
+
+	private EObject getSourceOccurrence(Edge connection) {
+		return getOccurrence(connection.getSourceAnchor(), connection.getSource());
+	}
+
+	private EObject getOccurrence(Anchor anchor, View targetElement) {
+		String anchorId = ((IdentityAnchor) anchor).getId();
+		EObject element = targetElement.getElement();
+		if (AnchorConstants.END_TERMINAL.equals(anchorId)) {
+			if (element instanceof Message) {
+				return ((Message) element).getReceiveEvent();
+			} else if (element instanceof ExecutionSpecification) {
+				return ((ExecutionSpecification) element).getFinish();
+			}
+		} else if (AnchorConstants.START_TERMINAL.equals(anchorId)) {
+			if (element instanceof Message) {
+				return ((Message) element).getSendEvent();
+			} else if (element instanceof ExecutionSpecification) {
+				return ((ExecutionSpecification) element).getStart();
+			}
+		}
+		return null;
+	}
+
+}