io_anim_nuke_chan/__init__.py

   1 # SPDX-License-Identifier: GPL-2.0-or-later
   2
   3 bl_info = {
   4     "name": "Nuke Animation Format (.chan)",
   5     "author": "Michael Krupa",
   6     "version": (1, 0),
   7     "blender": (2, 80, 0),
   8     "location": "File > Import/Export > Nuke (.chan)",
   9     "description": "Import/Export object's animation with nuke",
  10     "warning": "",
  11     "doc_url": "{BLENDER_MANUAL_URL}/addons/import_export/anim_nuke_chan.html",
  12     "category": "Import-Export",
  13 }
  14
  15
  16 # To support reload properly, try to access a package var,
  17 # if it's there, reload everything
  18 if "bpy" in locals():
  19     import importlib
  20     if "import_nuke_chan" in locals():
  21         importlib.reload(import_nuke_chan)
  22     if "export_nuke_chan" in locals():
  23         importlib.reload(export_nuke_chan)
  24
  25
  26 import bpy
  27 from bpy.types import Operator
  28 from bpy_extras.io_utils import ImportHelper, ExportHelper
  29 from bpy.props import (
  30         StringProperty,
  31         BoolProperty,
  32         EnumProperty,
  33         FloatProperty,
  34         )
  35
  36 # property shared by both operators
  37 rotation_order = EnumProperty(
  38         name="Rotation order",
  39         description="Choose the export rotation order",
  40         items=(('XYZ', "XYZ", "XYZ"),
  41                ('XZY', "XZY", "XZY"),
  42                ('YXZ', "YXZ", "YXZ"),
  43                ('YZX', "YZX", "YZX"),
  44                ('ZXY', "ZXY", "ZXY"),
  45                ('ZYX', "ZYX", "ZYX"),
  46                ),
  47         default='XYZ')
  48
  49
  50 class ImportChan(Operator, ImportHelper):
  51     """Import animation from .chan file, exported from nuke or houdini """ \
  52     """(the importer uses frame numbers from the file)"""
  53     bl_idname = "import_scene.import_chan"
  54     bl_label = "Import chan file"
  55
  56     filename_ext = ".chan"
  57
  58     filter_glob: StringProperty(default="*.chan", options={'HIDDEN'})
  59
  60     rotation_order: rotation_order
  61     z_up: BoolProperty(
  62             name="Make Z up",
  63             description="Switch the Y and Z axis",
  64             default=True)
  65
  66     sensor_width: FloatProperty(
  67             name="Camera sensor width",
  68             description="Imported camera sensor width",
  69             default=32.0)
  70
  71     sensor_height: FloatProperty(
  72             name="Camera sensor height",
  73             description="Imported camera sensor height",
  74             default=18.0)
  75
  76     @classmethod
  77     def poll(cls, context):
  78         return context.active_object is not None
  79
  80     def execute(self, context):
  81         from . import import_nuke_chan
  82         return import_nuke_chan.read_chan(context,
  83                                           self.filepath,
  84                                           self.z_up,
  85                                           self.rotation_order,
  86                                           self.sensor_width,
  87                                           self.sensor_height)
  88
  89
  90 class ExportChan(Operator, ExportHelper):
  91     """Export the animation to .chan file, readable by nuke and houdini """ \
  92     """(the exporter uses frames from the frames range)"""
  93     bl_idname = "export.export_chan"
  94     bl_label = "Export chan file"
  95
  96     filename_ext = ".chan"
  97     filter_glob: StringProperty(default="*.chan", options={'HIDDEN'})
  98     y_up: BoolProperty(
  99             name="Make Y up",
 100             description="Switch the Y and Z axis",
 101             default=True)
 102     rotation_order: rotation_order
 103
 104     @classmethod
 105     def poll(cls, context):
 106         return context.active_object is not None
 107
 108     def execute(self, context):
 109         from . import export_nuke_chan
 110         return export_nuke_chan.save_chan(context,
 111                                           self.filepath,
 112                                           self.y_up,
 113                                           self.rotation_order)
 114
 115
 116 def menu_func_import(self, context):
 117     self.layout.operator(ImportChan.bl_idname, text="Nuke (.chan)")
 118
 119
 120 def menu_func_export(self, context):
 121     self.layout.operator(ExportChan.bl_idname, text="Nuke (.chan)")
 122
 123
 124 def register():
 125     bpy.utils.register_class(ImportChan)
 126     bpy.utils.register_class(ExportChan)
 127     bpy.types.TOPBAR_MT_file_import.append(menu_func_import)
 128     bpy.types.TOPBAR_MT_file_export.append(menu_func_export)
 129
 130
 131 def unregister():
 132     bpy.utils.unregister_class(ImportChan)
 133     bpy.utils.unregister_class(ExportChan)
 134     bpy.types.TOPBAR_MT_file_import.remove(menu_func_import)
 135     bpy.types.TOPBAR_MT_file_export.remove(menu_func_export)
 136
 137
 138 if __name__ == "__main__":
 139     register()