Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.modeling / src / org / simantics / modeling / export / DiscoverModels.java

diff --git a/bundles/org.simantics.modeling/src/org/simantics/modeling/export/DiscoverModels.java b/bundles/org.simantics.modeling/src/org/simantics/modeling/export/DiscoverModels.java
new file mode 100644 (file)
index 0000000..dffd96b
--- /dev/null
+++ b/bundles/org.simantics.modeling/src/org/simantics/modeling/export/DiscoverModels.java
@@ -0,0 +1,53 @@
+package org.simantics.modeling.export;\r
+\r
+import java.util.Collection;\r
+\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.common.utils.traverser.TraverseQueryBuilder;\r
+import org.simantics.db.common.utils.traverser.TraverseResult;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.db.request.Read;\r
+import org.simantics.export.core.ExportContext;\r
+import org.simantics.export.core.error.ExportException;\r
+import org.simantics.export.core.intf.DiscoverAction;\r
+import org.simantics.export.core.util.ExportQueries;\r
+import org.simantics.layer0.Layer0;\r
+import org.simantics.simulation.ontology.SimulationResource;\r
+\r
+/**\r
+ * Discovers all models in context ( session, project )\r
+ *\r
+ * @author toni.kalajainen@semantum.fi\r
+ */\r
+public class DiscoverModels implements DiscoverAction {\r
+\r
+       public Read<Collection<String>> discoverRequest(\r
+                       final Collection<String> startLocations) throws ExportException {\r
+               return new Read<Collection<String>>() {\r
+                       public Collection<String> perform(ReadGraph graph) throws DatabaseException {\r
+                               \r
+                               Layer0 L0 = Layer0.getInstance(graph);\r
+                               SimulationResource SIM = SimulationResource.getInstance(graph);  \r
+                               TraverseQueryBuilder builder = new TraverseQueryBuilder();\r
+                               builder.setStartResources( graph.syncRequest( ExportQueries.toResources(startLocations) ) );\r
+                               builder.followRelation( L0.ConsistsOf );\r
+                               builder.followAndReturnInstanceOf( SIM.Model );\r
+                               TraverseResult traverseResult = graph.syncRequest( builder.build() );\r
+                               return graph.syncRequest( ExportQueries.toUris( traverseResult.result ) );\r
+                       }\r
+                       \r
+               };\r
+       }\r
+\r
+       @Override\r
+       public Collection<String> discoverContent(ExportContext ctx, Collection<String> startLocations) throws ExportException {                \r
+               try {\r
+                       return ctx.session.syncRequest( discoverRequest(startLocations) );\r
+               } catch (DatabaseException e) {\r
+                       if ( e.getCause() instanceof ExportException ) throw (ExportException) e.getCause();\r
+                       throw new ExportException( e );\r
+               }\r
+       }\r
+       \r
+\r
+}\r