123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910 |
- #
- # The Python Imaging Library.
- # $Id$
- #
- # GIF file handling
- #
- # History:
- # 1995-09-01 fl Created
- # 1996-12-14 fl Added interlace support
- # 1996-12-30 fl Added animation support
- # 1997-01-05 fl Added write support, fixed local colour map bug
- # 1997-02-23 fl Make sure to load raster data in getdata()
- # 1997-07-05 fl Support external decoder (0.4)
- # 1998-07-09 fl Handle all modes when saving (0.5)
- # 1998-07-15 fl Renamed offset attribute to avoid name clash
- # 2001-04-16 fl Added rewind support (seek to frame 0) (0.6)
- # 2001-04-17 fl Added palette optimization (0.7)
- # 2002-06-06 fl Added transparency support for save (0.8)
- # 2004-02-24 fl Disable interlacing for small images
- #
- # Copyright (c) 1997-2004 by Secret Labs AB
- # Copyright (c) 1995-2004 by Fredrik Lundh
- #
- # See the README file for information on usage and redistribution.
- #
- import itertools
- import math
- import os
- import subprocess
- from . import Image, ImageChops, ImageFile, ImagePalette, ImageSequence
- from ._binary import i16le as i16
- from ._binary import o8
- from ._binary import o16le as o16
- # --------------------------------------------------------------------
- # Identify/read GIF files
- def _accept(prefix):
- return prefix[:6] in [b"GIF87a", b"GIF89a"]
- ##
- # Image plugin for GIF images. This plugin supports both GIF87 and
- # GIF89 images.
- class GifImageFile(ImageFile.ImageFile):
- format = "GIF"
- format_description = "Compuserve GIF"
- _close_exclusive_fp_after_loading = False
- global_palette = None
- def data(self):
- s = self.fp.read(1)
- if s and s[0]:
- return self.fp.read(s[0])
- return None
- def _open(self):
- # Screen
- s = self.fp.read(13)
- if not _accept(s):
- raise SyntaxError("not a GIF file")
- self.info["version"] = s[:6]
- self._size = i16(s, 6), i16(s, 8)
- self.tile = []
- flags = s[10]
- bits = (flags & 7) + 1
- if flags & 128:
- # get global palette
- self.info["background"] = s[11]
- # check if palette contains colour indices
- p = self.fp.read(3 << bits)
- for i in range(0, len(p), 3):
- if not (i // 3 == p[i] == p[i + 1] == p[i + 2]):
- p = ImagePalette.raw("RGB", p)
- self.global_palette = self.palette = p
- break
- self.__fp = self.fp # FIXME: hack
- self.__rewind = self.fp.tell()
- self._n_frames = None
- self._is_animated = None
- self._seek(0) # get ready to read first frame
- @property
- def n_frames(self):
- if self._n_frames is None:
- current = self.tell()
- try:
- while True:
- self.seek(self.tell() + 1)
- except EOFError:
- self._n_frames = self.tell() + 1
- self.seek(current)
- return self._n_frames
- @property
- def is_animated(self):
- if self._is_animated is None:
- if self._n_frames is not None:
- self._is_animated = self._n_frames != 1
- else:
- current = self.tell()
- try:
- self.seek(1)
- self._is_animated = True
- except EOFError:
- self._is_animated = False
- self.seek(current)
- return self._is_animated
- def seek(self, frame):
- if not self._seek_check(frame):
- return
- if frame < self.__frame:
- if frame != 0:
- self.im = None
- self._seek(0)
- last_frame = self.__frame
- for f in range(self.__frame + 1, frame + 1):
- try:
- self._seek(f)
- except EOFError as e:
- self.seek(last_frame)
- raise EOFError("no more images in GIF file") from e
- def _seek(self, frame):
- if frame == 0:
- # rewind
- self.__offset = 0
- self.dispose = None
- self.dispose_extent = [0, 0, 0, 0] # x0, y0, x1, y1
- self.__frame = -1
- self.__fp.seek(self.__rewind)
- self.disposal_method = 0
- else:
- # ensure that the previous frame was loaded
- if self.tile:
- self.load()
- if frame != self.__frame + 1:
- raise ValueError(f"cannot seek to frame {frame}")
- self.__frame = frame
- self.tile = []
- self.fp = self.__fp
- if self.__offset:
- # backup to last frame
- self.fp.seek(self.__offset)
- while self.data():
- pass
- self.__offset = 0
- if self.dispose:
- self.im.paste(self.dispose, self.dispose_extent)
- from copy import copy
- self.palette = copy(self.global_palette)
- info = {}
- frame_transparency = None
- interlace = None
- while True:
- s = self.fp.read(1)
- if not s or s == b";":
- break
- elif s == b"!":
- #
- # extensions
- #
- s = self.fp.read(1)
- block = self.data()
- if s[0] == 249:
- #
- # graphic control extension
- #
- flags = block[0]
- if flags & 1:
- frame_transparency = block[3]
- info["duration"] = i16(block, 1) * 10
- # disposal method - find the value of bits 4 - 6
- dispose_bits = 0b00011100 & flags
- dispose_bits = dispose_bits >> 2
- if dispose_bits:
- # only set the dispose if it is not
- # unspecified. I'm not sure if this is
- # correct, but it seems to prevent the last
- # frame from looking odd for some animations
- self.disposal_method = dispose_bits
- elif s[0] == 254:
- #
- # comment extension
- #
- while block:
- if "comment" in info:
- info["comment"] += block
- else:
- info["comment"] = block
- block = self.data()
- continue
- elif s[0] == 255:
- #
- # application extension
- #
- info["extension"] = block, self.fp.tell()
- if block[:11] == b"NETSCAPE2.0":
- block = self.data()
- if len(block) >= 3 and block[0] == 1:
- info["loop"] = i16(block, 1)
- while self.data():
- pass
- elif s == b",":
- #
- # local image
- #
- s = self.fp.read(9)
- # extent
- x0, y0 = i16(s, 0), i16(s, 2)
- x1, y1 = x0 + i16(s, 4), y0 + i16(s, 6)
- if x1 > self.size[0] or y1 > self.size[1]:
- self._size = max(x1, self.size[0]), max(y1, self.size[1])
- self.dispose_extent = x0, y0, x1, y1
- flags = s[8]
- interlace = (flags & 64) != 0
- if flags & 128:
- bits = (flags & 7) + 1
- self.palette = ImagePalette.raw("RGB", self.fp.read(3 << bits))
- # image data
- bits = self.fp.read(1)[0]
- self.__offset = self.fp.tell()
- break
- else:
- pass
- # raise OSError, "illegal GIF tag `%x`" % s[0]
- try:
- if self.disposal_method < 2:
- # do not dispose or none specified
- self.dispose = None
- elif self.disposal_method == 2:
- # replace with background colour
- # only dispose the extent in this frame
- x0, y0, x1, y1 = self.dispose_extent
- dispose_size = (x1 - x0, y1 - y0)
- Image._decompression_bomb_check(dispose_size)
- # by convention, attempt to use transparency first
- color = (
- frame_transparency
- if frame_transparency is not None
- else self.info.get("background", 0)
- )
- self.dispose = Image.core.fill("P", dispose_size, color)
- else:
- # replace with previous contents
- if self.im:
- # only dispose the extent in this frame
- self.dispose = self._crop(self.im, self.dispose_extent)
- elif frame_transparency is not None:
- x0, y0, x1, y1 = self.dispose_extent
- dispose_size = (x1 - x0, y1 - y0)
- Image._decompression_bomb_check(dispose_size)
- self.dispose = Image.core.fill(
- "P", dispose_size, frame_transparency
- )
- except AttributeError:
- pass
- if interlace is not None:
- transparency = -1
- if frame_transparency is not None:
- if frame == 0:
- self.info["transparency"] = frame_transparency
- else:
- transparency = frame_transparency
- self.tile = [
- (
- "gif",
- (x0, y0, x1, y1),
- self.__offset,
- (bits, interlace, transparency),
- )
- ]
- else:
- # self.__fp = None
- raise EOFError
- for k in ["duration", "comment", "extension", "loop"]:
- if k in info:
- self.info[k] = info[k]
- elif k in self.info:
- del self.info[k]
- self.mode = "L"
- if self.palette:
- self.mode = "P"
- def load_prepare(self):
- if not self.im and "transparency" in self.info:
- self.im = Image.core.fill(self.mode, self.size, self.info["transparency"])
- super(GifImageFile, self).load_prepare()
- def tell(self):
- return self.__frame
- def _close__fp(self):
- try:
- if self.__fp != self.fp:
- self.__fp.close()
- except AttributeError:
- pass
- finally:
- self.__fp = None
- # --------------------------------------------------------------------
- # Write GIF files
- RAWMODE = {"1": "L", "L": "L", "P": "P"}
- def _normalize_mode(im, initial_call=False):
- """
- Takes an image (or frame), returns an image in a mode that is appropriate
- for saving in a Gif.
- It may return the original image, or it may return an image converted to
- palette or 'L' mode.
- UNDONE: What is the point of mucking with the initial call palette, for
- an image that shouldn't have a palette, or it would be a mode 'P' and
- get returned in the RAWMODE clause.
- :param im: Image object
- :param initial_call: Default false, set to true for a single frame.
- :returns: Image object
- """
- if im.mode in RAWMODE:
- im.load()
- return im
- if Image.getmodebase(im.mode) == "RGB":
- if initial_call:
- palette_size = 256
- if im.palette:
- palette_size = len(im.palette.getdata()[1]) // 3
- return im.convert("P", palette=Image.ADAPTIVE, colors=palette_size)
- else:
- return im.convert("P")
- return im.convert("L")
- def _normalize_palette(im, palette, info):
- """
- Normalizes the palette for image.
- - Sets the palette to the incoming palette, if provided.
- - Ensures that there's a palette for L mode images
- - Optimizes the palette if necessary/desired.
- :param im: Image object
- :param palette: bytes object containing the source palette, or ....
- :param info: encoderinfo
- :returns: Image object
- """
- source_palette = None
- if palette:
- # a bytes palette
- if isinstance(palette, (bytes, bytearray, list)):
- source_palette = bytearray(palette[:768])
- if isinstance(palette, ImagePalette.ImagePalette):
- source_palette = bytearray(palette.palette)
- if im.mode == "P":
- if not source_palette:
- source_palette = im.im.getpalette("RGB")[:768]
- else: # L-mode
- if not source_palette:
- source_palette = bytearray(i // 3 for i in range(768))
- im.palette = ImagePalette.ImagePalette("RGB", palette=source_palette)
- used_palette_colors = _get_optimize(im, info)
- if used_palette_colors is not None:
- return im.remap_palette(used_palette_colors, source_palette)
- im.palette.palette = source_palette
- return im
- def _write_single_frame(im, fp, palette):
- im_out = _normalize_mode(im, True)
- for k, v in im_out.info.items():
- im.encoderinfo.setdefault(k, v)
- im_out = _normalize_palette(im_out, palette, im.encoderinfo)
- for s in _get_global_header(im_out, im.encoderinfo):
- fp.write(s)
- # local image header
- flags = 0
- if get_interlace(im):
- flags = flags | 64
- _write_local_header(fp, im, (0, 0), flags)
- im_out.encoderconfig = (8, get_interlace(im))
- ImageFile._save(im_out, fp, [("gif", (0, 0) + im.size, 0, RAWMODE[im_out.mode])])
- fp.write(b"\0") # end of image data
- def _write_multiple_frames(im, fp, palette):
- duration = im.encoderinfo.get("duration", im.info.get("duration"))
- disposal = im.encoderinfo.get("disposal", im.info.get("disposal"))
- im_frames = []
- frame_count = 0
- background_im = None
- for imSequence in itertools.chain([im], im.encoderinfo.get("append_images", [])):
- for im_frame in ImageSequence.Iterator(imSequence):
- # a copy is required here since seek can still mutate the image
- im_frame = _normalize_mode(im_frame.copy())
- if frame_count == 0:
- for k, v in im_frame.info.items():
- im.encoderinfo.setdefault(k, v)
- im_frame = _normalize_palette(im_frame, palette, im.encoderinfo)
- encoderinfo = im.encoderinfo.copy()
- if isinstance(duration, (list, tuple)):
- encoderinfo["duration"] = duration[frame_count]
- if isinstance(disposal, (list, tuple)):
- encoderinfo["disposal"] = disposal[frame_count]
- frame_count += 1
- if im_frames:
- # delta frame
- previous = im_frames[-1]
- if encoderinfo.get("disposal") == 2:
- if background_im is None:
- color = im.encoderinfo.get(
- "transparency", im.info.get("transparency", (0, 0, 0))
- )
- background = _get_background(im_frame, color)
- background_im = Image.new("P", im_frame.size, background)
- background_im.putpalette(im_frames[0]["im"].palette)
- base_im = background_im
- else:
- base_im = previous["im"]
- if _get_palette_bytes(im_frame) == _get_palette_bytes(base_im):
- delta = ImageChops.subtract_modulo(im_frame, base_im)
- else:
- delta = ImageChops.subtract_modulo(
- im_frame.convert("RGB"), base_im.convert("RGB")
- )
- bbox = delta.getbbox()
- if not bbox:
- # This frame is identical to the previous frame
- if duration:
- previous["encoderinfo"]["duration"] += encoderinfo["duration"]
- continue
- else:
- bbox = None
- im_frames.append({"im": im_frame, "bbox": bbox, "encoderinfo": encoderinfo})
- if len(im_frames) > 1:
- for frame_data in im_frames:
- im_frame = frame_data["im"]
- if not frame_data["bbox"]:
- # global header
- for s in _get_global_header(im_frame, frame_data["encoderinfo"]):
- fp.write(s)
- offset = (0, 0)
- else:
- # compress difference
- frame_data["encoderinfo"]["include_color_table"] = True
- im_frame = im_frame.crop(frame_data["bbox"])
- offset = frame_data["bbox"][:2]
- _write_frame_data(fp, im_frame, offset, frame_data["encoderinfo"])
- return True
- elif "duration" in im.encoderinfo and isinstance(
- im.encoderinfo["duration"], (list, tuple)
- ):
- # Since multiple frames will not be written, add together the frame durations
- im.encoderinfo["duration"] = sum(im.encoderinfo["duration"])
- def _save_all(im, fp, filename):
- _save(im, fp, filename, save_all=True)
- def _save(im, fp, filename, save_all=False):
- # header
- if "palette" in im.encoderinfo or "palette" in im.info:
- palette = im.encoderinfo.get("palette", im.info.get("palette"))
- else:
- palette = None
- im.encoderinfo["optimize"] = im.encoderinfo.get("optimize", True)
- if not save_all or not _write_multiple_frames(im, fp, palette):
- _write_single_frame(im, fp, palette)
- fp.write(b";") # end of file
- if hasattr(fp, "flush"):
- fp.flush()
- def get_interlace(im):
- interlace = im.encoderinfo.get("interlace", 1)
- # workaround for @PIL153
- if min(im.size) < 16:
- interlace = 0
- return interlace
- def _write_local_header(fp, im, offset, flags):
- transparent_color_exists = False
- try:
- transparency = im.encoderinfo["transparency"]
- except KeyError:
- pass
- else:
- transparency = int(transparency)
- # optimize the block away if transparent color is not used
- transparent_color_exists = True
- used_palette_colors = _get_optimize(im, im.encoderinfo)
- if used_palette_colors is not None:
- # adjust the transparency index after optimize
- try:
- transparency = used_palette_colors.index(transparency)
- except ValueError:
- transparent_color_exists = False
- if "duration" in im.encoderinfo:
- duration = int(im.encoderinfo["duration"] / 10)
- else:
- duration = 0
- disposal = int(im.encoderinfo.get("disposal", 0))
- if transparent_color_exists or duration != 0 or disposal:
- packed_flag = 1 if transparent_color_exists else 0
- packed_flag |= disposal << 2
- if not transparent_color_exists:
- transparency = 0
- fp.write(
- b"!"
- + o8(249) # extension intro
- + o8(4) # length
- + o8(packed_flag) # packed fields
- + o16(duration) # duration
- + o8(transparency) # transparency index
- + o8(0)
- )
- if "comment" in im.encoderinfo and 1 <= len(im.encoderinfo["comment"]):
- fp.write(b"!" + o8(254)) # extension intro
- comment = im.encoderinfo["comment"]
- if isinstance(comment, str):
- comment = comment.encode()
- for i in range(0, len(comment), 255):
- subblock = comment[i : i + 255]
- fp.write(o8(len(subblock)) + subblock)
- fp.write(o8(0))
- if "loop" in im.encoderinfo:
- number_of_loops = im.encoderinfo["loop"]
- fp.write(
- b"!"
- + o8(255) # extension intro
- + o8(11)
- + b"NETSCAPE2.0"
- + o8(3)
- + o8(1)
- + o16(number_of_loops) # number of loops
- + o8(0)
- )
- include_color_table = im.encoderinfo.get("include_color_table")
- if include_color_table:
- palette_bytes = _get_palette_bytes(im)
- color_table_size = _get_color_table_size(palette_bytes)
- if color_table_size:
- flags = flags | 128 # local color table flag
- flags = flags | color_table_size
- fp.write(
- b","
- + o16(offset[0]) # offset
- + o16(offset[1])
- + o16(im.size[0]) # size
- + o16(im.size[1])
- + o8(flags) # flags
- )
- if include_color_table and color_table_size:
- fp.write(_get_header_palette(palette_bytes))
- fp.write(o8(8)) # bits
- def _save_netpbm(im, fp, filename):
- # Unused by default.
- # To use, uncomment the register_save call at the end of the file.
- #
- # If you need real GIF compression and/or RGB quantization, you
- # can use the external NETPBM/PBMPLUS utilities. See comments
- # below for information on how to enable this.
- tempfile = im._dump()
- try:
- with open(filename, "wb") as f:
- if im.mode != "RGB":
- subprocess.check_call(
- ["ppmtogif", tempfile], stdout=f, stderr=subprocess.DEVNULL
- )
- else:
- # Pipe ppmquant output into ppmtogif
- # "ppmquant 256 %s | ppmtogif > %s" % (tempfile, filename)
- quant_cmd = ["ppmquant", "256", tempfile]
- togif_cmd = ["ppmtogif"]
- quant_proc = subprocess.Popen(
- quant_cmd, stdout=subprocess.PIPE, stderr=subprocess.DEVNULL
- )
- togif_proc = subprocess.Popen(
- togif_cmd,
- stdin=quant_proc.stdout,
- stdout=f,
- stderr=subprocess.DEVNULL,
- )
- # Allow ppmquant to receive SIGPIPE if ppmtogif exits
- quant_proc.stdout.close()
- retcode = quant_proc.wait()
- if retcode:
- raise subprocess.CalledProcessError(retcode, quant_cmd)
- retcode = togif_proc.wait()
- if retcode:
- raise subprocess.CalledProcessError(retcode, togif_cmd)
- finally:
- try:
- os.unlink(tempfile)
- except OSError:
- pass
- # Force optimization so that we can test performance against
- # cases where it took lots of memory and time previously.
- _FORCE_OPTIMIZE = False
- def _get_optimize(im, info):
- """
- Palette optimization is a potentially expensive operation.
- This function determines if the palette should be optimized using
- some heuristics, then returns the list of palette entries in use.
- :param im: Image object
- :param info: encoderinfo
- :returns: list of indexes of palette entries in use, or None
- """
- if im.mode in ("P", "L") and info and info.get("optimize", 0):
- # Potentially expensive operation.
- # The palette saves 3 bytes per color not used, but palette
- # lengths are restricted to 3*(2**N) bytes. Max saving would
- # be 768 -> 6 bytes if we went all the way down to 2 colors.
- # * If we're over 128 colors, we can't save any space.
- # * If there aren't any holes, it's not worth collapsing.
- # * If we have a 'large' image, the palette is in the noise.
- # create the new palette if not every color is used
- optimise = _FORCE_OPTIMIZE or im.mode == "L"
- if optimise or im.width * im.height < 512 * 512:
- # check which colors are used
- used_palette_colors = []
- for i, count in enumerate(im.histogram()):
- if count:
- used_palette_colors.append(i)
- if optimise or (
- len(used_palette_colors) <= 128
- and max(used_palette_colors) > len(used_palette_colors)
- ):
- return used_palette_colors
- def _get_color_table_size(palette_bytes):
- # calculate the palette size for the header
- if not palette_bytes:
- return 0
- elif len(palette_bytes) < 9:
- return 1
- else:
- return math.ceil(math.log(len(palette_bytes) // 3, 2)) - 1
- def _get_header_palette(palette_bytes):
- """
- Returns the palette, null padded to the next power of 2 (*3) bytes
- suitable for direct inclusion in the GIF header
- :param palette_bytes: Unpadded palette bytes, in RGBRGB form
- :returns: Null padded palette
- """
- color_table_size = _get_color_table_size(palette_bytes)
- # add the missing amount of bytes
- # the palette has to be 2<<n in size
- actual_target_size_diff = (2 << color_table_size) - len(palette_bytes) // 3
- if actual_target_size_diff > 0:
- palette_bytes += o8(0) * 3 * actual_target_size_diff
- return palette_bytes
- def _get_palette_bytes(im):
- """
- Gets the palette for inclusion in the gif header
- :param im: Image object
- :returns: Bytes, len<=768 suitable for inclusion in gif header
- """
- return im.palette.palette
- def _get_background(im, infoBackground):
- background = 0
- if infoBackground:
- background = infoBackground
- if isinstance(background, tuple):
- # WebPImagePlugin stores an RGBA value in info["background"]
- # So it must be converted to the same format as GifImagePlugin's
- # info["background"] - a global color table index
- try:
- background = im.palette.getcolor(background, im)
- except ValueError as e:
- if str(e) == "cannot allocate more than 256 colors":
- # If all 256 colors are in use,
- # then there is no need for the background color
- return 0
- else:
- raise
- return background
- def _get_global_header(im, info):
- """Return a list of strings representing a GIF header"""
- # Header Block
- # http://www.matthewflickinger.com/lab/whatsinagif/bits_and_bytes.asp
- version = b"87a"
- for extensionKey in ["transparency", "duration", "loop", "comment"]:
- if info and extensionKey in info:
- if (extensionKey == "duration" and info[extensionKey] == 0) or (
- extensionKey == "comment" and not (1 <= len(info[extensionKey]) <= 255)
- ):
- continue
- version = b"89a"
- break
- else:
- if im.info.get("version") == b"89a":
- version = b"89a"
- background = _get_background(im, info.get("background"))
- palette_bytes = _get_palette_bytes(im)
- color_table_size = _get_color_table_size(palette_bytes)
- return [
- b"GIF" # signature
- + version # version
- + o16(im.size[0]) # canvas width
- + o16(im.size[1]), # canvas height
- # Logical Screen Descriptor
- # size of global color table + global color table flag
- o8(color_table_size + 128), # packed fields
- # background + reserved/aspect
- o8(background) + o8(0),
- # Global Color Table
- _get_header_palette(palette_bytes),
- ]
- def _write_frame_data(fp, im_frame, offset, params):
- try:
- im_frame.encoderinfo = params
- # local image header
- _write_local_header(fp, im_frame, offset, 0)
- ImageFile._save(
- im_frame, fp, [("gif", (0, 0) + im_frame.size, 0, RAWMODE[im_frame.mode])]
- )
- fp.write(b"\0") # end of image data
- finally:
- del im_frame.encoderinfo
- # --------------------------------------------------------------------
- # Legacy GIF utilities
- def getheader(im, palette=None, info=None):
- """
- Legacy Method to get Gif data from image.
- Warning:: May modify image data.
- :param im: Image object
- :param palette: bytes object containing the source palette, or ....
- :param info: encoderinfo
- :returns: tuple of(list of header items, optimized palette)
- """
- used_palette_colors = _get_optimize(im, info)
- if info is None:
- info = {}
- if "background" not in info and "background" in im.info:
- info["background"] = im.info["background"]
- im_mod = _normalize_palette(im, palette, info)
- im.palette = im_mod.palette
- im.im = im_mod.im
- header = _get_global_header(im, info)
- return header, used_palette_colors
- # To specify duration, add the time in milliseconds to getdata(),
- # e.g. getdata(im_frame, duration=1000)
- def getdata(im, offset=(0, 0), **params):
- """
- Legacy Method
- Return a list of strings representing this image.
- The first string is a local image header, the rest contains
- encoded image data.
- :param im: Image object
- :param offset: Tuple of (x, y) pixels. Defaults to (0,0)
- :param \\**params: E.g. duration or other encoder info parameters
- :returns: List of Bytes containing gif encoded frame data
- """
- class Collector:
- data = []
- def write(self, data):
- self.data.append(data)
- im.load() # make sure raster data is available
- fp = Collector()
- _write_frame_data(fp, im, offset, params)
- return fp.data
- # --------------------------------------------------------------------
- # Registry
- Image.register_open(GifImageFile.format, GifImageFile, _accept)
- Image.register_save(GifImageFile.format, _save)
- Image.register_save_all(GifImageFile.format, _save_all)
- Image.register_extension(GifImageFile.format, ".gif")
- Image.register_mime(GifImageFile.format, "image/gif")
- #
- # Uncomment the following line if you wish to use NETPBM/PBMPLUS
- # instead of the built-in "uncompressed" GIF encoder
- # Image.register_save(GifImageFile.format, _save_netpbm)
|