#include "CGameArea.h"
#include "Core/Resource/Script/CScriptLayer.h"
#include "Core/Render/CRenderer.h"

CGameArea::CGameArea()
    : CResource()
    , mWorldIndex(-1)
    , mVertexCount(0)
    , mTriangleCount(0)
    , mTerrainMerged(false)
    , mOriginalWorldMeshCount(0)
    , mUsesCompression(false)
    , mMaterialSet(nullptr)
    , mpGeneratorLayer(nullptr)
    , mpCollision(nullptr)
{
}

CGameArea::~CGameArea()
{
    ClearTerrain();

    delete mpCollision;
    delete mpGeneratorLayer;

    for (u32 iSCLY = 0; iSCLY < mScriptLayers.size(); iSCLY++)
        delete mScriptLayers[iSCLY];

    for (u32 lyr = 0; lyr < mLightLayers.size(); lyr++)
        for (u32 lit = 0; lit < mLightLayers[lyr].size(); lit++)
            delete mLightLayers[lyr][lit];
}

void CGameArea::AddWorldModel(CModel *mdl)
{
    mTerrainModels.push_back(mdl);
    mVertexCount += mdl->GetVertexCount();
    mTriangleCount += mdl->GetTriangleCount();
    mAABox.ExpandBounds(mdl->AABox());
}

void CGameArea::MergeTerrain()
{
    if (mTerrainMerged) return;

    // Nothing really complicated here - iterate through every terrain submesh, add each to a static model
    for (u32 iMdl = 0; iMdl < mTerrainModels.size(); iMdl++)
    {
        CModel *pMdl = mTerrainModels[iMdl];
        u32 SubmeshCount = pMdl->GetSurfaceCount();

        for (u32 iSurf = 0; iSurf < SubmeshCount; iSurf++)
        {
            SSurface *pSurf = pMdl->GetSurface(iSurf);
            CMaterial *pMat = mMaterialSet->MaterialByIndex(pSurf->MaterialID);

            bool newMat = true;
            for (std::vector<CStaticModel*>::iterator it = mStaticTerrainModels.begin(); it != mStaticTerrainModels.end(); it++)
            {
                if ((*it)->GetMaterial() == pMat)
                {
                    // When we append a new submesh to an existing static model, we bump it to the back of the vector.
                    // This is because mesh ordering actually matters sometimes
                    // (particularly with multi-layered transparent meshes)
                    // so we need to at least try to maintain it.
                    // This is maybe not the most efficient way to do this, but it works.
                    CStaticModel *pStatic = *it;
                    pStatic->AddSurface(pSurf);
                    mStaticTerrainModels.erase(it);
                    mStaticTerrainModels.push_back(pStatic);
                    newMat = false;
                    break;
                }
            }

            if (newMat)
            {
                CStaticModel *pStatic = new CStaticModel(pMat);
                pStatic->AddSurface(pSurf);
                mStaticTerrainModels.push_back(pStatic);
            }
        }
    }
}

void CGameArea::ClearTerrain()
{
    for (u32 t = 0; t < mTerrainModels.size(); t++)
        delete mTerrainModels[t];
    mTerrainModels.clear();

    for (u32 s = 0; s < mStaticTerrainModels.size(); s++)
        delete mStaticTerrainModels[s];
    mStaticTerrainModels.clear();

    if (mMaterialSet) delete mMaterialSet;

    mVertexCount = 0;
    mTriangleCount = 0;
    mTerrainMerged = false;
    mAABox = CAABox::skInfinite;
}

void CGameArea::ClearScriptLayers()
{
    for (auto it = mScriptLayers.begin(); it != mScriptLayers.end(); it++)
        delete *it;
    mScriptLayers.clear();
    delete mpGeneratorLayer;
    mpGeneratorLayer = nullptr;
}

u32 CGameArea::TotalInstanceCount() const
{
    u32 Num = 0;

    for (u32 iLyr = 0; iLyr < mScriptLayers.size(); iLyr++)
        Num += mScriptLayers[iLyr]->NumInstances();

    return Num;
}

CScriptObject* CGameArea::InstanceByID(u32 InstanceID)
{
    auto it = mObjectMap.find(InstanceID);
    if (it != mObjectMap.end()) return it->second;
    else return nullptr;
}


CScriptObject* CGameArea::SpawnInstance(CScriptTemplate *pTemplate,
                                        CScriptLayer *pLayer,
                                        const CVector3f& rkPosition /*= CVector3f::skZero*/,
                                        const CQuaternion& rkRotation /*= CQuaternion::skIdentity*/,
                                        const CVector3f& rkScale /*= CVector3f::skOne*/,
                                        u32 SuggestedID /*= -1*/,
                                        u32 SuggestedLayerIndex /*= -1*/ )
{
    // Verify we can fit another instance in this area.
    u32 NumInstances = TotalInstanceCount();

    if (NumInstances >= 0xFFFF)
    {
        Log::Error("Unable to spawn a new script instance; too many instances in area (" + TString::FromInt32(NumInstances, 0, 10) + ")");
        return nullptr;
    }

    // Check whether the suggested instance ID is valid
    u32 InstanceID = SuggestedID;

    if (InstanceID != -1)
    {
        if (mObjectMap.find(InstanceID) == mObjectMap.end())
            InstanceID = -1;
    }

    // If not valid (or if there's no suggested ID) then determine a new instance ID
    if (InstanceID == -1)
    {
        // Determine layer index
        u32 LayerIndex = -1;

        for (u32 iLyr = 0; iLyr < mScriptLayers.size(); iLyr++)
        {
            if (mScriptLayers[iLyr] == pLayer)
            {
                LayerIndex = iLyr;
                break;
            }
        }

        if (LayerIndex == -1)
        {
            Log::Error("Unable to spawn a new script instance; invalid script layer passed in");
            return nullptr;
        }

        // Look for a valid instance ID
        InstanceID = (LayerIndex << 26) | (mWorldIndex << 16) | 1;

        while (true)
        {
            auto it = mObjectMap.find(InstanceID);

            if (it == mObjectMap.end())
                break;
            else
                InstanceID++;
        }
    }

    // Spawn instance
    CScriptObject *pInstance = new CScriptObject(InstanceID, this, pLayer, pTemplate);
    pInstance->EvaluateProperties();
    pInstance->SetPosition(rkPosition);
    pInstance->SetRotation(rkRotation.ToEuler());
    pInstance->SetScale(rkScale);
    pInstance->SetName(pTemplate->Name());
    if (pTemplate->Game() < eEchoesDemo) pInstance->SetActive(true);
    pLayer->AddInstance(pInstance, SuggestedLayerIndex);
    mObjectMap[InstanceID] = pInstance;
    return pInstance;
}

void CGameArea::DeleteInstance(CScriptObject *pInstance)
{
    pInstance->Layer()->RemoveInstance(pInstance);
    pInstance->Template()->RemoveObject(pInstance);

    auto it = mObjectMap.find(pInstance->InstanceID());
    if (it != mObjectMap.end()) mObjectMap.erase(it);

    if (mpPoiToWorldMap && mpPoiToWorldMap->HasPoiMappings(pInstance->InstanceID()))
        mpPoiToWorldMap->RemovePoi(pInstance->InstanceID());

    pInstance->BreakAllLinks();
    delete pInstance;
}