package org.eclipse.gmf.examples.ocldriven.toe.diagram.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.Shape; | |
import org.eclipse.draw2d.StackLayout; | |
import org.eclipse.draw2d.geometry.Dimension; | |
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.handles.MoveHandle; | |
import org.eclipse.gef.requests.CreateRequest; | |
import org.eclipse.gmf.examples.ocldriven.diagram.figures.StickManWithNumber; | |
import org.eclipse.gmf.examples.ocldriven.toe.diagram.edit.policies.ManagerItemSemanticEditPolicy; | |
import org.eclipse.gmf.examples.ocldriven.toe.diagram.edit.policies.ManagerPinBackgroundRGB2004Policy; | |
import org.eclipse.gmf.examples.ocldriven.toe.diagram.part.TOEVisualIDRegistry; | |
import org.eclipse.gmf.examples.ocldriven.toe.diagram.providers.TOEElementTypes; | |
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.EditPolicyRoles; | |
import org.eclipse.gmf.runtime.diagram.ui.figures.BorderItemLocator; | |
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; | |
/** | |
* @generated | |
*/ | |
public class ManagerEditPart extends AbstractBorderedShapeEditPart { | |
/** | |
* @generated | |
*/ | |
public static final int VISUAL_ID = 2004; | |
/** | |
* @generated | |
*/ | |
protected IFigure contentPane; | |
/** | |
* @generated | |
*/ | |
protected IFigure primaryShape; | |
/** | |
* @generated | |
*/ | |
public ManagerEditPart(View view) { | |
super(view); | |
} | |
/** | |
* @generated | |
*/ | |
protected void createDefaultEditPolicies() { | |
super.createDefaultEditPolicies(); | |
installEditPolicy(EditPolicyRoles.SEMANTIC_ROLE, new ManagerItemSemanticEditPolicy()); | |
installEditPolicy(EditPolicy.LAYOUT_ROLE, createLayoutEditPolicy()); | |
installEditPolicy(ManagerPinBackgroundRGB2004Policy.KEY, new ManagerPinBackgroundRGB2004Policy()); | |
// 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 (TOEVisualIDRegistry.getVisualID(childView)) { | |
case ManagerNameEditPart.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 ManagerFigure(); | |
} | |
/** | |
* @generated | |
*/ | |
public ManagerFigure getPrimaryShape() { | |
return (ManagerFigure) primaryShape; | |
} | |
/** | |
* @generated | |
*/ | |
protected void addBorderItem(IFigure borderItemContainer, IBorderItemEditPart borderItemEditPart) { | |
if (borderItemEditPart instanceof ManagerNameEditPart) { | |
BorderItemLocator locator = new BorderItemLocator(getMainFigure(), PositionConstants.SOUTH); | |
locator.setBorderItemOffset(new Dimension(-20, -20)); | |
borderItemContainer.add(borderItemEditPart.getFigure(), locator); | |
} else { | |
super.addBorderItem(borderItemContainer, borderItemEditPart); | |
} | |
} | |
/** | |
* @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) { | |
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(TOEVisualIDRegistry.getType(ManagerNameEditPart.VISUAL_ID)); | |
} | |
/** | |
* @generated | |
*/ | |
public List<IElementType> getMARelTypesOnSource() { | |
ArrayList<IElementType> types = new ArrayList<IElementType>(3); | |
types.add(TOEElementTypes.ManagerManagedDepartment_4001); | |
types.add(TOEElementTypes.ManagerLeads_4002); | |
types.add(TOEElementTypes.Contribution_4003); | |
return types; | |
} | |
/** | |
* @generated | |
*/ | |
public List<IElementType> getMARelTypesOnSourceAndTarget(IGraphicalEditPart targetEditPart) { | |
LinkedList<IElementType> types = new LinkedList<IElementType>(); | |
if (targetEditPart instanceof DepartmentEditPart) { | |
types.add(TOEElementTypes.ManagerManagedDepartment_4001); | |
} | |
if (targetEditPart instanceof Department2EditPart) { | |
types.add(TOEElementTypes.ManagerManagedDepartment_4001); | |
} | |
if (targetEditPart instanceof ProjectEditPart) { | |
types.add(TOEElementTypes.ManagerLeads_4002); | |
} | |
if (targetEditPart instanceof ProjectEditPart) { | |
types.add(TOEElementTypes.Contribution_4003); | |
} | |
return types; | |
} | |
/** | |
* @generated | |
*/ | |
public List<IElementType> getMATypesForTarget(IElementType relationshipType) { | |
LinkedList<IElementType> types = new LinkedList<IElementType>(); | |
if (relationshipType == TOEElementTypes.ManagerManagedDepartment_4001) { | |
types.add(TOEElementTypes.Department_2002); | |
types.add(TOEElementTypes.Department_3002); | |
} else if (relationshipType == TOEElementTypes.ManagerLeads_4002) { | |
types.add(TOEElementTypes.Project_2003); | |
} else if (relationshipType == TOEElementTypes.Contribution_4003) { | |
types.add(TOEElementTypes.Project_2003); | |
} | |
return types; | |
} | |
/** | |
* @generated | |
*/ | |
public class ManagerFigure extends StickManWithNumber { | |
/** | |
* @generated | |
*/ | |
public ManagerFigure() { | |
this.setForegroundColor(ColorConstants.red); | |
this.setBackgroundColor(ColorConstants.yellow); | |
} | |
} | |
} |