prettify
This commit is contained in:
121
day6.c
121
day6.c
@@ -1,6 +1,7 @@
|
||||
#include "lib.h"
|
||||
|
||||
#define DBG 1
|
||||
#define DBG 0
|
||||
#define PRETTY 1
|
||||
|
||||
static
|
||||
ch input[] = {
|
||||
@@ -12,18 +13,44 @@ ch test[] = {
|
||||
#embed "day6_test.txt"
|
||||
};
|
||||
|
||||
#if PRETTY
|
||||
struct line {
|
||||
union {
|
||||
ch l[24];
|
||||
struct {
|
||||
ch vs[16];
|
||||
ch pad[4];
|
||||
num v;
|
||||
};
|
||||
};
|
||||
} __attribute__((aligned(0x40)));
|
||||
static_assert(alignof(struct line) == 0x40);
|
||||
#endif
|
||||
|
||||
#if PRETTY
|
||||
alignas(0x40) static
|
||||
struct line linemem[6];
|
||||
struct line *const op_line = &linemem[1];
|
||||
struct line *const a_line = &linemem[2];
|
||||
struct line *const b_line = &linemem[3];
|
||||
struct line *const result_line = &linemem[4];
|
||||
#endif
|
||||
|
||||
static
|
||||
num do_part1(size_t file_len, ch file[file_len]) {
|
||||
ch *s = file;
|
||||
num result = 0;
|
||||
|
||||
#if PRETTY
|
||||
memset(linemem, 0, sizeof(linemem));
|
||||
snprintd(result, 16, result_line->vs);
|
||||
#endif
|
||||
#if DBG
|
||||
print("\n");
|
||||
#endif
|
||||
|
||||
while (*s != 10) s++;
|
||||
num stride = s - file + 1;
|
||||
s = memchr(s, '\n', file_len);
|
||||
num stride = s - file;
|
||||
|
||||
s = file;
|
||||
while (*s != '+' && *s != '*') s += stride;
|
||||
@@ -45,38 +72,74 @@ num do_part1(size_t file_len, ch file[file_len]) {
|
||||
if (s[op_off] == '+') {
|
||||
num a = 0;
|
||||
num rem = height;
|
||||
#if PRETTY
|
||||
op_line->vs[15] = '+';
|
||||
a_line->v = a;
|
||||
snprintd(a, 16, a_line->vs);
|
||||
*b_line = (struct line){ };
|
||||
#endif
|
||||
do {
|
||||
ch *t = s + ledge_off;
|
||||
while (*t < '0') t++;
|
||||
num b = grabnum(t, NULL);
|
||||
#if PRETTY
|
||||
b_line->v = b;
|
||||
snprintd(b, 16, b_line->vs);
|
||||
#endif
|
||||
#if DBG
|
||||
if (rem < height) { print("\t+\t"); }
|
||||
printd(b);
|
||||
#endif
|
||||
a += b;
|
||||
#if PRETTY
|
||||
a_line->v = a;
|
||||
snprintd(a, 16, a_line->vs);
|
||||
#endif
|
||||
ledge_off += stride;
|
||||
rem--;
|
||||
} while (rem);
|
||||
result += a;
|
||||
#if PRETTY
|
||||
result_line->v = result;
|
||||
snprintd(result, 16, result_line->vs);
|
||||
#endif
|
||||
#if DBG
|
||||
print("\t= "); printd(a); print("\n");
|
||||
#endif
|
||||
} else if (s[op_off] == '*') {
|
||||
num a = 1;
|
||||
#if PRETTY
|
||||
op_line->vs[15] = '*';
|
||||
a_line->v = a;
|
||||
snprintd(a, 16, a_line->vs);
|
||||
*b_line = (struct line){ };
|
||||
#endif
|
||||
num rem = height;
|
||||
do {
|
||||
ch *t = s + ledge_off;
|
||||
while (*t < '0') t++;
|
||||
num b = grabnum(t, NULL);
|
||||
#if PRETTY
|
||||
b_line->v = b;
|
||||
snprintd(b, 16, b_line->vs);
|
||||
#endif
|
||||
#if DBG
|
||||
if (rem < height) { print("\t*\t"); }
|
||||
printd(b);
|
||||
#endif
|
||||
a *= b;
|
||||
#if PRETTY
|
||||
a_line->v = a;
|
||||
snprintd(a, 16, a_line->vs);
|
||||
#endif
|
||||
ledge_off += stride;
|
||||
rem--;
|
||||
} while (rem);
|
||||
result += a;
|
||||
#if PRETTY
|
||||
result_line->v = result;
|
||||
snprintd(result, 16, result_line->vs);
|
||||
#endif
|
||||
#if DBG
|
||||
print("\t= "); printd(a); print("\n");
|
||||
#endif
|
||||
@@ -95,6 +158,12 @@ num do_part1(size_t file_len, ch file[file_len]) {
|
||||
}
|
||||
} while (op_off < file_len);
|
||||
|
||||
#if PRETTY
|
||||
*op_line = (struct line){ };
|
||||
*a_line = (struct line){ };
|
||||
*b_line = (struct line){ };
|
||||
#endif
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
@@ -103,12 +172,16 @@ num do_part2(size_t file_len, unsigned char file[file_len]) {
|
||||
ch *s = file;
|
||||
num result = 0;
|
||||
|
||||
#if PRETTY
|
||||
memset(linemem, 0, sizeof(linemem));
|
||||
snprintd(result, 16, result_line->vs);
|
||||
#endif
|
||||
#if DBG
|
||||
print("\n");
|
||||
#endif
|
||||
|
||||
while (*s != 10) s++;
|
||||
num stride = s - file + 1;
|
||||
s = memchr(s, '\n', file_len);
|
||||
num stride = s - file;
|
||||
|
||||
s = file;
|
||||
while (*s != '+' && *s != '*') s += stride;
|
||||
@@ -139,6 +212,12 @@ num do_part2(size_t file_len, unsigned char file[file_len]) {
|
||||
num a = 0;
|
||||
num rem = redge_off - ledge_off + 1;
|
||||
num frem = rem;
|
||||
#if PRETTY
|
||||
op_line->vs[15] = '+';
|
||||
a_line->v = a;
|
||||
snprintd(a, 16, a_line->vs);
|
||||
*b_line = (struct line){ };
|
||||
#endif
|
||||
do {
|
||||
num b = 0;
|
||||
ch *t = s + redge_off;
|
||||
@@ -148,6 +227,10 @@ num do_part2(size_t file_len, unsigned char file[file_len]) {
|
||||
b += *t - '0';
|
||||
}
|
||||
}
|
||||
#if PRETTY
|
||||
b_line->v = b;
|
||||
snprintd(b, 16, b_line->vs);
|
||||
#endif
|
||||
#if DBG
|
||||
if (rem < frem) { print("\t+\t"); }
|
||||
printd(b);
|
||||
@@ -157,6 +240,10 @@ num do_part2(size_t file_len, unsigned char file[file_len]) {
|
||||
rem--;
|
||||
} while (rem);
|
||||
result += a;
|
||||
#if PRETTY
|
||||
result_line->v = result;
|
||||
snprintd(result, 16, result_line->vs);
|
||||
#endif
|
||||
#if DBG
|
||||
print("\t=\t"); printd(a); print("\n");
|
||||
#endif
|
||||
@@ -164,6 +251,12 @@ num do_part2(size_t file_len, unsigned char file[file_len]) {
|
||||
num a = 1;
|
||||
num rem = redge_off - ledge_off + 1;
|
||||
num frem = rem;
|
||||
#if PRETTY
|
||||
op_line->vs[15] = '*';
|
||||
a_line->v = a;
|
||||
snprintd(a, 16, a_line->vs);
|
||||
*b_line = (struct line){ };
|
||||
#endif
|
||||
do {
|
||||
num b = 0;
|
||||
ch *t = s + redge_off;
|
||||
@@ -173,6 +266,10 @@ num do_part2(size_t file_len, unsigned char file[file_len]) {
|
||||
b += *t - '0';
|
||||
}
|
||||
}
|
||||
#if PRETTY
|
||||
b_line->v = b;
|
||||
snprintd(b, 16, b_line->vs);
|
||||
#endif
|
||||
#if DBG
|
||||
if (rem < frem) { print("\t*\t"); }
|
||||
printd(b);
|
||||
@@ -182,6 +279,10 @@ num do_part2(size_t file_len, unsigned char file[file_len]) {
|
||||
rem--;
|
||||
} while (rem);
|
||||
result += a;
|
||||
#if PRETTY
|
||||
result_line->v = result;
|
||||
snprintd(result, 16, result_line->vs);
|
||||
#endif
|
||||
#if DBG
|
||||
print("\t=\t"); printd(a); print("\n");
|
||||
#endif
|
||||
@@ -195,11 +296,17 @@ num do_part2(size_t file_len, unsigned char file[file_len]) {
|
||||
prev_op_off = op_off;
|
||||
} while (prev_op_off != fop_off);
|
||||
|
||||
#if PRETTY
|
||||
*op_line = (struct line){ };
|
||||
*a_line = (struct line){ };
|
||||
*b_line = (struct line){ };
|
||||
#endif
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
#define RUN_TEST1 1
|
||||
#define RUN_PART1 1
|
||||
#define RUN_TEST1 0
|
||||
#define RUN_PART1 0
|
||||
#define RUN_TEST2 1
|
||||
#define RUN_PART2 1
|
||||
|
||||
|
||||
8
lib.h
8
lib.h
@@ -9,6 +9,14 @@
|
||||
|
||||
#define STDOUT_FILENO 1
|
||||
|
||||
__attribute__((always_inline))
|
||||
void *memchr(const void *s, int c, size_t n) {
|
||||
uint64_t dispose;
|
||||
void *d;
|
||||
asm("repnz scasb":"=D"(d),"=c"(dispose):"D"(s),"c"(n),"a"(c):"memory","cc");
|
||||
return d;
|
||||
}
|
||||
|
||||
static
|
||||
void write(uint64_t fileno, void *buffer, size_t len) {
|
||||
uint64_t a,d;
|
||||
|
||||
Reference in New Issue
Block a user