KarrasRadixTreeAABB.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 "shambackends/DeviceBuffer.hpp"
#include "shambackends/DeviceScheduler.hpp"
#include "shambackends/EventList.hpp"
#include "shambackends/kernel_call.hpp"
#include "shambackends/math.hpp"
#include "shamsys/NodeInstance.hpp"
#include "shamtree/CellIterator.hpp"
#include "shamtree/KarrasRadixTree.hpp"
#include "shamtree/LeafCellIterator.hpp"
#include <functional>
#include <utility>
 
namespace shamtree {
    template<class Tvec>
    class KarrasRadixTreeAABB;
} // namespace shamtree
 
template<class Tvec>
class shamtree::KarrasRadixTreeAABB {
 
    public:
    inline u32 get_total_cell_count() { return buf_aabb_min.get_size(); }
 
    sham::DeviceBuffer<Tvec> buf_aabb_min; 
    sham::DeviceBuffer<Tvec> buf_aabb_max; 

    KarrasRadixTreeAABB(
        sham::DeviceBuffer<Tvec> &&buf_cell_min, sham::DeviceBuffer<Tvec> &&buf_cell_max)
        : buf_aabb_min(std::move(buf_cell_min)), buf_aabb_max(std::move(buf_cell_max)) {}
 
    static inline KarrasRadixTreeAABB make_empty(sham::DeviceScheduler_ptr dev_sched) {
        return KarrasRadixTreeAABB{
            sham::DeviceBuffer<Tvec>(0, dev_sched), sham::DeviceBuffer<Tvec>(0, dev_sched)};
    }
};
 
namespace shamtree {
 
    template<class Tvec>
    KarrasRadixTreeAABB<Tvec> new_empty_karras_radix_tree_aabb() {
        auto dev_sched = shamsys::instance::get_compute_scheduler_ptr();
        return KarrasRadixTreeAABB<Tvec>(
            sham::DeviceBuffer<Tvec>(0, dev_sched), sham::DeviceBuffer<Tvec>(0, dev_sched));
    }

    template<class Tvec>
    KarrasRadixTreeAABB<Tvec> prepare_karras_radix_tree_aabb(
        const KarrasRadixTree &tree, KarrasRadixTreeAABB<Tvec> &&recycled_tree_aabb);

    template<class Tvec>
    void propagate_aabb_up(KarrasRadixTreeAABB<Tvec> &tree_aabb, const KarrasRadixTree &tree);

    template<class Tvec>
    KarrasRadixTreeAABB<Tvec> compute_tree_aabb(
        const KarrasRadixTree &tree,
        KarrasRadixTreeAABB<Tvec> &&recycled_tree_aabb,
        const std::function<void(KarrasRadixTreeAABB<Tvec> &, u32)> &fct_fill_leaf);

    template<class Tvec>
    KarrasRadixTreeAABB<Tvec> compute_tree_aabb_from_positions(
        const KarrasRadixTree &tree,
        const LeafCellIterator &cell_it,
        KarrasRadixTreeAABB<Tvec> &&recycled_tree_aabb,
        sham::DeviceBuffer<Tvec> &positions);

    template<class Tvec>
    KarrasRadixTreeAABB<Tvec> compute_tree_aabb_from_position_ranges(
        const KarrasRadixTree &tree,
        const LeafCellIterator &cell_it,
        KarrasRadixTreeAABB<Tvec> &&recycled_tree_aabb,
        sham::DeviceBuffer<Tvec> &min,
        sham::DeviceBuffer<Tvec> &max);
 
} // namespace shamtree