org.simantics.district.imports.ui/src/org/simantics/district/imports/ui/CSVImportWizardFirstPage.java

   1 package org.simantics.district.imports.ui;
   2
   3 import java.nio.file.Files;
   4 import java.nio.file.Path;
   5 import java.nio.file.Paths;
   6 import java.util.Collection;
   7 import java.util.HashMap;
   8 import java.util.Map;
   9
  10 import org.eclipse.jface.layout.GridDataFactory;
  11 import org.eclipse.jface.layout.GridLayoutFactory;
  12 import org.eclipse.jface.wizard.WizardPage;
  13 import org.eclipse.swt.SWT;
  14 import org.eclipse.swt.events.SelectionEvent;
  15 import org.eclipse.swt.events.SelectionListener;
  16 import org.eclipse.swt.layout.GridLayout;
  17 import org.eclipse.swt.widgets.Combo;
  18 import org.eclipse.swt.widgets.Composite;
  19 import org.eclipse.swt.widgets.Group;
  20 import org.simantics.Simantics;
  21 import org.simantics.db.ReadGraph;
  22 import org.simantics.db.Resource;
  23 import org.simantics.db.common.request.ReadRequest;
  24 import org.simantics.db.exception.DatabaseException;
  25 import org.simantics.district.imports.CSVImportModel;
  26 import org.simantics.district.network.ui.function.Functions;
  27 import org.simantics.modeling.ModelingResources;
  28 import org.simantics.utils.ui.widgets.FileOrDirectorySelectionWidget;
  29 import org.simantics.utils.ui.widgets.FileSelectionListener;
  30 import org.simantics.utils.ui.widgets.FileSelectionWidget;
  31 import org.slf4j.Logger;
  32 import org.slf4j.LoggerFactory;
  33
  34 public class CSVImportWizardFirstPage extends WizardPage implements FileSelectionListener {
  35
  36     private static final Logger LOGGER = LoggerFactory.getLogger(CSVImportWizardFirstPage.class);
  37
  38     private final CSVImportModel model;
  39
  40     private Map<String, Resource> diagrams = new HashMap<>();
  41
  42     private Combo diagramCombo;
  43     private FileSelectionWidget csvSelection;
  44
  45     public CSVImportWizardFirstPage(CSVImportModel model) {
  46         super("Select source for CSV data");
  47         this.model = model;
  48         setMessage("Select target network diagram and source for CSV-data");
  49     }
  50
  51     @Override
  52     public void createControl(Composite parent) {
  53         Composite composite = new Composite(parent, SWT.NONE);
  54         composite.setLayout(new GridLayout(1,false));
  55         GridDataFactory.fillDefaults().grab(true, true).align(SWT.FILL, SWT.FILL).applyTo(composite);
  56
  57         Group parentConfigurationGroup = new Group(composite, SWT.NONE);
  58         parentConfigurationGroup.setText("Select target network diagram");
  59         GridDataFactory.fillDefaults().grab(true, false).span(2, 1).applyTo(parentConfigurationGroup);
  60
  61         diagramCombo = new Combo(parentConfigurationGroup, SWT.DROP_DOWN | SWT.READ_ONLY);
  62         diagramCombo.addSelectionListener(new SelectionListener() {
  63
  64             @Override
  65             public void widgetSelected(SelectionEvent e) {
  66                 widgetDefaultSelected(e);
  67             }
  68
  69             @Override
  70             public void widgetDefaultSelected(SelectionEvent e) {
  71                 String selection = diagramCombo.getItem(diagramCombo.getSelectionIndex());
  72                 model.setParentDiagram(diagrams.get(selection));
  73                 validateComplete();
  74             }
  75         });
  76
  77         GridLayoutFactory.fillDefaults().numColumns(1).equalWidth(false).extendedMargins(12, 12, 12, 12).spacing(5, 4).applyTo(parentConfigurationGroup);
  78
  79         csvSelection = new FileSelectionWidget(composite, "CSV file", SWT.OPEN);
  80         csvSelection.addListener(this);
  81
  82         updateAvailableParents();
  83
  84         validateComplete();
  85
  86         setControl(composite);
  87     }
  88
  89     private void updateAvailableParents() {
  90         try {
  91             Simantics.getSession().syncRequest(new ReadRequest() {
  92
  93                 @Override
  94                 public void run(ReadGraph graph) throws DatabaseException {
  95                     Collection<Resource> diagrams = Functions.getDistrictDiagrams(graph);
  96                     ModelingResources MOD = ModelingResources.getInstance(graph);
  97                     Resource projectResource = Simantics.getProjectResource();
  98                     String projectURI = graph.getURI(projectResource);
  99                     for (Resource diagram : diagrams) {
 100                         Resource composite = graph.getSingleObject(diagram, MOD.DiagramToComposite);
 101                         String compositeURI = graph.getURI(composite);
 102                         String path = compositeURI.replace(projectURI, "");
 103                         CSVImportWizardFirstPage.this.diagrams.put(path, diagram);
 104                     }
 105                 }
 106             });
 107             diagramCombo.setItems(diagrams.keySet().toArray(new String[diagrams.size()]));
 108             if (diagrams.size() > 0) {
 109                 diagramCombo.select(0);
 110                 model.setParentDiagram(diagrams.get(diagramCombo.getItem(0)));
 111                 diagramCombo.getDisplay().asyncExec(() -> validateComplete());
 112             }
 113         } catch (DatabaseException e) {
 114             LOGGER.error("Could not update available parents", e);
 115         }
 116     }
 117
 118     @Override
 119     public void fileSelected(FileOrDirectorySelectionWidget source, String[] filename) {
 120         String[] selection = csvSelection.getFilename();
 121         if (selection != null && selection.length > 0) {
 122             Path csvFile = Paths.get(selection[0]);
 123             if (!Files.exists(csvFile)) {
 124                 setErrorMessage("File " + csvFile.toAbsolutePath() + " does not exist");
 125             } else {
 126                 model.setSource(csvFile);
 127                 validateComplete();
 128             }
 129         }
 130     }
 131
 132     private void validateComplete() {
 133         setPageComplete(model.getSource() != null && model.getParentDiagram() != null);
 134     }
 135 }