RankGetter.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 "shamrock/solvergraph/IEdgeNamed.hpp"
#include <functional>
#include <utility>
 
namespace shamrock::solvergraph {
 
    class RankGetter : public IEdgeNamed {
 
        std::function<u32(u64)> rank_getter_func;
 
        public:
        RankGetter(std::function<u32(u64)> getter_func, std::string label, std::string tex_symbol)
            : rank_getter_func(std::move(getter_func)),
              IEdgeNamed(std::move(label), std::move(tex_symbol)) {}
 
        inline u32 get_rank_owner(u64 patch_id) const { return rank_getter_func(patch_id); }
 
        inline void free_alloc() {};
    };
 
} // namespace shamrock::solvergraph