pyvisi/py_src/colormap.py

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

import vtk  

class Lut:
    """
    Class that provides the functions to create a map spectrum.
    """

    def __init__(self):
        self.vtk_lut = vtk.vtkLookupTable() 
        
    def setHue(self, lower_range, upper_range):
        """
        Set the upper and lower hue(color) range.
        @type lower_range: Number
        @param lower_range: Lower range of the hue 
        @type upper_range: Number
        @param upper_range: Upper range of the hue 
        """

        self.vtk_lut.SetHueRange(lower_range, upper_range)

    def setSaturation(self, lower_range, upper_range):
        """
        Set the upper and lower saturation(vibrancy) range.
        @type lower_range: Number
        @param lower_range: Lower range of the saturation
        @type upper_range: Number
        @param upper_range: Higher range of the saturation
        """

        self.vtk_lut.SetSaturationRange(lower_range, upper_range)

    def setValue(self, lower_range, upper_range):
        """
        Set the upper and lower value(brightness) range.

        @type lower_range: Number
        @param lower_range: Lower range of the value 
        @type upper_range: Number
        @param upper_range: Upper range of the value
        """
    
        self.vtk_lut.SetValueRange(lower_range, upper_range)
        
    def getLut(self):
        """
        Return the lookup table.

        @rtype: vtkLookupTable
        @return: VTK Lookup table that changes the color spectrum
        """

        return self.vtk_lut

from colormap import Lut

class BlueToRed(Lut):
    """
    Class that creates a map with spectrum from blue to red.
    """

    def __init__(self):
        Lut.__init__(self)
        self.setHue(0.667, 0.0) 

from colormap import Lut

class RedToBlue(Lut):
    """
    Class that creates a map with spectrum from red to blue.
    """

    def __init__(self):
        Lut.__init__(self)
        self.setHue(0.0, 0.667)

1	"""
2	@author: John Ngui
3	@author: Lutz Gross
4	"""
5
6	import vtk
7
8	class Lut:
9	"""
10	Class that provides the functions to create a map spectrum.
11	"""
12
13	def __init__(self):
14	self.vtk_lut = vtk.vtkLookupTable()
15
16	def setHue(self, lower_range, upper_range):
17	"""
18	Set the upper and lower hue(color) range.
19	@type lower_range: Number
20	@param lower_range: Lower range of the hue
21	@type upper_range: Number
22	@param upper_range: Upper range of the hue
23	"""
24
25	self.vtk_lut.SetHueRange(lower_range, upper_range)
26
27	def setSaturation(self, lower_range, upper_range):
28	"""
29	Set the upper and lower saturation(vibrancy) range.
30	@type lower_range: Number
31	@param lower_range: Lower range of the saturation
32	@type upper_range: Number
33	@param upper_range: Higher range of the saturation
34	"""
35
36	self.vtk_lut.SetSaturationRange(lower_range, upper_range)
37
38	def setValue(self, lower_range, upper_range):
39	"""
40	Set the upper and lower value(brightness) range.
41
42	@type lower_range: Number
43	@param lower_range: Lower range of the value
44	@type upper_range: Number
45	@param upper_range: Upper range of the value
46	"""
47
48	self.vtk_lut.SetValueRange(lower_range, upper_range)
49
50	def getLut(self):
51	"""
52	Return the lookup table.
53
54	@rtype: vtkLookupTable
55	@return: VTK Lookup table that changes the color spectrum
56	"""
57
58	return self.vtk_lut
59
60	from colormap import Lut
61
62	class BlueToRed(Lut):
63	"""
64	Class that creates a map with spectrum from blue to red.
65	"""
66
67	def __init__(self):
68	Lut.__init__(self)
69	self.setHue(0.667, 0.0)
70
71	from colormap import Lut
72
73	class RedToBlue(Lut):
74	"""
75	Class that creates a map with spectrum from red to blue.
76	"""
77
78	def __init__(self):
79	Lut.__init__(self)
80	self.setHue(0.0, 0.667)
81