package org.eclipse.uml2.diagram.csd.edit.parts; import java.util.ArrayList; import java.util.Collections; import java.util.LinkedList; import java.util.List; import org.eclipse.draw2d.ColorConstants; import org.eclipse.draw2d.IFigure; import org.eclipse.draw2d.PositionConstants; import org.eclipse.draw2d.RectangleFigure; import org.eclipse.draw2d.Shape; import org.eclipse.draw2d.StackLayout; import org.eclipse.draw2d.geometry.Dimension; import org.eclipse.emf.common.notify.Notification; import org.eclipse.emf.ecore.EObject; import org.eclipse.gef.EditPart; import org.eclipse.gef.EditPolicy; import org.eclipse.gef.GraphicalEditPart; import org.eclipse.gef.Request; import org.eclipse.gef.commands.Command; import org.eclipse.gef.editpolicies.LayoutEditPolicy; import org.eclipse.gef.editpolicies.NonResizableEditPolicy; import org.eclipse.gef.editpolicies.ResizableEditPolicy; import org.eclipse.gef.handles.MoveHandle; import org.eclipse.gef.requests.CreateRequest; import org.eclipse.gmf.runtime.diagram.ui.editparts.BorderedBorderItemEditPart; import org.eclipse.gmf.runtime.diagram.ui.editparts.IBorderItemEditPart; import org.eclipse.gmf.runtime.diagram.ui.editparts.IGraphicalEditPart; import org.eclipse.gmf.runtime.diagram.ui.editpolicies.BorderItemSelectionEditPolicy; import org.eclipse.gmf.runtime.diagram.ui.editpolicies.CanonicalEditPolicy; import org.eclipse.gmf.runtime.diagram.ui.editpolicies.EditPolicyRoles; import org.eclipse.gmf.runtime.diagram.ui.figures.BorderItemLocator; import org.eclipse.gmf.runtime.draw2d.ui.figures.IBorderItemLocator; import org.eclipse.gmf.runtime.emf.type.core.IElementType; import org.eclipse.gmf.runtime.gef.ui.figures.DefaultSizeNodeFigure; import org.eclipse.gmf.runtime.gef.ui.figures.NodeFigure; import org.eclipse.gmf.runtime.notation.View; import org.eclipse.swt.graphics.Color; import org.eclipse.uml2.diagram.common.async.AsyncDiagramComponentEditPolicy; import org.eclipse.uml2.diagram.common.draw2d.UnrestrictedOutsideBorderItemLocator; import org.eclipse.uml2.diagram.common.editparts.PrimaryShapeEditPart; import org.eclipse.uml2.diagram.common.editpolicies.PortVisualEffectEditPolicy; import org.eclipse.uml2.diagram.common.editpolicies.U2TResizableShapeEditPolicy; import org.eclipse.uml2.diagram.common.editpolicies.UpdateDescriptionEditPolicy; import org.eclipse.uml2.diagram.csd.edit.policies.PortIsBehaviorVisualEffectEditPolicy; import org.eclipse.uml2.diagram.csd.edit.policies.PortItemSemanticEditPolicy; import org.eclipse.uml2.diagram.csd.part.UMLDiagramUpdateCommand; import org.eclipse.uml2.diagram.csd.part.UMLDiagramUpdater; import org.eclipse.uml2.diagram.csd.part.UMLVisualIDRegistry; import org.eclipse.uml2.diagram.csd.providers.UMLElementTypes; import org.eclipse.uml2.uml.UMLPackage; /** * @generated */ public class PortEditPart extends BorderedBorderItemEditPart implements PrimaryShapeEditPart { /** * @generated */ public static final int VISUAL_ID = 3011; /** * @generated */ protected IFigure contentPane; /** * @generated */ protected IFigure primaryShape; /** * @generated */ public PortEditPart(View view) { super(view); } /** * @generated */ protected void createDefaultEditPolicies() { if (UMLVisualIDRegistry.isShortcutDescendant(getNotationView())) { installEditPolicy(UpdateDescriptionEditPolicy.ROLE, new UpdateDescriptionEditPolicy(UMLDiagramUpdater.TYPED_ADAPTER, true)); } super.createDefaultEditPolicies(); installEditPolicy(EditPolicy.PRIMARY_DRAG_ROLE, getPrimaryDragEditPolicy()); installEditPolicy(EditPolicyRoles.SEMANTIC_ROLE, new PortItemSemanticEditPolicy()); installEditPolicy(EditPolicy.LAYOUT_ROLE, createLayoutEditPolicy()); installEditPolicy(EditPolicy.COMPONENT_ROLE, new AsyncDiagramComponentEditPolicy()); installEditPolicy("VisualEffect.Port", new PortVisualEffectEditPolicy()); //$NON-NLS-1$ installEditPolicy("VisualEffect.BehaviorPort", new PortIsBehaviorVisualEffectEditPolicy()); //$NON-NLS-1$ // XXX need an SCR to runtime to have another abstract superclass that would let children add reasonable editpolicies // removeEditPolicy(org.eclipse.gmf.runtime.diagram.ui.editpolicies.EditPolicyRoles.CONNECTION_HANDLES_ROLE); } /** * @generated */ protected LayoutEditPolicy createLayoutEditPolicy() { org.eclipse.gmf.runtime.diagram.ui.editpolicies.LayoutEditPolicy lep = new org.eclipse.gmf.runtime.diagram.ui.editpolicies.LayoutEditPolicy() { protected EditPolicy createChildEditPolicy(EditPart child) { View childView = (View) child.getModel(); switch (UMLVisualIDRegistry.getVisualID(childView)) { case PortNameEditPart.VISUAL_ID: case PortIsBehaviorEditPart.VISUAL_ID: return new BorderItemSelectionEditPolicy() { protected List createSelectionHandles() { MoveHandle mh = new MoveHandle((GraphicalEditPart) getHost()); mh.setBorder(null); return Collections.singletonList(mh); } }; } EditPolicy result = child.getEditPolicy(EditPolicy.PRIMARY_DRAG_ROLE); if (result == null) { result = new NonResizableEditPolicy(); } return result; } protected Command getMoveChildrenCommand(Request request) { return null; } protected Command getCreateCommand(CreateRequest request) { return null; } }; return lep; } /** * @generated */ protected IFigure createNodeShape() { return primaryShape = new PortFigure(); } /** * @generated */ public PortFigure getPrimaryShape() { return (PortFigure) primaryShape; } /** * @generated */ protected void addBorderItem(IFigure borderItemContainer, IBorderItemEditPart borderItemEditPart) { if (borderItemEditPart instanceof PortNameEditPart) { IBorderItemLocator locator = new UnrestrictedOutsideBorderItemLocator(getMainFigure()); borderItemContainer.add(borderItemEditPart.getFigure(), locator); return; } if (borderItemEditPart instanceof PortIsBehaviorEditPart) { IBorderItemLocator locator = new UnrestrictedOutsideBorderItemLocator(getMainFigure()); borderItemContainer.add(borderItemEditPart.getFigure(), locator); return; } super.addBorderItem(borderItemContainer, borderItemEditPart); } /** * @generated */ protected NodeFigure createNodePlate() { DefaultSizeNodeFigure result = new DefaultSizeNodeFigure(15, 15); //FIXME: workaround for #154536 result.getBounds().setSize(result.getPreferredSize()); return result; } /** * @generated */ public EditPolicy getPrimaryDragEditPolicy() { // #265822 Improve appearance of selection feedback ResizableEditPolicy result = new U2TResizableShapeEditPolicy(); return result; } /** * Creates figure for this edit part. * * Body of this method does not depend on settings in generation model * so you may safely remove <i>generated</i> tag and modify it. * * @generated */ protected NodeFigure createMainFigure() { NodeFigure figure = createNodePlate(); figure.setLayoutManager(new StackLayout()); IFigure shape = createNodeShape(); figure.add(shape); contentPane = setupContentPane(shape); return figure; } /** * Default implementation treats passed figure as content pane. * Respects layout one may have set for generated figure. * @param nodeShape instance of generated figure class * @generated */ protected IFigure setupContentPane(IFigure nodeShape) { return nodeShape; // use nodeShape itself as contentPane } /** * @generated */ public IFigure getContentPane() { if (contentPane != null) { return contentPane; } return super.getContentPane(); } /** * @generated */ protected void setForegroundColor(Color color) { if (primaryShape != null) { primaryShape.setForegroundColor(color); } } /** * @generated */ protected void setBackgroundColor(Color color) { if (primaryShape != null) { primaryShape.setBackgroundColor(color); } } /** * @generated */ protected void setLineWidth(int width) { if (primaryShape instanceof Shape) { ((Shape) primaryShape).setLineWidth(width); } } /** * @generated */ protected void setLineType(int style) { if (primaryShape instanceof Shape) { ((Shape) primaryShape).setLineStyle(style); } } /** * @generated */ public EditPart getPrimaryChildEditPart() { return getChildBySemanticHint(UMLVisualIDRegistry.getType(PortNameEditPart.VISUAL_ID)); } /** * @generated */ public List<IElementType> getMARelTypesOnSource() { ArrayList<IElementType> types = new ArrayList<IElementType>(6); types.add(UMLElementTypes.Connector_4005); types.add(UMLElementTypes.Dependency_4006); types.add(UMLElementTypes.Usage_4008); types.add(UMLElementTypes.PortProvided_4010); types.add(UMLElementTypes.PortRequired_4014); types.add(UMLElementTypes.Dependency_4017); return types; } /** * @generated */ public List<IElementType> getMARelTypesOnTarget() { ArrayList<IElementType> types = new ArrayList<IElementType>(6); types.add(UMLElementTypes.Connector_4005); types.add(UMLElementTypes.Dependency_4006); types.add(UMLElementTypes.Usage_4008); types.add(UMLElementTypes.ConstraintConstrainedElement_4012); types.add(UMLElementTypes.CommentAnnotatedElement_4016); types.add(UMLElementTypes.Dependency_4017); return types; } /** * @generated */ public List<IElementType> getMARelTypesOnSourceAndTarget(IGraphicalEditPart targetEditPart) { LinkedList<IElementType> types = new LinkedList<IElementType>(); if (targetEditPart instanceof PropertyEditPart) { types.add(UMLElementTypes.Connector_4005); } if (targetEditPart instanceof org.eclipse.uml2.diagram.csd.edit.parts.PortEditPart) { types.add(UMLElementTypes.Connector_4005); } if (targetEditPart instanceof Property3EditPart) { types.add(UMLElementTypes.Connector_4005); } if (targetEditPart instanceof Port2EditPart) { types.add(UMLElementTypes.Connector_4005); } if (targetEditPart instanceof Port3EditPart) { types.add(UMLElementTypes.Connector_4005); } if (targetEditPart instanceof CollaborationEditPart) { types.add(UMLElementTypes.Dependency_4006); } if (targetEditPart instanceof ClassEditPart) { types.add(UMLElementTypes.Dependency_4006); } if (targetEditPart instanceof Package2EditPart) { types.add(UMLElementTypes.Dependency_4006); } if (targetEditPart instanceof Class3EditPart) { types.add(UMLElementTypes.Dependency_4006); } if (targetEditPart instanceof InterfaceEditPart) { types.add(UMLElementTypes.Dependency_4006); } if (targetEditPart instanceof InstanceSpecificationEditPart) { types.add(UMLElementTypes.Dependency_4006); } if (targetEditPart instanceof ConstraintEditPart) { types.add(UMLElementTypes.Dependency_4006); } if (targetEditPart instanceof Interface2EditPart) { types.add(UMLElementTypes.Dependency_4006); } if (targetEditPart instanceof CollaborationUse2EditPart) { types.add(UMLElementTypes.Dependency_4006); } if (targetEditPart instanceof PropertyEditPart) { types.add(UMLElementTypes.Dependency_4006); } if (targetEditPart instanceof org.eclipse.uml2.diagram.csd.edit.parts.PortEditPart) { types.add(UMLElementTypes.Dependency_4006); } if (targetEditPart instanceof Property3EditPart) { types.add(UMLElementTypes.Dependency_4006); } if (targetEditPart instanceof Port2EditPart) { types.add(UMLElementTypes.Dependency_4006); } if (targetEditPart instanceof Port3EditPart) { types.add(UMLElementTypes.Dependency_4006); } if (targetEditPart instanceof CollaborationEditPart) { types.add(UMLElementTypes.Usage_4008); } if (targetEditPart instanceof ClassEditPart) { types.add(UMLElementTypes.Usage_4008); } if (targetEditPart instanceof Package2EditPart) { types.add(UMLElementTypes.Usage_4008); } if (targetEditPart instanceof Class3EditPart) { types.add(UMLElementTypes.Usage_4008); } if (targetEditPart instanceof InterfaceEditPart) { types.add(UMLElementTypes.Usage_4008); } if (targetEditPart instanceof InstanceSpecificationEditPart) { types.add(UMLElementTypes.Usage_4008); } if (targetEditPart instanceof ConstraintEditPart) { types.add(UMLElementTypes.Usage_4008); } if (targetEditPart instanceof Interface2EditPart) { types.add(UMLElementTypes.Usage_4008); } if (targetEditPart instanceof CollaborationUse2EditPart) { types.add(UMLElementTypes.Usage_4008); } if (targetEditPart instanceof PropertyEditPart) { types.add(UMLElementTypes.Usage_4008); } if (targetEditPart instanceof org.eclipse.uml2.diagram.csd.edit.parts.PortEditPart) { types.add(UMLElementTypes.Usage_4008); } if (targetEditPart instanceof Property3EditPart) { types.add(UMLElementTypes.Usage_4008); } if (targetEditPart instanceof Port2EditPart) { types.add(UMLElementTypes.Usage_4008); } if (targetEditPart instanceof Port3EditPart) { types.add(UMLElementTypes.Usage_4008); } if (targetEditPart instanceof InterfaceEditPart) { types.add(UMLElementTypes.PortProvided_4010); } if (targetEditPart instanceof Interface2EditPart) { types.add(UMLElementTypes.PortProvided_4010); } if (targetEditPart instanceof InterfaceEditPart) { types.add(UMLElementTypes.PortRequired_4014); } if (targetEditPart instanceof Interface2EditPart) { types.add(UMLElementTypes.PortRequired_4014); } if (targetEditPart instanceof CollaborationEditPart) { types.add(UMLElementTypes.Dependency_4017); } if (targetEditPart instanceof ClassEditPart) { types.add(UMLElementTypes.Dependency_4017); } if (targetEditPart instanceof Package2EditPart) { types.add(UMLElementTypes.Dependency_4017); } if (targetEditPart instanceof Class3EditPart) { types.add(UMLElementTypes.Dependency_4017); } if (targetEditPart instanceof InterfaceEditPart) { types.add(UMLElementTypes.Dependency_4017); } if (targetEditPart instanceof InstanceSpecificationEditPart) { types.add(UMLElementTypes.Dependency_4017); } if (targetEditPart instanceof ConstraintEditPart) { types.add(UMLElementTypes.Dependency_4017); } if (targetEditPart instanceof Interface2EditPart) { types.add(UMLElementTypes.Dependency_4017); } if (targetEditPart instanceof CollaborationUse2EditPart) { types.add(UMLElementTypes.Dependency_4017); } if (targetEditPart instanceof PropertyEditPart) { types.add(UMLElementTypes.Dependency_4017); } if (targetEditPart instanceof org.eclipse.uml2.diagram.csd.edit.parts.PortEditPart) { types.add(UMLElementTypes.Dependency_4017); } if (targetEditPart instanceof Property3EditPart) { types.add(UMLElementTypes.Dependency_4017); } if (targetEditPart instanceof Port2EditPart) { types.add(UMLElementTypes.Dependency_4017); } if (targetEditPart instanceof Port3EditPart) { types.add(UMLElementTypes.Dependency_4017); } return types; } /** * @generated */ public List<IElementType> getMATypesForSource(IElementType relationshipType) { LinkedList<IElementType> types = new LinkedList<IElementType>(); if (relationshipType == UMLElementTypes.Connector_4005) { types.add(UMLElementTypes.Property_3007); types.add(UMLElementTypes.Port_3011); types.add(UMLElementTypes.Property_3014); types.add(UMLElementTypes.Port_3016); types.add(UMLElementTypes.Port_3017); } else if (relationshipType == UMLElementTypes.Dependency_4006) { types.add(UMLElementTypes.Collaboration_2005); types.add(UMLElementTypes.Class_2006); types.add(UMLElementTypes.Package_2003); types.add(UMLElementTypes.Class_2007); types.add(UMLElementTypes.Interface_2009); types.add(UMLElementTypes.InstanceSpecification_2011); types.add(UMLElementTypes.Constraint_2012); types.add(UMLElementTypes.Interface_2014); types.add(UMLElementTypes.CollaborationUse_3002); types.add(UMLElementTypes.Property_3007); types.add(UMLElementTypes.Port_3011); types.add(UMLElementTypes.Property_3014); types.add(UMLElementTypes.Port_3016); types.add(UMLElementTypes.Port_3017); } else if (relationshipType == UMLElementTypes.Usage_4008) { types.add(UMLElementTypes.Collaboration_2005); types.add(UMLElementTypes.Class_2006); types.add(UMLElementTypes.Package_2003); types.add(UMLElementTypes.Class_2007); types.add(UMLElementTypes.Interface_2009); types.add(UMLElementTypes.InstanceSpecification_2011); types.add(UMLElementTypes.Constraint_2012); types.add(UMLElementTypes.Interface_2014); types.add(UMLElementTypes.CollaborationUse_3002); types.add(UMLElementTypes.Property_3007); types.add(UMLElementTypes.Port_3011); types.add(UMLElementTypes.Property_3014); types.add(UMLElementTypes.Port_3016); types.add(UMLElementTypes.Port_3017); } else if (relationshipType == UMLElementTypes.ConstraintConstrainedElement_4012) { types.add(UMLElementTypes.Constraint_2012); } else if (relationshipType == UMLElementTypes.CommentAnnotatedElement_4016) { types.add(UMLElementTypes.Comment_2013); } else if (relationshipType == UMLElementTypes.Dependency_4017) { types.add(UMLElementTypes.Collaboration_2005); types.add(UMLElementTypes.Class_2006); types.add(UMLElementTypes.Package_2003); types.add(UMLElementTypes.Class_2007); types.add(UMLElementTypes.Interface_2009); types.add(UMLElementTypes.InstanceSpecification_2011); types.add(UMLElementTypes.Constraint_2012); types.add(UMLElementTypes.Interface_2014); types.add(UMLElementTypes.CollaborationUse_3002); types.add(UMLElementTypes.Property_3007); types.add(UMLElementTypes.Port_3011); types.add(UMLElementTypes.Property_3014); types.add(UMLElementTypes.Port_3016); types.add(UMLElementTypes.Port_3017); } return types; } /** * @generated */ public List<IElementType> getMATypesForTarget(IElementType relationshipType) { LinkedList<IElementType> types = new LinkedList<IElementType>(); if (relationshipType == UMLElementTypes.Connector_4005) { types.add(UMLElementTypes.Property_3007); types.add(UMLElementTypes.Port_3011); types.add(UMLElementTypes.Property_3014); types.add(UMLElementTypes.Port_3016); types.add(UMLElementTypes.Port_3017); } else if (relationshipType == UMLElementTypes.Dependency_4006) { types.add(UMLElementTypes.Collaboration_2005); types.add(UMLElementTypes.Class_2006); types.add(UMLElementTypes.Package_2003); types.add(UMLElementTypes.Class_2007); types.add(UMLElementTypes.Interface_2009); types.add(UMLElementTypes.InstanceSpecification_2011); types.add(UMLElementTypes.Constraint_2012); types.add(UMLElementTypes.Interface_2014); types.add(UMLElementTypes.CollaborationUse_3002); types.add(UMLElementTypes.Property_3007); types.add(UMLElementTypes.Port_3011); types.add(UMLElementTypes.Property_3014); types.add(UMLElementTypes.Port_3016); types.add(UMLElementTypes.Port_3017); } else if (relationshipType == UMLElementTypes.Usage_4008) { types.add(UMLElementTypes.Collaboration_2005); types.add(UMLElementTypes.Class_2006); types.add(UMLElementTypes.Package_2003); types.add(UMLElementTypes.Class_2007); types.add(UMLElementTypes.Interface_2009); types.add(UMLElementTypes.InstanceSpecification_2011); types.add(UMLElementTypes.Constraint_2012); types.add(UMLElementTypes.Interface_2014); types.add(UMLElementTypes.CollaborationUse_3002); types.add(UMLElementTypes.Property_3007); types.add(UMLElementTypes.Port_3011); types.add(UMLElementTypes.Property_3014); types.add(UMLElementTypes.Port_3016); types.add(UMLElementTypes.Port_3017); } else if (relationshipType == UMLElementTypes.PortProvided_4010) { types.add(UMLElementTypes.Interface_2009); types.add(UMLElementTypes.Interface_2014); } else if (relationshipType == UMLElementTypes.PortRequired_4014) { types.add(UMLElementTypes.Interface_2009); types.add(UMLElementTypes.Interface_2014); } else if (relationshipType == UMLElementTypes.Dependency_4017) { types.add(UMLElementTypes.Collaboration_2005); types.add(UMLElementTypes.Class_2006); types.add(UMLElementTypes.Package_2003); types.add(UMLElementTypes.Class_2007); types.add(UMLElementTypes.Interface_2009); types.add(UMLElementTypes.InstanceSpecification_2011); types.add(UMLElementTypes.Constraint_2012); types.add(UMLElementTypes.Interface_2014); types.add(UMLElementTypes.CollaborationUse_3002); types.add(UMLElementTypes.Property_3007); types.add(UMLElementTypes.Port_3011); types.add(UMLElementTypes.Property_3014); types.add(UMLElementTypes.Port_3016); types.add(UMLElementTypes.Port_3017); } return types; } /** * @generated */ protected void handleNotificationEvent(Notification event) { super.handleNotificationEvent(event); handleFeatureLinkModification(event); } /** * @generated */ private void handleFeatureLinkModification(Notification event) { if (event.getFeature() == UMLPackage.eINSTANCE.getPort_Provided()) { guardedRefreshDiagram(); return; } if (event.getFeature() == UMLPackage.eINSTANCE.getPort_Required()) { guardedRefreshDiagram(); return; } } /** * @generated */ private boolean isCanonicalDisabled() { if (isCanonicalDisabled(getEditPolicy(EditPolicyRoles.CANONICAL_ROLE))) { return true; } if (getParent() != null && isCanonicalDisabled(getParent().getEditPolicy(EditPolicyRoles.CANONICAL_ROLE))) { return true; } //this particular edit part may not have editpolicy at all, //but its compartments still may have it EObject semantic = resolveSemanticElement(); if (semantic != null) { for (Object next : CanonicalEditPolicy.getRegisteredEditPolicies(semantic)) { if (next instanceof EditPolicy) { EditPolicy nextEP = (EditPolicy) next; if (isCanonicalDisabled(nextEP)) { return true; } } } } return false; } /** * @generated */ private static boolean isCanonicalDisabled(EditPolicy editPolicy) { return editPolicy instanceof CanonicalEditPolicy && !((CanonicalEditPolicy) editPolicy).isEnabled(); } /** * @generated */ private void guardedRefreshDiagram() { if (!isCanonicalDisabled()) { UMLDiagramUpdateCommand.performCanonicalUpdate(getDiagramView().getElement()); } } /** * @generated */ public class PortFigure extends RectangleFigure { /** * @generated */ public PortFigure() { this.setLineWidth(1); this.setForegroundColor(ColorConstants.gray); this.setPreferredSize(new Dimension(getMapMode().DPtoLP(15), getMapMode().DPtoLP(15))); this.setMaximumSize(new Dimension(getMapMode().DPtoLP(15), getMapMode().DPtoLP(15))); this.setMinimumSize(new Dimension(getMapMode().DPtoLP(15), getMapMode().DPtoLP(15))); } } }