pyvisi/py_src/mapper.py

"""
@author: John NGUI
"""

import vtk

class DataSetMapper:
    """
    Class that defines a data set mapper.
    """

    # 'lookup_table = None' is used only by the outline.
    def __init__(self, object, lookup_table = None):
        """
        Initialise the data set mapper.

        @type object: vtkDataSet (i.e. vtkUnstructuredGrid, vtkPolyData, etc) 
        @param object: Data source map
        @type lookup_table: vtkLookupTable
        @param lookup_table: Maps scalar values to colors
        """

        self.__object = object
        self.__vtk_data_set_mapper = vtk.vtkDataSetMapper()
        self.__setInput()

        if(lookup_table != None): # False for the outline.
            self.__setLookupTable(lookup_table)

    def __setInput(self):
        """
        Set the input for the data set mapper.
        """

        self.__vtk_data_set_mapper.SetInput(self.__object)

    def __setLookupTable(self, lookup_table):
        """
        Set the lookup table for the data set mapper.
    
        @type lookup_table: vtkLookupTable
        @param lookup_table: Map scalar values to colors
        """

        self.__vtk_data_set_mapper.SetLookupTable(lookup_table)

    def _setScalarRange(self, range):
        """
        Set the minimum and maximum scalar range for the data set mapper.
        
        @type range: Two column tuple containing numbers
        @param range: Minimum and maximum data set mapper scalar range
        """

        self.__vtk_data_set_mapper.SetScalarRange(range) 
    
    def _setScalarVisibilityOn(self):
        """
        Scalar data is used to color the rendered object.
        """

        self.__vtk_data_set_mapper.ScalarVisibilityOn()
    
    def _getDataSetMapper(self):
        """
        Return the data set mapper.

        @rtype: vtkDataSetMapper    
        @return: Data set mapper
        """

        return self.__vtk_data_set_mapper


###############################################################################


class ImageMapper:
    def __init__(self, object):
        self.__object = object
        self.__vtk_image_mapper = vtk.vtkImageMapper()

        self.__setupImageMapper()

    def __setupImageMapper(self):
        self.__setInput()
        self.__setColorWindow(255)
        self.__setColorLevel(127.5)

    def __setInput(self):
        self.__vtk_image_mapper.SetInput(self.__object)

    def __setColorWindow(self, color):
        self.__vtk_image_mapper.SetColorWindow(color)

    def __setColorLevel(self, color):
        self.__vtk_image_mapper.SetColorLevel(color)

    def _getImageMapper(self):
        return self.__vtk_image_mapper
1	jongui	1037	"""
2			@author: John NGUI
3			"""
4
5			import vtk
6
7			class DataSetMapper:
8			"""
9			Class that defines a data set mapper.
10			"""
11
12			# 'lookup_table = None' is used only by the outline.
13			def __init__(self, object, lookup_table = None):
14			"""
15			Initialise the data set mapper.
16
17			@type object: vtkDataSet (i.e. vtkUnstructuredGrid, vtkPolyData, etc)
18			@param object: Data source map
19			@type lookup_table: vtkLookupTable
20			@param lookup_table: Maps scalar values to colors
21			"""
22
23			self.__object = object
24			self.__vtk_data_set_mapper = vtk.vtkDataSetMapper()
25			self.__setInput()
26
27			if(lookup_table != None): # False for the outline.
28			self.__setLookupTable(lookup_table)
29
30			def __setInput(self):
31			"""
32			Set the input for the data set mapper.
33			"""
34
35			self.__vtk_data_set_mapper.SetInput(self.__object)
36
37			def __setLookupTable(self, lookup_table):
38			"""
39			Set the lookup table for the data set mapper.
40
41			@type lookup_table: vtkLookupTable
42			@param lookup_table: Map scalar values to colors
43			"""
44
45			self.__vtk_data_set_mapper.SetLookupTable(lookup_table)
46
47			def _setScalarRange(self, range):
48			"""
49			Set the minimum and maximum scalar range for the data set mapper.
50
51			@type range: Two column tuple containing numbers
52			@param range: Minimum and maximum data set mapper scalar range
53			"""
54
55			self.__vtk_data_set_mapper.SetScalarRange(range)
56
57			def _setScalarVisibilityOn(self):
58			"""
59			Scalar data is used to color the rendered object.
60			"""
61
62			self.__vtk_data_set_mapper.ScalarVisibilityOn()
63
64			def _getDataSetMapper(self):
65			"""
66			Return the data set mapper.
67
68			@rtype: vtkDataSetMapper
69			@return: Data set mapper
70			"""
71
72			return self.__vtk_data_set_mapper
73
74	jongui	1108
75			###############################################################################
76
77
78			class ImageMapper:
79			def __init__(self, object):
80			self.__object = object
81			self.__vtk_image_mapper = vtk.vtkImageMapper()
82
83			self.__setupImageMapper()
84
85			def __setupImageMapper(self):
86			self.__setInput()
87	jongui	1112	self.__setColorWindow(255)
88			self.__setColorLevel(127.5)
89	jongui	1108
90			def __setInput(self):
91			self.__vtk_image_mapper.SetInput(self.__object)
92
93	jongui	1112	def __setColorWindow(self, color):
94	jongui	1108	self.__vtk_image_mapper.SetColorWindow(color)
95
96	jongui	1112	def __setColorLevel(self, color):
97	jongui	1108	self.__vtk_image_mapper.SetColorLevel(color)
98
99			def _getImageMapper(self):
100			return self.__vtk_image_mapper