244 lines
9.4 KiB
Python
Executable File
244 lines
9.4 KiB
Python
Executable File
#!/usr/bin/python
|
|
|
|
from builtins import zip
|
|
|
|
import unittest
|
|
import math
|
|
|
|
#import logging
|
|
#logging.basicConfig(level = logging.DEBUG)
|
|
|
|
from gi.repository import Vips
|
|
|
|
unsigned_formats = [Vips.BandFormat.UCHAR,
|
|
Vips.BandFormat.USHORT,
|
|
Vips.BandFormat.UINT]
|
|
signed_formats = [Vips.BandFormat.CHAR,
|
|
Vips.BandFormat.SHORT,
|
|
Vips.BandFormat.INT]
|
|
float_formats = [Vips.BandFormat.FLOAT,
|
|
Vips.BandFormat.DOUBLE]
|
|
complex_formats = [Vips.BandFormat.COMPLEX,
|
|
Vips.BandFormat.DPCOMPLEX]
|
|
int_formats = unsigned_formats + signed_formats
|
|
noncomplex_formats = int_formats + float_formats
|
|
all_formats = int_formats + float_formats + complex_formats
|
|
|
|
colour_colourspaces = [Vips.Interpretation.XYZ,
|
|
Vips.Interpretation.LAB,
|
|
Vips.Interpretation.LCH,
|
|
Vips.Interpretation.CMC,
|
|
Vips.Interpretation.LABS,
|
|
Vips.Interpretation.SCRGB,
|
|
Vips.Interpretation.SRGB,
|
|
Vips.Interpretation.RGB16,
|
|
Vips.Interpretation.YXY]
|
|
coded_colourspaces = [Vips.Interpretation.LABQ]
|
|
mono_colourspaces = [Vips.Interpretation.GREY16,
|
|
Vips.Interpretation.B_W]
|
|
all_colourspaces = colour_colourspaces + mono_colourspaces + coded_colourspaces
|
|
|
|
# an expanding zip ... if either of the args is not a list, duplicate it down
|
|
# the other
|
|
def zip_expand(x, y):
|
|
if isinstance(x, list) and isinstance(y, list):
|
|
return list(zip(x, y))
|
|
elif isinstance(x, list):
|
|
return [[i, y] for i in x]
|
|
elif isinstance(y, list):
|
|
return [[x, j] for j in y]
|
|
else:
|
|
return [[x, y]]
|
|
|
|
# run a 1-ary function on a thing -- loop over elements if the
|
|
# thing is a list
|
|
def run_fn(fn, x):
|
|
if isinstance(x, list):
|
|
return [fn(i) for i in x]
|
|
else:
|
|
return fn(x)
|
|
|
|
# run a 2-ary function on two things -- loop over elements pairwise if the
|
|
# things are lists
|
|
def run_fn2(fn, x, y):
|
|
if isinstance(x, Vips.Image) or isinstance(y, Vips.Image):
|
|
return fn(x, y)
|
|
elif isinstance(x, list) or isinstance(y, list):
|
|
return [fn(i, j) for i, j in zip_expand(x, y)]
|
|
else:
|
|
return fn(x, y)
|
|
|
|
class TestColour(unittest.TestCase):
|
|
# test a pair of things which can be lists for approx. equality
|
|
def assertAlmostEqualObjects(self, a, b, places = 4, msg = ''):
|
|
#print 'assertAlmostEqualObjects %s = %s' % (a, b)
|
|
for x, y in zip_expand(a, b):
|
|
self.assertAlmostEqual(x, y, places = places, msg = msg)
|
|
|
|
# run a function on an image and on a single pixel, the results
|
|
# should match
|
|
def run_cmp(self, message, im, x, y, fn):
|
|
a = im.getpoint(x, y)
|
|
v1 = fn(a)
|
|
im2 = fn(im)
|
|
v2 = im2.getpoint(x, y)
|
|
self.assertAlmostEqualObjects(v1, v2, msg = message)
|
|
|
|
# run a function on a pair of images and on a pair of pixels, the results
|
|
# should match
|
|
def run_cmp2(self, message, left, right, x, y, fn):
|
|
a = left.getpoint(x, y)
|
|
b = right.getpoint(x, y)
|
|
v1 = fn(a, b)
|
|
after = fn(left, right)
|
|
v2 = after.getpoint(x, y)
|
|
self.assertAlmostEqualObjects(v1, v2, msg = message)
|
|
|
|
# run a function on a pair of images
|
|
# 50,50 and 10,10 should have different values on the test image
|
|
def run_test2(self, message, left, right, fn):
|
|
self.run_cmp2(message, left, right, 50, 50,
|
|
lambda x, y: run_fn2(fn, x, y))
|
|
self.run_cmp2(message, left, right, 10, 10,
|
|
lambda x, y: run_fn2(fn, x, y))
|
|
|
|
def setUp(self):
|
|
im = Vips.Image.mask_ideal(100, 100, 0.5, reject = True, optical = True)
|
|
self.colour = im * [1, 2, 3] + [2, 3, 4]
|
|
self.mono = self.colour.extract_band(1)
|
|
self.all_images = [self.mono, self.colour]
|
|
|
|
def test_colourspace(self):
|
|
# mid-grey in Lab ... put 42 in the extra band, it should be copied
|
|
# unmodified
|
|
test = Vips.Image.black(100, 100) + [50, 0, 0, 42]
|
|
test = test.copy(interpretation = Vips.Interpretation.LAB)
|
|
|
|
# a long series should come in a circle
|
|
im = test
|
|
for col in colour_colourspaces + [Vips.Interpretation.LAB]:
|
|
im = im.colourspace(col)
|
|
self.assertEqual(im.interpretation, col)
|
|
pixel = im.getpoint(10, 10)
|
|
self.assertAlmostEqual(pixel[3], 42, places = 2)
|
|
|
|
before = test.getpoint(10, 10)
|
|
after = im.getpoint(10, 10)
|
|
self.assertAlmostEqualObjects(before, after, places = 1)
|
|
|
|
# go between every pair of colour spaces
|
|
for start in colour_colourspaces:
|
|
for end in colour_colourspaces:
|
|
im = test.colourspace(start)
|
|
im2 = im.colourspace(end)
|
|
im3 = im2.colourspace(Vips.Interpretation.LAB)
|
|
|
|
before = test.getpoint(10, 10)
|
|
after = im3.getpoint(10, 10)
|
|
self.assertAlmostEqualObjects(before, after, places = 1)
|
|
|
|
# test Lab->XYZ on mid-grey
|
|
# checked against http://www.brucelindbloom.com
|
|
im = test.colourspace(Vips.Interpretation.XYZ)
|
|
after = im.getpoint(10, 10)
|
|
self.assertAlmostEqualObjects(after, [17.5064, 18.4187, 20.0547, 42])
|
|
|
|
# grey->colour->grey should be equal
|
|
for mono_fmt in mono_colourspaces:
|
|
test_grey = test.colourspace(mono_fmt)
|
|
im = test_grey
|
|
for col in colour_colourspaces + [mono_fmt]:
|
|
im = im.colourspace(col)
|
|
self.assertEqual(im.interpretation, col)
|
|
[before, alpha_before] = test_grey.getpoint(10, 10)
|
|
[after, alpha_after] = im.getpoint(10, 10)
|
|
self.assertLess(abs(alpha_after - alpha_before), 1)
|
|
if mono_fmt == Vips.Interpretation.GREY16:
|
|
# GREY16 can wind up rather different due to rounding
|
|
self.assertLess(abs(after - before), 30)
|
|
else:
|
|
# but 8-bit we should hit exactly
|
|
self.assertLess(abs(after - before), 1)
|
|
|
|
# test results from Bruce Lindbloom's calculator:
|
|
# http://www.brucelindbloom.com
|
|
|
|
def test_dE00(self):
|
|
# put 42 in the extra band, it should be copied unmodified
|
|
reference = Vips.Image.black(100, 100) + [50, 10, 20, 42]
|
|
reference = reference.copy(interpretation = Vips.Interpretation.LAB)
|
|
sample = Vips.Image.black(100, 100) + [40, -20, 10]
|
|
sample = sample.copy(interpretation = Vips.Interpretation.LAB)
|
|
|
|
difference = reference.dE00(sample)
|
|
result, alpha = difference.getpoint(10, 10)
|
|
self.assertAlmostEqual(result, 30.238, places = 3)
|
|
self.assertAlmostEqual(alpha, 42.0, places = 3)
|
|
|
|
def test_dE76(self):
|
|
# put 42 in the extra band, it should be copied unmodified
|
|
reference = Vips.Image.black(100, 100) + [50, 10, 20, 42]
|
|
reference = reference.copy(interpretation = Vips.Interpretation.LAB)
|
|
sample = Vips.Image.black(100, 100) + [40, -20, 10]
|
|
sample = sample.copy(interpretation = Vips.Interpretation.LAB)
|
|
|
|
difference = reference.dE76(sample)
|
|
result, alpha = difference.getpoint(10, 10)
|
|
self.assertAlmostEqual(result, 33.166, places = 3)
|
|
self.assertAlmostEqual(alpha, 42.0, places = 3)
|
|
|
|
# the vips CMC calculation is based on distance in a colorspace derived from
|
|
# the CMC formula, so it won't match exactly ... see vips_LCh2CMC() for
|
|
# details
|
|
def test_dECMC(self):
|
|
reference = Vips.Image.black(100, 100) + [50, 10, 20, 42]
|
|
reference = reference.copy(interpretation = Vips.Interpretation.LAB)
|
|
sample = Vips.Image.black(100, 100) + [55, 11, 23]
|
|
sample = sample.copy(interpretation = Vips.Interpretation.LAB)
|
|
|
|
difference = reference.dECMC(sample)
|
|
result, alpha = difference.getpoint(10, 10)
|
|
self.assertLess(abs(result - 4.97), 0.5)
|
|
self.assertAlmostEqual(alpha, 42.0, places = 3)
|
|
|
|
def test_icc(self):
|
|
test = Vips.Image.new_from_file("images/IMG_4618.jpg")
|
|
|
|
im = test.icc_import().icc_export()
|
|
self.assertLess(im.dE76(test).max(), 6)
|
|
|
|
im = test.icc_import()
|
|
im2 = im.icc_export(depth = 16)
|
|
self.assertEqual(im2.format, Vips.BandFormat.USHORT)
|
|
im3 = im2.icc_import()
|
|
self.assertLess((im - im3).abs().max(), 3)
|
|
|
|
im = test.icc_import(intent = Vips.Intent.ABSOLUTE)
|
|
im2 = im.icc_export(intent = Vips.Intent.ABSOLUTE)
|
|
self.assertLess(im2.dE76(test).max(), 6)
|
|
|
|
im = test.icc_import()
|
|
im2 = im.icc_export(output_profile = "images/sRGB.icm")
|
|
im3 = im.colourspace(Vips.Interpretation.SRGB)
|
|
self.assertLess(im2.dE76(im3).max(), 6)
|
|
|
|
before_profile = test.get_value("icc-profile-data")
|
|
im = test.icc_transform("images/sRGB.icm")
|
|
after_profile = im.get_value("icc-profile-data")
|
|
im2 = test.icc_import()
|
|
im3 = im2.colourspace(Vips.Interpretation.SRGB)
|
|
self.assertLess(im.dE76(im3).max(), 6)
|
|
self.assertNotEqual(len(before_profile), len(after_profile))
|
|
|
|
im = test.icc_import(input_profile = "images/sRGB.icm")
|
|
im2 = test.icc_import()
|
|
self.assertLess(6, im.dE76(im2).max())
|
|
|
|
im = test.icc_import(pcs = Vips.PCS.XYZ)
|
|
self.assertEqual(im.interpretation, Vips.Interpretation.XYZ)
|
|
im = test.icc_import()
|
|
self.assertEqual(im.interpretation, Vips.Interpretation.LAB)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|