feat: experimental bendy_face

utils: strip_def

Author: eigen-value

rigs/experimental/bendy_face.py

@@ -1,7 +1,7 @@
 import bpy, re
 from   mathutils      import Vector
 from   ...utils       import copy_bone, flip_bone, put_bone
-from   ...utils       import org, strip_org, make_deformer_name, connected_children_names, make_mechanism_name
+from   ...utils       import org, strip_org, strip_def, make_deformer_name, connected_children_names, make_mechanism_name
 from   ...utils       import create_circle_widget, create_sphere_widget, create_widget, create_cube_widget
 from   ...utils       import MetarigError
 from   rna_prop_ui    import rna_idprop_ui_prop_get
@@ -13,6 +13,8 @@ class Rig:
     CTRL_SCALE = 0.1
     MCH_SCALE = 0.3
 
+    POSITION_RELATIVE_ERROR = 1e-3  # error below which two positions are considered equal (relative to bone len)
+
     def __init__(self, obj, bone_name, params):
         self.obj = obj
         self.bones = dict()
@@ -130,6 +132,54 @@ def aggregate_ctrls(self):
 
         return True
 
+    def make_constraints(self):
+        """
+        Make constraints for each bone subgroup
+        :return:
+        """
+
+        bpy.ops.object.mode_set(mode='OBJECT')
+        pose_bones = self.obj.pose.bones
+
+        ### Constrain DEF-bones ###
+        for subchain in self.bones['def']:
+            for name in self.bones['def'][subchain]:
+                owner_pb = pose_bones[name]
+
+                subtarget = make_mechanism_name(strip_def(name))
+                const = owner_pb.constraints.new('COPY_LOCATION')
+                const.target = self.obj
+                const.subtarget = subtarget
+
+                tail_subtargets = self.get_ctrls_by_position(owner_pb.bone.tail, subchain=subchain)
+
+                if tail_subtargets:
+                    const = owner_pb.constraints.new('DAMPED_TRACK')
+                    const.target = self.obj
+                    const.subtarget = tail_subtargets[0]
+
+                    const = owner_pb.constraints.new('STRETCH_TO')
+                    const.target = self.obj
+                    const.subtarget = tail_subtargets[0]
+
+    def parent_bones(self):
+        """
+        Specify bone parenting
+        :return:
+        """
+
+        bpy.ops.object.mode_set(mode='EDIT')
+        edit_bones = self.obj.data.edit_bones
+
+        ### PARENT MCH-bones ###
+        for subchain in self.bones['mch']:
+            for name in self.bones['mch'][subchain]:
+                mch_bone = edit_bones[name]
+                parents = self.get_ctrls_by_position(mch_bone.head, subchain=subchain)
+
+                if parents:
+                    mch_bone.parent = edit_bones[parents[0]]
+
     def get_aggregate_name(self, aggregate):
 
         total = '.'.join(aggregate)
@@ -182,10 +232,48 @@ def get_start_bones(self):
 
         return start_bones
 
+    def get_ctrls_by_position(self, position, subchain='', relative_error=0):
+        """
+        Returns the controls closest to position in given relative_error range and subchain
+        checking subchain first and then aggregates
+        :param subchain:
+        :type subchain: str
+        :param position:
+        :type subchain: Vector
+        :return:
+        :rtype: list(str)
+        """
+
+        bpy.ops.object.mode_set(mode='EDIT')
+        edit_bones = self.obj.data.edit_bones
+
+        bones_in_range = []
+
+        if subchain:
+            keys = [subchain]
+            if 'aggregate' in self.bones['ctrl']:
+                keys.append('aggregate')
+        else:
+            keys = self.bones['ctrl'].keys()
+
+        if not relative_error:
+            relative_error = self.POSITION_RELATIVE_ERROR
+
+        for chain in keys:
+            for name in self.bones['ctrl'][chain]:
+                error = edit_bones[name].length * relative_error
+                if (edit_bones[name].head - position).magnitude <= error:
+                    bones_in_range.append(name)
+
+        return bones_in_range
+
     def generate(self):
 
         self.create_mch()
         self.create_def()
         self.create_controls()
 
+        self.make_constraints()
+        self.parent_bones()
+
         return [""]

utils.py

@@ -125,6 +125,14 @@ def strip_mch(name):
     else:
         return name
 
+def strip_def(name):
+    """ Returns the name with DEF_PREFIX stripped from it.
+        """
+    if name.startswith(DEF_PREFIX):
+        return name[len(DEF_PREFIX):]
+    else:
+        return name
+
 def org(name):
     """ Prepends the ORG_PREFIX to a name if it doesn't already have
         it, and returns it.