#include "dynos.cpp.h"
extern "C" {
#include "behavior_table.h"
#include "levels/scripts.h"
}

  /////////////
 // Writing //
/////////////

typedef Pair<String, u32> PointerData;
static PointerData GetDataFromPointer(const void* aPtr, GfxData* aGfxData) {
    // Lights
    for (auto& _Node : aGfxData->mLights) {
        if (&_Node->mData->l[0] == aPtr) { // Light *, not Lights1 *
            return { _Node->mName, 1 };
        }
        if (&_Node->mData->a == aPtr) { // Ambient *, not Lights1 *
            return { _Node->mName, 2 };
        }
    }

    // Light_ts
    for (auto& _Node : aGfxData->mLightTs) {
        if (&_Node->mData->col[0] == aPtr) {
            return { _Node->mName, 1 };
        }
        if (&_Node->mData->colc[0] == aPtr) {
            return { _Node->mName, 2 };
        }
        if (&_Node->mData->dir[0] == aPtr) {
            return { _Node->mName, 3 };
        }
    }

    // Ambient_ts
    for (auto& _Node : aGfxData->mAmbientTs) {
        if (&_Node->mData->col[0] == aPtr) {
            return { _Node->mName, 1 };
        }
        if (&_Node->mData->colc[0] == aPtr) {
            return { _Node->mName, 2 };
        }
    }

    // Textures
    for (auto& _Node : aGfxData->mTextures) {
        if (_Node == aPtr) {
            return { _Node->mName, 0 };
        }
    }

    // Texture Lists
    for (auto& _Node : aGfxData->mTextureLists) {
        if (_Node == aPtr) {
            return { _Node->mName, 0 };
        }
    }

    // Display lists
    for (auto& _Node : aGfxData->mDisplayLists) {
        if (_Node == aPtr) {
            return { _Node->mName, 0 };
        }
    }

    // Geo layouts
    for (auto& _Node : aGfxData->mGeoLayouts) {
        if (_Node->mData == aPtr) {
            return { _Node->mName, 0 };
        }
    }

    // Collisions
    for (auto& _Node : aGfxData->mCollisions) {
        if (_Node->mData == aPtr) {
            return { _Node->mName, 0 };
        }
    }

    // Macro objects
    for (auto& _Node : aGfxData->mMacroObjects) {
        if (_Node->mData == aPtr) {
            return { _Node->mName, 0 };
        }
    }

    // Trajectories
    for (auto& _Node : aGfxData->mTrajectories) {
        if (_Node->mData == aPtr) {
            return { _Node->mName, 0 };
        }
    }

    // Movtexs
    for (auto& _Node : aGfxData->mMovtexs) {
        if (_Node->mData == aPtr) {
            return { _Node->mName, 0 };
        }
    }

    // MovtexQCs
    for (auto& _Node : aGfxData->mMovtexQCs) {
        if (_Node->mData == aPtr) {
            return { _Node->mName, 0 };
        }
    }

    // Rooms
    for (auto& _Node : aGfxData->mRooms) {
        if (_Node->mData == aPtr) {
            return { _Node->mName, 0 };
        }
    }

    // Behaviors
    enum BehaviorId id = get_id_from_vanilla_behavior((const BehaviorScript*) aPtr);
    if (id >= 0 && id < id_bhv_max_count) {
        return { get_behavior_name_from_id(id), 0 };
    }

    // Vanilla Actors
    s32 actorCount = DynOS_Geo_GetActorCount();
    for (s32 i = 0; i < actorCount; i++) {
        if (DynOS_Geo_IsCustomActor(i)) { break; }
        if (aPtr == DynOS_Geo_GetActorLayout(i)) {
            return { DynOS_Geo_GetActorName(i), 0 };
        }
    }

    // Vanilla Lvl Geos
    auto vanillaGeo = DynOS_Mgr_VanillaLvlGeo_GetFromData((const GeoLayout*)aPtr);
    if (vanillaGeo != NULL) {
        return { vanillaGeo, 0 };
    }

    // Vanilla Lvl Cols
    auto vanillaCol = DynOS_Mgr_VanillaLvlCol_GetFromData((const Collision*)aPtr);
    if (vanillaCol != NULL) {
        return { vanillaCol, 0 };
    }

    // Vanilla Script Pointers
    auto vanillaScriptPtr = DynOS_Mgr_VanillaScriptPtr_GetFromData(aPtr);
    if (vanillaScriptPtr != NULL) {
        return { vanillaScriptPtr, 0 };
    }

    // Vanilla Textures
    auto vanillaTex = DynOS_Mgr_VanillaTex_GetFromData((const Texture*)aPtr);
    if (vanillaTex != NULL) {
        return { vanillaTex, 0 };
    }

    // the ones below use a saved offset

    s32 _Offset = 0;
    for (auto& pair : aGfxData->mPointerOffsetList) {
        if (pair.first == aPtr) {
            _Offset = (s32)((const LevelScript*)pair.first - (const LevelScript*)pair.second);
            aPtr = pair.second;
            break;
        }
    }

    // Level scripts
    for (auto& _Node : aGfxData->mLevelScripts) {
        if (_Node->mData == aPtr) {
            return { _Node->mName, _Offset };
        }
    }

    // Vertices
    String _VtxArrayName = "";
    uintptr_t _VtxArrayStart = 0;
    for (auto& _Node : aGfxData->mVertices) {
        if (_Node->mData == aPtr) {
            return { _Node->mName, _Offset };
        }
    }

    PrintError("Unable to find pointer!");
    return { "", 0 };
}

void DynOS_Pointer_Lua_Write(FILE* aFile, u32 index, GfxData* aGfxData) {
    String& token = aGfxData->mLuaTokenList[index];
    WriteBytes<u32>(aFile, LUA_VAR_CODE);
    token.Write(aFile);
}

void DynOS_Pointer_Write(FILE* aFile, const void* aPtr, GfxData* aGfxData) {

    // NULL
    if (!aPtr) {
        WriteBytes<u32>(aFile, 0);
        return;
    }

    // Lua variable
    for (s32 i = 0; i < aGfxData->mLuaPointerList.Count(); i++) {
        if (aPtr == aGfxData->mLuaPointerList[i]) {
            u32 index = *((u32*)aPtr);
            String& token = aGfxData->mLuaTokenList[index];
            WriteBytes<u32>(aFile, LUA_VAR_CODE);
            token.Write(aFile);
            return;
        }
    }

    // Built-in functions
    s32 _GeoFunctionIndex = DynOS_Mgr_VanillaFunc_GetIndexFromData(aPtr);
    if (_GeoFunctionIndex != -1) {
        WriteBytes<u32>(aFile, FUNCTION_CODE);
        WriteBytes<s32>(aFile, _GeoFunctionIndex);
        return;
    }

    // Pointer
    PointerData _PtrData = GetDataFromPointer(aPtr, aGfxData);
    WriteBytes<u32>(aFile, POINTER_CODE);
    _PtrData.first.Write(aFile);
    WriteBytes<u32>(aFile, _PtrData.second);
}

  /////////////
 // Reading //
/////////////

static void *GetPointerFromData(GfxData *aGfxData, const String &aPtrName, u32 aPtrData) {

    // Lights
    for (auto& _Node : aGfxData->mLights) {
        if (_Node->mName == aPtrName) {
            if (aPtrData == 1) {
                return (void *) &_Node->mData->l[0];
            }
            if (aPtrData == 2) {
                return (void *) &_Node->mData->a;
            }
            sys_fatal("Unknown Light type: %u", aPtrData);
        }
    }

    // Light_ts
    for (auto& _Node : aGfxData->mLightTs) {
        if (_Node->mName == aPtrName) {
            if (aPtrData == 1) {
                return (void *) &_Node->mData->col[0];
            }
            if (aPtrData == 2) {
                return (void *) &_Node->mData->colc[0];
            }
            if (aPtrData == 3) {
                return (void *) &_Node->mData->dir[0];
            }
            sys_fatal("Unknown Light type: %u", aPtrData);
        }
    }

    // Ambient_ts
    for (auto& _Node : aGfxData->mAmbientTs) {
        if (_Node->mName == aPtrName) {
            if (aPtrData == 1) {
                return (void *) &_Node->mData->col[0];
            }
            if (aPtrData == 2) {
                return (void *) &_Node->mData->colc[0];
            }
            sys_fatal("Unknown Light type: %u", aPtrData);
        }
    }

    // Textures
    for (auto& _Node : aGfxData->mTextures) {
        if (_Node->mName == aPtrName) {
            return (void *) _Node;
        }
    }

    // Texture Lists
    for (auto& _Node : aGfxData->mTextureLists) {
        if (_Node->mName == aPtrName) {
            return (void *) _Node;
        }
    }

    // Display lists
    for (auto &_Node : aGfxData->mDisplayLists) {
        if (_Node->mName == aPtrName) {
            return (void *) _Node->mData;
        }
    }

    // Geo layouts
    for (auto &_Node : aGfxData->mGeoLayouts) {
        if (_Node->mName == aPtrName) {
            return (void *) _Node->mData;
        }
    }

    // Vertices
    for (auto &_Node : aGfxData->mVertices) {
        if (_Node->mName == aPtrName) {
            return (void *) (_Node->mData + aPtrData);
        }
    }

    // Collisions
    for (auto &_Node : aGfxData->mCollisions) {
        if (_Node->mName == aPtrName) {
            return (void *) _Node->mData;
        }
    }

    // Level scripts
    for (auto &_Node : aGfxData->mLevelScripts) {
        if (_Node->mName == aPtrName) {
            return (void *) (_Node->mData + aPtrData);
        }
    }

    // Macro objects
    for (auto &_Node : aGfxData->mMacroObjects) {
        if (_Node->mName == aPtrName) {
            return (void *) _Node->mData;
        }
    }

    // Trajectories
    for (auto &_Node : aGfxData->mTrajectories) {
        if (_Node->mName == aPtrName) {
            return (void *) _Node->mData;
        }
    }

    // Movtexs
    for (auto &_Node : aGfxData->mMovtexs) {
        if (_Node->mName == aPtrName) {
            return (void *) _Node->mData;
        }
    }

    // MovtexQCs
    for (auto &_Node : aGfxData->mMovtexQCs) {
        if (_Node->mName == aPtrName) {
            return (void *) _Node->mData;
        }
    }

    // Rooms
    for (auto &_Node : aGfxData->mRooms) {
        if (_Node->mName == aPtrName) {
            return (void *) _Node->mData;
        }
    }

    // Behaviors
    enum BehaviorId id = get_id_from_behavior_name(aPtrName.begin());
    if (id >= 0 && id < id_bhv_max_count) {
        return (void*)get_behavior_from_id(id);
    }

    // Vanilla Actors
    s32 actorCount = DynOS_Geo_GetActorCount();
    for (s32 i = 0; i < actorCount; i++) {
        if (DynOS_Geo_IsCustomActor(i)) { break; }
        if (!strcmp(aPtrName.begin(), DynOS_Geo_GetActorName(i))) {
            return (void*)DynOS_Geo_GetActorLayout(i);
        }
    }

    // Vanilla Lvl Geos
    auto vanillaGeo = DynOS_Mgr_VanillaLvlGeo_GetFromName(aPtrName.begin());
    if (vanillaGeo != NULL) {
        return (void*)vanillaGeo;
    }

    // Vanilla Lvl Cols
    auto vanillaCol = DynOS_Mgr_VanillaLvlCol_GetFromName(aPtrName.begin());
    if (vanillaCol != NULL) {
        return (void*)vanillaCol;
    }

    // Vanilla Script Pointers
    auto vanillaScriptPtr = DynOS_Mgr_VanillaScriptPtr_GetFromName(aPtrName.begin());
    if (vanillaScriptPtr != NULL) {
        return (void*)vanillaScriptPtr;
    }

    // Vanilla Textures
    auto vanillaTex = DynOS_Mgr_VanillaTex_GetFromName(aPtrName.begin());
    if (vanillaTex != NULL) {
        return (void*)vanillaTex;
    }

    // Error
    sys_fatal("Pointer not found: %s", aPtrName.begin());
    return NULL;
}

void *DynOS_Pointer_Load(FILE *aFile, GfxData *aGfxData, u32 aValue) {

    // LUAV
    if (aValue == LUA_VAR_CODE) {
        String token; token.Read(aFile);
        for (s32 i = 0; i < aGfxData->mLuaTokenList.Count(); i++) {
            if (token == aGfxData->mLuaTokenList[i]) {
                return (void*)(uintptr_t)(i+1);
            }
        }
        u32 index = aGfxData->mLuaTokenList.Count();
        aGfxData->mLuaTokenList.Add(token);
        return (void*)(uintptr_t)(index+1);
    }

    // FUNC
    if (aValue == FUNCTION_CODE) {
        s32 _FunctionIndex = ReadBytes<s32>(aFile);
        return (void*) DynOS_Mgr_VanillaFunc_GetFromIndex(_FunctionIndex);
    }

    // PNTR
    if (aValue == POINTER_CODE) {
        String _PtrName; _PtrName.Read(aFile);
        u32   _PtrData = ReadBytes<u32>(aFile);
        return GetPointerFromData(aGfxData, _PtrName, _PtrData);
    }

    // Not a pointer
    return NULL;
}