Send event for performed queries + some refactoring

[simantics/district.git] / org.simantics.district.selection.ui / src / org / simantics / district / selection / ui / ElementSelectionTools.java

diff --git a/org.simantics.district.selection.ui/src/org/simantics/district/selection/ui/ElementSelectionTools.java b/org.simantics.district.selection.ui/src/org/simantics/district/selection/ui/ElementSelectionTools.java
new file mode 100644 (file)
index 0000000..130e3f4
--- /dev/null
+++ b/org.simantics.district.selection.ui/src/org/simantics/district/selection/ui/ElementSelectionTools.java
@@ -0,0 +1,84 @@
+package org.simantics.district.selection.ui;
+
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Collections;
+import java.util.List;
+
+import org.simantics.Simantics;
+import org.simantics.browsing.ui.common.AdaptableHintContext;
+import org.simantics.db.ReadGraph;
+import org.simantics.db.Resource;
+import org.simantics.db.common.request.ResourceRead;
+import org.simantics.db.exception.DatabaseException;
+import org.simantics.db.layer0.QueryIndexUtils;
+import org.simantics.db.layer0.SelectionHints;
+import org.simantics.db.layer0.request.ActiveModels;
+import org.simantics.db.layer0.variable.Variable;
+import org.simantics.db.request.Read;
+import org.simantics.district.selection.ElementSelectionResource;
+import org.simantics.district.selection.ElementSelector;
+import org.simantics.layer0.Layer0;
+import org.simantics.ui.selection.AnyResource;
+import org.simantics.ui.selection.AnyVariable;
+import org.simantics.ui.selection.WorkbenchSelectionContentType;
+
+public class ElementSelectionTools {
+
+       public static final class SelectionElement extends AdaptableHintContext {
+               public SelectionElement(Key[] keys) {
+                       super(keys);
+               }
+
+               @SuppressWarnings("unchecked")
+               public <T> T getContent(WorkbenchSelectionContentType<T> contentType) {
+                       Resource element = getHint(SelectionHints.KEY_MAIN);
+                       if (contentType instanceof AnyResource) {
+                               return (T)element;
+                       }
+                       else if (contentType instanceof AnyVariable) {
+                               try {
+                                       return (T) Simantics.getSession().syncRequest(new ResourceRead<Variable>(element) {
+                                               public Variable perform(ReadGraph graph) throws DatabaseException {
+                                                       return ElementSelector.getVariableForElement(graph, resource);
+                                               }
+                                       });
+                               } catch (DatabaseException e) {
+                                       return null;
+                               }
+                       }
+                       
+                       return null;
+               }
+       }
+
+       public static final class SelectionsRequest implements Read<Collection<ElementSelector>> {
+               @Override
+               public Collection<ElementSelector> perform(ReadGraph graph) throws DatabaseException {
+                       Layer0 L0 = Layer0.getInstance(graph);
+                       ElementSelectionResource ES = ElementSelectionResource.getInstance(graph);
+                       
+                       Resource model = ActiveModels.getPossibleActiveModel(graph, Simantics.getProjectResource());
+                       if (model == null) {
+                               return Collections.emptyList();
+                       }
+                       
+                       List<Resource> libs = QueryIndexUtils.searchByType(graph, model, ES.SelectionLibrary);
+                       if (libs.isEmpty())
+                               return Collections.emptyList();
+                       
+                       Resource lib = libs.get(0);
+                       
+                       List<ElementSelector> result = new ArrayList<>();
+                       for (Resource selection : graph.getObjects(lib, L0.ConsistsOf)) {
+                               if (!graph.isInstanceOf(selection, ES.Selection))
+                                       continue;
+                               
+                               result.add(ElementSelector.getSelector(graph, selection));
+                       }
+                       
+                       return result;
+               }
+       }
+
+}