Merge pull request #6 from danimal141/calc-support

Calculation Support

README.md

@@ -139,7 +139,9 @@ This will create a summary that can be easily shared with others or used for doc
 
 Create a file `hello.pgo`:
 ```go
-package main {
+package main
+
+func main() {
     print("hello")
 }
 ```

examples/calc.pgo

@@ -0,0 +1,6 @@
+package main
+
+func main() {
+  print(4 + 3 * 2)
+  print(10 - 6 / 2)
+}

src/codegen/generator.test.ts

@@ -27,6 +27,7 @@ func main() {
         "declare i32 @printf",
         "@.str.fmt = private unnamed_addr constant [3 x i8]",
         "define i32 @main()",
+        "entry:",
         "call i32 (i8*, ...) @printf",
         "ret i32 0",
       ];
@@ -47,15 +48,76 @@ func main() {
   print("hello\\nworld")
 }`;
       const ir = generateIR(input);
-
-      // Debug output
-      console.log("Generated IR:", ir);
-
       assertEquals(
         ir.includes("\\0A"),
         true,
         `IR should contain "\\0A" for newline.\nGenerated IR:\n${ir}`,
       );
     });
   });
+
+  describe("arithmetic expressions", () => {
+    it("should generate correct IR for integer expressions", () => {
+      const input = `package main
+      func main() {
+        print(42)
+      }`;
+      const ir = generateIR(input);
+
+      const requiredParts = [
+        '@.str.int.fmt = private unnamed_addr constant [4 x i8] c"%d\\0A\\00"',
+        "call i32 (i8*, ...) @printf",
+      ];
+
+      for (const part of requiredParts) {
+        assertEquals(
+          ir.includes(part),
+          true,
+          `Expected IR to contain "${part}"\nActual IR:\n${ir}`,
+        );
+      }
+    });
+
+    it("should generate correct IR for complex arithmetic expressions", () => {
+      const input = `package main
+      func main() {
+        print(4 + 3 * 2)
+        print(10 - 6 / 2)
+      }`;
+      const ir = generateIR(input);
+
+      // Check instruction ordering and numbering
+      const instructions = [
+        "%1 = mul nsw i32 3, 2", // First multiplication: 3 * 2
+        "%2 = add nsw i32 4, %1", // Then addition: 4 + result
+        "%3 = call i32 (i8*, ...)", // First print call
+        "%4 = sdiv i32 6, 2", // Division: 6 / 2
+        "%5 = sub nsw i32 10, %4", // Then subtraction: 10 - result
+        "%6 = call i32 (i8*, ...)", // Second print call
+      ];
+
+      for (const instruction of instructions) {
+        assertEquals(
+          ir.includes(instruction),
+          true,
+          `Expected IR to contain instruction: "${instruction}"\nActual IR:\n${ir}`,
+        );
+      }
+
+      // Check instruction order
+      const irLines = ir.split("\n");
+      let lastIndex = -1;
+      for (const instruction of instructions) {
+        const currentIndex = irLines.findIndex((line) =>
+          line.trim().startsWith(instruction)
+        );
+        assertEquals(
+          currentIndex > lastIndex,
+          true,
+          `Instruction "${instruction}" is not in the correct order`,
+        );
+        lastIndex = currentIndex;
+      }
+    });
+  });
 });