Pillow/src/PIL/FliImagePlugin.py

172 lines
4.3 KiB
Python
Raw Normal View History

2010-07-31 06:52:47 +04:00
#
# The Python Imaging Library.
# $Id$
#
# FLI/FLC file handling.
#
# History:
# 95-09-01 fl Created
# 97-01-03 fl Fixed parser, setup decoder tile
# 98-07-15 fl Renamed offset attribute to avoid name clash
#
# Copyright (c) Secret Labs AB 1997-98.
# Copyright (c) Fredrik Lundh 1995-97.
#
# See the README file for information on usage and redistribution.
#
2022-09-29 14:29:16 +03:00
import os
2010-07-31 06:52:47 +04:00
from . import Image, ImageFile, ImagePalette
2020-09-01 20:16:46 +03:00
from ._binary import i16le as i16
from ._binary import i32le as i32
from ._binary import o8
2010-07-31 06:52:47 +04:00
#
# decoder
2019-03-21 16:28:20 +03:00
2010-07-31 06:52:47 +04:00
def _accept(prefix):
2022-02-26 09:34:13 +03:00
return (
len(prefix) >= 6
and i16(prefix, 4) in [0xAF11, 0xAF12]
and i16(prefix, 14) in [0, 3] # flags
)
2010-07-31 06:52:47 +04:00
2014-08-28 15:44:19 +04:00
2010-07-31 06:52:47 +04:00
##
# Image plugin for the FLI/FLC animation format. Use the <b>seek</b>
# method to load individual frames.
2019-03-21 16:28:20 +03:00
2010-07-31 06:52:47 +04:00
class FliImageFile(ImageFile.ImageFile):
format = "FLI"
format_description = "Autodesk FLI/FLC Animation"
_close_exclusive_fp_after_loading = False
2017-08-12 05:11:36 +03:00
2010-07-31 06:52:47 +04:00
def _open(self):
# HEAD
s = self.fp.read(128)
2022-02-26 09:34:13 +03:00
if not (_accept(s) and s[20:22] == b"\x00\x00"):
msg = "not an FLI/FLC file"
raise SyntaxError(msg)
2010-07-31 06:52:47 +04:00
2017-08-12 05:11:36 +03:00
# frames
self.n_frames = i16(s, 6)
self.is_animated = self.n_frames > 1
2017-08-12 05:11:36 +03:00
2010-07-31 06:52:47 +04:00
# image characteristics
self.mode = "P"
self._size = i16(s, 8), i16(s, 10)
2010-07-31 06:52:47 +04:00
# animation speed
duration = i32(s, 16)
magic = i16(s, 4)
2010-07-31 06:52:47 +04:00
if magic == 0xAF11:
2017-08-04 14:51:02 +03:00
duration = (duration * 1000) // 70
2010-07-31 06:52:47 +04:00
self.info["duration"] = duration
# look for palette
2014-08-28 15:44:19 +04:00
palette = [(a, a, a) for a in range(256)]
2010-07-31 06:52:47 +04:00
s = self.fp.read(16)
self.__offset = 128
if i16(s, 4) == 0xF100:
2010-07-31 06:52:47 +04:00
# prefix chunk; ignore it
self.__offset = self.__offset + i32(s)
s = self.fp.read(16)
if i16(s, 4) == 0xF1FA:
2010-07-31 06:52:47 +04:00
# look for palette chunk
2022-09-29 14:29:16 +03:00
number_of_subchunks = i16(s, 6)
chunk_size = None
for _ in range(number_of_subchunks):
if chunk_size is not None:
self.fp.seek(chunk_size - 6, os.SEEK_CUR)
s = self.fp.read(6)
chunk_type = i16(s, 4)
if chunk_type in (4, 11):
self._palette(palette, 2 if chunk_type == 11 else 0)
break
chunk_size = i32(s)
if not chunk_size:
break
2010-07-31 06:52:47 +04:00
2019-03-21 16:28:20 +03:00
palette = [o8(r) + o8(g) + o8(b) for (r, g, b) in palette]
self.palette = ImagePalette.raw("RGB", b"".join(palette))
2010-07-31 06:52:47 +04:00
# set things up to decode first frame
self.__frame = -1
2022-04-13 02:54:17 +03:00
self._fp = self.fp
self.__rewind = self.fp.tell()
2010-07-31 06:52:47 +04:00
self.seek(0)
def _palette(self, palette, shift):
# load palette
i = 0
for e in range(i16(self.fp.read(2))):
s = self.fp.read(2)
i = i + s[0]
n = s[1]
2010-07-31 06:52:47 +04:00
if n == 0:
n = 256
s = self.fp.read(n * 3)
for n in range(0, len(s), 3):
r = s[n] << shift
g = s[n + 1] << shift
b = s[n + 2] << shift
2010-07-31 06:52:47 +04:00
palette[i] = (r, g, b)
i += 1
2010-07-31 06:52:47 +04:00
def seek(self, frame):
if not self._seek_check(frame):
return
if frame < self.__frame:
self._seek(0)
for f in range(self.__frame + 1, frame + 1):
2017-08-12 05:11:36 +03:00
self._seek(f)
def _seek(self, frame):
if frame == 0:
self.__frame = -1
2022-04-13 02:54:17 +03:00
self._fp.seek(self.__rewind)
2015-06-07 18:01:50 +03:00
self.__offset = 128
else:
# ensure that the previous frame was loaded
self.load()
2010-07-31 06:52:47 +04:00
if frame != self.__frame + 1:
msg = f"cannot seek to frame {frame}"
raise ValueError(msg)
self.__frame = frame
2010-07-31 06:52:47 +04:00
# move to next frame
2022-04-13 02:54:17 +03:00
self.fp = self._fp
2010-07-31 06:52:47 +04:00
self.fp.seek(self.__offset)
s = self.fp.read(4)
if not s:
raise EOFError
framesize = i32(s)
self.decodermaxblock = framesize
2019-03-21 16:28:20 +03:00
self.tile = [("fli", (0, 0) + self.size, self.__offset, None)]
2010-07-31 06:52:47 +04:00
2015-06-07 18:01:50 +03:00
self.__offset += framesize
2010-07-31 06:52:47 +04:00
def tell(self):
return self.__frame
2010-07-31 06:52:47 +04:00
2017-08-04 14:51:02 +03:00
2010-07-31 06:52:47 +04:00
#
# registry
Image.register_open(FliImageFile.format, FliImageFile, _accept)
2010-07-31 06:52:47 +04:00
2016-04-25 07:59:02 +03:00
Image.register_extensions(FliImageFile.format, [".fli", ".flc"])