X-Git-Url: http://users.mur.at/ms/git/gitweb/?p=OpenColorIO-Configs.git;a=blobdiff_plain;f=aces_1.0.0%2Fpython%2Faces_ocio%2Fgenerate_lut.py;h=959e8a4f8e45349c18a24c0a111111db6cc0e4a6;hp=605963651302faa5e9ed0ee8e54385e7e8f9bf8f;hb=cad9d48a0f1067769435904348e9fffeffd25eb9;hpb=1ef6829d081edc2cdb9e7610f33ae52a0a7f0ee7 diff --git a/aces_1.0.0/python/aces_ocio/generate_lut.py b/aces_1.0.0/python/aces_ocio/generate_lut.py index 6059636..959e8a4 100755 --- a/aces_1.0.0/python/aces_ocio/generate_lut.py +++ b/aces_1.0.0/python/aces_ocio/generate_lut.py @@ -78,12 +78,12 @@ def generate_1d_LUT_image(ramp_1d_path, ramp.close() -def write_SPI_1d(filename, - from_min, - from_max, - data, - entries, - channels, +def write_SPI_1d(filename, + from_min, + from_max, + data, + entries, + channels, components=3): """ Object description. @@ -120,12 +120,12 @@ def write_SPI_1d(filename, fp.write('}\n') -def write_CSP_1d(filename, - from_min, - from_max, - data, - entries, - channels, +def write_CSP_1d(filename, + from_min, + from_max, + data, + entries, + channels, components=3): """ Object description. @@ -149,8 +149,8 @@ def write_CSP_1d(filename, fp.write('CSPLUTV100\n') fp.write('1D\n') fp.write('\n') - fp.write('BEGIN METADATA') - fp.write('END METADATA') + fp.write('BEGIN METADATA\n') + fp.write('END METADATA\n') fp.write('\n') @@ -168,25 +168,113 @@ def write_CSP_1d(filename, fp.write('%d\n' % entries) if components == 1: - for i in range(0, entries): - entry = '' - for j in range(3): - entry = '%s %s' % (entry, data[i * channels]) - fp.write('%s\n' % entry) + for i in range(0, entries): + entry = '' + for j in range(3): + entry = '%s %s' % (entry, data[i * channels]) + fp.write('%s\n' % entry) else: - for i in range(entries): - entry = '' - for j in range(components): - entry = '%s %s' % (entry, data[i * channels + j]) - fp.write('%s\n' % entry) + for i in range(entries): + entry = '' + for j in range(components): + entry = '%s %s' % (entry, data[i * channels + j]) + fp.write('%s\n' % entry) fp.write('\n') -def write_1d(filename, - from_min, - from_max, - data, - data_entries, - data_channels, + +def write_CTL_1d(filename, + from_min, + from_max, + data, + entries, + channels, + components=3): + """ + Object description. + + Parameters + ---------- + parameter : type + Parameter description. + + Returns + ------- + type + Return value description. + """ + + # May want to use fewer components than there are channels in the data + # Most commonly used for single channel LUTs + components = min(3, components, channels) + + with open(filename, 'w') as fp: + fp.write('// %d x %d LUT generated by "generate_lut"\n' % ( + entries, components)) + fp.write('\n') + fp.write('const float min1d = %3.9f;\n' % from_min) + fp.write('const float max1d = %3.9f;\n' % from_max) + fp.write('\n') + + # Write LUT + if components == 1: + fp.write('const float lut[] = {\n') + for i in range(0, entries): + fp.write('%s' % data[i * channels]) + if i != (entries - 1): + fp.write(',') + fp.write('\n') + fp.write('};\n') + fp.write('\n') + else: + for j in range(components): + fp.write('const float lut%d[] = {\n' % j) + for i in range(0, entries): + fp.write('%s' % data[i * channels]) + if i != (entries - 1): + fp.write(',') + fp.write('\n') + fp.write('};\n') + fp.write('\n') + + fp.write('void main\n') + fp.write('(\n') + fp.write(' input varying float rIn,\n') + fp.write(' input varying float gIn,\n') + fp.write(' input varying float bIn,\n') + fp.write(' input varying float aIn,\n') + fp.write(' output varying float rOut,\n') + fp.write(' output varying float gOut,\n') + fp.write(' output varying float bOut,\n') + fp.write(' output varying float aOut\n') + fp.write(')\n') + fp.write('{\n') + fp.write(' float r = rIn;\n') + fp.write(' float g = gIn;\n') + fp.write(' float b = bIn;\n') + fp.write('\n') + fp.write(' // Apply LUT\n') + if components == 1: + fp.write(' r = lookup1D(lut, min1d, max1d, r);\n') + fp.write(' g = lookup1D(lut, min1d, max1d, g);\n') + fp.write(' b = lookup1D(lut, min1d, max1d, b);\n') + elif components == 3: + fp.write(' r = lookup1D(lut0, min1d, max1d, r);\n') + fp.write(' g = lookup1D(lut1, min1d, max1d, g);\n') + fp.write(' b = lookup1D(lut2, min1d, max1d, b);\n') + fp.write('\n') + fp.write(' rOut = r;\n') + fp.write(' gOut = g;\n') + fp.write(' bOut = b;\n') + fp.write(' aOut = aIn;\n') + fp.write('}\n') + + +def write_1d(filename, + from_min, + from_max, + data, + data_entries, + data_channels, lut_components=3, format='spi1d'): """ @@ -203,29 +291,39 @@ def write_1d(filename, Return value description. """ - ocioFormatsToExtensions = {'cinespace' : 'csp', - 'flame' : '3dl', - 'icc' : 'icc', - 'houdini' : 'lut', - 'lustre' : '3dl'} + ocioFormatsToExtensions = {'cinespace': 'csp', + 'flame': '3dl', + 'icc': 'icc', + 'houdini': 'lut', + 'lustre': '3dl', + 'ctl': 'ctl'} if format in ocioFormatsToExtensions: - if ocioFormatsToExtensions[format] == 'csp': - write_CSP_1d(filename, + if ocioFormatsToExtensions[format] == 'csp': + write_CSP_1d(filename, + from_min, + from_max, + data, + data_entries, + data_channels, + lut_components) + elif ocioFormatsToExtensions[format] == 'ctl': + write_CTL_1d(filename, + from_min, + from_max, + data, + data_entries, + data_channels, + lut_components) + else: + write_SPI_1d(filename, from_min, from_max, data, data_entries, data_channels, lut_components) - else: - write_SPI_1d(filename, - from_min, - from_max, - data, - data_entries, - data_channels, - lut_components) + def generate_1d_LUT_from_image(ramp_1d_path, output_path=None, @@ -261,8 +359,8 @@ def generate_1d_LUT_from_image(ramp_1d_path, type = oiio.FLOAT ramp_data = ramp.read_image(type) - write_1d(output_path, min_value, max_value, - ramp_data, ramp_width, ramp_channels, channels, format) + write_1d(output_path, min_value, max_value, + ramp_data, ramp_width, ramp_channels, channels, format) def generate_3d_LUT_image(ramp_3d_path, resolution=32): @@ -293,8 +391,8 @@ def generate_3d_LUT_image(ramp_3d_path, resolution=32): lut_extract.execute() -def generate_3d_LUT_from_image(ramp_3d_path, - output_path=None, +def generate_3d_LUT_from_image(ramp_3d_path, + output_path=None, resolution=32, format='spi3d'): """ @@ -314,56 +412,56 @@ def generate_3d_LUT_from_image(ramp_3d_path, if output_path is None: output_path = '%s.%s' % (ramp_3d_path, 'spi3d') - ocioFormatsToExtensions = {'cinespace' : 'csp', - 'flame' : '3dl', - 'icc' : 'icc', - 'houdini' : 'lut', - 'lustre' : '3dl'} + ocioFormatsToExtensions = {'cinespace': 'csp', + 'flame': '3dl', + 'icc': 'icc', + 'houdini': 'lut', + 'lustre': '3dl'} if format == 'spi3d' or not (format in ocioFormatsToExtensions): - # Extract a spi3d LUT - args = ['--extract', - '--cubesize', - str(resolution), - '--maxwidth', - str(resolution * resolution), - '--input', - ramp_3d_path, - '--output', - output_path] - lut_extract = Process(description='extract a 3d LUT', - cmd='ociolutimage', - args=args) - lut_extract.execute() + # Extract a spi3d LUT + args = ['--extract', + '--cubesize', + str(resolution), + '--maxwidth', + str(resolution * resolution), + '--input', + ramp_3d_path, + '--output', + output_path] + lut_extract = Process(description='extract a 3d LUT', + cmd='ociolutimage', + args=args) + lut_extract.execute() else: - output_path_spi3d = '%s.%s' % (output_path, 'spi3d') - - # Extract a spi3d LUT - args = ['--extract', - '--cubesize', - str(resolution), - '--maxwidth', - str(resolution * resolution), - '--input', - ramp_3d_path, - '--output', - output_path_spi3d] - lut_extract = Process(description='extract a 3d LUT', - cmd='ociolutimage', - args=args) - lut_extract.execute() - - # Convert to a different format - args = ['--lut', - output_path_spi3d, - '--format', - format, - output_path] - lut_convert = Process(description='convert a 3d LUT', - cmd='ociobakelut', - args=args) - lut_convert.execute() + output_path_spi3d = '%s.%s' % (output_path, 'spi3d') + + # Extract a spi3d LUT + args = ['--extract', + '--cubesize', + str(resolution), + '--maxwidth', + str(resolution * resolution), + '--input', + ramp_3d_path, + '--output', + output_path_spi3d] + lut_extract = Process(description='extract a 3d LUT', + cmd='ociolutimage', + args=args) + lut_extract.execute() + + # Convert to a different format + args = ['--lut', + output_path_spi3d, + '--format', + format, + output_path] + lut_convert = Process(description='convert a 3d LUT', + cmd='ociobakelut', + args=args) + lut_convert.execute() def apply_CTL_to_image(input_image, @@ -644,9 +742,9 @@ def generate_3d_LUT_from_CTL(lut_path, corrected_LUT_image, lut_resolution) - generate_3d_LUT_from_image(corrected_LUT_image, - lut_path, - lut_resolution, + generate_3d_LUT_from_image(corrected_LUT_image, + lut_path, + lut_resolution, format) if cleanup: @@ -660,6 +758,7 @@ def generate_3d_LUT_from_CTL(lut_path, lut_path_spi3d = '%s.%s' % (lut_path, 'spi3d') os.remove(lut_path_spi3d) + def main(): """ Object description. @@ -781,4 +880,3 @@ def main(): if __name__ == '__main__': main() -