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 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.neck01Ctrl = Control('neck01', parent=self.ctrlCmpGrp, shape="pin") self.neck01Ctrl.setColor("orange") self.neck01Ctrl.lockTranslation(True, True, True) self.neck01Ctrl.lockScale(True, True, True) self.neck01CtrlSpace = self.neck01Ctrl.insertCtrlSpace(name='neck01') self.neck02Ctrl = Control('neck02', parent=self.neck01Ctrl, shape="pin") self.neck02Ctrl.setColor("orange") self.neck02Ctrl.lockTranslation(True, True, True) self.neck02Ctrl.lockScale(True, True, True) self.neck02CtrlSpace = self.neck02Ctrl.insertCtrlSpace(name='neck02') # ========== # Deformers # ========== deformersLayer = self.getOrCreateLayer('deformers') self.defCmpGrp = ComponentGroup(self.getName(), self, parent=deformersLayer) self.addItem('defCmpGrp', self.defCmpGrp) self.neck01Def = Joint('neck01', parent=self.defCmpGrp) self.neck01Def.setComponent(self) self.neck02Def = Joint('neck02', parent=self.defCmpGrp) self.neck02Def.setComponent(self) # ============== # Constrain I/O # ============== # Constraint inputs neckInputConstraintName = '_'.join([ self.neck01CtrlSpace.getName(), 'To', self.neckBaseInputTgt.getName() ]) self.neckInputCnstr = self.neck01CtrlSpace.constrainTo( self.neckBaseInputTgt, 'Pose', maintainOffset=True, name=neckInputConstraintName) # Constraint outputs neck01OutCnstrName = '_'.join( [self.neck01OutputTgt.getName(), 'To', self.neck01Ctrl.getName()]) self.neck01OutCnstr = self.neck01OutputTgt.constrainTo( self.neck01Ctrl, 'Pose', maintainOffset=False, name=neck01OutCnstrName) neck02OutCnstrName = '_'.join( [self.neck02OutputTgt.getName(), 'To', self.neck02Ctrl.getName()]) self.neck02OutCnstr = self.neck02OutputTgt.constrainTo( self.neck02Ctrl, 'Pose', maintainOffset=False, name=neck02OutCnstrName) neckEndCnstrName = '_'.join( [self.neckEndOutputTgt.getName(), 'To', self.neck02Ctrl.getName()]) self.neckEndCnstr = self.neckEndOutputTgt.constrainTo( self.neck02Ctrl, 'Pose', maintainOffset=True, name=neckEndCnstrName) # ============== # Add Operators # ============== # Add Deformer KL Op self.neckDeformerKLOp = KLOperator('neckDeformerKLOp', 'MultiPoseConstraintSolver', 'Kraken') self.addOperator(self.neckDeformerKLOp) # Add Att Inputs self.neckDeformerKLOp.setInput('drawDebug', self.drawDebugInputAttr) self.neckDeformerKLOp.setInput('rigScale', self.rigScaleInputAttr) # Add Xfo Inputstrl) self.neckDeformerKLOp.setInput('constrainers', [self.neck01Ctrl, self.neck02Ctrl]) # Add Xfo Outputs self.neckDeformerKLOp.setOutput('constrainees', [self.neck01Def, self.neck02Def]) 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) neckXfo = data.get('neckXfo') neckCrvData = data.get('neckCrvData') neckMidXfo = data.get('neckMidXfo') neckMidCrvData = data.get('neckMidCrvData') neckEndXfo = data.get('neckEndXfo') self.neck01CtrlSpace.xfo = neckXfo self.neck01Ctrl.xfo = neckXfo self.neck01Ctrl.setCurveData(neckCrvData) self.neck02CtrlSpace.xfo = neckMidXfo self.neck02Ctrl.xfo = neckMidXfo self.neck02Ctrl.setCurveData(neckMidCrvData) # ============ # Set IO Xfos # ============ self.neckBaseInputTgt.xfo = neckXfo self.neck01OutputTgt.xfo = neckXfo self.neck02OutputTgt.xfo = neckMidXfo self.neckEndOutputTgt.xfo = neckEndXfo # Evaluate Constraints self.neckInputCnstr.evaluate() self.neck01OutCnstr.evaluate() self.neck02OutCnstr.evaluate() self.neckEndCnstr.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 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 FKCollisionComponentRig(FKCollisionComponent): """FK Collision Rig""" def __init__(self, name='FKCollision', parent=None): Profiler.getInstance().push("Construct FK Collision Rig Component:" + name) super(FKCollisionComponentRig, self).__init__(name, parent) # ========= # Controls # ========= # Ground self.groundCtrl = Control('ground', parent=self.ctrlCmpGrp, shape='square') self.groundCtrl.insertCtrlSpace() groundSettingsAttrGrp = AttributeGroup("DisplayInfo_ChainSettings", parent=self.groundCtrl) self.groundOffsetYInputAttr = ScalarAttribute( 'ground_offsetY', 0.0, minValue=0.0, maxValue=10.0, parent=groundSettingsAttrGrp) # FK self.fkCtrlSpaces = [] self.fkCtrls = [] self.transforms = [] self.setNumControls(4) self.setNumTransforms(4) # Add Component Params to FK control chainSettingsAttrGrp = AttributeGroup("DisplayInfo_ChainSettings", parent=self.fkCtrls[0]) chainDrawDebugInputAttr = BoolAttribute('drawDebug', value=False, parent=chainSettingsAttrGrp) # Connect IO to controls self.drawDebugInputAttr.connect(chainDrawDebugInputAttr) # ========== # 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 # ============== # Constrain inputs self.rootInputConstraint = PoseConstraint('_'.join([ self.fkCtrlSpaces[0].getName(), 'To', self.rootInputTgt.getName() ])) self.rootInputConstraint.setMaintainOffset(True) self.rootInputConstraint.addConstrainer(self.rootInputTgt) self.fkCtrlSpaces[0].addConstraint(self.rootInputConstraint) # =============== # Add Canvas Ops # =============== # Add Output Canvas Op self.collisionCanvasOp = CanvasOperator( 'collisionCanvasOp', 'Kraken.Solvers.CollideChainSolver') self.addOperator(self.collisionCanvasOp) # Add Att Inputs self.collisionCanvasOp.setInput('drawDebug', self.drawDebugInputAttr) self.collisionCanvasOp.setInput('rigScale', self.rigScaleInputAttr) self.collisionCanvasOp.setInput('ground_offsetY', self.groundOffsetYInputAttr) # Add Xfo Inputs self.collisionCanvasOp.setInput('ground', self.groundCtrl) self.collisionCanvasOp.setInput('joints', self.transforms) # Add Xfo Outputs self.collisionCanvasOp.setOutput('outputs', self.boneOutputsTgt) # Add Deformer Canvas Op self.deformersToOutputsKLOp = KLOperator('fkChainDeformerKLOp', 'MultiPoseConstraintSolver', 'Kraken') self.addOperator(self.deformersToOutputsKLOp) # Add Att Inputs self.deformersToOutputsKLOp.setInput('drawDebug', False) self.deformersToOutputsKLOp.setInput('rigScale', self.rigScaleInputAttr) # Add Xfo Inputs self.deformersToOutputsKLOp.setInput('constrainers', self.boneOutputsTgt) # Add Xfo Outputs self.deformersToOutputsKLOp.setOutput('constrainees', self.deformerJoints) Profiler.getInstance().pop() def setNumControls(self, numControls): # Add more controls if numControls > len(self.fkCtrlSpaces): 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' boneFKCtrlSpace = CtrlSpace(boneName, parent=parent) boneFKCtrl = Control(boneName, parent=boneFKCtrlSpace, shape="cube") boneFKCtrl.alignOnXAxis() self.fkCtrlSpaces.append(boneFKCtrlSpace) self.fkCtrls.append(boneFKCtrl) # Remove extra ctrls elif numControls < len(self.fkCtrlSpaces): numExtraCtrls = len(self.fkCtrls) - numControls for i in xrange(numExtraCtrls): extraCtrlSpace = self.fkCtrlSpaces.pop() extraCtrl = self.fkCtrls.pop() extraCtrlSpace.getParent().removeChild(extraCtrlSpace) extraCtrl.getParent().removeChild(extraCtrl) def setNumTransforms(self, numTransforms): numTransforms += 1 # Add more transforms if numTransforms > len(self.transforms): for i in xrange(len(self.transforms), numTransforms): name = 'transform' + str(i + 1).zfill(2) transform = Transform(name, parent=self.outputHrcGrp) self.transforms.append(transform) # Remove extra transforms elif numTransforms < len(self.transforms): numExtraTransforms = len(self.transforms) - numTransforms for i in xrange(numExtraTransforms): extraTransform = self.transforms.pop() extraTransform.getParent().removeChild(extraTransform) return True def setNumDeformers(self, numDeformers): # Add more deformers and outputs if numDeformers > len(self.boneOutputsTgt): for i in xrange(len(self.boneOutputsTgt), numDeformers): name = 'bone' + str(i + 1).zfill(2) defOutput = ComponentOutput(name, parent=self.outputHrcGrp) self.boneOutputsTgt.append(defOutput) boneDef = Joint(name, parent=self.defCmpGrp) boneDef.setComponent(self) self.deformerJoints.append(boneDef) # Remove extra deformers and outputs elif numDeformers < len(self.boneOutputsTgt): numExtraOutputs = len(self.boneOutputsTgt) - numDeformers numExtraDefs = len(self.deformerJoints) - numDeformers for i in xrange(numExtraOutputs): extraOutput = self.boneOutputsTgt.pop() extraDef = self.deformerJoints.pop() extraOutput.getParent().removeChild(extraOutput) extraDef.getParent().removeChild(extraDef) 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(FKCollisionComponentRig, self).loadData(data) jointXfos = data['jointXfos'] boneXfos = data['boneXfos'] boneLengths = data['boneLengths'] numJoints = data['numJoints'] # Add extra controls and outputs self.setNumControls(numJoints) self.setNumTransforms(numJoints) self.setNumDeformers(numJoints) for i in xrange(numJoints): self.fkCtrlSpaces[i].xfo = boneXfos[i] self.fkCtrls[i].xfo = boneXfos[i] self.fkCtrls[i].scalePoints( Vec3(boneLengths[i], boneLengths[i] * 0.45, boneLengths[i] * 0.45)) for i in xrange(len(self.transforms)): self.transforms[i].xfo = jointXfos[i] # ========================== # Create Output Constraints # ========================== # This needs to be done here since the 'numJoints' attribute resizes the # number of controls and outputs self.chainEndXfoOutputConstraint = PoseConstraint('_'.join([ self.chainEndXfoOutputTgt.getName(), 'To', self.boneOutputsTgt[-1].getName() ])) self.chainEndXfoOutputConstraint.setMaintainOffset(True) self.chainEndXfoOutputConstraint.addConstrainer( self.boneOutputsTgt[-1]) self.chainEndXfoOutputTgt.addConstraint( self.chainEndXfoOutputConstraint) self.chainEndPosOutputConstraint = PositionConstraint('_'.join([ self.chainEndPosOutputTgt.getName(), 'To', self.boneOutputsTgt[-1].getName() ])) self.chainEndPosOutputConstraint.setMaintainOffset(True) self.chainEndPosOutputConstraint.addConstrainer( self.boneOutputsTgt[-1]) self.chainEndPosOutputTgt.addConstraint( self.chainEndPosOutputConstraint) self.transformConstraints = [] for i in xrange(len(self.transforms)): if i == len(self.transforms) - 1: constrainer = self.fkCtrls[i - 1] else: constrainer = self.fkCtrls[i] transformConstraint = PoseConstraint('_'.join( [self.transforms[i].getName(), 'To', constrainer.getName()])) transformConstraint.setMaintainOffset(True) transformConstraint.addConstrainer(constrainer) self.transforms[i].addConstraint(transformConstraint) self.transformConstraints.append(transformConstraint) # ============ # Set IO Xfos # ============ self.rootInputTgt.xfo = boneXfos[0] for i in xrange(numJoints): self.boneOutputsTgt[i].xfo = boneXfos[i] self.chainEndXfoOutputTgt.xfo = data['endXfo'] self.chainEndPosOutputTgt.xfo = data['endXfo'] # ============= # Set IO Attrs # ============= # ==================== # Evaluate Canvas Ops # ==================== # Eval Collision Op to evaulate with new outputs and controls self.collisionCanvasOp.evaluate() # evaluate the output splice op to evaluate with new outputs and deformers self.deformersToOutputsKLOp.evaluate() # evaluate the constraints to ensure the outputs are now in the correct location. for constraint in self.transformConstraints: constraint.evaluate() self.rootInputConstraint.evaluate() self.chainEndXfoOutputConstraint.evaluate()
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 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.neck01Ctrl = Control('neck01', parent=self.ctrlCmpGrp, shape="pin") self.neck01Ctrl.setColor("orange") self.neck01Ctrl.lockTranslation(True, True, True) self.neck01Ctrl.lockScale(True, True, True) self.neck01CtrlSpace = self.neck01Ctrl.insertCtrlSpace(name='neck01') self.neck02Ctrl = Control('neck02', parent=self.neck01Ctrl, shape="pin") self.neck02Ctrl.setColor("orange") self.neck02Ctrl.lockTranslation(True, True, True) self.neck02Ctrl.lockScale(True, True, True) self.neck02CtrlSpace = self.neck02Ctrl.insertCtrlSpace(name='neck02') # ========== # Deformers # ========== deformersLayer = self.getOrCreateLayer('deformers') self.defCmpGrp = ComponentGroup(self.getName(), self, parent=deformersLayer) self.addItem('defCmpGrp', self.defCmpGrp) self.neck01Def = Joint('neck01', parent=self.defCmpGrp) self.neck01Def.setComponent(self) self.neck02Def = Joint('neck02', parent=self.defCmpGrp) self.neck02Def.setComponent(self) # ============== # Constrain I/O # ============== # Constraint inputs neckInputConstraintName = '_'.join([self.neck01CtrlSpace.getName(), 'To', self.neckBaseInputTgt.getName()]) self.neckInputCnstr = self.neck01CtrlSpace.constrainTo( self.neckBaseInputTgt, 'Pose', maintainOffset=True, name=neckInputConstraintName) # Constraint outputs neck01OutCnstrName = '_'.join([self.neck01OutputTgt.getName(), 'To', self.neck01Ctrl.getName()]) self.neck01OutCnstr = self.neck01OutputTgt.constrainTo( self.neck01Ctrl, 'Pose', maintainOffset=False, name=neck01OutCnstrName) neck02OutCnstrName = '_'.join([self.neck02OutputTgt.getName(), 'To', self.neck02Ctrl.getName()]) self.neck02OutCnstr = self.neck02OutputTgt.constrainTo( self.neck02Ctrl, 'Pose', maintainOffset=False, name=neck02OutCnstrName) neckEndCnstrName = '_'.join([self.neckEndOutputTgt.getName(), 'To', self.neck02Ctrl.getName()]) self.neckEndCnstr = self.neckEndOutputTgt.constrainTo( self.neck02Ctrl, 'Pose', maintainOffset=True, name=neckEndCnstrName) # ============== # Add Operators # ============== # Add Deformer KL Op self.neckDeformerKLOp = KLOperator('neckDeformerKLOp', 'MultiPoseConstraintSolver', 'Kraken') self.addOperator(self.neckDeformerKLOp) # Add Att Inputs self.neckDeformerKLOp.setInput('drawDebug', self.drawDebugInputAttr) self.neckDeformerKLOp.setInput('rigScale', self.rigScaleInputAttr) # Add Xfo Inputstrl) self.neckDeformerKLOp.setInput('constrainers', [self.neck01Ctrl, self.neck02Ctrl]) # Add Xfo Outputs self.neckDeformerKLOp.setOutput('constrainees', [self.neck01Def, self.neck02Def]) 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) neckXfo = data.get('neckXfo') neckCrvData = data.get('neckCrvData') neckMidXfo = data.get('neckMidXfo') neckMidCrvData = data.get('neckMidCrvData') neckEndXfo = data.get('neckEndXfo') self.neck01CtrlSpace.xfo = neckXfo self.neck01Ctrl.xfo = neckXfo self.neck01Ctrl.setCurveData(neckCrvData) self.neck02CtrlSpace.xfo = neckMidXfo self.neck02Ctrl.xfo = neckMidXfo self.neck02Ctrl.setCurveData(neckMidCrvData) # ============ # Set IO Xfos # ============ self.neckBaseInputTgt.xfo = neckXfo self.neck01OutputTgt.xfo = neckXfo self.neck02OutputTgt.xfo = neckMidXfo self.neckEndOutputTgt.xfo = neckEndXfo # Evaluate Constraints self.neckInputCnstr.evaluate() self.neck01OutCnstr.evaluate() self.neck02OutCnstr.evaluate() self.neckEndCnstr.evaluate()