Check for keyframes before applying pose

This commit is contained in:
2025-01-18 21:41:24 -06:00
parent cb3999de33
commit 6fbfc22956
5 changed files with 936 additions and 923 deletions

View File

@ -1,48 +1,48 @@
# Damn Simple Pose Library # Damn Simple Pose Library
In Blender 3.x, a new Asset-based Pose system was introduced. That's fine and dandy, but it was not (and still is not as of Jan 2025) ready to replace the old Action-based Pose Library system. In Blender 3.x, a new Asset-based Pose system was introduced. That's fine and dandy, but it was not (and still is not as of Jan 2025) ready to replace the old Action-based Pose Library system.
The former Pose Library was deprecated and gutted rapidly during Blender 3.x, leaving legacy users high and dry. This addon brings the feature back as best as it can in a modern panel. Most, if not all existing Pose Libraries should work without any modifications, little new is introduced outside of string suffixes and mitigations (read further). The former Pose Library was deprecated and gutted rapidly during Blender 3.x, leaving legacy users high and dry. This addon brings the feature back as best as it can in a modern panel. Most, if not all existing Pose Libraries should work without any modifications, little new is introduced outside of string suffixes and mitigations (read further).
## Features ## Features
- Supports Blender 3.3 - 4.3 (4.1 recommended). - Supports Blender 3.3 - 4.3 (4.1 recommended).
- New panel interface for interacting with pose libraries, inspired by [gret's Actions Panel feature](https://github.com/greisane/gret?tab=readme-ov-file#animation-actions-panel). - New panel interface for interacting with pose libraries, inspired by [gret's Actions Panel feature](https://github.com/greisane/gret?tab=readme-ov-file#animation-actions-panel).
- The older Pose Library list layout is provided as option. - The older Pose Library list layout is provided as option.
- Operators and data property [that were removed in 3.5](https://projects.blender.org/blender/blender/issues/93406) are ported from C to Python: - Operators and data property [that were removed in 3.5](https://projects.blender.org/blender/blender/issues/93406) are ported from C to Python:
- `dspl.apply_pose` - `dspl.apply_pose`
- `dspl.browse_poses` - `dspl.browse_poses`
- `dspl.create_pose_library` - `dspl.create_pose_library`
- `dspl.convert_pose_library` - `dspl.convert_pose_library`
- `dspl.add_pose` - `dspl.add_pose`
- `dspl.move_pose` - `dspl.move_pose`
- `dspl.remove_pose` - `dspl.remove_pose`
- `dspl.rename_pose` - `dspl.rename_pose`
- `dspl.unlink_pose_library` - `dspl.unlink_pose_library`
- `Object.pose_library` - `Object.pose_library`
## Installation ## Installation
1. [Download the repository as a zip](https://git.bkspl.me/breakingspell/DamnSimplePoseLibrary/archive/develop.zip), or otherwise clone the repository. 1. [Download the repository as a zip](https://git.bkspl.me/breakingspell/DamnSimplePoseLibrary/archive/develop.zip), or otherwise clone the repository.
2. Install as an Add-on in Blender via Install -> Zip, and enable. 2. Install as an Add-on in Blender via Install -> Zip, and enable.
3. Optionally configure the suffix strings to fit your workflow, and whether the orhpan checker should run at startup. 3. Optionally configure the suffix strings to fit your workflow, and whether the orhpan checker should run at startup.
## Usage ## Usage
1. Open Sidebar (`N`), and choose the `Pose` tab. 1. Open Sidebar (`N`), and choose the `Pose` tab.
2. Select existing Pose Library from drop-down, or create a new library. 2. Select existing Pose Library from drop-down, or create a new library.
### Hotkeys: ### Hotkeys:
- `Shift + L` - Add/Replace Pose - `Shift + L` - Add/Replace Pose
- `Alt + L` - Browse Poses with Arrow Keys - `Alt + L` - Browse Poses with Arrow Keys
### Menu Controls ### Menu Controls
- `Single click` - Apply Pose - `Single click` - Apply Pose
- `Shift + Click` - Rename Pose - `Shift + Click` - Rename Pose
- `Alt + Click` - Remove Pose - `Alt + Click` - Remove Pose
- `Ctrl + Click ` - Select Pose - `Ctrl + Click ` - Select Pose
- Choose `Edit` for fast Move/Rename/Removal - Choose `Edit` for fast Move/Rename/Removal
## Considerations ## Considerations
- Opening older scenes will cause existing Pose Libraries to unlink from their former targets and fall into orphan state. They can be re-linked and will retain their link when saved afterwards, but would otherwise disappear if saved without linking or protecting the datablock. - Opening older scenes will cause existing Pose Libraries to unlink from their former targets and fall into orphan state. They can be re-linked and will retain their link when saved afterwards, but would otherwise disappear if saved without linking or protecting the datablock.
- This is due to the core DNA type `poselib` having been removed, so objects will drop the data. [There is no way to prevent this](https://developer.blender.org/docs/features/core/rna/#internals) as Python ID properties cannot be present at program runtime, only once the add-on is initialized. - This is due to the core DNA type `poselib` having been removed, so objects will drop the data. [There is no way to prevent this](https://developer.blender.org/docs/features/core/rna/#internals) as Python ID properties cannot be present at program runtime, only once the add-on is initialized.
- To mitgate, an operator is provided to protect orphaned pose libraries: `dspl.protect_orphan_pose_library`. - To mitgate, an operator is provided to protect orphaned pose libraries: `dspl.protect_orphan_pose_library`.
- This add-on was made much easier by the `pose_markers` property being retained for converting old pose libraries to the new asset system. If they decide to remove that property as well, there will be a need to improvise an index-based lookup. - This add-on was made much easier by the `pose_markers` property being retained for converting old pose libraries to the new asset system. If they decide to remove that property as well, there will be a need to improvise an index-based lookup.

409
common.py
View File

@ -1,199 +1,212 @@
import bpy import bpy
import mathutils import mathutils
def getArmatureData(context): def getArmatureData(context):
try: try:
arm_object = context.active_object arm_object = context.active_object
if arm_object and arm_object.type == "ARMATURE": if arm_object and arm_object.type == "ARMATURE":
return arm_object, getattr(arm_object, "pose_library", None) return arm_object, getattr(arm_object, "pose_library", None)
elif arm_object and arm_object.parent and arm_object.parent.type == "ARMATURE": elif arm_object and arm_object.parent and arm_object.parent.type == "ARMATURE":
return arm_object.parent, getattr(arm_object.parent, "pose_library", None) return arm_object.parent, getattr(arm_object.parent, "pose_library", None)
else: else:
return None, None return None, None
except: except:
return None, None return None, None
def getArmatureAction(context): def getArmatureAction(context):
try: try:
arm_object, pose_library = getArmatureData(context) arm_object, pose_library = getArmatureData(context)
if getattr(arm_object.animation_data.action, "pose_markers", None): if getattr(arm_object.animation_data.action, "pose_markers", None):
return getattr(arm_object.animation_data, "action", None) return getattr(arm_object.animation_data, "action", None)
except: except:
pass pass
def searchPoseMarker(context, posename, type): def searchPoseMarker(context, posename, type):
try: try:
arm_object, pose_library = getArmatureData(context) arm_object, pose_library = getArmatureData(context)
if type == "marker": if type == "marker":
return pose_library.pose_markers.get(posename, None) return pose_library.pose_markers.get(posename, None)
if type == "frame": if type == "frame":
return pose_library.pose_markers.get(posename, None).frame return pose_library.pose_markers.get(posename, None).frame
if type == "index": if type == "index":
return pose_library.pose_markers.find(posename) return pose_library.pose_markers.find(posename)
except: except:
pass pass
def findFcurve(context, bone_name, transform, index_int): def findFcurve(context, bone_name, transform, index_int):
arm_object, pose_library = getArmatureData(context) arm_object, pose_library = getArmatureData(context)
pose_markers = pose_library.pose_markers pose_markers = pose_library.pose_markers
active_frame = pose_markers.active.frame active_frame = pose_markers.active.frame
fcurve_object = pose_library.fcurves.find( fcurve_object = pose_library.fcurves.find(
'pose.bones["'+bone_name+'"].'+transform+'', index=index_int) 'pose.bones["'+bone_name+'"].'+transform+'', index=index_int)
if hasattr(fcurve_object, 'evaluate'): if hasattr(fcurve_object, 'evaluate'):
return fcurve_object.evaluate(active_frame) return fcurve_object.evaluate(active_frame)
else: else:
return None return None
def createFcurve(context, bone_name, transform, index_int): def createFcurve(context, bone_name, transform, index_int):
arm_object, pose_library = getArmatureData(context) arm_object, pose_library = getArmatureData(context)
pose_markers = pose_library.pose_markers pose_markers = pose_library.pose_markers
try: try:
pose_library.fcurves.new( pose_library.fcurves.new(
'pose.bones["'+bone_name+'"].'+transform+'', index=index_int, action_group=bone_name) 'pose.bones["'+bone_name+'"].'+transform+'', index=index_int, action_group=bone_name)
return return
except: except:
pass pass
def createKeyframe(context, bone_name, transform, index_int, new_marker, loc): def createKeyframe(context, bone_name, transform, index_int, new_marker, loc):
arm_object, pose_library = getArmatureData(context) arm_object, pose_library = getArmatureData(context)
pose_markers = pose_library.pose_markers pose_markers = pose_library.pose_markers
try: try:
pose_library.fcurves.find( pose_library.fcurves.find(
'pose.bones["'+bone_name+'"].'+transform+'', index=index_int).keyframe_points.insert(new_marker, loc) 'pose.bones["'+bone_name+'"].'+transform+'', index=index_int).keyframe_points.insert(new_marker, loc)
return return
except: except:
pass pass
def setKeyframesFromBones(context, arm_object, new_marker): def findKeyframe(context, bone, active_frame):
none_selected = True arm_object, pose_library = getArmatureData(context)
for bone in arm_object.pose.bones:
if bone.bone.select: for fcu in pose_library.fcurves:
none_selected = False if fcu.data_path.startswith('pose.bones["'+bone.name+'"]'):
for kp in fcu.keyframe_points:
for bone in arm_object.pose.bones: if kp.co.x == active_frame:
if bone.bone.select or none_selected == True: return fcu.data_path
bone_name = bone.name
if bone.rotation_mode == "XYZ": def setKeyframesFromBones(context, arm_object, new_marker):
rot_mode = "rotation_euler" none_selected = True
elif bone.rotation_mode == "YZX": for bone in arm_object.pose.bones:
rot_mode = "rotation_euler" if bone.bone.select:
elif bone.rotation_mode == "ZXY": none_selected = False
rot_mode = "rotation_euler"
elif bone.rotation_mode == "QUATERNION": for bone in arm_object.pose.bones:
rot_mode = "rotation_quaternion" if bone.bone.select or none_selected == True:
else: bone_name = bone.name
self.report({'WARNING'}, "DSPL: Unsupported bone: " + bone.name + ": " + bone.rotation_mode)
rot_mode = None if bone.rotation_mode == "XYZ":
rot_mode = "rotation_euler"
loc_x = bone.location[0] elif bone.rotation_mode == "YZX":
loc_y = bone.location[1] rot_mode = "rotation_euler"
loc_z = bone.location[2] elif bone.rotation_mode == "ZXY":
createFcurve(context, bone_name, "location", 0) rot_mode = "rotation_euler"
createFcurve(context, bone_name, "location", 1) elif bone.rotation_mode == "QUATERNION":
createFcurve(context, bone_name, "location", 2) rot_mode = "rotation_quaternion"
createKeyframe(context, bone_name, "location", 0, new_marker, loc_x) else:
createKeyframe(context, bone_name, "location", 1, new_marker, loc_y) self.report({'WARNING'}, "DSPL: Unsupported bone: " + bone.name + ": " + bone.rotation_mode)
createKeyframe(context, bone_name, "location", 2, new_marker, loc_z) rot_mode = None
if rot_mode == "rotation_quaternion":
rot_w = bone.rotation_quaternion[0] loc_x = bone.location[0]
rot_x = bone.rotation_quaternion[1] loc_y = bone.location[1]
rot_y = bone.rotation_quaternion[2] loc_z = bone.location[2]
rot_z = bone.rotation_quaternion[3] createFcurve(context, bone_name, "location", 0)
createFcurve(context, bone_name, rot_mode, 0) createFcurve(context, bone_name, "location", 1)
createFcurve(context, bone_name, rot_mode, 1) createFcurve(context, bone_name, "location", 2)
createFcurve(context, bone_name, rot_mode, 2) createKeyframe(context, bone_name, "location", 0, new_marker, loc_x)
createFcurve(context, bone_name, rot_mode, 3) createKeyframe(context, bone_name, "location", 1, new_marker, loc_y)
createKeyframe(context, bone_name, rot_mode, 0, new_marker, rot_w) createKeyframe(context, bone_name, "location", 2, new_marker, loc_z)
createKeyframe(context, bone_name, rot_mode, 1, new_marker, rot_x) if rot_mode == "rotation_quaternion":
createKeyframe(context, bone_name, rot_mode, 2, new_marker, rot_y) rot_w = bone.rotation_quaternion[0]
createKeyframe(context, bone_name, rot_mode, 3, new_marker, rot_z) rot_x = bone.rotation_quaternion[1]
elif rot_mode == "rotation_euler": rot_y = bone.rotation_quaternion[2]
rot_x = bone.rotation_euler[0] rot_z = bone.rotation_quaternion[3]
rot_y = bone.rotation_euler[1] createFcurve(context, bone_name, rot_mode, 0)
rot_z = bone.rotation_euler[2] createFcurve(context, bone_name, rot_mode, 1)
createFcurve(context, bone_name, rot_mode, 0) createFcurve(context, bone_name, rot_mode, 2)
createFcurve(context, bone_name, rot_mode, 1) createFcurve(context, bone_name, rot_mode, 3)
createFcurve(context, bone_name, rot_mode, 2) createKeyframe(context, bone_name, rot_mode, 0, new_marker, rot_w)
createKeyframe(context, bone_name, rot_mode, 0, new_marker, rot_x) createKeyframe(context, bone_name, rot_mode, 1, new_marker, rot_x)
createKeyframe(context, bone_name, rot_mode, 1, new_marker, rot_y) createKeyframe(context, bone_name, rot_mode, 2, new_marker, rot_y)
createKeyframe(context, bone_name, rot_mode, 2, new_marker, rot_z) createKeyframe(context, bone_name, rot_mode, 3, new_marker, rot_z)
scl_x = bone.scale[0] elif rot_mode == "rotation_euler":
scl_y = bone.scale[1] rot_x = bone.rotation_euler[0]
scl_z = bone.scale[2] rot_y = bone.rotation_euler[1]
createFcurve(context, bone_name, "scale", 0) rot_z = bone.rotation_euler[2]
createFcurve(context, bone_name, "scale", 1) createFcurve(context, bone_name, rot_mode, 0)
createFcurve(context, bone_name, "scale", 2) createFcurve(context, bone_name, rot_mode, 1)
createKeyframe(context, bone_name, "scale", 0, new_marker, scl_x) createFcurve(context, bone_name, rot_mode, 2)
createKeyframe(context, bone_name, "scale", 1, new_marker, scl_y) createKeyframe(context, bone_name, rot_mode, 0, new_marker, rot_x)
createKeyframe(context, bone_name, "scale", 2, new_marker, scl_z) createKeyframe(context, bone_name, rot_mode, 1, new_marker, rot_y)
createKeyframe(context, bone_name, rot_mode, 2, new_marker, rot_z)
scl_x = bone.scale[0]
def setBonesfromKeyframes(context, arm_object, active_marker): scl_y = bone.scale[1]
none_selected = True scl_z = bone.scale[2]
for bone in arm_object.pose.bones: createFcurve(context, bone_name, "scale", 0)
if bone.bone.select: createFcurve(context, bone_name, "scale", 1)
none_selected = False createFcurve(context, bone_name, "scale", 2)
createKeyframe(context, bone_name, "scale", 0, new_marker, scl_x)
for bone in arm_object.pose.bones: createKeyframe(context, bone_name, "scale", 1, new_marker, scl_y)
if bone.bone.select or none_selected == True: createKeyframe(context, bone_name, "scale", 2, new_marker, scl_z)
bone_name = bone.name
if bone.rotation_mode == "XYZ": def setBonesfromKeyframes(context, arm_object, active_marker):
rot_mode = "rotation_euler" none_selected = True
elif bone.rotation_mode == "YZX": for bone in arm_object.pose.bones:
rot_mode = "rotation_euler" if bone.bone.select:
elif bone.rotation_mode == "ZXY": none_selected = False
rot_mode = "rotation_euler"
elif bone.rotation_mode == "QUATERNION": for bone in arm_object.pose.bones:
rot_mode = "rotation_quaternion" if bone.bone.select or none_selected == True:
else: bone_name = bone.name
self.report({'WARNING'}, "DSPL: Unsupported bone: " + bone.name + ": " + bone.rotation_mode)
rot_mode = None if findKeyframe(context, bone, active_marker.frame) is None:
continue
loc_x = findFcurve(context, bone_name, "location", 0) or 0.0
loc_y = findFcurve(context, bone_name, "location", 1) or 0.0 if bone.rotation_mode == "XYZ":
loc_z = findFcurve(context, bone_name, "location", 2) or 0.0 rot_mode = "rotation_euler"
if rot_mode == "rotation_quaternion": elif bone.rotation_mode == "YZX":
rot_w = findFcurve(context, bone_name, rot_mode, 0) or 1.0 rot_mode = "rotation_euler"
rot_x = findFcurve(context, bone_name, rot_mode, 1) or 0.0 elif bone.rotation_mode == "ZXY":
rot_y = findFcurve(context, bone_name, rot_mode, 2) or 0.0 rot_mode = "rotation_euler"
rot_z = findFcurve(context, bone_name, rot_mode, 3) or 0.0 elif bone.rotation_mode == "QUATERNION":
elif rot_mode == "rotation_euler": rot_mode = "rotation_quaternion"
rot_x = findFcurve(context, bone_name, rot_mode, 0) or 0.0 else:
rot_y = findFcurve(context, bone_name, rot_mode, 1) or 0.0 self.report({'WARNING'}, "DSPL: Unsupported bone: " + bone.name + ": " + bone.rotation_mode)
rot_z = findFcurve(context, bone_name, rot_mode, 2) or 0.0 rot_mode = None
scl_x = findFcurve(context, bone_name, "scale", 0) or 1.0
scl_y = findFcurve(context, bone_name, "scale", 1) or 1.0 loc_x = findFcurve(context, bone_name, "location", 0) or 0.0
scl_z = findFcurve(context, bone_name, "scale", 2) or 1.0 loc_y = findFcurve(context, bone_name, "location", 1) or 0.0
loc_z = findFcurve(context, bone_name, "location", 2) or 0.0
bone.location = mathutils.Vector((loc_x, loc_y, loc_z)) if rot_mode == "rotation_quaternion":
if bone.rotation_mode == "XYZ": rot_w = findFcurve(context, bone_name, rot_mode, 0) or 1.0
bone.rotation_euler = mathutils.Euler( rot_x = findFcurve(context, bone_name, rot_mode, 1) or 0.0
(rot_x, rot_y, rot_z)) rot_y = findFcurve(context, bone_name, rot_mode, 2) or 0.0
elif bone.rotation_mode == "YZX": rot_z = findFcurve(context, bone_name, rot_mode, 3) or 0.0
bone.rotation_euler = mathutils.Euler( elif rot_mode == "rotation_euler":
(rot_x, rot_y, rot_z)) rot_x = findFcurve(context, bone_name, rot_mode, 0) or 0.0
elif bone.rotation_mode == "ZXY": rot_y = findFcurve(context, bone_name, rot_mode, 1) or 0.0
bone.rotation_euler = mathutils.Euler( rot_z = findFcurve(context, bone_name, rot_mode, 2) or 0.0
(rot_z, rot_x, rot_y)) scl_x = findFcurve(context, bone_name, "scale", 0) or 1.0
elif bone.rotation_mode == "YXZ": scl_y = findFcurve(context, bone_name, "scale", 1) or 1.0
bone.rotation_euler = mathutils.Euler( scl_z = findFcurve(context, bone_name, "scale", 2) or 1.0
(rot_y, rot_x, rot_z))
elif bone.rotation_mode == "XZY": bone.location = mathutils.Vector((loc_x, loc_y, loc_z))
bone.rotation_euler = mathutils.Euler( if bone.rotation_mode == "XYZ":
(rot_x, rot_z, rot_y)) bone.rotation_euler = mathutils.Euler(
elif rot_mode == "rotation_quaternion": (rot_x, rot_y, rot_z))
bone.rotation_quaternion = mathutils.Quaternion( elif bone.rotation_mode == "YZX":
(rot_w, rot_x, rot_y, rot_z)) bone.rotation_euler = mathutils.Euler(
(rot_x, rot_y, rot_z))
elif bone.rotation_mode == "ZXY":
bone.rotation_euler = mathutils.Euler(
(rot_z, rot_x, rot_y))
elif bone.rotation_mode == "YXZ":
bone.rotation_euler = mathutils.Euler(
(rot_y, rot_x, rot_z))
elif bone.rotation_mode == "XZY":
bone.rotation_euler = mathutils.Euler(
(rot_x, rot_z, rot_y))
elif rot_mode == "rotation_quaternion":
bone.rotation_quaternion = mathutils.Quaternion(
(rot_w, rot_x, rot_y, rot_z))
bone.scale = mathutils.Vector((scl_x, scl_y, scl_z)) bone.scale = mathutils.Vector((scl_x, scl_y, scl_z))

380
gui.py
View File

@ -1,190 +1,190 @@
import bpy import bpy
from .common import * from .common import *
class DATA_PT_DSPLPanel(bpy.types.Panel): class DATA_PT_DSPLPanel(bpy.types.Panel):
bl_label = "Damn Simple Pose Library" bl_label = "Damn Simple Pose Library"
bl_id = "DATA_PT_DSPLPanel" bl_id = "DATA_PT_DSPLPanel"
bl_space_type = 'VIEW_3D' bl_space_type = 'VIEW_3D'
bl_region_type = 'UI' bl_region_type = 'UI'
bl_category = 'Pose' bl_category = 'Pose'
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
return len(bpy.context.selected_objects) return len(bpy.context.selected_objects)
def draw(self, context): def draw(self, context):
dspl_panel_layout = self.layout dspl_panel_layout = self.layout
dsplsettings = bpy.context.scene.dsplSettings dsplsettings = bpy.context.scene.dsplSettings
# Detect Armature object and parent # Detect Armature object and parent
armature_layout = dspl_panel_layout.column(align=True) armature_layout = dspl_panel_layout.column(align=True)
active_obj = context.active_object active_obj = context.active_object
arm_object, pose_library = getArmatureData(context) arm_object, pose_library = getArmatureData(context)
pose_library_action = getArmatureAction(context) pose_library_action = getArmatureAction(context)
if arm_object: if arm_object:
if arm_object == active_obj: if arm_object == active_obj:
armature_layout.label(text=arm_object.name + " (Active)") armature_layout.label(text=arm_object.name + " (Active)")
elif arm_object == active_obj.parent: elif arm_object == active_obj.parent:
armature_layout.label(text=arm_object.name + " (Parent)") armature_layout.label(text=arm_object.name + " (Parent)")
# Attach or create pose library # Attach or create pose library
if pose_library or pose_library_action: if pose_library or pose_library_action:
if pose_library and not pose_library_action: if pose_library and not pose_library_action:
armature_layout.template_ID( armature_layout.template_ID(
arm_object, "pose_library", new="dspl.create_pose_library", unlink="dspl.unlink_pose_library") arm_object, "pose_library", new="dspl.create_pose_library", unlink="dspl.unlink_pose_library")
elif pose_library_action and not pose_library: elif pose_library_action and not pose_library:
armature_layout.template_ID( armature_layout.template_ID(
arm_object.animation_data, "action", new="dspl.create_pose_library") arm_object.animation_data, "action", new="dspl.create_pose_library")
armature_layout.label( armature_layout.label(
text="Pose Library detected as Action") text="Pose Library detected as Action")
armature_layout.label( armature_layout.label(
text="You should convert to avoid problems") text="You should convert to avoid problems")
armature_layout.operator( armature_layout.operator(
"dspl.convert_pose_library", icon='PLUGIN', text="Convert to Pose Library") "dspl.convert_pose_library", icon='PLUGIN', text="Convert to Pose Library")
elif pose_library and pose_library_action: elif pose_library and pose_library_action:
armature_layout.template_ID( armature_layout.template_ID(
arm_object, "pose_library", new="dspl.create_pose_library") arm_object, "pose_library", new="dspl.create_pose_library")
armature_layout.label(text="Pose Library is opened as Action") armature_layout.label(text="Pose Library is opened as Action")
armature_layout.label(text="Keyframes will affect pose") armature_layout.label(text="Keyframes will affect pose")
armature_layout.operator( armature_layout.operator(
"dspl.convert_pose_library", icon='PLUGIN', text="Unlink from Action") "dspl.convert_pose_library", icon='PLUGIN', text="Unlink from Action")
# List poses in pose library # List poses in pose library
if pose_library: if pose_library:
pose_box_layout = dspl_panel_layout.column() pose_box_layout = dspl_panel_layout.column()
# Menu switcher # Menu switcher
pose_box_menu_switcher_layout = pose_box_layout.row() pose_box_menu_switcher_layout = pose_box_layout.row()
pose_box_menu_switcher_layout.prop( pose_box_menu_switcher_layout.prop(
dsplsettings, "new_menu", icon='PMARKER_ACT', text="New Menu", toggle=True) dsplsettings, "new_menu", icon='PMARKER_ACT', text="New Menu", toggle=True)
# Quick controls # Quick controls
quick_pose_controls_layout = pose_box_layout.column() quick_pose_controls_layout = pose_box_layout.column()
quick_pose_controls_layout.menu( quick_pose_controls_layout.menu(
OBJECT_MT_AddPoseMenu.bl_idname, icon='ADD', text="New Pose") OBJECT_MT_AddPoseMenu.bl_idname, icon='ADD', text="New Pose")
if pose_library.pose_markers.active: if pose_library.pose_markers.active:
quick_apply_layout = quick_pose_controls_layout.split( quick_apply_layout = quick_pose_controls_layout.split(
align=True) align=True)
quick_apply_layout.operator( quick_apply_layout.operator(
"dspl.browse_poses", icon='CON_ARMATURE', text="Browse") "dspl.browse_poses", icon='CON_ARMATURE', text="Browse")
if dsplsettings.new_menu == False: if dsplsettings.new_menu == False:
quick_apply_layout.operator( quick_apply_layout.operator(
"dspl.apply_pose", icon='ARMATURE_DATA', text="Apply Pose").posename = pose_library.pose_markers.active.name "dspl.apply_pose", icon='ARMATURE_DATA', text="Apply Pose").posename = pose_library.pose_markers.active.name
else: else:
quick_apply_layout.prop(dsplsettings, quick_apply_layout.prop(dsplsettings,
"edit_mode", icon='GREASEPENCIL', text="Edit", toggle=True) "edit_mode", icon='GREASEPENCIL', text="Edit", toggle=True)
# New menu # New menu
if dsplsettings.new_menu == True: if dsplsettings.new_menu == True:
pose_button_layout = pose_box_layout.row() pose_button_layout = pose_box_layout.row()
pose_button_entries_layout = pose_button_layout.column(align=True) pose_button_entries_layout = pose_button_layout.column(align=True)
for pm in pose_library.pose_markers: for pm in pose_library.pose_markers:
row = pose_button_entries_layout.row(align=True) row = pose_button_entries_layout.row(align=True)
# Selected indicator # Selected indicator
selected = pm.frame == pose_library.pose_markers.active.frame selected = pm.frame == pose_library.pose_markers.active.frame
if dsplsettings.edit_mode == False: if dsplsettings.edit_mode == False:
row.label(text="", icon='PMARKER_ACT' if selected else 'PMARKER_SEL') row.label(text="", icon='PMARKER_ACT' if selected else 'PMARKER_SEL')
# Pose operator buttons # Pose operator buttons
if dsplsettings.edit_mode == True: if dsplsettings.edit_mode == True:
row.operator('dspl.rename_pose', text=pm.name).posename = pm.name row.operator('dspl.rename_pose', text=pm.name).posename = pm.name
else: else:
row.operator('dspl.apply_pose', text=pm.name).posename = pm.name row.operator('dspl.apply_pose', text=pm.name).posename = pm.name
if dsplsettings.edit_mode == True: if dsplsettings.edit_mode == True:
movebuttondown = row.operator("dspl.move_pose", icon='TRIA_DOWN', text="") movebuttondown = row.operator("dspl.move_pose", icon='TRIA_DOWN', text="")
movebuttondown.direction = "DOWN" movebuttondown.direction = "DOWN"
movebuttondown.posename = pm.name movebuttondown.posename = pm.name
movebuttonup = row.operator("dspl.move_pose", icon='TRIA_UP', text="") movebuttonup = row.operator("dspl.move_pose", icon='TRIA_UP', text="")
movebuttonup.direction = "UP" movebuttonup.direction = "UP"
movebuttonup.posename = pm.name movebuttonup.posename = pm.name
row.operator("dspl.remove_pose", icon='REMOVE', text="").posename = pm.name row.operator("dspl.remove_pose", icon='REMOVE', text="").posename = pm.name
# Old menu # Old menu
elif dsplsettings.new_menu == False: elif dsplsettings.new_menu == False:
pose_list_layout = pose_box_layout.column() pose_list_layout = pose_box_layout.column()
# Pose list # Pose list
pose_list_entries_layout = pose_list_layout.row() pose_list_entries_layout = pose_list_layout.row()
pose_list_entries_layout.template_list("UI_UL_list","pose_markers", pose_list_entries_layout.template_list("UI_UL_list","pose_markers",
pose_library, "pose_markers", pose_library, "pose_markers",
pose_library.pose_markers, "active_index", rows=4) pose_library.pose_markers, "active_index", rows=4)
# Pose operators # Pose operators
pose_ops_layout = pose_list_entries_layout.column(align=True) pose_ops_layout = pose_list_entries_layout.column(align=True)
pose_ops_layout.operator( pose_ops_layout.operator(
"wm.call_menu", icon='ADD', text="").name = "OBJECT_MT_AddPoseMenu" "wm.call_menu", icon='ADD', text="").name = "OBJECT_MT_AddPoseMenu"
if pose_library.pose_markers.active: if pose_library.pose_markers.active:
pose_ops_layout.operator( pose_ops_layout.operator(
"dspl.remove_pose", icon='REMOVE', text="") "dspl.remove_pose", icon='REMOVE', text="")
pose_ops_layout.operator( pose_ops_layout.operator(
"dspl.apply_pose", icon='ARMATURE_DATA', text="" "dspl.apply_pose", icon='ARMATURE_DATA', text=""
).posename = pose_library.pose_markers.active.name ).posename = pose_library.pose_markers.active.name
pose_ops_layout.operator( pose_ops_layout.operator(
"dspl.move_pose", icon='TRIA_UP', text="").direction = "UP" "dspl.move_pose", icon='TRIA_UP', text="").direction = "UP"
pose_ops_layout.operator( pose_ops_layout.operator(
"dspl.move_pose", icon='TRIA_DOWN', text="").direction = "DOWN" "dspl.move_pose", icon='TRIA_DOWN', text="").direction = "DOWN"
else: else:
armature_layout.label( armature_layout.label(
text="No Action or Pose Library detected") text="No Action or Pose Library detected")
armature_layout.template_ID( armature_layout.template_ID(
arm_object, "pose_library", new="dspl.create_pose_library") arm_object, "pose_library", new="dspl.create_pose_library")
else: else:
armature_layout.label(text="No armature or parent selected") armature_layout.label(text="No armature or parent selected")
class OBJECT_MT_AddPoseMenu(bpy.types.Menu): class OBJECT_MT_AddPoseMenu(bpy.types.Menu):
bl_idname = "OBJECT_MT_AddPoseMenu" bl_idname = "OBJECT_MT_AddPoseMenu"
bl_label = "Add Pose" bl_label = "Add Pose"
def draw(self, context): def draw(self, context):
arm_object, pose_library = getArmatureData(context) arm_object, pose_library = getArmatureData(context)
dspl_add_menu_layout = self.layout dspl_add_menu_layout = self.layout
dspl_add_menu_layout.operator( dspl_add_menu_layout.operator(
"dspl.add_pose", icon='ADD', text="Add New Pose") "dspl.add_pose", icon='ADD', text="Add New Pose")
if len(pose_library.pose_markers): if len(pose_library.pose_markers):
dspl_add_menu_layout.menu( dspl_add_menu_layout.menu(
"OBJECT_MT_ReplacePoseMenu", text="Replace Existing Pose", icon="DECORATE_OVERRIDE") "OBJECT_MT_ReplacePoseMenu", text="Replace Existing Pose", icon="DECORATE_OVERRIDE")
class OBJECT_MT_ReplacePoseMenu(bpy.types.Menu): class OBJECT_MT_ReplacePoseMenu(bpy.types.Menu):
bl_idname = "OBJECT_MT_ReplacePoseMenu" bl_idname = "OBJECT_MT_ReplacePoseMenu"
bl_label = "Add Pose" bl_label = "Add Pose"
def draw(self, context): def draw(self, context):
arm_object, pose_library = getArmatureData(context) arm_object, pose_library = getArmatureData(context)
dspl_replace_menu_layout = self.layout dspl_replace_menu_layout = self.layout
for pm in pose_library.pose_markers: for pm in pose_library.pose_markers:
op = dspl_replace_menu_layout.operator("dspl.add_pose", text=pm.name, icon="PMARKER") op = dspl_replace_menu_layout.operator("dspl.add_pose", text=pm.name, icon="PMARKER")
op.replace = True op.replace = True
op.posename = pm.name op.posename = pm.name
classes = ( classes = (
DATA_PT_DSPLPanel, DATA_PT_DSPLPanel,
OBJECT_MT_AddPoseMenu, OBJECT_MT_AddPoseMenu,
OBJECT_MT_ReplacePoseMenu, OBJECT_MT_ReplacePoseMenu,
) )
def register(): def register():
from bpy.utils import register_class from bpy.utils import register_class
for cls in classes: for cls in classes:
register_class(cls) register_class(cls)
def unregister(): def unregister():
from bpy.utils import unregister_class from bpy.utils import unregister_class
for cls in classes: for cls in classes:
unregister_class(cls) unregister_class(cls)

View File

@ -1,30 +1,30 @@
import bpy import bpy
addon_keymaps = [] addon_keymaps = []
def register_keymaps(): def register_keymaps():
wm = bpy.context.window_manager wm = bpy.context.window_manager
# Add Pose # Add Pose
km = wm.keyconfigs.addon.keymaps.new(name='Pose', space_type='EMPTY') km = wm.keyconfigs.addon.keymaps.new(name='Pose', space_type='EMPTY')
kmi = km.keymap_items.new('wm.call_menu', 'L', 'PRESS', shift=True) kmi = km.keymap_items.new('wm.call_menu', 'L', 'PRESS', shift=True)
kmi.properties.name = "OBJECT_MT_AddPoseMenu" kmi.properties.name = "OBJECT_MT_AddPoseMenu"
addon_keymaps.append((km, kmi)) addon_keymaps.append((km, kmi))
# Browse Poses # Browse Poses
km = wm.keyconfigs.addon.keymaps.new(name='Pose', space_type='EMPTY') km = wm.keyconfigs.addon.keymaps.new(name='Pose', space_type='EMPTY')
kmi = km.keymap_items.new('dspl.browse_poses', 'L', 'PRESS', alt=True) kmi = km.keymap_items.new('dspl.browse_poses', 'L', 'PRESS', alt=True)
addon_keymaps.append((km, kmi)) addon_keymaps.append((km, kmi))
def unregister_keymaps(): def unregister_keymaps():
for km, kmi in addon_keymaps: for km, kmi in addon_keymaps:
km.keymap_items.remove(kmi) km.keymap_items.remove(kmi)
addon_keymaps.clear() addon_keymaps.clear()
def register(): def register():
register_keymaps() register_keymaps()
def unregister(): def unregister():
unregister_keymaps() unregister_keymaps()

View File

@ -1,458 +1,458 @@
import bpy import bpy
import blf import blf
from .common import * from .common import *
# Operator to create a new pose library # Operator to create a new pose library
class DSPL_OT_CreatePoseLibrary(bpy.types.Operator): class DSPL_OT_CreatePoseLibrary(bpy.types.Operator):
bl_idname = "dspl.create_pose_library" bl_idname = "dspl.create_pose_library"
bl_label = "Create Pose Library" bl_label = "Create Pose Library"
bl_description = "Create Pose Library" bl_description = "Create Pose Library"
bl_options = {'REGISTER', 'UNDO'} bl_options = {'REGISTER', 'UNDO'}
def execute(self, context): def execute(self, context):
arm_object, pose_library = getArmatureData(context) arm_object, pose_library = getArmatureData(context)
arm_object.pose_library = bpy.data.actions.new( arm_object.pose_library = bpy.data.actions.new(
name=arm_object.name + "_PoseLib") name=arm_object.name + "_PoseLib")
arm_object.pose_library.use_fake_user = True arm_object.pose_library.use_fake_user = True
return {'FINISHED'} return {'FINISHED'}
# Operator to convert an action to pose library # Operator to convert an action to pose library
class DSPL_OT_ConvertPoseLibrary(bpy.types.Operator): class DSPL_OT_ConvertPoseLibrary(bpy.types.Operator):
bl_idname = "dspl.convert_pose_library" bl_idname = "dspl.convert_pose_library"
bl_label = "Convert Pose Library" bl_label = "Convert Pose Library"
bl_description = "Convert Pose Library" bl_description = "Convert Pose Library"
bl_options = {'REGISTER', 'UNDO'} bl_options = {'REGISTER', 'UNDO'}
def execute(self, context): def execute(self, context):
arm_object, pose_library = getArmatureData(context) arm_object, pose_library = getArmatureData(context)
if pose_library is None: if pose_library is None:
arm_object.pose_library = arm_object.animation_data.action arm_object.pose_library = arm_object.animation_data.action
arm_object.animation_data.action = None arm_object.animation_data.action = None
return {'FINISHED'} return {'FINISHED'}
# Operator to add keyframes and marker to pose library # Operator to add keyframes and marker to pose library
class DSPL_OT_AddPose(bpy.types.Operator): class DSPL_OT_AddPose(bpy.types.Operator):
bl_idname = "dspl.add_pose" bl_idname = "dspl.add_pose"
bl_label = "Add Pose" bl_label = "Add Pose"
bl_description = "Add Pose" bl_description = "Add Pose"
bl_options = {'REGISTER', 'UNDO'} bl_options = {'REGISTER', 'UNDO'}
posename: bpy.props.StringProperty(default="Pose") posename: bpy.props.StringProperty(default="Pose")
replace: bpy.props.BoolProperty(name="Replace", description="Replace existing pose", default=False, options={'SKIP_SAVE'}) replace: bpy.props.BoolProperty(name="Replace", description="Replace existing pose", default=False, options={'SKIP_SAVE'})
def execute(self, context): def execute(self, context):
arm_object, pose_library = getArmatureData(context) arm_object, pose_library = getArmatureData(context)
if self.replace == False: if self.replace == False:
pose_markers = pose_library.pose_markers pose_markers = pose_library.pose_markers
new_name = self.posename new_name = self.posename
counter = 1 counter = 1
# Find first unused marker frame # Find first unused marker frame
for f in range(0, len(pose_markers) + 1): for f in range(0, len(pose_markers) + 1):
f += 1 f += 1
for pm in pose_markers: for pm in pose_markers:
name_check = pm.name name_check = pm.name
frame_check = pm.frame frame_check = pm.frame
if frame_check == f: if frame_check == f:
break break
else: else:
new_marker = f new_marker = f
break break
# Check for duplicate names # Check for duplicate names
while pose_markers.find(new_name) > -1: while pose_markers.find(new_name) > -1:
new_name = self.posename + ".{:03d}".format(counter) new_name = self.posename + ".{:03d}".format(counter)
counter += 1 counter += 1
else: else:
pose_name = new_name pose_name = new_name
pose_markers.new(name=pose_name) pose_markers.new(name=pose_name)
pose_markers[pose_name].frame = new_marker pose_markers[pose_name].frame = new_marker
setKeyframesFromBones(context, arm_object, new_marker) setKeyframesFromBones(context, arm_object, new_marker)
pose_library.pose_markers.active = pose_markers[pose_name] pose_library.pose_markers.active = pose_markers[pose_name]
bpy.context.area.tag_redraw() bpy.context.area.tag_redraw()
self.report({'INFO'}, "DSPL: Added " + pose_markers[new_name].name + " to frame " + str(pose_markers[new_name].frame)) self.report({'INFO'}, "DSPL: Added " + pose_markers[new_name].name + " to frame " + str(pose_markers[new_name].frame))
else: else:
pose_markers = pose_library.pose_markers pose_markers = pose_library.pose_markers
active_marker = pose_markers.active active_marker = pose_markers.active
new_name = self.posename new_name = self.posename
counter = 1 counter = 1
for pm in pose_markers: for pm in pose_markers:
name_check = pm.name name_check = pm.name
frame_check = pm.frame frame_check = pm.frame
if name_check == new_name: if name_check == new_name:
target_name = name_check target_name = name_check
target_frame = frame_check target_frame = frame_check
break break
else: else:
return return
new_marker = target_frame new_marker = target_frame
setKeyframesFromBones(context, arm_object, new_marker) setKeyframesFromBones(context, arm_object, new_marker)
self.report({'INFO'}, "DSPL: Replaced " + pose_markers[new_name].name + " on frame " + str(pose_markers[new_name].frame)) self.report({'INFO'}, "DSPL: Replaced " + pose_markers[new_name].name + " on frame " + str(pose_markers[new_name].frame))
return {'FINISHED'} return {'FINISHED'}
# Operator to remove keyframes and marker # Operator to remove keyframes and marker
class DSPL_OT_RemovePose(bpy.types.Operator): class DSPL_OT_RemovePose(bpy.types.Operator):
bl_idname = "dspl.remove_pose" bl_idname = "dspl.remove_pose"
bl_label = "Remove Pose" bl_label = "Remove Pose"
bl_description = "Remove Pose" bl_description = "Remove Pose"
bl_options = {'REGISTER', 'UNDO'} bl_options = {'REGISTER', 'UNDO'}
posename: bpy.props.StringProperty() posename: bpy.props.StringProperty()
def execute(self, context): def execute(self, context):
arm_object, pose_library = getArmatureData(context) arm_object, pose_library = getArmatureData(context)
pose_markers = pose_library.pose_markers pose_markers = pose_library.pose_markers
if self.posename: if self.posename:
pose_library.pose_markers.active_index = searchPoseMarker(context, posename=self.posename, type="index") pose_library.pose_markers.active_index = searchPoseMarker(context, posename=self.posename, type="index")
active_index = pose_library.pose_markers.active_index active_index = pose_library.pose_markers.active_index
else: else:
active_index = pose_markers.active_index active_index = pose_markers.active_index
active_marker = pose_markers.active active_marker = pose_markers.active
active_frame = active_marker.frame active_frame = active_marker.frame
fcurves = pose_library.fcurves fcurves = pose_library.fcurves
for fcu in fcurves: for fcu in fcurves:
for kf in fcu.keyframe_points.values(): for kf in fcu.keyframe_points.values():
if kf.co.x == active_frame: if kf.co.x == active_frame:
fcu.keyframe_points.remove(kf) fcu.keyframe_points.remove(kf)
if active_index <= 0: if active_index <= 0:
next_index = active_index next_index = active_index
next_marker = pose_markers[next_index] next_marker = pose_markers[next_index]
elif (active_index + 1) >= len(pose_markers): elif (active_index + 1) >= len(pose_markers):
next_index = active_index - 1 next_index = active_index - 1
next_marker = pose_markers[next_index] next_marker = pose_markers[next_index]
else: else:
next_index = active_index next_index = active_index
next_marker = pose_markers[next_index] next_marker = pose_markers[next_index]
pose_markers.remove(marker=active_marker) pose_markers.remove(marker=active_marker)
pose_library.pose_markers.active = next_marker pose_library.pose_markers.active = next_marker
pose_library.pose_markers.active_index = next_index pose_library.pose_markers.active_index = next_index
self.report({'INFO'}, "DSPL: Removed " + self.posename) self.report({'INFO'}, "DSPL: Removed " + self.posename)
return {'FINISHED'} return {'FINISHED'}
# Operator to rename the current pose # Operator to rename the current pose
class DSPL_OT_RenamePose(bpy.types.Operator): class DSPL_OT_RenamePose(bpy.types.Operator):
bl_idname = "dspl.rename_pose" bl_idname = "dspl.rename_pose"
bl_label = "Rename Pose" bl_label = "Rename Pose"
bl_description = "Rename Pose" bl_description = "Rename Pose"
bl_options = {'REGISTER', 'UNDO'} bl_options = {'REGISTER', 'UNDO'}
posename: bpy.props.StringProperty() posename: bpy.props.StringProperty()
pose_new_name: bpy.props.StringProperty() pose_new_name: bpy.props.StringProperty()
def execute(self, context): def execute(self, context):
arm_object, pose_library = getArmatureData(context) arm_object, pose_library = getArmatureData(context)
pose_markers = pose_library.pose_markers pose_markers = pose_library.pose_markers
active_marker = pose_markers.active active_marker = pose_markers.active
if self.posename: if self.posename:
target_marker = searchPoseMarker(context, posename=self.posename, type="marker") target_marker = searchPoseMarker(context, posename=self.posename, type="marker")
else: else:
target_marker = active_marker target_marker = active_marker
if self.pose_new_name: if self.pose_new_name:
target_marker.name = self.pose_new_name target_marker.name = self.pose_new_name
context.area.tag_redraw() context.area.tag_redraw()
self.report({'INFO'}, "DSPL: Renamed " + self.posename + " to " + self.pose_new_name + " on frame " + str(active_marker.frame)) self.report({'INFO'}, "DSPL: Renamed " + self.posename + " to " + self.pose_new_name + " on frame " + str(active_marker.frame))
return {'FINISHED'} return {'FINISHED'}
else: else:
return {'FINISHED'} return {'FINISHED'}
def invoke(self, context, event): def invoke(self, context, event):
self.pose_new_name = self.posename self.pose_new_name = self.posename
return context.window_manager.invoke_props_dialog(self) return context.window_manager.invoke_props_dialog(self)
# Operator to reorder pose markers # Operator to reorder pose markers
class DSPL_OT_MovePose(bpy.types.Operator): class DSPL_OT_MovePose(bpy.types.Operator):
bl_idname = "dspl.move_pose" bl_idname = "dspl.move_pose"
bl_label = "Move Pose" bl_label = "Move Pose"
bl_description = "Move pose" bl_description = "Move pose"
bl_options = {'REGISTER', 'UNDO'} bl_options = {'REGISTER', 'UNDO'}
direction: bpy.props.StringProperty(name="Direction", default="DOWN") direction: bpy.props.StringProperty(name="Direction", default="DOWN")
posename: bpy.props.StringProperty(name="Pose Name", default="", options={'SKIP_SAVE'}) posename: bpy.props.StringProperty(name="Pose Name", default="", options={'SKIP_SAVE'})
def execute(self, context): def execute(self, context):
arm_object, pose_library = getArmatureData(context) arm_object, pose_library = getArmatureData(context)
pose_markers = pose_library.pose_markers pose_markers = pose_library.pose_markers
if self.posename: if self.posename:
active_index = searchPoseMarker(context, posename=self.posename, type="index") active_index = searchPoseMarker(context, posename=self.posename, type="index")
active_marker = searchPoseMarker(context, posename=self.posename, type="marker") active_marker = searchPoseMarker(context, posename=self.posename, type="marker")
active_frame = active_marker.frame active_frame = active_marker.frame
active_posename = active_marker.name active_posename = active_marker.name
else: else:
active_marker = pose_markers.active active_marker = pose_markers.active
active_index = pose_markers.active_index active_index = pose_markers.active_index
if self.direction == "UP": if self.direction == "UP":
move_dir = -1 move_dir = -1
elif self.direction == "DOWN": elif self.direction == "DOWN":
move_dir = 1 move_dir = 1
swap_index = active_index + move_dir swap_index = active_index + move_dir
if swap_index < 0: if swap_index < 0:
return {'FINISHED'} return {'FINISHED'}
elif swap_index >= len(pose_library.pose_markers): elif swap_index >= len(pose_library.pose_markers):
return {'FINISHED'} return {'FINISHED'}
swap_marker = pose_markers[swap_index] swap_marker = pose_markers[swap_index]
tmp_marker_name = swap_marker.name tmp_marker_name = swap_marker.name
tmp_marker_frame = swap_marker.frame tmp_marker_frame = swap_marker.frame
pose_library.pose_markers[swap_index].name = active_marker.name pose_library.pose_markers[swap_index].name = active_marker.name
pose_library.pose_markers[swap_index].frame = active_marker.frame pose_library.pose_markers[swap_index].frame = active_marker.frame
pose_library.pose_markers[active_index].name = tmp_marker_name pose_library.pose_markers[active_index].name = tmp_marker_name
pose_library.pose_markers[active_index].frame = tmp_marker_frame pose_library.pose_markers[active_index].frame = tmp_marker_frame
pose_library.pose_markers.active_index = swap_index pose_library.pose_markers.active_index = swap_index
return {'FINISHED'} return {'FINISHED'}
# Operator to apply a pose from active marker # Operator to apply a pose from active marker
class DSPL_OT_ApplyPose(bpy.types.Operator): class DSPL_OT_ApplyPose(bpy.types.Operator):
bl_idname = "dspl.apply_pose" bl_idname = "dspl.apply_pose"
bl_label = "Apply Pose" bl_label = "Apply Pose"
bl_description = "Apply Pose (Ctrl+Click to select, Shift+Click to rename, Alt+Click to remove)" bl_description = "Apply Pose (Ctrl+Click to select, Shift+Click to rename, Alt+Click to remove)"
bl_options = {'REGISTER', 'UNDO'} bl_options = {'REGISTER', 'UNDO'}
posename: bpy.props.StringProperty() posename: bpy.props.StringProperty()
def execute(self, context): def execute(self, context):
arm_object, pose_library = getArmatureData(context) arm_object, pose_library = getArmatureData(context)
pose_markers = pose_library.pose_markers pose_markers = pose_library.pose_markers
if self.posename: if self.posename:
active_marker = searchPoseMarker(context, posename=self.posename, type="marker") active_marker = searchPoseMarker(context, posename=self.posename, type="marker")
active_frame = active_marker.frame active_frame = active_marker.frame
active_posename = active_marker.name active_posename = active_marker.name
pose_library.pose_markers.active_index = searchPoseMarker(context, posename=self.posename, type="index") pose_library.pose_markers.active_index = searchPoseMarker(context, posename=self.posename, type="index")
else: else:
active_index = pose_markers.active_index active_index = pose_markers.active_index
active_marker = pose_markers.active active_marker = pose_markers.active
active_frame = active_marker.frame active_frame = active_marker.frame
active_posename = active_marker.name active_posename = active_marker.name
setBonesfromKeyframes(context, arm_object, active_marker) setBonesfromKeyframes(context, arm_object, active_marker)
self.report({'INFO'}, "DSPL: Applied " + active_posename) self.report({'INFO'}, "DSPL: Applied " + active_posename)
return {'FINISHED'} return {'FINISHED'}
def invoke(self, context, event): def invoke(self, context, event):
if event.ctrl: if event.ctrl:
# Select # Select
arm_object, pose_library = getArmatureData(context) arm_object, pose_library = getArmatureData(context)
pose_library.pose_markers.active_index = searchPoseMarker(context, posename=self.posename, type="index") pose_library.pose_markers.active_index = searchPoseMarker(context, posename=self.posename, type="index")
return {'FINISHED'} return {'FINISHED'}
elif event.alt: elif event.alt:
# Remove # Remove
bpy.ops.dspl.remove_pose(posename = self.posename) bpy.ops.dspl.remove_pose(posename = self.posename)
return {'FINISHED'} return {'FINISHED'}
elif event.shift: elif event.shift:
# Rename # Rename
bpy.ops.dspl.rename_pose('INVOKE_DEFAULT', posename = self.posename) bpy.ops.dspl.rename_pose('INVOKE_DEFAULT', posename = self.posename)
return {'FINISHED'} return {'FINISHED'}
else: else:
return self.execute(context) return self.execute(context)
# Operator to preview up and down pose list # Operator to preview up and down pose list
class DSPL_OT_BrowsePoses(bpy.types.Operator): class DSPL_OT_BrowsePoses(bpy.types.Operator):
bl_idname = "dspl.browse_poses" bl_idname = "dspl.browse_poses"
bl_label = "Browse Poses" bl_label = "Browse Poses"
bl_description = "Browse Poses" bl_description = "Browse Poses"
bl_options = {'REGISTER', 'UNDO'} bl_options = {'REGISTER', 'UNDO'}
def draw_callback_px(self, context, test): def draw_callback_px(self, context, test):
font_id = 1 font_id = 1
font_size = 24 font_size = 24
font_dpi = 72 font_dpi = 72
blf.position(font_id, 15, 30, 0) blf.position(font_id, 15, 30, 0)
mod_var = self.pose_library.pose_markers.active.name mod_var = self.pose_library.pose_markers.active.name
blf.color(font_id, 1.0, 1.0, 1.0, 1.0) blf.color(font_id, 1.0, 1.0, 1.0, 1.0)
blf.size(font_id, font_size) blf.size(font_id, font_size)
blf.draw(font_id, "Previewing pose: " + mod_var) blf.draw(font_id, "Previewing pose: " + mod_var)
def modal(self, context, event): def modal(self, context, event):
context.area.tag_redraw() context.area.tag_redraw()
if event.value == 'PRESS': if event.value == 'PRESS':
if event.type in {'LEFT_ARROW', 'UP_ARROW'}: if event.type in {'LEFT_ARROW', 'UP_ARROW'}:
if self.pose_library.pose_markers.active_index <= 0: if self.pose_library.pose_markers.active_index <= 0:
self.pose_library.pose_markers.active_index = len(self.pose_library.pose_markers) - 1 self.pose_library.pose_markers.active_index = len(self.pose_library.pose_markers) - 1
else: else:
self.pose_library.pose_markers.active_index = self.pose_library.pose_markers.active_index - 1 self.pose_library.pose_markers.active_index = self.pose_library.pose_markers.active_index - 1
bpy.ops.dspl.apply_pose() bpy.ops.dspl.apply_pose()
elif event.type in {'RIGHT_ARROW', 'DOWN_ARROW'}: elif event.type in {'RIGHT_ARROW', 'DOWN_ARROW'}:
if self.pose_library.pose_markers.active_index + 1 >= len(self.pose_library.pose_markers): if self.pose_library.pose_markers.active_index + 1 >= len(self.pose_library.pose_markers):
self.pose_library.pose_markers.active_index = 0 self.pose_library.pose_markers.active_index = 0
else: else:
self.pose_library.pose_markers.active_index = self.pose_library.pose_markers.active_index + 1 self.pose_library.pose_markers.active_index = self.pose_library.pose_markers.active_index + 1
bpy.ops.dspl.apply_pose() bpy.ops.dspl.apply_pose()
if event.type in {'LEFTMOUSE', 'RET', 'NUMPAD_ENTER'}: if event.type in {'LEFTMOUSE', 'RET', 'NUMPAD_ENTER'}:
bpy.types.SpaceView3D.draw_handler_remove(self._handle, 'WINDOW') bpy.types.SpaceView3D.draw_handler_remove(self._handle, 'WINDOW')
return {'FINISHED'} return {'FINISHED'}
elif event.type in {'RIGHTMOUSE', 'ESC'}: elif event.type in {'RIGHTMOUSE', 'ESC'}:
self.arm_object.pose.backup_restore() self.arm_object.pose.backup_restore()
self.pose_library.pose_markers.active_index = self.backup_index self.pose_library.pose_markers.active_index = self.backup_index
bpy.types.SpaceView3D.draw_handler_remove(self._handle, 'WINDOW') bpy.types.SpaceView3D.draw_handler_remove(self._handle, 'WINDOW')
return {'CANCELLED'} return {'CANCELLED'}
return {'RUNNING_MODAL'} return {'RUNNING_MODAL'}
def invoke(self, context, event): def invoke(self, context, event):
bpy.context.area.tag_redraw() bpy.context.area.tag_redraw()
self.arm_object, self.pose_library = getArmatureData(context) self.arm_object, self.pose_library = getArmatureData(context)
if self.pose_library is None: if self.pose_library is None:
self.report({'WARNING'}, "DSPL: Pose Library not active") self.report({'WARNING'}, "DSPL: Pose Library not active")
return {'CANCELLED'} return {'CANCELLED'}
self.arm_object.pose.backup_create(self.pose_library) self.arm_object.pose.backup_create(self.pose_library)
self.backup_index = self.pose_library.pose_markers.active_index self.backup_index = self.pose_library.pose_markers.active_index
bpy.ops.dspl.apply_pose() bpy.ops.dspl.apply_pose()
if context.area.type == 'VIEW_3D': if context.area.type == 'VIEW_3D':
self.report({'INFO'}, "DSPL: Browsing Poses") self.report({'INFO'}, "DSPL: Browsing Poses")
args = (self, context) args = (self, context)
self._handle = bpy.types.SpaceView3D.draw_handler_add(self.draw_callback_px, args, 'WINDOW', 'POST_PIXEL') self._handle = bpy.types.SpaceView3D.draw_handler_add(self.draw_callback_px, args, 'WINDOW', 'POST_PIXEL')
context.window_manager.modal_handler_add(self) context.window_manager.modal_handler_add(self)
return {'RUNNING_MODAL'} return {'RUNNING_MODAL'}
return {'CANCELLED'} return {'CANCELLED'}
# Operator to unlink a pose library and mark for removal # Operator to unlink a pose library and mark for removal
class DSPL_OT_UnlinkPoseLibrary(bpy.types.Operator): class DSPL_OT_UnlinkPoseLibrary(bpy.types.Operator):
bl_idname = "dspl.unlink_pose_library" bl_idname = "dspl.unlink_pose_library"
bl_label = "Unlink Pose Library" bl_label = "Unlink Pose Library"
bl_description = "Unlink Pose Library" bl_description = "Unlink Pose Library"
bl_options = {'REGISTER', 'UNDO'} bl_options = {'REGISTER', 'UNDO'}
def execute(self, context): def execute(self, context):
arm_object, pose_library = getArmatureData(context) arm_object, pose_library = getArmatureData(context)
try: try:
arm_object.pose_library = None arm_object.pose_library = None
# arm_object.pose_library.use_fake_user = False # arm_object.pose_library.use_fake_user = False
# if not arm_object.pose_library.name.startswith("del_"): # if not arm_object.pose_library.name.startswith("del_"):
# arm_object.pose_library.name = "del_{}".format(arm_object.pose_library.name) # arm_object.pose_library.name = "del_{}".format(arm_object.pose_library.name)
except: except:
pass pass
return {'FINISHED'} return {'FINISHED'}
# Operator to protect orphaned legacy pose libraries # Operator to protect orphaned legacy pose libraries
class DSPL_OT_ProtectOrphanPoseLibrary(bpy.types.Operator): class DSPL_OT_ProtectOrphanPoseLibrary(bpy.types.Operator):
bl_idname = "dspl.protect_orphan_pose_library" bl_idname = "dspl.protect_orphan_pose_library"
bl_label = "Protect Orphaned Pose Libraries" bl_label = "Protect Orphaned Pose Libraries"
bl_description = "Protect Orphaned Pose Libraries" bl_description = "Protect Orphaned Pose Libraries"
bl_options = {'REGISTER', 'UNDO'} bl_options = {'REGISTER', 'UNDO'}
only_poselibs: bpy.props.BoolProperty(name="Only Poselibs", description="Only named poselibs", default=False, options={'SKIP_SAVE'}) only_poselibs: bpy.props.BoolProperty(name="Only Poselibs", description="Only named poselibs", default=False, options={'SKIP_SAVE'})
protect: bpy.props.BoolProperty(name="Protect", description="Or not", default=True, options={'SKIP_SAVE'}) protect: bpy.props.BoolProperty(name="Protect", description="Or not", default=True, options={'SKIP_SAVE'})
def check(self, context): def check(self, context):
return True return True
def invoke(self, context, event): def invoke(self, context, event):
return context.window_manager.invoke_props_dialog(self) return context.window_manager.invoke_props_dialog(self)
def draw(self, context): def draw(self, context):
layout = self.layout layout = self.layout
orphaned_act = [act for act in bpy.data.actions if act.users == 0] orphaned_act = [act for act in bpy.data.actions if act.users == 0]
if orphaned_act: if orphaned_act:
# bpy.types.Scene.my_prop = bpy.props.BoolVectorProperty(name="boo", size = len(orphaned_act), default=True) # bpy.types.Scene.my_prop = bpy.props.BoolVectorProperty(name="boo", size = len(orphaned_act), default=True)
col = layout.column() col = layout.column()
col.label(text="Orphaned actions", icon="ORPHAN_DATA") col.label(text="Orphaned actions", icon="ORPHAN_DATA")
for act in orphaned_act: for act in orphaned_act:
entryrow = col.row() entryrow = col.row()
protectbutton = entryrow.prop(self, "protect", text=act.name) protectbutton = entryrow.prop(self, "protect", text=act.name)
col.split() col.split()
else: else:
layout.label(text="No orphans here") layout.label(text="No orphans here")
def execute(self, context): def execute(self, context):
orphaned_act = [act for act in bpy.data.actions if act.users == 0] orphaned_act = [act for act in bpy.data.actions if act.users == 0]
if orphaned_act: if orphaned_act:
for act in orphaned_act: for act in orphaned_act:
if "_loc" in act.name or "PoseLib" in act.name: if "_loc" in act.name or "PoseLib" in act.name:
self.report({'INFO'}, "DSPL: Protecting orphaned action: " + act.name) self.report({'INFO'}, "DSPL: Protecting orphaned action: " + act.name)
act.use_fake_user = True act.use_fake_user = True
return {'FINISHED'} return {'FINISHED'}
classes = ( classes = (
DSPL_OT_CreatePoseLibrary, DSPL_OT_CreatePoseLibrary,
DSPL_OT_ConvertPoseLibrary, DSPL_OT_ConvertPoseLibrary,
DSPL_OT_AddPose, DSPL_OT_AddPose,
DSPL_OT_RemovePose, DSPL_OT_RemovePose,
DSPL_OT_RenamePose, DSPL_OT_RenamePose,
DSPL_OT_MovePose, DSPL_OT_MovePose,
DSPL_OT_ApplyPose, DSPL_OT_ApplyPose,
DSPL_OT_BrowsePoses, DSPL_OT_BrowsePoses,
DSPL_OT_UnlinkPoseLibrary, DSPL_OT_UnlinkPoseLibrary,
DSPL_OT_ProtectOrphanPoseLibrary DSPL_OT_ProtectOrphanPoseLibrary
) )
register, unregister = bpy.utils.register_classes_factory(classes) register, unregister = bpy.utils.register_classes_factory(classes)