1 /*******************************************************************************
2 * Copyright (c) 2012, 2013 Association for Decentralized Information Management in
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
10 * VTT Technical Research Centre of Finland - initial API and implementation
11 *******************************************************************************/
12 package org.simantics.g3d.vtk.common;
14 import java.util.ArrayList;
15 import java.util.Collection;
16 import java.util.List;
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;
36 public class NodeSelectionProvider2<DBObject,JavaObject> implements ISelectionProvider, ISelectionChangedListener, ISelectionListener{
37 protected ISelection selection = new StructuredSelection();
38 protected List<ISelectionChangedListener> listeners = new ArrayList<ISelectionChangedListener>();
40 protected List<IG3DNode> selectedNodes = new ArrayList<IG3DNode>(); // selection is ordered
41 //List<Resource> selectedResources = new ArrayList<Resource>();
42 protected List<VTKSelectionItem<DBObject>> selectedItems = new ArrayList<VTKSelectionItem<DBObject>>();
44 protected IEditorPart part;
45 protected IMapping<DBObject,JavaObject> mapping;
46 protected VTKNodeMap nodeMap;
48 public NodeSelectionProvider2(IEditorPart part, IMapping<DBObject,JavaObject> mapping, VTKNodeMap nodeMap) {
50 this.mapping = mapping;
51 this.nodeMap = nodeMap;
55 public void addSelectionChangedListener(
56 ISelectionChangedListener listener) {
57 listeners.add(listener);
61 public void removeSelectionChangedListener(
62 ISelectionChangedListener listener) {
63 listeners.remove(listener);
67 public ISelection getSelection() {
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 JavaObject n = mapping.get((DBObject)r);
82 if (n instanceof IG3DNode) {
83 nodes.add((IG3DNode)n);
86 processSelection(new StructuredSelection(nodes.toArray()));
87 fireSelectionChanged(this);
92 // events coming from vtk
94 public void selectionChanged(SelectionChangedEvent event) {
95 ISelection s = event.getSelection();
98 fireSelectionChanged(event.getSource());
101 @SuppressWarnings("unchecked")
102 protected void processSelection(ISelection s) {
103 selectedNodes.clear();
104 selectedItems.clear();
106 Collection<vtkProp> selectedActors = AdaptationUtils.adaptToCollection(s, vtkProp.class);
107 if (selectedActors.size() > 0) {
108 for (vtkProp a : selectedActors) {
109 IG3DNode node = (IG3DNode)nodeMap.getNode((vtkProp)a);
112 if (!selectedNodes.contains(node))
113 selectedNodes.add(node);
114 DBObject r = mapping.inverseGet((JavaObject)node);
115 selectedItems.add(new VTKSelectionItem<DBObject>(a, node,r));
118 Collection<IG3DNode> selectedNds = AdaptationUtils.adaptToCollection(s, IG3DNode.class);
119 for (INode node : selectedNds) {
120 if (!selectedNodes.contains(node))
121 selectedNodes.add((IG3DNode)node);
122 DBObject r = mapping.inverseGet((JavaObject)node);
123 selectedItems.add(new VTKSelectionItem<DBObject>(null, (IG3DNode)node,r));
128 selection = new StructuredSelection(selectedItems);
131 protected void fireSelectionChanged(Object source) {
132 SelectionChangedEvent evt = new SelectionChangedEvent((ISelectionProvider)source, selection);
133 for (ISelectionChangedListener l : listeners) {
134 l.selectionChanged(evt);
140 // events coming from workbench
142 public void selectionChanged(IWorkbenchPart part, ISelection selection) {
143 if (part == this.part)
145 processSelection(selection);
149 public List<IG3DNode> getSelectedNodes() {
150 return selectedNodes;
153 @SuppressWarnings("unchecked")
154 public List<DBObject> getSelectedResources() {
156 List<DBObject> list = new ArrayList<DBObject>();
157 for (VTKSelectionItem<DBObject> i : selectedItems) {
158 DBObject r = (DBObject)i.getAdapter(Resource.class);
160 r = (DBObject)i.getAdapter(StructuralResource.class);
163 if (!list.contains(r))
169 protected INode getNode(DBObject r) {
170 return (INode)mapping.get(r);