feat(Gen): 现在可以使用全局变量了

wangziwenhk · Jul 29, 2024 · 0d445c9 · 0d445c9
1 parent 3aa160b
commit 0d445c9
Show file tree

Hide file tree

Showing 8 changed files with 66 additions and 32 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -36,6 +36,7 @@ file(GLOB AST_FILES "parser/*.cpp" "parser/*.h")
 file(GLOB SOURCE_FILES
         "src/*"
         "src/Tools/*"
+        "src/Tools/Managers/*"
         "src/Types/*"
         "src/Visitors/*")
 

diff --git a/src/Tools/Managers/ClassManager.cpp b/src/Tools/Managers/ClassManager.cpp
@@ -0,0 +1,8 @@
+//
+// Created by wangz on 24-7-29.
+//
+
+#include "ClassManager.h"
+
+namespace Riddle {
+}// namespace Riddle
diff --git a/src/Tools/Managers/ClassManager.h b/src/Tools/Managers/ClassManager.h
@@ -0,0 +1,18 @@
+//
+// Created by wangz on 24-7-29.
+//
+
+#ifndef RIDDLE_LANGUAGE_CLASSMANAGER_H
+#define RIDDLE_LANGUAGE_CLASSMANAGER_H
+#include <llvm/IR/IRBuilder.h>
+#include <llvm/IR/Value.h>
+namespace Riddle {
+    /// @brief 管理类
+    class ClassManager {
+        /// @brief 存储数据字段
+        llvm::StructType *data;
+    };
+
+}// namespace Riddle
+
+#endif//RIDDLE_LANGUAGE_CLASSMANAGER_H
diff --git a/src/Tools/VarManager.cpp → src/Tools/Managers/VarManager.cpp b/src/Tools/VarManager.cpp → src/Tools/Managers/VarManager.cpp
@@ -22,20 +22,21 @@ namespace Riddle {
         }
         Defined.pop();
     }
-    void VarManager::defineVar(const std::string &name, const bool &isConst, llvm::AllocaInst *value, const std::string &type) {
+    void VarManager::defineVar(const std::string &name, const bool &isConst, llvm::Value *value) {
         if(Defined.top().count(name)) {
             throw std::logic_error("The variable has been defined multiple times");
         }
-        NamedVar[name].push(Variable(name, value, type, isConst));
+        NamedVar[name].push(Variable(name, value, isConst));
         Defined.top()[name] = true;
     }
     Variable VarManager::getVar(const std::string &name) {
         if(!isDefined(name))
             throw std::logic_error("The variable does not exist");
-        if(NamedVar[name].top().type == Null)
-            throw std::logic_error("Unclear variables");
 
         return NamedVar[name].top();
     }
+    bool VarManager::isGlobal() {
+        return Defined.size() == 1;
+    }
 
 }// namespace Riddle
diff --git a/src/Tools/VarManager.h → src/Tools/Managers/VarManager.h b/src/Tools/VarManager.h → src/Tools/Managers/VarManager.h
@@ -25,11 +25,14 @@ namespace Riddle {
         /// @param type 类型，当类型为 null 时说明在下一次赋值时判断，在被引用之前必须有类型
         /// @param isConst 是否不变
         /// @param value 指某个局部变量的地址
-        void defineVar(const std::string &name, const bool &isConst, llvm::AllocaInst *value = nullptr, const std::string &type = "null");
+        void defineVar(const std::string &name, const bool &isConst, llvm::Value *value = nullptr);
         /// @brief 获取一个变量的属性
         /// @param name 变量名
         /// @returns 变量的属性
         Variable getVar(const std::string &name);
+        /// @brief 获取当前是否为全局
+        /// @returns 是否为全局
+        bool isGlobal();
     };
 
 }// namespace Riddle

diff --git a/src/Types/Variable.h b/src/Types/Variable.h
@@ -13,14 +13,12 @@ namespace Riddle {
     class Variable {
     public:
         const std::string name;
-        std::string type;
-        llvm::AllocaInst *value;
+        llvm::Value *value;
         const bool isConst;
         Variable() = delete;
         Variable(const std::string &name,
-                 llvm::AllocaInst *value,
-                 const std::string &type = Null,
-                 const bool isConst = false): name(name), value(value), type(type), isConst(isConst){};
+                 llvm::Value *value,
+                 const bool isConst = false): name(name), value(value), isConst(isConst){};
     };
 
 }// namespace Riddle

diff --git a/src/Visitors/GenVisitor.cpp b/src/Visitors/GenVisitor.cpp
@@ -42,9 +42,8 @@ namespace Riddle {
         return p;
     }
     std::any GenVisitor::visitObjectExpr(RiddleParser::ObjectExprContext *ctx) {
-        llvm::AllocaInst *value = varManager.getVar(ctx->id()->getText()).value;
-        llvm::Value *rv = value;
-        return rv;
+        llvm::Value *value = varManager.getVar(ctx->id()->getText()).value;
+        return value;
     }
     std::any GenVisitor::visitProgram(RiddleParser::ProgramContext *ctx) {
         varManager.push();
@@ -96,7 +95,7 @@ namespace Riddle {
         for(int i = 0; i < args.names.size(); i++, argIter++) {
             llvm::AllocaInst *Alloca = Builder.CreateAlloca(args.types[i], nullptr, args.names[i]);
             Builder.CreateStore(argIter, Alloca);
-            varManager.defineVar(args.names[i], false, Alloca, getTypeName(args.types[i]));
+            varManager.defineVar(args.names[i], false, Alloca);
         }
 
         visit(ctx->funcBody());
@@ -114,29 +113,35 @@ namespace Riddle {
     }
     std::any GenVisitor::visitVarDefineStatement(RiddleParser::VarDefineStatementContext *ctx) {
         std::string name = ctx->name->getText();
-        if(ctx->type == nullptr) {
-            auto value = any_cast<llvm::Value *>(visit(ctx->value));
-            llvm::AllocaInst *Alloca = Builder.CreateAlloca(value->getType(), nullptr, name);
-            assignBinaryOp(Alloca, value, "=");
-            varManager.defineVar(name, false, Alloca, getTypeName(value->getType()));
-        } else if(ctx->value == nullptr) {
+        llvm::Value *value = nullptr;
+        llvm::Type *type = nullptr;
+        llvm::Value *size = nullptr;
+
+        if(ctx->type != nullptr) {
             auto tuple = any_cast<std::tuple<llvm::Type *, llvm::Value *>>(visit(ctx->type));
-            auto [type, size] = tuple;
-            llvm::AllocaInst *Alloca = Builder.CreateAlloca(type, size, name);
-            varManager.defineVar(name, false, Alloca, getTypeName(type));
+            auto [_type, _size] = tuple;
+            type = _type;
+            size = _size;
+        } else if(ctx->value != nullptr) {
+            value = any_cast<llvm::Value *>(visit(ctx->value));
+            type = value->getType();
+        }
+
+        llvm::Value *var;
+        if(varManager.isGlobal()) {
+            auto CV = llvm::dyn_cast<llvm::Constant>(value);
+            var = new llvm::GlobalVariable(*module, type, false, llvm::GlobalVariable::LinkageTypes::ExternalLinkage, CV, name);
         } else {
-            auto tuple = any_cast<std::tuple<llvm::Type *, llvm::Value *>>(visit(ctx->type));
-            auto [type, size] = tuple;
-            auto value = any_cast<llvm::Value *>(visit(ctx->value));
-            llvm::AllocaInst *Alloca = Builder.CreateAlloca(type, size, name);
-            assignBinaryOp(Alloca, value, "=");
-            varManager.defineVar(name, false, Alloca, getTypeName(type));
+            var = Builder.CreateAlloca(type, size, name);
+            if(ctx->value != nullptr)
+                assignBinaryOp(var, value, "=");
         }
+        varManager.defineVar(name, false, var);
         return nullptr;
     }
     std::any GenVisitor::visitObjValExpr(RiddleParser::ObjValExprContext *ctx) {
-        llvm::AllocaInst *var = varManager.getVar(ctx->id()->getText()).value;
-        llvm::Value *LoadedValue = Builder.CreateLoad(var->getAllocatedType(), var, "tempVar");
+        llvm::Value *var = varManager.getVar(ctx->id()->getText()).value;
+        llvm::Value *LoadedValue = Builder.CreateLoad(var->getType(), var, "tempVar");
         return LoadedValue;
     }
     std::any GenVisitor::visitIfStatement(RiddleParser::IfStatementContext *ctx) {

diff --git a/src/Visitors/GenVisitor.h b/src/Visitors/GenVisitor.h
@@ -1,8 +1,8 @@
 #ifndef RIDDLE_LANGUAGE_GENVISITOR_H
 #define RIDDLE_LANGUAGE_GENVISITOR_H
 
+#include "Tools/Managers/VarManager.h"
 #include "Tools/Setup.h"
-#include "Tools/VarManager.h"
 #include <RiddleParserBaseVisitor.h>
 #include <llvm/IR/IRBuilder.h>
 #include <llvm/IR/LLVMContext.h>