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

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