import_anim.py

import bpy
from .anim import import_anim


def do_import(context, filepath, rig_name, use_current_frame, start_frame, baseline_offsets):
    if use_current_frame:
        start_frame = context.scene.frame_current
    import_anim(filepath, rig_name=rig_name, start_frame=start_frame, baseline_offsets=baseline_offsets)
    return {'FINISHED'}


from bpy_extras.io_utils import ImportHelper
from bpy.props import StringProperty, BoolProperty, IntProperty
from bpy.types import Operator


class ImportWadToolAnim(Operator, ImportHelper):
    """Import a Tomb Editor / WadTool Animation (.anim) as a new Action on an armature"""

    bl_idname = "import_anim.data"
    bl_label = "Import Wad Tool Animation"

    filename_ext = ".anim"

    filter_glob: StringProperty(
        default="*.anim;*.ANIM",
        options={'HIDDEN'},
        maxlen=255,
    )

    rig_name: StringProperty(
        default="LARA_RIG",
        description="Armature object name. If not found, the active armature is used.",
        maxlen=255,
    )

    use_current_frame: BoolProperty(
        name="Start at Current Frame",
        description="Insert imported keys starting at the current scene frame",
        default=True,
    )

    start_frame: IntProperty(
        name="Start Frame",
        description="Frame to start inserting keys at (ignored if 'Start at Current Frame' is enabled)",
        default=1,
        min=-100000,
        max=100000,
    )

    baseline_offsets: BoolProperty(
        name="Baseline Offsets",
        description="Treat the first frame's offsets as baseline and import deltas only (recommended)",
        default=True,
    )

    def draw(self, context):
        layout = self.layout
        layout.prop(self, "rig_name")
        layout.prop(self, "use_current_frame")
        if not self.use_current_frame:
            layout.prop(self, "start_frame")
        layout.prop(self, "baseline_offsets")

    def execute(self, context):
        return do_import(context, self.filepath, self.rig_name, self.use_current_frame, self.start_frame, self.baseline_offsets)


def menu_func_import(self, context):
    self.layout.operator(ImportWadToolAnim.bl_idname, text="Tomb Editor Animation (.anim)")


def register():
    bpy.utils.register_class(ImportWadToolAnim)
    bpy.types.TOPBAR_MT_file_import.append(menu_func_import)


def unregister():
    bpy.utils.unregister_class(ImportWadToolAnim)
    bpy.types.TOPBAR_MT_file_import.remove(menu_func_import)


if __name__ == "__main__":
    register()

    # test call
    bpy.ops.import_anim.data('INVOKE_DEFAULT')