home *** CD-ROM | disk | FTP | other *** search
/ Enter 2004 April / enter-2004-04.iso / files / EVE_1424_100181.exe / PpmImagePlugin.py < prev    next >
Encoding:
Text File  |  2004-04-20  |  3.3 KB  |  129 lines
  1. #
  2. # The Python Imaging Library.
  3. # $Id: //modules/pil/PIL/PpmImagePlugin.py#2 $
  4. #
  5. # PPM support for PIL
  6. #
  7. # History:
  8. #       96-03-24 fl     Created
  9. #       98-03-06 fl     Write RGBA images (as RGB, that is)
  10. #
  11. # Copyright (c) Secret Labs AB 1997-98.
  12. # Copyright (c) Fredrik Lundh 1996.
  13. #
  14. # See the README file for information on usage and redistribution.
  15. #
  16.  
  17.  
  18. __version__ = "0.2"
  19.  
  20. import string
  21.  
  22. import Image, ImageFile
  23.  
  24. #
  25. # --------------------------------------------------------------------
  26.  
  27. MODES = {
  28.     # standard
  29.     "P4": "1",
  30.     "P5": "L",
  31.     "P6": "RGB",
  32.     # extensions
  33.     "P0CMYK": "CMYK",
  34.     # PIL extensions (for test purposes only)
  35.     "PyP": "P",
  36.     "PyRGBA": "RGBA",
  37.     "PyCMYK": "CMYK"
  38. }
  39.  
  40. def _accept(prefix):
  41.     return prefix[0] == "P" and prefix[1] in "0456y"
  42.  
  43. class PpmImageFile(ImageFile.ImageFile):
  44.  
  45.     format = "PPM"
  46.     format_description = "Pbmplus image"
  47.  
  48.     def _token(self, s = ""):
  49.         while 1: # read until next whitespace
  50.             c = self.fp.read(1)
  51.             if not c or c in string.whitespace:
  52.                 break
  53.             s = s + c
  54.         return s
  55.  
  56.     def _open(self):
  57.  
  58.         # check magic
  59.         s = self.fp.read(1)
  60.         if s != "P":
  61.             raise SyntaxError, "not a PPM file"
  62.         mode = MODES[self._token(s)]
  63.  
  64.         if mode == "1":
  65.             self.mode = "1"
  66.             rawmode = "1;I"
  67.         else:
  68.             self.mode = rawmode = mode
  69.  
  70.         for ix in range(3):
  71.             while 1:
  72.                 while 1:
  73.                     s = self.fp.read(1)
  74.                     if s not in string.whitespace:
  75.                         break
  76.                 if s != "#":
  77.                     break
  78.                 s = self.fp.readline()
  79.             s = int(self._token(s))
  80.             if ix == 0:
  81.                 xsize = s
  82.             elif ix == 1:
  83.                 ysize = s
  84.                 if mode == "1":
  85.                     break
  86.  
  87.         self.size = xsize, ysize
  88.         self.tile = [("raw",
  89.                      (0, 0, xsize, ysize),
  90.                      self.fp.tell(),
  91.                      (rawmode, 0, 1))]
  92.  
  93.         # ALTERNATIVE: load via builtin debug function
  94.         # self.im = Image.core.open_ppm(self.filename)
  95.         # self.mode = self.im.mode
  96.         # self.size = self.im.size
  97.  
  98. #
  99. # --------------------------------------------------------------------
  100.  
  101. def _save(im, fp, filename):
  102.     if im.mode == "1":
  103.         rawmode, head = "1;I", "P4"
  104.     elif im.mode == "L":
  105.         rawmode, head = "L", "P5"
  106.     elif im.mode == "RGB":
  107.         rawmode, head = "RGB", "P6"
  108.     elif im.mode == "RGBA":
  109.         rawmode, head = "RGB", "P6"
  110.     else:
  111.         raise IOError, "cannot write mode %s as PPM" % im.mode
  112.     fp.write(head + "\n%d %d\n" % im.size)
  113.     if head != "P4":
  114.         fp.write("255\n")
  115.     ImageFile._save(im, fp, [("raw", (0,0)+im.size, 0, (rawmode, 0, 1))])
  116.  
  117.     # ALTERNATIVE: save via builtin debug function
  118.     # im._dump(filename)
  119.  
  120. #
  121. # --------------------------------------------------------------------
  122.  
  123. Image.register_open("PPM", PpmImageFile, _accept)
  124. Image.register_save("PPM", _save)
  125.  
  126. Image.register_extension("PPM", ".pbm")
  127. Image.register_extension("PPM", ".pgm")
  128. Image.register_extension("PPM", ".ppm")
  129.