io_shape_mdd/__init__.py

   1 # SPDX-License-Identifier: GPL-2.0-or-later
   2
   3 bl_info = {
   4     "name": "NewTek MDD format",
   5     "author": "Bill L.Nieuwendorp",
   6     "version": (1, 0, 1),
   7     "blender": (2, 80, 0),
   8     "location": "File > Import-Export",
   9     "description": "Import-Export MDD as mesh shape keys",
  10     "warning": "",
  11     "doc_url": "{BLENDER_MANUAL_URL}/addons/import_export/shape_mdd.html",
  12     "support": 'OFFICIAL',
  13     "category": "Import-Export",
  14 }
  15
  16 if "bpy" in locals():
  17     import importlib
  18     if "import_mdd" in locals():
  19         importlib.reload(import_mdd)
  20     if "export_mdd" in locals():
  21         importlib.reload(export_mdd)
  22
  23
  24 import bpy
  25 from bpy.props import (
  26         BoolProperty,
  27         FloatProperty,
  28         IntProperty,
  29         StringProperty,
  30         )
  31 from bpy_extras.io_utils import ExportHelper, ImportHelper
  32
  33
  34 class ImportMDD(bpy.types.Operator, ImportHelper):
  35     """Import MDD vertex keyframe file to shape keys"""
  36     bl_idname = "import_shape.mdd"
  37     bl_label = "Import MDD"
  38     bl_options = {'UNDO'}
  39
  40     filename_ext = ".mdd"
  41
  42     filter_glob: StringProperty(
  43             default="*.mdd",
  44             options={'HIDDEN'},
  45             )
  46     frame_start: IntProperty(
  47             name="Start Frame",
  48             description="Start frame for inserting animation",
  49             min=-300000, max=300000,
  50             default=0,
  51             )
  52     frame_step: IntProperty(
  53             name="Step",
  54             min=1, max=1000,
  55             default=1,
  56             )
  57
  58     @classmethod
  59     def poll(cls, context):
  60         obj = context.active_object
  61         return (obj and obj.type == 'MESH')
  62
  63     def invoke(self, context, event):
  64         scene = context.scene
  65         self.frame_start = scene.frame_start
  66
  67         return super().invoke(context, event)
  68
  69     def execute(self, context):
  70         keywords = self.as_keywords(ignore=("filter_glob",))
  71
  72         from . import import_mdd
  73         return import_mdd.load(context, **keywords)
  74
  75
  76 class ExportMDD(bpy.types.Operator, ExportHelper):
  77     """Animated mesh to MDD vertex keyframe file"""
  78     bl_idname = "export_shape.mdd"
  79     bl_label = "Export MDD"
  80
  81     filename_ext = ".mdd"
  82     filter_glob: StringProperty(default="*.mdd", options={'HIDDEN'})
  83
  84     # get first scene to get min and max properties for frames, fps
  85
  86     minframe = 0
  87     maxframe = 300000
  88     minfps = 1.0
  89     maxfps = 120.0
  90
  91     # List of operator properties, the attributes will be assigned
  92     # to the class instance from the operator settings before calling.
  93     fps: FloatProperty(
  94             name="Frames Per Second",
  95             description="Number of frames/second",
  96             min=minfps, max=maxfps,
  97             default=25.0,
  98             )
  99     frame_start: IntProperty(
 100             name="Start Frame",
 101             description="Start frame for baking",
 102             min=minframe, max=maxframe,
 103             default=1,
 104             )
 105     frame_end: IntProperty(
 106             name="End Frame",
 107             description="End frame for baking",
 108             min=minframe, max=maxframe,
 109             default=250,
 110             )
 111     use_rest_frame: BoolProperty(
 112             name="Rest Frame",
 113             description="Write the rest state at the first frame",
 114             default=False,
 115             )
 116
 117     @classmethod
 118     def poll(cls, context):
 119         obj = context.active_object
 120         return (obj and obj.type == 'MESH')
 121
 122     def invoke(self, context, event):
 123         scene = context.scene
 124         self.frame_start = scene.frame_start
 125         self.frame_end = scene.frame_end
 126         self.fps = scene.render.fps / scene.render.fps_base
 127
 128         return super().invoke(context, event)
 129
 130     def execute(self, context):
 131         keywords = self.as_keywords(ignore=("check_existing", "filter_glob"))
 132
 133         from . import export_mdd
 134         return export_mdd.save(context, **keywords)
 135
 136
 137 def menu_func_import(self, context):
 138     self.layout.operator(ImportMDD.bl_idname,
 139                          text="Lightwave Point Cache (.mdd)",
 140                          )
 141
 142
 143 def menu_func_export(self, context):
 144     self.layout.operator(ExportMDD.bl_idname,
 145                          text="Lightwave Point Cache (.mdd)",
 146                          )
 147
 148
 149 classes = (
 150     ImportMDD,
 151     ExportMDD
 152 )
 153
 154 def register():
 155     for cls in classes:
 156         bpy.utils.register_class(cls)
 157
 158     bpy.types.TOPBAR_MT_file_import.append(menu_func_import)
 159     bpy.types.TOPBAR_MT_file_export.append(menu_func_export)
 160
 161
 162 def unregister():
 163     for cls in classes:
 164         bpy.utils.unregister_class(cls)
 165
 166     bpy.types.TOPBAR_MT_file_import.remove(menu_func_import)
 167     bpy.types.TOPBAR_MT_file_export.remove(menu_func_export)
 168
 169 if __name__ == "__main__":
 170     register()