+++ /dev/null
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
-
-"""
-Implements support for general colorspaces conversions and transfer functions.
-"""
-
-from __future__ import division
-
-import array
-import os
-
-import PyOpenColorIO as ocio
-
-import aces_ocio.generate_lut as genlut
-from aces_ocio.colorspaces import aces
-from aces_ocio.utilities import ColorSpace, mat44_from_mat33
-
-__author__ = 'ACES Developers'
-__copyright__ = 'Copyright (C) 2014 - 2015 - ACES Developers'
-__license__ = ''
-__maintainer__ = 'ACES Developers'
-__email__ = 'aces@oscars.org'
-__status__ = 'Production'
-
-__all__ = ['create_matrix_colorspace',
- 'create_transfer_colorspace',
- 'create_matrix_plus_transfer_colorspace',
- 'transfer_function_sRGB_to_linear',
- 'transfer_function_Rec709_to_linear',
- 'transfer_function_Rec2020_10bit_to_linear',
- 'transfer_function_Rec2020_12bit_to_linear',
- 'transfer_function_Rec1886_to_linear',
- 'create_colorspaces',
- 'create_raw']
-
-
-# -------------------------------------------------------------------------
-# *Matrix Transform*
-# -------------------------------------------------------------------------
-def create_matrix_colorspace(name='matrix',
- from_reference_values=None,
- to_reference_values=None,
- aliases=None):
- """
- Creates a ColorSpace that only uses *Matrix Transforms*
-
- Parameters
- ----------
- name : str, optional
- Aliases for this colorspace
- from_reference_values : list of matrices
- List of matrices to convert from the reference colorspace to this space
- to_reference_values : list of matrices
- List of matrices to convert to the reference colorspace from this space
- aliases : list of str, optional
- Aliases for this colorspace
-
- Returns
- -------
- ColorSpace
- A *Matrix Transform*-based ColorSpace
- """
-
- if from_reference_values is None:
- from_reference_values = []
-
- if to_reference_values is None:
- to_reference_values = []
-
- if aliases is None:
- aliases = []
-
- cs = ColorSpace(name)
- cs.description = 'The %s color space' % name
- cs.aliases = aliases
- cs.equality_group = name
- cs.family = 'Utility'
- cs.is_data = False
-
- # A linear space needs allocation variables.
- cs.allocation_type = ocio.Constants.ALLOCATION_UNIFORM
- cs.allocation_vars = [0, 1]
-
- cs.to_reference_transforms = []
- if to_reference_values:
- for matrix in to_reference_values:
- cs.to_reference_transforms.append({
- 'type': 'matrix',
- 'matrix': mat44_from_mat33(matrix),
- 'direction': 'forward'})
-
- cs.from_reference_transforms = []
- if from_reference_values:
- for matrix in from_reference_values:
- cs.from_reference_transforms.append({
- 'type': 'matrix',
- 'matrix': mat44_from_mat33(matrix),
- 'direction': 'forward'})
-
- return cs
-
-
-# -------------------------------------------------------------------------
-# *Transfer Function Transform*
-# -------------------------------------------------------------------------
-def create_transfer_colorspace(name='transfer',
- transfer_function_name='transfer_function',
- transfer_function=lambda x: x,
- lut_directory='/tmp',
- lut_resolution_1d=1024,
- aliases=None):
- """
- Creates a ColorSpace that only uses transfer functions encoded as 1D LUTs
-
- Parameters
- ----------
- name : str, optional
- Aliases for this colorspace
- transfer_function_name : str, optional
- The name of the transfer function
- transfer_function : function, optional
- The transfer function to be evaluated
- lut_directory : str or unicode
- The directory to use when generating LUTs
- lut_resolution_1d : int
- The resolution of generated 1D LUTs
- aliases : list of str
- Aliases for this colorspace
-
- Returns
- -------
- ColorSpace
- A *LUT1D Transform*-based ColorSpace representing a transfer function
- """
-
- if aliases is None:
- aliases = []
-
- cs = ColorSpace(name)
- cs.description = 'The %s color space' % name
- cs.aliases = aliases
- cs.equality_group = name
- cs.family = 'Utility'
- cs.is_data = False
-
- # A linear space needs allocation variables.
- cs.allocation_type = ocio.Constants.ALLOCATION_UNIFORM
- cs.allocation_vars = [0, 1]
-
- # Sampling the transfer function.
- data = array.array('f', '\0' * lut_resolution_1d * 4)
- for c in range(lut_resolution_1d):
- data[c] = transfer_function(c / (lut_resolution_1d - 1))
-
- # Writing the sampled data to a *LUT*.
- lut = '%s_to_linear.spi1d' % transfer_function_name
- genlut.write_SPI_1d(
- os.path.join(lut_directory, lut),
- 0,
- 1,
- data,
- lut_resolution_1d,
- 1)
-
- # Creating the *to_reference* transforms.
- cs.to_reference_transforms = []
- cs.to_reference_transforms.append({
- 'type': 'lutFile',
- 'path': lut,
- 'interpolation': 'linear',
- 'direction': 'forward'})
-
- # Creating the *from_reference* transforms.
- cs.from_reference_transforms = []
-
- return cs
-
-
-# -------------------------------------------------------------------------
-# *Transfer Function + Matrix Transform*
-# -------------------------------------------------------------------------
-def create_matrix_plus_transfer_colorspace(
- name='matrix_plus_transfer',
- transfer_function_name='transfer_function',
- transfer_function=lambda x: x,
- lut_directory='/tmp',
- lut_resolution_1d=1024,
- from_reference_values=None,
- to_reference_values=None,
- aliases=None):
- """
- Creates a ColorSpace that uses transfer functions encoded as 1D LUTs and
- matrice
-
- Parameters
- ----------
- name : str, optional
- Aliases for this colorspace
- transfer_function_name : str, optional
- The name of the transfer function
- transfer_function : function, optional
- The transfer function to be evaluated
- lut_directory : str or unicode
- The directory to use when generating LUTs
- lut_resolution_1d : int
- The resolution of generated 1D LUTs
- from_reference_values : list of matrices
- List of matrices to convert from the reference colorspace to this space
- to_reference_values : list of matrices
- List of matrices to convert to the reference colorspace from this space
- aliases : list of str
- Aliases for this colorspace
-
- Returns
- -------
- ColorSpace
- A *Matrx and LUT1D Transform*-based ColorSpace representing a transfer
- function and matrix
- """
-
- if from_reference_values is None:
- from_reference_values = []
-
- if to_reference_values is None:
- to_reference_values = []
-
- if aliases is None:
- aliases = []
-
- cs = ColorSpace(name)
- cs.description = 'The %s color space' % name
- cs.aliases = aliases
- cs.equality_group = name
- cs.family = 'Utility'
- cs.is_data = False
-
- # A linear space needs allocation variables.
- cs.allocation_type = ocio.Constants.ALLOCATION_UNIFORM
- cs.allocation_vars = [0, 1]
-
- # Sampling the transfer function.
- data = array.array('f', '\0' * lut_resolution_1d * 4)
- for c in range(lut_resolution_1d):
- data[c] = transfer_function(c / (lut_resolution_1d - 1))
-
- # Writing the sampled data to a *LUT*.
- lut = '%s_to_linear.spi1d' % transfer_function_name
- genlut.write_SPI_1d(
- os.path.join(lut_directory, lut),
- 0,
- 1,
- data,
- lut_resolution_1d,
- 1)
-
- # Creating the *to_reference* transforms.
- cs.to_reference_transforms = []
- if to_reference_values:
- cs.to_reference_transforms.append({
- 'type': 'lutFile',
- 'path': lut,
- 'interpolation': 'linear',
- 'direction': 'forward'})
-
- for matrix in to_reference_values:
- cs.to_reference_transforms.append({
- 'type': 'matrix',
- 'matrix': mat44_from_mat33(matrix),
- 'direction': 'forward'})
-
- # Creating the *from_reference* transforms.
- cs.from_reference_transforms = []
- if from_reference_values:
- for matrix in from_reference_values:
- cs.from_reference_transforms.append({
- 'type': 'matrix',
- 'matrix': mat44_from_mat33(matrix),
- 'direction': 'forward'})
-
- cs.from_reference_transforms.append({
- 'type': 'lutFile',
- 'path': lut,
- 'interpolation': 'linear',
- 'direction': 'inverse'})
-
- return cs
-
-
-# Transfer functions for standard colorspaces.
-def transfer_function_sRGB_to_linear(v):
- """
- The sRGB (IEC 61966-2-1) transfer function
-
- Parameters
- ----------
- v : float
- The normalized value to pass through the function
-
- Returns
- -------
- float
- A converted value
- """
- a = 1.055
- b = 0.04045
- d = 12.92
- g = 2.4
-
- if v < b:
- return v / d
- return pow(((v + (a - 1)) / a), g)
-
-
-def transfer_function_Rec709_to_linear(v):
- """
- The Rec.709 transfer function
-
- Parameters
- ----------
- v : float
- The normalized value to pass through the function
-
- Returns
- -------
- float
- A converted value
- """
- a = 1.099
- b = 0.018
- d = 4.5
- g = (1.0 / 0.45)
-
- if v < b * d:
- return v / d
-
- return pow(((v + (a - 1)) / a), g)
-
-
-def transfer_function_Rec2020_10bit_to_linear(v):
- """
- The Rec.2020 10-bit transfer function
-
- Parameters
- ----------
- v : float
- The normalized value to pass through the function
-
- Returns
- -------
- float
- A converted value
- """
- a = 1.099
- b = 0.018
- d = 4.5
- g = (1.0 / 0.45)
-
- if v < b * d:
- return v / d
-
- return pow(((v + (a - 1)) / a), g)
-
-
-def transfer_function_Rec2020_12bit_to_linear(v):
- """
- The Rec.2020 12-bit transfer function
-
- Parameters
- ----------
- v : float
- The normalized value to pass through the function
-
- Returns
- -------
- float
- A converted value
- """
- a = 1.0993
- b = 0.0181
- d = 4.5
- g = (1.0 / 0.45)
-
- if v < b * d:
- return v / d
-
- return pow(((v + (a - 1)) / a), g)
-
-
-def transfer_function_Rec1886_to_linear(v):
- """
- The Rec.1886 transfer function
-
- Parameters
- ----------
- v : float
- The normalized value to pass through the function
-
- Returns
- -------
- float
- A converted value
- """
- g = 2.4
- Lw = 1
- Lb = 0
-
- # Ignoring legal to full scaling for now.
- # v = (1023.0*v - 64.0)/876.0
-
- t = pow(Lw, 1.0 / g) - pow(Lb, 1.0 / g)
- a = pow(t, g)
- b = pow(Lb, 1.0 / g) / t
-
- return a * pow(max((v + b), 0.0), g)
-
-
-def create_colorspaces(lut_directory,
- lut_resolution_1d):
- """
- Generates the colorspace conversions.
-
- Parameters
- ----------
- lut_directory : str or unicode
- The directory to use when generating LUTs
- lut_resolution_1d : int
- The resolution of generated 1D LUTs
-
- Returns
- -------
- list
- A list of colorspaces for general colorspaces and encodings
- """
-
- colorspaces = []
-
- # -------------------------------------------------------------------------
- # XYZ
- # -------------------------------------------------------------------------
- cs = create_matrix_colorspace('XYZ - D60',
- to_reference_values=[aces.ACES_XYZ_TO_AP0],
- from_reference_values=[aces.ACES_AP0_TO_XYZ],
- aliases=['lin_xyz_d60'])
- colorspaces.append(cs)
-
- # -------------------------------------------------------------------------
- # P3-D60
- # -------------------------------------------------------------------------
- # *ACES* to *Linear*, *P3D60* primaries
- XYZ_to_P3D60 = [2.4027414142, -0.8974841639, -0.3880533700,
- -0.8325796487, 1.7692317536, 0.0237127115,
- 0.0388233815, -0.0824996856, 1.0363685997]
-
- cs = create_matrix_colorspace(
- 'Linear - P3-D60',
- from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_P3D60],
- aliases=['lin_p3d60'])
- colorspaces.append(cs)
-
- # -------------------------------------------------------------------------
- # P3-DCI
- # -------------------------------------------------------------------------
- # *ACES* to *Linear*, *P3DCI* primaries, using Bradford chromatic
- # adaptation
- XYZ_to_P3DCI = [2.66286135, -1.11031783, -0.42271635,
- -0.82282376, 1.75861704, 0.02502194,
- 0.03932561, -0.08383448, 1.0372175]
-
- cs = create_matrix_colorspace(
- 'Linear - P3-DCI',
- from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_P3DCI],
- aliases=['lin_p3dci'])
- colorspaces.append(cs)
-
- # -------------------------------------------------------------------------
- # sRGB
- # -------------------------------------------------------------------------
- # *sRGB* and *Rec 709* use the same gamut.
-
- # *ACES* to *Linear*, *Rec. 709* primaries, D65 white point, using
- # Bradford chromatic adaptation
- XYZ_to_Rec709 = [3.20959735, -1.55742955, -0.49580497,
- -0.97098887, 1.88517118, 0.03948941,
- 0.05971934, -0.21010444, 1.14312482]
-
- cs = create_matrix_colorspace(
- 'Linear - sRGB',
- from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_Rec709],
- aliases=['lin_srgb'])
- colorspaces.append(cs)
-
- # *Linear* to *sRGB* Transfer Function*
- cs = create_transfer_colorspace(
- 'Curve - sRGB',
- 'sRGB',
- transfer_function_sRGB_to_linear,
- lut_directory,
- lut_resolution_1d,
- aliases=['crv_srgb'])
- colorspaces.append(cs)
-
- # *ACES* to *sRGB* Primaries + Transfer Function*
- cs = create_matrix_plus_transfer_colorspace(
- 'sRGB - Texture',
- 'sRGB',
- transfer_function_sRGB_to_linear,
- lut_directory,
- lut_resolution_1d,
- from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_Rec709],
- aliases=['srgb_texture'])
- colorspaces.append(cs)
-
- # -------------------------------------------------------------------------
- # Rec 709
- # -------------------------------------------------------------------------
- # *sRGB* and *Rec 709* use the same gamut.
- cs = create_matrix_colorspace(
- 'Linear - Rec.709',
- from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_Rec709],
- aliases=['lin_rec709'])
- colorspaces.append(cs)
-
- # *Linear* to *Rec. 709* Transfer Function*
- cs = create_transfer_colorspace(
- 'Curve - Rec.709',
- 'rec709',
- transfer_function_Rec709_to_linear,
- lut_directory,
- lut_resolution_1d,
- aliases=['crv_rec709'])
- colorspaces.append(cs)
-
- # *ACES* to *Rec. 709* Primaries + Transfer Function*
- cs = create_matrix_plus_transfer_colorspace(
- 'Rec.709 - Camera',
- 'rec709',
- transfer_function_Rec709_to_linear,
- lut_directory,
- lut_resolution_1d,
- from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_Rec709],
- aliases=['rec709_camera'])
- colorspaces.append(cs)
-
- # -------------------------------------------------------------------------
- # Rec 2020
- # -------------------------------------------------------------------------
- # *ACES* to *Linear*, *Rec. 2020* primaries, D65 white point, using
- # Bradford chromatic adaptation
- XYZ_to_Rec2020 = [1.69662619, -0.36551982, -0.24857099,
- -0.67039877, 1.62348187, 0.01503821,
- 0.02063163, -0.04775634, 1.01910818]
-
- cs = create_matrix_colorspace(
- 'Linear - Rec.2020',
- from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_Rec2020],
- aliases=['lin_rec2020'])
- colorspaces.append(cs)
-
- # *Linear* to *Rec. 2020 10 bit* Transfer Function*
- cs = create_transfer_colorspace(
- 'Curve - Rec.2020',
- 'rec2020',
- transfer_function_Rec2020_10bit_to_linear,
- lut_directory,
- lut_resolution_1d,
- aliases=['crv_rec2020'])
- colorspaces.append(cs)
-
- # *ACES* to *Rec. 2020 10 bit* Primaries + Transfer Function*
- cs = create_matrix_plus_transfer_colorspace(
- 'Rec.2020 - Camera',
- 'rec2020',
- transfer_function_Rec2020_10bit_to_linear,
- lut_directory,
- lut_resolution_1d,
- from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_Rec2020],
- aliases=['rec2020_camera'])
- colorspaces.append(cs)
-
- # -------------------------------------------------------------------------
- # Rec 1886
- # -------------------------------------------------------------------------
- # *Linear* to *Rec.1886* Transfer Function*
- cs = create_transfer_colorspace(
- 'Curve - Rec.1886',
- 'rec1886',
- transfer_function_Rec1886_to_linear,
- lut_directory,
- lut_resolution_1d,
- aliases=['crv_rec1886'])
- colorspaces.append(cs)
-
- # *ACES* to *Rec. 709* Primaries + Transfer Function*
- cs = create_matrix_plus_transfer_colorspace(
- 'Rec.709 - Display',
- 'rec1886',
- transfer_function_Rec1886_to_linear,
- lut_directory,
- lut_resolution_1d,
- from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_Rec709],
- aliases=['rec709_display'])
- colorspaces.append(cs)
-
- # *ACES* to *Rec. 2020* Primaries + Transfer Function*
- cs = create_matrix_plus_transfer_colorspace(
- 'Rec.2020 - Display',
- 'rec1886',
- transfer_function_Rec1886_to_linear,
- lut_directory,
- lut_resolution_1d,
- from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_Rec2020],
- aliases=['rec2020_display'])
- colorspaces.append(cs)
-
- # -------------------------------------------------------------------------
- # ProPhoto
- # -------------------------------------------------------------------------
- # *ACES* to *Linear*, *Pro Photo* primaries, D50 white point, using
- # Bradford chromatic adaptation
- AP0_to_RIMM = [1.2412367771, -0.1685692287, -0.0726675484,
- 0.0061203066, 1.083151174, -0.0892714806,
- -0.0032853314, 0.0099796402, 0.9933056912]
-
- cs = create_matrix_colorspace(
- 'Linear - RIMM ROMM (ProPhoto)',
- from_reference_values=[AP0_to_RIMM],
- aliases=['lin_prophoto', 'lin_rimm'])
- colorspaces.append(cs)
-
- # -------------------------------------------------------------------------
- # Adobe RGB
- # -------------------------------------------------------------------------
- # *ACES* to *Linear*, *Adobe RGB* primaries, D65 white point, using
- # Bradford chromatic adaptation
- AP0_to_ADOBERGB = [1.7245603168, -0.4199935942, -0.3045667227,
- -0.2764799142, 1.3727190877, -0.0962391734,
- -0.0261255258, -0.0901747807, 1.1163003065]
-
- cs = create_matrix_colorspace(
- 'Linear - Adobe RGB',
- from_reference_values=[AP0_to_ADOBERGB],
- aliases=['lin_adobergb'])
- colorspaces.append(cs)
-
- # -------------------------------------------------------------------------
- # Adobe Wide Gamut RGB
- # -------------------------------------------------------------------------
- # *ACES* to *Linear*, *Adobe Wide Gamut RGB* primaries, D50 white point,
- # using Bradford chromatic adaptation
- AP0_to_ADOBEWIDEGAMUT = [1.3809814778, -0.1158594573, -0.2651220205,
- 0.0057015535, 1.0402949043, -0.0459964578,
- -0.0038908746, -0.0597091815, 1.0636000561]
-
- cs = create_matrix_colorspace(
- 'Linear - Adobe Wide Gamut RGB',
- from_reference_values=[AP0_to_ADOBEWIDEGAMUT],
- aliases=['lin_adobewidegamutrgb'])
- colorspaces.append(cs)
-
- return colorspaces
-
-
-def create_raw():
- """
- Creates the *raw* color space
-
- Parameters
- ----------
- None
-
- Returns
- -------
- ColorSpace
- *raw* and all its identifying information
- """
- # *Raw* utility space
- name = 'Raw'
- raw = ColorSpace(name)
- raw.description = 'The %s color space' % name
- raw.aliases = ['raw']
- raw.equality_group = name
- raw.family = 'Utility'
- raw.is_data = True
-
- return raw