### DEscription ###

## This script is to create AOV pass system like mental ray contribution passes

## This script is still deving/improving

## really welcome anyone can give/ mail me any comments and suggestions

## Thanks! :)

## Kenzie Chen | kenziec@themill.com

### Limits ###

## Due to use aiUserData, the shape node has some custome attributes

## so, there are 2 limits as below

## 01. When update new reference model, those custom attributes are gone!

## 02. when using deformer, there is a new shape node generates, so siUserData will ignore the old shape node

## which has cutome attributes.

## load lib

import pymel.core as pm

import sys

import copy

import math

# check if current render is Arnold

if( pm.getAttr( 'defaultRenderGlobals.currentRenderer' ) != 'arnold' ):

pm.confirmDialog( t="Error", message="Please use Arnold render", icon='critical' )

sys.exit( "Please use Arnold render!" )

import mtoa.aovs as aovs

# declare ui

uiLayout = {}

# declaire global variable

prefixAOV = 'mtoa_constant_'

isProgress = False

sel = []

''' =============== sub-function =============== '''

## check if any object is seleted

def isSelEmpty(*args):

return 1

## check is selection has unsupport type

def isObjType(*args):

return 1

## add Color/ String attributes

def addUserAttr( obj, attrType ):

return 1

## Creates AOV render pass

def addAOV( name ):

return 1

## add AOV Attribute for objects

def doAddAOVAttr(*args):

return 1

def doAddColorAttr( inColor ):

return 1

## delete custom mtoa* attributes

def doDelAttrAOV(*args):

return 1

## delete unused AOVs

def doDelEmptyAOVs(*args):

return 1

## update/ collect each AOVs containing objects list

def updateAOVStrAttr(*args):

return 1

def doStopUpdateAOV(*args):

if isProgress:

pm.progressWindow(endProgress=1)

return 1

## update scene id/ AOV

def doUpdateScnAOV(type, *args):

return 1

def doSelObjInAOV(*args):

return 1

def doDelAOV(*args):

return 1;

## rebuild object Attr from AOVs

## when updateing model assets

def doRebuildObjData(*args):

pm.confirmDialog( t='Objects Missing!', message=('Original Objects Missing Target: \n======================\n%s' % "\n".join(miss_obj) ), b=['OK'], icon='critical' )

## create shading network

def doIDShdNetwork(*args):

writeColor.outColor >> aiIDShd.color

''' =============== main function =============== '''

def main():

pm.showWindow( uiLayout['window'] )

main()