Bvh.h

#pragma once
#ifndef DISABLE_BVH
#include <AtomsUtils/AtomsMath.h>
#include <AtomsUtils/Mesh.h>
#include <vector>
 
namespace AtomsUtils {
 
    enum BVHNodeType { INTERNAL, LEAF };
    struct BVHNode {
 
        int leftChildIndex;
        int rightChildIndex;
        int start;
        int end;
        BVHNodeType leftType;
        BVHNodeType rightType;
    };
 
    struct MortonCode {
        unsigned int code;
        int index;
    };
 
    class Bvh {
    public:
        Bvh();
 
        ~Bvh();
 
        void init(Mesh *mesh);
        void clear();
 
        bool intersect(const AtomsMath::Vector3f &orig, const AtomsMath::Vector3f &dir, float &param,
            unsigned int &outFaceId, float &outU, float &outV,
            bool bothDirection) const;
        bool RayIntersectsTriangle(const AtomsMath::Vector3f rayOrigin,
            const AtomsMath::Vector3f rayVector, int i,
            AtomsMath::Vector3f &outIntersectionPoint,
            float &param) const;
 
        std::vector<MortonCode> mortons;
        std::vector<BVHNode> internalNodes;
        std::vector<AtomsMath::Vector3f> treeAABBs;
        AtomsUtils::Mesh *m_mesh;
        std::vector<AtomsMath::Vector3f> m_outMesh;
    };
} // namespace AtomsUtils
#else
namespace AtomsUtils {
 
class Bvh {
public:
    Bvh() {}
 
    ~Bvh() {}
 
    void init(Mesh *mesh) {}
    void clear() {}
 
    bool intersect(const AtomsMath::Vector3f &orig, const AtomsMath::Vector3f &dir, float &param,
        unsigned int &outFaceId, float &outU, float &outV,
        bool bothDirection) const {
        return false;
    }
    bool RayIntersectsTriangle(const AtomsMath::Vector3f rayOrigin,
        const AtomsMath::Vector3f rayVector, int i,
        AtomsMath::Vector3f &outIntersectionPoint,
        float &param) const {
        return false;
    }
};
}
#endif