mirror of
https://github.com/moparisthebest/SickRage
synced 2024-12-16 13:02:19 -05:00
212 lines
8.2 KiB
Python
212 lines
8.2 KiB
Python
|
"""
|
||
|
TIFF image parser.
|
||
|
|
||
|
Authors: Victor Stinner and Sebastien Ponce
|
||
|
Creation date: 30 september 2006
|
||
|
"""
|
||
|
|
||
|
from lib.hachoir_parser import Parser
|
||
|
from lib.hachoir_core.field import (FieldSet, SeekableFieldSet, ParserError, RootSeekableFieldSet,
|
||
|
UInt16, UInt32, Bytes, String)
|
||
|
from lib.hachoir_core.endian import LITTLE_ENDIAN, BIG_ENDIAN
|
||
|
from lib.hachoir_parser.image.exif import BasicIFDEntry
|
||
|
from lib.hachoir_core.tools import createDict
|
||
|
|
||
|
MAX_COUNT = 250
|
||
|
|
||
|
class IFDEntry(BasicIFDEntry):
|
||
|
static_size = 12*8
|
||
|
|
||
|
TAG_INFO = {
|
||
|
254: ("new_subfile_type", "New subfile type"),
|
||
|
255: ("subfile_type", "Subfile type"),
|
||
|
256: ("img_width", "Image width in pixels"),
|
||
|
257: ("img_height", "Image height in pixels"),
|
||
|
258: ("bits_per_sample", "Bits per sample"),
|
||
|
259: ("compression", "Compression method"),
|
||
|
262: ("photo_interpret", "Photometric interpretation"),
|
||
|
263: ("thres", "Thresholding"),
|
||
|
264: ("cell_width", "Cellule width"),
|
||
|
265: ("cell_height", "Cellule height"),
|
||
|
266: ("fill_order", "Fill order"),
|
||
|
269: ("doc_name", "Document name"),
|
||
|
270: ("description", "Image description"),
|
||
|
271: ("make", "Make"),
|
||
|
272: ("model", "Model"),
|
||
|
273: ("strip_ofs", "Strip offsets"),
|
||
|
274: ("orientation", "Orientation"),
|
||
|
277: ("sample_pixel", "Samples per pixel"),
|
||
|
278: ("row_per_strip", "Rows per strip"),
|
||
|
279: ("strip_byte", "Strip byte counts"),
|
||
|
280: ("min_sample_value", "Min sample value"),
|
||
|
281: ("max_sample_value", "Max sample value"),
|
||
|
282: ("xres", "X resolution"),
|
||
|
283: ("yres", "Y resolution"),
|
||
|
284: ("planar_conf", "Planar configuration"),
|
||
|
285: ("page_name", "Page name"),
|
||
|
286: ("xpos", "X position"),
|
||
|
287: ("ypos", "Y position"),
|
||
|
288: ("free_ofs", "Free offsets"),
|
||
|
289: ("free_byte", "Free byte counts"),
|
||
|
290: ("gray_resp_unit", "Gray response unit"),
|
||
|
291: ("gray_resp_curve", "Gray response curve"),
|
||
|
292: ("group3_opt", "Group 3 options"),
|
||
|
293: ("group4_opt", "Group 4 options"),
|
||
|
296: ("res_unit", "Resolution unit"),
|
||
|
297: ("page_nb", "Page number"),
|
||
|
301: ("color_respt_curve", "Color response curves"),
|
||
|
305: ("software", "Software"),
|
||
|
306: ("date_time", "Date time"),
|
||
|
315: ("artist", "Artist"),
|
||
|
316: ("host_computer", "Host computer"),
|
||
|
317: ("predicator", "Predicator"),
|
||
|
318: ("white_pt", "White point"),
|
||
|
319: ("prim_chomat", "Primary chromaticities"),
|
||
|
320: ("color_map", "Color map"),
|
||
|
321: ("half_tone_hints", "Halftone Hints"),
|
||
|
322: ("tile_width", "TileWidth"),
|
||
|
323: ("tile_length", "TileLength"),
|
||
|
324: ("tile_offsets", "TileOffsets"),
|
||
|
325: ("tile_byte_counts", "TileByteCounts"),
|
||
|
332: ("ink_set", "InkSet"),
|
||
|
333: ("ink_names", "InkNames"),
|
||
|
334: ("number_of_inks", "NumberOfInks"),
|
||
|
336: ("dot_range", "DotRange"),
|
||
|
337: ("target_printer", "TargetPrinter"),
|
||
|
338: ("extra_samples", "ExtraSamples"),
|
||
|
339: ("sample_format", "SampleFormat"),
|
||
|
340: ("smin_sample_value", "SMinSampleValue"),
|
||
|
341: ("smax_sample_value", "SMaxSampleValue"),
|
||
|
342: ("transfer_range", "TransferRange"),
|
||
|
512: ("jpeg_proc", "JPEGProc"),
|
||
|
513: ("jpeg_interchange_format", "JPEGInterchangeFormat"),
|
||
|
514: ("jpeg_interchange_format_length", "JPEGInterchangeFormatLength"),
|
||
|
515: ("jpeg_restart_interval", "JPEGRestartInterval"),
|
||
|
517: ("jpeg_lossless_predictors", "JPEGLosslessPredictors"),
|
||
|
518: ("jpeg_point_transforms", "JPEGPointTransforms"),
|
||
|
519: ("jpeg_qtables", "JPEGQTables"),
|
||
|
520: ("jpeg_dctables", "JPEGDCTables"),
|
||
|
521: ("jpeg_actables", "JPEGACTables"),
|
||
|
529: ("ycbcr_coefficients", "YCbCrCoefficients"),
|
||
|
530: ("ycbcr_subsampling", "YCbCrSubSampling"),
|
||
|
531: ("ycbcr_positioning", "YCbCrPositioning"),
|
||
|
532: ("reference_blackwhite", "ReferenceBlackWhite"),
|
||
|
33432: ("copyright", "Copyright"),
|
||
|
0x8769: ("ifd_pointer", "Pointer to next IFD entry"),
|
||
|
}
|
||
|
TAG_NAME = createDict(TAG_INFO, 0)
|
||
|
|
||
|
def __init__(self, *args):
|
||
|
FieldSet.__init__(self, *args)
|
||
|
tag = self["tag"].value
|
||
|
if tag in self.TAG_INFO:
|
||
|
self._name, self._description = self.TAG_INFO[tag]
|
||
|
else:
|
||
|
self._parser = None
|
||
|
|
||
|
class IFD(FieldSet):
|
||
|
def __init__(self, *args):
|
||
|
FieldSet.__init__(self, *args)
|
||
|
self._size = 16 + self["count"].value * IFDEntry.static_size
|
||
|
self._has_offset = False
|
||
|
|
||
|
def createFields(self):
|
||
|
yield UInt16(self, "count")
|
||
|
if MAX_COUNT < self["count"].value:
|
||
|
raise ParserError("TIFF IFD: Invalid count (%s)"
|
||
|
% self["count"].value)
|
||
|
for index in xrange(self["count"].value):
|
||
|
yield IFDEntry(self, "entry[]")
|
||
|
|
||
|
class ImageFile(SeekableFieldSet):
|
||
|
def __init__(self, parent, name, description, ifd):
|
||
|
SeekableFieldSet.__init__(self, parent, name, description, None)
|
||
|
self._has_offset = False
|
||
|
self._ifd = ifd
|
||
|
|
||
|
def createFields(self):
|
||
|
datas = {}
|
||
|
for entry in self._ifd:
|
||
|
if type(entry) != IFDEntry:
|
||
|
continue
|
||
|
for c in entry:
|
||
|
if c.name != "offset":
|
||
|
continue
|
||
|
self.seekByte(c.value, False)
|
||
|
desc = "data of ifd entry " + entry.name,
|
||
|
entryType = BasicIFDEntry.ENTRY_FORMAT[entry["type"].value]
|
||
|
count = entry["count"].value
|
||
|
if entryType == String:
|
||
|
yield String(self, entry.name, count, desc, "\0", "ISO-8859-1")
|
||
|
else:
|
||
|
d = Data(self, entry.name, desc, entryType, count)
|
||
|
datas[d.name] = d
|
||
|
yield d
|
||
|
break
|
||
|
# image data
|
||
|
if "strip_ofs" in datas and "strip_byte" in datas:
|
||
|
for i in xrange(datas["strip_byte"]._count):
|
||
|
self.seekByte(datas["strip_ofs"]["value["+str(i)+"]"].value, False)
|
||
|
yield Bytes(self, "strip[]", datas["strip_byte"]["value["+str(i)+"]"].value)
|
||
|
|
||
|
class Data(FieldSet):
|
||
|
|
||
|
def __init__(self, parent, name, desc, type, count):
|
||
|
size = type.static_size * count
|
||
|
FieldSet.__init__(self, parent, name, desc, size)
|
||
|
self._count = count
|
||
|
self._type = type
|
||
|
|
||
|
def createFields(self):
|
||
|
for i in xrange(self._count):
|
||
|
yield self._type(self, "value[]")
|
||
|
|
||
|
class TiffFile(RootSeekableFieldSet, Parser):
|
||
|
PARSER_TAGS = {
|
||
|
"id": "tiff",
|
||
|
"category": "image",
|
||
|
"file_ext": ("tif", "tiff"),
|
||
|
"mime": (u"image/tiff",),
|
||
|
"min_size": 8*8,
|
||
|
# TODO: Re-enable magic
|
||
|
"magic": (("II\x2A\0", 0), ("MM\0\x2A", 0)),
|
||
|
"description": "TIFF picture"
|
||
|
}
|
||
|
|
||
|
# Correct endian is set in constructor
|
||
|
endian = LITTLE_ENDIAN
|
||
|
|
||
|
def __init__(self, stream, **args):
|
||
|
RootSeekableFieldSet.__init__(self, None, "root", stream, None, stream.askSize(self))
|
||
|
if self.stream.readBytes(0, 2) == "MM":
|
||
|
self.endian = BIG_ENDIAN
|
||
|
Parser.__init__(self, stream, **args)
|
||
|
|
||
|
def validate(self):
|
||
|
endian = self.stream.readBytes(0, 2)
|
||
|
if endian not in ("MM", "II"):
|
||
|
return "Invalid endian (%r)" % endian
|
||
|
if self["version"].value != 42:
|
||
|
return "Unknown TIFF version"
|
||
|
return True
|
||
|
|
||
|
def createFields(self):
|
||
|
yield String(self, "endian", 2, 'Endian ("II" or "MM")', charset="ASCII")
|
||
|
yield UInt16(self, "version", "TIFF version number")
|
||
|
offset = UInt32(self, "img_dir_ofs[]", "Next image directory offset (in bytes from the beginning)")
|
||
|
yield offset
|
||
|
ifds = []
|
||
|
while True:
|
||
|
if offset.value == 0:
|
||
|
break
|
||
|
|
||
|
self.seekByte(offset.value, relative=False)
|
||
|
ifd = IFD(self, "ifd[]", "Image File Directory", None)
|
||
|
ifds.append(ifd)
|
||
|
yield ifd
|
||
|
offset = UInt32(self, "img_dir_ofs[]", "Next image directory offset (in bytes from the beginning)")
|
||
|
yield offset
|
||
|
for ifd in ifds:
|
||
|
image = ImageFile(self, "image[]", "Image File", ifd)
|
||
|
yield image
|