#include "ao_lisp.h"
-#if 0
-#define DBG(...) printf (__VA_ARGS__)
-#else
-#define DBG(...)
-#endif
-
struct ao_lisp_funcs {
- void (*print)(ao_poly);
- void (*patom)(ao_poly);
+ void (*write)(ao_poly);
+ void (*display)(ao_poly);
};
static const struct ao_lisp_funcs ao_lisp_funcs[AO_LISP_NUM_TYPE] = {
[AO_LISP_CONS] = {
- .print = ao_lisp_cons_print,
- .patom = ao_lisp_cons_patom,
+ .write = ao_lisp_cons_write,
+ .display = ao_lisp_cons_display,
},
[AO_LISP_STRING] = {
- .print = ao_lisp_string_print,
- .patom = ao_lisp_string_patom,
+ .write = ao_lisp_string_write,
+ .display = ao_lisp_string_display,
},
[AO_LISP_INT] = {
- .print = ao_lisp_int_print,
- .patom = ao_lisp_int_print,
+ .write = ao_lisp_int_write,
+ .display = ao_lisp_int_write,
},
[AO_LISP_ATOM] = {
- .print = ao_lisp_atom_print,
- .patom = ao_lisp_atom_print,
+ .write = ao_lisp_atom_write,
+ .display = ao_lisp_atom_write,
},
[AO_LISP_BUILTIN] = {
- .print = ao_lisp_builtin_print,
- .patom = ao_lisp_builtin_print,
+ .write = ao_lisp_builtin_write,
+ .display = ao_lisp_builtin_write,
},
[AO_LISP_FRAME] = {
- .print = ao_lisp_frame_print,
- .patom = ao_lisp_frame_print,
+ .write = ao_lisp_frame_write,
+ .display = ao_lisp_frame_write,
},
[AO_LISP_LAMBDA] = {
- .print = ao_lisp_lambda_print,
- .patom = ao_lisp_lambda_print,
+ .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,
},
};
}
void
-ao_lisp_poly_print(ao_poly p)
+ao_lisp_poly_write(ao_poly p)
{
const struct ao_lisp_funcs *f = funcs(p);
- if (f && f->print)
- f->print(p);
+ if (f && f->write)
+ f->write(p);
}
void
-ao_lisp_poly_patom(ao_poly p)
+ao_lisp_poly_display(ao_poly p)
{
const struct ao_lisp_funcs *f = funcs(p);
- if (f && f->patom)
- f->patom(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;
+}