From 0ff37c1640bd79a3a9b43cb6fb886480f85fcabd Mon Sep 17 00:00:00 2001
From: Nathan Vegdahl <cessen@cessen.com>
Date: Tue, 28 Jun 2011 22:47:44 +0000
Subject: Rigify: fixed bug where generating failed on metarig with no
 animation data.

---
 rigify/generate.py | 79 +++++++++++++++++++++++++++---------------------------
 1 file changed, 40 insertions(+), 39 deletions(-)

(limited to 'rigify/generate.py')

diff --git a/rigify/generate.py b/rigify/generate.py
index a2ac6f7a..153162bc 100644
--- a/rigify/generate.py
+++ b/rigify/generate.py
@@ -168,45 +168,46 @@ def generate_rig(context, metarig):
                     con2.target = obj
 
     # Copy drivers
-    for d1 in metarig.animation_data.drivers:
-        d2 = obj.driver_add(d1.data_path)
-        copy_attributes(d1, d2)
-        copy_attributes(d1.driver, d2.driver)
-
-        # Remove default modifiers, variables, etc.
-        for m in d2.modifiers:
-            d2.modifiers.remove(m)
-        for v in d2.driver.variables:
-            d2.driver.variables.remove(v)
-
-        # Copy modifiers
-        for m1 in d1.modifiers:
-            m2 = d2.modifiers.new(type=m1.type)
-            copy_attributes(m1, m2)
-
-        # Copy variables
-        for v1 in d1.driver.variables:
-            v2 = d2.driver.variables.new()
-            copy_attributes(v1, v2)
-            for i in range(len(v1.targets)):
-                copy_attributes(v1.targets[i], v2.targets[i])
-                # Switch metarig targets to rig targets
-                if v2.targets[i].id == metarig:
-                    v2.targets[i].id = obj
-
-                # Mark targets that may need to be altered after rig generation
-                tar = v2.targets[i]
-                # If a custom property
-                if v2.type == 'SINGLE_PROP' \
-                and re.match('^pose.bones\["[^"\]]*"\]\["[^"\]]*"\]$', tar.data_path):
-                    tar.data_path = "RIGIFY-" + tar.data_path
-
-        # Copy key frames
-        for i in range(len(d1.keyframe_points)):
-            d2.keyframe_points.add()
-            k1 = d1.keyframe_points[i]
-            k2 = d2.keyframe_points[i]
-            copy_attributes(k1, k2)
+    if metarig.animation_data:
+        for d1 in metarig.animation_data.drivers:
+            d2 = obj.driver_add(d1.data_path)
+            copy_attributes(d1, d2)
+            copy_attributes(d1.driver, d2.driver)
+
+            # Remove default modifiers, variables, etc.
+            for m in d2.modifiers:
+                d2.modifiers.remove(m)
+            for v in d2.driver.variables:
+                d2.driver.variables.remove(v)
+
+            # Copy modifiers
+            for m1 in d1.modifiers:
+                m2 = d2.modifiers.new(type=m1.type)
+                copy_attributes(m1, m2)
+
+            # Copy variables
+            for v1 in d1.driver.variables:
+                v2 = d2.driver.variables.new()
+                copy_attributes(v1, v2)
+                for i in range(len(v1.targets)):
+                    copy_attributes(v1.targets[i], v2.targets[i])
+                    # Switch metarig targets to rig targets
+                    if v2.targets[i].id == metarig:
+                        v2.targets[i].id = obj
+
+                    # Mark targets that may need to be altered after rig generation
+                    tar = v2.targets[i]
+                    # If a custom property
+                    if v2.type == 'SINGLE_PROP' \
+                    and re.match('^pose.bones\["[^"\]]*"\]\["[^"\]]*"\]$', tar.data_path):
+                        tar.data_path = "RIGIFY-" + tar.data_path
+
+            # Copy key frames
+            for i in range(len(d1.keyframe_points)):
+                d2.keyframe_points.add()
+                k1 = d1.keyframe_points[i]
+                k2 = d2.keyframe_points[i]
+                copy_attributes(k1, k2)
 
     t.tick("Duplicate rig: ")
     #----------------------------------
-- 
cgit v1.2.3