#include <global.h>
#include <vt.h>
#include <padmgr.h>
#include <sched.h>

u32 gScreenWidth = SCREEN_WIDTH;
u32 gScreenHeight = SCREEN_HEIGHT;
u32 gSystemHeapSize = 0;

PreNmiBuff* gAppNmiBufferPtr;
SchedContext gSchedContext;
PadMgr gPadMgr;
IrqMgr gIrqMgr;
u32 gSegments[NUM_SEGMENTS];
OSThread sGraphThread;
u8 sGraphStack[0x1800];
u8 sSchedStack[0x600];
u8 sAudioStack[0x800];
u8 sPadMgrStack[0x500];
u8 sIrqMgrStack[0x500];
StackEntry sGraphStackInfo;
StackEntry sSchedStackInfo;
StackEntry sAudioStackInfo;
StackEntry sPadMgrStackInfo;
StackEntry sIrqMgrStackInfo;
u8 gAudioMgr[0x298]; //type should be AudioMgr
OSMesgQueue sSiIntMsgQ;
OSMesg sSiIntMsgBuf[1];


void Main_LogSystemHeap()
{
    osSyncPrintf(VT_FGCOL(GREEN));
    //System heap size% 08x (% dKB) Start address% 08x
    osSyncPrintf("システムヒープサイズ %08x(%dKB) 開始アドレス %08x\n", gSystemHeapSize, gSystemHeapSize / 1024, gSystemHeap);
    osSyncPrintf(VT_RST);
}

void Main(void* arg0)
{
    IrqMgrClient irqClient;
    OSMesgQueue irqMgrMsgQ;
    OSMesg irqMgrMsgBuf[60];
    u32 sysHeap;
    u32 fb;
    s32 debugHeap;
    s32 debugHeapSize;
    s16 *msg;

    osSyncPrintf("mainproc 実行開始\n"); //Start running
    gScreenWidth = SCREEN_WIDTH;
    gScreenHeight = SCREEN_HEIGHT;
    gAppNmiBufferPtr = (PreNmiBuff*)osAppNmiBuffer;
    PreNmiBuff_Init(gAppNmiBufferPtr);
    Fault_Start();
    SysCfb_Init(0);
    sysHeap = (u32)gSystemHeap;
    fb = SysCfb_GetFbPtr(0);
    gSystemHeapSize = (fb - sysHeap);
    osSyncPrintf("システムヒープ初期化 %08x-%08x %08x\n", sysHeap, fb, gSystemHeapSize); //System heap initalization
    SystemHeap_Init(sysHeap, gSystemHeapSize); //initializes the system heap
    if (osMemSize >= 0x800000U)
    {
        debugHeap = SysCfb_GetFbEnd();
        debugHeapSize = (s32) (0x80600000 - debugHeap);
    }
    else
    {
        debugHeapSize = 0x400;
        debugHeap = SystemArena_MallocDebug(debugHeapSize, "../main.c", 0x235);
    }
    osSyncPrintf("debug_InitArena(%08x, %08x)\n", debugHeap, debugHeapSize);
    DebugArena_Init(debugHeap, debugHeapSize);
    func_800636C0();

    SREG(0) = 0;

    osCreateMesgQueue(&sSiIntMsgQ, sSiIntMsgBuf, 1);
    osSetEventMesg(5, &sSiIntMsgQ, 0);

    Main_LogSystemHeap();

    osCreateMesgQueue(&irqMgrMsgQ, irqMgrMsgBuf, 0x3c);
    StackCheck_Init(&sIrqMgrStackInfo, sIrqMgrStack, sIrqMgrStack+sizeof(sIrqMgrStack), 0, 0x100, "irqmgr");
    IrqMgr_Create(&gIrqMgr, &sGraphStackInfo, 0x11, 1);

    osSyncPrintf("タスクスケジューラの初期化\n"); //Initialize the task scheduler
    StackCheck_Init(&sSchedStackInfo, sSchedStack, sSchedStack+sizeof(sSchedStack), 0, 0x100, "sched");
    func_800C9874(&gSchedContext, &sAudioStack, 0xf, D_80013960, 1, &gIrqMgr);

    IrqMgr_AddClient(&gIrqMgr, &irqClient, &irqMgrMsgQ);

    StackCheck_Init(&sAudioStackInfo, sAudioStack, sAudioStack+sizeof(sAudioStack), 0, 0x100, "audio");
    func_800C3FEC(&gAudioMgr, sAudioStack+sizeof(sAudioStack), 0xc, 0xa, &gSchedContext, &gIrqMgr);

    StackCheck_Init(&sPadMgrStackInfo, sPadMgrStack, sPadMgrStack+sizeof(sPadMgrStack), 0, 0x100, "padmgr");
    PadMgr_Init(&gPadMgr, &sSiIntMsgQ, &gIrqMgr, 7, 0xe, &sIrqMgrStack);

    func_800C3FC4(&gAudioMgr);

    StackCheck_Init(&sGraphStackInfo, sGraphStack, sGraphStack+sizeof(sGraphStack), 0, 0x100, "graph");
    osCreateThread(&sGraphThread, 4, Graph_ThreadEntry, arg0, sGraphStack+sizeof(sGraphStack), 0xb);
    osStartThread(&sGraphThread);
    osSetThreadPri(0, 0xf);

    while (true)
    {
        msg = NULL;
        osRecvMesg(&irqMgrMsgQ, &msg, OS_MESG_BLOCK);
        if (msg == NULL)
            break;
        if (*msg == OS_SC_PRE_NMI_MSG)
        {
            osSyncPrintf("main.c: リセットされたみたいだよ\n"); //Looks like it's been reset
            PreNmiBuff_SetReset(gAppNmiBufferPtr);
        }
    }

    osSyncPrintf("mainproc 後始末\n"); //Cleanup
    osDestroyThread(&sGraphThread);
    func_800FBFD8();
    osSyncPrintf("mainproc 実行終了\n"); //End of execution
}