#include "ui/CSimpleModelFFX.hpp"
#include "gx/Draw.hpp"
#include "ui/CRenderBatch.hpp"
#include "ui/CSimpleModelFFXScript.hpp"
#include <storm/Memory.hpp>

int32_t CSimpleModelFFX::s_metatable;

CSimpleFrame* CSimpleModelFFX::Create(CSimpleFrame* parent) {
    // TODO
    // auto m = CDataAllocator::GetData(CSimpleModelFFX::s_simpleModelFFXHeap, 0, __FILE__, __LINE__);
    auto m = SMemAlloc(sizeof(CSimpleModelFFX), __FILE__, __LINE__, 0x0);
    return new (m) CSimpleModelFFX(parent);
}

void CSimpleModelFFX::Render(void* arg) {
    CSimpleModelFFX* simpleModel = static_cast<CSimpleModelFFX*>(arg);

    CImVector clearColor = { 0x00, 0x00, 0x00, 0xFF };
    GxSceneClear(0x1 | 0x2, clearColor);

    // TODO

    GxSceneClear(0x1 | 0x2, clearColor);

    if (simpleModel->m_model) {
        CSimpleModel::RenderModel(simpleModel);
    }

    // TODO
}

void CSimpleModelFFX::CreateScriptMetaTable() {
    lua_State* L = FrameScript_GetContext();
    int32_t ref = FrameScript_Object::CreateScriptMetaTable(L, &CSimpleModelFFX::RegisterScriptMethods);
    CSimpleModelFFX::s_metatable = ref;
}

void CSimpleModelFFX::RegisterScriptMethods(lua_State* L) {
    CSimpleModel::RegisterScriptMethods(L);
    FrameScript_Object::FillScriptMethodTable(L, SimpleModelFFXMethods, NUM_SIMPLE_MODEL_FFX_SCRIPT_METHODS);
}

CSimpleModelFFX::CSimpleModelFFX(CSimpleFrame* parent) : CSimpleModel(parent) {
    // TODO
}

int32_t CSimpleModelFFX::GetScriptMetaTable() {
    return CSimpleModelFFX::s_metatable;
}

void CSimpleModelFFX::OnFrameRender(CRenderBatch* batch, uint32_t layer) {
    CSimpleFrame::OnFrameRender(batch, layer);

    if (layer == DRAWLAYER_ARTWORK) {
        batch->QueueCallback(CSimpleModelFFX::Render, this);
    }
}