git-svn-id: https://www.simantics.org/svn/simantics/3d/trunk@22283 ac1ea38d-2e2b...

[simantics/3d.git] / org.simantics.g3d.shapeeditor / src / org / simantics / proconf / g3d / shapeeditor / actions / ExportAction.java

diff --git a/org.simantics.g3d.shapeeditor/src/org/simantics/proconf/g3d/shapeeditor/actions/ExportAction.java b/org.simantics.g3d.shapeeditor/src/org/simantics/proconf/g3d/shapeeditor/actions/ExportAction.java
new file mode 100644 (file)
index 0000000..152bcf9
--- /dev/null
+++ b/org.simantics.g3d.shapeeditor/src/org/simantics/proconf/g3d/shapeeditor/actions/ExportAction.java
@@ -0,0 +1,110 @@
+/*******************************************************************************\r
+ * Copyright (c) 2007- VTT Technical Research Centre of Finland.\r
+ * All rights reserved. This program and the accompanying materials\r
+ * are made available under the terms of the Eclipse Public License v1.0\r
+ * which accompanies this distribution, and is available at\r
+ * http://www.eclipse.org/legal/epl-v10.html\r
+ *\r
+ * Contributors:\r
+ *     VTT Technical Research Centre of Finland - initial API and implementation\r
+ *******************************************************************************/\r
+package org.simantics.proconf.g3d.shapeeditor.actions;\r
+\r
+import java.io.File;\r
+import java.io.FileOutputStream;\r
+import java.io.IOException;\r
+import java.util.Collection;\r
+\r
+import org.eclipse.jface.action.Action;\r
+import org.eclipse.swt.SWT;\r
+import org.eclipse.swt.widgets.FileDialog;\r
+import org.simantics.db.Graph;\r
+import org.simantics.db.GraphRequestAdapter;\r
+import org.simantics.db.GraphRequestStatus;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.Statement;\r
+import org.simantics.layer0.utils.direct.GraphUtils;\r
+import org.simantics.layer0.utils.extent.Extent;\r
+import org.simantics.layer0.utils.extent.ExtentUtils;\r
+import org.simantics.layer0.utils.extent.IExtentAdvisor;\r
+import org.simantics.layer0.utils.serialization.ConnectionPointMap;\r
+import org.simantics.layer0.utils.serialization.TransferableGraph;\r
+import org.simantics.layer0.utils.serialization.TransferableGraphUtils;\r
+import org.simantics.proconf.g3d.shapeeditor.views.ShapeEditorBase;\r
+import org.simantics.utils.ui.ErrorLogger;\r
+\r
+\r
+public class ExportAction extends Action{\r
+       private ShapeEditorBase parent;\r
+       \r
+       public ExportAction(ShapeEditorBase parent) {\r
+               this.parent = parent;\r
+               this.setText("Export");\r
+               this.setId("g3d shape export");\r
+       }\r
+       \r
+       @Override\r
+       public void run() {\r
+               try {\r
+                       doExport();\r
+               } catch (IOException e) {\r
+                       ErrorLogger.defaultLogError(e);\r
+               }\r
+       }\r
+       \r
+       private void doExport() throws IOException {\r
+               FileDialog dialog = new FileDialog(parent.getRenderingComposite().getShell(),SWT.SAVE);\r
+               String filename = dialog.open();\r
+               if (filename == null)\r
+                       return;\r
+               \r
+               final File file = new File(filename);\r
+               final FileOutputStream fos = new FileOutputStream(file);\r
+               \r
+               parent.getSession().asyncRead(new GraphRequestAdapter() {\r
+                       @Override\r
+                       public GraphRequestStatus perform(Graph g) throws Exception {\r
+                               Resource modelResource = parent.getModelResource();\r
+                               System.out.println("Exporting " + modelResource);\r
+                               Collection<Statement> model = ExtentUtils.determineExtent(g, new ExtentAdvisor(), modelResource);\r
+                               \r
+                               for (Statement s : model)                \r
+                                       System.out.println(GraphUtils.getReadableName(g, s.getSubject())+", "+GraphUtils.getReadableName(g, s.getPredicate())+", "+GraphUtils.getReadableName(g, s.getObject()));                \r
+                               \r
+                               ConnectionPointMap purposeProvider = new ConnectionPointMap();\r
+                               purposeProvider.put(modelResource, TransferableGraphUtils.CP_OBJECT);\r
+                               \r
+                               TransferableGraph dbIndependentSubgraph = TransferableGraphUtils.extractTransferableGraph(g, model, purposeProvider, null); \r
+                               \r
+                               byte[] data = TransferableGraphUtils.serialize(dbIndependentSubgraph);\r
+                               \r
+                               fos.write(data);\r
+                               \r
+                               System.out.println("Exporting done.");\r
+                               return GraphRequestStatus.transactionComplete();\r
+                       }\r
+                       \r
+                       @Override\r
+                       public void requestCompleted(GraphRequestStatus status) {\r
+                               try {\r
+                                       fos.close();\r
+                               } catch (IOException e) {\r
+                                       ErrorLogger.defaultLogError(e);\r
+                               }\r
+                               \r
+                       }\r
+               });\r
+       }\r
+       \r
+       public class ExtentAdvisor implements IExtentAdvisor {\r
+               @Override\r
+               public ExtentAdvice getAdvice(Graph g, Extent currentState, Resource extent) {\r
+                       System.out.println("Extent advice : " + currentState + " : " + extent);\r
+                       \r
+                       //if (extent.equals(ShapeEditorResources.equationResource.Expression)) return ExtentAdvice.Include;\r
+                       //return ExtentAdvice.Exclude;\r
+                       return ExtentAdvice.Include;\r
+               }\r
+       }\r
+\r
+}\r