#include "global.h"

s32 gFramerateDivisor = 1;
f32 gFramerateDivisorF = 1.0f;
f32 gFramerateDivisorHalf = 1.0f / 2.0f;
f32 gFramerateDivisorThird = 1.0f / 3.0f;

u32 D_801D1510 = 0x0000000A;
u32 D_801D1514[3] = { 0 };

void Game_UpdateFramerateVariables(s32 divisor) {
    gFramerateDivisor = divisor;
    gFramerateDivisorF = (f32)divisor;
    gFramerateDivisorHalf = (f32)(divisor * 0.5f);
    gFramerateDivisorThird = (f32)(divisor / 3.0f);
}

void Game_SetFramerateDivisor(GameState* gameState, s32 divisor) {
    R_UPDATE_RATE = (s16)divisor;
    gameState->framerateDivisor = divisor;
    Game_UpdateFramerateVariables(divisor);
}

void GameState_SetFBFilter(Gfx** gfx, u32 arg1) {
    Gfx* _gfx = *gfx;

    if ((R_FB_FILTER_TYPE > 0) && (R_FB_FILTER_TYPE < 5)) {
        D_801F8010.type = R_FB_FILTER_TYPE;
        D_801F8010.color.r = R_FB_FILTER_PRIM_COLOR(0);
        D_801F8010.color.g = R_FB_FILTER_PRIM_COLOR(1);
        D_801F8010.color.b = R_FB_FILTER_PRIM_COLOR(2);
        D_801F8010.color.a = R_FB_FILTER_A;
        func_80140D10(&D_801F8010, &_gfx, arg1);
    } else {
        if ((R_FB_FILTER_TYPE == 5) || (R_FB_FILTER_TYPE == 6)) {
            D_801F8020.useRgba = (R_FB_FILTER_TYPE == 6);
            D_801F8020.primColor.r = R_FB_FILTER_PRIM_COLOR(0);
            D_801F8020.primColor.g = R_FB_FILTER_PRIM_COLOR(1);
            D_801F8020.primColor.b = R_FB_FILTER_PRIM_COLOR(2);
            D_801F8020.primColor.a = R_FB_FILTER_A;
            D_801F8020.envColor.r = R_FB_FILTER_ENV_COLOR(0);
            D_801F8020.envColor.g = R_FB_FILTER_ENV_COLOR(1);
            D_801F8020.envColor.b = R_FB_FILTER_ENV_COLOR(2);
            D_801F8020.envColor.a = R_FB_FILTER_A;
            func_80142100(&D_801F8020, &_gfx, arg1);
        } else {
            if (R_FB_FILTER_TYPE == 7) {
                sMonoColors.unk_00 = 0;
                sMonoColors.primColor.r = R_FB_FILTER_PRIM_COLOR(0);
                sMonoColors.primColor.g = R_FB_FILTER_PRIM_COLOR(1);
                sMonoColors.primColor.b = R_FB_FILTER_PRIM_COLOR(2);
                sMonoColors.primColor.a = R_FB_FILTER_A;
                sMonoColors.envColor.r = R_FB_FILTER_ENV_COLOR(0);
                sMonoColors.envColor.g = R_FB_FILTER_ENV_COLOR(1);
                sMonoColors.envColor.b = R_FB_FILTER_ENV_COLOR(2);
                sMonoColors.envColor.a = R_FB_FILTER_A;
                VisMono_Draw(&sMonoColors, &_gfx, arg1);
            }
        }
    }

    *gfx = _gfx;
}

void Game_Nop80173534(GameState* gamestate) {
    ;
}

void GameState_Draw(GameState* gamestate, GraphicsContext* gfxCtx) {
    Gfx* nextDisplayList;
    Gfx* _polyOpa;
    // Unused vars impact regalloc
    Gfx* temp_t2;
    u32 temp_v1;

    _polyOpa = gfxCtx->polyOpa.p;
    nextDisplayList = Graph_GfxPlusOne(gfxCtx->polyOpa.p);
    gSPDisplayList(gfxCtx->overlay.p++, nextDisplayList);

    if (R_FB_FILTER_TYPE && R_FB_FILTER_ENV_COLOR(3) == 0) {
        GameState_SetFBFilter(&nextDisplayList, (u32)gfxCtx->zbuffer);
    }

    if (R_ENABLE_ARENA_DBG < 0) {
        R_ENABLE_ARENA_DBG = 0;
    }

    gSPEndDisplayList(nextDisplayList++);
    Graph_BranchDlist(_polyOpa, nextDisplayList);
    gfxCtx->polyOpa.p = nextDisplayList;

lblUnk:; // Label prevents reordering, if(1) around the above block don't seem to help unlike in OoT
    func_800E9F78(gfxCtx);

    if (R_ENABLE_ARENA_DBG != 0) {
        SpeedMeter_DrawTimeEntries(&D_801F7FF0, gfxCtx);
        SpeedMeter_DrawAllocEntries(&D_801F7FF0, gfxCtx, gamestate);
    }
}

void Game_ResetSegments(GraphicsContext* gfxCtx) {
    gSPSegment(gfxCtx->polyOpa.p++, 0, 0);
    gSPSegment(gfxCtx->polyOpa.p++, 0xF, gfxCtx->framebuffer);
    gSPSegment(gfxCtx->polyXlu.p++, 0, 0);
    gSPSegment(gfxCtx->polyXlu.p++, 0xF, gfxCtx->framebuffer);
    gSPSegment(gfxCtx->overlay.p++, 0, 0);
    gSPSegment(gfxCtx->overlay.p++, 0xF, gfxCtx->framebuffer);
}

void func_801736DC(GraphicsContext* gfxCtx) {
    Gfx* nextDisplayList;
    Gfx* _polyOpa;

    nextDisplayList = Graph_GfxPlusOne(_polyOpa = gfxCtx->polyOpa.p);
    gSPDisplayList(gfxCtx->overlay.p++, nextDisplayList);
    gSPEndDisplayList(nextDisplayList++);
    Graph_BranchDlist(_polyOpa, nextDisplayList);

    gfxCtx->polyOpa.p = nextDisplayList;
}

void Game_UpdateInput(GameState* gamestate) {
    Padmgr_GetInput(gamestate->input, 1);
}

void Game_Update(GameState* gamestate) {
    GraphicsContext* _gCtx;
    _gCtx = gamestate->gfxCtx;

    Game_ResetSegments(gamestate->gfxCtx);

    gamestate->main(gamestate);

    if (R_PAUSE_MENU_MODE != 2) {
        GameState_Draw(gamestate, _gCtx);
        func_801736DC(_gCtx);
    }
}

void Game_IncrementFrameCount(GameState* gamestate) {
    Game_Nop80173534(gamestate);
    gamestate->frames++;
}

void Game_InitHeap(GameState* gamestate, u32 size) {
    GameState* _ctx;
    void* buf;

    _ctx = gamestate;
    buf = Gamealloc_Alloc(&_ctx->alloc, size);

    if (buf) {
        THA_Ct(&gamestate->heap, buf, size);
        return;
    }

    THA_Ct(&gamestate->heap, NULL, 0);
    __assert("../game.c", 0x40B);
}

void Game_ResizeHeap(GameState* gamestate, u32 size) {
    GameAlloc* alloc;
    void* buf;
    u32 systemMaxFree;
    u32 bytesFree;
    u32 bytesAllocated;
    void* heapStart;

    heapStart = gamestate->heap.bufp;
    alloc = &gamestate->alloc;
    THA_Dt(&gamestate->heap);
    Gamealloc_Free(alloc, heapStart);
    StartHeap_AnalyzeArena(&systemMaxFree, &bytesFree, &bytesAllocated);
    size = ((systemMaxFree - (sizeof(ArenaNode))) < size) ? (0) : (size);
    if (!size) {
        size = systemMaxFree - (sizeof(ArenaNode));
    }

    if ((buf = Gamealloc_Alloc(alloc, size)) != NULL) {
        THA_Ct(&gamestate->heap, buf, size);
    } else {
        THA_Ct(&gamestate->heap, 0, 0);
        __assert("../game.c", 0x432);
    }
}

void Game_StateInit(GameState* gamestate, GameStateFunc gameStateInit, GraphicsContext* gfxCtx) {
    gamestate->gfxCtx = gfxCtx;
    gamestate->frames = 0U;
    gamestate->main = NULL;
    gamestate->destroy = NULL;
    gamestate->running = 1;
    gfxCtx->unk274 = D_801FBB88;
    gfxCtx->viConfigFeatures = gViConfigFeatures;
    gfxCtx->viConfigXScale = gViConfigXScale;
    gfxCtx->viConfigYScale = gViConfigYScale;
    gamestate->nextGameStateInit = NULL;
    gamestate->nextGameStateSize = 0U;

lblUnk:;
    Gamealloc_Init(&gamestate->alloc);
    Game_InitHeap(gamestate, 0x100000);
    Game_SetFramerateDivisor(gamestate, 3);

    gameStateInit(gamestate);

    func_80140CE0(&D_801F8010);
    func_801420C0(&D_801F8020);
    func_801418B0(&sMonoColors);
    func_80140898(&D_801F8048);
    func_801773A0(&D_801F7FF0);
    func_8013ED9C();

    osSendMesg(&gamestate->gfxCtx->unk5C, NULL, 1);
}

void Game_StateFini(GameState* gamestate) {
    func_80172BC0();
    func_8019E014();
    osRecvMesg(&gamestate->gfxCtx->unk5C, 0, 1);

    if (gamestate->destroy != 0) {
        gamestate->destroy(gamestate);
    }

    func_8013EDD0();
    func_801773C4(&D_801F7FF0);
    func_80140D04(&D_801F8010);
    func_801420F4(&D_801F8020);
    func_80141900(&sMonoColors);
    func_80140900(&D_801F8048);
    THA_Dt(&gamestate->heap);
    Gamealloc_FreeAll(&gamestate->alloc);
}

GameStateFunc Game_GetNextStateInit(GameState* gamestate) {
    return gamestate->nextGameStateInit;
}

u32 Game_GetNextStateSize(GameState* gamestate) {
    return gamestate->nextGameStateSize;
}

u32 Game_GetShouldContinue(GameState* gamestate) {
    return gamestate->running;
}

s32 Game_GetHeapFreeSize(GameState* gamestate) {
    return THA_GetSize(&gamestate->heap);
}

s32 func_80173B48(GameState* gamestate) {
    s32 result;
    result = OS_CYCLES_TO_NSEC(gamestate->framerateDivisor * sIrqMgrRetraceTime) - OS_CYCLES_TO_NSEC(D_801FBAF0);
    return result;
}