100 lines
3 KiB
Python
100 lines
3 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.
|
||
|
|
#
|
||
|
|
|
||
|
|
from PIL import Image, JpegImagePlugin
|
||
|
|
|
||
|
|
__version__ = "0.1"
|
||
|
|
|
||
|
|
|
||
|
|
def _accept(prefix):
|
||
|
|
return JpegImagePlugin._accept(prefix)
|
||
|
|
|
||
|
|
|
||
|
|
def _save(im, fp, filename):
|
||
|
|
# Note that we can only save the current frame at present
|
||
|
|
return JpegImagePlugin._save(im, fp, filename)
|
||
|
|
|
||
|
|
|
||
|
|
##
|
||
|
|
# Image plugin for MPO images.
|
||
|
|
|
||
|
|
class MpoImageFile(JpegImagePlugin.JpegImageFile):
|
||
|
|
|
||
|
|
format = "MPO"
|
||
|
|
format_description = "MPO (CIPA DC-007)"
|
||
|
|
|
||
|
|
def _open(self):
|
||
|
|
self.fp.seek(0) # prep the fp in order to pass the JPEG test
|
||
|
|
JpegImagePlugin.JpegImageFile._open(self)
|
||
|
|
self.mpinfo = self._getmp()
|
||
|
|
self.__framecount = 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.__framecount == len(self.__mpoffsets)
|
||
|
|
del self.info['mpoffset'] # no longer needed
|
||
|
|
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)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def n_frames(self):
|
||
|
|
return self.__framecount
|
||
|
|
|
||
|
|
@property
|
||
|
|
def is_animated(self):
|
||
|
|
return self.__framecount > 1
|
||
|
|
|
||
|
|
def seek(self, frame):
|
||
|
|
if frame < 0 or frame >= self.__framecount:
|
||
|
|
raise EOFError("no more images in MPO file")
|
||
|
|
else:
|
||
|
|
self.fp = self.__fp
|
||
|
|
self.offset = self.__mpoffsets[frame]
|
||
|
|
self.tile = [
|
||
|
|
("jpeg", (0, 0) + self.size, self.offset, (self.mode, ""))
|
||
|
|
]
|
||
|
|
self.__frame = frame
|
||
|
|
|
||
|
|
def tell(self):
|
||
|
|
return self.__frame
|
||
|
|
|
||
|
|
|
||
|
|
# -------------------------------------------------------------------q-
|
||
|
|
# 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_extension(MpoImageFile.format, ".mpo")
|
||
|
|
|
||
|
|
Image.register_mime(MpoImageFile.format, "image/mpo")
|