]> gerrit.simantics Code Review - simantics/district.git/blob
3f32f445265a84325082b195f417b40ab2278fc6
[simantics/district.git] /
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 }