code_gallery

Web News Blog Press Pricing

Learn

Training

Templates

VFX

Code

Code Gallery Designer

v5.09

Maya

Set Drivers and AOVs for Rendering

Luis Miguel 31/10/2024

Tags

Set AOVs

Set Drivers and AOVs for Rendering

Luis Miguel

Action v1

31/10/2024

Maya

Pseudo-Code

50	Iterate for each driver to create
75	Iterate for each AOV to create in the current driver.
86	Check filter of the AOV and remove it if necessary.
100	Connect the new AOV to the driver and the filter.
117	General render settings.

Parameters

Talk with TD Support

Import to my Pipeline

Code

import maya.cmds as cmds
import mtoa.aovs as aovs

# AOVs and drivers configuration
drivers_data = {
    "Beauty": {
        "aovs": [
        {
            "name": "RGBA"
        },{
            "name": "albedo"
        },{
            "name": "diffuse"
        },{
            "name": "specular"
        }],
    },
    "Tech": {
        "filter": {
            "type": "closest"
        },
        "aovs": [
        {
            "name": "N"
        },{
            "name": "Z"
        },{
            "name": "P"
        },{
            "name": "Pref"
        }]
    },
    "Utility": {
        "filter": {
            "type": "closest"
        },
        "aovs": [
        {
            "name": "ID"
        },{
            "name": "crypto_object"
        },{
            "name": "crypto_material"
        }]
    }
}

def create_aovs(drivers_data):
    # Create drivers, filters and AOVs
    for driver_name, driver_data in drivers_data.items():
        aov_list = driver_data.get("aovs", [])
        # Create driver
        driver_node = cmds.createNode("aiAOVDriver", name=f"{driver_name}_Driver")

# Config Driver render output
        cmds.setAttr(f"{driver_node}.aiTranslator", "exr", type="string")  # Format EXR
        cmds.setAttr(f"{driver_node}.exrCompression", 2)  # Compresion ZIP (2 = ZIP)
        cmds.setAttr(f"{driver_node}.mergeAOVs", 1)  # 1 file for all AOVs

# output folder structure for the driver
        cmds.setAttr(
            f"{driver_node}.prefix", 
            f"<RenderLayer>/{driver_name}/<Scene>", 
            type="string")

# Custom Arnold filter for all AOVs of this Driver
        filter_node = cmds.createNode("aiAOVFilter", name=f"{driver_name}_Filter")
        filter_type = driver_data.get("filter", {}).get("type", "gaussian")
        cmds.setAttr(
            f"{filter_node}.aiTranslator", 
            f"{filter_type}", 
            type="string")

# Create AOVs of this driver
        for aov_data in aov_list:
            aov_name = aov_data["name"]
            aov_interface = aovs.AOVInterface()
            aov_interface.addAOV(aov_name) # Create AOV

# AOV node name
            aov_node = f"aiAOV_{aov_name}"

if not cmds.objExists(aov_node):
                print(f"AOV node name {aov_node} could not be found.The AOV creation failed.")
                continue
            # Check for AOV filters
            existing_filter = cmds.listConnections(
                f"{aov_node}.outputs[0].filter", source=True, destination=False)

if existing_filter:
                existing_filter = existing_filter[0]
                # Remove existing filters if do not have more connections  
                if existing_filter != "defaultArnoldFilter" and \
                        len(cmds.listConnections(f"{existing_filter}.message", \
                            source=False, destination=True)) == 1:
                    cmds.disconnectAttr(f"{existing_filter}.message", f"{aov_node}.outputs[0].filter")
                    cmds.delete(existing_filter)
                    print(f"Remove previous filter: {existing_filter}")

# Connect driver and filter to the AOV
            cmds.connectAttr(
                f"{driver_node}.message", 
                f"{aov_node}.outputs[0].driver", 
                force=True)  # Conectar al Driver
            cmds.connectAttr(
                f"{filter_node}.message", 
                f"{aov_node}.outputs[0].filter", 
                force=True)  # Conectar al Filter

print(f"AOV {aov_name} created and connected to: "
                  f"filter - {filter_node} & driver - {driver_node}")

print(f"Configuration for driver \"{driver_node}\" complete.\n")

print("All Drivers, Filters and AOVs configured successfully.\n")

def set_render_output_settings():
    # Output filename
    file_prefix = "MasterLayer/<Scene>"
    cmds.setAttr("defaultRenderGlobals.imageFilePrefix", file_prefix, type="string")
    # Output format -> exr
    image_format = 51  # 51 = EXR
    cmds.setAttr("defaultRenderGlobals.imageFormat", image_format)
    # Frame format -> .name.#.ext & padding of 4
    cmds.setAttr("defaultRenderGlobals.animation", 1)
    cmds.setAttr("defaultRenderGlobals.putFrameBeforeExt", 1)
    cmds.setAttr("defaultRenderGlobals.periodInExt", 1)
    cmds.setAttr("defaultRenderGlobals.outFormatControl", 0)
    cmds.setAttr("defaultRenderGlobals.extensionPadding", 4)

# File output
set_render_output_settings()
# AOVS
create_aovs(drivers_data)

Import Action

1. Copy Key

Copy

2. Go to your Action pannel

Go