package org.eclipse.gmf.tooling.examples.affixednode.diagram.edit.parts; import org.eclipse.draw2d.FlowLayout; 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.gef.EditPart; import org.eclipse.gef.EditPolicy; import org.eclipse.gef.commands.Command; import org.eclipse.gef.editpolicies.LayoutEditPolicy; import org.eclipse.gef.requests.CreateRequest; import org.eclipse.gmf.runtime.diagram.ui.editparts.AbstractBorderedShapeEditPart; 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.DragDropEditPolicy; import org.eclipse.gmf.runtime.diagram.ui.editpolicies.EditPolicyRoles; import org.eclipse.gmf.runtime.diagram.ui.editpolicies.FlowLayoutEditPolicy; import org.eclipse.gmf.runtime.diagram.ui.figures.BorderItemLocator; import org.eclipse.gmf.runtime.draw2d.ui.figures.ConstrainedToolbarLayout; import org.eclipse.gmf.runtime.draw2d.ui.figures.WrappingLabel; 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.gmf.tooling.examples.affixednode.diagram.edit.policies.BundleCanonicalEditPolicy; import org.eclipse.gmf.tooling.examples.affixednode.diagram.edit.policies.BundleItemSemanticEditPolicy; import org.eclipse.gmf.tooling.examples.affixednode.diagram.part.AffixednodeVisualIDRegistry; import org.eclipse.gmf.tooling.runtime.edit.policies.reparent.CreationEditPolicyWithCustomReparent; import org.eclipse.swt.graphics.Color; /** * @generated */ public class BundleEditPart extends AbstractBorderedShapeEditPart { /** * @generated */ public static final int VISUAL_ID = 2001; /** * @generated */ protected IFigure contentPane; /** * @generated */ protected IFigure primaryShape; /** * @generated */ public BundleEditPart(View view) { super(view); } /** * @generated */ protected void createDefaultEditPolicies() { installEditPolicy(EditPolicyRoles.CREATION_ROLE, new CreationEditPolicyWithCustomReparent(AffixednodeVisualIDRegistry.TYPED_INSTANCE)); super.createDefaultEditPolicies(); installEditPolicy(EditPolicyRoles.SEMANTIC_ROLE, new BundleItemSemanticEditPolicy()); installEditPolicy(EditPolicyRoles.DRAG_DROP_ROLE, new DragDropEditPolicy()); installEditPolicy(EditPolicyRoles.CANONICAL_ROLE, new BundleCanonicalEditPolicy()); installEditPolicy(EditPolicy.LAYOUT_ROLE, createLayoutEditPolicy()); // 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() { FlowLayoutEditPolicy lep = new FlowLayoutEditPolicy() { protected EditPolicy createChildEditPolicy(EditPart child) { View childView = (View) child.getModel(); switch (AffixednodeVisualIDRegistry.getVisualID(childView)) { case IncomingCommunicatorEditPart.VISUAL_ID: case OutcomingCommunicatorEditPart.VISUAL_ID: return new BorderItemSelectionEditPolicy(); } return super.createChildEditPolicy(child); } protected Command createAddCommand(EditPart child, EditPart after) { return null; } protected Command createMoveChildCommand(EditPart child, EditPart after) { return null; } protected Command getCreateCommand(CreateRequest request) { return null; } }; return lep; } /** * @generated */ protected IFigure createNodeShape() { return primaryShape = new BundleFigure(); } /** * @generated */ public BundleFigure getPrimaryShape() { return (BundleFigure) primaryShape; } /** * @generated */ protected boolean addFixedChild(EditPart childEditPart) { if (childEditPart instanceof BundleNameEditPart) { ((BundleNameEditPart) childEditPart).setLabel(getPrimaryShape().getFigureBundleNameFigure()); return true; } if (childEditPart instanceof IncomingCommunicatorEditPart) { BorderItemLocator locator = new BorderItemLocator(getMainFigure(), PositionConstants.WEST); getBorderedFigure().getBorderItemContainer().add(((IncomingCommunicatorEditPart) childEditPart).getFigure(), locator); return true; } if (childEditPart instanceof OutcomingCommunicatorEditPart) { BorderItemLocator locator = new BorderItemLocator(getMainFigure(), PositionConstants.EAST); getBorderedFigure().getBorderItemContainer().add(((OutcomingCommunicatorEditPart) childEditPart).getFigure(), locator); return true; } return false; } /** * @generated */ protected boolean removeFixedChild(EditPart childEditPart) { if (childEditPart instanceof BundleNameEditPart) { return true; } if (childEditPart instanceof IncomingCommunicatorEditPart) { getBorderedFigure().getBorderItemContainer().remove(((IncomingCommunicatorEditPart) childEditPart).getFigure()); return true; } if (childEditPart instanceof OutcomingCommunicatorEditPart) { getBorderedFigure().getBorderItemContainer().remove(((OutcomingCommunicatorEditPart) childEditPart).getFigure()); return true; } return false; } /** * @generated */ protected void addChildVisual(EditPart childEditPart, int index) { if (addFixedChild(childEditPart)) { return; } super.addChildVisual(childEditPart, -1); } /** * @generated */ protected void removeChildVisual(EditPart childEditPart) { if (removeFixedChild(childEditPart)) { return; } super.removeChildVisual(childEditPart); } /** * @generated */ protected IFigure getContentPaneFor(IGraphicalEditPart editPart) { if (editPart instanceof IBorderItemEditPart) { return getBorderedFigure().getBorderItemContainer(); } return getContentPane(); } /** * @generated */ protected NodeFigure createNodePlate() { DefaultSizeNodeFigure result = new DefaultSizeNodeFigure(40, 40); 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) { if (nodeShape.getLayoutManager() == null) { ConstrainedToolbarLayout layout = new ConstrainedToolbarLayout(); layout.setSpacing(5); nodeShape.setLayoutManager(layout); } 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(AffixednodeVisualIDRegistry.getType(BundleNameEditPart.VISUAL_ID)); } /** * @generated */ public class BundleFigure extends RectangleFigure { /** * @generated */ private WrappingLabel fFigureBundleNameFigure; /** * @generated */ public BundleFigure() { FlowLayout layoutThis = new FlowLayout(); layoutThis.setStretchMinorAxis(false); layoutThis.setMinorAlignment(FlowLayout.ALIGN_LEFTTOP); layoutThis.setMajorAlignment(FlowLayout.ALIGN_LEFTTOP); layoutThis.setMajorSpacing(5); layoutThis.setMinorSpacing(5); layoutThis.setHorizontal(true); this.setLayoutManager(layoutThis); createContents(); } /** * @generated */ private void createContents() { WrappingLabel bUNDLE1 = new WrappingLabel(); bUNDLE1.setText("BUNDLE"); this.add(bUNDLE1); fFigureBundleNameFigure = new WrappingLabel(); fFigureBundleNameFigure.setText("<...>"); this.add(fFigureBundleNameFigure); } /** * @generated */ public WrappingLabel getFigureBundleNameFigure() { return fFigureBundleNameFigure; } } }