def __init__(self, parent, partClass):
MelFormLayout.__init__(self, parent)
self.partClass = partClass
self.UI_create = cmd.button(
l=self.BUTTON_LBL_TEMPLATE %
names.camelCaseToNice(partClass.GetPartName()),
c=self.on_create,
w=160)
#now populate the ui for the part's args
self.kwarg_UIs = {} #keyed by arg name
kwargList = partClass.GetDefaultBuildKwargList()
self.UI_argsLayout = MelFormLayout(self)
#everything has a parent attribute, so build it first
prevUI = None
for arg, default in kwargList:
#skip UI for parent - assume selection always
if arg == 'parent':
continue
setParent(self.UI_argsLayout)
lbl = cmd.text(l=names.camelCaseToNice(arg, self.ABBRVS_TO_EXPAND))
#determine the function to use for building the UI for the arg
buildMethodFromName = UI_FOR_NAMED_RIG_ARGS.get(arg, None)
buildMethodFromType = getBuildUIMethodForObject(
default) or MelTextField
buildMethod = buildMethodFromName or buildMethodFromType
self.kwarg_UIs[arg] = argUI = buildMethod(self.UI_argsLayout)
argUI.setValue(default)
#perform layout
if prevUI is None:
self.UI_argsLayout(e=True, af=((lbl, 'left', 15)))
else:
self.UI_argsLayout(e=True, ac=((lbl, 'left', 15, prevUI)))
if isinstance(argUI, MelCheckBox):
self.UI_argsLayout(e=True, af=((argUI, 'top', 3)))
self.UI_argsLayout(e=True,
af=((lbl, 'top', 3)),
ac=((argUI, 'left', 5, lbl)))
prevUI = argUI
setParent(self)
self(e=True,
af=((self.UI_create, 'left', 0), (self.UI_argsLayout, 'right',
0)),
ac=((self.UI_argsLayout, 'left', 0, self.UI_create)))
def __init__(self, parent, partClass):
MelFormLayout.__init__(self, parent)
self.partClass = partClass
self.UI_create = cmd.button(
l=self.BUTTON_LBL_TEMPLATE % names.camelCaseToNice(partClass.GetPartName()), c=self.on_create, w=160
)
# now populate the ui for the part's args
self.kwarg_UIs = {} # keyed by arg name
kwargList = partClass.GetDefaultBuildKwargList()
self.UI_argsLayout = MelFormLayout(self)
# everything has a parent attribute, so build it first
prevUI = None
for arg, default in kwargList:
# skip UI for parent - assume selection always
if arg == "parent":
continue
setParent(self.UI_argsLayout)
lbl = cmd.text(l=names.camelCaseToNice(arg, self.ABBRVS_TO_EXPAND))
# determine the function to use for building the UI for the arg
buildMethodFromName = UI_FOR_NAMED_RIG_ARGS.get(arg, None)
buildMethodFromType = getBuildUIMethodForObject(default) or MelTextField
buildMethod = buildMethodFromName or buildMethodFromType
self.kwarg_UIs[arg] = argUI = buildMethod(self.UI_argsLayout)
argUI.setValue(default)
# perform layout
if prevUI is None:
self.UI_argsLayout(e=True, af=((lbl, "left", 15)))
else:
self.UI_argsLayout(e=True, ac=((lbl, "left", 15, prevUI)))
if isinstance(argUI, MelCheckBox):
self.UI_argsLayout(e=True, af=((argUI, "top", 3)))
self.UI_argsLayout(e=True, af=((lbl, "top", 3)), ac=((argUI, "left", 5, lbl)))
prevUI = argUI
setParent(self)
self(
e=True,
af=((self.UI_create, "left", 0), (self.UI_argsLayout, "right", 0)),
ac=((self.UI_argsLayout, "left", 0, self.UI_create)),
)
def populate(self):
# remove any existing children
self.clear()
part = self.part
assert isinstance(part, SkeletonPart)
# pimp out the UI
lbl = MelButton(self, l=part.getPartName(), w=140, c=self.on_select)
# grab the args the rigging method takes
argsForm = MelFormLayout(self)
argsUIs = []
buildKwargs = part.getBuildKwargs()
for arg in self.ARGS_TO_HIDE:
buildKwargs.pop(arg, None)
for arg, argValue in buildKwargs.iteritems():
argLbl = MelLabel(argsForm, label=names.camelCaseToNice(arg))
# determine the function to use for building the UI for the arg
buildMethodFromName = UI_FOR_NAMED_RIG_ARGS.get(arg, None)
buildMethodFromType = getBuildUIMethodForObject(argValue) or MelTextField
buildMethod = (
buildMethodFromName or buildMethodFromType
) # prioritize the method from name over the method from type
argWidget = buildMethod(argsForm)
argWidget.setValue(argValue)
argsUIs.append(argLbl)
argsUIs.append(argWidget)
self.argUIs[arg] = argWidget
try:
inc = 1.0 / len(argsUIs)
except ZeroDivisionError:
inc = 1.0
for n, ui in enumerate(argsUIs):
p = n * inc
if n:
argsForm(e=True, ac=((ui, "left", 5, argsUIs[n - 1])))
else:
argsForm(e=True, af=((ui, "left", 0)))
# finally build the "rebuild" button
reButt = MelButton(self, l="rebuild", c=self.on_rebuild, w=100)
# perform layout...
self.setWidth(50)
argsForm.setWidth(50)
self(
e=True,
af=((lbl, "left", 0), (reButt, "right", 0)),
ac=((argsForm, "left", 5, lbl), (argsForm, "right", 0, reButt)),
)
def GetPartName( cls ): ''' can be used to get a "nice" name for the part class ''' if cls.DISPLAY_NAME is None: return camelCaseToNice( cls.__name__ ) return cls.DISPLAY_NAME
def populate(self):
#remove any existing children
self.clear()
part = self.part
assert isinstance(part, SkeletonPart)
#pimp out the UI
lbl = MelButton(self, l=part.getPartName(), w=140, c=self.on_select)
#grab the args the rigging method takes
argsForm = MelFormLayout(self)
argsUIs = []
buildKwargs = part.getBuildKwargs()
for arg in self.ARGS_TO_HIDE:
buildKwargs.pop(arg, None)
for arg, argValue in buildKwargs.iteritems():
argLbl = MelLabel(argsForm, label=names.camelCaseToNice(arg))
#determine the function to use for building the UI for the arg
buildMethodFromName = UI_FOR_NAMED_RIG_ARGS.get(arg, None)
buildMethodFromType = getBuildUIMethodForObject(
argValue) or MelTextField
buildMethod = buildMethodFromName or buildMethodFromType #prioritize the method from name over the method from type
argWidget = buildMethod(argsForm)
argWidget.setValue(argValue)
argsUIs.append(argLbl)
argsUIs.append(argWidget)
self.argUIs[arg] = argWidget
try:
inc = 1.0 / len(argsUIs)
except ZeroDivisionError:
inc = 1.0
for n, ui in enumerate(argsUIs):
p = n * inc
if n:
argsForm(e=True, ac=((ui, 'left', 5, argsUIs[n - 1])))
else:
argsForm(e=True, af=((ui, 'left', 0)))
#finally build the "rebuild" button
reButt = MelButton(self, l="rebuild", c=self.on_rebuild, w=100)
#perform layout...
self.setWidth(50)
argsForm.setWidth(50)
self(e=True,
af=((lbl, 'left', 0), (reButt, 'right', 0)),
ac=((argsForm, 'left', 5, lbl), (argsForm, 'right', 0, reButt)))
def populate(self):
if not bool(self.rigTypes):
self.setVisibility(False)
return
#remove any existing children
self.clearArgs()
part = self.part
rigKwargs = part.getRigKwargs()
#build the disable and optionbox for the rig method
disableState = rigKwargs.get('disable', False)
#grab the args the rigging method takes
argsForm = self.UI_argsForm
argsUIs = []
rigMethodName = rigKwargs.get('rigMethodName',
self.rigTypes[0].__name__)
rigClass = rigPrimitives.RigPart.GetNamedSubclass(rigMethodName)
if rigClass is None:
rigClass = part.RigTypes[0]
zeroWeightTypes = MelCheckBox, MelOptionMenu
argNamesAndDefaults = rigClass.GetDefaultBuildKwargList()
for arg, default in argNamesAndDefaults:
argValue = rigKwargs.get(arg, default)
argLbl = MelLabel(argsForm, label=names.camelCaseToNice(arg))
#determine the function to use for building the UI for the arg
buildMethodFromName = UI_FOR_NAMED_RIG_ARGS.get(arg, None)
buildMethodFromType = getBuildUIMethodForObject(
default) or MelTextField
buildMethod = buildMethodFromName or buildMethodFromType
argWidget = buildMethod(argsForm)
argWidget.setValue(argValue, False)
argWidget.setChangeCB(self.on_argCB)
argLbl.enable(not disableState)
argWidget.enable(not disableState)
argsUIs.append(argLbl)
argsUIs.append(argWidget)
self.argUIs[arg] = argWidget
#if there are no args - create an empty text widget otherwise maya will crash. yay!
argsUIs.append(MelLabel(argsForm, label=''))
argsForm.layout()
def populate(self):
if not bool(self.rigTypes):
self.setVisibility(False)
return
# remove any existing children
self.clearArgs()
part = self.part
rigKwargs = part.getRigKwargs()
# build the disable and optionbox for the rig method
disableState = rigKwargs.get("disable", False)
# grab the args the rigging method takes
argsForm = self.UI_argsForm
argsUIs = []
rigMethodName = rigKwargs.get("rigMethodName", self.rigTypes[0].__name__)
rigClass = rigPrimitives.RigPart.GetNamedSubclass(rigMethodName)
if rigClass is None:
rigClass = part.RigTypes[0]
zeroWeightTypes = MelCheckBox, MelOptionMenu
argNamesAndDefaults = rigClass.GetDefaultBuildKwargList()
for arg, default in argNamesAndDefaults:
argValue = rigKwargs.get(arg, default)
argLbl = MelLabel(argsForm, label=names.camelCaseToNice(arg))
# determine the function to use for building the UI for the arg
buildMethodFromName = UI_FOR_NAMED_RIG_ARGS.get(arg, None)
buildMethodFromType = getBuildUIMethodForObject(default) or MelTextField
buildMethod = buildMethodFromName or buildMethodFromType
argWidget = buildMethod(argsForm)
argWidget.setValue(argValue, False)
argWidget.setChangeCB(self.on_argCB)
argLbl.enable(not disableState)
argWidget.enable(not disableState)
argsUIs.append(argLbl)
argsUIs.append(argWidget)
self.argUIs[arg] = argWidget
# if there are no args - create an empty text widget otherwise maya will crash. yay!
argsUIs.append(MelLabel(argsForm, label=""))
argsForm.layout()
def addParent( self, parent, name=None ): #sanity check - make sure the parent isn't either the control, the space, already in the list or a child of either the control or space if parent == self.getControl(): return elif parent == self.getSpace(): return if name is None: name = spaceSwitching.getSpaceName( self.getControl(), parent ) if name is None: name = names.camelCaseToNice( str( parent ) ) self.UI_parents.append( [parent, name] )
def __init__(self, parent, preset):
MelHSingleStretchLayout.__init__(self, parent)
self.preset = preset
MelButton(self, l=names.camelCaseToNice(preset.name()), c=self.on_load, w=160)
spac = MelSpacer(self)
MelButton(self, l="Overwrite", c=self.on_overwrite, w=100)
if preset.locale == presetsUI.LOCAL:
self.UI_publish = MelButton(self, l="Publish", c=self.on_publish, w=100)
# MelButton( self, l='Delete', c=self.on_remove, w=100 )
self.setStretchWidget(spac)
self.layout()
def __init__( self, parent, preset ): MelHSingleStretchLayout.__init__( self, parent ) self.preset = preset MelButton( self, l=names.camelCaseToNice( preset.name() ), c=self.on_load, w=160 ) spac = MelSpacer( self ) MelButton( self, l='Overwrite', c=self.on_overwrite, w=100 ) if preset.locale == presetsUI.LOCAL: self.UI_publish = MelButton( self, l='Publish', c=self.on_publish, w=100 ) #MelButton( self, l='Delete', c=self.on_remove, w=100 ) self.setStretchWidget( spac ) self.layout()
def generateNiceControlName( control ): niceName = getNiceName( control ) if niceName is not None: return niceName try: rigPart = RigPart.InitFromItem( control ) if rigPart is None: raise RigPartError( "null" ) controlName = rigPart.getControlName( control ) except RigPartError: controlName = str( control ) controlName = Name( controlName ) parity = controlName.get_parity() if parity == Parity.LEFT: controlName = 'Left '+ str( stripParity( controlName ) ) if parity == Parity.RIGHT: controlName = 'Right '+ str( stripParity( controlName ) ) else: controlName = str( controlName ) return camelCaseToNice( controlName )
def add(src,
tgt,
name=None,
space=None,
maintainOffset=True,
nodeWithParentAttr=None,
skipTranslationAxes=(),
skipRotationAxes=(),
constraintType=CONSTRAINT_PARENT):
global AXES
AXES = list(AXES)
if space is None:
space = listRelatives(src, p=True, pa=True)[0]
if nodeWithParentAttr is None:
nodeWithParentAttr = src
if not name:
name = getNiceName(tgt)
if name is None:
name = camelCaseToNice(str(tgt))
#if there is an existing constraint, check to see if the target already exists in its target list - if it does, return the condition used it uses
attrState(space, ('t', 'r'), lock=False)
existingConstraint = findConstraint(src)
if existingConstraint:
constraintType = nodeType(existingConstraint)
constraintFunc = getattr(cmd, constraintType)
targetsOnConstraint = constraintFunc(existingConstraint,
q=True,
tl=True)
if tgt in targetsOnConstraint:
idx = targetsOnConstraint.index(tgt)
aliases = constraintFunc(existingConstraint,
q=True,
weightAliasList=True)
cons = listConnections('%s.%s' %
(existingConstraint, aliases[idx]),
type='condition',
d=False)
return cons[0]
#when skip axes are specified maya doesn't handle things properly - so make sure
#ALL transform channels are connected, and remove unwanted channels at the end...
preT, preR = getAttr('%s.t' % space)[0], getAttr('%s.r' % space)[0]
if existingConstraint:
chans = CONSTRAINT_CHANNELS[constraintType]
for channel, constraintAttr in zip(*chans):
for axis in AXES:
spaceAttr = '%s.%s%s' % (space, channel, axis)
conAttr = '%s.%s%s' % (existingConstraint, constraintAttr,
axis)
if not isConnected(conAttr, spaceAttr):
connectAttr(conAttr, spaceAttr)
#get the names for the parents from the parent enum attribute
cmdOptionKw = {'mo': True} if maintainOffset else {}
if objExists('%s.parent' % nodeWithParentAttr):
srcs, names = getSpaceTargetsNames(src)
addAttr('%s.parent' % nodeWithParentAttr,
e=True,
enumName=':'.join(names + [name]))
#if we're building a pointConstraint instead of a parent constraint AND we already
#have spaces on the object, we need to turn the -mo flag off regardless of what the
#user set it to, as the pointConstraint maintain offset has different behaviour to
#the parent constraint
if constraintType in (CONSTRAINT_POINT, CONSTRAINT_ORIENT):
cmdOptionKw = {}
else:
addAttr(nodeWithParentAttr, ln='parent', at="enum", en=name)
setAttr('%s.parent' % nodeWithParentAttr, keyable=True)
#now build the constraint
constraintFunction = getattr(cmd, constraintType)
constraint = constraintFunction(tgt, space, **cmdOptionKw)[0]
weightAliasList = constraintFunction(constraint,
q=True,
weightAliasList=True)
targetCount = len(weightAliasList)
constraintAttr = weightAliasList[-1]
condition = shadingNode('condition', asUtility=True)
condition = rename(
condition,
'%s_to_space_%s#' % (cleanShortName(src), cleanShortName(tgt)))
setAttr('%s.secondTerm' % condition, targetCount - 1)
setAttr('%s.colorIfTrue' % condition, 1, 1, 1)
setAttr('%s.colorIfFalse' % condition, 0, 0, 0)
connectAttr('%s.parent' % nodeWithParentAttr, '%s.firstTerm' % condition)
connectAttr('%s.outColorR' % condition,
'%s.%s' % (constraint, constraintAttr))
#find out what symbol to use to find the parent attribute
parentAttrIdx = 0
if not apiExtensions.cmpNodes(space, src):
srcTrigger = triggered.Trigger(src)
parentAttrIdx = srcTrigger.connect(nodeWithParentAttr)
#add the zooObjMenu commands to the object for easy space switching
Trigger.CreateMenu(src, "parent to %s" % name,
ChangeSpaceCmd.Create(targetCount - 1, parentAttrIdx))
#when skip axes are specified maya doesn't handle things properly - so make sure
#ALL transform channels are connected, and remove unwanted channels at the end...
for axis, value in zip(AXES, preT):
if axis in skipTranslationAxes:
attr = '%s.t%s' % (space, axis)
delete(attr, icn=True)
setAttr(attr, value)
for axis, value in zip(AXES, preR):
if axis in skipRotationAxes:
attr = '%s.r%s' % (space, axis)
delete(attr, icn=True)
setAttr(attr, value)
#make the space node non-keyable and lock visibility
attrState(space, ['t', 'r', 's'], lock=True)
attrState(space, 'v', *control.HIDE)
return condition
def add( src, tgt,
name=None,
space=None,
maintainOffset=True,
nodeWithParentAttr=None,
skipTranslationAxes=(),
skipRotationAxes=(),
constraintType=CONSTRAINT_PARENT ):
global AXES
AXES = list( AXES )
if space is None:
space = listRelatives( src, p=True, pa=True )[ 0 ]
if nodeWithParentAttr is None:
nodeWithParentAttr = src
if not name:
name = getNiceName( tgt )
if name is None:
name = camelCaseToNice( str( tgt ) )
#if there is an existing constraint, check to see if the target already exists in its target list - if it does, return the condition used it uses
attrState( space, ('t', 'r'), lock=False )
existingConstraint = findConstraint( src )
if existingConstraint:
constraintType = nodeType( existingConstraint )
constraintFunc = getattr( cmd, constraintType )
targetsOnConstraint = constraintFunc( existingConstraint, q=True, tl=True )
if tgt in targetsOnConstraint:
idx = targetsOnConstraint.index( tgt )
aliases = constraintFunc( existingConstraint, q=True, weightAliasList=True )
cons = listConnections( '%s.%s' % (existingConstraint, aliases[ idx ]), type='condition', d=False )
return cons[ 0 ]
#when skip axes are specified maya doesn't handle things properly - so make sure
#ALL transform channels are connected, and remove unwanted channels at the end...
preT, preR = getAttr( '%s.t' % space )[0], getAttr( '%s.r' % space )[0]
if existingConstraint:
chans = CONSTRAINT_CHANNELS[ constraintType ]
for channel, constraintAttr in zip( *chans ):
for axis in AXES:
spaceAttr = '%s.%s%s' %( space, channel, axis)
conAttr = '%s.%s%s' % (existingConstraint, constraintAttr, axis)
if not isConnected( conAttr, spaceAttr ):
connectAttr( conAttr, spaceAttr )
#get the names for the parents from the parent enum attribute
cmdOptionKw = { 'mo': True } if maintainOffset else {}
if objExists( '%s.parent' % nodeWithParentAttr ):
srcs, names = getSpaceTargetsNames( src )
addAttr( '%s.parent' % nodeWithParentAttr, e=True, enumName=':'.join( names + [name] ) )
#if we're building a pointConstraint instead of a parent constraint AND we already
#have spaces on the object, we need to turn the -mo flag off regardless of what the
#user set it to, as the pointConstraint maintain offset has different behaviour to
#the parent constraint
if constraintType in ( CONSTRAINT_POINT, CONSTRAINT_ORIENT ):
cmdOptionKw = {}
else:
addAttr( nodeWithParentAttr, ln='parent', at="enum", en=name )
setAttr( '%s.parent' % nodeWithParentAttr, keyable=True )
#now build the constraint
constraintFunction = getattr( cmd, constraintType )
constraint = constraintFunction( tgt, space, **cmdOptionKw )[ 0 ]
weightAliasList = constraintFunction( constraint, q=True, weightAliasList=True )
targetCount = len( weightAliasList )
constraintAttr = weightAliasList[ -1 ]
condition = shadingNode( 'condition', asUtility=True )
condition = rename( condition, '%s_to_space_%s#' % (cleanShortName( src ), cleanShortName( tgt )) )
setAttr( '%s.secondTerm' % condition, targetCount-1 )
setAttr( '%s.colorIfTrue' % condition, 1, 1, 1 )
setAttr( '%s.colorIfFalse' % condition, 0, 0, 0 )
connectAttr( '%s.parent' % nodeWithParentAttr, '%s.firstTerm' % condition )
connectAttr( '%s.outColorR' % condition, '%s.%s' % (constraint, constraintAttr) )
#find out what symbol to use to find the parent attribute
parentAttrIdx = 0
if not apiExtensions.cmpNodes( space, src ):
parentAttrIdx = triggered.addConnect( src, nodeWithParentAttr )
#add the zooObjMenu commands to the object for easy space switching
Trigger.CreateMenu( src,
"parent to %s" % name,
ChangeSpaceCmd.Create( targetCount-1, parentAttrIdx ) )
#when skip axes are specified maya doesn't handle things properly - so make sure
#ALL transform channels are connected, and remove unwanted channels at the end...
for axis, value in zip( AXES, preT ):
if axis in skipTranslationAxes:
attr = '%s.t%s' % (space, axis)
delete( attr, icn=True )
setAttr( attr, value )
for axis, value in zip( AXES, preR ):
if axis in skipRotationAxes:
attr = '%s.r%s' % (space, axis)
delete( attr, icn=True )
setAttr( attr, value )
#make the space node non-keyable and lock visibility
attrState( space, [ 't', 'r', 's' ], lock=True )
attrState( space, 'v', *control.HIDE )
return condition
