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

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