def create_clusters(self):
for i in range(0, self._curve.numCVs):
pm.select(self._curve.curveNode.cv[i])
tempClstr = DrawNode(Shape.cluster,
self._limbName + "IK_SpineCl_#")
tempClstr.create_axialCor()
self.clusters.append(tempClstr)
def orient_joint_frontAxis(self, joint, aimAxis=[0, 1, 0],
upAxis=[0, 0, 1],
worldUpType="object",
frontAxis="z"):
#TODO : Validate frontAxis
# orient_joint Function OverLoading
# Creates a temporal Trasnform node for WorldUpVector.
# Calls orient_joint method
jointUnder = self.jointUnder(joint)
if jointUnder is None:
return 0
moveAxis = [0, 0]
if frontAxis == "x":
moveAxis[0] = -1
elif frontAxis == "z":
moveAxis[1] = -1
temporalTrans = DrawNode(Shape.transform, 'temporalTransform')
temporalTrans.temp_constrain(jointUnder)
temporalTrans.freeze_transformations()
temporalTransMove = self.orient_choose_direction(joint, jointUnder,
frontAxis)
temporalTrans.move(pm.dt.Vector([moveAxis[0],
(temporalTransMove * 0.001),
moveAxis[1]]))
worldUpVector = temporalTrans.drawnNode
self.orient_joint(joint, aimAxis, upAxis, worldUpType, worldUpVector)
temporalTrans.delete()
def create_clusters(self):
for i in range(0, self._curve.numCVs):
pm.select(self._curve.curveNode.cv[i])
tempClstr = DrawNode(
Shape.cluster,
self._limbName + "IK_SpineCl_#"
)
tempClstr.create_axialCor()
self.clusters.append(tempClstr)
def orient_joint(self, joint, aimAxis=[1, 0, 0], upAxis=[0, 0, 1],
worldUpType="vector",
worldUpVector=[0, 1, 0]):
#joint should be pm.nt.Joint type
if not isinstance(joint, pm.nt.Joint):
raise TypeError("%s sholud be an instance of pm.nt.Joint Class"
% joint)
jointUnder = self.jointUnder(joint)
if jointUnder is None:
return 0
temporalGroup = DrawNode(Shape.transform, 'temporalGroup')
pm.parent(jointUnder, temporalGroup.drawnNode)
pm.setAttr(joint.jointOrient, (0, 0, 0))
if worldUpType == "object":
aimConst = pm.aimConstraint(jointUnder, joint, aimVector=aimAxis,
upVector=upAxis,
worldUpType=worldUpType,
worldUpObject=worldUpVector)
elif worldUpType == "vector":
aimConst = pm.aimConstraint(jointUnder, joint, aimVector=aimAxis,
upVector=upAxis,
worldUpType=worldUpType,
worldUpVector=worldUpVector)
pm.delete(aimConst)
pm.parent(jointUnder, joint)
pm.setAttr(joint.jointOrient, (pm.getAttr(joint.rotate)))
pm.setAttr((joint.rotate), [0, 0, 0])
pm.delete(temporalGroup.drawnNode)
def set_zero_joint(self):
#Removes Zero Joint from Joint Chain
pm.joint(self.jointChain[0], e=True, zso=True, oj='xyz', sao='xup')
self.zeroJoint = self.jointChain[0]
self._zeroPos = pm.dt.Point(pm.getAttr(self._zeroJoint.translate))
self.jointChain.remove(self.jointChain[0])
self.jointPos.remove(self.jointPos[0])
pm.joint(self.jointChain[1], e=True, zso=True, oj='xyz', sao='yup')
for i in range(1, len(self.jointChain)):
pm.joint(self.jointChain[i], e=True, zso=True, oj='xyz', sao='yup')
#sets Start End Num Of Joints again
self._numOfJoints = len(self._jointChain)
#Orient Zero Joint
temporalGroup = DrawNode(Shape.transform, 'temporalGroup')
pm.parent(self.startJoint, temporalGroup.drawnNode)
print (pm.getAttr(self.zeroJoint.jointOrient))
pm.setAttr(self.zeroJoint.jointOrientX, 0)
pm.parent(self.startJoint, self.zeroJoint)
temporalGroup.delete()
class IkSpineLimb(object):
def __init__(self):
self._limbName = None
self._joints = None
self._curve = None
self._ikHandle = None
self._effector = None
self._clusters = []
self._network = None
self._scaleMD = None
self._factors = None
self._hipCtrl = None
self._shoulderCtrl = None
self._COGCtrl = None
self._stuff = []
# *************************************************************************
# IKSPINE BASE SETUP METHODS
def create_spine(self, name_in, curve_in, frontAxis="z"):
#self._network = Network(name_in)
self._limbName = name_in
# You can change createion method with a Joint Chain Class
# JointChain(name_in, jointPositions)
# self.joint.orientChain
self.joints = SpineJoints(name_in, curve_in)
self.joints.orient_spine(frontAxis)
ikSolver = pm.ikHandle(sj=self.joints.startJoint,
ee=self.joints.endJoint,
tws="linear",
cra=True,
pcv=False,
ns=2,
sol="ikSplineSolver",
name=(name_in + "_IKSpine"))
self._ikHandle = pm.rename(ikSolver[0], (name_in + "_IK_Spine"))
self._effector = pm.rename(ikSolver[0],
(name_in + "_IK_SpineEffector"))
self._curve = Curve((name_in + "_IKSpineCurve"), ikSolver[2])
def create_clusters(self):
for i in range(0, self._curve.numCVs):
pm.select(self._curve.curveNode.cv[i])
tempClstr = DrawNode(Shape.cluster,
self._limbName + "IK_SpineCl_#")
tempClstr.create_axialCor()
self.clusters.append(tempClstr)
#self.clusters[i].create_axialCor()
def make_stretchy(self):
#check joints
"""
"""
self._scaleMD = pm.createNode("multiplyDivide",
n=self.limbName + "_scaleMD")
pm.connectAttr(self.curve.curveInfo.arcLength, self.scaleMD.input1X)
pm.setAttr(self.scaleMD.input2X, self.curve.arclen)
pm.setAttr(self.scaleMD.operation, 2)
for jnt in self.joints.jointChain:
factor = pm.createNode("multiplyDivide", n="factor_" + jnt)
pm.connectAttr(self.scaleMD.outputX, factor.input1X)
pm.setAttr(factor.input2X, (pm.getAttr(jnt.ty)))
pm.connectAttr(factor.outputX, jnt.ty)
def create_controllers(self):
#Check if clusters is not an empty list
# Hip Ctrl Create
"""
"""
self._hipCtrl = DrawNode(Shape.ikCtrl, 'hip_ctrl')
self.hipCtrl.temp_constrain(self.clusters[0].drawnNode)
self.hipCtrl.create_axialCor()
#parent Hip Clusters to Hip Control
pm.parent(self.clusters[0].axialCor, self.clusters[1].axialCor,
self.hipCtrl.drawnNode)
# Shoulder Ctrl Create
self._shoulderCtrl = DrawNode(Shape.circle, 'shoulder_ctrl')
self.shoulderCtrl.temp_constrain(self.clusters[(len(self.clusters) -
1)].drawnNode)
self.shoulderCtrl.create_axialCor()
# COG Ctrl Create
self._COGCtrl = DrawNode(Shape.cube, 'COG_ctrl')
self._COGCtrl.temp_constrain(self.hipCtrl.drawnNode)
self._COGCtrl.create_axialCor()
#parent Shoulder Clusters to Shoulder Control
pm.parent(self.clusters[(len(self.clusters) - 1)].axialCor,
self.clusters[(len(self.clusters) - 2)].axialCor,
self._shoulderCtrl.drawnNode)
# Create Mid Cluster Control Transforms and Constrains
mid_cluster = self.clusters[2].drawnNode
tempCluster_const_1 = DrawNode(Shape.transform,
"C_IK_SpineCl_ConstGrp")
tempCluster_const_1.temp_constrain(mid_cluster)
pm.parent(tempCluster_const_1.drawnNode, self.hipCtrl.drawnNode)
tempCluster_const_2 = DrawNode(Shape.transform,
"C_IK_SpineCl_ConstGrp")
tempCluster_const_2.temp_constrain(mid_cluster)
pm.parent(tempCluster_const_2.drawnNode, self.shoulderCtrl.drawnNode)
tempCluster_const_1.constrain(mid_cluster, targetType='targetObj')
tempCluster_const_2.constrain(mid_cluster, targetType='targetObj')
self.stuff = tempCluster_const_1
self.stuff = tempCluster_const_2
#if spine has zero joint it calls an unique function
self.unique_spine_zero_controller()
def unique_spine_zero_controller(self):
# Create Root Costrain Jnt Unde Hip cotrol
# Duplicate zero Jnt
tempConst = pm.duplicate(self.joints.zeroJoint,
po=True,
name=("Const_" + self.joints.zeroJoint))
rootConst_jnt = tempConst[0]
pm.parent(rootConst_jnt, self.hipCtrl.drawnNode)
pm.pointConstraint(rootConst_jnt, self.joints.zeroJoint)
pm.orientConstraint(rootConst_jnt, self.joints.zeroJoint)
pm.setAttr(rootConst_jnt.visibility, 0)
self._stuff.append(rootConst_jnt)
def organize_DAG(self):
pass
def fk_create(self, numOfFkCtrl=3):
fkJointsPos = []
fkJointsPos.append(self.joints.zeroPos)
for i in xrange(numOfFkCtrl, self.joints._numOfJoints - numOfFkCtrl,
numOfFkCtrl):
fkJointsPos.append(self.joints.jointPos[i])
fkJointsPos.append(self.joints.endPos)
print fkJointsPos
fkSetup = FkLimb()
fkSetup.create_fk_limb("back_FK_", fkJointsPos)
# *************************************************************************
# PROPERTIES
@property
def spineJoints(self):
return self._joints
@spineJoints.setter
def spineJoints(self, joints_in):
self._joints = joints_in
@property
def curve(self):
return self._curve
@property
def clusters(self):
return self._clusters
@clusters.setter
def clusters(self, node_in):
self._clusters.append(node_in)
@property
def hipCtrl(self):
return self._hipCtrl
@hipCtrl.setter
def hipCtrl(self, name_in):
if self._hipCtrl is not None:
pm.rename(self._hipCtrl.drawnNode, name_in)
@property
def shoulderCtrl(self):
return self._shoulderCtrl
@shoulderCtrl.setter
def shoulderCtrl(self, name_in):
if self._shoulderCtrl != None:
pm.rename(self._shoulderCtrl.drawnNode, name_in)
@property
def stuff(self):
return self._stuff
@stuff.setter
def stuff(self, stuff_in):
self._stuff.append(stuff_in)
@property
def limbName(self):
return self._limbName
@property
def scaleMD(self):
return self._scaleMD
def create_controllers(self):
#Check if clusters is not an empty list
# Hip Ctrl Create
"""
"""
self._hipCtrl = DrawNode(Shape.ikCtrl, 'hip_ctrl')
self.hipCtrl.temp_constrain(self.clusters[0].drawnNode)
self.hipCtrl.create_axialCor()
#parent Hip Clusters to Hip Control
pm.parent(self.clusters[0].axialCor, self.clusters[1].axialCor,
self.hipCtrl.drawnNode)
# Shoulder Ctrl Create
self._shoulderCtrl = DrawNode(Shape.circle, 'shoulder_ctrl')
self.shoulderCtrl.temp_constrain(self.clusters[(len(self.clusters) -
1)].drawnNode)
self.shoulderCtrl.create_axialCor()
# COG Ctrl Create
self._COGCtrl = DrawNode(Shape.cube, 'COG_ctrl')
self._COGCtrl.temp_constrain(self.hipCtrl.drawnNode)
self._COGCtrl.create_axialCor()
#parent Shoulder Clusters to Shoulder Control
pm.parent(self.clusters[(len(self.clusters) - 1)].axialCor,
self.clusters[(len(self.clusters) - 2)].axialCor,
self._shoulderCtrl.drawnNode)
# Create Mid Cluster Control Transforms and Constrains
mid_cluster = self.clusters[2].drawnNode
tempCluster_const_1 = DrawNode(Shape.transform,
"C_IK_SpineCl_ConstGrp")
tempCluster_const_1.temp_constrain(mid_cluster)
pm.parent(tempCluster_const_1.drawnNode, self.hipCtrl.drawnNode)
tempCluster_const_2 = DrawNode(Shape.transform,
"C_IK_SpineCl_ConstGrp")
tempCluster_const_2.temp_constrain(mid_cluster)
pm.parent(tempCluster_const_2.drawnNode, self.shoulderCtrl.drawnNode)
tempCluster_const_1.constrain(mid_cluster, targetType='targetObj')
tempCluster_const_2.constrain(mid_cluster, targetType='targetObj')
self.stuff = tempCluster_const_1
self.stuff = tempCluster_const_2
#if spine has zero joint it calls an unique function
self.unique_spine_zero_controller()
def orient_choose_direction(self, joint, jointUnder, frontAxis):
#TODO : Validate frontAxis
if frontAxis == "x":
frontInt = 0
elif frontAxis == "z":
frontInt = 2
returnVal = 1
transform_1 = DrawNode(Shape.transform, 'direction1')
transform_1.temp_constrain(joint)
transform_2 = DrawNode(Shape.transform, "direction2")
transform_2.temp_constrain(jointUnder)
frontTransform = transform_1.transform[frontInt] - \
transform_2.transform[frontInt]
if frontTransform > 0:
returnVal = -1
transform_1.delete()
transform_2.delete()
return returnVal
class IkSpineLimb(object):
def __init__(self):
self._limbName = None
self._joints = None
self._curve = None
self._ikHandle = None
self._effector = None
self._clusters = []
self._network = None
self._scaleMD = None
self._factors = None
self._hipCtrl = None
self._shoulderCtrl = None
self._COGCtrl = None
self._stuff = []
# *************************************************************************
# IKSPINE BASE SETUP METHODS
def create_spine(self, name_in, curve_in, frontAxis="z"):
#self._network = Network(name_in)
self._limbName = name_in
# You can change createion method with a Joint Chain Class
# JointChain(name_in, jointPositions)
# self.joint.orientChain
self.joints = SpineJoints(name_in, curve_in)
self.joints.orient_spine(frontAxis)
ikSolver = pm.ikHandle(sj=self.joints.startJoint,
ee=self.joints.endJoint,
tws="linear",
cra=True,
pcv=False,
ns=2,
sol="ikSplineSolver",
name=(name_in + "_IKSpine"))
self._ikHandle = pm.rename(ikSolver[0], (name_in + "_IK_Spine"))
self._effector = pm.rename(ikSolver[0],
(name_in + "_IK_SpineEffector"))
self._curve = Curve((name_in + "_IKSpineCurve"), ikSolver[2])
def create_clusters(self):
for i in range(0, self._curve.numCVs):
pm.select(self._curve.curveNode.cv[i])
tempClstr = DrawNode(
Shape.cluster,
self._limbName + "IK_SpineCl_#"
)
tempClstr.create_axialCor()
self.clusters.append(tempClstr)
#self.clusters[i].create_axialCor()
def make_stretchy(self):
#check joints
"""
"""
self._scaleMD = pm.createNode("multiplyDivide",
n=self.limbName + "_scaleMD")
pm.connectAttr(self.curve.curveInfo.arcLength, self.scaleMD.input1X)
pm.setAttr(self.scaleMD.input2X, self.curve.arclen)
pm.setAttr(self.scaleMD.operation, 2)
for jnt in self.joints.jointChain:
factor = pm.createNode("multiplyDivide", n="factor_" + jnt)
pm.connectAttr(self.scaleMD.outputX, factor.input1X)
pm.setAttr(factor.input2X, (pm.getAttr(jnt.ty)))
pm.connectAttr(factor.outputX, jnt.ty)
def create_controllers(self):
#Check if clusters is not an empty list
# Hip Ctrl Create
"""
"""
self._hipCtrl = DrawNode(Shape.ikCtrl, 'hip_ctrl')
self.hipCtrl.temp_constrain(self.clusters[0].drawnNode)
self.hipCtrl.create_axialCor()
#parent Hip Clusters to Hip Control
pm.parent(self.clusters[0].axialCor, self.clusters[1].axialCor,
self.hipCtrl.drawnNode)
# Shoulder Ctrl Create
self._shoulderCtrl = DrawNode(Shape.circle, 'shoulder_ctrl')
self.shoulderCtrl.temp_constrain(
self.clusters[(len(self.clusters) - 1)].drawnNode)
self.shoulderCtrl.create_axialCor()
# COG Ctrl Create
self._COGCtrl = DrawNode(Shape.cube, 'COG_ctrl')
self._COGCtrl.temp_constrain(self.hipCtrl.drawnNode)
self._COGCtrl.create_axialCor()
#parent Shoulder Clusters to Shoulder Control
pm.parent(self.clusters[(len(self.clusters) - 1)].axialCor,
self.clusters[(len(self.clusters) - 2)].axialCor,
self._shoulderCtrl.drawnNode)
# Create Mid Cluster Control Transforms and Constrains
mid_cluster = self.clusters[2].drawnNode
tempCluster_const_1 = DrawNode(Shape.transform,
"C_IK_SpineCl_ConstGrp")
tempCluster_const_1.temp_constrain(mid_cluster)
pm.parent(tempCluster_const_1.drawnNode, self.hipCtrl.drawnNode)
tempCluster_const_2 = DrawNode(Shape.transform,
"C_IK_SpineCl_ConstGrp")
tempCluster_const_2.temp_constrain(mid_cluster)
pm.parent(tempCluster_const_2.drawnNode, self.shoulderCtrl.drawnNode)
tempCluster_const_1.constrain(mid_cluster, targetType='targetObj')
tempCluster_const_2.constrain(mid_cluster, targetType='targetObj')
self.stuff = tempCluster_const_1
self.stuff = tempCluster_const_2
#if spine has zero joint it calls an unique function
self.unique_spine_zero_controller()
def unique_spine_zero_controller(self):
# Create Root Costrain Jnt Unde Hip cotrol
# Duplicate zero Jnt
tempConst = pm.duplicate(self.joints.zeroJoint, po=True,
name=("Const_" + self.joints.zeroJoint ))
rootConst_jnt = tempConst[0]
pm.parent(rootConst_jnt, self.hipCtrl.drawnNode)
pm.pointConstraint(rootConst_jnt, self.joints.zeroJoint)
pm.orientConstraint(rootConst_jnt, self.joints.zeroJoint)
pm.setAttr(rootConst_jnt.visibility, 0)
self._stuff.append(rootConst_jnt)
def organize_DAG(self):
pass
def fk_create(self, numOfFkCtrl = 3):
fkJointsPos = []
fkJointsPos.append(self.joints.zeroPos)
for i in xrange(numOfFkCtrl, self.joints._numOfJoints - numOfFkCtrl,
numOfFkCtrl):
fkJointsPos.append(self.joints.jointPos[i])
fkJointsPos.append(self.joints.endPos)
print fkJointsPos
fkSetup = FkLimb()
fkSetup.create_fk_limb("back_FK_", fkJointsPos)
# *************************************************************************
# PROPERTIES
@property
def spineJoints(self):
return self._joints
@spineJoints.setter
def spineJoints(self, joints_in):
self._joints = joints_in
@property
def curve(self):
return self._curve
@property
def clusters(self):
return self._clusters
@clusters.setter
def clusters(self, node_in):
self._clusters.append(node_in)
@property
def hipCtrl(self):
return self._hipCtrl
@hipCtrl.setter
def hipCtrl(self, name_in):
if self._hipCtrl is not None:
pm.rename(self._hipCtrl.drawnNode, name_in)
@property
def shoulderCtrl(self):
return self._shoulderCtrl
@shoulderCtrl.setter
def shoulderCtrl(self, name_in):
if self._shoulderCtrl != None:
pm.rename(self._shoulderCtrl.drawnNode, name_in)
@property
def stuff(self):
return self._stuff
@stuff.setter
def stuff(self, stuff_in):
self._stuff.append(stuff_in)
@property
def limbName(self):
return self._limbName
@property
def scaleMD(self):
return self._scaleMD
def create_controllers(self):
#Check if clusters is not an empty list
# Hip Ctrl Create
"""
"""
self._hipCtrl = DrawNode(Shape.ikCtrl, 'hip_ctrl')
self.hipCtrl.temp_constrain(self.clusters[0].drawnNode)
self.hipCtrl.create_axialCor()
#parent Hip Clusters to Hip Control
pm.parent(self.clusters[0].axialCor, self.clusters[1].axialCor,
self.hipCtrl.drawnNode)
# Shoulder Ctrl Create
self._shoulderCtrl = DrawNode(Shape.circle, 'shoulder_ctrl')
self.shoulderCtrl.temp_constrain(
self.clusters[(len(self.clusters) - 1)].drawnNode)
self.shoulderCtrl.create_axialCor()
# COG Ctrl Create
self._COGCtrl = DrawNode(Shape.cube, 'COG_ctrl')
self._COGCtrl.temp_constrain(self.hipCtrl.drawnNode)
self._COGCtrl.create_axialCor()
#parent Shoulder Clusters to Shoulder Control
pm.parent(self.clusters[(len(self.clusters) - 1)].axialCor,
self.clusters[(len(self.clusters) - 2)].axialCor,
self._shoulderCtrl.drawnNode)
# Create Mid Cluster Control Transforms and Constrains
mid_cluster = self.clusters[2].drawnNode
tempCluster_const_1 = DrawNode(Shape.transform,
"C_IK_SpineCl_ConstGrp")
tempCluster_const_1.temp_constrain(mid_cluster)
pm.parent(tempCluster_const_1.drawnNode, self.hipCtrl.drawnNode)
tempCluster_const_2 = DrawNode(Shape.transform,
"C_IK_SpineCl_ConstGrp")
tempCluster_const_2.temp_constrain(mid_cluster)
pm.parent(tempCluster_const_2.drawnNode, self.shoulderCtrl.drawnNode)
tempCluster_const_1.constrain(mid_cluster, targetType='targetObj')
tempCluster_const_2.constrain(mid_cluster, targetType='targetObj')
self.stuff = tempCluster_const_1
self.stuff = tempCluster_const_2
#if spine has zero joint it calls an unique function
self.unique_spine_zero_controller()
