split into separate packages/classes

1 files changed, 97 insertions, 0 deletions

diff --git a/src/main/java/no/eliashaugsbakk/kompilator/asmGeneration/AssemblyBuilder.java b/src/main/java/no/eliashaugsbakk/kompilator/asmGeneration/AssemblyBuilder.java
new file mode 100644
index 0000000..94a753b
--- /dev/null
+++ b/src/main/java/no/eliashaugsbakk/kompilator/asmGeneration/AssemblyBuilder.java
@@ -0,0 +1,97 @@
+package no.eliashaugsbakk.kompilator.asmGeneration;
+
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+public class AssemblyBuilder {
+  Map<String, String> stringVariables = new HashMap<>();
+
+  StringBuilder finalAssembly;
+  StringBuilder text;
+  StringBuilder rodata;
+  StringBuilder data;
+
+  private final static String exit = """
+      mov rax, 60
+      xor rdi, rdi
+      syscall
+      """;
+
+  public AssemblyBuilder() {
+    this.finalAssembly = new StringBuilder(".intel_syntax noprefix\n.global _start\n");
+    // CPU instructions
+    this.text = new StringBuilder(".text\n_start:\n\n");
+    // Read only data
+    this.rodata = new StringBuilder("\n.section .rodata\n");
+    // Global initialized variables, Read - Write
+    this.data = new StringBuilder("\n.data\n");
+  }
+
+  public String createAssembly(List<String> IR) {
+    for (String line : IR) {
+      line = line.trim();
+
+      if (line.startsWith("print")) {
+        handlePrint(line);
+      } else if (line.contains("=")) {
+        handleAssignment(line);
+      }
+    }
+    this.text.append(exit);
+    this.finalAssembly.append(text);
+    this.finalAssembly.append(rodata);
+    this.finalAssembly.append(data);
+    return this.finalAssembly.toString();
+  }
+
+  private void handleAssignment(String line) {
+    /* Example IR:
+    variable_name: type = data
+    str1: string = "Hello world"
+     */
+    String name = line.substring(0, line.indexOf(":"));
+
+    String dataType = line.substring(line.indexOf(":") + 2, line.indexOf("=") - 1);
+
+    String value = line.substring(line.indexOf("=") + 2);
+
+    if (dataType.contains("string")) {
+      assignString(name, value);
+    } else {
+      IO.println("err: Unknown data type: " + dataType);
+    }
+  }
+
+  private void assignString(String name, String value) {
+    value = value.substring(1, value.length() - 1);
+    this.rodata.append(String.format("""
+        %s: .ascii "%s"
+        """, name, value));
+    this.stringVariables.put(name, value);
+  }
+
+  void handlePrint(String line) {
+    /* Takes in a print statement and constructs the assembly
+    example input:
+    print(x)
+     */
+    String variableName = line.substring(line.indexOf("(") + 1, line.lastIndexOf(")"));
+
+    int stringLength = 0;
+    try {
+      stringLength = this.stringVariables.get(variableName).length();
+    } catch (RuntimeException e) {
+      IO.println("No variable " + variableName + " initialized in IR.");
+    }
+
+    this.text.append(String.format("""
+        mov rax, 1
+        mov rdi, 1
+        lea rsi, [%s]
+        lea rdx, [%d]
+        syscall
+        
+        """, variableName, stringLength));
+  }
+}