Bug 521422 - [Composite Structure Diagram] DnD of Classifier to create

port

 - add a new strategy to drop a type to a class in composite diagram

Change-Id: Iedd3c237ac63df7bdb067d3a7dafe934b1aa10e8
Signed-off-by: Benoit Maggi <benoit.maggi@cea.fr>

2 files changed, 107 insertions, 0 deletions

diff --git a/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.dnd/plugin.xml b/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.dnd/plugin.xml
index f9de7585a1c..bf8ec66889f 100644
--- a/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.dnd/plugin.xml
+++ b/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.dnd/plugin.xml
@@ -33,5 +33,8 @@
        <strategy
              strategy="org.eclipse.papyrus.uml.diagram.dnd.strategy.ReorderListItemsStrategy">
        </strategy>
+       <strategy
+             strategy="org.eclipse.papyrus.uml.diagram.dnd.strategy.TypeToClassDropStrategy">
+       </strategy>       
    </extension>
 </plugin>
diff --git a/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.dnd/src/org/eclipse/papyrus/uml/diagram/dnd/strategy/TypeToClassDropStrategy.java b/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.dnd/src/org/eclipse/papyrus/uml/diagram/dnd/strategy/TypeToClassDropStrategy.java
new file mode 100755
index 00000000000..19fc79408a6
--- /dev/null
+++ b/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.dnd/src/org/eclipse/papyrus/uml/diagram/dnd/strategy/TypeToClassDropStrategy.java
@@ -0,0 +1,104 @@
+/*****************************************************************************
+ * Copyright (c) 2017 CEA LIST.
+ *
+ *
+ * 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:
+ *  Benoit Maggi (CEA LIST) benoit.maggi@cea.fr - Initial API and implementation
+ *
+ *****************************************************************************/
+package org.eclipse.papyrus.uml.diagram.dnd.strategy;
+
+import java.util.List;
+
+import org.eclipse.emf.ecore.EObject;
+import org.eclipse.gef.EditPart;
+import org.eclipse.gef.Request;
+import org.eclipse.gef.commands.Command;
+import org.eclipse.gmf.runtime.diagram.core.services.ViewService;
+import org.eclipse.gmf.runtime.diagram.ui.editparts.GraphicalEditPart;
+import org.eclipse.papyrus.infra.gmfdiag.dnd.strategy.TransactionalDropStrategy;
+import org.eclipse.papyrus.uml.diagram.composite.edit.parts.ClassCompositeEditPart;
+import org.eclipse.swt.graphics.Image;
+import org.eclipse.uml2.uml.Port;
+import org.eclipse.uml2.uml.Type;
+
+/**
+ * This strategy is is a specialization in order to be able to drop a Type as a Port in a Classifier
+ */
+public class TypeToClassDropStrategy extends TransactionalDropStrategy {
+
+	@Override
+	public String getLabel() {
+		return "Drop as Port";
+	}
+
+	@Override
+	public String getID() {
+		return "org.eclipse.papyrus.uml.diagram.parametric.dnd.PortToTypesPortDropStrategy"; //$NON-NLS-1$
+	}
+
+	@Override
+	public String getDescription() {
+		return "This strategy is is a specialization in order to be able to drop a Type as a Port in a Classifier.";
+	}
+
+	/**
+	 * @see org.eclipse.papyrus.infra.gmfdiag.dnd.strategy.DropStrategy#getImage()
+	 *
+	 * @return
+	 */
+	@Override
+	public Image getImage() {
+		return null;
+	}
+
+	/**
+	 * @see org.eclipse.papyrus.infra.gmfdiag.dnd.strategy.DropStrategy#getPriority()
+	 *
+	 * @return
+	 * @deprecated
+	 */
+	@Override
+	public int getPriority() {
+		return 0;
+	}
+
+	@Override
+	public Command doGetCommand(Request request, final EditPart targetEditPart) {
+		List<EObject> sourceElements = getSourceEObjects(request);
+		if (!sourceElements.isEmpty() && targetEditPart instanceof GraphicalEditPart) {
+			GraphicalEditPart graphicalEditPart = (GraphicalEditPart) targetEditPart;
+			EObject semanticElement = graphicalEditPart.resolveSemanticElement();
+			if (semanticElement instanceof org.eclipse.uml2.uml.Class) {
+				org.eclipse.uml2.uml.Class clazz = (org.eclipse.uml2.uml.Class) semanticElement;
+				EObject eObject = sourceElements.get(0);
+				if (eObject instanceof Type) {
+					if (graphicalEditPart instanceof ClassCompositeEditPart) {
+						return getDropTypeOnClassifierCommand((ClassCompositeEditPart) graphicalEditPart, (Type) eObject, clazz);
+					} else if (graphicalEditPart.getParent() instanceof ClassCompositeEditPart) {
+						return getDropTypeOnClassifierCommand((ClassCompositeEditPart) graphicalEditPart.getParent(), (Type) eObject, clazz);
+					}
+				}
+			}
+		}
+		return null; // UnexecutableCommand.INSTANCE;
+	}
+
+	private Command getDropTypeOnClassifierCommand(ClassCompositeEditPart classCompositeEditPart, Type type, org.eclipse.uml2.uml.Class clazz) {
+		Command cmd = new Command() {
+			@Override
+			public void execute() {
+				Port port = clazz.createOwnedPort(type.getName(), type);
+				ViewService.createNode(classCompositeEditPart.getNotationView(), port, "Port_Shape", classCompositeEditPart.getDiagramPreferencesHint());
+			}
+
+		};
+		return cmd;
+	}
+
+}