@Override
public Image getImage(Object element) {
EObject object = getBusinessObject(element);
if (object != null) {
CustomTaskDescriptor ctd = getCustomTaskDescriptor(object);
if (ctd != null) {
return PropertyUtil.getImage("CustomTask"); // $NON-NLS-1$
}
return PropertyUtil.getImage(object);
}
return super.getImage(element);
}
/**
* This method is called from refreshVisuals(), to display the image of the TreeItem.
*
* <p>By default this method displays the image of the FIRST attribute of the ModelObject as the
* TreeItem.
*
* @see org.eclipse.gef.editparts.AbstractEditPart#refreshVisuals()
*/
@Override
protected Image getImage() {
EObject o = (EObject) getModel();
Image img = labelProvider.getImage(o);
if (img != null) return img;
return PropertyUtil.getImage(o);
}