Rigify 0.5 with legacy mode

1 files changed, 128 insertions, 4 deletions

diff --git a/rigify/__init__.py b/rigify/__init__.py
index 6e6f751c..638260e1 100644
--- a/rigify/__init__.py
+++ b/rigify/__init__.py
@@ -20,9 +20,9 @@
 
 bl_info = {
     "name": "Rigify",
-    "version": (0, 4),
-    "author": "Nathan Vegdahl",
-    "blender": (2, 66, 0),
+    "version": (0, 5),
+    "author": "Nathan Vegdahl, Lucio Rossi, Ivan Cappiello",
+    "blender": (2, 78, 0),
     "description": "Automatic rigging from building-block components",
     "location": "Armature properties, Bone properties, View3d tools panel, Armature Add menu",
     "wiki_url": "http://wiki.blender.org/index.php/Extensions:2.5/Py/"
@@ -42,6 +42,120 @@ else:
     from . import utils, rig_lists, generate, ui, metarig_menu
 
 import bpy
+import sys
+import os
+from bpy.types import AddonPreferences
+from bpy.props import BoolProperty
+
+
+class RigifyPreferences(AddonPreferences):
+    # this must match the addon name, use '__package__'
+    # when defining this in a submodule of a python package.
+    bl_idname = __name__
+
+    def update_legacy(self, context):
+        if self.legacy_mode:
+
+            if 'ui' in globals() and 'legacy' in str(globals()['ui']):    # already in legacy mode. needed when rigify is reloaded
+                return
+            else:
+                rigify_dir = os.path.dirname(os.path.realpath(__file__))
+                if rigify_dir not in sys.path:
+                    sys.path.append(rigify_dir)
+
+                unregister()
+
+                globals().pop('utils')
+                globals().pop('rig_lists')
+                globals().pop('generate')
+                globals().pop('ui')
+                globals().pop('metarig_menu')
+
+                import legacy.utils
+                import legacy.rig_lists
+                import legacy.generate
+                import legacy.ui
+                import legacy.metarig_menu
+
+                print("ENTERING RIGIFY LEGACY\r\n")
+
+                globals()['utils'] = legacy.utils
+                globals()['rig_lists'] = legacy.rig_lists
+                globals()['generate'] = legacy.generate
+                globals()['ui'] = legacy.ui
+                globals()['metarig_menu'] = legacy.metarig_menu
+
+                register()
+
+        else:
+
+            rigify_dir = os.path.dirname(os.path.realpath(__file__))
+
+            if rigify_dir in sys.path:
+                id = sys.path.index(rigify_dir)
+                sys.path.pop(id)
+
+            unregister()
+
+            globals().pop('utils')
+            globals().pop('rig_lists')
+            globals().pop('generate')
+            globals().pop('ui')
+            globals().pop('metarig_menu')
+
+            from . import utils
+            from . import rig_lists
+            from . import generate
+            from . import ui
+            from . import metarig_menu
+
+            print("EXIT RIGIFY LEGACY\r\n")
+
+            globals()['utils'] = utils
+            globals()['rig_lists'] = rig_lists
+            globals()['generate'] = generate
+            globals()['ui'] = ui
+            globals()['metarig_menu'] = metarig_menu
+
+            register()
+
+    legacy_mode = BoolProperty(
+        name='Rigify Legacy Mode',
+        description='Select if you want to use Rigify in legacy mode',
+        default=False,
+        update=update_legacy
+    )
+
+    show_expanded = BoolProperty()
+
+    def draw(self, context):
+        layout = self.layout
+        column = layout.column()
+        box = column.box()
+
+        # first stage
+        expand = getattr(self, 'show_expanded')
+        icon = 'TRIA_DOWN' if expand else 'TRIA_RIGHT'
+        col = box.column()
+        row = col.row()
+        sub = row.row()
+        sub.context_pointer_set('addon_prefs', self)
+        sub.alignment = 'LEFT'
+        op = sub.operator('wm.context_toggle', text='', icon=icon,
+                          emboss=False)
+        op.data_path = 'addon_prefs.show_expanded'
+        sub.label('{}: {}'.format('Rigify', 'Enable Legacy Mode'))
+        sub = row.row()
+        sub.alignment = 'RIGHT'
+        sub.prop(self, 'legacy_mode')
+
+        if expand:
+            split = col.row().split(percentage=0.15)
+            split.label('Description:')
+            split.label(text='When enabled the add-on will run in legacy mode using the old 2.76b feature set.')
+
+        row = layout.row()
+        row.label("End of Rigify Preferences")
 
 
 class RigifyName(bpy.types.PropertyGroup):
@@ -66,6 +180,8 @@ def register():
     bpy.utils.register_class(RigifyName)
     bpy.utils.register_class(RigifyParameters)
     bpy.utils.register_class(RigifyArmatureLayer)
+    bpy.utils.register_class(RigifyPreferences)
+    bpy.types.Armature.rigify_layers = bpy.props.CollectionProperty(type=RigifyArmatureLayer)
 
     bpy.types.PoseBone.rigify_type = bpy.props.StringProperty(name="Rigify Type", description="Rig type for this bone")
     bpy.types.PoseBone.rigify_parameters = bpy.props.PointerProperty(type=RigifyParameters)
@@ -73,10 +189,17 @@ def register():
     bpy.types.Armature.rigify_layers = bpy.props.CollectionProperty(type=RigifyArmatureLayer)
 
     IDStore = bpy.types.WindowManager
-    IDStore.rigify_collection = bpy.props.EnumProperty(items=rig_lists.col_enum_list, default="All", name="Rigify Active Collection", description="The selected rig collection")
+    IDStore.rigify_collection = bpy.props.EnumProperty(items=rig_lists.col_enum_list, default="All",
+                                                       name="Rigify Active Collection",
+                                                       description="The selected rig collection")
+
     IDStore.rigify_types = bpy.props.CollectionProperty(type=RigifyName)
     IDStore.rigify_active_type = bpy.props.IntProperty(name="Rigify Active Type", description="The selected rig type")
 
+    if (ui and 'legacy' in str(ui)) or bpy.context.user_preferences.addons['rigify'].preferences.legacy_mode:
+        # update legacy on restart or reload
+        bpy.context.user_preferences.addons['rigify'].preferences.legacy_mode = True
+
     # Add rig parameters
     for rig in rig_lists.rig_list:
         r = utils.get_rig_type(rig)
@@ -98,6 +221,7 @@ def unregister():
     bpy.utils.unregister_class(RigifyName)
     bpy.utils.unregister_class(RigifyParameters)
     bpy.utils.unregister_class(RigifyArmatureLayer)
+    bpy.utils.unregister_class(RigifyPreferences)
 
     metarig_menu.unregister()
     ui.unregister()