/** @see LabelProvider#getImage(Object) */ @Override public Image getImage(Object element) { if (element instanceof IStructuredSelection) { element = ((IStructuredSelection) element).getFirstElement(); } element = TransformationTreeUtil.extractObject(element); if (element instanceof Entity) { element = ((Entity) element).getDefinition(); } if (element instanceof EntityDefinition || element instanceof Definition<?>) { return definitionLabels.getImage(element); } if (element instanceof Cell) { Cell cell = (Cell) element; AbstractFunction<?> function = FunctionUtil.getFunction(cell.getTransformationIdentifier()); if (function != null) { element = function; } } if (element instanceof Function) { return functionLabels.getImage(element); } return super.getImage(element); }