Rigify: internal API enhancements.

Add a LazyRigComponent class that has to be explicitly enabled
to start receiving callbacks.

Support calling GeneratorPlugin instance callbacks in the same
stage they were created.

1 files changed, 25 insertions, 2 deletions

diff --git a/rigify/base_generate.py b/rigify/base_generate.py
index ce04b8fc..da4949b2 100644
--- a/rigify/base_generate.py
+++ b/rigify/base_generate.py
@@ -31,6 +31,7 @@ from .utils.misc import clone_parameters, assign_parameters
 
 from . import base_rig
 
+from itertools import count
 
 #=============================================
 # Generator Plugin
@@ -284,13 +285,24 @@ class BaseGenerator:
 
         self.stage = method_name
 
-        for rig in [*self.rig_list, *self.plugin_list]:
+        for rig in self.rig_list:
             rig.rigify_invoke_stage(method_name)
 
             assert(self.context.active_object == self.obj)
             assert(self.obj.mode == 'OBJECT')
             assert(num_bones == len(self.obj.data.bones))
 
+        # Allow plugins to be added to the end of the list on the fly
+        for i in count(0):
+            if i >= len(self.plugin_list):
+                break
+
+            self.plugin_list[i].rigify_invoke_stage(method_name)
+
+            assert(self.context.active_object == self.obj)
+            assert(self.obj.mode == 'OBJECT')
+            assert(num_bones == len(self.obj.data.bones))
+
 
     def __run_edit_stage(self, method_name):
         assert(self.context.active_object == self.obj)
@@ -299,13 +311,24 @@ class BaseGenerator:
 
         self.stage = method_name
 
-        for rig in [*self.rig_list, *self.plugin_list]:
+        for rig in self.rig_list:
             rig.rigify_invoke_stage(method_name)
 
             assert(self.context.active_object == self.obj)
             assert(self.obj.mode == 'EDIT')
             assert(num_bones == len(self.obj.data.edit_bones))
 
+        # Allow plugins to be added to the end of the list on the fly
+        for i in count(0):
+            if i >= len(self.plugin_list):
+                break
+
+            self.plugin_list[i].rigify_invoke_stage(method_name)
+
+            assert(self.context.active_object == self.obj)
+            assert(self.obj.mode == 'EDIT')
+            assert(num_bones == len(self.obj.data.edit_bones))
+
 
     def invoke_initialize(self):
         self.__run_object_stage('initialize')