/*
===========================================================================
Copyright (C) 2024 the OpenMoHAA team

This file is part of OpenMoHAA source code.

OpenMoHAA source code is free software; you can redistribute it
and/or modify it under the terms of the GNU General Public License as
published by the Free Software Foundation; either version 2 of the License,
or (at your option) any later version.

OpenMoHAA source code is distributed in the hope that it will be
useful, but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with OpenMoHAA source code; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
===========================================================================
*/

// tiki_skel.cpp : TIKI skeletor loader

#include "q_shared.h"
#include "qcommon.h"
#include "tiki.h"
#include "../tiki/tiki_shared.h"

int         cache_numskel = 0;
int         cache_maxskel = 0;
skelcache_t skelcache[TIKI_MAX_SKELCACHE];

/*
===============
TIKI_AddPointToBounds
===============
*/
void TIKI_AddPointToBounds(float *v, float *mins, float *maxs)
{
    int   i;
    vec_t val;

    for (i = 0; i < 3; i++) {
        val = v[i];

        if (val < mins[i]) {
            mins[i] = val;
        }

        if (val > maxs[i]) {
            maxs[i] = val;
        }
    }
}

/*
===============
TIKI_Message
===============
*/
void TIKI_Message(const char *fmt, ...)
{
    char    msg[1024];
    va_list va;

    va_start(va, fmt);
    Q_vsnprintf(msg, sizeof(msg), fmt, va);
    va_end(va);
    TIKI_DPrintf(msg);
}

/*
===============
TIKI_Warning
===============
*/
void TIKI_Warning(const char *fmt, ...)
{
    char    msg[1024];
    va_list va;

    va_start(va, fmt);
    Q_vsnprintf(msg, sizeof(msg), fmt, va);
    va_end(va);
    TIKI_DPrintf(msg);
}

/*
===============
TIKI_Error
===============
*/
void TIKI_Error(const char *fmt, ...)
{
    char    msg[1024];
    va_list va;

    va_start(va, fmt);
    Q_vsnprintf(msg, sizeof(msg), fmt, va);
    va_end(va);
    TIKI_DPrintf(msg);
}

/*
===============
TIKI_SetupIndividualSurface
===============
*/
void TIKI_SetupIndividualSurface(const char *filename, dtikisurface_t *surf, const char *name, dloadsurface_t *loadsurf)
{
    int j;

    surf->numskins = 0;
    for (j = 0; j < loadsurf->numskins; j++) {
        if (surf->numskins >= MAX_TIKI_SHADER) {
            TIKI_Error(
                "TIKI_SetupIndividualSurface: Too many skins defined for surface %s in %s.\n", loadsurf->name, filename
            );
        } else {
            strncpy(surf->name, name, sizeof(surf->name));
            strncpy(surf->shader[surf->numskins], loadsurf->shader[j], sizeof(surf->shader[surf->numskins]));
            surf->numskins++;
        }
    }

    surf->flags             = loadsurf->flags;
    surf->damage_multiplier = loadsurf->damage_multiplier;
}

/*
===============
TIKI_CalcAnimDefSize
===============
*/
size_t TIKI_CalcAnimDefSize(dloaddef_t *ld)
{
    int i, j, k;

    // initial cmds
    size_t defsize = 0;
    // other animations
    defsize += sizeof(dtikianimdef_t) * ld->numanims;
    defsize += ld->numclientinitcmds * sizeof(dtikicmd_t);
    defsize += ld->numserverinitcmds * sizeof(dtikicmd_t);

    for (i = 0; i < ld->numserverinitcmds; i++) {
        const dloadinitcmd_t *initcmd = ld->loadserverinitcmds[i];
        defsize += sizeof(*initcmd->args) * initcmd->num_args;

        for (j = 0; j < initcmd->num_args; j++) {
            defsize += strlen(initcmd->args[j]) + 1;
        }

        defsize = PAD(defsize, sizeof(void *));
    }

    for (i = 0; i < ld->numclientinitcmds; i++) {
        const dloadinitcmd_t *initcmd = ld->loadclientinitcmds[i];
        defsize += sizeof(*initcmd->args) * initcmd->num_args;

        for (j = 0; j < initcmd->num_args; j++) {
            defsize += strlen(initcmd->args[j]) + 1;
        }

        defsize = PAD(defsize, sizeof(void *));
    }

    for (i = 0; i < ld->numanims; i++) {
        const dloadanim_t *loadanim = ld->loadanims[i];

        defsize += sizeof(dtikianimdef_t);
        defsize += loadanim->num_server_cmds * sizeof(dtikicmd_t);
        defsize += loadanim->num_client_cmds * sizeof(dtikianimdef_t);

        for (j = 0; j < loadanim->num_server_cmds; j++) {
            const dloadframecmd_t *loadframecmd = loadanim->loadservercmds[j];
            defsize += sizeof(*loadframecmd->args) * loadframecmd->num_args;

            for (k = 0; k < loadframecmd->num_args; k++) {
                defsize += strlen(loadframecmd->args[k]) + 1;
            }

            defsize = PAD(defsize, sizeof(void *));
        }

        for (j = 0; j < loadanim->num_client_cmds; j++) {
            const dloadframecmd_t *loadframecmd = loadanim->loadclientcmds[j];
            defsize += sizeof(*loadframecmd->args) * loadframecmd->num_args;

            for (k = 0; k < loadframecmd->num_args; k++) {
                defsize += strlen(loadframecmd->args[k]) + 1;
            }

            defsize = PAD(defsize, sizeof(void *));
        }

        defsize = PAD(defsize, sizeof(void *));
    }

    defsize += ld->modelBuf->cursize + strlen(ld->headmodels) + 1 + strlen(ld->headskins) + 1;
    defsize = PAD(defsize, sizeof(void *));
    return defsize;
}

/*
===============
TIKI_FillTIKIStructureSkel
===============
*/
dtikianim_t *TIKI_FillTIKIStructureSkel(dloaddef_t *ld)
{
    size_t animdefsize;
    size_t defsize;

    animdefsize = TIKI_CalcAnimDefSize(ld);
    defsize     = sizeof(dtikianim_t);
    // anim name
    defsize += strlen(ld->path) + 1;

    return TIKI_InitTiki(ld, PAD(animdefsize + defsize, sizeof(void *)));
}

/*
===============
TIKI_SwapSkel

Swap the skeleton header on Big-Endian systems
===============
*/
void TIKI_SwapSkel(skelHeader_t *pheader)
{
#ifdef Q3_BIG_ENDIAN
    skelSurface_t *pSurf;
    int            i, j, k;

    //pheader->ident = LittleLong(pheader->ident);
    pheader->version     = LittleLong(pheader->version);
    pheader->numSurfaces = LittleLong(pheader->numSurfaces);
    pheader->numBones    = LittleLong(pheader->numBones);
    pheader->ofsBones    = LittleLong(pheader->ofsBones);
    pheader->ofsSurfaces = LittleLong(pheader->ofsSurfaces);
    pheader->ofsEnd      = LittleLong(pheader->ofsEnd);

    if (pheader->version >= TIKI_SKB_HEADER_VERSION) {
        for (i = 0; i < TIKI_SKEL_LOD_INDEXES; i++) {
            pheader->lodIndex[i] = LittleLong(pheader->lodIndex[i]);
        }

        pheader->numBoxes = LittleLong(pheader->numBoxes);
        pheader->ofsBoxes = LittleLong(pheader->ofsBoxes);
    }

    if (pheader->version >= TIKI_SKD_HEADER_OLD_VERSION) {
        pheader->numMorphTargets = LittleLong(pheader->numMorphTargets);
        pheader->ofsMorphTargets = LittleLong(pheader->ofsMorphTargets);
    }

    if (pheader->version >= TIKI_SKD_HEADER_VERSION) {
        pheader->scale = LittleFloat(pheader->scale);
    }

    if (pheader->version >= TIKI_SKD_HEADER_OLD_VERSION) {
        boneFileData_t *boneBuffer;

        boneBuffer = (boneFileData_t *)((byte *)pheader + pheader->ofsBones);
        for (i = 0; i < pheader->numBones; i++) {
            boneBuffer->boneType        = (boneType_t)LittleLong(boneBuffer->boneType);
            boneBuffer->ofsBaseData     = LittleLong(boneBuffer->ofsBaseData);
            boneBuffer->ofsChannelNames = LittleLong(boneBuffer->ofsChannelNames);
            boneBuffer->ofsBoneNames    = LittleLong(boneBuffer->ofsBoneNames);
            boneBuffer->ofsEnd          = LittleLong(boneBuffer->ofsEnd);

            boneBuffer = (boneFileData_t *)((byte *)boneBuffer + boneBuffer->ofsEnd);
        }
    } else {
        skelBoneName_t *TIKI_bones;

        TIKI_bones = (skelBoneName_t *)((byte *)pheader + pheader->ofsBones);

        for (i = 0; i < pheader->numBones; i++) {
            TIKI_bones->parent   = LittleShort(TIKI_bones->parent);
            TIKI_bones->boxIndex = LittleShort(TIKI_bones->boxIndex);
            TIKI_bones->flags    = LittleLong(TIKI_bones->flags);

            TIKI_bones++;
        }
    }

    pSurf = (skelSurface_t *)((byte *)pheader + pheader->ofsSurfaces);
    for (i = 0; i < pheader->numSurfaces; i++) {
        skeletorMorph_t *pMorph;
        skelWeight_t    *pWeight;
        int             *pTriangles;
        int             *pCollapse, *pCollapseIndex;
        int              numVerts, numTriangles;

        CopyLittleLong(&pSurf->ident, &pSurf->ident);
        CopyLittleLong(&pSurf->numTriangles, &pSurf->numTriangles);
        CopyLittleLong(&pSurf->numVerts, &pSurf->numVerts);
        CopyLittleLong(&pSurf->staticSurfProcessed, &pSurf->staticSurfProcessed);
        CopyLittleLong(&pSurf->ofsTriangles, &pSurf->ofsTriangles);
        CopyLittleLong(&pSurf->ofsVerts, &pSurf->ofsVerts);
        CopyLittleLong(&pSurf->ofsCollapse, &pSurf->ofsCollapse);
        CopyLittleLong(&pSurf->ofsEnd, &pSurf->ofsEnd);

        numVerts     = LongNoSwapPtr(&pSurf->numVerts);
        numTriangles = LongNoSwapPtr(&pSurf->numTriangles);

        if (pheader->version >= TIKI_SKB_HEADER_VERSION) {
            CopyLittleLong(&pSurf->ofsCollapseIndex, &pSurf->ofsCollapseIndex);
        }

        pTriangles = (int *)((byte *)pSurf + LongNoSwapPtr(&pSurf->ofsTriangles));
        for (j = 0; j < numTriangles; j++) {
            CopyLittleLong(&pTriangles[j * 3 + 0], &pTriangles[j * 3 + 0]);
            CopyLittleLong(&pTriangles[j * 3 + 1], &pTriangles[j * 3 + 1]);
            CopyLittleLong(&pTriangles[j * 3 + 2], &pTriangles[j * 3 + 2]);
        }

        if (pheader->version >= TIKI_SKD_HEADER_OLD_VERSION) {
            skeletorVertex_t *pVert = (skeletorVertex_t *)((byte *)pSurf + LongNoSwapPtr(&pSurf->ofsVerts));

            for (j = 0; j < numVerts; j++) {
                int numMorphs, numWeights;

                CopyLittleLong(&pVert->normal[0], &pVert->normal[0]);
                CopyLittleLong(&pVert->normal[1], &pVert->normal[1]);
                CopyLittleLong(&pVert->normal[2], &pVert->normal[2]);
                CopyLittleLong(&pVert->texCoords[0], &pVert->texCoords[0]);
                CopyLittleLong(&pVert->texCoords[1], &pVert->texCoords[1]);

                numMorphs = LittleLong(pVert->numMorphs);
                memcpy(&pVert->numMorphs, &numMorphs, sizeof(pVert->numMorphs));

                pMorph = (skeletorMorph_t *)((byte *)pVert + sizeof(*pVert));

                for (k = 0; k < numMorphs; k++, pMorph++) {
                    int    morphIndex;
                    vec3_t offset;

                    CopyLittleLong(&pMorph->morphIndex, &pMorph->morphIndex);
                    CopyLittleLong(&pMorph->offset[0], &pMorph->offset[0]);
                    CopyLittleLong(&pMorph->offset[1], &pMorph->offset[1]);
                    CopyLittleLong(&pMorph->offset[2], &pMorph->offset[2]);
                }

                numWeights = LittleLong(pVert->numWeights);
                memcpy(&pVert->numWeights, &numWeights, sizeof(pVert->numWeights));

                pWeight = (skelWeight_t *)pMorph;

                for (k = 0; k < numWeights; k++, pWeight++) {
                    CopyLittleLong(&pWeight->boneIndex, &pWeight->boneIndex);
                    CopyLittleLong(&pWeight->boneWeight, &pWeight->boneWeight);
                    CopyLittleLong(&pWeight->offset[0], &pWeight->offset[0]);
                    CopyLittleLong(&pWeight->offset[1], &pWeight->offset[1]);
                    CopyLittleLong(&pWeight->offset[2], &pWeight->offset[2]);
                }

                pVert = (skeletorVertex_t *)pWeight;
            }
        } else {
            skelVertex_t *pVert = (skelVertex_t *)((byte *)pSurf + LongNoSwapPtr(&pSurf->ofsVerts));

            for (j = 0; j < numVerts; j++) {
                vec3_t normal, texCoords;
                int    numWeights;

                CopyLittleLong(&pVert->normal[0], &pVert->normal[0]);
                CopyLittleLong(&pVert->normal[1], &pVert->normal[1]);
                CopyLittleLong(&pVert->normal[2], &pVert->normal[2]);
                CopyLittleLong(&pVert->texCoords[0], &pVert->texCoords[0]);
                CopyLittleLong(&pVert->texCoords[1], &pVert->texCoords[1]);

                numWeights = LittleLong(pVert->numWeights);
                memcpy(&pVert->numWeights, &numWeights, sizeof(pVert->numWeights));

                pWeight = pVert->weights;

                for (k = 0; k < numWeights; k++, pWeight++) {
                    CopyLittleLong(&pWeight->boneIndex, &pWeight->boneIndex);
                    CopyLittleLong(&pWeight->boneWeight, &pWeight->boneWeight);
                    CopyLittleLong(&pWeight->offset[0], &pWeight->offset[0]);
                    CopyLittleLong(&pWeight->offset[1], &pWeight->offset[1]);
                    CopyLittleLong(&pWeight->offset[2], &pWeight->offset[2]);
                }

                pVert = (skelVertex_t *)pWeight;
            }
        }

        pCollapse = (int *)((byte *)pSurf + LongNoSwapPtr(&pSurf->ofsCollapse));
        for (j = 0; j < numVerts; j++) {
            CopyLittleLong(&pCollapse[j], &pCollapse[j]);
        }

        if (pheader->version >= TIKI_SKB_HEADER_VERSION) {
            pCollapseIndex = (int *)((byte *)pSurf + LongNoSwapPtr(&pSurf->ofsCollapseIndex));
            for (j = 0; j < numVerts; j++) {
                CopyLittleLong(&pCollapseIndex[j], &pCollapseIndex[j]);
            }
        }

        pSurf = (skelSurface_t *)((byte *)pSurf + LongNoSwapPtr(&pSurf->ofsEnd));
    }
#endif
}