Support putting tweak controls on different layers for super_finger.

Author: angavrilov

rigs/limbs/super_finger.py

@@ -3,6 +3,7 @@
 from ...utils import strip_org, make_deformer_name, connected_children_names, make_mechanism_name
 from ...utils import create_circle_widget, create_widget
 from ...utils import MetarigError, align_bone_x_axis
+from ...utils import layout_layer_selection_ui
 from rna_prop_ui import rna_idprop_ui_prop_get
 
 script = """
@@ -20,6 +21,11 @@ def __init__(self, obj, bone_name, params):
         self.org_bones = [bone_name] + connected_children_names(obj, bone_name)
         self.params = params
 
+        if params.tweak_extra_layers:
+            self.tweak_layers = list(params.tweak_layers)
+        else:
+            self.tweak_layers = None
+
         if len(self.org_bones) <= 1:
             raise MetarigError("RIGIFY ERROR: Bone '%s': listen bro, that finger rig jusaint put tugetha rite. A little hint, use more than one bone!!" % (strip_org(bone_name)))
 
@@ -303,6 +309,9 @@ def generate(self):
             # Assigning shapes to control bones
             create_circle_widget(self.obj, ctrl, radius=0.3, head_tail=0.5)
 
+            if self.tweak_layers:
+                pb[ctrl].bone.layers = self.tweak_layers
+
         # Create ctrl master widget
         w = create_widget(self.obj, master_name)
         if w is not None:
@@ -321,6 +330,9 @@ def generate(self):
         # Create tip control widget
         create_circle_widget(self.obj, tip_name, radius=0.3, head_tail=0.0)
 
+        if self.tweak_layers:
+            pb[tip_name].bone.layers = self.tweak_layers
+
         # Create UI
         controls_string = ", ".join(
             ["'" + x + "'" for x in ctrl_chain]
@@ -336,6 +348,19 @@ def add_parameters(params):
              ('-X', '-X manual', ''), ('-Y', '-Y manual', ''), ('-Z', '-Z manual', '')]
     params.primary_rotation_axis = bpy.props.EnumProperty(items=items, name="Primary Rotation Axis", default='automatic')
 
+    # Setting up extra tweak layers
+    params.tweak_extra_layers = bpy.props.BoolProperty(
+        name="tweak_extra_layers",
+        default=False,
+        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)])
+        )
+
 
 def parameters_ui(layout, params):
     """ Create the ui for the rig parameters.
@@ -344,6 +369,7 @@ def parameters_ui(layout, params):
     r.label(text="Bend rotation axis:")
     r.prop(params, "primary_rotation_axis", text="")
 
+    layout_layer_selection_ui(layout, params, "tweak_extra_layers", "tweak_layers")
 
 def create_sample(obj):
     # generated by rigify.utils.write_metarig

utils.py

@@ -1427,3 +1427,50 @@ def make_track_constraint_from_string(owner, target, subtarget, fstring):
         const.target_space = constraint_space[cns_props[3][0]] if bool(cns_props[3]) else "LOCAL"
         const.owner_space = constraint_space[cns_props[3][1]] if bool(cns_props[3]) else "LOCAL"
         const.head_tail = float(cns_props[4]) if bool(cns_props[4]) else 0.0
+
+
+#=============================================
+# UI utilities
+#=============================================
+
+def layout_layer_selection_ui(layout, params, enable_option, layers_option):
+    r = layout.row()
+    r.prop(params, enable_option)
+    r.active = getattr(params, enable_option)
+
+    col = r.column(align=True)
+    row = col.row(align=True)
+
+    bone_layers = bpy.context.active_pose_bone.bone.layers[:]
+
+    for i in range(8):    # Layers 0-7
+        icon = "NONE"
+        if bone_layers[i]:
+            icon = "LAYER_ACTIVE"
+        row.prop(params, layers_option, index=i, toggle=True, text="", icon=icon)
+
+    row = col.row(align=True)
+
+    for i in range(16, 24):     # Layers 16-23
+        icon = "NONE"
+        if bone_layers[i]:
+            icon = "LAYER_ACTIVE"
+        row.prop(params, layers_option, index=i, toggle=True, text="", icon=icon)
+
+    col = r.column(align=True)
+    row = col.row(align=True)
+
+    for i in range(8, 16):  # Layers 8-15
+        icon = "NONE"
+        if bone_layers[i]:
+            icon = "LAYER_ACTIVE"
+        row.prop(params, layers_option, index=i, toggle=True, text="", icon=icon)
+
+    row = col.row(align=True)
+
+    for i in range(24, 32):     # Layers 24-31
+        icon = "NONE"
+        if bone_layers[i]:
+            icon = "LAYER_ACTIVE"
+        row.prop(params, layers_option, index=i, toggle=True, text="", icon=icon)
+