astbuilder.hpp

 
#pragma once
 
#include <memory>
 
#include "antlr4-runtime.h"
#include <TorqueBaseVisitor.h>
 
#include <torquescript/codeblock.hpp>
#include <torquescript/stringtable.hpp>
#include <torquescript/instructions.hpp>
#include <torquescript/ast.hpp>
#include <torquescript/instructionsequence.hpp>
 
namespace TorqueScript
{
    typedef std::vector<InstructionSequence> GeneratedInstructions;
 
    namespace AST
    {
        class ASTBuilder : public TorqueBaseVisitor
        {
            public:
                ASTBuilder(StringTable* stringTable);
 
                virtual antlrcpp::Any defaultResult() override;
 
                virtual antlrcpp::Any visitChildren(antlr4::tree::ParseTree *node) override;
 
                virtual antlrcpp::Any visitProgram(TorqueParser::ProgramContext* context) override;
                virtual antlrcpp::Any visitPackage_declaration(TorqueParser::Package_declarationContext* context) override;
                virtual antlrcpp::Any visitFunction_declaration(TorqueParser::Function_declarationContext* context) override;
                virtual antlrcpp::Any visitCall(TorqueParser::CallContext* context) override;
                virtual antlrcpp::Any visitSubcall(TorqueParser::SubcallContext* context) override;
                virtual antlrcpp::Any visitValue(TorqueParser::ValueContext* context) override;
                virtual antlrcpp::Any visitArithmetic(TorqueParser::ArithmeticContext* context) override;
                virtual antlrcpp::Any visitAssign(TorqueParser::AssignContext* context) override;
                virtual antlrcpp::Any visitUnary(TorqueParser::UnaryContext* context) override;
                virtual antlrcpp::Any visitWhile_control(TorqueParser::While_controlContext* context) override;
                virtual antlrcpp::Any visitIf_control(TorqueParser::If_controlContext* context) override;
                virtual antlrcpp::Any visitFor_control(TorqueParser::For_controlContext* context) override;
                virtual antlrcpp::Any visitLocalvariable(TorqueParser::LocalvariableContext* context) override;
                virtual antlrcpp::Any visitGlobalvariable(TorqueParser::GlobalvariableContext* context) override;
                virtual antlrcpp::Any visitIncrement(TorqueParser::IncrementContext* context) override;
                virtual antlrcpp::Any visitReturn_control(TorqueParser::Return_controlContext* context) override;
                virtual antlrcpp::Any visitEquality(TorqueParser::EqualityContext* context) override;
                virtual antlrcpp::Any visitConcat(TorqueParser::ConcatContext* context) override;
                virtual antlrcpp::Any visitTernary(TorqueParser::TernaryContext* context) override;
                virtual antlrcpp::Any visitSubfield(TorqueParser::SubfieldContext* context) override;
                virtual antlrcpp::Any visitArray(TorqueParser::ArrayContext* context) override;
                virtual antlrcpp::Any visitRelational(TorqueParser::RelationalContext* context) override;
                virtual antlrcpp::Any visitSwitch_control(TorqueParser::Switch_controlContext* context) override;
                virtual antlrcpp::Any visitSubarray(TorqueParser::SubarrayContext* context) override;
 
                virtual antlrcpp::Any visitField_assign(TorqueParser::Field_assignContext* context) override;
                virtual antlrcpp::Any visitDatablock_declaration(TorqueParser::Datablock_declarationContext* context) override;
                virtual antlrcpp::Any visitObject_declaration(TorqueParser::Object_declarationContext* context) override;
 
            private:
                StringTable* mStringTable;
        };
    }
}