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

   1 package org.simantics.district.imports.ui;
   2
   3 import java.io.IOException;
   4 import java.util.Collection;
   5 import java.util.Map;
   6
   7 import org.eclipse.jface.dialogs.IPageChangeProvider;
   8 import org.eclipse.jface.dialogs.IPageChangedListener;
   9 import org.eclipse.jface.dialogs.PageChangedEvent;
  10 import org.eclipse.jface.layout.GridDataFactory;
  11 import org.eclipse.jface.wizard.IWizardContainer;
  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.Label;
  20 import org.simantics.Simantics;
  21 import org.simantics.db.ReadGraph;
  22 import org.simantics.db.Resource;
  23 import org.simantics.db.common.request.IndexRoot;
  24 import org.simantics.db.common.request.ReadRequest;
  25 import org.simantics.db.exception.DatabaseException;
  26 import org.simantics.district.imports.DistrictImportUtils;
  27 import org.simantics.district.network.ui.function.Functions;
  28
  29 public class ComponentMappingPage extends WizardPage {
  30
  31     private CSVImportModel model;
  32     private Composite composite;
  33     private Composite childComposite;
  34     protected Map<String, Resource> componentMappings;
  35     private Collection<String> distinctMappingIndexColumnValues;
  36
  37     public ComponentMappingPage(CSVImportModel model) {
  38         super("Select component mappings");
  39         this.model = model;
  40         setMessage("Select component mappings");
  41     }
  42
  43     @Override
  44     public void createControl(Composite parent) {
  45         composite = new Composite(parent, SWT.NONE);
  46         composite.setLayout(new GridLayout(1,false));
  47         GridDataFactory.fillDefaults().grab(true, true).align(SWT.FILL, SWT.FILL).applyTo(composite);
  48
  49         setControl(composite);
  50
  51         validatePageComplete();
  52
  53         IWizardContainer container = getContainer();
  54         if (container instanceof IPageChangeProvider) {
  55             ((IPageChangeProvider) container).addPageChangedListener(new IPageChangedListener() {
  56
  57                 @Override
  58                 public void pageChanged(PageChangedEvent event) {
  59                     if (container.getCurrentPage().equals(ComponentMappingPage.this)) {
  60                         ComponentMappingPage.this.updateComponentMappings();
  61                     }
  62                 }
  63             });
  64         }
  65     }
  66
  67     private void updateComponentMappings() {
  68         try {
  69             int mappingIndex = model.getComponentMappingIndex();
  70             distinctMappingIndexColumnValues = DistrictImportUtils.readDistinctValuesOfColumn(model.getSource(), model.getDelimiter(), mappingIndex);
  71
  72             try {
  73                 Simantics.getSession().sync(new ReadRequest() {
  74
  75                     @Override
  76                     public void run(ReadGraph graph) throws DatabaseException {
  77                         Resource indexRoot = graph.sync(new IndexRoot(model.getParentDiagram()));
  78                         if (model.isVertexImport())
  79                             componentMappings = Functions.getVertexMappings(graph, indexRoot);
  80                         else
  81                             componentMappings = Functions.getEdgeMappings(graph, indexRoot);
  82                     }
  83                 });
  84             } catch (DatabaseException e) {
  85                 e.printStackTrace();
  86             }
  87
  88             System.out.println(distinctMappingIndexColumnValues);
  89             if (childComposite != null)
  90                 childComposite.dispose();
  91             childComposite = new Composite(composite, SWT.NONE);
  92             childComposite.setLayout(new GridLayout(1,false));
  93             GridDataFactory.fillDefaults().grab(true, true).align(SWT.FILL, SWT.FILL).applyTo(childComposite);
  94
  95             for (String value : distinctMappingIndexColumnValues) {
  96
  97                 Composite newCompo = new Composite(childComposite, SWT.NONE);
  98                 newCompo.setLayout(new GridLayout(2,false));
  99                 GridDataFactory.fillDefaults().grab(true, true).align(SWT.FILL, SWT.FILL).applyTo(newCompo);
 100
 101                 Label t = new Label(newCompo, SWT.NONE);
 102                 t.setText(value);
 103
 104                 Combo c = new Combo(newCompo, SWT.READ_ONLY);
 105                 c.setItems(componentMappings.keySet().toArray(new String[componentMappings.size()]));
 106                 c.addSelectionListener(new SelectionListener() {
 107
 108                     @Override
 109                     public void widgetSelected(SelectionEvent e) {
 110                         model.setComponentMappings(value, componentMappings.get(c.getItem(c.getSelectionIndex())));
 111                         validatePageComplete();
 112                     }
 113
 114                     @Override
 115                     public void widgetDefaultSelected(SelectionEvent e) {
 116                         widgetSelected(e);
 117                     }
 118                 });
 119             }
 120             composite.layout(true, true);
 121         } catch (IOException e) {
 122             e.printStackTrace();
 123         }
 124     }
 125
 126     private void validatePageComplete() {
 127         Map<String, Resource> currentMappings = model.getComponentMappings();
 128         if (currentMappings != null && distinctMappingIndexColumnValues != null && currentMappings.keySet().containsAll(distinctMappingIndexColumnValues))
 129             setPageComplete(true);
 130         else
 131             setPageComplete(false);
 132     }
 133
 134 }