Updated to ACES 1.0.1

[OpenColorIO-Configs.git] / aces_1.0.1 / python / aces_ocio / colorspaces / panasonic.py

diff --git a/aces_1.0.1/python/aces_ocio/colorspaces/panasonic.py b/aces_1.0.1/python/aces_ocio/colorspaces/panasonic.py
new file mode 100644 (file)
index 0000000..9e5e094
--- /dev/null
+++ b/aces_1.0.1/python/aces_ocio/colorspaces/panasonic.py
@@ -0,0 +1,167 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+Implements support for *Panasonic* colorspaces conversions and transfer
+functions.
+"""
+
+import array
+import os
+
+import PyOpenColorIO as ocio
+
+import aces_ocio.generate_lut as genlut
+from aces_ocio.utilities import ColorSpace
+
+__author__ = 'ACES Developers'
+__copyright__ = 'Copyright (C) 2014 - 2015 - ACES Developers'
+__license__ = ''
+__maintainer__ = 'ACES Developers'
+__email__ = 'aces@oscars.org'
+__status__ = 'Production'
+
+__all__ = ['create_v_log',
+           'create_colorspaces']
+
+
+def create_v_log(gamut,
+                 transfer_function,
+                 lut_directory,
+                 lut_resolution_1d,
+                 aliases):
+    """
+    Creates colorspace covering the conversion from VLog to ACES, with various transfer 
+    functions and encoding gamuts covered
+
+    Parameters
+    ----------
+    gamut : str
+        The name of the encoding gamut to use.
+    transfer_function : str
+        The name of the transfer function to use
+    lut_directory : str or unicode 
+        The directory to use when generating LUTs
+    lut_resolution_1d : int
+        The resolution of generated 1D LUTs
+    aliases : list of str
+        Aliases for this colorspace
+
+    Returns
+    -------
+    ColorSpace
+         A ColorSpace container class referencing the LUTs, matrices and identifying
+         information for the requested colorspace.
+    """
+
+    name = '%s - %s' % (transfer_function, gamut)
+    if transfer_function == '':
+        name = 'Linear - %s' % gamut
+    if gamut == '':
+        name = 'Curve - %s' % transfer_function
+
+    cs = ColorSpace(name)
+    cs.description = name
+    cs.aliases = aliases
+    cs.equality_group = ''
+    cs.family = 'Input/Panasonic'
+    cs.is_data = False
+
+    # A linear space needs allocation variables
+    if transfer_function == '':
+        cs.allocation_type = ocio.Constants.ALLOCATION_LG2
+        cs.allocation_vars = [-8, 5, 0.00390625]
+
+    def v_log_to_linear(x):
+        cut_inv = 0.181
+        b = 0.00873
+        c = 0.241514
+        d = 0.598206
+
+        if x <= cut_inv:
+            return (x - 0.125) / 5.6
+        else:
+            return pow(10, (x - d) / c) - b
+
+    cs.to_reference_transforms = []
+
+    if transfer_function == 'V-Log':
+        data = array.array('f', '\0' * lut_resolution_1d * 4)
+        for c in range(lut_resolution_1d):
+            data[c] = v_log_to_linear(float(c) / (lut_resolution_1d - 1))
+
+        lut = '%s_to_linear.spi1d' % transfer_function
+        genlut.write_SPI_1d(
+            os.path.join(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 == 'V-Gamut':
+        cs.to_reference_transforms.append({
+            'type': 'matrix',
+            'matrix': [0.724382758, 0.166748484, 0.108497411, 0.0,
+                       0.021354009, 0.985138372, -0.006319092, 0.0,
+                       -0.009234278, -0.00104295, 1.010272625, 0.0,
+                       0, 0, 0, 1.0],
+            'direction': 'forward'})
+
+    cs.from_reference_transforms = []
+    return cs
+
+
+def create_colorspaces(lut_directory, lut_resolution_1d):
+    """
+    Generates the colorspace conversions.
+
+    Parameters
+    ----------
+    lut_directory : str or unicode 
+        The directory to use when generating LUTs
+    lut_resolution_1d : int
+        The resolution of generated 1D LUTs
+
+    Returns
+    -------
+    list
+         A list of colorspaces for Panasonic cameras and encodings 
+    """
+
+    colorspaces = []
+
+    # Full conversion
+    v_log_1 = create_v_log(
+        'V-Gamut',
+        'V-Log',
+        lut_directory,
+        lut_resolution_1d,
+        ['vlog_vgamut'])
+    colorspaces.append(v_log_1)
+
+    # Linearization Only
+    v_log_2 = create_v_log(
+        '',
+        'V-Log',
+        lut_directory,
+        lut_resolution_1d,
+        ['crv_vlog'])
+    colorspaces.append(v_log_2)
+
+    # Primaries Only
+    v_log_3 = create_v_log(
+        'V-Gamut',
+        '',
+        lut_directory,
+        lut_resolution_1d,
+        ['lin_vgamut'])
+    colorspaces.append(v_log_3)
+
+    return colorspaces