set structure for IR

include/IR/IR.h

@@ -0,0 +1,3 @@
+#pragma once
+#include "IR_basic.h"
+#include "IRBuilder.h"

include/IR/IRBuilder.h

@@ -0,0 +1,54 @@
+#include <memory>
+#include "IR_basic.h"
+#include "ast/astnode_visitor.h"
+class IRBuilder : public ASTNodeVirturalVisitor {
+  std::shared_ptr<ModuleItem> prog;
+
+ public:
+  // Structural AST Nodes
+  void ActuralVisit(FuncDef_ASTNode *node) override;
+  void ActuralVisit(ClassDef_ASTNode *node) override;
+  void ActuralVisit(Program_ASTNode *node) override;
+
+  // Statement AST Nodes
+  void ActuralVisit(EmptyStatement_ASTNode *node) override;
+  void ActuralVisit(DefinitionStatement_ASTNode *node) override;
+  void ActuralVisit(ExprStatement_ASTNode *node) override;
+  void ActuralVisit(IfStatement_ASTNode *node) override;
+  void ActuralVisit(WhileStatement_ASTNode *node) override;
+  void ActuralVisit(ForStatement_ASTNode *node) override;
+  void ActuralVisit(JmpStatement_ASTNode *node) override;
+  void ActuralVisit(SuiteStatement_ASTNode *node) override;
+
+  // Expression AST Nodes
+  void ActuralVisit(NewArrayExpr_ASTNode *node) override;
+  void ActuralVisit(NewConstructExpr_ASTNode *node) override;
+  void ActuralVisit(NewExpr_ASTNode *node) override;
+  void ActuralVisit(AccessExpr_ASTNode *node) override;
+  void ActuralVisit(IndexExpr_ASTNode *node) override;
+  void ActuralVisit(SuffixExpr_ASTNode *node) override;
+  void ActuralVisit(PrefixExpr_ASTNode *node) override;
+  void ActuralVisit(OppositeExpr_ASTNode *node) override;
+  void ActuralVisit(LNotExpr_ASTNode *node) override;
+  void ActuralVisit(BNotExpr_ASTNode *node) override;
+  void ActuralVisit(MDMExpr_ASTNode *node) override;
+  void ActuralVisit(PMExpr_ASTNode *node) override;
+  void ActuralVisit(RLExpr_ASTNode *node) override;
+  void ActuralVisit(GGLLExpr_ASTNode *node) override;
+  void ActuralVisit(NEExpr_ASTNode *node) override;
+  void ActuralVisit(BAndExpr_ASTNode *node) override;
+  void ActuralVisit(BXorExpr_ASTNode *node) override;
+  void ActuralVisit(BOrExpr_ASTNode *node) override;
+  void ActuralVisit(LAndExpr_ASTNode *node) override;
+  void ActuralVisit(LOrExpr_ASTNode *node) override;
+  void ActuralVisit(TernaryExpr_ASTNode *node) override;
+  void ActuralVisit(AssignExpr_ASTNode *node) override;
+  void ActuralVisit(ThisExpr_ASTNode *node) override;
+  void ActuralVisit(ParenExpr_ASTNode *node) override;
+  void ActuralVisit(IDExpr_ASTNode *node) override;
+  void ActuralVisit(FunctionCallExpr_ASTNode *node) override;
+  void ActuralVisit(FormattedStringExpr_ASTNode *node) override;
+  void ActuralVisit(ConstantExpr_ASTNode *node) override;
+};
+
+std::shared_ptr<ModuleItem> BuildIR(std::shared_ptr<Program_ASTNode> src);

include/IR/IR_basic.h

@@ -0,0 +1,143 @@
+#pragma once
+#include <ios>
+#include <memory>
+#include <string>
+#include <variant>
+#include <vector>
+#include "ast/astnode.h"
+struct LLVMIRIntType {
+  size_t bits;
+};
+struct LLVMIRPTRType {};
+struct LLVMIRCLASSTYPE {};
+using LLVMType = std::variant<LLVMIRIntType, LLVMIRPTRType, LLVMIRCLASSTYPE>;
+class LLVMIRItemBase {
+ public:
+  LLVMIRItemBase() = default;
+  virtual ~LLVMIRItemBase() = default;
+  virtual void RecursivePrint(std::ostream &os) const;
+};
+
+class TypeDefItem : public LLVMIRItemBase {
+ public:
+  void RecursivePrint(std::ostream &os) const { ; }
+};
+class GlobalVarDefItem : public LLVMIRItemBase {
+  LLVMType type;
+  std::string name;
+
+ public:
+  void RecursivePrint(std::ostream &os) const { ; }
+};
+class ActionItem : public LLVMIRItemBase {
+ public:
+  void RecursivePrint(std::ostream &os) const { ; }
+};
+class JMPActionItem : public ActionItem {
+  std::string label;
+ public:
+  void RecursivePrint(std::ostream &os) const { ; }
+};
+class BRAction: public JMPActionItem {
+  std::string cond;
+  std::string true_label;
+  std::string false_label;
+ public:
+  void RecursivePrint(std::ostream &os) const { ; }
+};
+class UNConditionJMPAction: public JMPActionItem {
+  std::string label;
+ public:
+  void RecursivePrint(std::ostream &os) const { ; }
+};
+class RETAction : public JMPActionItem {
+  std::string value;
+ public:
+  void RecursivePrint(std::ostream &os) const { ; }
+};
+class BinaryOperationAction : public ActionItem {
+  std::string op;
+  std::string lhs;
+  std::string rhs;
+  std::string result;
+  LLVMType type;
+ public:
+  void RecursivePrint(std::ostream &os) const { ; }
+};
+class AllocaAction : public ActionItem {
+  std::string name;
+  LLVMType type;
+  size_t num;
+ public:
+  void RecursivePrint(std::ostream &os) const { ; }
+};
+class LoadAction : public ActionItem {
+  std::string result;
+  LLVMType ty;
+  std::string ptr;
+ public:
+  void RecursivePrint(std::ostream &os) const { ; }
+};
+class StoreAction : public ActionItem {
+  LLVMType ty;
+  std::string value;
+  std::string ptr;
+ public:
+  void RecursivePrint(std::ostream &os) const { ; }
+};
+class GetElementPtrAction : public ActionItem {
+  std::string result;
+  LLVMType ty;
+  std::string ptr;
+  std::vector<std::string> indices;
+ public:
+  void RecursivePrint(std::ostream &os) const { ; }
+};
+class ICMPAction : public ActionItem {
+  std::string op;
+  std::string lhs;
+  std::string rhs;
+  std::string result;
+  LLVMType ty;
+ public:
+  void RecursivePrint(std::ostream &os) const { ; }
+};
+class BlockItem : public LLVMIRItemBase {
+  std::string label;
+  std::vector<std::shared_ptr<ActionItem>> actions;
+  std::shared_ptr<JMPActionItem> exit_action;
+ public:
+  void RecursivePrint(std::ostream &os) const { ; }
+};
+class FunctionDefItem : public LLVMIRItemBase {
+  std::vector<std::shared_ptr<BlockItem>> basic_blocks;
+
+ public:
+  void RecursivePrint(std::ostream &os) const {
+    for (auto &item : basic_blocks) {
+      item->RecursivePrint(os);
+      os << '\n';
+    }
+  }
+};
+class ModuleItem : public LLVMIRItemBase {
+  std::vector<std::shared_ptr<TypeDefItem>> type_defs;
+  std::vector<std::shared_ptr<GlobalVarDefItem>> global_var_defs;
+  std::vector<std::shared_ptr<FunctionDefItem>> function_defs;
+
+ public:
+  void RecursivePrint(std::ostream &os) const {
+    for (auto &item : type_defs) {
+      item->RecursivePrint(os);
+      os << '\n';
+    }
+    for (auto &item : global_var_defs) {
+      item->RecursivePrint(os);
+      os << '\n';
+    }
+    for (auto &item : function_defs) {
+      item->RecursivePrint(os);
+      os << '\n';
+    }
+  }
+};