MortonCodeSet.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 "shambase/aliases_int.hpp"
#include "shambackends/DeviceBuffer.hpp"
#include "shambackends/DeviceScheduler.hpp"
#include "shambackends/vec.hpp"
#include "shammath/AABB.hpp"
 
namespace shamtree {
 
    template<class Tmorton, class Tvec, u32 dim = shambase::VectorProperties<Tvec>::dimension>
    class MortonCodeSet {
        public:
        shammath::AABB<Tvec> bounding_box;
 
        u32 cnt_obj;
 
        u32 morton_count;
 
        sham::DeviceBuffer<Tmorton> morton_codes;
 
        MortonCodeSet(
            shammath::AABB<Tvec> &&bounding_box,
            u32 &&cnt_obj,
            u32 &&morton_count,
            sham::DeviceBuffer<Tmorton> &&morton_codes)
            : bounding_box(std::move(bounding_box)), cnt_obj(std::move(cnt_obj)),
              morton_count(std::move(morton_count)), morton_codes(std::move(morton_codes)) {}
    };
 
    template<class Tmorton, class Tvec, u32 dim>
    MortonCodeSet<Tmorton, Tvec, dim> morton_code_set_from_positions(
        const sham::DeviceScheduler_ptr &dev_sched,
        shammath::AABB<Tvec> bounding_box,
        sham::DeviceBuffer<Tvec> &pos_buf,
        u32 cnt_obj,
        u32 morton_count,
        sham::DeviceBuffer<Tmorton> &&cache_buf_morton_codes);
 
    template<class Tmorton, class Tvec, u32 dim>
    MortonCodeSet<Tmorton, Tvec, dim> morton_code_set_from_positions(
        const sham::DeviceScheduler_ptr &dev_sched,
        shammath::AABB<Tvec> bounding_box,
        sham::DeviceBuffer<Tvec> &pos_buf,
        u32 cnt_obj,
        u32 morton_count);
 
} // namespace shamtree