2 # -*- coding: utf-8 -*-
5 Implements support for general 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.colorspaces import aces
17 from aces_ocio.utilities import ColorSpace, mat44_from_mat33
19 __author__ = 'ACES Developers'
20 __copyright__ = 'Copyright (C) 2014 - 2015 - ACES Developers'
22 __maintainer__ = 'ACES Developers'
23 __email__ = 'aces@oscars.org'
24 __status__ = 'Production'
26 __all__ = ['create_matrix_colorspace',
30 # -------------------------------------------------------------------------
32 # -------------------------------------------------------------------------
33 def create_matrix_colorspace(name='matrix',
34 from_reference_values=None,
35 to_reference_values=None,
43 Parameter description.
48 Return value description.
51 if from_reference_values is None:
52 from_reference_values = []
54 if to_reference_values is None:
55 to_reference_values = []
61 cs.description = 'The %s color space' % name
63 cs.equality_group = name
67 # A linear space needs allocation variables.
68 cs.allocation_type = ocio.Constants.ALLOCATION_UNIFORM
69 cs.allocation_vars = [0, 1]
71 cs.to_reference_transforms = []
72 if to_reference_values:
73 for matrix in to_reference_values:
74 cs.to_reference_transforms.append({
76 'matrix': mat44_from_mat33(matrix),
77 'direction': 'forward'})
79 cs.from_reference_transforms = []
80 if from_reference_values:
81 for matrix in from_reference_values:
82 cs.from_reference_transforms.append({
84 'matrix': mat44_from_mat33(matrix),
85 'direction': 'forward'})
90 # -------------------------------------------------------------------------
91 # *Transfer Function Transform*
92 # -------------------------------------------------------------------------
93 def create_transfer_colorspace(name='transfer',
94 transfer_function_name='transfer_function',
95 transfer_function=lambda x: x,
97 lut_resolution_1d=1024,
105 Parameter description.
110 Return value description.
116 cs = ColorSpace(name)
117 cs.description = 'The %s color space' % name
119 cs.equality_group = name
120 cs.family = 'Utility'
123 # A linear space needs allocation variables.
124 cs.allocation_type = ocio.Constants.ALLOCATION_UNIFORM
125 cs.allocation_vars = [0, 1]
127 # Sampling the transfer function.
128 data = array.array('f', '\0' * lut_resolution_1d * 4)
129 for c in range(lut_resolution_1d):
130 data[c] = transfer_function(c / (lut_resolution_1d - 1))
132 # Writing the sampled data to a *LUT*.
133 lut = '%s_to_linear.spi1d' % transfer_function_name
135 os.path.join(lut_directory, lut),
142 # Creating the *to_reference* transforms.
143 cs.to_reference_transforms = []
144 cs.to_reference_transforms.append({
147 'interpolation': 'linear',
148 'direction': 'forward'})
150 # Creating the *from_reference* transforms.
151 cs.from_reference_transforms = []
156 # -------------------------------------------------------------------------
157 # *Transfer Function + Matrix Transform*
158 # -------------------------------------------------------------------------
159 def create_matrix_plus_transfer_colorspace(
160 name='matrix_plus_transfer',
161 transfer_function_name='transfer_function',
162 transfer_function=lambda x: x,
163 lut_directory='/tmp',
164 lut_resolution_1d=1024,
165 from_reference_values=None,
166 to_reference_values=None,
174 Parameter description.
179 Return value description.
182 if from_reference_values is None:
183 from_reference_values = []
185 if to_reference_values is None:
186 to_reference_values = []
191 cs = ColorSpace(name)
192 cs.description = 'The %s color space' % name
194 cs.equality_group = name
195 cs.family = 'Utility'
198 # A linear space needs allocation variables.
199 cs.allocation_type = ocio.Constants.ALLOCATION_UNIFORM
200 cs.allocation_vars = [0, 1]
202 # Sampling the transfer function.
203 data = array.array('f', '\0' * lut_resolution_1d * 4)
204 for c in range(lut_resolution_1d):
205 data[c] = transfer_function(c / (lut_resolution_1d - 1))
207 # Writing the sampled data to a *LUT*.
208 lut = '%s_to_linear.spi1d' % transfer_function_name
210 os.path.join(lut_directory, lut),
217 # Creating the *to_reference* transforms.
218 cs.to_reference_transforms = []
219 if to_reference_values:
220 cs.to_reference_transforms.append({
223 'interpolation': 'linear',
224 'direction': 'forward'})
226 for matrix in to_reference_values:
227 cs.to_reference_transforms.append({
229 'matrix': mat44_from_mat33(matrix),
230 'direction': 'forward'})
232 # Creating the *from_reference* transforms.
233 cs.from_reference_transforms = []
234 if from_reference_values:
235 for matrix in from_reference_values:
236 cs.from_reference_transforms.append({
238 'matrix': mat44_from_mat33(matrix),
239 'direction': 'forward'})
241 cs.from_reference_transforms.append({
244 'interpolation': 'linear',
245 'direction': 'inverse'})
250 # Transfer functions for standard colorspaces.
251 def transfer_function_sRGB_to_linear(v):
259 return pow(((v + (a - 1)) / a), g)
262 def transfer_function_Rec709_to_linear(v):
271 return pow(((v + (a - 1)) / a), g)
274 def transfer_function_Rec2020_10bit_to_linear(v):
283 return pow(((v + (a - 1)) / a), g)
286 def transfer_function_Rec2020_12bit_to_linear(v):
295 return pow(((v + (a - 1)) / a), g)
298 def transfer_function_Rec1886_to_linear(v):
303 # Ignoring legal to full scaling for now.
304 # v = (1023.0*v - 64.0)/876.0
306 t = pow(Lw, 1.0 / g) - pow(Lb, 1.0 / g)
308 b = pow(Lb, 1.0 / g) / t
310 return a * pow(max((v + b), 0.0), g)
313 def create_colorspaces(lut_directory,
316 Generates the colorspace conversions.
321 Parameter description.
326 Return value description.
331 # -------------------------------------------------------------------------
333 # -------------------------------------------------------------------------
334 cs = create_matrix_colorspace('XYZ-D60',
335 to_reference_values=[aces.ACES_XYZ_TO_AP0],
336 from_reference_values=[aces.ACES_AP0_TO_XYZ],
337 aliases=['lin_xyz_d60'])
338 colorspaces.append(cs)
340 # -------------------------------------------------------------------------
342 # -------------------------------------------------------------------------
343 # *ACES* to *Linear*, *P3D60* primaries
344 XYZ_to_P3D60 = [2.4027414142, -0.8974841639, -0.3880533700,
345 -0.8325796487, 1.7692317536, 0.0237127115,
346 0.0388233815, -0.0824996856, 1.0363685997]
348 cs = create_matrix_colorspace(
350 from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_P3D60],
351 aliases=['lin_p3d60'])
352 colorspaces.append(cs)
354 # -------------------------------------------------------------------------
356 # -------------------------------------------------------------------------
357 # *ACES* to *Linear*, *P3DCI* primaries
358 XYZ_to_P3DCI = [2.7253940305, -1.0180030062, -0.4401631952,
359 -0.7951680258, 1.6897320548, 0.0226471906,
360 0.0412418914, -0.0876390192, 1.1009293786]
362 cs = create_matrix_colorspace(
364 from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_P3DCI],
365 aliases=['lin_p3dci'])
366 colorspaces.append(cs)
368 # -------------------------------------------------------------------------
370 # -------------------------------------------------------------------------
371 # *ACES* to *Linear*, *Rec. 709* primaries.
372 # *sRGB* and *Rec 709* use the same gamut.
373 XYZ_to_Rec709 = [3.2409699419, -1.5373831776, -0.4986107603,
374 -0.9692436363, 1.8759675015, 0.0415550574,
375 0.0556300797, -0.2039769589, 1.0569715142]
377 cs = create_matrix_colorspace(
379 from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_Rec709],
380 aliases=['lin_srgb'])
381 colorspaces.append(cs)
383 # *Linear* to *sRGB* Transfer Function*
384 cs = create_transfer_colorspace(
387 transfer_function_sRGB_to_linear,
390 aliases=['crv_srgb'])
391 colorspaces.append(cs)
393 # *ACES* to *sRGB* Primaries + Transfer Function*
394 cs = create_matrix_plus_transfer_colorspace(
397 transfer_function_sRGB_to_linear,
400 from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_Rec709],
402 colorspaces.append(cs)
404 # -------------------------------------------------------------------------
406 # -------------------------------------------------------------------------
407 # *ACES* to *Linear*, *Rec. 709* primaries
408 XYZ_to_Rec709 = [3.2409699419, -1.5373831776, -0.4986107603,
409 -0.9692436363, 1.8759675015, 0.0415550574,
410 0.0556300797, -0.2039769589, 1.0569715142]
412 cs = create_matrix_colorspace(
414 from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_Rec709],
415 aliases=['lin_rec709'])
416 colorspaces.append(cs)
418 # *Linear* to *Rec. 709* Transfer Function*
419 cs = create_transfer_colorspace(
422 transfer_function_Rec709_to_linear,
425 aliases=['crv_rec709'])
426 colorspaces.append(cs)
428 # *ACES* to *Rec. 709* Primaries + Transfer Function*
429 cs = create_matrix_plus_transfer_colorspace(
432 transfer_function_Rec709_to_linear,
435 from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_Rec709],
436 aliases=['rec709_camera'])
437 colorspaces.append(cs)
439 # -------------------------------------------------------------------------
441 # -------------------------------------------------------------------------
442 # *ACES* to *Linear*, *Rec. 2020* primaries
443 XYZ_to_Rec2020 = [1.7166511880, -0.3556707838, -0.2533662814,
444 -0.6666843518, 1.6164812366, 0.0157685458,
445 0.0176398574, -0.0427706133, 0.9421031212]
447 cs = create_matrix_colorspace(
449 from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_Rec2020],
450 aliases=['lin_rec2020'])
451 colorspaces.append(cs)
453 # *Linear* to *Rec. 2020 10 bit* Transfer Function*
454 cs = create_transfer_colorspace(
457 transfer_function_Rec2020_10bit_to_linear,
460 aliases=['crv_rec2020'])
461 colorspaces.append(cs)
463 # *ACES* to *Rec. 2020 10 bit* Primaries + Transfer Function*
464 cs = create_matrix_plus_transfer_colorspace(
467 transfer_function_Rec2020_10bit_to_linear,
470 from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_Rec2020],
471 aliases=['rec2020_camera'])
472 colorspaces.append(cs)
474 # -------------------------------------------------------------------------
476 # -------------------------------------------------------------------------
477 # *Linear* to *Rec.1886* Transfer Function*
478 cs = create_transfer_colorspace(
481 transfer_function_Rec1886_to_linear,
484 aliases=['crv_rec1886'])
485 colorspaces.append(cs)
487 # *ACES* to *sRGB* Primaries + Transfer Function*
488 cs = create_matrix_plus_transfer_colorspace(
491 transfer_function_Rec1886_to_linear,
494 from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_Rec709],
495 aliases=['rec709_display'])
496 colorspaces.append(cs)
498 # *ACES* to *sRGB* Primaries + Transfer Function*
499 cs = create_matrix_plus_transfer_colorspace(
500 'Rec.2020 - Display',
502 transfer_function_Rec1886_to_linear,
505 from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_Rec2020],
506 aliases=['rec2020_display'])
507 colorspaces.append(cs)
509 # -------------------------------------------------------------------------
511 # -------------------------------------------------------------------------
512 # *ACES* to *Linear*, *Pro Photo* primaries
513 AP0_to_RIMM = [1.2412367771, -0.1685692287, -0.0726675484,
514 0.0061203066, 1.083151174, -0.0892714806,
515 -0.0032853314, 0.0099796402, 0.9933056912]
517 cs = create_matrix_colorspace(
518 'Linear - RIMM ROMM (ProPhoto)',
519 from_reference_values=[AP0_to_RIMM],
520 aliases=['lin_prophoto', 'lin_rimm'])
521 colorspaces.append(cs)
523 # -------------------------------------------------------------------------
525 # -------------------------------------------------------------------------
526 # *ACES* to *Linear*, *Adobe RGB* primaries
527 AP0_to_ADOBERGB = [1.7245603168, -0.4199935942, -0.3045667227,
528 -0.2764799142, 1.3727190877, -0.0962391734,
529 -0.0261255258, -0.0901747807, 1.1163003065]
531 cs = create_matrix_colorspace(
532 'Linear - Adobe RGB',
533 from_reference_values=[AP0_to_ADOBERGB],
534 aliases=['lin_adobergb'])
535 colorspaces.append(cs)
537 # -------------------------------------------------------------------------
538 # Adobe Wide Gamut RGB
539 # -------------------------------------------------------------------------
540 # *ACES* to *Linear*, *Adobe Wide Gamut RGB* primaries
541 AP0_to_ADOBERGB = [1.3809814778, -0.1158594573, -0.2651220205,
542 0.0057015535, 1.0402949043, -0.0459964578,
543 -0.0038908746, -0.0597091815, 1.0636000561]
545 cs = create_matrix_colorspace(
546 'Linear - Adobe Wide Gamut RGB',
547 from_reference_values=[AP0_to_ADOBERGB],
548 aliases=['lin_adobewidegamutrgb'])
549 colorspaces.append(cs)
555 # *Raw* utility space
557 raw = ColorSpace(name)
558 raw.description = 'The %s color space' % name
559 raw.aliases = ['raw']
560 raw.equality_group = name
561 raw.family = 'Utility'