diff options
Diffstat (limited to 'plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.common/src/org/eclipse/papyrus/uml/diagram/common/editpolicies/AppliedStereotypeNodeLabelDisplayEditPolicy.java')
| -rw-r--r-- | plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.common/src/org/eclipse/papyrus/uml/diagram/common/editpolicies/AppliedStereotypeNodeLabelDisplayEditPolicy.java | 31 |
1 files changed, 29 insertions, 2 deletions
diff --git a/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.common/src/org/eclipse/papyrus/uml/diagram/common/editpolicies/AppliedStereotypeNodeLabelDisplayEditPolicy.java b/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.common/src/org/eclipse/papyrus/uml/diagram/common/editpolicies/AppliedStereotypeNodeLabelDisplayEditPolicy.java index 243b254cd64..89e4566efa3 100644 --- a/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.common/src/org/eclipse/papyrus/uml/diagram/common/editpolicies/AppliedStereotypeNodeLabelDisplayEditPolicy.java +++ b/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.common/src/org/eclipse/papyrus/uml/diagram/common/editpolicies/AppliedStereotypeNodeLabelDisplayEditPolicy.java @@ -1,5 +1,5 @@ /***************************************************************************** - * Copyright (c) 2009 CEA LIST. + * Copyright (c) 2009, 2014 CEA LIST and others. * * * All rights reserved. This program and the accompanying materials @@ -9,16 +9,21 @@ * * Contributors: * Patrick Tessier (CEA LIST) Patrick.tessier@cea.fr - Initial API and implementation + * Christian W. Damus (CEA) - bug 429275 * *****************************************************************************/ package org.eclipse.papyrus.uml.diagram.common.editpolicies; import org.eclipse.draw2d.IFigure; +import org.eclipse.gmf.runtime.notation.BooleanValueStyle; +import org.eclipse.gmf.runtime.notation.NotationPackage; import org.eclipse.gmf.runtime.notation.View; import org.eclipse.papyrus.infra.gmfdiag.common.editpart.IPapyrusEditPart; import org.eclipse.papyrus.uml.appearance.helper.AppliedStereotypeHelper; import org.eclipse.papyrus.uml.appearance.helper.UMLVisualInformationPapyrusConstant; import org.eclipse.papyrus.uml.diagram.common.Activator; +import org.eclipse.papyrus.uml.diagram.common.editparts.NamedElementEditPart; +import org.eclipse.papyrus.uml.diagram.common.figure.node.IPapyrusNodeNamedElementFigure; import org.eclipse.papyrus.uml.diagram.common.figure.node.IPapyrusNodeUMLElementFigure; import org.eclipse.papyrus.uml.tools.utils.StereotypeUtil; import org.eclipse.swt.graphics.Image; @@ -55,7 +60,12 @@ public class AppliedStereotypeNodeLabelDisplayEditPolicy extends AppliedStereoty // if (stereotypesToDisplay != "" || imageToDisplay != null) { if(figure instanceof IPapyrusNodeUMLElementFigure) { ((IPapyrusNodeUMLElementFigure)figure).setStereotypeDisplay(tag + (stereotypesOnlyToDisplay().equals("") ? stereotypesToDisplay : stereotypesToDisplay), imageToDisplay); - refreshAppliedStereotypesProperties(((IPapyrusNodeUMLElementFigure)figure)); + + if(figure instanceof IPapyrusNodeNamedElementFigure) { + refreshAppliedStereotypesProperties((IPapyrusNodeNamedElementFigure)figure); + } else { + refreshAppliedStereotypesProperties((IPapyrusNodeUMLElementFigure)figure); + } } // TODO we should manage PapyrusNodeFigure here too (and // WrappingLabel ?) @@ -65,6 +75,23 @@ public class AppliedStereotypeNodeLabelDisplayEditPolicy extends AppliedStereoty /** * Refreshes the displayed stereotypes properties for this edit part. */ + protected void refreshAppliedStereotypesProperties(IPapyrusNodeNamedElementFigure figure) { + final String stereotypesPropertiesToDisplay = AppliedStereotypeHelper.getAppliedStereotypesPropertiesToDisplay((View)getHost().getModel()); + + refreshAppliedStereotypesPropertiesInCompartment(stereotypesPropertiesToDisplay, figure); + refreshAppliedStereotypesPropertiesInBrace(stereotypesPropertiesToDisplay, figure); + + BooleanValueStyle displayStereotypes = (BooleanValueStyle) getView().getNamedStyle(NotationPackage.eINSTANCE.getBooleanValueStyle(), NamedElementEditPart.DISPLAY_STEREOTYPES); + if (displayStereotypes != null && !displayStereotypes.isBooleanValue()) { + figure.removeStereotypeLabel(); + } else { + figure.restoreStereotypeLabel(); + } + } + + /** + * Refreshes the displayed stereotypes properties for this edit part. + */ protected void refreshAppliedStereotypesProperties(IPapyrusNodeUMLElementFigure figure) { final String stereotypesPropertiesToDisplay = AppliedStereotypeHelper.getAppliedStereotypesPropertiesToDisplay((View)getHost().getModel()); |