MortonCodeSortedSet.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 "shamtree/MortonCodeSet.hpp"
 
namespace shamtree {
 
    template<class Tmorton, class Tvec, u32 dim = shambase::VectorProperties<Tvec>::dimension>
    class MortonCodeSortedSet {
        public:
        shammath::AABB<Tvec> bounding_box;
 
        u32 cnt_obj;
 
        u32 morton_count;
 
        sham::DeviceBuffer<Tmorton> sorted_morton_codes;
 
        sham::DeviceBuffer<u32> map_morton_id_to_obj_id;
 
        MortonCodeSortedSet(
            shammath::AABB<Tvec> &&bounding_box,
            u32 &&cnt_obj,
            u32 &&morton_count,
            sham::DeviceBuffer<Tmorton> &&sorted_morton_codes,
            sham::DeviceBuffer<u32> &&map_morton_id_to_obj_id)
            : bounding_box(std::move(bounding_box)), cnt_obj(std::move(cnt_obj)),
              morton_count(std::move(morton_count)),
              sorted_morton_codes(std::move(sorted_morton_codes)),
              map_morton_id_to_obj_id(std::move(map_morton_id_to_obj_id)) {}
 
        inline static MortonCodeSortedSet make_empty(sham::DeviceScheduler_ptr dev_sched) {
            return MortonCodeSortedSet(
                shammath::AABB<Tvec>(),
                0_u32,
                0_u32,
                sham::DeviceBuffer<Tmorton>(0, dev_sched),
                sham::DeviceBuffer<u32>(0, dev_sched));
        }
    };
 
    template<class Tmorton, class Tvec, u32 dim>
    MortonCodeSortedSet<Tmorton, Tvec, dim> sort_morton_set(
        const sham::DeviceScheduler_ptr &dev_sched,
        MortonCodeSet<Tmorton, Tvec, dim> &&morton_codes_set);
 
    template<class Tmorton, class Tvec, u32 dim>
    MortonCodeSortedSet<Tmorton, Tvec, dim> sort_morton_set(
        const sham::DeviceScheduler_ptr &dev_sched,
        MortonCodeSet<Tmorton, Tvec, dim> &&morton_codes_set,
        sham::DeviceBuffer<u32> &&cached_map_morton_id_to_obj_id);
 
} // namespace shamtree