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()
def __init__(self, name='neck', parent=None, data=None): Profiler.getInstance().push("Construct Neck Component:" + name) super(NeckComponentGuide, self).__init__(name, parent) # ========= # Controls # ======== guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self) # Guide Controls self.neckCtrl = Control('neck', parent=self.ctrlCmpGrp, shape="sphere") self.neckEndCtrl = Control('neckEnd', parent=self.ctrlCmpGrp, shape="sphere") if data is None: data = { "name": name, "location": "M", "neckPosition": Vec3(0.0, 16.5572, -0.6915), "neckEndPosition": Vec3(0.0, 17.4756, -0.421) } self.loadData(data) Profiler.getInstance().pop()
def __init__(self, name='clavicle', parent=None, data=None): Profiler.getInstance().push("Construct Clavicle Guide Component:" + name) super(ClavicleComponentGuide, self).__init__(name, parent) # ========= # Controls # ========= # Guide Controls guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self) self.clavicleCtrl = Control('clavicle', parent=self.ctrlCmpGrp, shape="sphere") self.clavicleUpVCtrl = Control('clavicleUpV', parent=self.ctrlCmpGrp, shape="triangle") self.clavicleUpVCtrl.setColor('red') self.clavicleEndCtrl = Control('clavicleEnd', parent=self.ctrlCmpGrp, shape="sphere") if data is None: data = { "name": name, "location": "L", "clavicleXfo": Xfo(Vec3(0.1322, 15.403, -0.5723)), "clavicleUpVXfo": Xfo(Vec3(0.0, 1.0, 0.0)), "clavicleEndXfo": Xfo(Vec3(2.27, 15.295, -0.753)) } self.loadData(data) Profiler.getInstance().pop()
def __init__(self, name='clavicle', parent=None): Profiler.getInstance().push("Construct Clavicle Guide Component:" + name) super(FabriceClavicleGuide, self).__init__(name, parent) # ========= # Controls # ========= # Guide Controls guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self) self.clavicleCtrl = Control('clavicle', parent=self.ctrlCmpGrp, shape="cube") self.clavicleCtrl.alignOnXAxis() self.clavicleCtrl.scalePoints(Vec3(1.0, 0.25, 0.25)) data = { "name": name, "location": "L", "clavicleXfo": Xfo(Vec3(0.1322, 15.403, -0.5723)), 'clavicleCtrlCrvData': self.clavicleCtrl.getCurveData() } self.loadData(data) Profiler.getInstance().pop()
def __init__(self, name='limb', parent=None, *args, **kwargs): Profiler.getInstance().push("Construct StretchyLimb Guide Component:" + name) super(StretchyLimbComponentGuide, self).__init__(name, parent, *args, **kwargs) # ========= # Controls # ======== guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self) # Guide Controls self.upperCtl = Control('upper', parent=self.ctrlCmpGrp, shape="sphere") self.lowerCtl = Control('lower', parent=self.ctrlCmpGrp, shape="sphere") self.endCtl = Control('end', parent=self.ctrlCmpGrp, shape="sphere") self.default_data = { "name": name, "location": "L", "upperXfo": Xfo(Vec3(0.9811, 9.769, -0.4572)), "lowerXfo": Xfo(Vec3(1.4488, 5.4418, -0.5348)), "endXfo": Xfo(Vec3(1.841, 1.1516, -1.237)) } self.loadData(self.default_data) Profiler.getInstance().pop()
def updateNumJointControls(self, numSprings): """Load a saved guide representation from persisted data. Arguments: numSprings -- object, The number of joints inthe chain. Return: True if successful. """ if numSprings == 0: raise IndexError("'numSprings' must be > 0") if numSprings > len(self.springCtrls): for i in xrange(len(self.springCtrls), numSprings): newCtrl = Control('spring' + str(i + 1).zfill(2), parent=self.ctrlCmpGrp, shape="squarePointed") newCtrl.alignOnXAxis() self.springCtrls.append(newCtrl) elif numSprings < len(self.springCtrls): numExtraCtrls = len(self.springCtrls) - numSprings for i in xrange(numExtraCtrls): extraCtrl = self.springCtrls.pop() extraCtrl.getParent().removeChild(extraCtrl) # Reset the control positions based on new number of joints springXfos = self.generateGuideXfos(numSprings) for i in xrange(len(self.springCtrls)): self.springCtrls[i].xfo = springXfos[i] return True
def __init__(self, name='leg', parent=None, data=None): Profiler.getInstance().push("Construct Leg Guide Component:" + name) super(LegComponentGuide, self).__init__(name, parent) # ========= # Controls # ======== guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self) # Guide Controls self.femurCtrl = Control('femur', parent=self.ctrlCmpGrp, shape="sphere") self.kneeCtrl = Control('knee', parent=self.ctrlCmpGrp, shape="sphere") self.ankleCtrl = Control('ankle', parent=self.ctrlCmpGrp, shape="sphere") self.toeCtrl = Control('toe', parent=self.ctrlCmpGrp, shape="sphere") self.toeTipCtrl = Control('toeTip', parent=self.ctrlCmpGrp, shape="sphere") if data is None: data = { "name": name, "location": "L", "femurXfo": Xfo(Vec3(0.9811, 9.769, -0.4572)), "kneeXfo": Xfo(Vec3(1.4488, 5.4418, -0.5348)), "ankleXfo": Xfo(Vec3(1.841, 1.1516, -1.237)), "toeXfo": Xfo(Vec3(1.85, 0.4, 0.25)), "toeTipXfo": Xfo(Vec3(1.85, 0.4, 1.5)) } self.loadData(data) Profiler.getInstance().pop()
def __init__(self, name='head', parent=None, data=None): Profiler.getInstance().push("Construct Head Guide Component:" + name) super(HeadComponentGuide, self).__init__(name, parent) # ========= # Controls # ========= guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self) self.headCtrl = Control('head', parent=self.ctrlCmpGrp, shape="cube") self.headEndCtrl = Control('headEnd', parent=self.ctrlCmpGrp, shape="sphere") self.eyeLeftCtrl = Control('eyeLeft', parent=self.ctrlCmpGrp, shape="sphere") self.eyeRightCtrl = Control('eyeRight', parent=self.ctrlCmpGrp, shape="sphere") self.jawCtrl = Control('jaw', parent=self.ctrlCmpGrp, shape="cube") if data is None: data = { "name": name, "location": "M", "headPosition": Vec3(0.0, 17.4756, -0.421), "headEndPosition": Vec3(0.0, 19.5, -0.421), "eyeLeftPosition": Vec3(0.3497, 18.0878, 0.6088), "eyeRightPosition": Vec3(-0.3497, 18.0878, 0.6088), "jawPosition": Vec3(0.0, 17.613, -0.2731) } self.loadData(data) 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 __init__(self, name='mainSrt', parent=None): Profiler.getInstance().push("Construct MainSrt Guide Component:" + name) super(MainSrtComponentGuide, self).__init__(name, parent) # ========= # Attributes # ========= # Add Component Params to IK control guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self) self.mainSrtSizeInputAttr = ScalarAttribute('mainSrtSize', value=5.0, minValue=1.0, maxValue=50.0, parent=guideSettingsAttrGrp) # ========= # Controls # ========= # Guide Controls self.mainSrtCtrl = Control('mainSrt', parent=self.ctrlCmpGrp, shape="circle") data = { "location": 'M', "mainSrtSize": self.mainSrtSizeInputAttr.getValue(), "mainSrtXfo": Xfo(tr=Vec3(0.0, 0.0, 0.0)) } self.loadData(data) Profiler.getInstance().pop()
def __init__(self, name='Clavicle', parent=None): Profiler.getInstance().push("Construct Clavicle Rig Component:" + name) super(FabriceClavicleRig, self).__init__(name, parent) # ========= # Controls # ========= # Clavicle self.clavicleCtrlSpace = CtrlSpace('clavicle', parent=self.ctrlCmpGrp) self.clavicleCtrl = Control('clavicle', parent=self.clavicleCtrlSpace, shape="cube") self.clavicleCtrl.alignOnXAxis() # ========== # Deformers # ========== deformersLayer = self.getOrCreateLayer('deformers') defCmpGrp = ComponentGroup(self.getName(), self, parent=deformersLayer) self.ctrlCmpGrp.setComponent(self) self.clavicleDef = Joint('clavicle', parent=defCmpGrp) self.clavicleDef.setComponent(self) # ============== # Constrain I/O # ============== # Constraint inputs clavicleInputConstraint = PoseConstraint('_'.join([self.clavicleCtrl.getName(), 'To', self.spineEndInputTgt.getName()])) clavicleInputConstraint.setMaintainOffset(True) clavicleInputConstraint.addConstrainer(self.spineEndInputTgt) self.clavicleCtrlSpace.addConstraint(clavicleInputConstraint) # Constraint outputs clavicleConstraint = PoseConstraint('_'.join([self.clavicleOutputTgt.getName(), 'To', self.clavicleCtrl.getName()])) clavicleConstraint.addConstrainer(self.clavicleCtrl) self.clavicleOutputTgt.addConstraint(clavicleConstraint) # =============== # Add Splice Ops # =============== # Add Deformer Splice Op spliceOp = KLOperator('clavicleDeformerKLOp', 'PoseConstraintSolver', 'Kraken') self.addOperator(spliceOp) # Add Att Inputs spliceOp.setInput('drawDebug', self.drawDebugInputAttr) spliceOp.setInput('rigScale', self.rigScaleInputAttr) # Add Xfo Inputs spliceOp.setInput('constrainer', self.clavicleOutputTgt) # Add Xfo Outputs spliceOp.setOutput('constrainee', self.clavicleDef) Profiler.getInstance().pop()
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 __init__(self, name='leg', parent=None, *args, **kwargs): Profiler.getInstance().push("Construct Leg Guide Component:" + name) super(LegComponentGuide, self).__init__(name, parent, *args, **kwargs) # ========= # Controls # ========= guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self) # Guide Controls self.femurCtrl = Control('femur', parent=self.ctrlCmpGrp, shape="sphere") self.kneeCtrl = Control('knee', parent=self.ctrlCmpGrp, shape="sphere") self.ankleCtrl = Control('ankle', parent=self.ctrlCmpGrp, shape="sphere") armGuideSettingsAttrGrp = AttributeGroup("DisplayInfo_ArmSettings", parent=self.femurCtrl) self.armGuideDebugAttr = BoolAttribute('drawDebug', value=True, parent=armGuideSettingsAttrGrp) self.guideOpHost = Transform('guideOpHost', self.ctrlCmpGrp) # Guide Operator self.legGuideKLOp = KLOperator('guide', 'TwoBoneIKGuideSolver', 'Kraken') self.addOperator(self.legGuideKLOp) # Add Att Inputs self.legGuideKLOp.setInput('drawDebug', self.armGuideDebugAttr) self.legGuideKLOp.setInput('rigScale', self.rigScaleInputAttr) # Add Source Inputs self.legGuideKLOp.setInput('root', self.femurCtrl) self.legGuideKLOp.setInput('mid', self.kneeCtrl) self.legGuideKLOp.setInput('end', self.ankleCtrl) # Add Target Outputs self.legGuideKLOp.setOutput('guideOpHost', self.guideOpHost) self.default_data = { "name": name, "location": "L", "createIKHandle": False, "femurXfo": Xfo(Vec3(1.0, 9.75, -0.5)), "kneeXfo": Xfo(Vec3(1.5, 5.5, -0.5)), "ankleXfo": Xfo(Vec3(1.75, 1.15, -1.25)) } self.loadData(self.default_data) Profiler.getInstance().pop()
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()
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 __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()
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()
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='spine', parent=None, *args, **kwargs): Profiler.getInstance().push("Construct Spine Guide Component:" + name) super(SpineComponentGuide, self).__init__(name, parent, *args, **kwargs) # ========= # Controls # ======== guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self) self.numDeformersAttr = IntegerAttribute('numDeformers', value=1, minValue=0, maxValue=20, parent=guideSettingsAttrGrp) # Guide Controls self.cog = Control('cogPosition', parent=self.ctrlCmpGrp, shape="sphere") self.cog.scalePoints(Vec3(1.2, 1.2, 1.2)) self.cog.setColor('red') self.spine01Ctrl = Control('spine01Position', parent=self.ctrlCmpGrp, shape='sphere') self.spine02Ctrl = Control('spine02Position', parent=self.ctrlCmpGrp, shape='sphere') self.spine03Ctrl = Control('spine03Position', parent=self.ctrlCmpGrp, shape='sphere') self.spine04Ctrl = Control('spine04Position', parent=self.ctrlCmpGrp, shape='sphere') data = { 'name': name, 'location': 'M', 'cogPosition': Vec3(0.0, 11.1351, -0.1382), 'spine01Position': Vec3(0.0, 11.1351, -0.1382), 'spine02Position': Vec3(0.0, 11.8013, -0.1995), 'spine03Position': Vec3(0.0, 12.4496, -0.3649), 'spine04Position': Vec3(0.0, 13.1051, -0.4821), 'numDeformers': 6 } self.loadData(data) Profiler.getInstance().pop()
def __init__(self, name='InsectLeg', parent=None, data=None): Profiler.getInstance().push("Construct InsectLeg Guide Component:" + name) super(InsectLegComponentGuide, self).__init__(name, parent) # ========= # Controls # ========= guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self) self.numJoints = IntegerAttribute('numJoints', value=5, minValue=2, maxValue=20, parent=guideSettingsAttrGrp) self.numJoints.setValueChangeCallback(self.updateNumLegControls) self.jointCtrls = [] if data is None: numJoints = self.numJoints.getValue() jointPositions = self.generateGuidePositions(numJoints) for i in xrange(numJoints): self.jointCtrls.append(Control('leg' + str(i + 1).zfill(2), parent=self.ctrlCmpGrp, shape="sphere")) data = { "location": "L", "jointPositions": jointPositions, "numJoints": self.numJoints.getValue() } self.loadData(data) Profiler.getInstance().pop()
def __init__(self, name="mainSrt", parent=None, data=None): Profiler.getInstance().push("Construct MainSrt Guide Component:" + name) super(MainSrtComponentGuide, self).__init__(name, parent) # ========= # Attributes # ========= # Add Component Params to IK control guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self) self.mainSrtSizeInputAttr = ScalarAttribute( "mainSrtSize", value=5.0, minValue=1.0, maxValue=50.0, parent=guideSettingsAttrGrp ) # ========= # Controls # ========= # Guide Controls self.mainSrtCtrl = Control("mainSrt", parent=self.ctrlCmpGrp, shape="circle") if data is None: data = { "location": "M", "mainSrtSize": self.mainSrtSizeInputAttr.getValue(), "mainSrtXfo": Xfo(tr=Vec3(0.0, 0.0, 0.0)), } self.loadData(data) Profiler.getInstance().pop()
def updateNumLegControls(self, numJoints): """Generate the guide controls for the variable outputes array. Arguments: numJoints -- object, The number of joints in the chain. Return: True if successful. """ if numJoints == 0: raise IndexError("'numJoints' must be > 0") if numJoints + 1 > len(self.jointCtrls): for i in xrange(len(self.jointCtrls), numJoints + 1): newCtrl = Control('leg' + str(i + 1).zfill(2), parent=self.ctrlCmpGrp, shape="sphere") self.jointCtrls.append(newCtrl) elif numJoints + 1 < len(self.jointCtrls): numExtraCtrls = len(self.jointCtrls) - (numJoints + 1) for i in xrange(numExtraCtrls): extraCtrl = self.jointCtrls.pop() self.ctrlCmpGrp.removeChild(extraCtrl) # Reset the control positions based on new number of joints jointPositions = self.generateGuidePositions(numJoints) for i in xrange(len(self.jointCtrls)): self.jointCtrls[i].xfo.tr = jointPositions[i] return True
def __init__(self, name="arm", parent=None): Profiler.getInstance().push("Construct Arm Guide Component:" + name) super(ArmComponentGuide, self).__init__(name, parent) # =========== # Attributes # =========== # Add Component Params to IK control guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self) self.bicepFKCtrlSizeInputAttr = ScalarAttribute( "bicepFKCtrlSize", value=1.75, minValue=0.0, maxValue=10.0, parent=guideSettingsAttrGrp ) self.forearmFKCtrlSizeInputAttr = ScalarAttribute( "forearmFKCtrlSize", value=1.5, minValue=0.0, maxValue=10.0, parent=guideSettingsAttrGrp ) # ========= # Controls # ========= # Guide Controls self.bicepCtrl = Control("bicepFK", parent=self.ctrlCmpGrp, shape="sphere") self.bicepCtrl.setColor("blue") self.forearmCtrl = Control("forearmFK", parent=self.ctrlCmpGrp, shape="sphere") self.forearmCtrl.setColor("blue") self.wristCtrl = Control("wristFK", parent=self.ctrlCmpGrp, shape="sphere") self.wristCtrl.setColor("blue") self.handCtrl = Control("hand", parent=self.ctrlCmpGrp, shape="cube") self.handCtrl.setColor("blue") data = { "name": name, "location": "L", "bicepXfo": Xfo(Vec3(2.27, 15.295, -0.753)), "forearmXfo": Xfo(Vec3(5.039, 13.56, -0.859)), "wristXfo": Xfo(Vec3(7.1886, 12.2819, 0.4906)), "handXfo": Xfo(tr=Vec3(7.1886, 12.2819, 0.4906), ori=Quat(Vec3(-0.0865, -0.2301, -0.2623), 0.9331)), "bicepFKCtrlSize": 1.75, "forearmFKCtrlSize": 1.5, } self.loadData(data) 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() boneFKCtrl.lockScale(x=True, y=True, z=True) boneFKCtrl.lockTranslation(x=True, y=True, z=True) 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 __init__(self, name='FKChain', parent=None, data=None): Profiler.getInstance().push("Construct FKCHain Guide Component:" + name) super(FKChainComponentGuide, self).__init__(name, parent) # ========= # Controls # ========= guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self) self.numJoints = IntegerAttribute('numJoints', value=4, minValue=1, maxValue=20, parent=guideSettingsAttrGrp) self.numJoints.setValueChangeCallback(self.updateNumJointControls) self.jointCtrls = [] if data is None: numJoints = self.numJoints.getValue() jointPositions = self.generateGuidePositions(numJoints) for i in xrange(numJoints + 1): if i == 0: ctrlParent = self.ctrlCmpGrp else: ctrlParent = self.jointCtrls[i - 1] newCtrl = Control('chain' + str(i + 1).zfill(2), parent=ctrlParent, shape="sphere") newCtrl.scalePoints(Vec3(0.25, 0.25, 0.25)) self.jointCtrls.append(newCtrl) data = { "location": "L", "jointPositions": jointPositions, "numJoints": self.numJoints.getValue() } self.loadData(data) Profiler.getInstance().pop()
def updateNumJointControls(self, numJoints): """Load a saved guide representation from persisted data. Arguments: numJoints -- object, The number of joints inthe chain. Return: True if successful. """ if numJoints == 0: raise IndexError("'numJoints' must be > 0") if numJoints + 1 > len(self.jointCtrls): for i in xrange(len(self.jointCtrls), numJoints + 1): if i == 0: ctrlParent = self.ctrlCmpGrp else: ctrlParent = self.jointCtrls[i - 1] newCtrl = Control('chain' + str(i + 1).zfill(2), parent=ctrlParent, shape="sphere") newCtrl.scalePoints(Vec3(0.25, 0.25, 0.25)) # Generate thew new ctrl off the end of the existing one. newCtrl.xfo = self.jointCtrls[i - 1].xfo.multiply( Xfo(Vec3(10.0, 0.0, 0.0))) self.jointCtrls.append(newCtrl) elif numJoints + 1 < len(self.jointCtrls): numExtraCtrls = len(self.jointCtrls) - (numJoints + 1) for i in xrange(numExtraCtrls): extraCtrl = self.jointCtrls.pop() extraCtrl.getParent().removeChild(extraCtrl) # Reset the control positions based on new number of joints jointPositions = self.generateGuidePositions(numJoints) for i in xrange(len(self.jointCtrls)): self.jointCtrls[i].xfo.tr = jointPositions[i] return True
def updateNumJointControls(self, numJoints): """Load a saved guide representation from persisted data. Arguments: numJoints -- object, The number of joints inthe chain. Return: True if successful. """ if numJoints == 0: raise IndexError("'numJoints' must be > 0") if numJoints + 1 > len(self.jointCtrls): for i in xrange(len(self.jointCtrls), numJoints + 1): if i == 0: ctrlParent = self.ctrlCmpGrp else: ctrlParent = self.jointCtrls[i - 1] newCtrl = Control('chain' + str(i + 1).zfill(2), parent=ctrlParent, shape="sphere") newCtrl.scalePoints(Vec3(0.25, 0.25, 0.25)) # Generate thew new ctrl off the end of the existing one. newCtrl.xfo = self.jointCtrls[i-1].xfo.multiply(Xfo(Vec3(10.0, 0.0, 0.0))) self.jointCtrls.append(newCtrl) elif numJoints + 1 < len(self.jointCtrls): numExtraCtrls = len(self.jointCtrls) - (numJoints + 1) for i in xrange(numExtraCtrls): extraCtrl = self.jointCtrls.pop() extraCtrl.getParent().removeChild(extraCtrl) # Reset the control positions based on new number of joints jointPositions = self.generateGuidePositions(numJoints) for i in xrange(len(self.jointCtrls)): self.jointCtrls[i].xfo.tr = jointPositions[i] return True
def __init__(self, name='MultiSpring', parent=None): Profiler.getInstance().push("Construct FKCHain Guide Component:" + name) super(MultiSpringComponentGuide, self).__init__(name, parent) # ========= # Controls # ========= guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self) self.numSprings = IntegerAttribute('numSprings', value=5, minValue=1, maxValue=100, parent=guideSettingsAttrGrp) self.numSprings.setValueChangeCallback(self.updateNumJointControls) self.springCtrls = [] numSprings = self.numSprings.getValue() springXfos = self.generateGuideXfos(numSprings) for i in xrange(numSprings): newCtrl = Control('spring' + str(i + 1).zfill(2), parent=self.ctrlCmpGrp, shape="squarePointed") newCtrl.alignOnXAxis() self.springCtrls.append(newCtrl) data = { "location": "L", "springXfos": springXfos, "numSprings": self.numSprings.getValue() } self.loadData(data) Profiler.getInstance().pop()
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()
def setNumControls(self, numControls): # Add more controls if numControls > len(self.springCtrlSpaces): for i in xrange(len(self.springCtrlSpaces), numControls): boneName = 'spring' + str(i + 1).zfill(2) boneFKCtrlSpace = CtrlSpace(boneName, parent=self.ctrlCmpGrp) boneFKCtrl = Control(boneName, parent=boneFKCtrlSpace, shape='squarePointed') boneFKCtrl.alignOnXAxis() self.springCtrlSpaces.append(boneFKCtrlSpace) self.springCtrls.append(boneFKCtrl) # Remove extra ctrls elif numControls < len(self.springCtrlSpaces): numExtraCtrls = len(self.springCtrls) - numControls for i in xrange(numExtraCtrls): extraCtrlSpace = self.springCtrlSpaces.pop() extraCtrl = self.springCtrls.pop() extraCtrlSpace.getParent().removeChild(extraCtrlSpace) extraCtrl.getParent().removeChild(extraCtrl)
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 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 __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()
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()
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
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 updateNumControls(self, numJoints): """Load a saved guide representation from persisted data. Arguments: numJoints -- object, The number of joints inthe chain. Return: True if successful. """ if numJoints == 0: raise IndexError("'numJoints' must be > 0") if numJoints + 1 > len(self.jointCtrls): for i in xrange(len(self.jointCtrls), numJoints + 1): newCtrl = Control('tentacle' + str(i + 1).zfill(2), parent=self.ctrlCmpGrp, shape="sphere") self.jointCtrls.append(newCtrl) newOutput = ComponentOutput('tentacle' + str(i + 1).zfill(2), parent=self.outputHrcGrp) self.tentacleOutputs.append(newOutput) elif numJoints + 1 < len(self.jointCtrls): numExtraCtrls = len(self.jointCtrls) - (numJoints + 1) for i in xrange(numExtraCtrls): extraCtrl = self.jointCtrls.pop() self.ctrlCmpGrp.removeChild(extraCtrl) extraOutput = self.tentacleOutputs.pop() self.outputHrcGrp.removeChild(extraOutput) # Reset the control positions based on new number of joints jointPositions = self.generateGuidePositions(numJoints) for i in xrange(len(self.jointCtrls)): self.jointCtrls[i].xfo.tr = jointPositions[i] return True
def __init__(self, name="spine", parent=None): Profiler.getInstance().push("Construct Spine Guide Component:" + name) super(SpineComponentGuide, self).__init__(name, parent) # ========= # Controls # ======== guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self) self.numDeformersAttr = IntegerAttribute( "numDeformers", value=1, minValue=0, maxValue=20, parent=guideSettingsAttrGrp ) # Guide Controls self.cog = Control("cogPosition", parent=self.ctrlCmpGrp, shape="sphere") self.cog.scalePoints(Vec3(1.2, 1.2, 1.2)) self.cog.setColor("red") self.spine01Ctrl = Control("spine01Position", parent=self.ctrlCmpGrp, shape="sphere") self.spine02Ctrl = Control("spine02Position", parent=self.ctrlCmpGrp, shape="sphere") self.spine03Ctrl = Control("spine03Position", parent=self.ctrlCmpGrp, shape="sphere") self.spine04Ctrl = Control("spine04Position", parent=self.ctrlCmpGrp, shape="sphere") self.loadData( { "name": name, "location": "M", "cogPosition": Vec3(0.0, 11.1351, -0.1382), "spine01Position": Vec3(0.0, 11.1351, -0.1382), "spine02Position": Vec3(0.0, 11.8013, -0.1995), "spine03Position": Vec3(0.0, 12.4496, -0.3649), "spine04Position": Vec3(0.0, 13.1051, -0.4821), "numDeformers": 6, } ) Profiler.getInstance().pop()
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()
def __init__(self, name="fabriceTail", parent=None): Profiler.getInstance().push("Construct Tail Rig Component:" + name) super(FabriceTailRig, self).__init__(name, parent) # ========= # Controls # ========= # Tail Base # self.tailBaseCtrlSpace = CtrlSpace('tailBase', parent=self.ctrlCmpGrp) # self.tailBaseCtrl = Control('tailBase', parent=self.tailBaseCtrlSpace, shape="circle") # self.tailBaseCtrl.rotatePoints(90, 0, 0) # self.tailBaseCtrl.scalePoints(Vec3(2.0, 2.0, 2.0)) # self.tailBaseCtrl.setColor("greenBlue") # Tail Base Handle self.tailBaseHandleCtrlSpace = CtrlSpace('tailBaseHandle', parent=self.ctrlCmpGrp) self.tailBaseHandleCtrl = Control('tailBaseHandle', parent=self.tailBaseHandleCtrlSpace, shape="pin") self.tailBaseHandleCtrl.lockScale(x=True, y=True, z=True) self.tailBaseHandleCtrl.setColor("turqoise") # Tail End Handle self.tailEndHandleCtrlSpace = CtrlSpace('tailEndHandle', parent=self.ctrlCmpGrp) self.tailEndHandleCtrl = Control('tailEndHandle', parent=self.tailEndHandleCtrlSpace, shape="pin") self.tailEndHandleCtrl.lockScale(x=True, y=True, z=True) self.tailEndHandleCtrl.setColor("turqoise") # Tail End self.tailEndCtrlSpace = CtrlSpace('tailEnd', parent=self.tailEndHandleCtrl) self.tailEndCtrl = Control('tailEnd', parent=self.tailEndCtrlSpace, shape="pin") self.tailEndCtrl.lockScale(x=True, y=True, z=True) self.tailEndCtrl.setColor("greenBlue") # ========== # Deformers # ========== deformersLayer = self.getOrCreateLayer('deformers') self.defCmpGrp = ComponentGroup(self.getName(), self, parent=deformersLayer) self.deformerJoints = [] self.tailOutputs = [] self.setNumDeformers(1) # ===================== # Create Component I/O # ===================== # Setup component Xfo I/O's self.tailVertebraeOutput.setTarget(self.tailOutputs) # ============== # Constrain I/O # ============== # Constraint inputs self.tailBaseHandleInputConstraint = PoseConstraint('_'.join([self.tailBaseHandleCtrlSpace.getName(), 'To', self.spineEndCtrlInputTgt.getName()])) self.tailBaseHandleInputConstraint.addConstrainer(self.spineEndCtrlInputTgt) self.tailBaseHandleInputConstraint.setMaintainOffset(True) self.tailBaseHandleCtrlSpace.addConstraint(self.tailBaseHandleInputConstraint) self.tailEndHandleInputConstraint = PoseConstraint('_'.join([self.tailEndHandleCtrlSpace.getName(), 'To', self.cogInputTgt.getName()])) self.tailEndHandleInputConstraint.addConstrainer(self.cogInputTgt) self.tailEndHandleInputConstraint.setMaintainOffset(True) self.tailEndHandleCtrlSpace.addConstraint(self.tailEndHandleInputConstraint) # Constraint outputs self.tailBaseOutputConstraint = PoseConstraint('_'.join([self.tailBaseOutputTgt.getName(), 'To', 'spineBase'])) self.tailBaseOutputConstraint.addConstrainer(self.tailOutputs[0]) self.tailBaseOutputTgt.addConstraint(self.tailBaseOutputConstraint) self.tailEndOutputConstraint = PoseConstraint('_'.join([self.tailEndOutputTgt.getName(), 'To', 'spineEnd'])) self.tailEndOutputConstraint.addConstrainer(self.tailOutputs[0]) self.tailEndOutputTgt.addConstraint(self.tailEndOutputConstraint) # =============== # Add Splice Ops # =============== # Add Tail Splice Op self.bezierTailKLOp = KLOperator('tailKLOp', 'BezierSpineSolver', 'Kraken') self.addOperator(self.bezierTailKLOp) # Add Att Inputs self.bezierTailKLOp.setInput('drawDebug', self.drawDebugInputAttr) self.bezierTailKLOp.setInput('rigScale', self.rigScaleInputAttr) self.bezierTailKLOp.setInput('length', self.lengthInputAttr) # Add Xfo Inputs self.bezierTailKLOp.setInput('base', self.spineEndInputTgt) self.bezierTailKLOp.setInput('baseHandle', self.tailBaseHandleCtrl) self.bezierTailKLOp.setInput('tipHandle', self.tailEndHandleCtrl) self.bezierTailKLOp.setInput('tip', self.tailEndCtrl) # Add Xfo Outputs self.bezierTailKLOp.setOutput('outputs', self.tailOutputs) # Add Deformer Splice Op self.deformersToOutputsKLOp = KLOperator('tailDeformerKLOp', 'MultiPoseConstraintSolver', 'Kraken') self.addOperator(self.deformersToOutputsKLOp) # Add Att Inputs self.deformersToOutputsKLOp.setInput('drawDebug', self.drawDebugInputAttr) self.deformersToOutputsKLOp.setInput('rigScale', self.rigScaleInputAttr) # Add Xfo Outputs self.deformersToOutputsKLOp.setInput('constrainers', self.tailOutputs) # Add Xfo Outputs self.deformersToOutputsKLOp.setOutput('constrainees', self.deformerJoints) Profiler.getInstance().pop()
class FabriceTailRig(FabriceTail): """Fabrice Tail Component""" def __init__(self, name="fabriceTail", parent=None): Profiler.getInstance().push("Construct Tail Rig Component:" + name) super(FabriceTailRig, self).__init__(name, parent) # ========= # Controls # ========= # Tail Base # self.tailBaseCtrlSpace = CtrlSpace('tailBase', parent=self.ctrlCmpGrp) # self.tailBaseCtrl = Control('tailBase', parent=self.tailBaseCtrlSpace, shape="circle") # self.tailBaseCtrl.rotatePoints(90, 0, 0) # self.tailBaseCtrl.scalePoints(Vec3(2.0, 2.0, 2.0)) # self.tailBaseCtrl.setColor("greenBlue") # Tail Base Handle self.tailBaseHandleCtrlSpace = CtrlSpace('tailBaseHandle', parent=self.ctrlCmpGrp) self.tailBaseHandleCtrl = Control('tailBaseHandle', parent=self.tailBaseHandleCtrlSpace, shape="pin") self.tailBaseHandleCtrl.lockScale(x=True, y=True, z=True) self.tailBaseHandleCtrl.setColor("turqoise") # Tail End Handle self.tailEndHandleCtrlSpace = CtrlSpace('tailEndHandle', parent=self.ctrlCmpGrp) self.tailEndHandleCtrl = Control('tailEndHandle', parent=self.tailEndHandleCtrlSpace, shape="pin") self.tailEndHandleCtrl.lockScale(x=True, y=True, z=True) self.tailEndHandleCtrl.setColor("turqoise") # Tail End self.tailEndCtrlSpace = CtrlSpace('tailEnd', parent=self.tailEndHandleCtrl) self.tailEndCtrl = Control('tailEnd', parent=self.tailEndCtrlSpace, shape="pin") self.tailEndCtrl.lockScale(x=True, y=True, z=True) self.tailEndCtrl.setColor("greenBlue") # ========== # Deformers # ========== deformersLayer = self.getOrCreateLayer('deformers') self.defCmpGrp = ComponentGroup(self.getName(), self, parent=deformersLayer) self.deformerJoints = [] self.tailOutputs = [] self.setNumDeformers(1) # ===================== # Create Component I/O # ===================== # Setup component Xfo I/O's self.tailVertebraeOutput.setTarget(self.tailOutputs) # ============== # Constrain I/O # ============== # Constraint inputs self.tailBaseHandleInputConstraint = PoseConstraint('_'.join([self.tailBaseHandleCtrlSpace.getName(), 'To', self.spineEndCtrlInputTgt.getName()])) self.tailBaseHandleInputConstraint.addConstrainer(self.spineEndCtrlInputTgt) self.tailBaseHandleInputConstraint.setMaintainOffset(True) self.tailBaseHandleCtrlSpace.addConstraint(self.tailBaseHandleInputConstraint) self.tailEndHandleInputConstraint = PoseConstraint('_'.join([self.tailEndHandleCtrlSpace.getName(), 'To', self.cogInputTgt.getName()])) self.tailEndHandleInputConstraint.addConstrainer(self.cogInputTgt) self.tailEndHandleInputConstraint.setMaintainOffset(True) self.tailEndHandleCtrlSpace.addConstraint(self.tailEndHandleInputConstraint) # Constraint outputs self.tailBaseOutputConstraint = PoseConstraint('_'.join([self.tailBaseOutputTgt.getName(), 'To', 'spineBase'])) self.tailBaseOutputConstraint.addConstrainer(self.tailOutputs[0]) self.tailBaseOutputTgt.addConstraint(self.tailBaseOutputConstraint) self.tailEndOutputConstraint = PoseConstraint('_'.join([self.tailEndOutputTgt.getName(), 'To', 'spineEnd'])) self.tailEndOutputConstraint.addConstrainer(self.tailOutputs[0]) self.tailEndOutputTgt.addConstraint(self.tailEndOutputConstraint) # =============== # Add Splice Ops # =============== # Add Tail Splice Op self.bezierTailKLOp = KLOperator('tailKLOp', 'BezierSpineSolver', 'Kraken') self.addOperator(self.bezierTailKLOp) # Add Att Inputs self.bezierTailKLOp.setInput('drawDebug', self.drawDebugInputAttr) self.bezierTailKLOp.setInput('rigScale', self.rigScaleInputAttr) self.bezierTailKLOp.setInput('length', self.lengthInputAttr) # Add Xfo Inputs self.bezierTailKLOp.setInput('base', self.spineEndInputTgt) self.bezierTailKLOp.setInput('baseHandle', self.tailBaseHandleCtrl) self.bezierTailKLOp.setInput('tipHandle', self.tailEndHandleCtrl) self.bezierTailKLOp.setInput('tip', self.tailEndCtrl) # Add Xfo Outputs self.bezierTailKLOp.setOutput('outputs', self.tailOutputs) # Add Deformer Splice Op self.deformersToOutputsKLOp = KLOperator('tailDeformerKLOp', 'MultiPoseConstraintSolver', 'Kraken') self.addOperator(self.deformersToOutputsKLOp) # Add Att Inputs self.deformersToOutputsKLOp.setInput('drawDebug', self.drawDebugInputAttr) self.deformersToOutputsKLOp.setInput('rigScale', self.rigScaleInputAttr) # Add Xfo Outputs self.deformersToOutputsKLOp.setInput('constrainers', self.tailOutputs) # Add Xfo Outputs self.deformersToOutputsKLOp.setOutput('constrainees', self.deformerJoints) Profiler.getInstance().pop() def setNumDeformers(self, numDeformers): # Add new deformers and outputs for i in xrange(len(self.tailOutputs), numDeformers): name = 'tail' + str(i + 1).zfill(2) tailOutput = ComponentOutput(name, parent=self.outputHrcGrp) self.tailOutputs.append(tailOutput) for i in xrange(len(self.deformerJoints), numDeformers): name = 'tail' + str(i + 1).zfill(2) tailDef = Joint(name, parent=self.defCmpGrp) tailDef.setComponent(self) self.deformerJoints.append(tailDef) 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(FabriceTailRig, self).loadData( data ) tailBasePos = data['tailBasePos'] tailBaseHandlePos = data['tailBaseHandlePos'] tailBaseHandleCtrlCrvData = data['tailBaseHandleCtrlCrvData'] tailEndHandlePos = data['tailEndHandlePos'] tailEndHandleCtrlCrvData = data['tailEndHandleCtrlCrvData'] tailEndPos = data['tailEndPos'] tailEndCtrlCrvData = data['tailEndCtrlCrvData'] numDeformers = data['numDeformers'] # Set Xfos self.spineEndInputTgt.xfo.tr = tailBasePos self.spineEndCtrlInputTgt.xfo.tr = tailBasePos self.tailBaseHandleCtrlSpace.xfo.tr = tailBaseHandlePos self.tailBaseHandleCtrl.xfo.tr = tailBaseHandlePos self.tailBaseHandleCtrl.setCurveData(tailBaseHandleCtrlCrvData) self.tailEndHandleCtrlSpace.xfo.tr = tailEndHandlePos self.tailEndHandleCtrl.xfo.tr = tailEndHandlePos self.tailEndHandleCtrl.setCurveData(tailEndHandleCtrlCrvData) self.tailEndCtrlSpace.xfo.tr = tailEndPos self.tailEndCtrl.xfo.tr = tailEndPos self.tailEndCtrl.setCurveData(tailEndCtrlCrvData) length = tailBasePos.distanceTo(tailBaseHandlePos) + tailBaseHandlePos.distanceTo(tailEndHandlePos) + tailEndHandlePos.distanceTo(tailEndPos) 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.tailEndOutputConstraint.setConstrainer(self.tailOutputs[-1], index=0) # ============ # Set IO Xfos # ============ # ==================== # Evaluate Splice Ops # ==================== # evaluate the spine op so that all the output transforms are updated. self.bezierTailKLOp.evaluate() # evaluate the constraint op so that all the joint transforms are updated. self.deformersToOutputsKLOp.evaluate() # evaluate the constraints to ensure the outputs are now in the correct location. self.tailBaseHandleInputConstraint.evaluate() self.tailBaseOutputConstraint.evaluate() self.tailEndOutputConstraint.evaluate()
class FabriceClavicleRig(FabriceClavicle): """Clavicle Component""" def __init__(self, name='Clavicle', parent=None): Profiler.getInstance().push("Construct Clavicle Rig Component:" + name) super(FabriceClavicleRig, self).__init__(name, parent) # ========= # Controls # ========= # Clavicle self.clavicleCtrlSpace = CtrlSpace('clavicle', parent=self.ctrlCmpGrp) self.clavicleCtrl = Control('clavicle', parent=self.clavicleCtrlSpace, shape="cube") self.clavicleCtrl.alignOnXAxis() # ========== # Deformers # ========== deformersLayer = self.getOrCreateLayer('deformers') defCmpGrp = ComponentGroup(self.getName(), self, parent=deformersLayer) self.ctrlCmpGrp.setComponent(self) self.clavicleDef = Joint('clavicle', parent=defCmpGrp) self.clavicleDef.setComponent(self) # ============== # Constrain I/O # ============== # Constraint inputs clavicleInputConstraint = PoseConstraint('_'.join([self.clavicleCtrl.getName(), 'To', self.spineEndInputTgt.getName()])) clavicleInputConstraint.setMaintainOffset(True) clavicleInputConstraint.addConstrainer(self.spineEndInputTgt) self.clavicleCtrlSpace.addConstraint(clavicleInputConstraint) # Constraint outputs clavicleConstraint = PoseConstraint('_'.join([self.clavicleOutputTgt.getName(), 'To', self.clavicleCtrl.getName()])) clavicleConstraint.addConstrainer(self.clavicleCtrl) self.clavicleOutputTgt.addConstraint(clavicleConstraint) # =============== # Add Splice Ops # =============== # Add Deformer Splice Op spliceOp = KLOperator('clavicleDeformerKLOp', 'PoseConstraintSolver', 'Kraken') self.addOperator(spliceOp) # Add Att Inputs spliceOp.setInput('drawDebug', self.drawDebugInputAttr) spliceOp.setInput('rigScale', self.rigScaleInputAttr) # Add Xfo Inputs spliceOp.setInput('constrainer', self.clavicleOutputTgt) # Add Xfo Outputs spliceOp.setOutput('constrainee', self.clavicleDef) 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(FabriceClavicleRig, self).loadData( data ) self.clavicleCtrlSpace.xfo = data['clavicleXfo'] self.clavicleCtrl.xfo = data['clavicleXfo'] self.clavicleCtrl.setCurveData(data['clavicleCtrlCrvData']) # ============ # Set IO Xfos # ============ self.spineEndInputTgt.xfo = data['clavicleXfo'] self.clavicleOutputTgt.xfo = data['clavicleXfo']
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)
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
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 __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()
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()
def __init__(self, name='spine', parent=None): Profiler.getInstance().push("Construct Fabrice Spine Guide Component:" + name) super(FabriceSpineGuide, 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.cogCtrl = Control('cog', parent=self.ctrlCmpGrp, shape="circle") self.cogCtrl.rotatePoints(90, 0, 0) self.cogCtrl.scalePoints(Vec3(3.0, 3.0, 3.0)) self.cogCtrl.setColor('red') self.spineBaseCtrl = Control('spineBase', parent=self.ctrlCmpGrp, shape='pin') self.spineBaseCtrl.rotatePoints(90, 0, 0) self.spineBaseCtrl.translatePoints(Vec3(0, 1.0, 0)) self.spineBaseHandleCtrl = Control('spineBaseHandle', parent=self.ctrlCmpGrp, shape='pin') self.spineBaseHandleCtrl.rotatePoints(90, 0, 0) self.spineBaseHandleCtrl.translatePoints(Vec3(0, 1.0, 0)) self.spineEndHandleCtrl = Control('spineEndHandle', parent=self.ctrlCmpGrp, shape='pin') self.spineEndHandleCtrl.rotatePoints(90, 0, 0) self.spineEndHandleCtrl.translatePoints(Vec3(0, 1.0, 0)) self.spineEndCtrl = Control('spineEnd', parent=self.ctrlCmpGrp, shape='pin') self.spineEndCtrl.rotatePoints(90, 0, 0) self.spineEndCtrl.translatePoints(Vec3(0, 1.0, 0)) # =============== # Add Splice Ops # =============== # Add Spine Splice Op self.bezierSpineSpliceOp = SpliceOperator('spineGuideSpliceOp', 'BezierSpineSolver', 'Kraken', alwaysEval=True) 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.spineVertebraeOutput.getTarget()) self.loadData({ 'name': name, 'location': 'M', 'cogPos': Vec3(0.0, 1.65, 0.75), 'cogCtrlCrvData': self.cogCtrl.getCurveData(), 'spineBasePos': Vec3(0.0, 1.65, 0.75), 'spineBaseCtrlCrvData': self.spineBaseCtrl.getCurveData(), 'spineBaseHandlePos': Vec3(0.0, 1.6, -0.7), 'spineBaseHandleCtrlCrvData': self.spineBaseHandleCtrl.getCurveData(), 'spineEndHandlePos': Vec3(0.0, 1.15, -2.0), 'spineEndHandleCtrlCrvData': self.spineEndHandleCtrl.getCurveData(), 'spineEndPos': Vec3(0.0, 0.65, -3.1), 'spineEndCtrlCrvData': self.spineEndCtrl.getCurveData(), 'numDeformers': 6 }) Profiler.getInstance().pop()
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()
class FabriceClavicleGuide(FabriceClavicle): """Clavicle Component Guide""" def __init__(self, name='clavicle', parent=None): Profiler.getInstance().push("Construct Clavicle Guide Component:" + name) super(FabriceClavicleGuide, self).__init__(name, parent) # ========= # Controls # ========= # Guide Controls guideSettingsAttrGrp = AttributeGroup("GuideSettings", parent=self) self.clavicleCtrl = Control('clavicle', parent=self.ctrlCmpGrp, shape="cube") self.clavicleCtrl.alignOnXAxis() self.clavicleCtrl.scalePoints(Vec3(1.0, 0.25, 0.25)) data = { "name": name, "location": "L", "clavicleXfo": Xfo(Vec3(0.1322, 15.403, -0.5723)), 'clavicleCtrlCrvData': self.clavicleCtrl.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(FabriceClavicleGuide, self).saveData() data['clavicleXfo'] = self.clavicleCtrl.xfo data['clavicleCtrlCrvData'] = self.clavicleCtrl.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(FabriceClavicleGuide, self).loadData( data ) self.clavicleCtrl.xfo = data['clavicleXfo'] self.clavicleCtrl.setCurveData(data['clavicleCtrlCrvData']) 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(FabriceClavicleGuide, self).getRigBuildData() data['clavicleXfo'] = self.clavicleCtrl.xfo data['clavicleCtrlCrvData'] = self.clavicleCtrl.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 FabriceClavicleRig