|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197 |
- #
- # The Python Imaging Library.
- # $Id$
- #
- # MPO file handling
- #
- # See "Multi-Picture Format" (CIPA DC-007-Translation 2009, Standard of the
- # Camera & Imaging Products Association)
- #
- # The multi-picture object combines multiple JPEG images (with a modified EXIF
- # data format) into a single file. While it can theoretically be used much like
- # a GIF animation, it is commonly used to represent 3D photographs and is (as
- # of this writing) the most commonly used format by 3D cameras.
- #
- # History:
- # 2014-03-13 Feneric Created
- #
- # See the README file for information on usage and redistribution.
- #
-
- import itertools
- import os
- import struct
-
- from . import (
- ExifTags,
- Image,
- ImageFile,
- ImageSequence,
- JpegImagePlugin,
- TiffImagePlugin,
- )
- from ._binary import i16be as i16
- from ._binary import o32le
-
- # def _accept(prefix):
- # return JpegImagePlugin._accept(prefix)
-
-
- def _save(im, fp, filename):
- JpegImagePlugin._save(im, fp, filename)
-
-
- def _save_all(im, fp, filename):
- append_images = im.encoderinfo.get("append_images", [])
- if not append_images:
- try:
- animated = im.is_animated
- except AttributeError:
- animated = False
- if not animated:
- _save(im, fp, filename)
- return
-
- mpf_offset = 28
- offsets = []
- for imSequence in itertools.chain([im], append_images):
- for im_frame in ImageSequence.Iterator(imSequence):
- if not offsets:
- # APP2 marker
- im_frame.encoderinfo["extra"] = (
- b"\xFF\xE2" + struct.pack(">H", 6 + 82) + b"MPF\0" + b" " * 82
- )
- exif = im_frame.encoderinfo.get("exif")
- if isinstance(exif, Image.Exif):
- exif = exif.tobytes()
- im_frame.encoderinfo["exif"] = exif
- if exif:
- mpf_offset += 4 + len(exif)
-
- JpegImagePlugin._save(im_frame, fp, filename)
- offsets.append(fp.tell())
- else:
- im_frame.save(fp, "JPEG")
- offsets.append(fp.tell() - offsets[-1])
-
- ifd = TiffImagePlugin.ImageFileDirectory_v2()
- ifd[0xB000] = b"0100"
- ifd[0xB001] = len(offsets)
-
- mpentries = b""
- data_offset = 0
- for i, size in enumerate(offsets):
- if i == 0:
- mptype = 0x030000 # Baseline MP Primary Image
- else:
- mptype = 0x000000 # Undefined
- mpentries += struct.pack("<LLLHH", mptype, size, data_offset, 0, 0)
- if i == 0:
- data_offset -= mpf_offset
- data_offset += size
- ifd[0xB002] = mpentries
-
- fp.seek(mpf_offset)
- fp.write(b"II\x2A\x00" + o32le(8) + ifd.tobytes(8))
- fp.seek(0, os.SEEK_END)
-
-
- ##
- # Image plugin for MPO images.
-
-
- class MpoImageFile(JpegImagePlugin.JpegImageFile):
- format = "MPO"
- format_description = "MPO (CIPA DC-007)"
- _close_exclusive_fp_after_loading = False
-
- def _open(self):
- self.fp.seek(0) # prep the fp in order to pass the JPEG test
- JpegImagePlugin.JpegImageFile._open(self)
- self._after_jpeg_open()
-
- def _after_jpeg_open(self, mpheader=None):
- self._initial_size = self.size
- self.mpinfo = mpheader if mpheader is not None else self._getmp()
- self.n_frames = self.mpinfo[0xB001]
- self.__mpoffsets = [
- mpent["DataOffset"] + self.info["mpoffset"] for mpent in self.mpinfo[0xB002]
- ]
- self.__mpoffsets[0] = 0
- # Note that the following assertion will only be invalid if something
- # gets broken within JpegImagePlugin.
- assert self.n_frames == len(self.__mpoffsets)
- del self.info["mpoffset"] # no longer needed
- self.is_animated = self.n_frames > 1
- self._fp = self.fp # FIXME: hack
- self._fp.seek(self.__mpoffsets[0]) # get ready to read first frame
- self.__frame = 0
- self.offset = 0
- # for now we can only handle reading and individual frame extraction
- self.readonly = 1
-
- def load_seek(self, pos):
- self._fp.seek(pos)
-
- def seek(self, frame):
- if not self._seek_check(frame):
- return
- self.fp = self._fp
- self.offset = self.__mpoffsets[frame]
-
- self.fp.seek(self.offset + 2) # skip SOI marker
- segment = self.fp.read(2)
- if not segment:
- msg = "No data found for frame"
- raise ValueError(msg)
- self._size = self._initial_size
- if i16(segment) == 0xFFE1: # APP1
- n = i16(self.fp.read(2)) - 2
- self.info["exif"] = ImageFile._safe_read(self.fp, n)
- self._reload_exif()
-
- mptype = self.mpinfo[0xB002][frame]["Attribute"]["MPType"]
- if mptype.startswith("Large Thumbnail"):
- exif = self.getexif().get_ifd(ExifTags.IFD.Exif)
- if 40962 in exif and 40963 in exif:
- self._size = (exif[40962], exif[40963])
- elif "exif" in self.info:
- del self.info["exif"]
- self._reload_exif()
-
- self.tile = [("jpeg", (0, 0) + self.size, self.offset, (self.mode, ""))]
- self.__frame = frame
-
- def tell(self):
- return self.__frame
-
- @staticmethod
- def adopt(jpeg_instance, mpheader=None):
- """
- Transform the instance of JpegImageFile into
- an instance of MpoImageFile.
- After the call, the JpegImageFile is extended
- to be an MpoImageFile.
-
- This is essentially useful when opening a JPEG
- file that reveals itself as an MPO, to avoid
- double call to _open.
- """
- jpeg_instance.__class__ = MpoImageFile
- jpeg_instance._after_jpeg_open(mpheader)
- return jpeg_instance
-
-
- # ---------------------------------------------------------------------
- # Registry stuff
-
- # Note that since MPO shares a factory with JPEG, we do not need to do a
- # separate registration for it here.
- # Image.register_open(MpoImageFile.format,
- # JpegImagePlugin.jpeg_factory, _accept)
- Image.register_save(MpoImageFile.format, _save)
- Image.register_save_all(MpoImageFile.format, _save_all)
-
- Image.register_extension(MpoImageFile.format, ".mpo")
-
- Image.register_mime(MpoImageFile.format, "image/mpo")
|