Bone Selection Groups commit from gooseberry.

This add-on allows storing and restoring a group of bones for later use.
The groups can be named and linked to new files.

The groups can be found in the object properties tab of the properties
area for armature objects.

1 files changed, 244 insertions, 0 deletions

diff --git a/bone_selection_groups.py b/bone_selection_groups.py
new file mode 100644
index 00000000..1c00bb13
--- /dev/null
+++ b/bone_selection_groups.py
@@ -0,0 +1,244 @@
+# ***** BEGIN GPL LICENSE BLOCK *****
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of the GNU General Public License
+# as published by the Free Software Foundation; either version 2
+# of the License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software Foundation,
+# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+#
+# ***** END GPL LICENCE BLOCK *****
+
+bl_info = {
+    "name": "Bone Selection Groups",
+    "author": "Antony Riakiotakis",
+    "version": (1, 0, 2),
+    "blender": (2, 75, 0),
+    "location": "Properties > Object Buttons",
+    "description": "Operator and storage for restoration of bone selection state.",
+    "category": "Animation",
+}
+
+import bpy
+from bpy.types import (
+    Operator,
+    Panel,
+    UIList,
+)
+
+from bpy.props import (
+    StringProperty,
+    BoolProperty,
+    IntProperty,
+    FloatProperty,
+    EnumProperty,
+    CollectionProperty,
+    BoolVectorProperty,
+    FloatVectorProperty,
+)
+
+
+class POSE_PT_selection_sets(Panel):
+    bl_label = "Selection Sets"
+    bl_space_type = 'PROPERTIES'
+    bl_region_type = 'WINDOW'
+    bl_context = "data"
+
+    @classmethod
+    def poll(cls, context):
+        return context.object.type == 'ARMATURE'
+
+    def draw(self, context):
+        layout = self.layout
+
+        armature = context.object
+
+        # Rig type list
+        row = layout.row()
+        row.template_list(
+            "POSE_UL_selection_set", "",
+            armature, "selection_sets",
+            armature, "active_selection_set")
+
+        col = row.column()
+        colsub = col.column(align=True)
+        colsub.operator("pose.selection_set_add", icon='ZOOMIN', text="")
+        colsub.operator("pose.selection_set_remove", icon='ZOOMOUT', text="")
+
+        layout.operator("pose.selection_set_toggle")
+
+
+class POSE_UL_selection_set(UIList):
+    def draw_item(self, context, layout, data, set, icon, active_data, active_propname, index):
+        layout.prop(set, "name", text="", emboss=False)
+
+
+class SelectionEntry(bpy.types.PropertyGroup):
+    name = StringProperty(name="Bone Name")
+
+
+class SelectionSet(bpy.types.PropertyGroup):
+    name = StringProperty(name="Set Name")
+    bone_ids = CollectionProperty(type=SelectionEntry)
+
+
+class PluginOperator(Operator):
+    @classmethod
+    def poll(self, context):
+        return (context.object and
+                context.object.type == 'ARMATURE' and
+                context.mode == 'POSE')
+
+
+class POSE_OT_selection_set_add(PluginOperator):
+    bl_idname = "pose.selection_set_add"
+    bl_label = "Add Selection Set"
+    bl_options = {'UNDO', 'REGISTER'}
+
+    def execute(self, context):
+        keep = False
+        armature = context.object
+        pose = armature.pose
+
+        selection_set = armature.selection_sets.add()
+        selection_set.name = "SelectionSet.%d" % len(armature.selection_sets)
+        armature.active_selection_set = len(armature.selection_sets) - 1
+        for bone in pose.bones:
+            if (bone.bone.select):
+                bone_id = selection_set.bone_ids.add()
+                bone_id.name = bone.name
+                keep = True
+
+        if (not keep):
+            armature.selection_sets.remove(armature.active_selection_set)
+            numsets = len(armature.selection_sets)
+            if (armature.active_selection_set > (numsets - 1) and numsets > 0):
+                armature.active_selection_set = len(armature.selection_sets) - 1
+            return {'CANCELLED'}
+
+        return {'FINISHED'}
+
+
+class POSE_OT_selection_set_remove(PluginOperator):
+    bl_idname = "pose.selection_set_remove"
+    bl_label = "Delete Selection Set"
+    bl_options = {'UNDO', 'REGISTER'}
+
+    def execute(self, context):
+        armature = context.object
+
+        armature.selection_sets.remove(armature.active_selection_set)
+        numsets = len(armature.selection_sets)
+        if (armature.active_selection_set > (numsets - 1) and numsets > 0):
+            armature.active_selection_set = len(armature.selection_sets) - 1
+        return {'FINISHED'}
+
+
+class POSE_OT_selection_set_toggle(PluginOperator):
+    bl_idname = "pose.selection_set_toggle"
+    bl_label = "Toggle Selection Set"
+    bl_options = {'UNDO', 'REGISTER'}
+
+    def execute(self, context):
+        armature = context.object
+        pose = armature.pose
+
+        selection_set = armature.selection_sets[armature.active_selection_set]
+        for bone in pose.bones:
+            bone.bone.select = False
+
+        for bone in selection_set.bone_ids:
+            pose.bones[bone.name].bone.select = True
+
+        return {'FINISHED'}
+
+
+class MotionPathsCopyStartFrame(Operator):
+    bl_idname = "anim.motionpaths_copy_scene_startframe"
+    bl_label = "Copy Scene Start Frame"
+    bl_options = {'REGISTER', 'UNDO'}
+
+    armature_paths = BoolProperty()
+
+    @classmethod
+    def poll(cls, context):
+        return (context.object)
+
+    def execute(self, context):
+        avs = None
+        motionpath = None
+        ob = context.object
+        scene = context.scene
+
+        if (self.armature_paths):
+            avs = ob.pose.animation_visualization
+        else:
+            avs = ob.animation_visualization
+
+        preview = scene.use_preview_range
+
+        if (avs):
+            motionpath = avs.motion_path
+
+        if (motionpath):
+            if (preview):
+                motionpath.frame_start = scene.frame_preview_start
+                motionpath.frame_end = scene.frame_preview_end
+            else:
+                motionpath.frame_start = scene.frame_start
+                motionpath.frame_end = scene.frame_end
+        else:
+            return {'CANCELLED'}
+
+        return {'FINISHED'}
+
+
+classes = (
+    POSE_PT_selection_sets,
+    POSE_UL_selection_set,
+    SelectionEntry,
+    SelectionSet,
+    POSE_OT_selection_set_add,
+    POSE_OT_selection_set_remove,
+    POSE_OT_selection_set_toggle,
+    MotionPathsCopyStartFrame
+)
+
+
+def copy_frames_armature(self, context):
+    layout = self.layout
+    layout.operator("anim.motionpaths_copy_scene_startframe").armature_paths = True
+
+
+def copy_frames_object(self, context):
+    layout = self.layout
+    layout.operator("anim.motionpaths_copy_scene_startframe").armature_paths = False
+
+
+def register():
+    for cls in classes:
+        bpy.utils.register_class(cls)
+
+    bpy.types.Object.selection_sets = CollectionProperty(type=SelectionSet)
+    bpy.types.Object.active_selection_set = IntProperty()
+    bpy.types.DATA_PT_motion_paths.append(copy_frames_armature)
+    bpy.types.OBJECT_PT_motion_paths.append(copy_frames_object)
+
+
+def unregister():
+    for cls in classes:
+        bpy.utils.unregister_class(cls)
+
+    del bpy.types.Object.selection_sets
+    del bpy.types.Object.active_selection_set
+
+
+if __name__ == "__main__":
+    register()