org.simantics.g3d/src/org/simantics/g3d/wizard/ModelExportWizardPage.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2012, 2013 Association for Decentralized Information Management in\r
   3  * Industry THTH ry.\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
   8  *\r
   9  * Contributors:\r
  10  *     VTT Technical Research Centre of Finland - initial API and implementation\r
  11  *******************************************************************************/\r
  12 package org.simantics.g3d.wizard;\r
  13 \r
  14 import java.io.File;\r
  15 import java.util.ArrayList;\r
  16 import java.util.Collections;\r
  17 import java.util.List;\r
  18 \r
  19 import org.eclipse.jface.layout.GridDataFactory;\r
  20 import org.eclipse.jface.resource.ImageDescriptor;\r
  21 import org.eclipse.jface.wizard.WizardPage;\r
  22 import org.eclipse.swt.SWT;\r
  23 import org.eclipse.swt.custom.CCombo;\r
  24 import org.eclipse.swt.events.ModifyEvent;\r
  25 import org.eclipse.swt.events.ModifyListener;\r
  26 import org.eclipse.swt.events.SelectionAdapter;\r
  27 import org.eclipse.swt.events.SelectionEvent;\r
  28 import org.eclipse.swt.layout.GridData;\r
  29 import org.eclipse.swt.layout.GridLayout;\r
  30 import org.eclipse.swt.widgets.Button;\r
  31 import org.eclipse.swt.widgets.Composite;\r
  32 import org.eclipse.swt.widgets.DirectoryDialog;\r
  33 import org.eclipse.swt.widgets.FileDialog;\r
  34 import org.eclipse.swt.widgets.Label;\r
  35 import org.simantics.Simantics;\r
  36 import org.simantics.db.ReadGraph;\r
  37 import org.simantics.db.Resource;\r
  38 import org.simantics.db.common.NamedResource;\r
  39 import org.simantics.db.common.request.ReadRequest;\r
  40 import org.simantics.db.exception.DatabaseException;\r
  41 import org.simantics.layer0.Layer0;\r
  42 import org.simantics.ui.utils.ResourceAdaptionUtils;\r
  43 \r
  44 \r
  45 public abstract class ModelExportWizardPage<T extends IExportModel> extends WizardPage{\r
  46         T exportData;\r
  47 \r
  48         CCombo model;\r
  49         CCombo exportLocation;\r
  50         \r
  51         boolean exportToFile = true;\r
  52 \r
  53         private List<NamedResource> models = new ArrayList<NamedResource>();\r
  54 \r
  55         private Button overwrite;\r
  56         \r
  57         \r
  58         \r
  59         public ModelExportWizardPage(String pageName, String title, ImageDescriptor titleImage, T data) {\r
  60                 super(pageName, title, titleImage);\r
  61                 this.exportData = data;\r
  62                 exportToFile = data.usesFile();\r
  63         }\r
  64         \r
  65         protected abstract List<NamedResource> getSupportedModels(ReadGraph graph, Resource project) throws DatabaseException;\r
  66         \r
  67         public String[] getFilterExtensions() {\r
  68                 return new String[0];\r
  69         }\r
  70         \r
  71         public String[] getFilterNames() {\r
  72                 return new String[0];\r
  73         }\r
  74 \r
  75         @Override\r
  76         public void createControl(Composite parent) {\r
  77                 Composite container = new Composite(parent, SWT.NONE);\r
  78                 {\r
  79                         GridLayout layout = new GridLayout();\r
  80                         layout.horizontalSpacing = 20;\r
  81                         layout.verticalSpacing = 10;\r
  82                         layout.numColumns = 3;\r
  83                         container.setLayout(layout);\r
  84                 }\r
  85 \r
  86                 new Label(container, SWT.NONE).setText("Exported &model:");\r
  87                 model = new CCombo(container, SWT.BORDER);\r
  88                 {\r
  89                         model.setEditable(false);\r
  90                         model.setText("");\r
  91                         model.setToolTipText("Selects the model to export.");\r
  92                         GridDataFactory.fillDefaults().grab(true, false).span(2, 1).applyTo(model);\r
  93                         model.addSelectionListener(new SelectionAdapter() {\r
  94                                 @Override\r
  95                                 public void widgetSelected(SelectionEvent e) {\r
  96                                         exportData.setModel(((NamedResource) model.getData(Integer.toString(model.getSelectionIndex()))));\r
  97                                         validatePage();\r
  98                                 }\r
  99                         });\r
 100 \r
 101                 }\r
 102                 if (exportToFile) {\r
 103                          new Label(container, SWT.NONE).setText("&Target file:");\r
 104                      exportLocation = new CCombo(container, SWT.BORDER);\r
 105                      {\r
 106                          exportLocation.setText("");\r
 107                          GridDataFactory.fillDefaults().grab(true, false).span(1, 1).applyTo(exportLocation);\r
 108                          exportLocation.addModifyListener(new ModifyListener(){\r
 109                              @Override\r
 110                              public void modifyText(ModifyEvent e) {\r
 111                                  validatePage();\r
 112                              }\r
 113                          });\r
 114                      }\r
 115                      Button browseFileButton = new Button(container, SWT.PUSH);\r
 116                      {\r
 117                          browseFileButton.setText("Browse...");\r
 118                          browseFileButton.setLayoutData(new GridData(SWT.RIGHT, SWT.CENTER, false, false));\r
 119                          browseFileButton.addSelectionListener(new SelectionAdapter() {\r
 120                              @Override\r
 121                              public void widgetSelected(SelectionEvent e) {\r
 122                                  FileDialog dialog = new FileDialog(getShell(), SWT.SAVE);\r
 123                                  dialog.setText("Choose Export Target File");\r
 124                                  dialog.setFilterExtensions(getFilterExtensions());\r
 125                                  dialog.setFilterNames(getFilterNames());\r
 126                                  String loc = exportLocation.getText();\r
 127                                  dialog.setFilterPath(loc);\r
 128                                  String file = dialog.open();\r
 129                                  if (file == null)\r
 130                                      return;\r
 131                                  exportLocation.setText(file);\r
 132                                  validatePage();\r
 133                              }\r
 134                          });\r
 135                      }\r
 136 \r
 137                     \r
 138 \r
 139                     \r
 140                 } else {\r
 141                      new Label(container, SWT.NONE).setText("&Target folder:");\r
 142                      exportLocation = new CCombo(container, SWT.BORDER);\r
 143                      {\r
 144                          exportLocation.setText("");\r
 145                          GridDataFactory.fillDefaults().grab(true, false).span(1, 1).applyTo(exportLocation);\r
 146                          exportLocation.addModifyListener(new ModifyListener(){\r
 147                              @Override\r
 148                              public void modifyText(ModifyEvent e) {\r
 149                                  validatePage();\r
 150                              }\r
 151                          });\r
 152                      }\r
 153                      Button browseFileButton = new Button(container, SWT.PUSH);\r
 154                      {\r
 155                          browseFileButton.setText("Browse...");\r
 156                          browseFileButton.setLayoutData(new GridData(SWT.RIGHT, SWT.CENTER, false, false));\r
 157                          browseFileButton.addSelectionListener(new SelectionAdapter() {\r
 158                              @Override\r
 159                              public void widgetSelected(SelectionEvent e) {\r
 160                                  DirectoryDialog dialog = new DirectoryDialog(getShell(), SWT.SAVE);\r
 161                                  dialog.setText("Choose Export Target Folder");\r
 162                                  String loc = exportLocation.getText();\r
 163                                  dialog.setFilterPath(loc);\r
 164                                  String file = dialog.open();\r
 165                                  if (file == null)\r
 166                                      return;\r
 167                                  exportLocation.setText(file);\r
 168                                  validatePage();\r
 169                              }\r
 170                          });\r
 171                      }\r
 172                 }\r
 173                 \r
 174                  Label horizRule = new Label(container, SWT.BORDER);\r
 175              GridDataFactory.fillDefaults().hint(SWT.DEFAULT, 0).grab(true, false).span(3, 1).applyTo(horizRule);\r
 176                 \r
 177                  overwrite = new Button(container, SWT.CHECK);\r
 178              overwrite.setText("&Overwrite existing files without warning");\r
 179              overwrite.setSelection(exportData.isOverwrite());\r
 180              GridDataFactory.fillDefaults().grab(true, false).span(3, 1).applyTo(overwrite);\r
 181              overwrite.addSelectionListener(new SelectionAdapter() {\r
 182                  @Override\r
 183                  public void widgetSelected(SelectionEvent e) {\r
 184                      validatePage();\r
 185                  }\r
 186              });\r
 187 \r
 188              try {\r
 189                  initializeData();\r
 190              } catch (DatabaseException e) {\r
 191                  e.printStackTrace();\r
 192              }\r
 193 \r
 194              setControl(container);\r
 195              validatePage();\r
 196                 \r
 197         }\r
 198         \r
 199 \r
 200         \r
 201         \r
 202         protected void initializeData() throws DatabaseException {\r
 203                   final Resource selection = ResourceAdaptionUtils.toSingleResource(exportData.getSelection());\r
 204 \r
 205              Simantics.getSessionContext().getSession().syncRequest(\r
 206                      new ReadRequest() {\r
 207                          @Override\r
 208                          public void run(ReadGraph graph) throws DatabaseException {\r
 209                              Layer0 L0 = Layer0.getInstance(graph);\r
 210                              models = getSupportedModels(graph, Simantics.getProject().get());\r
 211                          }\r
 212                      });\r
 213 \r
 214                      Collections.sort(models);\r
 215 \r
 216                      // Populate combo boxes\r
 217                      int i = 0;\r
 218                      boolean selected = false;\r
 219                      for (NamedResource s : models) {\r
 220                          model.add(s.getName());\r
 221                          model.setData(String.valueOf(i), s);\r
 222                          if (s.getResource().equals(selection)) {\r
 223                              model.select(i);\r
 224                              selected = true;\r
 225                          }\r
 226                          ++i;\r
 227                      }\r
 228                      if (!selected && i > 0)\r
 229                          model.select(0);\r
 230                      \r
 231                      if (model.getSelectionIndex() >= 0) {\r
 232                          exportData.setModel((NamedResource)model.getData(Integer.toString(model.getSelectionIndex())));\r
 233                      }\r
 234                      for (String path : exportData.getRecentLocations()) {\r
 235                          exportLocation.add(path);\r
 236                      }\r
 237                      if (exportLocation.getItemCount() > 0)\r
 238                          exportLocation.select(0);\r
 239         }\r
 240         \r
 241         \r
 242         \r
 243         protected void validatePage() {\r
 244                 if (exportData.getModel() == null) {\r
 245                         setMessage("Select model to export.");\r
 246                         setErrorMessage(null);\r
 247                         setPageComplete(false);\r
 248                         return;\r
 249                 }\r
 250                 String exportLoc = exportLocation.getText();\r
 251                 File file;\r
 252                 if (exportToFile) {\r
 253 \r
 254                         if (exportLoc.isEmpty()) {\r
 255                                 setMessage("Select target file.");\r
 256                                 setErrorMessage(null);\r
 257                                 setPageComplete(false);\r
 258                                 return;\r
 259                         }\r
 260                         file = new File(exportLoc);\r
 261                         if (file.exists() && !file.isFile()) {\r
 262                                 setErrorMessage("The target must be a file, an existing directory was given.");\r
 263                                 setPageComplete(false);\r
 264                                 return;\r
 265                         }\r
 266                         File parent = file.getParentFile();\r
 267                         if (parent == null || !parent.isDirectory()) {\r
 268                                 setErrorMessage("The target directory does not exist.");\r
 269                                 setPageComplete(false);\r
 270                                 return;\r
 271                         }\r
 272 \r
 273                 } else {\r
 274                         if (exportLoc.isEmpty()) {\r
 275                                 setMessage("Select target directory.");\r
 276                                 setErrorMessage(null);\r
 277                                 setPageComplete(false);\r
 278                                 return;\r
 279                         }\r
 280                         file = new File(exportLoc);\r
 281                         if (file.exists() && !file.isDirectory()) {\r
 282                                 setErrorMessage("The target must be a directory, an existing file was given.");\r
 283                                 setPageComplete(false);\r
 284                                 return;\r
 285                         }\r
 286 \r
 287                 }\r
 288 \r
 289                 exportData.setExportLocation(file);\r
 290                 exportData.setOverwrite(overwrite.getSelection());\r
 291 \r
 292                 setErrorMessage(null);\r
 293                 setMessage("Export selected model.");\r
 294                 setPageComplete(true);\r
 295         }\r
 296 }\r