def create_trans_ctrl_at_joint(self,joint):
pos_ctrl_file = os.environ["mlrig_tool"]+"/controlobjects/blueprint/translation_control.ma"
cmds.file(pos_ctrl_file, i=True)
container = cmds.rename("translation_control_container", joint+"_translation_control_container")
utils.add_node_to_container(self.container_name, container)
for node in cmds.container(container, q=True, nodeList=True):
cmds.rename(node, joint+"_"+node, ignoreShape=True)
control = joint+"_translation_control"
cmds.parent(control, self.module_trans, absolute=True)
joint_pos = cmds.xform(joint, q=True, worldSpace=True, translation=True)
cmds.xform(control, worldSpace=True, absolute=True, translation=joint_pos)
nice_name = utils.strip_leading_namespace(joint)[1]
attr_name = nice_name+"_T"
cmds.container(container, edit=True, publishAndBind=[control+".translate", attr_name])
cmds.container(self.container_name, edit=True, publishAndBind=[container+"."+attr_name, attr_name])
return control
def lock_phase3(self, hook_obj):
module_container = self.module_namespace + ":module_container"
if hook_obj != None:
hook_obj_module_node = utils.strip_leading_namespace(hook_obj)
hook_obj_module = hook_obj_module_node[0]
hook_obj_joint = hook_obj_module_node[1].split(
"_translation_control")[0]
hook_obj = hook_obj_module + ":blueprint_" + hook_obj_joint
parent_con = cmds.parentConstraint(
hook_obj,
self.module_namespace + ":HOOK_IN",
maintainOffset=True,
name=self.module_namespace + ":hook_parent_constraint")[0]
scale_con = cmds.scaleConstraint(
hook_obj,
self.module_namespace + ":HOOK_IN",
maintainOffset=True,
name=self.module_namespace + ":hook_scale_constraint")[0]
module_container = self.module_namespace + ":module_container"
utils.add_node_to_container(module_container,
[parent_con, scale_con])
cmds.lockNode(module_container, lock=True, lockUnpublished=True)
def create_stretchy_obj(self,obj_relative_filepath,obj_container_name,obj_name,parent_joint,child_joint): obj_file = os.environ["mlrig_tool"]+obj_relative_filepath cmds.file(obj_file, i=True) obj_container = cmds.rename(obj_container_name, parent_joint+"_"+obj_container_name) for node in cmds.container(obj_container, q=True, nodeList=True): cmds.rename(node, parent_joint+"_"+node, ignoreShape=True) obj = parent_joint+"_"+obj_name con_grp = cmds.group(empty=True, name=obj+"_parentConstraint_grp") cmds.parent(obj, con_grp, absolute=True) parent_con = cmds.parentConstraint(parent_joint, con_grp, maintainOffset=False)[0] cmds.connectAttr(child_joint+".translateX", con_grp+".scaleX") scale_con = cmds.scaleConstraint(self.module_trans, con_grp, skip=["x"], maintainOffset=0)[0] utils.add_node_to_container(obj_container, [con_grp, parent_con, scale_con], ihb=True) utils.add_node_to_container(self.container_name, obj_container) return(obj_container, obj, con_grp)
def lock_phase3(self, hook_obj):
module_container = self.module_namespace+":module_container"
if hook_obj != None:
hook_obj_module_node = utils.strip_leading_namespace(hook_obj)
hook_obj_module = hook_obj_module_node[0]
hook_obj_joint = hook_obj_module_node[1].split("_translation_control")[0]
hook_obj = hook_obj_module+":blueprint_"+hook_obj_joint
parent_con = cmds.parentConstraint(
hook_obj,
self.module_namespace+":HOOK_IN",
maintainOffset=True,
name=self.module_namespace+":hook_parent_constraint"
)[0]
scale_con = cmds.scaleConstraint(
hook_obj,
self.module_namespace+":HOOK_IN",
maintainOffset=True,
name=self.module_namespace+":hook_scale_constraint"
)[0]
module_container = self.module_namespace+":module_container"
utils.add_node_to_container(module_container, [parent_con, scale_con])
cmds.lockNode(module_container, lock=True, lockUnpublished=True)
def create_stretchy_obj(self, obj_relative_filepath, obj_container_name,
obj_name, parent_joint, child_joint):
obj_file = os.environ["mlrig_tool"] + obj_relative_filepath
cmds.file(obj_file, i=True)
obj_container = cmds.rename(obj_container_name,
parent_joint + "_" + obj_container_name)
for node in cmds.container(obj_container, q=True, nodeList=True):
cmds.rename(node, parent_joint + "_" + node, ignoreShape=True)
obj = parent_joint + "_" + obj_name
con_grp = cmds.group(empty=True, name=obj + "_parentConstraint_grp")
cmds.parent(obj, con_grp, absolute=True)
parent_con = cmds.parentConstraint(parent_joint,
con_grp,
maintainOffset=False)[0]
cmds.connectAttr(child_joint + ".translateX", con_grp + ".scaleX")
scale_con = cmds.scaleConstraint(self.module_trans,
con_grp,
skip=["x"],
maintainOffset=0)[0]
utils.add_node_to_container(obj_container,
[con_grp, parent_con, scale_con],
ihb=True)
utils.add_node_to_container(self.container_name, obj_container)
return (obj_container, obj, con_grp)
def add_group_to_container(self, group):
group_container = "Group_container"
utils.add_node_to_container(group_container, group, includeShapes=True)
group_name = group.partition("Group__")[2]
cmds.container(group_container, edit=True, publishAndBind=[group+".translate", group_name+"_t"])
cmds.container(group_container, edit=True, publishAndBind=[group+".rotate", group_name+"_r"])
cmds.container(group_container, edit=True, publishAndBind=[group+".globalScale", group_name+"_globalScale"])
def init_module_trans(self,root_pos):
ctrl_grp_file = os.environ["mlrig_tool"]+"/controlobjects/blueprint/controlGroup_control.ma"
cmds.file(ctrl_grp_file, i=True)
self.module_trans = cmds.rename("controlGroup_control", self.module_namespace+":module_transform")
cmds.xform(self.module_trans, worldSpace=True, absolute=True, translation=root_pos)
# mirroring method
if self.mirrored:
duplicate_transform = cmds.duplicate(self.original_module+":module_transform", parentOnly=True, name="TEMP_TRANSFORM")[0]
empty_group = cmds.group(empty=True)
cmds.parent(duplicate_transform, empty_group, absolute=True)
scale_attr = ".scaleX"
if self.mirror_plane == "XZ":
scale_attr = ".scaleY"
elif self.mirror_plane == "XY":
scale_attr = ".scaleZ"
cmds.setAttr(empty_group+scale_attr, -1)
parent_constraint = cmds.parentConstraint(duplicate_transform, self.module_trans, maintainOffset=False)
cmds.delete(parent_constraint)
cmds.delete(empty_group)
temp_locator = cmds.spaceLocator()[0]
scale_constraint = cmds.scaleConstraint(self.original_module+":module_transform", temp_locator, maintainOffset=False)[0]
scale = cmds.getAttr(temp_locator+".scaleX")
cmds.delete([temp_locator, scale_constraint])
cmds.xform(self.module_trans, objectSpace=True, scale=[scale, scale, scale])
utils.add_node_to_container(self.container_name, self.module_trans, ihb=True)
# Setup global scaling
cmds.connectAttr(self.module_trans+".scaleY", self.module_trans+".scaleX")
cmds.connectAttr(self.module_trans+".scaleY", self.module_trans+".scaleZ")
cmds.aliasAttr("globalScale", self.module_trans+".scaleY")
cmds.container(self.container_name, edit=True, publishAndBind=[self.module_trans+".translate", "moduleTransform_T"])
cmds.container(self.container_name, edit=True, publishAndBind=[self.module_trans+".rotate", "moduleTransform_R"])
cmds.container(self.container_name, edit=True, publishAndBind=[self.module_trans+".globalScale", "moduleTransform_globalScale"])
def setup_stretchy_jnt_segment(self, parent_joint, child_joint):
parent_trans_control = self.get_trans_ctrl(parent_joint)
child_trans_control = self.get_trans_ctrl(child_joint)
pole_vector_loc = cmds.spaceLocator(n=parent_trans_control +
"_poleVectorLocator")[0]
pole_vector_loc_grp = cmds.group(pole_vector_loc,
n=pole_vector_loc +
"_parentConstraint_grp")
cmds.parent(pole_vector_loc_grp, self.module_grp, absolute=True)
parent_con = cmds.parentConstraint(parent_trans_control,
pole_vector_loc_grp,
maintainOffset=False)[0]
cmds.setAttr(pole_vector_loc + ".visibility", 0)
cmds.setAttr(pole_vector_loc + ".ty", -0.5)
ik_nodes = utils.basic_stretchy_ik(parent_joint,
child_joint,
container=self.container_name,
lock_min_len=False,
pole_vector_obj=pole_vector_loc,
scale_correct_atrr=None)
ik_handle = ik_nodes["ik_handle"]
root_loc = ik_nodes["root_loc"]
end_loc = ik_nodes["end_loc"]
# Mirroring Method
if self.mirrored:
if self.mirror_plane == "XZ":
cmds.setAttr(ik_handle + ".twist", 90)
child_point_con = cmds.pointConstraint(child_trans_control,
end_loc,
maintainOffset=False,
n=end_loc +
"_pointConstraint")[0]
utils.add_node_to_container(
self.container_name,
[pole_vector_loc_grp, parent_con, child_point_con],
ihb=True)
for node in [ik_handle, root_loc, end_loc]:
cmds.parent(node, self.joints_grp, absolute=True)
cmds.setAttr(node + ".visibility", 0)
self.create_hierarchy(parent_joint, child_joint)
def setup_stretchy_jnt_segment(self,parent_joint,child_joint): parent_trans_control = self.get_trans_ctrl(parent_joint) child_trans_control = self.get_trans_ctrl(child_joint) pole_vector_loc = cmds.spaceLocator(n=parent_trans_control+"_poleVectorLocator")[0] pole_vector_loc_grp = cmds.group(pole_vector_loc, n=pole_vector_loc+"_parentConstraint_grp") cmds.parent(pole_vector_loc_grp, self.module_grp, absolute=True) parent_con = cmds.parentConstraint(parent_trans_control, pole_vector_loc_grp, maintainOffset=False)[0] cmds.setAttr(pole_vector_loc+".visibility", 0) cmds.setAttr(pole_vector_loc+".ty", -0.5) ik_nodes = utils.basic_stretchy_ik( parent_joint, child_joint, container=self.container_name, lock_min_len=False, pole_vector_obj=pole_vector_loc, scale_correct_atrr=None ) ik_handle = ik_nodes["ik_handle"] root_loc = ik_nodes["root_loc"] end_loc = ik_nodes["end_loc"] # Mirroring Method if self.mirrored: if self.mirror_plane == "XZ": cmds.setAttr(ik_handle+".twist", 90) child_point_con = cmds.pointConstraint(child_trans_control, end_loc, maintainOffset=False, n=end_loc+"_pointConstraint")[0] utils.add_node_to_container(self.container_name, [pole_vector_loc_grp, parent_con, child_point_con], ihb=True) for node in [ik_handle, root_loc, end_loc]: cmds.parent(node, self.joints_grp, absolute=True) cmds.setAttr(node+".visibility", 0) self.create_hierarchy(parent_joint, child_joint)
def add_group_to_container(self, group):
group_container = "Group_container"
utils.add_node_to_container(group_container, group, includeShapes=True)
group_name = group.partition("Group__")[2]
cmds.container(
group_container,
edit=True,
publishAndBind=[group + ".translate", group_name + "_t"])
cmds.container(group_container,
edit=True,
publishAndBind=[group + ".rotate", group_name + "_r"])
cmds.container(group_container,
edit=True,
publishAndBind=[
group + ".globalScale", group_name + "_globalScale"
])
def create_trans_ctrl_at_joint(self, joint):
pos_ctrl_file = os.environ[
"mlrig_tool"] + "/controlobjects/blueprint/translation_control.ma"
cmds.file(pos_ctrl_file, i=True)
container = cmds.rename("translation_control_container",
joint + "_translation_control_container")
utils.add_node_to_container(self.container_name, container)
for node in cmds.container(container, q=True, nodeList=True):
cmds.rename(node, joint + "_" + node, ignoreShape=True)
control = joint + "_translation_control"
cmds.parent(control, self.module_trans, absolute=True)
joint_pos = cmds.xform(joint,
q=True,
worldSpace=True,
translation=True)
cmds.xform(control,
worldSpace=True,
absolute=True,
translation=joint_pos)
nice_name = utils.strip_leading_namespace(joint)[1]
attr_name = nice_name + "_T"
cmds.container(container,
edit=True,
publishAndBind=[control + ".translate", attr_name])
cmds.container(self.container_name,
edit=True,
publishAndBind=[container + "." + attr_name, attr_name])
return control
def initialize_hook(self, root_trans_control): unhooked_locator = cmds.spaceLocator(name=self.module_namespace+":unhookedTarget")[0] cmds.pointConstraint(root_trans_control, unhooked_locator, offset=[0, 0.001, 0]) cmds.setAttr(unhooked_locator+".visibility", 0) if self.hook_obj == None: self.hook_obj = unhooked_locator root_pos = cmds.xform(root_trans_control, q=True, worldSpace=True, translation=True) target_pos = cmds.xform(self.hook_obj, q=True, worldSpace=True, translation=True) cmds.select(clear=True) root_joint_without_namespace = "hook_root_joint" root_joint = cmds.joint(name=self.module_namespace+":"+root_joint_without_namespace, p=root_pos) cmds.setAttr(root_joint+".visibility", 0) target_joint_without_namespace = "hook_target_joint" target_joint = cmds.joint(name=self.module_namespace+":"+target_joint_without_namespace, p=target_pos) cmds.setAttr(target_joint+".visibility", 0) cmds.joint(root_joint, edit=True, orientJoint="xyz", sao="yup") # Container method for the objects being created in the hook phase hook_grp = cmds.group([root_joint, unhooked_locator], name=self.module_namespace+":hook_grp", parent=self.module_grp) hook_container = cmds.container(name=self.module_namespace+":hook_container") utils.add_node_to_container(hook_container, hook_grp, ihb=True) utils.add_node_to_container(self.container_name, hook_container) for joint in [root_joint, target_joint]: joint_name = utils.strip_all_namespaces(joint)[1] cmds.container(hook_container, edit=True, publishAndBind=[joint+".rotate", joint_name+"_R"]) # Ik functionality for the hook joints ik_nodes = utils.basic_stretchy_ik(root_joint, target_joint, hook_container, lock_min_len=False) ik_handle = ik_nodes["ik_handle"] root_loc = ik_nodes["root_loc"] end_loc = ik_nodes["end_loc"] pole_vector_loc = ik_nodes["pole_vector_obj"] root_point_con = cmds.pointConstraint( root_trans_control, root_joint, maintainOffset=False, name=root_joint+"_pointConstraint" )[0] target_point_con = cmds.pointConstraint( self.hook_obj, end_loc, maintainOffset=False, name=self.module_namespace+":hook_pointConstraint" )[0] utils.add_node_to_container(hook_container, [root_point_con, target_point_con]) for node in [ik_handle, root_loc, end_loc, pole_vector_loc]: cmds.parent(node, hook_grp, absolute=True) cmds.setAttr(node+".visibility", 0) object_nodes = self.create_stretchy_obj( "/controlobjects/blueprint/hook_representation.ma", "hook_representation_container", "hook_representation", root_joint, target_joint ) constrained_grp = object_nodes[2] cmds.parent(constrained_grp, hook_grp, absolute=True) hook_representation_container = object_nodes[0] cmds.container(self.container_name, edit=True, removeNode=hook_representation_container) utils.add_node_to_container(hook_container, hook_representation_container)
def install(self): cmds.namespace(setNamespace = ":") cmds.namespace(add=self.module_namespace) self.joints_grp = cmds.group(empty=True, name=self.module_namespace+":joints_grp") self.hierarchy_grp = cmds.group(empty=True, name=self.module_namespace+":hierarchy_grp") self.ori_ctrl_grp = cmds.group(empty=True, name=self.module_namespace+":orientationControls_grp") self.module_grp = cmds.group([self.joints_grp, self.hierarchy_grp, self.ori_ctrl_grp], name=self.module_namespace+":module_grp") cmds.container(name=self.container_name, addNode=self.module_grp, ihb=True) cmds.select(clear=True) index = 0 joints = [] for joint in self.joint_info: joint_name = joint[0] joint_pos = joint[1] parent_joint = "" if index > 0: parent_joint = self.module_namespace+":"+self.joint_info[index-1][0] cmds.select(parent_joint, replace=True) joint_name_full = cmds.joint(n=self.module_namespace+":"+joint_name, p=joint_pos) joints.append(joint_name_full) cmds.setAttr(joint_name_full+".visibility", 0) utils.add_node_to_container(self.container_name, joint_name_full) cmds.container( self.container_name, edit=True, publishAndBind=[joint_name_full+".rotate", joint_name+"_R"] ) cmds.container( self.container_name, edit=True, publishAndBind=[joint_name_full+".rotateOrder", joint_name+"_rotateOrder"] ) if index > 0: cmds.joint(parent_joint, edit=True, orientJoint="xyz", sao="yup") index += 1 if self.mirrored: mirror_xy = False mirror_yz = False mirror_xz = False if self.mirror_plane == "XY": mirror_xy = True elif self.mirror_plane == "YZ": mirror_yz = True elif self.mirror_plane == "XZ": mirror_xz = True mirror_behavior = False if self.rotation_function == "behavior": mirror_behavior = True mirror_nodes = cmds.mirrorJoint( joints[0], mirrorXY=mirror_xy, mirrorYZ=mirror_yz, mirrorXZ=mirror_xz, mirrorBehavior=mirror_behavior ) cmds.delete(joints) mirrored_joints = [] for node in mirror_nodes: if cmds.objectType(node, isType="joint"): mirrored_joints.append(node) else: cmds.delete(node) index = 0 for joint in mirrored_joints: joint_name = self.joint_info[index][0] new_joint_name = cmds.rename(joint, self.module_namespace+":"+joint_name) self.joint_info[index][1] = cmds.xform(new_joint_name, query=True, worldSpace=True, translation=True) index += 1 cmds.parent(joints[0], self.joints_grp, absolute=True) self.init_module_trans(self.joint_info[0][1]) trans_ctrl = [] for joint in joints: trans_ctrl.append(self.create_trans_ctrl_at_joint(joint)) root_joint_point_con = cmds.pointConstraint( trans_ctrl[0], joints[0], maintainOffset=False, name=joints[0]+"_pointConstraint" ) utils.add_node_to_container(self.container_name, root_joint_point_con) self.initialize_hook(trans_ctrl[0]) # Set up stretchy joint segment for index in range(len(joints)-1): self.setup_stretchy_jnt_segment(joints[index], joints[index+1]) self.install_custom(joints) utils.force_scene_update() cmds.lockNode(self.container_name, lock=True, lockUnpublished=True)
def lock_phase2(self, module_info):
joint_pos = module_info[0]
num_joints = len(joint_pos)
joint_ories = module_info[1]
ori_with_axis = False
pure_ori = False
if joint_ories[0] == None:
ori_with_axis = True
joint_ories = joint_ories[1]
else:
pure_ori = True
joint_ories = joint_ories[0]
num_ori = len(joint_ories)
joint_rotation_orders = module_info[2]
num_rotation_orders = len(joint_rotation_orders)
joint_pref_angle = module_info[3]
num_pref_angle = 0
if joint_pref_angle != None:
num_pref_angle = len(joint_pref_angle)
hook_obj = module_info[4]
root_trans = module_info[5]
# delete our blueprint controls
cmds.lockNode(self.container_name, lock=False, lockUnpublished=False)
cmds.delete(self.container_name)
cmds.namespace(setNamespace=":")
joint_radius = 1
if num_joints == 1:
joint_radius = 1.5
new_joints = []
for i in range(num_joints):
new_joint = ""
cmds.select(clear=True)
if ori_with_axis:
new_joint = cmds.joint(
n=self.module_namespace+":blueprint_"+self.joint_info[i][0],
p=joint_pos[i],
rotationOrder="xyz",
radius=joint_radius
)
if i != 0:
cmds.parent(new_joint, new_joints[i-1], absolute=True)
offset_index = i - 1
if offset_index < num_ori:
print joint_ories[offset_index][0]
cmds.joint(
new_joints[offset_index],
edit=True,
oj=joint_ories[offset_index][0],
sao=joint_ories[offset_index][1]
)
cmds.makeIdentity(new_joint, rotate=True, apply=True)
else:
if i != 0:
cmds.select(new_joints[i-1])
joint_ori = [0.0, 0.0, 0.0]
if i < num_ori:
joint_ori = [joint_ories[i][0], joint_ories[i][1], joint_ories[i][2]]
new_joint = cmds.joint(
n=self.module_namespace+":blueprint_"+self.joint_info[i][0],
p=joint_pos[i],
orientation=joint_ori,
rotationOrder="xyz",
radius=joint_radius
)
new_joints.append(new_joint)
if i < num_rotation_orders:
cmds.setAttr(new_joint+".rotateOrder", int(joint_rotation_orders[i]))
if i < num_pref_angle:
cmds.setAttr(new_joint+".preferredAngleX", joint_rotation_orders[i][0])
cmds.setAttr(new_joint+".preferredAngleY", joint_rotation_orders[i][1])
cmds.setAttr(new_joint+".preferredAngleZ", joint_rotation_orders[i][2])
cmds.setAttr(new_joint+".segmentScaleCompensate", 0)
blueprint_grp = cmds.group(empty=True, name=self.module_namespace+":blueprint_joints_grp")
cmds.parent(new_joints[0], blueprint_grp, absolute=True)
creation_pose_grp_nodes = cmds.duplicate(
blueprint_grp,
name=self.module_namespace+":creationPose_joints_grp",
renameChildren=True
)
creation_pose_grp = creation_pose_grp_nodes[0]
creation_pose_grp_nodes.pop(0)
i = 0
for node in creation_pose_grp_nodes:
rename_node = cmds.rename(node, self.module_namespace+":creationPose_"+self.joint_info[i][0])
cmds.setAttr(rename_node+".visibility", 0)
i +=1
cmds.select(blueprint_grp, replace=True)
cmds.addAttr(at="bool", defaultValue=0, ln="controlModuleInstalled", k=False)
hook_grp = cmds.group(empty=True, name=self.module_namespace+":HOOK_IN")
for obj in [blueprint_grp, creation_pose_grp]:
cmds.parent(obj, hook_grp, absolute=True)
settings_locator = cmds.spaceLocator(n=self.module_namespace+":SETTINGS")[0]
cmds.setAttr(settings_locator+".visibility", 0)
cmds.select(settings_locator, replace=True)
cmds.addAttr(at="enum", ln="activeModule", en="None:", k=False)
cmds.addAttr(at="float", ln="creationPoseWeight", defaultValue=1, k=False)
i = 0
utility_nodes = []
for joint in new_joints:
if i < (num_joints-1) or num_joints == 1:
add_node = cmds.shadingNode("plusMinusAverage", n=joint+"_addRotations", asUtility=True)
cmds.connectAttr(add_node+".output3D", joint+".rotate", force=True)
utility_nodes.append(add_node)
dummy_rotations_multiply = cmds.shadingNode(
"multiplyDivide",
n=joint+"_dummyRotationsMultiply",
asUtility=True
)
cmds.connectAttr(dummy_rotations_multiply+".output", add_node+".input3D[0]", force=True)
utility_nodes.append(dummy_rotations_multiply)
if i > 0:
original_tx = cmds.getAttr(joint+".tx")
add_tx_node = cmds.shadingNode("plusMinusAverage", n=joint+"_addTx", asUtility=True)
cmds.connectAttr(add_tx_node+".output1D", joint+".translateX", force=True)
utility_nodes.append(add_tx_node)
original_tx_multiply = cmds.shadingNode("multiplyDivide", n=joint+"_original_tx", asUtility=True)
cmds.setAttr(original_tx_multiply+".input1X", original_tx, lock=True)
cmds.connectAttr(settings_locator+".creationPoseWeight", original_tx_multiply+".input2X", force=True)
cmds.connectAttr(original_tx_multiply+".outputX", add_tx_node+".input1D[0]", force=True)
utility_nodes.append(original_tx_multiply)
else:
if root_trans:
original_translates = cmds.getAttr(joint+".translate")[0]
add_translate_node = cmds.shadingNode("plusMinusAverage", n=joint+"_addTranslate", asUtility=True)
cmds.connectAttr(add_translate_node+".output3D", joint+".translate", force=True)
utility_nodes.append(add_translate_node)
original_translate_multiply = cmds.shadingNode("multiplyDivide", n=joint+"_original_Translate", asUtility=True)
cmds.setAttr(
original_translate_multiply+".input1",
original_translates[0],
original_translates[1],
original_translates[2],
type="double3"
)
for attr in ["X", "Y", "Z"]:
cmds.connectAttr(settings_locator+".creationPoseWeight", original_translate_multiply+".input2"+attr)
cmds.connectAttr(original_translate_multiply+".output", add_translate_node+".input3D[0]", force=True)
utility_nodes.append(original_translate_multiply)
#Scale
original_scale = cmds.getAttr(joint+".scale")[0]
add_scale_node = cmds.shadingNode("plusMinusAverage", n=joint+"_addScale", asUtility=True)
cmds.connectAttr(add_scale_node+".output3D", joint+".scale", force=True)
utility_nodes.append(add_scale_node)
original_scale_multiply = cmds.shadingNode("multiplyDivide", n=joint+"_original_scale", asUtility=True)
cmds.setAttr(
original_scale_multiply+".input1",
original_scale[0],
original_scale[1],
original_scale[2],
type="double3"
)
for attr in ["X", "Y", "Z"]:
cmds.connectAttr(settings_locator+".creationPoseWeight", original_scale_multiply+".input2"+attr)
cmds.connectAttr(original_scale_multiply+".output", add_scale_node+".input3D[0]", force=True)
utility_nodes.append(original_scale_multiply)
i += 1
blueprint_nodes = utility_nodes
blueprint_nodes.append(blueprint_grp)
blueprint_nodes.append(creation_pose_grp)
blueprint_container = cmds.container(n=self.module_namespace+":blueprint_container")
utils.add_node_to_container(blueprint_container, blueprint_nodes, ihb=True)
module_grp = cmds.group(empty=True, name=self.module_namespace+":module_grp")
for obj in [hook_grp, settings_locator]:
cmds.parent(obj, module_grp, absolute=True)
module_container = cmds.container(n=self.module_namespace+":module_container")
utils.add_node_to_container(
module_container,
[module_grp, settings_locator, blueprint_container, hook_grp],
includeShapes=True
)
cmds.container(
module_container,
edit=True,
publishAndBind=[settings_locator+".activeModule", "activeModule"]
)
cmds.container(
module_container,
edit=True,
publishAndBind=[settings_locator+".creationPoseWeight", "creationPoseWeight"]
)
cmds.select(module_grp)
cmds.addAttr(at="float", longName="hierarchicalScale")
cmds.connectAttr(hook_grp+".scaleY", module_grp+".hierarchicalScale")
def initialize_hook(self, root_trans_control):
unhooked_locator = cmds.spaceLocator(name=self.module_namespace +
":unhookedTarget")[0]
cmds.pointConstraint(root_trans_control,
unhooked_locator,
offset=[0, 0.001, 0])
cmds.setAttr(unhooked_locator + ".visibility", 0)
if self.hook_obj == None:
self.hook_obj = unhooked_locator
root_pos = cmds.xform(root_trans_control,
q=True,
worldSpace=True,
translation=True)
target_pos = cmds.xform(self.hook_obj,
q=True,
worldSpace=True,
translation=True)
cmds.select(clear=True)
root_joint_without_namespace = "hook_root_joint"
root_joint = cmds.joint(name=self.module_namespace + ":" +
root_joint_without_namespace,
p=root_pos)
cmds.setAttr(root_joint + ".visibility", 0)
target_joint_without_namespace = "hook_target_joint"
target_joint = cmds.joint(name=self.module_namespace + ":" +
target_joint_without_namespace,
p=target_pos)
cmds.setAttr(target_joint + ".visibility", 0)
cmds.joint(root_joint, edit=True, orientJoint="xyz", sao="yup")
# Container method for the objects being created in the hook phase
hook_grp = cmds.group([root_joint, unhooked_locator],
name=self.module_namespace + ":hook_grp",
parent=self.module_grp)
hook_container = cmds.container(name=self.module_namespace +
":hook_container")
utils.add_node_to_container(hook_container, hook_grp, ihb=True)
utils.add_node_to_container(self.container_name, hook_container)
for joint in [root_joint, target_joint]:
joint_name = utils.strip_all_namespaces(joint)[1]
cmds.container(
hook_container,
edit=True,
publishAndBind=[joint + ".rotate", joint_name + "_R"])
# Ik functionality for the hook joints
ik_nodes = utils.basic_stretchy_ik(root_joint,
target_joint,
hook_container,
lock_min_len=False)
ik_handle = ik_nodes["ik_handle"]
root_loc = ik_nodes["root_loc"]
end_loc = ik_nodes["end_loc"]
pole_vector_loc = ik_nodes["pole_vector_obj"]
root_point_con = cmds.pointConstraint(root_trans_control,
root_joint,
maintainOffset=False,
name=root_joint +
"_pointConstraint")[0]
target_point_con = cmds.pointConstraint(self.hook_obj,
end_loc,
maintainOffset=False,
name=self.module_namespace +
":hook_pointConstraint")[0]
utils.add_node_to_container(hook_container,
[root_point_con, target_point_con])
for node in [ik_handle, root_loc, end_loc, pole_vector_loc]:
cmds.parent(node, hook_grp, absolute=True)
cmds.setAttr(node + ".visibility", 0)
object_nodes = self.create_stretchy_obj(
"/controlobjects/blueprint/hook_representation.ma",
"hook_representation_container", "hook_representation", root_joint,
target_joint)
constrained_grp = object_nodes[2]
cmds.parent(constrained_grp, hook_grp, absolute=True)
hook_representation_container = object_nodes[0]
cmds.container(self.container_name,
edit=True,
removeNode=hook_representation_container)
utils.add_node_to_container(hook_container,
hook_representation_container)
def install(self):
cmds.namespace(setNamespace=":")
cmds.namespace(add=self.module_namespace)
self.joints_grp = cmds.group(empty=True,
name=self.module_namespace +
":joints_grp")
self.hierarchy_grp = cmds.group(empty=True,
name=self.module_namespace +
":hierarchy_grp")
self.ori_ctrl_grp = cmds.group(empty=True,
name=self.module_namespace +
":orientationControls_grp")
self.module_grp = cmds.group(
[self.joints_grp, self.hierarchy_grp, self.ori_ctrl_grp],
name=self.module_namespace + ":module_grp")
cmds.container(name=self.container_name,
addNode=self.module_grp,
ihb=True)
cmds.select(clear=True)
index = 0
joints = []
for joint in self.joint_info:
joint_name = joint[0]
joint_pos = joint[1]
parent_joint = ""
if index > 0:
parent_joint = self.module_namespace + ":" + self.joint_info[
index - 1][0]
cmds.select(parent_joint, replace=True)
joint_name_full = cmds.joint(n=self.module_namespace + ":" +
joint_name,
p=joint_pos)
joints.append(joint_name_full)
cmds.setAttr(joint_name_full + ".visibility", 0)
utils.add_node_to_container(self.container_name, joint_name_full)
cmds.container(self.container_name,
edit=True,
publishAndBind=[
joint_name_full + ".rotate", joint_name + "_R"
])
cmds.container(self.container_name,
edit=True,
publishAndBind=[
joint_name_full + ".rotateOrder",
joint_name + "_rotateOrder"
])
if index > 0:
cmds.joint(parent_joint,
edit=True,
orientJoint="xyz",
sao="yup")
index += 1
if self.mirrored:
mirror_xy = False
mirror_yz = False
mirror_xz = False
if self.mirror_plane == "XY":
mirror_xy = True
elif self.mirror_plane == "YZ":
mirror_yz = True
elif self.mirror_plane == "XZ":
mirror_xz = True
mirror_behavior = False
if self.rotation_function == "behavior":
mirror_behavior = True
mirror_nodes = cmds.mirrorJoint(joints[0],
mirrorXY=mirror_xy,
mirrorYZ=mirror_yz,
mirrorXZ=mirror_xz,
mirrorBehavior=mirror_behavior)
cmds.delete(joints)
mirrored_joints = []
for node in mirror_nodes:
if cmds.objectType(node, isType="joint"):
mirrored_joints.append(node)
else:
cmds.delete(node)
index = 0
for joint in mirrored_joints:
joint_name = self.joint_info[index][0]
new_joint_name = cmds.rename(
joint, self.module_namespace + ":" + joint_name)
self.joint_info[index][1] = cmds.xform(new_joint_name,
query=True,
worldSpace=True,
translation=True)
index += 1
cmds.parent(joints[0], self.joints_grp, absolute=True)
self.init_module_trans(self.joint_info[0][1])
trans_ctrl = []
for joint in joints:
trans_ctrl.append(self.create_trans_ctrl_at_joint(joint))
root_joint_point_con = cmds.pointConstraint(trans_ctrl[0],
joints[0],
maintainOffset=False,
name=joints[0] +
"_pointConstraint")
utils.add_node_to_container(self.container_name, root_joint_point_con)
self.initialize_hook(trans_ctrl[0])
# Set up stretchy joint segment
for index in range(len(joints) - 1):
self.setup_stretchy_jnt_segment(joints[index], joints[index + 1])
self.install_custom(joints)
utils.force_scene_update()
cmds.lockNode(self.container_name, lock=True, lockUnpublished=True)
def lock_phase2(self, module_info):
joint_pos = module_info[0]
num_joints = len(joint_pos)
joint_ories = module_info[1]
ori_with_axis = False
pure_ori = False
if joint_ories[0] == None:
ori_with_axis = True
joint_ories = joint_ories[1]
else:
pure_ori = True
joint_ories = joint_ories[0]
num_ori = len(joint_ories)
joint_rotation_orders = module_info[2]
num_rotation_orders = len(joint_rotation_orders)
joint_pref_angle = module_info[3]
num_pref_angle = 0
if joint_pref_angle != None:
num_pref_angle = len(joint_pref_angle)
hook_obj = module_info[4]
root_trans = module_info[5]
# delete our blueprint controls
cmds.lockNode(self.container_name, lock=False, lockUnpublished=False)
cmds.delete(self.container_name)
cmds.namespace(setNamespace=":")
joint_radius = 1
if num_joints == 1:
joint_radius = 1.5
new_joints = []
for i in range(num_joints):
new_joint = ""
cmds.select(clear=True)
if ori_with_axis:
new_joint = cmds.joint(n=self.module_namespace +
":blueprint_" + self.joint_info[i][0],
p=joint_pos[i],
rotationOrder="xyz",
radius=joint_radius)
if i != 0:
cmds.parent(new_joint, new_joints[i - 1], absolute=True)
offset_index = i - 1
if offset_index < num_ori:
print joint_ories[offset_index][0]
cmds.joint(new_joints[offset_index],
edit=True,
oj=joint_ories[offset_index][0],
sao=joint_ories[offset_index][1])
cmds.makeIdentity(new_joint, rotate=True, apply=True)
else:
if i != 0:
cmds.select(new_joints[i - 1])
joint_ori = [0.0, 0.0, 0.0]
if i < num_ori:
joint_ori = [
joint_ories[i][0], joint_ories[i][1], joint_ories[i][2]
]
new_joint = cmds.joint(n=self.module_namespace +
":blueprint_" + self.joint_info[i][0],
p=joint_pos[i],
orientation=joint_ori,
rotationOrder="xyz",
radius=joint_radius)
new_joints.append(new_joint)
if i < num_rotation_orders:
cmds.setAttr(new_joint + ".rotateOrder",
int(joint_rotation_orders[i]))
if i < num_pref_angle:
cmds.setAttr(new_joint + ".preferredAngleX",
joint_rotation_orders[i][0])
cmds.setAttr(new_joint + ".preferredAngleY",
joint_rotation_orders[i][1])
cmds.setAttr(new_joint + ".preferredAngleZ",
joint_rotation_orders[i][2])
cmds.setAttr(new_joint + ".segmentScaleCompensate", 0)
blueprint_grp = cmds.group(empty=True,
name=self.module_namespace +
":blueprint_joints_grp")
cmds.parent(new_joints[0], blueprint_grp, absolute=True)
creation_pose_grp_nodes = cmds.duplicate(blueprint_grp,
name=self.module_namespace +
":creationPose_joints_grp",
renameChildren=True)
creation_pose_grp = creation_pose_grp_nodes[0]
creation_pose_grp_nodes.pop(0)
i = 0
for node in creation_pose_grp_nodes:
rename_node = cmds.rename(
node, self.module_namespace + ":creationPose_" +
self.joint_info[i][0])
cmds.setAttr(rename_node + ".visibility", 0)
i += 1
cmds.select(blueprint_grp, replace=True)
cmds.addAttr(at="bool",
defaultValue=0,
ln="controlModuleInstalled",
k=False)
hook_grp = cmds.group(empty=True,
name=self.module_namespace + ":HOOK_IN")
for obj in [blueprint_grp, creation_pose_grp]:
cmds.parent(obj, hook_grp, absolute=True)
settings_locator = cmds.spaceLocator(n=self.module_namespace +
":SETTINGS")[0]
cmds.setAttr(settings_locator + ".visibility", 0)
cmds.select(settings_locator, replace=True)
cmds.addAttr(at="enum", ln="activeModule", en="None:", k=False)
cmds.addAttr(at="float",
ln="creationPoseWeight",
defaultValue=1,
k=False)
i = 0
utility_nodes = []
for joint in new_joints:
if i < (num_joints - 1) or num_joints == 1:
add_node = cmds.shadingNode("plusMinusAverage",
n=joint + "_addRotations",
asUtility=True)
cmds.connectAttr(add_node + ".output3D",
joint + ".rotate",
force=True)
utility_nodes.append(add_node)
dummy_rotations_multiply = cmds.shadingNode(
"multiplyDivide",
n=joint + "_dummyRotationsMultiply",
asUtility=True)
cmds.connectAttr(dummy_rotations_multiply + ".output",
add_node + ".input3D[0]",
force=True)
utility_nodes.append(dummy_rotations_multiply)
if i > 0:
original_tx = cmds.getAttr(joint + ".tx")
add_tx_node = cmds.shadingNode("plusMinusAverage",
n=joint + "_addTx",
asUtility=True)
cmds.connectAttr(add_tx_node + ".output1D",
joint + ".translateX",
force=True)
utility_nodes.append(add_tx_node)
original_tx_multiply = cmds.shadingNode("multiplyDivide",
n=joint +
"_original_tx",
asUtility=True)
cmds.setAttr(original_tx_multiply + ".input1X",
original_tx,
lock=True)
cmds.connectAttr(settings_locator + ".creationPoseWeight",
original_tx_multiply + ".input2X",
force=True)
cmds.connectAttr(original_tx_multiply + ".outputX",
add_tx_node + ".input1D[0]",
force=True)
utility_nodes.append(original_tx_multiply)
else:
if root_trans:
original_translates = cmds.getAttr(joint + ".translate")[0]
add_translate_node = cmds.shadingNode("plusMinusAverage",
n=joint +
"_addTranslate",
asUtility=True)
cmds.connectAttr(add_translate_node + ".output3D",
joint + ".translate",
force=True)
utility_nodes.append(add_translate_node)
original_translate_multiply = cmds.shadingNode(
"multiplyDivide",
n=joint + "_original_Translate",
asUtility=True)
cmds.setAttr(original_translate_multiply + ".input1",
original_translates[0],
original_translates[1],
original_translates[2],
type="double3")
for attr in ["X", "Y", "Z"]:
cmds.connectAttr(
settings_locator + ".creationPoseWeight",
original_translate_multiply + ".input2" + attr)
cmds.connectAttr(original_translate_multiply + ".output",
add_translate_node + ".input3D[0]",
force=True)
utility_nodes.append(original_translate_multiply)
#Scale
original_scale = cmds.getAttr(joint + ".scale")[0]
add_scale_node = cmds.shadingNode("plusMinusAverage",
n=joint + "_addScale",
asUtility=True)
cmds.connectAttr(add_scale_node + ".output3D",
joint + ".scale",
force=True)
utility_nodes.append(add_scale_node)
original_scale_multiply = cmds.shadingNode(
"multiplyDivide",
n=joint + "_original_scale",
asUtility=True)
cmds.setAttr(original_scale_multiply + ".input1",
original_scale[0],
original_scale[1],
original_scale[2],
type="double3")
for attr in ["X", "Y", "Z"]:
cmds.connectAttr(
settings_locator + ".creationPoseWeight",
original_scale_multiply + ".input2" + attr)
cmds.connectAttr(original_scale_multiply + ".output",
add_scale_node + ".input3D[0]",
force=True)
utility_nodes.append(original_scale_multiply)
i += 1
blueprint_nodes = utility_nodes
blueprint_nodes.append(blueprint_grp)
blueprint_nodes.append(creation_pose_grp)
blueprint_container = cmds.container(n=self.module_namespace +
":blueprint_container")
utils.add_node_to_container(blueprint_container,
blueprint_nodes,
ihb=True)
module_grp = cmds.group(empty=True,
name=self.module_namespace + ":module_grp")
for obj in [hook_grp, settings_locator]:
cmds.parent(obj, module_grp, absolute=True)
module_container = cmds.container(n=self.module_namespace +
":module_container")
utils.add_node_to_container(
module_container,
[module_grp, settings_locator, blueprint_container, hook_grp],
includeShapes=True)
cmds.container(module_container,
edit=True,
publishAndBind=[
settings_locator + ".activeModule", "activeModule"
])
cmds.container(module_container,
edit=True,
publishAndBind=[
settings_locator + ".creationPoseWeight",
"creationPoseWeight"
])
cmds.select(module_grp)
cmds.addAttr(at="float", longName="hierarchicalScale")
cmds.connectAttr(hook_grp + ".scaleY",
module_grp + ".hierarchicalScale")
def init_module_trans(self, root_pos):
ctrl_grp_file = os.environ[
"mlrig_tool"] + "/controlobjects/blueprint/controlGroup_control.ma"
cmds.file(ctrl_grp_file, i=True)
self.module_trans = cmds.rename(
"controlGroup_control",
self.module_namespace + ":module_transform")
cmds.xform(self.module_trans,
worldSpace=True,
absolute=True,
translation=root_pos)
# mirroring method
if self.mirrored:
duplicate_transform = cmds.duplicate(self.original_module +
":module_transform",
parentOnly=True,
name="TEMP_TRANSFORM")[0]
empty_group = cmds.group(empty=True)
cmds.parent(duplicate_transform, empty_group, absolute=True)
scale_attr = ".scaleX"
if self.mirror_plane == "XZ":
scale_attr = ".scaleY"
elif self.mirror_plane == "XY":
scale_attr = ".scaleZ"
cmds.setAttr(empty_group + scale_attr, -1)
parent_constraint = cmds.parentConstraint(duplicate_transform,
self.module_trans,
maintainOffset=False)
cmds.delete(parent_constraint)
cmds.delete(empty_group)
temp_locator = cmds.spaceLocator()[0]
scale_constraint = cmds.scaleConstraint(self.original_module +
":module_transform",
temp_locator,
maintainOffset=False)[0]
scale = cmds.getAttr(temp_locator + ".scaleX")
cmds.delete([temp_locator, scale_constraint])
cmds.xform(self.module_trans,
objectSpace=True,
scale=[scale, scale, scale])
utils.add_node_to_container(self.container_name,
self.module_trans,
ihb=True)
# Setup global scaling
cmds.connectAttr(self.module_trans + ".scaleY",
self.module_trans + ".scaleX")
cmds.connectAttr(self.module_trans + ".scaleY",
self.module_trans + ".scaleZ")
cmds.aliasAttr("globalScale", self.module_trans + ".scaleY")
cmds.container(self.container_name,
edit=True,
publishAndBind=[
self.module_trans + ".translate",
"moduleTransform_T"
])
cmds.container(self.container_name,
edit=True,
publishAndBind=[
self.module_trans + ".rotate", "moduleTransform_R"
])
cmds.container(self.container_name,
edit=True,
publishAndBind=[
self.module_trans + ".globalScale",
"moduleTransform_globalScale"
])