Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.annotation.ui / src / org / simantics / annotation / ui / wizard / AnnotationTypeExporter.java

diff --git a/bundles/org.simantics.annotation.ui/src/org/simantics/annotation/ui/wizard/AnnotationTypeExporter.java b/bundles/org.simantics.annotation.ui/src/org/simantics/annotation/ui/wizard/AnnotationTypeExporter.java
new file mode 100644 (file)
index 0000000..64274d0
--- /dev/null
+++ b/bundles/org.simantics.annotation.ui/src/org/simantics/annotation/ui/wizard/AnnotationTypeExporter.java
@@ -0,0 +1,118 @@
+/*******************************************************************************\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.annotation.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.utils.ui.dialogs.ShowMessage;\r
+\r
+/**\r
+ * @author Tuukka Lehtonen\r
+ * @author Teemu Mätäsniemi\r
+ */\r
+public class AnnotationTypeExporter implements IRunnableWithProgress {\r
+\r
+    ExportPlan exportModel;\r
+\r
+    public AnnotationTypeExporter(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
+        try {\r
+            doExport(mon, exportModel.exportLocation, exportModel.model.getResource());\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
+        } finally {\r
+            mon.setWorkRemaining(0);\r
+        }\r
+    }\r
+    \r
+    public static void doExport(File location, Resource annotation)  throws DatabaseException, IOException {\r
+       NullProgressMonitor monitor = new NullProgressMonitor();\r
+       doExport(monitor, location, annotation);\r
+    }\r
+    \r
+    public static void doExport(IProgressMonitor monitor, File location, final Resource annotation)  throws DatabaseException, IOException {\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
+        monitor.beginTask("Exporting annotation type...", 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(annotation, 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
+            monitor.worked(95);\r
+\r
+            monitor.setTaskName("Writing transferable graph...");\r
+            DataContainers.writeFile(location, new DataContainer(\r
+                    Constants.ANNOTATION_TYPE_FORMAT, Constants.ANNOTATION_TYPE_CURRENT_VERSION,\r
+                    new Variant(TransferableGraph1.BINDING, tg)));\r
+\r
+            monitor.worked(5);\r
+        }\r
+    }          \r
+}\r