package org.eclipse.papyrus.uml.diagram.component.sheet;
import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.gmf.runtime.emf.type.core.IElementType;
import org.eclipse.gmf.runtime.notation.View;
import org.eclipse.jface.viewers.BaseLabelProvider;
import org.eclipse.jface.viewers.ILabelProvider;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.papyrus.uml.diagram.component.part.UMLVisualIDRegistry;
import org.eclipse.papyrus.uml.diagram.component.providers.UMLElementTypes;
import org.eclipse.swt.graphics.Image;
/**
* @generated
*/
public class UMLSheetLabelProvider extends BaseLabelProvider implements ILabelProvider {
/**
* @generated
*/
public String getText(Object element) {
element = unwrap(element);
IElementType etype = getElementType(getView(element));
return etype == null ? "" : etype.getDisplayName();
}
/**
* @generated
*/
public Image getImage(Object element) {
IElementType etype = getElementType(getView(unwrap(element)));
return etype == null ? null : UMLElementTypes.getImage(etype);
}
/**
* @generated
*/
private Object unwrap(Object element) {
if(element instanceof IStructuredSelection) {
return ((IStructuredSelection)element).getFirstElement();
}
return element;
}
/**
* @generated
*/
private View getView(Object element) {
if(element instanceof View) {
return (View)element;
}
if(element instanceof IAdaptable) {
return (View)((IAdaptable)element).getAdapter(View.class);
}
return null;
}
/**
* @generated
*/
private IElementType getElementType(View view) {
// For intermediate views climb up the containment hierarchy to find the one associated with an element type.
while(view != null) {
int vid = UMLVisualIDRegistry.getVisualID(view);
IElementType etype = UMLElementTypes.getElementType(vid);
if(etype != null) {
return etype;
}
view = view.eContainer() instanceof View ? (View)view.eContainer() : null;
}
return null;
}
}