diff options
Diffstat (limited to 'release/scripts/modules/rigify/delta.py')
-rw-r--r-- | release/scripts/modules/rigify/delta.py | 162 |
1 files changed, 162 insertions, 0 deletions
diff --git a/release/scripts/modules/rigify/delta.py b/release/scripts/modules/rigify/delta.py new file mode 100644 index 00000000000..475c64ab317 --- /dev/null +++ b/release/scripts/modules/rigify/delta.py @@ -0,0 +1,162 @@ +# ##### 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 ##### + +# <pep8 compliant> + +import bpy +from rigify import RigifyError + +# not used, defined for completeness +METARIG_NAMES = tuple() + + +def metarig_template(): + # generated by rigify.write_meta_rig + bpy.ops.object.mode_set(mode='EDIT') + obj = bpy.context.active_object + arm = obj.data + bone = arm.edit_bones.new('bonesker') + bone.head[:] = 0.0000, 0.0000, 0.0000 + bone.tail[:] = -0.0000, 0.7382, 0.1895 + bone.roll = -0.0000 + bone.connected = False + bone = arm.edit_bones.new('delta') + bone.head[:] = -0.0497, 0.8414, 0.3530 + bone.tail[:] = -0.2511, 1.1588, 0.9653 + bone.roll = 2.6044 + bone.connected = False + bone.parent = arm.edit_bones['bonesker'] + bone = arm.edit_bones.new('boney') + bone.head[:] = 0.7940, 2.5592, 0.4134 + bone.tail[:] = 0.7940, 3.3975, 0.4890 + bone.roll = 3.1416 + bone.connected = False + bone.parent = arm.edit_bones['delta'] + + bpy.ops.object.mode_set(mode='OBJECT') + pbone = obj.pose.bones['delta'] + pbone['type'] = 'delta' + + +def metarig_definition(obj, orig_bone_name): + ''' + The bone given is the head, its parent is the body, + # its only child the first of a chain with matching basenames. + eg. + body -> head -> neck_01 -> neck_02 -> neck_03.... etc + ''' + arm = obj.data + delta = arm.bones[orig_bone_name] + children = delta.children + + if len(children) != 1: + raise RigifyError("only 1 child supported for delta on bone '%s'" % delta.name) + + if delta.connected: + raise RigifyError("bone cannot be connected to its parent '%s'" % delta.name) + + bone_definition = [delta.name, children[0].name] + + return bone_definition + + +def main(obj, bone_definition, base_names, options): + ''' + Use this bone to define a delta thats applied to its child in pose mode. + ''' + mode_orig = obj.mode + bpy.ops.object.mode_set(mode='OBJECT') + + delta_name, child_name = bone_definition + + delta_pbone = obj.pose.bones[delta_name] + + arm = obj.data + child_pbone = obj.pose.bones[child_name] + + delta_phead = delta_pbone.head.copy() + delta_ptail = delta_pbone.tail.copy() + delta_pmatrix = delta_pbone.matrix.copy() + + child_phead = child_pbone.head.copy() + child_ptail = child_pbone.tail.copy() + child_pmatrix = child_pbone.matrix.copy() + + + children = delta_pbone.children + + bpy.ops.object.mode_set(mode='EDIT') + + delta_ebone = arm.edit_bones[delta_name] + child_ebone = arm.edit_bones[child_name] + + delta_head = delta_ebone.head.copy() + delta_tail = delta_ebone.tail.copy() + + child_head = child_ebone.head.copy() + child_tail = child_ebone.tail.copy() + + #arm.edit_bones.remove(delta_ebone) + #del delta_ebone # cant use this + del child_pbone + + bpy.ops.object.mode_set(mode='OBJECT') + + + # Move the child bone to the deltas location + obj.animation_data_create() + delta_pbone = obj.pose.bones[delta_name] + # child_pbone = obj.pose.bones[child_name] + + # ------------------- drivers + + delta_pbone.rotation_mode = 'XYZ' + + rot = delta_pmatrix.invert().rotationPart() * child_pmatrix.rotationPart() + rot = rot.invert().toEuler() + + fcurve_drivers = delta_pbone.driver_add("rotation_euler", -1) + for i, fcurve_driver in enumerate(fcurve_drivers): + driver = fcurve_driver.driver + driver.type = 'AVERAGE' + #mod = fcurve_driver.modifiers.new('GENERATOR') + mod = fcurve_driver.modifiers[0] + mod.poly_order = 1 + mod.coefficients[0] = rot[i] + mod.coefficients[1] = 0.0 + + # tricky, find the transform to drive the bone to this location. + delta_head_offset = child_pmatrix.rotationPart() * (delta_phead - child_phead) + + fcurve_drivers = delta_pbone.driver_add("location", -1) + for i, fcurve_driver in enumerate(fcurve_drivers): + driver = fcurve_driver.driver + driver.type = 'AVERAGE' + #mod = fcurve_driver.modifiers.new('GENERATOR') + mod = fcurve_driver.modifiers[0] + mod.poly_order = 1 + mod.coefficients[0] = delta_head_offset[i] + mod.coefficients[1] = 0.0 + + + bpy.ops.object.mode_set(mode='EDIT') + + bpy.ops.object.mode_set(mode=mode_orig) + + # no blendeing + return None |