invoke_node.h

Go to the documentation of this file.

#pragma once
 
#include "sdfg/data_flow/library_node.h"
 
#include "sdfg/codegen/dispatchers/block_dispatcher.h"
#include "sdfg/serializer/json_serializer.h"
 
namespace sdfg {
namespace data_flow {
 
inline LibraryNodeCode LibraryNodeType_Invoke("Invoke");
 
class InvokeNode : public LibraryNode {
protected:
    std::string callee_name_;
 
public:
    InvokeNode(
        size_t element_id,
        const DebugInfo& debug_info,
        const graph::Vertex vertex,
        DataFlowGraph& parent,
        const std::string& callee_name,
        const std::vector<std::string>& outputs,
        const std::vector<std::string>& inputs
    );
 
    const std::string& callee_name() const;
 
    bool is_void(const Function& sdfg) const;
 
    bool is_indirect_call(const Function& sdfg) const;
 
    void validate(const Function& function) const override;
 
    symbolic::SymbolSet symbols() const override;
 
    void replace(const symbolic::Expression old_expression, const symbolic::Expression new_expression) override;
 
    void replace(const symbolic::ExpressionMapping& replacements) override;
 
    std::unique_ptr<DataFlowNode> clone(size_t element_id, const graph::Vertex vertex, DataFlowGraph& parent) const override;
};
 
class InvokeNodeSerializer : public serializer::LibraryNodeSerializer {
public:
    nlohmann::json serialize(const LibraryNode& library_node) override;
 
    LibraryNode& deserialize(
        const nlohmann::json& j, builder::StructuredSDFGBuilder& builder, structured_control_flow::Block& parent
    ) override;
};
 
class InvokeNodeDispatcher : public codegen::LibraryNodeDispatcher {
public:
    InvokeNodeDispatcher(
        codegen::LanguageExtension& language_extension,
        const Function& function,
        const DataFlowGraph& data_flow_graph,
        const InvokeNode& node
    );
 
    void dispatch_code(
        codegen::PrettyPrinter& stream,
        codegen::PrettyPrinter& globals_stream,
        codegen::CodeSnippetFactory& library_snippet_factory
    ) override;
};
 
} // namespace data_flow
} // namespace sdfg