cpython/Lib/dos-8x3/test_img.py

126 lines
4.4 KiB
Python
Raw Normal View History

1997-04-02 02:13:34 -04:00
#! /usr/bin/env python
1997-05-08 20:14:57 -03:00
1997-04-02 02:13:34 -04:00
"""Simple test script for imgfile.c
Roger E. Masse
"""
1997-05-08 20:14:57 -03:00
from test_support import verbose, unlink
import imgfile, uu, os
1997-04-02 02:13:34 -04:00
1997-05-08 20:14:57 -03:00
def main():
uu.decode(findfile('testrgb.uue'), 'test.rgb')
uu.decode(findfile('greyrgb.uue'), 'greytest.rgb')
1997-04-02 02:13:34 -04:00
# Test a 3 byte color image
testimage('test.rgb')
# Test a 1 byte greyscale image
testimage('greytest.rgb')
1997-05-08 20:14:57 -03:00
unlink('test.rgb')
unlink('greytest.rgb')
def findfile(file):
1998-03-26 18:14:20 -04:00
if os.path.isabs(file): return file
import sys
for dn in sys.path:
fn = os.path.join(dn, file)
if os.path.exists(fn): return fn
return file
1997-04-02 02:13:34 -04:00
def testimage(name):
"""Run through the imgfile's battery of possible methods
on the image passed in name.
"""
import sys
import os
import string
outputfile = '/tmp/deleteme'
# try opening the name directly
try:
1998-03-26 18:14:20 -04:00
# This function returns a tuple (x, y, z) where x and y are the size
# of the image in pixels and z is the number of bytes per pixel. Only
# 3 byte RGB pixels and 1 byte greyscale pixels are supported.
sizes = imgfile.getsizes(name)
1997-04-02 02:13:34 -04:00
except imgfile.error:
1998-03-26 18:14:20 -04:00
# get a more qualified path component of the script...
if __name__ == '__main__':
ourname = sys.argv[0]
else: # ...or the full path of the module
ourname = sys.modules[__name__].__file__
parts = string.splitfields(ourname, os.sep)
parts[-1] = name
name = string.joinfields(parts, os.sep)
sizes = imgfile.getsizes(name)
1997-04-02 02:13:34 -04:00
if verbose:
1998-03-26 18:14:20 -04:00
print 'Opening test image: %s, sizes: %s' % (name, str(sizes))
1997-04-02 02:13:34 -04:00
# This function reads and decodes the image on the specified file,
# and returns it as a python string. The string has either 1 byte
# greyscale pixels or 4 byte RGBA pixels. The bottom left pixel
# is the first in the string. This format is suitable to pass
# to gl.lrectwrite, for instance.
image = imgfile.read(name)
# This function writes the RGB or greyscale data in data to
# image file file. x and y give the size of the image, z is
# 1 for 1 byte greyscale images or 3 for RGB images (which
# are stored as 4 byte values of which only the lower three
# bytes are used). These are the formats returned by gl.lrectread.
if verbose:
1998-03-26 18:14:20 -04:00
print 'Writing output file'
1997-04-02 02:13:34 -04:00
imgfile.write (outputfile, image, sizes[0], sizes[1], sizes[2])
if verbose:
1998-03-26 18:14:20 -04:00
print 'Opening scaled test image: %s, sizes: %s' % (name, str(sizes))
1997-04-02 02:13:34 -04:00
# This function is identical to read but it returns an image that
# is scaled to the given x and y sizes. If the filter and blur
# parameters are omitted scaling is done by simply dropping
# or duplicating pixels, so the result will be less than perfect,
# especially for computer-generated images. Alternatively,
# you can specify a filter to use to smoothen the image after
# scaling. The filter forms supported are 'impulse', 'box',
# 'triangle', 'quadratic' and 'gaussian'. If a filter is
# specified blur is an optional parameter specifying the
# blurriness of the filter. It defaults to 1.0. readscaled
# makes no attempt to keep the aspect ratio correct, so that
# is the users' responsibility.
if verbose:
1998-03-26 18:14:20 -04:00
print 'Filtering with "impulse"'
1997-04-02 02:13:34 -04:00
simage = imgfile.readscaled (name, sizes[0]/2, sizes[1]/2, 'impulse', 2.0)
# This function sets a global flag which defines whether the
# scan lines of the image are read or written from bottom to
# top (flag is zero, compatible with SGI GL) or from top to
# bottom(flag is one, compatible with X). The default is zero.
if verbose:
1998-03-26 18:14:20 -04:00
print 'Switching to X compatibility'
1997-04-02 02:13:34 -04:00
imgfile.ttob (1)
if verbose:
1998-03-26 18:14:20 -04:00
print 'Filtering with "triangle"'
1997-04-02 02:13:34 -04:00
simage = imgfile.readscaled (name, sizes[0]/2, sizes[1]/2, 'triangle', 3.0)
if verbose:
1998-03-26 18:14:20 -04:00
print 'Switching back to SGI compatibility'
1997-04-02 02:13:34 -04:00
imgfile.ttob (0)
if verbose: print 'Filtering with "quadratic"'
simage = imgfile.readscaled (name, sizes[0]/2, sizes[1]/2, 'quadratic')
if verbose: print 'Filtering with "gaussian"'
simage = imgfile.readscaled (name, sizes[0]/2, sizes[1]/2, 'gaussian', 1.0)
if verbose:
1998-03-26 18:14:20 -04:00
print 'Writing output file'
1997-04-02 02:13:34 -04:00
imgfile.write (outputfile, simage, sizes[0]/2, sizes[1]/2, sizes[2])
os.unlink(outputfile)
main()