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: String
        @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 _getOutput(self):
        """
        Return the output of the image reader.

        @rtype: vtkImageData
        @return: Image Data 
        """

        return self.__vtk_image_reader.GetOutput()

1	"""
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: String
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	def _getOutput(self):
56	"""
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