1 files changed, 87 insertions, 0 deletions
diff --git a/plugins/infra/gmfdiag/org.eclipse.papyrus.infra.gmfdiag.common/src/org/eclipse/papyrus/infra/gmfdiag/common/providers/NotationLabelProvider.java b/plugins/infra/gmfdiag/org.eclipse.papyrus.infra.gmfdiag.common/src/org/eclipse/papyrus/infra/gmfdiag/common/providers/NotationLabelProvider.java
new file mode 100644
index 00000000000..18e3c9f2c76
--- /dev/null
+++ b/plugins/infra/gmfdiag/org.eclipse.papyrus.infra.gmfdiag.common/src/org/eclipse/papyrus/infra/gmfdiag/common/providers/NotationLabelProvider.java
@@ -0,0 +1,87 @@
+/*****************************************************************************
+ * Copyright (c) 2012 CEA LIST.
+ * 
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * which accompanies this distribution, and is available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ *
+ * Contributors:
+ *  Camille Letavernier (CEA LIST) camille.letavernier@cea.fr - Initial API and implementation
+ *****************************************************************************/
+package org.eclipse.papyrus.infra.gmfdiag.common.providers;
+
+import org.eclipse.emf.ecore.EObject;
+import org.eclipse.gef.EditPart;
+import org.eclipse.gmf.runtime.diagram.ui.editparts.ResizableCompartmentEditPart;
+import org.eclipse.gmf.runtime.diagram.ui.services.editpart.EditPartService;
+import org.eclipse.gmf.runtime.notation.BasicCompartment;
+import org.eclipse.gmf.runtime.notation.DecorationNode;
+import org.eclipse.gmf.runtime.notation.Diagram;
+import org.eclipse.gmf.runtime.notation.View;
+import org.eclipse.papyrus.infra.core.editorsfactory.IPageIconsRegistry;
+import org.eclipse.papyrus.infra.core.editorsfactory.PageIconsRegistry;
+import org.eclipse.papyrus.infra.core.services.ServiceException;
+import org.eclipse.papyrus.infra.emf.providers.EMFLabelProvider;
+import org.eclipse.papyrus.infra.emf.utils.ServiceUtilsForEObject;
+import org.eclipse.papyrus.uml.tools.Activator;
+import org.eclipse.swt.graphics.Image;
+
+/**
+ * A Label Provider for GMF Notation model
+ */
+public class NotationLabelProvider extends EMFLabelProvider {
+
+	/** icon for a compartment */
+	public static final String ICON_COMPARTMENT = "/icons/none_comp_vis.gif"; //$NON-NLS-1$
+
+	@Override
+	protected Image getImage(EObject element) {
+		if(element instanceof Diagram) {
+			IPageIconsRegistry registry = null;
+			try {
+				registry = ServiceUtilsForEObject.getInstance().getService(IPageIconsRegistry.class, element);
+			} catch (ServiceException e) {
+				// nothing to do
+			}
+			if(registry == null) {
+				registry = new PageIconsRegistry();
+			}
+			return registry.getEditorIcon(element);
+		}
+
+		// if the element is a compartment
+		if(element instanceof BasicCompartment || element instanceof DecorationNode) {
+			return org.eclipse.papyrus.infra.widgets.Activator.getDefault().getImage(Activator.PLUGIN_ID, ICON_COMPARTMENT);
+		}
+
+		return super.getImage(element);
+	}
+
+	/**
+	 * 
+	 * @see org.eclipse.jface.viewers.ILabelProvider#getText(java.lang.Object)
+	 * 
+	 * @param element
+	 * @return <ul>
+	 *         <li>if element is a {@link NamedElement}, we return its name</li>
+	 *         <li>else if element is a {@link Element}, we return its type + a index</li>
+	 *         <li>else return Messages#EditorLabelProvider_No_name</li>
+	 *         </ul>
+	 */
+	@Override
+	protected String getText(EObject element) {
+		if(element instanceof Diagram) {
+			return ((Diagram)element).getName();
+		}
+
+		if(element instanceof View) { // maybe it is a view of a compartment
+			EditPart dummyEP = EditPartService.getInstance().createGraphicEditPart((View)element);
+			if(dummyEP instanceof ResizableCompartmentEditPart) {
+				return ((ResizableCompartmentEditPart)dummyEP).getCompartmentName();
+			}
+		}
+
+		return super.getText(element);
+	}
+}