2009-11-30 12:31:11 +00:00
|
|
|
# ##### BEGIN GPL LICENSE BLOCK #####
|
|
|
|
#
|
|
|
|
# This program is free software; you can redistribute it and/or
|
|
|
|
# modify it under the terms of the GNU General Public License
|
|
|
|
# as published by the Free Software Foundation; either version 2
|
|
|
|
# of the License, or (at your option) any later version.
|
|
|
|
#
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
# GNU General Public License for more details.
|
|
|
|
#
|
|
|
|
# You should have received a copy of the GNU General Public License
|
|
|
|
# along with this program; if not, write to the Free Software Foundation,
|
|
|
|
# Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
|
|
#
|
|
|
|
# ##### END GPL LICENSE BLOCK #####
|
|
|
|
|
2009-12-05 22:03:07 +00:00
|
|
|
# <pep8 compliant>
|
|
|
|
|
2009-11-30 12:31:11 +00:00
|
|
|
import bpy
|
2009-12-10 12:58:03 +00:00
|
|
|
from rigify_utils import bone_class_instance, copy_bone_simple, add_pole_target_bone, add_stretch_to, blend_bone_list, get_side_name, get_base_name
|
2009-12-05 22:03:07 +00:00
|
|
|
from rna_prop_ui import rna_idprop_ui_prop_get
|
2009-12-09 12:00:28 +00:00
|
|
|
from Mathutils import Vector
|
2009-11-30 12:31:11 +00:00
|
|
|
|
2009-12-05 19:26:28 +00:00
|
|
|
METARIG_NAMES = "shoulder", "arm", "forearm", "hand"
|
2009-11-30 12:31:11 +00:00
|
|
|
|
2009-12-05 22:03:07 +00:00
|
|
|
|
2009-12-02 11:55:47 +00:00
|
|
|
def metarig_template():
|
2009-12-09 12:00:28 +00:00
|
|
|
# generated by rigify.write_meta_rig
|
2009-12-02 11:55:47 +00:00
|
|
|
bpy.ops.object.mode_set(mode='EDIT')
|
2009-12-10 11:56:31 +00:00
|
|
|
obj = bpy.context.active_object
|
2009-12-02 11:55:47 +00:00
|
|
|
arm = obj.data
|
|
|
|
bone = arm.edit_bones.new('shoulder')
|
|
|
|
bone.head[:] = 0.0000, -0.4515, 0.0000
|
|
|
|
bone.tail[:] = 1.0000, -0.0794, 0.3540
|
|
|
|
bone.roll = -0.2227
|
|
|
|
bone.connected = False
|
|
|
|
bone = arm.edit_bones.new('upper_arm')
|
|
|
|
bone.head[:] = 1.1319, -0.0808, -0.0101
|
|
|
|
bone.tail[:] = 3.0319, 0.2191, -0.1101
|
|
|
|
bone.roll = 1.6152
|
|
|
|
bone.connected = False
|
|
|
|
bone.parent = arm.edit_bones['shoulder']
|
|
|
|
bone = arm.edit_bones.new('forearm')
|
|
|
|
bone.head[:] = 3.0319, 0.2191, -0.1101
|
|
|
|
bone.tail[:] = 4.8319, -0.0809, -0.0242
|
|
|
|
bone.roll = 1.5153
|
|
|
|
bone.connected = True
|
|
|
|
bone.parent = arm.edit_bones['upper_arm']
|
|
|
|
bone = arm.edit_bones.new('hand')
|
|
|
|
bone.head[:] = 4.8319, -0.0809, -0.0242
|
|
|
|
bone.tail[:] = 5.7590, -0.1553, -0.1392
|
|
|
|
bone.roll = -3.0083
|
|
|
|
bone.connected = True
|
|
|
|
bone.parent = arm.edit_bones['forearm']
|
|
|
|
|
|
|
|
bpy.ops.object.mode_set(mode='OBJECT')
|
|
|
|
pbone = obj.pose.bones['upper_arm']
|
|
|
|
pbone['type'] = 'arm'
|
|
|
|
|
|
|
|
|
2009-12-05 19:26:28 +00:00
|
|
|
def metarig_definition(obj, orig_bone_name):
|
|
|
|
mt = bone_class_instance(obj, METARIG_NAMES) # meta
|
|
|
|
mt.arm = orig_bone_name
|
|
|
|
mt.update()
|
2009-12-05 22:03:07 +00:00
|
|
|
|
2009-12-05 19:26:28 +00:00
|
|
|
mt.shoulder_p = mt.arm_p.parent
|
2009-12-08 07:11:43 +00:00
|
|
|
|
2009-12-05 19:26:28 +00:00
|
|
|
if not mt.shoulder_p:
|
2009-12-09 12:00:28 +00:00
|
|
|
raise Exception("could not find '%s' parent, skipping:" % orig_bone_name)
|
|
|
|
|
2009-12-07 17:21:30 +00:00
|
|
|
mt.shoulder = mt.shoulder_p.name
|
2009-12-05 19:26:28 +00:00
|
|
|
|
|
|
|
# We could have some bones attached, find the bone that has this as its 2nd parent
|
|
|
|
hands = []
|
|
|
|
for pbone in obj.pose.bones:
|
|
|
|
index = pbone.parent_index(mt.arm_p)
|
2009-12-09 22:44:26 +00:00
|
|
|
if index == 2 and pbone.bone.connected and pbone.bone.parent.connected:
|
2009-12-05 19:26:28 +00:00
|
|
|
hands.append(pbone)
|
|
|
|
|
2009-12-08 17:00:53 +00:00
|
|
|
if len(hands) != 1:
|
2009-12-09 22:44:26 +00:00
|
|
|
raise Exception("Found %s possible hands attached to this arm, expected 1 from bone: %s" % ([pbone.name for pbone in hands], orig_bone_name))
|
2009-12-05 19:26:28 +00:00
|
|
|
|
|
|
|
# first add the 2 new bones
|
|
|
|
mt.hand_p = hands[0]
|
|
|
|
mt.hand = mt.hand_p.name
|
|
|
|
|
|
|
|
mt.forearm_p = mt.hand_p.parent
|
|
|
|
mt.forearm = mt.forearm_p.name
|
2009-12-05 22:03:07 +00:00
|
|
|
|
2009-12-05 19:26:28 +00:00
|
|
|
return mt.names()
|
|
|
|
|
|
|
|
|
2009-12-09 12:00:28 +00:00
|
|
|
def ik(obj, definitions, base_names):
|
|
|
|
mt = bone_class_instance(obj, METARIG_NAMES)
|
2009-12-05 19:26:28 +00:00
|
|
|
mt.shoulder, mt.arm, mt.forearm, mt.hand = definitions
|
2009-12-09 12:00:28 +00:00
|
|
|
mt.update()
|
|
|
|
|
|
|
|
ik = bone_class_instance(obj, ["pole", "pole_vis", "hand_vis"])
|
|
|
|
ik_chain = mt.copy(to_fmt="MCH-%s_ik", base_names=base_names, exclude_attrs=["shoulder"])
|
|
|
|
|
|
|
|
# IK needs no parent_index
|
|
|
|
ik_chain.hand_e.connected = False
|
|
|
|
ik_chain.hand_e.parent = None
|
2009-12-09 22:44:26 +00:00
|
|
|
ik_chain.hand_e.local_location = False
|
|
|
|
ik_chain.rename("hand", get_base_name(base_names[mt.hand]) + "_ik" + get_side_name(mt.hand))
|
2009-12-09 12:00:28 +00:00
|
|
|
|
|
|
|
ik_chain.arm_e.connected = False
|
|
|
|
ik_chain.arm_e.parent = mt.shoulder_e
|
|
|
|
|
|
|
|
# Add the bone used for the arms poll target
|
2009-12-09 22:44:26 +00:00
|
|
|
#ik.pole = add_pole_target_bone(obj, mt.forearm, get_base_name(base_names[mt.forearm]) + "_target" + get_side_name(mt.forearm), mode='ZAVERAGE')
|
|
|
|
ik.pole = add_pole_target_bone(obj, mt.forearm, "elbow_target" + get_side_name(mt.forearm), mode='ZAVERAGE')
|
|
|
|
|
|
|
|
ik.update()
|
|
|
|
ik.pole_e.local_location = False
|
2009-12-09 12:00:28 +00:00
|
|
|
|
|
|
|
# update bones after this!
|
|
|
|
ik.hand_vis = add_stretch_to(obj, mt.hand, ik_chain.hand, "VIS-%s_ik" % base_names[mt.hand])
|
|
|
|
ik.pole_vis = add_stretch_to(obj, mt.forearm, ik.pole, "VIS-%s_ik" % base_names[mt.forearm])
|
|
|
|
|
|
|
|
ik.update()
|
|
|
|
ik.hand_vis_e.restrict_select = True
|
|
|
|
ik.pole_vis_e.restrict_select = True
|
|
|
|
|
|
|
|
bpy.ops.object.mode_set(mode='OBJECT')
|
|
|
|
|
|
|
|
mt.update()
|
|
|
|
ik.update()
|
|
|
|
ik_chain.update()
|
2009-12-09 14:29:55 +00:00
|
|
|
|
|
|
|
# Set IK dof
|
|
|
|
ik_chain.forearm_p.ik_dof_x = True
|
|
|
|
ik_chain.forearm_p.ik_dof_y = False
|
|
|
|
ik_chain.forearm_p.ik_dof_z = False
|
|
|
|
|
2009-12-09 12:00:28 +00:00
|
|
|
con = ik_chain.forearm_p.constraints.new('IK')
|
|
|
|
con.target = obj
|
|
|
|
con.subtarget = ik_chain.hand
|
|
|
|
con.pole_target = obj
|
|
|
|
con.pole_subtarget = ik.pole
|
|
|
|
|
|
|
|
con.use_tail = True
|
|
|
|
con.use_stretch = True
|
|
|
|
con.use_target = True
|
|
|
|
con.use_rotation = False
|
|
|
|
con.chain_length = 2
|
|
|
|
con.pole_angle = -90.0 # XXX, RAD2DEG
|
|
|
|
|
|
|
|
# ID Propery on the hand for IK/FK switch
|
|
|
|
|
|
|
|
prop = rna_idprop_ui_prop_get(ik_chain.hand_p, "ik", create=True)
|
|
|
|
ik_chain.hand_p["ik"] = 0.5
|
|
|
|
prop["soft_min"] = 0.0
|
|
|
|
prop["soft_max"] = 1.0
|
2009-12-05 22:03:07 +00:00
|
|
|
|
2009-12-09 12:00:28 +00:00
|
|
|
bpy.ops.object.mode_set(mode='EDIT')
|
|
|
|
|
|
|
|
# don't blend the shoulder
|
|
|
|
return [None] + ik_chain.names()
|
2009-12-05 22:03:07 +00:00
|
|
|
|
|
|
|
|
2009-12-09 12:00:28 +00:00
|
|
|
def fk(obj, definitions, base_names):
|
|
|
|
|
|
|
|
arm = obj.data
|
|
|
|
|
|
|
|
mt = bone_class_instance(obj, METARIG_NAMES)
|
|
|
|
mt.shoulder, mt.arm, mt.forearm, mt.hand = definitions
|
|
|
|
mt.update()
|
2009-12-05 22:03:07 +00:00
|
|
|
|
2009-12-09 12:00:28 +00:00
|
|
|
ex = bone_class_instance(obj, ["socket", "arm_hinge", "hand_delta"])
|
|
|
|
fk_chain = mt.copy(base_names=base_names)
|
|
|
|
|
|
|
|
# shoulder is used as a hinge
|
|
|
|
fk_chain.rename("shoulder", "MCH-%s_hinge" % base_names[mt.arm])
|
|
|
|
fk_chain.shoulder_e.translate(Vector(0.0, fk_chain.shoulder_e.length / 2, 0.0))
|
|
|
|
|
|
|
|
# upper arm constrains to this.
|
|
|
|
ex.socket_e = copy_bone_simple(arm, mt.arm, "MCH-%s_socket" % base_names[mt.arm])
|
|
|
|
ex.socket = ex.socket_e.name
|
|
|
|
ex.socket_e.connected = False
|
|
|
|
ex.socket_e.parent = mt.shoulder_e
|
2009-12-09 22:44:26 +00:00
|
|
|
ex.socket_e.length *= 0.5
|
2009-12-09 12:00:28 +00:00
|
|
|
|
|
|
|
# insert the 'DLT-hand', between the forearm and the hand
|
|
|
|
# copies forarm rotation
|
|
|
|
ex.hand_delta_e = copy_bone_simple(arm, fk_chain.hand, "DLT-%s" % base_names[mt.hand], parent=True)
|
|
|
|
ex.hand_delta = ex.hand_delta_e.name
|
|
|
|
ex.hand_delta_e.length *= 0.5
|
|
|
|
ex.hand_delta_e.connected = False
|
|
|
|
|
|
|
|
fk_chain.hand_e.connected = False
|
|
|
|
fk_chain.hand_e.parent = ex.hand_delta_e
|
2009-12-05 22:03:07 +00:00
|
|
|
|
2009-12-09 12:00:28 +00:00
|
|
|
bpy.ops.object.mode_set(mode='OBJECT')
|
|
|
|
|
|
|
|
mt.update()
|
|
|
|
ex.update()
|
|
|
|
fk_chain.update()
|
|
|
|
|
2009-12-09 14:29:55 +00:00
|
|
|
# Set rotation modes and axis locks
|
|
|
|
fk_chain.forearm_p.rotation_mode = 'XYZ'
|
|
|
|
fk_chain.forearm_p.lock_rotation = (False, True, True)
|
|
|
|
fk_chain.hand_p.rotation_mode = 'ZXY'
|
|
|
|
|
2009-12-09 12:00:28 +00:00
|
|
|
con = fk_chain.arm_p.constraints.new('COPY_LOCATION')
|
|
|
|
con.target = obj
|
|
|
|
con.subtarget = ex.socket
|
|
|
|
|
|
|
|
fk_chain.hand_p.lock_location = True, True, True
|
|
|
|
con = ex.hand_delta_p.constraints.new('COPY_ROTATION')
|
|
|
|
con.target = obj
|
|
|
|
con.subtarget = fk_chain.forearm
|
|
|
|
|
|
|
|
def hinge_setup():
|
2009-11-30 12:31:11 +00:00
|
|
|
# Hinge constraint & driver
|
2009-12-09 12:00:28 +00:00
|
|
|
con = fk_chain.shoulder_p.constraints.new('COPY_ROTATION')
|
2009-11-30 12:31:11 +00:00
|
|
|
con.name = "hinge"
|
|
|
|
con.target = obj
|
2009-12-09 12:00:28 +00:00
|
|
|
con.subtarget = mt.shoulder
|
2009-11-30 12:31:11 +00:00
|
|
|
driver_fcurve = con.driver_add("influence", 0)
|
|
|
|
driver = driver_fcurve.driver
|
|
|
|
|
2009-12-05 22:03:07 +00:00
|
|
|
|
2009-12-09 12:00:28 +00:00
|
|
|
controller_path = fk_chain.arm_p.path_to_id()
|
2009-11-30 12:31:11 +00:00
|
|
|
# add custom prop
|
2009-12-09 12:00:28 +00:00
|
|
|
fk_chain.arm_p["hinge"] = 0.0
|
|
|
|
prop = rna_idprop_ui_prop_get(fk_chain.arm_p, "hinge", create=True)
|
2009-11-30 12:31:11 +00:00
|
|
|
prop["soft_min"] = 0.0
|
|
|
|
prop["soft_max"] = 1.0
|
2009-12-05 22:03:07 +00:00
|
|
|
|
|
|
|
|
2009-11-30 12:31:11 +00:00
|
|
|
# *****
|
|
|
|
driver = driver_fcurve.driver
|
|
|
|
driver.type = 'AVERAGE'
|
2009-12-05 22:03:07 +00:00
|
|
|
|
2009-11-30 12:31:11 +00:00
|
|
|
tar = driver.targets.new()
|
|
|
|
tar.name = "hinge"
|
|
|
|
tar.id_type = 'OBJECT'
|
|
|
|
tar.id = obj
|
|
|
|
tar.rna_path = controller_path + '["hinge"]'
|
2009-12-05 22:03:07 +00:00
|
|
|
|
2009-12-09 12:00:28 +00:00
|
|
|
mod = driver_fcurve.modifiers[0]
|
|
|
|
mod.poly_order = 1
|
|
|
|
mod.coefficients[0] = 1.0
|
|
|
|
mod.coefficients[1] = -1.0
|
2009-12-05 22:03:07 +00:00
|
|
|
|
2009-12-09 12:00:28 +00:00
|
|
|
hinge_setup()
|
|
|
|
|
|
|
|
bpy.ops.object.mode_set(mode='EDIT')
|
2009-12-05 22:03:07 +00:00
|
|
|
|
2009-12-09 12:00:28 +00:00
|
|
|
return None, fk_chain.arm, fk_chain.forearm, fk_chain.hand
|
2009-12-05 20:45:51 +00:00
|
|
|
|
2009-12-05 22:03:07 +00:00
|
|
|
|
2009-12-09 12:00:28 +00:00
|
|
|
def main(obj, bone_definition, base_names):
|
|
|
|
bones_ik = ik(obj, bone_definition, base_names)
|
|
|
|
bones_fk = fk(obj, bone_definition, base_names)
|
2009-12-05 22:03:07 +00:00
|
|
|
|
2009-12-09 12:00:28 +00:00
|
|
|
bpy.ops.object.mode_set(mode='OBJECT')
|
2009-12-09 12:36:13 +00:00
|
|
|
blend_bone_list(obj, bone_definition, bones_ik, bones_fk, target_bone=bone_definition[1])
|