class LegComponentRig(LegComponent):
"""Leg Component"""
def __init__(self, name='leg', parent=None):
Profiler.getInstance().push("Construct Leg Rig Component:" + name)
super(LegComponentRig, self).__init__(name, parent)
# =========
# Controls
# =========
# Femur
self.femurFKCtrlSpace = CtrlSpace('femurFK', parent=self.ctrlCmpGrp)
self.femurFKCtrl = Control('femurFK', parent=self.femurFKCtrlSpace, shape="cube")
self.femurFKCtrl.alignOnXAxis()
self.femurFKCtrl.lockTranslation(True, True, True)
self.femurFKCtrl.lockScale(True, True, True)
# Shin
self.shinFKCtrlSpace = CtrlSpace('shinFK', parent=self.femurFKCtrl)
self.shinFKCtrl = Control('shinFK', parent=self.shinFKCtrlSpace, shape="cube")
self.shinFKCtrl.alignOnXAxis()
self.shinFKCtrl.lockTranslation(True, True, True)
self.shinFKCtrl.lockScale(True, True, True)
# IK Handle
self.legIKCtrlSpace = CtrlSpace('IK', parent=self.ctrlCmpGrp)
self.legIKCtrl = Control('IK', parent=self.legIKCtrlSpace, shape="pin")
self.legIKCtrl.lockScale(True, True, True)
# Add Component Params to IK control
legSettingsAttrGrp = AttributeGroup("DisplayInfo_LegSettings", parent=self.legIKCtrl)
legDrawDebugInputAttr = BoolAttribute('drawDebug', value=False, parent=legSettingsAttrGrp)
self.legRightSideInputAttr = BoolAttribute('rightSide', value=False, parent=legSettingsAttrGrp)
self.legBone0LenInputAttr = ScalarAttribute('bone0Len', value=1.0, parent=legSettingsAttrGrp)
self.legBone1LenInputAttr = ScalarAttribute('bone1Len', value=1.0, parent=legSettingsAttrGrp)
legIKBlendInputAttr = ScalarAttribute('ikblend', value=1.0, minValue=0.0, maxValue=1.0, parent=legSettingsAttrGrp)
# Util Objects
self.ikRootPosition = Transform("ikRootPosition", parent=self.ctrlCmpGrp)
# Connect Input Attrs
self.drawDebugInputAttr.connect(legDrawDebugInputAttr)
# Connect Output Attrs
self.drawDebugOutputAttr.connect(legDrawDebugInputAttr)
self.ikBlendOutputAttr.connect(legIKBlendInputAttr)
# UpV
self.legUpVCtrlSpace = CtrlSpace('UpV', parent=self.ctrlCmpGrp)
self.legUpVCtrl = Control('UpV', parent=self.legUpVCtrlSpace, shape="triangle")
self.legUpVCtrl.alignOnZAxis()
self.legUpVCtrl.lockRotation(True, True, True)
self.legUpVCtrl.lockScale(True, True, True)
# ==========
# Deformers
# ==========
deformersLayer = self.getOrCreateLayer('deformers')
self.defCmpGrp = ComponentGroup(self.getName(), self, parent=deformersLayer)
self.addItem('defCmpGrp', self.defCmpGrp)
femurDef = Joint('femur', parent=self.defCmpGrp)
femurDef.setComponent(self)
kneeDef = Joint('knee', parent=self.defCmpGrp)
kneeDef.setComponent(self)
shinDef = Joint('shin', parent=self.defCmpGrp)
shinDef.setComponent(self)
# ==============
# Constrain I/O
# ==============
# Constraint inputs
self.legIKCtrlSpaceInputConstraint = PoseConstraint('_'.join([self.legIKCtrlSpace.getName(), 'To', self.globalSRTInputTgt.getName()]))
self.legIKCtrlSpaceInputConstraint.setMaintainOffset(True)
self.legIKCtrlSpaceInputConstraint.addConstrainer(self.globalSRTInputTgt)
self.legIKCtrlSpace.addConstraint(self.legIKCtrlSpaceInputConstraint)
self.legUpVCtrlSpaceInputConstraint = PoseConstraint('_'.join([self.legUpVCtrlSpace.getName(), 'To', self.globalSRTInputTgt.getName()]))
self.legUpVCtrlSpaceInputConstraint.setMaintainOffset(True)
self.legUpVCtrlSpaceInputConstraint.addConstrainer(self.globalSRTInputTgt)
self.legUpVCtrlSpace.addConstraint(self.legUpVCtrlSpaceInputConstraint)
self.legRootInputConstraint = PoseConstraint('_'.join([self.femurFKCtrlSpace.getName(), 'To', self.legPelvisInputTgt.getName()]))
self.legRootInputConstraint.setMaintainOffset(True)
self.legRootInputConstraint.addConstrainer(self.legPelvisInputTgt)
self.femurFKCtrlSpace.addConstraint(self.legRootInputConstraint)
self.ikRootPosInputConstraint = PoseConstraint('_'.join([self.ikRootPosition.getName(), 'To', self.legPelvisInputTgt.getName()]))
self.ikRootPosInputConstraint.setMaintainOffset(True)
self.ikRootPosInputConstraint.addConstrainer(self.legPelvisInputTgt)
self.ikRootPosition.addConstraint(self.ikRootPosInputConstraint)
# Constraint outputs
self.legEndFKOutputConstraint = PoseConstraint('_'.join([self.legEndFKOutputTgt.getName(), 'To', self.shinFKCtrl.getName()]))
self.legEndFKOutputConstraint.setMaintainOffset(True)
self.legEndFKOutputConstraint.addConstrainer(self.shinFKCtrl)
self.legEndFKOutputTgt.addConstraint(self.legEndFKOutputConstraint)
self.ikHandleOutputConstraint = PoseConstraint('_'.join([self.ikHandleOutputTgt.getName(), 'To', self.legIKCtrl.getName()]))
self.ikHandleOutputConstraint.setMaintainOffset(True)
self.ikHandleOutputConstraint.addConstrainer(self.legIKCtrl)
self.ikHandleOutputTgt.addConstraint(self.ikHandleOutputConstraint)
# ===============
# Add Splice Ops
# ===============
# Add Leg Splice Op
self.legIKKLOp = KLOperator('legKLOp', 'TwoBoneIKSolver', 'Kraken')
self.addOperator(self.legIKKLOp)
# Add Att Inputs
self.legIKKLOp.setInput('drawDebug', self.drawDebugInputAttr)
self.legIKKLOp.setInput('rigScale', self.rigScaleInputAttr)
self.legIKKLOp.setInput('bone0Len', self.legBone0LenInputAttr)
self.legIKKLOp.setInput('bone1Len', self.legBone1LenInputAttr)
self.legIKKLOp.setInput('ikblend', legIKBlendInputAttr)
self.legIKKLOp.setInput('rightSide', self.legRightSideInputAttr)
# Add Xfo Inputs
self.legIKKLOp.setInput('root', self.ikRootPosition)
self.legIKKLOp.setInput('bone0FK', self.femurFKCtrl)
self.legIKKLOp.setInput('bone1FK', self.shinFKCtrl)
self.legIKKLOp.setInput('ikHandle', self.legIKTargetInputTgt)
self.legIKKLOp.setInput('upV', self.legUpVCtrl)
# Add Xfo Outputs
self.legIKKLOp.setOutput('bone0Out', self.femurOutputTgt)
self.legIKKLOp.setOutput('bone1Out', self.shinOutputTgt)
self.legIKKLOp.setOutput('bone2Out', self.legEndOutputTgt)
self.legIKKLOp.setOutput('midJointOut', self.kneeOutputTgt)
# Add Leg Deformer Splice Op
self.outputsToDeformersKLOp = KLOperator('legDeformerKLOp', 'MultiPoseConstraintSolver', 'Kraken')
self.addOperator(self.outputsToDeformersKLOp)
# Add Att Inputs
self.outputsToDeformersKLOp.setInput('drawDebug', self.drawDebugInputAttr)
self.outputsToDeformersKLOp.setInput('rigScale', self.rigScaleInputAttr)
# Add Xfo Inputs
self.outputsToDeformersKLOp.setInput('constrainers', [self.femurOutputTgt, self.kneeOutputTgt, self.shinOutputTgt])
# Add Xfo Outputs
self.outputsToDeformersKLOp.setOutput('constrainees', [femurDef, kneeDef, shinDef])
Profiler.getInstance().pop()
# =============
# Data Methods
# =============
def loadData(self, data=None):
"""Load a saved guide representation from persisted data.
Arguments:
data -- object, The JSON data object.
Return:
True if successful.
"""
super(LegComponentRig, self).loadData( data )
createIKHandle = data.get('createIKHandle')
femurXfo = data.get('femurXfo')
kneeXfo = data.get('kneeXfo')
handleXfo = data.get('handleXfo')
upVXfo = data.get('upVXfo')
femurLen = data.get('femurLen')
shinLen = data.get('shinLen')
self.femurFKCtrlSpace.xfo = femurXfo
self.femurFKCtrl.xfo = femurXfo
self.femurFKCtrl.scalePoints(Vec3(femurLen, 1.75, 1.75))
self.femurOutputTgt.xfo = femurXfo
self.shinOutputTgt.xfo = kneeXfo
self.shinFKCtrlSpace.xfo = kneeXfo
self.shinFKCtrl.xfo = kneeXfo
self.shinFKCtrl.scalePoints(Vec3(shinLen, 1.5, 1.5))
self.legEndFKOutputTgt.xfo.tr = handleXfo.tr
self.legEndFKOutputTgt.xfo.ori = kneeXfo.ori
self.ikHandleOutputTgt.xfo = handleXfo
self.ikRootPosition.xfo = femurXfo
self.legIKCtrlSpace.xfo = handleXfo
self.legIKCtrl.xfo = handleXfo
if self.getLocation() == 'R':
self.legIKCtrl.rotatePoints(0, 90, 0)
self.legIKCtrl.translatePoints(Vec3(-1.0, 0.0, 0.0))
else:
self.legIKCtrl.rotatePoints(0, -90, 0)
self.legIKCtrl.translatePoints(Vec3(1.0, 0.0, 0.0))
self.legUpVCtrlSpace.xfo.tr = upVXfo.tr
self.legUpVCtrl.xfo.tr = upVXfo.tr
self.legRightSideInputAttr.setValue(self.getLocation() is 'R')
self.legBone0LenInputAttr.setMin(0.0)
self.legBone0LenInputAttr.setMax(femurLen * 3.0)
self.legBone0LenInputAttr.setValue(femurLen)
self.legBone1LenInputAttr.setMin(0.0)
self.legBone1LenInputAttr.setMax(shinLen * 3.0)
self.legBone1LenInputAttr.setValue(shinLen)
self.legPelvisInputTgt.xfo = femurXfo
self.legIKTargetInputTgt.xfo = handleXfo
# TODO: We need the Rig class to be modified to handle the ability to
# query if the ports are connected during loadData. Currently just a
# place holder until that happens.
# If IK Target input is not connected, switch to legIKCtrl
# ikTargetInput = self.getInputByName('ikTarget')
# if not ikTargetInput.isConnected():
# self.legIKKLOp.setInput('ikHandle', self.legIKCtrl)
# Eval Input Constraints
self.ikRootPosInputConstraint.evaluate()
self.legIKCtrlSpaceInputConstraint.evaluate()
self.legUpVCtrlSpaceInputConstraint.evaluate()
self.legRootInputConstraint.evaluate()
# Eval Operators
self.legIKKLOp.evaluate()
self.outputsToDeformersKLOp.evaluate()
# Eval Output Constraints
self.legEndFKOutputConstraint.evaluate()
self.ikHandleOutputConstraint.evaluate()
class ArmComponentRig(ArmComponent):
"""Arm Component Rig"""
def __init__(self, name='arm', parent=None):
Profiler.getInstance().push("Construct Arm Rig Component:" + name)
super(ArmComponentRig, self).__init__(name, parent)
# =========
# Controls
# =========
# Bicep
self.bicepFKCtrlSpace = CtrlSpace('bicepFK', parent=self.ctrlCmpGrp)
self.bicepFKCtrl = Control('bicepFK',
parent=self.bicepFKCtrlSpace,
shape="cube")
self.bicepFKCtrl.alignOnXAxis()
self.bicepFKCtrl.lockScale(True, True, True)
self.bicepFKCtrl.lockTranslation(True, True, True)
# Forearm
self.forearmFKCtrlSpace = CtrlSpace('forearmFK',
parent=self.bicepFKCtrl)
self.forearmFKCtrl = Control('forearmFK',
parent=self.forearmFKCtrlSpace,
shape="cube")
self.forearmFKCtrl.alignOnXAxis()
self.forearmFKCtrl.lockScale(True, True, True)
self.forearmFKCtrl.lockTranslation(True, True, True)
# Arm IK
self.armIKCtrlSpace = CtrlSpace('IK', parent=self.ctrlCmpGrp)
self.armIKCtrl = Control('IK',
parent=self.armIKCtrlSpace,
shape="jack")
self.armIKCtrl.scalePoints(Vec3(2.0, 2.0, 2.0))
self.armIKCtrl.lockScale(True, True, True)
# Add Params to IK control
armSettingsAttrGrp = AttributeGroup("DisplayInfo_ArmSettings",
parent=self.armIKCtrl)
self.armDebugInputAttr = BoolAttribute('drawDebug',
value=False,
parent=armSettingsAttrGrp)
self.armBone0LenInputAttr = ScalarAttribute('bone1Len',
value=0.0,
parent=armSettingsAttrGrp)
self.armBone1LenInputAttr = ScalarAttribute('bone2Len',
value=0.0,
parent=armSettingsAttrGrp)
self.armIKBlendInputAttr = ScalarAttribute('fkik',
value=0.0,
minValue=0.0,
maxValue=1.0,
parent=armSettingsAttrGrp)
# Util Objects
self.ikRootPosition = Transform("ikPosition", parent=self.ctrlCmpGrp)
# Connect Input Attrs
self.drawDebugInputAttr.connect(self.armDebugInputAttr)
# Connect Output Attrs
self.drawDebugOutputAttr.connect(self.armDebugInputAttr)
self.ikBlendOutputAttr.connect(self.armIKBlendInputAttr)
# UpV
self.armUpVCtrlSpace = CtrlSpace('UpV', parent=self.ctrlCmpGrp)
self.armUpVCtrl = Control('UpV',
parent=self.armUpVCtrlSpace,
shape="triangle")
self.armUpVCtrl.alignOnZAxis()
self.armUpVCtrl.rotatePoints(180, 0, 0)
self.armIKCtrl.lockScale(True, True, True)
self.armIKCtrl.lockRotation(True, True, True)
# ==========
# Deformers
# ==========
self.deformersLayer = self.getOrCreateLayer('deformers')
self.defCmpGrp = ComponentGroup(self.getName(),
self,
parent=self.deformersLayer)
self.addItem('defCmpGrp', self.defCmpGrp)
self.bicepDef = Joint('bicep', parent=self.defCmpGrp)
self.bicepDef.setComponent(self)
self.elbowDef = Joint('elbow', parent=self.defCmpGrp)
self.elbowDef.setComponent(self)
self.forearmDef = Joint('forearm', parent=self.defCmpGrp)
self.forearmDef.setComponent(self)
self.wristDef = Joint('wrist', parent=self.defCmpGrp)
self.wristDef.setComponent(self)
# ==============
# Constrain I/O
# ==============
# Constraint inputs
self.armIKCtrlSpaceInputConstraint = PoseConstraint('_'.join([
self.armIKCtrlSpace.getName(), 'To',
self.globalSRTInputTgt.getName()
]))
self.armIKCtrlSpaceInputConstraint.setMaintainOffset(True)
self.armIKCtrlSpaceInputConstraint.addConstrainer(
self.globalSRTInputTgt)
self.armIKCtrlSpace.addConstraint(self.armIKCtrlSpaceInputConstraint)
self.armUpVCtrlSpaceInputConstraint = PoseConstraint('_'.join([
self.armUpVCtrlSpace.getName(), 'To',
self.globalSRTInputTgt.getName()
]))
self.armUpVCtrlSpaceInputConstraint.setMaintainOffset(True)
self.armUpVCtrlSpaceInputConstraint.addConstrainer(
self.globalSRTInputTgt)
self.armUpVCtrlSpace.addConstraint(self.armUpVCtrlSpaceInputConstraint)
self.armRootInputConstraint = PoseConstraint('_'.join([
self.bicepFKCtrlSpace.getName(), 'To',
self.rootInputTgt.getName()
]))
self.armRootInputConstraint.setMaintainOffset(True)
self.armRootInputConstraint.addConstrainer(self.rootInputTgt)
self.bicepFKCtrlSpace.addConstraint(self.armRootInputConstraint)
self.ikPosInputConstraint = PoseConstraint('_'.join(
[self.ikRootPosition.getName(), 'To',
self.rootInputTgt.getName()]))
self.ikPosInputConstraint.setMaintainOffset(True)
self.ikPosInputConstraint.addConstrainer(self.rootInputTgt)
self.ikRootPosition.addConstraint(self.ikPosInputConstraint)
# Constraint outputs
# ===============
# Add Splice Ops
# ===============
# Add Splice Op
self.armSolverKLOperator = KLOperator('ikSolver', 'TwoBoneIKSolver',
'Kraken')
self.addOperator(self.armSolverKLOperator)
# Add Att Inputs
self.armSolverKLOperator.setInput('drawDebug', self.drawDebugInputAttr)
self.armSolverKLOperator.setInput('rigScale', self.rigScaleInputAttr)
self.armSolverKLOperator.setInput('bone0Len',
self.armBone0LenInputAttr)
self.armSolverKLOperator.setInput('bone1Len',
self.armBone1LenInputAttr)
self.armSolverKLOperator.setInput('ikblend', self.armIKBlendInputAttr)
self.armSolverKLOperator.setInput('rightSide', self.rightSideInputAttr)
# Add Xfo Inputs
self.armSolverKLOperator.setInput('root', self.ikRootPosition)
self.armSolverKLOperator.setInput('bone0FK', self.bicepFKCtrl)
self.armSolverKLOperator.setInput('bone1FK', self.forearmFKCtrl)
self.armSolverKLOperator.setInput('ikHandle', self.armIKCtrl)
self.armSolverKLOperator.setInput('upV', self.armUpVCtrl)
# Add Xfo Outputs
self.armSolverKLOperator.setOutput('bone0Out', self.bicepOutputTgt)
self.armSolverKLOperator.setOutput('bone1Out', self.forearmOutputTgt)
self.armSolverKLOperator.setOutput('bone2Out', self.wristOutputTgt)
self.armSolverKLOperator.setOutput('midJointOut', self.elbowOutputTgt)
# Add Deformer Splice Op
self.outputsToDeformersKLOp = KLOperator('defConstraint',
'MultiPoseConstraintSolver',
'Kraken')
self.addOperator(self.outputsToDeformersKLOp)
# Add Att Inputs
self.outputsToDeformersKLOp.setInput('drawDebug',
self.drawDebugInputAttr)
self.outputsToDeformersKLOp.setInput('rigScale',
self.rigScaleInputAttr)
# Add Xfo Inputs
self.outputsToDeformersKLOp.setInput('constrainers', [
self.bicepOutputTgt, self.elbowOutputTgt, self.forearmOutputTgt,
self.wristOutputTgt
])
# Add Xfo Outputs
self.outputsToDeformersKLOp.setOutput(
'constrainees',
[self.bicepDef, self.elbowDef, self.forearmDef, self.wristDef])
Profiler.getInstance().pop()
def loadData(self, data=None):
"""Load a saved guide representation from persisted data.
Arguments:
data -- object, The JSON data object.
Return:
True if successful.
"""
super(ArmComponentRig, self).loadData(data)
bicepXfo = data.get('bicepXfo')
forearmXfo = data.get('forearmXfo')
wristXfo = data.get('wristXfo')
upVXfo = data.get('upVXfo')
bicepLen = data.get('bicepLen')
forearmLen = data.get('forearmLen')
bicepFKCtrlSize = data.get('bicepFKCtrlSize')
forearmFKCtrlSize = data.get('forearmFKCtrlSize')
self.rootInputTgt.xfo.tr = bicepXfo.tr
self.bicepFKCtrlSpace.xfo = bicepXfo
self.bicepFKCtrl.xfo = bicepXfo
self.bicepFKCtrl.scalePoints(
Vec3(bicepLen, bicepFKCtrlSize, bicepFKCtrlSize))
self.forearmFKCtrlSpace.xfo = forearmXfo
self.forearmFKCtrl.xfo = forearmXfo
self.forearmFKCtrl.scalePoints(
Vec3(forearmLen, forearmFKCtrlSize, forearmFKCtrlSize))
self.ikRootPosition.xfo = bicepXfo
self.armIKCtrlSpace.xfo.tr = wristXfo.tr
self.armIKCtrl.xfo.tr = wristXfo.tr
if self.getLocation() == "R":
self.armIKCtrl.rotatePoints(0, 90, 0)
else:
self.armIKCtrl.rotatePoints(0, -90, 0)
self.armUpVCtrlSpace.xfo.tr = upVXfo.tr
self.armUpVCtrl.xfo.tr = upVXfo.tr
self.rightSideInputAttr.setValue(self.getLocation() is 'R')
self.armBone0LenInputAttr.setMin(0.0)
self.armBone0LenInputAttr.setMax(bicepLen * 3.0)
self.armBone0LenInputAttr.setValue(bicepLen)
self.armBone1LenInputAttr.setMin(0.0)
self.armBone1LenInputAttr.setMax(forearmLen * 3.0)
self.armBone1LenInputAttr.setValue(forearmLen)
# Outputs
self.bicepOutputTgt.xfo = bicepXfo
self.forearmOutputTgt.xfo = forearmXfo
self.wristOutputTgt.xfo = wristXfo
# Eval Constraints
self.ikPosInputConstraint.evaluate()
self.armIKCtrlSpaceInputConstraint.evaluate()
self.armUpVCtrlSpaceInputConstraint.evaluate()
self.armRootInputConstraint.evaluate()
self.armRootInputConstraint.evaluate()
# Eval Operators
self.armSolverKLOperator.evaluate()
self.outputsToDeformersKLOp.evaluate()