about summary refs log tree commit diff stats
path: root/linux/advent2020/5b.mu
blob: e6a3520f8b2fa66b76376ecc95d75d9c676501b2 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
# https://adventofcode.com/2020/day/5
#
# To run (on Linux):
#   $ git clone https://github.com/akkartik/mu
#   $ cd mu
#   $ ./translate advent2020/5b.mu
#   $ ./a.elf < input
#
# You'll need to register to download the 'input' file for yourself.

fn main -> _/ebx: int {
  var pass-storage: (array int 0x400)  # 1k ints
  var pass/esi: (addr array int) <- address pass-storage
  # phase 1: populate pass array
  var line-storage: (stream byte 0x10)  # 16 bytes is enough
  var line/edx: (addr stream byte) <- address line-storage
  {
    # read line from stdin
    clear-stream line
    read-line-from-real-keyboard line
    # if line is empty (not even a newline), quit
    var done?/eax: boolean <- stream-empty? line
    compare done?, 0/false
    break-if-!=
    # process line
    var seat-id/eax: int <- convert-from-binary line
    var dest/eax: (addr int) <- index pass, seat-id
    copy-to *dest, 1
    loop
  }
  # phase 2: skip empty seats
  var i/eax: int <- copy 0
  {
    compare i, 0x400
    break-if->=
    var src/ecx: (addr int) <- index pass, i
    compare *src, 0
    break-if-!=
    i <- increment
    loop
  }
  # phase 3: skip non-empty seats
  {
    compare i, 0x400
    break-if->=
    var src/ecx: (addr int) <- index pass, i
    compare *src, 0
    break-if-=
    i <- increment
    loop
  }
  print-int32-decimal 0, i
  print-string 0, "\n"
  return 0
}

fn convert-from-binary in: (addr stream byte) -> _/eax: int {
  var result/edi: int <- copy 0
  var i/ecx: int <- copy 9  # loop counter and also exponent
  {
    compare i, 0
    break-if-<
    var c/eax: byte <- read-byte in
    var bit/edx: int <- copy 0
    {
      compare c, 0x42/B
      break-if-!=
      bit <- copy 1
    }
    {
      compare c, 0x52/R
      break-if-!=
      bit <- copy 1
    }
    var bit-value/eax: int <- repeated-shift-left bit, i
    result <- add bit-value
    i <- decrement
    loop
  }
  return result
}
> compare c, 0x52/R break-if-!= bit <- copy 1 } #? print-string 0, "bit: " #? print-int32-decimal 0, bit #? print-string 0, "\n" var bit-value/eax: int <- repeated-shift-left bit, i #? print-string 0, "bit value: " #? print-int32-decimal 0, bit-value #? print-string 0, "\n" result <- add bit-value #? print-string 0, "result: " #? print-int32-decimal 0, result #? print-string 0, "\n" i <- decrement loop } print-int32-decimal 0, result print-string 0, "\n" return result }