def updateNumDeformers(self, count):
"""Generate the guide controls for the variable outputes array.
Arguments:
count -- object, The number of joints inthe chain.
Return:
True if successful.
"""
if count == 0:
raise IndexError("'count' must be > 0")
vertebraeOutputs = self.tailVertebraeOutput.getTarget()
if count > len(vertebraeOutputs):
for i in xrange(len(vertebraeOutputs), count):
debugCtrl = Control('spine' + str(i+1).zfill(2), parent=self.outputHrcGrp, shape="vertebra")
debugCtrl.rotatePoints(0, -90, 0)
debugCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
debugCtrl.setColor('turqoise')
vertebraeOutputs.append(debugCtrl)
elif count < len(vertebraeOutputs):
numExtraCtrls = len(vertebraeOutputs) - count
for i in xrange(numExtraCtrls):
extraCtrl = vertebraeOutputs.pop()
self.outputHrcGrp.removeChild(extraCtrl)
return True
def updateNumDeformers(self, count):
"""Generate the guide controls for the variable outputes array.
Arguments:
count -- object, The number of joints inthe chain.
Return:
True if successful.
"""
if count == 0:
raise IndexError("'count' must be > 0")
vertebraeOutputs = self.tailVertebraeOutput.getTarget()
if count > len(vertebraeOutputs):
for i in xrange(len(vertebraeOutputs), count):
debugCtrl = Control('spine' + str(i+1).zfill(2), parent=self.outputHrcGrp, shape="vertebra")
debugCtrl.rotatePoints(0, -90, 0)
debugCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
debugCtrl.setColor('turqoise')
vertebraeOutputs.append(debugCtrl)
elif count < len(vertebraeOutputs):
numExtraCtrls = len(vertebraeOutputs) - count
for i in xrange(numExtraCtrls):
extraCtrl = vertebraeOutputs.pop()
self.outputHrcGrp.removeChild(extraCtrl)
return True
def __init__(self, name='twist', parent=None):
Profiler.getInstance().push('Construct Spine Guide Component:' + name)
super(TwistComponentGuide, self).__init__(name, parent)
# =========
# Controls
# ========
guideSettingsAttrGrp = AttributeGroup('GuideSettings', parent=self)
self.numDeformersAttr = IntegerAttribute('numDeformers',
value=1,
minValue=0,
maxValue=20,
parent=guideSettingsAttrGrp)
self.blendBiasAttr = ScalarAttribute('blendBias',
value=0.0,
minValue=0,
maxValue=1.0,
parent=guideSettingsAttrGrp)
# Guide Controls
triangleCtrl = Control('triangle', shape='triangle')
triangleCtrl.rotatePoints(90, 0, 0)
triangleCtrl.scalePoints(Vec3(0.25, 0.25, 0.25))
triangleCtrl.scalePoints(Vec3(1.0, 0.5, 1.0))
triangleCtrl.translatePoints(Vec3(0.0, 1.25, 0.0))
triangleCtrl.rotatePoints(0, 90, 0)
self.originCtrl = Control('origin',
parent=self.ctrlCmpGrp,
shape='circle')
self.originCtrl.rotatePoints(90, 0, 0)
self.originCtrl.rotatePoints(0, 90, 0)
self.originCtrl.appendCurveData(triangleCtrl.getCurveData())
self.insertCtrl = Control('insert',
parent=self.ctrlCmpGrp,
shape='circle')
self.insertCtrl.rotatePoints(90, 0, 0)
self.insertCtrl.rotatePoints(0, 90, 0)
self.insertCtrl.appendCurveData(triangleCtrl.getCurveData())
self.default_data = {
'name': name,
'location': 'M',
'blendBias': 0.5,
'originXfo': Xfo(Vec3(0.0, 0.0, 0.0)),
'insertXfo': Xfo(Vec3(5.0, 0.0, 0.0)),
'numDeformers': 5
}
self.loadData(self.default_data)
Profiler.getInstance().pop()
def __init__(self, name='twist', parent=None):
Profiler.getInstance().push('Construct Spine Guide Component:' + name)
super(TwistComponentGuide, self).__init__(name, parent)
# =========
# Controls
# ========
guideSettingsAttrGrp = AttributeGroup('GuideSettings', parent=self)
self.numDeformersAttr = IntegerAttribute('numDeformers', value=1, minValue=0, maxValue=20, parent=guideSettingsAttrGrp)
self.blendBiasAttr = ScalarAttribute('blendBias', value=0.0, minValue=0, maxValue=1.0, parent=guideSettingsAttrGrp)
# Guide Controls
triangleCtrl = Control('triangle', shape='triangle')
triangleCtrl.rotatePoints(90, 0, 0)
triangleCtrl.scalePoints(Vec3(0.25, 0.25, 0.25))
triangleCtrl.scalePoints(Vec3(1.0, 0.5, 1.0))
triangleCtrl.translatePoints(Vec3(0.0, 1.25, 0.0))
triangleCtrl.rotatePoints(0, 90, 0)
self.originCtrl = Control('origin', parent=self.ctrlCmpGrp, shape='circle')
self.originCtrl.rotatePoints(90, 0, 0)
self.originCtrl.rotatePoints(0, 90, 0)
self.originCtrl.appendCurveData(triangleCtrl.getCurveData())
self.insertCtrl = Control('insert', parent=self.ctrlCmpGrp, shape='circle')
self.insertCtrl.rotatePoints(90, 0, 0)
self.insertCtrl.rotatePoints(0, 90, 0)
self.insertCtrl.appendCurveData(triangleCtrl.getCurveData())
self.default_data = {
'name': name,
'location': 'M',
'blendBias': 0.5,
'originXfo': Xfo(Vec3(0.0, 0.0, 0.0)),
'insertXfo': Xfo(Vec3(5.0, 0.0, 0.0)),
'numDeformers': 5
}
self.loadData(self.default_data)
Profiler.getInstance().pop()
def updateNumLowDeformers(self, countLow):
if countLow == 0:
raise IndexError("'count' must be > 0")
#Lip Low
lidLowOutputs = self.eyelidLowOutput.getTarget()
if countLow > len(lidLowOutputs):
for i in xrange(len(lidLowOutputs), countLow):
debugLowCtrl = Control('Lid_Low_' + str(i+1).zfill(2), parent=self.outputHrcGrp, shape="sphere")
debugLowCtrl.rotatePoints(90, -90, 180)
debugLowCtrl.scalePoints(Vec3(0.01, 0.01, 0.01))
debugLowCtrl.setColor("yellowLight")
lidLowOutputs.append(debugLowCtrl)
elif countLow < len(lidLowOutputs):
numExtraLowCtrls = len(lidLowOutputs) - countLow
for i in xrange(numExtraLowCtrls):
extraLowCtrl = lidLowOutputs.pop()
self.outputHrcGrp.removeChild(extraLowCtrl)
return True
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()
# Shin
self.shinFKCtrlSpace = CtrlSpace('shinFK', parent=self.femurFKCtrl)
self.shinFKCtrl = Control('shinFK', parent=self.shinFKCtrlSpace, shape="cube")
self.shinFKCtrl.alignOnXAxis()
# Ankle
self.legIKCtrlSpace = CtrlSpace('IK', parent=self.ctrlCmpGrp)
self.legIKCtrl = Control('IK', parent=self.legIKCtrlSpace, shape="pin")
# FK Foot
self.footCtrlSpace = CtrlSpace('foot', parent=self.ctrlCmpGrp)
self.footCtrl = Control('foot', parent=self.footCtrlSpace, shape="cube")
self.footCtrl.alignOnXAxis()
# FK Toe
self.toeCtrlSpace = CtrlSpace('toe', parent=self.footCtrl)
self.toeCtrl = Control('toe', parent=self.toeCtrlSpace, shape="cube")
self.toeCtrl.alignOnXAxis()
# Rig Ref objects
self.footRefSrt = Locator('footRef', parent=self.ctrlCmpGrp)
# Add Component Params to IK control
footSettingsAttrGrp = AttributeGroup("DisplayInfo_FootSettings", parent=self.footCtrl)
footLinkToWorldInputAttr = ScalarAttribute('linkToWorld', 1.0, maxValue=1.0, parent=footSettingsAttrGrp)
# Add Component Params to IK control
legSettingsAttrGrp = AttributeGroup("DisplayInfo_LegSettings", parent=self.legIKCtrl)
legDrawDebugInputAttr = BoolAttribute('drawDebug', 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)
legSoftIKInputAttr = BoolAttribute('softIK', value=True, parent=legSettingsAttrGrp)
legSoftDistInputAttr = ScalarAttribute('softDist', value=0.0, minValue=0.0, parent=legSettingsAttrGrp)
legStretchInputAttr = BoolAttribute('stretch', value=True, parent=legSettingsAttrGrp)
legStretchBlendInputAttr = ScalarAttribute('stretchBlend', value=0.0, minValue=0.0, maxValue=1.0, parent=legSettingsAttrGrp)
self.drawDebugInputAttr.connect(legDrawDebugInputAttr)
# UpV
self.legUpVCtrlSpace = CtrlSpace('UpV', parent=self.ctrlCmpGrp)
self.legUpVCtrl = Control('UpV', parent=self.legUpVCtrlSpace, shape="triangle")
self.legUpVCtrl.alignOnZAxis()
# ==========
# Deformers
# ==========
deformersLayer = self.getOrCreateLayer('deformers')
self.defCmpGrp = ComponentGroup(self.getName(), self, parent=deformersLayer)
femurDef = Joint('femur', parent=self.defCmpGrp)
femurDef.setComponent(self)
shinDef = Joint('shin', parent=self.defCmpGrp)
shinDef.setComponent(self)
ankleDef = Joint('ankle', parent=self.defCmpGrp)
ankleDef.setComponent(self)
self.footDef = Joint('foot', parent=self.defCmpGrp)
self.footDef.setComponent(self)
self.toeDef = Joint('toe', parent=self.defCmpGrp)
self.toeDef.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.legIKCtrl.getName(), 'To', self.legPelvisInputTgt.getName()]))
self.legRootInputConstraint.setMaintainOffset(True)
self.legRootInputConstraint.addConstrainer(self.legPelvisInputTgt)
self.femurFKCtrlSpace.addConstraint(self.legRootInputConstraint)
# Constraint outputs
self.footOutputConstraint = PoseConstraint('_'.join([self.footOutputTgt.getName(), 'To', self.footCtrl.getName()]))
self.footOutputConstraint.addConstrainer(self.footCtrl)
self.footOutputTgt.addConstraint(self.footOutputConstraint)
self.toeOutputConstraint = PoseConstraint('_'.join([self.toeOutputTgt.getName(), 'To', self.toeCtrl.getName()]))
self.toeOutputConstraint.addConstrainer(self.toeCtrl)
self.toeOutputTgt.addConstraint(self.toeOutputConstraint)
# ===============
# Add Splice Ops
# ===============
# Add Leg Splice Op
self.legIKSpliceOp = SpliceOperator('legSpliceOp', 'TwoBoneIKSolver', 'Kraken')
self.addOperator(self.legIKSpliceOp)
# Add Att Inputs
self.legIKSpliceOp.setInput('drawDebug', self.drawDebugInputAttr)
self.legIKSpliceOp.setInput('rigScale', self.rigScaleInputAttr)
self.legIKSpliceOp.setInput('bone0Len', self.legBone0LenInputAttr)
self.legIKSpliceOp.setInput('bone1Len', self.legBone1LenInputAttr)
self.legIKSpliceOp.setInput('ikblend', legIKBlendInputAttr)
self.legIKSpliceOp.setInput('softIK', legSoftIKInputAttr)
self.legIKSpliceOp.setInput('softDist', legSoftDistInputAttr)
self.legIKSpliceOp.setInput('stretch', legStretchInputAttr)
self.legIKSpliceOp.setInput('stretchBlend', legStretchBlendInputAttr)
self.legIKSpliceOp.setInput('rightSide', self.rightSideInputAttr)
# Add Xfo Inputs
self.legIKSpliceOp.setInput('root', self.legPelvisInputTgt)
self.legIKSpliceOp.setInput('bone0FK', self.femurFKCtrl)
self.legIKSpliceOp.setInput('bone1FK', self.shinFKCtrl)
self.legIKSpliceOp.setInput('ikHandle', self.legIKCtrl)
self.legIKSpliceOp.setInput('upV', self.legUpVCtrl)
# Add Xfo Outputs
self.legIKSpliceOp.setOutput('bone0Out', self.femurOutputTgt)
self.legIKSpliceOp.setOutput('bone1Out', self.shinOutputTgt)
self.legIKSpliceOp.setOutput('bone2Out', self.legEndXfoOutputTgt)
# Add Leg Deformer Splice Op
self.outputsToDeformersSpliceOp = SpliceOperator('legDeformerSpliceOp', 'MultiPoseConstraintSolver', 'Kraken')
self.addOperator(self.outputsToDeformersSpliceOp)
# Add Att Inputs
self.outputsToDeformersSpliceOp.setInput('drawDebug', self.drawDebugInputAttr)
self.outputsToDeformersSpliceOp.setInput('rigScale', self.rigScaleInputAttr)
# Add Xfo Inputs
self.outputsToDeformersSpliceOp.setInput('constrainers', [self.femurOutputTgt, self.shinOutputTgt, self.legEndXfoOutputTgt])
# Add Xfo Outputs
self.outputsToDeformersSpliceOp.setOutput('constrainees', [femurDef, shinDef, ankleDef])
# Add Foot Deformer Splice Op
self.footDefSpliceOp = SpliceOperator('footDeformerSpliceOp', 'PoseConstraintSolver', 'Kraken')
self.addOperator(self.footDefSpliceOp)
# Add Att Inputs
self.footDefSpliceOp.setInput('drawDebug', self.drawDebugInputAttr)
self.footDefSpliceOp.setInput('rigScale', self.rigScaleInputAttr)
# Add Xfo Inputs)
self.footDefSpliceOp.setInput('constrainer', self.footOutputTgt)
# Add Xfo Outputs
self.footDefSpliceOp.setOutput('constrainee', self.footDef)
# Add Toe Deformer Splice Op
self.toeDefSpliceOp = SpliceOperator('toeDeformerSpliceOp', 'PoseConstraintSolver', 'Kraken')
self.addOperator(self.toeDefSpliceOp)
# Add Att Inputs
self.toeDefSpliceOp.setInput('drawDebug', self.drawDebugInputAttr)
self.toeDefSpliceOp.setInput('rigScale', self.rigScaleInputAttr)
# Add Xfo Inputs
self.toeDefSpliceOp.setInput('constrainer', self.toeOutputTgt)
# Add Xfo Outputs
self.toeDefSpliceOp.setOutput('constrainee', self.toeDef)
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 )
self.femurFKCtrlSpace.xfo = data['femurXfo']
self.femurFKCtrl.xfo = data['femurXfo']
self.femurFKCtrl.scalePoints(Vec3(data['femurLen'], 1.75, 1.75))
self.femurOutputTgt.xfo = data['femurXfo']
self.shinOutputTgt.xfo = data['kneeXfo']
self.shinFKCtrlSpace.xfo = data['kneeXfo']
self.shinFKCtrl.xfo = data['kneeXfo']
self.shinFKCtrl.scalePoints(Vec3(data['shinLen'], 1.5, 1.5))
self.footCtrlSpace.xfo.tr = data['ankleXfo'].tr
self.footCtrl.xfo.tr = data['ankleXfo'].tr
self.toeCtrlSpace.xfo = data['toeXfo']
self.toeCtrl.xfo = data['toeXfo']
self.legIKCtrlSpace.xfo.tr = data['ankleXfo'].tr
self.legIKCtrl.xfo.tr = data['ankleXfo'].tr
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 = data['upVXfo']
self.legUpVCtrl.xfo = data['upVXfo']
self.rightSideInputAttr.setValue(self.getLocation() is 'R')
self.legBone0LenInputAttr.setMin(0.0)
self.legBone0LenInputAttr.setMax(data['femurLen'] * 3.0)
self.legBone0LenInputAttr.setValue(data['femurLen'])
self.legBone1LenInputAttr.setMin(0.0)
self.legBone1LenInputAttr.setMax(data['shinLen'] * 3.0)
self.legBone1LenInputAttr.setValue(data['shinLen'])
self.legPelvisInputTgt.xfo = data['femurXfo']
# Eval Constraints
self.legIKCtrlSpaceInputConstraint.evaluate()
self.legUpVCtrlSpaceInputConstraint.evaluate()
self.legRootInputConstraint.evaluate()
self.footOutputConstraint.evaluate()
self.toeOutputConstraint.evaluate()
# Eval Operators
self.legIKSpliceOp.evaluate()
self.outputsToDeformersSpliceOp.evaluate()
self.footDefSpliceOp.evaluate()
class FabriceTailGuide(FabriceTail):
"""Fabrice Tail Component Guide"""
def __init__(self, name='tail', parent=None):
Profiler.getInstance().push("Construct Fabrice Tail Guide Component:" + name)
super(FabriceTailGuide, self).__init__(name, parent)
# =========
# Controls
# ========
guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self)
self.numDeformersAttr = IntegerAttribute('numDeformers', value=1, minValue=0, maxValue=20, parent=guideSettingsAttrGrp)
self.numDeformersAttr.setValueChangeCallback(self.updateNumDeformers)
# Guide Controls
self.tailBaseCtrl = Control('tailBase', parent=self.ctrlCmpGrp, shape='sphere')
self.tailBaseCtrl.scalePoints(Vec3(1.2, 1.2, 1.2))
self.tailBaseCtrl.lockScale(x=True, y=True, z=True)
self.tailBaseCtrl.setColor("turqoise")
self.tailBaseHandleCtrl = Control('tailBaseHandle', parent=self.ctrlCmpGrp, shape='pin')
self.tailBaseHandleCtrl.rotatePoints(90, 0, 0)
self.tailBaseHandleCtrl.translatePoints(Vec3(0, 1.0, 0))
self.tailBaseHandleCtrl.lockScale(x=True, y=True, z=True)
self.tailBaseHandleCtrl.setColor("turqoise")
self.tailEndHandleCtrl = Control('tailEndHandle', parent=self.ctrlCmpGrp, shape='pin')
self.tailEndHandleCtrl.rotatePoints(90, 0, 0)
self.tailEndHandleCtrl.translatePoints(Vec3(0, 1.0, 0))
self.tailEndHandleCtrl.lockScale(x=True, y=True, z=True)
self.tailEndHandleCtrl.setColor("turqoise")
self.tailEndCtrl = Control('tailEnd', parent=self.ctrlCmpGrp, shape='pin')
self.tailEndCtrl.rotatePoints(90, 0, 0)
self.tailEndCtrl.translatePoints(Vec3(0, 1.0, 0))
self.tailEndCtrl.lockScale(x=True, y=True, z=True)
self.tailEndCtrl.setColor("turqoise")
# ===============
# Add Splice Ops
# ===============
# Add Tail Splice Op
self.bezierSpineSpliceOp = SpliceOperator('spineGuideSpliceOp', 'BezierSpineSolver', 'Kraken', alwaysEval=True)
self.bezierSpineSpliceOp.setOutput('outputs', self.tailVertebraeOutput.getTarget())
self.addOperator(self.bezierSpineSpliceOp)
# Add Att Inputs
self.bezierSpineSpliceOp.setInput('drawDebug', self.drawDebugInputAttr)
self.bezierSpineSpliceOp.setInput('rigScale', self.rigScaleInputAttr)
self.bezierSpineSpliceOp.setInput('length', self.lengthInputAttr)
# Add Xfo Inputs
self.bezierSpineSpliceOp.setInput('base', self.tailBaseCtrl)
self.bezierSpineSpliceOp.setInput('baseHandle', self.tailBaseHandleCtrl)
self.bezierSpineSpliceOp.setInput('tipHandle', self.tailEndHandleCtrl)
self.bezierSpineSpliceOp.setInput('tip', self.tailEndCtrl)
self.loadData({
'name': name,
'location': 'M',
'tailBasePos': Vec3(0.0, 0.65, -3.1),
'tailBaseHandlePos': Vec3(0.0, 0.157, -4.7),
'tailBaseHandleCtrlCrvData': self.tailBaseHandleCtrl.getCurveData(),
'tailEndHandlePos': Vec3(0.0, 0.0625, -6.165),
'tailEndHandleCtrlCrvData': self.tailEndHandleCtrl.getCurveData(),
'tailEndPos': Vec3(0.0, -0.22, -7.42),
'tailEndCtrlCrvData': self.tailEndCtrl.getCurveData(),
'numDeformers': 6
})
Profiler.getInstance().pop()
# ==========
# Callbacks
# ==========
def updateNumDeformers(self, count):
"""Generate the guide controls for the variable outputes array.
Arguments:
count -- object, The number of joints inthe chain.
Return:
True if successful.
"""
if count == 0:
raise IndexError("'count' must be > 0")
vertebraeOutputs = self.tailVertebraeOutput.getTarget()
if count > len(vertebraeOutputs):
for i in xrange(len(vertebraeOutputs), count):
debugCtrl = Control('spine' + str(i+1).zfill(2), parent=self.outputHrcGrp, shape="vertebra")
debugCtrl.rotatePoints(0, -90, 0)
debugCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
debugCtrl.setColor('turqoise')
vertebraeOutputs.append(debugCtrl)
elif count < len(vertebraeOutputs):
numExtraCtrls = len(vertebraeOutputs) - count
for i in xrange(numExtraCtrls):
extraCtrl = vertebraeOutputs.pop()
self.outputHrcGrp.removeChild(extraCtrl)
return True
# =============
# Data Methods
# =============
def saveData(self):
"""Save the data for the component to be persisted.
Return:
The JSON data object
"""
data = super(FabriceTailGuide, self).saveData()
data['tailBasePos'] = self.tailBaseCtrl.xfo.tr
data['tailBaseHandlePos'] = self.tailBaseHandleCtrl.xfo.tr
data['tailBaseHandleCtrlCrvData'] = self.tailBaseHandleCtrl.getCurveData()
data['tailEndHandlePos'] = self.tailEndHandleCtrl.xfo.tr
data['tailEndHandleCtrlCrvData'] = self.tailEndHandleCtrl.getCurveData()
data['tailEndPos'] = self.tailEndCtrl.xfo.tr
data['tailEndCtrlCrvData'] = self.tailEndCtrl.getCurveData()
data['numDeformers'] = self.numDeformersAttr.getValue()
return data
def loadData(self, data):
"""Load a saved guide representation from persisted data.
Arguments:
data -- object, The JSON data object.
Return:
True if successful.
"""
super(FabriceTailGuide, self).loadData( data )
self.tailBaseCtrl.xfo.tr = data["tailBasePos"]
self.tailBaseHandleCtrl.xfo.tr = data["tailBaseHandlePos"]
self.tailBaseHandleCtrl.setCurveData(data['tailBaseHandleCtrlCrvData'])
self.tailEndHandleCtrl.xfo.tr = data["tailEndHandlePos"]
self.tailEndHandleCtrl.setCurveData(data['tailEndHandleCtrlCrvData'])
self.tailEndCtrl.xfo.tr = data["tailEndPos"]
self.tailEndCtrl.setCurveData(data['tailEndCtrlCrvData'])
self.numDeformersAttr.setValue(data["numDeformers"])
length = data["tailBasePos"].distanceTo(data["tailBaseHandlePos"]) + data["tailBaseHandlePos"].distanceTo(data["tailEndHandlePos"]) + data["tailEndHandlePos"].distanceTo(data["tailEndPos"])
self.lengthInputAttr.setMax(length * 3.0)
self.lengthInputAttr.setValue(length)
self.bezierSpineSpliceOp.evaluate()
return True
def getRigBuildData(self):
"""Returns the Guide data used by the Rig Component to define the layout of the final rig.
Return:
The JSON rig data object.
"""
data = super(FabriceTailGuide, self).getRigBuildData()
data['tailBasePos'] = self.tailBaseCtrl.xfo.tr
data['tailBaseHandlePos'] = self.tailBaseHandleCtrl.xfo.tr
data['tailBaseHandleCtrlCrvData'] = self.tailBaseHandleCtrl.getCurveData()
data['tailEndHandlePos'] = self.tailEndHandleCtrl.xfo.tr
data['tailEndHandleCtrlCrvData'] = self.tailEndHandleCtrl.getCurveData()
data['tailEndPos'] = self.tailEndCtrl.xfo.tr
data['tailEndCtrlCrvData'] = self.tailEndCtrl.getCurveData()
data['numDeformers'] = self.numDeformersAttr.getValue()
return data
# ==============
# Class Methods
# ==============
@classmethod
def getComponentType(cls):
"""Enables introspection of the class prior to construction to determine if it is a guide component.
Return:
The true if this component is a guide component.
"""
return 'Guide'
@classmethod
def getRigComponentClass(cls):
"""Returns the corresponding rig component class for this guide component class
Return:
The rig component class.
"""
return FabriceTailRig
class StretchyLimbComponentRig(StretchyLimbComponent):
"""StretchyLimb Component"""
def __init__(self, name='limb', parent=None):
Profiler.getInstance().push("Construct StretchyLimb Rig Component:" +
name)
super(StretchyLimbComponentRig, self).__init__(name, parent)
# =========
# Controls
# =========
# Upper (FK)
self.upperFKCtrlSpace = CtrlSpace('upperFK', parent=self.ctrlCmpGrp)
self.upperFKCtrl = Control('upperFK',
parent=self.upperFKCtrlSpace,
shape="cube")
self.upperFKCtrl.alignOnXAxis()
# Lower (FK)
self.lowerFKCtrlSpace = CtrlSpace('lowerFK', parent=self.upperFKCtrl)
self.lowerFKCtrl = Control('lowerFK',
parent=self.lowerFKCtrlSpace,
shape="cube")
self.lowerFKCtrl.alignOnXAxis()
# End (IK)
self.limbIKCtrlSpace = CtrlSpace('IK', parent=self.ctrlCmpGrp)
self.limbIKCtrl = Control('IK',
parent=self.limbIKCtrlSpace,
shape="pin")
# Add Component Params to IK control
# TODO: Move these separate control
limbSettingsAttrGrp = AttributeGroup(
"DisplayInfo_StretchyLimbSettings", parent=self.limbIKCtrl)
limbDrawDebugInputAttr = BoolAttribute('drawDebug',
value=False,
parent=limbSettingsAttrGrp)
self.limbBone0LenInputAttr = ScalarAttribute(
'bone0Len', value=1.0, parent=limbSettingsAttrGrp)
self.limbBone1LenInputAttr = ScalarAttribute(
'bone1Len', value=1.0, parent=limbSettingsAttrGrp)
limbIKBlendInputAttr = ScalarAttribute('ikblend',
value=1.0,
minValue=0.0,
maxValue=1.0,
parent=limbSettingsAttrGrp)
limbSoftIKInputAttr = BoolAttribute('softIK',
value=True,
parent=limbSettingsAttrGrp)
limbSoftRatioInputAttr = ScalarAttribute('softRatio',
value=0.0,
minValue=0.0,
maxValue=1.0,
parent=limbSettingsAttrGrp)
limbStretchInputAttr = BoolAttribute('stretch',
value=True,
parent=limbSettingsAttrGrp)
limbStretchBlendInputAttr = ScalarAttribute('stretchBlend',
value=0.0,
minValue=0.0,
maxValue=1.0,
parent=limbSettingsAttrGrp)
limbSlideInputAttr = ScalarAttribute('slide',
value=0.0,
minValue=-1.0,
maxValue=1.0,
parent=limbSettingsAttrGrp)
limbPinInputAttr = ScalarAttribute('pin',
value=0.0,
minValue=0.0,
maxValue=1.0,
parent=limbSettingsAttrGrp)
self.rightSideInputAttr = BoolAttribute('rightSide',
value=False,
parent=limbSettingsAttrGrp)
self.drawDebugInputAttr.connect(limbDrawDebugInputAttr)
# UpV (IK Pole Vector)
self.limbUpVCtrlSpace = CtrlSpace('UpV', parent=self.ctrlCmpGrp)
self.limbUpVCtrl = Control('UpV',
parent=self.limbUpVCtrlSpace,
shape="triangle")
self.limbUpVCtrl.alignOnZAxis()
# ==========
# Deformers
# ==========
deformersLayer = self.getOrCreateLayer('deformers')
self.defCmpGrp = ComponentGroup(self.getName(),
self,
parent=deformersLayer)
self.addItem('defCmpGrp', self.defCmpGrp)
upperDef = Joint('upper', parent=self.defCmpGrp)
upperDef.setComponent(self)
lowerDef = Joint('lower', parent=self.defCmpGrp)
lowerDef.setComponent(self)
endDef = Joint('end', parent=self.defCmpGrp)
endDef.setComponent(self)
# ==============
# Constrain I/O
# ==============
# Constraint inputs
self.limbIKCtrlSpaceInputConstraint = PoseConstraint('_'.join([
self.limbIKCtrlSpace.getName(), 'To',
self.globalSRTInputTgt.getName()
]))
self.limbIKCtrlSpaceInputConstraint.setMaintainOffset(True)
self.limbIKCtrlSpaceInputConstraint.addConstrainer(
self.globalSRTInputTgt)
self.limbIKCtrlSpace.addConstraint(self.limbIKCtrlSpaceInputConstraint)
self.limbUpVCtrlSpaceInputConstraint = PoseConstraint('_'.join([
self.limbUpVCtrlSpace.getName(), 'To',
self.globalSRTInputTgt.getName()
]))
self.limbUpVCtrlSpaceInputConstraint.setMaintainOffset(True)
self.limbUpVCtrlSpaceInputConstraint.addConstrainer(
self.globalSRTInputTgt)
self.limbUpVCtrlSpace.addConstraint(
self.limbUpVCtrlSpaceInputConstraint)
self.limbRootInputConstraint = PoseConstraint('_'.join([
self.limbIKCtrl.getName(), 'To',
self.limbParentInputTgt.getName()
]))
self.limbRootInputConstraint.setMaintainOffset(True)
self.limbRootInputConstraint.addConstrainer(self.limbParentInputTgt)
self.upperFKCtrlSpace.addConstraint(self.limbRootInputConstraint)
# ===============
# Add Splice Ops
# ===============
# Add StretchyLimb Splice Op
self.limbIKKLOp = KLOperator('limbKLOp', 'TwoBoneStretchyIKSolver',
'Kraken')
self.addOperator(self.limbIKKLOp)
# Add Att Inputs
self.limbIKKLOp.setInput('drawDebug', self.drawDebugInputAttr)
self.limbIKKLOp.setInput('rigScale', self.rigScaleInputAttr)
self.limbIKKLOp.setInput('bone0Len', self.limbBone0LenInputAttr)
self.limbIKKLOp.setInput('bone1Len', self.limbBone1LenInputAttr)
self.limbIKKLOp.setInput('ikblend', limbIKBlendInputAttr)
self.limbIKKLOp.setInput('softIK', limbSoftIKInputAttr)
self.limbIKKLOp.setInput('softRatio', limbSoftRatioInputAttr)
self.limbIKKLOp.setInput('stretch', limbStretchInputAttr)
self.limbIKKLOp.setInput('stretchBlend', limbStretchBlendInputAttr)
self.limbIKKLOp.setInput('slide', limbSlideInputAttr)
self.limbIKKLOp.setInput('pin', limbPinInputAttr)
self.limbIKKLOp.setInput('rightSide', self.rightSideInputAttr)
# Add Xfo Inputs
self.limbIKKLOp.setInput('root', self.limbParentInputTgt)
self.limbIKKLOp.setInput('bone0FK', self.upperFKCtrl)
self.limbIKKLOp.setInput('bone1FK', self.lowerFKCtrl)
self.limbIKKLOp.setInput('ikHandle', self.limbIKCtrl)
self.limbIKKLOp.setInput('upV', self.limbUpVCtrl)
# Add Xfo Outputs
self.limbIKKLOp.setOutput('bone0Out', self.limbUpperOutputTgt)
self.limbIKKLOp.setOutput('bone1Out', self.limbLowerOutputTgt)
self.limbIKKLOp.setOutput('bone2Out', self.limbEndOutputTgt)
# =====================
# Connect the deformers
# =====================
# Add StretchyLimb Deformer Splice Op
self.outputsToDeformersKLOp = KLOperator('limbDeformerKLOp',
'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.limbUpperOutputTgt, self.limbLowerOutputTgt,
self.limbEndOutputTgt
])
# Add Xfo Outputs
self.outputsToDeformersKLOp.setOutput('constrainees',
[upperDef, lowerDef, endDef])
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(StretchyLimbComponentRig, self).loadData(data)
upperXfo = data.get('upperXfo')
upperLen = data.get('upperLen')
lowerXfo = data.get('lowerXfo')
lowerLen = data.get('lowerLen')
endXfo = data.get('endXfo')
upVXfo = data.get('upVXfo')
self.upperFKCtrlSpace.xfo = upperXfo
self.upperFKCtrl.xfo = upperXfo
self.upperFKCtrl.scalePoints(Vec3(upperLen, 1.75, 1.75))
self.limbUpperOutputTgt.xfo = upperXfo
self.limbLowerOutputTgt.xfo = lowerXfo
self.lowerFKCtrlSpace.xfo = lowerXfo
self.lowerFKCtrl.xfo = lowerXfo
self.lowerFKCtrl.scalePoints(Vec3(lowerLen, 1.5, 1.5))
self.limbIKCtrlSpace.xfo.tr = endXfo.tr
self.limbIKCtrl.xfo.tr = endXfo.tr
if self.getLocation() == "R":
self.limbIKCtrl.rotatePoints(0, 90, 0)
self.limbIKCtrl.translatePoints(Vec3(-1.0, 0.0, 0.0))
else:
self.limbIKCtrl.rotatePoints(0, -90, 0)
self.limbIKCtrl.translatePoints(Vec3(1.0, 0.0, 0.0))
self.limbUpVCtrlSpace.xfo = upVXfo
self.limbUpVCtrl.xfo = upVXfo
self.limbBone0LenInputAttr.setMin(0.0)
self.limbBone0LenInputAttr.setMax(upperLen * 3.0)
self.limbBone0LenInputAttr.setValue(upperLen)
self.limbBone1LenInputAttr.setMin(0.0)
self.limbBone1LenInputAttr.setMax(lowerLen * 3.0)
self.limbBone1LenInputAttr.setValue(lowerLen)
self.limbParentInputTgt.xfo = upperXfo
# Set Attrs
self.rightSideInputAttr.setValue(self.getLocation() is 'R')
# Eval Constraints
self.limbIKCtrlSpaceInputConstraint.evaluate()
self.limbUpVCtrlSpaceInputConstraint.evaluate()
self.limbRootInputConstraint.evaluate()
# Eval Operators
self.limbIKKLOp.evaluate()
self.outputsToDeformersKLOp.evaluate()
class SimpleControlComponentRig(SimpleControlComponent):
"""Simple Control Component Rig"""
def __init__(self, name='SimpleControl', parent=None):
Profiler.getInstance().push("Construct Simple Control Rig Component:" +
name)
super(SimpleControlComponentRig, self).__init__(name, parent)
# =========
# Controls
# =========
# Add Controls
self.mainCtrl = Control('main', shape='square', parent=self.ctrlCmpGrp)
self.mainCtrlSpace = self.mainCtrl.insertCtrlSpace()
self.mainCtrl.lockScale(x=True, y=True, z=True)
# Add Component Params to Main control
mainSrtSettingsAttrGrp = AttributeGroup('DisplayInfo_MainSrtSettings',
parent=self.mainCtrl)
self.rigScaleAttr = ScalarAttribute('rigScale',
value=1.0,
parent=mainSrtSettingsAttrGrp,
minValue=0.1,
maxValue=100.0)
self.rigScaleOutputAttr.connect(self.rigScaleAttr)
# ==========
# Deformers
# ==========
deformersLayer = self.getOrCreateLayer('deformers')
self.defCmpGrp = ComponentGroup(self.getName(),
self,
parent=deformersLayer)
self.addItem('defCmpGrp', self.defCmpGrp)
self.mainDef = Joint('main', parent=self.defCmpGrp)
self.mainDef.setComponent(self)
# ==============
# Constrain I/O
# ==============
# Constrain inputs
self.mainInputConstraint = PoseConstraint('_'.join(
[self.mainCtrlSpace.getName(), 'To',
self.mainInputTgt.getName()]))
self.mainInputConstraint.setMaintainOffset(True)
self.mainInputConstraint.addConstrainer(self.mainInputTgt)
self.mainCtrlSpace.addConstraint(self.mainInputConstraint)
# Constrain outputs
self.mainOutputConstraint = PoseConstraint('_'.join(
[self.outputTgt.getName(), 'To',
self.mainCtrl.getName()]))
self.mainOutputConstraint.addConstrainer(self.mainCtrl)
self.outputTgt.addConstraint(self.mainOutputConstraint)
# Constrain deformers
self.mainDefConstraint = PoseConstraint('_'.join(
[self.mainDef.getName(), 'To',
self.mainCtrl.getName()]))
self.mainDefConstraint.addConstrainer(self.mainCtrl)
self.mainDef.addConstraint(self.mainDefConstraint)
# ===============
# Add Canvas Ops
# ===============
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(SimpleControlComponentRig, self).loadData(data)
ctrlSize = data.get('ctrlSize', 1.0)
ctrlXfo = data.get('ctrlXfo', Xfo())
# ================
# Resize Controls
# ================
self.mainCtrl.setShape('square')
self.mainCtrl.rotatePoints(90, 0, 0)
self.mainCtrl.scalePoints(Vec3(ctrlSize, ctrlSize, ctrlSize))
# =======================
# Set Control Transforms
# =======================
self.mainCtrlSpace.xfo = ctrlXfo
self.mainCtrl.xfo = ctrlXfo
# ============
# Set IO Xfos
# ============
self.mainInputTgt.xfo = ctrlXfo
self.mainDef.xfo = ctrlXfo
self.outputTgt.xfo = ctrlXfo
# ====================
# Evaluate Constraints
# ====================
self.mainInputConstraint.evaluate()
self.mainOutputConstraint.evaluate()
self.mainDefConstraint.evaluate()
class FabriceSpineRig(FabriceSpine):
"""Fabrice Spine Component"""
def __init__(self, name="spine", parent=None):
Profiler.getInstance().push("Construct Spine Rig Component:" + name)
super(FabriceSpineRig, self).__init__(name, parent)
# =========
# Controls
# =========
# COG
self.cogCtrlSpace = CtrlSpace('cog', parent=self.ctrlCmpGrp)
self.cogCtrl = Control('cog', parent=self.cogCtrlSpace, shape="circle")
self.cogCtrl.rotatePoints(90, 0, 0)
self.cogCtrl.scalePoints(Vec3(3.0, 3.0, 3.0))
self.cogCtrl.translatePoints(Vec3(0.0, 0.0, 0.2))
self.cogCtrl.lockScale(x=True, y=True, z=True)
self.cogCtrl.setColor("orange")
# Spine Base
self.spineBaseCtrlSpace = CtrlSpace('spineBase', parent=self.cogCtrl)
self.spineBaseCtrl = Control('spineBase', parent=self.spineBaseCtrlSpace, shape="pin")
self.spineBaseCtrl.rotatePoints(90, 0, 0)
self.spineBaseCtrl.translatePoints(Vec3(0, 1.0, 0))
self.spineBaseCtrl.lockScale(x=True, y=True, z=True)
# Spine Base Handle
self.spineBaseHandleCtrlSpace = CtrlSpace('spineBaseHandle', parent=self.spineBaseCtrl)
self.spineBaseHandleCtrl = Control('spineBaseHandle', parent=self.spineBaseHandleCtrlSpace, shape="pin")
self.spineBaseHandleCtrl.rotatePoints(90, 0, 0)
self.spineBaseHandleCtrl.translatePoints(Vec3(0, 1.0, 0))
self.spineBaseHandleCtrl.lockScale(x=True, y=True, z=True)
self.spineBaseHandleCtrl.setColor("orange")
# Spine End
self.spineEndCtrlSpace = CtrlSpace('spineEnd', parent=self.cogCtrl)
self.spineEndCtrl = Control('spineEnd', parent=self.spineEndCtrlSpace, shape="pin")
self.spineEndCtrl.rotatePoints(90, 0, 0)
self.spineEndCtrl.lockScale(x=True, y=True, z=True)
self.spineEndCtrl.translatePoints(Vec3(0, 1.0, 0))
# Spine End Handle
self.spineEndHandleCtrlSpace = CtrlSpace('spineEndHandle', parent=self.spineEndCtrl)
self.spineEndHandleCtrl = Control('spineEndHandle', parent=self.spineEndHandleCtrlSpace, shape="pin")
self.spineEndHandleCtrl.rotatePoints(90, 0, 0)
self.spineEndHandleCtrl.translatePoints(Vec3(0, 1.0, 0))
self.spineEndHandleCtrl.lockScale(x=True, y=True, z=True)
self.spineEndHandleCtrl.setColor("orange")
# ==========
# Deformers
# ==========
deformersLayer = self.getOrCreateLayer('deformers')
self.defCmpGrp = ComponentGroup(self.getName(), self, parent=deformersLayer)
self.chestDef = Joint('chest', parent=self.defCmpGrp)
self.chestDef.setComponent(self)
self.deformerJoints = []
self.spineOutputs = []
self.setNumDeformers(1)
# =====================
# Create Component I/O
# =====================
# Setup component Xfo I/O's
self.spineVertebraeOutput.setTarget(self.spineOutputs)
# =====================
# Constraint Deformers
# =====================
self.chestDefConstraint = PoseConstraint('_'.join([self.chestDef.getName(), 'To', self.spineBaseOutputTgt.getName()]))
self.chestDefConstraint.addConstrainer(self.spineBaseOutputTgt)
self.chestDef.addConstraint(self.chestDefConstraint)
# ==============
# Constrain I/O
# ==============
# Constraint inputs
self.spineSrtInputConstraint = PoseConstraint('_'.join([self.cogCtrlSpace.getName(), 'To', self.spineMainSrtInputTgt.getName()]))
self.spineSrtInputConstraint.addConstrainer(self.spineMainSrtInputTgt)
self.spineSrtInputConstraint.setMaintainOffset(True)
self.cogCtrlSpace.addConstraint(self.spineSrtInputConstraint)
# Constraint outputs
self.spineCogOutputConstraint = PoseConstraint('_'.join([self.spineCogOutputTgt.getName(), 'To', self.cogCtrl.getName()]))
self.spineCogOutputConstraint.addConstrainer(self.cogCtrl)
self.spineCogOutputTgt.addConstraint(self.spineCogOutputConstraint)
# Spine Base
self.spineBaseOutputPosConstraint = PositionConstraint('_'.join([self.spineBaseOutputTgt.getName(), 'PosTo', self.spineOutputs[0].getName()]))
self.spineBaseOutputPosConstraint.addConstrainer(self.spineOutputs[0])
self.spineBaseOutputTgt.addConstraint(self.spineBaseOutputPosConstraint)
self.spineBaseOutputOriConstraint = OrientationConstraint('_'.join([self.spineBaseOutputTgt.getName(), 'PosTo', self.cogCtrl.getName()]))
self.spineBaseOutputOriConstraint.addConstrainer(self.cogCtrl)
self.spineBaseOutputTgt.addConstraint(self.spineBaseOutputOriConstraint)
# Spine End
self.spineEndOutputConstraint = PoseConstraint('_'.join([self.spineEndOutputTgt.getName(), 'To', 'spineEnd']))
self.spineEndOutputConstraint.addConstrainer(self.spineOutputs[0])
self.spineEndOutputTgt.addConstraint(self.spineEndOutputConstraint)
self.spineEndCtrlOutputConstraint = PoseConstraint('_'.join([self.spineEndCtrlOutputTgt.getName(), 'To', self.spineEndCtrl.getName()]))
self.spineEndCtrlOutputConstraint.addConstrainer(self.spineEndCtrl)
self.spineEndCtrlOutputTgt.addConstraint(self.spineEndCtrlOutputConstraint)
# ===============
# Add Splice Ops
# ===============
# Add Spine Splice Op
self.bezierSpineSpliceOp = SpliceOperator('spineSpliceOp', 'BezierSpineSolver', 'Kraken')
self.addOperator(self.bezierSpineSpliceOp)
# Add Att Inputs
self.bezierSpineSpliceOp.setInput('drawDebug', self.drawDebugInputAttr)
self.bezierSpineSpliceOp.setInput('rigScale', self.rigScaleInputAttr)
self.bezierSpineSpliceOp.setInput('length', self.lengthInputAttr)
# Add Xfo Inputs
self.bezierSpineSpliceOp.setInput('base', self.spineBaseCtrl)
self.bezierSpineSpliceOp.setInput('baseHandle', self.spineBaseHandleCtrl)
self.bezierSpineSpliceOp.setInput('tipHandle', self.spineEndHandleCtrl)
self.bezierSpineSpliceOp.setInput('tip', self.spineEndCtrl)
# Add Xfo Outputs
self.bezierSpineSpliceOp.setOutput('outputs', self.spineOutputs)
# Add Deformer Splice Op
self.deformersToOutputsSpliceOp = SpliceOperator('spineDeformerSpliceOp', 'MultiPoseConstraintSolver', 'Kraken', alwaysEval=True)
self.addOperator(self.deformersToOutputsSpliceOp)
# Add Att Inputs
self.deformersToOutputsSpliceOp.setInput('drawDebug', self.drawDebugInputAttr)
self.deformersToOutputsSpliceOp.setInput('rigScale', self.rigScaleInputAttr)
# Add Xfo Outputs
self.deformersToOutputsSpliceOp.setInput('constrainers', self.spineOutputs)
# Add Xfo Outputs
self.deformersToOutputsSpliceOp.setOutput('constrainees', self.deformerJoints)
Profiler.getInstance().pop()
def setNumDeformers(self, numDeformers):
# Add new deformers and outputs
for i in xrange(len(self.spineOutputs), numDeformers):
name = 'spine' + str(i + 1).zfill(2)
spineOutput = ComponentOutput(name, parent=self.outputHrcGrp)
self.spineOutputs.append(spineOutput)
for i in xrange(len(self.deformerJoints), numDeformers):
name = 'spine' + str(i + 1).zfill(2)
spineDef = Joint(name, parent=self.defCmpGrp)
spineDef.setComponent(self)
self.deformerJoints.append(spineDef)
return True
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(FabriceSpineRig, self).loadData( data )
# Get Data
cogPos = data['cogPos']
cogCtrlCrvData = data['cogCtrlCrvData']
spineBasePos = data['spineBasePos']
spineBaseCtrlCrvData = data['spineBaseCtrlCrvData']
spineBaseHandlePos = data['spineBaseHandlePos']
spineBaseHandleCtrlCrvData = data['spineBaseHandleCtrlCrvData']
spineEndHandlePos = data['spineEndHandlePos']
spineEndHandleCtrlCrvData = data['spineEndHandleCtrlCrvData']
spineEndPos = data['spineEndPos']
spineEndCtrlCrvData = data['spineEndCtrlCrvData']
numDeformers = data['numDeformers']
# Set Xfos
self.cogCtrlSpace.xfo.tr = cogPos
self.cogCtrl.xfo.tr = cogPos
self.cogCtrl.setCurveData(cogCtrlCrvData)
self.spineBaseCtrlSpace.xfo.tr = spineBasePos
self.spineBaseCtrl.xfo.tr = spineBasePos
self.spineBaseCtrl.setCurveData(spineBaseCtrlCrvData)
self.spineBaseHandleCtrlSpace.xfo.tr = spineBaseHandlePos
self.spineBaseHandleCtrl.xfo.tr = spineBaseHandlePos
self.spineBaseHandleCtrl.setCurveData(spineBaseHandleCtrlCrvData)
self.spineEndHandleCtrlSpace.xfo.tr = spineEndHandlePos
self.spineEndHandleCtrl.xfo.tr = spineEndHandlePos
self.spineEndHandleCtrl.setCurveData(spineEndHandleCtrlCrvData)
self.spineEndCtrlSpace.xfo.tr = spineEndPos
self.spineEndCtrl.xfo.tr = spineEndPos
self.spineEndCtrl.setCurveData(spineEndCtrlCrvData)
length = spineBasePos.distanceTo(spineBaseHandlePos) + spineBaseHandlePos.distanceTo(spineEndHandlePos) + spineEndHandlePos.distanceTo(spineEndPos)
self.lengthInputAttr.setMax(length * 3.0)
self.lengthInputAttr.setValue(length)
# Update number of deformers and outputs
self.setNumDeformers(numDeformers)
# Updating constraint to use the updated last output.
self.spineEndOutputConstraint.setConstrainer(self.spineOutputs[-1], index=0)
# ============
# Set IO Xfos
# ============
# ====================
# Evaluate Splice Ops
# ====================
# evaluate the spine op so that all the output transforms are updated.
self.bezierSpineSpliceOp.evaluate()
# evaluate the constraint op so that all the joint transforms are updated.
self.deformersToOutputsSpliceOp.evaluate()
# evaluate the constraints to ensure the outputs are now in the correct location.
self.spineSrtInputConstraint.evaluate()
self.spineCogOutputConstraint.evaluate()
self.spineBaseOutputPosConstraint.evaluate()
self.spineBaseOutputOriConstraint.evaluate()
self.spineEndOutputConstraint.evaluate()
self.spineEndCtrlOutputConstraint.evaluate()
class TwistComponentGuide(TwistComponent):
"""Twist Component Guide"""
def __init__(self, name='twist', parent=None):
Profiler.getInstance().push('Construct Spine Guide Component:' + name)
super(TwistComponentGuide, self).__init__(name, parent)
# =========
# Controls
# ========
guideSettingsAttrGrp = AttributeGroup('GuideSettings', parent=self)
self.numDeformersAttr = IntegerAttribute('numDeformers',
value=1,
minValue=0,
maxValue=20,
parent=guideSettingsAttrGrp)
self.blendBiasAttr = ScalarAttribute('blendBias',
value=0.0,
minValue=0,
maxValue=1.0,
parent=guideSettingsAttrGrp)
# Guide Controls
triangleCtrl = Control('triangle', shape='triangle')
triangleCtrl.rotatePoints(90, 0, 0)
triangleCtrl.scalePoints(Vec3(0.25, 0.25, 0.25))
triangleCtrl.scalePoints(Vec3(1.0, 0.5, 1.0))
triangleCtrl.translatePoints(Vec3(0.0, 1.25, 0.0))
triangleCtrl.rotatePoints(0, 90, 0)
self.originCtrl = Control('origin',
parent=self.ctrlCmpGrp,
shape='circle')
self.originCtrl.rotatePoints(90, 0, 0)
self.originCtrl.rotatePoints(0, 90, 0)
self.originCtrl.appendCurveData(triangleCtrl.getCurveData())
self.insertCtrl = Control('insert',
parent=self.ctrlCmpGrp,
shape='circle')
self.insertCtrl.rotatePoints(90, 0, 0)
self.insertCtrl.rotatePoints(0, 90, 0)
self.insertCtrl.appendCurveData(triangleCtrl.getCurveData())
self.default_data = {
'name': name,
'location': 'M',
'blendBias': 0.5,
'originXfo': Xfo(Vec3(0.0, 0.0, 0.0)),
'insertXfo': Xfo(Vec3(5.0, 0.0, 0.0)),
'numDeformers': 5
}
self.loadData(self.default_data)
Profiler.getInstance().pop()
# =============
# Data Methods
# =============
def saveData(self):
"""Save the data for the component to be persisted.
Return:
The JSON data object
"""
data = super(TwistComponentGuide, self).saveData()
data['blendBias'] = self.blendBiasAttr.getValue()
data['originXfo'] = self.originCtrl.xfo
data['insertXfo'] = self.insertCtrl.xfo
data['numDeformers'] = self.numDeformersAttr.getValue()
return data
def loadData(self, data):
"""Load a saved guide representation from persisted data.
Arguments:
data -- object, The JSON data object.
Return:
True if successful.
"""
super(TwistComponentGuide, self).loadData(data)
self.blendBiasAttr.setValue(data.get('blendBias', 0.0))
self.originCtrl.xfo = data.get('originXfo', Xfo())
self.insertCtrl.xfo = data.get('insertXfo', Xfo())
self.numDeformersAttr.setValue(data.get('numDeformers', 5))
return True
def getRigBuildData(self):
"""Returns the Guide data used by the Rig Component to define the layout of the final rig.
Return:
The JSON rig data object.
"""
data = super(TwistComponentGuide, self).getRigBuildData()
data['blendBias'] = self.blendBiasAttr.getValue()
data['originXfo'] = self.originCtrl.xfo
data['insertXfo'] = self.insertCtrl.xfo
originUpVXfo = Xfo(Vec3(0.0, 1.0, 0.0))
insertUpVXfo = Xfo(Vec3(0.0, 1.0, 0.0))
data['originUpVXfo'] = self.originCtrl.xfo * originUpVXfo
data['insertUpVXfo'] = self.insertCtrl.xfo * insertUpVXfo
data['numDeformers'] = self.numDeformersAttr.getValue()
return data
# ==============
# Class Methods
# ==============
@classmethod
def getComponentType(cls):
"""Enables introspection of the class prior to construction to determine if it is a guide component.
Return:
The true if this component is a guide component.
"""
return 'Guide'
@classmethod
def getRigComponentClass(cls):
"""Returns the corresponding rig component class for this guide component class
Return:
The rig component class.
"""
return TwistComponentRig
class NeckComponentGuide(NeckComponent):
"""Neck Component Guide"""
def __init__(self, name='neck', parent=None, *args, **kwargs):
Profiler.getInstance().push('Construct Neck Component:' + name)
super(NeckComponentGuide, self).__init__(name, parent, *args, **kwargs)
# =========
# Controls
# =========
# Guide Controls
self.neckCtrl = Control('neck', parent=self.ctrlCmpGrp, shape='sphere')
self.neckCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.neckMidCtrl = Control('neckMid',
parent=self.ctrlCmpGrp,
shape='sphere')
self.neckMidCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.neckEndCtrl = Control('neckEnd',
parent=self.ctrlCmpGrp,
shape='sphere')
self.neckEndCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.neckCtrlShape = Control('neck',
parent=self.ctrlCmpGrp,
shape='pin')
self.neckCtrlShape.rotatePoints(90.0, 0.0, 0.0)
self.neckCtrlShape.rotatePoints(0.0, 90.0, 0.0)
self.neckCtrlShape.setColor('orange')
self.neckMidCtrlShape = Control('neckMid',
parent=self.ctrlCmpGrp,
shape='pin')
self.neckMidCtrlShape.rotatePoints(90.0, 0.0, 0.0)
self.neckMidCtrlShape.rotatePoints(0.0, 90.0, 0.0)
self.neckMidCtrlShape.setColor('orange')
# Guide Operator
self.neckGuideKLOp = KLOperator(name + 'GuideKLOp', 'NeckGuideSolver',
'Kraken')
self.addOperator(self.neckGuideKLOp)
# Add Att Inputs
self.neckGuideKLOp.setInput('drawDebug', self.drawDebugInputAttr)
self.neckGuideKLOp.setInput('rigScale', self.rigScaleInputAttr)
# Add Source Inputs
self.neckGuideKLOp.setInput(
'sources', [self.neckCtrl, self.neckMidCtrl, self.neckEndCtrl])
# Add Target Outputs
self.neckGuideKLOp.setOutput(
'targets', [self.neckCtrlShape, self.neckMidCtrlShape])
# Calculate default values
neckVec = Vec3(0.0, 16.00, -0.75)
neckMidVec = Vec3(0.0, 16.50, -0.50)
neckEndVec = Vec3(0.0, 17.00, -0.25)
upVector = Vec3(0.0, 0.0, -1.0)
neckOri = Quat()
neckOri.setFromDirectionAndUpvector(
(neckMidVec - neckVec).unit(),
((neckVec + upVector) - neckVec).unit())
neckMidOri = Quat()
neckMidOri.setFromDirectionAndUpvector(
(neckEndVec - neckMidVec).unit(),
((neckMidVec + upVector) - neckMidVec).unit())
self.default_data = {
"name": name,
"location": "M",
"neckXfo": Xfo(tr=neckVec, ori=neckOri),
"neckMidXfo": Xfo(tr=neckMidVec, ori=neckMidOri),
"neckEndXfo": Xfo(tr=neckEndVec, ori=neckMidOri),
"neckCrvData": self.neckCtrlShape.getCurveData(),
"neckMidCrvData": self.neckMidCtrlShape.getCurveData()
}
self.loadData(self.default_data)
Profiler.getInstance().pop()
# =============
# Data Methods
# =============
def saveData(self):
"""Save the data for the component to be persisted.
Return:
The JSON data object
"""
data = super(NeckComponentGuide, self).saveData()
data['neckXfo'] = self.neckCtrl.xfo
data['neckMidXfo'] = self.neckMidCtrl.xfo
data['neckEndXfo'] = self.neckEndCtrl.xfo
data['neckCrvData'] = self.neckCtrlShape.getCurveData()
data['neckMidCrvData'] = self.neckMidCtrlShape.getCurveData()
return data
def loadData(self, data):
"""Load a saved guide representation from persisted data.
Arguments:
data (object): The JSON data object.
Returns:
bool: True if successful.
"""
super(NeckComponentGuide, self).loadData(data)
self.neckCtrl.xfo = data.get('neckXfo')
self.neckMidCtrl.xfo = data.get('neckMidXfo')
self.neckEndCtrl.xfo = data.get('neckEndXfo')
self.neckCtrlShape.setCurveData(data.get('neckCrvData'))
self.neckMidCtrlShape.setCurveData(data.get('neckMidCrvData'))
# Evaluate guide operators
self.neckGuideKLOp.evaluate()
return True
def getRigBuildData(self):
"""Returns the Guide data used by the Rig Component to define the layout
of the final rig.
Return:
The JSON rig data object.
"""
data = super(NeckComponentGuide, self).getRigBuildData()
neckEndXfo = Xfo(tr=self.neckEndCtrl.xfo.tr,
ori=self.neckMidCtrlShape.xfo.ori)
data['neckXfo'] = self.neckCtrlShape.xfo
data['neckCrvData'] = self.neckCtrlShape.getCurveData()
data['neckMidXfo'] = self.neckMidCtrlShape.xfo
data['neckMidCrvData'] = self.neckMidCtrlShape.getCurveData()
data['neckEndXfo'] = neckEndXfo
return data
# ==============
# Class Methods
# ==============
@classmethod
def getComponentType(cls):
"""Enables introspection of the class prior to construction to determine
if it is a guide component.
Returns:
bool: Whether the component is a guide component.
"""
return 'Guide'
@classmethod
def getRigComponentClass(cls):
"""Returns the corresponding rig component class for this guide
component class.
Returns:
class: The rig component class.
"""
return NeckComponentRig
class HeadComponentRig(HeadComponent):
"""Head Component Rig"""
def __init__(self, name='head', parent=None):
Profiler.getInstance().push("Construct Head Rig Component:" + name)
super(HeadComponentRig, self).__init__(name, parent)
# =========
# Controls
# =========
# Head
self.headCtrlSpace = CtrlSpace('head', parent=self.ctrlCmpGrp)
self.headCtrl = Control('head', parent=self.headCtrlSpace, shape="circle")
self.headCtrl.rotatePoints(0, 0, 90)
self.headCtrl.scalePoints(Vec3(3, 3, 3))
self.headCtrl.translatePoints(Vec3(0, 1, 0.25))
# Eye Left
self.eyeLeftCtrlSpace = CtrlSpace('eyeLeft', parent=self.headCtrl)
self.eyeLeftCtrl = Control('eyeLeft', parent=self.eyeLeftCtrlSpace, shape="sphere")
self.eyeLeftCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.eyeLeftCtrl.setColor("blueMedium")
# Eye Right
self.eyeRightCtrlSpace = CtrlSpace('eyeRight', parent=self.headCtrl)
self.eyeRightCtrl = Control('eyeRight', parent=self.eyeRightCtrlSpace, shape="sphere")
self.eyeRightCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.eyeRightCtrl.setColor("blueMedium")
# Jaw
self.jawCtrlSpace = CtrlSpace('jawCtrlSpace', parent=self.headCtrl)
self.jawCtrl = Control('jaw', parent=self.jawCtrlSpace, shape="cube")
self.jawCtrl.alignOnYAxis(negative=True)
self.jawCtrl.alignOnZAxis()
self.jawCtrl.scalePoints(Vec3(1.45, 0.65, 1.25))
self.jawCtrl.translatePoints(Vec3(0, -0.25, 0))
self.jawCtrl.setColor("orange")
# ==========
# Deformers
# ==========
deformersLayer = self.getOrCreateLayer('deformers')
defCmpGrp = ComponentGroup(self.getName(), self, parent=deformersLayer)
headDef = Joint('head', parent=defCmpGrp)
headDef.setComponent(self)
jawDef = Joint('jaw', parent=defCmpGrp)
jawDef.setComponent(self)
eyeLeftDef = Joint('eyeLeft', parent=defCmpGrp)
eyeLeftDef.setComponent(self)
eyeRightDef = Joint('eyeRight', parent=defCmpGrp)
eyeRightDef.setComponent(self)
# ==============
# Constrain I/O
# ==============
# Constraint inputs
headInputConstraint = PoseConstraint('_'.join([self.headCtrlSpace.getName(), 'To', self.headBaseInputTgt.getName()]))
headInputConstraint.setMaintainOffset(True)
headInputConstraint.addConstrainer(self.headBaseInputTgt)
self.headCtrlSpace.addConstraint(headInputConstraint)
# # Constraint outputs
headOutputConstraint = PoseConstraint('_'.join([self.headOutputTgt.getName(), 'To', self.headCtrl.getName()]))
headOutputConstraint.addConstrainer(self.headCtrl)
self.headOutputTgt.addConstraint(headOutputConstraint)
jawOutputConstraint = PoseConstraint('_'.join([self.jawOutputTgt.getName(), 'To', self.jawCtrl.getName()]))
jawOutputConstraint.addConstrainer(self.jawCtrl)
self.jawOutputTgt.addConstraint(jawOutputConstraint)
eyeLOutputConstraint = PoseConstraint('_'.join([self.eyeLOutputTgt.getName(), 'To', self.eyeLeftCtrl.getName()]))
eyeLOutputConstraint.addConstrainer(self.eyeLeftCtrl)
self.eyeLOutputTgt.addConstraint(eyeLOutputConstraint)
eyeROutputConstraint = PoseConstraint('_'.join([self.eyeROutputTgt.getName(), 'To', self.eyeRightCtrl.getName()]))
eyeROutputConstraint.addConstrainer(self.eyeRightCtrl)
self.eyeROutputTgt.addConstraint(eyeROutputConstraint)
# ==================
# Add Component I/O
# ==================
# Add Xfo I/O's
# self.addInput(self.headBaseInputTgt)
# self.addOutput(self.headOutputTgt)
# self.addOutput(self.jawOutputTgt)
# self.addOutput(self.eyeLOutputTgt)
# self.addOutput(self.eyeROutputTgt)
# Add Attribute I/O's
# self.addInput(self.drawDebugInputAttr)
# ===============
# Add Splice Ops
# ===============
# Add Deformer Splice Op
# spliceOp = SpliceOperator('headDeformerSpliceOp', 'HeadConstraintSolver', 'KrakenHeadConstraintSolver')
# self.addOperator(spliceOp)
# # Add Att Inputs
# spliceOp.setInput('drawDebug', self.drawDebugInputAttr)
# spliceOp.setInput('rigScale', self.rigScaleInputAttr)
# # Add Xfo Inputstrl)
# spliceOp.setInput('headConstrainer', self.headOutputTgt)
# spliceOp.setInput('jawConstrainer', self.jawOutputTgt)
# spliceOp.setInput('eyeLeftConstrainer', self.eyeLOutputTgt)
# spliceOp.setInput('eyeRightConstrainer', self.eyeROutputTgt)
# # Add Xfo Outputs
# spliceOp.setOutput('headDeformer', headDef)
# spliceOp.setOutput('jawDeformer', jawDef)
# spliceOp.setOutput('eyeLeftDeformer', eyeLeftDef)
# spliceOp.setOutput('eyeRightDeformer', eyeRightDef)
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(HeadComponentRig, self).loadData( data )
self.headCtrlSpace.xfo.tr = data['headPosition']
self.headCtrl.xfo.tr = data['headPosition']
self.eyeLeftCtrlSpace.xfo.tr = data['eyeLeftPosition']
self.eyeLeftCtrl.xfo.tr = data['eyeLeftPosition']
self.eyeRightCtrlSpace.xfo.tr = data['eyeRightPosition']
self.eyeRightCtrl.xfo.tr = data['eyeRightPosition']
self.jawCtrlSpace.xfo.tr = data['jawPosition']
self.jawCtrl.xfo.tr = data['jawPosition']
# ============
# Set IO Xfos
# ============
self.headBaseInputTgt.xfo.tr = data['headPosition']
self.headOutputTgt.xfo.tr = data['headPosition']
self.jawOutputTgt.xfo.tr = data['jawPosition']
self.eyeLOutputTgt.xfo.tr = data['eyeLeftPosition']
self.eyeROutputTgt.xfo.tr = data['eyeRightPosition']
class StretchyLimbComponentRig(StretchyLimbComponent):
"""StretchyLimb Component"""
def __init__(self, name='limb', parent=None):
Profiler.getInstance().push("Construct StretchyLimb Rig Component:" + name)
super(StretchyLimbComponentRig, self).__init__(name, parent)
# =========
# Controls
# =========
# Upper (FK)
self.upperFKCtrlSpace = CtrlSpace('upperFK', parent=self.ctrlCmpGrp)
self.upperFKCtrl = Control('upperFK', parent=self.upperFKCtrlSpace, shape="cube")
self.upperFKCtrl.alignOnXAxis()
# Lower (FK)
self.lowerFKCtrlSpace = CtrlSpace('lowerFK', parent=self.upperFKCtrl)
self.lowerFKCtrl = Control('lowerFK', parent=self.lowerFKCtrlSpace, shape="cube")
self.lowerFKCtrl.alignOnXAxis()
# End (IK)
self.limbIKCtrlSpace = CtrlSpace('IK', parent=self.ctrlCmpGrp)
self.limbIKCtrl = Control('IK', parent=self.limbIKCtrlSpace, shape="pin")
# Add Component Params to IK control
# TODO: Move these separate control
limbSettingsAttrGrp = AttributeGroup("DisplayInfo_StretchyLimbSettings", parent=self.limbIKCtrl)
limbDrawDebugInputAttr = BoolAttribute('drawDebug', value=False, parent=limbSettingsAttrGrp)
self.limbBone0LenInputAttr = ScalarAttribute('bone0Len', value=1.0, parent=limbSettingsAttrGrp)
self.limbBone1LenInputAttr = ScalarAttribute('bone1Len', value=1.0, parent=limbSettingsAttrGrp)
limbIKBlendInputAttr = ScalarAttribute('ikblend', value=1.0, minValue=0.0, maxValue=1.0, parent=limbSettingsAttrGrp)
limbSoftIKInputAttr = BoolAttribute('softIK', value=True, parent=limbSettingsAttrGrp)
limbSoftRatioInputAttr = ScalarAttribute('softRatio', value=0.0, minValue=0.0, maxValue=1.0, parent=limbSettingsAttrGrp)
limbStretchInputAttr = BoolAttribute('stretch', value=True, parent=limbSettingsAttrGrp)
limbStretchBlendInputAttr = ScalarAttribute('stretchBlend', value=0.0, minValue=0.0, maxValue=1.0, parent=limbSettingsAttrGrp)
limbSlideInputAttr = ScalarAttribute('slide', value=0.0, minValue=-1.0, maxValue=1.0, parent=limbSettingsAttrGrp)
limbPinInputAttr = ScalarAttribute('pin', value=0.0, minValue=0.0, maxValue=1.0, parent=limbSettingsAttrGrp)
self.rightSideInputAttr = BoolAttribute('rightSide', value=False, parent=limbSettingsAttrGrp)
self.drawDebugInputAttr.connect(limbDrawDebugInputAttr)
# UpV (IK Pole Vector)
self.limbUpVCtrlSpace = CtrlSpace('UpV', parent=self.ctrlCmpGrp)
self.limbUpVCtrl = Control('UpV', parent=self.limbUpVCtrlSpace, shape="triangle")
self.limbUpVCtrl.alignOnZAxis()
# ==========
# Deformers
# ==========
deformersLayer = self.getOrCreateLayer('deformers')
self.defCmpGrp = ComponentGroup(self.getName(), self, parent=deformersLayer)
self.addItem('defCmpGrp', self.defCmpGrp)
upperDef = Joint('upper', parent=self.defCmpGrp)
upperDef.setComponent(self)
lowerDef = Joint('lower', parent=self.defCmpGrp)
lowerDef.setComponent(self)
endDef = Joint('end', parent=self.defCmpGrp)
endDef.setComponent(self)
# ==============
# Constrain I/O
# ==============
# Constraint inputs
self.limbIKCtrlSpaceInputConstraint = PoseConstraint('_'.join([self.limbIKCtrlSpace.getName(), 'To', self.globalSRTInputTgt.getName()]))
self.limbIKCtrlSpaceInputConstraint.setMaintainOffset(True)
self.limbIKCtrlSpaceInputConstraint.addConstrainer(self.globalSRTInputTgt)
self.limbIKCtrlSpace.addConstraint(self.limbIKCtrlSpaceInputConstraint)
self.limbUpVCtrlSpaceInputConstraint = PoseConstraint('_'.join([self.limbUpVCtrlSpace.getName(), 'To', self.globalSRTInputTgt.getName()]))
self.limbUpVCtrlSpaceInputConstraint.setMaintainOffset(True)
self.limbUpVCtrlSpaceInputConstraint.addConstrainer(self.globalSRTInputTgt)
self.limbUpVCtrlSpace.addConstraint(self.limbUpVCtrlSpaceInputConstraint)
self.limbRootInputConstraint = PoseConstraint('_'.join([self.limbIKCtrl.getName(), 'To', self.limbParentInputTgt.getName()]))
self.limbRootInputConstraint.setMaintainOffset(True)
self.limbRootInputConstraint.addConstrainer(self.limbParentInputTgt)
self.upperFKCtrlSpace.addConstraint(self.limbRootInputConstraint)
# ===============
# Add Splice Ops
# ===============
# Add StretchyLimb Splice Op
self.limbIKKLOp = KLOperator('limbKLOp', 'TwoBoneStretchyIKSolver', 'Kraken')
self.addOperator(self.limbIKKLOp)
# Add Att Inputs
self.limbIKKLOp.setInput('drawDebug', self.drawDebugInputAttr)
self.limbIKKLOp.setInput('rigScale', self.rigScaleInputAttr)
self.limbIKKLOp.setInput('bone0Len', self.limbBone0LenInputAttr)
self.limbIKKLOp.setInput('bone1Len', self.limbBone1LenInputAttr)
self.limbIKKLOp.setInput('ikblend', limbIKBlendInputAttr)
self.limbIKKLOp.setInput('softIK', limbSoftIKInputAttr)
self.limbIKKLOp.setInput('softRatio', limbSoftRatioInputAttr)
self.limbIKKLOp.setInput('stretch', limbStretchInputAttr)
self.limbIKKLOp.setInput('stretchBlend', limbStretchBlendInputAttr)
self.limbIKKLOp.setInput('slide', limbSlideInputAttr)
self.limbIKKLOp.setInput('pin', limbPinInputAttr)
self.limbIKKLOp.setInput('rightSide', self.rightSideInputAttr)
# Add Xfo Inputs
self.limbIKKLOp.setInput('root', self.limbParentInputTgt)
self.limbIKKLOp.setInput('bone0FK', self.upperFKCtrl)
self.limbIKKLOp.setInput('bone1FK', self.lowerFKCtrl)
self.limbIKKLOp.setInput('ikHandle', self.limbIKCtrl)
self.limbIKKLOp.setInput('upV', self.limbUpVCtrl)
# Add Xfo Outputs
self.limbIKKLOp.setOutput('bone0Out', self.limbUpperOutputTgt)
self.limbIKKLOp.setOutput('bone1Out', self.limbLowerOutputTgt)
self.limbIKKLOp.setOutput('bone2Out', self.limbEndOutputTgt)
# =====================
# Connect the deformers
# =====================
# Add StretchyLimb Deformer Splice Op
self.outputsToDeformersKLOp = KLOperator('limbDeformerKLOp', '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.limbUpperOutputTgt, self.limbLowerOutputTgt, self.limbEndOutputTgt])
# Add Xfo Outputs
self.outputsToDeformersKLOp.setOutput('constrainees', [upperDef, lowerDef, endDef])
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(StretchyLimbComponentRig, self).loadData(data)
upperXfo = data.get('upperXfo')
upperLen = data.get('upperLen')
lowerXfo = data.get('lowerXfo')
lowerLen = data.get('lowerLen')
endXfo = data.get('endXfo')
upVXfo = data.get('upVXfo')
self.upperFKCtrlSpace.xfo = upperXfo
self.upperFKCtrl.xfo = upperXfo
self.upperFKCtrl.scalePoints(Vec3(upperLen, 1.75, 1.75))
self.limbUpperOutputTgt.xfo = upperXfo
self.limbLowerOutputTgt.xfo = lowerXfo
self.lowerFKCtrlSpace.xfo = lowerXfo
self.lowerFKCtrl.xfo = lowerXfo
self.lowerFKCtrl.scalePoints(Vec3(lowerLen, 1.5, 1.5))
self.limbIKCtrlSpace.xfo.tr = endXfo.tr
self.limbIKCtrl.xfo.tr = endXfo.tr
if self.getLocation() == "R":
self.limbIKCtrl.rotatePoints(0, 90, 0)
self.limbIKCtrl.translatePoints(Vec3(-1.0, 0.0, 0.0))
else:
self.limbIKCtrl.rotatePoints(0, -90, 0)
self.limbIKCtrl.translatePoints(Vec3(1.0, 0.0, 0.0))
self.limbUpVCtrlSpace.xfo = upVXfo
self.limbUpVCtrl.xfo = upVXfo
self.limbBone0LenInputAttr.setMin(0.0)
self.limbBone0LenInputAttr.setMax(upperLen * 3.0)
self.limbBone0LenInputAttr.setValue(upperLen)
self.limbBone1LenInputAttr.setMin(0.0)
self.limbBone1LenInputAttr.setMax(lowerLen * 3.0)
self.limbBone1LenInputAttr.setValue(lowerLen)
self.limbParentInputTgt.xfo = upperXfo
# Set Attrs
self.rightSideInputAttr.setValue(self.getLocation() is 'R')
# Eval Constraints
self.limbIKCtrlSpaceInputConstraint.evaluate()
self.limbUpVCtrlSpaceInputConstraint.evaluate()
self.limbRootInputConstraint.evaluate()
# Eval Operators
self.limbIKKLOp.evaluate()
self.outputsToDeformersKLOp.evaluate()
class FootComponentGuide(FootComponent):
"""Foot Component Guide"""
def __init__(self, name='foot', parent=None, *args, **kwargs):
Profiler.getInstance().push("Construct Foot Component:" + name)
super(FootComponentGuide, self).__init__(name, parent, *args, **kwargs)
# =========
# Controls
# ========
guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self)
# Guide Controls
self.ankleCtrl = Control('ankle', parent=self.ctrlCmpGrp, shape="pin")
self.toeCtrl = Control('toe', parent=self.ctrlCmpGrp, shape="pin")
self.toeCtrl.rotatePoints(-90.0, 0.0, 0.0)
self.toeTipCtrl = Control('toeTip', parent=self.ctrlCmpGrp, shape="pin")
self.toeTipCtrl.rotatePoints(-90.0, 0.0, 0.0)
self.backPivotCtrl = Control('backPivot', parent=self.ctrlCmpGrp, shape="axesHalfTarget")
self.backPivotCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.frontPivotCtrl = Control('frontPivot', parent=self.ctrlCmpGrp, shape="axesHalfTarget")
self.frontPivotCtrl.rotatePoints(0.0, 180.0, 0.0)
self.frontPivotCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.outerPivotCtrl = Control('outerPivot', parent=self.ctrlCmpGrp, shape="axesHalfTarget")
self.outerPivotCtrl.rotatePoints(0.0, -90.0, 0.0)
self.outerPivotCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.innerPivotCtrl = Control('innerPivot', parent=self.ctrlCmpGrp, shape="axesHalfTarget")
self.innerPivotCtrl.rotatePoints(0.0, 90.0, 0.0)
self.innerPivotCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.default_data = {
"name": name,
"location": 'L',
"ankleXfo": Xfo(Vec3(1.75, 1.15, -1.25)),
"toeXfo": Xfo(Vec3(1.75, 0.4, 0.25)),
"toeTipXfo": Xfo(Vec3(1.75, 0.4, 1.5)),
"backPivotXfo": Xfo(Vec3(1.75, 0.0, -2.5)),
"frontPivotXfo": Xfo(Vec3(1.75, 0.0, 2.0)),
"outerPivotXfo": Xfo(Vec3(2.5, 0.0, -1.25)),
"innerPivotXfo": Xfo(Vec3(1.0, 0.0, -1.25))
}
self.loadData(self.default_data)
Profiler.getInstance().pop()
# =============
# Data Methods
# =============
def saveData(self):
"""Save the data for the component to be persisted.
Return:
The JSON data object
"""
data = super(FootComponentGuide, self).saveData()
# data['footCtrlCrvData'] = self.footCtrl.getCurveData()
data['ankleXfo'] = self.ankleCtrl.xfo
data['toeXfo'] = self.toeCtrl.xfo
data['toeTipXfo'] = self.toeTipCtrl.xfo
data['backPivotXfo'] = self.backPivotCtrl.xfo
data['frontPivotXfo'] = self.frontPivotCtrl.xfo
data['outerPivotXfo'] = self.outerPivotCtrl.xfo
data['innerPivotXfo'] = self.innerPivotCtrl.xfo
return data
def loadData(self, data):
"""Load a saved guide representation from persisted data.
Arguments:
data -- object, The JSON data object.
Return:
True if successful.
"""
super(FootComponentGuide, self).loadData(data)
self.ankleCtrl.xfo = data.get('ankleXfo')
self.toeCtrl.xfo = data.get('toeXfo')
self.toeTipCtrl.xfo = data.get('toeTipXfo')
self.backPivotCtrl.xfo = data.get('backPivotXfo')
self.frontPivotCtrl.xfo = data.get('frontPivotXfo')
self.outerPivotCtrl.xfo = data.get('outerPivotXfo')
self.innerPivotCtrl.xfo = data.get('innerPivotXfo')
if self.getLocation() == 'R':
self.outerPivotCtrl.rotatePoints(0.0, 180, 0.0)
self.innerPivotCtrl.rotatePoints(0.0, 180, 0.0)
if self.outerPivotCtrl.xfo.toMat44().equal(self.default_data['outerPivotXfo'].toMat44()):
self.outerPivotCtrl.xfo = self.default_data['innerPivotXfo']
if self.innerPivotCtrl.xfo.toMat44().equal(self.default_data['innerPivotXfo'].toMat44()):
self.innerPivotCtrl.xfo = self.default_data['outerPivotXfo']
return True
def getRigBuildData(self):
"""Returns the Guide data used by the Rig Component to define the layout of the final rig.
Return:
The JSON rig data object.
"""
data = super(FootComponentGuide, self).getRigBuildData()
# Values
anklePos = self.ankleCtrl.xfo.tr
toePos = self.toeCtrl.xfo.tr
toeTipPos = self.toeTipCtrl.xfo.tr
backPivotXfo = self.backPivotCtrl.xfo
frontPivotXfo = self.frontPivotCtrl.xfo
outerPivotXfo = self.outerPivotCtrl.xfo
innerPivotXfo = self.innerPivotCtrl.xfo
# Calculate Ankle Xfo
rootToEnd = toePos.subtract(anklePos).unit()
rootToUpV = Vec3(0.0, -1.0, 0.0).add(anklePos).subtract(anklePos).unit()
zAxis = rootToUpV.cross(rootToEnd).unit()
normal = zAxis.cross(rootToEnd).unit()
fkOffsetQuat = Quat()
fkOffsetQuat.setFromAxisAndAngle(Vec3(1.0, 0., 0.0), (PI / 2))
ankleXfo = Xfo()
ankleXfo.setFromVectors(rootToEnd, normal, zAxis, toePos)
ankleFKXfo = Xfo(ankleXfo)
ankleFKXfo.ori = ankleFKXfo.ori.multiply(fkOffsetQuat)
# Calculate Toe Xfo
rootToEnd = toeTipPos.subtract(toePos).unit()
rootToUpV = Vec3(0.0, -1.0, 0.0).add(toePos).subtract(toePos).unit()
zAxis = rootToUpV.cross(rootToEnd).unit()
normal = zAxis.cross(rootToEnd).unit()
toeXfo = Xfo()
toeXfo.setFromVectors(rootToEnd, normal, zAxis, toePos)
toeFKXfo = Xfo(toeXfo)
toeFKXfo.ori = toeFKXfo.ori.multiply(fkOffsetQuat)
data['footXfo'] = self.ankleCtrl.xfo
data['ankleXfo'] = ankleXfo
data['ankleFKXfo'] = ankleFKXfo
data['ankleLen'] = anklePos.subtract(toePos).length()
data['toeXfo'] = toeXfo
data['toeFKXfo'] = toeFKXfo
data['toeLen'] = toePos.subtract(toeTipPos).length()
data['backPivotXfo'] = backPivotXfo
data['frontPivotXfo'] = frontPivotXfo
data['outerPivotXfo'] = outerPivotXfo
data['innerPivotXfo'] = innerPivotXfo
return data
# ==============
# Class Methods
# ==============
@classmethod
def getComponentType(cls):
"""Enables introspection of the class prior to construction to determine if it is a guide component.
Return:
The true if this component is a guide component.
"""
return 'Guide'
@classmethod
def getRigComponentClass(cls):
"""Returns the corresponding rig component class for this guide component class
Return:
The rig component class.
"""
return FootComponentRig
class FootComponentRig(FootComponent):
"""Foot Component"""
def __init__(self, name="foot", parent=None):
Profiler.getInstance().push("Construct Neck Rig Component:" + name)
super(FootComponentRig, self).__init__(name, parent)
# =========
# Controls
# =========
self.ankleLenInputAttr = ScalarAttribute('ankleLen', 1.0, maxValue=1.0, parent=self.cmpInputAttrGrp)
self.toeLenInputAttr = ScalarAttribute('toeLen', 1.0, maxValue=1.0, parent=self.cmpInputAttrGrp)
self.rightSideInputAttr = BoolAttribute('rightSide', False, parent=self.cmpInputAttrGrp)
self.footAll = Locator('footAll', parent=self.ctrlCmpGrp)
self.footAll.setShapeVisibility(False)
self.ankleIKCtrlSpace = CtrlSpace('ankleIK', parent=self.footAll)
self.ankleIKCtrl = Control('ankleIK', parent=self.ankleIKCtrlSpace, shape="square")
self.ankleIKCtrl.alignOnXAxis(negative=True)
self.ankleIKCtrl.lockTranslation(True, True, True)
self.ankleIKCtrl.lockScale(True, True, True)
self.toeIKCtrlSpace = CtrlSpace('toeIK', parent=self.footAll)
self.toeIKCtrl = Control('toeIK', parent=self.toeIKCtrlSpace, shape="square")
self.toeIKCtrl.alignOnXAxis()
self.toeIKCtrl.lockTranslation(True, True, True)
self.toeIKCtrl.lockScale(True, True, True)
self.ankleFKCtrlSpace = CtrlSpace('ankleFK', parent=self.ctrlCmpGrp)
self.ankleFKCtrl = Control('ankleFK', parent=self.ankleFKCtrlSpace, shape="cube")
self.ankleFKCtrl.alignOnXAxis()
self.ankleFKCtrl.lockTranslation(True, True, True)
self.ankleFKCtrl.lockScale(True, True, True)
self.toeFKCtrlSpace = CtrlSpace('toeFK', parent=self.ankleFKCtrl)
self.toeFKCtrl = Control('toeFK', parent=self.toeFKCtrlSpace, shape="cube")
self.toeFKCtrl.alignOnXAxis()
self.toeFKCtrl.lockTranslation(True, True, True)
self.toeFKCtrl.lockScale(True, True, True)
self.footSettingsAttrGrp = AttributeGroup("DisplayInfo_FootSettings", parent=self.ankleIKCtrl)
self.footDebugInputAttr = BoolAttribute('drawDebug', value=False, parent=self.footSettingsAttrGrp)
self.footRockInputAttr = ScalarAttribute('footRock', value=0.0, minValue=-1.0, maxValue=1.0, parent=self.footSettingsAttrGrp)
self.footBankInputAttr = ScalarAttribute('footBank', value=0.0, minValue=-1.0, maxValue=1.0, parent=self.footSettingsAttrGrp)
self.drawDebugInputAttr.connect(self.footDebugInputAttr)
self.pivotAll = Locator('pivotAll', parent=self.ctrlCmpGrp)
self.pivotAll.setShapeVisibility(False)
self.backPivotCtrl = Control('backPivot', parent=self.pivotAll, shape="axesHalfTarget")
self.backPivotCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.backPivotCtrl.lockScale(True, True, True)
self.backPivotCtrlSpace = self.backPivotCtrl.insertCtrlSpace()
self.frontPivotCtrl = Control('frontPivot', parent=self.pivotAll, shape="axesHalfTarget")
self.frontPivotCtrl.rotatePoints(0.0, 180.0, 0.0)
self.frontPivotCtrl.lockScale(True, True, True)
self.frontPivotCtrlSpace = self.frontPivotCtrl.insertCtrlSpace()
self.frontPivotCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.outerPivotCtrl = Control('outerPivot', parent=self.pivotAll, shape="axesHalfTarget")
self.outerPivotCtrl.rotatePoints(0.0, -90.0, 0.0)
self.outerPivotCtrl.lockScale(True, True, True)
self.outerPivotCtrlSpace = self.outerPivotCtrl.insertCtrlSpace()
self.outerPivotCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.innerPivotCtrl = Control('innerPivot', parent=self.pivotAll, shape="axesHalfTarget")
self.innerPivotCtrl.rotatePoints(0.0, 90.0, 0.0)
self.innerPivotCtrl.lockScale(True, True, True)
self.innerPivotCtrlSpace = self.innerPivotCtrl.insertCtrlSpace()
self.innerPivotCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
# ==========
# Deformers
# ==========
deformersLayer = self.getOrCreateLayer('deformers')
self.defCmpGrp = ComponentGroup(self.getName(), self, parent=deformersLayer)
self.addItem('defCmpGrp', self.defCmpGrp)
self.ankleDef = Joint('ankle', parent=self.defCmpGrp)
self.ankleDef.setComponent(self)
self.toeDef = Joint('toe', parent=self.defCmpGrp)
self.toeDef.setComponent(self)
# ==============
# Constrain I/O
# ==============
# Constraint to inputs
self.pivotAllInputConstraint = PoseConstraint('_'.join([self.pivotAll.getName(), 'To', self.ikHandleInputTgt.getName()]))
self.pivotAllInputConstraint.setMaintainOffset(True)
self.pivotAllInputConstraint.addConstrainer(self.ikHandleInputTgt)
self.pivotAll.addConstraint(self.pivotAllInputConstraint)
self.ankleFKInputConstraint = PoseConstraint('_'.join([self.ankleFKCtrlSpace.getName(), 'To', self.legEndFKInputTgt.getName()]))
self.ankleFKInputConstraint.setMaintainOffset(True)
self.ankleFKInputConstraint.addConstrainer(self.legEndFKInputTgt)
self.ankleFKCtrlSpace.addConstraint(self.ankleFKInputConstraint)
# Constraint outputs
self.ikTargetOutputConstraint = PoseConstraint('_'.join([self.ikTargetOutputTgt.getName(), 'To', self.ankleIKCtrl.getName()]))
self.ikTargetOutputConstraint.setMaintainOffset(True)
self.ikTargetOutputConstraint.addConstrainer(self.ankleIKCtrl)
self.ikTargetOutputTgt.addConstraint(self.ikTargetOutputConstraint)
# =========================
# Add Foot Pivot Canvas Op
# =========================
# self.footPivotCanvasOp = CanvasOperator('footPivotCanvasOp', 'Kraken.Solvers.Biped.BipedFootPivotSolver')
self.footPivotCanvasOp = KLOperator('footPivotKLOp', 'BipedFootPivotSolver', 'Kraken')
self.addOperator(self.footPivotCanvasOp)
# Add Att Inputs
self.footPivotCanvasOp.setInput('drawDebug', self.drawDebugInputAttr)
self.footPivotCanvasOp.setInput('rigScale', self.rigScaleInputAttr)
self.footPivotCanvasOp.setInput('rightSide', self.rightSideInputAttr)
self.footPivotCanvasOp.setInput('footRock', self.footRockInputAttr)
self.footPivotCanvasOp.setInput('footBank', self.footBankInputAttr)
# Add Xfo Inputs
self.footPivotCanvasOp.setInput('pivotAll', self.pivotAll)
self.footPivotCanvasOp.setInput('backPivot', self.backPivotCtrl)
self.footPivotCanvasOp.setInput('frontPivot', self.frontPivotCtrl)
self.footPivotCanvasOp.setInput('outerPivot', self.outerPivotCtrl)
self.footPivotCanvasOp.setInput('innerPivot', self.innerPivotCtrl)
# Add Xfo Outputs
self.footPivotCanvasOp.setOutput('result', self.footAll)
# =========================
# Add Foot Solver Canvas Op
# =========================
# self.footSolverCanvasOp = CanvasOperator('footSolverCanvasOp', 'Kraken.Solvers.Biped.BipedFootSolver')
self.footSolverCanvasOp = KLOperator('footSolverKLOp', 'BipedFootSolver', 'Kraken')
self.addOperator(self.footSolverCanvasOp)
# Add Att Inputs
self.footSolverCanvasOp.setInput('drawDebug', self.drawDebugInputAttr)
self.footSolverCanvasOp.setInput('rigScale', self.rigScaleInputAttr)
self.footSolverCanvasOp.setInput('ikBlend', self.ikBlendInputAttr)
self.footSolverCanvasOp.setInput('ankleLen', self.ankleLenInputAttr)
self.footSolverCanvasOp.setInput('toeLen', self.toeLenInputAttr)
# Add Xfo Inputs
self.footSolverCanvasOp.setInput('legEnd', self.legEndInputTgt)
self.footSolverCanvasOp.setInput('ankleIK', self.ankleIKCtrl)
self.footSolverCanvasOp.setInput('toeIK', self.toeIKCtrl)
self.footSolverCanvasOp.setInput('ankleFK', self.ankleFKCtrl)
self.footSolverCanvasOp.setInput('toeFK', self.toeFKCtrl)
# Add Xfo Outputs
self.footSolverCanvasOp.setOutput('ankle_result', self.ankleOutputTgt)
self.footSolverCanvasOp.setOutput('toe_result', self.toeOutputTgt)
# ===================
# Add Deformer KL Op
# ===================
self.outputsToDeformersKLOp = KLOperator('foot' + self.getLocation() + 'DeformerKLOp', '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.ankleOutputTgt, self.toeOutputTgt])
# Add Xfo Outputs
self.outputsToDeformersKLOp.setOutput('constrainees', [self.ankleDef, self.toeDef])
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(FootComponentRig, self).loadData( data )
footXfo = data.get('footXfo')
ankleXfo = data.get('ankleXfo')
toeXfo = data.get('toeXfo')
ankleFKXfo = data.get('ankleFKXfo')
toeFKXfo = data.get('toeFKXfo')
ankleLen = data.get('ankleLen')
toeLen = data.get('toeLen')
backPivotXfo = data.get('backPivotXfo')
frontPivotXfo = data.get('frontPivotXfo')
outerPivotXfo = data.get('outerPivotXfo')
innerPivotXfo = data.get('innerPivotXfo')
self.footAll.xfo = footXfo
self.ankleIKCtrlSpace.xfo = ankleXfo
self.ankleIKCtrl.xfo = ankleXfo
self.toeIKCtrlSpace.xfo = toeXfo
self.toeIKCtrl.xfo = toeXfo
self.ankleFKCtrl.scalePoints(Vec3(ankleLen, 1.0, 1.0))
self.toeFKCtrl.scalePoints(Vec3(toeLen, 1.0, 1.0))
self.ankleFKCtrlSpace.xfo.tr = footXfo.tr
self.ankleFKCtrlSpace.xfo.ori = ankleFKXfo.ori
self.ankleFKCtrl.xfo.tr = footXfo.tr
self.ankleFKCtrl.xfo.ori = ankleFKXfo.ori
self.toeFKCtrlSpace.xfo = toeFKXfo
self.toeFKCtrl.xfo = toeFKXfo
self.pivotAll.xfo = footXfo
self.backPivotCtrlSpace.xfo = backPivotXfo
self.backPivotCtrl.xfo = backPivotXfo
self.frontPivotCtrlSpace.xfo = frontPivotXfo
self.frontPivotCtrl.xfo = frontPivotXfo
self.outerPivotCtrlSpace.xfo = outerPivotXfo
self.outerPivotCtrl.xfo = outerPivotXfo
self.innerPivotCtrlSpace.xfo = innerPivotXfo
self.innerPivotCtrl.xfo = innerPivotXfo
if self.getLocation() == 'R':
self.outerPivotCtrl.rotatePoints(0.0, 180.0, 0.0)
self.innerPivotCtrl.rotatePoints(0.0, 180.0, 0.0)
self.ankleIKCtrl.scalePoints(Vec3(ankleLen, 1.0, 1.5))
self.toeIKCtrl.scalePoints(Vec3(toeLen, 1.0, 1.5))
# Set Attribute Values
self.rightSideInputAttr.setValue(self.getLocation() is 'R')
self.ankleLenInputAttr.setValue(ankleLen)
self.ankleLenInputAttr.setMax(ankleLen * 3.0)
self.toeLenInputAttr.setValue(toeLen)
self.toeLenInputAttr.setMax(toeLen * 3.0)
# Set IO Xfos
self.ikHandleInputTgt.xfo = footXfo
self.legEndInputTgt.xfo.tr = footXfo.tr
self.legEndInputTgt.xfo.ori = ankleXfo.ori
self.legEndFKInputTgt.xfo.tr = footXfo.tr
self.legEndFKInputTgt.xfo.ori = ankleXfo.ori
self.ikTargetOutputTgt.xfo.tr = footXfo.tr
self.ikTargetOutputTgt.xfo.ori = ankleXfo.ori
# Eval Canvas Ops
self.footPivotCanvasOp.evaluate()
self.footSolverCanvasOp.evaluate()
# Eval Constraints
self.ikTargetOutputConstraint.evaluate()
self.ankleFKInputConstraint.evaluate()
class TwistComponentGuide(TwistComponent):
"""Twist Component Guide"""
def __init__(self, name='twist', parent=None):
Profiler.getInstance().push('Construct Spine Guide Component:' + name)
super(TwistComponentGuide, self).__init__(name, parent)
# =========
# Controls
# ========
guideSettingsAttrGrp = AttributeGroup('GuideSettings', parent=self)
self.numDeformersAttr = IntegerAttribute('numDeformers', value=1, minValue=0, maxValue=20, parent=guideSettingsAttrGrp)
self.blendBiasAttr = ScalarAttribute('blendBias', value=0.0, minValue=0, maxValue=1.0, parent=guideSettingsAttrGrp)
# Guide Controls
triangleCtrl = Control('triangle', shape='triangle')
triangleCtrl.rotatePoints(90, 0, 0)
triangleCtrl.scalePoints(Vec3(0.25, 0.25, 0.25))
triangleCtrl.scalePoints(Vec3(1.0, 0.5, 1.0))
triangleCtrl.translatePoints(Vec3(0.0, 1.25, 0.0))
triangleCtrl.rotatePoints(0, 90, 0)
self.originCtrl = Control('origin', parent=self.ctrlCmpGrp, shape='circle')
self.originCtrl.rotatePoints(90, 0, 0)
self.originCtrl.rotatePoints(0, 90, 0)
self.originCtrl.appendCurveData(triangleCtrl.getCurveData())
self.insertCtrl = Control('insert', parent=self.ctrlCmpGrp, shape='circle')
self.insertCtrl.rotatePoints(90, 0, 0)
self.insertCtrl.rotatePoints(0, 90, 0)
self.insertCtrl.appendCurveData(triangleCtrl.getCurveData())
self.default_data = {
'name': name,
'location': 'M',
'blendBias': 0.5,
'originXfo': Xfo(Vec3(0.0, 0.0, 0.0)),
'insertXfo': Xfo(Vec3(5.0, 0.0, 0.0)),
'numDeformers': 5
}
self.loadData(self.default_data)
Profiler.getInstance().pop()
# =============
# Data Methods
# =============
def saveData(self):
"""Save the data for the component to be persisted.
Return:
The JSON data object
"""
data = super(TwistComponentGuide, self).saveData()
data['blendBias'] = self.blendBiasAttr.getValue()
data['originXfo'] = self.originCtrl.xfo
data['insertXfo'] = self.insertCtrl.xfo
data['numDeformers'] = self.numDeformersAttr.getValue()
return data
def loadData(self, data):
"""Load a saved guide representation from persisted data.
Arguments:
data -- object, The JSON data object.
Return:
True if successful.
"""
super(TwistComponentGuide, self).loadData(data)
self.blendBiasAttr.setValue(data.get('blendBias', 0.0))
self.originCtrl.xfo = data.get('originXfo', Xfo())
self.insertCtrl.xfo = data.get('insertXfo', Xfo())
self.numDeformersAttr.setValue(data.get('numDeformers', 5))
return True
def getRigBuildData(self):
"""Returns the Guide data used by the Rig Component to define the layout of the final rig.
Return:
The JSON rig data object.
"""
data = super(TwistComponentGuide, self).getRigBuildData()
data['blendBias'] = self.blendBiasAttr.getValue()
data['originXfo'] = self.originCtrl.xfo
data['insertXfo'] = self.insertCtrl.xfo
originUpVXfo = Xfo(Vec3(0.0, 1.0, 0.0))
insertUpVXfo = Xfo(Vec3(0.0, 1.0, 0.0))
data['originUpVXfo'] = self.originCtrl.xfo * originUpVXfo
data['insertUpVXfo'] = self.insertCtrl.xfo * insertUpVXfo
data['numDeformers'] = self.numDeformersAttr.getValue()
return data
# ==============
# Class Methods
# ==============
@classmethod
def getComponentType(cls):
"""Enables introspection of the class prior to construction to determine if it is a guide component.
Return:
The true if this component is a guide component.
"""
return 'Guide'
@classmethod
def getRigComponentClass(cls):
"""Returns the corresponding rig component class for this guide component class
Return:
The rig component class.
"""
return TwistComponentRig
class HandComponentRig(HandComponent):
"""Hand Component"""
def __init__(self, name='Hand', parent=None):
Profiler.getInstance().push("Construct Hand Rig Component:" + name)
super(HandComponentRig, self).__init__(name, parent)
# =========
# Controls
# =========
# Hand
self.handCtrlSpace = CtrlSpace('hand', parent=self.ctrlCmpGrp)
self.handCtrl = Control('hand', parent=self.handCtrlSpace, shape="square")
self.handCtrl.rotatePoints(0, 0, 90.0)
self.handCtrl.lockScale(True, True, True)
self.handCtrl.lockTranslation(True, True, True)
# ==========
# Deformers
# ==========
self.deformersLayer = self.getOrCreateLayer('deformers')
self.defCmpGrp = ComponentGroup(self.getName(), self, parent=self.deformersLayer)
self.addItem('defCmpGrp', self.defCmpGrp)
self.handDef = Joint('hand', parent=self.defCmpGrp)
self.handDef.setComponent(self)
# ==============
# Constrain I/O
# ==============
# Constraint inputs
self.armEndInputConstraint = PoseConstraint('_'.join([self.handCtrlSpace.getName(), 'To', self.armEndInputTgt.getName()]))
self.armEndInputConstraint.setMaintainOffset(True)
self.armEndInputConstraint.addConstrainer(self.armEndInputTgt)
self.handCtrlSpace.addConstraint(self.armEndInputConstraint)
# Constraint outputs
self.handOutputConstraint = PoseConstraint('_'.join([self.handOutputTgt.getName(), 'To', self.handCtrl.getName()]))
self.handOutputConstraint.addConstrainer(self.handCtrl)
self.handOutputTgt.addConstraint(self.handOutputConstraint)
# Constraint deformers
self.handDefConstraint = PoseConstraint('_'.join([self.handDef.getName(), 'To', self.handCtrl.getName()]))
self.handDefConstraint.addConstrainer(self.handCtrl)
self.handDef.addConstraint(self.handDefConstraint)
Profiler.getInstance().pop()
def addFinger(self, name, data):
fingerCtrls = []
fingerJoints = []
parentCtrl = self.handCtrl
for i, joint in enumerate(data):
if i == 0:
jointName = name + 'Meta'
else:
jointName = name + str(i).zfill(2)
jointXfo = joint.get('xfo', Xfo())
jointCrvData = joint.get('curveData')
# Create Controls
newJointCtrlSpace = CtrlSpace(jointName, parent=parentCtrl)
newJointCtrl = Control(jointName, parent=newJointCtrlSpace, shape='square')
newJointCtrl.lockScale(True, True, True)
newJointCtrl.lockTranslation(True, True, True)
if jointCrvData is not None:
newJointCtrl.setCurveData(jointCrvData)
fingerCtrls.append(newJointCtrl)
# Create Deformers
jointDef = Joint(jointName, parent=self.defCmpGrp)
fingerJoints.append(jointDef)
# Create Constraints
# Set Xfos
newJointCtrlSpace.xfo = jointXfo
newJointCtrl.xfo = jointXfo
parentCtrl = newJointCtrl
# =================
# Create Operators
# =================
# Add Deformer KL Op
deformersToCtrlsKLOp = KLOperator(name + 'DefConstraint', 'MultiPoseConstraintSolver', 'Kraken')
self.addOperator(deformersToCtrlsKLOp)
# Add Att Inputs
deformersToCtrlsKLOp.setInput('drawDebug', self.drawDebugInputAttr)
deformersToCtrlsKLOp.setInput('rigScale', self.rigScaleInputAttr)
# Add Xfo Inputs
deformersToCtrlsKLOp.setInput('constrainers', fingerCtrls)
# Add Xfo Outputs
deformersToCtrlsKLOp.setOutput('constrainees', fingerJoints)
return deformersToCtrlsKLOp
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(HandComponentRig, self).loadData(data)
# Data
fingerData = data.get('fingerData')
handXfo = data.get('handXfo', Xfo())
self.handCtrlSpace.xfo = handXfo
self.handCtrl.xfo = handXfo
fingerOps = []
for finger in fingerData.keys():
fingerOp = self.addFinger(finger, fingerData[finger])
fingerOps.append(fingerOp)
# ============
# Set IO Xfos
# ============
self.armEndInputTgt.xfo = handXfo
self.handOutputTgt.xfo = handXfo
# Eval Constraints
self.armEndInputConstraint.evaluate()
self.handOutputConstraint.evaluate()
self.handDefConstraint.evaluate()
# Eval Operators
for op in fingerOps:
op.evaluate()
class HeadComponentRig(HeadComponent):
"""Head Component Rig"""
def __init__(self, name='head', parent=None):
Profiler.getInstance().push("Construct Head Rig Component:" + name)
super(HeadComponentRig, self).__init__(name, parent)
# =========
# Controls
# =========
# Head
self.headCtrl = Control('head', parent=self.ctrlCmpGrp, shape='circle')
self.headCtrl.lockScale(x=True, y=True, z=True)
self.headCtrl.lockTranslation(x=True, y=True, z=True)
self.headCtrlSpace = self.headCtrl.insertCtrlSpace()
self.headCtrl.rotatePoints(0, 0, 90)
self.headCtrl.scalePoints(Vec3(3, 3, 3))
self.headCtrl.translatePoints(Vec3(0, 1, 0.25))
# Eye Left
self.eyeLeftCtrl = Control('eyeLeft',
parent=self.ctrlCmpGrp,
shape='sphere')
self.eyeLeftCtrl.lockScale(x=True, y=True, z=True)
self.eyeLeftCtrl.lockTranslation(x=True, y=True, z=True)
self.eyeLeftCtrlSpace = self.eyeLeftCtrl.insertCtrlSpace()
self.eyeLeftCtrl.rotatePoints(0, 90, 0)
self.eyeLeftCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.eyeLeftCtrl.setColor('blueMedium')
# Eye Right
self.eyeRightCtrl = Control('eyeRight',
parent=self.ctrlCmpGrp,
shape='sphere')
self.eyeRightCtrl.lockScale(x=True, y=True, z=True)
self.eyeRightCtrl.lockTranslation(x=True, y=True, z=True)
self.eyeRightCtrlSpace = self.eyeRightCtrl.insertCtrlSpace()
self.eyeRightCtrl.rotatePoints(0, 90, 0)
self.eyeRightCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.eyeRightCtrl.setColor('blueMedium')
# LookAt Control
self.lookAtCtrl = Control('lookAt',
parent=self.ctrlCmpGrp,
shape='square')
self.lookAtCtrl.lockScale(x=True, y=True, z=True)
self.lookAtCtrl.rotatePoints(90, 0, 0)
self.lookAtCtrlSpace = self.lookAtCtrl.insertCtrlSpace()
self.eyeLeftBase = Transform('eyeLeftBase', parent=self.headCtrl)
self.eyeRightBase = Transform('eyeRightBase', parent=self.headCtrl)
self.eyeLeftUpV = Transform('eyeLeftUpV', parent=self.headCtrl)
self.eyeRightUpV = Transform('eyeRightUpV', parent=self.headCtrl)
self.eyeLeftAtV = Transform('eyeLeftAtV', parent=self.lookAtCtrl)
self.eyeRightAtV = Transform('eyeRightAtV', parent=self.lookAtCtrl)
# Jaw
self.jawCtrl = Control('jaw', parent=self.headCtrl, shape='cube')
self.jawCtrlSpace = self.jawCtrl.insertCtrlSpace()
self.jawCtrl.lockScale(x=True, y=True, z=True)
self.jawCtrl.lockTranslation(x=True, y=True, z=True)
self.jawCtrl.alignOnYAxis(negative=True)
self.jawCtrl.alignOnZAxis()
self.jawCtrl.scalePoints(Vec3(1.45, 0.65, 1.25))
self.jawCtrl.translatePoints(Vec3(0, -0.25, 0))
self.jawCtrl.setColor('orange')
# ==========
# Deformers
# ==========
deformersLayer = self.getOrCreateLayer('deformers')
self.defCmpGrp = ComponentGroup(self.getName(),
self,
parent=deformersLayer)
self.addItem('defCmpGrp', self.defCmpGrp)
headDef = Joint('head', parent=self.defCmpGrp)
headDef.setComponent(self)
jawDef = Joint('jaw', parent=self.defCmpGrp)
jawDef.setComponent(self)
eyeLeftDef = Joint('eyeLeft', parent=self.defCmpGrp)
eyeLeftDef.setComponent(self)
eyeRightDef = Joint('eyeRight', parent=self.defCmpGrp)
eyeRightDef.setComponent(self)
# ==============
# Constrain I/O
# ==============
# Constraint inputs
self.headInputConstraint = PoseConstraint('_'.join([
self.headCtrlSpace.getName(), 'To',
self.neckRefInputTgt.getName()
]))
self.headInputConstraint.setMaintainOffset(True)
self.headInputConstraint.addConstrainer(self.neckRefInputTgt)
self.headCtrlSpace.addConstraint(self.headInputConstraint)
# Constraint outputs
self.headOutputConstraint = PoseConstraint('_'.join(
[self.headOutputTgt.getName(), 'To',
self.headCtrl.getName()]))
self.headOutputConstraint.addConstrainer(self.headCtrl)
self.headOutputTgt.addConstraint(self.headOutputConstraint)
self.jawOutputConstraint = PoseConstraint('_'.join(
[self.jawOutputTgt.getName(), 'To',
self.jawCtrl.getName()]))
self.jawOutputConstraint.addConstrainer(self.jawCtrl)
self.jawOutputTgt.addConstraint(self.jawOutputConstraint)
self.eyeLOutputConstraint = PoseConstraint('_'.join(
[self.eyeLOutputTgt.getName(), 'To',
self.eyeLeftCtrl.getName()]))
self.eyeLOutputConstraint.addConstrainer(self.eyeLeftCtrl)
self.eyeLOutputTgt.addConstraint(self.eyeLOutputConstraint)
self.eyeROutputConstraint = PoseConstraint('_'.join(
[self.eyeROutputTgt.getName(), 'To',
self.eyeRightCtrl.getName()]))
self.eyeROutputConstraint.addConstrainer(self.eyeRightCtrl)
self.eyeROutputTgt.addConstraint(self.eyeROutputConstraint)
# Add Eye Left Direction KL Op
self.eyeLeftDirKLOp = KLOperator('eyeLeftDirKLOp',
'DirectionConstraintSolver', 'Kraken')
self.addOperator(self.eyeLeftDirKLOp)
# Add Att Inputs
self.eyeLeftDirKLOp.setInput('drawDebug', self.drawDebugInputAttr)
self.eyeLeftDirKLOp.setInput('rigScale', self.rigScaleInputAttr)
# Add Xfo Inputs
self.eyeLeftDirKLOp.setInput('position', self.eyeLeftBase)
self.eyeLeftDirKLOp.setInput('upVector', self.eyeLeftUpV)
self.eyeLeftDirKLOp.setInput('atVector', self.eyeLeftAtV)
# Add Xfo Outputs
self.eyeLeftDirKLOp.setOutput('constrainee', self.eyeLeftCtrlSpace)
# Add Eye Right Direction KL Op
self.eyeRightDirKLOp = KLOperator('eyeRightDirKLOp',
'DirectionConstraintSolver',
'Kraken')
self.addOperator(self.eyeRightDirKLOp)
# Add Att Inputs
self.eyeRightDirKLOp.setInput('drawDebug', self.drawDebugInputAttr)
self.eyeRightDirKLOp.setInput('rigScale', self.rigScaleInputAttr)
# Add Xfo Inputs
self.eyeRightDirKLOp.setInput('position', self.eyeRightBase)
self.eyeRightDirKLOp.setInput('upVector', self.eyeRightUpV)
self.eyeRightDirKLOp.setInput('atVector', self.eyeRightAtV)
# Add Xfo Outputs
self.eyeRightDirKLOp.setOutput('constrainee', self.eyeRightCtrlSpace)
# Add Deformer Joints KL Op
self.outputsToDeformersKLOp = KLOperator('headDeformerKLOp',
'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.headOutputTgt, self.jawOutputTgt, self.eyeROutputTgt,
self.eyeLOutputTgt
])
# Add Xfo Outputs
self.outputsToDeformersKLOp.setOutput(
'constrainees', [headDef, jawDef, eyeRightDef, eyeLeftDef])
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(HeadComponentRig, self).loadData(data)
headXfo = data.get('headXfo')
headCrvData = data.get('headCrvData')
eyeLeftXfo = data.get('eyeLeftXfo')
eyeLeftCrvData = data.get('eyeLeftCrvData')
eyeRightXfo = data.get('eyeRightXfo')
eyeRightCrvData = data.get('eyeRightCrvData')
jawXfo = data.get('jawXfo')
jawCrvData = data.get('jawCrvData')
self.headCtrlSpace.xfo = headXfo
self.headCtrl.xfo = headXfo
self.headCtrl.setCurveData(headCrvData)
# self.eyeLeftCtrlSpace.xfo = eyeLeftXfo
# self.eyeLeftCtrl.xfo = eyeLeftXfo
self.eyeLeftCtrl.setCurveData(eyeLeftCrvData)
# self.eyeRightCtrlSpace.xfo = eyeRightXfo
# self.eyeRightCtrl.xfo = eyeRightXfo
self.eyeRightCtrl.setCurveData(eyeRightCrvData)
# LookAt
eyeLeftRelXfo = headXfo.inverse() * eyeLeftXfo
eyeRightRelXfo = headXfo.inverse() * eyeRightXfo
eyeMidRelPos = eyeLeftRelXfo.tr.linearInterpolate(
eyeRightRelXfo.tr, 0.5)
eyeMidRelPos = eyeMidRelPos + Vec3(0.0, 0.0, 8.0)
eyeLen = eyeLeftRelXfo.tr.distanceTo(eyeRightRelXfo.tr)
self.eyeLeftBase.xfo = eyeLeftXfo
self.eyeRightBase.xfo = eyeRightXfo
self.eyeLeftUpV.xfo = eyeLeftXfo * Xfo(Vec3(0, 1, 0))
self.eyeRightUpV.xfo = eyeRightXfo * Xfo(Vec3(0, 1, 0))
self.eyeLeftAtV.xfo.tr = eyeLeftXfo.transformVector(Vec3(
8.0, 0.0, 0.0))
self.eyeRightAtV.xfo.tr = eyeRightXfo.transformVector(
Vec3(8.0, 0.0, 0.0))
lookAtXfo = headXfo.clone()
lookAtXfo.tr = headXfo.transformVector(eyeMidRelPos)
self.lookAtCtrl.scalePoints(Vec3(eyeLen * 1.6, eyeLen * 0.65, 1.0))
self.lookAtCtrl.xfo = lookAtXfo
self.lookAtCtrlSpace.xfo = lookAtXfo
self.jawCtrlSpace.xfo = jawXfo
self.jawCtrl.xfo = jawXfo
self.jawCtrl.setCurveData(jawCrvData)
# ============
# Set IO Xfos
# ============
self.neckRefInputTgt.xfo = headXfo
self.worldRefInputTgt.xfo = headXfo
self.headOutputTgt.xfo = headXfo
self.jawOutputTgt.xfo = jawXfo
self.eyeLOutputTgt.xfo = eyeLeftXfo
self.eyeROutputTgt.xfo = eyeRightXfo
# Eval Constraints
self.headInputConstraint.evaluate()
self.headOutputConstraint.evaluate()
self.jawOutputConstraint.evaluate()
self.eyeLOutputConstraint.evaluate()
self.eyeROutputConstraint.evaluate()
# Eval Operators
self.eyeLeftDirKLOp.evaluate()
self.eyeRightDirKLOp.evaluate()
self.outputsToDeformersKLOp.evaluate()
# Have to set the eye control xfos to match the evaluated xfos from
self.eyeLeftCtrl.xfo = self.eyeLeftCtrlSpace.xfo
self.eyeRightCtrl.xfo = self.eyeRightCtrlSpace.xfo
class FabriceSpineRig(FabriceSpine):
"""Fabrice Spine Component"""
def __init__(self, name="spine", parent=None):
Profiler.getInstance().push("Construct Spine Rig Component:" + name)
super(FabriceSpineRig, self).__init__(name, parent)
# =========
# Controls
# =========
# COG
self.cogCtrlSpace = CtrlSpace('cog', parent=self.ctrlCmpGrp)
self.cogCtrl = Control('cog', parent=self.cogCtrlSpace, shape="circle")
self.cogCtrl.rotatePoints(90, 0, 0)
self.cogCtrl.scalePoints(Vec3(3.0, 3.0, 3.0))
self.cogCtrl.translatePoints(Vec3(0.0, 0.0, 0.2))
self.cogCtrl.lockScale(x=True, y=True, z=True)
self.cogCtrl.setColor("orange")
# Spine Base
self.spineBaseCtrlSpace = CtrlSpace('spineBase', parent=self.cogCtrl)
self.spineBaseCtrl = Control('spineBase',
parent=self.spineBaseCtrlSpace,
shape="pin")
self.spineBaseCtrl.rotatePoints(90, 0, 0)
self.spineBaseCtrl.translatePoints(Vec3(0, 1.0, 0))
self.spineBaseCtrl.lockScale(x=True, y=True, z=True)
# Spine Base Handle
self.spineBaseHandleCtrlSpace = CtrlSpace('spineBaseHandle',
parent=self.spineBaseCtrl)
self.spineBaseHandleCtrl = Control(
'spineBaseHandle',
parent=self.spineBaseHandleCtrlSpace,
shape="pin")
self.spineBaseHandleCtrl.rotatePoints(90, 0, 0)
self.spineBaseHandleCtrl.translatePoints(Vec3(0, 1.0, 0))
self.spineBaseHandleCtrl.lockScale(x=True, y=True, z=True)
self.spineBaseHandleCtrl.setColor("orange")
# Spine End
self.spineEndCtrlSpace = CtrlSpace('spineEnd', parent=self.cogCtrl)
self.spineEndCtrl = Control('spineEnd',
parent=self.spineEndCtrlSpace,
shape="pin")
self.spineEndCtrl.rotatePoints(90, 0, 0)
self.spineEndCtrl.lockScale(x=True, y=True, z=True)
self.spineEndCtrl.translatePoints(Vec3(0, 1.0, 0))
# Spine End Handle
self.spineEndHandleCtrlSpace = CtrlSpace('spineEndHandle',
parent=self.spineEndCtrl)
self.spineEndHandleCtrl = Control('spineEndHandle',
parent=self.spineEndHandleCtrlSpace,
shape="pin")
self.spineEndHandleCtrl.rotatePoints(90, 0, 0)
self.spineEndHandleCtrl.translatePoints(Vec3(0, 1.0, 0))
self.spineEndHandleCtrl.lockScale(x=True, y=True, z=True)
self.spineEndHandleCtrl.setColor("orange")
# ==========
# Deformers
# ==========
deformersLayer = self.getOrCreateLayer('deformers')
self.defCmpGrp = ComponentGroup(self.getName(),
self,
parent=deformersLayer)
self.chestDef = Joint('chest', parent=self.defCmpGrp)
self.chestDef.setComponent(self)
self.deformerJoints = []
self.spineOutputs = []
self.setNumDeformers(1)
# =====================
# Create Component I/O
# =====================
# Setup component Xfo I/O's
self.spineVertebraeOutput.setTarget(self.spineOutputs)
# =====================
# Constraint Deformers
# =====================
self.chestDefConstraint = PoseConstraint('_'.join(
[self.chestDef.getName(), 'To',
self.spineBaseOutputTgt.getName()]))
self.chestDefConstraint.addConstrainer(self.spineBaseOutputTgt)
self.chestDef.addConstraint(self.chestDefConstraint)
# ==============
# Constrain I/O
# ==============
# Constraint inputs
self.spineSrtInputConstraint = PoseConstraint('_'.join([
self.cogCtrlSpace.getName(), 'To',
self.spineMainSrtInputTgt.getName()
]))
self.spineSrtInputConstraint.addConstrainer(self.spineMainSrtInputTgt)
self.spineSrtInputConstraint.setMaintainOffset(True)
self.cogCtrlSpace.addConstraint(self.spineSrtInputConstraint)
# Constraint outputs
self.spineCogOutputConstraint = PoseConstraint('_'.join(
[self.spineCogOutputTgt.getName(), 'To',
self.cogCtrl.getName()]))
self.spineCogOutputConstraint.addConstrainer(self.cogCtrl)
self.spineCogOutputTgt.addConstraint(self.spineCogOutputConstraint)
# Spine Base
self.spineBaseOutputPosConstraint = PositionConstraint('_'.join([
self.spineBaseOutputTgt.getName(), 'PosTo',
self.spineOutputs[0].getName()
]))
self.spineBaseOutputPosConstraint.addConstrainer(self.spineOutputs[0])
self.spineBaseOutputTgt.addConstraint(
self.spineBaseOutputPosConstraint)
self.spineBaseOutputOriConstraint = OrientationConstraint('_'.join([
self.spineBaseOutputTgt.getName(), 'PosTo',
self.cogCtrl.getName()
]))
self.spineBaseOutputOriConstraint.addConstrainer(self.cogCtrl)
self.spineBaseOutputTgt.addConstraint(
self.spineBaseOutputOriConstraint)
# Spine End
self.spineEndOutputConstraint = PoseConstraint('_'.join(
[self.spineEndOutputTgt.getName(), 'To', 'spineEnd']))
self.spineEndOutputConstraint.addConstrainer(self.spineOutputs[0])
self.spineEndOutputTgt.addConstraint(self.spineEndOutputConstraint)
self.spineEndCtrlOutputConstraint = PoseConstraint('_'.join([
self.spineEndCtrlOutputTgt.getName(), 'To',
self.spineEndCtrl.getName()
]))
self.spineEndCtrlOutputConstraint.addConstrainer(self.spineEndCtrl)
self.spineEndCtrlOutputTgt.addConstraint(
self.spineEndCtrlOutputConstraint)
# ===============
# Add Splice Ops
# ===============
# Add Spine Splice Op
self.bezierSpineSpliceOp = SpliceOperator('spineSpliceOp',
'BezierSpineSolver',
'Kraken')
self.addOperator(self.bezierSpineSpliceOp)
# Add Att Inputs
self.bezierSpineSpliceOp.setInput('drawDebug', self.drawDebugInputAttr)
self.bezierSpineSpliceOp.setInput('rigScale', self.rigScaleInputAttr)
self.bezierSpineSpliceOp.setInput('length', self.lengthInputAttr)
# Add Xfo Inputs
self.bezierSpineSpliceOp.setInput('base', self.spineBaseCtrl)
self.bezierSpineSpliceOp.setInput('baseHandle',
self.spineBaseHandleCtrl)
self.bezierSpineSpliceOp.setInput('tipHandle', self.spineEndHandleCtrl)
self.bezierSpineSpliceOp.setInput('tip', self.spineEndCtrl)
# Add Xfo Outputs
self.bezierSpineSpliceOp.setOutput('outputs', self.spineOutputs)
# Add Deformer Splice Op
self.deformersToOutputsSpliceOp = SpliceOperator(
'spineDeformerSpliceOp',
'MultiPoseConstraintSolver',
'Kraken',
alwaysEval=True)
self.addOperator(self.deformersToOutputsSpliceOp)
# Add Att Inputs
self.deformersToOutputsSpliceOp.setInput('drawDebug',
self.drawDebugInputAttr)
self.deformersToOutputsSpliceOp.setInput('rigScale',
self.rigScaleInputAttr)
# Add Xfo Outputs
self.deformersToOutputsSpliceOp.setInput('constrainers',
self.spineOutputs)
# Add Xfo Outputs
self.deformersToOutputsSpliceOp.setOutput('constrainees',
self.deformerJoints)
Profiler.getInstance().pop()
def setNumDeformers(self, numDeformers):
# Add new deformers and outputs
for i in xrange(len(self.spineOutputs), numDeformers):
name = 'spine' + str(i + 1).zfill(2)
spineOutput = ComponentOutput(name, parent=self.outputHrcGrp)
self.spineOutputs.append(spineOutput)
for i in xrange(len(self.deformerJoints), numDeformers):
name = 'spine' + str(i + 1).zfill(2)
spineDef = Joint(name, parent=self.defCmpGrp)
spineDef.setComponent(self)
self.deformerJoints.append(spineDef)
return True
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(FabriceSpineRig, self).loadData(data)
# Get Data
cogPos = data['cogPos']
cogCtrlCrvData = data['cogCtrlCrvData']
spineBasePos = data['spineBasePos']
spineBaseCtrlCrvData = data['spineBaseCtrlCrvData']
spineBaseHandlePos = data['spineBaseHandlePos']
spineBaseHandleCtrlCrvData = data['spineBaseHandleCtrlCrvData']
spineEndHandlePos = data['spineEndHandlePos']
spineEndHandleCtrlCrvData = data['spineEndHandleCtrlCrvData']
spineEndPos = data['spineEndPos']
spineEndCtrlCrvData = data['spineEndCtrlCrvData']
numDeformers = data['numDeformers']
# Set Xfos
self.cogCtrlSpace.xfo.tr = cogPos
self.cogCtrl.xfo.tr = cogPos
self.cogCtrl.setCurveData(cogCtrlCrvData)
self.spineBaseCtrlSpace.xfo.tr = spineBasePos
self.spineBaseCtrl.xfo.tr = spineBasePos
self.spineBaseCtrl.setCurveData(spineBaseCtrlCrvData)
self.spineBaseHandleCtrlSpace.xfo.tr = spineBaseHandlePos
self.spineBaseHandleCtrl.xfo.tr = spineBaseHandlePos
self.spineBaseHandleCtrl.setCurveData(spineBaseHandleCtrlCrvData)
self.spineEndHandleCtrlSpace.xfo.tr = spineEndHandlePos
self.spineEndHandleCtrl.xfo.tr = spineEndHandlePos
self.spineEndHandleCtrl.setCurveData(spineEndHandleCtrlCrvData)
self.spineEndCtrlSpace.xfo.tr = spineEndPos
self.spineEndCtrl.xfo.tr = spineEndPos
self.spineEndCtrl.setCurveData(spineEndCtrlCrvData)
length = spineBasePos.distanceTo(
spineBaseHandlePos) + spineBaseHandlePos.distanceTo(
spineEndHandlePos) + spineEndHandlePos.distanceTo(spineEndPos)
self.lengthInputAttr.setMax(length * 3.0)
self.lengthInputAttr.setValue(length)
# Update number of deformers and outputs
self.setNumDeformers(numDeformers)
# Updating constraint to use the updated last output.
self.spineEndOutputConstraint.setConstrainer(self.spineOutputs[-1],
index=0)
# ============
# Set IO Xfos
# ============
# ====================
# Evaluate Splice Ops
# ====================
# evaluate the spine op so that all the output transforms are updated.
self.bezierSpineSpliceOp.evaluate()
# evaluate the constraint op so that all the joint transforms are updated.
self.deformersToOutputsSpliceOp.evaluate()
# evaluate the constraints to ensure the outputs are now in the correct location.
self.spineSrtInputConstraint.evaluate()
self.spineCogOutputConstraint.evaluate()
self.spineBaseOutputPosConstraint.evaluate()
self.spineBaseOutputOriConstraint.evaluate()
self.spineEndOutputConstraint.evaluate()
self.spineEndCtrlOutputConstraint.evaluate()
def addFinger(self, name):
digitSizeAttributes = []
fingerGuideCtrls = []
firstDigitCtrl = Control(name + "01", parent=self.handCtrl, shape='sphere')
firstDigitCtrl.scalePoints(Vec3(0.125, 0.125, 0.125))
firstDigitShapeCtrl = Control(name + "Shp01", parent=self.guideCtrlHrcGrp, shape='square')
firstDigitShapeCtrl.setColor('yellow')
firstDigitShapeCtrl.scalePoints(Vec3(0.175, 0.175, 0.175))
firstDigitShapeCtrl.translatePoints(Vec3(0.0, 0.125, 0.0))
fingerGuideCtrls.append(firstDigitShapeCtrl)
firstDigitCtrl.shapeCtrl = firstDigitShapeCtrl
firstDigitVisAttr = firstDigitShapeCtrl.getVisibilityAttr()
firstDigitVisAttr.connect(self.ctrlShapeToggle)
triangleCtrl = Control('tempCtrl', parent=None, shape='triangle')
triangleCtrl.rotatePoints(90.0, 0.0, 0.0)
triangleCtrl.scalePoints(Vec3(0.025, 0.025, 0.025))
triangleCtrl.translatePoints(Vec3(0.0, 0.0875, 0.0))
firstDigitCtrl.appendCurveData(triangleCtrl.getCurveData())
firstDigitCtrl.lockScale(True, True, True)
digitSettingsAttrGrp = AttributeGroup("DigitSettings", parent=firstDigitCtrl)
digitSizeAttr = ScalarAttribute('size', value=0.25, parent=digitSettingsAttrGrp)
digitSizeAttributes.append(digitSizeAttr)
# Set Finger
self.fingers[name] = []
self.fingers[name].append(firstDigitCtrl)
parent = firstDigitCtrl
numJoints = self.numJointsAttr.getValue()
if name == "thumb":
numJoints = 3
for i in xrange(2, numJoints + 2):
digitCtrl = Control(name + str(i).zfill(2), parent=parent, shape='sphere')
if i != numJoints + 1:
digitCtrl.scalePoints(Vec3(0.125, 0.125, 0.125))
digitCtrl.appendCurveData(triangleCtrl.getCurveData())
digitShapeCtrl = Control(name + 'Shp' + str(i).zfill(2), parent=self.guideCtrlHrcGrp, shape='circle')
digitShapeCtrl.setColor('yellow')
digitShapeCtrl.scalePoints(Vec3(0.175, 0.175, 0.175))
digitShapeCtrl.getVisibilityAttr().connect(self.ctrlShapeToggle)
digitCtrl.shapeCtrl = digitShapeCtrl
if i == 2:
digitShapeCtrl.translatePoints(Vec3(0.0, 0.125, 0.0))
else:
digitShapeCtrl.rotatePoints(0.0, 0.0, 90.0)
fingerGuideCtrls.append(digitShapeCtrl)
# Add size attr to all but last guide control
digitSettingsAttrGrp = AttributeGroup("DigitSettings", parent=digitCtrl)
digitSizeAttr = ScalarAttribute('size', value=0.25, parent=digitSettingsAttrGrp)
digitSizeAttributes.append(digitSizeAttr)
else:
digitCtrl.scalePoints(Vec3(0.0875, 0.0875, 0.0875))
digitCtrl.lockScale(True, True, True)
self.fingers[name].append(digitCtrl)
parent = digitCtrl
# ===========================
# Create Canvas Operators
# ===========================
# Add Finger Guide Canvas Op
fingerGuideCanvasOp = CanvasOperator(name + 'FingerGuideOp', 'Kraken.Solvers.Biped.BipedFingerGuideSolver')
self.addOperator(fingerGuideCanvasOp)
# Add Att Inputs
fingerGuideCanvasOp.setInput('drawDebug', self.drawDebugInputAttr)
fingerGuideCanvasOp.setInput('rigScale', self.rigScaleInputAttr)
# Add Xfo Inputs
fingerGuideCanvasOp.setInput('controls', self.fingers[name])
fingerGuideCanvasOp.setInput('planeSizes', digitSizeAttributes)
# Add Xfo Outputs
fingerGuideCanvasOp.setOutput('result', fingerGuideCtrls)
fingerGuideCanvasOp.setOutput('forceEval', firstDigitCtrl.getVisibilityAttr())
return firstDigitCtrl
class NeckComponentGuide(NeckComponent):
"""Neck Component Guide"""
def __init__(self, name='neck', parent=None, *args, **kwargs):
Profiler.getInstance().push('Construct Neck Component:' + name)
super(NeckComponentGuide, self).__init__(name, parent, *args, **kwargs)
# =========
# Controls
# =========
# Guide Controls
self.neckCtrl = Control('neck', parent=self.ctrlCmpGrp, shape='sphere')
self.neckCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.neckMidCtrl = Control('neckMid', parent=self.ctrlCmpGrp, shape='sphere')
self.neckMidCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.neckEndCtrl = Control('neckEnd', parent=self.ctrlCmpGrp, shape='sphere')
self.neckEndCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.neckCtrlShape = Control('neck', parent=self.ctrlCmpGrp, shape='pin')
self.neckCtrlShape.rotatePoints(90.0, 0.0, 0.0)
self.neckCtrlShape.rotatePoints(0.0, 90.0, 0.0)
self.neckCtrlShape.setColor('orange')
self.neckMidCtrlShape = Control('neckMid', parent=self.ctrlCmpGrp, shape='pin')
self.neckMidCtrlShape.rotatePoints(90.0, 0.0, 0.0)
self.neckMidCtrlShape.rotatePoints(0.0, 90.0, 0.0)
self.neckMidCtrlShape.setColor('orange')
# Guide Operator
self.neckGuideKLOp = KLOperator(name + 'GuideKLOp', 'NeckGuideSolver', 'Kraken')
self.addOperator(self.neckGuideKLOp)
# Add Att Inputs
self.neckGuideKLOp.setInput('drawDebug', self.drawDebugInputAttr)
self.neckGuideKLOp.setInput('rigScale', self.rigScaleInputAttr)
# Add Source Inputs
self.neckGuideKLOp.setInput('sources', [self.neckCtrl, self.neckMidCtrl, self.neckEndCtrl])
# Add Target Outputs
self.neckGuideKLOp.setOutput('targets', [self.neckCtrlShape, self.neckMidCtrlShape])
# Calculate default values
neckVec = Vec3(0.0, 16.00, -0.75)
neckMidVec = Vec3(0.0, 16.50, -0.50)
neckEndVec = Vec3(0.0, 17.00, -0.25)
upVector = Vec3(0.0, 0.0, -1.0)
neckOri = Quat()
neckOri.setFromDirectionAndUpvector((neckMidVec - neckVec).unit(),
((neckVec + upVector) - neckVec).unit())
neckMidOri = Quat()
neckMidOri.setFromDirectionAndUpvector((neckEndVec - neckMidVec).unit(),
((neckMidVec + upVector) - neckMidVec).unit())
self.default_data = {
"name": name,
"location": "M",
"neckXfo": Xfo(tr=neckVec, ori=neckOri),
"neckMidXfo": Xfo(tr=neckMidVec, ori=neckMidOri),
"neckEndXfo": Xfo(tr=neckEndVec, ori=neckMidOri),
"neckCrvData": self.neckCtrlShape.getCurveData(),
"neckMidCrvData": self.neckMidCtrlShape.getCurveData()
}
self.loadData(self.default_data)
Profiler.getInstance().pop()
# =============
# Data Methods
# =============
def saveData(self):
"""Save the data for the component to be persisted.
Return:
The JSON data object
"""
data = super(NeckComponentGuide, self).saveData()
data['neckXfo'] = self.neckCtrl.xfo
data['neckMidXfo'] = self.neckMidCtrl.xfo
data['neckEndXfo'] = self.neckEndCtrl.xfo
data['neckCrvData'] = self.neckCtrlShape.getCurveData()
data['neckMidCrvData'] = self.neckMidCtrlShape.getCurveData()
return data
def loadData(self, data):
"""Load a saved guide representation from persisted data.
Arguments:
data (object): The JSON data object.
Returns:
bool: True if successful.
"""
super(NeckComponentGuide, self).loadData(data)
self.neckCtrl.xfo = data.get('neckXfo')
self.neckMidCtrl.xfo = data.get('neckMidXfo')
self.neckEndCtrl.xfo = data.get('neckEndXfo')
self.neckCtrlShape.setCurveData(data.get('neckCrvData'))
self.neckMidCtrlShape.setCurveData(data.get('neckMidCrvData'))
# Evaluate guide operators
self.neckGuideKLOp.evaluate()
return True
def getRigBuildData(self):
"""Returns the Guide data used by the Rig Component to define the layout
of the final rig.
Return:
The JSON rig data object.
"""
data = super(NeckComponentGuide, self).getRigBuildData()
neckEndXfo = Xfo(tr=self.neckEndCtrl.xfo.tr,
ori=self.neckMidCtrlShape.xfo.ori)
data['neckXfo'] = self.neckCtrlShape.xfo
data['neckCrvData'] = self.neckCtrlShape.getCurveData()
data['neckMidXfo'] = self.neckMidCtrlShape.xfo
data['neckMidCrvData'] = self.neckMidCtrlShape.getCurveData()
data['neckEndXfo'] = neckEndXfo
return data
# ==============
# Class Methods
# ==============
@classmethod
def getComponentType(cls):
"""Enables introspection of the class prior to construction to determine
if it is a guide component.
Returns:
bool: Whether the component is a guide component.
"""
return 'Guide'
@classmethod
def getRigComponentClass(cls):
"""Returns the corresponding rig component class for this guide
component class.
Returns:
class: The rig component class.
"""
return NeckComponentRig
class ClavicleComponentGuide(ClavicleComponent):
"""Clavicle Component Guide"""
def __init__(self, name='clavicle', parent=None, *args, **kwargs):
Profiler.getInstance().push("Construct Clavicle Guide Component:" + name)
super(ClavicleComponentGuide, self).__init__(name, parent, *args, **kwargs)
# =========
# Controls
# =========
# Guide Controls
guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self)
self.clavicleCtrl = Control('clavicle', parent=self.ctrlCmpGrp, shape="circle")
self.clavicleCtrl.scalePoints(Vec3(0.75, 0.75, 0.75))
self.clavicleCtrl.rotatePoints(0.0, 0.0, 90.0)
self.clavicleGuideSettingsAttrGrp = AttributeGroup("Settings", parent=self.clavicleCtrl)
self.handDebugInputAttr = BoolAttribute('drawDebug', value=False, parent=self.clavicleGuideSettingsAttrGrp)
self.drawDebugInputAttr.connect(self.handDebugInputAttr)
self.clavicleUpVCtrl = Control('clavicleUpV', parent=self.clavicleCtrl, shape="triangle")
self.clavicleUpVCtrl.setColor('red')
self.clavicleUpVCtrl.rotatePoints(-90.0, 0.0, 0.0)
self.clavicleUpVCtrl.rotatePoints(0.0, 90.0, 0.0)
self.clavicleUpVCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.clavicleEndCtrl = Control('clavicleEnd', parent=self.ctrlCmpGrp, shape="cube")
self.clavicleEndCtrl.scalePoints(Vec3(0.25, 0.25, 0.25))
self.default_data = {
"name": name,
"location": "L",
"clavicleXfo": Xfo(Vec3(0.15, 15.5, -0.5)),
"clavicleUpVXfo": Xfo(Vec3(0.15, 16.5, -0.5)),
"clavicleEndXfo": Xfo(Vec3(2.25, 15.5, -0.75))
}
self.loadData(self.default_data)
Profiler.getInstance().pop()
# =============
# Data Methods
# =============
def saveData(self):
"""Save the data for the component to be persisted.
Return:
The JSON data object
"""
data = super(ClavicleComponentGuide, self).saveData()
data['clavicleXfo'] = self.clavicleCtrl.xfo
data['clavicleUpVXfo'] = self.clavicleUpVCtrl.xfo
data['clavicleEndXfo'] = self.clavicleEndCtrl.xfo
return data
def loadData(self, data):
"""Load a saved guide representation from persisted data.
Arguments:
data -- object, The JSON data object.
Return:
True if successful.
"""
super(ClavicleComponentGuide, self).loadData( data )
self.clavicleCtrl.xfo = data['clavicleXfo']
self.clavicleUpVCtrl.xfo = data['clavicleUpVXfo']
self.clavicleEndCtrl.xfo = data['clavicleEndXfo']
return True
def getRigBuildData(self):
"""Returns the Guide data used by the Rig Component to define the layout of the final rig..
Return:
The JSON rig data object.
"""
data = super(ClavicleComponentGuide, self).getRigBuildData()
# Values
claviclePosition = self.clavicleCtrl.xfo.tr
clavicleUpV = self.clavicleUpVCtrl.xfo.tr
clavicleEndPosition = self.clavicleEndCtrl.xfo.tr
# Calculate Clavicle Xfo
clavicleXfo = Xfo()
clavicleOri = Quat()
clavicleOri.setFromDirectionAndUpvector((clavicleEndPosition - claviclePosition).unit(),
(clavicleUpV - claviclePosition).unit())
xAlignOffset = Quat()
xAlignOffset.setFromAxisAndAngle(Vec3(0, 1, 0), Math_degToRad(-90))
clavicleXfo.ori = clavicleOri * xAlignOffset
clavicleXfo.tr = claviclePosition
clavicleLen = claviclePosition.subtract(clavicleEndPosition).length()
data['clavicleXfo'] = clavicleXfo
data['clavicleLen'] = clavicleLen
return data
# ==============
# Class Methods
# ==============
@classmethod
def getComponentType(cls):
"""Enables introspection of the class prior to construction to determine if it is a guide component.
Return:
The true if this component is a guide component.
"""
return 'Guide'
@classmethod
def getRigComponentClass(cls):
"""Returns the corresponding rig component class for this guide component class
Return:
The rig component class.
"""
return ClavicleComponentRig
class HandComponentRig(HandComponent):
"""Hand Component"""
def __init__(self, name='Hand', parent=None):
Profiler.getInstance().push("Construct Hand Rig Component:" + name)
super(HandComponentRig, self).__init__(name, parent)
# =========
# Controls
# =========
# Hand
self.handCtrlSpace = CtrlSpace('hand', parent=self.ctrlCmpGrp)
self.handCtrl = Control('hand', parent=self.handCtrlSpace, shape="square")
self.handCtrl.rotatePoints(0, 0, 90.0)
self.handCtrl.lockScale(True, True, True)
self.handCtrl.lockTranslation(True, True, True)
# ==========
# Deformers
# ==========
self.deformersLayer = self.getOrCreateLayer('deformers')
self.defCmpGrp = ComponentGroup(self.getName(), self, parent=self.deformersLayer)
self.addItem('defCmpGrp', self.defCmpGrp)
self.handDef = Joint('hand', parent=self.defCmpGrp)
self.handDef.setComponent(self)
# ==============
# Constrain I/O
# ==============
# Constraint inputs
self.armEndInputConstraint = PoseConstraint('_'.join([self.handCtrlSpace.getName(), 'To', self.armEndInputTgt.getName()]))
self.armEndInputConstraint.setMaintainOffset(True)
self.armEndInputConstraint.addConstrainer(self.armEndInputTgt)
self.handCtrlSpace.addConstraint(self.armEndInputConstraint)
# Constraint outputs
self.handOutputConstraint = PoseConstraint('_'.join([self.handOutputTgt.getName(), 'To', self.handCtrl.getName()]))
self.handOutputConstraint.addConstrainer(self.handCtrl)
self.handOutputTgt.addConstraint(self.handOutputConstraint)
# Constraint deformers
self.handDefConstraint = PoseConstraint('_'.join([self.handDef.getName(), 'To', self.handCtrl.getName()]))
self.handDefConstraint.addConstrainer(self.handCtrl)
self.handDef.addConstraint(self.handDefConstraint)
Profiler.getInstance().pop()
def addFinger(self, name, data):
fingerCtrls = []
fingerJoints = []
parentCtrl = self.handCtrl
for i, joint in enumerate(data):
if i == 0:
jointName = name + 'Meta'
else:
jointName = name + str(i).zfill(2)
jointXfo = joint.get('xfo', Xfo())
jointCrvData = joint.get('curveData')
# Create Controls
newJointCtrlSpace = CtrlSpace(jointName, parent=parentCtrl)
newJointCtrl = Control(jointName, parent=newJointCtrlSpace, shape='square')
newJointCtrl.lockScale(True, True, True)
newJointCtrl.lockTranslation(True, True, True)
if jointCrvData is not None:
newJointCtrl.setCurveData(jointCrvData)
fingerCtrls.append(newJointCtrl)
# Create Deformers
jointDef = Joint(jointName, parent=self.defCmpGrp)
fingerJoints.append(jointDef)
# Create Constraints
# Set Xfos
newJointCtrlSpace.xfo = jointXfo
newJointCtrl.xfo = jointXfo
parentCtrl = newJointCtrl
# =================
# Create Operators
# =================
# Add Deformer KL Op
deformersToCtrlsKLOp = KLOperator(name + 'DeformerKLOp', 'MultiPoseConstraintSolver', 'Kraken')
self.addOperator(deformersToCtrlsKLOp)
# Add Att Inputs
deformersToCtrlsKLOp.setInput('drawDebug', self.drawDebugInputAttr)
deformersToCtrlsKLOp.setInput('rigScale', self.rigScaleInputAttr)
# Add Xfo Inputs
deformersToCtrlsKLOp.setInput('constrainers', fingerCtrls)
# Add Xfo Outputs
deformersToCtrlsKLOp.setOutput('constrainees', fingerJoints)
return deformersToCtrlsKLOp
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(HandComponentRig, self).loadData(data)
# Data
fingerData = data.get('fingerData')
handXfo = data.get('handXfo', Xfo())
self.handCtrlSpace.xfo = handXfo
self.handCtrl.xfo = handXfo
fingerOps = []
for finger in fingerData.keys():
fingerOp = self.addFinger(finger, fingerData[finger])
fingerOps.append(fingerOp)
# ============
# Set IO Xfos
# ============
self.armEndInputTgt.xfo = handXfo
self.handOutputTgt.xfo = handXfo
# Eval Constraints
self.armEndInputConstraint.evaluate()
self.handOutputConstraint.evaluate()
self.handDefConstraint.evaluate()
# Eval Operators
for op in fingerOps:
op.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()
class HandComponentGuide(HandComponent):
"""Hand Component Guide"""
def __init__(self, name='hand', parent=None, *args, **kwargs):
Profiler.getInstance().push("Construct Hand Guide Component:" + name)
super(HandComponentGuide, self).__init__(name, parent, *args, **kwargs)
# =========
# Controls
# =========
# Guide Controls
self.guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self)
self.digitNamesAttr = StringAttribute('digitNames', value="thumb,index,middle,ring,pinky", parent=self.guideSettingsAttrGrp)
self.digitNamesAttr.setValueChangeCallback(self.updateFingers)
self.numJointsAttr = IntegerAttribute('numJoints', value=4, minValue=2, maxValue=20, parent=self.guideSettingsAttrGrp)
self.numJointsAttr.setValueChangeCallback(self.resizeDigits)
self.fingers = OrderedDict()
self.handCtrl = Control('hand', parent=self.ctrlCmpGrp, shape="square")
self.handCtrl.rotatePoints(0.0, 0.0, 90.0)
self.handCtrl.scalePoints(Vec3(1.0, 0.75, 1.0))
self.handCtrl.setColor('yellow')
self.handGuideSettingsAttrGrp = AttributeGroup("Settings", parent=self.handCtrl)
self.ctrlShapeToggle = BoolAttribute('ctrlShape_vis', value=False, parent=self.handGuideSettingsAttrGrp)
self.handDebugInputAttr = BoolAttribute('drawDebug', value=False, parent=self.handGuideSettingsAttrGrp)
self.drawDebugInputAttr.connect(self.handDebugInputAttr)
self.guideCtrlHrcGrp = HierarchyGroup('controlShapes', parent=self.ctrlCmpGrp)
self.default_data = {
"name": name,
"location": "L",
"handXfo": Xfo(Vec3(7.1886, 12.2819, 0.4906)),
"digitNames": self.digitNamesAttr.getValue(),
"numJoints": self.numJointsAttr.getValue(),
"fingers": self.fingers
}
self.loadData(self.default_data)
Profiler.getInstance().pop()
# =============
# Data Methods
# =============
def saveData(self):
"""Save the data for the component to be persisted.
Return:
The JSON data object
"""
data = super(HandComponentGuide, self).saveData()
data['handXfo'] = self.handCtrl.xfo
data['digitNames'] = self.digitNamesAttr.getValue()
data['numJoints'] = self.numJointsAttr.getValue()
fingerXfos = {}
fingerShapeCtrlData = {}
for finger in self.fingers.keys():
fingerXfos[finger] = [x.xfo for x in self.fingers[finger]]
fingerShapeCtrlData[finger] = []
for i, digit in enumerate(self.fingers[finger]):
if i != len(self.fingers[finger]) - 1:
fingerShapeCtrlData[finger].append(digit.shapeCtrl.getCurveData())
data['fingersGuideXfos'] = fingerXfos
data['fingerShapeCtrlData'] = fingerShapeCtrlData
return data
def loadData(self, data):
"""Load a saved guide representation from persisted data.
Arguments:
data -- object, The JSON data object.
Return:
True if successful.
"""
super(HandComponentGuide, self).loadData(data)
self.handCtrl.xfo = data.get('handXfo')
self.numJointsAttr.setValue(data.get('numJoints'))
self.digitNamesAttr.setValue(data.get('digitNames'))
fingersGuideXfos = data.get('fingersGuideXfos')
fingerShapeCtrlData = data.get('fingerShapeCtrlData')
if fingersGuideXfos is not None:
for finger in self.fingers.keys():
for i in xrange(len(self.fingers[finger])):
self.fingers[finger][i].xfo = fingersGuideXfos[finger][i]
if hasattr(self.fingers[finger][i], 'shapeCtrl'):
if fingerShapeCtrlData is not None:
if finger in fingerShapeCtrlData:
self.fingers[finger][i].shapeCtrl.setCurveData(fingerShapeCtrlData[finger][i])
for op in self.getOperators():
guideOpName = ''.join([op.getName().split('FingerGuideOp')[0], self.getLocation(), 'FingerGuideOp'])
op.setName(guideOpName)
return True
def getRigBuildData(self):
"""Returns the Guide data used by the Rig Component to define the layout of the final rig..
Return:
The JSON rig data object.
"""
data = super(HandComponentGuide, self).getRigBuildData()
data['handXfo'] = self.handCtrl.xfo
fingerData = {}
for finger in self.fingers.keys():
fingerData[finger] = []
for i, joint in enumerate(self.fingers[finger]):
if i == len(self.fingers[finger]) - 1:
continue
# Calculate Xfo
boneVec = self.fingers[finger][i + 1].xfo.tr - self.fingers[finger][i].xfo.tr
bone1Normal = self.fingers[finger][i].xfo.ori.getZaxis().cross(boneVec).unit()
bone1ZAxis = boneVec.cross(bone1Normal).unit()
jointXfo = Xfo()
jointXfo.setFromVectors(boneVec.unit(), bone1Normal, bone1ZAxis, self.fingers[finger][i].xfo.tr)
jointData = {
'curveData': self.fingers[finger][i].shapeCtrl.getCurveData(),
'length': self.fingers[finger][i].xfo.tr.distanceTo(self.fingers[finger][i + 1].xfo.tr),
'xfo': jointXfo
}
fingerData[finger].append(jointData)
data['fingerData'] = fingerData
return data
# ==========
# Callbacks
# ==========
def addFinger(self, name):
digitSizeAttributes = []
fingerGuideCtrls = []
firstDigitCtrl = Control(name + "01", parent=self.handCtrl, shape='sphere')
firstDigitCtrl.scalePoints(Vec3(0.125, 0.125, 0.125))
firstDigitShapeCtrl = Control(name + "Shp01", parent=self.guideCtrlHrcGrp, shape='square')
firstDigitShapeCtrl.setColor('yellow')
firstDigitShapeCtrl.scalePoints(Vec3(0.175, 0.175, 0.175))
firstDigitShapeCtrl.translatePoints(Vec3(0.0, 0.125, 0.0))
fingerGuideCtrls.append(firstDigitShapeCtrl)
firstDigitCtrl.shapeCtrl = firstDigitShapeCtrl
firstDigitVisAttr = firstDigitShapeCtrl.getVisibilityAttr()
firstDigitVisAttr.connect(self.ctrlShapeToggle)
triangleCtrl = Control('tempCtrl', parent=None, shape='triangle')
triangleCtrl.rotatePoints(90.0, 0.0, 0.0)
triangleCtrl.scalePoints(Vec3(0.025, 0.025, 0.025))
triangleCtrl.translatePoints(Vec3(0.0, 0.0875, 0.0))
firstDigitCtrl.appendCurveData(triangleCtrl.getCurveData())
firstDigitCtrl.lockScale(True, True, True)
digitSettingsAttrGrp = AttributeGroup("DigitSettings", parent=firstDigitCtrl)
digitSizeAttr = ScalarAttribute('size', value=0.25, parent=digitSettingsAttrGrp)
digitSizeAttributes.append(digitSizeAttr)
# Set Finger
self.fingers[name] = []
self.fingers[name].append(firstDigitCtrl)
parent = firstDigitCtrl
numJoints = self.numJointsAttr.getValue()
if name == "thumb":
numJoints = 3
for i in xrange(2, numJoints + 2):
digitCtrl = Control(name + str(i).zfill(2), parent=parent, shape='sphere')
if i != numJoints + 1:
digitCtrl.scalePoints(Vec3(0.125, 0.125, 0.125))
digitCtrl.appendCurveData(triangleCtrl.getCurveData())
digitShapeCtrl = Control(name + 'Shp' + str(i).zfill(2), parent=self.guideCtrlHrcGrp, shape='circle')
digitShapeCtrl.setColor('yellow')
digitShapeCtrl.scalePoints(Vec3(0.175, 0.175, 0.175))
digitShapeCtrl.getVisibilityAttr().connect(self.ctrlShapeToggle)
digitCtrl.shapeCtrl = digitShapeCtrl
if i == 2:
digitShapeCtrl.translatePoints(Vec3(0.0, 0.125, 0.0))
else:
digitShapeCtrl.rotatePoints(0.0, 0.0, 90.0)
fingerGuideCtrls.append(digitShapeCtrl)
# Add size attr to all but last guide control
digitSettingsAttrGrp = AttributeGroup("DigitSettings", parent=digitCtrl)
digitSizeAttr = ScalarAttribute('size', value=0.25, parent=digitSettingsAttrGrp)
digitSizeAttributes.append(digitSizeAttr)
else:
digitCtrl.scalePoints(Vec3(0.0875, 0.0875, 0.0875))
digitCtrl.lockScale(True, True, True)
self.fingers[name].append(digitCtrl)
parent = digitCtrl
# ===========================
# Create Canvas Operators
# ===========================
# Add Finger Guide Canvas Op
fingerGuideCanvasOp = CanvasOperator(name + 'FingerGuideOp', 'Kraken.Solvers.Biped.BipedFingerGuideSolver')
self.addOperator(fingerGuideCanvasOp)
# Add Att Inputs
fingerGuideCanvasOp.setInput('drawDebug', self.drawDebugInputAttr)
fingerGuideCanvasOp.setInput('rigScale', self.rigScaleInputAttr)
# Add Xfo Inputs
fingerGuideCanvasOp.setInput('controls', self.fingers[name])
fingerGuideCanvasOp.setInput('planeSizes', digitSizeAttributes)
# Add Xfo Outputs
fingerGuideCanvasOp.setOutput('result', fingerGuideCtrls)
fingerGuideCanvasOp.setOutput('forceEval', firstDigitCtrl.getVisibilityAttr())
return firstDigitCtrl
def removeFinger(self, name):
self.handCtrl.removeChild(self.fingers[name][0])
del self.fingers[name]
def placeFingers(self):
spacing = 0.25
length = spacing * (len(self.fingers.keys()) - 1)
mid = length / 2.0
startOffset = length - mid
for i, finger in enumerate(self.fingers.keys()):
parentCtrl = self.handCtrl
numJoints = self.numJointsAttr.getValue()
if finger == "thumb":
numJoints = 3
for y in xrange(numJoints + 1):
if y == 1:
xOffset = 0.375
else:
xOffset = 0.25
if y == 0:
offsetVec = Vec3(xOffset, 0, startOffset - (i * spacing))
else:
offsetVec = Vec3(xOffset, 0, 0)
fingerPos = parentCtrl.xfo.transformVector(offsetVec)
fingerXfo = Xfo(tr=fingerPos, ori=self.handCtrl.xfo.ori)
self.fingers[finger][y].xfo = fingerXfo
parentCtrl = self.fingers[finger][y]
def updateFingers(self, fingers):
if " " in fingers:
self.digitNamesAttr.setValue(fingers.replace(" ", ""))
return
fingerNames = fingers.split(',')
# Delete fingers that don't exist any more
for finger in list(set(self.fingers.keys()) - set(fingerNames)):
self.removeFinger(finger)
# Add Fingers
for finger in fingerNames:
if finger in self.fingers.keys():
continue
self.addFinger(finger)
self.placeFingers()
def resizeDigits(self, numJoints):
initNumJoints = numJoints
for finger in self.fingers.keys():
if finger == "thumb":
numJoints = 3
else:
numJoints = initNumJoints
if numJoints + 1 == len(self.fingers[finger]):
continue
elif numJoints + 1 > len(self.fingers[finger]):
for i in xrange(len(self.fingers[finger]), numJoints + 1):
prevDigit = self.fingers[finger][i - 1]
digitCtrl = Control(finger + str(i + 1).zfill(2), parent=prevDigit, shape='sphere')
digitCtrl.setColor('orange')
digitCtrl.scalePoints(Vec3(0.25, 0.25, 0.25))
digitCtrl.lockScale(True, True, True)
self.fingers[finger].append(digitCtrl)
elif numJoints + 1 < len(self.fingers[finger]):
numExtraCtrls = len(self.fingers[finger]) - (numJoints + 1)
for i in xrange(numExtraCtrls):
removedJoint = self.fingers[finger].pop()
removedJoint.getParent().removeChild(removedJoint)
self.placeFingers()
# ==============
# Class Methods
# ==============
@classmethod
def getComponentType(cls):
"""Enables introspection of the class prior to construction to determine if it is a guide component.
Return:
The true if this component is a guide component.
"""
return 'Guide'
@classmethod
def getRigComponentClass(cls):
"""Returns the corresponding rig component class for this guide component class
Return:
The rig component class.
"""
return HandComponentRig
class InsectLegComponentRig(InsectLegComponent):
"""Insect Leg Rig"""
def __init__(self, name='InsectLeg', parent=None):
Profiler.getInstance().push("Construct InsectLeg Rig Component:" +
name)
super(InsectLegComponentRig, self).__init__(name, parent)
# =========
# Controls
# =========
# Chain Base
self.chainBase = Locator('ChainBase', parent=self.ctrlCmpGrp)
self.chainBase.setShapeVisibility(False)
# FK
self.fkCtrlSpaces = []
self.fkCtrls = []
self.setNumControls(4)
# IK Control
self.legIKCtrlSpace = CtrlSpace('IK', parent=self.ctrlCmpGrp)
self.legIKCtrl = Control('IK', parent=self.legIKCtrlSpace, shape="pin")
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))
# Add Component Params to IK control
legSettingsAttrGrp = AttributeGroup("DisplayInfo_LegSettings",
parent=self.legIKCtrl)
legdrawDebugInputAttr = BoolAttribute('drawDebug',
value=False,
parent=legSettingsAttrGrp)
legUseInitPoseInputAttr = BoolAttribute('useInitPose',
value=True,
parent=legSettingsAttrGrp)
self.rootIndexInputAttr = IntegerAttribute('rootIndex',
value=0,
parent=legSettingsAttrGrp)
legFkikInputAttr = ScalarAttribute('fkik',
value=1.0,
minValue=0.0,
maxValue=1.0,
parent=legSettingsAttrGrp)
# Connect IO to controls
self.drawDebugInputAttr.connect(legdrawDebugInputAttr)
# UpV
self.legUpVCtrlSpace = CtrlSpace('UpV', parent=self.ctrlCmpGrp)
self.legUpVCtrl = Control('UpV',
parent=self.legUpVCtrlSpace,
shape="triangle")
self.legUpVCtrl.alignOnZAxis()
self.legUpVCtrl.rotatePoints(0, 90, 0)
# ==========
# Deformers
# ==========
deformersLayer = self.getOrCreateLayer('deformers')
self.defCmpGrp = ComponentGroup(self.getName(),
self,
parent=deformersLayer)
self.addItem('defCmpGrp', self.defCmpGrp)
self.deformerJoints = []
self.boneOutputsTgt = []
self.setNumDeformers(4)
# =====================
# Create Component I/O
# =====================
# Set IO Targets
self.boneOutputs.setTarget(self.boneOutputsTgt)
# ==============
# Constrain I/O
# ==============
# Constraint inputs
legRootInputConstraint = PoseConstraint('_'.join([
self.fkCtrlSpaces[0].getName(), 'To',
self.rootInputTgt.getName()
]))
legRootInputConstraint.setMaintainOffset(True)
legRootInputConstraint.addConstrainer(self.rootInputTgt)
self.fkCtrlSpaces[0].addConstraint(legRootInputConstraint)
chainBaseInputConstraint = PoseConstraint('_'.join(
[self.chainBase.getName(), 'To',
self.rootInputTgt.getName()]))
chainBaseInputConstraint.setMaintainOffset(True)
chainBaseInputConstraint.addConstrainer(self.rootInputTgt)
self.chainBase.addConstraint(chainBaseInputConstraint)
# ===============
# Add Canvas Ops
# ===============
# Add Canvas Op
self.nBoneSolverKLOp = KLOperator('leg', 'NBoneIKSolver', 'Kraken')
self.addOperator(self.nBoneSolverKLOp)
# # Add Att Inputs
self.nBoneSolverKLOp.setInput('drawDebug', self.drawDebugInputAttr)
self.nBoneSolverKLOp.setInput('rigScale', self.rigScaleInputAttr)
self.nBoneSolverKLOp.setInput('useInitPose', legUseInitPoseInputAttr)
self.nBoneSolverKLOp.setInput('ikblend', legFkikInputAttr)
self.nBoneSolverKLOp.setInput('rootIndex', self.rootIndexInputAttr)
self.nBoneSolverKLOp.setInput('tipBoneLen', self.tipBoneLenInputAttr)
# Add Xfo Inputs
self.nBoneSolverKLOp.setInput('chainBase', self.chainBase)
self.nBoneSolverKLOp.setInput('ikgoal', self.legIKCtrl)
self.nBoneSolverKLOp.setInput('upVector', self.legUpVCtrl)
self.nBoneSolverKLOp.setInput('fkcontrols', self.fkCtrls)
# Add Xfo Outputs
self.nBoneSolverKLOp.setOutput('pose', self.boneOutputsTgt)
self.nBoneSolverKLOp.setOutput('legEnd', self.legEndPosOutputTgt)
# Add Deformer Canvas 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.boneOutputsTgt)
# Add Xfo Outputs
self.outputsToDeformersKLOp.setOutput('constrainees',
self.deformerJoints)
Profiler.getInstance().pop()
def setNumControls(self, numControls):
# Add new control spaces and controls
for i in xrange(len(self.fkCtrlSpaces), numControls):
if i == 0:
parent = self.ctrlCmpGrp
else:
parent = self.fkCtrls[i - 1]
boneName = 'bone' + str(i + 1).zfill(2) + 'FK'
fkCtrlSpace = CtrlSpace(boneName, parent=parent)
fkCtrl = Control(boneName, parent=fkCtrlSpace, shape="cube")
fkCtrl.alignOnXAxis()
fkCtrl.lockScale(x=True, y=True, z=True)
fkCtrl.lockTranslation(x=True, y=True, z=True)
self.fkCtrlSpaces.append(fkCtrlSpace)
self.fkCtrls.append(fkCtrl)
def setNumDeformers(self, numDeformers):
# Add new deformers and outputs
for i in xrange(len(self.boneOutputsTgt), numDeformers):
name = 'bone' + str(i + 1).zfill(2)
legOutput = ComponentOutput(name, parent=self.outputHrcGrp)
self.boneOutputsTgt.append(legOutput)
for i in xrange(len(self.deformerJoints), numDeformers):
name = 'bone' + str(i + 1).zfill(2)
boneDef = Joint(name, parent=self.defCmpGrp)
boneDef.setComponent(self)
self.deformerJoints.append(boneDef)
return True
def calculateUpVXfo(self, boneXfos, endXfo):
"""Calculates the transform for the UpV control.
Args:
boneXfos (list): Bone transforms.
endXfo (Xfo): Transform for the end of the chain.
Returns:
Xfo: Up Vector transform.
"""
# Calculate FW
toFirst = boneXfos[1].tr.subtract(boneXfos[0].tr).unit()
toTip = endXfo.tr.subtract(boneXfos[0].tr).unit()
fw = toTip.cross(toFirst).unit()
chainNormal = fw.cross(toTip).unit()
chainZAxis = toTip.cross(chainNormal).unit()
chainXfo = Xfo()
chainXfo.setFromVectors(toTip.unit(), chainNormal, chainZAxis,
boneXfos[0].tr)
rootToTip = endXfo.tr.subtract(boneXfos[0].tr).length()
upVXfo = Xfo()
upVXfo.tr = chainXfo.transformVector(
Vec3(rootToTip / 2.0, rootToTip / 2.0, 0.0))
return upVXfo
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(InsectLegComponentRig, self).loadData(data)
boneXfos = data['boneXfos']
boneLengths = data['boneLengths']
numJoints = data['numJoints']
endXfo = data['endXfo']
# Add extra controls and outputs
self.setNumControls(numJoints)
self.setNumDeformers(numJoints)
# Scale controls based on bone lengths
for i, each in enumerate(self.fkCtrlSpaces):
self.fkCtrlSpaces[i].xfo = boneXfos[i]
self.fkCtrls[i].xfo = boneXfos[i]
self.fkCtrls[i].scalePoints(Vec3(boneLengths[i], 1.75, 1.75))
self.chainBase.xfo = boneXfos[0]
self.legIKCtrlSpace.xfo = endXfo
self.legIKCtrl.xfo = endXfo
upVXfo = self.calculateUpVXfo(boneXfos, endXfo)
self.legUpVCtrlSpace.xfo = upVXfo
self.legUpVCtrl.xfo = upVXfo
# Set max on the rootIndex attribute
self.rootIndexInputAttr.setMax(len(boneXfos))
# ============
# Set IO Xfos
# ============
self.rootInputTgt.xfo = boneXfos[0]
for i in xrange(len(boneLengths)):
self.boneOutputsTgt[i].xfo = boneXfos[i]
self.legEndXfoOutputTgt.xfo = endXfo
self.legEndPosOutputTgt.xfo = endXfo
# =============
# Set IO Attrs
# =============
tipBoneLen = boneLengths[len(boneLengths) - 1]
self.tipBoneLenInputAttr.setMax(tipBoneLen * 2.0)
self.tipBoneLenInputAttr.setValue(tipBoneLen)
# ====================
# Evaluate Splice Ops
# ====================
# evaluate the nbone op so that all the output transforms are updated.
self.nBoneSolverKLOp.evaluate()
self.outputsToDeformersKLOp.evaluate()
class InsectLegComponentRig(InsectLegComponent):
"""Insect Leg Rig"""
def __init__(self, name='InsectLeg', parent=None):
Profiler.getInstance().push("Construct InsectLeg Rig Component:" + name)
super(InsectLegComponentRig, self).__init__(name, parent)
# =========
# Controls
# =========
# Chain Base
self.chainBase = Locator('ChainBase', parent=self.ctrlCmpGrp)
self.chainBase.setShapeVisibility(False)
# FK
self.fkCtrlSpaces = []
self.fkCtrls = []
self.setNumControls(4)
# IK Control
self.legIKCtrlSpace = CtrlSpace('IK', parent=self.ctrlCmpGrp)
self.legIKCtrl = Control('IK', parent=self.legIKCtrlSpace, shape="pin")
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))
# Add Component Params to IK control
legSettingsAttrGrp = AttributeGroup("DisplayInfo_LegSettings", parent=self.legIKCtrl)
legdrawDebugInputAttr = BoolAttribute('drawDebug', value=False, parent=legSettingsAttrGrp)
legUseInitPoseInputAttr = BoolAttribute('useInitPose', value=True, parent=legSettingsAttrGrp)
self.rootIndexInputAttr = IntegerAttribute('rootIndex', value=0, parent=legSettingsAttrGrp)
legFkikInputAttr = ScalarAttribute('fkik', value=1.0, minValue=0.0, maxValue=1.0, parent=legSettingsAttrGrp)
# Connect IO to controls
self.drawDebugInputAttr.connect(legdrawDebugInputAttr)
# UpV
self.legUpVCtrlSpace = CtrlSpace('UpV', parent=self.ctrlCmpGrp)
self.legUpVCtrl = Control('UpV', parent=self.legUpVCtrlSpace, shape="triangle")
self.legUpVCtrl.alignOnZAxis()
self.legUpVCtrl.rotatePoints(0, 90, 0)
# ==========
# Deformers
# ==========
deformersLayer = self.getOrCreateLayer('deformers')
self.defCmpGrp = ComponentGroup(self.getName(), self, parent=deformersLayer)
self.deformerJoints = []
self.boneOutputsTgt = []
self.setNumDeformers(4)
# =====================
# Create Component I/O
# =====================
# Set IO Targets
self.boneOutputs.setTarget(self.boneOutputsTgt)
# ==============
# Constrain I/O
# ==============
# Constraint inputs
legRootInputConstraint = PoseConstraint('_'.join([self.fkCtrlSpaces[0].getName(), 'To', self.rootInputTgt.getName()]))
legRootInputConstraint.setMaintainOffset(True)
legRootInputConstraint.addConstrainer(self.rootInputTgt)
self.fkCtrlSpaces[0].addConstraint(legRootInputConstraint)
chainBaseInputConstraint = PoseConstraint('_'.join([self.chainBase.getName(), 'To', self.rootInputTgt.getName()]))
chainBaseInputConstraint.setMaintainOffset(True)
chainBaseInputConstraint.addConstrainer(self.rootInputTgt)
self.chainBase.addConstraint(chainBaseInputConstraint)
# ===============
# Add Splice Ops
# ===============
# Add Splice Op
self.nBoneSolverSpliceOp = SpliceOperator('legSpliceOp', 'NBoneIKSolver', 'Kraken')
self.addOperator(self.nBoneSolverSpliceOp)
# # Add Att Inputs
self.nBoneSolverSpliceOp.setInput('drawDebug', self.drawDebugInputAttr)
self.nBoneSolverSpliceOp.setInput('rigScale', self.rigScaleInputAttr)
self.nBoneSolverSpliceOp.setInput('useInitPose', legUseInitPoseInputAttr)
self.nBoneSolverSpliceOp.setInput('ikblend', legFkikInputAttr)
self.nBoneSolverSpliceOp.setInput('rootIndex', self.rootIndexInputAttr)
self.nBoneSolverSpliceOp.setInput('tipBoneLen', self.tipBoneLenInputAttr)
# Add Xfo Inputs
self.nBoneSolverSpliceOp.setInput('chainBase', self.chainBase)
self.nBoneSolverSpliceOp.setInput('ikgoal', self.legIKCtrl)
self.nBoneSolverSpliceOp.setInput('upVector', self.legUpVCtrl)
self.nBoneSolverSpliceOp.setInput('fkcontrols', self.fkCtrls)
# Add Xfo Outputs
self.nBoneSolverSpliceOp.setOutput('pose', self.boneOutputsTgt)
self.nBoneSolverSpliceOp.setOutput('legEnd', self.legEndPosOutputTgt)
# Add Deformer Splice Op
self.outputsToDeformersSpliceOp = SpliceOperator('insectLegDeformerSpliceOp', 'MultiPoseConstraintSolver', 'Kraken')
self.addOperator(self.outputsToDeformersSpliceOp)
# Add Att Inputs
self.outputsToDeformersSpliceOp.setInput('drawDebug', self.drawDebugInputAttr)
self.outputsToDeformersSpliceOp.setInput('rigScale', self.rigScaleInputAttr)
# Add Xfo Inputs
self.outputsToDeformersSpliceOp.setInput('constrainers', self.boneOutputsTgt)
# Add Xfo Outputs
self.outputsToDeformersSpliceOp.setOutput('constrainees', self.deformerJoints)
Profiler.getInstance().pop()
def setNumControls(self, numControls):
# Add new control spaces and controls
for i in xrange(len(self.fkCtrlSpaces), numControls):
if i==0:
parent = self.ctrlCmpGrp
else:
parent = self.fkCtrls[i - 1]
boneName = 'bone' + str(i + 1).zfill(2) + 'FK'
fkCtrlSpace = CtrlSpace(boneName, parent=parent)
fkCtrl = Control(boneName, parent=fkCtrlSpace, shape="cube")
fkCtrl.alignOnXAxis()
fkCtrl.lockScale(x=True, y=True, z=True)
fkCtrl.lockTranslation(x=True, y=True, z=True)
self.fkCtrlSpaces.append(fkCtrlSpace)
self.fkCtrls.append(fkCtrl)
def setNumDeformers(self, numDeformers):
# Add new deformers and outputs
for i in xrange(len(self.boneOutputsTgt), numDeformers):
name = 'bone' + str(i + 1).zfill(2)
legOutput = ComponentOutput(name, parent=self.outputHrcGrp)
self.boneOutputsTgt.append(legOutput)
for i in xrange(len(self.deformerJoints), numDeformers):
name = 'bone' + str(i + 1).zfill(2)
boneDef = Joint(name, parent=self.defCmpGrp)
boneDef.setComponent(self)
self.deformerJoints.append(boneDef)
return True
def calculateUpVXfo(self, boneXfos, endXfo):
"""Calculates the transform for the UpV control.
Args:
boneXfos (list): Bone transforms.
endXfo (Xfo): Transform for the end of the chain.
Returns:
Xfo: Up Vector transform.
"""
# Calculate FW
toFirst = boneXfos[1].tr.subtract(boneXfos[0].tr).unit()
toTip = endXfo.tr.subtract(boneXfos[0].tr).unit()
fw = toTip.cross(toFirst).unit()
chainNormal = fw.cross(toTip).unit()
chainZAxis = toTip.cross(chainNormal).unit()
chainXfo = Xfo()
chainXfo.setFromVectors(toTip.unit(), chainNormal, chainZAxis, boneXfos[0].tr)
rootToTip = endXfo.tr.subtract(boneXfos[0].tr).length()
upVXfo = Xfo()
upVXfo.tr = chainXfo.transformVector(Vec3(rootToTip / 2.0, rootToTip / 2.0, 0.0))
return upVXfo
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(InsectLegComponentRig, self).loadData( data )
boneXfos = data['boneXfos']
boneLengths = data['boneLengths']
numJoints = data['numJoints']
endXfo = data['endXfo']
# Add extra controls and outputs
self.setNumControls(numJoints)
self.setNumDeformers(numJoints)
# Scale controls based on bone lengths
for i, each in enumerate(self.fkCtrlSpaces):
self.fkCtrlSpaces[i].xfo = boneXfos[i]
self.fkCtrls[i].xfo = boneXfos[i]
self.fkCtrls[i].scalePoints(Vec3(boneLengths[i], 1.75, 1.75))
self.chainBase.xfo = boneXfos[0]
self.legIKCtrlSpace.xfo = endXfo
self.legIKCtrl.xfo = endXfo
upVXfo = self.calculateUpVXfo(boneXfos, endXfo)
self.legUpVCtrlSpace.xfo = upVXfo
self.legUpVCtrl.xfo = upVXfo
# Set max on the rootIndex attribute
self.rootIndexInputAttr.setMax(len(boneXfos))
# ============
# Set IO Xfos
# ============
self.rootInputTgt.xfo = boneXfos[0]
for i in xrange(len(boneLengths)):
self.boneOutputsTgt[i].xfo = boneXfos[i]
self.legEndXfoOutputTgt.xfo = endXfo
self.legEndPosOutputTgt.xfo = endXfo
# =============
# Set IO Attrs
# =============
tipBoneLen = boneLengths[len(boneLengths) - 1]
self.tipBoneLenInputAttr.setMax(tipBoneLen * 2.0)
self.tipBoneLenInputAttr.setValue(tipBoneLen)
# ====================
# Evaluate Splice Ops
# ====================
# evaluate the nbone op so that all the output transforms are updated.
self.nBoneSolverSpliceOp.evaluate()
self.outputsToDeformersSpliceOp.evaluate()
class SimpleControlComponentGuide(SimpleControlComponent):
"""Simple Control Component Guide"""
def __init__(self, name='SimpleControl', parent=None):
Profiler.getInstance().push(
"Construct Simple Control Guide Component:" + name)
super(SimpleControlComponentGuide, self).__init__(name, parent)
# =========
# Attributes
# =========
# Add Component Params to IK control
guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self)
self.ctrlSizeInputAttr = ScalarAttribute('ctrlSize',
value=5.0,
minValue=1.0,
maxValue=50.0,
parent=guideSettingsAttrGrp)
self.ctrlSizeInputAttr.setValueChangeCallback(self.resizeMainCtrl)
# =========
# Controls
# =========
# Guide Controls
self.mainCtrl = Control('main', parent=self.ctrlCmpGrp, shape='square')
self.mainCtrl.rotatePoints(90, 0, 0)
data = {
"location": 'M',
"ctrlSize": self.ctrlSizeInputAttr.getValue(),
"ctrlXfo": Xfo(tr=Vec3(0.0, 0.0, 0.0))
}
self.loadData(data)
Profiler.getInstance().pop()
# =============
# Data Methods
# =============
def saveData(self):
"""Save the data for the component to be persisted.
Return:
The JSON data object
"""
data = super(SimpleControlComponentGuide, self).saveData()
data["ctrlSize"] = self.ctrlSizeInputAttr.getValue()
data["ctrlXfo"] = self.mainCtrl.xfo
return data
def loadData(self, data):
"""Load a saved guide representation from persisted data.
Arguments:
data -- object, The JSON data object.
Return:
True if successful.
"""
super(SimpleControlComponentGuide, self).loadData(data)
self.ctrlSizeInputAttr.setValue(data["ctrlSize"])
self.mainCtrl.xfo = data["ctrlXfo"]
scaleValue = data["ctrlSize"]
self.mainCtrl.setShape('square')
self.mainCtrl.rotatePoints(90, 0, 0)
self.mainCtrl.scalePoints(Vec3(scaleValue, scaleValue, scaleValue))
return True
def getRigBuildData(self):
"""Returns the Guide data used by the Rig Component to define the layout of the final rig.
Return:
The JSON rig data object.
"""
data = super(SimpleControlComponentGuide, self).getRigBuildData()
data["ctrlSize"] = self.ctrlSizeInputAttr.getValue()
data["ctrlXfo"] = self.mainCtrl.xfo
return data
# ==========
# Callbacks
# ==========
def resizeMainCtrl(self, newSize):
self.mainCtrl.setShape('square')
self.mainCtrl.rotatePoints(90, 0, 0)
self.mainCtrl.scalePoints(Vec3(newSize, newSize, newSize))
# ==============
# Class Methods
# ==============
@classmethod
def getComponentType(cls):
"""Enables introspection of the class prior to construction to determine if it is a guide component.
Return:
The true if this component is a guide component.
"""
return 'Guide'
@classmethod
def getRigComponentClass(cls):
"""Returns the corresponding rig component class for this guide component class
Return:
The rig component class.
"""
return SimpleControlComponentRig
class FabriceHeadGuide(FabriceHead):
"""Fabrice Head Component Guide"""
def __init__(self, name='head', parent=None):
Profiler.getInstance().push("Construct Head Guide Component:" + name)
super(FabriceHeadGuide, self).__init__(name, parent)
# =========
# Controls
# =========
guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self)
self.headCtrl = Control('head', parent=self.ctrlCmpGrp, shape="circle")
self.headCtrl.rotatePoints(90.0, 0.0, 0.0)
self.headCtrl.scalePoints(Vec3(3.5, 3.5, 3.5))
self.jawCtrl = Control('jaw', parent=self.ctrlCmpGrp, shape="cube")
self.jawCtrl.alignOnZAxis()
self.jawCtrl.scalePoints(Vec3(2.0, 0.5, 2.0))
self.jawCtrl.alignOnYAxis(negative=True)
self.jawCtrl.setColor('orange')
data = {
"name": name,
"location": "M",
"headXfo": Xfo(Vec3(0.0, 1.67, 1.75)),
"headCtrlCrvData": self.headCtrl.getCurveData(),
"jawPosition": Vec3(0.0, 1.2787, 2.0078),
"jawCtrlCrvData": self.jawCtrl.getCurveData(),
}
self.loadData(data)
Profiler.getInstance().pop()
# =============
# Data Methods
# =============
def saveData(self):
"""Save the data for the component to be persisted.
Return:
The JSON data object
"""
data = super(FabriceHeadGuide, self).saveData()
data['headXfo'] = self.headCtrl.xfo
data['headCtrlCrvData'] = self.headCtrl.getCurveData()
data['jawPosition'] = self.jawCtrl.xfo.tr
data['jawCtrlCrvData'] = self.jawCtrl.getCurveData()
return data
def loadData(self, data):
"""Load a saved guide representation from persisted data.
Arguments:
data -- object, The JSON data object.
Return:
True if successful.
"""
super(FabriceHeadGuide, self).loadData( data )
self.headCtrl.xfo = data['headXfo']
self.headCtrl.setCurveData(data['headCtrlCrvData'])
self.jawCtrl.xfo.tr = data['jawPosition']
self.jawCtrl.setCurveData(data['jawCtrlCrvData'])
return True
def getRigBuildData(self):
"""Returns the Guide data used by the Rig Component to define the layout of the final rig..
Return:
The JSON rig data object.
"""
data = super(FabriceHeadGuide, self).getRigBuildData()
data['headXfo'] = self.headCtrl.xfo
data['headCtrlCrvData'] = self.headCtrl.getCurveData()
data['jawPosition'] = self.jawCtrl.xfo.tr
data['jawCtrlCrvData'] = self.jawCtrl.getCurveData()
return data
# ==============
# Class Methods
# ==============
@classmethod
def getComponentType(cls):
"""Enables introspection of the class prior to construction to determine if it is a guide component.
Return:
The true if this component is a guide component.
"""
return 'Guide'
@classmethod
def getRigComponentClass(cls):
"""Returns the corresponding rig component class for this guide component class
Return:
The rig component class.
"""
return FabriceHeadRig
class HeadComponentRig(HeadComponent):
"""Head Component Rig"""
def __init__(self, name='head', parent=None):
Profiler.getInstance().push("Construct Head Rig Component:" + name)
super(HeadComponentRig, self).__init__(name, parent)
# =========
# Controls
# =========
# Head
self.headCtrlSpace = CtrlSpace('head', parent=self.ctrlCmpGrp)
self.headCtrl = Control('head',
parent=self.headCtrlSpace,
shape="circle")
self.headCtrl.rotatePoints(0, 0, 90)
self.headCtrl.scalePoints(Vec3(3, 3, 3))
self.headCtrl.translatePoints(Vec3(0, 1, 0.25))
# Eye Left
self.eyeLeftCtrlSpace = CtrlSpace('eyeLeft', parent=self.headCtrl)
self.eyeLeftCtrl = Control('eyeLeft',
parent=self.eyeLeftCtrlSpace,
shape="sphere")
self.eyeLeftCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.eyeLeftCtrl.setColor("blueMedium")
# Eye Right
self.eyeRightCtrlSpace = CtrlSpace('eyeRight', parent=self.headCtrl)
self.eyeRightCtrl = Control('eyeRight',
parent=self.eyeRightCtrlSpace,
shape="sphere")
self.eyeRightCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.eyeRightCtrl.setColor("blueMedium")
# Jaw
self.jawCtrlSpace = CtrlSpace('jawCtrlSpace', parent=self.headCtrl)
self.jawCtrl = Control('jaw', parent=self.jawCtrlSpace, shape="cube")
self.jawCtrl.alignOnYAxis(negative=True)
self.jawCtrl.alignOnZAxis()
self.jawCtrl.scalePoints(Vec3(1.45, 0.65, 1.25))
self.jawCtrl.translatePoints(Vec3(0, -0.25, 0))
self.jawCtrl.setColor("orange")
# ==========
# Deformers
# ==========
deformersLayer = self.getOrCreateLayer('deformers')
defCmpGrp = ComponentGroup(self.getName(), self, parent=deformersLayer)
headDef = Joint('head', parent=defCmpGrp)
headDef.setComponent(self)
jawDef = Joint('jaw', parent=defCmpGrp)
jawDef.setComponent(self)
eyeLeftDef = Joint('eyeLeft', parent=defCmpGrp)
eyeLeftDef.setComponent(self)
eyeRightDef = Joint('eyeRight', parent=defCmpGrp)
eyeRightDef.setComponent(self)
# ==============
# Constrain I/O
# ==============
# Constraint inputs
headInputConstraint = PoseConstraint('_'.join([
self.headCtrlSpace.getName(), 'To',
self.headBaseInputTgt.getName()
]))
headInputConstraint.setMaintainOffset(True)
headInputConstraint.addConstrainer(self.headBaseInputTgt)
self.headCtrlSpace.addConstraint(headInputConstraint)
# # Constraint outputs
headOutputConstraint = PoseConstraint('_'.join(
[self.headOutputTgt.getName(), 'To',
self.headCtrl.getName()]))
headOutputConstraint.addConstrainer(self.headCtrl)
self.headOutputTgt.addConstraint(headOutputConstraint)
jawOutputConstraint = PoseConstraint('_'.join(
[self.jawOutputTgt.getName(), 'To',
self.jawCtrl.getName()]))
jawOutputConstraint.addConstrainer(self.jawCtrl)
self.jawOutputTgt.addConstraint(jawOutputConstraint)
eyeLOutputConstraint = PoseConstraint('_'.join(
[self.eyeLOutputTgt.getName(), 'To',
self.eyeLeftCtrl.getName()]))
eyeLOutputConstraint.addConstrainer(self.eyeLeftCtrl)
self.eyeLOutputTgt.addConstraint(eyeLOutputConstraint)
eyeROutputConstraint = PoseConstraint('_'.join(
[self.eyeROutputTgt.getName(), 'To',
self.eyeRightCtrl.getName()]))
eyeROutputConstraint.addConstrainer(self.eyeRightCtrl)
self.eyeROutputTgt.addConstraint(eyeROutputConstraint)
# ==================
# Add Component I/O
# ==================
# Add Xfo I/O's
# self.addInput(self.headBaseInputTgt)
# self.addOutput(self.headOutputTgt)
# self.addOutput(self.jawOutputTgt)
# self.addOutput(self.eyeLOutputTgt)
# self.addOutput(self.eyeROutputTgt)
# Add Attribute I/O's
# self.addInput(self.drawDebugInputAttr)
# ===============
# Add Splice Ops
# ===============
# Add Deformer Splice Op
# spliceOp = SpliceOperator('headDeformerSpliceOp', 'HeadConstraintSolver', 'KrakenHeadConstraintSolver')
# self.addOperator(spliceOp)
# # Add Att Inputs
# spliceOp.setInput('drawDebug', self.drawDebugInputAttr)
# spliceOp.setInput('rigScale', self.rigScaleInputAttr)
# # Add Xfo Inputstrl)
# spliceOp.setInput('headConstrainer', self.headOutputTgt)
# spliceOp.setInput('jawConstrainer', self.jawOutputTgt)
# spliceOp.setInput('eyeLeftConstrainer', self.eyeLOutputTgt)
# spliceOp.setInput('eyeRightConstrainer', self.eyeROutputTgt)
# # Add Xfo Outputs
# spliceOp.setOutput('headDeformer', headDef)
# spliceOp.setOutput('jawDeformer', jawDef)
# spliceOp.setOutput('eyeLeftDeformer', eyeLeftDef)
# spliceOp.setOutput('eyeRightDeformer', eyeRightDef)
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(HeadComponentRig, self).loadData(data)
self.headCtrlSpace.xfo.tr = data['headPosition']
self.headCtrl.xfo.tr = data['headPosition']
self.eyeLeftCtrlSpace.xfo.tr = data['eyeLeftPosition']
self.eyeLeftCtrl.xfo.tr = data['eyeLeftPosition']
self.eyeRightCtrlSpace.xfo.tr = data['eyeRightPosition']
self.eyeRightCtrl.xfo.tr = data['eyeRightPosition']
self.jawCtrlSpace.xfo.tr = data['jawPosition']
self.jawCtrl.xfo.tr = data['jawPosition']
# ============
# Set IO Xfos
# ============
self.headBaseInputTgt.xfo.tr = data['headPosition']
self.headOutputTgt.xfo.tr = data['headPosition']
self.jawOutputTgt.xfo.tr = data['jawPosition']
self.eyeLOutputTgt.xfo.tr = data['eyeLeftPosition']
self.eyeROutputTgt.xfo.tr = data['eyeRightPosition']
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()
# Forearm
self.forearmFKCtrlSpace = CtrlSpace('forearmFK',
parent=self.bicepFKCtrl)
self.forearmFKCtrl = Control('forearmFK',
parent=self.forearmFKCtrlSpace,
shape="cube")
self.forearmFKCtrl.alignOnXAxis()
self.handCtrlSpace = CtrlSpace('hand', parent=self.ctrlCmpGrp)
self.handCtrl = Control('hand',
parent=self.handCtrlSpace,
shape="circle")
self.handCtrl.rotatePoints(0, 0, 90)
self.handCtrl.scalePoints(Vec3(1.0, 0.75, 0.75))
# Arm IK
self.armIKCtrlSpace = CtrlSpace('IK', parent=self.ctrlCmpGrp)
self.armIKCtrl = Control('IK', parent=self.armIKCtrlSpace, shape="pin")
# Add Params to IK control
armSettingsAttrGrp = AttributeGroup("DisplayInfo_ArmSettings",
parent=self.armIKCtrl)
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)
armIKBlendInputAttr = ScalarAttribute('fkik',
value=0.0,
minValue=0.0,
maxValue=1.0,
parent=armSettingsAttrGrp)
armSoftIKInputAttr = BoolAttribute('softIK',
value=True,
parent=armSettingsAttrGrp)
armSoftDistInputAttr = ScalarAttribute('softDist',
value=0.0,
minValue=0.0,
parent=armSettingsAttrGrp)
armStretchInputAttr = BoolAttribute('stretch',
value=True,
parent=armSettingsAttrGrp)
armStretchBlendInputAttr = ScalarAttribute('stretchBlend',
value=0.0,
minValue=0.0,
maxValue=1.0,
parent=armSettingsAttrGrp)
# Hand Params
handSettingsAttrGrp = AttributeGroup("DisplayInfo_HandSettings",
parent=self.handCtrl)
handLinkToWorldInputAttr = ScalarAttribute('linkToWorld',
0.0,
maxValue=1.0,
parent=handSettingsAttrGrp)
self.drawDebugInputAttr.connect(armDebugInputAttr)
# 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)
# ==========
# Deformers
# ==========
deformersLayer = self.getOrCreateLayer('deformers')
defCmpGrp = ComponentGroup(self.getName(), self, parent=deformersLayer)
bicepDef = Joint('bicep', parent=defCmpGrp)
bicepDef.setComponent(self)
forearmDef = Joint('forearm', parent=defCmpGrp)
forearmDef.setComponent(self)
wristDef = Joint('wrist', parent=defCmpGrp)
wristDef.setComponent(self)
handDef = Joint('hand', parent=defCmpGrp)
handDef.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.clavicleEndInputTgt.getName()
]))
self.armRootInputConstraint.setMaintainOffset(True)
self.armRootInputConstraint.addConstrainer(self.clavicleEndInputTgt)
self.bicepFKCtrlSpace.addConstraint(self.armRootInputConstraint)
# Constraint outputs
self.handConstraint = PoseConstraint('_'.join(
[self.handOutputTgt.getName(), 'To',
self.handCtrl.getName()]))
self.handConstraint.addConstrainer(self.handCtrl)
self.handOutputTgt.addConstraint(self.handConstraint)
self.handCtrlSpaceConstraint = PoseConstraint('_'.join([
self.handCtrlSpace.getName(), 'To',
self.armEndXfoOutputTgt.getName()
]))
self.handCtrlSpaceConstraint.setMaintainOffset(True)
self.handCtrlSpaceConstraint.addConstrainer(self.armEndXfoOutputTgt)
self.handCtrlSpace.addConstraint(self.handCtrlSpaceConstraint)
# ===============
# Add Splice Ops
# ===============
# Add Splice Op
self.spliceOp = KLOperator('armKLOp', 'TwoBoneIKSolver', 'Kraken')
self.addOperator(self.spliceOp)
# Add Att Inputs
self.spliceOp.setInput('drawDebug', self.drawDebugInputAttr)
self.spliceOp.setInput('rigScale', self.rigScaleInputAttr)
self.spliceOp.setInput('bone0Len', self.armBone0LenInputAttr)
self.spliceOp.setInput('bone1Len', self.armBone1LenInputAttr)
self.spliceOp.setInput('ikblend', armIKBlendInputAttr)
self.spliceOp.setInput('softIK', armSoftIKInputAttr)
self.spliceOp.setInput('softDist', armSoftDistInputAttr)
self.spliceOp.setInput('stretch', armStretchInputAttr)
self.spliceOp.setInput('stretchBlend', armStretchBlendInputAttr)
self.spliceOp.setInput('rightSide', self.rightSideInputAttr)
# Add Xfo Inputs
self.spliceOp.setInput('root', self.clavicleEndInputTgt)
self.spliceOp.setInput('bone0FK', self.bicepFKCtrl)
self.spliceOp.setInput('bone1FK', self.forearmFKCtrl)
self.spliceOp.setInput('ikHandle', self.armIKCtrl)
self.spliceOp.setInput('upV', self.armUpVCtrl)
# Add Xfo Outputs
self.spliceOp.setOutput('bone0Out', self.bicepOutputTgt)
self.spliceOp.setOutput('bone1Out', self.forearmOutputTgt)
self.spliceOp.setOutput('bone2Out', self.armEndXfoOutputTgt)
# Add Deformer Splice Op
self.outputsToDeformersKLOp = KLOperator('armDeformerKLOp',
'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.forearmOutputTgt,
self.armEndXfoOutputTgt, self.handOutputTgt
])
# Add Xfo Outputs
self.outputsToDeformersKLOp.setOutput(
'constrainees', [bicepDef, forearmDef, wristDef, handDef])
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)
self.clavicleEndInputTgt.xfo.tr = data['bicepXfo'].tr
self.bicepFKCtrlSpace.xfo = data['bicepXfo']
self.bicepFKCtrl.xfo = data['bicepXfo']
self.bicepFKCtrl.scalePoints(
Vec3(data['bicepLen'], data['bicepFKCtrlSize'],
data['bicepFKCtrlSize']))
self.bicepOutputTgt.xfo = data['bicepXfo']
self.forearmOutputTgt.xfo = data['forearmXfo']
self.forearmFKCtrlSpace.xfo = data['forearmXfo']
self.forearmFKCtrl.xfo = data['forearmXfo']
self.forearmFKCtrl.scalePoints(
Vec3(data['forearmLen'], data['forearmFKCtrlSize'],
data['forearmFKCtrlSize']))
self.handCtrlSpace.xfo = data['handXfo']
self.handCtrl.xfo = data['handXfo']
self.armIKCtrlSpace.xfo.tr = data['armEndXfo'].tr
self.armIKCtrl.xfo.tr = data['armEndXfo'].tr
if self.getLocation() == "R":
self.armIKCtrl.rotatePoints(0, 90, 0)
else:
self.armIKCtrl.rotatePoints(0, -90, 0)
self.armUpVCtrlSpace.xfo = data['upVXfo']
self.armUpVCtrl.xfo = data['upVXfo']
self.rightSideInputAttr.setValue(self.getLocation() is 'R')
self.armBone0LenInputAttr.setMin(0.0)
self.armBone0LenInputAttr.setMax(data['bicepLen'] * 3.0)
self.armBone0LenInputAttr.setValue(data['bicepLen'])
self.armBone1LenInputAttr.setMin(0.0)
self.armBone1LenInputAttr.setMax(data['forearmLen'] * 3.0)
self.armBone1LenInputAttr.setValue(data['forearmLen'])
# Outputs
self.handOutputTgt.xfo = data['handXfo']
# Eval Constraints
self.armIKCtrlSpaceInputConstraint.evaluate()
self.armUpVCtrlSpaceInputConstraint.evaluate()
self.armRootInputConstraint.evaluate()
self.armRootInputConstraint.evaluate()
self.handConstraint.evaluate()
self.handCtrlSpaceConstraint.evaluate()
# Eval Operators
self.spliceOp.evaluate()
self.outputsToDeformersKLOp.evaluate()
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 HeadComponentRig(HeadComponent):
"""Head Component Rig"""
def __init__(self, name='head', parent=None):
Profiler.getInstance().push("Construct Head Rig Component:" + name)
super(HeadComponentRig, self).__init__(name, parent)
# =========
# Controls
# =========
# Head
self.headCtrl = Control('head', parent=self.ctrlCmpGrp, shape='circle')
self.headCtrl.lockScale(x=True, y=True, z=True)
self.headCtrl.lockTranslation(x=True, y=True, z=True)
self.headCtrlSpace = self.headCtrl.insertCtrlSpace()
self.headCtrl.rotatePoints(0, 0, 90)
self.headCtrl.scalePoints(Vec3(3, 3, 3))
self.headCtrl.translatePoints(Vec3(0, 1, 0.25))
# Eye Left
self.eyeLeftCtrl = Control('eyeLeft', parent=self.ctrlCmpGrp, shape='sphere')
self.eyeLeftCtrl.lockScale(x=True, y=True, z=True)
self.eyeLeftCtrl.lockTranslation(x=True, y=True, z=True)
self.eyeLeftCtrlSpace = self.eyeLeftCtrl.insertCtrlSpace()
self.eyeLeftCtrl.rotatePoints(0, 90, 0)
self.eyeLeftCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.eyeLeftCtrl.setColor('blueMedium')
# Eye Right
self.eyeRightCtrl = Control('eyeRight', parent=self.ctrlCmpGrp, shape='sphere')
self.eyeRightCtrl.lockScale(x=True, y=True, z=True)
self.eyeRightCtrl.lockTranslation(x=True, y=True, z=True)
self.eyeRightCtrlSpace = self.eyeRightCtrl.insertCtrlSpace()
self.eyeRightCtrl.rotatePoints(0, 90, 0)
self.eyeRightCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
self.eyeRightCtrl.setColor('blueMedium')
# LookAt Control
self.lookAtCtrl = Control('lookAt', parent=self.ctrlCmpGrp, shape='square')
self.lookAtCtrl.lockScale(x=True, y=True, z=True)
self.lookAtCtrl.rotatePoints(90, 0, 0)
self.lookAtCtrlSpace = self.lookAtCtrl.insertCtrlSpace()
self.eyeLeftBase = Transform('eyeLeftBase', parent=self.headCtrl)
self.eyeRightBase = Transform('eyeRightBase', parent=self.headCtrl)
self.eyeLeftUpV = Transform('eyeLeftUpV', parent=self.headCtrl)
self.eyeRightUpV = Transform('eyeRightUpV', parent=self.headCtrl)
self.eyeLeftAtV = Transform('eyeLeftAtV', parent=self.lookAtCtrl)
self.eyeRightAtV = Transform('eyeRightAtV', parent=self.lookAtCtrl)
# Jaw
self.jawCtrl = Control('jaw', parent=self.headCtrl, shape='cube')
self.jawCtrlSpace = self.jawCtrl.insertCtrlSpace()
self.jawCtrl.lockScale(x=True, y=True, z=True)
self.jawCtrl.lockTranslation(x=True, y=True, z=True)
self.jawCtrl.alignOnYAxis(negative=True)
self.jawCtrl.alignOnZAxis()
self.jawCtrl.scalePoints(Vec3(1.45, 0.65, 1.25))
self.jawCtrl.translatePoints(Vec3(0, -0.25, 0))
self.jawCtrl.setColor('orange')
# ==========
# Deformers
# ==========
deformersLayer = self.getOrCreateLayer('deformers')
self.defCmpGrp = ComponentGroup(self.getName(), self, parent=deformersLayer)
self.addItem('defCmpGrp', self.defCmpGrp)
headDef = Joint('head', parent=self.defCmpGrp)
headDef.setComponent(self)
jawDef = Joint('jaw', parent=self.defCmpGrp)
jawDef.setComponent(self)
eyeLeftDef = Joint('eyeLeft', parent=self.defCmpGrp)
eyeLeftDef.setComponent(self)
eyeRightDef = Joint('eyeRight', parent=self.defCmpGrp)
eyeRightDef.setComponent(self)
# ==============
# Constrain I/O
# ==============
# Constraint inputs
self.headInputConstraint = PoseConstraint('_'.join([self.headCtrlSpace.getName(), 'To', self.neckRefInputTgt.getName()]))
self.headInputConstraint.setMaintainOffset(True)
self.headInputConstraint.addConstrainer(self.neckRefInputTgt)
self.headCtrlSpace.addConstraint(self.headInputConstraint)
# Constraint outputs
self.headOutputConstraint = PoseConstraint('_'.join([self.headOutputTgt.getName(), 'To', self.headCtrl.getName()]))
self.headOutputConstraint.addConstrainer(self.headCtrl)
self.headOutputTgt.addConstraint(self.headOutputConstraint)
self.jawOutputConstraint = PoseConstraint('_'.join([self.jawOutputTgt.getName(), 'To', self.jawCtrl.getName()]))
self.jawOutputConstraint.addConstrainer(self.jawCtrl)
self.jawOutputTgt.addConstraint(self.jawOutputConstraint)
self.eyeLOutputConstraint = PoseConstraint('_'.join([self.eyeLOutputTgt.getName(), 'To', self.eyeLeftCtrl.getName()]))
self.eyeLOutputConstraint.addConstrainer(self.eyeLeftCtrl)
self.eyeLOutputTgt.addConstraint(self.eyeLOutputConstraint)
self.eyeROutputConstraint = PoseConstraint('_'.join([self.eyeROutputTgt.getName(), 'To', self.eyeRightCtrl.getName()]))
self.eyeROutputConstraint.addConstrainer(self.eyeRightCtrl)
self.eyeROutputTgt.addConstraint(self.eyeROutputConstraint)
# Add Eye Left Direction KL Op
self.eyeLeftDirKLOp = KLOperator('eyeLeftDirKLOp', 'DirectionConstraintSolver', 'Kraken')
self.addOperator(self.eyeLeftDirKLOp)
# Add Att Inputs
self.eyeLeftDirKLOp.setInput('drawDebug', self.drawDebugInputAttr)
self.eyeLeftDirKLOp.setInput('rigScale', self.rigScaleInputAttr)
# Add Xfo Inputs
self.eyeLeftDirKLOp.setInput('position', self.eyeLeftBase)
self.eyeLeftDirKLOp.setInput('upVector', self.eyeLeftUpV)
self.eyeLeftDirKLOp.setInput('atVector', self.eyeLeftAtV)
# Add Xfo Outputs
self.eyeLeftDirKLOp.setOutput('constrainee', self.eyeLeftCtrlSpace)
# Add Eye Right Direction KL Op
self.eyeRightDirKLOp = KLOperator('eyeRightDirKLOp', 'DirectionConstraintSolver', 'Kraken')
self.addOperator(self.eyeRightDirKLOp)
# Add Att Inputs
self.eyeRightDirKLOp.setInput('drawDebug', self.drawDebugInputAttr)
self.eyeRightDirKLOp.setInput('rigScale', self.rigScaleInputAttr)
# Add Xfo Inputs
self.eyeRightDirKLOp.setInput('position', self.eyeRightBase)
self.eyeRightDirKLOp.setInput('upVector', self.eyeRightUpV)
self.eyeRightDirKLOp.setInput('atVector', self.eyeRightAtV)
# Add Xfo Outputs
self.eyeRightDirKLOp.setOutput('constrainee', self.eyeRightCtrlSpace)
# Add Deformer Joints KL Op
self.outputsToDeformersKLOp = KLOperator('headDeformerKLOp', '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.headOutputTgt, self.jawOutputTgt, self.eyeROutputTgt, self.eyeLOutputTgt])
# Add Xfo Outputs
self.outputsToDeformersKLOp.setOutput('constrainees', [headDef, jawDef, eyeRightDef, eyeLeftDef])
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(HeadComponentRig, self).loadData(data)
headXfo = data.get('headXfo')
headCrvData = data.get('headCrvData')
eyeLeftXfo = data.get('eyeLeftXfo')
eyeLeftCrvData = data.get('eyeLeftCrvData')
eyeRightXfo = data.get('eyeRightXfo')
eyeRightCrvData = data.get('eyeRightCrvData')
jawXfo = data.get('jawXfo')
jawCrvData = data.get('jawCrvData')
self.headCtrlSpace.xfo = headXfo
self.headCtrl.xfo = headXfo
self.headCtrl.setCurveData(headCrvData)
# self.eyeLeftCtrlSpace.xfo = eyeLeftXfo
# self.eyeLeftCtrl.xfo = eyeLeftXfo
self.eyeLeftCtrl.setCurveData(eyeLeftCrvData)
# self.eyeRightCtrlSpace.xfo = eyeRightXfo
# self.eyeRightCtrl.xfo = eyeRightXfo
self.eyeRightCtrl.setCurveData(eyeRightCrvData)
# LookAt
eyeLeftRelXfo = headXfo.inverse() * eyeLeftXfo
eyeRightRelXfo = headXfo.inverse() * eyeRightXfo
eyeMidRelPos = eyeLeftRelXfo.tr.linearInterpolate(eyeRightRelXfo.tr, 0.5)
eyeMidRelPos = eyeMidRelPos + Vec3(0.0, 0.0, 8.0)
eyeLen = eyeLeftRelXfo.tr.distanceTo(eyeRightRelXfo.tr)
self.eyeLeftBase.xfo = eyeLeftXfo
self.eyeRightBase.xfo = eyeRightXfo
self.eyeLeftUpV.xfo = eyeLeftXfo * Xfo(Vec3(0, 1, 0))
self.eyeRightUpV.xfo = eyeRightXfo * Xfo(Vec3(0, 1, 0))
self.eyeLeftAtV.xfo.tr = eyeLeftXfo.transformVector(Vec3(8.0, 0.0, 0.0))
self.eyeRightAtV.xfo.tr = eyeRightXfo.transformVector(Vec3(8.0, 0.0, 0.0))
lookAtXfo = headXfo.clone()
lookAtXfo.tr = headXfo.transformVector(eyeMidRelPos)
self.lookAtCtrl.scalePoints(Vec3(eyeLen * 1.6, eyeLen * 0.65, 1.0))
self.lookAtCtrl.xfo = lookAtXfo
self.lookAtCtrlSpace.xfo = lookAtXfo
self.jawCtrlSpace.xfo = jawXfo
self.jawCtrl.xfo = jawXfo
self.jawCtrl.setCurveData(jawCrvData)
# ============
# Set IO Xfos
# ============
self.neckRefInputTgt.xfo = headXfo
self.worldRefInputTgt.xfo = headXfo
self.headOutputTgt.xfo = headXfo
self.jawOutputTgt.xfo = jawXfo
self.eyeLOutputTgt.xfo = eyeLeftXfo
self.eyeROutputTgt.xfo = eyeRightXfo
# Eval Constraints
self.headInputConstraint.evaluate()
self.headOutputConstraint.evaluate()
self.jawOutputConstraint.evaluate()
self.eyeLOutputConstraint.evaluate()
self.eyeROutputConstraint.evaluate()
# Eval Operators
self.eyeLeftDirKLOp.evaluate()
self.eyeRightDirKLOp.evaluate()
self.outputsToDeformersKLOp.evaluate()
# Have to set the eye control xfos to match the evaluated xfos from
self.eyeLeftCtrl.xfo = self.eyeLeftCtrlSpace.xfo
self.eyeRightCtrl.xfo = self.eyeRightCtrlSpace.xfo
def addFinger(self, name):
digitSizeAttributes = []
fingerGuideCtrls = []
firstDigitCtrl = Control(name + "01", parent=self.handCtrl, shape='sphere')
firstDigitCtrl.scalePoints(Vec3(0.125, 0.125, 0.125))
firstDigitShapeCtrl = Control(name + "Shp01", parent=self.guideCtrlHrcGrp, shape='square')
firstDigitShapeCtrl.setColor('yellow')
firstDigitShapeCtrl.scalePoints(Vec3(0.175, 0.175, 0.175))
firstDigitShapeCtrl.translatePoints(Vec3(0.0, 0.125, 0.0))
fingerGuideCtrls.append(firstDigitShapeCtrl)
firstDigitCtrl.shapeCtrl = firstDigitShapeCtrl
firstDigitVisAttr = firstDigitShapeCtrl.getVisibilityAttr()
firstDigitVisAttr.connect(self.ctrlShapeToggle)
triangleCtrl = Control('tempCtrl', parent=None, shape='triangle')
triangleCtrl.rotatePoints(90.0, 0.0, 0.0)
triangleCtrl.scalePoints(Vec3(0.025, 0.025, 0.025))
triangleCtrl.translatePoints(Vec3(0.0, 0.0875, 0.0))
firstDigitCtrl.appendCurveData(triangleCtrl.getCurveData())
firstDigitCtrl.lockScale(True, True, True)
digitSettingsAttrGrp = AttributeGroup("DigitSettings", parent=firstDigitCtrl)
digitSizeAttr = ScalarAttribute('size', value=0.25, parent=digitSettingsAttrGrp)
digitSizeAttributes.append(digitSizeAttr)
# Set Finger
self.fingers[name] = []
self.fingers[name].append(firstDigitCtrl)
parent = firstDigitCtrl
numJoints = self.numJointsAttr.getValue()
if name == "thumb":
numJoints = 3
for i in xrange(2, numJoints + 2):
digitCtrl = Control(name + str(i).zfill(2), parent=parent, shape='sphere')
if i != numJoints + 1:
digitCtrl.scalePoints(Vec3(0.125, 0.125, 0.125))
digitCtrl.appendCurveData(triangleCtrl.getCurveData())
digitShapeCtrl = Control(name + 'Shp' + str(i).zfill(2), parent=self.guideCtrlHrcGrp, shape='circle')
digitShapeCtrl.setColor('yellow')
digitShapeCtrl.scalePoints(Vec3(0.175, 0.175, 0.175))
digitShapeCtrl.getVisibilityAttr().connect(self.ctrlShapeToggle)
digitCtrl.shapeCtrl = digitShapeCtrl
if i == 2:
digitShapeCtrl.translatePoints(Vec3(0.0, 0.125, 0.0))
else:
digitShapeCtrl.rotatePoints(0.0, 0.0, 90.0)
fingerGuideCtrls.append(digitShapeCtrl)
# Add size attr to all but last guide control
digitSettingsAttrGrp = AttributeGroup("DigitSettings", parent=digitCtrl)
digitSizeAttr = ScalarAttribute('size', value=0.25, parent=digitSettingsAttrGrp)
digitSizeAttributes.append(digitSizeAttr)
else:
digitCtrl.scalePoints(Vec3(0.0875, 0.0875, 0.0875))
digitCtrl.lockScale(True, True, True)
self.fingers[name].append(digitCtrl)
parent = digitCtrl
# ===========================
# Create Canvas Operators
# ===========================
# Add Finger Guide Canvas Op
fingerGuideCanvasOp = CanvasOperator(name + 'FingerGuide', 'Kraken.Solvers.Biped.BipedFingerGuideSolver')
self.addOperator(fingerGuideCanvasOp)
# Add Att Inputs
fingerGuideCanvasOp.setInput('drawDebug', self.drawDebugInputAttr)
fingerGuideCanvasOp.setInput('rigScale', self.rigScaleInputAttr)
# Add Xfo Inputs
fingerGuideCanvasOp.setInput('controls', self.fingers[name])
fingerGuideCanvasOp.setInput('planeSizes', digitSizeAttributes)
# Add Xfo Outputs
fingerGuideCanvasOp.setOutput('result', fingerGuideCtrls)
fingerGuideCanvasOp.setOutput('forceEval', firstDigitCtrl.getVisibilityAttr())
return firstDigitCtrl
class HeadComponentGuide(HeadComponent):
"""Head Component Guide"""
def __init__(self, name='head', parent=None, *args, **kwargs):
Profiler.getInstance().push("Construct Head Guide Component:" + name)
super(HeadComponentGuide, self).__init__(name, parent, *args, **kwargs)
# =========
# Controls
# =========
guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self)
sphereCtrl = Control('sphere', shape='sphere')
sphereCtrl.scalePoints(Vec3(0.375, 0.375, 0.375))
self.headCtrl = Control('head', parent=self.ctrlCmpGrp, shape='square')
self.headCtrl.rotatePoints(90, 0, 0)
self.headCtrl.translatePoints(Vec3(0.0, 0.5, 0.0))
self.headCtrl.scalePoints(Vec3(1.8, 2.0, 2.0))
self.eyeLeftCtrl = Control('eyeLeft', parent=self.headCtrl, shape='arrow_thin')
self.eyeLeftCtrl.translatePoints(Vec3(0, 0, 0.5))
self.eyeLeftCtrl.rotatePoints(0, 90, 0)
self.eyeLeftCtrl.appendCurveData(sphereCtrl.getCurveData())
self.eyeRightCtrl = Control('eyeRight', parent=self.headCtrl, shape='arrow_thin')
self.eyeRightCtrl.translatePoints(Vec3(0, 0, 0.5))
self.eyeRightCtrl.rotatePoints(0, 90, 0)
self.eyeRightCtrl.appendCurveData(sphereCtrl.getCurveData())
self.jawCtrl = Control('jaw', parent=self.headCtrl, shape='square')
self.jawCtrl.rotatePoints(90, 0, 0)
self.jawCtrl.rotatePoints(0, 90, 0)
self.jawCtrl.translatePoints(Vec3(0.0, -0.5, 0.5))
self.jawCtrl.scalePoints(Vec3(1.0, 0.8, 1.5))
self.jawCtrl.setColor('orange')
eyeXAlignOri = Quat()
eyeXAlignOri.setFromAxisAndAngle(Vec3(0, 1, 0), Math_degToRad(-90))
self.default_data = {
"name": name,
"location": "M",
"headXfo": Xfo(Vec3(0.0, 17.5, -0.5)),
"headCrvData": self.headCtrl.getCurveData(),
"eyeLeftXfo": Xfo(tr=Vec3(0.375, 18.5, 0.5), ori=eyeXAlignOri),
"eyeLeftCrvData": self.eyeLeftCtrl.getCurveData(),
"eyeRightXfo": Xfo(tr=Vec3(-0.375, 18.5, 0.5), ori=eyeXAlignOri),
"eyeRightCrvData": self.eyeRightCtrl.getCurveData(),
"jawXfo": Xfo(Vec3(0.0, 17.875, -0.275)),
"jawCrvData": self.jawCtrl.getCurveData()
}
self.loadData(self.default_data)
Profiler.getInstance().pop()
# =============
# Data Methods
# =============
def saveData(self):
"""Save the data for the component to be persisted.
Return:
The JSON data object
"""
data = super(HeadComponentGuide, self).saveData()
data['headXfo'] = self.headCtrl.xfo
data['headCrvData'] = self.headCtrl.getCurveData()
data['eyeLeftXfo'] = self.eyeLeftCtrl.xfo
data['eyeLeftCrvData'] = self.eyeLeftCtrl.getCurveData()
data['eyeRightXfo'] = self.eyeRightCtrl.xfo
data['eyeRightCrvData'] = self.eyeRightCtrl.getCurveData()
data['jawXfo'] = self.jawCtrl.xfo
data['jawCrvData'] = self.jawCtrl.getCurveData()
return data
def loadData(self, data):
"""Load a saved guide representation from persisted data.
Arguments:
data -- object, The JSON data object.
Return:
True if successful.
"""
super(HeadComponentGuide, self).loadData(data)
self.headCtrl.xfo = data.get('headXfo', self.default_data['headXfo'])
self.headCtrl.setCurveData(data.get('headCrvData', self.default_data['headCrvData']))
self.eyeLeftCtrl.xfo = data.get('eyeLeftXfo', self.default_data['eyeLeftXfo'])
self.eyeLeftCtrl.setCurveData(data.get('eyeLeftCrvData', self.default_data['eyeLeftCrvData']))
self.eyeRightCtrl.xfo = data.get('eyeRightXfo', self.default_data['eyeRightXfo'])
self.eyeRightCtrl.setCurveData(data.get('eyeRightCrvData', self.default_data['eyeRightCrvData']))
self.jawCtrl.xfo = data.get('jawXfo', self.default_data['jawXfo'])
self.jawCtrl.setCurveData(data.get('jawCrvData', self.default_data['jawCrvData']))
return True
def getRigBuildData(self):
"""Returns the Guide data used by the Rig Component to define the layout of the final rig..
Return:
The JSON rig data object.
"""
data = super(HeadComponentGuide, self).getRigBuildData()
data['headXfo'] = self.headCtrl.xfo
data['headCrvData'] = self.headCtrl.getCurveData()
data['eyeLeftXfo'] = self.eyeLeftCtrl.xfo
data['eyeLeftCrvData'] = self.eyeLeftCtrl.getCurveData()
data['eyeRightXfo'] = self.eyeRightCtrl.xfo
data['eyeRightCrvData'] = self.eyeRightCtrl.getCurveData()
data['jawXfo'] = self.jawCtrl.xfo
data['jawCrvData'] = self.jawCtrl.getCurveData()
return data
# ==============
# Class Methods
# ==============
@classmethod
def getComponentType(cls):
"""Enables introspection of the class prior to construction to determine if it is a guide component.
Return:
The true if this component is a guide component.
"""
return 'Guide'
@classmethod
def getRigComponentClass(cls):
"""Returns the corresponding rig component class for this guide component class
Return:
The rig component class.
"""
return HeadComponentRig
class HandComponentGuide(HandComponent):
"""Hand Component Guide"""
def __init__(self, name='hand', parent=None, *args, **kwargs):
Profiler.getInstance().push("Construct Hand Guide Component:" + name)
super(HandComponentGuide, self).__init__(name, parent, *args, **kwargs)
# =========
# Controls
# =========
# Guide Controls
self.guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self)
self.digitNamesAttr = StringAttribute('digitNames', value="thumb,index,middle,ring,pinky", parent=self.guideSettingsAttrGrp)
self.digitNamesAttr.setValueChangeCallback(self.updateFingers)
self.numJointsAttr = IntegerAttribute('numJoints', value=4, minValue=2, maxValue=20, parent=self.guideSettingsAttrGrp)
self.numJointsAttr.setValueChangeCallback(self.resizeDigits)
self.fingers = OrderedDict()
self.handCtrl = Control('hand', parent=self.ctrlCmpGrp, shape="square")
self.handCtrl.rotatePoints(0.0, 0.0, 90.0)
self.handCtrl.scalePoints(Vec3(1.0, 0.75, 1.0))
self.handCtrl.setColor('yellow')
self.handGuideSettingsAttrGrp = AttributeGroup("Settings", parent=self.handCtrl)
self.ctrlShapeToggle = BoolAttribute('ctrlShape_vis', value=False, parent=self.handGuideSettingsAttrGrp)
self.handDebugInputAttr = BoolAttribute('drawDebug', value=False, parent=self.handGuideSettingsAttrGrp)
self.drawDebugInputAttr.connect(self.handDebugInputAttr)
self.guideCtrlHrcGrp = HierarchyGroup('controlShapes', parent=self.ctrlCmpGrp)
self.default_data = {
"name": name,
"location": "L",
"handXfo": Xfo(Vec3(7.1886, 12.2819, 0.4906)),
"digitNames": self.digitNamesAttr.getValue(),
"numJoints": self.numJointsAttr.getValue(),
"fingers": self.fingers
}
self.loadData(self.default_data)
Profiler.getInstance().pop()
# =============
# Data Methods
# =============
def saveData(self):
"""Save the data for the component to be persisted.
Return:
The JSON data object
"""
data = super(HandComponentGuide, self).saveData()
data['handXfo'] = self.handCtrl.xfo
data['digitNames'] = self.digitNamesAttr.getValue()
data['numJoints'] = self.numJointsAttr.getValue()
fingerXfos = {}
fingerShapeCtrlData = {}
for finger in self.fingers.keys():
fingerXfos[finger] = [x.xfo for x in self.fingers[finger]]
fingerShapeCtrlData[finger] = []
for i, digit in enumerate(self.fingers[finger]):
if i != len(self.fingers[finger]) - 1:
fingerShapeCtrlData[finger].append(digit.shapeCtrl.getCurveData())
data['fingersGuideXfos'] = fingerXfos
data['fingerShapeCtrlData'] = fingerShapeCtrlData
return data
def loadData(self, data):
"""Load a saved guide representation from persisted data.
Arguments:
data -- object, The JSON data object.
Return:
True if successful.
"""
super(HandComponentGuide, self).loadData(data)
self.handCtrl.xfo = data.get('handXfo')
self.numJointsAttr.setValue(data.get('numJoints'))
self.digitNamesAttr.setValue(data.get('digitNames'))
fingersGuideXfos = data.get('fingersGuideXfos')
fingerShapeCtrlData = data.get('fingerShapeCtrlData')
if fingersGuideXfos is not None:
for finger in self.fingers.keys():
for i in xrange(len(self.fingers[finger])):
self.fingers[finger][i].xfo = fingersGuideXfos[finger][i]
if hasattr(self.fingers[finger][i], 'shapeCtrl'):
if fingerShapeCtrlData is not None:
if finger in fingerShapeCtrlData:
self.fingers[finger][i].shapeCtrl.setCurveData(fingerShapeCtrlData[finger][i])
for op in self.getOperators():
guideOpName = ''.join([op.getName().split('FingerGuideOp')[0], self.getLocation(), 'FingerGuideOp'])
op.setName(guideOpName)
return True
def getRigBuildData(self):
"""Returns the Guide data used by the Rig Component to define the layout of the final rig..
Return:
The JSON rig data object.
"""
data = super(HandComponentGuide, self).getRigBuildData()
data['handXfo'] = self.handCtrl.xfo
fingerData = {}
for finger in self.fingers.keys():
fingerData[finger] = []
for i, joint in enumerate(self.fingers[finger]):
if i == len(self.fingers[finger]) - 1:
continue
# Calculate Xfo
boneVec = self.fingers[finger][i + 1].xfo.tr - self.fingers[finger][i].xfo.tr
bone1Normal = self.fingers[finger][i].xfo.ori.getZaxis().cross(boneVec).unit()
bone1ZAxis = boneVec.cross(bone1Normal).unit()
jointXfo = Xfo()
jointXfo.setFromVectors(boneVec.unit(), bone1Normal, bone1ZAxis, self.fingers[finger][i].xfo.tr)
jointData = {
'curveData': self.fingers[finger][i].shapeCtrl.getCurveData(),
'length': self.fingers[finger][i].xfo.tr.distanceTo(self.fingers[finger][i + 1].xfo.tr),
'xfo': jointXfo
}
fingerData[finger].append(jointData)
data['fingerData'] = fingerData
return data
# ==========
# Callbacks
# ==========
def addFinger(self, name):
digitSizeAttributes = []
fingerGuideCtrls = []
firstDigitCtrl = Control(name + "01", parent=self.handCtrl, shape='sphere')
firstDigitCtrl.scalePoints(Vec3(0.125, 0.125, 0.125))
firstDigitShapeCtrl = Control(name + "Shp01", parent=self.guideCtrlHrcGrp, shape='square')
firstDigitShapeCtrl.setColor('yellow')
firstDigitShapeCtrl.scalePoints(Vec3(0.175, 0.175, 0.175))
firstDigitShapeCtrl.translatePoints(Vec3(0.0, 0.125, 0.0))
fingerGuideCtrls.append(firstDigitShapeCtrl)
firstDigitCtrl.shapeCtrl = firstDigitShapeCtrl
firstDigitVisAttr = firstDigitShapeCtrl.getVisibilityAttr()
firstDigitVisAttr.connect(self.ctrlShapeToggle)
triangleCtrl = Control('tempCtrl', parent=None, shape='triangle')
triangleCtrl.rotatePoints(90.0, 0.0, 0.0)
triangleCtrl.scalePoints(Vec3(0.025, 0.025, 0.025))
triangleCtrl.translatePoints(Vec3(0.0, 0.0875, 0.0))
firstDigitCtrl.appendCurveData(triangleCtrl.getCurveData())
firstDigitCtrl.lockScale(True, True, True)
digitSettingsAttrGrp = AttributeGroup("DigitSettings", parent=firstDigitCtrl)
digitSizeAttr = ScalarAttribute('size', value=0.25, parent=digitSettingsAttrGrp)
digitSizeAttributes.append(digitSizeAttr)
# Set Finger
self.fingers[name] = []
self.fingers[name].append(firstDigitCtrl)
parent = firstDigitCtrl
numJoints = self.numJointsAttr.getValue()
if name == "thumb":
numJoints = 3
for i in xrange(2, numJoints + 2):
digitCtrl = Control(name + str(i).zfill(2), parent=parent, shape='sphere')
if i != numJoints + 1:
digitCtrl.scalePoints(Vec3(0.125, 0.125, 0.125))
digitCtrl.appendCurveData(triangleCtrl.getCurveData())
digitShapeCtrl = Control(name + 'Shp' + str(i).zfill(2), parent=self.guideCtrlHrcGrp, shape='circle')
digitShapeCtrl.setColor('yellow')
digitShapeCtrl.scalePoints(Vec3(0.175, 0.175, 0.175))
digitShapeCtrl.getVisibilityAttr().connect(self.ctrlShapeToggle)
digitCtrl.shapeCtrl = digitShapeCtrl
if i == 2:
digitShapeCtrl.translatePoints(Vec3(0.0, 0.125, 0.0))
else:
digitShapeCtrl.rotatePoints(0.0, 0.0, 90.0)
fingerGuideCtrls.append(digitShapeCtrl)
# Add size attr to all but last guide control
digitSettingsAttrGrp = AttributeGroup("DigitSettings", parent=digitCtrl)
digitSizeAttr = ScalarAttribute('size', value=0.25, parent=digitSettingsAttrGrp)
digitSizeAttributes.append(digitSizeAttr)
else:
digitCtrl.scalePoints(Vec3(0.0875, 0.0875, 0.0875))
digitCtrl.lockScale(True, True, True)
self.fingers[name].append(digitCtrl)
parent = digitCtrl
# ===========================
# Create Canvas Operators
# ===========================
# Add Finger Guide Canvas Op
fingerGuideCanvasOp = CanvasOperator(name + 'FingerGuide', 'Kraken.Solvers.Biped.BipedFingerGuideSolver')
self.addOperator(fingerGuideCanvasOp)
# Add Att Inputs
fingerGuideCanvasOp.setInput('drawDebug', self.drawDebugInputAttr)
fingerGuideCanvasOp.setInput('rigScale', self.rigScaleInputAttr)
# Add Xfo Inputs
fingerGuideCanvasOp.setInput('controls', self.fingers[name])
fingerGuideCanvasOp.setInput('planeSizes', digitSizeAttributes)
# Add Xfo Outputs
fingerGuideCanvasOp.setOutput('result', fingerGuideCtrls)
fingerGuideCanvasOp.setOutput('forceEval', firstDigitCtrl.getVisibilityAttr())
return firstDigitCtrl
def removeFinger(self, name):
self.handCtrl.removeChild(self.fingers[name][0])
del self.fingers[name]
def placeFingers(self):
spacing = 0.25
length = spacing * (len(self.fingers.keys()) - 1)
mid = length / 2.0
startOffset = length - mid
for i, finger in enumerate(self.fingers.keys()):
parentCtrl = self.handCtrl
numJoints = self.numJointsAttr.getValue()
if finger == "thumb":
numJoints = 3
for y in xrange(numJoints + 1):
if y == 1:
xOffset = 0.375
else:
xOffset = 0.25
if y == 0:
offsetVec = Vec3(xOffset, 0, startOffset - (i * spacing))
else:
offsetVec = Vec3(xOffset, 0, 0)
fingerPos = parentCtrl.xfo.transformVector(offsetVec)
fingerXfo = Xfo(tr=fingerPos, ori=self.handCtrl.xfo.ori)
self.fingers[finger][y].xfo = fingerXfo
parentCtrl = self.fingers[finger][y]
def updateFingers(self, fingers):
if " " in fingers:
self.digitNamesAttr.setValue(fingers.replace(" ", ""))
return
fingerNames = fingers.split(',')
# Delete fingers that don't exist any more
for finger in list(set(self.fingers.keys()) - set(fingerNames)):
self.removeFinger(finger)
# Add Fingers
for finger in fingerNames:
if finger in self.fingers.keys():
continue
self.addFinger(finger)
self.placeFingers()
def resizeDigits(self, numJoints):
initNumJoints = numJoints
for finger in self.fingers.keys():
if finger == "thumb":
numJoints = 3
else:
numJoints = initNumJoints
if numJoints + 1 == len(self.fingers[finger]):
continue
elif numJoints + 1 > len(self.fingers[finger]):
for i in xrange(len(self.fingers[finger]), numJoints + 1):
prevDigit = self.fingers[finger][i - 1]
digitCtrl = Control(finger + str(i + 1).zfill(2), parent=prevDigit, shape='sphere')
digitCtrl.setColor('orange')
digitCtrl.scalePoints(Vec3(0.25, 0.25, 0.25))
digitCtrl.lockScale(True, True, True)
self.fingers[finger].append(digitCtrl)
elif numJoints + 1 < len(self.fingers[finger]):
numExtraCtrls = len(self.fingers[finger]) - (numJoints + 1)
for i in xrange(numExtraCtrls):
removedJoint = self.fingers[finger].pop()
removedJoint.getParent().removeChild(removedJoint)
self.placeFingers()
# ==============
# Class Methods
# ==============
@classmethod
def getComponentType(cls):
"""Enables introspection of the class prior to construction to determine if it is a guide component.
Return:
The true if this component is a guide component.
"""
return 'Guide'
@classmethod
def getRigComponentClass(cls):
"""Returns the corresponding rig component class for this guide component class
Return:
The rig component class.
"""
return HandComponentRig
class NeckComponentRig(NeckComponent):
"""Neck Component"""
def __init__(self, name="neck", parent=None):
Profiler.getInstance().push("Construct Neck Rig Component:" + name)
super(NeckComponentRig, self).__init__(name, parent)
# =========
# Controls
# =========
# Neck
self.neckCtrlSpace = CtrlSpace('neck', parent=self.ctrlCmpGrp)
self.neckCtrl = Control('neck', parent=self.neckCtrlSpace, shape="pin")
self.neckCtrl.scalePoints(Vec3(1.25, 1.25, 1.25))
self.neckCtrl.translatePoints(Vec3(0, 0, -0.5))
self.neckCtrl.rotatePoints(90, 0, 90)
self.neckCtrl.setColor("orange")
# ==========
# Deformers
# ==========
deformersLayer = self.getOrCreateLayer('deformers')
defCmpGrp = ComponentGroup(self.getName(), self, parent=deformersLayer)
neckDef = Joint('neck', parent=defCmpGrp)
neckDef.setComponent(self)
# ==============
# Constrain I/O
# ==============
# Constraint inputs
clavicleInputConstraint = PoseConstraint('_'.join([self.neckCtrlSpace.getName(), 'To', self.neckBaseInputTgt.getName()]))
clavicleInputConstraint.setMaintainOffset(True)
clavicleInputConstraint.addConstrainer(self.neckBaseInputTgt)
self.neckCtrlSpace.addConstraint(clavicleInputConstraint)
# Constraint outputs
neckOutputConstraint = PoseConstraint('_'.join([self.neckOutputTgt.getName(), 'To', self.neckCtrl.getName()]))
neckOutputConstraint.addConstrainer(self.neckCtrl)
self.neckOutputTgt.addConstraint(neckOutputConstraint)
neckEndConstraint = PoseConstraint('_'.join([self.neckEndOutputTgt.getName(), 'To', self.neckCtrl.getName()]))
neckEndConstraint.addConstrainer(self.neckCtrl)
self.neckEndOutputTgt.addConstraint(neckEndConstraint)
# ===============
# Add Splice Ops
# ===============
#Add Deformer Splice Op
spliceOp = KLOperator('neckDeformerKLOp', 'PoseConstraintSolver', 'Kraken')
self.addOperator(spliceOp)
# Add Att Inputs
spliceOp.setInput('drawDebug', self.drawDebugInputAttr)
spliceOp.setInput('rigScale', self.rigScaleInputAttr)
# Add Xfo Inputstrl)
spliceOp.setInput('constrainer', self.neckEndOutputTgt)
# Add Xfo Outputs
spliceOp.setOutput('constrainee', neckDef)
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(NeckComponentRig, self).loadData( data )
self.neckCtrlSpace.xfo = data['neckXfo']
self.neckCtrl.xfo = data['neckXfo']
# ============
# Set IO Xfos
# ============
self.neckBaseInputTgt.xfo = data['neckXfo']
self.neckEndOutputTgt.xfo = data['neckXfo']
self.neckOutputTgt.xfo = data['neckXfo']
class HeadComponentGuide(HeadComponent):
"""Head Component Guide"""
def __init__(self, name='head', parent=None, *args, **kwargs):
Profiler.getInstance().push("Construct Head Guide Component:" + name)
super(HeadComponentGuide, self).__init__(name, parent, *args, **kwargs)
# =========
# Controls
# =========
guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self)
sphereCtrl = Control('sphere', shape='sphere')
sphereCtrl.scalePoints(Vec3(0.375, 0.375, 0.375))
self.headCtrl = Control('head', parent=self.ctrlCmpGrp, shape='square')
self.headCtrl.rotatePoints(90, 0, 0)
self.headCtrl.translatePoints(Vec3(0.0, 0.5, 0.0))
self.headCtrl.scalePoints(Vec3(1.8, 2.0, 2.0))
self.eyeLeftCtrl = Control('eyeLeft',
parent=self.headCtrl,
shape='arrow_thin')
self.eyeLeftCtrl.translatePoints(Vec3(0, 0, 0.5))
self.eyeLeftCtrl.rotatePoints(0, 90, 0)
self.eyeLeftCtrl.appendCurveData(sphereCtrl.getCurveData())
self.eyeRightCtrl = Control('eyeRight',
parent=self.headCtrl,
shape='arrow_thin')
self.eyeRightCtrl.translatePoints(Vec3(0, 0, 0.5))
self.eyeRightCtrl.rotatePoints(0, 90, 0)
self.eyeRightCtrl.appendCurveData(sphereCtrl.getCurveData())
self.jawCtrl = Control('jaw', parent=self.headCtrl, shape='square')
self.jawCtrl.rotatePoints(90, 0, 0)
self.jawCtrl.rotatePoints(0, 90, 0)
self.jawCtrl.translatePoints(Vec3(0.0, -0.5, 0.5))
self.jawCtrl.scalePoints(Vec3(1.0, 0.8, 1.5))
self.jawCtrl.setColor('orange')
eyeXAlignOri = Quat()
eyeXAlignOri.setFromAxisAndAngle(Vec3(0, 1, 0), Math_degToRad(-90))
self.default_data = {
"name": name,
"location": "M",
"headXfo": Xfo(Vec3(0.0, 17.5, -0.5)),
"headCrvData": self.headCtrl.getCurveData(),
"eyeLeftXfo": Xfo(tr=Vec3(0.375, 18.5, 0.5), ori=eyeXAlignOri),
"eyeLeftCrvData": self.eyeLeftCtrl.getCurveData(),
"eyeRightXfo": Xfo(tr=Vec3(-0.375, 18.5, 0.5), ori=eyeXAlignOri),
"eyeRightCrvData": self.eyeRightCtrl.getCurveData(),
"jawXfo": Xfo(Vec3(0.0, 17.875, -0.275)),
"jawCrvData": self.jawCtrl.getCurveData()
}
self.loadData(self.default_data)
Profiler.getInstance().pop()
# =============
# Data Methods
# =============
def saveData(self):
"""Save the data for the component to be persisted.
Return:
The JSON data object
"""
data = super(HeadComponentGuide, self).saveData()
data['headXfo'] = self.headCtrl.xfo
data['headCrvData'] = self.headCtrl.getCurveData()
data['eyeLeftXfo'] = self.eyeLeftCtrl.xfo
data['eyeLeftCrvData'] = self.eyeLeftCtrl.getCurveData()
data['eyeRightXfo'] = self.eyeRightCtrl.xfo
data['eyeRightCrvData'] = self.eyeRightCtrl.getCurveData()
data['jawXfo'] = self.jawCtrl.xfo
data['jawCrvData'] = self.jawCtrl.getCurveData()
return data
def loadData(self, data):
"""Load a saved guide representation from persisted data.
Arguments:
data -- object, The JSON data object.
Return:
True if successful.
"""
super(HeadComponentGuide, self).loadData(data)
self.headCtrl.xfo = data.get('headXfo', self.default_data['headXfo'])
self.headCtrl.setCurveData(
data.get('headCrvData', self.default_data['headCrvData']))
self.eyeLeftCtrl.xfo = data.get('eyeLeftXfo',
self.default_data['eyeLeftXfo'])
self.eyeLeftCtrl.setCurveData(
data.get('eyeLeftCrvData', self.default_data['eyeLeftCrvData']))
self.eyeRightCtrl.xfo = data.get('eyeRightXfo',
self.default_data['eyeRightXfo'])
self.eyeRightCtrl.setCurveData(
data.get('eyeRightCrvData', self.default_data['eyeRightCrvData']))
self.jawCtrl.xfo = data.get('jawXfo', self.default_data['jawXfo'])
self.jawCtrl.setCurveData(
data.get('jawCrvData', self.default_data['jawCrvData']))
return True
def getRigBuildData(self):
"""Returns the Guide data used by the Rig Component to define the layout of the final rig..
Return:
The JSON rig data object.
"""
data = super(HeadComponentGuide, self).getRigBuildData()
data['headXfo'] = self.headCtrl.xfo
data['headCrvData'] = self.headCtrl.getCurveData()
data['eyeLeftXfo'] = self.eyeLeftCtrl.xfo
data['eyeLeftCrvData'] = self.eyeLeftCtrl.getCurveData()
data['eyeRightXfo'] = self.eyeRightCtrl.xfo
data['eyeRightCrvData'] = self.eyeRightCtrl.getCurveData()
data['jawXfo'] = self.jawCtrl.xfo
data['jawCrvData'] = self.jawCtrl.getCurveData()
return data
# ==============
# Class Methods
# ==============
@classmethod
def getComponentType(cls):
"""Enables introspection of the class prior to construction to determine if it is a guide component.
Return:
The true if this component is a guide component.
"""
return 'Guide'
@classmethod
def getRigComponentClass(cls):
"""Returns the corresponding rig component class for this guide component class
Return:
The rig component class.
"""
return HeadComponentRig
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()
# Forearm
self.forearmFKCtrlSpace = CtrlSpace("forearmFK", parent=self.bicepFKCtrl)
self.forearmFKCtrl = Control("forearmFK", parent=self.forearmFKCtrlSpace, shape="cube")
self.forearmFKCtrl.alignOnXAxis()
self.handCtrlSpace = CtrlSpace("hand", parent=self.ctrlCmpGrp)
self.handCtrl = Control("hand", parent=self.handCtrlSpace, shape="circle")
self.handCtrl.rotatePoints(0, 0, 90)
self.handCtrl.scalePoints(Vec3(1.0, 0.75, 0.75))
# Arm IK
self.armIKCtrlSpace = CtrlSpace("IK", parent=self.ctrlCmpGrp)
self.armIKCtrl = Control("IK", parent=self.armIKCtrlSpace, shape="pin")
# Add Params to IK control
armSettingsAttrGrp = AttributeGroup("DisplayInfo_ArmSettings", parent=self.armIKCtrl)
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)
armIKBlendInputAttr = ScalarAttribute("fkik", value=0.0, minValue=0.0, maxValue=1.0, parent=armSettingsAttrGrp)
armSoftIKInputAttr = BoolAttribute("softIK", value=True, parent=armSettingsAttrGrp)
armSoftDistInputAttr = ScalarAttribute("softDist", value=0.0, minValue=0.0, parent=armSettingsAttrGrp)
armStretchInputAttr = BoolAttribute("stretch", value=True, parent=armSettingsAttrGrp)
armStretchBlendInputAttr = ScalarAttribute(
"stretchBlend", value=0.0, minValue=0.0, maxValue=1.0, parent=armSettingsAttrGrp
)
# Hand Params
handSettingsAttrGrp = AttributeGroup("DisplayInfo_HandSettings", parent=self.handCtrl)
handLinkToWorldInputAttr = ScalarAttribute("linkToWorld", 0.0, maxValue=1.0, parent=handSettingsAttrGrp)
self.drawDebugInputAttr.connect(armDebugInputAttr)
# 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)
# ==========
# Deformers
# ==========
deformersLayer = self.getOrCreateLayer("deformers")
defCmpGrp = ComponentGroup(self.getName(), self, parent=deformersLayer)
bicepDef = Joint("bicep", parent=defCmpGrp)
bicepDef.setComponent(self)
forearmDef = Joint("forearm", parent=defCmpGrp)
forearmDef.setComponent(self)
wristDef = Joint("wrist", parent=defCmpGrp)
wristDef.setComponent(self)
handDef = Joint("hand", parent=defCmpGrp)
handDef.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.clavicleEndInputTgt.getName()])
)
self.armRootInputConstraint.setMaintainOffset(True)
self.armRootInputConstraint.addConstrainer(self.clavicleEndInputTgt)
self.bicepFKCtrlSpace.addConstraint(self.armRootInputConstraint)
# Constraint outputs
self.handConstraint = PoseConstraint("_".join([self.handOutputTgt.getName(), "To", self.handCtrl.getName()]))
self.handConstraint.addConstrainer(self.handCtrl)
self.handOutputTgt.addConstraint(self.handConstraint)
self.handCtrlSpaceConstraint = PoseConstraint(
"_".join([self.handCtrlSpace.getName(), "To", self.armEndXfoOutputTgt.getName()])
)
self.handCtrlSpaceConstraint.setMaintainOffset(True)
self.handCtrlSpaceConstraint.addConstrainer(self.armEndXfoOutputTgt)
self.handCtrlSpace.addConstraint(self.handCtrlSpaceConstraint)
# ===============
# Add Splice Ops
# ===============
# Add Splice Op
self.spliceOp = SpliceOperator("armSpliceOp", "TwoBoneIKSolver", "Kraken")
self.addOperator(self.spliceOp)
# Add Att Inputs
self.spliceOp.setInput("drawDebug", self.drawDebugInputAttr)
self.spliceOp.setInput("rigScale", self.rigScaleInputAttr)
self.spliceOp.setInput("bone0Len", self.armBone0LenInputAttr)
self.spliceOp.setInput("bone1Len", self.armBone1LenInputAttr)
self.spliceOp.setInput("ikblend", armIKBlendInputAttr)
self.spliceOp.setInput("softIK", armSoftIKInputAttr)
self.spliceOp.setInput("softDist", armSoftDistInputAttr)
self.spliceOp.setInput("stretch", armStretchInputAttr)
self.spliceOp.setInput("stretchBlend", armStretchBlendInputAttr)
self.spliceOp.setInput("rightSide", self.rightSideInputAttr)
# Add Xfo Inputs
self.spliceOp.setInput("root", self.clavicleEndInputTgt)
self.spliceOp.setInput("bone0FK", self.bicepFKCtrl)
self.spliceOp.setInput("bone1FK", self.forearmFKCtrl)
self.spliceOp.setInput("ikHandle", self.armIKCtrl)
self.spliceOp.setInput("upV", self.armUpVCtrl)
# Add Xfo Outputs
self.spliceOp.setOutput("bone0Out", self.bicepOutputTgt)
self.spliceOp.setOutput("bone1Out", self.forearmOutputTgt)
self.spliceOp.setOutput("bone2Out", self.armEndXfoOutputTgt)
# Add Deformer Splice Op
self.outputsToDeformersSpliceOp = SpliceOperator("armDeformerSpliceOp", "MultiPoseConstraintSolver", "Kraken")
self.addOperator(self.outputsToDeformersSpliceOp)
# Add Att Inputs
self.outputsToDeformersSpliceOp.setInput("drawDebug", self.drawDebugInputAttr)
self.outputsToDeformersSpliceOp.setInput("rigScale", self.rigScaleInputAttr)
# Add Xfo Inputs
self.outputsToDeformersSpliceOp.setInput(
"constrainers", [self.bicepOutputTgt, self.forearmOutputTgt, self.armEndXfoOutputTgt, self.handOutputTgt]
)
# Add Xfo Outputs
self.outputsToDeformersSpliceOp.setOutput("constrainees", [bicepDef, forearmDef, wristDef, handDef])
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)
self.clavicleEndInputTgt.xfo.tr = data["bicepXfo"].tr
self.bicepFKCtrlSpace.xfo = data["bicepXfo"]
self.bicepFKCtrl.xfo = data["bicepXfo"]
self.bicepFKCtrl.scalePoints(Vec3(data["bicepLen"], data["bicepFKCtrlSize"], data["bicepFKCtrlSize"]))
self.bicepOutputTgt.xfo = data["bicepXfo"]
self.forearmOutputTgt.xfo = data["forearmXfo"]
self.forearmFKCtrlSpace.xfo = data["forearmXfo"]
self.forearmFKCtrl.xfo = data["forearmXfo"]
self.forearmFKCtrl.scalePoints(Vec3(data["forearmLen"], data["forearmFKCtrlSize"], data["forearmFKCtrlSize"]))
self.handCtrlSpace.xfo = data["handXfo"]
self.handCtrl.xfo = data["handXfo"]
self.armIKCtrlSpace.xfo.tr = data["armEndXfo"].tr
self.armIKCtrl.xfo.tr = data["armEndXfo"].tr
if self.getLocation() == "R":
self.armIKCtrl.rotatePoints(0, 90, 0)
else:
self.armIKCtrl.rotatePoints(0, -90, 0)
self.armUpVCtrlSpace.xfo = data["upVXfo"]
self.armUpVCtrl.xfo = data["upVXfo"]
self.rightSideInputAttr.setValue(self.getLocation() is "R")
self.armBone0LenInputAttr.setMin(0.0)
self.armBone0LenInputAttr.setMax(data["bicepLen"] * 3.0)
self.armBone0LenInputAttr.setValue(data["bicepLen"])
self.armBone1LenInputAttr.setMin(0.0)
self.armBone1LenInputAttr.setMax(data["forearmLen"] * 3.0)
self.armBone1LenInputAttr.setValue(data["forearmLen"])
# Outputs
self.handOutputTgt.xfo = data["handXfo"]
# Eval Constraints
self.armIKCtrlSpaceInputConstraint.evaluate()
self.armUpVCtrlSpaceInputConstraint.evaluate()
self.armRootInputConstraint.evaluate()
self.armRootInputConstraint.evaluate()
self.handConstraint.evaluate()
self.handCtrlSpaceConstraint.evaluate()
# Eval Operators
self.spliceOp.evaluate()
self.outputsToDeformersSpliceOp.evaluate()
class FabriceHeadGuide(FabriceHead):
"""Fabrice Head Component Guide"""
def __init__(self, name='head', parent=None):
Profiler.getInstance().push("Construct Head Guide Component:" + name)
super(FabriceHeadGuide, self).__init__(name, parent)
# =========
# Controls
# =========
guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self)
self.headCtrl = Control('head', parent=self.ctrlCmpGrp, shape="circle")
self.headCtrl.rotatePoints(90.0, 0.0, 0.0)
self.headCtrl.scalePoints(Vec3(3.5, 3.5, 3.5))
self.jawCtrl = Control('jaw', parent=self.ctrlCmpGrp, shape="cube")
self.jawCtrl.alignOnZAxis()
self.jawCtrl.scalePoints(Vec3(2.0, 0.5, 2.0))
self.jawCtrl.alignOnYAxis(negative=True)
self.jawCtrl.setColor('orange')
data = {
"name": name,
"location": "M",
"headXfo": Xfo(Vec3(0.0, 1.67, 1.75)),
"headCtrlCrvData": self.headCtrl.getCurveData(),
"jawPosition": Vec3(0.0, 1.2787, 2.0078),
"jawCtrlCrvData": self.jawCtrl.getCurveData(),
}
self.loadData(data)
Profiler.getInstance().pop()
# =============
# Data Methods
# =============
def saveData(self):
"""Save the data for the component to be persisted.
Return:
The JSON data object
"""
data = super(FabriceHeadGuide, self).saveData()
data['headXfo'] = self.headCtrl.xfo
data['headCtrlCrvData'] = self.headCtrl.getCurveData()
data['jawPosition'] = self.jawCtrl.xfo.tr
data['jawCtrlCrvData'] = self.jawCtrl.getCurveData()
return data
def loadData(self, data):
"""Load a saved guide representation from persisted data.
Arguments:
data -- object, The JSON data object.
Return:
True if successful.
"""
super(FabriceHeadGuide, self).loadData(data)
self.headCtrl.xfo = data['headXfo']
self.headCtrl.setCurveData(data['headCtrlCrvData'])
self.jawCtrl.xfo.tr = data['jawPosition']
self.jawCtrl.setCurveData(data['jawCtrlCrvData'])
return True
def getRigBuildData(self):
"""Returns the Guide data used by the Rig Component to define the layout of the final rig..
Return:
The JSON rig data object.
"""
data = super(FabriceHeadGuide, self).getRigBuildData()
data['headXfo'] = self.headCtrl.xfo
data['headCtrlCrvData'] = self.headCtrl.getCurveData()
data['jawPosition'] = self.jawCtrl.xfo.tr
data['jawCtrlCrvData'] = self.jawCtrl.getCurveData()
return data
# ==============
# Class Methods
# ==============
@classmethod
def getComponentType(cls):
"""Enables introspection of the class prior to construction to determine if it is a guide component.
Return:
The true if this component is a guide component.
"""
return 'Guide'
@classmethod
def getRigComponentClass(cls):
"""Returns the corresponding rig component class for this guide component class
Return:
The rig component class.
"""
return FabriceHeadRig
from kraken import plugins
from kraken.core.maths import *
from kraken.core.objects.control import Control
builder = plugins.getBuilder()
config = builder.getConfig()
config.setExplicitNaming(True)
nullControl = Control("NullControl", shape="null")
nullControl.rotatePoints(0, 45, 0)
nullControl.scalePoints(Vec3(3, 3, 3))
nullControl.translatePoints(Vec3(0, 1, 0.25))
nullControl.xfo.tr = Vec3(0.0, 1.0, 0.0)
builder.build(nullControl)
class FabriceTailGuide(FabriceTail):
"""Fabrice Tail Component Guide"""
def __init__(self, name='tail', parent=None):
Profiler.getInstance().push("Construct Fabrice Tail Guide Component:" + name)
super(FabriceTailGuide, self).__init__(name, parent)
# =========
# Controls
# ========
guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self)
self.numDeformersAttr = IntegerAttribute('numDeformers', value=1, minValue=0, maxValue=20, parent=guideSettingsAttrGrp)
self.numDeformersAttr.setValueChangeCallback(self.updateNumDeformers)
# Guide Controls
self.tailBaseCtrl = Control('tailBase', parent=self.ctrlCmpGrp, shape='sphere')
self.tailBaseCtrl.scalePoints(Vec3(1.2, 1.2, 1.2))
self.tailBaseCtrl.lockScale(x=True, y=True, z=True)
self.tailBaseCtrl.setColor("turqoise")
self.tailBaseHandleCtrl = Control('tailBaseHandle', parent=self.ctrlCmpGrp, shape='pin')
self.tailBaseHandleCtrl.rotatePoints(90, 0, 0)
self.tailBaseHandleCtrl.translatePoints(Vec3(0, 1.0, 0))
self.tailBaseHandleCtrl.lockScale(x=True, y=True, z=True)
self.tailBaseHandleCtrl.setColor("turqoise")
self.tailEndHandleCtrl = Control('tailEndHandle', parent=self.ctrlCmpGrp, shape='pin')
self.tailEndHandleCtrl.rotatePoints(90, 0, 0)
self.tailEndHandleCtrl.translatePoints(Vec3(0, 1.0, 0))
self.tailEndHandleCtrl.lockScale(x=True, y=True, z=True)
self.tailEndHandleCtrl.setColor("turqoise")
self.tailEndCtrl = Control('tailEnd', parent=self.ctrlCmpGrp, shape='pin')
self.tailEndCtrl.rotatePoints(90, 0, 0)
self.tailEndCtrl.translatePoints(Vec3(0, 1.0, 0))
self.tailEndCtrl.lockScale(x=True, y=True, z=True)
self.tailEndCtrl.setColor("turqoise")
# ===============
# Add Splice Ops
# ===============
# Add Tail Splice Op
self.bezierSpineKLOp = KLOperator('spineGuideKLOp', 'BezierSpineSolver', 'Kraken')
self.bezierSpineKLOp.setOutput('outputs', self.tailVertebraeOutput.getTarget())
self.addOperator(self.bezierSpineKLOp)
# Add Att Inputs
self.bezierSpineKLOp.setInput('drawDebug', self.drawDebugInputAttr)
self.bezierSpineKLOp.setInput('rigScale', self.rigScaleInputAttr)
self.bezierSpineKLOp.setInput('length', self.lengthInputAttr)
# Add Xfo Inputs
self.bezierSpineKLOp.setInput('base', self.tailBaseCtrl)
self.bezierSpineKLOp.setInput('baseHandle', self.tailBaseHandleCtrl)
self.bezierSpineKLOp.setInput('tipHandle', self.tailEndHandleCtrl)
self.bezierSpineKLOp.setInput('tip', self.tailEndCtrl)
self.loadData({
'name': name,
'location': 'M',
'tailBasePos': Vec3(0.0, 0.65, -3.1),
'tailBaseHandlePos': Vec3(0.0, 0.157, -4.7),
'tailBaseHandleCtrlCrvData': self.tailBaseHandleCtrl.getCurveData(),
'tailEndHandlePos': Vec3(0.0, 0.0625, -6.165),
'tailEndHandleCtrlCrvData': self.tailEndHandleCtrl.getCurveData(),
'tailEndPos': Vec3(0.0, -0.22, -7.42),
'tailEndCtrlCrvData': self.tailEndCtrl.getCurveData(),
'numDeformers': 6
})
Profiler.getInstance().pop()
# ==========
# Callbacks
# ==========
def updateNumDeformers(self, count):
"""Generate the guide controls for the variable outputes array.
Arguments:
count -- object, The number of joints inthe chain.
Return:
True if successful.
"""
if count == 0:
raise IndexError("'count' must be > 0")
vertebraeOutputs = self.tailVertebraeOutput.getTarget()
if count > len(vertebraeOutputs):
for i in xrange(len(vertebraeOutputs), count):
debugCtrl = Control('spine' + str(i+1).zfill(2), parent=self.outputHrcGrp, shape="vertebra")
debugCtrl.rotatePoints(0, -90, 0)
debugCtrl.scalePoints(Vec3(0.5, 0.5, 0.5))
debugCtrl.setColor('turqoise')
vertebraeOutputs.append(debugCtrl)
elif count < len(vertebraeOutputs):
numExtraCtrls = len(vertebraeOutputs) - count
for i in xrange(numExtraCtrls):
extraCtrl = vertebraeOutputs.pop()
self.outputHrcGrp.removeChild(extraCtrl)
return True
# =============
# Data Methods
# =============
def saveData(self):
"""Save the data for the component to be persisted.
Return:
The JSON data object
"""
data = super(FabriceTailGuide, self).saveData()
data['tailBasePos'] = self.tailBaseCtrl.xfo.tr
data['tailBaseHandlePos'] = self.tailBaseHandleCtrl.xfo.tr
data['tailBaseHandleCtrlCrvData'] = self.tailBaseHandleCtrl.getCurveData()
data['tailEndHandlePos'] = self.tailEndHandleCtrl.xfo.tr
data['tailEndHandleCtrlCrvData'] = self.tailEndHandleCtrl.getCurveData()
data['tailEndPos'] = self.tailEndCtrl.xfo.tr
data['tailEndCtrlCrvData'] = self.tailEndCtrl.getCurveData()
data['numDeformers'] = self.numDeformersAttr.getValue()
return data
def loadData(self, data):
"""Load a saved guide representation from persisted data.
Arguments:
data -- object, The JSON data object.
Return:
True if successful.
"""
super(FabriceTailGuide, self).loadData( data )
self.tailBaseCtrl.xfo.tr = data["tailBasePos"]
self.tailBaseHandleCtrl.xfo.tr = data["tailBaseHandlePos"]
self.tailBaseHandleCtrl.setCurveData(data['tailBaseHandleCtrlCrvData'])
self.tailEndHandleCtrl.xfo.tr = data["tailEndHandlePos"]
self.tailEndHandleCtrl.setCurveData(data['tailEndHandleCtrlCrvData'])
self.tailEndCtrl.xfo.tr = data["tailEndPos"]
self.tailEndCtrl.setCurveData(data['tailEndCtrlCrvData'])
self.numDeformersAttr.setValue(data["numDeformers"])
length = data["tailBasePos"].distanceTo(data["tailBaseHandlePos"]) + data["tailBaseHandlePos"].distanceTo(data["tailEndHandlePos"]) + data["tailEndHandlePos"].distanceTo(data["tailEndPos"])
self.lengthInputAttr.setMax(length * 3.0)
self.lengthInputAttr.setValue(length)
self.bezierSpineKLOp.evaluate()
return True
def getRigBuildData(self):
"""Returns the Guide data used by the Rig Component to define the layout of the final rig.
Return:
The JSON rig data object.
"""
data = super(FabriceTailGuide, self).getRigBuildData()
data['tailBasePos'] = self.tailBaseCtrl.xfo.tr
data['tailBaseHandlePos'] = self.tailBaseHandleCtrl.xfo.tr
data['tailBaseHandleCtrlCrvData'] = self.tailBaseHandleCtrl.getCurveData()
data['tailEndHandlePos'] = self.tailEndHandleCtrl.xfo.tr
data['tailEndHandleCtrlCrvData'] = self.tailEndHandleCtrl.getCurveData()
data['tailEndPos'] = self.tailEndCtrl.xfo.tr
data['tailEndCtrlCrvData'] = self.tailEndCtrl.getCurveData()
data['numDeformers'] = self.numDeformersAttr.getValue()
return data
# ==============
# Class Methods
# ==============
@classmethod
def getComponentType(cls):
"""Enables introspection of the class prior to construction to determine if it is a guide component.
Return:
The true if this component is a guide component.
"""
return 'Guide'
@classmethod
def getRigComponentClass(cls):
"""Returns the corresponding rig component class for this guide component class
Return:
The rig component class.
"""
return FabriceTailRig
from kraken import plugins
from kraken.core.maths import *
from kraken.core.objects.control import Control
builder = plugins.getBuilder()
config = builder.getConfig()
config.setExplicitNaming(True)
nullControl = Control("NullControl", shape="null")
nullControl.rotatePoints(0, 45, 0)
nullControl.scalePoints(Vec3(3, 3, 3))
nullControl.translatePoints(Vec3(0, 1, 0.25))
nullControl.xfo.tr = Vec3(0.0, 1.0, 0.0)
builder.build(nullControl)
