blob: e99b75873f300f66486d64b22b6d90fea84178b7 [file] [log] [blame]
Roger E. Masse4455cd81996-12-20 21:52:29 +00001#! /usr/bin/env python
Guido van Rossum7aa9fc51997-04-16 00:30:45 +00002
Roger E. Masse4455cd81996-12-20 21:52:29 +00003"""Simple test script for imgfile.c
4 Roger E. Masse
5"""
Roger E. Masse033d91b1996-12-20 21:41:50 +00006
Fredrik Lundhf7850422001-01-17 21:51:36 +00007from test_support import verbose, unlink, findfile
Roger E. Masse033d91b1996-12-20 21:41:50 +00008
Guido van Rossum7aa9fc51997-04-16 00:30:45 +00009import imgfile, uu, os
10
Fred Drake004d5e62000-10-23 17:22:08 +000011
Guido van Rossum7aa9fc51997-04-16 00:30:45 +000012def main():
13
14 uu.decode(findfile('testrgb.uue'), 'test.rgb')
15 uu.decode(findfile('greyrgb.uue'), 'greytest.rgb')
Roger E. Masse033d91b1996-12-20 21:41:50 +000016
Roger E. Masse4455cd81996-12-20 21:52:29 +000017 # Test a 3 byte color image
Roger E. Masse3507f501996-12-20 21:47:58 +000018 testimage('test.rgb')
Fred Drake004d5e62000-10-23 17:22:08 +000019
Roger E. Masse4455cd81996-12-20 21:52:29 +000020 # Test a 1 byte greyscale image
Roger E. Masse3507f501996-12-20 21:47:58 +000021 testimage('greytest.rgb')
Roger E. Masse033d91b1996-12-20 21:41:50 +000022
Guido van Rossum7aa9fc51997-04-16 00:30:45 +000023 unlink('test.rgb')
24 unlink('greytest.rgb')
25
Roger E. Masse3507f501996-12-20 21:47:58 +000026def testimage(name):
Roger E. Masse4455cd81996-12-20 21:52:29 +000027 """Run through the imgfile's battery of possible methods
28 on the image passed in name.
Roger E. Masse033d91b1996-12-20 21:41:50 +000029 """
30
31 import sys
32 import os
Roger E. Masse033d91b1996-12-20 21:41:50 +000033
34 outputfile = '/tmp/deleteme'
35
36 # try opening the name directly
37 try:
Guido van Rossum41360a41998-03-26 19:42:58 +000038 # This function returns a tuple (x, y, z) where x and y are the size
39 # of the image in pixels and z is the number of bytes per pixel. Only
40 # 3 byte RGB pixels and 1 byte greyscale pixels are supported.
41 sizes = imgfile.getsizes(name)
Roger E. Masse033d91b1996-12-20 21:41:50 +000042 except imgfile.error:
Guido van Rossum41360a41998-03-26 19:42:58 +000043 # get a more qualified path component of the script...
44 if __name__ == '__main__':
45 ourname = sys.argv[0]
46 else: # ...or the full path of the module
47 ourname = sys.modules[__name__].__file__
Roger E. Masse033d91b1996-12-20 21:41:50 +000048
Eric S. Raymondfc170b12001-02-09 11:51:27 +000049 parts = ourname.split(os.sep)
Guido van Rossum41360a41998-03-26 19:42:58 +000050 parts[-1] = name
Eric S. Raymonddbbbaf22001-02-09 17:05:53 +000051 name = os.sep.join(parts)
Guido van Rossum41360a41998-03-26 19:42:58 +000052 sizes = imgfile.getsizes(name)
Roger E. Masse033d91b1996-12-20 21:41:50 +000053 if verbose:
Guido van Rossum41360a41998-03-26 19:42:58 +000054 print 'Opening test image: %s, sizes: %s' % (name, str(sizes))
Roger E. Masse033d91b1996-12-20 21:41:50 +000055 # This function reads and decodes the image on the specified file,
56 # and returns it as a python string. The string has either 1 byte
57 # greyscale pixels or 4 byte RGBA pixels. The bottom left pixel
58 # is the first in the string. This format is suitable to pass
Fred Drake004d5e62000-10-23 17:22:08 +000059 # to gl.lrectwrite, for instance.
Roger E. Masse033d91b1996-12-20 21:41:50 +000060 image = imgfile.read(name)
Fred Drake004d5e62000-10-23 17:22:08 +000061
Roger E. Masse033d91b1996-12-20 21:41:50 +000062 # This function writes the RGB or greyscale data in data to
63 # image file file. x and y give the size of the image, z is
64 # 1 for 1 byte greyscale images or 3 for RGB images (which
65 # are stored as 4 byte values of which only the lower three
Fred Drake004d5e62000-10-23 17:22:08 +000066 # bytes are used). These are the formats returned by gl.lrectread.
Roger E. Masse033d91b1996-12-20 21:41:50 +000067 if verbose:
Guido van Rossum41360a41998-03-26 19:42:58 +000068 print 'Writing output file'
Fred Drake004d5e62000-10-23 17:22:08 +000069 imgfile.write (outputfile, image, sizes[0], sizes[1], sizes[2])
70
Roger E. Masse033d91b1996-12-20 21:41:50 +000071
72 if verbose:
Guido van Rossum41360a41998-03-26 19:42:58 +000073 print 'Opening scaled test image: %s, sizes: %s' % (name, str(sizes))
Roger E. Masse033d91b1996-12-20 21:41:50 +000074 # This function is identical to read but it returns an image that
75 # is scaled to the given x and y sizes. If the filter and blur
76 # parameters are omitted scaling is done by simply dropping
77 # or duplicating pixels, so the result will be less than perfect,
78 # especially for computer-generated images. Alternatively,
79 # you can specify a filter to use to smoothen the image after
80 # scaling. The filter forms supported are 'impulse', 'box',
81 # 'triangle', 'quadratic' and 'gaussian'. If a filter is
82 # specified blur is an optional parameter specifying the
83 # blurriness of the filter. It defaults to 1.0. readscaled
84 # makes no attempt to keep the aspect ratio correct, so that
85 # is the users' responsibility.
86 if verbose:
Guido van Rossum41360a41998-03-26 19:42:58 +000087 print 'Filtering with "impulse"'
Roger E. Masse033d91b1996-12-20 21:41:50 +000088 simage = imgfile.readscaled (name, sizes[0]/2, sizes[1]/2, 'impulse', 2.0)
89
90 # This function sets a global flag which defines whether the
91 # scan lines of the image are read or written from bottom to
92 # top (flag is zero, compatible with SGI GL) or from top to
Fred Drake004d5e62000-10-23 17:22:08 +000093 # bottom(flag is one, compatible with X). The default is zero.
Roger E. Masse033d91b1996-12-20 21:41:50 +000094 if verbose:
Guido van Rossum41360a41998-03-26 19:42:58 +000095 print 'Switching to X compatibility'
Fred Drake004d5e62000-10-23 17:22:08 +000096 imgfile.ttob (1)
Roger E. Masse033d91b1996-12-20 21:41:50 +000097
98 if verbose:
Guido van Rossum41360a41998-03-26 19:42:58 +000099 print 'Filtering with "triangle"'
Roger E. Masse033d91b1996-12-20 21:41:50 +0000100 simage = imgfile.readscaled (name, sizes[0]/2, sizes[1]/2, 'triangle', 3.0)
101 if verbose:
Guido van Rossum41360a41998-03-26 19:42:58 +0000102 print 'Switching back to SGI compatibility'
Fred Drake004d5e62000-10-23 17:22:08 +0000103 imgfile.ttob (0)
104
Roger E. Masse033d91b1996-12-20 21:41:50 +0000105 if verbose: print 'Filtering with "quadratic"'
106 simage = imgfile.readscaled (name, sizes[0]/2, sizes[1]/2, 'quadratic')
107 if verbose: print 'Filtering with "gaussian"'
108 simage = imgfile.readscaled (name, sizes[0]/2, sizes[1]/2, 'gaussian', 1.0)
109
110 if verbose:
Guido van Rossum41360a41998-03-26 19:42:58 +0000111 print 'Writing output file'
Fred Drake004d5e62000-10-23 17:22:08 +0000112 imgfile.write (outputfile, simage, sizes[0]/2, sizes[1]/2, sizes[2])
Roger E. Masse033d91b1996-12-20 21:41:50 +0000113
114 os.unlink(outputfile)
115
116main()