1 package org.simantics.g3d.vtk.shape;
\r
3 import javax.vecmath.AxisAngle4d;
\r
4 import javax.vecmath.Matrix4d;
\r
5 import javax.vecmath.Point3d;
\r
6 import javax.vecmath.Vector3d;
\r
8 import org.simantics.g3d.math.MathTools;
\r
9 import org.simantics.utils.threads.AWTThread;
\r
10 import org.simantics.utils.threads.ThreadUtils;
\r
12 import vtk.vtkActor;
\r
13 import vtk.vtkAssembly;
\r
14 import vtk.vtkConeSource;
\r
15 import vtk.vtkLineSource;
\r
16 import vtk.vtkLinearTransform;
\r
17 import vtk.vtkMatrix4x4;
\r
18 import vtk.vtkPolyDataMapper;
\r
19 import vtk.vtkProp3D;
\r
20 import vtk.vtkRenderer;
\r
21 import vtk.vtkTextActor;
\r
22 import vtk.vtkTubeFilter;
\r
24 public class DualHeadArrowActor extends vtkAssembly implements IvtkVisualObject{
\r
26 private vtkRenderer ren;
\r
27 private Vector3d axisDir = new Vector3d(1,0,0);
\r
28 private vtkTextActor tactor;
\r
29 private vtkActor tubeActor;
\r
30 private vtkActor coneActor;
\r
31 private vtkActor coneActor2;
\r
32 private boolean rendered = false;
\r
34 public DualHeadArrowActor(vtkRenderer _ren, String label) {
\r
40 public DualHeadArrowActor(vtkRenderer _ren, String label, Vector3d dir) {
\r
44 if (dir.lengthSquared() < MathTools.NEAR_ZERO)
\r
45 throw new IllegalArgumentException("Direction vector length must mer larger than zero");
\r
49 public void createAxis(String label) {
\r
50 double coneScale = 0.3 * axisDir.length();
\r
51 Vector3d coneOffset = new Vector3d(axisDir);
\r
52 coneOffset.normalize();
\r
53 coneOffset.scale(coneScale*0.5);
\r
55 vtkLineSource line = new vtkLineSource();
\r
56 //line.SetPoint1(0.0,0.0,0.0);
\r
57 line.SetPoint1(coneOffset.x,coneOffset.y,coneOffset.z);
\r
58 line.SetPoint2(axisDir.x-coneOffset.x,axisDir.y-coneOffset.y,axisDir.z-coneOffset.z);
\r
60 tactor = new vtkTextActor();
\r
62 tactor.SetInput(label);
\r
64 tactor.SetTextScaleModeToNone();
\r
65 tactor.GetTextProperty().SetColor(0.0, 0.0, 0.0);
\r
66 tactor.GetTextProperty().ShadowOff();
\r
67 tactor.GetTextProperty().ItalicOff();
\r
68 tactor.GetTextProperty().BoldOff();
\r
70 tactor.GetTextProperty().Delete();
\r
72 tactor.SetMaximumLineHeight(0.25);
\r
74 tactor.SetPickable(0);
\r
76 vtkTubeFilter tube = new vtkTubeFilter();
\r
77 tube.SetInput(line.GetOutput());
\r
78 tube.SetRadius(0.05 * axisDir.length());
\r
79 tube.SetNumberOfSides(8);
\r
81 vtkPolyDataMapper tubeMapper = new vtkPolyDataMapper();
\r
82 tubeMapper.SetInput(tube.GetOutput());
\r
84 tubeActor = new vtkActor();
\r
85 tubeActor.SetMapper(tubeMapper);
\r
86 tubeActor.PickableOff();
\r
94 vtkConeSource cone = new vtkConeSource();
\r
95 cone.SetResolution(coneRes);
\r
96 vtkPolyDataMapper coneMapper = new vtkPolyDataMapper();
\r
97 coneMapper.SetInput(cone.GetOutput());
\r
98 coneActor = new vtkActor();
\r
99 coneActor.SetMapper(coneMapper);
\r
100 coneActor.GetProperty().SetColor(1, 0, 0);
\r
101 coneActor.SetScale(coneScale, coneScale, coneScale);
\r
102 coneActor.SetPosition(axisDir.x-coneOffset.x,axisDir.y-coneOffset.y,axisDir.z-coneOffset.z);
\r
103 coneActor.SetPickable(0);
\r
105 AxisAngle4d aa = MathTools.createRotation(new Vector3d(1,0,0), axisDir);
\r
107 coneActor.RotateWXYZ(MathTools.radToDeg(aa.angle), aa.x, aa.y, aa.z);
\r
110 coneActor2 = new vtkActor();
\r
111 coneActor2.SetMapper(coneMapper);
\r
112 coneActor2.GetProperty().SetColor(1, 0, 0);
\r
113 coneActor2.SetScale(coneScale, coneScale, coneScale);
\r
114 coneActor2.SetPosition(coneOffset.x,coneOffset.y,coneOffset.z);
\r
115 coneActor2.SetPickable(0);
\r
117 aa = MathTools.createRotation(new Vector3d(-1,0,0), axisDir);
\r
119 coneActor2.RotateWXYZ(MathTools.radToDeg(aa.angle), aa.x, aa.y, aa.z);
\r
121 this.AddPart(tubeActor);
\r
122 this.AddPart(coneActor);
\r
123 this.AddPart(coneActor2);
\r
125 tube.GetOutput().Delete();
\r
126 cone.GetOutput().Delete();
\r
127 line.GetOutput().Delete();
\r
129 tubeMapper.Delete();
\r
134 coneMapper.Delete();
\r
136 coneActor.GetProperty().Delete();
\r
137 coneActor2.GetProperty().Delete();
\r
140 public void addToRenderer() {
\r
145 ren.AddActor2D(tactor);
\r
147 ren.AddActor(this);
\r
150 public void removeFromRenderer() {
\r
154 ren.RemoveActor2D(tactor);
\r
155 ren.RemoveActor(this);
\r
158 public boolean isRendered() {
\r
162 public void setAxesVisibility(boolean ison) {
\r
163 this.SetVisibility(ison ? 1 : 0);
\r
164 tactor.SetVisibility(ison ? 1 : 0);
\r
167 public void setLabelVisibility(boolean ison) {
\r
168 tactor.SetVisibility(ison ? 1 : 0);
\r
171 double mat[] = new double[16];
\r
172 Matrix4d m = new Matrix4d();
\r
173 private void updateTextLoc() {
\r
174 tactor.GetPositionCoordinate().SetCoordinateSystemToWorld();
\r
178 MathTools.set(m,mat);
\r
179 Point3d p = new Point3d(axisDir.x*0.5, axisDir.y*0.5,axisDir.z*0.5);
\r
181 tactor.GetPositionCoordinate().SetValue(p.x, p.y, p.z);
\r
183 tactor.GetPositionCoordinate().Delete();
\r
187 public void SetPickable(int id0) {
\r
188 super.SetPickable(id0);
\r
189 tubeActor.SetPickable(id0);
\r
190 coneActor.SetPickable(id0);
\r
191 coneActor2.SetPickable(id0);
\r
195 public void SetOrientation(double id0, double id1, double id2) {
\r
196 super.SetOrientation(id0, id1, id2);
\r
201 public void RotateWXYZ(double id0, double id1, double id2, double id3) {
\r
202 super.RotateWXYZ(id0, id1, id2, id3);
\r
207 public void SetPosition(double[] id0) {
\r
208 super.SetPosition(id0);
\r
213 public void SetPosition(double id0, double id1, double id2) {
\r
214 super.SetPosition(id0, id1, id2);
\r
219 public void SetOrientation(double[] id0) {
\r
220 super.SetOrientation(id0);
\r
225 public void SetScale(double id0) {
\r
226 super.SetScale(id0);
\r
231 public void SetScale(double id0, double id1, double id2) {
\r
232 super.SetScale(id0, id1, id2);
\r
237 public void SetScale(double[] id0) {
\r
238 super.SetScale(id0);
\r
242 public void SetColor(double r, double g, double b) {
\r
243 coneActor.GetProperty().SetColor(r, g, b);
\r
244 coneActor2.GetProperty().SetColor(r, g, b);
\r
245 tubeActor.GetProperty().SetColor(r, g, b);
\r
246 coneActor.GetProperty().Delete();
\r
247 coneActor2.GetProperty().Delete();
\r
248 tubeActor.GetProperty().Delete();
\r
251 public void SetTextColor(double r, double g, double b) {
\r
252 tactor.GetTextProperty().SetColor(r, g, b);
\r
253 tactor.GetTextProperty().Delete();
\r
257 public void SetUserMatrix(vtkMatrix4x4 id0) {
\r
258 super.SetUserMatrix(id0);
\r
263 public void SetUserTransform(vtkLinearTransform id0) {
\r
264 super.SetUserTransform(id0);
\r
269 public void Delete() {
\r
270 ren.RemoveActor(tactor);
\r
271 ren.RemoveActor(tubeActor);
\r
272 ren.RemoveActor(coneActor);
\r
273 ren.RemoveActor(coneActor2);
\r
275 tubeActor.Delete();
\r
276 coneActor.Delete();
\r
277 coneActor2.Delete();
\r
281 public void dispose() {
\r
282 ThreadUtils.asyncExec(AWTThread.getThreadAccess(), new Runnable() {
\r
285 public void run() {
\r
286 removeFromRenderer();
\r
293 public vtkProp3D getVtkProp() {
\r