mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-25 17:36:18 +03:00
338 lines
14 KiB
Python
338 lines
14 KiB
Python
from helper import unittest, PillowTestCase, hopper
|
|
import datetime
|
|
|
|
from PIL import Image
|
|
|
|
from io import BytesIO
|
|
import os
|
|
|
|
try:
|
|
from PIL import ImageCms
|
|
from PIL.ImageCms import ImageCmsProfile
|
|
ImageCms.core.profile_open
|
|
except ImportError as v:
|
|
# Skipped via setUp()
|
|
pass
|
|
|
|
|
|
SRGB = "Tests/icc/sRGB_IEC61966-2-1_black_scaled.icc"
|
|
HAVE_PROFILE = os.path.exists(SRGB)
|
|
|
|
|
|
class TestImageCms(PillowTestCase):
|
|
|
|
def setUp(self):
|
|
try:
|
|
from PIL import ImageCms
|
|
# need to hit getattr to trigger the delayed import error
|
|
ImageCms.core.profile_open
|
|
except ImportError as v:
|
|
self.skipTest(v)
|
|
|
|
def skip_missing(self):
|
|
if not HAVE_PROFILE:
|
|
self.skipTest("SRGB profile not available")
|
|
|
|
def test_sanity(self):
|
|
|
|
# basic smoke test.
|
|
# this mostly follows the cms_test outline.
|
|
|
|
v = ImageCms.versions() # should return four strings
|
|
self.assertEqual(v[0], '1.0.0 pil')
|
|
self.assertEqual(list(map(type, v)), [str, str, str, str])
|
|
|
|
# internal version number
|
|
self.assertRegexpMatches(ImageCms.core.littlecms_version, r"\d+\.\d+$")
|
|
|
|
self.skip_missing()
|
|
i = ImageCms.profileToProfile(hopper(), SRGB, SRGB)
|
|
self.assert_image(i, "RGB", (128, 128))
|
|
|
|
i = hopper()
|
|
ImageCms.profileToProfile(i, SRGB, SRGB, inPlace=True)
|
|
self.assert_image(i, "RGB", (128, 128))
|
|
|
|
t = ImageCms.buildTransform(SRGB, SRGB, "RGB", "RGB")
|
|
i = ImageCms.applyTransform(hopper(), t)
|
|
self.assert_image(i, "RGB", (128, 128))
|
|
|
|
i = hopper()
|
|
t = ImageCms.buildTransform(SRGB, SRGB, "RGB", "RGB")
|
|
ImageCms.applyTransform(hopper(), t, inPlace=True)
|
|
self.assert_image(i, "RGB", (128, 128))
|
|
|
|
p = ImageCms.createProfile("sRGB")
|
|
o = ImageCms.getOpenProfile(SRGB)
|
|
t = ImageCms.buildTransformFromOpenProfiles(p, o, "RGB", "RGB")
|
|
i = ImageCms.applyTransform(hopper(), t)
|
|
self.assert_image(i, "RGB", (128, 128))
|
|
|
|
t = ImageCms.buildProofTransform(SRGB, SRGB, SRGB, "RGB", "RGB")
|
|
self.assertEqual(t.inputMode, "RGB")
|
|
self.assertEqual(t.outputMode, "RGB")
|
|
i = ImageCms.applyTransform(hopper(), t)
|
|
self.assert_image(i, "RGB", (128, 128))
|
|
|
|
# test PointTransform convenience API
|
|
hopper().point(t)
|
|
|
|
def test_name(self):
|
|
self.skip_missing()
|
|
# get profile information for file
|
|
self.assertEqual(
|
|
ImageCms.getProfileName(SRGB).strip(),
|
|
'IEC 61966-2-1 Default RGB Colour Space - sRGB')
|
|
|
|
def test_info(self):
|
|
self.skip_missing()
|
|
self.assertEqual(
|
|
ImageCms.getProfileInfo(SRGB).splitlines(), [
|
|
'sRGB IEC61966-2-1 black scaled', '',
|
|
'Copyright International Color Consortium, 2009', ''])
|
|
|
|
def test_copyright(self):
|
|
self.skip_missing()
|
|
self.assertEqual(
|
|
ImageCms.getProfileCopyright(SRGB).strip(),
|
|
'Copyright International Color Consortium, 2009')
|
|
|
|
def test_manufacturer(self):
|
|
self.skip_missing()
|
|
self.assertEqual(
|
|
ImageCms.getProfileManufacturer(SRGB).strip(),
|
|
'')
|
|
|
|
def test_model(self):
|
|
self.skip_missing()
|
|
self.assertEqual(
|
|
ImageCms.getProfileModel(SRGB).strip(),
|
|
'IEC 61966-2-1 Default RGB Colour Space - sRGB')
|
|
|
|
def test_description(self):
|
|
self.skip_missing()
|
|
self.assertEqual(
|
|
ImageCms.getProfileDescription(SRGB).strip(),
|
|
'sRGB IEC61966-2-1 black scaled')
|
|
|
|
def test_intent(self):
|
|
self.skip_missing()
|
|
self.assertEqual(ImageCms.getDefaultIntent(SRGB), 0)
|
|
self.assertEqual(ImageCms.isIntentSupported(
|
|
SRGB, ImageCms.INTENT_ABSOLUTE_COLORIMETRIC,
|
|
ImageCms.DIRECTION_INPUT), 1)
|
|
|
|
def test_profile_object(self):
|
|
# same, using profile object
|
|
p = ImageCms.createProfile("sRGB")
|
|
# self.assertEqual(ImageCms.getProfileName(p).strip(),
|
|
# 'sRGB built-in - (lcms internal)')
|
|
# self.assertEqual(ImageCms.getProfileInfo(p).splitlines(),
|
|
# ['sRGB built-in', '', 'WhitePoint : D65 (daylight)', '', ''])
|
|
self.assertEqual(ImageCms.getDefaultIntent(p), 0)
|
|
self.assertEqual(ImageCms.isIntentSupported(
|
|
p, ImageCms.INTENT_ABSOLUTE_COLORIMETRIC,
|
|
ImageCms.DIRECTION_INPUT), 1)
|
|
|
|
def test_extensions(self):
|
|
# extensions
|
|
|
|
i = Image.open("Tests/images/rgb.jpg")
|
|
p = ImageCms.getOpenProfile(BytesIO(i.info["icc_profile"]))
|
|
self.assertEqual(
|
|
ImageCms.getProfileName(p).strip(),
|
|
'IEC 61966-2.1 Default RGB colour space - sRGB')
|
|
|
|
def test_exceptions(self):
|
|
# the procedural pyCMS API uses PyCMSError for all sorts of errors
|
|
self.assertRaises(
|
|
ImageCms.PyCMSError,
|
|
lambda: ImageCms.profileToProfile(hopper(), "foo", "bar"))
|
|
self.assertRaises(
|
|
ImageCms.PyCMSError,
|
|
lambda: ImageCms.buildTransform("foo", "bar", "RGB", "RGB"))
|
|
self.assertRaises(
|
|
ImageCms.PyCMSError,
|
|
lambda: ImageCms.getProfileName(None))
|
|
self.skip_missing()
|
|
self.assertRaises(
|
|
ImageCms.PyCMSError,
|
|
lambda: ImageCms.isIntentSupported(SRGB, None, None))
|
|
|
|
def test_display_profile(self):
|
|
# try fetching the profile for the current display device
|
|
ImageCms.get_display_profile()
|
|
|
|
def test_lab_color_profile(self):
|
|
ImageCms.createProfile("LAB", 5000)
|
|
ImageCms.createProfile("LAB", 6500)
|
|
|
|
def test_simple_lab(self):
|
|
i = Image.new('RGB', (10, 10), (128, 128, 128))
|
|
|
|
psRGB = ImageCms.createProfile("sRGB")
|
|
pLab = ImageCms.createProfile("LAB")
|
|
t = ImageCms.buildTransform(psRGB, pLab, "RGB", "LAB")
|
|
|
|
i_lab = ImageCms.applyTransform(i, t)
|
|
|
|
self.assertEqual(i_lab.mode, 'LAB')
|
|
|
|
k = i_lab.getpixel((0, 0))
|
|
# not a linear luminance map. so L != 128:
|
|
self.assertEqual(k, (137, 128, 128))
|
|
|
|
l = i_lab.getdata(0)
|
|
a = i_lab.getdata(1)
|
|
b = i_lab.getdata(2)
|
|
|
|
self.assertEqual(list(l), [137] * 100)
|
|
self.assertEqual(list(a), [128] * 100)
|
|
self.assertEqual(list(b), [128] * 100)
|
|
|
|
def test_lab_color(self):
|
|
psRGB = ImageCms.createProfile("sRGB")
|
|
pLab = ImageCms.createProfile("LAB")
|
|
t = ImageCms.buildTransform(psRGB, pLab, "RGB", "LAB")
|
|
|
|
# Need to add a type mapping for some PIL type to TYPE_Lab_8 in
|
|
# findLCMSType, and have that mapping work back to a PIL mode
|
|
# (likely RGB).
|
|
i = ImageCms.applyTransform(hopper(), t)
|
|
self.assert_image(i, "LAB", (128, 128))
|
|
|
|
# i.save('temp.lab.tif') # visually verified vs PS.
|
|
|
|
target = Image.open('Tests/images/hopper.Lab.tif')
|
|
|
|
self.assert_image_similar(i, target, 30)
|
|
|
|
def test_lab_srgb(self):
|
|
psRGB = ImageCms.createProfile("sRGB")
|
|
pLab = ImageCms.createProfile("LAB")
|
|
t = ImageCms.buildTransform(pLab, psRGB, "LAB", "RGB")
|
|
|
|
img = Image.open('Tests/images/hopper.Lab.tif')
|
|
|
|
img_srgb = ImageCms.applyTransform(img, t)
|
|
|
|
# img_srgb.save('temp.srgb.tif') # visually verified vs ps.
|
|
|
|
self.assert_image_similar(hopper(), img_srgb, 30)
|
|
self.assertTrue(img_srgb.info['icc_profile'])
|
|
|
|
profile = ImageCmsProfile(BytesIO(img_srgb.info['icc_profile']))
|
|
self.assertTrue('sRGB' in ImageCms.getProfileDescription(profile))
|
|
|
|
def test_lab_roundtrip(self):
|
|
# check to see if we're at least internally consistent.
|
|
psRGB = ImageCms.createProfile("sRGB")
|
|
pLab = ImageCms.createProfile("LAB")
|
|
t = ImageCms.buildTransform(psRGB, pLab, "RGB", "LAB")
|
|
|
|
t2 = ImageCms.buildTransform(pLab, psRGB, "LAB", "RGB")
|
|
|
|
i = ImageCms.applyTransform(hopper(), t)
|
|
|
|
self.assertEqual(i.info['icc_profile'],
|
|
ImageCmsProfile(pLab).tobytes())
|
|
|
|
out = ImageCms.applyTransform(i, t2)
|
|
|
|
self.assert_image_similar(hopper(), out, 2)
|
|
|
|
def test_profile_tobytes(self):
|
|
i = Image.open("Tests/images/rgb.jpg")
|
|
p = ImageCms.getOpenProfile(BytesIO(i.info["icc_profile"]))
|
|
|
|
p2 = ImageCms.getOpenProfile(BytesIO(p.tobytes()))
|
|
|
|
# not the same bytes as the original icc_profile,
|
|
# but it does roundtrip
|
|
self.assertEqual(p.tobytes(), p2.tobytes())
|
|
self.assertEqual(ImageCms.getProfileName(p),
|
|
ImageCms.getProfileName(p2))
|
|
self.assertEqual(ImageCms.getProfileDescription(p),
|
|
ImageCms.getProfileDescription(p2))
|
|
|
|
def test_extended_information(self):
|
|
self.skip_missing()
|
|
o = ImageCms.getOpenProfile(SRGB)
|
|
p = o.profile
|
|
|
|
def assert_truncated_tuple_equal(tup1, tup2, digits=10):
|
|
# Helper function to reduce precision of tuples of floats
|
|
# recursively and then check equality.
|
|
power = 10 ** digits
|
|
|
|
def truncate_tuple(tuple_or_float):
|
|
return tuple(
|
|
truncate_tuple(val) if isinstance(val, tuple)
|
|
else int(val * power) / power for val in tuple_or_float)
|
|
self.assertEqual(truncate_tuple(tup1), truncate_tuple(tup2))
|
|
|
|
self.assertEqual(p.attributes, 4294967296)
|
|
assert_truncated_tuple_equal(p.blue_colorant, ((0.14306640625, 0.06060791015625, 0.7140960693359375), (0.1558847490315394, 0.06603820639433387, 0.06060791015625)))
|
|
assert_truncated_tuple_equal(p.blue_primary, ((0.14306641366715667, 0.06060790921083026, 0.7140960805782015), (0.15588475410450106, 0.06603820408959558, 0.06060790921083026)))
|
|
assert_truncated_tuple_equal(p.chromatic_adaptation, (((1.04791259765625, 0.0229339599609375, -0.050201416015625), (0.02960205078125, 0.9904632568359375, -0.0170745849609375), (-0.009246826171875, 0.0150604248046875, 0.7517852783203125)), ((1.0267159024652783, 0.022470062342089134, 0.0229339599609375), (0.02951378324103937, 0.9875098886387147, 0.9904632568359375), (-0.012205438066465256, 0.01987915407854985, 0.0150604248046875))))
|
|
self.assertEqual(p.chromaticity, None)
|
|
self.assertEqual(p.clut, {0: (False, False, True), 1: (False, False, True), 2: (False, False, True), 3: (False, False, True)})
|
|
self.assertEqual(p.color_space, 'RGB')
|
|
self.assertEqual(p.colorant_table, None)
|
|
self.assertEqual(p.colorant_table_out, None)
|
|
self.assertEqual(p.colorimetric_intent, None)
|
|
self.assertEqual(p.connection_space, 'XYZ ')
|
|
self.assertEqual(p.copyright, 'Copyright International Color Consortium, 2009')
|
|
self.assertEqual(p.creation_date, datetime.datetime(2009, 2, 27, 21, 36, 31))
|
|
self.assertEqual(p.device_class, 'mntr')
|
|
assert_truncated_tuple_equal(p.green_colorant, ((0.3851470947265625, 0.7168731689453125, 0.097076416015625), (0.32119769927720654, 0.5978443449048152, 0.7168731689453125)))
|
|
assert_truncated_tuple_equal(p.green_primary, ((0.3851470888162112, 0.7168731974161346, 0.09707641738998518), (0.32119768793686687, 0.5978443567149709, 0.7168731974161346)))
|
|
self.assertEqual(p.header_flags, 0)
|
|
self.assertEqual(p.header_manufacturer, '\x00\x00\x00\x00')
|
|
self.assertEqual(p.header_model, '\x00\x00\x00\x00')
|
|
self.assertEqual(p.icc_measurement_condition, {'backing': (0.0, 0.0, 0.0), 'flare': 0.0, 'geo': 'unknown', 'observer': 1, 'illuminant_type': 'D65'})
|
|
self.assertEqual(p.icc_version, 33554432)
|
|
self.assertEqual(p.icc_viewing_condition, None)
|
|
self.assertEqual(p.intent_supported, {0: (True, True, True), 1: (True, True, True), 2: (True, True, True), 3: (True, True, True)})
|
|
self.assertEqual(p.is_matrix_shaper, True)
|
|
self.assertEqual(p.luminance, ((0.0, 80.0, 0.0), (0.0, 1.0, 80.0)))
|
|
self.assertEqual(p.manufacturer, None)
|
|
assert_truncated_tuple_equal(p.media_black_point, ((0.012054443359375, 0.0124969482421875, 0.01031494140625), (0.34573304157549234, 0.35842450765864337, 0.0124969482421875)))
|
|
assert_truncated_tuple_equal(p.media_white_point, ((0.964202880859375, 1.0, 0.8249053955078125), (0.3457029219802284, 0.3585375327567059, 1.0)))
|
|
assert_truncated_tuple_equal((p.media_white_point_temperature,), (5000.722328847392,))
|
|
self.assertEqual(p.model, 'IEC 61966-2-1 Default RGB Colour Space - sRGB')
|
|
self.assertEqual(p.pcs, 'XYZ')
|
|
self.assertEqual(p.perceptual_rendering_intent_gamut, None)
|
|
self.assertEqual(p.product_copyright, 'Copyright International Color Consortium, 2009')
|
|
self.assertEqual(p.product_desc, 'sRGB IEC61966-2-1 black scaled')
|
|
self.assertEqual(p.product_description, 'sRGB IEC61966-2-1 black scaled')
|
|
self.assertEqual(p.product_manufacturer, '')
|
|
self.assertEqual(p.product_model, 'IEC 61966-2-1 Default RGB Colour Space - sRGB')
|
|
self.assertEqual(p.profile_description, 'sRGB IEC61966-2-1 black scaled')
|
|
self.assertEqual(p.profile_id, b')\xf8=\xde\xaf\xf2U\xaexB\xfa\xe4\xca\x839\r')
|
|
assert_truncated_tuple_equal(p.red_colorant, ((0.436065673828125, 0.2224884033203125, 0.013916015625), (0.6484536316398539, 0.3308524880306778, 0.2224884033203125)))
|
|
assert_truncated_tuple_equal(p.red_primary, ((0.43606566581047446, 0.22248840582960838, 0.013916015621759925), (0.6484536250319214, 0.3308524944738204, 0.22248840582960838)))
|
|
self.assertEqual(p.rendering_intent, 0)
|
|
self.assertEqual(p.saturation_rendering_intent_gamut, None)
|
|
self.assertEqual(p.screening_description, None)
|
|
self.assertEqual(p.target, None)
|
|
self.assertEqual(p.technology, 'CRT ')
|
|
self.assertEqual(p.version, 2.0)
|
|
self.assertEqual(p.viewing_condition, 'Reference Viewing Condition in IEC 61966-2-1')
|
|
self.assertEqual(p.xcolor_space, 'RGB ')
|
|
|
|
def test_profile_typesafety(self):
|
|
""" Profile init type safety
|
|
|
|
prepatch, these would segfault, postpatch they should emit a typeerror
|
|
"""
|
|
|
|
with self.assertRaises(TypeError):
|
|
ImageCms.ImageCmsProfile(0).tobytes()
|
|
with self.assertRaises(TypeError):
|
|
ImageCms.ImageCmsProfile(1).tobytes()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|