Page MenuHome
Differential D4624 Diff 18077 rigify/rigs/limbs/super_limb.py

Changeset View

Standalone View

View Options

rigify/rigs/limbs/super_limb.py

#====================== 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
#
#======================= END GPL LICENSE BLOCK ========================
# <pep8 compliant>
import bpy import bpy
from .arm import Rig as armRig from ...base_generate import SubstitutionRig
from .leg import Rig as legRig
from .paw import Rig as pawRig
from ...utils import ControlLayersOption
class Rig:
def __init__(self, obj, bone_name, params):
""" Initialize super_limb rig wrapper class """
self.obj = obj
self.params = params
if params.limb_type == 'arm':
self.limb = armRig(obj, bone_name, params)
elif params.limb_type == 'leg':
self.limb = legRig(obj, bone_name, params)
elif params.limb_type == 'paw':
self.limb = pawRig(obj, bone_name, params)
def generate(self):
return self.limb.generate()
@staticmethod
def get_future_names(bones):
if bones[0].rigify_parameters.limb_type == 'arm':
return armRig.get_future_names(bones)
elif bones[0].rigify_parameters.limb_type == 'leg':
return legRig.get_future_names(bones)
elif bones[0].rigify_parameters.limb_type == 'paw':
return pawRig.get_future_names(bones)
from .limb_rigs import BaseLimbRig
def add_parameters(params): from . import arm, leg, paw
""" Add the parameters of this rig type to the
RigifyParameters PropertyGroup
""" RIGS = { 'arm': arm.Rig, 'leg': leg.Rig, 'paw': paw.Rig }
class Rig(SubstitutionRig):
def substitute(self):
return [ self.instantiate_rig(RIGS[self.params.limb_type], self.base_bone) ]
def add_parameters(params):
items = [ items = [
('arm', 'Arm', ''), ('arm', 'Arm', ''),
('leg', 'Leg', ''), ('leg', 'Leg', ''),
('paw', 'Paw', '') ('paw', 'Paw', '')
] ]
params.limb_type = bpy.props.EnumProperty( params.limb_type = bpy.props.EnumProperty(
items = items, items = items,
name = "Limb Type", name = "Limb Type",
default = 'arm' default = 'arm'
) )
items = [ BaseLimbRig.add_parameters(params)
('x', 'X manual', ''),
('z', 'Z manual', ''),
('automatic', 'Automatic', '')
]
params.rotation_axis = bpy.props.EnumProperty(
items = items,
name = "Rotation Axis",
default = 'automatic'
)
params.auto_align_extremity = bpy.props.BoolProperty(
name='auto_align_extremity',
default=False,
description="Auto Align Extremity Bone"
)
params.segments = bpy.props.IntProperty(
name = 'limb segments',
default = 2,
min = 1,
description = 'Number of segments'
)
params.bbones = bpy.props.IntProperty(
name = 'bbone segments',
default = 10,
min = 1,
description = 'Number of segments'
)
# Setting up extra layers for the FK and tweak
ControlLayersOption.FK.add_parameters(params)
ControlLayersOption.TWEAK.add_parameters(params)
def parameters_ui(layout, params): def parameters_ui(layout, params):
""" Create the ui for the rig parameters."""
r = layout.row() r = layout.row()
r.prop(params, "limb_type") r.prop(params, "limb_type")
r = layout.row() RIGS[params.limb_type].parameters_ui(layout, params)
r.prop(params, "rotation_axis")
if 'auto' not in params.rotation_axis.lower():
r = layout.row()
etremities = {'arm': 'Hand', 'leg': 'Foot', 'paw': 'Claw'}
text = "Auto align " + etremities[params.limb_type]
r.prop(params, "auto_align_extremity", text=text)
r = layout.row()
r.prop(params, "segments")
r = layout.row()
r.prop(params, "bbones")
ControlLayersOption.FK.parameters_ui(layout, params)
ControlLayersOption.TWEAK.parameters_ui(layout, params)
def create_sample(obj): def create_sample(obj):
# generated by rigify.utils.write_metarig arm.create_sample(obj, limb=True)
bpy.ops.object.mode_set(mode='EDIT')
arm = obj.data
bones = {}
bone = arm.edit_bones.new('upper_arm.L')
bone.head[:] = -0.0016, 0.0060, -0.0012
bone.tail[:] = 0.2455, 0.0678, -0.1367
bone.roll = 2.0724
bone.use_connect = False
bones['upper_arm.L'] = bone.name
bone = arm.edit_bones.new('forearm.L')
bone.head[:] = 0.2455, 0.0678, -0.1367
bone.tail[:] = 0.4625, 0.0285, -0.2797
bone.roll = 2.1535
bone.use_connect = True
bone.parent = arm.edit_bones[bones['upper_arm.L']]
bones['forearm.L'] = bone.name
bone = arm.edit_bones.new('hand.L')
bone.head[:] = 0.4625, 0.0285, -0.2797
bone.tail[:] = 0.5265, 0.0205, -0.3273
bone.roll = 2.2103
bone.use_connect = True
bone.parent = arm.edit_bones[bones['forearm.L']]
bones['hand.L'] = bone.name
bpy.ops.object.mode_set(mode='OBJECT')
pbone = obj.pose.bones[bones['upper_arm.L']]
pbone.rigify_type = 'limbs.super_limb'
pbone.lock_location = (False, False, False)
pbone.lock_rotation = (False, False, False)
pbone.lock_rotation_w = False
pbone.lock_scale = (False, False, False)
pbone.rotation_mode = 'QUATERNION'
try:
pbone.rigify_parameters.separate_ik_layers = True
except AttributeError:
pass
try:
pbone.rigify_parameters.ik_layers = [
False, False, False, False, False, False, False, False, True, False,
False, False, False, False, False, False, False, False, False, False,
False, False, False, False, False, False, False, False, False, False,
False, False
]
except AttributeError:
pass
try:
pbone.rigify_parameters.separate_hose_layers = True
except AttributeError:
pass
try:
pbone.rigify_parameters.hose_layers = [
False, False, False, False, False, False, False, False, False, True,
False, False, False, False, False, False, False, False, False, False,
False, False, False, False, False, False, False, False, False, False,
False, False
]
except AttributeError:
pass
try:
pbone.rigify_parameters.tweak_layers = [
False, False, False, False, False, False, False, False, False, True,
False, False, False, False, False, False, False, False, False, False,
False, False, False, False, False, False, False, False, False, False,
False, False
]
except AttributeError:
pass
try:
pbone.rigify_parameters.fk_layers = [
False, False, False, False, False, False, False, False, True, False,
False, False, False, False, False, False, False, False, False, False,
False, False, False, False, False, False, False, False, False, False,
False, False
]
except AttributeError:
pass
pbone = obj.pose.bones[bones['forearm.L']]
pbone.rigify_type = ''
pbone.lock_location = (False, False, False)
pbone.lock_rotation = (False, False, False)
pbone.lock_rotation_w = False
pbone.lock_scale = (False, False, False)
pbone.rotation_mode = 'QUATERNION'
pbone = obj.pose.bones[bones['hand.L']]
pbone.rigify_type = ''
pbone.lock_location = (False, False, False)
pbone.lock_rotation = (False, False, False)
pbone.lock_rotation_w = False
pbone.lock_scale = (False, False, False)
pbone.rotation_mode = 'QUATERNION'
bpy.ops.object.mode_set(mode='EDIT')
for bone in arm.edit_bones:
bone.select = False
bone.select_head = False
bone.select_tail = False
for b in bones:
bone = arm.edit_bones[bones[b]]
bone.select = True
bone.select_head = True
bone.select_tail = True
arm.edit_bones.active = bone