CompressedLeafBVH.hpp

Go to the documentation of this file.

// -------------------------------------------------------//
//
// SHAMROCK code for hydrodynamics
// Copyright (c) 2021-2026 Timothée David--Cléris <tim.shamrock@proton.me>
// SPDX-License-Identifier: CeCILL Free Software License Agreement v2.1
// Shamrock is licensed under the CeCILL 2.1 License, see LICENSE for more information
//
// -------------------------------------------------------//
 
#pragma once
 
#include "shammath/sfc/morton.hpp"
#include "shamtree/CLBVHObjectIterator.hpp"
#include "shamtree/CellIterator.hpp"
#include "shamtree/KarrasRadixTree.hpp"
#include "shamtree/KarrasRadixTreeAABB.hpp"
#include "shamtree/MortonReducedSet.hpp"
 
namespace shamtree {
 
    template<class Tmorton, class Tvec, u32 dim>
    class CompressedLeafBVH;
 
} // namespace shamtree
 
template<class Tmorton, class Tvec, u32 dim>
class shamtree::CompressedLeafBVH {
 
    void internal_rebuild_from_positions_no_aabb(
        sham::DeviceBuffer<Tvec> &positions,
        u32 obj_cnt,
        const shammath::AABB<Tvec> &bounding_box,
        u32 compression_level);
 
    public:
    inline u32 get_total_cell_count() { return structure.get_total_cell_count(); }
 
    inline u32 get_internal_cell_count() { return structure.get_internal_cell_count(); }
 
    inline bool is_root_leaf() const { return structure.is_root_leaf(); }
 
    inline u32 get_leaf_cell_count() { return structure.get_leaf_count(); }
 
    MortonReducedSet<Tmorton, Tvec, dim> reduced_morton_set;
 
    KarrasRadixTree structure;
 
    KarrasRadixTreeAABB<Tvec> aabbs;
 
    CompressedLeafBVH(
        MortonReducedSet<Tmorton, Tvec, dim> &&reduced_morton_set,
        KarrasRadixTree &&structure,
        KarrasRadixTreeAABB<Tvec> &&aabbs)
        : reduced_morton_set(
              std::forward<MortonReducedSet<Tmorton, Tvec, dim>>(reduced_morton_set)),
          structure(std::forward<KarrasRadixTree>(structure)),
          aabbs(std::forward<KarrasRadixTreeAABB<Tvec>>(aabbs)) {}
 
    static CompressedLeafBVH make_empty(sham::DeviceScheduler_ptr dev_sched);
 
    inline bool is_empty() const { return reduced_morton_set.is_empty(); }
 
    void rebuild_from_positions(
        sham::DeviceBuffer<Tvec> &positions,
        const shammath::AABB<Tvec> &bounding_box,
        u32 compression_level);
 
    void rebuild_from_positions(
        sham::DeviceBuffer<Tvec> &positions,
        u32 obj_cnt,
        const shammath::AABB<Tvec> &bounding_box,
        u32 compression_level);
 
    void rebuild_from_position_range(
        sham::DeviceBuffer<Tvec> &min,
        sham::DeviceBuffer<Tvec> &max,
        u32 obj_cnt,
        shammath::AABB<Tvec> &bounding_box,
        u32 compression_level);
 
    void rebuild_from_position_range(
        sham::DeviceBuffer<Tvec> &min,
        sham::DeviceBuffer<Tvec> &max,
        shammath::AABB<Tvec> &bounding_box,
        u32 compression_level);
 
    inline shamtree::CLBVHTraverser<Tmorton, Tvec, dim> get_traverser() const {
        return {structure.get_structure_traverser(), aabbs.buf_aabb_min, aabbs.buf_aabb_max};
    }
 
    inline shamtree::CLBVHTraverserHost<Tmorton, Tvec, dim> get_traverser_host() const {
        return {
            structure.get_structure_traverser_host(),
            aabbs.buf_aabb_min.copy_to_stdvec(),
            aabbs.buf_aabb_max.copy_to_stdvec()};
    }
 
    inline shamtree::CLBVHObjectIterator<Tmorton, Tvec, dim> get_object_iterator() const {
        return {reduced_morton_set.get_leaf_cell_iterator(), get_traverser()};
    }
 
    inline shamtree::CLBVHObjectIteratorHost<Tmorton, Tvec, dim> get_object_iterator_host() const {
        return {reduced_morton_set.get_leaf_cell_iterator_host(), get_traverser_host()};
    }
 
    inline CellIterator get_cell_iterator() const {
        return {reduced_morton_set.get_cell_iterator(
            structure.buf_endrange, structure.get_internal_cell_count())};
    }
 
    inline CellIteratorHost get_cell_iterator_host() const {
        return {reduced_morton_set.get_cell_iterator_host(
            structure.buf_endrange, structure.get_internal_cell_count())};
    }
};