Rigify: support choosing widgets out of a list in super_copy and pivot.

Register the most generic widgets in the list and add a few more.
Use it to assign a nicer looking shoulder widget in human metarigs.

1 files changed, 5 insertions, 4 deletions

diff --git a/rigify/generate.py b/rigify/generate.py
index f1586873..613ef59f 100644
--- a/rigify/generate.py
+++ b/rigify/generate.py
@@ -262,13 +262,14 @@ class Generator(base_generate.BaseGenerator):
 
 
     def __assign_layers(self):
-        bones = self.obj.data.bones
+        pbones = self.obj.pose.bones
 
-        bones[self.root_bone].layers = ROOT_LAYER
+        pbones[self.root_bone].bone.layers = ROOT_LAYER
 
         # Every bone that has a name starting with "DEF-" make deforming.  All the
         # others make non-deforming.
-        for bone in bones:
+        for pbone in pbones:
+            bone = pbone.bone
             name = bone.name
             layers = None
 
@@ -288,7 +289,7 @@ class Generator(base_generate.BaseGenerator):
                 bone.layers = layers
 
                 # Remove custom shapes from non-control bones
-                bone.custom_shape = None
+                pbone.custom_shape = None
 
             bone.bbone_x = bone.bbone_z = bone.length * 0.05