Skip to content

Commit

Permalink
Merge pull request #7706 from radarhere/psd
Browse files Browse the repository at this point in the history
  • Loading branch information
hugovk committed Mar 11, 2024
2 parents fc30eba + 2888f76 commit 94f319c
Show file tree
Hide file tree
Showing 4 changed files with 23 additions and 8 deletions.
Binary file added Tests/images/negative_top_left_layer.psd
Binary file not shown.
5 changes: 5 additions & 0 deletions Tests/test_file_psd.py
Expand Up @@ -113,6 +113,11 @@ def test_rgba() -> None:
assert_image_equal_tofile(im, "Tests/images/imagedraw_square.png")


def test_negative_top_left_layer() -> None:
with Image.open("Tests/images/negative_top_left_layer.psd") as im:
assert im.layers[0][2] == (-50, -50, 50, 50)


def test_layer_skip() -> None:
with Image.open("Tests/images/five_channels.psd") as im:
assert im.n_frames == 1
Expand Down
16 changes: 8 additions & 8 deletions src/PIL/PsdImagePlugin.py
Expand Up @@ -24,6 +24,7 @@
from ._binary import i16be as i16
from ._binary import i32be as i32
from ._binary import si16be as si16
from ._binary import si32be as si32

MODES = {
# (photoshop mode, bits) -> (pil mode, required channels)
Expand Down Expand Up @@ -177,22 +178,21 @@ def read(size):

for _ in range(abs(ct)):
# bounding box
y0 = i32(read(4))
x0 = i32(read(4))
y1 = i32(read(4))
x1 = i32(read(4))
y0 = si32(read(4))
x0 = si32(read(4))
y1 = si32(read(4))
x1 = si32(read(4))

# image info
mode = []
ct_types = i16(read(2))
types = list(range(ct_types))
if len(types) > 4:
fp.seek(len(types) * 6 + 12, io.SEEK_CUR)
if ct_types > 4:
fp.seek(ct_types * 6 + 12, io.SEEK_CUR)
size = i32(read(4))
fp.seek(size, io.SEEK_CUR)
continue

for _ in types:
for _ in range(ct_types):
type = i16(read(2))

if type == 65535:
Expand Down
10 changes: 10 additions & 0 deletions src/PIL/_binary.py
Expand Up @@ -77,6 +77,16 @@ def si32le(c: bytes, o: int = 0) -> int:
return unpack_from("<i", c, o)[0]


def si32be(c: bytes, o: int = 0) -> int:
"""
Converts a 4-bytes (32 bits) string to a signed integer, big endian.
:param c: string containing bytes to convert
:param o: offset of bytes to convert in string
"""
return unpack_from(">i", c, o)[0]


def i16be(c: bytes, o: int = 0) -> int:
return unpack_from(">H", c, o)[0]

Expand Down

0 comments on commit 94f319c

Please sign in to comment.