mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-26 18:06:18 +03:00
a7e3b2e47b
There are two main issues fixed with this commit: * bytes vs. str: All file, image, and palette data are now handled as bytes. A new _binary module consolidates the hacks needed to do this across Python versions. tostring/fromstring methods have been renamed to tobytes/frombytes, but the Python 2.6/2.7 versions alias them to the old names for compatibility. Users should move to tobytes/frombytes. One other potentially-breaking change is that text data in image files (such as tags, comments) are now explicitly handled with a specific character encoding in mind. This works well with the Unicode str in Python 3, but may trip up old code expecting a straight byte-for-byte translation to a Python string. This also required a change to Gohlke's tags tests (in Tests/test_file_png.py) to expect Unicode strings from the code. * True div vs. floor div: Many division operations used the "/" operator to do floor division, which is now the "//" operator in Python 3. These were fixed. As of this commit, on the first pass, I have one failing test (improper handling of a slice object in a C module, test_imagepath.py) in Python 3, and three that that I haven't tried running yet (test_imagegl, test_imagegrab, and test_imageqt). I also haven't tested anything on Windows. All but the three skipped tests run flawlessly against Pythons 2.6 and 2.7.
125 lines
3.1 KiB
Python
125 lines
3.1 KiB
Python
#
|
|
# Python Imaging Library
|
|
# $Id$
|
|
#
|
|
# stuff to read (and render) GIMP gradient files
|
|
#
|
|
# History:
|
|
# 97-08-23 fl Created
|
|
#
|
|
# Copyright (c) Secret Labs AB 1997.
|
|
# Copyright (c) Fredrik Lundh 1997.
|
|
#
|
|
# See the README file for information on usage and redistribution.
|
|
#
|
|
|
|
from math import pi, log, sin, sqrt
|
|
from ._binary import o8
|
|
|
|
# --------------------------------------------------------------------
|
|
# Stuff to translate curve segments to palette values (derived from
|
|
# the corresponding code in GIMP, written by Federico Mena Quintero.
|
|
# See the GIMP distribution for more information.)
|
|
#
|
|
|
|
EPSILON = 1e-10
|
|
|
|
def linear(middle, pos):
|
|
if pos <= middle:
|
|
if middle < EPSILON:
|
|
return 0.0
|
|
else:
|
|
return 0.5 * pos / middle
|
|
else:
|
|
pos = pos - middle
|
|
middle = 1.0 - middle
|
|
if middle < EPSILON:
|
|
return 1.0
|
|
else:
|
|
return 0.5 + 0.5 * pos / middle
|
|
|
|
def curved(middle, pos):
|
|
return pos ** (log(0.5) / log(max(middle, EPSILON)))
|
|
|
|
def sine(middle, pos):
|
|
return (sin((-pi / 2.0) + pi * linear(middle, pos)) + 1.0) / 2.0
|
|
|
|
def sphere_increasing(middle, pos):
|
|
return sqrt(1.0 - (linear(middle, pos) - 1.0) ** 2)
|
|
|
|
def sphere_decreasing(middle, pos):
|
|
return 1.0 - sqrt(1.0 - linear(middle, pos) ** 2)
|
|
|
|
SEGMENTS = [ linear, curved, sine, sphere_increasing, sphere_decreasing ]
|
|
|
|
class GradientFile:
|
|
|
|
gradient = None
|
|
|
|
def getpalette(self, entries = 256):
|
|
|
|
palette = []
|
|
|
|
ix = 0
|
|
x0, x1, xm, rgb0, rgb1, segment = self.gradient[ix]
|
|
|
|
for i in range(entries):
|
|
|
|
x = i / float(entries-1)
|
|
|
|
while x1 < x:
|
|
ix = ix + 1
|
|
x0, x1, xm, rgb0, rgb1, segment = self.gradient[ix]
|
|
|
|
w = x1 - x0
|
|
|
|
if w < EPSILON:
|
|
scale = segment(0.5, 0.5)
|
|
else:
|
|
scale = segment((xm - x0) / w, (x - x0) / w)
|
|
|
|
# expand to RGBA
|
|
r = o8(int(255 * ((rgb1[0] - rgb0[0]) * scale + rgb0[0]) + 0.5))
|
|
g = o8(int(255 * ((rgb1[1] - rgb0[1]) * scale + rgb0[1]) + 0.5))
|
|
b = o8(int(255 * ((rgb1[2] - rgb0[2]) * scale + rgb0[2]) + 0.5))
|
|
a = o8(int(255 * ((rgb1[3] - rgb0[3]) * scale + rgb0[3]) + 0.5))
|
|
|
|
# add to palette
|
|
palette.append(r + g + b + a)
|
|
|
|
return b"".join(palette), "RGBA"
|
|
|
|
##
|
|
# File handler for GIMP's gradient format.
|
|
|
|
class GimpGradientFile(GradientFile):
|
|
|
|
def __init__(self, fp):
|
|
|
|
if fp.readline()[:13] != b"GIMP Gradient":
|
|
raise SyntaxError("not a GIMP gradient file")
|
|
|
|
count = int(fp.readline())
|
|
|
|
gradient = []
|
|
|
|
for i in range(count):
|
|
|
|
s = fp.readline().split()
|
|
w = [float(x) for x in s[:11]]
|
|
|
|
x0, x1 = w[0], w[2]
|
|
xm = w[1]
|
|
rgb0 = w[3:7]
|
|
rgb1 = w[7:11]
|
|
|
|
segment = SEGMENTS[int(s[11])]
|
|
cspace = int(s[12])
|
|
|
|
if cspace != 0:
|
|
raise IOError("cannot handle HSV colour space")
|
|
|
|
gradient.append((x0, x1, xm, rgb0, rgb1, segment))
|
|
|
|
self.gradient = gradient
|