#include "famicom_emu.h"

#include "m_malloc.h"
#include "_mem.h"
#include "dolphin/gx.h"
#include "Famicom/famicom.h"
#include "m_scene.h"
#include "jsyswrap.h"
#include "m_common_data.h"
#include "m_debug.h"
#include "libc64/sprintf.h"

int famicom_done;
int famicom_done_countdown;
void* freeXfbBase;
u32 freeXfbSize;

void* my_malloc_func[] = {
    my_zelda_malloc_align,
    my_zelda_free,
    my_zelda_getmemblocksize,
    my_zelda_gettotalfreesize,
};



void my_alloc_init(GAME_FAMICOM_EMU* famicom, void* start, size_t size){

    u32 freebytes;
    u32 alloc;
    u32 aligned;
    u32 tsize;
    
    freebytes = game_getFreeBytes(&famicom->game);
    alloc = (u32)THA_alloc16(&famicom->game.tha, freebytes);
    aligned = ALIGN_NEXT(alloc, 16);
    tsize = aligned - alloc;

    zelda_InitArena((void*)aligned, freebytes - tsize);

    if((start != NULL) && (size != 0)){

        zelda_AddBlockArena(start, size); 
    } 

}

void my_alloc_cleanup(){
    
    zelda_CleanupArena();
}

void my_zelda_getmemblocksize(void* ptr){

    zelda_GetMemBlockSize(ptr);
}

void my_zelda_gettotalfreesize(){
    
    zelda_GetTotalFreeSize();
}

void* my_zelda_malloc_align(size_t size, u32 align){

    void* aligned = zelda_malloc_align(size, align);

    if(aligned != NULL){

        memset(aligned, 0xFF, size);
    }
    return aligned;
}

void my_zelda_free(void* ptr){

    zelda_free(ptr);
}

void famicom_emu_main(GAME* famicom){
    static GXColor black_color = {0,0,0,0};
    GXColor t;
    int i;
    int padid;
    pad_t* current_pad;
    u32 combo;
    void* manager;

    if(famicom_done == 0){
        if(famicom_rom_load_check() < 0){
            Common_Set(famicom_2DBAC, Common_Get(famicom_2DBAC) | 1);
            famicom_done = 1;
            famicom_done_countdown = 0;
        }
        else{
            for(padid = 0,i = 4; i != 0; i--, padid++){

                current_pad = &gamePT->pads[padid]; 
                combo = current_pad->now.button | current_pad->on.button;

                if(combo == (BUTTON_Z | BUTTON_R | BUTTON_L) || (combo == 0xF0)){
                    
                    famicom_done = 1;
                    famicom_done_countdown = 60;
                    JC_JFWDisplay_startFadeOut(JC_JFWDisplay_getManager(), famicom_done_countdown);
                    break;
                }
            }
        }
    }
    if (famicom_done != 0) {
        if ( famicom_done_countdown == 0) {
            return_emu_game(famicom);
        } else {
            famicom_done_countdown -= 1;
        }
    }

    JW_BeginFrame();
    famicom->disable_display = 1;

    if(famicom_done == 0){
 
        famicom_1frame();
    }
    else{
        manager = JC_JFWDisplay_getManager();
        t = black_color; 
        JC_JFWDisplay_clearEfb(manager, &t);
    }
    JW_EndFrame();
}


void famicom_emu_init(GAME_FAMICOM_EMU* famicom){

    int rom_id;
    u8 player;
    int debug;
    void* manager;
    GXRenderModeObj* render;
    
    famicom_done = 0;
    famicom_done_countdown = 0;
    game_resize_hyral(&famicom->game, 0);
    Common_Set(famicom_287F9, 0); 
    
    rom_id = Common_Get(current_famicom_rom); 
    player = Common_Get(player_no);
    save_game_image = (GETREG(HREG, 4) == 1);
    debug = GETREG(HREG,3);

    if((rom_id > 0) && ( debug != 0)){
        rom_id = debug & (-debug & ~debug) >> 0x1f;
    }
 
    famicom->game.exec = famicom_emu_main;
    famicom->game.cleanup = famicom_emu_cleanup;
    

    while(sAdo_SubGameOK() == 0){
        VIWaitForRetrace();
        sAdo_GameFrame();
    }

    manager = JC_JFWDisplay_getManager();
    render = JC_JFWDisplay_getRenderMode(manager); 
    freeXfbBase = JC_JFWDisplay_changeToSingleXfb(manager, 1);
    freeXfbSize = render->fbWidth * render->xfbHeight * sizeof(u16);

    my_alloc_init(famicom, freeXfbBase, freeXfbSize);

    if(famicom_init(rom_id, my_malloc_func, player) != 0){
        Common_Set(famicom_2DBAC, Common_Get(famicom_2DBAC) | 1);
        return_emu_game(&famicom->game);
    }
}


void famicom_emu_cleanup(GAME* game){

    JC_JFWDisplay_startFadeIn(JC_JFWDisplay_getManager(), 1);

    if(famicom_cleanup() != 0){
        
        Common_Set(famicom_2DBAC, Common_Get(famicom_2DBAC) | 2);
    }

    my_alloc_cleanup();

    if(freeXfbBase != NULL){

        JC_JFWDisplay_changeToDoubleXfb(JC_JFWDisplay_getManager());    
        freeXfbBase = NULL;
        freeXfbSize = 0;
    }
    sAdo_SubGameEnd();
}

int famicom_gba_getImage(u32 rom_id, int* ptr){
    static char* names[] = {
        "cluclu",
        "usa_balloon",
        "donkey",
        "usa_jr_math",
        "pinball",
        "tennis",
        "usa_golf",
        NULL,
        "usa_baseball",
        NULL,
        "usa_donkey3",
        "donkeyjr",
        "soccer",
        "exbike",
        NULL,
        "usa_icecl",
        "mario",
        "smario",
        NULL
    };
    
    char buf[256];
    u32 resource;
    int block;
    char* rom;
    

    if(rom_id > 0x13){
        return 0;
    }
    

    rom = names[(rom_id == 0) ? 0 : rom_id -1]; 
    
    if(rom == NULL){
        return 0;
    }
    else{
    sprintf(buf, "/FAMICOM/GBA/jb_%s.bin.szs", rom);
    resource = JC__JKRGetResource(buf);
        if((resource != 0) && (ptr != NULL)){
            block = JC__JKRGetMemBlockSize(0, resource);
            *ptr = block;
        }
    }
    return resource;
}

void famicom_gba_removeImage(void* p){

    JC__JKRRemoveResource(p);
}