MRAABBTreeBase.h

Go to the documentation of this file.

#pragma once
 
#include "MRAABBTreeNode.h"
#include "MRVector.h"
 
namespace MR
{
 

template <typename T>
class MRMESH_CLASS AABBTreeBase
{
public:
    using Traits = T;
    using Node = AABBTreeNode<Traits>;
    using NodeVec = Vector<Node, NodeId>;
    using LeafTag = typename T::LeafTag;
    using LeafId = typename T::LeafId;
    using LeafBitSet = TaggedBitSet<LeafTag>;
    using LeafBMap = BMap<LeafId, LeafId>;
    using BoxT = typename T::BoxT;
 
public:
    [[nodiscard]] const NodeVec & nodes() const { return nodes_; }

    [[nodiscard]] const Node & operator[]( NodeId nid ) const { return nodes_[nid]; }

    [[nodiscard]] static NodeId rootNodeId() { return NodeId{ 0 }; }

    [[nodiscard]] BoxT getBoundingBox() const { return nodes_.empty() ? BoxT{} : nodes_[rootNodeId()].box; }

    [[nodiscard]] size_t heapBytes() const { return nodes_.heapBytes(); }

    [[nodiscard]] size_t numLeaves() const { return nodes_.empty() ? 0 : ( nodes_.size() + 1 ) / 2; }

    [[nodiscard]] MRMESH_API std::vector<NodeId> getSubtrees( int minNum ) const;

    [[nodiscard]] MRMESH_API LeafBitSet getSubtreeLeaves( NodeId subtreeRoot ) const;

    [[nodiscard]] MRMESH_API NodeBitSet getNodesFromLeaves( const LeafBitSet & leaves ) const;

    MRMESH_API void getLeafOrder( LeafBMap & leafMap ) const;

    MRMESH_API void getLeafOrderAndReset( LeafBMap & leafMap );
 
protected:
    NodeVec nodes_;
};
 
}