openmw-tes3mp/components/detournavigator/recastmeshmanager.cpp

#include "recastmeshmanager.hpp"

#include <BulletCollision/CollisionShapes/btCompoundShape.h>
#include <BulletCollision/CollisionShapes/btHeightfieldTerrainShape.h>

namespace DetourNavigator
{
    RecastMeshManager::RecastMeshManager(const Settings& settings, const TileBounds& bounds)
        : mShouldRebuild(false)
        , mMeshBuilder(settings, bounds)
    {
    }

    bool RecastMeshManager::addObject(const ObjectId id, const btCollisionShape& shape, const btTransform& transform,
                                      const AreaType areaType)
    {
        if (!mObjects.emplace(id, RecastMeshObject(shape, transform, areaType)).second)
            return false;
        mShouldRebuild = true;
        return mShouldRebuild;
    }

    bool RecastMeshManager::updateObject(const ObjectId id, const btTransform& transform, const AreaType areaType)
    {
        const auto object = mObjects.find(id);
        if (object == mObjects.end())
            return false;
        if (!object->second.update(transform, areaType))
            return false;
        mShouldRebuild = true;
        return mShouldRebuild;
    }

    boost::optional<RemovedRecastMeshObject> RecastMeshManager::removeObject(const ObjectId id)
    {
        const auto object = mObjects.find(id);
        if (object == mObjects.end())
            return boost::none;
        const RemovedRecastMeshObject result {object->second.getShape(), object->second.getTransform()};
        mObjects.erase(object);
        mShouldRebuild = true;
        return result;
    }

    bool RecastMeshManager::addWater(const osg::Vec2i& cellPosition, const int cellSize,
        const btTransform& transform)
    {
        if (!mWater.insert(std::make_pair(cellPosition, Water {cellSize, transform})).second)
            return false;
        mShouldRebuild = true;
        return true;
    }

    boost::optional<RecastMeshManager::Water> RecastMeshManager::removeWater(const osg::Vec2i& cellPosition)
    {
        const auto water = mWater.find(cellPosition);
        if (water == mWater.end())
            return boost::none;
        mShouldRebuild = true;
        const auto result = water->second;
        mWater.erase(water);
        return result;
    }

    std::shared_ptr<RecastMesh> RecastMeshManager::getMesh()
    {
        rebuild();
        return mMeshBuilder.create();
    }

    bool RecastMeshManager::isEmpty() const
    {
        return mObjects.empty();
    }

    void RecastMeshManager::rebuild()
    {
        if (!mShouldRebuild)
            return;
        mMeshBuilder.reset();
        for (const auto& v : mWater)
            mMeshBuilder.addWater(v.second.mCellSize, v.second.mTransform);
        for (const auto& v : mObjects)
            mMeshBuilder.addObject(v.second.getShape(), v.second.getTransform(), v.second.getAreaType());
        mShouldRebuild = false;
    }
}