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.SelectionChangedEvent;
21 import org.simantics.g3d.scenegraph.IG3DNode;
22 import org.simantics.g3d.scenegraph.NodeHighlighter;
23 import org.simantics.g3d.scenegraph.NodeHighlighter.HighlightEventType;
24 import org.simantics.g3d.scenegraph.NodeHighlighter.HighlightObjectType;
25 import org.simantics.g3d.scenegraph.base.INode;
26 import org.simantics.g3d.tools.AdaptationUtils;
27 import org.simantics.utils.threads.AWTThread;
28 import org.simantics.utils.threads.ThreadUtils;
31 import vtk.vtkAlgorithm;
32 import vtk.vtkAlgorithmOutput;
33 import vtk.vtkFeatureEdges;
37 import vtk.vtkProperty;
39 public class SelectionHighlighter<E extends IG3DNode> implements ISelectionChangedListener{
44 VTKNodeMap<E> nodeMap;
46 List<IG3DNode> selectedNodes = new ArrayList<IG3DNode>();
47 List<vtkActor> selectedActors = new ArrayList<vtkActor>();
49 HighlightObjectType type = HighlightObjectType.Node;
51 public SelectionHighlighter(vtkPanel panel, VTKNodeMap<E> nodeMap) {
53 this.nodeMap = nodeMap;
57 public void selectionChanged(SelectionChangedEvent event) {
58 final ISelection s = event.getSelection();
60 if (Thread.currentThread().equals(AWTThread.getThreadAccess().getThread()))
63 ThreadUtils.asyncExec(AWTThread.getThreadAccess(), new Runnable() {
67 //System.out.println(this.getClass().getName() + " highlight ");
76 protected void hilight(E node, HighlightEventType type) {
77 if (node instanceof NodeHighlighter) {
78 ((NodeHighlighter)node).highlight(type);
81 if (type == HighlightEventType.Selection) {
82 setSelectedColor(node);
83 } else if (type == HighlightEventType.ClearSelection) {
84 setDefaultColor(node);
88 protected void hilight(vtkActor actor, HighlightEventType type) {
89 if (type == HighlightEventType.Selection) {
90 setColor(actor,false,new double[]{1,0,0});
91 setColor(actor,true,new double[]{1,0,1});
92 } else if (type == HighlightEventType.ClearSelection) {
93 setColor(actor,false,new double[]{1,1,0});
94 setColor(actor,true,new double[]{0,0,0});
98 protected void highlight(ISelection s) {
99 highlight(s, HighlightEventType.Selection, HighlightEventType.ClearSelection);
102 protected void highlight(ISelection s, HighlightEventType apply, HighlightEventType clear) {
104 boolean changed = false;
105 if (type == HighlightObjectType.Node) {
106 Collection<IG3DNode> currentSelectedNodes = AdaptationUtils.adaptToCollection(s,IG3DNode.class);//getSelectedNodes(currentSelectedActors);
107 if (currentSelectedNodes.size() == 0) {
108 Collection<vtkProp> currentSelectedActors = AdaptationUtils.adaptToCollection(s, vtkProp.class);
109 currentSelectedNodes = getSelectedNodes(currentSelectedActors);
111 for (IG3DNode node : selectedNodes) {
112 if (!currentSelectedNodes.contains(node)) {
113 hilight((E)node, clear);
117 for (IG3DNode node : currentSelectedNodes) {
118 if (!selectedNodes.contains(node)) {
119 hilight((E)node, apply);
123 selectedNodes.clear();
124 selectedNodes.addAll(currentSelectedNodes);
125 //selectedNodes = currentSelectedNodes;
129 Collection<vtkActor> currentSelectedActors = AdaptationUtils.adaptToCollection(s, vtkActor.class);
131 for (vtkActor act : selectedActors) {
132 if (!currentSelectedActors.contains(act)) {
137 for (vtkActor act : currentSelectedActors) {
138 if (!selectedActors.contains(act)) {
143 selectedActors.clear();
144 selectedActors.addAll(currentSelectedActors);
151 protected List<IG3DNode> getSelectedNodes(Collection<vtkProp> selectedActors) {
152 List<IG3DNode> currentSelectedNodes = new ArrayList<IG3DNode>();
154 for (vtkProp a : selectedActors) {
155 INode node = nodeMap.getNode((vtkProp)a);
156 if (node == null || !(node instanceof IG3DNode))
158 if (!currentSelectedNodes.contains(node))
159 currentSelectedNodes.add((IG3DNode)node);
161 return currentSelectedNodes;
164 protected void setDefaultColor(E node) {
165 double color[] = new double[]{1,1,0};
166 setColor(node, false, color);
169 protected void setSelectedColor(E node) {
170 double color[] = new double[]{1,0,0};
171 setColor(node, false, color);
175 protected void setColor(E node, boolean edge, double color[]) {
176 for (vtkProp prop : nodeMap.getRenderObjects(node)) {
177 if (prop instanceof vtkActor) {
178 vtkActor act = (vtkActor)prop;
179 setColor(act, edge, color);
184 protected void setColor(vtkActor act, boolean edge, double color[]) {
186 vtkMapper mapper = act.GetMapper();
187 vtkAlgorithmOutput out = mapper.GetInputConnection(0, 0);
188 vtkAlgorithm producer = out.GetProducer();
189 boolean isEdge = (producer instanceof vtkFeatureEdges);
191 if (isEdge == edge) {
192 vtkProperty property = act.GetProperty();
193 property.SetColor(color);