4 |
""" |
""" |
5 |
|
|
6 |
import vtk |
import vtk |
7 |
from common import * |
from common import Common |
8 |
|
|
9 |
class Arrows(Common): |
class Arrows(Common): |
10 |
""" |
""" |
21 |
""" |
""" |
22 |
|
|
23 |
Common.__init__(self, scene, data_collector) |
Common.__init__(self, scene, data_collector) |
24 |
self.vtk_glyph = None |
self.vtk_glyph = vtk.vtkGlyph3D() |
25 |
self.setArrows() |
self.setArrows() |
26 |
|
|
27 |
Common.setMapper(self, "self.vtk_glyph.GetOutput()", lut) |
Common.setMapperInput(self, self.vtk_glyph.GetOutput(), lut) |
28 |
Common.setActor(self) |
Common.setActorInput(self) |
29 |
Common.addActor(self) |
Common.addActor(self) |
30 |
|
|
31 |
def setArrows(self): |
def setArrows(self): |
35 |
|
|
36 |
vtk_arrows = vtk.vtkArrowSource() |
vtk_arrows = vtk.vtkArrowSource() |
37 |
|
|
|
self.vtk_glyph = vtk.vtkGlyph3D() |
|
38 |
self.vtk_glyph.SetInput(self.data_collector.getReader().GetOutput()) |
self.vtk_glyph.SetInput(self.data_collector.getReader().GetOutput()) |
39 |
self.vtk_glyph.SetSource(vtk_arrows.GetOutput()) |
self.vtk_glyph.SetSource(vtk_arrows.GetOutput()) |
40 |
self.vtk_glyph.SetVectorModeToUseVector() # Default vector mode |
self.setVectorMode("Vector") |
41 |
self.vtk_glyph.SetScaleModeToScaleByVector() # Default scale mode |
self.setScaleMode("Vector") |
42 |
self.setColorMode("Scalar") # Default color mode |
self.setColorMode("Scalar") |
43 |
self.setScaleFactor(0.2) # Default scale factor |
self.setScaleFactor(0.3) |
44 |
|
|
45 |
|
def setVectorMode(self, vector_mode): |
46 |
|
eval("self.vtk_glyph.SetVectorModeToUse%s" % vector_mode) |
47 |
|
|
48 |
|
def setScaleMode(self, scale_mode): |
49 |
|
eval("self.vtk_glyph.SetScaleModeToScaleBy%s" % scale_mode) |
50 |
|
|
51 |
def setScaleFactor(self, scale_factor): |
def setScaleFactor(self, scale_factor): |
52 |
""" |
""" |
70 |
|
|
71 |
|
|
72 |
from arrows import Arrows |
from arrows import Arrows |
73 |
from geo import Plane |
from plane import Plane |
74 |
|
|
75 |
class ArrowsOnPlane(Arrows, Plane): |
class ArrowsOnPlane(Arrows, Plane): |
76 |
""" |
""" |
87 |
@param data_collector: Source of data for visualization |
@param data_collector: Source of data for visualization |
88 |
""" |
""" |
89 |
|
|
90 |
Common.__init__(self, scene, data_collector) |
self.data_collector = data_collector |
91 |
self.vtk_glyph = None |
self.vtk_glyph = vtk.vtkGlyph3D() |
92 |
self.setArrows() |
Arrows.setArrows(self) |
93 |
|
|
94 |
Plane.__init__(self, scene, data_collector, |
Plane.__init__(self, scene, data_collector, |
95 |
"self.vtk_glyph.GetOutput()") |
self.vtk_glyph.GetOutput(), lut) |
96 |
|
|