AOC 2022 day 1!

binary-banter · Nov 30, 2023 · 59dcf5b · 59dcf5b
1 parent 24fad29
commit 59dcf5b
Show file tree

Hide file tree

Showing 3 changed files with 61 additions and 5 deletions.
diff --git a/compiler/src/passes/explicate/display.rs b/compiler/src/passes/explicate/display.rs
@@ -60,8 +60,12 @@ impl Display for CExpr<'_> {
             CExpr::AccessField { strct, field } => {
                 write!(f, "{strct}.{field}")
             }
-            CExpr::Asm { .. } => {
-                todo!()
+            CExpr::Asm { instrs } => {
+                writeln!(f, "asm {{")?;
+                for instr in instrs {
+                    writeln!(indented(f), "{instr}")?;
+                }
+                write!(f, "}}")
             }
         }
     }

diff --git a/compiler/src/passes/select/mod.rs b/compiler/src/passes/select/mod.rs
@@ -72,7 +72,7 @@ pub enum Instr<Arg: Display, IdentVars: Display> {
     Popq { dst: Arg },
     #[display(fmt = "retq")]
     Retq,
-    #[display(fmt = "syscall\t// arity: {arity}")]
+    #[display(fmt = "syscall\t{arity}")]
     Syscall { arity: usize },
     #[display(fmt = "cmpq\t{src}\t{dst}")]
     Cmpq { src: Arg, dst: Arg },
@@ -92,9 +92,9 @@ pub enum Instr<Arg: Display, IdentVars: Display> {
     Setcc { cnd: Cnd }, //TODO allow setting other byteregs
     #[display(fmt = "loadlbl\t{sym}\t{dst}")]
     LoadLbl { sym: IdentVars, dst: Arg },
-    #[display(fmt = "call_direct\t{lbl}\t// arity: {arity}")]
+    #[display(fmt = "call_direct\t{lbl}\t{arity}")]
     CallqDirect { lbl: IdentVars, arity: usize },
-    #[display(fmt = "call_indirect\t{src}\t// arity: {arity}")]
+    #[display(fmt = "call_indirect\t{src}\t{arity}")]
     CallqIndirect { src: Arg, arity: usize },
 }
 

diff --git a/programs/aoc/y2022d01.test b/programs/aoc/y2022d01.test
@@ -0,0 +1,52 @@
+fn read_char() -> I64 {
+    let mut v = 0;
+    let mut res = 0;
+    asm {
+        subq $8 %RSP    // allocate stack space for reading char
+        movq $0 %RAX    // read
+        movq $0 %RDI    // stdin
+        movq %RSP %RSI  // put read char at top of stack
+        movq $1 %RDX    // read 1 byte
+        syscall 4       // arity of 4
+        movq %RAX {res} // result of system call
+        popq {v}        // pop read char
+    };
+    if res == 0 {
+        return res
+    };
+    v
+}
+
+fn main() {
+    let ASCII_NEWLINE = 10;
+    let ASCII_ZERO = 48;
+    let ASCII_NULL = 0;
+
+    let mut next = read_char();
+
+    let mut best = 0;
+    let mut sum = 0;
+    let mut current = 0;
+    let mut last_was_newline = false;
+
+    while next != ASCII_NULL {
+        if next == ASCII_NEWLINE {
+            if last_was_newline {
+                // Found empty line
+                if sum > best {
+                    best = sum;
+                };
+                sum = 0;
+            } else {
+                last_was_newline = true;
+                sum = sum + current;
+                current = 0;
+            }
+        } else {
+            last_was_newline = false;
+            current = current * 10 + next - ASCII_ZERO;
+        };
+        next = read_char();
+    };
+    print(best);
+}