bvhTriangleMeshShape.h

Go to the documentation of this file.

/*******************************************************************************
** Copyright (c) 2023 MAK Technologies, Inc.
** All rights reserved.
*******************************************************************************/


#pragma once

#include <vantageTerrainImplementation/vantageTerrainImplementationDefines.h>
#include <vantageTerrainImplementation/bulletUtil.h>
#include <BulletCollision/CollisionDispatch/btCollisionObject.h>
#include <BulletCollision/CollisionShapes/btBvhTriangleMeshShape.h>
#include <BulletCollision/CollisionShapes/btTriangleMesh.h>

#include <boost/shared_ptr.hpp>
#include <boost/enable_shared_from_this.hpp>

#include <osg/Vec3>

#include <tbb/spin_mutex.h>

class DtSurface;
class btVector3;

class DT_DLL_vantageTerrainImplementation DtTriangleMeshWithSoil : public btTriangleMesh
{
public:
   DtTriangleMeshWithSoil(bool use32bitIndices = true, bool use4componentVertices = true);
   virtual ~DtTriangleMeshWithSoil();

   bool valid(const btVector3& vertex0, const btVector3& vertex1, const btVector3& vertex2) const;
   bool valid(const btVector3& vertex) const;
   bool valid(const btScalar& val) const;

   bool checkAddTriangle(const btVector3& vertex0, const btVector3& vertex1, const btVector3& vertex2,
      bool removeDuplicateVertices = false);

   void addTriangle(const btVector3& vertex0, const btVector3& vertex1, const btVector3& vertex2,
      bool removeDuplicateVertices = false);

   void addTriangleWithSurface(const btVector3& vertex0, const btVector3& vertex1, const btVector3& vertex2,
      const DtSurface& surface, bool removeDuplicateVertices = false);

   DtSurface triangleSurface(unsigned int triangleIndex) const;

   const btVector3& max() const;
   const btVector3& min() const;

   bool isAabbValid() const;

   double maxDistance() const;

protected:
   void updateBounds(const btVector3& val);

   void initializeSurfaceArray();

   void syncSurfaceArray();

   bool myUseSurfaceArray;
   btAlignedObjectArray<DtSurface> mySurfaceTypes;
   btVector3 myMin;
   btVector3 myMax;
   double myMaxPointDistanceSq;
};

class DT_DLL_vantageTerrainImplementation DtBvhTriangleMeshShape
   : public btBvhTriangleMeshShape
   , public boost::enable_shared_from_this<DtBvhTriangleMeshShape>
{
public:
   using Ptr = boost::shared_ptr<DtBvhTriangleMeshShape>;

   DtBvhTriangleMeshShape(
      DtTriangleMeshWithSoil* mesh,
      bool useQuantization = true, bool buildBvhNow = false,
      bool filterTransparentPolygons = false);

   ~DtBvhTriangleMeshShape();

   bool buildBvh();
   bool isBvhBuilt() const;

   int getNumTriangles() const;

private:
   using Mutex = tbb::spin_mutex;

   mutable Mutex myMutex;
   DtTriangleMeshWithSoil* myMesh;

   int myTrianglesCount;
   tbb::atomic<bool> myIsBvhBuilt;
};

class DT_DLL_vantageTerrainImplementation DtBvhCollisionObject
   : public btCollisionObject
   , public DtCollisionObjectInterface
{
public:
   DtBvhCollisionObject();

   void setCollisionShape(btCollisionShape* shape) override;

   bool buildBvh() override;
   bool enabled() const override;
   bool isBvhBuilt() const;

   boost::shared_ptr<btCollisionShape> bulletCollisionShape() const override;

   unsigned int numberOfTriangles() const override;
   double radius() const override;
   DtSurface triangleSurface(unsigned int triangleIndex) const override;

   DynamicFeatures dynamicFeatures() const override;

   const btTransform& worldTransform() const override;

protected:
   DtBvhTriangleMeshShape::Ptr myShape;
};

struct ProcessTriangle
{
   ProcessTriangle();
   void operator() (const osg::Vec3& vec1, const osg::Vec3& vec2, const osg::Vec3& vec3, bool) const;

   DtTriangleMeshWithSoil* myTriMesh;
};

struct ProcessTriangleWithSurface : public ProcessTriangle
{
   void operator() (const osg::Vec3& vec1, const osg::Vec3& vec2, const osg::Vec3& vec3) const;

   DtSurface mySurface;
   osg::Vec3d myScale;
   bool myIsScale;
};