#include "JSystem/JSystem.h" // IWYU pragma: keep

#include "JSystem/JParticle/JPAResourceLoader.h"
#include "JSystem/JKernel/JKRHeap.h"
#include "JSystem/JParticle/JPABaseShape.h"
#include "JSystem/JParticle/JPAChildShape.h"
#include "JSystem/JParticle/JPADynamicsBlock.h"
#include "JSystem/JParticle/JPAExTexShape.h"
#include "JSystem/JParticle/JPAExtraShape.h"
#include "JSystem/JParticle/JPAFieldBlock.h"
#include "JSystem/JParticle/JPAKeyBlock.h"
#include "JSystem/JParticle/JPAResource.h"
#include "JSystem/JParticle/JPAResourceManager.h"
#include "os_report.h"

static void dummy1() {
    JUT_WARN(0, "JPA : wrong version file %s\n");
    OS_REPORT("JPA : no jpa file\n");
}

JPAResourceLoader::JPAResourceLoader(u8 const* data, JPAResourceManager* mgr) {
    if (*(BE(u32)*)(data + 4) == '2-10') {
        load_jpc(data, mgr);
    } else {
        JUT_WARN(48, "JPA : wrong version file\n");
    }
}

static void dummy2() {
    OS_REPORT("tex_num && \"a resource have to use one or more texture\"");
    OS_REPORT("p_res != 0");
    OS_REPORT("(p_res->ppFld != 0) || (p_res->fldNum == 0)");
    OS_REPORT("(p_res->ppKey != 0) || (p_res->keyNum == 0)");
    OS_REPORT("(p_res->texIdxTbl != 0) || (p_res->texNum == 0)");
    OS_REPORT("p_res->ppFld[fld_no] != 0");
    OS_REPORT("p_res->ppKey[key_no] != 0");
    OS_REPORT("p_res->pDyn != 0");
    OS_REPORT("p_res->pBsp != 0");
    OS_REPORT("p_res->pEsp != 0");
    OS_REPORT("p_res->pCsp != 0");
    OS_REPORT("p_res->pEts != 0");
    OS_REPORT("JPA : wrong type block in jpc file %x\n");
}

struct JPAResourceHeader {
    /* 0x0 */ BE(u16) mUsrIdx;
    /* 0x2 */ BE(u16) mBlockNum;
    /* 0x4 */ u8 mFieldBlockNum;
    /* 0x5 */ u8 mKeyBlockNum;
    /* 0x6 */ u8 mTDB1Num;
};

void JPAResourceLoader::load_jpc(u8 const* data, JPAResourceManager* p_res_mgr) {
    JKRHeap* heap = p_res_mgr->mpHeap;
    p_res_mgr->resMaxNum = *(BE(u16)*)(data + 8);
    p_res_mgr->texMaxNum = *(BE(u16)*)(data + 0xA);
    p_res_mgr->pResAry = JKR_NEW_ARRAY_ARGS(JPAResource*, p_res_mgr->resMaxNum, heap, 0);
    p_res_mgr->pTexAry = JKR_NEW_ARRAY_ARGS(JPATexture*, p_res_mgr->texMaxNum, heap, 0);
    JUT_ASSERT(199, (p_res_mgr->pResAry != NULL) && (p_res_mgr->pTexAry != 0));

    u32 offset = 0x10;
    for (int i = 0; i < *(BE(u16)*)(data + 8); i++) {
        JPAResourceHeader* header = (JPAResourceHeader*)(data + offset);
        JPAResource* p_res = JKR_NEW_ARGS (heap, 0) JPAResource();
        JUT_ASSERT(211, p_res != NULL);
        p_res->fldNum = header->mFieldBlockNum;
        p_res->ppFld = p_res->fldNum != 0 ?
            JKR_NEW_ARRAY_ARGS(JPAFieldBlock*, p_res->fldNum, heap, 0) : NULL;
        JUT_ASSERT(216, (p_res->ppFld != NULL) || (p_res->fldNum == 0));
        p_res->keyNum = header->mKeyBlockNum;
        p_res->ppKey = p_res->keyNum != 0 ?
            JKR_NEW_ARRAY_ARGS(JPAKeyBlock*, p_res->keyNum, heap, 0) : NULL;
        JUT_ASSERT(221, (p_res->ppKey != NULL) || (p_res->keyNum == 0));
        p_res->texNum = header->mTDB1Num;
        p_res->mpTDB1 = NULL;
        p_res->mUsrIdx = header->mUsrIdx;

        offset += 8;
        u32 fld_no = 0;
        u32 key_no = 0;

        for (int j = 0; j < header->mBlockNum; j++) {
            u32 magic = *(BE(u32)*)(data + offset);
            u32 size = *(BE(u32)*)(data + offset + 4);
            switch (magic) {
            case 'FLD1':
                p_res->ppFld[fld_no] = JKR_NEW_ARGS (heap, 0) JPAFieldBlock(data + offset, heap);
                JUT_ASSERT(244, p_res->ppFld[fld_no] != NULL);
                fld_no++;
                break;
            case 'KFA1':
                p_res->ppKey[key_no] = JKR_NEW_ARGS (heap, 0) JPAKeyBlock(data + offset);
                JUT_ASSERT(249, p_res->ppKey[key_no] != NULL);
                key_no++;
                break;
            case 'BEM1':
                p_res->pDyn = JKR_NEW_ARGS (heap, 0) JPADynamicsBlock(data + offset);
                JUT_ASSERT(254, p_res->pDyn != NULL);
                break;
            case 'BSP1':
                p_res->pBsp = JKR_NEW_ARGS (heap, 0) JPABaseShape(data + offset, heap);
                JUT_ASSERT(258, p_res->pBsp != NULL);
                break;
            case 'ESP1':
                p_res->pEsp = JKR_NEW_ARGS (heap, 0) JPAExtraShape(data + offset);
                JUT_ASSERT(262, p_res->pEsp != NULL);
                break;
            case 'SSP1':
                p_res->pCsp = JKR_NEW_ARGS (heap, 0) JPAChildShape(data + offset);
                JUT_ASSERT(266, p_res->pCsp != NULL);
                break;
            case 'ETX1':
                p_res->pEts = JKR_NEW_ARGS (heap, 0) JPAExTexShape(data + offset);
                JUT_ASSERT(270, p_res->pEts != NULL);
                break;
            case 'TDB1':
                p_res->mpTDB1 = (const BE(u16)*)(data + offset + 8);
                break;
            default:
                JUT_WARN(275, "JPA : wrong type block in jpc file %d %x\n", header->mBlockNum, offset);
                break;
            }
            offset += size;
        }

        p_res->init(heap);
        p_res_mgr->registRes(p_res);
    }

    offset = *(BE(u32)*)(data + 0xC);
    for (int i = 0; i < *(BE(u16)*)(data + 0xA); i++) {
        u32 size = *(BE(u32)*)(data + offset + 4);
        JPATexture* p_tex = JKR_NEW_ARGS (heap, 0) JPATexture(data + offset);
        JUT_ASSERT(298, p_tex != NULL);
        p_res_mgr->registTex(p_tex);
        offset += size;
    }
}