-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
-
-"""
-Implements support for *RED* colorspaces conversions and transfer functions.
-"""
-
-import array
-
-import aces_ocio.generate_lut as genlut
-from aces_ocio.utilities import ColorSpace, mat44_from_mat33
-
-__author__ = 'ACES Developers'
-__copyright__ = 'Copyright (C) 2014 - 2015 - ACES Developers'
-__license__ = ''
-__maintainer__ = 'ACES Developers'
-__email__ = 'aces@oscars.org'
-__status__ = 'Production'
-
-__all__ = ['create_RED_log_film',
- 'create_colorspaces']
-
-
-def create_RED_log_film(gamut,
- transfer_function,
- name,
- lut_directory,
- lut_resolution_1d):
- """
- Object description.
-
- RED colorspaces to ACES.
-
- Parameters
- ----------
- parameter : type
- Parameter description.
-
- Returns
- -------
- type
- Return value description.
- """
-
- name = "%s - %s" % (transfer_function, gamut)
- if transfer_function == "":
- name = "Linear - %s" % gamut
- if gamut == "":
- name = "%s" % transfer_function
-
- cs = ColorSpace(name)
- cs.description = name
- cs.equality_group = ''
- cs.family = 'RED'
- cs.is_data = False
-
- def cineon_to_linear(code_value):
- n_gamma = 0.6
- black_point = 95.0
- white_point = 685.0
- code_value_to_density = 0.002
-
- black_linear = pow(10.0, (black_point - white_point) * (
- code_value_to_density / n_gamma))
- code_linear = pow(10.0, (code_value - white_point) * (
- code_value_to_density / n_gamma))
-
- return (code_linear - black_linear) / (1.0 - black_linear)
-
- cs.to_reference_transforms = []
-
- if transfer_function == 'REDlogFilm':
- data = array.array('f', "\0" * lut_resolution_1d * 4)
- for c in range(lut_resolution_1d):
- data[c] = cineon_to_linear(1023.0 * c / (lut_resolution_1d - 1))
-
- lut = "CineonLog_to_linear.spi1d"
- genlut.write_SPI_1d(lut_directory + "/" + lut,
- 0.0,
- 1.0,
- data,
- lut_resolution_1d,
- 1)
-
- cs.to_reference_transforms.append({
- 'type': 'lutFile',
- 'path': lut,
- 'interpolation': 'linear',
- 'direction': 'forward'})
-
- if gamut == 'DRAGONcolor':
- cs.to_reference_transforms.append({
- 'type': 'matrix',
- 'matrix': mat44_from_mat33([0.532279, 0.376648, 0.091073,
- 0.046344, 0.974513, -0.020860,
- -0.053976, -0.000320, 1.054267]),
- 'direction': 'forward'})
- elif gamut == 'DRAGONcolor2':
- cs.to_reference_transforms.append({
- 'type': 'matrix',
- 'matrix': mat44_from_mat33([0.468452, 0.331484, 0.200064,
- 0.040787, 0.857658, 0.101553,
- -0.047504, -0.000282, 1.047756]),
- 'direction': 'forward'})
- elif gamut == 'REDcolor2':
- cs.to_reference_transforms.append({
- 'type': 'matrix',
- 'matrix': mat44_from_mat33([0.480997, 0.402289, 0.116714,
- -0.004938, 1.000154, 0.004781,
- -0.105257, 0.025320, 1.079907]),
- 'direction': 'forward'})
- elif gamut == 'REDcolor3':
- cs.to_reference_transforms.append({
- 'type': 'matrix',
- 'matrix': mat44_from_mat33([0.512136, 0.360370, 0.127494,
- 0.070377, 0.903884, 0.025737,
- -0.020824, 0.017671, 1.003123]),
- 'direction': 'forward'})
- elif gamut == 'REDcolor4':
- cs.to_reference_transforms.append({
- 'type': 'matrix',
- 'matrix': mat44_from_mat33([0.474202, 0.333677, 0.192121,
- 0.065164, 0.836932, 0.097901,
- -0.019281, 0.016362, 1.002889]),
- 'direction': 'forward'})
-
- cs.from_reference_transforms = []
- return cs
-
-
-def create_colorspaces(lut_directory, lut_resolution_1d):
- """
- Generates the colorspace conversions.
-
- Parameters
- ----------
- parameter : type
- Parameter description.
-
- Returns
- -------
- type
- Return value description.
- """
-
- colorspaces = []
-
- # Full conversion
- RED_log_film_dragon = create_RED_log_film(
- "DRAGONcolor",
- "REDlogFilm",
- "REDlogFilm",
- lut_directory,
- lut_resolution_1d)
- colorspaces.append(RED_log_film_dragon)
-
- RED_log_film_dragon2 = create_RED_log_film(
- "DRAGONcolor2",
- "REDlogFilm",
- "REDlogFilm",
- lut_directory,
- lut_resolution_1d)
- colorspaces.append(RED_log_film_dragon2)
-
- RED_log_film_color2 = create_RED_log_film(
- "REDcolor2",
- "REDlogFilm",
- "REDlogFilm",
- lut_directory,
- lut_resolution_1d)
- colorspaces.append(RED_log_film_color2)
-
- RED_log_film_color3 = create_RED_log_film(
- "REDcolor3",
- "REDlogFilm",
- "REDlogFilm",
- lut_directory,
- lut_resolution_1d)
- colorspaces.append(RED_log_film_color3)
-
- RED_log_film_color4 = create_RED_log_film(
- "REDcolor4",
- "REDlogFilm",
- "REDlogFilm",
- lut_directory,
- lut_resolution_1d)
- colorspaces.append(RED_log_film_color4)
-
- # Linearization only
- RED_log_film_dragon = create_RED_log_film(
- "",
- "REDlogFilm",
- "REDlogFilm",
- lut_directory,
- lut_resolution_1d)
- colorspaces.append(RED_log_film_dragon)
-
- # Primaries only
- RED_log_film_dragon = create_RED_log_film(
- "DRAGONcolor",
- "",
- "REDlogFilm",
- lut_directory,
- lut_resolution_1d)
- colorspaces.append(RED_log_film_dragon)
-
- RED_log_film_dragon2 = create_RED_log_film(
- "DRAGONcolor2",
- "",
- "REDlogFilm",
- lut_directory,
- lut_resolution_1d)
- colorspaces.append(RED_log_film_dragon2)
-
- RED_log_film_color2 = create_RED_log_film(
- "REDcolor2",
- "",
- "REDlogFilm",
- lut_directory,
- lut_resolution_1d)
- colorspaces.append(RED_log_film_color2)
-
- RED_log_film_color3 = create_RED_log_film(
- "REDcolor3",
- "",
- "REDlogFilm",
- lut_directory,
- lut_resolution_1d)
- colorspaces.append(RED_log_film_color3)
-
- RED_log_film_color4 = create_RED_log_film(
- "REDcolor4",
- "",
- "REDlogFilm",
- lut_directory,
- lut_resolution_1d)
- colorspaces.append(RED_log_film_color4)
-
- return colorspaces