blob: f283063c000c7936bc4b468db380c0e8e79c120d [file] [log] [blame]
Guido van Rossum2db91351992-10-18 17:09:59 +00001# Convert "arbitrary" image files to rgb files (SGI's image format).
2# Input may be compressed.
3# The uncompressed file type may be PBM, PGM, PPM, GIF, TIFF, or Sun raster.
4# An exception is raised if the file is not of a recognized type.
5# Returned filename is either the input filename or a temporary filename;
6# in the latter case the caller must ensure that it is removed.
7# Other temporary files used are removed by the function.
8
9import os
10import tempfile
11import pipes
12import imghdr
13
14table = {}
15
Guido van Rossum7bc817d1993-12-17 15:25:27 +000016t = pipes.Template()
Guido van Rossum2db91351992-10-18 17:09:59 +000017t.append('fromppm $IN $OUT', 'ff')
18table['ppm'] = t
19
Guido van Rossum7bc817d1993-12-17 15:25:27 +000020t = pipes.Template()
Sjoerd Mullender35fe6ec1993-10-11 12:39:15 +000021t.append('(PATH=$PATH:/ufs/guido/bin/sgi; exec pnmtoppm)', '--')
Guido van Rossum2db91351992-10-18 17:09:59 +000022t.append('fromppm $IN $OUT', 'ff')
23table['pnm'] = t
24table['pgm'] = t
25table['pbm'] = t
26
Guido van Rossum7bc817d1993-12-17 15:25:27 +000027t = pipes.Template()
Guido van Rossum2db91351992-10-18 17:09:59 +000028t.append('fromgif $IN $OUT', 'ff')
29table['gif'] = t
30
Guido van Rossum7bc817d1993-12-17 15:25:27 +000031t = pipes.Template()
Guido van Rossum2db91351992-10-18 17:09:59 +000032t.append('tifftopnm', '--')
33t.append('(PATH=$PATH:/ufs/guido/bin/sgi; exec pnmtoppm)', '--')
34t.append('fromppm $IN $OUT', 'ff')
35table['tiff'] = t
36
Guido van Rossum7bc817d1993-12-17 15:25:27 +000037t = pipes.Template()
Guido van Rossum2db91351992-10-18 17:09:59 +000038t.append('rasttopnm', '--')
Sjoerd Mullender35fe6ec1993-10-11 12:39:15 +000039t.append('(PATH=$PATH:/ufs/guido/bin/sgi; exec pnmtoppm)', '--')
Guido van Rossum2db91351992-10-18 17:09:59 +000040t.append('fromppm $IN $OUT', 'ff')
41table['rast'] = t
42
Guido van Rossum7bc817d1993-12-17 15:25:27 +000043t = pipes.Template()
Guido van Rossum5cfa5df1993-06-23 09:30:50 +000044t.append('djpeg', '--')
Sjoerd Mullender35fe6ec1993-10-11 12:39:15 +000045t.append('(PATH=$PATH:/ufs/guido/bin/sgi; exec pnmtoppm)', '--')
Guido van Rossum5cfa5df1993-06-23 09:30:50 +000046t.append('fromppm $IN $OUT', 'ff')
47table['jpeg'] = t
48
Guido van Rossum7bc817d1993-12-17 15:25:27 +000049uncompress = pipes.Template()
Guido van Rossum2db91351992-10-18 17:09:59 +000050uncompress.append('uncompress', '--')
51
52
53error = 'torgb.error' # Exception
54
55def torgb(filename):
56 temps = []
57 ret = None
58 try:
59 ret = _torgb(filename, temps)
60 finally:
61 for temp in temps[:]:
62 if temp <> ret:
63 try:
64 os.unlink(temp)
65 except os.error:
66 pass
67 temps.remove(temp)
68 return ret
69
70def _torgb(filename, temps):
71 if filename[-2:] == '.Z':
72 fname = tempfile.mktemp()
73 temps.append(fname)
74 sts = uncompress.copy(filename, fname)
75 if sts:
76 raise error, filename + ': uncompress failed'
77 else:
78 fname = filename
79 try:
80 ftype = imghdr.what(fname)
81 except IOError, msg:
82 if type(msg) == type(()) and len(msg) == 2 and \
83 type(msg[0]) == type(0) and type(msg[1]) == type(''):
84 msg = msg[1]
85 if type(msg) <> type(''):
86 msg = `msg`
87 raise error, filename + ': ' + msg
88 if ftype == 'rgb':
89 return fname
90 if ftype == None or not table.has_key(ftype):
91 raise error, \
92 filename + ': unsupported image file type ' + `ftype`
93 temp = tempfile.mktemp()
94 sts = table[ftype].copy(fname, temp)
95 if sts:
96 raise error, filename + ': conversion to rgb failed'
97 return temp