--- /dev/null
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+Implements support for *Canon* 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.utilities import ColorSpace
+
+__author__ = 'ACES Developers'
+__copyright__ = 'Copyright (C) 2014 - 2015 - ACES Developers'
+__license__ = ''
+__maintainer__ = 'ACES Developers'
+__email__ = 'aces@oscars.org'
+__status__ = 'Production'
+
+__all__ = ['create_c_log',
+ 'create_colorspaces']
+
+
+def create_c_log(gamut,
+ transfer_function,
+ lut_directory,
+ lut_resolution_1d,
+ aliases):
+ """
+ Creates colorspace covering the conversion from CLog to ACES, with various transfer
+ functions and encoding gamuts covered
+
+ Parameters
+ ----------
+ gamut : str
+ The name of the encoding gamut to use.
+ transfer_function : str
+ The name of the transfer function to use
+ 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 ColorSpace container class referencing the LUTs, matrices and identifying
+ information for the requested colorspace.
+ """
+
+ name = '%s - %s' % (transfer_function, gamut)
+ if transfer_function == '':
+ name = 'Linear - Canon %s' % gamut
+ if gamut == '':
+ name = 'Curve - %s' % transfer_function
+
+ cs = ColorSpace(name)
+ cs.description = name
+ cs.aliases = aliases
+ cs.equality_group = ''
+ cs.family = 'Input/Canon'
+ cs.is_data = False
+
+ # A linear space needs allocation variables.
+ if transfer_function == '':
+ cs.allocation_type = ocio.Constants.ALLOCATION_LG2
+ cs.allocation_vars = [-8, 5, 0.00390625]
+
+ def legal_to_full(code_value):
+ return (code_value - 64) / (940 - 64)
+
+ def c_log_to_linear(code_value):
+ # log = fullToLegal(c1 * log10(c2*linear + 1) + c3)
+ # linear = (pow(10, (legalToFul(log) - c3)/c1) - 1)/c2
+ c1 = 0.529136
+ c2 = 10.1596
+ c3 = 0.0730597
+
+ linear = (pow(10, (legal_to_full(code_value) - c3) / c1) - 1) / c2
+ linear *= 0.9
+
+ return linear
+
+ cs.to_reference_transforms = []
+
+ if transfer_function == 'Canon-Log':
+ data = array.array('f', '\0' * lut_resolution_1d * 4)
+ for c in range(lut_resolution_1d):
+ data[c] = c_log_to_linear(1023 * c / (lut_resolution_1d - 1))
+
+ lut = '%s_to_linear.spi1d' % transfer_function
+ genlut.write_SPI_1d(
+ os.path.join(lut_directory, lut),
+ 0,
+ 1,
+ data,
+ lut_resolution_1d,
+ 1)
+
+ cs.to_reference_transforms.append({
+ 'type': 'lutFile',
+ 'path': lut,
+ 'interpolation': 'linear',
+ 'direction': 'forward'})
+
+ if gamut == 'Rec. 709 Daylight':
+ cs.to_reference_transforms.append({
+ 'type': 'matrix',
+ 'matrix': [0.561538969, 0.402060105, 0.036400926, 0,
+ 0.092739623, 0.924121198, -0.016860821, 0,
+ 0.084812961, 0.006373835, 0.908813204, 0,
+ 0, 0, 0, 1],
+ 'direction': 'forward'})
+ elif gamut == 'Rec. 709 Tungsten':
+ cs.to_reference_transforms.append({
+ 'type': 'matrix',
+ 'matrix': [0.566996399, 0.365079418, 0.067924183, 0,
+ 0.070901044, 0.880331008, 0.048767948, 0,
+ 0.073013542, -0.066540862, 0.99352732, 0,
+ 0, 0, 0, 1],
+ 'direction': 'forward'})
+ elif gamut == 'DCI-P3 Daylight':
+ cs.to_reference_transforms.append({
+ 'type': 'matrix',
+ 'matrix': [0.607160575, 0.299507286, 0.093332140, 0,
+ 0.004968120, 1.050982224, -0.055950343, 0,
+ -0.007839939, 0.000809127, 1.007030813, 0,
+ 0, 0, 0, 1],
+ 'direction': 'forward'})
+ elif gamut == 'DCI-P3 Tungsten':
+ cs.to_reference_transforms.append({
+ 'type': 'matrix',
+ 'matrix': [0.650279125, 0.253880169, 0.095840706, 0,
+ -0.026137986, 1.017900530, 0.008237456, 0,
+ 0.007757558, -0.063081669, 1.055324110, 0,
+ 0, 0, 0, 1],
+ 'direction': 'forward'})
+ elif gamut == 'Cinema Gamut Daylight':
+ cs.to_reference_transforms.append({
+ 'type': 'matrix',
+ 'matrix': [0.763064455, 0.149021161, 0.087914384, 0,
+ 0.003657457, 1.10696038, -0.110617837, 0,
+ -0.009407794, -0.218383305, 1.227791099, 0,
+ 0, 0, 0, 1],
+ 'direction': 'forward'})
+ elif gamut == 'Cinema Gamut Tungsten':
+ cs.to_reference_transforms.append({
+ 'type': 'matrix',
+ 'matrix': [0.817416293, 0.090755698, 0.091828009, 0,
+ -0.035361374, 1.065690585, -0.030329211, 0,
+ 0.010390366, -0.299271107, 1.288880741, 0,
+ 0, 0, 0, 1],
+ 'direction': 'forward'})
+
+ cs.from_reference_transforms = []
+ return cs
+
+
+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 Canon cameras and encodings
+ """
+
+ colorspaces = []
+
+ # Full conversion
+ c_log_1 = create_c_log(
+ 'Rec. 709 Daylight',
+ 'Canon-Log',
+ lut_directory,
+ lut_resolution_1d,
+ ['canonlog_rec709day'])
+ colorspaces.append(c_log_1)
+
+ c_log_2 = create_c_log(
+ 'Rec. 709 Tungsten',
+ 'Canon-Log',
+ lut_directory,
+ lut_resolution_1d,
+ ['canonlog_rec709tung'])
+ colorspaces.append(c_log_2)
+
+ c_log_3 = create_c_log(
+ 'DCI-P3 Daylight',
+ 'Canon-Log',
+ lut_directory,
+ lut_resolution_1d,
+ ['canonlog_dcip3day'])
+ colorspaces.append(c_log_3)
+
+ c_log_4 = create_c_log(
+ 'DCI-P3 Tungsten',
+ 'Canon-Log',
+ lut_directory,
+ lut_resolution_1d,
+ ['canonlog_dcip3tung'])
+ colorspaces.append(c_log_4)
+
+ c_log_5 = create_c_log(
+ 'Cinema Gamut Daylight',
+ 'Canon-Log',
+ lut_directory,
+ lut_resolution_1d,
+ ['canonlog_cgamutday'])
+ colorspaces.append(c_log_5)
+
+ c_log_6 = create_c_log(
+ 'Cinema Gamut Tungsten',
+ 'Canon-Log',
+ lut_directory,
+ lut_resolution_1d,
+ ['canonlog_cgamuttung'])
+ colorspaces.append(c_log_6)
+
+ # Linearization Only
+ c_log_7 = create_c_log(
+ '',
+ 'Canon-Log',
+ lut_directory,
+ lut_resolution_1d,
+ ['crv_canonlog'])
+ colorspaces.append(c_log_7)
+
+ # Primaries Only
+ c_log_8 = create_c_log(
+ 'Rec. 709 Daylight',
+ '',
+ lut_directory,
+ lut_resolution_1d,
+ ['lin_canonrec709day'])
+ colorspaces.append(c_log_8)
+
+ c_log_9 = create_c_log(
+ 'Rec. 709 Tungsten',
+ '',
+ lut_directory,
+ lut_resolution_1d,
+ ['lin_canonrec709tung'])
+ colorspaces.append(c_log_9)
+
+ c_log_10 = create_c_log(
+ 'DCI-P3 Daylight',
+ '',
+ lut_directory,
+ lut_resolution_1d,
+ ['lin_canondcip3day'])
+ colorspaces.append(c_log_10)
+
+ c_log_11 = create_c_log(
+ 'DCI-P3 Tungsten',
+ '',
+ lut_directory,
+ lut_resolution_1d,
+ ['lin_canondcip3tung'])
+ colorspaces.append(c_log_11)
+
+ c_log_12 = create_c_log(
+ 'Cinema Gamut Daylight',
+ '',
+ lut_directory,
+ lut_resolution_1d,
+ ['lin_canoncgamutday'])
+ colorspaces.append(c_log_12)
+
+ c_log_13 = create_c_log(
+ 'Cinema Gamut Tungsten',
+ '',
+ lut_directory,
+ lut_resolution_1d,
+ ['lin_canoncgamuttung'])
+ colorspaces.append(c_log_13)
+
+ return colorspaces