pyvisi/py_src/common.py

"""
@author: John Ngui
@author: Lutz Gross
"""

import vtk

class Common:
    """
    Class that defines the common operations invoked by the components. 
    """

    def __init__(self, scene, data_collector = None):
        """
        @type scene: L{Scene <scene.Scene>} object
        @param scene: Scene in which components are to be added to
        @type data_collector: L{DataCollector <datacollector.DataCollector>} 
            object
        @param data_collector: Source of data for visualization
        """

        self.scene = scene
        self.data_collector = data_collector
        self.vtk_mapper = vtk.vtkDataSetMapper() 
        self.vtk_actor = vtk.vtkActor() 

    def setMapperInput(self, component, lut = None):
        """
        Set up the mapper.
        @type component: String
        @param component: Component to be mapped
        @type lut: L{BlueToRed <colormap.BlueToRed>} or 
            L{RedToBlue <colormap.RedToBlue>} object
        @param lut: Lookup table to be used by the mapper
        """
        
        # Convert unstructured grid data to polygonal data. 
        vtk_geometry = vtk.vtkGeometryFilter()
        vtk_geometry.SetInput(component)

        # Compute normals to ensure consistent orientation across neighbours.
        # This results in a better object being rendered. 
        #vtk_normals = vtk.vtkPolyDataNormals()
        #vtk_normals.SetInput(vtk_geometry.GetOutput())

        #self.vtk_mapper.SetInput(vtk_normals.GetOutput())
        #self.vtk_mapper.SetInput(vtk_geometry.GetOutput())
        self.vtk_mapper.SetInput(component)

    
        # Mapper uses the customized lookup table only if it is specified. 
        # Otherwise, the default one is used.
        if(lut != None):
            self.vtk_mapper.SetLookupTable(lut.getLut())    

    def setActorTexture(self, texture):
        """
        Set the texture of the actor.
        @type texture: vtkTexture
        @param texture: Texture map of the image
        """

        self.vtk_actor.SetTexture(texture)

    def setActorInput(self):
        """
        Set up the actor.
        """

        self.vtk_actor.SetMapper(self.vtk_mapper)


    def addActor(self):
        """
        Add the actor to the renderer.
        """

        self.scene.getRenderer().AddActor(self.vtk_actor) 

    def setActorOpacity(self, opacity):
        """
        Set the opacity (transparency) of the actor.
        @type opacity: Number
        @param opacity: Opacity (transparency) of the actor
        """

        self.vtk_actor.GetProperty().SetOpacity(opacity)

    def setActorColor(self, color):
        """
        Set the color of the actor.
        @type color: RGB list
        @param color: Color of the actor
        """
        
        self.vtk_actor.GetProperty().SetColor(color[0], color[1],
            color[2])

    def setActorRepresentation(self, representation):
        """
        Set the representation of the actor.
        @type representation: String
        @param representation: Actor representation type (I{i.e. Wireframe})
        """

        eval("self.vtk_actor.GetProperty().SetRepresentationTo%s()" % 
            representation)
    

class Component:
     pass
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	@type scene: L{Scene <scene.Scene>} object
16	@param scene: Scene in which components are to be added to
17	@type data_collector: L{DataCollector <datacollector.DataCollector>}
18	object
19	@param data_collector: Source of data for visualization
20	"""
21
22	self.scene = scene
23	self.data_collector = data_collector
24	self.vtk_mapper = vtk.vtkDataSetMapper()
25	self.vtk_actor = vtk.vtkActor()
26
27	def setMapperInput(self, component, lut = None):
28	"""
29	Set up the mapper.
30	@type component: String
31	@param component: Component to be mapped
32	@type lut: L{BlueToRed <colormap.BlueToRed>} or
33	L{RedToBlue <colormap.RedToBlue>} object
34	@param lut: Lookup table to be used by the mapper
35	"""
36
37	# Convert unstructured grid data to polygonal data.
38	vtk_geometry = vtk.vtkGeometryFilter()
39	vtk_geometry.SetInput(component)
40
41	# Compute normals to ensure consistent orientation across neighbours.
42	# This results in a better object being rendered.
43	#vtk_normals = vtk.vtkPolyDataNormals()
44	#vtk_normals.SetInput(vtk_geometry.GetOutput())
45
46	#self.vtk_mapper.SetInput(vtk_normals.GetOutput())
47	#self.vtk_mapper.SetInput(vtk_geometry.GetOutput())
48	self.vtk_mapper.SetInput(component)
49
50
51	# Mapper uses the customized lookup table only if it is specified.
52	# Otherwise, the default one is used.
53	if(lut != None):
54	self.vtk_mapper.SetLookupTable(lut.getLut())
55
56	def setActorTexture(self, texture):
57	"""
58	Set the texture of the actor.
59	@type texture: vtkTexture
60	@param texture: Texture map of the image
61	"""
62
63	self.vtk_actor.SetTexture(texture)
64
65	def setActorInput(self):
66	"""
67	Set up the actor.
68	"""
69
70	self.vtk_actor.SetMapper(self.vtk_mapper)
71
72
73	def addActor(self):
74	"""
75	Add the actor to the renderer.
76	"""
77
78	self.scene.getRenderer().AddActor(self.vtk_actor)
79
80	def setActorOpacity(self, opacity):
81	"""
82	Set the opacity (transparency) of the actor.
83	@type opacity: Number
84	@param opacity: Opacity (transparency) of the actor
85	"""
86
87	self.vtk_actor.GetProperty().SetOpacity(opacity)
88
89	def setActorColor(self, color):
90	"""
91	Set the color of the actor.
92	@type color: RGB list
93	@param color: Color of the actor
94	"""
95
96	self.vtk_actor.GetProperty().SetColor(color[0], color[1],
97	color[2])
98
99	def setActorRepresentation(self, representation):
100	"""
101	Set the representation of the actor.
102	@type representation: String
103	@param representation: Actor representation type (I{i.e. Wireframe})
104	"""
105
106	eval("self.vtk_actor.GetProperty().SetRepresentationTo%s()" %
107	representation)
108
109
110	class Component:
111	pass