mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-26 18:06:18 +03:00
93abbd0caa
To have the old API that always returns tuples, and fractions as pairs, set the `legacy_api` attribute of the IFD to True. This should alleviate concerns about backwards compatibility.
130 lines
4.6 KiB
Python
130 lines
4.6 KiB
Python
from __future__ import division
|
|
|
|
from helper import unittest, PillowTestCase, hopper
|
|
|
|
from PIL import Image, TiffImagePlugin, TiffTags
|
|
|
|
tag_ids = dict((info.name, info.value) for info in TiffTags.TAGS.values())
|
|
|
|
|
|
class TestFileTiffMetadata(PillowTestCase):
|
|
|
|
def test_rt_metadata(self):
|
|
""" Test writing arbitrary metadata into the tiff image directory
|
|
Use case is ImageJ private tags, one numeric, one arbitrary
|
|
data. https://github.com/python-pillow/Pillow/issues/291
|
|
"""
|
|
|
|
img = hopper()
|
|
|
|
basetextdata = "This is some arbitrary metadata for a text field"
|
|
textdata = basetextdata + " \xff"
|
|
floatdata = 12.345
|
|
doubledata = 67.89
|
|
info = TiffImagePlugin.ImageFileDirectory()
|
|
|
|
info[tag_ids['ImageJMetaDataByteCounts']] = len(textdata)
|
|
info[tag_ids['ImageJMetaData']] = textdata
|
|
info[tag_ids['RollAngle']] = floatdata
|
|
info.tagtype[tag_ids['RollAngle']] = 11
|
|
info[tag_ids['YawAngle']] = doubledata
|
|
info.tagtype[tag_ids['YawAngle']] = 12
|
|
|
|
f = self.tempfile("temp.tif")
|
|
|
|
img.save(f, tiffinfo=info)
|
|
|
|
for legacy_api in [False, True]:
|
|
loaded = Image.open(f)
|
|
loaded.tag.legacy_api = legacy_api
|
|
|
|
self.assertEqual(loaded.tag[50838],
|
|
(len(basetextdata + " ?"),) if legacy_api else len(basetextdata + " ?"))
|
|
self.assertEqual(loaded.tag[50839], basetextdata + " ?")
|
|
loaded_float = loaded.tag[tag_ids['RollAngle']]
|
|
if legacy_api:
|
|
loaded_float = loaded_float[0]
|
|
self.assertAlmostEqual(loaded_float, floatdata, places=5)
|
|
loaded_double = loaded.tag[tag_ids['YawAngle']]
|
|
if legacy_api:
|
|
loaded_double = loaded_double[0]
|
|
self.assertAlmostEqual(loaded_double, doubledata)
|
|
|
|
|
|
def test_read_metadata(self):
|
|
for legacy_api in [False, True]:
|
|
img = Image.open('Tests/images/hopper_g4.tif')
|
|
img.tag.legacy_api = legacy_api
|
|
|
|
known = {'YResolution': ((4294967295, 113653537),),
|
|
'PlanarConfiguration': (1,),
|
|
'BitsPerSample': (1,),
|
|
'ImageLength': (128,),
|
|
'Compression': (4,),
|
|
'FillOrder': (1,),
|
|
'RowsPerStrip': (128,),
|
|
'ResolutionUnit': (3,),
|
|
'PhotometricInterpretation': (0,),
|
|
'PageNumber': (0, 1),
|
|
'XResolution': ((4294967295, 113653537),),
|
|
'ImageWidth': (128,),
|
|
'Orientation': (1,),
|
|
'StripByteCounts': (1968,),
|
|
'SamplesPerPixel': (1,),
|
|
'StripOffsets': (8,)
|
|
} if legacy_api else {
|
|
'YResolution': 4294967295 / 113653537,
|
|
'PlanarConfiguration': 1,
|
|
'BitsPerSample': (1,),
|
|
'ImageLength': 128,
|
|
'Compression': 4,
|
|
'FillOrder': 1,
|
|
'RowsPerStrip': 128,
|
|
'ResolutionUnit': 3,
|
|
'PhotometricInterpretation': 0,
|
|
'PageNumber': (0, 1),
|
|
'XResolution': 4294967295 / 113653537,
|
|
'ImageWidth': 128,
|
|
'Orientation': 1,
|
|
'StripByteCounts': (1968,),
|
|
'SamplesPerPixel': 1,
|
|
'StripOffsets': (8,)
|
|
}
|
|
|
|
self.assertEqual(known, img.tag.named())
|
|
|
|
def test_write_metadata(self):
|
|
""" Test metadata writing through the python code """
|
|
img = Image.open('Tests/images/hopper.tif')
|
|
|
|
f = self.tempfile('temp.tiff')
|
|
img.save(f, tiffinfo=img.tag)
|
|
|
|
loaded = Image.open(f)
|
|
|
|
original = img.tag.named()
|
|
reloaded = loaded.tag.named()
|
|
|
|
ignored = [
|
|
'StripByteCounts', 'RowsPerStrip', 'PageNumber', 'StripOffsets']
|
|
|
|
for tag, value in reloaded.items():
|
|
if tag not in ignored:
|
|
self.assertEqual(
|
|
original[tag], value, "%s didn't roundtrip" % tag)
|
|
|
|
for tag, value in original.items():
|
|
if tag not in ignored:
|
|
self.assertEqual(
|
|
value, reloaded[tag], "%s didn't roundtrip" % tag)
|
|
|
|
def test_no_duplicate_50741_tag(self):
|
|
self.assertEqual(tag_ids['MakerNoteSafety'], 50741)
|
|
self.assertEqual(tag_ids['BestQualityScale'], 50780)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|
|
|
|
# End of file
|