private void updateIcon(EObject element) {
String className = entry.getMessage();
IEditorPart editorPart =
PlatformUI.getWorkbench().getActiveWorkbenchWindow().getActivePage().getActiveEditor();
if (editorPart instanceof IDiagramWorkbenchPart) {
DiagramEditPart diagramPart = ((IDiagramWorkbenchPart) editorPart).getDiagramEditPart();
List<EditPart> editPartCollector = new ArrayList<EditPart>();
Neuro4jDiagramEditorUtil.findElementsInDiagramByID(diagramPart, element, editPartCollector);
if (editPartCollector.size() > 0) {
LogicNodeEditPart editpart = (LogicNodeEditPart) editPartCollector.get(0);
editpart.updateImageForClass(className);
}
// editPartCollector
}
}
