pyvisi/py_src/imagereader.py

"""
@author: John NGUI
"""

import vtk
from constant import ImageFormat

class ImageReader:
    """
    Class that defines an image reader. An image reader is used to read
    data from an image in a variety of formats.
    """

    def __init__(self, format):
        """ 
        Initialise the image reader.

        @type format:  L{ImageFormat <constant.ImageFormat>} constant
        @param format: Format of the image 
        """

        self.__format = format
        self.__vtk_image_reader = self.__getImageReader()

    def __getImageReader(self):
        """
        Return the corresponding image reader based on the supplied image 
        format.

        @rtype: vtkImageReader2 (i.e. vtkJPEGReader, etc)
        @return: Image reader 
        """

        if(self.__format == ImageFormat.JPG):
            return vtk.vtkJPEGReader()  
        elif(self.__format == ImageFormat.BMP):
            return vtk.vtkBMPReader()
        elif(self.__format == ImageFormat.PNM):
            return vtk.vtkPNMReader()
        elif(self.__format == ImageFormat.PNG):
            return vtk.vtkPNGReader()
        elif(self.__format == ImageFormat.TIF):
            return vtk.vtkTIFFReader()

    def setImageName(self, image_name):
        """
        Set the image file name to be read.

        @type image_name: String
        @param image_name: Image name from which data is to be read 
        """

        self.__vtk_image_reader.SetFileName(image_name)

    def _getImageReaderOutput(self):
        """
        Return the output of the image reader.

        @rtype: vtkImageData
        @return: Image data 
        """

        return self.__vtk_image_reader.GetOutput()

1	ksteube	1147	"""
2			@author: John NGUI
3			"""
4
5			import vtk
6			from constant import ImageFormat
7
8			class ImageReader:
9			"""
10			Class that defines an image reader. An image reader is used to read
11			data from an image in a variety of formats.
12			"""
13
14			def __init__(self, format):
15			"""
16			Initialise the image reader.
17
18			@type format: L{ImageFormat <constant.ImageFormat>} constant
19			@param format: Format of the image
20			"""
21
22			self.__format = format
23			self.__vtk_image_reader = self.__getImageReader()
24
25			def __getImageReader(self):
26			"""
27			Return the corresponding image reader based on the supplied image
28			format.
29
30			@rtype: vtkImageReader2 (i.e. vtkJPEGReader, etc)
31			@return: Image reader
32			"""
33
34			if(self.__format == ImageFormat.JPG):
35			return vtk.vtkJPEGReader()
36			elif(self.__format == ImageFormat.BMP):
37			return vtk.vtkBMPReader()
38			elif(self.__format == ImageFormat.PNM):
39			return vtk.vtkPNMReader()
40			elif(self.__format == ImageFormat.PNG):
41			return vtk.vtkPNGReader()
42			elif(self.__format == ImageFormat.TIF):
43			return vtk.vtkTIFFReader()
44
45			def setImageName(self, image_name):
46			"""
47			Set the image file name to be read.
48
49			@type image_name: String
50			@param image_name: Image name from which data is to be read
51			"""
52
53			self.__vtk_image_reader.SetFileName(image_name)
54
55	jongui	1148	def _getImageReaderOutput(self):
56	ksteube	1147	"""
57			Return the output of the image reader.
58
59			@rtype: vtkImageData
60			@return: Image data
61			"""
62
63			return self.__vtk_image_reader.GetOutput()
64