diff options
Diffstat (limited to 'src/stdlib/pointers.c')
| -rw-r--r-- | src/stdlib/pointers.c | 123 |
1 files changed, 123 insertions, 0 deletions
diff --git a/src/stdlib/pointers.c b/src/stdlib/pointers.c new file mode 100644 index 00000000..76e882ec --- /dev/null +++ b/src/stdlib/pointers.c @@ -0,0 +1,123 @@ +// Type infos and methods for Pointer types +#include <ctype.h> +#include <err.h> +#include <gc.h> +#include <stdbool.h> +#include <stdint.h> +#include <stdlib.h> +#include <sys/param.h> + +#include "integers.h" +#include "metamethods.h" +#include "tables.h" +#include "text.h" +#include "types.h" +#include "util.h" + +public Text_t Pointer$as_text(const void *x, bool colorize, const TypeInfo_t *type) { + auto ptr_info = type->PointerInfo; + if (!x) { + Text_t typename = generic_as_text(NULL, false, ptr_info.pointed); + if (colorize) + return Text$concat(Text("\x1b[34;1m"), Text$from_str(ptr_info.sigil), typename, Text("\x1b[m")); + else + return Text$concat(Text$from_str(ptr_info.sigil), typename); + } + const void *ptr = *(const void**)x; + if (!ptr) { + Text_t typename = generic_as_text(NULL, false, ptr_info.pointed); + if (colorize) + return Text$concat(Text("\x1b[34;1m!"), typename, Text("\x1b[m")); + else + return Text$concat(Text("!"), typename); + } + + static const void *root = NULL; + static Table_t pending = {}; + bool top_level = (root == NULL); + + // Check for recursive references, so if `x.foo = x`, then it prints as + // `@Foo{foo=@~1}` instead of overflowing the stack: + if (top_level) { + root = ptr; + } else if (ptr == root) { + return Text$format(colorize ? "\x1b[34;1m%s~1\x1b[m" : "%s~1", ptr_info.sigil); + } else { + TypeInfo_t rec_table = *Table$info(type, &Int64$info); + int64_t *id = Table$get(pending, x, &rec_table); + if (id) + return Text$format(colorize ? "\x1b[34;1m%s~%ld\x1b[m" : "%s~%ld", ptr_info.sigil, *id); + int64_t next_id = pending.entries.length + 2; + Table$set(&pending, x, &next_id, &rec_table); + } + + Text_t pointed = generic_as_text(ptr, colorize, ptr_info.pointed); + + if (top_level) { + pending = (Table_t){}; // Restore + root = NULL; + } + + Text_t text; + if (colorize) + text = Text$concat(Text("\x1b[34;1m"), Text$from_str(ptr_info.sigil), Text("\x1b[m"), pointed); + else + text = Text$concat(Text$from_str(ptr_info.sigil), pointed); + return text; +} + +PUREFUNC public int32_t Pointer$compare(const void *x, const void *y, const TypeInfo_t*) { + const void *xp = *(const void**)x, *yp = *(const void**)y; + return (xp > yp) - (xp < yp); +} + +PUREFUNC public bool Pointer$equal(const void *x, const void *y, const TypeInfo_t*) { + const void *xp = *(const void**)x, *yp = *(const void**)y; + return xp == yp; +} + +PUREFUNC public bool Pointer$is_none(const void *x, const TypeInfo_t*) +{ + return *(void**)x == NULL; +} + +public void Pointer$serialize(const void *obj, FILE *out, Table_t *pointers, const TypeInfo_t *type) +{ + void *ptr = *(void**)obj; + assert(ptr != NULL); + + const TypeInfo_t ptr_to_int_table = {.size=sizeof(Table_t), .align=__alignof__(Table_t), + .tag=TableInfo, .TableInfo.key=type, .TableInfo.value=&Int64$info}; + + int64_t *id_ptr = Table$get(*pointers, &ptr, &ptr_to_int_table); + int64_t id; + if (id_ptr) { + id = *id_ptr; + } else { + id = pointers->entries.length + 1; + Table$set(pointers, &ptr, &id, &ptr_to_int_table); + } + + Int64$serialize(&id, out, pointers, &Int64$info); + + if (!id_ptr) + _serialize(ptr, out, pointers, type->PointerInfo.pointed); +} + +public void Pointer$deserialize(FILE *in, void *outval, Array_t *pointers, const TypeInfo_t *type) +{ + int64_t id = 0; + Int64$deserialize(in, &id, pointers, &Int64$info); + assert(id != 0); + + if (id > pointers->length) { + void *obj = GC_MALLOC((size_t)type->PointerInfo.pointed->size); + Array$insert(pointers, &obj, I(0), sizeof(void*)); + _deserialize(in, obj, pointers, type->PointerInfo.pointed); + *(void**)outval = obj; + } else { + *(void**)outval = *(void**)(pointers->data + (id-1)*pointers->stride); + } +} + +// vim: ts=4 sw=0 et cino=L2,l1,(0,W4,m1,\:0 |