private Diagram getDiagramForSemanticElement(EObject state) {
Collection<Diagram> diagrams =
EcoreUtil2.getObjectsByType(
state.eResource().getContents(), NotationPackage.Literals.DIAGRAM);
for (Diagram diagram : diagrams) {
if (EcoreUtil.equals(diagram.getElement(), state)) {
return diagram;
}
}
return null;
}
