Viano-Core/src/tools/mesh_extractor/ContinentBuilder.cpp

#include "ContinentBuilder.h"
#include "TileBuilder.h"
#include "WDT.h"
#include "Utils.h"
#include "DetourNavMesh.h"
#include "Cache.h"
#include "ace/Task.h"
#include "Recast.h"
#include "DetourCommon.h"

class BuilderThread : public ACE_Task_Base
{
private:
    int X, Y, MapId;
    bool Instance;
    std::string Continent;
    dtNavMeshParams Params;
    ContinentBuilder* cBuilder;
    WorldModelRoot* Model;
    const WorldModelDefinition* Definition;
public:
    BuilderThread(ContinentBuilder* _cBuilder, dtNavMeshParams& params) : Params(params), cBuilder(_cBuilder), Free(true) {}
    void SetData(int x, int y, int map, const std::string& cont, bool instance, WorldModelRoot* model, const WorldModelDefinition* def) 
    { 
        X = x; 
        Y = y; 
        MapId = map; 
        Continent = cont; 
        Instance = instance; 
        if (Instance) 
        {
            Model = model; 
            Definition = def;
        }
        else 
            Model = NULL; 
    }

    int svc()
    {
        if (Instance)
        {
            // Build a WMO
            printf("Building WMO %s ( %u )", Continent.c_str(), MapId);
            TileBuilder builder(cBuilder, Continent, X, Y, MapId);
            char buff[100];
            sprintf(buff, "mmaps/%03u%02i%02i.mmtile", MapId, Y, X);
            FILE* f = fopen(buff, "r");
            if (f) // Check if file already exists.
            {
                printf("Instance %s ( %u ) skipped, file already exists\n", Continent.c_str(), MapId);
                fclose(f);
                return 0;
            }

            uint8* nav = builder.BuildInstance(Params);
            if (nav)
            {
                f = fopen(buff, "wb");
                if (!f)
                {
                    printf("Could not create file %s. Check that you have write permissions to the destination folder and try again\n", buff);
                    return 0;
                }
                MmapTileHeader header;
                header.size = builder.DataSize;
                fwrite(&header, sizeof(MmapTileHeader), 1, f);
                fwrite(nav, sizeof(unsigned char), builder.DataSize, f);
                fclose(f);
            }
            dtFree(nav);
            return 0;
        }
        else
        {
            Free = false;
            printf("[%02i,%02i] Building tile\n", X, Y);
            TileBuilder builder(cBuilder, Continent, X, Y, MapId);
            char buff[100];
            sprintf(buff, "mmaps/%03u%02i%02i.mmtile", MapId, Y, X);
            FILE* f = fopen(buff, "r");
            if (f) // Check if file already exists.
            {
                printf("[%02i,%02i] Tile skipped, file already exists\n", X, Y);
                fclose(f);
                Free = true;
                return 0;
            }
            uint8* nav = builder.Build(Params);
            if (nav)
            {
                f = fopen(buff, "wb");
                if (!f)
                {
                    printf("Could not create file %s. Check that you have write permissions to the destination folder and try again\n", buff);
                    return 0;
                }
                MmapTileHeader header;
                header.size = builder.DataSize;
                fwrite(&header, sizeof(MmapTileHeader), 1, f);
                fwrite(nav, sizeof(unsigned char), builder.DataSize, f);
                fclose(f);
            }
            dtFree(nav);
            printf("[%02i,%02i] Tile Built!\n", X, Y);
            Free = true;
        }
        return 0;
    }

    bool Free;
};

void ContinentBuilder::getTileBounds(uint32 tileX, uint32 tileY, float* verts, int vertCount, float* bmin, float* bmax)
{
    // this is for elevation
    if (verts && vertCount)
        rcCalcBounds(verts, vertCount, bmin, bmax);
    else
    {
        bmin[1] = FLT_MIN;
        bmax[1] = FLT_MAX;
    }

    // this is for width and depth
    bmax[0] = (32 - int(tileX)) * Constants::TileSize;
    bmax[2] = (32 - int(tileY)) * Constants::TileSize;
    bmin[0] = bmax[0] - Constants::TileSize;
    bmin[2] = bmax[2] - Constants::TileSize;
}

void ContinentBuilder::CalculateTileBounds()
{
    for (std::vector<TilePos>::iterator itr = TileMap->TileTable.begin(); itr != TileMap->TileTable.end(); ++itr)
    {
        tileXMax = std::max(itr->X, tileXMax);
        tileXMin = std::min(itr->X, tileXMin);

        tileYMax = std::max(itr->Y, tileYMax);
        tileYMin = std::min(itr->Y, tileYMin);
    }
    getTileBounds(tileXMax, tileYMax, NULL, 0, bmin, bmax);
}

void ContinentBuilder::Build()
{
    char buff[50];
    sprintf(buff, "mmaps/%03u.mmap", MapId);
    FILE* mmap = fopen(buff, "wb");
    if (!mmap)
    {
        printf("Could not create file %s. Check that you have write permissions to the destination folder and try again\n", buff);
        return;
    }

    CalculateTileBounds();

    dtNavMeshParams params;
    
    std::vector<BuilderThread*> Threads;

    if (TileMap->IsGlobalModel)
    {
        printf("Map %s ( %u ) is a WMO. Building with 1 thread.\n", Continent.c_str(), MapId);
        
        TileBuilder* builder = new TileBuilder(this, Continent, 0, 0, MapId);
        builder->AddGeometry(TileMap->Model, TileMap->ModelDefinition);
        builder->SetCoords(0, 0);
        uint8* nav = builder->BuildInstance(params);
        if (nav)
        {
            // Set some params for the navmesh
            dtMeshHeader* header = (dtMeshHeader*)nav;
            dtVcopy(params.orig, header->bmin);
            params.tileWidth = header->bmax[0] - header->bmin[0];
            params.tileHeight = header->bmax[2] - header->bmin[2];
            params.maxTiles = 1;
            params.maxPolys = header->polyCount;
            fwrite(&params, sizeof(dtNavMeshParams), 1, mmap);
            fclose(mmap);

            char buff[100];
            sprintf(buff, "mmaps/%03u%02i%02i.mmtile", MapId, 0, 0);
            FILE* f = fopen(buff, "wb");
            if (!f)
            {
                printf("Could not create file %s. Check that you have write permissions to the destination folder and try again\n", buff);
                return;
            }

            MmapTileHeader mheader;
            mheader.size = builder->DataSize;
            fwrite(&mheader, sizeof(MmapTileHeader), 1, f);
            fwrite(nav, sizeof(unsigned char), builder->DataSize, f);
            fclose(f);
        }

        dtFree(nav);
        delete builder;
    }
    else
    {
        params.maxPolys = 1 << STATIC_POLY_BITS;
        params.maxTiles = TileMap->TileTable.size();
        rcVcopy(params.orig, bmin);
        params.tileHeight = Constants::TileSize;
        params.tileWidth = Constants::TileSize;
        fwrite(&params, sizeof(dtNavMeshParams), 1, mmap);
        fclose(mmap);

        for (uint32 i = 0; i < NumberOfThreads; ++i)
            Threads.push_back(new BuilderThread(this, params));
        printf("Map %s ( %u ) has %u tiles. Building them with %u threads\n", Continent.c_str(), MapId, uint32(TileMap->TileTable.size()), NumberOfThreads);
        for (std::vector<TilePos>::iterator itr = TileMap->TileTable.begin(); itr != TileMap->TileTable.end(); ++itr)
        {
            bool next = false;
            while (!next)
            {
                for (std::vector<BuilderThread*>::iterator _th = Threads.begin(); _th != Threads.end(); ++_th)
                {
                    if ((*_th)->Free)
                    {
                        (*_th)->SetData(itr->X, itr->Y, MapId, Continent, false, NULL, NULL);
                        (*_th)->activate();
                        next = true;
                        break;
                    }
                }
                // Wait for 20 seconds
                ACE_OS::sleep(ACE_Time_Value (0, 20000));
            }
        }
    }

    Cache->Clear();

    // Free memory
    for (std::vector<BuilderThread*>::iterator _th = Threads.begin(); _th != Threads.end(); ++_th)
    {
        (*_th)->wait();
        delete *_th;
    }
}