[Bf-extensions-cvs] SVN commit: /data/svn/bf-extensions [4272] trunk/py/scripts/addons/rigify/ ui.py: add button to initialize armature layers, since this isn' t allowed on draw anymore.

[Campbell Barton]

Revision: 4272
          http://projects.blender.org/scm/viewvc.php?view=rev&root=bf-extensions&revision=4272
Author:   campbellbarton
Date:     2013-02-15 03:48:46 +0000 (Fri, 15 Feb 2013)
Log Message:
-----------
add button to initialize armature layers, since this isn't allowed on draw anymore.

Modified Paths:
--------------
    trunk/py/scripts/addons/rigify/ui.py

Modified: trunk/py/scripts/addons/rigify/ui.py
===================================================================
--- trunk/py/scripts/addons/rigify/ui.py	2013-02-15 00:09:26 UTC (rev 4271)
+++ trunk/py/scripts/addons/rigify/ui.py	2013-02-15 03:48:46 UTC (rev 4272)
@@ -101,8 +101,14 @@
         obj = context.object
 
         # Ensure that the layers exist
-        for i in range(1 + len(obj.data.rigify_layers), 29):
-            obj.data.rigify_layers.add()
+        if 0:
+            for i in range(1 + len(obj.data.rigify_layers), 29):
+                obj.data.rigify_layers.add()
+        else:
+            # Can't add while drawing, just use button
+            if len(obj.data.rigify_layers) < 28:
+                layout.operator("pose.rigify_layer_init")
+                return
 
         # UI
         for i in range(28):
@@ -229,6 +235,20 @@
     operator.report({'INFO'}, '\n'.join(message))
 
 
+class LayerInit(bpy.types.Operator):
+    """Initialize armature rigify layers"""
+
+    bl_idname = "pose.rigify_layer_init"
+    bl_label = "Add Rigify Layers"
+    bl_options = {'UNDO'}
+
+    def execute(self, context):
+        obj = context.object
+        for i in range(1 + len(obj.data.rigify_layers), 29):
+            obj.data.rigify_layers.add()
+        return {'FINISHED'}
+
+
 class Generate(bpy.types.Operator):
     """Generates a rig from the active metarig armature"""
 
@@ -292,6 +312,7 @@
     bpy.utils.register_class(DATA_PT_rigify_layer_names)
     bpy.utils.register_class(DATA_PT_rigify_buttons)
     bpy.utils.register_class(BONE_PT_rigify_buttons)
+    bpy.utils.register_class(LayerInit)
     bpy.utils.register_class(Generate)
     bpy.utils.register_class(Sample)
 
@@ -302,5 +323,6 @@
     bpy.utils.unregister_class(DATA_PT_rigify_layer_names)
     bpy.utils.unregister_class(DATA_PT_rigify_buttons)
     bpy.utils.unregister_class(BONE_PT_rigify_buttons)
+    bpy.utils.unregister_class(LayerInit)
     bpy.utils.unregister_class(Generate)
     bpy.utils.unregister_class(Sample)