fix: bendy tail ctrls not showing on chosen layers

+ control_layers_generator tweak_layers builder

Author: eigen-value

rigs/experimental/bendy_eye.py

@@ -477,7 +477,7 @@ def assign_layers(self):
         if 'common' in self.bones['eye_ctrl']:
             primary_ctrls.append(self.bones['eye_ctrl']['common'])
 
-        all_ctrls = self.control_snapper.flatten(self.bones['ctrl'])
+        all_ctrls = self.get_all_ctrls()
         self.layer_generator.assign_layer(primary_ctrls, all_ctrls)
 
     def make_constraints(self):

rigs/experimental/bendy_tail.py

@@ -112,6 +112,17 @@ def parent_bones(self):
 
         edit_bones[self.bones['tail_mch']['rot_tail_mch']].parent = edit_bones[tweak_chain[-1]]
 
+    def assign_layers(self):
+
+        primary_ctrls = []
+        primary_ctrls.append(self.bones['tail_ctrl']['tail_master'])
+
+        all_ctrls = self.get_all_ctrls()
+        self.layer_generator.assign_layer(primary_ctrls, all_ctrls)
+
+        tweaks = self.flatten(self.bones['tweaks'])
+        self.layer_generator.assign_tweak_layers(tweaks)
+
     def make_constraints(self):
 
         bpy.ops.object.mode_set(mode='OBJECT')
@@ -289,9 +300,11 @@ def add_parameters(params):
     """
 
     ControlLayersGenerator.add_layer_parameters(params)
+    ControlLayersGenerator.add_tweak_layer_parameters(params)
 
 
 def parameters_ui(layout, params):
     """ Create the ui for the rig parameters."""
 
     ControlLayersGenerator.add_layers_ui(layout, params)
+    ControlLayersGenerator.add_tweak_layers_ui(layout, params)

rigs/experimental/control_layers_generator.py

@@ -19,9 +19,14 @@ def __init__(self, rig):
         else:
             self.rig.secondary_layers = None
 
+        if self.rig.params.tweak_layers_extra:
+            self.rig.tweak_layers = list(self.rig.params.tweak_layers)
+        else:
+            self.rig.tweak_layers = None
+
     def assign_layer(self, primary_ctrls, all_ctrls):
         """
-        Assign bones to layer
+        Assign ctrl bones to layer
         :param primary_ctrls:
         :type primary_ctrls: list(str)
         :param all_ctrls:
@@ -39,17 +44,33 @@ def assign_layer(self, primary_ctrls, all_ctrls):
             if self.rig.secondary_layers and bone not in primary_ctrls:
                 edit_bones[bone].layers = self.rig.params.secondary_layers
 
+    def assign_tweak_layers(self, tweaks):
+        """
+        Assign tweak bones to layer
+        :param tweaks:
+        :type tweaks: list(str)
+        :return:
+        """
+
+        bpy.ops.object.mode_set(mode='EDIT')
+        edit_bones = self.obj.data.edit_bones
+
+        for bone in tweaks:
+            if self.rig.tweak_layers:
+                edit_bones[bone].layers = self.rig.params.tweak_layers
+
+
     @staticmethod
     def add_layer_parameters(params):
-        # Setting up extra layers for the tweak bones
+        # Setting up extra layers for the primary & secondary controls
         params.primary_layers_extra = bpy.props.BoolProperty(
             name="primary_layers_extra",
             default=True,
             description=""
         )
         params.primary_layers = bpy.props.BoolVectorProperty(
             size=32,
-            description="Layers for the 1st tweak controls to be on",
+            description="Layers for the primary controls to be on",
             default=tuple([i == 1 for i in range(0, 32)])
         )
         params.secondary_layers_extra = bpy.props.BoolProperty(
@@ -59,7 +80,20 @@ def add_layer_parameters(params):
         )
         params.secondary_layers = bpy.props.BoolVectorProperty(
             size=32,
-            description="Layers for the 2nd tweak controls to be on",
+            description="Layers for the secondary controls to be on",
+            default=tuple([i == 1 for i in range(0, 32)])
+        )
+
+    @staticmethod
+    def add_tweak_layer_parameters(params):
+        params.tweak_layers_extra = bpy.props.BoolProperty(
+            name="tweak_layers_extra",
+            default=True,
+            description=""
+        )
+        params.tweak_layers = bpy.props.BoolVectorProperty(
+            size=32,
+            description="Layers for the tweak controls to be on",
             default=tuple([i == 1 for i in range(0, 32)])
         )
 
@@ -104,3 +138,44 @@ def add_layers_ui(layout, params):
                 if bone_layers[i]:
                     icon = "LAYER_ACTIVE"
                 row.prop(params, layer, index=i, toggle=True, text="", icon=icon)
+
+    @staticmethod
+    def add_tweak_layers_ui(layout, params):
+
+        layer = "tweak_layers"
+        bone_layers = bpy.context.active_pose_bone.bone.layers[:]
+
+        r = layout.row()
+        r.prop(params, layer + "_extra")
+        r.active = getattr(params, layer + "_extra")
+
+        col = r.column(align=True)
+        row = col.row(align=True)
+        for i in range(8):
+            icon = "NONE"
+            if bone_layers[i]:
+                icon = "LAYER_ACTIVE"
+            row.prop(params, layer, index=i, toggle=True, text="", icon=icon)
+
+        row = col.row(align=True)
+        for i in range(16, 24):
+            icon = "NONE"
+            if bone_layers[i]:
+                icon = "LAYER_ACTIVE"
+            row.prop(params, layer, index=i, toggle=True, text="", icon=icon)
+
+        col = r.column(align=True)
+        row = col.row(align=True)
+
+        for i in range(8, 16):
+            icon = "NONE"
+            if bone_layers[i]:
+                icon = "LAYER_ACTIVE"
+            row.prop(params, layer, index=i, toggle=True, text="", icon=icon)
+
+        row = col.row(align=True)
+        for i in range(24, 32):
+            icon = "NONE"
+            if bone_layers[i]:
+                icon = "LAYER_ACTIVE"
+            row.prop(params, layer, index=i, toggle=True, text="", icon=icon)