ASTNode.h

Go to the documentation of this file.

 
#pragma once
 
#include <cstddef>
#include <memory>
#include <vector>
#include <string>
#include <iostream>
 
#include <AST/NodeTypes.h>
#include <AST/Position.h>
#include <AST/Token.h>
 
#define PRETTYPRINT_INDENTATION_AMOUNT 4
 
namespace AST {
 
class ASTNode {
    protected:
        std::vector<std::shared_ptr<ASTNode>> children;
        AST::FilePosition position;
        AST::FilePosition end_position;
 
    public:
        ASTNode() = default;
        
        virtual constexpr AST::NodeType getType() const = 0;
 
        void addChild(const std::shared_ptr<ASTNode>& child);
        void addChildren(const std::vector<std::shared_ptr<ASTNode>>& childs);
        const std::vector<std::shared_ptr<ASTNode>>& getChildren() const;
        void setPosition(const AST::FilePosition& pos);
        void setPosition(uint32_t line, uint32_t column);
        const AST::FilePosition& getPosition() const;
        void setEndPosition(const AST::FilePosition& pos);
        void setEndPosition(uint32_t line, uint32_t column);
        const AST::FilePosition& getEndPosition() const;
 
        uint32_t getLine() const;
        uint32_t getCharPositionInLine() const;
        
        std::shared_ptr<ASTNode> getChildAt(size_t index) const;
        std::shared_ptr<ASTNode> getFirstChild() const;
        std::shared_ptr<ASTNode> getLastChild() const;
        size_t getChildrenCount() const;
 
        void clear();
        void clearChildren();
 
        virtual std::ostream& prettyPrint(std::ostream& os, int indentation_level = 0) const = 0;
        friend std::ostream& operator<<(std::ostream& os, const ASTNode& node) {
            return node.prettyPrint(os, 0);
        }
};
 
} // namespace AST