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<INode> selectedNodes = new ArrayList<>(); // 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);
89 processSelection(selection);
90 fireSelectionChanged(this);
95 // events coming from vtk
97 public void selectionChanged(SelectionChangedEvent event) {
98 ISelection s = event.getSelection();
101 fireSelectionChanged(event.getSource());
104 @SuppressWarnings("unchecked")
105 protected void processSelection(ISelection s) {
106 selectedNodes.clear();
107 selectedItems.clear();
108 _processSelection(s);
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 IG3DNode node = (IG3DNode)nodeMap.getNode((vtkProp)a);
118 if (!selectedNodes.contains(node))
119 selectedNodes.add(node);
120 DBObject r = mapping.inverseGet((JavaObject)node);
121 selectedItems.add(new VTKSelectionItem<DBObject>(a, node,r));
124 Collection<IG3DNode> selectedNds = AdaptationUtils.adaptToCollection(s, IG3DNode.class);
125 for (INode node : selectedNds) {
126 if (!selectedNodes.contains(node))
127 selectedNodes.add((IG3DNode)node);
128 DBObject r = mapping.inverseGet((JavaObject)node);
129 selectedItems.add(new VTKSelectionItem<DBObject>(null, (IG3DNode)node,r));
134 selection = new StructuredSelection(selectedItems);
137 protected void fireSelectionChanged(Object source) {
138 SelectionChangedEvent evt = new SelectionChangedEvent((ISelectionProvider)source, selection);
139 for (ISelectionChangedListener l : listeners) {
140 l.selectionChanged(evt);
146 // events coming from workbench
148 public void selectionChanged(IWorkbenchPart part, ISelection selection) {
149 if (part == this.part)
151 processSelection(selection);
155 public List<INode> getSelectedNodes() {
156 return selectedNodes;
159 public List<VTKSelectionItem<DBObject>> getSelectedItems() {
160 return selectedItems;
163 @SuppressWarnings("unchecked")
164 public List<DBObject> getSelectedResources() {
166 List<DBObject> list = new ArrayList<DBObject>();
167 for (VTKSelectionItem<DBObject> i : selectedItems) {
168 DBObject r = (DBObject)i.getAdapter(Resource.class);
170 r = (DBObject)i.getAdapter(StructuralResource.class);
173 if (!list.contains(r))
179 protected INode getNode(DBObject r) {
180 return (INode)mapping.get(r);