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%2Faces_config.py;h=3cc5727e5afa06db5f06b10103114ea66554ade0;hp=a63acf718abf325374913dd88507b551cacb7bb0;hb=7ed859e03c3b3dfb0732af17fb81524f9a5a1ea7;hpb=9404c06b163b18672794ced12638b6063fc2b8dd diff --git a/aces_1.0.0/python/aces_ocio/aces_config.py b/aces_1.0.0/python/aces_ocio/aces_config.py index a63acf7..3cc5727 100755 --- a/aces_1.0.0/python/aces_ocio/aces_config.py +++ b/aces_1.0.0/python/aces_ocio/aces_config.py @@ -7,7 +7,9 @@ Defines objects creating the *ACES* configuration. from __future__ import division +import copy import os +import shutil import sys import PyOpenColorIO as ocio @@ -21,6 +23,7 @@ from aces_ocio.colorspaces import red from aces_ocio.colorspaces import sony from aces_ocio.process import Process +from aces_ocio.utilities import replace, ColorSpace, compact __author__ = 'ACES Developers' __copyright__ = 'Copyright (C) 2014 - 2015 - ACES Developers' @@ -55,7 +58,9 @@ def set_config_default_roles(config, matte_paint='', reference='', scene_linear='', - texture_paint=''): + texture_paint='', + rendering='', + compositing_linear=''): """ Sets given *OCIO* configuration default roles. @@ -102,11 +107,22 @@ def set_config_default_roles(config, config.setRole(ocio.Constants.ROLE_MATTE_PAINT, matte_paint) if reference: config.setRole(ocio.Constants.ROLE_REFERENCE, reference) - if scene_linear: - config.setRole(ocio.Constants.ROLE_SCENE_LINEAR, scene_linear) if texture_paint: config.setRole(ocio.Constants.ROLE_TEXTURE_PAINT, texture_paint) + # 'rendering' and 'compositing_linear' roles default to the 'scene_linear' + # value if not set explicitly + if rendering: + config.setRole("rendering", rendering) + if compositing_linear: + config.setRole("compositing_linear", compositing_linear) + if scene_linear: + config.setRole(ocio.Constants.ROLE_SCENE_LINEAR, scene_linear) + if not rendering: + config.setRole("rendering", scene_linear) + if not compositing_linear: + config.setRole("compositing_linear", scene_linear) + return True @@ -167,11 +183,24 @@ def generate_OCIO_transform(transforms): # lutFile transform if transform['type'] == 'lutFile': - ocio_transform = ocio.FileTransform( - src=transform['path'], - interpolation=interpolation_options[ - transform['interpolation']], - direction=direction_options[transform['direction']]) + # Create transforms + ocio_transform = ocio.FileTransform() + + if 'path' in transform: + ocio_transform.setSrc(transform['path']) + + if 'cccid' in transform: + ocio_transform.setCCCId(transform['cccid']) + + if 'interpolation' in transform: + ocio_transform.setInterpolation(transform['interpolation']) + else: + ocio_transform.setInterpolation(ocio.Constants.INTERP_BEST) + + if 'direction' in transform: + ocio_transform.setDirection( + direction_options[transform['direction']]) + ocio_transforms.append(ocio_transform) # matrix transform @@ -193,25 +222,59 @@ def generate_OCIO_transform(transforms): # exponent transform elif transform['type'] == 'exponent': ocio_transform = ocio.ExponentTransform() - ocio_transform.setValue(transform['value']) + + if 'value' in transform: + ocio_transform.setValue(transform['value']) + ocio_transforms.append(ocio_transform) # log transform elif transform['type'] == 'log': - ocio_transform = ocio.LogTransform( - base=transform['base'], - direction=direction_options[transform['direction']]) + ocio_transform = ocio.LogTransform() + + if 'base' in transform: + ocio_transform.setBase(transform['base']) + + if 'direction' in transform: + ocio_transform.setDirection( + direction_options[transform['direction']]) ocio_transforms.append(ocio_transform) # color space transform elif transform['type'] == 'colorspace': - ocio_transform = ocio.ColorSpaceTransform(src=transform['src'], - dst=transform['dst'], - direction= - direction_options[ - 'forward']) + ocio_transform = ocio.ColorSpaceTransform() + + if 'src' in transform: + ocio_transform.setSrc(transform['src']) + + if 'dst' in transform: + ocio_transform.setDst(transform['dst']) + + if 'direction' in transform: + ocio_transform.setDirection( + direction_options[transform['direction']]) + + ocio_transforms.append(ocio_transform) + + # look transform + elif transform['type'] == 'look': + ocio_transform = ocio.LookTransform() + if 'look' in transform: + ocio_transform.setLooks(transform['look']) + + if 'src' in transform: + ocio_transform.setSrc(transform['src']) + + if 'dst' in transform: + ocio_transform.setDst(transform['dst']) + + if 'direction' in transform: + ocio_transform.setDirection( + direction_options[transform['direction']]) + ocio_transforms.append(ocio_transform) + # unknown type else: print("Ignoring unknown transform type : %s" % transform['type']) @@ -227,10 +290,11 @@ def generate_OCIO_transform(transforms): return transform -def add_colorspace_alias(config, - reference_colorspace, - colorspace, - colorspace_alias_names): +def add_colorspace_aliases(config, + reference_colorspace, + colorspace, + colorspace_alias_names, + family='Aliases'): """ Object description. @@ -246,26 +310,32 @@ def add_colorspace_alias(config, """ for alias_name in colorspace_alias_names: - if alias_name == colorspace.name.lower(): - return + if alias_name.lower() == colorspace.name.lower(): + print('Skipping alias creation for %s, alias %s, because lower cased names match' % ( + colorspace.name, alias_name) ) + continue print('Adding alias colorspace space %s, alias to %s' % ( alias_name, colorspace.name)) - compact_family_name = 'Aliases' + compact_family_name = family + + description = colorspace.description + if colorspace.aces_transform_id: + description += "\n\nACES Transform ID : %s" % colorspace.aces_transform_id ocio_colorspace_alias = ocio.ColorSpace( name=alias_name, bitDepth=colorspace.bit_depth, - description=colorspace.description, + description=description, equalityGroup=colorspace.equality_group, family=compact_family_name, isData=colorspace.is_data, allocation=colorspace.allocation_type, allocationVars=colorspace.allocation_vars) - if not colorspace.to_reference_transforms: - print('Generating To-Reference transforms') + if colorspace.to_reference_transforms: + print('\tGenerating To-Reference transforms') ocio_transform = generate_OCIO_transform( [{'type': 'colorspace', 'src': colorspace.name, @@ -275,8 +345,8 @@ def add_colorspace_alias(config, ocio_transform, ocio.Constants.COLORSPACE_DIR_TO_REFERENCE) - if not colorspace.from_reference_transforms: - print('Generating From-Reference transforms') + if colorspace.from_reference_transforms: + print('\tGenerating From-Reference transforms') ocio_transform = generate_OCIO_transform( [{'type': 'colorspace', 'src': reference_colorspace.name, @@ -288,8 +358,104 @@ def add_colorspace_alias(config, config.addColorSpace(ocio_colorspace_alias) +def colorspace_prefixed_name(colorspace): + prefix = colorspace.family.replace("/", " - ") + return "%s - %s" % (prefix, colorspace.name) + +def add_look(config, + look, + prefix, + custom_lut_dir, + reference_name, + config_data, + multiple_displays=False): + """ + Object description. -def create_config(config_data, nuke=False): + Parameters + ---------- + parameter : type + Parameter description. + + Returns + ------- + type + Return value description. + """ + + look_name = look[0] + look_colorspace = look[1] + look_lut = look[2] + look_cccid = None + if len(look) == 4: + look_cccid = look[3] + + print('Adding look %s - %s' % (look_name, ", ".join(look)) ) + + # + # Copy look lut + # + if custom_lut_dir: + if not '$' in look_lut: + print( "Getting ready to copy look lut : %s" % look_lut ) + shutil.copy2(look_lut, custom_lut_dir) + look_lut = os.path.split(look_lut)[1] + else: + print( "Skipping LUT copy because path contains a context variable" ) + + # + # Create OCIO Look + # + # Look 1 + print('Adding look to config' ) + lk1 = ocio.Look() + lk1.setName( look_name ) + lk1.setProcessSpace( look_colorspace ) + + keys = {'type': 'lutFile', + 'path': look_lut, + 'direction': 'forward'} + if look_cccid: + keys['cccid'] = look_cccid + + ocio_transform = generate_OCIO_transform([keys]) + lk1.setTransform( ocio_transform ) + + # add to config + config.addLook( lk1 ) + + print( "Creating aliased colorspace") + + # + # Create OCIO colorspace that references that look + # - Needed for some implementations that don't process looks well + # - Also needed for some implementations that don't expose looks well + # + look_aliases = ["look_%s" % compact(look_name)] + colorspace = ColorSpace(look_name, + aliases=look_aliases, + description="The %s Look colorspace" % look_name, + family='Look') + + colorspace.from_reference_transforms = [{'type': 'look', + 'look': look_name, + 'src': reference_name, + 'dst': reference_name, + 'direction': 'forward'}] + + print('Adding colorspace %s, alias to look %s to config data' % ( + look_name, look_name)) + + # Add this colorspace into the main list of colorspaces + config_data['colorSpaces'].append(colorspace) + + print("") + +def integrate_looks_into_views(config, + looks, + reference_name, + config_data, + multiple_displays=False): """ Object description. @@ -303,16 +469,118 @@ def create_config(config_data, nuke=False): type Return value description. """ + look_names = [look[0] for look in looks] + + # Option 1 - Add a 'look' to each Display + # - Assumes there is a Display for each ACES Output Transform + if multiple_displays: + for look_name in look_names: + config_data['looks'].append(look_name) + + # Option 2 + # - Copy each Output Transform colorspace + # - For each copy, add a LookTransform at the head of the from_reference + # transform list + # - Add these new copied colorspaces for the Displays / Views + else: + for display, view_list in config_data['displays'].iteritems(): + output_colorspace_copy = None + look_names_string = "" + for view_name, output_colorspace in view_list.iteritems(): + if view_name == "Output Transform": + + print( "Adding new View that incorporates looks" ) + + # Make a copy of the output colorspace + output_colorspace_copy = copy.deepcopy(output_colorspace) + + #for look_name in look_names: + for i in range(len(look_names)): + look_name = look_names[i] + + # Add the LookTransform to the head of the from_reference transform list + if output_colorspace_copy.from_reference_transforms: + output_colorspace_copy.from_reference_transforms.insert(i, {'type': 'look', + 'look': look_name, + 'src': reference_name, + 'dst': reference_name, + 'direction': 'forward'}) + + # Add the LookTransform to the end of the to_reference transform list + if output_colorspace_copy.to_reference_transforms: + inverse_look_name = look_names[len(look_names) -1 -i] + + output_colorspace_copy.to_reference_transforms.append({'type': 'look', + 'look': inverse_look_name, + 'src': reference_name, + 'dst': reference_name, + 'direction': 'inverse'}) + + if not look_name in config_data['looks']: + config_data['looks'].append(look_name) + + look_names_string = ", ".join(look_names) + output_colorspace_copy.name = "%s with %s" % (output_colorspace.name, look_names_string) + output_colorspace_copy.aliases = ["out_%s" % compact(output_colorspace_copy.name)] + + print( "Colorspace that incorporates looks created : %s" % output_colorspace_copy.name ) + + config_data['colorSpaces'].append(output_colorspace_copy) + + if output_colorspace_copy: + print( "Adding colorspace that incorporates looks into view list" ) + + # Change the name of the View + view_list["Output Transform with %s" % look_names_string] = output_colorspace_copy + config_data['displays'][display] = view_list + + #print( "Display : %s, View List : %s" % (display, ", ".join(view_list)) ) + +def create_config(config_data, + aliases=False, + prefix=False, + multiple_displays=False, + look_info=[], + custom_lut_dir=None): + """ + Object description. + + Parameters + ---------- + parameter : type + Parameter description. + + Returns + ------- + type + Return value description. + """ + + prefixed_names = {} + alias_colorspaces = [] # Creating the *OCIO* configuration. config = ocio.Config() - # Setting configuration overall values. + # Setting configuration description. config.setDescription('An ACES config generated from python') - config.setSearchPath('luts') + + # Setting configuration search path. + searchPath = ['luts'] + if custom_lut_dir: + searchPath.append('custom') + config.setSearchPath(':'.join(searchPath)) # Defining the reference colorspace. reference_data = config_data['referenceColorSpace'] + + # Adding the color space Family into the name + # Helps with applications that present colorspaces as one long list + if prefix: + prefixed_name = colorspace_prefixed_name(reference_data) + prefixed_names[reference_data.name] = prefixed_name + reference_data.name = prefixed_name + print('Adding the reference color space : %s' % reference_data.name) reference = ocio.ColorSpace( @@ -328,21 +596,65 @@ def create_config(config_data, nuke=False): config.addColorSpace(reference) # Add alias - if not nuke: + if aliases: if reference_data.aliases != []: - add_colorspace_alias(config, reference_data, - reference_data, reference_data.aliases) + #add_colorspace_alias(config, reference_data, + # reference_data, reference_data.aliases) + # defer adding alias colorspaces until end. Helps with some applications + alias_colorspaces.append([reference_data, reference_data, reference_data.aliases]) + print("") + #print( "color spaces : %s" % [x.name for x in sorted(config_data['colorSpaces'])]) + + # + # Add Looks and Look colorspaces + # + if look_info != []: + print('Adding looks') + + config_data['looks'] = [] + + # Add looks and colorspaces + for look in look_info: + add_look(config, + look, + prefix, + custom_lut_dir, + reference_data.name, + config_data) + + # Integrate looks with displays, views + integrate_looks_into_views(config, + look_info, + reference_data.name, + config_data, + multiple_displays) + + print("") + + print('Adding the regular color spaces') + # Creating the remaining colorspaces. for colorspace in sorted(config_data['colorSpaces']): + # Adding the color space Family into the name + # Helps with applications that present colorspaces as one long list + if prefix: + prefixed_name = colorspace_prefixed_name(colorspace) + prefixed_names[colorspace.name] = prefixed_name + colorspace.name = prefixed_name + print('Creating new color space : %s' % colorspace.name) + description = colorspace.description + if colorspace.aces_transform_id: + description += "\n\nACES Transform ID : %s" % colorspace.aces_transform_id + ocio_colorspace = ocio.ColorSpace( name=colorspace.name, bitDepth=colorspace.bit_depth, - description=colorspace.description, + description=description, equalityGroup=colorspace.equality_group, family=colorspace.family, isData=colorspace.is_data, @@ -350,7 +662,7 @@ def create_config(config_data, nuke=False): allocationVars=colorspace.allocation_vars) if colorspace.to_reference_transforms: - print('Generating To-Reference transforms') + print('\tGenerating To-Reference transforms') ocio_transform = generate_OCIO_transform( colorspace.to_reference_transforms) ocio_colorspace.setTransform( @@ -358,7 +670,7 @@ def create_config(config_data, nuke=False): ocio.Constants.COLORSPACE_DIR_TO_REFERENCE) if colorspace.from_reference_transforms: - print('Generating From-Reference transforms') + print('\tGenerating From-Reference transforms') ocio_transform = generate_OCIO_transform( colorspace.from_reference_transforms) ocio_colorspace.setTransform( @@ -370,70 +682,269 @@ def create_config(config_data, nuke=False): # # Add alias to normal colorspace, using compact name # - if not nuke: + if aliases: if colorspace.aliases != []: - add_colorspace_alias(config, reference_data, - colorspace, colorspace.aliases) + #add_colorspace_alias(config, reference_data, + # colorspace, colorspace.aliases) + # defer adding alias colorspaces until end. Helps with some applications + alias_colorspaces.append([reference_data, colorspace, colorspace.aliases]) print('') + print("") + + # + # We add roles early so we can create alias colorspaces with the names of the roles + # before the rest of the colorspace aliases are added to the config. + # + print('Setting the roles') + + if prefix: + set_config_default_roles( + config, + color_picking=prefixed_names[config_data['roles']['color_picking']], + color_timing=prefixed_names[config_data['roles']['color_timing']], + compositing_log=prefixed_names[config_data['roles']['compositing_log']], + data=prefixed_names[config_data['roles']['data']], + default=prefixed_names[config_data['roles']['default']], + matte_paint=prefixed_names[config_data['roles']['matte_paint']], + reference=prefixed_names[config_data['roles']['reference']], + scene_linear=prefixed_names[config_data['roles']['scene_linear']], + texture_paint=prefixed_names[config_data['roles']['texture_paint']]) + + # Not allowed for the moment. role names can not overlap with colorspace names. + ''' + # Add the aliased colorspaces for each role + for role_name, role_colorspace_name in config_data['roles'].iteritems(): + role_colorspace_prefixed_name = prefixed_names[role_colorspace_name] + + print( "Finding colorspace : %s" % role_colorspace_prefixed_name ) + # Find the colorspace pointed to by the role + role_colorspaces = [colorspace for colorspace in config_data['colorSpaces'] if colorspace.name == role_colorspace_prefixed_name] + role_colorspace = None + if len(role_colorspaces) > 0: + role_colorspace = role_colorspaces[0] + else: + if reference_data.name == role_colorspace_prefixed_name: + role_colorspace = reference_data + + if role_colorspace: + print( "Adding an alias colorspace named %s, pointing to %s" % ( + role_name, role_colorspace.name)) + + add_colorspace_aliases(config, reference_data, role_colorspace, [role_name], 'Roles') + ''' + + else: + set_config_default_roles( + config, + color_picking=config_data['roles']['color_picking'], + color_timing=config_data['roles']['color_timing'], + compositing_log=config_data['roles']['compositing_log'], + data=config_data['roles']['data'], + default=config_data['roles']['default'], + matte_paint=config_data['roles']['matte_paint'], + reference=config_data['roles']['reference'], + scene_linear=config_data['roles']['scene_linear'], + texture_paint=config_data['roles']['texture_paint']) + + # Not allowed for the moment. role names can not overlap with colorspace names. + ''' + # Add the aliased colorspaces for each role + for role_name, role_colorspace_name in config_data['roles'].iteritems(): + # Find the colorspace pointed to by the role + role_colorspaces = [colorspace for colorspace in config_data['colorSpaces'] if colorspace.name == role_colorspace_name] + role_colorspace = None + if len(role_colorspaces) > 0: + role_colorspace = role_colorspaces[0] + else: + if reference_data.name == role_colorspace_name: + role_colorspace = reference_data + + if role_colorspace: + print( "Adding an alias colorspace named %s, pointing to %s" % ( + role_name, role_colorspace.name)) + + add_colorspace_aliases(config, reference_data, role_colorspace, [role_name], 'Roles') + ''' + + print("") + + # We add these at the end as some applications use the order of the colorspaces + # definitions in the config to order the colorspaces in their selection lists. + # Other go alphabetically. This should keep the alias colorspaces out of the way + # for the apps that use the order of definition in the config. + print('Adding the alias colorspaces') + for reference, colorspace, aliases in alias_colorspaces: + add_colorspace_aliases(config, reference, colorspace, aliases) + + print("") + + print('Adding the diplays and views') + + # Set the color_picking role to be the first Display's Output Transform View + default_display_name = config_data['defaultDisplay'] + default_display_views = config_data['displays'][default_display_name] + default_display_colorspace = default_display_views['Output Transform'] + + set_config_default_roles( + config, + color_picking=default_display_colorspace.name) + # Defining the *views* and *displays*. displays = [] views = [] # Defining a *generic* *display* and *view* setup. - if not nuke: + if multiple_displays: + # Built list of looks to add to Displays + looks = config_data['looks'] if ('looks' in config_data) else [] + looks = ", ".join(looks) + print( "Creating multiple displays, with looks : %s" % looks) + + # Note: We don't reorder the Displays to put the 'defaultDisplay' first + # because OCIO will order them alphabetically when the config is written to disk. + + # Create Displays, Views for display, view_list in config_data['displays'].iteritems(): for view_name, colorspace in view_list.iteritems(): - config.addDisplay(display, view_name, colorspace.name) + config.addDisplay(display, view_name, colorspace.name, looks) + if 'Output Transform' in view_name and looks != "": + # Add normal View, without looks + config.addDisplay(display, view_name, colorspace.name) + + # Add View with looks + view_name_with_looks = "%s with %s" % (view_name, looks) + config.addDisplay(display, view_name_with_looks, colorspace.name, looks) + else: + config.addDisplay(display, view_name, colorspace.name) if not (view_name in views): views.append(view_name) displays.append(display) - # Defining the *Nuke* specific set of *views* and *displays*. + # Defining the set of *views* and *displays* useful in a *GUI* context. else: - display_name = 'ACES' - displays.append(display_name) + single_display_name = 'ACES' + #single_display_name = config_data['roles']['scene_linear'] + displays.append(single_display_name) + # Make sure the default display is first display_names = sorted(config_data['displays']) + display_names.insert(0, display_names.pop(display_names.index(default_display_name))) + + # Built list of looks to add to Displays + looks = config_data['looks'] if ('looks' in config_data) else [] + look_names = ", ".join(looks) + + displays_views_colorspaces = [] + + # Create Displays, Views for display in display_names: view_list = config_data['displays'][display] for view_name, colorspace in view_list.iteritems(): - if view_name == 'Output Transform': - config.addDisplay(display_name, display, colorspace.name) - if not (display in views): - views.append(display) + if 'Output Transform' in view_name: + #print( "Adding view for %s" % colorspace.name ) + + # We use the Display names as the View names in this case + # as there is a single Display that contains all views. + # This works for more applications than not, as of the time of this implementation. + + # Maya 2016 doesn't like parentheses in View names + display_cleaned = replace(display, {')': '', '(': ''}) + + # If View includes looks + if 'with' in view_name: + # Integrate looks into view name + display_cleaned = "%s with %s" % (display_cleaned, look_names) + + viewsWithLooksAtEnd = False + # Storing combo of display, view and colorspace name in a list so we can + # add them to the end of the list + if viewsWithLooksAtEnd: + displays_views_colorspaces.append([single_display_name, display_cleaned, colorspace.name]) + + # Or add as normal + else: + config.addDisplay(single_display_name, display_cleaned, colorspace.name) + + # Add to views list + if not (display_cleaned in views): + views.append(display_cleaned) + + # A normal View + else: + config.addDisplay(single_display_name, display_cleaned, colorspace.name) + + # Add to views list + if not (display_cleaned in views): + views.append(display_cleaned) + + # Add to config any display, view combinations that were saved for later + # This list will be empty unless viewsWithLooksAtEnd is set to True above + for display_view_colorspace in displays_views_colorspaces: + single_display_name, display_cleaned, colorspace_name = display_view_colorspace + + # Add to config + config.addDisplay(single_display_name, display_cleaned, colorspace_name) + + # Add to views list + if not (display_cleaned in views): + views.append(display_cleaned) + # Works with Nuke Studio and Mari, but not Nuke - # display_name = 'Utility' - # displays.append(display_name) + # single_display_name = 'Utility' + # displays.append(single_display_name) + + raw_display_space_name = config_data['roles']['data'] + log_display_space_name = config_data['roles']['compositing_log'] - linear_display_space_name = config_data['linearDisplaySpace'].name - log_display_space_name = config_data['logDisplaySpace'].name + # Find the newly-prefixed colorspace names + if prefix: + #print( prefixed_names ) + raw_display_space_name = prefixed_names[raw_display_space_name] + log_display_space_name = prefixed_names[log_display_space_name] - config.addDisplay(display_name, 'Linear', linear_display_space_name) - views.append('Linear') - config.addDisplay(display_name, 'Log', log_display_space_name) + config.addDisplay(single_display_name, 'Raw', raw_display_space_name) + views.append('Raw') + config.addDisplay(single_display_name, 'Log', log_display_space_name) views.append('Log') # Setting the active *displays* and *views*. config.setActiveDisplays(','.join(sorted(displays))) config.setActiveViews(','.join(views)) - set_config_default_roles( - config, - color_picking=config_data['roles']['color_picking'], - color_timing=config_data['roles']['color_timing'], - compositing_log=config_data['roles']['compositing_log'], - data=config_data['roles']['data'], - default=config_data['roles']['default'], - matte_paint=config_data['roles']['matte_paint'], - reference=config_data['roles']['reference'], - scene_linear=config_data['roles']['scene_linear'], - texture_paint=config_data['roles']['texture_paint']) + print("") + # Make sure we didn't create a bad config config.sanityCheck() + # Reset the colorspace names back to their non-prefixed versions + if prefix: + # Build the reverse lookup + prefixed_names_inverse = {} + for original, prefixed in prefixed_names.iteritems(): + prefixed_names_inverse[prefixed] = original + + # Reset the reference colorspace name + reference_data.name = prefixed_names_inverse[reference_data.name] + + # Reset the rest of the colorspace names + try: + for colorspace in config_data['colorSpaces']: + colorspace.name = prefixed_names_inverse[colorspace.name] + except: + print( "Prefixed names") + for original, prefixed in prefixed_names.iteritems(): + print( "%s, %s" % (original, prefixed) ) + + print( "\n") + + print( "Inverse Lookup of Prefixed names") + for prefixed, original in prefixed_names_inverse.iteritems(): + print( "%s, %s" % (prefixed, original) ) + raise + return config @@ -476,14 +987,15 @@ def generate_LUTs(odt_info, aces_colorspaces, aces_displays, aces_log_display_space, - aces_roles) = aces.create_colorspaces(aces_ctl_directory, - lut_directory, - lut_resolution_1d, - lut_resolution_3d, - lmt_info, - odt_info, - shaper_name, - cleanup) + aces_roles, + aces_default_display) = aces.create_colorspaces(aces_ctl_directory, + lut_directory, + lut_resolution_1d, + lut_resolution_3d, + lmt_info, + odt_info, + shaper_name, + cleanup) config_data['referenceColorSpace'] = aces_reference config_data['roles'] = aces_roles @@ -494,6 +1006,7 @@ def generate_LUTs(odt_info, for name, data in aces_displays.iteritems(): config_data['displays'][name] = data + config_data['defaultDisplay'] = aces_default_display config_data['linearDisplaySpace'] = aces_reference config_data['logDisplaySpace'] = aces_log_display_space @@ -565,7 +1078,8 @@ def generate_baked_LUTs(odt_info, config_path, lut_resolution_1d, lut_resolution_3d, - lut_resolution_shaper=1024): + lut_resolution_shaper=1024, + prefix=False): """ Object description. @@ -580,8 +1094,12 @@ def generate_baked_LUTs(odt_info, Return value description. """ - # Create two entries for ODTs that have full and legal range support odt_info_C = dict(odt_info) + + # Uncomment if you would like to support the older behavior where ODTs + # that have support for full and legal range output generate a LUT for each. + ''' + # Create two entries for ODTs that have full and legal range support for odt_ctl_name, odt_values in odt_info.iteritems(): if odt_values['transformHasFullLegalSwitch']: odt_name = odt_values['transformUserName'] @@ -595,6 +1113,7 @@ def generate_baked_LUTs(odt_info, odt_info_C['%s - Full' % odt_ctl_name] = odt_values_full del (odt_info_C[odt_ctl_name]) + ''' # Generate appropriate LUTs for each ODT for odt_ctl_name, odt_values in odt_info_C.iteritems(): @@ -604,15 +1123,24 @@ def generate_baked_LUTs(odt_info, # *Photoshop* for input_space in ['ACEScc', 'ACESproxy']: args = ['--iconfig', config_path, - '-v', - '--inputspace', input_space] - args += ['--outputspace', '%s' % odt_name] + '-v'] + if prefix: + args += ['--inputspace', "ACES - %s" % input_space] + args += ['--outputspace', "Output - %s" % odt_name] + else: + args += ['--inputspace', input_space] + args += ['--outputspace', odt_name] + args += ['--description', '%s - %s for %s data' % (odt_prefix, odt_name, input_space)] - args += ['--shaperspace', shaper_name, - '--shapersize', str(lut_resolution_shaper)] + if prefix: + args += ['--shaperspace', "Utility - %s" % shaper_name, + '--shapersize', str(lut_resolution_shaper)] + else: + args += ['--shaperspace', shaper_name, + '--shapersize', str(lut_resolution_shaper)] args += ['--cubesize', str(lut_resolution_3d)] args += ['--format', 'icc', @@ -628,14 +1156,22 @@ def generate_baked_LUTs(odt_info, # *Flame*, *Lustre* for input_space in ['ACEScc', 'ACESproxy']: args = ['--iconfig', config_path, - '-v', - '--inputspace', input_space] - args += ['--outputspace', '%s' % odt_name] + '-v'] + if prefix: + args += ['--inputspace', "ACES - %s" % input_space] + args += ['--outputspace', "Output - %s" % odt_name] + else: + args += ['--inputspace', input_space] + args += ['--outputspace', odt_name] args += ['--description', '%s - %s for %s data' % ( odt_prefix, odt_name, input_space)] - args += ['--shaperspace', shaper_name, - '--shapersize', str(lut_resolution_shaper)] + if prefix: + args += ['--shaperspace', "Utility - %s" % shaper_name, + '--shapersize', str(lut_resolution_shaper)] + else: + args += ['--shaperspace', shaper_name, + '--shapersize', str(lut_resolution_shaper)] args += ['--cubesize', str(lut_resolution_3d)] fargs = ['--format', @@ -663,16 +1199,22 @@ def generate_baked_LUTs(odt_info, # *Maya*, *Houdini* for input_space in ['ACEScg', 'ACES2065-1']: args = ['--iconfig', config_path, - '-v', - '--inputspace', input_space] - args += ['--outputspace', '%s' % odt_name] + '-v'] + if prefix: + args += ['--inputspace', "ACES - %s" % input_space] + args += ['--outputspace', "Output - %s" % odt_name] + else: + args += ['--inputspace', input_space] + args += ['--outputspace', odt_name] args += ['--description', '%s - %s for %s data' % ( odt_prefix, odt_name, input_space)] if input_space == 'ACEScg': - lin_shaper_name = '%s - AP1' % shaper_name + lin_shaper_name = "%s - AP1" % shaper_name else: lin_shaper_name = shaper_name + if prefix: + lin_shaper_name = "Utility - %s" % lin_shaper_name args += ['--shaperspace', lin_shaper_name, '--shapersize', str(lut_resolution_shaper)] @@ -701,7 +1243,9 @@ def generate_baked_LUTs(odt_info, bake_lut.execute() -def create_config_dir(config_directory, bake_secondary_LUTs): +def create_config_dir(config_directory, + bake_secondary_LUTs=False, + custom_lut_dir=None): """ Object description. @@ -718,6 +1262,7 @@ def create_config_dir(config_directory, bake_secondary_LUTs): lut_directory = os.path.join(config_directory, 'luts') dirs = [config_directory, lut_directory] + if bake_secondary_LUTs: dirs.extend([os.path.join(config_directory, 'baked'), os.path.join(config_directory, 'baked', 'flame'), @@ -726,6 +1271,9 @@ def create_config_dir(config_directory, bake_secondary_LUTs): os.path.join(config_directory, 'baked', 'lustre'), os.path.join(config_directory, 'baked', 'maya')]) + if custom_lut_dir: + dirs.append(os.path.join(config_directory, 'custom')) + for d in dirs: not os.path.exists(d) and os.mkdir(d) @@ -737,7 +1285,11 @@ def create_ACES_config(aces_ctl_directory, lut_resolution_1d=4096, lut_resolution_3d=64, bake_secondary_LUTs=True, - cleanup=True): + multiple_displays=False, + look_info=[], + copy_custom_luts=True, + cleanup=True, + prefix_colorspaces_with_family_names=True): """ Creates the ACES configuration. @@ -752,7 +1304,14 @@ def create_ACES_config(aces_ctl_directory, Return value description. """ - lut_directory = create_config_dir(config_directory, bake_secondary_LUTs) + # Directory for custom LUTs + custom_lut_dir = None + if copy_custom_luts: + custom_lut_dir = os.path.join(config_directory, "custom") + + lut_directory = create_config_dir(config_directory, + bake_secondary_LUTs, + custom_lut_dir) odt_info = aces.get_ODTs_info(aces_ctl_directory) lmt_info = aces.get_LMTs_info(aces_ctl_directory) @@ -767,20 +1326,18 @@ def create_ACES_config(aces_ctl_directory, lut_resolution_3d, cleanup) - print('Creating "generic" config') - config = create_config(config_data) + print('Creating config - with prefixes, with aliases') + config = create_config(config_data, + prefix=prefix_colorspaces_with_family_names, + aliases=True, + multiple_displays=multiple_displays, + look_info=look_info, + custom_lut_dir=custom_lut_dir) print('\n\n\n') write_config(config, os.path.join(config_directory, 'config.ocio')) - print('Creating "Nuke" config') - nuke_config = create_config(config_data, nuke=True) - print('\n\n\n') - - write_config(nuke_config, - os.path.join(config_directory, 'nuke_config.ocio')) - if bake_secondary_LUTs: generate_baked_LUTs(odt_info, shaper_name, @@ -788,7 +1345,8 @@ def create_ACES_config(aces_ctl_directory, os.path.join(config_directory, 'config.ocio'), lut_resolution_1d, lut_resolution_3d, - lut_resolution_1d) + lut_resolution_1d, + prefix=prefix_colorspaces_with_family_names) return True @@ -810,18 +1368,75 @@ def main(): import optparse - p = optparse.OptionParser(description='An OCIO config generation script', - prog='createACESConfig', - version='createACESConfig 0.1', - usage='%prog [options]') + usage = '%prog [options]\n' + usage += '\n' + usage += 'An OCIO config generation script for ACES 1.0\n' + usage += '\n' + usage += 'Command line examples' + usage += '\n' + usage += 'Create a GUI-friendly ACES 1.0 config with no secondary, baked LUTs : \n' + usage += '\tcreate_aces_config -a /path/to/aces-dev/transforms/ctl --lutResolution1d 1024 --lutResolution3d 33 -c aces_1.0.0 --dontBakeSecondaryLUTs' + usage += '\n' + usage += 'Create a more OCIO-compliant ACES 1.0 config : \n' + usage += '\tcreate_aces_config -a /path/to/aces-dev/transforms/ctl --lutResolution1d 1024 --lutResolution3d 33 -c aces_1.0.0 --createMultipleDisplays' + usage += '\n' + usage += '\n' + usage += 'Adding custom looks' + usage += '\n' + usage += 'Create a GUI-friendly ACES 1.0 config with an ACES-style CDL (will be applied in the ACEScc colorspace): \n' + usage += '\tcreate_aces_config -a /path/to/aces-dev/transforms/ctl --lutResolution1d 1024 --lutResolution3d 33 -c aces_1.0.0 \n\t\t--addACESLookCDL ACESCDLName /path/to/SampleCDL.ccc cc03345' + usage += '\n' + usage += 'Create a GUI-friendly ACES 1.0 config with an general CDL: \n' + usage += '\tcreate_aces_config -a /path/to/aces-dev/transforms/ctl --lutResolution1d 1024 --lutResolution3d 33 -c aces_1.0.0 \n\t\t--addCustomLookCDL CustomCDLName "ACES - ACEScc" /path/to/SampleCDL.ccc cc03345' + usage += '\n' + usage += '\tIn this example, the CDL will be applied in the ACEScc colorspace, but the user could choose other spaces by changing the argument after the name of the look. \n' + usage += '\n' + usage += 'Create a GUI-friendly ACES 1.0 config with an ACES-style LUT (will be applied in the ACEScc colorspace): \n' + usage += '\tcreate_aces_config -a /path/to/aces-dev/transforms/ctl --lutResolution1d 1024 --lutResolution3d 33 -c aces_1.0.0 \n\t\t--addACESLookLUT ACESLUTName /path/to/SampleCDL.ccc cc03345' + usage += '\n' + usage += 'Create a GUI-friendly ACES 1.0 config with an general LUT: \n' + usage += '\tcreate_aces_config -a /path/to/aces-dev/transforms/ctl --lutResolution1d 1024 --lutResolution3d 33 -c aces_1.0.0 \n\t\t--addCustomLookLUT CustomLUTName "ACES - ACEScc" /path/to/SampleCDL.ccc cc03345' + usage += '\n' + usage += '\tIn this example, the LUT will be applied in the ACEScc colorspace, but the user could choose other spaces by changing the argument after the name of the look. \n' + usage += '\n' + + look_info = [] + def look_info_callback(option, opt_str, value, parser): + print( "look_info_callback" ) + print( option, opt_str, value, parser ) + if opt_str == "--addCustomLookCDL": + look_info.append(value) + elif opt_str == "--addCustomLookLUT": + look_info.append(value) + elif opt_str == "--addACESLookCDL": + look_info.append([value[0], "ACES - ACEScc", value[1], value[2]]) + elif opt_str == "--addACESLookLUT": + look_info.append([value[0], "ACES - ACEScc", value[1]]) + + p = optparse.OptionParser(description='', + prog='create_aces_config', + version='create_aces_config 1.0', + usage=usage) p.add_option('--acesCTLDir', '-a', default=os.environ.get( ACES_OCIO_CTL_DIRECTORY_ENVIRON, None)) p.add_option('--configDir', '-c', default=os.environ.get( ACES_OCIO_CONFIGURATION_DIRECTORY_ENVIRON, None)) p.add_option('--lutResolution1d', default=4096) p.add_option('--lutResolution3d', default=64) - p.add_option('--dontBakeSecondaryLUTs', action='store_true') - p.add_option('--keepTempImages', action='store_true') + p.add_option('--dontBakeSecondaryLUTs', action='store_true', default=False) + p.add_option('--keepTempImages', action='store_true', default=False) + + p.add_option('--createMultipleDisplays', action='store_true', default=False) + + p.add_option('--addCustomLookLUT', '', type='string', nargs=3, + action="callback", callback=look_info_callback) + p.add_option('--addCustomLookCDL', '', type='string', nargs=4, + action="callback", callback=look_info_callback) + p.add_option('--addACESLookLUT', '', type='string', nargs=2, + action="callback", callback=look_info_callback) + p.add_option('--addACESLookCDL', '', type='string', nargs=3, + action="callback", callback=look_info_callback) + p.add_option('--copyCustomLUTs', action='store_true', default=False) options, arguments = p.parse_args() @@ -831,6 +1446,10 @@ def main(): lut_resolution_3d = int(options.lutResolution3d) bake_secondary_luts = not options.dontBakeSecondaryLUTs cleanup_temp_images = not options.keepTempImages + multiple_displays = options.createMultipleDisplays + copy_custom_luts = options.copyCustomLUTs + + print( look_info ) # TODO: Investigate the following statements. try: @@ -857,8 +1476,10 @@ def main(): lut_resolution_1d, lut_resolution_3d, bake_secondary_luts, + multiple_displays, + look_info, + copy_custom_luts, cleanup_temp_images) - if __name__ == '__main__': main()