Source code for colour.io.luts.iridas_cube

"""
Iridas .cube LUT Format Input / Output Utilities
================================================

Defines the *Iridas* *.cube* *LUT* format related input / output utilities
objects:

-   :func:`colour.io.read_LUT_IridasCube`
-   :func:`colour.io.write_LUT_IridasCube`

References
----------
-   :cite:`AdobeSystems2013b` : Adobe Systems. (2013). Cube LUT Specification.
    https://drive.google.com/open?id=143Eh08ZYncCAMwJ1q4gWxVOqR_OSWYvs
"""

from __future__ import annotations

import numpy as np

from colour.io.luts import LUT1D, LUT3D, LUT3x1D, LUTSequence
from colour.io.luts.common import path_to_title
from colour.utilities import (
    as_float_array,
    as_int_scalar,
    attest,
    format_array_as_row,
    usage_warning,
)

__author__ = "Colour Developers"
__copyright__ = "Copyright 2013 Colour Developers"
__license__ = "BSD-3-Clause - https://opensource.org/licenses/BSD-3-Clause"
__maintainer__ = "Colour Developers"
__email__ = "colour-developers@colour-science.org"
__status__ = "Production"

__all__ = [
    "read_LUT_IridasCube",
    "write_LUT_IridasCube",
]


[docs] def read_LUT_IridasCube(path: str) -> LUT3x1D | LUT3D: """ Read given *Iridas* *.cube* *LUT* file. Parameters ---------- path *LUT* path. Returns ------- :class:`LUT3x1D` or :class:`LUT3D`. :class:`LUT3x1D` or :class:`LUT3D` class instance. References ---------- :cite:`AdobeSystems2013b` Examples -------- Reading a 3x1D *Iridas* *.cube* *LUT*: >>> import os >>> path = os.path.join( ... os.path.dirname(__file__), ... "tests", ... "resources", ... "iridas_cube", ... "ACES_Proxy_10_to_ACES.cube", ... ) >>> print(read_LUT_IridasCube(path)) LUT3x1D - ACES Proxy 10 to ACES ------------------------------- <BLANKLINE> Dimensions : 2 Domain : [[ 0. 0. 0.] [ 1. 1. 1.]] Size : (32, 3) Reading a 3D *Iridas* *.cube* *LUT*: >>> path = os.path.join( ... os.path.dirname(__file__), ... "tests", ... "resources", ... "iridas_cube", ... "Colour_Correct.cube", ... ) >>> print(read_LUT_IridasCube(path)) LUT3D - Generated by Foundry::LUT --------------------------------- <BLANKLINE> Dimensions : 3 Domain : [[ 0. 0. 0.] [ 1. 1. 1.]] Size : (4, 4, 4, 3) Reading a 3D *Iridas* *.cube* *LUT* with comments: >>> path = os.path.join( ... os.path.dirname(__file__), ... "tests", ... "resources", ... "iridas_cube", ... "Demo.cube", ... ) >>> print(read_LUT_IridasCube(path)) LUT3x1D - Demo -------------- <BLANKLINE> Dimensions : 2 Domain : [[ 0. 0. 0.] [ 1. 2. 3.]] Size : (3, 3) Comment 01 : Comments can go anywhere """ title = path_to_title(path) domain_min, domain_max = np.array([0, 0, 0]), np.array([1, 1, 1]) dimensions: int = 3 size: int = 2 data = [] comments = [] with open(path) as cube_file: lines = cube_file.readlines() for line in lines: line = line.strip() # noqa: PLW2901 if len(line) == 0: continue if line.startswith("#"): comments.append(line[1:].strip()) continue tokens = line.split() if tokens[0] == "TITLE": title = " ".join(tokens[1:])[1:-1] elif tokens[0] == "DOMAIN_MIN": domain_min = as_float_array(tokens[1:]) elif tokens[0] == "DOMAIN_MAX": domain_max = as_float_array(tokens[1:]) elif tokens[0] == "LUT_1D_SIZE": dimensions = 2 size = as_int_scalar(tokens[1]) elif tokens[0] == "LUT_3D_SIZE": dimensions = 3 size = as_int_scalar(tokens[1]) else: data.append(tokens) table = as_float_array(data) LUT: LUT3x1D | LUT3D if dimensions == 2: LUT = LUT3x1D( table, title, np.vstack([domain_min, domain_max]), comments=comments, ) elif dimensions == 3: # The lines of table data shall be in ascending index order, # with the first component index (Red) changing most rapidly, # and the last component index (Blue) changing least rapidly. table = table.reshape([size, size, size, 3], order="F") LUT = LUT3D( table, title, np.vstack([domain_min, domain_max]), comments=comments, ) return LUT
[docs] def write_LUT_IridasCube( LUT: LUT3x1D | LUT3D | LUTSequence, path: str, decimals: int = 7 ) -> bool: """ Write given *LUT* to given *Iridas* *.cube* *LUT* file. Parameters ---------- LUT :class:`LUT3x1D`, :class:`LUT3D` or :class:`LUTSequence` class instance to write at given path. path *LUT* path. decimals Formatting decimals. Returns ------- :class:`bool` Definition success. Warnings -------- - If a :class:`LUTSequence` class instance is passed as ``LUT``, the first *LUT* in the *LUT* sequence will be used. References ---------- :cite:`AdobeSystems2013b` Examples -------- Writing a 3x1D *Iridas* *.cube* *LUT*: >>> from colour.algebra import spow >>> domain = np.array([[-0.1, -0.2, -0.4], [1.5, 3.0, 6.0]]) >>> LUT = LUT3x1D( ... spow(LUT3x1D.linear_table(16, domain), 1 / 2.2), ... "My LUT", ... domain, ... comments=["A first comment.", "A second comment."], ... ) >>> write_LUT_IridasCube(LUTxD, "My_LUT.cube") # doctest: +SKIP Writing a 3D *Iridas* *.cube* *LUT*: >>> domain = np.array([[-0.1, -0.2, -0.4], [1.5, 3.0, 6.0]]) >>> LUT = LUT3D( ... spow(LUT3D.linear_table(16, domain), 1 / 2.2), ... "My LUT", ... np.array([[-0.1, -0.2, -0.4], [1.5, 3.0, 6.0]]), ... comments=["A first comment.", "A second comment."], ... ) >>> write_LUT_IridasCube(LUTxD, "My_LUT.cube") # doctest: +SKIP """ if isinstance(LUT, LUTSequence): usage_warning( f'"LUT" is a "LUTSequence" instance was passed, ' f'using first sequence "LUT":\n{LUT}' ) LUTxD = LUT[0] elif isinstance(LUT, LUT1D): LUTxD = LUT.convert(LUT3x1D) else: LUTxD = LUT attest( isinstance(LUTxD, (LUT3x1D, LUT3D)), '"LUT" must be a 1D, 3x1D or 3D "LUT"!', ) attest(not LUTxD.is_domain_explicit(), '"LUT" domain must be implicit!') is_3x1D = isinstance(LUTxD, LUT3x1D) size = LUTxD.size if is_3x1D: attest(2 <= size <= 65536, '"LUT" size must be in domain [2, 65536]!') else: attest(2 <= size <= 256, '"LUT" size must be in domain [2, 256]!') with open(path, "w") as cube_file: cube_file.write(f'TITLE "{LUTxD.name}"\n') if LUTxD.comments: for comment in LUTxD.comments: cube_file.write(f"# {comment}\n") cube_file.write( f"{'LUT_1D_SIZE' if is_3x1D else 'LUT_3D_SIZE'} {LUTxD.table.shape[0]}\n" ) default_domain = np.array([[0, 0, 0], [1, 1, 1]]) if not np.array_equal(LUTxD.domain, default_domain): cube_file.write( f"DOMAIN_MIN {format_array_as_row(LUTxD.domain[0], decimals)}\n" ) cube_file.write( f"DOMAIN_MAX {format_array_as_row(LUTxD.domain[1], decimals)}\n" ) table = ( LUTxD.table.reshape([-1, 3], order="F") if not is_3x1D else LUTxD.table ) for array in table: cube_file.write(f"{format_array_as_row(array, decimals)}\n") return True