Use indexRoot directly for searching vertex/edge mappings

[simantics/district.git] / org.simantics.district.imports.ui / src / org / simantics / district / imports / ui / ComponentMappingPage.java

diff --git a/org.simantics.district.imports.ui/src/org/simantics/district/imports/ui/ComponentMappingPage.java b/org.simantics.district.imports.ui/src/org/simantics/district/imports/ui/ComponentMappingPage.java
index 5cf5551194a9fab2274fbe69edc8d70a62c1eec6..c74603be90b96949059cb1a659f1cd057f3b1976 100644 (file)
--- a/org.simantics.district.imports.ui/src/org/simantics/district/imports/ui/ComponentMappingPage.java
+++ b/org.simantics.district.imports.ui/src/org/simantics/district/imports/ui/ComponentMappingPage.java
@@ -20,6 +20,7 @@ import org.eclipse.swt.widgets.Label;
 import org.simantics.Simantics;
 import org.simantics.db.ReadGraph;
 import org.simantics.db.Resource;
+import org.simantics.db.common.request.IndexRoot;
 import org.simantics.db.common.request.ReadRequest;
 import org.simantics.db.exception.DatabaseException;
 import org.simantics.district.imports.DistrictImportUtils;
@@ -74,10 +75,11 @@ public class ComponentMappingPage extends WizardPage {
                     
                     @Override
                     public void run(ReadGraph graph) throws DatabaseException {
+                        Resource indexRoot = graph.sync(new IndexRoot(model.getParentDiagram()));
                         if (model.isVertexImport())
-                            componentMappings = Functions.getVertexMappings(graph, model.getParentDiagram());
+                            componentMappings = Functions.getVertexMappings(graph, indexRoot);
                         else
-                            componentMappings = Functions.getEdgeMappings(graph, model.getParentDiagram());
+                            componentMappings = Functions.getEdgeMappings(graph, indexRoot);
                     }
                 });
             } catch (DatabaseException e) {