143 lines
5.1 KiB
Python
143 lines
5.1 KiB
Python
import bpy
|
|
from . import fn
|
|
|
|
|
|
class AW_OT_world_space_copy(bpy.types.Operator):
|
|
bl_idname = "pose.world_space_copy"
|
|
bl_label = "World Copy"
|
|
bl_description = "Copy world space transforms. Store active bone matrix"
|
|
bl_options = {"REGISTER", "UNDO"}
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
return context.object and context.mode == 'POSE' and context.active_pose_bone
|
|
|
|
def execute(self, context):
|
|
bpy.types.ViewLayer.world_space_store = context.object.matrix_world @ context.active_pose_bone.matrix
|
|
|
|
return {"FINISHED"}
|
|
|
|
class AW_OT_world_space_paste(bpy.types.Operator):
|
|
bl_idname = "pose.world_space_paste"
|
|
bl_label = "World Paste"
|
|
bl_description = "Paste world space transforms. Apply stored matrix to active bone"
|
|
bl_options = {"REGISTER", "UNDO"}
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
if not hasattr(bpy.context.view_layer, 'world_space_store'):
|
|
cls.poll_message_set("Nothing To Paste")
|
|
return False
|
|
return context.object and context.mode == 'POSE' and context.active_pose_bone
|
|
|
|
|
|
def execute(self, context):
|
|
context.active_pose_bone.matrix = context.object.matrix_world.inverted() @ bpy.context.view_layer.world_space_store
|
|
|
|
if context.scene.tool_settings.use_keyframe_insert_auto:
|
|
bpy.ops.anim.keyframe_insert_menu(type='Available')
|
|
|
|
return {"FINISHED"}
|
|
|
|
class AW_OT_world_space_paste_next(bpy.types.Operator):
|
|
bl_idname = "pose.world_space_paste_next"
|
|
bl_label = "World Paste Jump"
|
|
bl_description = "Paste world space transforms and keyframe available chanels\nThen jump to prev/next key"
|
|
bl_options = {"REGISTER", "UNDO"}
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
if not hasattr(bpy.context.view_layer, 'world_space_store'):
|
|
cls.poll_message_set("Nothing To Paste")
|
|
return False
|
|
return context.object and context.mode == 'POSE' and context.active_pose_bone
|
|
|
|
prev: bpy.props.BoolProperty()
|
|
|
|
def execute(self, context):
|
|
# apply matrix
|
|
context.active_pose_bone.matrix = context.object.matrix_world.inverted() @ bpy.context.view_layer.world_space_store
|
|
|
|
# insert keyframe at value
|
|
# context.object.keyframe_insert(data_path, index=-1, frame=bpy.context.scene.frame_current, group="", options={'INSERTKEY_AVAILABLE'})
|
|
bpy.ops.anim.keyframe_insert_menu(type='Available')
|
|
|
|
# jump to next key
|
|
act = fn.get_obj_action(context.object)
|
|
if not act:
|
|
self.report({'ERROR'}, 'No action on armature')
|
|
return {'CANCELLED'}
|
|
|
|
kx = [k.co.x for fcu in act.fcurves
|
|
if fcu.data_path.split('"')[1] == context.active_pose_bone.bone.name
|
|
for k in fcu.keyframe_points]
|
|
|
|
if not kx:
|
|
self.report({'ERROR'}, 'No keys on action available (no keyframe added)')
|
|
return {'CANCELLED'}
|
|
|
|
# for fcu in act.fcurves:
|
|
# if fcu.data_path.split('"')[1] == context.active_pose_bone.bone.name:
|
|
|
|
if self.prev:
|
|
new_frame = next((k for k in reversed(kx) if k < context.scene.frame_current), None)
|
|
else:
|
|
new_frame = next((k for k in kx if k > context.scene.frame_current), None)
|
|
|
|
if not new_frame:
|
|
self.report({'WARNING'}, 'No next frame to jump on')
|
|
return {'FINISHED'}
|
|
|
|
context.scene.frame_current = int(new_frame)
|
|
return {"FINISHED"}
|
|
|
|
class AW_OT_world_space_paste_next_frame(bpy.types.Operator):
|
|
bl_idname = "pose.world_space_paste_next_frame"
|
|
bl_label = "World Paste Jump Frame"
|
|
bl_description = "Paste world space transforms and keyframe available chanels\nThen jump to prev/next frame"
|
|
bl_options = {"REGISTER", "UNDO"}
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
if not hasattr(bpy.context.view_layer, 'world_space_store'):
|
|
cls.poll_message_set("Nothing To Paste")
|
|
return False
|
|
return context.object and context.mode == 'POSE' and context.active_pose_bone
|
|
|
|
prev: bpy.props.BoolProperty()
|
|
|
|
def execute(self, context):
|
|
# apply matrix
|
|
context.active_pose_bone.matrix = context.object.matrix_world.inverted() @ bpy.context.view_layer.world_space_store
|
|
|
|
# insert keyframe at value
|
|
# context.object.keyframe_insert(data_path, index=-1, frame=bpy.context.scene.frame_current, group="", options={'INSERTKEY_AVAILABLE'})
|
|
bpy.ops.anim.keyframe_insert_menu(type='Available')
|
|
|
|
# jump to next frame
|
|
act = fn.get_obj_action(context.object)
|
|
if not act:
|
|
self.report({'ERROR'}, 'No action on armature')
|
|
return {'CANCELLED'}
|
|
|
|
offset = -1 if self.prev else 1
|
|
new_frame = context.scene.frame_current + offset
|
|
|
|
context.scene.frame_current = new_frame
|
|
return {"FINISHED"}
|
|
|
|
|
|
classes=(
|
|
AW_OT_world_space_copy,
|
|
AW_OT_world_space_paste,
|
|
AW_OT_world_space_paste_next,
|
|
AW_OT_world_space_paste_next_frame,
|
|
)
|
|
|
|
def register():
|
|
for cls in classes:
|
|
bpy.utils.register_class(cls)
|
|
|
|
def unregister():
|
|
for cls in reversed(classes):
|
|
bpy.utils.unregister_class(cls) |