[Bf-blender-cvs] SVN commit: /data/svn/bf-blender [33975] branches/blender2.4/release/ scripts/bvh_export.py: BVH Exporter, originally written by myself but exported bad rotations,
Campbell Barton
ideasman42 at gmail.com
Sat Jan 1 06:57:20 CET 2011
Revision: 33975
http://projects.blender.org/plugins/scmsvn/viewcvs.php?view=rev&root=bf-blender&revision=33975
Author: campbellbarton
Date: 2011-01-01 06:57:08 +0100 (Sat, 01 Jan 2011)
Log Message:
-----------
BVH Exporter, originally written by myself but exported bad rotations,
Andrea Rugliancich managed to fix these this bug so the script is finally ok usable.
Will port to 2.5x later.
Added Paths:
-----------
branches/blender2.4/release/scripts/bvh_export.py
Added: branches/blender2.4/release/scripts/bvh_export.py
===================================================================
--- branches/blender2.4/release/scripts/bvh_export.py (rev 0)
+++ branches/blender2.4/release/scripts/bvh_export.py 2011-01-01 05:57:08 UTC (rev 33975)
@@ -0,0 +1,307 @@
+#!BPY
+
+"""
+Name: 'Motion Capture (.bvh)...'
+Blender: 249.2
+Group: 'Export'
+Tooltip: 'Active Armature to BVH'
+"""
+
+
+__author__ = "Campbell Barton, Andrea Rugliancich"
+__url__ = ("blender.org", "blenderartists.org")
+__version__ = "10/12/30"
+__bpydoc__ = """\
+This script exports Blender armature motion data to BHV motion capture format
+"""
+
+
+# --------------------------------------------------------------------------
+# BVH Export by Campbell Barton (AKA Ideasman), Andrea Rugliancich
+# --------------------------------------------------------------------------
+# ***** 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+#
+# ***** END GPL LICENCE BLOCK *****
+# --------------------------------------------------------------------------
+
+
+import Blender
+from Blender import *
+TranslationMatrix = Mathutils.TranslationMatrix
+Matrix= Blender.Mathutils.Matrix
+Vector= Blender.Mathutils.Vector
+
+import BPyMessages
+
+
+
+def bvh_export(filepath, ob, PREF_STARTFRAME, PREF_ENDFRAME, PREF_SCALE= 1.0):
+
+ Window.EditMode(0)
+ Blender.Window.WaitCursor(1)
+
+ file= open(filepath, 'w')
+
+ # bvh_nodes= {}
+ arm_data= ob.data
+ bones= arm_data.bones.values()
+
+ # Build a dictionary of bone children.
+ # None is for parentless bones
+ bone_children= {None:[]}
+
+ # initialize with blank lists
+ for bone in bones:
+ bone_children[bone.name] = []
+
+ for bone in bones:
+ parent= bone.parent
+ bone_name= bone.name
+ if parent:
+ bone_children[parent.name].append( bone_name )
+ else: # root node
+ bone_children[None].append( bone_name )
+
+ # sort the children
+ for children_list in bone_children.itervalues():
+ children_list.sort()
+
+ # build a (name:bone) mapping dict
+ bone_dict= {}
+ for bone in bones: bone_dict[bone.name] = bone
+
+ # bone name list in the order that the bones are written
+ bones_serialized_names= []
+
+ bone_locs= {}
+
+ file.write('HIERARCHY\n')
+
+ def write_bones_recursive(bone_name, indent):
+ my_bone_children= bone_children[bone_name]
+
+ indent_str= '\t'*indent # cache?
+
+ bone= bone_dict[bone_name]
+ loc= bone.head['ARMATURESPACE']
+ bone_locs[bone_name] = loc
+
+ # make relative if we can
+ if bone.parent:
+ loc= loc - bone_locs[bone.parent.name]
+
+ if indent:
+ file.write('%sJOINT %s\n' % (indent_str, bone_name))
+ else:
+ file.write('%sROOT %s\n' % (indent_str, bone_name))
+
+ file.write('%s{\n' % indent_str)
+ file.write('%s\tOFFSET %.6f %.6f %.6f\n' % (indent_str, loc.x * PREF_SCALE, loc.y * PREF_SCALE, loc.z * PREF_SCALE))
+ file.write('%s\tCHANNELS 6 Xposition Yposition Zposition Xrotation Yrotation Zrotation\n' % indent_str)
+
+
+ if my_bone_children:
+
+ # store the location for the children
+ # to het their relative offset
+
+ # Write children
+ for child_bone in my_bone_children:
+ bones_serialized_names.append(child_bone)
+ write_bones_recursive(child_bone, indent+1)
+
+ else:
+ # Write the bone end.
+ file.write('%s\tEnd Site\n' % indent_str)
+ file.write('%s\t{\n' % indent_str)
+ loc= bone.tail['ARMATURESPACE'] - bone_locs[bone_name]
+ file.write('%s\t\tOFFSET %.6f %.6f %.6f\n' % (indent_str, loc.x * PREF_SCALE, loc.y * PREF_SCALE, loc.z * PREF_SCALE))
+ file.write('%s\t}\n' % indent_str)
+
+ file.write('%s}\n' % indent_str)
+
+
+
+ if len(bone_children[None])==1:
+ key= bone_children[None][0]
+ bones_serialized_names.append(key)
+ indent= 0
+
+ write_bones_recursive(key, indent)
+
+ else:
+ # Write a dummy parent node
+ file.write('ROOT %s\n' % key)
+ file.write('{\n')
+ file.write('\tOFFSET 0.0 0.0 0.0\n')
+ file.write('\tCHANNELS 0\n') # Xposition Yposition Zposition Xrotation Yrotation Zrotation
+ key= None
+ indent= 1
+
+ write_bones_recursive(key, indent)
+
+ file.write('}\n')
+
+
+ # redefine bones as sorted by bones_serialized_names
+ # se we can write motion
+
+ pose_dict= ob.getPose().bones
+ #pose_bones= [(pose_dict[bone_name], bone_dict[bone_name].matrix['ARMATURESPACE'].copy().invert() ) for bone_name in bones_serialized_names]
+
+ class decorated_bone(object):
+ __slots__=(\
+ 'name',# bone name, used as key in many places
+ 'parent',# decorated bone parent, set in a later loop
+ 'rest_bone',# blender armature bone
+ 'pose_bone',# blender pose bone
+ 'pose_mat',# blender pose matrix
+ 'rest_arm_mat',# blender rest matrix (armature space)
+ 'rest_local_mat',# blender rest batrix (local space)
+ 'pose_imat',# pose_mat inverted
+ 'rest_arm_imat',# rest_arm_mat inverted
+ 'rest_local_imat') # rest_local_mat inverted
+
+ def __init__(self, bone_name):
+ self.name= bone_name
+ self.rest_bone= bone_dict[bone_name]
+ self.pose_bone= pose_dict[bone_name]
+
+
+ self.pose_mat= self.pose_bone.poseMatrix
+
+ mat= self.rest_bone.matrix
+ self.rest_arm_mat= mat['ARMATURESPACE'].copy()
+ self.rest_local_mat= mat['BONESPACE'].copy().resize4x4()
+
+ # inverted mats
+ self.pose_imat= self.pose_mat.copy().invert()
+ self.rest_arm_imat= self.rest_arm_mat.copy().invert()
+ self.rest_local_imat= self.rest_local_mat.copy().invert()
+
+ self.parent= None
+
+ def update_posedata(self):
+ self.pose_mat= self.pose_bone.poseMatrix
+ self.pose_imat= self.pose_mat.copy().invert()
+
+ def __repr__(self):
+ if self.parent:
+ return '["%s" child on "%s"]\n' % (self.name, self.parent.name)
+ else:
+ return '["%s" root bone]\n' % (self.name)
+
+
+ bones_decorated= [ decorated_bone(bone_name) for bone_name in bones_serialized_names]
+ '''
+ for i, b in enumerate(bones_serialized_names):
+ print i,
+ print " " + b
+ '''
+ # Assign parents
+ bones_decorated_dict= {}
+ for db in bones_decorated:
+ bones_decorated_dict[db.name] = db
+
+ for db in bones_decorated:
+ parent= db.rest_bone.parent
+ if parent:
+ db.parent = bones_decorated_dict[parent.name]
+ del bones_decorated_dict
+ # finish assigning parents
+ #print bones_decorated
+ '''
+ for i, b in enumerate(bones_serialized_names):
+ print i,
+ print " " + b + " " + bones_decorated[i].name
+ #print len(bones), len(pose_bones)
+ '''
+ file.write('MOTION\n')
+ file.write('Frames: %d\n' % (PREF_ENDFRAME - PREF_STARTFRAME + 1))
+ file.write('Frame Time: %.6f\n' % 0.03)
+
+ triple= '%.6f %.6f %.6f '
+ for frame in xrange(PREF_STARTFRAME, PREF_ENDFRAME+1):
+ Blender.Set('curframe', frame)
+ for dbone in bones_decorated:
+ dbone.update_posedata()
+ for dbone in bones_decorated:
+ if dbone.parent:
+ trans = TranslationMatrix(dbone.rest_bone.head['ARMATURESPACE'])
+ itrans = TranslationMatrix(-dbone.rest_bone.head['ARMATURESPACE'])
+ mat2 = dbone.rest_arm_imat * dbone.pose_mat * dbone.parent.pose_imat *dbone.parent.rest_arm_mat #FASTER
+ mat2 = trans * mat2 * itrans
+ myloc = mat2.translationPart() + (dbone.rest_bone.head['ARMATURESPACE'] - dbone.parent.rest_bone.head['ARMATURESPACE'])
+ rot = mat2.copy().transpose().toEuler()
+ else:
+ trans = TranslationMatrix(dbone.rest_bone.head['ARMATURESPACE'])
+ itrans = TranslationMatrix(-dbone.rest_bone.head['ARMATURESPACE'])
+ mat2 = dbone.rest_arm_imat * dbone.pose_mat
+ mat2 = trans * mat2 * itrans
+ myloc = mat2.translationPart() + dbone.rest_bone.head['ARMATURESPACE']
+ rot= mat2.copy().transpose().toEuler()
+
+ file.write(triple % (myloc[0] * PREF_SCALE, myloc[1] * PREF_SCALE,myloc[2] * PREF_SCALE))
+ file.write(triple % (-rot[0],-rot[1],-rot[2])) #NEGATED
+
+ file.write('\n')
+
+ numframes = PREF_ENDFRAME - PREF_STARTFRAME + 1
+ file.close()
+
+ print'BVH Exported: %s frames:%d\n'% (filepath, numframes)
+ Blender.Window.WaitCursor(0)
+
+
+def bvh_export_ui(filepath):
+ # Dont overwrite
+ if not BPyMessages.Warning_SaveOver(filepath):
+ return
+
+ scn= Scene.GetCurrent()
+ ob_act= scn.objects.active
+ if not ob_act or ob_act.type != 'Armature':
+ BPyMessages.Error_NoArmatureActive()
+
+ arm_ob= scn.objects.active
+
+ if not arm_ob or arm_ob.type!='Armature':
+ Blender.Draw.PupMenu('No Armature object selected.')
+ return
+
+ ctx = scn.getRenderingContext()
+ orig_frame = Blender.Get('curframe')
@@ Diff output truncated at 10240 characters. @@
More information about the Bf-blender-cvs
mailing list