258 lines
7.1 KiB
Python
258 lines
7.1 KiB
Python
#
|
|
# THIS IS WORK IN PROGRESS
|
|
#
|
|
# The Python Imaging Library.
|
|
# $Id$
|
|
#
|
|
# FlashPix support for PIL
|
|
#
|
|
# History:
|
|
# 97-01-25 fl Created (reads uncompressed RGB images only)
|
|
#
|
|
# Copyright (c) Secret Labs AB 1997.
|
|
# Copyright (c) Fredrik Lundh 1997.
|
|
#
|
|
# See the README file for information on usage and redistribution.
|
|
#
|
|
from __future__ import annotations
|
|
|
|
import olefile
|
|
|
|
from . import Image, ImageFile
|
|
from ._binary import i32le as i32
|
|
|
|
# we map from colour field tuples to (mode, rawmode) descriptors
|
|
MODES = {
|
|
# opacity
|
|
(0x00007FFE,): ("A", "L"),
|
|
# monochrome
|
|
(0x00010000,): ("L", "L"),
|
|
(0x00018000, 0x00017FFE): ("RGBA", "LA"),
|
|
# photo YCC
|
|
(0x00020000, 0x00020001, 0x00020002): ("RGB", "YCC;P"),
|
|
(0x00028000, 0x00028001, 0x00028002, 0x00027FFE): ("RGBA", "YCCA;P"),
|
|
# standard RGB (NIFRGB)
|
|
(0x00030000, 0x00030001, 0x00030002): ("RGB", "RGB"),
|
|
(0x00038000, 0x00038001, 0x00038002, 0x00037FFE): ("RGBA", "RGBA"),
|
|
}
|
|
|
|
|
|
#
|
|
# --------------------------------------------------------------------
|
|
|
|
|
|
def _accept(prefix: bytes) -> bool:
|
|
return prefix.startswith(olefile.MAGIC)
|
|
|
|
|
|
##
|
|
# Image plugin for the FlashPix images.
|
|
|
|
|
|
class FpxImageFile(ImageFile.ImageFile):
|
|
format = "FPX"
|
|
format_description = "FlashPix"
|
|
|
|
def _open(self) -> None:
|
|
#
|
|
# read the OLE directory and see if this is a likely
|
|
# to be a FlashPix file
|
|
|
|
try:
|
|
self.ole = olefile.OleFileIO(self.fp)
|
|
except OSError as e:
|
|
msg = "not an FPX file; invalid OLE file"
|
|
raise SyntaxError(msg) from e
|
|
|
|
root = self.ole.root
|
|
if not root or root.clsid != "56616700-C154-11CE-8553-00AA00A1F95B":
|
|
msg = "not an FPX file; bad root CLSID"
|
|
raise SyntaxError(msg)
|
|
|
|
self._open_index(1)
|
|
|
|
def _open_index(self, index: int = 1) -> None:
|
|
#
|
|
# get the Image Contents Property Set
|
|
|
|
prop = self.ole.getproperties(
|
|
[f"Data Object Store {index:06d}", "\005Image Contents"]
|
|
)
|
|
|
|
# size (highest resolution)
|
|
|
|
assert isinstance(prop[0x1000002], int)
|
|
assert isinstance(prop[0x1000003], int)
|
|
self._size = prop[0x1000002], prop[0x1000003]
|
|
|
|
size = max(self.size)
|
|
i = 1
|
|
while size > 64:
|
|
size = size // 2
|
|
i += 1
|
|
self.maxid = i - 1
|
|
|
|
# mode. instead of using a single field for this, flashpix
|
|
# requires you to specify the mode for each channel in each
|
|
# resolution subimage, and leaves it to the decoder to make
|
|
# sure that they all match. for now, we'll cheat and assume
|
|
# that this is always the case.
|
|
|
|
id = self.maxid << 16
|
|
|
|
s = prop[0x2000002 | id]
|
|
|
|
if not isinstance(s, bytes) or (bands := i32(s, 4)) > 4:
|
|
msg = "Invalid number of bands"
|
|
raise OSError(msg)
|
|
|
|
# note: for now, we ignore the "uncalibrated" flag
|
|
colors = tuple(i32(s, 8 + i * 4) & 0x7FFFFFFF for i in range(bands))
|
|
|
|
self._mode, self.rawmode = MODES[colors]
|
|
|
|
# load JPEG tables, if any
|
|
self.jpeg = {}
|
|
for i in range(256):
|
|
id = 0x3000001 | (i << 16)
|
|
if id in prop:
|
|
self.jpeg[i] = prop[id]
|
|
|
|
self._open_subimage(1, self.maxid)
|
|
|
|
def _open_subimage(self, index: int = 1, subimage: int = 0) -> None:
|
|
#
|
|
# setup tile descriptors for a given subimage
|
|
|
|
stream = [
|
|
f"Data Object Store {index:06d}",
|
|
f"Resolution {subimage:04d}",
|
|
"Subimage 0000 Header",
|
|
]
|
|
|
|
fp = self.ole.openstream(stream)
|
|
|
|
# skip prefix
|
|
fp.read(28)
|
|
|
|
# header stream
|
|
s = fp.read(36)
|
|
|
|
size = i32(s, 4), i32(s, 8)
|
|
# tilecount = i32(s, 12)
|
|
tilesize = i32(s, 16), i32(s, 20)
|
|
# channels = i32(s, 24)
|
|
offset = i32(s, 28)
|
|
length = i32(s, 32)
|
|
|
|
if size != self.size:
|
|
msg = "subimage mismatch"
|
|
raise OSError(msg)
|
|
|
|
# get tile descriptors
|
|
fp.seek(28 + offset)
|
|
s = fp.read(i32(s, 12) * length)
|
|
|
|
x = y = 0
|
|
xsize, ysize = size
|
|
xtile, ytile = tilesize
|
|
self.tile = []
|
|
|
|
for i in range(0, len(s), length):
|
|
x1 = min(xsize, x + xtile)
|
|
y1 = min(ysize, y + ytile)
|
|
|
|
compression = i32(s, i + 8)
|
|
|
|
if compression == 0:
|
|
self.tile.append(
|
|
ImageFile._Tile(
|
|
"raw",
|
|
(x, y, x1, y1),
|
|
i32(s, i) + 28,
|
|
self.rawmode,
|
|
)
|
|
)
|
|
|
|
elif compression == 1:
|
|
# FIXME: the fill decoder is not implemented
|
|
self.tile.append(
|
|
ImageFile._Tile(
|
|
"fill",
|
|
(x, y, x1, y1),
|
|
i32(s, i) + 28,
|
|
(self.rawmode, s[12:16]),
|
|
)
|
|
)
|
|
|
|
elif compression == 2:
|
|
internal_color_conversion = s[14]
|
|
jpeg_tables = s[15]
|
|
rawmode = self.rawmode
|
|
|
|
if internal_color_conversion:
|
|
# The image is stored as usual (usually YCbCr).
|
|
if rawmode == "RGBA":
|
|
# For "RGBA", data is stored as YCbCrA based on
|
|
# negative RGB. The following trick works around
|
|
# this problem :
|
|
jpegmode, rawmode = "YCbCrK", "CMYK"
|
|
else:
|
|
jpegmode = None # let the decoder decide
|
|
|
|
else:
|
|
# The image is stored as defined by rawmode
|
|
jpegmode = rawmode
|
|
|
|
self.tile.append(
|
|
ImageFile._Tile(
|
|
"jpeg",
|
|
(x, y, x1, y1),
|
|
i32(s, i) + 28,
|
|
(rawmode, jpegmode),
|
|
)
|
|
)
|
|
|
|
# FIXME: jpeg tables are tile dependent; the prefix
|
|
# data must be placed in the tile descriptor itself!
|
|
|
|
if jpeg_tables:
|
|
self.tile_prefix = self.jpeg[jpeg_tables]
|
|
|
|
else:
|
|
msg = "unknown/invalid compression"
|
|
raise OSError(msg)
|
|
|
|
x = x + xtile
|
|
if x >= xsize:
|
|
x, y = 0, y + ytile
|
|
if y >= ysize:
|
|
break # isn't really required
|
|
|
|
self.stream = stream
|
|
self._fp = self.fp
|
|
self.fp = None
|
|
|
|
def load(self) -> Image.core.PixelAccess | None:
|
|
if not self.fp:
|
|
self.fp = self.ole.openstream(self.stream[:2] + ["Subimage 0000 Data"])
|
|
|
|
return ImageFile.ImageFile.load(self)
|
|
|
|
def close(self) -> None:
|
|
self.ole.close()
|
|
super().close()
|
|
|
|
def __exit__(self, *args: object) -> None:
|
|
self.ole.close()
|
|
super().__exit__()
|
|
|
|
|
|
#
|
|
# --------------------------------------------------------------------
|
|
|
|
|
|
Image.register_open(FpxImageFile.format, FpxImageFile, _accept)
|
|
|
|
Image.register_extension(FpxImageFile.format, ".fpx")
|