#pragma once

#define WIN32_LEAN_AND_MEAN
#include <windows.h>

#include "Common.h"
#include "vftable.h"

enum EngModelType : DWORD
{
    Object = 0,
    Virtual = 2,
};

struct EngModel
{
    EngModelType type; // 0x00
    BYTE x04[0xC];
    EngModel* parent; // 0x10
};

struct EngAnimation
{
    // The first parameter seems to be a pointer to a stack-struct with the first three DWORDS set to 0 and then a ModelBinary*.
    bool SetModel_Hook(PDWORD unk, const EngModel* model);
    bool SetModel(PDWORD unk, const EngModel* model);
};

struct IAnimation2
{
    FILL_VFTABLE(0);
    FILL_VFTABLE(1);
    virtual void Vftable_x20();
    virtual int __stdcall Open(int scriptIndex, long engineInstance, LPCSTR animationScript, int unk1 = 0, int unk2 = 0);

    int Open_Hook(LPCSTR animationScript, int scriptIndex, const CAttachedEquip& equip);
};

void InitWeaponAnimFix();