openmedialibrary_platform/Darwin/lib/python3.5/site-packages/PIL/PsdImagePlugin.py

313 lines
7.4 KiB
Python
Raw Normal View History

2013-10-11 17:28:32 +00:00
#
# The Python Imaging Library
# $Id$
#
# Adobe PSD 2.5/3.0 file handling
#
# History:
# 1995-09-01 fl Created
# 1997-01-03 fl Read most PSD images
# 1997-01-18 fl Fixed P and CMYK support
# 2001-10-21 fl Added seek/tell support (for layers)
#
# Copyright (c) 1997-2001 by Secret Labs AB.
# Copyright (c) 1995-2001 by Fredrik Lundh
#
# See the README file for information on usage and redistribution.
#
__version__ = "0.4"
2014-09-30 16:15:32 +00:00
from PIL import Image, ImageFile, ImagePalette, _binary
2013-10-11 17:28:32 +00:00
MODES = {
# (photoshop mode, bits) -> (pil mode, required channels)
(0, 1): ("1", 1),
(0, 8): ("L", 1),
(1, 8): ("L", 1),
(2, 8): ("P", 1),
(3, 8): ("RGB", 3),
(4, 8): ("CMYK", 4),
2015-11-25 01:25:01 +00:00
(7, 8): ("L", 1), # FIXME: multilayer
(8, 8): ("L", 1), # duotone
2013-10-11 17:28:32 +00:00
(9, 8): ("LAB", 3)
}
#
# helpers
2014-09-30 16:15:32 +00:00
i8 = _binary.i8
i16 = _binary.i16be
i32 = _binary.i32be
2013-10-11 17:28:32 +00:00
2015-11-25 01:25:01 +00:00
2013-10-11 17:28:32 +00:00
# --------------------------------------------------------------------.
# read PSD images
def _accept(prefix):
2014-09-30 16:15:32 +00:00
return prefix[:4] == b"8BPS"
2013-10-11 17:28:32 +00:00
2015-11-25 01:25:01 +00:00
2013-10-11 17:28:32 +00:00
##
# Image plugin for Photoshop images.
class PsdImageFile(ImageFile.ImageFile):
format = "PSD"
format_description = "Adobe Photoshop"
def _open(self):
read = self.fp.read
#
# header
s = read(26)
2014-09-30 16:15:32 +00:00
if s[:4] != b"8BPS" or i16(s[4:]) != 1:
raise SyntaxError("not a PSD file")
2013-10-11 17:28:32 +00:00
psd_bits = i16(s[22:])
psd_channels = i16(s[12:])
psd_mode = i16(s[24:])
mode, channels = MODES[(psd_mode, psd_bits)]
if channels > psd_channels:
2014-09-30 16:15:32 +00:00
raise IOError("not enough channels")
2013-10-11 17:28:32 +00:00
self.mode = mode
self.size = i32(s[18:]), i32(s[14:])
#
# color mode data
size = i32(read(4))
if size:
data = read(size)
if mode == "P" and size == 768:
self.palette = ImagePalette.raw("RGB;L", data)
#
# image resources
self.resources = []
size = i32(read(4))
if size:
# load resources
end = self.fp.tell() + size
while self.fp.tell() < end:
signature = read(4)
id = i16(read(2))
2014-09-30 16:15:32 +00:00
name = read(i8(read(1)))
2013-10-11 17:28:32 +00:00
if not (len(name) & 1):
2015-11-25 01:25:01 +00:00
read(1) # padding
2013-10-11 17:28:32 +00:00
data = read(i32(read(4)))
if (len(data) & 1):
2015-11-25 01:25:01 +00:00
read(1) # padding
2013-10-11 17:28:32 +00:00
self.resources.append((id, name, data))
2015-11-25 01:25:01 +00:00
if id == 1039: # ICC profile
2013-10-11 17:28:32 +00:00
self.info["icc_profile"] = data
#
# layer and mask information
self.layers = []
size = i32(read(4))
if size:
end = self.fp.tell() + size
size = i32(read(4))
if size:
self.layers = _layerinfo(self.fp)
self.fp.seek(end)
#
# image descriptor
self.tile = _maketile(self.fp, mode, (0, 0) + self.size, channels)
# keep the file open
self._fp = self.fp
self.frame = 0
2015-11-25 01:25:01 +00:00
@property
def n_frames(self):
return len(self.layers)
@property
def is_animated(self):
return len(self.layers) > 1
2013-10-11 17:28:32 +00:00
def seek(self, layer):
# seek to given layer (1..max)
if layer == self.frame:
return
try:
if layer <= 0:
raise IndexError
name, mode, bbox, tile = self.layers[layer-1]
self.mode = mode
self.tile = tile
self.frame = layer
self.fp = self._fp
return name, bbox
except IndexError:
2014-09-30 16:15:32 +00:00
raise EOFError("no such layer")
2013-10-11 17:28:32 +00:00
def tell(self):
# return layer number (0=image, 1..max=layers)
return self.frame
def load_prepare(self):
# create image memory if necessary
if not self.im or\
self.im.mode != self.mode or self.im.size != self.size:
self.im = Image.core.fill(self.mode, self.size, 0)
# create palette (optional)
if self.mode == "P":
Image.Image.load(self)
2015-11-25 01:25:01 +00:00
2013-10-11 17:28:32 +00:00
def _layerinfo(file):
# read layerinfo block
layers = []
read = file.read
for i in range(abs(i16(read(2)))):
# bounding box
2015-11-25 01:25:01 +00:00
y0 = i32(read(4))
x0 = i32(read(4))
y1 = i32(read(4))
x1 = i32(read(4))
2013-10-11 17:28:32 +00:00
# image info
info = []
mode = []
2014-09-30 16:15:32 +00:00
types = list(range(i16(read(2))))
if len(types) > 4:
continue
for i in types:
2013-10-11 17:28:32 +00:00
type = i16(read(2))
2014-09-30 16:15:32 +00:00
2013-10-11 17:28:32 +00:00
if type == 65535:
m = "A"
else:
2014-09-30 16:15:32 +00:00
m = "RGBA"[type]
2013-10-11 17:28:32 +00:00
mode.append(m)
size = i32(read(4))
info.append((m, size))
# figure out the image mode
mode.sort()
if mode == ["R"]:
mode = "L"
elif mode == ["B", "G", "R"]:
mode = "RGB"
elif mode == ["A", "B", "G", "R"]:
mode = "RGBA"
else:
2015-11-25 01:25:01 +00:00
mode = None # unknown
2013-10-11 17:28:32 +00:00
# skip over blend flags and extra information
filler = read(12)
name = ""
size = i32(read(4))
combined = 0
if size:
length = i32(read(4))
if length:
2015-11-25 01:25:01 +00:00
mask_y = i32(read(4))
mask_x = i32(read(4))
mask_h = i32(read(4)) - mask_y
mask_w = i32(read(4)) - mask_x
2013-10-11 17:28:32 +00:00
file.seek(length - 16, 1)
combined += length + 4
length = i32(read(4))
if length:
file.seek(length, 1)
combined += length + 4
2014-09-30 16:15:32 +00:00
length = i8(read(1))
2013-10-11 17:28:32 +00:00
if length:
2015-11-25 01:25:01 +00:00
# Don't know the proper encoding,
# Latin-1 should be a good guess
2014-09-30 16:15:32 +00:00
name = read(length).decode('latin-1', 'replace')
2013-10-11 17:28:32 +00:00
combined += length + 1
file.seek(size - combined, 1)
layers.append((name, mode, (x0, y0, x1, y1)))
# get tiles
i = 0
for name, mode, bbox in layers:
tile = []
for m in mode:
t = _maketile(file, m, bbox, 1)
if t:
tile.extend(t)
layers[i] = name, mode, bbox, tile
2014-09-30 16:15:32 +00:00
i += 1
2013-10-11 17:28:32 +00:00
return layers
2015-11-25 01:25:01 +00:00
2013-10-11 17:28:32 +00:00
def _maketile(file, mode, bbox, channels):
tile = None
read = file.read
compression = i16(read(2))
xsize = bbox[2] - bbox[0]
ysize = bbox[3] - bbox[1]
offset = file.tell()
if compression == 0:
#
# raw compression
tile = []
for channel in range(channels):
layer = mode[channel]
if mode == "CMYK":
2014-09-30 16:15:32 +00:00
layer += ";I"
2013-10-11 17:28:32 +00:00
tile.append(("raw", bbox, offset, layer))
offset = offset + xsize*ysize
elif compression == 1:
#
# packbits compression
i = 0
tile = []
bytecount = read(channels * ysize * 2)
offset = file.tell()
for channel in range(channels):
layer = mode[channel]
if mode == "CMYK":
2014-09-30 16:15:32 +00:00
layer += ";I"
2013-10-11 17:28:32 +00:00
tile.append(
("packbits", bbox, offset, layer)
)
for y in range(ysize):
offset = offset + i16(bytecount[i:i+2])
2014-09-30 16:15:32 +00:00
i += 2
2013-10-11 17:28:32 +00:00
file.seek(offset)
if offset & 1:
2015-11-25 01:25:01 +00:00
read(1) # padding
2013-10-11 17:28:32 +00:00
return tile
# --------------------------------------------------------------------
# registry
2015-11-25 01:25:01 +00:00
Image.register_open(PsdImageFile.format, PsdImageFile, _accept)
2013-10-11 17:28:32 +00:00
2015-11-25 01:25:01 +00:00
Image.register_extension(PsdImageFile.format, ".psd")