NodeSetEdge.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 "shamrock/solvergraph/IEdge.hpp"
#include "shamrock/solvergraph/INode.hpp"
#include <functional>
 
namespace shamrock::solvergraph {
 
    template<class Tedge>
    class NodeSetEdge : public INode {
 
        std::function<void(Tedge &)> set_edge;
 
        public:
        NodeSetEdge(std::function<void(Tedge &)> set_edge) : set_edge(std::move(set_edge)) {}
 
        inline void set_edges(std::shared_ptr<IEdge> to_set) {
            __internal_set_ro_edges({});
            __internal_set_rw_edges({to_set});
        }
 
        inline void _impl_evaluate_internal() {
            __shamrock_stack_entry();
            set_edge(get_rw_edge<Tedge>(0));
        }
 
        inline virtual std::string _impl_get_label() const { return "SetEdge"; };
 
        inline virtual std::string _impl_get_tex() const {
 
            auto to_set = get_rw_edge_base(0).get_tex_symbol();
 
            std::string tex = R"tex(
                Set edge ${to_set}$
            )tex";
 
            shambase::replace_all(tex, "{to_set}", to_set);
 
            return tex;
        }
    };
 
} // namespace shamrock::solvergraph