From cb51710f25fe6736328d6a1b5c1cfccbfd7e567c Mon Sep 17 00:00:00 2001
From: Campbell Barton <ideasman42@gmail.com>
Date: Mon, 23 Nov 2009 14:58:27 +0000
Subject: [PATCH] pose_bone attributes, children, children_recursive,
 parent_recursive & parent_index() function.

---
 release/scripts/modules/bpy_types.py | 52 ++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)

diff --git a/release/scripts/modules/bpy_types.py b/release/scripts/modules/bpy_types.py
index 70fecec0235..221e4d29ae8 100644
--- a/release/scripts/modules/bpy_types.py
+++ b/release/scripts/modules/bpy_types.py
@@ -42,6 +42,58 @@ class Object(bpy_types.ID):
     children = property(_get_children)
 
 
+class PoseBone(StructRNA):
+
+    def _get_children(self):
+        import bpy
+        obj = self.id_data
+        return [child for child in obj.pose.bones if child.parent == self]
+
+    children = property(_get_children)
+
+    def _get_parent_recursive(self):
+        parent_list = []
+        parent = self.parent
+        
+        while parent:
+            if parent:
+                parent_list.append(parent)
+            
+            parent = parent.parent
+        
+        return parent_list
+
+    parent_recursive = property(_get_parent_recursive)
+    
+    def parent_index(self, parent_test):
+        '''
+        The same as 'bone in other_bone.parent_recursive' but saved generating a list.
+        '''
+        parent = self.parent
+        i = 1
+        while parent:
+            if parent == parent_test:
+                return i
+            parent = parent.parent
+            i += 1
+        
+        return 0
+    
+    def _get_children_recursive(self):
+        obj = self.id_data
+        bones_children = []
+        for bone in obj.pose.bones:
+            index = bone.parent_index(self)
+            if index:
+                bones_children.append((index, bone))
+        
+        # sort by distance to parent
+        bones_children.sort(key=lambda bone_pair: bone_pair[0])
+        return [bone for index, bone in bones_children]
+
+    children_recursive = property(_get_children_recursive)
+
+
 def ord_ind(i1,i2):
     if i1<i2: return i1,i2
     return i2,i1