python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-01-26 17:24:31 +03:00
Code Issues Packages Projects Releases Wiki Activity

Simplified use of struct

Browse Source
This commit is contained in:
Andrew Murray 2021-06-29 20:20:34 +10:00
parent f5558f42f4
commit d07a085757
1 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/PIL/IcnsImagePlugin.py
View File

@ -331,24 +331,22 @@ def _save(im, fp, filename):
entries.append(
{"type": _to_int(size_str[index]), "size": len(stream), "stream": stream}
)
file_size = sum(entry["size"] for entry in entries)
# Header
fp.write(struct.pack("<i", _to_int(MAGIC))[::-1])
fp.write(struct.pack("<i", file_size)[::-1])
fp.write(struct.pack(">i", _to_int(MAGIC)))
fp.write(struct.pack(">i", sum(entry["size"] for entry in entries)))
# TOC
toc_size = HEADERSIZE + (len(entries) * HEADERSIZE)
fp.write(struct.pack("<i", _to_int(TOC))[::-1])
fp.write(struct.pack("<i", toc_size)[::-1])
fp.write(struct.pack(">i", _to_int(TOC)))
fp.write(struct.pack(">i", HEADERSIZE + len(entries) * HEADERSIZE))
for entry in entries:
fp.write(struct.pack("<i", entry["type"])[::-1])
fp.write(struct.pack("<i", HEADERSIZE + entry["size"])[::-1])
fp.write(struct.pack(">i", entry["type"]))
fp.write(struct.pack(">i", HEADERSIZE + entry["size"]))
# Data
for entry in entries:
fp.write(struct.pack("<i", entry["type"])[::-1])
fp.write(struct.pack("<i", HEADERSIZE + entry["size"])[::-1])
fp.write(struct.pack(">i", entry["type"]))
fp.write(struct.pack(">i", HEADERSIZE + entry["size"]))
fp.write(entry["stream"])
if hasattr(fp, "flush"):