← Back
Editing: MpoImagePlugin.py
# # 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")
Save File
Cancel