Put all printer state into a global struct

This commit is contained in:
Lassi Kortela 2019-08-25 16:56:02 +03:00
parent 6defff3696
commit 49e0a3ad1f
2 changed files with 57 additions and 55 deletions

View File

@ -2562,7 +2562,7 @@ static void lisp_init(size_t initial_heapsize)
curheap = fromspace; curheap = fromspace;
lim = curheap + heapsize - sizeof(struct cons); lim = curheap + heapsize - sizeof(struct cons);
consflags = bitvector_new(heapsize / sizeof(struct cons), 1); consflags = bitvector_new(heapsize / sizeof(struct cons), 1);
htable_new(&cycle_visited_pairs, 32); htable_new(&pr.cycle_traversed, 32);
comparehash_init(); comparehash_init();
N_STACK = 262144; N_STACK = 262144;
Stack = malloc(N_STACK * sizeof(value_t)); Stack = malloc(N_STACK * sizeof(value_t));

110
c/print.h
View File

@ -1,5 +1,14 @@
extern void *memrchr(const void *s, int c, size_t n); extern void *memrchr(const void *s, int c, size_t n);
// Printer state
static struct printer {
int line;
int column;
unsigned int level;
unsigned int cycle_labels;
struct htable cycle_traversed;
} pr;
// *print-readably* -- use `write` repr instead of `display` repr // *print-readably* -- use `write` repr instead of `display` repr
static int print_readably; static int print_readably;
@ -15,31 +24,25 @@ static fixnum_t print_length;
// *print-level* -- print only the outermost N levels of nested structures // *print-level* -- print only the outermost N levels of nested structures
static fixnum_t print_level; static fixnum_t print_level;
static fixnum_t cur_line;
static fixnum_t cur_column = 0;
static fixnum_t cur_level;
static uint32_t cycle_used_labels;
static struct htable cycle_visited_pairs;
static void outc(char c, struct ios *f) static void outc(char c, struct ios *f)
{ {
ios_putc(c, f); ios_putc(c, f);
if (c == '\n') if (c == '\n')
cur_column = 0; pr.column = 0;
else else
cur_column++; pr.column++;
} }
static void outs(char *s, struct ios *f) static void outs(char *s, struct ios *f)
{ {
ios_puts(s, f); ios_puts(s, f);
cur_column += u8_strwidth(s); pr.column += u8_strwidth(s);
} }
static void outsn(char *s, struct ios *f, size_t n) static void outsn(char *s, struct ios *f, size_t n)
{ {
ios_write(f, s, n); ios_write(f, s, n);
cur_column += u8_strwidth(s); pr.column += u8_strwidth(s);
} }
static int outindent(int n, struct ios *f) static int outindent(int n, struct ios *f)
@ -51,8 +54,8 @@ static int outindent(int n, struct ios *f)
n = 2; n = 2;
n0 = n; n0 = n;
ios_putc('\n', f); ios_putc('\n', f);
cur_line++; pr.line++;
cur_column = n; pr.column = n;
while (n) { while (n) {
ios_putc(' ', f); ios_putc(' ', f);
n--; n--;
@ -70,9 +73,9 @@ void print_traverse(value_t v)
while (iscons(v)) { while (iscons(v)) {
if (ismarked(v)) { if (ismarked(v)) {
bp = (value_t *)ptrhash_bp(&cycle_visited_pairs, (void *)v); bp = (value_t *)ptrhash_bp(&pr.cycle_traversed, (void *)v);
if (*bp == (value_t)HT_NOTFOUND) if (*bp == (value_t)HT_NOTFOUND)
*bp = fixnum(cycle_used_labels++); *bp = fixnum(pr.cycle_labels++);
return; return;
} }
mark_cons(v); mark_cons(v);
@ -82,9 +85,9 @@ void print_traverse(value_t v)
if (!ismanaged(v) || issymbol(v)) if (!ismanaged(v) || issymbol(v))
return; return;
if (ismarked(v)) { if (ismarked(v)) {
bp = (value_t *)ptrhash_bp(&cycle_visited_pairs, (void *)v); bp = (value_t *)ptrhash_bp(&pr.cycle_traversed, (void *)v);
if (*bp == (value_t)HT_NOTFOUND) if (*bp == (value_t)HT_NOTFOUND)
*bp = fixnum(cycle_used_labels++); *bp = fixnum(pr.cycle_labels++);
return; return;
} }
if (isvector(v)) { if (isvector(v)) {
@ -281,7 +284,7 @@ static void print_pair(struct ios *f, value_t v)
op = NULL; op = NULL;
if (iscons(cdr_(v)) && cdr_(cdr_(v)) == NIL && if (iscons(cdr_(v)) && cdr_(cdr_(v)) == NIL &&
!ptrhash_has(&cycle_visited_pairs, (void *)cdr_(v)) && !ptrhash_has(&pr.cycle_traversed, (void *)cdr_(v)) &&
(((car_(v) == QUOTE) && (op = "'")) || (((car_(v) == QUOTE) && (op = "'")) ||
((car_(v) == BACKQUOTE) && (op = "`")) || ((car_(v) == BACKQUOTE) && (op = "`")) ||
((car_(v) == COMMA) && (op = ",")) || ((car_(v) == COMMA) && (op = ",")) ||
@ -294,9 +297,9 @@ static void print_pair(struct ios *f, value_t v)
fl_print_child(f, car_(cdr_(v))); fl_print_child(f, car_(cdr_(v)));
return; return;
} }
startpos = cur_column; startpos = pr.column;
outc('(', f); outc('(', f);
newindent = cur_column; newindent = pr.column;
blk = blockindent(v); blk = blockindent(v);
n = ind = always = 0; n = ind = always = 0;
if (!blk) if (!blk)
@ -311,10 +314,10 @@ static void print_pair(struct ios *f, value_t v)
outsn("...)", f, 4); outsn("...)", f, 4);
break; break;
} }
last_line = cur_line; last_line = pr.line;
unmark_cons(v); unmark_cons(v);
fl_print_child(f, car_(v)); fl_print_child(f, car_(v));
if (!iscons(cd) || ptrhash_has(&cycle_visited_pairs, (void *)cd)) { if (!iscons(cd) || ptrhash_has(&pr.cycle_traversed, (void *)cd)) {
if (cd != NIL) { if (cd != NIL) {
outsn(" . ", f, 3); outsn(" . ", f, 3);
fl_print_child(f, cd); fl_print_child(f, cd);
@ -331,12 +334,12 @@ static void print_pair(struct ios *f, value_t v)
nextsmall = smallp(car_(cd)); nextsmall = smallp(car_(cd));
thistiny = tinyp(car_(v)); thistiny = tinyp(car_(v));
ind = ind =
(((cur_line > last_line) || (cur_column > print_width / 2 && (((pr.line > last_line) || (pr.column > print_width / 2 &&
!nextsmall && !thistiny && n > 0)) || !nextsmall && !thistiny && n > 0)) ||
(cur_column > print_width - 4) || (pr.column > print_width - 4) ||
(est != -1 && (cur_column + est > print_width - 2)) || (est != -1 && (pr.column + est > print_width - 2)) ||
((head == LAMBDA) && !nextsmall) || ((head == LAMBDA) && !nextsmall) ||
@ -361,7 +364,7 @@ static void print_pair(struct ios *f, value_t v)
if (si != -1) if (si != -1)
newindent = startpos + si; newindent = startpos + si;
else if (!blk) else if (!blk)
newindent = cur_column; newindent = pr.column;
} }
} }
n++; n++;
@ -375,13 +378,13 @@ static int write_cycle_prefix(struct ios *f, value_t v)
{ {
value_t label; value_t label;
if ((label = (value_t)ptrhash_get(&cycle_visited_pairs, (void *)v)) != if ((label = (value_t)ptrhash_get(&pr.cycle_traversed, (void *)v)) !=
(value_t)HT_NOTFOUND) { (value_t)HT_NOTFOUND) {
if (!ismarked(v)) { if (!ismarked(v)) {
cur_column += ios_printf(f, "#%ld#", numval(label)); pr.column += ios_printf(f, "#%ld#", numval(label));
return 1; return 1;
} }
cur_column += ios_printf(f, "#%ld=", numval(label)); pr.column += ios_printf(f, "#%ld=", numval(label));
} }
if (ismanaged(v)) if (ismanaged(v))
unmark_cons(v); unmark_cons(v);
@ -392,16 +395,16 @@ void fl_print_child(struct ios *f, value_t v)
{ {
char *name; char *name;
if (print_level >= 0 && cur_level >= print_level && if (print_level >= 0 && pr.level >= print_level &&
(iscons(v) || isvector(v) || isclosure(v))) { (iscons(v) || isvector(v) || isclosure(v))) {
outc('#', f); outc('#', f);
return; return;
} }
cur_level++; pr.level++;
switch (tag(v)) { switch (tag(v)) {
case TAG_NUM: case TAG_NUM:
case TAG_NUM1: case TAG_NUM1:
cur_column += ios_printf(f, "%ld", numval(v)); pr.column += ios_printf(f, "%ld", numval(v));
break; break;
case TAG_SYM: case TAG_SYM:
name = symbol_name(v); name = symbol_name(v);
@ -475,7 +478,7 @@ void fl_print_child(struct ios *f, value_t v)
int newindent, est, sz, i; int newindent, est, sz, i;
outc('[', f); outc('[', f);
newindent = cur_column; newindent = pr.column;
sz = vector_size(v); sz = vector_size(v);
for (i = 0; i < sz; i++) { for (i = 0; i < sz; i++) {
if (print_length >= 0 && i >= print_length && i < sz - 1) { if (print_length >= 0 && i >= print_length && i < sz - 1) {
@ -488,10 +491,10 @@ void fl_print_child(struct ios *f, value_t v)
outc(' ', f); outc(' ', f);
} else { } else {
est = lengthestimate(vector_elt(v, i + 1)); est = lengthestimate(vector_elt(v, i + 1));
if (cur_column > print_width - 4 || if (pr.column > print_width - 4 ||
(est != -1 && (est != -1 &&
(cur_column + est > print_width - 2)) || (pr.column + est > print_width - 2)) ||
(cur_column > print_width / 2 && (pr.column > print_width / 2 &&
!smallp(vector_elt(v, i + 1)) && !smallp(vector_elt(v, i + 1)) &&
!tinyp(vector_elt(v, i)))) !tinyp(vector_elt(v, i))))
newindent = outindent(newindent, f); newindent = outindent(newindent, f);
@ -509,7 +512,7 @@ void fl_print_child(struct ios *f, value_t v)
print_pair(f, v); print_pair(f, v);
break; break;
} }
cur_level--; pr.level--;
} }
static void print_string(struct ios *f, char *str, size_t sz) static void print_string(struct ios *f, char *str, size_t sz)
@ -644,9 +647,9 @@ static void cvalue_printdata(struct ios *f, void *data, size_t len,
if (!print_readably) if (!print_readably)
outc(ch, f); outc(ch, f);
else if (weak) else if (weak)
cur_column += ios_printf(f, "#x%hhx", ch); pr.column += ios_printf(f, "#x%hhx", ch);
else else
cur_column += ios_printf(f, "#byte(#x%hhx)", ch); pr.column += ios_printf(f, "#byte(#x%hhx)", ch);
} else if (type == wcharsym) { } else if (type == wcharsym) {
char seq[8]; char seq[8];
uint32_t wc = *(uint32_t *)data; uint32_t wc = *(uint32_t *)data;
@ -686,7 +689,7 @@ static void cvalue_printdata(struct ios *f, void *data, size_t len,
else if (iswprint(wc)) else if (iswprint(wc))
outs(seq, f); outs(seq, f);
else else
cur_column += ios_printf(f, "x%04x", (int)wc); pr.column += ios_printf(f, "x%04x", (int)wc);
} }
} else if (type == floatsym || type == doublesym) { } else if (type == floatsym || type == doublesym) {
char buf[64]; char buf[64];
@ -708,8 +711,7 @@ static void cvalue_printdata(struct ios *f, void *data, size_t len,
else else
rep = sign_bit(d) ? "-inf.0" : "+inf.0"; rep = sign_bit(d) ? "-inf.0" : "+inf.0";
if (type == floatsym && print_readably && !weak) if (type == floatsym && print_readably && !weak)
cur_column += pr.column += ios_printf(f, "#%s(%s)", symbol_name(type), rep);
ios_printf(f, "#%s(%s)", symbol_name(type), rep);
else else
outs(rep, f); outs(rep, f);
} else if (d == 0) { } else if (d == 0) {
@ -737,21 +739,21 @@ static void cvalue_printdata(struct ios *f, void *data, size_t len,
) { ) {
uint64_t ui64 = *(uint64_t *)data; uint64_t ui64 = *(uint64_t *)data;
if (weak || !print_readably) if (weak || !print_readably)
cur_column += ios_printf(f, "%llu", ui64); pr.column += ios_printf(f, "%llu", ui64);
else else
cur_column += ios_printf(f, "#%s(%llu)", symbol_name(type), ui64); pr.column += ios_printf(f, "#%s(%llu)", symbol_name(type), ui64);
} else if (issymbol(type)) { } else if (issymbol(type)) {
// handle other integer prims. we know it's smaller than uint64 // handle other integer prims. we know it's smaller than uint64
// at this point, so int64 is big enough to capture everything. // at this point, so int64 is big enough to capture everything.
numerictype_t nt = sym_to_numtype(type); numerictype_t nt = sym_to_numtype(type);
if (nt == N_NUMTYPES) { if (nt == N_NUMTYPES) {
cur_column += ios_printf(f, "#<%s>", symbol_name(type)); pr.column += ios_printf(f, "#<%s>", symbol_name(type));
} else { } else {
int64_t i64 = conv_to_int64(data, nt); int64_t i64 = conv_to_int64(data, nt);
if (weak || !print_readably) if (weak || !print_readably)
cur_column += ios_printf(f, "%lld", i64); pr.column += ios_printf(f, "%lld", i64);
else else
cur_column += pr.column +=
ios_printf(f, "#%s(%lld)", symbol_name(type), i64); ios_printf(f, "#%s(%lld)", symbol_name(type), i64);
} }
} else if (iscons(type)) { } else if (iscons(type)) {
@ -774,9 +776,9 @@ static void cvalue_printdata(struct ios *f, void *data, size_t len,
/* /*
char *nl = memrchr(data, '\n', len); char *nl = memrchr(data, '\n', len);
if (nl) if (nl)
cur_column = u8_strwidth(nl+1); pr.column = u8_strwidth(nl+1);
else else
cur_column += u8_strwidth(data); pr.column += u8_strwidth(data);
*/ */
} else { } else {
print_string(f, (char *)data, len); print_string(f, (char *)data, len);
@ -838,7 +840,7 @@ static void cvalue_print(struct ios *f, value_t v)
void *fptr = *(void **)data; void *fptr = *(void **)data;
label = (value_t)ptrhash_get(&reverse_dlsym_lookup_table, cv); label = (value_t)ptrhash_get(&reverse_dlsym_lookup_table, cv);
if (label == (value_t)HT_NOTFOUND) { if (label == (value_t)HT_NOTFOUND) {
cur_column += pr.column +=
ios_printf(f, "#<builtin @#x%08zx>", (size_t)(builtin_t)fptr); ios_printf(f, "#<builtin @#x%08zx>", (size_t)(builtin_t)fptr);
} else { } else {
if (!print_readably) { if (!print_readably) {
@ -886,11 +888,11 @@ void fl_print(struct ios *f, value_t v)
if (print_width < 20) if (print_width < 20)
print_width = 20; print_width = 20;
cur_level = 0; pr.level = 0;
cycle_used_labels = 0; pr.cycle_labels = 0;
if (print_readably) if (print_readably)
print_traverse(v); print_traverse(v);
cur_line = cur_column = 0; pr.line = pr.column = 0;
fl_print_child(f, v); fl_print_child(f, v);
@ -901,6 +903,6 @@ void fl_print(struct ios *f, value_t v)
if ((iscons(v) || isvector(v) || isfunction(v) || iscvalue(v)) && if ((iscons(v) || isvector(v) || isfunction(v) || iscvalue(v)) &&
!fl_isstring(v) && v != FL_T && v != FL_F && v != FL_NIL) { !fl_isstring(v) && v != FL_T && v != FL_F && v != FL_NIL) {
htable_reset(&cycle_visited_pairs, 32); htable_reset(&pr.cycle_traversed, 32);
} }
} }