From f66f8ad7119207b99f00ea2ea389550ee65db5b3 Mon Sep 17 00:00:00 2001
From: Bruce Hill <bruce@bruce-hill.com>
Date: Fri, 29 Nov 2024 18:09:12 -0500
Subject: Add serialization and deserialization

---
 stdlib/integers.c | 77 ++++++++++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 74 insertions(+), 3 deletions(-)

(limited to 'stdlib/integers.c')

diff --git a/stdlib/integers.c b/stdlib/integers.c
index 6c00527d..b40657bc 100644
--- a/stdlib/integers.c
+++ b/stdlib/integers.c
@@ -1,4 +1,6 @@
 // Integer type infos and methods
+#include <stdio.h> // Must be before gmp.h
+
 #include <ctype.h>
 #include <gc.h>
 #include <gmp.h>
@@ -367,6 +369,21 @@ static bool Int$is_none(const void *i, const TypeInfo_t*)
     return ((Int_t*)i)->small == 0;
 }
 
+static void Int$serialize(const void *obj, FILE *out, Table_t*, const TypeInfo_t*)
+{
+    mpz_t n;
+    mpz_init_set_int(n, *(Int_t*)obj);
+    mpz_out_raw(out, n);
+}
+
+static void Int$deserialize(FILE *in, void *obj, Array_t*, const TypeInfo_t*)
+{
+    mpz_t n;
+    mpz_init(n);
+    mpz_inp_raw(n, in);
+    *(Int_t*)obj = Int$from_mpz(n);
+}
+
 public const TypeInfo_t Int$info = {
     .size=sizeof(Int_t),
     .align=__alignof__(Int_t),
@@ -376,9 +393,61 @@ public const TypeInfo_t Int$info = {
         .hash=Int$hash,
         .as_text=Int$as_text,
         .is_none=Int$is_none,
+        .serialize=Int$serialize,
+        .deserialize=Int$deserialize,
     },
 };
 
+public void Int64$serialize(const void *obj, FILE *out, Table_t*, const TypeInfo_t*)
+{
+    int64_t i = *(int64_t*)obj;
+    uint64_t z = (uint64_t)((i << 1) ^ (i >> 63)); // Zigzag encode
+    while (z >= 0x80) {
+        fputc((uint8_t)(z | 0x80), out);
+        z >>= 7;
+    }
+    fputc((uint8_t)z, out);
+}
+
+public void Int64$deserialize(FILE *in, void *outval, Array_t*, const TypeInfo_t*)
+{
+    uint64_t z = 0;
+    for(size_t shift = 0; ; shift += 7) {
+        uint8_t byte = (uint8_t)fgetc(in);
+        z |= ((uint64_t)(byte & 0x7F)) << shift;
+        if ((byte & 0x80) == 0) break;
+    }
+    *(int64_t*)outval = (int64_t)((z >> 1) ^ -(z & 1)); // Zigzag decode
+}
+
+public void Int32$serialize(const void *obj, FILE *out, Table_t*, const TypeInfo_t*) 
+{
+    int32_t i = *(int32_t*)obj;
+    uint32_t z = (uint32_t)((i << 1) ^ (i >> 31)); // Zigzag encode
+    while (z >= 0x80) {
+        fputc((uint8_t)(z | 0x80), out);
+        z >>= 7;
+    }
+    fputc((uint8_t)z, out);
+}
+
+public void Int32$deserialize(FILE *in, void *outval, Array_t*, const TypeInfo_t*)
+{
+    uint32_t z = 0;
+    for(size_t shift = 0; ; shift += 7) {
+        uint8_t byte = (uint8_t)fgetc(in);
+        z |= ((uint32_t)(byte & 0x7F)) << shift;
+        if ((byte & 0x80) == 0) break;
+    }
+    *(int32_t*)outval = (int32_t)((z >> 1) ^ -(z & 1)); // Zigzag decode
+}
+
+// The space savings for smaller ints are not worth having:
+#define Int16$serialize NULL
+#define Int16$deserialize NULL
+#define Int8$serialize NULL
+#define Int8$deserialize NULL
+
 #define DEFINE_INT_TYPE(c_type, KindOfInt, fmt, min_val, max_val, to_attr)\
     public Text_t KindOfInt ## $as_text(const void *i, bool colorize, const TypeInfo_t*) { \
         if (!i) return Text(#KindOfInt); \
@@ -416,12 +485,12 @@ public const TypeInfo_t Int$info = {
     } \
     public PUREFUNC Optional ## KindOfInt ## _t KindOfInt ## $parse(Text_t text) { \
         OptionalInt_t full_int = Int$parse(text); \
-        if (full_int.small == 0) return (Optional ## KindOfInt ## _t){.is_null=true}; \
+        if (full_int.small == 0) return (Optional ## KindOfInt ## _t){.is_none=true}; \
         if (Int$compare_value(full_int, I(min_val)) < 0) { \
-            return (Optional ## KindOfInt ## _t){.is_null=true}; \
+            return (Optional ## KindOfInt ## _t){.is_none=true}; \
         } \
         if (Int$compare_value(full_int, I(max_val)) > 0) { \
-            return (Optional ## KindOfInt ## _t){.is_null=true}; \
+            return (Optional ## KindOfInt ## _t){.is_none=true}; \
         } \
         return (Optional ## KindOfInt ## _t){.i=Int_to_ ## KindOfInt(full_int, true)}; \
     } \
@@ -433,6 +502,8 @@ public const TypeInfo_t Int$info = {
         .metamethods={ \
             .compare=KindOfInt##$compare, \
             .as_text=KindOfInt##$as_text, \
+            .serialize=KindOfInt##$serialize, \
+            .deserialize=KindOfInt##$deserialize, \
         }, \
     };
 
-- 
cgit v1.2.3