TreeNode.h

Go to the documentation of this file.

#ifndef UTILS_TREE_NODE_H
#define UTILS_TREE_NODE_H
 
#include "tdb.h"
 
#include "utils/misc.h"
#include "utils/tree/NodeTag.h"
 
namespace taco {
 
class TreeNode {
public:
 
    template<class ...UniquePtrs>
    TreeNode(NodeTag tag, UniquePtrs&& ...input):
        m_tag(tag),
        m_input() {
        m_input.reserve(sizeof...(UniquePtrs));
        emplace_back_parameter_pack(m_input,
                                    std::forward<UniquePtrs>(input)...);
    }
 
    virtual ~TreeNode() {}
 
    TreeNode(TreeNode&&) = default;
    TreeNode &operator=(TreeNode&&) = default;
    TreeNode(const TreeNode&) = delete;
    TreeNode &operator=(const TreeNode &) = delete;
 
    constexpr NodeTag
    get_tag() const {
        return m_tag;
    }
 
    TreeNode *
    get_input(size_t i) const {
        if (i < m_input.size())
            return m_input[i].get();
        return nullptr;
    }
 
    template<class Node>
    Node *
    get_input_as(size_t i) const {
        if (i < m_input.size())
            return static_cast<Node*>(m_input[i].get());
        return nullptr;
    }
 
    std::string
    to_string() const {
        std::string buf;
        node_to_string(buf, 0);
        return buf;
    }
 
    void node_to_string(std::string &buf, int indent) const;
 
    const char *node_name() const {
        return node_tag_name(get_tag());
    }
 
    virtual void node_properties_to_string(std::string &buf,
                                           int indent) const = 0;
 
    static void append_indent(std::string &buf, int indent);
private:
    NodeTag m_tag;
 
    std::vector<std::unique_ptr<TreeNode>> m_input;
 
    static constexpr const int TO_STRING_INDENT_SIZE = 4;
};
 
}   // namespace taco
 
#endif  // UTILS_TREE_NODE_H