Allow Diagram API to work for Document instances.

[simantics/interop.git] / org.simantics.interop.diagram / src / org / simantics / interop / diagram / DiagramUtils.java

diff --git a/org.simantics.interop.diagram/src/org/simantics/interop/diagram/DiagramUtils.java b/org.simantics.interop.diagram/src/org/simantics/interop/diagram/DiagramUtils.java
index 29f3e21186761cc846daecffa2b22e1f2a0beed0..6a1aa1010cd496eb4ed11d78d46dd6e987122fae 100644 (file)
--- a/org.simantics.interop.diagram/src/org/simantics/interop/diagram/DiagramUtils.java
+++ b/org.simantics.interop.diagram/src/org/simantics/interop/diagram/DiagramUtils.java
@@ -78,6 +78,8 @@ public class DiagramUtils {
                while (r != null) {
                        if (graph.isInstanceOf(r, sim.Model))
                                return r;
+                       if (graph.isInstanceOf(r, l0.SharedOntology))
+                               return r;
                        r = graph.getPossibleObject(r, l0.PartOf);
                }
                return r;
@@ -85,6 +87,6 @@ public class DiagramUtils {
        
        public static Resource getConfiguration(ReadGraph graph, Resource model) throws DatabaseException {
                SimulationResource sim = SimulationResource.getInstance(graph);
-               return graph.getSingleObject(model, sim.HasConfiguration);
+               return graph.getPossibleObject(model, sim.HasConfiguration);
        }
 }