home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / pypil112.zip / Scripts / gifmaker.py < prev    next >
Text File  |  2001-05-03  |  3KB  |  136 lines
  1. #
  2. # The Python Imaging Library
  3. # $Id$
  4. #
  5. # convert sequence format to GIF animation
  6. #
  7. # history:
  8. #       97-01-03 fl     created
  9. #
  10. # Copyright (c) Secret Labs AB 1997.  All rights reserved.
  11. # Copyright (c) Fredrik Lundh 1997.
  12. #
  13. # See the README file for information on usage and redistribution.
  14. #
  15.  
  16. #
  17. # For special purposes, you can import this module and call
  18. # the makedelta or compress functions yourself.  For example,
  19. # if you have an application that generates a sequence of
  20. # images, you can convert it to a GIF animation using some-
  21. # thing like the following code:
  22. #
  23. #       import Image
  24. #       import gifmaker
  25. #       
  26. #       sequence = []
  27. #       
  28. #       # generate sequence
  29. #       for i in range(100):
  30. #           im = <generate image i>
  31. #           sequence.append(im)
  32. #
  33. #       # write GIF animation
  34. #       fp = open("out.gif", "wb")
  35. #       gifmaker.makedelta(fp, sequence)
  36. #       fp.close()
  37. # 
  38. # Alternatively, use an iterator to generate the sequence, and
  39. # write data directly to a socket.  Or something...
  40. #
  41.  
  42. import Image, ImageChops
  43. import string
  44.  
  45. from GifImagePlugin import getheader, getdata
  46.  
  47. # --------------------------------------------------------------------
  48. # sequence iterator
  49.  
  50. class image_sequence:
  51.     def __init__(self, im):
  52.         self.im = im
  53.     def __getitem__(self, ix):
  54.         try:
  55.             if ix:
  56.                 self.im.seek(ix)
  57.             return self.im
  58.         except EOFError:
  59.             raise IndexError # end of sequence
  60.  
  61. # --------------------------------------------------------------------
  62. # straightforward delta encoding
  63.  
  64. def makedelta(fp, sequence):
  65.     """Convert list of image frames to a GIF animation file"""
  66.  
  67.     frames = 0
  68.  
  69.     previous = None
  70.  
  71.     for im in sequence:
  72.  
  73.         #
  74.         # FIXME: write graphics control block before each frame
  75.  
  76.         if not previous:
  77.  
  78.             # global header
  79.             for s in getheader(im) + getdata(im):
  80.                 fp.write(s)
  81.  
  82.         else:
  83.  
  84.             # delta frame
  85.             delta = ImageChops.subtract_modulo(im, previous)
  86.  
  87.             bbox = delta.getbbox()
  88.  
  89.             if bbox:
  90.  
  91.                 # compress difference
  92.                 for s in getdata(im.crop(bbox), offset = bbox[:2]):
  93.                     fp.write(s)
  94.  
  95.             else:
  96.                 # FIXME: what should we do in this case?
  97.                 pass
  98.  
  99.         previous = im.copy()
  100.  
  101.         frames = frames + 1
  102.  
  103.     fp.write(";")
  104.  
  105.     return frames
  106.  
  107. # --------------------------------------------------------------------
  108. # main hack
  109.  
  110. def compress(infile, outfile):
  111.  
  112.     # open input image, and force loading of first frame
  113.     im = Image.open(infile)
  114.     im.load()
  115.  
  116.     # open output file
  117.     fp = open(outfile, "wb")
  118.  
  119.     seq = image_sequence(im)
  120.  
  121.     makedelta(fp, seq)
  122.  
  123.     fp.close()
  124.  
  125.  
  126. if __name__ == "__main__":
  127.  
  128.     import sys
  129.  
  130.     if len(sys.argv) < 3:
  131.         print "GIFMAKER -- create GIF animations"
  132.         print "Usage: gifmaker infile outfile"
  133.         sys.exit(1)
  134.  
  135.     compress(sys.argv[1], sys.argv[2])
  136.