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.shape;
14 import javax.vecmath.AxisAngle4d;
15 import javax.vecmath.Matrix4d;
16 import javax.vecmath.Point3d;
17 import javax.vecmath.Vector3d;
19 import org.simantics.g3d.math.MathTools;
20 import org.simantics.utils.threads.AWTThread;
21 import org.simantics.utils.threads.ThreadUtils;
24 import vtk.vtkAssembly;
25 import vtk.vtkConeSource;
26 import vtk.vtkLineSource;
27 import vtk.vtkLinearTransform;
28 import vtk.vtkMatrix4x4;
29 import vtk.vtkPolyDataMapper;
31 import vtk.vtkRenderer;
32 import vtk.vtkTextActor;
33 import vtk.vtkTubeFilter;
35 public class DualHeadArrowActor extends vtkAssembly implements IvtkVisualObject{
37 private vtkRenderer ren;
38 private Vector3d axisDir = new Vector3d(1,0,0);
39 private vtkTextActor tactor;
40 private vtkActor tubeActor;
41 private vtkActor coneActor;
42 private vtkActor coneActor2;
43 private boolean rendered = false;
45 public DualHeadArrowActor(vtkRenderer _ren, String label) {
51 public DualHeadArrowActor(vtkRenderer _ren, String label, Vector3d dir) {
55 if (dir.lengthSquared() < MathTools.NEAR_ZERO)
56 throw new IllegalArgumentException("Direction vector length must mer larger than zero");
60 public void createAxis(String label) {
61 double coneScale = 0.3 * axisDir.length();
62 Vector3d coneOffset = new Vector3d(axisDir);
63 coneOffset.normalize();
64 coneOffset.scale(coneScale*0.5);
66 vtkLineSource line = new vtkLineSource();
67 //line.SetPoint1(0.0,0.0,0.0);
68 line.SetPoint1(coneOffset.x,coneOffset.y,coneOffset.z);
69 line.SetPoint2(axisDir.x-coneOffset.x,axisDir.y-coneOffset.y,axisDir.z-coneOffset.z);
71 tactor = new vtkTextActor();
73 tactor.SetInput(label);
75 tactor.SetTextScaleModeToNone();
76 tactor.GetTextProperty().SetColor(0.0, 0.0, 0.0);
77 tactor.GetTextProperty().ShadowOff();
78 tactor.GetTextProperty().ItalicOff();
79 tactor.GetTextProperty().BoldOff();
81 tactor.GetTextProperty().Delete();
83 tactor.SetMaximumLineHeight(0.25);
85 tactor.SetPickable(0);
87 vtkTubeFilter tube = new vtkTubeFilter();
88 tube.SetInputData(line.GetOutput());
89 tube.SetRadius(0.05 * axisDir.length());
90 tube.SetNumberOfSides(8);
92 vtkPolyDataMapper tubeMapper = new vtkPolyDataMapper();
93 tubeMapper.SetInputData(tube.GetOutput());
95 tubeActor = new vtkActor();
96 tubeActor.SetMapper(tubeMapper);
97 tubeActor.PickableOff();
105 vtkConeSource cone = new vtkConeSource();
106 cone.SetResolution(coneRes);
107 vtkPolyDataMapper coneMapper = new vtkPolyDataMapper();
108 coneMapper.SetInputData(cone.GetOutput());
109 coneActor = new vtkActor();
110 coneActor.SetMapper(coneMapper);
111 coneActor.GetProperty().SetColor(1, 0, 0);
112 coneActor.SetScale(coneScale, coneScale, coneScale);
113 coneActor.SetPosition(axisDir.x-coneOffset.x,axisDir.y-coneOffset.y,axisDir.z-coneOffset.z);
114 coneActor.SetPickable(0);
116 AxisAngle4d aa = MathTools.createRotation(new Vector3d(1,0,0), axisDir);
118 coneActor.RotateWXYZ(MathTools.radToDeg(aa.angle), aa.x, aa.y, aa.z);
121 coneActor2 = new vtkActor();
122 coneActor2.SetMapper(coneMapper);
123 coneActor2.GetProperty().SetColor(1, 0, 0);
124 coneActor2.SetScale(coneScale, coneScale, coneScale);
125 coneActor2.SetPosition(coneOffset.x,coneOffset.y,coneOffset.z);
126 coneActor2.SetPickable(0);
128 aa = MathTools.createRotation(new Vector3d(-1,0,0), axisDir);
130 coneActor2.RotateWXYZ(MathTools.radToDeg(aa.angle), aa.x, aa.y, aa.z);
132 this.AddPart(tubeActor);
133 this.AddPart(coneActor);
134 this.AddPart(coneActor2);
136 tube.GetOutput().Delete();
137 cone.GetOutput().Delete();
138 line.GetOutput().Delete();
147 coneActor.GetProperty().Delete();
148 coneActor2.GetProperty().Delete();
151 public void addToRenderer() {
156 ren.AddActor2D(tactor);
161 public void removeFromRenderer() {
165 ren.RemoveActor2D(tactor);
166 ren.RemoveActor(this);
169 public boolean isRendered() {
173 public void setAxesVisibility(boolean ison) {
174 this.SetVisibility(ison ? 1 : 0);
175 tactor.SetVisibility(ison ? 1 : 0);
178 public void setLabelVisibility(boolean ison) {
179 tactor.SetVisibility(ison ? 1 : 0);
182 double mat[] = new double[16];
183 Matrix4d m = new Matrix4d();
184 private void updateTextLoc() {
185 tactor.GetPositionCoordinate().SetCoordinateSystemToWorld();
189 MathTools.set(m,mat);
190 Point3d p = new Point3d(axisDir.x*0.5, axisDir.y*0.5,axisDir.z*0.5);
192 tactor.GetPositionCoordinate().SetValue(p.x, p.y, p.z);
194 tactor.GetPositionCoordinate().Delete();
198 public void SetPickable(int id0) {
199 super.SetPickable(id0);
200 tubeActor.SetPickable(id0);
201 coneActor.SetPickable(id0);
202 coneActor2.SetPickable(id0);
206 public void SetOrientation(double id0, double id1, double id2) {
207 super.SetOrientation(id0, id1, id2);
212 public void RotateWXYZ(double id0, double id1, double id2, double id3) {
213 super.RotateWXYZ(id0, id1, id2, id3);
218 public void SetPosition(double[] id0) {
219 super.SetPosition(id0);
224 public void SetPosition(double id0, double id1, double id2) {
225 super.SetPosition(id0, id1, id2);
230 public void SetOrientation(double[] id0) {
231 super.SetOrientation(id0);
236 public void SetScale(double id0) {
242 public void SetScale(double id0, double id1, double id2) {
243 super.SetScale(id0, id1, id2);
248 public void SetScale(double[] id0) {
253 public void SetColor(double r, double g, double b) {
254 coneActor.GetProperty().SetColor(r, g, b);
255 coneActor2.GetProperty().SetColor(r, g, b);
256 tubeActor.GetProperty().SetColor(r, g, b);
257 coneActor.GetProperty().Delete();
258 coneActor2.GetProperty().Delete();
259 tubeActor.GetProperty().Delete();
262 public void SetTextColor(double r, double g, double b) {
263 tactor.GetTextProperty().SetColor(r, g, b);
264 tactor.GetTextProperty().Delete();
268 public void SetUserMatrix(vtkMatrix4x4 id0) {
269 super.SetUserMatrix(id0);
274 public void SetUserTransform(vtkLinearTransform id0) {
275 super.SetUserTransform(id0);
280 public void Delete() {
281 ren.RemoveActor(tactor);
282 ren.RemoveActor(tubeActor);
283 ren.RemoveActor(coneActor);
284 ren.RemoveActor(coneActor2);
292 public void dispose() {
293 ThreadUtils.asyncExec(AWTThread.getThreadAccess(), new Runnable() {
297 removeFromRenderer();
304 public vtkProp3D getVtkProp() {