Array support

.devcontainer/devcontainer.json

@@ -1,22 +1,21 @@
 {
   "name": "SysY Compiler Dev",
   "image": "maxxing/compiler-dev",
-
   "workspaceMount": "source=${localWorkspaceFolder},target=${localWorkspaceFolder},type=bind",
   "workspaceFolder": "${localWorkspaceFolder}",
-
   "mounts": [
     "source=${localEnv:HOME}/.ssh,target=/home/ubuntu/.ssh,type=bind,consistency=cached"
   ],
-
   "customizations": {
     "vscode": {
       "extensions": [
         "llvm-vs-code-extensions.vscode-clangd",
         "ms-vscode.cmake-tools",
         "twxs.cmake",
         "daohong-emilio.yash",
-        "mikawi.flex-language"
+        "mikawi.flex-language",
+        "aaron-bond.better-comments",
+        "eamodio.gitlens"
       ],
       "settings": {
         "clangd.path": "/usr/bin/clangd-21",
@@ -35,9 +34,7 @@
       }
     }
   },
-
   "remoteUser": "ubuntu",
-
   "features": {
     "ghcr.io/devcontainers/features/common-utils:2": {
       "username": "ubuntu",
@@ -46,13 +43,12 @@
       "configureZshAsDefaultShell": true
     }
   },
-
-  "postCreateCommand": "sudo apt-get update && sudo apt-get install -y clangd-21",
-
+  // "postCreateCommand": "sudo apt-get update && sudo apt-get install -y clangd-21",
+  "postCreateCommand": "sudo apt-get update && sudo apt-get install -y clangd-21 python3-pip && sudo pip3 install ninja --break-system-packages",
   "runArgs": [
     "--network=host",
     "--cap-add=SYS_PTRACE",
     "--security-opt",
     "seccomp=unconfined"
   ]
-}
+}

.github/workflows/ci.yml

@@ -1,35 +1,60 @@
 name: compiler auto test
 run-name: ${{ github.actor }} is testing out compiler function 🚀
+
 on:
   push:
+    paths:
+      - 'project/**'
   pull_request:
   workflow_dispatch:
-  
+
 jobs:
-  build:
+  build_compiler:
     runs-on: ubuntu-latest
     container:
       image: maxxing/compiler-dev
     defaults:
       run:
         working-directory: ./project
+    outputs:
+      artifact_name: compiler-binary
     steps:
-      - name: Check out repository code
-        uses: actions/checkout@v5
-      - name: test koopa
+      - uses: actions/checkout@v5
+
+      - name: Install Ninja
         run: |
-          autotest -w debug -koopa -s lv1 .
-          autotest -w debug -koopa -s lv3 .
-          autotest -w debug -koopa -s lv4 .
-          autotest -w debug -koopa -s lv5 .
-          autotest -w debug -koopa -s lv6 .
-          autotest -w debug -koopa -s lv7 .
-          autotest -w debug -koopa -s lv8 .
-      - name: test riscv
+          apt-get update && apt-get install -y ninja-build
+          ninja --version
+
+      - name: Build Binary
+        run: make
+
+  test_compiler:
+    needs: build_compiler
+    runs-on: ubuntu-latest
+    container:
+      image: maxxing/compiler-dev
+    strategy:
+      fail-fast: false
+      matrix:
+        test_type: [koopa, riscv]
+    defaults:
+      run:
+        working-directory: ./project
+    steps:
+      - uses: actions/checkout@v5
+
+      - name: Install Ninja
         run: |
-          autotest -w debug -riscv -s lv1 .
-          autotest -w debug -riscv -s lv3 .
-          autotest -w debug -riscv -s lv4 .
-          autotest -w debug -riscv -s lv5 .
-          autotest -w debug -riscv -s lv6 .
-          autotest -w debug -riscv -s lv7 .
+          apt-get update && apt-get install -y ninja-build
+          ninja --version
+
+      - name: Run Test
+        run: autotest -w debug -${{ matrix.test_type }} .
+
+      - name: Upload Test Logs
+        if: failure()
+        uses: actions/upload-artifact@v5
+        with:
+          name: test-logs-${{ matrix.test_type }}
+          path: project/debug/

project/.clang-format

@@ -1 +1,2 @@
-AllowShortCaseLabelsOnASingleLine: true
+AllowShortCaseLabelsOnASingleLine: true
+AllowShortIfStatementsOnASingleLine: AllIfsAndElse

project/Makefile

@@ -8,7 +8,7 @@ PWD := $(shell pwd)
 all: build
 
 config:
-	cmake -S . -B $(BUILD_DIR) -G Ninja -DCMAKE_MAKE_PROGRAM=$(PWD)/ninja
+	cmake -S . -B $(BUILD_DIR) -G Ninja -DCMAKE_CXX_COMPILER=clang++
 
 build: config
 	cmake --build $(BUILD_DIR) -j12
@@ -28,7 +28,8 @@ run:
 	./cmake-build/compiler -riscv ./tests/hello.c -o ./debug/hello.asm
 
 test:
-	cd cmake-build && ctest && cd ..
+	python3 scripts/test_runner.py koopa
+	python3 scripts/test_runner.py riscv
 
 docker-build:
 	docker run --rm \

project/include/backend/backend.cppm

@@ -81,14 +81,14 @@ private:
    */
   auto visit(const koopa_raw_return_t &ret) -> void;
   auto visit(const koopa_raw_binary_t &binary) -> void;
-  auto visit(const koopa_raw_integer_t &integer) -> void;
   auto visit(const koopa_raw_jump_t &jump) -> void;
   auto visit(const koopa_raw_branch_t &branch) -> void;
   auto visit(const koopa_raw_load_t &load) -> void;
   auto visit(const koopa_raw_store_t &store) -> void;
   auto visit(const koopa_raw_call_t &call) -> void;
-  auto visit(const koopa_raw_func_arg_ref_t &func_arg_ref) -> void;
   auto visit(const koopa_raw_global_alloc_t &global_alloc) -> void;
+  auto visit(const koopa_raw_get_elem_ptr_t &get_elem_ptr) -> void;
+  auto visit(const koopa_raw_get_ptr_t &get_ptr) -> void;
   /** @} */
 };
 } // namespace backend

project/include/ir/ast.cppm

@@ -15,6 +15,7 @@ export module ir.ast;
 
 import symbol_table;
 import ir_builder;
+import ir.type;
 
 /**
  * @brief Namespace containing all AST related classes and functions.
@@ -59,6 +60,7 @@ public:
 
 class LValAST;
 class BlockAST;
+class InitValStmtAST;
 
 //* enum class
 // clang-format off
@@ -98,23 +100,32 @@ public:
   std::string btype;
   std::string ident;
   bool is_const;
+  bool is_ptr;
+  std::vector<std::unique_ptr<ExprAST>> indices;
   /**
    * @brief Constructs a function parameter.
    * @param _btype The type of the parameter (e.g., "int").
    * @param _ident The name of the parameter.
    * @param _is_const Whether the parameter is constant.
    */
-  FuncParamAST(std::string _btype, std::string _ident, bool _is_const)
-      : btype(std::move(_btype)), ident(std::move(_ident)),
-        is_const(_is_const) {}
+  FuncParamAST(std::string _btype, std::string _ident, bool _is_const,
+               bool _is_ptr, std::vector<std::unique_ptr<ExprAST>> _indices)
+      : btype(std::move(_btype)), ident(std::move(_ident)), is_const(_is_const),
+        is_ptr(_is_ptr), indices(std::move(_indices)) {}
   auto dump(int depth) const -> void override;
   auto codeGen(ir::KoopaBuilder &builder) const -> std::string override;
+  auto toKoopa(ir::KoopaBuilder &builder) const -> std::string;
 };
 
+/**
+ * @brief Base class for definition AST nodes (variables, functions, arrays).
+ */
+class DefAST : public BaseAST {};
+
 /**
  * @brief AST node for function definitions.
  */
-class FuncDefAST : public BaseAST {
+class FuncDefAST : public DefAST {
 public:
   ~FuncDefAST() override;
   std::string btype;
@@ -135,10 +146,29 @@ public:
   auto codeGen(ir::KoopaBuilder &builder) const -> std::string override;
 };
 
+/**
+ * @brief AST node for array definitions.
+ *
+ * Handles both constant and non-constant array definitions, including their sizes and initializers.
+ */
+class ArrayDefAST : public DefAST {
+public:
+  ~ArrayDefAST() override;
+  bool is_const;
+  std::string ident;
+  std::vector<std::unique_ptr<ExprAST>> array_suffix;
+  std::unique_ptr<InitValStmtAST> init_val;
+  ArrayDefAST(bool _is_const, std::string _ident,
+              std::vector<std::unique_ptr<ExprAST>> _array_suffix,
+              InitValStmtAST *_init_val);
+  auto dump(int depth) const -> void override;
+  auto codeGen(ir::KoopaBuilder &builder) const -> std::string override;
+};
+
 /**
  * @brief AST node for a single variable definition.
  */
-class DefAST : public BaseAST {
+class ScalarDefAST : public DefAST {
 public:
   bool is_const;
   std::string ident;
@@ -149,7 +179,7 @@ public:
    * @param _ident The name of the variable.
    * @param _initVal The initial value expression (optional).
    */
-  DefAST(bool _is_const, std::string _ident, BaseAST *_initVal)
+  ScalarDefAST(bool _is_const, std::string _ident, BaseAST *_initVal)
       : is_const(_is_const), ident(std::move(_ident)) {
     if (_initVal) {
       initVal.reset(static_cast<ExprAST *>(_initVal));
@@ -206,6 +236,38 @@ public:
   auto dump(int depth) const -> void override;
   auto codeGen(ir::KoopaBuilder &builder) const -> std::string override;
 };
+
+/**
+ * @brief Represents an initialization value statement in the abstract syntax
+ * tree.
+ *
+ * This class is used to model initialization values, which can be either a
+ * single expression or a list of nested initializers (e.g., for arrays or
+ * aggregate types).
+ */
+class InitValStmtAST : public StmtAST {
+public:
+  std::unique_ptr<ExprAST> expr;
+  std::vector<std::unique_ptr<InitValStmtAST>> initialize_list;
+  /**
+   * @brief Constructs an InitValStmtAST object.
+   *
+   * @param _expr Pointer to a BaseAST that will be statically cast to ExprAST.
+   * It represents the primary initialization expression.
+   * @param _initialize_list Vector of unique pointers to InitValStmtAST for
+   * nested initialization. This list is moved into the member to avoid copying.
+   */
+  InitValStmtAST(BaseAST *_expr,
+                 std::vector<std::unique_ptr<InitValStmtAST>> _initialize_list)
+      : initialize_list(std::move(_initialize_list)) {
+    expr.reset(static_cast<ExprAST *>(_expr));
+  }
+
+  auto dump(int depth) const -> void override;
+  auto codeGen(ir::KoopaBuilder &builder) const -> std::string override;
+  auto flatten(std::shared_ptr<type::Type>, ir::KoopaBuilder &) const
+      -> std::vector<std::string>;
+};
 /** @} */
 
 /** @name Data Action
@@ -242,9 +304,9 @@ public:
    * @param _btype The type of the variables.
    * @param _defs The list of variable definitions.
    */
-  DeclAST(bool _isConst, std::string _btype,
+  DeclAST(bool _is_const, std::string _btype,
           std::vector<std::unique_ptr<DefAST>> _defs)
-      : is_const(_isConst), btype(std::move(_btype)), defs(std::move(_defs)) {}
+      : is_const(_is_const), btype(std::move(_btype)), defs(std::move(_defs)) {}
   auto dump(int depth) const -> void override;
   auto codeGen(ir::KoopaBuilder &builder) const -> std::string override;
 };
@@ -345,11 +407,13 @@ public:
 class LValAST : public ExprAST {
 public:
   std::string ident;
+  std::vector<std::unique_ptr<ExprAST>> indices;
   /**
    * @brief Constructs an LVal node.
    * @param _ident The variable name.
    */
-  LValAST(std::string _ident) : ident(std::move(_ident)) {};
+  LValAST(std::string _ident, std::vector<std::unique_ptr<ExprAST>> _indices)
+      : ident(std::move(_ident)), indices(std::move(_indices)) {};
   auto dump(int depth) const -> void override;
   auto codeGen(ir::KoopaBuilder &builder) const -> std::string override;
   auto CalcValue(ir::KoopaBuilder &builder) const -> int override;

project/include/ir/ir_builder.cppm

@@ -151,8 +151,6 @@ public:
    */
   auto resetCount() -> void {
     count_reg = 0;
-    count_name = 0;
-    count_label = 0;
     _is_block_closed = false;
   }
 

project/include/ir/symbol_table.cppm

@@ -33,7 +33,7 @@ struct Symbol {
   std::string irName;               ///< IR-level name (e.g., "@x_1")
   std::shared_ptr<type::Type> type; ///< Data type of the symbol
   SymbolKind kind;                  ///< Variable or Function
-  bool is_const;                     ///< True if it's a compile-time constant
+  bool is_const;                    ///< True if it's a compile-time constant
   int constValue; ///< The value of the constant, if applicable
 };
 
@@ -94,8 +94,8 @@ public:
    * @param val     Initial value if constant.
    */
   auto define(const std::string &name, const std::string &irName,
-              std::shared_ptr<type::Type> type, SymbolKind kind,
-              bool is_const, int val = 0) -> void {
+              std::shared_ptr<type::Type> type, SymbolKind kind, bool is_const,
+              int val = 0) -> void {
 
     if (scopes.back().contains(name)) {
       Log::panic("Semantic Error: Redefinition of " + name);
@@ -127,7 +127,7 @@ public:
    * @return Symbol* Pointer to the symbol if found, else nullptr.
    */
   auto lookup(const std::string &name) -> Symbol * {
-    for (auto &scope : std::views::reverse(scopes)) {
+    for (auto &scope : scopes | std::views::reverse) {
       auto it = scope.find(name);
       if (it != scope.end()) {
         return &(it->second);