arrays

src/codegen_x86_64.rs

@@ -58,7 +58,7 @@ macro_rules! emit {
 }
 
 static REGISTERS: [&str; 6] = ["rdi", "rsi", "rdx", "rcx", "r8", "r9"];
-static TYPES: [&str; 5] = ["I64", "String", "Bool", "Ptr", "Char"];
+static TYPES: [&str; 6] = ["I64", "String", "Bool", "Ptr", "Char", "Array"];
 
 pub struct CodegenX86_64 {
     output: String,
@@ -97,6 +97,8 @@ impl CodegenX86_64 {
             "
 section .text
 extern malloc
+extern calloc
+extern realloc
 extern free
 extern printf
 extern sprintf
@@ -117,11 +119,11 @@ extern exit
 extern gettimeofday
 copystr equ strdup
 
-nth:
+String.nth:
     movzx rax, byte [rdi + rsi]
     ret
 
-set:
+String.set:
     mov [rdi + rsi], dl
     ret
 
@@ -142,7 +144,58 @@ time:
     pop rbx
     ret
 
-isqrt:
+Array.new:
+    mov rdi, 1
+    mov rsi, 24
+    jmp calloc
+
+Array.nth:
+    mov rax, [rdi]
+    mov rax, [rax + rsi*8]
+    ret
+
+Array.push:
+    push r14
+    push rbx
+    push rax
+    mov r14, rsi
+    mov rbx, rdi
+    mov rax, [rdi]
+    mov rcx, [rdi + 16]
+    cmp rcx, [rdi + 8]
+    jne .no_realloc
+    lea rdx, [rcx + rcx]
+    mov rsi, 4
+    test rcx, rcx
+    cmovnz rsi, rdx
+    mov [rbx + 8], rsi
+    shl rsi, 3
+    mov rdi, rax
+    call realloc
+    mov [rbx], rax
+    mov rcx, [rbx + 16]
+.no_realloc:
+    mov [rax + rcx*8], r14
+    inc qword [rbx + 16]
+    add rsp, 8
+    pop rbx
+    pop r14
+    ret
+
+Array.size:
+    mov rax, [rdi + 16]
+    ret
+
+Array.free:
+    push rbx
+    mov rbx, rdi
+    mov rdi, [rdi]
+    call free
+    mov rdi, rbx
+    pop rbx
+    jmp free
+
+Math.isqrt:
     xor rax, rax
     mov rcx, 1
     mov rbx, rdi
@@ -400,7 +453,12 @@ isqrt:
                 }
                 TokenType::String => {
                     // TODO: actual string parsing in the tokenizer
-                    let value = &token.lexeme[1..token.lexeme.len() - 1].replace("\\n", "\n");
+                    let value = &token.lexeme[1..token.lexeme.len() - 1]
+                        .replace("\\n", "\n")
+                        .replace("\\r", "\r")
+                        .replace("\\t", "\t")
+                        .replace("\\033", "\x1b")
+                        .replace("\\0", "\0");
 
                     if value.is_empty() {
                         emit!(&mut self.data_section, "    S{} db 0", self.data_counter);

src/std.zr

@@ -23,12 +23,10 @@ func strrev[s: String] : String
     let len: I64 = strlen(s)
     let out: String = malloc(len + 1)
 
-    let i: I64 = 0
-    while i < len
-        let c: Char = nth(s, len - i - 1)
-        set(out, i, c)
-        i = i + 1
-    set(out, len, 0)
+    for i in 0:len
+        let c: Char = String.nth(s, len - i - 1)
+        String.set(out, i, c)
+    String.set(out, len, 0)
     return out
 
 func IO.read_file[path: String]: String
@@ -43,7 +41,7 @@ func IO.read_file[path: String]: String
     let buffer: String = malloc(size + 1)
 
     let n: I64 = fread(buffer, 1, size, file)
-    set(buffer, n, 0)
+    String.set(buffer, n, 0)
     return buffer
 
 func Char.to_i64[c: Char]: I64
@@ -78,10 +76,8 @@ func Math.abs[n: I64] : I64
 
 func Math.pow[b: I64, e: I64] : I64
     let out: I64 = 1
-    let i: I64 = 0
-    while i < e
+    for i in 0:e
         out = out * b
-        i = i + 1
     return out
 
 func Math.lcm[a: I64, b: I64] : I64