bundles/org.simantics.selectionview/src/org/simantics/selectionview/StandardSelectionProcessor.java

   1 package org.simantics.selectionview;
   2
   3 import java.util.ArrayList;
   4 import java.util.Collection;
   5 import java.util.Collections;
   6 import java.util.HashSet;
   7 import java.util.Set;
   8
   9 import org.simantics.db.ReadGraph;
  10 import org.simantics.db.Resource;
  11 import org.simantics.db.common.request.PossibleIndexRoot;
  12 import org.simantics.db.common.utils.Functions;
  13 import org.simantics.db.common.utils.NameUtils;
  14 import org.simantics.db.exception.DatabaseException;
  15 import org.simantics.db.layer0.SelectionHints;
  16 import org.simantics.db.layer0.adapter.Instances;
  17 import org.simantics.db.layer0.request.VariableIndexRoot;
  18 import org.simantics.db.layer0.variable.Variable;
  19 import org.simantics.db.layer0.variable.Variables;
  20 import org.simantics.layer0.Layer0;
  21 import org.simantics.ui.selection.WorkbenchSelectionUtils;
  22 import org.simantics.utils.ui.ISelectionUtils;
  23 import org.slf4j.Logger;
  24 import org.slf4j.LoggerFactory;
  25
  26 public class StandardSelectionProcessor implements SelectionProcessor<Object, ReadGraph> {
  27     private static final Logger LOGGER = LoggerFactory.getLogger(StandardSelectionProcessor.class);
  28
  29         private static Resource getIndexRoot(ReadGraph graph, Object selection) throws DatabaseException {
  30
  31                 Variable variable = WorkbenchSelectionUtils.getPossibleVariable(graph, selection);
  32                 if (variable != null) {
  33                         return graph.sync(new VariableIndexRoot(variable));
  34                 }
  35
  36                 variable = ISelectionUtils.filterSingleSelection(selection, Variable.class);
  37                 if (variable != null) {
  38                         return graph.sync(new VariableIndexRoot(variable));
  39                 }
  40
  41                 Resource indexRoot = ISelectionUtils.getSinglePossibleKey(selection, SelectionHints.KEY_MODEL, Resource.class, false);
  42                 if (indexRoot == null) {
  43                         Resource r = ISelectionUtils.getSinglePossibleKey(selection, SelectionHints.KEY_MAIN, Resource.class);
  44                         if (r != null)
  45                                 indexRoot = graph.sync(new PossibleIndexRoot(r));
  46                 }
  47
  48                 return indexRoot != null && graph.isInstanceOf(indexRoot, Layer0.getInstance(graph).IndexRoot) ? indexRoot : null;
  49         }
  50
  51         @Override
  52         public Collection<?> process(Object selection, ReadGraph graph) {
  53                 return processStatic(selection, graph);
  54         }
  55
  56         public static Collection<?> processStatic(Object selection, ReadGraph graph) {
  57
  58                 try {
  59
  60                         SelectionViewResources SEL = SelectionViewResources.getInstance(graph);
  61                         Instances selectionProcessorFinder = graph.adapt(SEL.SelectionProcessorContribution, Instances.class);
  62                         Instances contributionFinder = graph.adapt(SEL.TabContribution, Instances.class);
  63                         Instances transformationFinder = graph.adapt(SEL.SelectionTransformation, Instances.class);
  64
  65                         ArrayList<TabContribution<?>> contributions = new ArrayList<>();
  66
  67                         Resource index = getIndexRoot(graph, selection);
  68                         if(index != null) {
  69
  70                                 for(Resource r : contributionFinder.find(graph, index)) {
  71                                         TabContribution<?> contribution = graph.adapt(r, TabContribution.class);
  72                                         contributions.add(contribution);
  73                                         if(DebugPolicy.DEBUG) LOGGER.debug("-contribution " + graph.getURI(r));
  74                                 }
  75
  76                                 ArrayList<Resource> transforms = new ArrayList<>();
  77                                 transforms.addAll(transformationFinder.find(graph, index));
  78                                 if(DebugPolicy.DEBUG) {
  79                                         for(Resource r : transforms)
  80                                                 LOGGER.debug("-transform " + NameUtils.getSafeLabel(graph, r));
  81                                 }
  82
  83                                 HashSet<Object> inputs = new HashSet<>();
  84                                 HashSet<Object> newInputs = new HashSet<>();
  85                                 inputs.add(selection);
  86                                 boolean changed = true;
  87                                 while(changed) {
  88                                         for(Resource tr : transforms) {
  89                                                 for(Object input : inputs) {
  90                                                         try {
  91                                                                 Object newInput = Functions.exec(graph, tr, graph, input);
  92                                                                 if(newInput != null)
  93                                                                         newInputs.add(newInput);
  94                                                         } catch (DatabaseException e) {
  95                                                         }
  96                                                 }
  97                                         }
  98                                         changed = inputs.addAll(newInputs);
  99                                         newInputs.clear();
 100                                 }
 101
 102                                 if(DebugPolicy.DEBUG) {
 103                                         for(Object o : inputs)
 104                                                 LOGGER.debug("-input " + inputName(graph, o));
 105                                 }
 106
 107                                 Set<ComparableTabContributor> result = new HashSet<>();
 108                                 for(TabContribution c : contributions) {
 109                                         for(Object input : inputs) {
 110                                                 try {
 111                                                         if(DebugPolicy.DEBUG) {
 112                                                                 LOGGER.debug("contribution: " + c);
 113                                                                 LOGGER.debug("->input: " + input);
 114                                                                 LOGGER.debug("->input name: " + inputName(graph, input));
 115                                                         }
 116                                                         if (c.accept(graph, input))
 117                                                                 c.contribute(graph, input, result);
 118                                                 } catch (Exception e) {
 119                                                         LOGGER.error("Selection view tab contribution failed for contribution " + c, e);
 120                                                 }
 121                                         }
 122                                 }
 123
 124                                 for(Resource r : selectionProcessorFinder.find(graph, index)) {
 125                                         if (DebugPolicy.DEBUG)
 126                                                 LOGGER.debug("-SelectionProcessor contribution " + graph.getURI(r));
 127                                         @SuppressWarnings("unchecked")
 128                                         SelectionProcessor<Object, ReadGraph> processor = graph.adapt(r, SelectionProcessor.class);
 129                                         Collection<?> contribs = processor.process(selection, graph);
 130                                         if (contribs != null && !contribs.isEmpty()) {
 131                                                 for (Object contrib : contribs) {
 132                                                         if (contrib instanceof ComparableTabContributor) {
 133                                                                 result.add((ComparableTabContributor) contrib);
 134                                                         }
 135                                                 }
 136                                         }
 137                                 }
 138
 139                                 if(DebugPolicy.DEBUG) {
 140                                         LOGGER.debug("result: " + result);
 141                                 }
 142
 143                                 return result;
 144
 145                         }
 146
 147                 } catch (DatabaseException e) {
 148                         LOGGER.error("Selection view tab contribution failed unexpectedly.", e);
 149                 }
 150
 151                 return Collections.emptyList();
 152         }
 153
 154         private static String inputName(ReadGraph graph, Object o) throws DatabaseException {
 155                 if(o instanceof Resource) {
 156                         return "Resource: " + NameUtils.getURIOrSafeNameInternal(graph, (Resource)o);
 157                 } else if(o instanceof Variable) {
 158                         return "Variable: " + ((Variable)o).getURI(graph) + " " + NameUtils.getSafeLabel(graph, (Resource)((Variable)o).getPossiblePropertyValue(graph, Variables.TYPE));
 159                 } else if(o instanceof SelectionInput) {
 160                         return "SelectionInput: " + o.toString();
 161                 } else {
 162                         return "Object(" + o.getClass().getSimpleName() + "): " + o.toString();
 163                 }
 164         }
 165
 166 }