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 import org.simantics.utils.ui.AdaptionUtils;
  33
  34 import vtk.vtkProp;
  35
  36 public class NodeSelectionProvider2<DBObject,JavaObject> implements ISelectionProvider, ISelectionChangedListener, ISelectionListener{
  37     protected ISelection selection = new StructuredSelection();
  38         protected List<ISelectionChangedListener> listeners = new ArrayList<ISelectionChangedListener>();
  39
  40         protected List<INode> selectedNodes = new ArrayList<>(); // selection is ordered
  41         //List<Resource> selectedResources = new ArrayList<Resource>();
  42         protected List<VTKSelectionItem<DBObject>> selectedItems = new ArrayList<VTKSelectionItem<DBObject>>();
  43
  44         protected IEditorPart part;
  45         protected IMapping<DBObject,JavaObject> mapping;
  46         protected VTKNodeMap<DBObject, ? extends INode> nodeMap;
  47
  48         public NodeSelectionProvider2(IEditorPart part, IMapping<DBObject,JavaObject> mapping, VTKNodeMap<DBObject, ? extends INode> nodeMap) {
  49                 this.part = part;
  50                 this.mapping = mapping;
  51                 this.nodeMap = nodeMap;
  52         }
  53
  54         @Override
  55         public void addSelectionChangedListener(
  56                         ISelectionChangedListener listener) {
  57                 listeners.add(listener);
  58         }
  59
  60         @Override
  61         public void removeSelectionChangedListener(
  62                         ISelectionChangedListener listener) {
  63                 listeners.remove(listener);
  64         }
  65
  66         @Override
  67         public ISelection getSelection() {
  68                 return selection;
  69         }
  70
  71         @Override
  72         public void setSelection(ISelection selection) {
  73             Collection<IG3DNode> nodes = AdaptionUtils.adaptToCollection(selection, IG3DNode.class);
  74             Collection<Resource> resources= AdaptionUtils.adaptToCollection(selection, Resource.class);
  75             if (nodes.size() > 0) {
  76                 processSelection(selection);
  77                 fireSelectionChanged(this);
  78             } else if (resources.size() > 0) {
  79                 nodes = new ArrayList<IG3DNode>();
  80                 for (Resource r : resources) {
  81                     @SuppressWarnings("unchecked")
  82                     JavaObject n  = mapping.get((DBObject)r);
  83                     if (n instanceof IG3DNode) {
  84                         nodes.add((IG3DNode)n);
  85                     }
  86                 }
  87                 processSelection(new StructuredSelection(nodes.toArray()));
  88                 fireSelectionChanged(this);
  89             } else {
  90                 processSelection(selection);
  91             fireSelectionChanged(this);
  92             }
  93
  94         }
  95
  96         // events coming from vtk
  97         @Override
  98         public void selectionChanged(SelectionChangedEvent event) {
  99                 ISelection s = event.getSelection();
 100
 101                 processSelection(s);
 102                 fireSelectionChanged(event.getSource());
 103         }
 104
 105         protected void processSelection(ISelection s) {
 106                 selectedNodes.clear();
 107                 selectedItems.clear();
 108                 _processSelection(s);
 109         }
 110
 111         protected void _processSelection(ISelection s) {
 112             Collection<vtkProp> selectedActors = AdaptationUtils.adaptToCollection(s, vtkProp.class);
 113         if (selectedActors.size() > 0) {
 114             for (vtkProp a : selectedActors) {
 115                 INode node = nodeMap.getNode(a);
 116                 if (node == null)
 117                     continue;
 118                 if (!selectedNodes.contains(node))
 119                     selectedNodes.add(node);
 120                 @SuppressWarnings("unchecked")
 121                 DBObject r = mapping.inverseGet((JavaObject)node);
 122                 selectedItems.add(new VTKSelectionItem<DBObject>(a, node,r));
 123             }
 124         } else {
 125             Collection<IG3DNode> selectedNds = AdaptationUtils.adaptToCollection(s, IG3DNode.class);
 126             for (INode node : selectedNds) {
 127                 if (!selectedNodes.contains(node))
 128                     selectedNodes.add((IG3DNode)node);
 129                 @SuppressWarnings("unchecked")
 130                 DBObject r = mapping.inverseGet((JavaObject)node);
 131                 selectedItems.add(new VTKSelectionItem<DBObject>(null, (IG3DNode)node,r));
 132             }
 133         }
 134
 135
 136         selection = new StructuredSelection(selectedItems);
 137         }
 138
 139         protected void fireSelectionChanged(Object source) {
 140                 SelectionChangedEvent evt = new SelectionChangedEvent((ISelectionProvider)source, selection);
 141                 for (ISelectionChangedListener l : listeners) {
 142                         l.selectionChanged(evt);
 143                 }
 144         }
 145
 146
 147
 148         // events coming from workbench
 149         @Override
 150         public void selectionChanged(IWorkbenchPart part, ISelection selection) {
 151                 if (part == this.part)
 152                         return;
 153                 processSelection(selection);
 154
 155         }
 156
 157         public List<INode> getSelectedNodes() {
 158                 return selectedNodes;
 159         }
 160
 161         public List<VTKSelectionItem<DBObject>> getSelectedItems() {
 162         return selectedItems;
 163     }
 164
 165         @SuppressWarnings("unchecked")
 166         public List<DBObject> getSelectedResources() {
 167
 168                 List<DBObject> list = new ArrayList<DBObject>();
 169                 for (VTKSelectionItem<DBObject> i : selectedItems) {
 170                         DBObject r = (DBObject)i.getAdapter(Resource.class);
 171                         if (r == null)
 172                                 r = (DBObject)i.getAdapter(StructuralResource.class);
 173                         if (r == null)
 174                                 continue;
 175                         if (!list.contains(r))
 176                                 list.add(r);
 177                 }
 178                 return list;
 179         }
 180
 181         protected INode getNode(DBObject r) {
 182                 return (INode)mapping.get(r);
 183         }
 184 }