198 lines
6.1 KiB
Python
198 lines
6.1 KiB
Python
|
#
|
||
|
# 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")
|