2 # -*- coding: utf-8 -*-
6 import aces_ocio.generateLUT as genlut
7 from aces_ocio.util import ColorSpace, mat44FromMat33
9 __author__ = 'ACES Developers'
10 __copyright__ = 'Copyright (C) 2014 - 2015 - ACES Developers'
12 __maintainer__ = 'ACES Developers'
13 __email__ = 'aces@oscars.org'
14 __status__ = 'Production'
16 __all__ = ['createREDlogFilm',
20 # RED color spaces to ACES
22 def createREDlogFilm(gamut, transferFunction, name, lutDir, lutResolution1d):
23 name = "%s - %s" % (transferFunction, gamut)
24 if transferFunction == "":
25 name = "Linear - %s" % gamut
27 name = "%s" % transferFunction
35 def cineonToLinear(codeValue):
39 codeValueToDensity = 0.002
41 blackLinear = pow(10.0, (blackPoint - whitePoint) * (
42 codeValueToDensity / nGamma))
43 codeLinear = pow(10.0, (codeValue - whitePoint) * (
44 codeValueToDensity / nGamma))
46 return (codeLinear - blackLinear) / (1.0 - blackLinear)
48 cs.toReferenceTransforms = []
50 if transferFunction == 'REDlogFilm':
51 data = array.array('f', "\0" * lutResolution1d * 4)
52 for c in range(lutResolution1d):
53 data[c] = cineonToLinear(1023.0 * c / (lutResolution1d - 1))
55 lut = "CineonLog_to_linear.spi1d"
56 genlut.writeSPI1D(lutDir + "/" + lut,
63 cs.toReferenceTransforms.append({
66 'interpolation': 'linear',
67 'direction': 'forward'
70 if gamut == 'DRAGONcolor':
71 cs.toReferenceTransforms.append({
73 'matrix': mat44FromMat33([0.532279, 0.376648, 0.091073,
74 0.046344, 0.974513, -0.020860,
75 -0.053976, -0.000320, 1.054267]),
76 'direction': 'forward'
78 elif gamut == 'DRAGONcolor2':
79 cs.toReferenceTransforms.append({
81 'matrix': mat44FromMat33([0.468452, 0.331484, 0.200064,
82 0.040787, 0.857658, 0.101553,
83 -0.047504, -0.000282, 1.047756]),
84 'direction': 'forward'
86 elif gamut == 'REDcolor2':
87 cs.toReferenceTransforms.append({
89 'matrix': mat44FromMat33([0.480997, 0.402289, 0.116714,
90 -0.004938, 1.000154, 0.004781,
91 -0.105257, 0.025320, 1.079907]),
92 'direction': 'forward'
94 elif gamut == 'REDcolor3':
95 cs.toReferenceTransforms.append({
97 'matrix': mat44FromMat33([0.512136, 0.360370, 0.127494,
98 0.070377, 0.903884, 0.025737,
99 -0.020824, 0.017671, 1.003123]),
100 'direction': 'forward'
102 elif gamut == 'REDcolor4':
103 cs.toReferenceTransforms.append({
105 'matrix': mat44FromMat33([0.474202, 0.333677, 0.192121,
106 0.065164, 0.836932, 0.097901,
107 -0.019281, 0.016362, 1.002889]),
108 'direction': 'forward'
111 cs.fromReferenceTransforms = []
115 # Generate all color spaces conversion
116 def createColorSpaces(lutDir, lutResolution1d):
120 REDlogFilmDRAGON = createREDlogFilm(
121 "DRAGONcolor", "REDlogFilm", "REDlogFilm", lutDir, lutResolution1d)
122 colorspaces.append(REDlogFilmDRAGON)
124 REDlogFilmDRAGON2 = createREDlogFilm(
125 "DRAGONcolor2", "REDlogFilm", "REDlogFilm", lutDir, lutResolution1d)
126 colorspaces.append(REDlogFilmDRAGON2)
128 REDlogFilmREDcolor2 = createREDlogFilm(
129 "REDcolor2", "REDlogFilm", "REDlogFilm", lutDir, lutResolution1d)
130 colorspaces.append(REDlogFilmREDcolor2)
132 REDlogFilmREDcolor3 = createREDlogFilm(
133 "REDcolor3", "REDlogFilm", "REDlogFilm", lutDir, lutResolution1d)
134 colorspaces.append(REDlogFilmREDcolor3)
136 REDlogFilmREDcolor4 = createREDlogFilm(
137 "REDcolor4", "REDlogFilm", "REDlogFilm", lutDir, lutResolution1d)
138 colorspaces.append(REDlogFilmREDcolor4)
141 REDlogFilmDRAGON = createREDlogFilm(
142 "", "REDlogFilm", "REDlogFilm", lutDir, lutResolution1d)
143 colorspaces.append(REDlogFilmDRAGON)
146 REDlogFilmDRAGON = createREDlogFilm(
147 "DRAGONcolor", "", "REDlogFilm", lutDir, lutResolution1d)
148 colorspaces.append(REDlogFilmDRAGON)
150 REDlogFilmDRAGON2 = createREDlogFilm(
151 "DRAGONcolor2", "", "REDlogFilm", lutDir, lutResolution1d)
152 colorspaces.append(REDlogFilmDRAGON2)
154 REDlogFilmREDcolor2 = createREDlogFilm(
155 "REDcolor2", "", "REDlogFilm", lutDir, lutResolution1d)
156 colorspaces.append(REDlogFilmREDcolor2)
158 REDlogFilmREDcolor3 = createREDlogFilm(
159 "REDcolor3", "", "REDlogFilm", lutDir, lutResolution1d)
160 colorspaces.append(REDlogFilmREDcolor3)
162 REDlogFilmREDcolor4 = createREDlogFilm(
163 "REDcolor4", "", "REDlogFilm", lutDir, lutResolution1d)
164 colorspaces.append(REDlogFilmREDcolor4)