def connect(obj, name_level):
'''
Hook the given obj's visibility to the `name` attribute on the sharedShape.
If the attr doesn't exist, it will be made.
Optionanal `level` will determine when the `obj` will become visible. For
example, 2 will not be visible at 1, but will at 2 and higher.
'''
name, level = name_level # Probably should just update this eventually to be 3 params
orig = obj
zero = pdil.dagObj.zero(obj, apply=False, make=False)
if zero:
obj = zero
shape = get()
if not shape:
warning(
'Unable to add vis control, no object exists named "main" or tagged with ".fossilMainControl"'
)
return
log.debug('Applying vis control to {}, was given {} using {}'.format(
obj, orig, shape))
plug = shape + '.' + name
if not cmds.objExists(plug):
cmds.addAttr(shape, ln=name, at='short', min=0, max=level, dv=1)
cmds.setAttr(shape + '.' + name, cb=True)
elif cmds.getAttr(shape + '.' + name, type=True) not in [
'bool', 'double', 'float', 'long', 'short'
]:
warning(
'{0} is not a good name for a vis group since the sharedShape has an attr already that is of the wrong type'
.format(name))
return
if cmds.addAttr(plug, q=True, max=True) < level:
cmds.addAttr(plug, e=True, max=level)
if level == 1:
connectAttr(plug, obj.visibility.name(), f=True)
else:
connectAttr(getConditionNode(plug, level).outColorR,
obj.visibility,
f=True)
obj.visibility.setKeyable(False)
if not pdil.sharedShape.find(orig, VIS_NODE_TYPE):
pdil.sharedShape.use(orig, shape)
# If we have a main controller, put the container in a subgroup to make
# the main group more organized.
''' 2021-11-25 Taking this out, I think redoing the hierarchery is probalby more confusing that having lots there.
def getSwitcherPlug(obj):
'''
Will return either a string like "Arm_L_FKIK_SWITCH" or empty string if no
switcher is found.
Can't use pymel to avoid warnings of invalid node (nurbs with no cvs).
This also means listRelatives returns None instead of []. Lame.
'''
shapes = cmds.listRelatives(str(obj), type='nurbsCurve', f=True)
if shapes:
for shape in shapes:
# Check for new style switch first
if cmds.objExists(shape + '.IkSwitch'):
return shape + '.IkSwitch'
# Fallback to old style switch
if cmds.objExists(shape + '.kinematicSwitch'):
attr = cmds.listAttr(shape, ud=1, st='*_Switch')
return cmds.ls(shape, l=False)[0] + '.' + attr[0] # noqa
return ''
def connect(obj, name):
'''
Hook the given obj's visibility to the `name` attribute on the sharedShape.
If the attr doesn't exist, it will be made.
'''
orig = obj
zero = core.dagObj.zero(obj, apply=False, make=False)
if zero:
obj = zero
shape = get()
plug = shape + '.' + name
if not cmds.objExists(plug):
cmds.addAttr(shape, ln=name, at='short', min=0, max=1, dv=1)
cmds.setAttr(shape + '.' + name, cb=True)
elif cmds.getAttr(shape + '.' + name, type=True) not in [
'bool', 'double', 'float', 'long', 'short'
]:
warning(
'{0} is not a good name for a vis group since the sharedShape has an attr already that is of the wrong type'
.format(name))
return
connectAttr(plug, obj.visibility.name(), f=True)
obj.visibility.setKeyable(False)
# If we have a main controller, put the container in a subgroup to make
# the main group more organized.
visGroupName = '_vis_' + name
if isinstance(orig, nodeApi.RigController):
if shortName(orig.container.getParent()) != visGroupName:
orig.setGroup(visGroupName)
Hook the given obj's visibility to the `name` attribute on the sharedShape.
If the attr doesn't exist, it will be made.
Optionanal `level` will determine when the `obj` will become visible. For
example, 2 will not be visible at 1, but will at 2 and higher.
'''
orig = obj
zero = core.dagObj.zero(obj, apply=False, make=False)
if zero:
obj = zero
shape = get()
plug = shape + '.' + name
if not cmds.objExists( plug ):
cmds.addAttr( shape, ln=name, at='short', min=0, max=level, dv=1 )
cmds.setAttr( shape + '.' + name, cb=True )
elif cmds.getAttr( shape + '.' + name, type=True) not in ['bool', 'double', 'float', 'long', 'short']:
warning( '{0} is not a good name for a vis group since the sharedShape has an attr already that is of the wrong type'.format(name) )
return
if cmds.addAttr(plug, q=True, max=True) < level:
cmds.addAttr(plug, e=True, max=level)
if level == 1:
connectAttr( plug, obj.visibility.name(), f=True)
else:
connectAttr( getConditionNode(plug, level).outColorR, obj.visibility, f=True)
obj.visibility.setKeyable(False)
def ikFkSwitch(name, ikRigController, ikPlugs, fkRigController, fkPlugs):
'''
Adds "<name>_Switch" attr to the shared shape, controlling the given ik
and fk plugs (which are expected to be 0=off and 1.0=on)
.. todo::
* Verify an existing switch doesn't exist
'''
# When adding the switcher shape, re-add the shared shape to ensure it is last.
lib.sharedShape.remove(ikRigController)
shape = lib.sharedShape._makeSharedShape(ikRigController,
'%s_FKIK_SWITCH' % name,
'kinematicSwitch')
lib.sharedShape.use(ikRigController)
for key, obj in itertools.chain(ikRigController.subControl.items(),
fkRigController.subControl.items()):
cmds.parent(shape, obj.longName(), add=True, shape=True)
lib.sharedShape.remove(obj)
lib.sharedShape.use(obj)
cmds.parent(shape, fkRigController.longName(), add=True, shape=True)
lib.sharedShape.remove(fkRigController)
lib.sharedShape.use(fkRigController)
attrName = 'IkSwitch'
if not cmds.objExists(shape + '.' + attrName):
cmds.addAttr(shape,
longName=attrName,
at='double',
min=0.0,
max=1.0,
k=True)
plug = Attribute(shape + '.' + attrName)
if ikPlugs and fkPlugs:
# Optionally skip in case retrofitting, where these connections are made elsewhere
for ikPlug in ikPlugs:
plug >> ikPlug
if not ikRigController.visibility.isDestination():
plug >> ikRigController.visibility
ikRigController.visibility.setKeyable(False)
for name, ctrl in ikRigController.subControl.items():
if not ctrl.visibility.isDestination():
plug >> ctrl.visibility
ctrl.visibility.setKeyable(False)
opposite = core.math.opposite(plug)
for fkPlug in fkPlugs:
opposite >> fkPlug
if not fkRigController.visibility.isDestination():
opposite >> fkRigController.visibility
fkRigController.visibility.setKeyable(False)
for name, ctrl in fkRigController.subControl.items():
if not ctrl.visibility.isDestination():
opposite >> ctrl.visibility
ctrl.visibility.setKeyable(False)
return plug