MortonReducedSet.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/CellIterator.hpp"
#include "shamtree/LeafCellIterator.hpp"
#include "shamtree/MortonCodeSortedSet.hpp"
 
namespace shamtree {

    template<class Tmorton, class Tvec, u32 dim = shambase::VectorProperties<Tvec>::dimension>
    class MortonReducedSet {
        public:
        MortonCodeSortedSet<Tmorton, Tvec> morton_codes_set;

        u32 reduce_code_count;

        sham::DeviceBuffer<u32> buf_reduc_index_map;

        sham::DeviceBuffer<Tmorton> reduced_morton_codes;

        MortonReducedSet(
            MortonCodeSortedSet<Tmorton, Tvec> &&morton_codes_set,
            u32 reduce_code_count,
            sham::DeviceBuffer<u32> &&buf_reduc_index_map,
            sham::DeviceBuffer<Tmorton> &&reduced_morton_codes)
            : morton_codes_set(std::move(morton_codes_set)), reduce_code_count(reduce_code_count),
              buf_reduc_index_map(std::move(buf_reduc_index_map)),
              reduced_morton_codes(std::move(reduced_morton_codes)) {}
 
        inline bool is_empty() const { return reduce_code_count == 0; }
 
        inline LeafCellIterator get_leaf_cell_iterator() const {
            return LeafCellIterator{morton_codes_set.map_morton_id_to_obj_id, buf_reduc_index_map};
        }
 
        inline LeafCellIteratorHost get_leaf_cell_iterator_host() const {
            return LeafCellIteratorHost{
                morton_codes_set.map_morton_id_to_obj_id.copy_to_stdvec(),
                buf_reduc_index_map.copy_to_stdvec()};
        }
 
        inline CellIterator get_cell_iterator(
            const sham::DeviceBuffer<u32> &buf_endrange, u32 offset_leaf) const {
            return CellIterator{
                morton_codes_set.map_morton_id_to_obj_id,
                buf_reduc_index_map,
                buf_endrange,
                offset_leaf};
        }
 
        inline CellIteratorHost get_cell_iterator_host(
            const sham::DeviceBuffer<u32> &buf_endrange, u32 offset_leaf) const {
            return CellIteratorHost{
                morton_codes_set.map_morton_id_to_obj_id.copy_to_stdvec(),
                buf_reduc_index_map.copy_to_stdvec(),
                buf_endrange.copy_to_stdvec(),
                offset_leaf};
        }
 
        inline static MortonReducedSet make_empty(sham::DeviceScheduler_ptr dev_sched) {
            return MortonReducedSet{
                MortonCodeSortedSet<Tmorton, Tvec>::make_empty(dev_sched),
                0,
                sham::DeviceBuffer<u32>(0, dev_sched),
                sham::DeviceBuffer<Tmorton>(0, dev_sched)};
        }
    };

    template<class Tmorton, class Tvec, u32 dim>
    MortonReducedSet<Tmorton, Tvec, dim> reduce_morton_set(
        const sham::DeviceScheduler_ptr &dev_sched,
        MortonCodeSortedSet<Tmorton, Tvec, dim> &&morton_codes_set,
        u32 reduction_level);

    template<class Tmorton, class Tvec, u32 dim>
    MortonReducedSet<Tmorton, Tvec, dim> reduce_morton_set(
        const sham::DeviceScheduler_ptr &dev_sched,
        MortonCodeSortedSet<Tmorton, Tvec, dim> &&morton_codes_set,
        u32 reduction_level,
        sham::DeviceBuffer<u32> &&cache_buf_reduc_index_map,
        sham::DeviceBuffer<Tmorton> &&cache_reduced_morton_codes);
 
} // namespace shamtree