2010-04-01 09:29:35 +00:00
|
|
|
# ##### 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 LICENSE BLOCK #####
|
|
|
|
|
|
|
|
# <pep8 compliant>
|
|
|
|
|
|
|
|
import bpy
|
|
|
|
|
2010-04-04 14:52:15 +00:00
|
|
|
|
2010-04-01 09:29:35 +00:00
|
|
|
def pose_info():
|
2010-04-11 14:22:27 +00:00
|
|
|
from mathutils import Matrix
|
2010-04-01 09:29:35 +00:00
|
|
|
|
|
|
|
info = {}
|
|
|
|
|
|
|
|
obj = bpy.context.object
|
|
|
|
pose = obj.pose
|
|
|
|
|
|
|
|
pose_items = pose.bones.items()
|
|
|
|
|
|
|
|
for name, pbone in pose_items:
|
|
|
|
binfo = {}
|
|
|
|
bone = pbone.bone
|
|
|
|
|
|
|
|
binfo["parent"] = getattr(bone.parent, "name", None)
|
|
|
|
binfo["bone"] = bone
|
|
|
|
binfo["pbone"] = pbone
|
|
|
|
binfo["matrix_local"] = bone.matrix_local.copy()
|
|
|
|
try:
|
|
|
|
binfo["matrix_local_inv"] = binfo["matrix_local"].copy().invert()
|
|
|
|
except:
|
|
|
|
binfo["matrix_local_inv"] = Matrix()
|
2010-04-04 14:52:15 +00:00
|
|
|
|
2010-04-01 09:29:35 +00:00
|
|
|
binfo["matrix"] = bone.matrix.copy()
|
|
|
|
binfo["matrix_pose"] = pbone.matrix.copy()
|
|
|
|
try:
|
|
|
|
binfo["matrix_pose_inv"] = binfo["matrix_pose"].copy().invert()
|
|
|
|
except:
|
|
|
|
binfo["matrix_pose_inv"] = Matrix()
|
2010-04-04 14:52:15 +00:00
|
|
|
|
2010-04-01 09:29:35 +00:00
|
|
|
print(binfo["matrix_pose"])
|
|
|
|
info[name] = binfo
|
|
|
|
|
|
|
|
for name, pbone in pose_items:
|
|
|
|
binfo = info[name]
|
|
|
|
binfo_parent = binfo.get("parent", None)
|
|
|
|
if binfo_parent:
|
|
|
|
binfo_parent = info[binfo_parent]
|
2010-04-04 14:52:15 +00:00
|
|
|
|
2010-04-01 09:29:35 +00:00
|
|
|
matrix = binfo["matrix_pose"]
|
|
|
|
rest_matrix = binfo["matrix_local"]
|
|
|
|
|
|
|
|
if binfo_parent:
|
2010-04-04 14:52:15 +00:00
|
|
|
matrix = binfo_parent["matrix_pose_inv"] * matrix
|
|
|
|
rest_matrix = binfo_parent["matrix_local_inv"] * rest_matrix
|
2010-04-01 09:29:35 +00:00
|
|
|
|
|
|
|
matrix = rest_matrix.copy().invert() * matrix
|
|
|
|
|
|
|
|
binfo["matrix_key"] = matrix.copy()
|
|
|
|
|
|
|
|
return info
|
|
|
|
|
|
|
|
|
2010-04-01 21:44:56 +00:00
|
|
|
def bake(frame_start, frame_end, step=1, only_selected=False):
|
2010-04-01 09:29:35 +00:00
|
|
|
# import nla; reload(nla); nla.bake()
|
|
|
|
|
|
|
|
scene = bpy.context.scene
|
|
|
|
obj = bpy.context.object
|
|
|
|
pose = obj.pose
|
|
|
|
|
|
|
|
info_ls = []
|
2010-04-04 14:52:15 +00:00
|
|
|
|
2010-04-01 21:44:56 +00:00
|
|
|
frame_range = range(frame_start, frame_end + 1, step)
|
2010-04-01 09:29:35 +00:00
|
|
|
|
|
|
|
# could spped this up by applying steps here too...
|
|
|
|
for f in frame_range:
|
|
|
|
scene.set_frame(f)
|
2010-04-04 14:52:15 +00:00
|
|
|
|
2010-04-01 09:29:35 +00:00
|
|
|
info = pose_info()
|
|
|
|
info_ls.append(info)
|
|
|
|
f += 1
|
|
|
|
|
|
|
|
action = bpy.data.actions.new("Action")
|
|
|
|
|
|
|
|
bpy.context.object.animation_data.action = action
|
|
|
|
|
|
|
|
pose_items = pose.bones.items()
|
|
|
|
|
|
|
|
for name, pbone in pose_items:
|
2010-07-15 16:56:04 +00:00
|
|
|
if only_selected and not pbone.select:
|
2010-04-01 09:29:35 +00:00
|
|
|
continue
|
|
|
|
|
|
|
|
for f in frame_range:
|
2010-04-04 14:52:15 +00:00
|
|
|
matrix = info_ls[int((f - frame_start) / step)][name]["matrix_key"]
|
2010-04-01 09:29:35 +00:00
|
|
|
|
|
|
|
#pbone.location = matrix.translation_part()
|
|
|
|
#pbone.rotation_quaternion = matrix.to_quat()
|
|
|
|
pbone.matrix_local = [f for v in matrix for f in v]
|
2010-04-17 19:05:53 +00:00
|
|
|
|
2010-04-06 10:07:24 +00:00
|
|
|
pbone.keyframe_insert("location", -1, f, name)
|
2010-04-04 14:52:15 +00:00
|
|
|
|
2010-04-01 09:29:35 +00:00
|
|
|
rotation_mode = pbone.rotation_mode
|
2010-04-04 14:52:15 +00:00
|
|
|
|
2010-04-01 09:29:35 +00:00
|
|
|
if rotation_mode == 'QUATERNION':
|
2010-04-06 10:07:24 +00:00
|
|
|
pbone.keyframe_insert("rotation_quaternion", -1, f, name)
|
2010-04-01 09:29:35 +00:00
|
|
|
elif rotation_mode == 'AXIS_ANGLE':
|
2010-04-06 10:07:24 +00:00
|
|
|
pbone.keyframe_insert("rotation_axis_angle", -1, f, name)
|
2010-04-01 09:29:35 +00:00
|
|
|
else: # euler, XYZ, ZXY etc
|
2010-04-06 10:07:24 +00:00
|
|
|
pbone.keyframe_insert("rotation_euler", -1, f, name)
|
2010-04-01 09:29:35 +00:00
|
|
|
|
2010-04-06 10:07:24 +00:00
|
|
|
pbone.keyframe_insert("scale", -1, f, name)
|
2010-04-01 09:29:35 +00:00
|
|
|
|
|
|
|
return action
|
|
|
|
|
|
|
|
|
|
|
|
from bpy.props import *
|
|
|
|
|
|
|
|
|
|
|
|
class BakeAction(bpy.types.Operator):
|
2010-05-14 07:20:16 +00:00
|
|
|
'''Bake animation to an Action'''
|
2010-04-01 09:29:35 +00:00
|
|
|
bl_idname = "nla.bake"
|
|
|
|
bl_label = "Bake Action"
|
|
|
|
bl_options = {'REGISTER', 'UNDO'}
|
|
|
|
|
2010-04-01 21:44:56 +00:00
|
|
|
frame_start = IntProperty(name="Start Frame",
|
2010-04-01 09:29:35 +00:00
|
|
|
description="Start frame for baking",
|
|
|
|
default=1, min=1, max=300000)
|
2010-04-01 21:44:56 +00:00
|
|
|
frame_end = IntProperty(name="End Frame",
|
2010-04-01 09:29:35 +00:00
|
|
|
description="End frame for baking",
|
|
|
|
default=250, min=1, max=300000)
|
|
|
|
step = IntProperty(name="Frame Step",
|
|
|
|
description="Frame Step",
|
|
|
|
default=1, min=1, max=120)
|
|
|
|
only_selected = BoolProperty(name="Only Selected",
|
|
|
|
default=True)
|
|
|
|
|
|
|
|
def execute(self, context):
|
|
|
|
props = self.properties
|
|
|
|
|
2010-08-17 07:49:53 +00:00
|
|
|
action = bake(props.frame_start, props.frame_end, props.step, props.show_only_selected)
|
2010-04-04 14:52:15 +00:00
|
|
|
|
2010-04-01 09:29:35 +00:00
|
|
|
# basic cleanup, could move elsewhere
|
|
|
|
for fcu in action.fcurves:
|
|
|
|
keyframe_points = fcu.keyframe_points
|
|
|
|
i = 1
|
|
|
|
while i < len(fcu.keyframe_points) - 1:
|
|
|
|
val_prev = keyframe_points[i - 1].co[1]
|
|
|
|
val_next = keyframe_points[i + 1].co[1]
|
|
|
|
val = keyframe_points[i].co[1]
|
|
|
|
|
|
|
|
if abs(val - val_prev) + abs(val - val_next) < 0.0001:
|
|
|
|
keyframe_points.remove(keyframe_points[i])
|
|
|
|
else:
|
|
|
|
i += 1
|
|
|
|
|
|
|
|
return {'FINISHED'}
|
|
|
|
|
|
|
|
def invoke(self, context, event):
|
|
|
|
wm = context.manager
|
|
|
|
return wm.invoke_props_dialog(self)
|
|
|
|
|
|
|
|
|
|
|
|
#def menu_func(self, context):
|
|
|
|
# self.layout.operator(BakeAction.bl_idname, text="Bake Armature Action")
|
|
|
|
|
|
|
|
|
|
|
|
def register():
|
2010-08-02 02:55:12 +00:00
|
|
|
pass
|
2010-04-01 09:29:35 +00:00
|
|
|
# bpy.types.INFO_MT_mesh_add.append(menu_func)
|
|
|
|
|
|
|
|
|
|
|
|
def unregister():
|
2010-08-02 02:55:12 +00:00
|
|
|
pass
|
2010-04-01 09:29:35 +00:00
|
|
|
# bpy.types.INFO_MT_mesh_add.remove(menu_func)
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
2010-04-04 14:52:15 +00:00
|
|
|
register()
|