LeafCellIterator.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/math.hpp"
#include "shamcomm/logs.hpp"
#include <utility>
 
namespace shamtree {
 
    struct LeafCellIteratorAccessed {
        const u32 *sort_index_map;  
        const u32 *reduc_index_map; 

        template<class Functor_iter>
        inline void for_each_in_leaf_cell(const u32 &cell_id, Functor_iter &&func_it) const {
            // loop on particle indexes
            uint min_ids = reduc_index_map[cell_id];
            uint max_ids = reduc_index_map[cell_id + 1];
 
            for (unsigned int id_s = min_ids; id_s < max_ids; id_s++) {
 
                // recover old index before morton sort
                uint id_b = sort_index_map[id_s];
 
                // iteration function
                func_it(id_b);
            }
        }
    };

    struct LeafCellIterator {
        const sham::DeviceBuffer<u32> &buf_sort_index_map;  
        const sham::DeviceBuffer<u32> &buf_reduc_index_map; 
 
        using acc = LeafCellIteratorAccessed;

        inline acc get_read_access(sham::EventList &deps) const {
            return acc{
                .sort_index_map  = buf_sort_index_map.get_read_access(deps),
                .reduc_index_map = buf_reduc_index_map.get_read_access(deps)};
        }

        inline void complete_event_state(sycl::event e) const {
            buf_sort_index_map.complete_event_state(e);
            buf_reduc_index_map.complete_event_state(e);
        }
    };

    struct LeafCellIteratorHost {
        std::vector<u32> sort_index_map;  
        std::vector<u32> reduc_index_map; 
 
        using acc = LeafCellIteratorAccessed;

        inline acc get_read_access() const {
            return acc{
                .sort_index_map = sort_index_map.data(), .reduc_index_map = reduc_index_map.data()};
        }
    };
 
} // namespace shamtree