aces_1.0.0/python/aces_ocio/create_canon_colorspaces.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3
   4 """
   5 Implements support for *Canon* colorspaces conversions and transfer functions.
   6 """
   7
   8 import array
   9 import os
  10
  11 import aces_ocio.generate_lut as genlut
  12 from aces_ocio.utilities import ColorSpace
  13
  14 __author__ = 'ACES Developers'
  15 __copyright__ = 'Copyright (C) 2014 - 2015 - ACES Developers'
  16 __license__ = ''
  17 __maintainer__ = 'ACES Developers'
  18 __email__ = 'aces@oscars.org'
  19 __status__ = 'Production'
  20
  21 __all__ = ['create_c_log',
  22            'create_colorspaces']
  23
  24
  25 def create_c_log(gamut,
  26                  transfer_function,
  27                  name,
  28                  lut_directory,
  29                  lut_resolution_1d,
  30                  aliases):
  31     """
  32     Object description.
  33
  34     Canon-Log to ACES.
  35
  36     Parameters
  37     ----------
  38     parameter : type
  39         Parameter description.
  40
  41     Returns
  42     -------
  43     type
  44          Return value description.
  45     """
  46
  47     name = '%s - %s' % (transfer_function, gamut)
  48     if transfer_function == '':
  49         name = 'Linear - %s' % gamut
  50     if gamut == '':
  51         name = '%s' % transfer_function
  52
  53     cs = ColorSpace(name)
  54     cs.description = name
  55     cs.aliases = aliases
  56     cs.equality_group = ''
  57     cs.family = 'Canon'
  58     cs.is_data = False
  59
  60     def legal_to_full(code_value):
  61         return (code_value - 64.0) / (940.0 - 64.0)
  62
  63     def c_log_to_linear(code_value):
  64         # log = fullToLegal(c1 * log10(c2*linear + 1) + c3)
  65         # linear = (pow(10, (legalToFul(log) - c3)/c1) - 1)/c2
  66         c1 = 0.529136
  67         c2 = 10.1596
  68         c3 = 0.0730597
  69
  70         linear = (pow(10.0, (legal_to_full(code_value) - c3) / c1) - 1.0) / c2
  71         linear *= 0.9
  72
  73         return linear
  74
  75     cs.to_reference_transforms = []
  76
  77     if transfer_function == 'Canon-Log':
  78         data = array.array('f', '\0' * lut_resolution_1d * 4)
  79         for c in range(lut_resolution_1d):
  80             data[c] = c_log_to_linear(1023.0 * c / (lut_resolution_1d - 1))
  81
  82         lut = '%s_to_linear.spi1d' % transfer_function
  83         genlut.write_SPI_1d(
  84             os.path.join(lut_directory, lut),
  85             0.0,
  86             1.0,
  87             data,
  88             lut_resolution_1d,
  89             1)
  90
  91         cs.to_reference_transforms.append({
  92             'type': 'lutFile',
  93             'path': lut,
  94             'interpolation': 'linear',
  95             'direction': 'forward'})
  96
  97     if gamut == 'Rec. 709 Daylight':
  98         cs.to_reference_transforms.append({
  99             'type': 'matrix',
 100             'matrix': [0.561538969, 0.402060105, 0.036400926, 0.0,
 101                        0.092739623, 0.924121198, -0.016860821, 0.0,
 102                        0.084812961, 0.006373835, 0.908813204, 0.0,
 103                        0, 0, 0, 1.0],
 104             'direction': 'forward'})
 105     elif gamut == 'Rec. 709 Tungsten':
 106         cs.to_reference_transforms.append({
 107             'type': 'matrix',
 108             'matrix': [0.566996399, 0.365079418, 0.067924183, 0.0,
 109                        0.070901044, 0.880331008, 0.048767948, 0.0,
 110                        0.073013542, -0.066540862, 0.99352732, 0.0,
 111                        0, 0, 0, 1.0],
 112             'direction': 'forward'})
 113     elif gamut == 'DCI-P3 Daylight':
 114         cs.to_reference_transforms.append({
 115             'type': 'matrix',
 116             'matrix': [0.607160575, 0.299507286, 0.093332140, 0.0,
 117                        0.004968120, 1.050982224, -0.055950343, 0.0,
 118                        -0.007839939, 0.000809127, 1.007030813, 0.0,
 119                        0, 0, 0, 1.0],
 120             'direction': 'forward'})
 121     elif gamut == 'DCI-P3 Tungsten':
 122         cs.to_reference_transforms.append({
 123             'type': 'matrix',
 124             'matrix': [0.650279125, 0.253880169, 0.095840706, 0.0,
 125                        -0.026137986, 1.017900530, 0.008237456, 0.0,
 126                        0.007757558, -0.063081669, 1.055324110, 0.0,
 127                        0, 0, 0, 1.0],
 128             'direction': 'forward'})
 129     elif gamut == 'Cinema Gamut Daylight':
 130         cs.to_reference_transforms.append({
 131             'type': 'matrix',
 132             'matrix': [0.763064455, 0.149021161, 0.087914384, 0.0,
 133                        0.003657457, 1.10696038, -0.110617837, 0.0,
 134                        -0.009407794, -0.218383305, 1.227791099, 0.0,
 135                        0, 0, 0, 1.0],
 136             'direction': 'forward'})
 137     elif gamut == 'Cinema Gamut Tungsten':
 138         cs.to_reference_transforms.append({
 139             'type': 'matrix',
 140             'matrix': [0.817416293, 0.090755698, 0.091828009, 0.0,
 141                        -0.035361374, 1.065690585, -0.030329211, 0.0,
 142                        0.010390366, -0.299271107, 1.288880741, 0.0,
 143                        0, 0, 0, 1.0],
 144             'direction': 'forward'})
 145
 146     cs.from_reference_transforms = []
 147     return cs
 148
 149
 150 def create_colorspaces(lut_directory, lut_resolution_1d):
 151     """
 152     Generates the colorspace conversions.
 153
 154     Parameters
 155     ----------
 156     parameter : type
 157         Parameter description.
 158
 159     Returns
 160     -------
 161     type
 162          Return value description.
 163     """
 164
 165     colorspaces = []
 166
 167     # Full conversion
 168     c_log_1 = create_c_log(
 169         'Rec. 709 Daylight',
 170         'Canon-Log',
 171         'Canon-Log',
 172         lut_directory,
 173         lut_resolution_1d,
 174         ["canonlog_rec709day"])
 175     colorspaces.append(c_log_1)
 176
 177     c_log_2 = create_c_log(
 178         'Rec. 709 Tungsten',
 179         'Canon-Log',
 180         'Canon-Log',
 181         lut_directory,
 182         lut_resolution_1d,
 183         ["canonlog_rec709tung"])
 184     colorspaces.append(c_log_2)
 185
 186     c_log_3 = create_c_log(
 187         'DCI-P3 Daylight',
 188         'Canon-Log',
 189         'Canon-Log',
 190         lut_directory,
 191         lut_resolution_1d,
 192         ["canonlog_dcip3day"])
 193     colorspaces.append(c_log_3)
 194
 195     c_log_4 = create_c_log(
 196         'DCI-P3 Tungsten',
 197         'Canon-Log',
 198         'Canon-Log',
 199         lut_directory,
 200         lut_resolution_1d,
 201         ["canonlog_dcip3tung"])
 202     colorspaces.append(c_log_4)
 203
 204     c_log_5 = create_c_log(
 205         'Cinema Gamut Daylight',
 206         'Canon-Log',
 207         'Canon-Log',
 208         lut_directory,
 209         lut_resolution_1d,
 210         ["canonlog_cgamutday"])
 211     colorspaces.append(c_log_5)
 212
 213     c_log_6 = create_c_log(
 214         'Cinema Gamut Tungsten',
 215         'Canon-Log',
 216         'Canon-Log',
 217         lut_directory,
 218         lut_resolution_1d,
 219         ["canonlog_cgamuttung"])
 220     colorspaces.append(c_log_6)
 221
 222     # Linearization Only
 223     c_log_7 = create_c_log(
 224         '',
 225         'Canon-Log',
 226         'Canon-Log',
 227         lut_directory,
 228         lut_resolution_1d,
 229         ["crv_canonlog"])
 230     colorspaces.append(c_log_7)
 231
 232     # Primaries Only
 233     c_log_8 = create_c_log(
 234         'Rec. 709 Daylight',
 235         '',
 236         'Canon-Log',
 237         lut_directory,
 238         lut_resolution_1d,
 239         ["lin_rec709day"])
 240     colorspaces.append(c_log_8)
 241
 242     c_log_9 = create_c_log(
 243         'Rec. 709 Tungsten',
 244         '',
 245         'Canon-Log',
 246         lut_directory,
 247         lut_resolution_1d,
 248         ["lin_rec709tung"])
 249     colorspaces.append(c_log_9)
 250
 251     c_log_10 = create_c_log(
 252         'DCI-P3 Daylight',
 253         '',
 254         'Canon-Log',
 255         lut_directory,
 256         lut_resolution_1d,
 257         ["lin_dcip3day"])
 258     colorspaces.append(c_log_10)
 259
 260     c_log_11 = create_c_log(
 261         'DCI-P3 Tungsten',
 262         '',
 263         'Canon-Log',
 264         lut_directory,
 265         lut_resolution_1d,
 266         ["lin_dcip3tung"])
 267     colorspaces.append(c_log_11)
 268
 269     c_log_12 = create_c_log(
 270         'Cinema Gamut Daylight',
 271         '',
 272         'Canon-Log',
 273         lut_directory,
 274         lut_resolution_1d,
 275         ["lin_cgamutday"])
 276     colorspaces.append(c_log_12)
 277
 278     c_log_13 = create_c_log(
 279         'Cinema Gamut Tungsten',
 280         '',
 281         'Canon-Log',
 282         lut_directory,
 283         lut_resolution_1d,
 284         ["lin_cgamuttung"])
 285     colorspaces.append(c_log_13)
 286
 287     return colorspaces