diff options
Diffstat (limited to 'deprecated/org.eclipse.papyrus.uml.diagram.sequence_initial-kepler/custom-src/org/eclipse/papyrus/uml/diagram/sequence/edit/parts/AbstractMessageEditPart.java')
| -rw-r--r-- | deprecated/org.eclipse.papyrus.uml.diagram.sequence_initial-kepler/custom-src/org/eclipse/papyrus/uml/diagram/sequence/edit/parts/AbstractMessageEditPart.java | 194 |
1 files changed, 194 insertions, 0 deletions
diff --git a/deprecated/org.eclipse.papyrus.uml.diagram.sequence_initial-kepler/custom-src/org/eclipse/papyrus/uml/diagram/sequence/edit/parts/AbstractMessageEditPart.java b/deprecated/org.eclipse.papyrus.uml.diagram.sequence_initial-kepler/custom-src/org/eclipse/papyrus/uml/diagram/sequence/edit/parts/AbstractMessageEditPart.java new file mode 100644 index 00000000000..353cd16c66a --- /dev/null +++ b/deprecated/org.eclipse.papyrus.uml.diagram.sequence_initial-kepler/custom-src/org/eclipse/papyrus/uml/diagram/sequence/edit/parts/AbstractMessageEditPart.java @@ -0,0 +1,194 @@ +package org.eclipse.papyrus.uml.diagram.sequence.edit.parts;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import org.eclipse.draw2d.ConnectionLocator;
+import org.eclipse.draw2d.IFigure;
+import org.eclipse.draw2d.Shape;
+import org.eclipse.emf.common.notify.Notification;
+import org.eclipse.emf.ecore.EObject;
+import org.eclipse.gef.ConnectionEditPart;
+import org.eclipse.gef.EditPart;
+import org.eclipse.gef.Request;
+import org.eclipse.gef.requests.ReconnectRequest;
+import org.eclipse.gmf.runtime.diagram.ui.editparts.ConnectionNodeEditPart;
+import org.eclipse.gmf.runtime.diagram.ui.editparts.LabelEditPart;
+import org.eclipse.gmf.runtime.diagram.ui.l10n.DiagramColorRegistry;
+import org.eclipse.gmf.runtime.diagram.ui.requests.CreateUnspecifiedTypeConnectionRequest;
+import org.eclipse.gmf.runtime.notation.Diagram;
+import org.eclipse.gmf.runtime.notation.FontStyle;
+import org.eclipse.gmf.runtime.notation.NotationPackage;
+import org.eclipse.gmf.runtime.notation.View;
+import org.eclipse.papyrus.infra.gmfdiag.common.editpolicies.IMaskManagedLabelEditPolicy;
+import org.eclipse.papyrus.uml.diagram.common.editparts.UMLConnectionNodeEditPart;
+import org.eclipse.papyrus.uml.diagram.common.figure.edge.UMLEdgeFigure;
+import org.eclipse.papyrus.uml.diagram.sequence.edit.policies.MessageLabelEditPolicy;
+import org.eclipse.papyrus.uml.diagram.sequence.providers.UMLElementTypes;
+import org.eclipse.swt.graphics.Color;
+import org.eclipse.uml2.uml.Message;
+
+public abstract class AbstractMessageEditPart extends UMLConnectionNodeEditPart {
+
+ private List messageEventParts;
+
+ public AbstractMessageEditPart(View view) {
+ super(view);
+ }
+
+ public View findChildByModel(EObject model) {
+ List list = getModelChildren();
+ if(list != null && list.size() > 0) {
+ for(Object o : list) {
+ if(!(o instanceof View) )
+ continue;
+
+ View view = (View) o;
+ if(view.getElement() == model){
+ return view;
+ }
+ }
+ }
+ return null;
+ }
+
+ @Override
+ public List getChildren() {
+ if (messageEventParts == null) {
+ initMessageEventPart();
+ }
+ return super.getChildren();
+ }
+
+ protected void initMessageEventPart() {
+ messageEventParts = new ArrayList();
+
+ EObject element = this.resolveSemanticElement();
+ if (!(element instanceof Message))
+ return;
+ Message message = (Message) element;
+ UMLEdgeFigure edgeFigure = (UMLEdgeFigure) this.getFigure();
+ final MessageEndEditPart sendEventPart = new MessageEndEditPart(
+ message.getSendEvent(), this, new ConnectionLocator(edgeFigure,
+ ConnectionLocator.SOURCE));
+ messageEventParts.add(sendEventPart);
+
+ final MessageEndEditPart receiveEventPart = new MessageEndEditPart(
+ message.getReceiveEvent(), this, new ConnectionLocator(
+ edgeFigure, ConnectionLocator.TARGET));
+ messageEventParts.add(receiveEventPart);
+
+ Diagram diagram = ((View) this.getModel()).getDiagram();
+ sendEventPart.rebuildLinks(diagram);
+ receiveEventPart.rebuildLinks(diagram);
+
+ addChild(sendEventPart, -1);
+ addChild(receiveEventPart, -1);
+ }
+
+ @Override
+ protected void createDefaultEditPolicies() {
+ super.createDefaultEditPolicies();
+ installEditPolicy(IMaskManagedLabelEditPolicy.MASK_MANAGED_LABEL_EDIT_POLICY, new MessageLabelEditPolicy());
+ }
+
+ @Override
+ public EditPart getTargetEditPart(Request request) {
+ if(request instanceof CreateUnspecifiedTypeConnectionRequest){
+ List types = ((CreateUnspecifiedTypeConnectionRequest) request).getElementTypes();
+ if(types.contains(UMLElementTypes.Message_4009) || types.contains(UMLElementTypes.Message_4008)){
+ return null;
+ }
+ }else if(request instanceof ReconnectRequest){
+ ConnectionEditPart con = ((ReconnectRequest)request).getConnectionEditPart();
+ if(con instanceof Message7EditPart || con instanceof Message6EditPart){
+ return null;
+ }
+ }
+ return super.getTargetEditPart(request);
+ }
+
+ protected void handleNotificationEvent(Notification notification) {
+ super.handleNotificationEvent(notification);
+ Object feature = notification.getFeature();
+
+ MessageLabelEditPart labelPart = getMessageLabelEditPart();
+ if(labelPart == null)
+ return;
+ if(NotationPackage.eINSTANCE.getFontStyle_FontColor().equals(feature)) {
+ labelPart.refreshFontColor();
+ }else if(NotationPackage.eINSTANCE.getFontStyle_FontHeight().equals(feature) || NotationPackage.eINSTANCE.getFontStyle_FontName().equals(feature) || NotationPackage.eINSTANCE.getFontStyle_Bold().equals(feature) || NotationPackage.eINSTANCE.getFontStyle_Italic().equals(feature)) {
+ labelPart.refreshFont();
+ }
+ }
+
+ public MessageLabelEditPart getMessageLabelEditPart(){
+ for(Object c : this.getChildren())
+ if(c instanceof MessageLabelEditPart) {
+ return (MessageLabelEditPart)c;
+ }
+ return null;
+ }
+
+ //public abstract IFigure getPrimaryShape() ;
+
+ public void setLineWidth(int width) {
+ if(getPrimaryShape() instanceof MessageFigure){
+ MessageFigure edge = (MessageFigure)getPrimaryShape();
+ edge.setLineWidth(width);
+ }
+ }
+
+ public static class MessageFigure extends UMLEdgeFigure{
+ @Override
+ public void setLineWidth(int w) {
+ super.setLineWidth(w);
+ if(getSourceDecoration() instanceof Shape){
+ ((Shape)getSourceDecoration()).setLineWidth(w);
+ }
+ if(getTargetDecoration() instanceof Shape){
+ ((Shape)getTargetDecoration()).setLineWidth(w);
+ }
+ }
+
+ @Override
+ public void setForegroundColor(Color c) {
+ super.setForegroundColor(c);
+ if(getSourceDecoration() instanceof Shape){
+ ((Shape)getSourceDecoration()).setForegroundColor(c);
+ ((Shape)getSourceDecoration()).setBackgroundColor(c);
+ }
+ if(getTargetDecoration() instanceof Shape){
+ ((Shape)getTargetDecoration()).setForegroundColor(c);
+ ((Shape)getTargetDecoration()).setBackgroundColor(c);
+ }
+ }
+ }
+
+ static abstract class MessageLabelEditPart extends LabelEditPart {
+
+ public MessageLabelEditPart(View view) {
+ super(view);
+ }
+
+ protected void handleNotificationEvent(Notification notification) {
+ Object feature = notification.getFeature();
+ if(NotationPackage.eINSTANCE.getLineStyle_LineColor().equals(feature)) {
+ refreshFontColor();
+ } else
+ super.handleNotificationEvent(notification);
+ }
+
+ @Override
+ public void refreshFontColor() {
+ FontStyle style = (FontStyle)((org.eclipse.gmf.runtime.diagram.ui.editparts.ConnectionEditPart)getParent()).getPrimaryView().getStyle(NotationPackage.Literals.FONT_STYLE);
+ if(style != null) {
+ setFontColor(DiagramColorRegistry.getInstance().getColor(Integer.valueOf(style.getFontColor())));
+ }
+ }
+
+ public void refreshFont() {
+ super.refreshFont();
+ }
+ }
+}
|