2 # -*- coding: utf-8 -*-
5 Implements support for *RED* colorspaces conversions and transfer functions.
8 from __future__ import division
13 import PyOpenColorIO as ocio
15 import aces_ocio.generate_lut as genlut
16 from aces_ocio.utilities import ColorSpace, mat44_from_mat33
18 __author__ = 'ACES Developers'
19 __copyright__ = 'Copyright (C) 2014 - 2015 - ACES Developers'
21 __maintainer__ = 'ACES Developers'
22 __email__ = 'aces@oscars.org'
23 __status__ = 'Production'
25 __all__ = ['create_RED_log_film',
29 def create_RED_log_film(gamut,
38 RED colorspaces to ACES.
43 Parameter description.
48 Return value description.
51 name = '%s - %s' % (transfer_function, gamut)
52 if transfer_function == '':
53 name = 'Linear - %s' % gamut
55 name = '%s' % transfer_function
60 cs.equality_group = ''
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]
69 def cineon_to_linear(code_value):
73 code_value_to_density = 0.002
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))
80 return (code_linear - black_linear) / (1 - black_linear)
82 cs.to_reference_transforms = []
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))
89 lut = 'CineonLog_to_linear.spi1d'
91 os.path.join(lut_directory, lut),
98 cs.to_reference_transforms.append({
101 'interpolation': 'linear',
102 'direction': 'forward'})
104 if gamut == 'DRAGONcolor':
105 cs.to_reference_transforms.append({
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({
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 == 'REDcolor2':
119 cs.to_reference_transforms.append({
121 'matrix': mat44_from_mat33([0.480997, 0.402289, 0.116714,
122 -0.004938, 1.000154, 0.004781,
123 -0.105257, 0.025320, 1.079907]),
124 'direction': 'forward'})
125 elif gamut == 'REDcolor3':
126 cs.to_reference_transforms.append({
128 'matrix': mat44_from_mat33([0.512136, 0.360370, 0.127494,
129 0.070377, 0.903884, 0.025737,
130 -0.020824, 0.017671, 1.003123]),
131 'direction': 'forward'})
132 elif gamut == 'REDcolor4':
133 cs.to_reference_transforms.append({
135 'matrix': mat44_from_mat33([0.474202, 0.333677, 0.192121,
136 0.065164, 0.836932, 0.097901,
137 -0.019281, 0.016362, 1.002889]),
138 'direction': 'forward'})
140 cs.from_reference_transforms = []
144 def create_colorspaces(lut_directory, lut_resolution_1d):
146 Generates the colorspace conversions.
151 Parameter description.
156 Return value description.
162 RED_log_film_dragon = create_RED_log_film(
169 colorspaces.append(RED_log_film_dragon)
171 RED_log_film_dragon2 = create_RED_log_film(
178 colorspaces.append(RED_log_film_dragon2)
180 RED_log_film_color2 = create_RED_log_film(
187 colorspaces.append(RED_log_film_color2)
189 RED_log_film_color3 = create_RED_log_film(
196 colorspaces.append(RED_log_film_color3)
198 RED_log_film_color4 = create_RED_log_film(
205 colorspaces.append(RED_log_film_color4)
208 RED_log_film = create_RED_log_film(
215 colorspaces.append(RED_log_film)
218 RED_dragon = create_RED_log_film(
225 colorspaces.append(RED_dragon)
227 RED_dragon2 = create_RED_log_film(
234 colorspaces.append(RED_dragon2)
236 RED_color2 = create_RED_log_film(
243 colorspaces.append(RED_color2)
245 RED_color3 = create_RED_log_film(
252 colorspaces.append(RED_color3)
254 RED_color4 = create_RED_log_film(
261 colorspaces.append(RED_color4)