-package org.simantics.district.imports.ui;\r
-\r
-import java.io.IOException;\r
-import java.nio.file.Files;\r
-import java.nio.file.Path;\r
-import java.nio.file.Paths;\r
-import java.util.Collections;\r
-import java.util.List;\r
-import java.util.Map;\r
-\r
-import org.eclipse.jface.layout.GridDataFactory;\r
-import org.eclipse.jface.layout.TableColumnLayout;\r
-import org.eclipse.jface.layout.TreeColumnLayout;\r
-import org.eclipse.jface.viewers.ColumnViewerToolTipSupport;\r
-import org.eclipse.jface.viewers.ColumnWeightData;\r
-import org.eclipse.jface.viewers.IContentProvider;\r
-import org.eclipse.jface.viewers.TableViewer;\r
-import org.eclipse.jface.viewers.TreeViewer;\r
-import org.eclipse.jface.viewers.Viewer;\r
-import org.eclipse.jface.wizard.WizardPage;\r
-import org.eclipse.swt.SWT;\r
-import org.eclipse.swt.events.SelectionEvent;\r
-import org.eclipse.swt.events.SelectionListener;\r
-import org.eclipse.swt.layout.GridData;\r
-import org.eclipse.swt.layout.GridLayout;\r
-import org.eclipse.swt.widgets.Button;\r
-import org.eclipse.swt.widgets.Combo;\r
-import org.eclipse.swt.widgets.Composite;\r
-import org.eclipse.swt.widgets.Layout;\r
-import org.eclipse.swt.widgets.Table;\r
-import org.eclipse.swt.widgets.TableColumn;\r
-import org.eclipse.swt.widgets.TableItem;\r
-import org.simantics.district.imports.DistrictImportUtils.CSVHeader;\r
-import org.simantics.utils.ui.widgets.FileOrDirectorySelectionWidget;\r
-import org.simantics.utils.ui.widgets.FileSelectionListener;\r
-import org.simantics.utils.ui.widgets.FileSelectionWidget;\r
-\r
-public class CSVImportWizardPage extends WizardPage implements FileSelectionListener {\r
-\r
- private CSVImportModel model;\r
- private FileSelectionWidget csvSelection;\r
- private Table headerTable;\r
- private Button firstAsHeader;\r
- private Combo delimiterCombo;\r
- private TableColumnLayout tableColumnLayout;\r
- private TableViewer tableViewer;\r
- private Composite tableComposite;\r
-\r
- protected CSVImportWizardPage(CSVImportModel model) {\r
- super("Import CSV Data");\r
- this.model = model;\r
- setMessage("Select CSV file to import");\r
- }\r
-\r
- @Override\r
- public void createControl(Composite parent) {\r
- Composite composite = new Composite(parent, SWT.NONE);\r
- composite.setLayout(new GridLayout(1,false));\r
- GridDataFactory.fillDefaults().grab(true, true).align(SWT.FILL, SWT.FILL).applyTo(composite);\r
- \r
- csvSelection = new FileSelectionWidget(composite, "CSV file", SWT.OPEN);\r
- csvSelection.addListener(this);\r
- \r
- delimiterCombo = new Combo(composite, SWT.DROP_DOWN | SWT.READ_ONLY);\r
- delimiterCombo.setItems(model.getDelimiterFormats());\r
- delimiterCombo.addSelectionListener(new SelectionListener() {\r
- \r
- @Override\r
- public void widgetSelected(SelectionEvent e) {\r
- model.setDelimiterByLabel(delimiterCombo.getItem(delimiterCombo.getSelectionIndex()));\r
- updateHeaders();\r
- }\r
- \r
- @Override\r
- public void widgetDefaultSelected(SelectionEvent e) {\r
- \r
- }\r
- });\r
- \r
- firstAsHeader = new Button(composite, SWT.CHECK);\r
- firstAsHeader.setText("Read first row as header");\r
- firstAsHeader.setSelection(model.getReadFirstAsHeader());\r
- firstAsHeader.addSelectionListener(new SelectionListener() {\r
- \r
- @Override\r
- public void widgetSelected(SelectionEvent e) {\r
- model.setReadFirstAsHeader(firstAsHeader.getSelection());\r
- updateHeaders();\r
- }\r
- \r
- @Override\r
- public void widgetDefaultSelected(SelectionEvent e) {\r
- \r
- }\r
- });\r
- \r
- \r
- tableComposite = new Composite(composite, SWT.BORDER);\r
- tableColumnLayout = new TableColumnLayout();\r
- tableComposite.setLayout(tableColumnLayout);\r
-\r
- updateHeaders();\r
- \r
- setControl(composite);\r
- \r
- \r
- \r
- setPageComplete(false);\r
- }\r
-\r
- @Override\r
- public void fileSelected(FileOrDirectorySelectionWidget source, String[] filename) {\r
- String[] selection = csvSelection.getFilename();\r
- if (selection != null && selection.length > 0) {\r
- Path csvFile = Paths.get(selection[0]);\r
- if (!Files.exists(csvFile)) {\r
- setErrorMessage("File " + csvFile.toAbsolutePath() + " does not exist");\r
- } else {\r
- model.setSource(csvFile);\r
- updateHeaders();\r
- setPageComplete(true);\r
- }\r
- } else {\r
- setPageComplete(false);\r
- }\r
- }\r
-\r
- private void updateHeaders() {\r
- if (headerTable != null)\r
- headerTable.dispose();\r
- headerTable = new Table(tableComposite, SWT.NONE);\r
- headerTable.setHeaderVisible(true);\r
- headerTable.setLinesVisible(true);\r
- GridDataFactory.fillDefaults().grab(true, true).align(SWT.FILL, SWT.FILL).applyTo(tableComposite);\r
- for (int i = 0; i < headerTable.getColumns().length; i++) {\r
- TableColumn column = headerTable.getColumns()[i];\r
- column.dispose();\r
- }\r
- for (int i = 0; i < headerTable.getItemCount(); i++) {\r
- TableItem item = headerTable.getItem(i);\r
- item.dispose();\r
- }\r
- try {\r
- Map<CSVHeader, List<String>> headerAndValues = model.getHeaderAndRows(5);\r
- \r
- int coo = headerTable.getColumns().length;\r
- int count = headerTable.getColumnCount();\r
- \r
- for (Map.Entry<CSVHeader, List<String>> entr : headerAndValues.entrySet()) {\r
- CSVHeader he = entr.getKey();\r
- int index = he.getIndex();\r
- TableColumn headerCol = new TableColumn(headerTable, SWT.NONE);\r
- headerCol.setText(he.getHeader());\r
- \r
- tableColumnLayout.setColumnData(headerCol, new ColumnWeightData(10));\r
- headerCol.pack();\r
- \r
- for (String val : entr.getValue()) {\r
- TableItem item = new TableItem(headerTable, SWT.NONE);\r
- item.setText(val);\r
- }\r
- }\r
- } catch (IOException e) {\r
- setErrorMessage(e.getMessage());\r
- }\r
- }\r
-\r
-}\r
+package org.simantics.district.imports.ui;
+
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+
+import org.apache.commons.csv.CSVRecord;
+import org.eclipse.jface.dialogs.IPageChangeProvider;
+import org.eclipse.jface.dialogs.IPageChangedListener;
+import org.eclipse.jface.dialogs.PageChangedEvent;
+import org.eclipse.jface.layout.GridDataFactory;
+import org.eclipse.jface.layout.TableColumnLayout;
+import org.eclipse.jface.viewers.ColumnWeightData;
+import org.eclipse.jface.wizard.IWizardContainer;
+import org.eclipse.jface.wizard.WizardPage;
+import org.eclipse.swt.SWT;
+import org.eclipse.swt.events.SelectionEvent;
+import org.eclipse.swt.events.SelectionListener;
+import org.eclipse.swt.layout.GridLayout;
+import org.eclipse.swt.widgets.Button;
+import org.eclipse.swt.widgets.Combo;
+import org.eclipse.swt.widgets.Composite;
+import org.eclipse.swt.widgets.Group;
+import org.eclipse.swt.widgets.Table;
+import org.eclipse.swt.widgets.TableColumn;
+import org.eclipse.swt.widgets.TableItem;
+import org.simantics.district.imports.ui.controls.DynamicComboFieldEditor;
+import org.simantics.utils.ui.widgets.FileOrDirectorySelectionWidget;
+import org.simantics.utils.ui.widgets.FileSelectionListener;
+import org.simantics.utils.ui.widgets.FileSelectionWidget;
+
+public class CSVImportWizardPage extends WizardPage {
+
+ private CSVImportModel model;
+
+ private Map<Integer, String> headerIndexAndValues = new HashMap<>();
+
+ private Table headerTable;
+ private Button firstAsHeader;
+
+ private Combo delimiterCombo;
+ private TableColumnLayout tableColumnLayout;
+ private Composite tableComposite;
+ private FileSelectionWidget wktFileSelection;
+
+ // Common for vertex and edge
+ private DynamicComboFieldEditor componentMappingSelector;
+ private DynamicComboFieldEditor labelSelector;
+
+ // For vertex import
+ private DynamicComboFieldEditor xCoordSelector;
+ private DynamicComboFieldEditor yCoordSelector;
+ private DynamicComboFieldEditor zValueSelector;
+
+ // For edge import
+ private DynamicComboFieldEditor startXCoordSelector;
+ private DynamicComboFieldEditor startYCoordSelector;
+ private DynamicComboFieldEditor startZValueSelector;
+ private DynamicComboFieldEditor endXCoordSelector;
+ private DynamicComboFieldEditor endYCoordSelector;
+ private DynamicComboFieldEditor endZValueSelector;
+
+ private Group indexMappingGroup;
+
+ private Composite composite;
+
+ private Button isVertexImport;
+
+ protected CSVImportWizardPage(CSVImportModel model) {
+ super("Import CSV Data");
+ this.model = model;
+ setMessage("Select columng index mappings");
+ }
+
+ @Override
+ public void createControl(Composite parent) {
+ composite = new Composite(parent, SWT.NONE);
+ composite.setLayout(new GridLayout(1, false));
+ GridDataFactory.fillDefaults().grab(true, true).align(SWT.FILL, SWT.FILL).applyTo(composite);
+
+ delimiterCombo = new Combo(composite, SWT.DROP_DOWN | SWT.READ_ONLY);
+ delimiterCombo.setItems(model.getDelimiterFormats());
+ delimiterCombo.addSelectionListener(new SelectionListener() {
+
+ @Override
+ public void widgetSelected(SelectionEvent e) {
+ model.setDelimiterByLabel(delimiterCombo.getItem(delimiterCombo.getSelectionIndex()));
+ updateHeaders();
+ updateCombos();
+ }
+
+ @Override
+ public void widgetDefaultSelected(SelectionEvent e) {
+ widgetSelected(e);
+ }
+ });
+
+ firstAsHeader = new Button(composite, SWT.CHECK);
+ firstAsHeader.setText("Read first row as header");
+ firstAsHeader.setSelection(model.getReadFirstAsHeader());
+ firstAsHeader.addSelectionListener(new SelectionListener() {
+
+ @Override
+ public void widgetSelected(SelectionEvent e) {
+ model.setReadFirstAsHeader(firstAsHeader.getSelection());
+ updateHeaders();
+ updateCombos();
+ }
+
+ @Override
+ public void widgetDefaultSelected(SelectionEvent e) {
+
+ }
+ });
+
+ tableComposite = new Composite(composite, SWT.BORDER);
+ tableColumnLayout = new TableColumnLayout();
+ tableComposite.setLayout(tableColumnLayout);
+
+ wktFileSelection = new FileSelectionWidget(composite, "WKT file", SWT.OPEN);
+ wktFileSelection.addListener(new FileSelectionListener() {
+
+ @Override
+ public void fileSelected(FileOrDirectorySelectionWidget source, String[] filename) {
+ String[] selection = wktFileSelection.getFilename();
+ if (selection != null && selection.length > 0) {
+ Path wktFile = Paths.get(selection[0]);
+ if (!Files.exists(wktFile)) {
+ setErrorMessage("File " + wktFile.toAbsolutePath() + " does not exist");
+ } else {
+ model.setWKTFile(wktFile);
+ validatePageComplete();
+ }
+ }
+ }
+ });
+
+ isVertexImport = new Button(composite, SWT.CHECK);
+ isVertexImport.setText("File contains vertices");
+ isVertexImport.setSelection(model.isVertexImport());
+ isVertexImport.addSelectionListener(new SelectionListener() {
+
+ @Override
+ public void widgetSelected(SelectionEvent e) {
+ model.setVertexImport(isVertexImport.getSelection());
+ updateControls(false);
+ }
+
+ @Override
+ public void widgetDefaultSelected(SelectionEvent e) {
+ widgetSelected(e);
+ }
+ });
+
+ updateControls(true);
+ updateHeaders();
+ setControl(composite);
+
+ final IWizardContainer container = getContainer();
+
+ if (container instanceof IPageChangeProvider) {
+ ((IPageChangeProvider) container).addPageChangedListener(new IPageChangedListener() {
+
+ @Override
+ public void pageChanged(PageChangedEvent event) {
+ if (isCurrentPage())
+ CSVImportWizardPage.this.updateControls(false);
+ }
+ });
+ }
+
+ validatePageComplete();
+ }
+
+ private void updateControls(boolean initial) {
+ createIndexMappingGroup();
+ updateCombos();
+ if (!initial)
+ composite.layout(true, true);
+ }
+
+ private void createIndexMappingGroup() {
+ if (indexMappingGroup != null)
+ indexMappingGroup.dispose();
+
+ indexMappingGroup = new Group(composite, SWT.NONE);
+ indexMappingGroup.setText("Column index mapping");
+ GridDataFactory.fillDefaults().grab(true, false).span(2, 1).applyTo(indexMappingGroup);
+
+ if (model.isVertexImport())
+ createVertexIndexMappingField(indexMappingGroup);
+ else
+ createEdgeIndexMappingField(indexMappingGroup);
+
+ createCommonIndexMappingField(indexMappingGroup);
+ }
+
+ private void createCommonIndexMappingField(Group parent) {
+
+ componentMappingSelector = new DynamicComboFieldEditor("componentMapping", "Apros component mapping", parent);
+ componentMappingSelector.addComboListener(new SelectionListener() {
+
+ @Override
+ public void widgetSelected(SelectionEvent e) {
+ widgetDefaultSelected(e);
+ }
+
+ @Override
+ public void widgetDefaultSelected(SelectionEvent e) {
+ model.setComponentMappingIndex(Integer.parseInt(componentMappingSelector.getValue()));
+ validatePageComplete();
+ }
+ });
+
+ }
+
+ private void createVertexIndexMappingField(Group parent) {
+ xCoordSelector = new DynamicComboFieldEditor("xCoord", "X Coordinate", parent);
+ xCoordSelector.addComboListener(new SelectionListener() {
+
+ @Override
+ public void widgetSelected(SelectionEvent e) {
+ widgetDefaultSelected(e);
+ }
+
+ @Override
+ public void widgetDefaultSelected(SelectionEvent e) {
+ model.setXCoordIndex(Integer.parseInt(xCoordSelector.getValue()));
+ validatePageComplete();
+ }
+ });
+
+ yCoordSelector = new DynamicComboFieldEditor("yCoord", "Y Coordinate", parent);
+ yCoordSelector.addComboListener(new SelectionListener() {
+
+ @Override
+ public void widgetSelected(SelectionEvent e) {
+ widgetDefaultSelected(e);
+ }
+
+ @Override
+ public void widgetDefaultSelected(SelectionEvent e) {
+ model.setYCoordIndex(Integer.parseInt(yCoordSelector.getValue()));
+ validatePageComplete();
+ }
+ });
+ zValueSelector = new DynamicComboFieldEditor("zValue", "Z Value", parent);
+ zValueSelector.addComboListener(new SelectionListener() {
+
+ @Override
+ public void widgetSelected(SelectionEvent e) {
+ widgetDefaultSelected(e);
+ }
+
+ @Override
+ public void widgetDefaultSelected(SelectionEvent e) {
+ model.setZCoordIndex(Integer.parseInt(zValueSelector.getValue()));
+ validatePageComplete();
+ }
+ });
+ }
+
+ private void createEdgeIndexMappingField(Group parent) {
+ startXCoordSelector = new DynamicComboFieldEditor("startxCoord", "Start X Coordinate", parent);
+ startXCoordSelector.addComboListener(new SelectionListener() {
+
+ @Override
+ public void widgetSelected(SelectionEvent e) {
+ widgetDefaultSelected(e);
+ }
+
+ @Override
+ public void widgetDefaultSelected(SelectionEvent e) {
+ model.setStartXCoordIndex(Integer.parseInt(startXCoordSelector.getValue()));
+ validatePageComplete();
+ }
+ });
+
+ startYCoordSelector = new DynamicComboFieldEditor("startyCoord", "Start Y Coordinate", parent);
+ startYCoordSelector.addComboListener(new SelectionListener() {
+
+ @Override
+ public void widgetSelected(SelectionEvent e) {
+ widgetDefaultSelected(e);
+ }
+
+ @Override
+ public void widgetDefaultSelected(SelectionEvent e) {
+ model.setStartYCoordIndex(Integer.parseInt(startYCoordSelector.getValue()));
+ validatePageComplete();
+ }
+ });
+ startZValueSelector = new DynamicComboFieldEditor("startzValue", "Start Z Value", parent);
+ startZValueSelector.addComboListener(new SelectionListener() {
+
+ @Override
+ public void widgetSelected(SelectionEvent e) {
+ widgetDefaultSelected(e);
+ }
+
+ @Override
+ public void widgetDefaultSelected(SelectionEvent e) {
+ model.setStartZCoordIndex(Integer.parseInt(startZValueSelector.getValue()));
+ validatePageComplete();
+ }
+ });
+
+ endXCoordSelector = new DynamicComboFieldEditor("endxCoord", "End X Coordinate", parent);
+ endXCoordSelector.addComboListener(new SelectionListener() {
+
+ @Override
+ public void widgetSelected(SelectionEvent e) {
+ widgetDefaultSelected(e);
+ }
+
+ @Override
+ public void widgetDefaultSelected(SelectionEvent e) {
+ model.setEndXCoordIndex(Integer.parseInt(endXCoordSelector.getValue()));
+ validatePageComplete();
+ }
+ });
+
+ endYCoordSelector = new DynamicComboFieldEditor("endyCoord", "End Y Coordinate", parent);
+ endYCoordSelector.addComboListener(new SelectionListener() {
+
+ @Override
+ public void widgetSelected(SelectionEvent e) {
+ widgetDefaultSelected(e);
+ }
+
+ @Override
+ public void widgetDefaultSelected(SelectionEvent e) {
+ model.setEndYCoordIndex(Integer.parseInt(endYCoordSelector.getValue()));
+ validatePageComplete();
+ }
+ });
+ endZValueSelector = new DynamicComboFieldEditor("endzValue", "End Z Value", parent);
+ endZValueSelector.addComboListener(new SelectionListener() {
+
+ @Override
+ public void widgetSelected(SelectionEvent e) {
+ widgetDefaultSelected(e);
+ }
+
+ @Override
+ public void widgetDefaultSelected(SelectionEvent e) {
+ model.setEndZCoordIndex(Integer.parseInt(endZValueSelector.getValue()));
+ validatePageComplete();
+ }
+ });
+ }
+
+ private void updateCombos() {
+ String[][] namesAndValues = new String[headerIndexAndValues.size()][];
+
+ int i = 0;
+ for (Entry<Integer, String> entry : headerIndexAndValues.entrySet()) {
+ int key = entry.getKey();
+ String value = entry.getValue();
+
+ String[] nameAndValue = new String[2];
+ nameAndValue[0] = value;
+ nameAndValue[1] = Integer.toString(key);
+ namesAndValues[i++] = nameAndValue;
+ }
+
+ if (model.isVertexImport())
+ updateVertexcombos(namesAndValues);
+ else
+ updateEdgeCombos(namesAndValues);
+
+ componentMappingSelector.updateCombo(namesAndValues);
+ }
+
+ private void updateEdgeCombos(String[][] namesAndValues) {
+ startXCoordSelector.updateCombo(namesAndValues);
+ endXCoordSelector.updateCombo(namesAndValues);
+ startYCoordSelector.updateCombo(namesAndValues);
+ endYCoordSelector.updateCombo(namesAndValues);
+ startZValueSelector.updateCombo(namesAndValues);
+ endZValueSelector.updateCombo(namesAndValues);
+ }
+
+ private void updateVertexcombos(String[][] namesAndValues) {
+ xCoordSelector.updateCombo(namesAndValues);
+ yCoordSelector.updateCombo(namesAndValues);
+ zValueSelector.updateCombo(namesAndValues);
+ }
+
+ private void updateHeaders() {
+ if (headerTable != null)
+ headerTable.dispose();
+ headerTable = new Table(tableComposite, SWT.NONE);
+ headerTable.setHeaderVisible(true);
+ headerTable.setLinesVisible(true);
+ GridDataFactory.fillDefaults().grab(true, true).align(SWT.FILL, SWT.FILL).applyTo(tableComposite);
+ for (int i = 0; i < headerTable.getColumns().length; i++) {
+ TableColumn column = headerTable.getColumns()[i];
+ column.dispose();
+ }
+ for (int i = 0; i < headerTable.getItemCount(); i++) {
+ TableItem item = headerTable.getItem(i);
+ item.dispose();
+ }
+
+ headerIndexAndValues.clear();
+
+ try {
+ List<CSVRecord> rows = model.getRows(5);
+
+ for (int k = 0; k < rows.size(); k++) {
+ CSVRecord row = rows.get(k);
+
+ int columnCount = row.size();
+ for (int i = 0; i < columnCount; i++) {
+ String value = row.get(i);
+ if (k == 0) {
+ TableColumn headerCol = new TableColumn(headerTable, SWT.NONE);
+ headerCol.setText(value);
+
+ tableColumnLayout.setColumnData(headerCol, new ColumnWeightData(10));
+ headerCol.pack();
+
+ headerIndexAndValues.put(i, value);
+ } else {
+ int actualK = k - 1;
+ TableItem item;
+ int itemCount = headerTable.getItemCount();
+ if (actualK >= itemCount) {
+ item = new TableItem(headerTable, SWT.NONE);
+ } else {
+ item = headerTable.getItem(actualK);
+ }
+ item.setText(i, value);
+ }
+ }
+ }
+
+ } catch (IOException e) {
+ setErrorMessage(e.getMessage());
+ }
+ }
+
+ protected void validatePageComplete() {
+ if (model.isVertexImport())
+ setPageComplete(model.getXCoordIndex() != -1 && model.getYCoordIndex() != -1 && model.getComponentMappingIndex() != -1);
+ else
+ setPageComplete(model.getStartXCoordIndex() != 1 && model.getStartYCoordIndex() != -1 && model.getEndXCoordIndex() != -1 && model.getEndYCoordIndex() != -1 && model.getComponentMappingIndex() != -1);
+ }
+
+}
Code Review / simantics / district.git / blobdiff