+++ /dev/null
-# -*- coding: utf-8 -*-
-#######################################################################################
-# Photo is a part of Plinn - http://plinn.org #
-# Copyright (C) 2008 Benoît PIN <benoit.pin@ensmp.fr> #
-# #
-# This program is free software; you can redistribute it and/or #
-# modify it under the terms of the GNU General Public License #
-# as published by the Free Software Foundation; either version 2 #
-# of the License, or (at your option) any later version. #
-# #
-# This program is distributed in the hope that it will be useful, #
-# but WITHOUT ANY WARRANTY; without even the implied warranty of #
-# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #
-# GNU General Public License for more details. #
-# #
-# You should have received a copy of the GNU General Public License #
-# along with this program; if not, write to the Free Software #
-# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. #
-#######################################################################################
-""" Jpeg plugin for xmp read/write support.
-
-
-"""
-
-from xmp import XMP
-from types import StringType
-
-class JpegXmpIO(object):
-
- JPEG_XMP_LEADIN = 'http://ns.adobe.com/xap/1.0/\x00'
- JPEG_XMP_LEADIN_LENGTH = len(JPEG_XMP_LEADIN)
-
- MARKERS = {
- 0xFFC0: ("SOF0", "Baseline DCT", True),
- 0xFFC1: ("SOF1", "Extended Sequential DCT", True),
- 0xFFC2: ("SOF2", "Progressive DCT", True),
- 0xFFC3: ("SOF3", "Spatial lossless", True),
- 0xFFC4: ("DHT", "Define Huffman table", True),
- 0xFFC5: ("SOF5", "Differential sequential DCT", True),
- 0xFFC6: ("SOF6", "Differential progressive DCT", True),
- 0xFFC7: ("SOF7", "Differential spatial", True),
- 0xFFC8: ("JPG", "Extension", False),
- 0xFFC9: ("SOF9", "Extended sequential DCT (AC)", True),
- 0xFFCA: ("SOF10", "Progressive DCT (AC)", True),
- 0xFFCB: ("SOF11", "Spatial lossless DCT (AC)", True),
- 0xFFCC: ("DAC", "Define arithmetic coding conditioning", True),
- 0xFFCD: ("SOF13", "Differential sequential DCT (AC)", True),
- 0xFFCE: ("SOF14", "Differential progressive DCT (AC)", True),
- 0xFFCF: ("SOF15", "Differential spatial (AC)", True),
- 0xFFD0: ("RST0", "Restart 0", False),
- 0xFFD1: ("RST1", "Restart 1", False),
- 0xFFD2: ("RST2", "Restart 2", False),
- 0xFFD3: ("RST3", "Restart 3", False),
- 0xFFD4: ("RST4", "Restart 4", False),
- 0xFFD5: ("RST5", "Restart 5", False),
- 0xFFD6: ("RST6", "Restart 6", False),
- 0xFFD7: ("RST7", "Restart 7", False),
- 0xFFD8: ("SOI", "Start of image", False),
- 0xFFD9: ("EOI", "End of image", False),
- 0xFFDA: ("SOS", "Start of scan", True),
- 0xFFDB: ("DQT", "Define quantization table", True),
- 0xFFDC: ("DNL", "Define number of lines", True),
- 0xFFDD: ("DRI", "Define restart interval", True),
- 0xFFDE: ("DHP", "Define hierarchical progression", True),
- 0xFFDF: ("EXP", "Expand reference component", True),
- 0xFFE0: ("APP0", "Application segment 0", True),
- 0xFFE1: ("APP1", "Application segment 1", True),
- 0xFFE2: ("APP2", "Application segment 2", True),
- 0xFFE3: ("APP3", "Application segment 3", True),
- 0xFFE4: ("APP4", "Application segment 4", True),
- 0xFFE5: ("APP5", "Application segment 5", True),
- 0xFFE6: ("APP6", "Application segment 6", True),
- 0xFFE7: ("APP7", "Application segment 7", True),
- 0xFFE8: ("APP8", "Application segment 8", True),
- 0xFFE9: ("APP9", "Application segment 9", True),
- 0xFFEA: ("APP10", "Application segment 10", True),
- 0xFFEB: ("APP11", "Application segment 11", True),
- 0xFFEC: ("APP12", "Application segment 12", True),
- 0xFFED: ("APP13", "Application segment 13", True),
- 0xFFEE: ("APP14", "Application segment 14", True),
- 0xFFEF: ("APP15", "Application segment 15", True),
- 0xFFF0: ("JPG0", "Extension 0", False),
- 0xFFF1: ("JPG1", "Extension 1", False),
- 0xFFF2: ("JPG2", "Extension 2", False),
- 0xFFF3: ("JPG3", "Extension 3", False),
- 0xFFF4: ("JPG4", "Extension 4", False),
- 0xFFF5: ("JPG5", "Extension 5", False),
- 0xFFF6: ("JPG6", "Extension 6", False),
- 0xFFF7: ("JPG7", "Extension 7", False),
- 0xFFF8: ("JPG8", "Extension 8", False),
- 0xFFF9: ("JPG9", "Extension 9", False),
- 0xFFFA: ("JPG10", "Extension 10", False),
- 0xFFFB: ("JPG11", "Extension 11", False),
- 0xFFFC: ("JPG12", "Extension 12", False),
- 0xFFFD: ("JPG13", "Extension 13", False),
- 0xFFFE: ("COM", "Comment", True)
- }
-
-
- @staticmethod
- def i16(c,o=0):
- return ord(c[o+1]) + (ord(c[o])<<8)
-
- @staticmethod
- def getBlockInfo(marker, f):
- start = f.tell()
- length = JpegXmpIO.i16(f.read(2))
-
- markerInfo = JpegXmpIO.MARKERS[marker]
- blockInfo = { 'name' : markerInfo[0]
- , 'description' : markerInfo[1]
- , 'start' : start
- , 'length' : length}
-
- jump = start + length
- f.seek(jump)
-
- return blockInfo
-
- @staticmethod
- def getBlockInfos(f) :
- f.seek(0)
- s = f.read(1)
-
- blockInfos = []
-
- while 1:
- s = s + f.read(1)
- i = JpegXmpIO.i16(s)
-
- if JpegXmpIO.MARKERS.has_key(i):
- name, desciption, handle = JpegXmpIO.MARKERS[i]
-
- if handle:
- blockInfo = JpegXmpIO.getBlockInfo(i, f)
- blockInfos.append(blockInfo)
- if i == 0xFFDA: # start of scan
- break
- s = f.read(1)
- elif i == 0 or i == 65535:
- # padded marker or junk; move on
- s = "\xff"
-
- return blockInfos
-
-
- @staticmethod
- def genJpegXmpBlock(uXmpData, paddingSize=2) :
- block = u''
-
- block += JpegXmpIO.JPEG_XMP_LEADIN
- block += XMP.genXMPPacket(uXmpData, paddingSize)
- # utf-8 mandatory in jpeg files (xmp specification)
- block = block.encode('utf-8')
-
- length = len(block) + 2
-
- # TODO : reduce padding size if this assertion occurs
- assert length <= 0xfffd, "Jpeg block too long: %d (max: 0xfffd)" % hex(length)
-
- chrlength = chr(length >> 8 & 0xff) + chr(length & 0xff)
-
- block = chrlength + block
-
- return block
-
-
-
- @staticmethod
- def read(f) :
-
- blockInfos = JpegXmpIO.getBlockInfos(f)
- app1BlockInfos = [b for b in blockInfos if b['name'] == 'APP1']
-
- xmpBlocks = []
-
- for info in app1BlockInfos :
- f.seek(info['start'])
- data = f.read(info['length'])[2:]
- if data.startswith(JpegXmpIO.JPEG_XMP_LEADIN) :
- xmpBlocks.append(data)
-
- assert len(xmpBlocks) <= 1, "Multiple xmp block data is not yet supported."
-
- if len(xmpBlocks) == 1 :
- data = xmpBlocks[0]
- packet = data[len(JpegXmpIO.JPEG_XMP_LEADIN):]
- return packet
- else :
- return None
-
- @staticmethod
- def write(original, new, uxmp) :
-
- blockInfos = JpegXmpIO.getBlockInfos(original)
- app1BlockInfos = [b for b in blockInfos if b['name'] == 'APP1']
-
- xmpBlockInfos = []
-
- for info in app1BlockInfos :
- original.seek(info['start'])
- lead = original.read(JpegXmpIO.JPEG_XMP_LEADIN_LENGTH+2)[2:]
- if lead == JpegXmpIO.JPEG_XMP_LEADIN :
- xmpBlockInfos.append(info)
-
-
- assert len(xmpBlockInfos) <= 1, "Multiple xmp block data is not yet supported."
-
- if isinstance(uxmp, StringType) :
- uxmp = unicode(uxmp, 'utf-8')
-
- if len(xmpBlockInfos) == 0 :
- blockInfo = [b for b in blockInfos if b['name'] == 'APP13']
-
- if not blockInfo :
- blockInfo = [b for b in blockInfos if b['name'] == 'APP1']
-
- if not blockInfo :
- blockInfo = [b for b in blockInfos if b['name'] == 'APP0']
-
- if not blockInfo : raise ValueError, "No suitable place to write xmp segment"
-
- info = blockInfo[0]
- print 'create xmp after: %s' % info['name']
-
- original.seek(0)
- before = original.read(info['start'] + info['length'])
- after = original.read()
-
- jpegBlock = '\xFF\xE1' + JpegXmpIO.genJpegXmpBlock(uxmp)
-
- else :
- info = xmpBlockInfos[0]
-
- original.seek(0)
- before = original.read(info['start'])
-
- original.seek(info['start'] + info['length'])
- after = original.read()
-
- jpegBlock = JpegXmpIO.genJpegXmpBlock(uxmp)
-
- new.seek(0)
- new.write(before)
- new.write(jpegBlock)
- new.write(after)
-
- # if original == new :
- # new.seek(0)
- # else :
- # new.close()
- # original.close()
-
-
-XMP.registerReader('image/jpeg', JpegXmpIO.read)
-XMP.registerWriter('image/jpeg', JpegXmpIO.write)
projects / Photo.git / blobdiff
