New rig_types

Author: eigen-value

__init__.py

@@ -301,7 +301,6 @@ def register():
 
     bpy.types.PoseBone.rigify_type = bpy.props.StringProperty(name="Rigify Type", description="Rig type for this bone")
     bpy.types.PoseBone.rigify_parameters = bpy.props.PointerProperty(type=RigifyParameters)
-    bpy.types.PoseBone.rigify_glue = bpy.props.StringProperty(name="Rigify Glue", description="Defines influence between controls")
 
     bpy.types.Armature.rigify_colors = bpy.props.CollectionProperty(type=RigifyColorSet)
 
@@ -396,7 +395,6 @@ def update_mode(self, context):
 def unregister():
     del bpy.types.PoseBone.rigify_type
     del bpy.types.PoseBone.rigify_parameters
-    del bpy.types.PoseBone.rigify_glue
 
     IDStore = bpy.types.WindowManager
     del IDStore.rigify_collection

generate.py

@@ -351,6 +351,7 @@ def generate_rig(context, metarig):
             if scripts is not None:
                 ui_scripts += [scripts[0]]
         t.tick("Generate rigs: ")
+
     except Exception as e:
         # Cleanup if something goes wrong
         print("Rigify: failed to generate rig.")
@@ -524,6 +525,15 @@ def generate_rig(context, metarig):
         ctrl = obj.game.controllers[-1]
         ctrl.text = bpy.data.texts[script.name]
 
+    # Do final gluing
+    for rig in rigs:
+        if hasattr(rig, "glue"):
+            # update glue_bone rigs
+            bpy.ops.object.mode_set(mode='EDIT')
+            rig = rig.__class__(rig.obj, rig.base_bone, rig.params)
+
+            rig.glue()
+    t.tick("Glue pass")
 
     t.tick("The rest: ")
     #----------------------------------

rigs/experimental/base_rig.py

@@ -0,0 +1,101 @@
+import bpy
+
+
+class BaseRig(object):
+
+    def __init__(self, obj, bone_name, params):
+        """
+        Rig Base class the bones struct is a dict with 'org' as a list and 'def' 'mch' and 'ctrl' dicts
+        ctrls mchs and defs must be organized in groups. If your Rig Class has just one group you can call it all_ctrls
+        :param obj:
+        :param bone_name:
+        :param params:
+        """
+
+        self.obj = obj
+        self.params = params
+        self.bones = dict()
+        self.bones['org'] = [bone_name]
+        self.base_bone = bone_name
+
+        # Get all the recursive ORG children of the base bone BUT the rig_type trees
+        for edit_bone in self.obj.data.edit_bones[bone_name].children:
+            if self.obj.pose.bones[edit_bone.name].rigify_type != "":
+                continue
+            else:
+                self.bones['org'].append(edit_bone.name)
+            for child in edit_bone.children_recursive:
+                self.bones['org'].append(child.name)
+
+        self.bones['ctrl'] = dict()
+        self.bones['mch'] = dict()
+        self.bones['def'] = dict()
+
+    def orient_org_bones(self):
+        """
+        This function re-orients org bones so that created bones are properly aligned and cns can work
+        :return:
+        """
+        pass
+
+    def create_mch(self):
+        pass
+
+    def create_def(self):
+        pass
+
+    def create_controls(self):
+        pass
+
+    def create_widgets(self):
+        pass
+
+    def make_constraints(self):
+        pass
+
+    def parent_bones(self):
+        pass
+
+    def generate(self):
+        pass
+
+    def flatten(self, bones):
+        """
+        Flattens a bones dictionary
+        :param bones:
+        :return:
+        :rtype: list
+        """
+
+        all_bones = []
+
+        if isinstance(bones, dict):
+            for key in bones:
+                all_bones.extend(self.flatten(bones[key]))
+            return all_bones
+        else:
+            return bones
+
+    def get_all_ctrls(self):
+        return self.flatten(self.bones['ctrl'])
+
+    @staticmethod
+    def add_parameters(params):
+        """
+        This method add more parameters to params
+        :param params: rigify_parameters of a pose_bone
+        :return:
+        """
+
+        pass
+
+    @staticmethod
+    def parameters_ui(layout, params):
+        """
+        This method draws the UI of the rigify_parameters defined on the pose_bone
+        :param layout:
+        :param params:
+        :return:
+        """
+
+        pass