day 5, pretty blinkenlights mem, awful performance

day5.c

@@ -0,0 +1,230 @@
+#include "lib.h"
+
+#define DBG 0
+#define PRETTY 1
+
+static
+unsigned char input[] = {
+#embed "day5_input.txt"
+};
+
+static
+unsigned char test[] = {
+#embed "day5_test.txt"
+};
+
+#if PRETTY
+struct line {
+    union {
+        unsigned char l[64];
+        struct {
+            unsigned char s[16];
+            unsigned char pad0[4];
+            unsigned char e[16];
+            unsigned char pad1[4];
+            num sv;
+            num ev;
+        };
+    };
+};
+static_assert(sizeof(struct line) == 0x40);
+#else
+struct line {
+    num sv;
+    num ev;
+};
+#endif
+
+alignas(0x40) static
+struct line ranges[256];
+
+static
+unsigned long do_part1(size_t file_len, unsigned char file[file_len]) {
+    unsigned char *s = file;
+    unsigned long result = 0;
+
+    memset(ranges, 0, sizeof(ranges));
+
+#if DBG
+    print("\n");
+#endif
+
+    struct line *d = ranges;
+    do {
+        unsigned char *tmp;
+        d->sv = grabbcd(s, &tmp);
+#if PRETTY
+        memcpy(d->s, s, tmp - s);
+#endif
+        s = tmp;
+        s++;
+        d->ev = grabbcd(s, &tmp);
+#if PRETTY
+        memcpy(d->e, s, tmp - s);
+#endif
+        s = tmp;
+        s++;
+        d++;
+    } while (*s != '\n');
+    d->sv = 0;
+#if PRETTY
+    d->pad0[0] = 'F';
+    d->pad0[1] = 'I';
+    d->pad0[2] = 'N';
+    d->pad0[3] = 'I';
+#endif
+    s++;
+
+    do {
+        bcdint id = grabbcd(s, &s);
+        for (d = ranges; d->sv; d++) {
+            bcdint ds = d->sv;
+            bcdint de = d->ev;
+            if (id >= ds && id <= de) {
+                result++;
+                goto next_id;
+            }
+        }
+next_id:
+        s++;
+    } while (s != &file[file_len]);
+
+    return result;
+}
+
+static
+unsigned long do_part2(size_t file_len, unsigned char file[file_len]) {
+    unsigned char *s = file;
+    unsigned long result = 0;
+
+    memset(ranges, 0, sizeof(ranges));
+
+#if DBG
+    print("\n");
+#endif
+
+    struct line *d = ranges;
+    do {
+        unsigned char *tmp;
+        d->sv = grabbcd(s, &tmp);
+#if PRETTY
+        snprintd(d->sv, sizeof(d->s), d->s);
+#endif
+        s = tmp;
+        s++;
+        d->ev = grabbcd(s, &tmp);
+#if PRETTY
+        snprintd(d->ev, sizeof(d->e), d->e);
+#endif
+        s = tmp;
+        s++;
+        d++;
+    } while (*s != '\n');
+    d->sv = 0;
+#if PRETTY
+    d->pad0[0] = 'F';
+    d->pad0[1] = 'I';
+    d->pad0[2] = 'N';
+    d->pad0[3] = 'I';
+#endif
+
+    for (struct line *di = ranges; di->sv; di++) {
+        bcdint dis = di->sv;
+        bcdint die = di->ev;
+        if (dis == -1) continue;
+        for (struct line *dj = ranges; dj->sv; dj++) {
+            if (dj == di) continue;
+            bcdint djs = dj->sv;
+            bcdint dje = dj->ev;
+            if (djs == -1) continue;
+            if (djs <= die && djs >= dis) {
+                djs = bcdadd(die, 1);
+            }
+            if (dje >= dis && dje <= die) {
+                dje = bcdsub(dis, 1);
+            }
+            if (djs == dj->sv && dje == dj->ev)
+                continue;
+            if (dje < djs) {
+#if DBG
+                print("FOR: "); printd(dis); print(" "); printd(die); print("\n");
+                print("     "); printd(dj->sv); print(" "); printd(dj->ev); print("\n");
+                print("  -> "); printd(djs); print(" "); printd(dje); print(" BAD\n");
+#endif
+                dj->sv = -1;
+                dj->ev = -1;
+#if PRETTY
+                snprinth(dj->sv, sizeof(dj->s), dj->s);
+                snprinth(dj->ev, sizeof(dj->e), dj->e);
+#endif
+            } else {
+#if DBG
+                print("FOR: "); printd(dis); print(" "); printd(die); print("\n");
+                print("     "); printd(dj->sv); print(" "); printd(dj->ev); print("\n");
+                print("  -> "); printd(djs); print(" "); printd(dje); print("\n");
+#endif
+                dj->sv = djs;
+                dj->ev = dje;
+#if PRETTY
+                snprinth(dj->sv, sizeof(dj->s), dj->s);
+                snprinth(dj->ev, sizeof(dj->e), dj->e);
+#endif
+            }
+        }
+    }
+
+    for (struct line *di = ranges; di->sv; di++) {
+        if (di->sv == -1) continue;
+        unsigned long dis = bcdtonum(di->sv);
+        unsigned long die = bcdtonum(di->ev);
+        result += die - dis + 1;
+    }
+
+    return result;
+}
+
+#define RUN_TEST1 1
+#define RUN_PART1 1
+#define RUN_TEST2 1
+#define RUN_PART2 1
+
+#define TEST1_EXPECT 3
+#define TEST2_EXPECT 14
+
+void run() {
+#if RUN_TEST1
+    print("PART 1 TEST: ");
+    if (unsigned long v = do_part1(countof(test), test); v != TEST1_EXPECT) {
+        print("FAILED (got ");
+        printd(v);
+        print(", expected " xstr(TEST1_EXPECT) ")\n");
+    } else {
+        print("PASSED\n");
+    }
+#endif
+
+#if RUN_PART1
+    print("PART 1 RESULT: ");
+    printd(do_part1(countof(input), input));
+    print("\n");
+#endif
+
+#if RUN_TEST2
+    print("PART 2 TEST: ");
+    if (unsigned long v = do_part2(countof(test), test); v != TEST2_EXPECT) {
+        print("FAILED (got ");
+        printd(v);
+        print(", expected " xstr(TEST2_EXPECT) ")\n");
+    } else {
+        print("PASSED\n");
+    }
+#endif
+
+#if RUN_PART2
+    print("PART 2 RESULT: ");
+    printd(do_part2(countof(input), input));
+    print("\n");
+#endif
+
+    exit_group(0);
+}