Mei ChuandPatrick Hodoul
c4a4b2bcd6
* Update Lut1DTransformTest and Lut3DTransformTest - Conform those tests with TransformsBaseTest. Signed-off-by: Mei Chu <meimchu@gmail.com> * -Lut3DTransformTest python bug fix. Signed-off-by: Mei Chu <meimchu@gmail.com> * - Adding Bernard Lefebvre to the COMMITTERS page! Signed-off-by: Mei Chu <meimchu@gmail.com> * - Fixed due to misunderstanding. Signed-off-by: Mei Chu <meimchu@gmail.com> Co-authored-by: Patrick Hodoul <patrick.hodoul@autodesk.com>
210 lines
7.3 KiB
Python
210 lines
7.3 KiB
Python
# SPDX-License-Identifier: BSD-3-Clause
|
|
# Copyright Contributors to the OpenColorIO Project.
|
|
|
|
import logging
|
|
import unittest
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
try:
|
|
import numpy as np
|
|
except ImportError:
|
|
logger.warning(
|
|
"NumPy could not be imported. "
|
|
"Test case will lack significant coverage!"
|
|
)
|
|
np = None
|
|
|
|
import PyOpenColorIO as OCIO
|
|
from TransformsBaseTest import TransformsBaseTest
|
|
|
|
|
|
class Lut1DTransformTest(unittest.TestCase, TransformsBaseTest):
|
|
|
|
def setUp(self):
|
|
self.tr = OCIO.Lut1DTransform()
|
|
|
|
def test_default_constructor(self):
|
|
"""
|
|
Test the default constructor.
|
|
"""
|
|
|
|
self.assertEqual(self.tr.getLength(), 2)
|
|
self.assertEqual(self.tr.getDirection(), OCIO.TRANSFORM_DIR_FORWARD)
|
|
self.assertEqual(self.tr.getHueAdjust(), OCIO.HUE_NONE)
|
|
self.assertFalse(self.tr.getInputHalfDomain())
|
|
self.assertFalse(self.tr.getOutputRawHalfs())
|
|
self.assertEqual(self.tr.getInterpolation(), OCIO.INTERP_DEFAULT)
|
|
self.assertEqual(self.tr.getFileOutputBitDepth(),
|
|
OCIO.BIT_DEPTH_UNKNOWN)
|
|
r, g, b = self.tr.getValue(0)
|
|
self.assertEqual([r, g, b], [0, 0, 0])
|
|
r, g, b = self.tr.getValue(1)
|
|
self.assertEqual([r, g, b], [1, 1, 1])
|
|
|
|
def test_format_metadata(self):
|
|
"""
|
|
Test the getFormatMetadata() method.
|
|
"""
|
|
|
|
format_metadata = self.tr.getFormatMetadata()
|
|
self.assertIsInstance(format_metadata, OCIO.FormatMetadata)
|
|
self.assertEqual(format_metadata.getElementName(), 'ROOT')
|
|
self.assertEqual(format_metadata.getName(), '')
|
|
self.assertEqual(format_metadata.getID(), '')
|
|
format_metadata.setName('name')
|
|
format_metadata.setID('id')
|
|
self.assertEqual(format_metadata.getName(), 'name')
|
|
self.assertEqual(format_metadata.getID(), 'id')
|
|
|
|
def test_file_output_bit_depth(self):
|
|
"""
|
|
Test the setFileOutputBitDepth() and getFileOutputBitDepth() methods.
|
|
"""
|
|
|
|
self.assertEqual(self.tr.getFileOutputBitDepth(),
|
|
OCIO.BIT_DEPTH_UNKNOWN)
|
|
self.tr.setFileOutputBitDepth(OCIO.BIT_DEPTH_UINT10)
|
|
self.assertEqual(self.tr.getFileOutputBitDepth(),
|
|
OCIO.BIT_DEPTH_UINT10)
|
|
|
|
def test_hue_adjust(self):
|
|
"""
|
|
Test the setHueAdjust() and getHueAdjust() methods.
|
|
"""
|
|
|
|
self.assertEqual(self.tr.getHueAdjust(), OCIO.HUE_NONE)
|
|
self.tr.setHueAdjust(OCIO.HUE_DW3)
|
|
self.assertEqual(self.tr.getHueAdjust(), OCIO.HUE_DW3)
|
|
with self.assertRaises(OCIO.Exception):
|
|
self.tr.setHueAdjust(OCIO.HUE_WYPN)
|
|
|
|
def test_input_half_domain(self):
|
|
"""
|
|
Test the setInputHalfDomain() and getInputHalfDomain() methods.
|
|
"""
|
|
|
|
self.assertFalse(self.tr.getInputHalfDomain())
|
|
self.tr.setInputHalfDomain(True)
|
|
self.assertTrue(self.tr.getInputHalfDomain())
|
|
|
|
def test_output_raw_halfs(self):
|
|
"""
|
|
Test the setOutputRawHalfs() and getOutputRawHalfs() methods.
|
|
"""
|
|
|
|
self.assertFalse(self.tr.getOutputRawHalfs())
|
|
self.tr.setOutputRawHalfs(True)
|
|
self.assertTrue(self.tr.getOutputRawHalfs())
|
|
|
|
def test_length(self):
|
|
"""
|
|
Test the setLength() and getLength() methods.
|
|
"""
|
|
|
|
self.assertEqual(self.tr.getLength(), 2)
|
|
self.tr.setValue(0, 0.1, 0.2, 0.3)
|
|
self.tr.setLength(3)
|
|
self.assertEqual(self.tr.getLength(), 3)
|
|
# Changing the length reset LUT values to identity.
|
|
r, g, b = self.tr.getValue(0)
|
|
self.assertEqual([r, g, b], [0, 0, 0])
|
|
|
|
def test_constructor_with_keywords(self):
|
|
"""
|
|
Test Lut1DTransform constructor with keywords and validate its values.
|
|
"""
|
|
|
|
lut = OCIO.Lut1DTransform(
|
|
length=65536,
|
|
inputHalfDomain=True,
|
|
outputRawHalfs=True,
|
|
fileOutputBitDepth=OCIO.BIT_DEPTH_UINT10,
|
|
hueAdjust=OCIO.HUE_DW3,
|
|
interpolation=OCIO.INTERP_BEST,
|
|
direction=OCIO.TRANSFORM_DIR_INVERSE)
|
|
self.assertEqual(lut.getLength(), 65536)
|
|
self.assertEqual(lut.getDirection(), OCIO.TRANSFORM_DIR_INVERSE)
|
|
self.assertEqual(lut.getHueAdjust(), OCIO.HUE_DW3)
|
|
self.assertTrue(lut.getInputHalfDomain())
|
|
self.assertTrue(lut.getOutputRawHalfs())
|
|
self.assertEqual(lut.getInterpolation(), OCIO.INTERP_BEST)
|
|
self.assertEqual(lut.getFileOutputBitDepth(), OCIO.BIT_DEPTH_UINT10)
|
|
|
|
lut = OCIO.Lut1DTransform(
|
|
length=4,
|
|
direction=OCIO.TRANSFORM_DIR_INVERSE)
|
|
self.assertEqual(lut.getLength(), 4)
|
|
self.assertEqual(lut.getDirection(), OCIO.TRANSFORM_DIR_INVERSE)
|
|
self.assertEqual(lut.getHueAdjust(), OCIO.HUE_NONE)
|
|
self.assertFalse(lut.getInputHalfDomain())
|
|
self.assertFalse(lut.getOutputRawHalfs())
|
|
self.assertEqual(lut.getInterpolation(), OCIO.INTERP_DEFAULT)
|
|
self.assertEqual(lut.getFileOutputBitDepth(), OCIO.BIT_DEPTH_UNKNOWN)
|
|
|
|
def test_constructor_with_positional(self):
|
|
"""
|
|
Test Lut1DTransform constructor without keywords and validate its values.
|
|
"""
|
|
|
|
lut = OCIO.Lut1DTransform(65536, True, True, OCIO.BIT_DEPTH_UINT10,
|
|
OCIO.HUE_DW3, OCIO.INTERP_BEST,
|
|
OCIO.TRANSFORM_DIR_INVERSE)
|
|
self.assertEqual(lut.getLength(), 65536)
|
|
self.assertEqual(lut.getDirection(), OCIO.TRANSFORM_DIR_INVERSE)
|
|
self.assertEqual(lut.getHueAdjust(), OCIO.HUE_DW3)
|
|
self.assertTrue(lut.getInputHalfDomain())
|
|
self.assertTrue(lut.getOutputRawHalfs())
|
|
self.assertEqual(lut.getInterpolation(), OCIO.INTERP_BEST)
|
|
self.assertEqual(lut.getFileOutputBitDepth(), OCIO.BIT_DEPTH_UINT10)
|
|
|
|
def test_array(self):
|
|
"""
|
|
Test the setValue() and getValue() methods.
|
|
"""
|
|
|
|
self.tr = OCIO.Lut1DTransform(length=3)
|
|
r, g, b = self.tr.getValue(0)
|
|
self.assertEqual([r, g, b], [0, 0, 0])
|
|
r, g, b = self.tr.getValue(1)
|
|
self.assertEqual([r, g, b], [0.5, 0.5, 0.5])
|
|
r, g, b = self.tr.getValue(2)
|
|
self.assertEqual([r, g, b], [1, 1, 1])
|
|
|
|
self.tr.setValue(0, 0.1, 0.2, 0.3)
|
|
r, g, b = self.tr.getValue(0)
|
|
# Values are stored as float.
|
|
self.assertAlmostEqual(r, 0.1, delta=1e-6)
|
|
self.assertAlmostEqual(g, 0.2, delta=1e-6)
|
|
self.assertAlmostEqual(b, 0.3, delta=1e-6)
|
|
|
|
if not np:
|
|
logger.warning("NumPy not found. Skipping part of test!")
|
|
return
|
|
|
|
data = self.tr.getData()
|
|
expected = np.array([0.1, 0.2, 0.3,
|
|
0.5, 0.5, 0.5,
|
|
1., 1., 1.]).astype(np.float32)
|
|
self.assertEqual(data.all(), expected.all())
|
|
data[6] = 0.9
|
|
data[7] = 1.1
|
|
data[8] = 1.2
|
|
self.tr.setData(data)
|
|
|
|
r, g, b = self.tr.getValue(2)
|
|
self.assertAlmostEqual(r, 0.9, delta=1e-6)
|
|
self.assertAlmostEqual(g, 1.1, delta=1e-6)
|
|
self.assertAlmostEqual(b, 1.2, delta=1e-6)
|
|
|
|
def test_equals(self):
|
|
"""
|
|
Test equals.
|
|
"""
|
|
|
|
lut = OCIO.Lut1DTransform()
|
|
lut2 = OCIO.Lut1DTransform()
|
|
self.assertTrue(lut.equals(lut2))
|
|
lut.setValue(0, 0.1, 0.2, 0.3)
|
|
self.assertFalse(lut.equals(lut2))
|