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 aces_ocio.create_aces_colorspaces as aces
   9 from aces_ocio.utilities import ColorSpace, mat44_from_mat33
  10
  11
  12 __author__ = 'ACES Developers'
  13 __copyright__ = 'Copyright (C) 2014 - 2015 - ACES Developers'
  14 __license__ = ''
  15 __maintainer__ = 'ACES Developers'
  16 __email__ = 'aces@oscars.org'
  17 __status__ = 'Production'
  18
  19 __all__ = ['create_generic_matrix',
  20            'create_colorspaces']
  21
  22
  23 def create_generic_matrix(name='matrix',
  24                           from_reference_values=None,
  25                           to_reference_values=None,
  26                           aliases=[]):
  27     """
  28     Object description.
  29
  30     Parameters
  31     ----------
  32     parameter : type
  33         Parameter description.
  34
  35     Returns
  36     -------
  37     type
  38          Return value description.
  39     """
  40
  41     if from_reference_values is None:
  42         from_reference_values = []
  43
  44     if to_reference_values is None:
  45         to_reference_values = []
  46
  47     cs = ColorSpace(name)
  48     cs.description = 'The %s color space' % name
  49     cs.aliases = aliases
  50     cs.equality_group = name
  51     cs.family = 'Utility'
  52     cs.is_data = False
  53
  54     cs.to_reference_transforms = []
  55     if to_reference_values:
  56         for matrix in to_reference_values:
  57             cs.to_reference_transforms.append({
  58                 'type': 'matrix',
  59                 'matrix': mat44_from_mat33(matrix),
  60                 'direction': 'forward'})
  61
  62     cs.from_reference_transforms = []
  63     if from_reference_values:
  64         for matrix in from_reference_values:
  65             cs.from_reference_transforms.append({
  66                 'type': 'matrix',
  67                 'matrix': mat44_from_mat33(matrix),
  68                 'direction': 'forward'})
  69
  70     return cs
  71
  72
  73 def create_colorspaces(lut_directory,
  74                        lut_resolution_1d,
  75                        lut_resolution_3d):
  76     """
  77     Generates the colorspace conversions.
  78
  79     Parameters
  80     ----------
  81     parameter : type
  82         Parameter description.
  83
  84     Returns
  85     -------
  86     type
  87          Return value description.
  88     """
  89
  90     colorspaces = []
  91
  92     cs = create_generic_matrix('XYZ',
  93                                from_reference_values=[aces.ACES_AP0_TO_XYZ],
  94                                aliases=["lin_xyz"])
  95     colorspaces.append(cs)
  96
  97     cs = create_generic_matrix(
  98         'Linear - AP1',
  99         to_reference_values=[aces.ACES_AP1_TO_AP0],
 100         aliases=["lin_ap1"])
 101     colorspaces.append(cs)
 102
 103     # *ACES* to *Linear*, *P3D60* primaries.
 104     XYZ_to_P3D60 = [2.4027414142, -0.8974841639, -0.3880533700,
 105                     -0.8325796487, 1.7692317536, 0.0237127115,
 106                     0.0388233815, -0.0824996856, 1.0363685997]
 107
 108     cs = create_generic_matrix(
 109         'Linear - P3-D60',
 110         from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_P3D60],
 111         aliases=["lin_p3d60"])
 112     colorspaces.append(cs)
 113
 114     # *ACES* to *Linear*, *P3DCI* primaries.
 115     XYZ_to_P3DCI = [2.7253940305, -1.0180030062, -0.4401631952,
 116                     -0.7951680258, 1.6897320548, 0.0226471906,
 117                     0.0412418914, -0.0876390192, 1.1009293786]
 118
 119     cs = create_generic_matrix(
 120         'Linear - P3-DCI',
 121         from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_P3DCI],
 122         aliases=["lin_p3dci"])
 123     colorspaces.append(cs)
 124
 125     # *ACES* to *Linear*, *Rec. 709* primaries.
 126     XYZ_to_Rec709 = [3.2409699419, -1.5373831776, -0.4986107603,
 127                      -0.9692436363, 1.8759675015, 0.0415550574,
 128                      0.0556300797, -0.2039769589, 1.0569715142]
 129
 130     cs = create_generic_matrix(
 131         'Linear - Rec.709',
 132         from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_Rec709],
 133         aliases=["lin_rec709"])
 134     colorspaces.append(cs)
 135
 136     # *ACES* to *Linear*, *Rec. 2020* primaries.
 137     XYZ_to_Rec2020 = [1.7166511880, -0.3556707838, -0.2533662814,
 138                       -0.6666843518, 1.6164812366, 0.0157685458,
 139                       0.0176398574, -0.0427706133, 0.9421031212]
 140
 141     cs = create_generic_matrix(
 142         'Linear - Rec.2020',
 143         from_reference_values=[aces.ACES_AP0_TO_XYZ, XYZ_to_Rec2020],
 144         aliases=["lin_rec2020"])
 145     colorspaces.append(cs)
 146
 147     # *ACES* to *Linear*, *Pro Photo* primaries.
 148     AP0_to_RIMM = [1.2412367771, -0.1685692287, -0.0726675484,
 149                    0.0061203066, 1.083151174, -0.0892714806,
 150                    -0.0032853314, 0.0099796402, 0.9933056912]
 151
 152     cs = create_generic_matrix(
 153         'Linear - ProPhoto',
 154         from_reference_values=[AP0_to_RIMM],
 155         aliases=["lin_prophoto", "lin_rimm"])
 156     colorspaces.append(cs)
 157
 158     # *ACES* to *Linear*, *Adobe RGB* primaries.
 159     AP0_to_ADOBERGB = [1.7245603168, -0.4199935942, -0.3045667227,
 160                        -0.2764799142, 1.3727190877, -0.0962391734,
 161                        -0.0261255258, -0.0901747807, 1.1163003065]
 162
 163     cs = create_generic_matrix(
 164         'Linear - Adobe RGB',
 165         from_reference_values=[AP0_to_ADOBERGB],
 166         aliases=["lin_adobergb"])
 167     colorspaces.append(cs)
 168
 169     # *ACES* to *Linear*, *Adobe Wide Gamut RGB* primaries.
 170     AP0_to_ADOBERGB = [1.3809814778, -0.1158594573, -0.2651220205,
 171                        0.0057015535, 1.0402949043, -0.0459964578,
 172                        -0.0038908746, -0.0597091815, 1.0636000561]
 173
 174     cs = create_generic_matrix(
 175         'Linear - Adobe Wide Gamut RGB',
 176         from_reference_values=[AP0_to_ADOBERGB],
 177         aliases=["lin_adobewidegamutrgb"])
 178     colorspaces.append(cs)
 179
 180     return colorspaces