aces_1.0.0/python/aces_ocio/colorspaces/red.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3
   4 """
   5 Implements support for *RED* colorspaces conversions and transfer functions.
   6 """
   7
   8 from __future__ import division
   9
  10 import array
  11 import os
  12
  13 import PyOpenColorIO as ocio
  14
  15 import aces_ocio.generate_lut as genlut
  16 from aces_ocio.utilities import ColorSpace, mat44_from_mat33
  17
  18 __author__ = 'ACES Developers'
  19 __copyright__ = 'Copyright (C) 2014 - 2015 - ACES Developers'
  20 __license__ = ''
  21 __maintainer__ = 'ACES Developers'
  22 __email__ = 'aces@oscars.org'
  23 __status__ = 'Production'
  24
  25 __all__ = ['create_RED_log_film',
  26            'create_colorspaces']
  27
  28
  29 def create_RED_log_film(gamut,
  30                         transfer_function,
  31                         name,
  32                         lut_directory,
  33                         lut_resolution_1d,
  34                         aliases=[]):
  35     """
  36     Object description.
  37
  38     RED colorspaces to ACES.
  39
  40     Parameters
  41     ----------
  42     parameter : type
  43         Parameter description.
  44
  45     Returns
  46     -------
  47     type
  48          Return value description.
  49     """
  50
  51     name = '%s - %s' % (transfer_function, gamut)
  52     if transfer_function == '':
  53         name = 'Linear - %s' % gamut
  54     if gamut == '':
  55         name = 'Curve - %s' % transfer_function
  56
  57     cs = ColorSpace(name)
  58     cs.description = name
  59     cs.aliases = aliases
  60     cs.equality_group = ''
  61     cs.family = 'Input/RED'
  62     cs.is_data = False
  63
  64     # A linear space needs allocation variables
  65     if transfer_function == '':
  66         cs.allocation_type = ocio.Constants.ALLOCATION_LG2
  67         cs.allocation_vars = [-8, 5, 0.00390625]
  68
  69     def cineon_to_linear(code_value):
  70         n_gamma = 0.6
  71         black_point = 95
  72         white_point = 685
  73         code_value_to_density = 0.002
  74
  75         black_linear = pow(10, (black_point - white_point) * (
  76             code_value_to_density / n_gamma))
  77         code_linear = pow(10, (code_value - white_point) * (
  78             code_value_to_density / n_gamma))
  79
  80         return (code_linear - black_linear) / (1 - black_linear)
  81
  82     cs.to_reference_transforms = []
  83
  84     if transfer_function == 'REDlogFilm':
  85         data = array.array('f', '\0' * lut_resolution_1d * 4)
  86         for c in range(lut_resolution_1d):
  87             data[c] = cineon_to_linear(1023 * c / (lut_resolution_1d - 1))
  88
  89         lut = 'CineonLog_to_linear.spi1d'
  90         genlut.write_SPI_1d(
  91             os.path.join(lut_directory, lut),
  92             0,
  93             1,
  94             data,
  95             lut_resolution_1d,
  96             1)
  97
  98         cs.to_reference_transforms.append({
  99             'type': 'lutFile',
 100             'path': lut,
 101             'interpolation': 'linear',
 102             'direction': 'forward'})
 103
 104     if gamut == 'DRAGONcolor':
 105         cs.to_reference_transforms.append({
 106             'type': 'matrix',
 107             'matrix': mat44_from_mat33([0.532279, 0.376648, 0.091073,
 108                                         0.046344, 0.974513, -0.020860,
 109                                         -0.053976, -0.000320, 1.054267]),
 110             'direction': 'forward'})
 111     elif gamut == 'DRAGONcolor2':
 112         cs.to_reference_transforms.append({
 113             'type': 'matrix',
 114             'matrix': mat44_from_mat33([0.468452, 0.331484, 0.200064,
 115                                         0.040787, 0.857658, 0.101553,
 116                                         -0.047504, -0.000282, 1.047756]),
 117             'direction': 'forward'})
 118     elif gamut == 'REDcolor':
 119         cs.to_reference_transforms.append({
 120             'type': 'matrix',
 121             'matrix': mat44_from_mat33([0.451464, 0.388498, 0.160038,
 122                                         0.062716, 0.866790, 0.070491,
 123                                         -0.017541, 0.086921, 0.930590]),
 124             'direction': 'forward'})
 125     elif gamut == 'REDcolor2':
 126         cs.to_reference_transforms.append({
 127             'type': 'matrix',
 128             'matrix': mat44_from_mat33([0.480997, 0.402289, 0.116714,
 129                                         -0.004938, 1.000154, 0.004781,
 130                                         -0.105257, 0.025320, 1.079907]),
 131             'direction': 'forward'})
 132     elif gamut == 'REDcolor3':
 133         cs.to_reference_transforms.append({
 134             'type': 'matrix',
 135             'matrix': mat44_from_mat33([0.512136, 0.360370, 0.127494,
 136                                         0.070377, 0.903884, 0.025737,
 137                                         -0.020824, 0.017671, 1.003123]),
 138             'direction': 'forward'})
 139     elif gamut == 'REDcolor4':
 140         cs.to_reference_transforms.append({
 141             'type': 'matrix',
 142             'matrix': mat44_from_mat33([0.474202, 0.333677, 0.192121,
 143                                         0.065164, 0.836932, 0.097901,
 144                                         -0.019281, 0.016362, 1.002889]),
 145             'direction': 'forward'})
 146
 147     cs.from_reference_transforms = []
 148     return cs
 149
 150
 151 def create_colorspaces(lut_directory, lut_resolution_1d):
 152     """
 153     Generates the colorspace conversions.
 154
 155     Parameters
 156     ----------
 157     parameter : type
 158         Parameter description.
 159
 160     Returns
 161     -------
 162     type
 163          Return value description.
 164     """
 165
 166     colorspaces = []
 167
 168     # Full conversion
 169     RED_log_film_dragon = create_RED_log_film(
 170         'DRAGONcolor',
 171         'REDlogFilm',
 172         'REDlogFilm',
 173         lut_directory,
 174         lut_resolution_1d,
 175         ["rlf_dgn"])
 176     colorspaces.append(RED_log_film_dragon)
 177
 178     RED_log_film_dragon2 = create_RED_log_film(
 179         'DRAGONcolor2',
 180         'REDlogFilm',
 181         'REDlogFilm',
 182         lut_directory,
 183         lut_resolution_1d,
 184         ["rlf_dgn2"])
 185     colorspaces.append(RED_log_film_dragon2)
 186
 187     RED_log_film_color = create_RED_log_film(
 188         'REDcolor',
 189         'REDlogFilm',
 190         'REDlogFilm',
 191         lut_directory,
 192         lut_resolution_1d,
 193         ["rlf_rc"])
 194     colorspaces.append(RED_log_film_color)
 195
 196     RED_log_film_color2 = create_RED_log_film(
 197         'REDcolor2',
 198         'REDlogFilm',
 199         'REDlogFilm',
 200         lut_directory,
 201         lut_resolution_1d,
 202         ["rlf_rc2"])
 203     colorspaces.append(RED_log_film_color2)
 204
 205     RED_log_film_color3 = create_RED_log_film(
 206         'REDcolor3',
 207         'REDlogFilm',
 208         'REDlogFilm',
 209         lut_directory,
 210         lut_resolution_1d,
 211         ["rlf_rc3"])
 212     colorspaces.append(RED_log_film_color3)
 213
 214     RED_log_film_color4 = create_RED_log_film(
 215         'REDcolor4',
 216         'REDlogFilm',
 217         'REDlogFilm',
 218         lut_directory,
 219         lut_resolution_1d,
 220         ["rlf_rc4"])
 221     colorspaces.append(RED_log_film_color4)
 222
 223     # Linearization only
 224     RED_log_film = create_RED_log_film(
 225         '',
 226         'REDlogFilm',
 227         'REDlogFilm',
 228         lut_directory,
 229         lut_resolution_1d,
 230         ["crv_rlf"])
 231     colorspaces.append(RED_log_film)
 232
 233     # Primaries only
 234     RED_dragon = create_RED_log_film(
 235         'DRAGONcolor',
 236         '',
 237         'REDlogFilm',
 238         lut_directory,
 239         lut_resolution_1d,
 240         ["lin_dgn"])
 241     colorspaces.append(RED_dragon)
 242
 243     RED_dragon2 = create_RED_log_film(
 244         'DRAGONcolor2',
 245         '',
 246         'REDlogFilm',
 247         lut_directory,
 248         lut_resolution_1d,
 249         ["lin_dgn2"])
 250     colorspaces.append(RED_dragon2)
 251
 252     RED_color = create_RED_log_film(
 253         'REDcolor',
 254         '',
 255         'REDlogFilm',
 256         lut_directory,
 257         lut_resolution_1d,
 258         ["lin_rc"])
 259     colorspaces.append(RED_color)
 260
 261     RED_color2 = create_RED_log_film(
 262         'REDcolor2',
 263         '',
 264         'REDlogFilm',
 265         lut_directory,
 266         lut_resolution_1d,
 267         ["lin_rc2"])
 268     colorspaces.append(RED_color2)
 269
 270     RED_color3 = create_RED_log_film(
 271         'REDcolor3',
 272         '',
 273         'REDlogFilm',
 274         lut_directory,
 275         lut_resolution_1d,
 276         ["lin_rc3"])
 277     colorspaces.append(RED_color3)
 278
 279     RED_color4 = create_RED_log_film(
 280         'REDcolor4',
 281         '',
 282         'REDlogFilm',
 283         lut_directory,
 284         lut_resolution_1d,
 285         ["lin_rc4"])
 286     colorspaces.append(RED_color4)
 287
 288     return colorspaces