1 /*******************************************************************************
\r
2 * Copyright (c) 2012 Association for Decentralized Information Management in
\r
4 * All rights reserved. This program and the accompanying materials
\r
5 * are made available under the terms of the Eclipse Public License v1.0
\r
6 * which accompanies this distribution, and is available at
\r
7 * http://www.eclipse.org/legal/epl-v10.html
\r
10 * VTT Technical Research Centre of Finland - initial API and implementation
\r
11 *******************************************************************************/
\r
12 package org.simantics.annotation.ui.wizard;
\r
14 import java.io.File;
\r
15 import java.io.IOException;
\r
16 import java.lang.reflect.InvocationTargetException;
\r
17 import java.util.Deque;
\r
18 import java.util.HashMap;
\r
20 import org.eclipse.core.runtime.IProgressMonitor;
\r
21 import org.eclipse.core.runtime.preferences.InstanceScope;
\r
22 import org.eclipse.jface.operation.IRunnableWithProgress;
\r
23 import org.eclipse.jface.preference.IPersistentPreferenceStore;
\r
24 import org.eclipse.jface.preference.IPreferenceStore;
\r
25 import org.eclipse.jface.viewers.IStructuredSelection;
\r
26 import org.eclipse.jface.wizard.Wizard;
\r
27 import org.eclipse.jface.wizard.WizardPage;
\r
28 import org.eclipse.ui.IImportWizard;
\r
29 import org.eclipse.ui.IWorkbench;
\r
30 import org.eclipse.ui.preferences.ScopedPreferenceStore;
\r
31 import org.simantics.annotation.ui.Activator;
\r
32 import org.simantics.databoard.binding.Binding;
\r
33 import org.simantics.databoard.container.DataContainer;
\r
34 import org.simantics.databoard.container.DataContainers;
\r
35 import org.simantics.databoard.container.DataFormatException;
\r
36 import org.simantics.databoard.container.FormatHandler;
\r
37 import org.simantics.databoard.serialization.SerializationException;
\r
38 import org.simantics.db.ReadGraph;
\r
39 import org.simantics.db.Resource;
\r
40 import org.simantics.db.Session;
\r
41 import org.simantics.db.exception.DatabaseException;
\r
42 import org.simantics.db.layer0.adapter.impl.DefaultPasteImportAdvisor;
\r
43 import org.simantics.db.management.ISessionContext;
\r
44 import org.simantics.graph.db.TransferableGraphs;
\r
45 import org.simantics.graph.representation.TransferableGraph1;
\r
46 import org.simantics.modeling.ui.utils.NoProjectPage;
\r
47 import org.simantics.project.IProject;
\r
48 import org.simantics.project.ProjectKeys;
\r
49 import org.simantics.ui.SimanticsUI;
\r
50 import org.simantics.ui.utils.ResourceAdaptionUtils;
\r
51 import org.simantics.utils.ui.ErrorLogger;
\r
52 import org.simantics.utils.ui.ExceptionUtils;
\r
55 * @author Tuukka Lehtonen
\r
57 public class AnnotationTypeImportWizard extends Wizard implements IImportWizard {
\r
59 private static final int MAX_RECENT_IMPORT_PATHS = 10;
\r
61 ImportPlan importModel;
\r
63 private boolean readPreferences(IStructuredSelection selection) {
\r
64 IPreferenceStore store = new ScopedPreferenceStore(InstanceScope.INSTANCE, Activator.PLUGIN_ID);
\r
66 String recentPathsPref = store.getString(Preferences.RECENT_ANNOTATION_TYPE_IMPORT_LOCATIONS);
\r
67 Deque<String> recentImportPaths = Preferences.decodePaths(recentPathsPref);
\r
69 ISessionContext ctx = SimanticsUI.getSessionContext();
\r
72 IProject project = ctx.getHint(ProjectKeys.KEY_PROJECT);
\r
73 if (project == null)
\r
76 importModel = new ImportPlan(ctx, recentImportPaths);
\r
77 importModel.project = project;
\r
78 importModel.selection = selection.getFirstElement();
\r
83 private void writePreferences() throws IOException {
\r
84 IPersistentPreferenceStore store = new ScopedPreferenceStore(InstanceScope.INSTANCE, Activator.PLUGIN_ID);
\r
86 store.putValue(Preferences.RECENT_ANNOTATION_TYPE_IMPORT_LOCATIONS, Preferences.encodePaths(importModel.recentLocations));
\r
88 if (store.needsSaving())
\r
92 public AnnotationTypeImportWizard() {
\r
93 setWindowTitle("Import Annotation Type");
\r
94 setNeedsProgressMonitor(true);
\r
98 public void init(IWorkbench workbench, IStructuredSelection selection) {
\r
99 readPreferences(selection);
\r
103 public void addPages() {
\r
105 if (importModel != null) {
\r
106 addPage(new AnnotationTypeImportPage(importModel));
\r
108 addPage(new NoProjectPage("Import Annotation Type"));
\r
113 public boolean performFinish() {
\r
115 importModel.recentLocations.addFirst(importModel.importLocation.getAbsolutePath());
\r
116 Preferences.removeDuplicates(importModel.recentLocations);
\r
117 if (importModel.recentLocations.size() > MAX_RECENT_IMPORT_PATHS)
\r
118 importModel.recentLocations.pollLast();
\r
120 writePreferences();
\r
121 } catch (IOException e) {
\r
122 ErrorLogger.defaultLogError("Failed to write preferences", e);
\r
126 getContainer().run(true, true, new IRunnableWithProgress() {
\r
128 public void run(IProgressMonitor monitor) throws InvocationTargetException, InterruptedException {
\r
130 Resource target = ResourceAdaptionUtils.toSingleResource(importModel.selection);
\r
131 doImport(monitor, importModel.importLocation, importModel.sessionContext.getSession(), target);
\r
132 } catch (Exception e) {
\r
133 throw new InvocationTargetException(e);
\r
137 } catch (InvocationTargetException e) {
\r
138 Throwable t = e.getTargetException();
\r
139 WizardPage cp = (WizardPage) getContainer().getCurrentPage();
\r
140 if (t instanceof IOException) {
\r
141 cp.setErrorMessage("An I/O problem occurred while importing annotation type.\n\nMessage: " + e.getMessage());
\r
143 ErrorLogger.defaultLogError(t);
\r
145 } catch (InterruptedException e) {
\r
146 ExceptionUtils.logAndShowError(e);
\r
154 public static void doImport(final IProgressMonitor monitor, File modelFile, final Session session, final Resource target)
\r
155 throws IOException, SerializationException, DatabaseException {
\r
157 final DefaultPasteImportAdvisor advisor = new DefaultPasteImportAdvisor(target);
\r
159 monitor.beginTask("Loading annotation type from disk", 1000);
\r
161 FormatHandler<Object> handler1 = new FormatHandler<Object>() {
\r
163 public Binding getBinding() {
\r
164 return TransferableGraph1.BINDING;
\r
168 public Object process(DataContainer container)
\r
170 monitor.worked(100);
\r
171 monitor.setTaskName("Importing annotation type into database");
\r
172 TransferableGraphs.importGraph1(session, (TransferableGraph1)container.content.getValue(),
\r
174 monitor.worked(850);
\r
179 HashMap<String, FormatHandler<Object>> handlers = new HashMap<String, FormatHandler<Object>>();
\r
180 handlers.put(Constants.ANNOTATION_TYPE_FORMAT_V1, handler1);
\r
183 DataContainers.readFile(modelFile, handlers);
\r
184 } catch(DataFormatException e) {
\r
185 throw new IOException(e);
\r
186 } catch(IOException e) {
\r
188 } catch(Exception e) {
\r
189 if(e instanceof RuntimeException)
\r
190 throw (RuntimeException)e;
\r
192 throw new RuntimeException(e);
\r
194 } catch(IOException e) {
\r
197 monitor.setTaskName("Postprocessing");
\r
198 monitor.worked(50);
\r
200 } catch (Throwable t) {
\r
201 t.printStackTrace();
\r