NodeMapEdge.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/stacktrace.hpp"
#include "shambase/string.hpp"
#include "shamsolvergraph/edge/IEdge.hpp"
#include "shamsolvergraph/node/INode.hpp"
#include <functional>
 
namespace shamrock::solvergraph {

    template<class Tin, class Tout>
    class NodeMapEdge : public INode {
 
        std::function<void(const Tin &, Tout &)> map_edge;
 
        public:
        NodeMapEdge(std::function<void(const Tin &, Tout &)> map_edge)
            : map_edge(std::move(map_edge)) {}

        inline void set_edges(std::shared_ptr<Tin> in_edge, std::shared_ptr<Tout> out_edge) {
            __internal_set_ro_edges({in_edge});
            __internal_set_rw_edges({out_edge});
        }

        inline void _impl_evaluate_internal() {
            __shamrock_stack_entry();
            map_edge(get_ro_edge<Tin>(0), get_rw_edge<Tout>(0));
        }

        inline virtual std::string _impl_get_label() const { return "MapEdge"; };

        inline virtual std::string _impl_get_tex() const {
 
            auto in_edge  = get_ro_edge_base(0).get_tex_symbol();
            auto out_edge = get_rw_edge_base(0).get_tex_symbol();
 
            std::string tex = R"tex(
                Map edge ${in_edge}$ to ${out_edge}$
            )tex";
 
            shambase::replace_all(tex, "{in_edge}", in_edge);
            shambase::replace_all(tex, "{out_edge}", out_edge);
 
            return tex;
        }
    };
 
} // namespace shamrock::solvergraph