altos/lisp: Switch to scheme formal syntax for varargs

[fw/altos] / src / lisp / ao_lisp_poly.c

diff --git a/src/lisp/ao_lisp_poly.c b/src/lisp/ao_lisp_poly.c
index c6ca0a975f0d1077a133a3de7b0087dc4dac53f0..d14f41513c1320837c70813b0c043adf3688996a 100644 (file)
--- a/src/lisp/ao_lisp_poly.c
+++ b/src/lisp/ao_lisp_poly.c
@@ -14,34 +14,105 @@
 
 #include "ao_lisp.h"
 
-/*
-
-static const struct ao_lisp_builtin builtin_plus = {
-       .type = AO_LISP_BUILTIN,
-       .func = ao_lisp_plus,
-       .name = "+"
+struct ao_lisp_funcs {
+       void (*write)(ao_poly);
+       void (*display)(ao_poly);
 };
 
-static const struct ao_lisp_atom atom_plus = {
-       .type = AO_LISP_ATOM,
-       .val = AO_LISP_OTHER_POLY(&builtin_plus),
-       .next = AO_LISP_ATOM_CONST,
-       .name = "plus"
+static const struct ao_lisp_funcs ao_lisp_funcs[AO_LISP_NUM_TYPE] = {
+       [AO_LISP_CONS] = {
+               .write = ao_lisp_cons_write,
+               .display = ao_lisp_cons_display,
+       },
+       [AO_LISP_STRING] = {
+               .write = ao_lisp_string_write,
+               .display = ao_lisp_string_display,
+       },
+       [AO_LISP_INT] = {
+               .write = ao_lisp_int_write,
+               .display = ao_lisp_int_write,
+       },
+       [AO_LISP_ATOM] = {
+               .write = ao_lisp_atom_write,
+               .display = ao_lisp_atom_write,
+       },
+       [AO_LISP_BUILTIN] = {
+               .write = ao_lisp_builtin_write,
+               .display = ao_lisp_builtin_write,
+       },
+       [AO_LISP_FRAME] = {
+               .write = ao_lisp_frame_write,
+               .display = ao_lisp_frame_write,
+       },
+       [AO_LISP_FRAME_VALS] = {
+               .write = NULL,
+               .display = NULL,
+       },
+       [AO_LISP_LAMBDA] = {
+               .write = ao_lisp_lambda_write,
+               .display = ao_lisp_lambda_write,
+       },
+       [AO_LISP_STACK] = {
+               .write = ao_lisp_stack_write,
+               .display = ao_lisp_stack_write,
+       },
+       [AO_LISP_BOOL] = {
+               .write = ao_lisp_bool_write,
+               .display = ao_lisp_bool_write,
+       },
+       [AO_LISP_BIGINT] = {
+               .write = ao_lisp_bigint_write,
+               .display = ao_lisp_bigint_write,
+       },
+       [AO_LISP_FLOAT] = {
+               .write = ao_lisp_float_write,
+               .display = ao_lisp_float_write,
+       },
 };
 
-static const struct ao_lisp_builtin builtin_minus = {
-       .type = AO_LISP_BUILTIN,
-       .func = ao_lisp_minus
-};
+static const struct ao_lisp_funcs *
+funcs(ao_poly p)
+{
+       uint8_t type = ao_lisp_poly_type(p);
 
-static const struct ao_lisp_builtin builtin_times = {
-       .type = AO_LISP_BUILTIN,
-       .func = ao_lisp_times
-};
+       if (type < AO_LISP_NUM_TYPE)
+               return &ao_lisp_funcs[type];
+       return NULL;
+}
 
+void
+ao_lisp_poly_write(ao_poly p)
+{
+       const struct ao_lisp_funcs *f = funcs(p);
 
-const struct ao_lisp_atom const *ao_lisp_builtins[] = {
-       &atom_plus,
-       0
-};
-*/
+       if (f && f->write)
+               f->write(p);
+}
+
+void
+ao_lisp_poly_display(ao_poly p)
+{
+       const struct ao_lisp_funcs *f = funcs(p);
+
+       if (f && f->display)
+               f->display(p);
+}
+
+void *
+ao_lisp_ref(ao_poly poly) {
+       if (poly == AO_LISP_NIL)
+               return NULL;
+       if (poly & AO_LISP_CONST)
+               return (void *) (ao_lisp_const + (poly & AO_LISP_REF_MASK) - 4);
+       return (void *) (ao_lisp_pool + (poly & AO_LISP_REF_MASK) - 4);
+}
+
+ao_poly
+ao_lisp_poly(const void *addr, ao_poly type) {
+       const uint8_t   *a = addr;
+       if (a == NULL)
+               return AO_LISP_NIL;
+       if (AO_LISP_IS_CONST(a))
+               return AO_LISP_CONST | (a - ao_lisp_const + 4) | type;
+       return (a - ao_lisp_pool + 4) | type;
+}