compiler.hpp

 
#pragma once
 
#include <memory>
 
#include "antlr4-runtime.h"
 
#include <torquescript/astvisitor.hpp>
#include <torquescript/codeblock.hpp>
#include <torquescript/stringtable.hpp>
#include <torquescript/instructions.hpp>
#include <torquescript/interpreterconfiguration.hpp>
#include <torquescript/instructionsequence.hpp>
 
namespace TorqueScript
{
    class Compiler : public AST::ASTVisitor
    {
        public:
            Compiler(const InterpreterConfiguration& config);
 
            CodeBlock* compileStream(std::istream& input, StringTable* stringTable);
 
            CodeBlock* compileString(const std::string& input, StringTable* stringTable);
 
            CodeBlock* compileFile(const std::string& path, StringTable* stringTable);
 
            virtual antlrcpp::Any defaultResult() override;
            virtual antlrcpp::Any aggregateResult(antlrcpp::Any& aggregate, antlrcpp::Any& nextResult) override;
 
            const InterpreterConfiguration mConfig;
 
        private:
            std::string mCurrentPackage;
            StringTable* mStringTable;
 
            /*
                Compiler Routines ==============================
            */
 
            virtual antlrcpp::Any visitPackageDeclarationNode(AST::PackageDeclarationNode* package) override;
            virtual antlrcpp::Any visitFunctionDeclarationNode(AST::FunctionDeclarationNode* function) override;
            virtual antlrcpp::Any visitFunctionCallNode(AST::FunctionCallNode* call) override;
            virtual antlrcpp::Any visitSubFunctionCallNode(AST::SubFunctionCallNode* call) override;
            virtual antlrcpp::Any visitSubFieldNode(AST::SubFieldNode* subfield) override;
            virtual antlrcpp::Any visitAddNode(AST::AddNode* expression) override;
            virtual antlrcpp::Any visitIntegerNode(AST::IntegerNode* value) override;
            virtual antlrcpp::Any visitFloatNode(AST::FloatNode* value) override;
            virtual antlrcpp::Any visitStringNode(AST::StringNode* value) override;
            virtual antlrcpp::Any visitTaggedStringNode(AST::TaggedStringNode* value) override;
            virtual antlrcpp::Any visitLocalVariableNode(AST::LocalVariableNode* value) override;
            virtual antlrcpp::Any visitGlobalVariableNode(AST::GlobalVariableNode* value) override;
            virtual antlrcpp::Any visitAssignmentNode(AST::AssignmentNode* expression) override;
            virtual antlrcpp::Any visitLessThanNode(AST::LessThanNode* expression) override;
            virtual antlrcpp::Any visitNegateNode(AST::NegateNode* expression) override;
            virtual antlrcpp::Any visitNotNode(AST::NotNode* expression) override;
            virtual antlrcpp::Any visitIncrementNode(AST::IncrementNode* expression) override;
            virtual antlrcpp::Any visitWhileNode(AST::WhileNode* node) override;
            virtual antlrcpp::Any visitForNode(AST::ForNode* node) override;
            virtual antlrcpp::Any visitBreakNode(AST::BreakNode* node) override;
            virtual antlrcpp::Any visitReturnNode(AST::ReturnNode* node) override;
            virtual antlrcpp::Any visitTernaryNode(AST::TernaryNode* node) override;
            virtual antlrcpp::Any visitSwitchNode(AST::SwitchNode* node) override;
            virtual antlrcpp::Any visitIfNode(AST::IfNode* node) override;
            virtual antlrcpp::Any visitArrayNode(AST::ArrayNode* array) override;
            virtual antlrcpp::Any visitEqualsNode(AST::EqualsNode* expression) override;
            virtual antlrcpp::Any visitConcatNode(AST::ConcatNode* expression) override;
            virtual antlrcpp::Any visitMultiplyNode(AST::MultiplyNode* expression) override;
            virtual antlrcpp::Any visitDivideNode(AST::DivideNode* expression) override;
            virtual antlrcpp::Any visitDatablockDeclarationNode(AST::DatablockDeclarationNode* datablock) override;
            virtual antlrcpp::Any visitObjectDeclarationNode(AST::ObjectDeclarationNode* object) override;
            virtual antlrcpp::Any visitFieldAssignNode(AST::FieldAssignNode* node) override;
    };
}