org.simantics.district.network.ui/src/org/simantics/district/network/ui/contributions/ChangeMappingTypeHandler.java

   1 package org.simantics.district.network.ui.contributions;
   2
   3 import java.util.Collection;
   4 import java.util.HashMap;
   5 import java.util.HashSet;
   6 import java.util.List;
   7 import java.util.Map;
   8 import java.util.concurrent.CompletableFuture;
   9 import java.util.concurrent.ExecutionException;
  10 import java.util.stream.Collectors;
  11
  12 import javax.inject.Named;
  13
  14 import org.eclipse.core.runtime.IProgressMonitor;
  15 import org.eclipse.core.runtime.IStatus;
  16 import org.eclipse.core.runtime.Status;
  17 import org.eclipse.core.runtime.jobs.Job;
  18 import org.eclipse.e4.core.di.annotations.CanExecute;
  19 import org.eclipse.e4.core.di.annotations.Execute;
  20 import org.eclipse.e4.ui.services.IServiceConstants;
  21 import org.eclipse.jface.dialogs.Dialog;
  22 import org.eclipse.jface.layout.GridDataFactory;
  23 import org.eclipse.jface.viewers.ISelection;
  24 import org.eclipse.swt.SWT;
  25 import org.eclipse.swt.layout.GridData;
  26 import org.eclipse.swt.layout.GridLayout;
  27 import org.eclipse.swt.widgets.Combo;
  28 import org.eclipse.swt.widgets.Composite;
  29 import org.eclipse.swt.widgets.Control;
  30 import org.eclipse.swt.widgets.Group;
  31 import org.eclipse.swt.widgets.Label;
  32 import org.eclipse.swt.widgets.Shell;
  33 import org.eclipse.ui.dialogs.SelectionStatusDialog;
  34 import org.eclipse.ui.progress.UIJob;
  35 import org.simantics.DatabaseJob;
  36 import org.simantics.Simantics;
  37 import org.simantics.db.ReadGraph;
  38 import org.simantics.db.Resource;
  39 import org.simantics.db.WriteGraph;
  40 import org.simantics.db.common.NamedResource;
  41 import org.simantics.db.common.request.IndexRoot;
  42 import org.simantics.db.common.request.ReadRequest;
  43 import org.simantics.db.common.request.UniqueRead;
  44 import org.simantics.db.common.request.WriteRequest;
  45 import org.simantics.db.common.utils.NameUtils;
  46 import org.simantics.db.exception.DatabaseException;
  47 import org.simantics.db.exception.ServiceException;
  48 import org.simantics.db.exception.ValidationException;
  49 import org.simantics.db.layer0.SelectionHints;
  50 import org.simantics.db.procedure.Procedure;
  51 import org.simantics.db.request.Read;
  52 import org.simantics.district.network.DistrictNetworkUtil;
  53 import org.simantics.district.network.ontology.DistrictNetworkResource;
  54 import org.simantics.district.network.ui.function.Functions;
  55 import org.simantics.district.network.ui.internal.Activator;
  56 import org.simantics.utils.ui.ISelectionUtils;
  57 import org.slf4j.Logger;
  58 import org.slf4j.LoggerFactory;
  59
  60 public class ChangeMappingTypeHandler {
  61
  62     private static final Logger LOGGER = LoggerFactory.getLogger(ChangeMappingTypeHandler.class);
  63
  64     @CanExecute
  65     public boolean canExecute(@Named(IServiceConstants.ACTIVE_SELECTION) ISelection selection) {
  66         List<Resource> elements = ISelectionUtils.getPossibleKeys(selection, SelectionHints.KEY_MAIN, Resource.class);
  67         if (elements.size() < 1)
  68             return false;
  69         try {
  70             return Simantics.getSession().syncRequest(new Read<Boolean>() {
  71
  72                 @Override
  73                 public Boolean perform(ReadGraph graph) throws DatabaseException {
  74                     DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
  75                     for (Resource selection : elements) {
  76                         if (!graph.isInstanceOf(selection, DN.Element)) {
  77                             return false;
  78                         }
  79                     }
  80                     return true;
  81                 }
  82             });
  83         } catch (DatabaseException e) {
  84             LOGGER.error("Could not evaluate if mapping can be changed for selection {}", elements, e);
  85             return false;
  86         }
  87     }
  88
  89     @Execute
  90     public void execute(@Named(IServiceConstants.ACTIVE_SELECTION) Object selection) {
  91         final List<Resource> elements = ISelectionUtils.getPossibleKeys(selection, SelectionHints.KEY_MAIN, Resource.class);
  92 //        if (elements.size() < 1)
  93 //            return;
  94
  95         CompletableFuture<Map<NamedResource, Collection<NamedResource>>> result = new CompletableFuture<>();
  96         Simantics.getSession().asyncRequest(new UniqueRead<Map<NamedResource, Collection<NamedResource>>>() {
  97
  98             @Override
  99             public Map<NamedResource, Collection<NamedResource>> perform(ReadGraph graph) throws DatabaseException {
 100                 DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
 101
 102                 Map<NamedResource, Collection<NamedResource>> currents = new HashMap<>();
 103                 for (Resource element : elements) {
 104                     Resource currentMapping = graph.getPossibleObject(element, DN.HasMapping);
 105                     NamedResource mapping = new NamedResource(NameUtils.getSafeName(graph, currentMapping), currentMapping);
 106                     currents.compute(mapping, (t, u) -> {
 107                         if (u == null)
 108                             u = new HashSet<>();
 109                         try {
 110                             u.add(new NamedResource(NameUtils.getSafeName(graph, element), element));
 111                         } catch (ValidationException | ServiceException e) {
 112                             LOGGER.error("Could not compute name for {}", element, e);
 113                         }
 114                         return u;
 115                     });
 116                 }
 117                 return currents;
 118             }
 119         }, new Procedure<Map<NamedResource, Collection<NamedResource>>>() {
 120
 121             @Override
 122             public void execute(Map<NamedResource, Collection<NamedResource>> results) {
 123                 result.complete(results);
 124             }
 125
 126             @Override
 127             public void exception(Throwable t) {
 128                 LOGGER.error("Could not compute mappings for selection {}", elements, t);
 129                 result.completeExceptionally(t);
 130             }
 131         });
 132
 133         UIJob uiJob = new UIJob("Change mappings..") {
 134
 135             @Override
 136             public IStatus runInUIThread(IProgressMonitor monitor) {
 137                 SelectMappingDialog dialog = new SelectMappingDialog(getDisplay().getActiveShell(), result);
 138                 if (dialog.open() != Dialog.OK)
 139                     return Status.OK_STATUS;
 140
 141                 Map<Resource, Collection<NamedResource>> results = dialog.getResults();
 142                 Job job = new DatabaseJob("Join selected vertices") {
 143
 144                     @Override
 145                     protected IStatus run(IProgressMonitor monitor) {
 146                         try {
 147                             Simantics.getSession().syncRequest(new WriteRequest() {
 148
 149                                 @Override
 150                                 public void perform(WriteGraph graph) throws DatabaseException {
 151                                     DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
 152
 153                                     for (Map.Entry<Resource, Collection<NamedResource>> entry : results.entrySet()) {
 154                                         List<Resource> elements = entry.getValue().stream()
 155                                             .map(NamedResource::getResource)
 156                                             .collect(Collectors.toList());
 157                                         DistrictNetworkUtil.changeMappingType(graph, entry.getKey(), elements);
 158                                     }
 159                                 }
 160                             });
 161                         } catch (DatabaseException e) {
 162                             return new Status(IStatus.ERROR, Activator.PLUGIN_ID, getName() + " failed.", e);
 163                         }
 164                         return Status.OK_STATUS;
 165                     }
 166                 };
 167                 job.setUser(true);
 168                 job.schedule();
 169                 return Status.OK_STATUS;
 170             }
 171         };
 172         uiJob.setUser(true);
 173         uiJob.schedule();
 174     }
 175
 176     private static class SelectMappingDialog extends SelectionStatusDialog {
 177
 178         private Map<NamedResource, Combo> mappingCombos = new HashMap<>();
 179
 180         private Composite composite;
 181
 182         private CompletableFuture<Map<NamedResource, Collection<NamedResource>>> elements;
 183
 184         private Map<NamedResource, Map<String, Resource>> possibleMappings = new HashMap<>();
 185
 186         protected SelectMappingDialog(Shell parentShell, CompletableFuture<Map<NamedResource, Collection<NamedResource>>> elements) {
 187             super(parentShell);
 188             this.elements = elements;
 189             setTitle("Change mappings");
 190         }
 191
 192         @Override
 193         protected Control createDialogArea(Composite parent) {
 194             composite = (Composite) super.createDialogArea(parent);
 195
 196             createMappingsGroup(composite);
 197
 198             // compute default values
 199             Simantics.getSession().asyncRequest(new ReadRequest() {
 200
 201                 @Override
 202                 public void run(ReadGraph graph) throws DatabaseException {
 203                     DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
 204                     try {
 205                         for (Map.Entry<NamedResource, Collection<NamedResource>> entry : elements.get().entrySet()) {
 206                             NamedResource currentMapping = entry.getKey();
 207                             Resource resource = entry.getValue().iterator().next().getResource();
 208                             Resource indexRoot = graph.sync(new IndexRoot(resource));
 209                             if (graph.isInstanceOf(currentMapping.getResource(), DN.Mapping_VertexMapping)) {
 210                                 possibleMappings.put(currentMapping, Functions.getVertexMappings(graph, indexRoot));
 211                             } else if (graph.isInstanceOf(currentMapping.getResource(), DN.Mapping_EdgeMapping)) {
 212                                 possibleMappings.put(currentMapping, Functions.getEdgeMappings(graph, indexRoot));
 213                             }
 214                         }
 215                     } catch (InterruptedException | ExecutionException e) {
 216                         e.printStackTrace();
 217                     }
 218                     composite.getDisplay().asyncExec(() -> {
 219                         for (Map.Entry<NamedResource, Map<String, Resource>> entry : possibleMappings.entrySet()) {
 220                             NamedResource key = entry.getKey();
 221                             Map<String, Resource> value = entry.getValue();
 222                             Combo combo = mappingCombos.get(key);
 223                             combo.setItems(value.keySet().toArray(new String[value.size()]));
 224                             combo.select(0);
 225                         }
 226                     });
 227                 }
 228             });
 229             return composite;
 230         }
 231
 232         private Map<Resource, Collection<NamedResource>> results = new HashMap<>();
 233
 234         @Override
 235         protected void computeResult() {
 236             Map<NamedResource, Collection<NamedResource>> currentElements = null;
 237             try {
 238                 currentElements = elements.get();
 239             } catch (InterruptedException | ExecutionException e) {
 240                 LOGGER.error("Could not get currentElements", e);
 241                 throw new RuntimeException("Could not get currentElements", e);
 242             }
 243             for (Map.Entry<NamedResource, Combo> combos : mappingCombos.entrySet()) {
 244                 NamedResource resource = combos.getKey();
 245                 Combo c = combos.getValue();
 246                 String item = c.getItem(c.getSelectionIndex());
 247                 Collection<NamedResource> collection = currentElements.get(resource);
 248                 Map<String, Resource> map = possibleMappings.get(resource);
 249                 Resource newMapping = map.get(item);
 250                 results.compute(newMapping, (t, u) -> {
 251                     if (u == null) {
 252                         u = new HashSet<>();
 253                     }
 254                     u.addAll(collection);
 255                     return u;
 256                 });
 257             }
 258         }
 259
 260         public Map<Resource, Collection<NamedResource>> getResults() {
 261             return results;
 262         }
 263
 264         private void createMappingsGroup(Composite parent) {
 265             try {
 266                 for (Map.Entry<NamedResource, Collection<NamedResource>> entry : elements.get().entrySet()) {
 267
 268                     NamedResource currentMapping = entry.getKey();
 269
 270                     Collection<NamedResource> mappedElements = entry.getValue();
 271
 272                     Group group= new Group(parent, SWT.NONE);
 273                     group.setFont(parent.getFont());
 274                     group.setText(currentMapping.getName() + " currently mapped to " + mappedElements.size() + " elements");
 275                     GridDataFactory.fillDefaults().grab(true, false).applyTo(group);
 276                     group.setLayout(new GridLayout(1, false));
 277
 278                     Composite cmposite = new Composite(group, SWT.NONE);
 279                     cmposite.setLayoutData(new GridData(SWT.FILL, SWT.TOP, true, false));
 280                     cmposite.setLayout(new GridLayout(2, false));
 281
 282                     Label vertexMappingLabel = new Label(cmposite, SWT.NONE);
 283                     vertexMappingLabel.setText("New mapping type");
 284
 285                     Combo c = new Combo(cmposite, SWT.READ_ONLY | SWT.BORDER);
 286                     GridDataFactory.fillDefaults().grab(true, false).applyTo(c);
 287                     mappingCombos.put(entry.getKey(), c);
 288                 }
 289             } catch (InterruptedException | ExecutionException e) {
 290                 e.printStackTrace();
 291             }
 292         }
 293     }
 294 }