1 package org.simantics.district.imports.ui;
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;
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;
34 public class CSVImportWizardFirstPage extends WizardPage implements FileSelectionListener {
36 private static final Logger LOGGER = LoggerFactory.getLogger(CSVImportWizardFirstPage.class);
38 private final CSVImportModel model;
40 private Map<String, Resource> diagrams = new HashMap<>();
42 private Combo diagramCombo;
43 private FileSelectionWidget csvSelection;
45 public CSVImportWizardFirstPage(CSVImportModel model) {
46 super("Select source for CSV data");
48 setMessage("Select target network diagram and source for CSV-data");
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);
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);
61 diagramCombo = new Combo(parentConfigurationGroup, SWT.DROP_DOWN | SWT.READ_ONLY);
62 diagramCombo.addSelectionListener(new SelectionListener() {
65 public void widgetSelected(SelectionEvent e) {
66 widgetDefaultSelected(e);
70 public void widgetDefaultSelected(SelectionEvent e) {
71 String selection = diagramCombo.getItem(diagramCombo.getSelectionIndex());
72 model.setParentDiagram(diagrams.get(selection));
77 GridLayoutFactory.fillDefaults().numColumns(1).equalWidth(false).extendedMargins(12, 12, 12, 12).spacing(5, 4).applyTo(parentConfigurationGroup);
79 csvSelection = new FileSelectionWidget(composite, "CSV file", SWT.OPEN);
80 csvSelection.addListener(this);
82 updateAvailableParents();
86 setControl(composite);
89 private void updateAvailableParents() {
91 Simantics.getSession().syncRequest(new ReadRequest() {
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);
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());
113 } catch (DatabaseException e) {
114 LOGGER.error("Could not update available parents", e);
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");
126 model.setSource(csvFile);
132 private void validateComplete() {
133 setPageComplete(model.getSource() != null && model.getParentDiagram() != null);