io_shape_mdd/export_mdd.py

   1 # ##### BEGIN GPL LICENSE BLOCK #####
   2 #
   3 #  This program is free software; you can redistribute it and/or
   4 #  modify it under the terms of the GNU General Public License
   5 #  as published by the Free Software Foundation; either version 2
   6 #  of the License, or (at your option) any later version.
   7 #
   8 #  This program is distributed in the hope that it will be useful,
   9 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 #  GNU General Public License for more details.
  12 #
  13 #  You should have received a copy of the GNU General Public License
  14 #  along with this program; if not, write to the Free Software Foundation,
  15 #  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  16 #
  17 # ##### END GPL LICENSE BLOCK #####
  18
  19 # <pep8 compliant>
  20
  21 # Contributors: Bill L.Nieuwendorp
  22
  23 """
  24 This script Exports Lightwaves MotionDesigner format.
  25
  26 The .mdd format has become quite a popular Pipeline format<br>
  27 for moving animations from package to package.
  28
  29 Be sure not to use modifiers that change the number or order of verts in the mesh
  30 """
  31
  32 import bpy
  33 import mathutils
  34 from struct import pack
  35
  36
  37 def zero_file(filepath):
  38     """
  39     If a file fails, this replaces it with 1 char, better not remove it?
  40     """
  41     file = open(filepath, 'w')
  42     file.write('\n')  # apparently macosx needs some data in a blank file?
  43     file.close()
  44
  45
  46 def check_vertcount(mesh, vertcount):
  47     """
  48     check and make sure the vertcount is consistent throughout the frame range
  49     """
  50     if len(mesh.vertices) != vertcount:
  51         raise Exception('Error, number of verts has changed during animation, cannot export')
  52
  53
  54 def save(context, filepath="", frame_start=1, frame_end=300, fps=25.0, use_rest_frame=False):
  55     """
  56     Blender.Window.WaitCursor(1)
  57
  58     mesh_orig = Mesh.New()
  59     mesh_orig.getFromObject(obj.name)
  60     """
  61
  62     scene = context.scene
  63     obj = context.object
  64
  65     if bpy.ops.object.mode_set.poll():
  66         bpy.ops.object.mode_set(mode='OBJECT')
  67
  68     orig_frame = scene.frame_current
  69     scene.frame_set(frame_start)
  70     depsgraph = context.evaluated_depsgraph_get()
  71     obj_eval = obj.evaluated_get(depsgraph)
  72     me = obj_eval.to_mesh()
  73
  74     #Flip y and z
  75     '''
  76     mat_flip = mathutils.Matrix(((1.0, 0.0, 0.0, 0.0),
  77                                  (0.0, 0.0, 1.0, 0.0),
  78                                  (0.0, 1.0, 0.0, 0.0),
  79                                  (0.0, 0.0, 0.0, 1.0),
  80                                  ))
  81     '''
  82     mat_flip = mathutils.Matrix()
  83
  84     numverts = len(me.vertices)
  85
  86     numframes = frame_end - frame_start + 1
  87     if use_rest_frame:
  88         numframes += 1
  89
  90     f = open(filepath, 'wb')  # no Errors yet:Safe to create file
  91
  92     # Write the header
  93     f.write(pack(">2i", numframes, numverts))
  94
  95     # Write the frame times (should we use the time IPO??)
  96     f.write(pack(">%df" % (numframes), *[frame / fps for frame in range(numframes)]))  # seconds
  97
  98     if use_rest_frame:
  99         check_vertcount(me, numverts)
 100         me.transform(mat_flip @ obj.matrix_world)
 101         f.write(pack(">%df" % (numverts * 3), *[axis for v in me.vertices for axis in v.co]))
 102
 103     obj_eval.to_mesh_clear()
 104
 105     for frame in range(frame_start, frame_end + 1):  # in order to start at desired frame
 106         scene.frame_set(frame)
 107         depsgraph = context.evaluated_depsgraph_get()
 108         obj_eval = obj.evaluated_get(depsgraph)
 109         me = obj_eval.to_mesh()
 110         check_vertcount(me, numverts)
 111         me.transform(mat_flip @ obj.matrix_world)
 112
 113         # Write the vertex data
 114         f.write(pack(">%df" % (numverts * 3), *[axis for v in me.vertices for axis in v.co]))
 115
 116         obj_eval.to_mesh_clear()
 117
 118     f.close()
 119
 120     print('MDD Exported: %r frames:%d\n' % (filepath, numframes - 1))
 121     scene.frame_set(orig_frame)
 122
 123     return {'FINISHED'}