home *** CD-ROM | disk | FTP | other *** search
/ Enter 2004 April / enter-2004-04.iso / files / EVE_1424_100181.exe / GimpGradientFile.py < prev    next >
Encoding:
Python Source  |  2004-04-20  |  3.3 KB  |  126 lines
  1. #
  2. # Python Imaging Library
  3. # $Id: //modules/pil/PIL/GimpGradientFile.py#2 $
  4. #
  5. # stuff to read (and render) GIMP gradient files
  6. #
  7. # History:
  8. #       97-08-23 fl     Created
  9. #
  10. # Copyright (c) Secret Labs AB 1997.
  11. # Copyright (c) Fredrik Lundh 1997.
  12. #
  13. # See the README file for information on usage and redistribution.
  14. #
  15.  
  16. from math import pi, log, sin, sqrt
  17. import string
  18.  
  19. # --------------------------------------------------------------------
  20. # Stuff to translate curve segments to palette values (derived from
  21. # the corresponding code in GIMP, written by Federico Mena Quintero.
  22. # See the GIMP distribution for more information.)
  23. #
  24.  
  25. EPSILON = 1e-10
  26.  
  27. def linear(middle, pos):
  28.     if pos <= middle:
  29.         if middle < EPSILON:
  30.             return 0.0
  31.         else:
  32.             return 0.5 * pos / middle
  33.     else:
  34.         pos = pos - middle
  35.         middle = 1.0 - middle
  36.         if middle < EPSILON:
  37.             return 1.0
  38.         else:
  39.             return 0.5 + 0.5 * pos / middle
  40.  
  41. def curved(middle, pos):
  42.     return pos ** (log(0.5) / log(max(middle, EPSILON)))
  43.  
  44. def sine(middle, pos):
  45.     return (sin((-pi / 2.0) + pi * linear(middle, pos)) + 1.0) / 2.0
  46.  
  47. def sphere_increasing(middle, pos):
  48.     return sqrt(1.0 - (linear(middle, pos) - 1.0) ** 2)
  49.  
  50. def sphere_decreasing(middle, pos):
  51.     return 1.0 - sqrt(1.0 - linear(middle, pos) ** 2)
  52.  
  53. SEGMENTS = [ linear, curved, sine, sphere_increasing, sphere_decreasing ]
  54.  
  55. # --------------------------------------------------------------------
  56. #
  57.  
  58. class GradientFile:
  59.  
  60.     gradient = None
  61.  
  62.     def getpalette(self, entries = 256):
  63.  
  64.         palette = []
  65.  
  66.         ix = 0
  67.         x0, x1, xm, rgb0, rgb1, segment = self.gradient[ix]
  68.  
  69.         for i in range(entries):
  70.  
  71.             x = i / float(entries-1)
  72.  
  73.             while x1 < x:
  74.                 ix = ix + 1
  75.                 x0, x1, xm, rgb0, rgb1, segment = self.gradient[ix]
  76.  
  77.             w = x1 - x0
  78.  
  79.             if w < EPSILON:
  80.                 scale = segment(0.5, 0.5)
  81.             else:
  82.                 scale = segment((xm - x0) / w, (x - x0) / w)
  83.  
  84.             # expand to RGBA
  85.             r = chr(int(255 * ((rgb1[0] - rgb0[0]) * scale + rgb0[0]) + 0.5))
  86.             g = chr(int(255 * ((rgb1[1] - rgb0[1]) * scale + rgb0[1]) + 0.5))
  87.             b = chr(int(255 * ((rgb1[2] - rgb0[2]) * scale + rgb0[2]) + 0.5))
  88.             a = chr(int(255 * ((rgb1[3] - rgb0[3]) * scale + rgb0[3]) + 0.5))
  89.  
  90.             # add to palette
  91.             palette.append(r + g + b + a)
  92.  
  93.         return string.join(palette, ""), "RGBA"
  94.  
  95.  
  96. class GimpGradientFile(GradientFile):
  97.  
  98.     def __init__(self, fp):
  99.  
  100.         if fp.readline()[:13] != "GIMP Gradient":
  101.             raise SyntaxError, "not a GIMP gradient file"
  102.  
  103.         count = string.atoi(fp.readline())
  104.  
  105.         gradient = []
  106.  
  107.         for i in range(count):
  108.  
  109.             s = string.split(fp.readline())
  110.             w = map(string.atof, s[:11])
  111.  
  112.             x0, x1  = w[0], w[2]
  113.             xm      = w[1]
  114.             rgb0    = w[3:7]
  115.             rgb1    = w[7:11]
  116.  
  117.             segment = SEGMENTS[string.atoi(s[11])]
  118.             cspace  = string.atoi(s[12])
  119.  
  120.             if cspace != 0:
  121.                 raise IOError, "cannot handle HSV colour space"
  122.  
  123.             gradient.append((x0, x1, xm, rgb0, rgb1, segment))
  124.  
  125.         self.gradient = gradient
  126.