org.simantics.g3d.vtk/src/org/simantics/g3d/vtk/common/NodeSelectionProvider2.java

   1 package org.simantics.g3d.vtk.common;\r
   2 \r
   3 import java.util.ArrayList;\r
   4 import java.util.Collection;\r
   5 import java.util.List;\r
   6 \r
   7 import org.eclipse.jface.viewers.ISelection;\r
   8 import org.eclipse.jface.viewers.ISelectionChangedListener;\r
   9 import org.eclipse.jface.viewers.ISelectionProvider;\r
  10 import org.eclipse.jface.viewers.SelectionChangedEvent;\r
  11 import org.eclipse.jface.viewers.StructuredSelection;\r
  12 import org.eclipse.ui.IEditorPart;\r
  13 import org.eclipse.ui.ISelectionListener;\r
  14 import org.eclipse.ui.IWorkbenchPart;\r
  15 import org.simantics.db.Resource;\r
  16 import org.simantics.g3d.scenegraph.IG3DNode;\r
  17 import org.simantics.g3d.scenegraph.base.INode;\r
  18 import org.simantics.g3d.tools.AdaptationUtils;\r
  19 import org.simantics.objmap.graph.IMapping;\r
  20 import org.simantics.objmap.structural.StructuralResource;\r
  21 \r
  22 import vtk.vtkProp;\r
  23 \r
  24 public class NodeSelectionProvider2<DBObject,JavaObject> implements ISelectionProvider, ISelectionChangedListener, ISelectionListener{\r
  25         private ISelection selection = new StructuredSelection();\r
  26         private List<ISelectionChangedListener> listeners = new ArrayList<ISelectionChangedListener>();\r
  27         \r
  28         List<IG3DNode> selectedNodes = new ArrayList<IG3DNode>(); // selection is ordered\r
  29         //List<Resource> selectedResources = new ArrayList<Resource>();\r
  30         List<VTKSelectionItem<DBObject>> selectedItems = new ArrayList<VTKSelectionItem<DBObject>>();\r
  31         \r
  32         IEditorPart part;\r
  33         IMapping<DBObject,JavaObject> mapping;\r
  34         VTKNodeMap nodeMap;\r
  35         \r
  36         public NodeSelectionProvider2(IEditorPart part, IMapping<DBObject,JavaObject> mapping, VTKNodeMap nodeMap) {\r
  37                 this.part = part;\r
  38                 this.mapping = mapping;\r
  39                 this.nodeMap = nodeMap;\r
  40         }\r
  41         \r
  42         @Override\r
  43         public void addSelectionChangedListener(\r
  44                         ISelectionChangedListener listener) {\r
  45                 listeners.add(listener);\r
  46         }\r
  47         \r
  48         @Override\r
  49         public void removeSelectionChangedListener(\r
  50                         ISelectionChangedListener listener) {\r
  51                 listeners.remove(listener);\r
  52         }\r
  53         \r
  54         @Override\r
  55         public ISelection getSelection() {\r
  56                 return selection;\r
  57         }\r
  58         \r
  59         @Override\r
  60         public void setSelection(ISelection selection) {\r
  61                 \r
  62         }\r
  63         \r
  64         // events coming from vtk       \r
  65         @Override\r
  66         public void selectionChanged(SelectionChangedEvent event) {\r
  67                 ISelection s = event.getSelection();\r
  68                 \r
  69                 processSelection(s);\r
  70                 fireSelectionChanged(event.getSource());\r
  71         }\r
  72         \r
  73         @SuppressWarnings("unchecked")\r
  74         private void processSelection(ISelection s) {\r
  75                 selectedNodes.clear();\r
  76                 selectedItems.clear();\r
  77                 \r
  78                 Collection<vtkProp> selectedActors = AdaptationUtils.adaptToCollection(s, vtkProp.class);\r
  79                 if (selectedActors.size() > 0) {\r
  80                         for (vtkProp a : selectedActors) {\r
  81                                 IG3DNode node = (IG3DNode)nodeMap.getNode((vtkProp)a);\r
  82                                 if (node == null)\r
  83                                         continue;\r
  84                                 if (!selectedNodes.contains(node))\r
  85                                         selectedNodes.add(node);\r
  86                                 DBObject r = mapping.inverseGet((JavaObject)node);\r
  87                                 selectedItems.add(new VTKSelectionItem<DBObject>(a, node,r));\r
  88                         }\r
  89                 } else {\r
  90                         Collection<IG3DNode> selectedNds = AdaptationUtils.adaptToCollection(s, IG3DNode.class);\r
  91                         for (INode node : selectedNds) {\r
  92                                 if (!selectedNodes.contains(node))\r
  93                                         selectedNodes.add((IG3DNode)node);\r
  94                                 DBObject r = mapping.inverseGet((JavaObject)node);\r
  95                                 selectedItems.add(new VTKSelectionItem<DBObject>(null, (IG3DNode)node,r));\r
  96                         }\r
  97                 }\r
  98                 \r
  99                 \r
 100                 selection = new StructuredSelection(selectedItems);\r
 101         }\r
 102         \r
 103         private void fireSelectionChanged(Object source) {\r
 104                 SelectionChangedEvent evt = new SelectionChangedEvent((ISelectionProvider)source, selection);\r
 105                 for (ISelectionChangedListener l : listeners) {\r
 106                         l.selectionChanged(evt);\r
 107                 }\r
 108         }\r
 109         \r
 110         \r
 111         \r
 112         // events coming from workbench\r
 113         @Override\r
 114         public void selectionChanged(IWorkbenchPart part, ISelection selection) {\r
 115                 if (part == this.part)\r
 116                         return;\r
 117                 processSelection(selection);\r
 118                 \r
 119         }\r
 120         \r
 121         public List<IG3DNode> getSelectedNodes() {\r
 122                 return selectedNodes;\r
 123         }\r
 124         \r
 125         @SuppressWarnings("unchecked")\r
 126         public List<DBObject> getSelectedResources() {\r
 127                 \r
 128                 List<DBObject> list = new ArrayList<DBObject>();\r
 129                 for (VTKSelectionItem<DBObject> i : selectedItems) {\r
 130                         DBObject r = (DBObject)i.getAdapter(Resource.class);\r
 131                         if (r == null)\r
 132                                 r = (DBObject)i.getAdapter(StructuralResource.class);\r
 133                         if (r == null)\r
 134                                 continue;\r
 135                         if (!list.contains(r))\r
 136                                 list.add(r);\r
 137                 }\r
 138                 return list;\r
 139         }\r
 140         \r
 141         protected INode getNode(DBObject r) {\r
 142                 return (INode)mapping.get(r);\r
 143         }\r
 144 }\r