Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.modeling.template2d.ui / src / org / simantics / modeling / template2d / ui / wizard / DrawingTemplateExporter.java

diff --git a/bundles/org.simantics.modeling.template2d.ui/src/org/simantics/modeling/template2d/ui/wizard/DrawingTemplateExporter.java b/bundles/org.simantics.modeling.template2d.ui/src/org/simantics/modeling/template2d/ui/wizard/DrawingTemplateExporter.java
new file mode 100644 (file)
index 0000000..f72c210
--- /dev/null
+++ b/bundles/org.simantics.modeling.template2d.ui/src/org/simantics/modeling/template2d/ui/wizard/DrawingTemplateExporter.java
@@ -0,0 +1,119 @@
+/*******************************************************************************\r
+ * Copyright (c) 2012 Association for Decentralized Information Management in\r
+ * Industry THTH ry.\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.modeling.template2d.ui.wizard;\r
+\r
+import java.io.File;\r
+import java.io.IOException;\r
+import java.lang.reflect.InvocationTargetException;\r
+import java.util.Set;\r
+\r
+import org.eclipse.core.runtime.IProgressMonitor;\r
+import org.eclipse.core.runtime.NullProgressMonitor;\r
+import org.eclipse.core.runtime.SubMonitor;\r
+import org.eclipse.jface.operation.IRunnableWithProgress;\r
+import org.simantics.Simantics;\r
+import org.simantics.databoard.binding.error.BindingException;\r
+import org.simantics.databoard.binding.mutable.Variant;\r
+import org.simantics.databoard.container.DataContainer;\r
+import org.simantics.databoard.container.DataContainers;\r
+import org.simantics.databoard.serialization.SerializationException;\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.Session;\r
+import org.simantics.db.common.utils.Logger;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.db.layer0.adapter.CopyHandler;\r
+import org.simantics.db.layer0.util.ClipboardUtils;\r
+import org.simantics.db.layer0.util.SimanticsClipboard;\r
+import org.simantics.db.layer0.util.SimanticsClipboard.Representation;\r
+import org.simantics.db.layer0.util.SimanticsClipboardImpl;\r
+import org.simantics.db.layer0.util.SimanticsKeys;\r
+import org.simantics.db.request.Read;\r
+import org.simantics.graph.representation.TransferableGraph1;\r
+import org.simantics.modeling.template2d.DiagramTemplateConstants;\r
+import org.simantics.utils.ui.dialogs.ShowMessage;\r
+\r
+/**\r
+ * @author Tuukka Lehtonen\r
+ * @author Teemu Mätäsniemi\r
+ */\r
+public class DrawingTemplateExporter implements IRunnableWithProgress {\r
+\r
+    ExportPlan exportModel;\r
+\r
+    public DrawingTemplateExporter(ExportPlan exportModel) {\r
+        this.exportModel = exportModel;\r
+    }\r
+\r
+    @Override\r
+    public void run(IProgressMonitor monitor) throws InvocationTargetException, InterruptedException {\r
+        SubMonitor progress = SubMonitor.convert(monitor, 50);\r
+        try {\r
+            exportModel(progress.newChild(50, SubMonitor.SUPPRESS_NONE));\r
+        } catch (IOException e) {\r
+            throw new InvocationTargetException(e);\r
+        } catch (DatabaseException e) {\r
+            throw new InvocationTargetException(e);\r
+        } catch (BindingException e) {\r
+            throw new InvocationTargetException(e);\r
+        } finally {\r
+            monitor.done();\r
+        }\r
+    }\r
+\r
+    void exportModel(SubMonitor mon) throws IOException, DatabaseException, SerializationException, BindingException{\r
+       doExport(mon, exportModel.exportLocation, exportModel.model.getResource());\r
+       mon.setWorkRemaining(0);\r
+    }\r
+    \r
+    public static void doExport(File location, Resource template) throws DatabaseException, IOException {\r
+       NullProgressMonitor monitor = new NullProgressMonitor();\r
+       doExport(monitor, location, template);\r
+    }\r
+    \r
+    public static void doExport(IProgressMonitor mon, File location, final Resource template) throws DatabaseException, IOException {\r
+        try {\r
+            // TODO: figure out a way to make the TG go directly into a file\r
+            // instead of having it all in memory at once.\r
+               \r
+            mon.beginTask("Exporting diagram template...", 100);\r
+            Session session = Simantics.getSession();\r
+            SimanticsClipboard clipboard = session.syncRequest(new Read<SimanticsClipboard>() {\r
+                @Override\r
+                public SimanticsClipboard perform(ReadGraph graph) throws DatabaseException {\r
+                    CopyHandler ch = graph.adapt(template, CopyHandler.class);\r
+                    SimanticsClipboardImpl clipboard = new SimanticsClipboardImpl();\r
+                    ch.copyToClipboard(graph, clipboard);\r
+                    return clipboard;\r
+                }\r
+            });\r
+            for (Set<Representation> object : clipboard.getContents()) {\r
+                TransferableGraph1 tg = ClipboardUtils.accept(object, SimanticsKeys.KEY_TRANSFERABLE_GRAPH);\r
+                mon.worked(95);\r
+\r
+                mon.setTaskName("Writing transferable graph...");\r
+                DataContainers.writeFile(location, new DataContainer(\r
+                        DiagramTemplateConstants.DRAWING_TEMPLATE_FORMAT, DiagramTemplateConstants.DRAWING_TEMPLATE_CURRENT_VERSION,\r
+                        new Variant(TransferableGraph1.BINDING, tg)));\r
+\r
+                mon.worked(5);\r
+            }\r
+\r
+        } catch (DatabaseException e) {\r
+            e.printStackTrace();\r
+            Logger.defaultLogError(e);\r
+            mon.setCanceled(true);\r
+            ShowMessage.showError("Export failed.", "Internal application error in export. See log for details.");\r
+        }\r
+    }\r
+\r
+}\r