aces_1.0.0/python/aces_ocio/create_general_colorspaces.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3
   4 """
   5 Implements support for general colorspaces conversions and transfer functions.
   6 """
   7
   8 import array
   9 import math
  10 import os
  11
  12 import aces_ocio.generate_lut as genlut
  13 import aces_ocio.create_aces_colorspaces as aces
  14 from aces_ocio.utilities import ColorSpace, mat44_from_mat33, sanitize_path, compact
  15
  16
  17 __author__ = 'ACES Developers'
  18 __copyright__ = 'Copyright (C) 2014 - 2015 - ACES Developers'
  19 __license__ = ''
  20 __maintainer__ = 'ACES Developers'
  21 __email__ = 'aces@oscars.org'
  22 __status__ = 'Production'
  23
  24 __all__ = ['create_generic_matrix',
  25            'create_colorspaces']
  26
  27 # -------------------------------------------------------------------------
  28 # Generic Matrix transform
  29 # -------------------------------------------------------------------------
  30 def create_generic_matrix(name='matrix',
  31                           from_reference_values=None,
  32                           to_reference_values=None,
  33                           aliases=[]):
  34
  35     if from_reference_values is None:
  36          from_reference_values = []
  37     if to_reference_values is None:
  38          to_reference_values = []
  39
  40     cs = ColorSpace(name)
  41     cs.description = 'The %s color space' % name
  42     cs.aliases = []
  43     cs.equality_group = name
  44     cs.family = 'Utility'
  45     cs.is_data = False
  46
  47     cs.to_reference_transforms = []
  48     if to_reference_values:
  49         for matrix in to_reference_values:
  50             cs.to_reference_transforms.append({
  51                 'type': 'matrix',
  52                 'matrix': mat44_from_mat33(matrix),
  53                 'direction': 'forward'})
  54
  55     cs.from_reference_transforms = []
  56     if from_reference_values:
  57         for matrix in from_reference_values:
  58             cs.from_reference_transforms.append({
  59                 'type': 'matrix',
  60                 'matrix': mat44_from_mat33(matrix),
  61                 'direction': 'forward'})
  62
  63     return cs
  64
  65 def create_colorspaces(lut_directory,
  66                        lut_resolution_1d,
  67                        lut_resolution_3d):
  68
  69     """
  70     Generates the colorspace conversions.
  71
  72     Parameters
  73     ----------
  74     parameter : type
  75         Parameter description.
  76
  77     Returns
  78     -------
  79     type
  80          Return value description.
  81     """
  82
  83     colorspaces = []
  84
  85     cs = create_generic_matrix('XYZ',
  86         from_reference_values=[aces.ACES_AP0_to_XYZ],
  87         aliases=["lin_xyz"])
  88     colorspaces.append(cs)
  89
  90     cs = create_generic_matrix(
  91         'Linear - AP1',
  92         to_reference_values=[aces.ACES_AP1_to_AP0],
  93         aliases=["lin_ap1"])
  94     colorspaces.append(cs)
  95
  96     # *ACES* to *Linear*, *P3D60* primaries.
  97     XYZ_to_P3D60 = [2.4027414142, -0.8974841639, -0.3880533700,
  98                     -0.8325796487, 1.7692317536, 0.0237127115,
  99                     0.0388233815, -0.0824996856, 1.0363685997]
 100
 101     cs = create_generic_matrix(
 102         'Linear - P3-D60',
 103         from_reference_values=[aces.ACES_AP0_to_XYZ, XYZ_to_P3D60],
 104         aliases=["lin_p3d60"])
 105     colorspaces.append(cs)
 106
 107     # *ACES* to *Linear*, *P3DCI* primaries.
 108     XYZ_to_P3DCI = [2.7253940305, -1.0180030062, -0.4401631952,
 109                     -0.7951680258, 1.6897320548, 0.0226471906,
 110                     0.0412418914, -0.0876390192, 1.1009293786]
 111
 112     cs = create_generic_matrix(
 113         'Linear - P3-DCI',
 114         from_reference_values=[aces.ACES_AP0_to_XYZ, XYZ_to_P3DCI],
 115         aliases=["lin_p3dci"])
 116     colorspaces.append(cs)
 117
 118     # *ACES* to *Linear*, *Rec. 709* primaries.
 119     XYZ_to_Rec709 = [3.2409699419, -1.5373831776, -0.4986107603,
 120                      -0.9692436363, 1.8759675015, 0.0415550574,
 121                      0.0556300797, -0.2039769589, 1.0569715142]
 122
 123     cs = create_generic_matrix(
 124         'Linear - Rec.709',
 125         from_reference_values=[aces.ACES_AP0_to_XYZ, XYZ_to_Rec709],
 126         aliases=["lin_rec709"])
 127     colorspaces.append(cs)
 128
 129     # *ACES* to *Linear*, *Rec. 2020* primaries.
 130     XYZ_to_Rec2020 = [1.7166511880, -0.3556707838, -0.2533662814,
 131                       -0.6666843518, 1.6164812366, 0.0157685458,
 132                       0.0176398574, -0.0427706133, 0.9421031212]
 133
 134     cs = create_generic_matrix(
 135         'Linear - Rec.2020',
 136         from_reference_values=[aces.ACES_AP0_to_XYZ, XYZ_to_Rec2020],
 137         aliases=["lin_rec2020"])
 138     colorspaces.append(cs)
 139
 140     return colorspaces