/[escript]/trunk/pyvisi/py_src/common.py
ViewVC logotype

Contents of /trunk/pyvisi/py_src/common.py

Parent Directory Parent Directory | Revision Log Revision Log


Revision 845 - (show annotations)
Fri Sep 8 07:13:01 2006 UTC (14 years ago) by jongui
File MIME type: text/x-python
File size: 2196 byte(s)
Display an image is partially completed.

1 """
2 @author: John Ngui
3 @author: Lutz Gross
4 """
5
6 import vtk
7
8 class Common:
9 """
10 Class that defines the common operations invoked by the components.
11 """
12
13 def __init__(self, scene, data_collector = None):
14 """
15 Initialize all the instance variables.
16
17 @type scene: L{Scene <scene.Scene>} object
18 @param scene: Scene in which components are to be added to
19 @type data_collector: L{DataCollector <datacollector.DataCollector>}
20 object
21 @param data_collector: Source of data for visualization
22 """
23
24 self.scene = scene
25 self.data_collector = data_collector
26 self.vtk_mapper = None
27 self.vtk_actor = None
28
29 def setMapper(self, component, lut = None):
30 """
31 Set up the mapper and its input.
32
33 @type component: String
34 @param component: Component to be mapped
35 @type lut: L{BlueToRed <colormap.BlueToRed>} or
36 L{RedToBlue <colormap.RedToBlue>} object
37 @param lut: Color lookup table to be used by the mapper
38 """
39
40 self.vtk_mapper = vtk.vtkDataSetMapper()
41 eval("self.vtk_mapper.SetInput(%s)" % component)
42
43 if(lut != None):
44 self.vtk_mapper.SetLookupTable(lut.getLut())
45
46 def setActor(self):
47 """
48 Set up the actor and its mapper.
49 """
50
51 self.vtk_actor = vtk.vtkActor()
52 self.vtk_actor.SetMapper(self.vtk_mapper)
53
54 def setTexture(self, texture):
55 self.vtk_actor.SetTexture(texture)
56
57 def addActor(self):
58 """
59 Add the actor to the renderer.
60 """
61
62 self.scene.getRenderer().AddActor(self.vtk_actor)
63
64 def setOpacity(self, opacity):
65 """
66 Set the opacity (transparency) of the actor.
67
68 @type opacity: Number
69 @param opacity: Opacity (transparency) of the actor
70 """
71
72 self.getProperty().SetOpacity(opacity)
73
74 def setColor(self, red, green, blue):
75 self.getProperty().SetColor(red, green, blue)
76
77 def setRepresentation(self, representation):
78 """
79 Set the representation of the actor.
80
81 @type representation: String
82 @param representation: Representation type (I{i.e. Wireframe})
83 """
84
85 eval("self.getProperty().SetRepresentationTo%s()" % representation)
86
87 def getProperty(self):
88 """
89 Return the property of the actor.
90
91 @rtype: vtkProperty
92 @return: VTK property
93 """
94
95 return self.vtk_actor.GetProperty()
96
97
98 class Component:
99 pass

  ViewVC Help
Powered by ViewVC 1.1.26