home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / pypil112.zip / PIL-1.1.2.zip / Lib / site-packages / PIL / SgiImagePlugin.py < prev    next >
Text File  |  2001-05-03  |  2KB  |  90 lines
  1. #
  2. # The Python Imaging Library.
  3. # $Id$
  4. #
  5. # SGI image file handling
  6. #
  7. # See "The SGI Image File Format (Draft version 0.97)", Paul Haeberli.
  8. # <ftp://ftp.sgi.com/graphics/SGIIMAGESPEC>
  9. #
  10. # History:
  11. # 1995-09-10 fl     Created
  12. #
  13. # Copyright (c) Secret Labs AB 1997.
  14. # Copyright (c) Fredrik Lundh 1995.
  15. #
  16. # See the README file for information on usage and redistribution.
  17. #
  18.  
  19.  
  20. __version__ = "0.1"
  21.  
  22.  
  23. import string
  24. import Image, ImageFile
  25.  
  26.  
  27. def i16(c):
  28.     return ord(c[1]) + (ord(c[0])<<8)
  29.  
  30. def i32(c):
  31.     return ord(c[3]) + (ord(c[2])<<8) + (ord(c[1])<<16) + (ord(c[0])<<24)
  32.  
  33.  
  34. def _accept(prefix):
  35.     return i16(prefix) == 474
  36.  
  37.  
  38. class SgiImageFile(ImageFile.ImageFile):
  39.  
  40.     format = "SGI"
  41.     format_description = "SGI Image File Format"
  42.  
  43.     def _open(self):
  44.  
  45.         # HEAD
  46.         s = self.fp.read(512)
  47.         if i16(s) != 474:
  48.             raise SyntaxError, "not an SGI image file"
  49.  
  50.         # relevant header entries
  51.         compression = ord(s[2])
  52.  
  53.         # bytes, dimension, zsize
  54.         layout = ord(s[3]), i16(s[4:]), i16(s[10:])
  55.  
  56.         # determine mode from bytes/zsize
  57.         if layout == (1, 2, 1):
  58.             self.mode = "L"
  59.         elif layout == (1, 3, 3):
  60.             self.mode = "RGB"
  61.         else:
  62.             raise SyntaxError, "unsupported SGI image mode"
  63.  
  64.         # size
  65.         self.size = i16(s[6:]), i16(s[8:])
  66.  
  67.         # decoder info
  68.         if compression == 0:
  69.             if self.mode == "RGB":
  70.                 # RGB images are band interleaved
  71.                 size = self.size[0]*self.size[1]
  72.                 self.tile = [("raw", (0,0)+self.size, 512, ("R",0,1)),
  73.                              ("raw", (0,0)+self.size, 512+size, ("G",0,1)),
  74.                              ("raw", (0,0)+self.size, 512+2*size, ("B",0,1))]
  75.             else:
  76.                 self.tile = [("raw", (0,0)+self.size, 512, (self.mode, 0, 1))]
  77.         if compression == 1:
  78.             self.tile = [("sgi_rle", (0,0)+self.size, 512, (self.mode, 0, 1))]
  79.  
  80. #
  81. # registry
  82.  
  83. Image.register_open("SGI", SgiImageFile, _accept)
  84.  
  85. Image.register_extension("SGI", ".bw")
  86. Image.register_extension("SGI", ".rgb")
  87. Image.register_extension("SGI", ".rgba")
  88.  
  89. Image.register_extension("SGI", ".sgi") # really?
  90.