1 files changed, 76 insertions, 0 deletions
diff --git a/plugins/sysml/diagram/org.eclipse.papyrus.sysml.diagram.internalblock/src/org/eclipse/papyrus/sysml/diagram/internalblock/ui/InternalBlockInterfaceManagerDialog.java b/plugins/sysml/diagram/org.eclipse.papyrus.sysml.diagram.internalblock/src/org/eclipse/papyrus/sysml/diagram/internalblock/ui/InternalBlockInterfaceManagerDialog.java
new file mode 100644
index 00000000000..4c835596ad9
--- /dev/null
+++ b/plugins/sysml/diagram/org.eclipse.papyrus.sysml.diagram.internalblock/src/org/eclipse/papyrus/sysml/diagram/internalblock/ui/InternalBlockInterfaceManagerDialog.java
@@ -0,0 +1,76 @@
+/*****************************************************************************
+ * Copyright (c) 2014 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:
+ *	Gabriel Pascual (ALL4TEC) gabriel.pascual@all4tec.net - Initial API and implementation
+ *****************************************************************************/
+
+package org.eclipse.papyrus.sysml.diagram.internalblock.ui;
+
+import java.util.Arrays;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+
+import org.eclipse.papyrus.sysml.portandflows.FlowSpecification;
+import org.eclipse.papyrus.uml.diagram.common.dialogs.InterfaceManagerDialog;
+import org.eclipse.papyrus.uml.diagram.common.util.Visitor;
+import org.eclipse.swt.widgets.Shell;
+import org.eclipse.uml2.uml.Element;
+import org.eclipse.uml2.uml.Interface;
+import org.eclipse.uml2.uml.Namespace;
+import org.eclipse.uml2.uml.Package;
+import org.eclipse.uml2.uml.Port;
+import org.eclipse.uml2.uml.util.UMLUtil;
+
+/**
+ * <pre>
+ * This class provides a Dialog to manage easily the provided and 
+ * the required Interfaces for a {@link Port}.
+ * </pre>
+ */
+public class InternalBlockInterfaceManagerDialog extends InterfaceManagerDialog {
+
+
+	/**
+	 * Constructor.
+	 *
+	 * @param parentShell
+	 * @param port
+	 */
+	public InternalBlockInterfaceManagerDialog(Shell parentShell, Port port) {
+		super(parentShell, port);
+	}
+
+	/**
+	 * Returns all the available Interfaces (except {@link FlowSpecification})
+	 * 
+	 * @return
+	 *         all the available Interfaces
+	 *         //TODO add the interfaces from the package import!
+	 */
+	protected List<Interface> getAllAvailableInterfaces(Package pack) {
+		Set<Interface> otherInterfaces = new HashSet<Interface>();
+		List<Element> interfaces = Visitor.getOwnedAndImportedElement(pack, Interface.class);
+		for(Namespace namespace : Visitor.getOwnedAndImportedNamespaces(pack)) {
+			interfaces.addAll(Visitor.getOwnedAndImportedElement(namespace, Interface.class));
+
+		}
+
+		for(Element element : interfaces) {
+			// Test if the interface is a FlowSpecification
+			FlowSpecification flowSpec = UMLUtil.getStereotypeApplication(element, FlowSpecification.class);
+			if(flowSpec == null) {
+				otherInterfaces.add((Interface)element);
+			}
+		}
+
+		return Arrays.asList(otherInterfaces.toArray(new Interface[otherInterfaces.size()]));
+	}
+
+}