#include "ao_lisp.h"
-ao_lisp_poly
-ao_lisp_poly_print(ao_lisp_poly p)
+#if 0
+#define DBG(...) printf (__VA_ARGS__)
+#else
+#define DBG(...)
+#endif
+
+struct ao_lisp_funcs {
+ void (*print)(ao_poly);
+ void (*patom)(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,
+ },
+ [AO_LISP_STRING] = {
+ .print = ao_lisp_string_print,
+ .patom = ao_lisp_string_patom,
+ },
+ [AO_LISP_INT] = {
+ .print = ao_lisp_int_print,
+ .patom = ao_lisp_int_print,
+ },
+ [AO_LISP_ATOM] = {
+ .print = ao_lisp_atom_print,
+ .patom = ao_lisp_atom_print,
+ },
+ [AO_LISP_BUILTIN] = {
+ .print = ao_lisp_builtin_print,
+ .patom = ao_lisp_builtin_print,
+ },
+ [AO_LISP_FRAME] = {
+ .print = ao_lisp_frame_print,
+ .patom = ao_lisp_frame_print,
+ },
+ [AO_LISP_LAMBDA] = {
+ .print = ao_lisp_lambda_print,
+ .patom = ao_lisp_lambda_print,
+ },
+};
+
+static const struct ao_lisp_funcs *
+funcs(ao_poly p)
{
- switch (ao_lisp_poly_type(p)) {
- case AO_LISP_CONS:
- ao_lisp_cons_print(ao_lisp_poly_cons(p));
- break;
- case AO_LISP_STRING:
- ao_lisp_string_print(ao_lisp_poly_string(p));
- break;
- case AO_LISP_INT:
- ao_lisp_int_print(ao_lisp_poly_int(p));
- break;
- case AO_LISP_ATOM:
- ao_lisp_atom_print(ao_lisp_poly_atom(p));
- break;
- case AO_LISP_BUILTIN:
- ao_lisp_builtin_print(ao_lisp_poly_builtin(p));
- break;
- }
- return AO_LISP_NIL;
+ uint8_t type = ao_lisp_poly_type(p);
+
+ if (type < AO_LISP_NUM_TYPE)
+ return &ao_lisp_funcs[type];
+ return NULL;
}
void
-ao_lisp_poly_mark(ao_lisp_poly p)
+ao_lisp_poly_print(ao_poly p)
{
- switch (ao_lisp_poly_type(p)) {
- case AO_LISP_CONS:
- ao_lisp_mark(&ao_lisp_cons_type, ao_lisp_poly_cons(p));
- break;
- case AO_LISP_STRING:
- ao_lisp_mark(&ao_lisp_string_type, ao_lisp_poly_string(p));
- break;
- case AO_LISP_ATOM:
- ao_lisp_mark(&ao_lisp_atom_type, ao_lisp_poly_atom(p));
- break;
- }
+ const struct ao_lisp_funcs *f = funcs(p);
+
+ if (f && f->print)
+ f->print(p);
}
-ao_lisp_poly
-ao_lisp_poly_move(ao_lisp_poly p)
+void
+ao_lisp_poly_patom(ao_poly p)
{
- switch (ao_lisp_poly_type(p)) {
- case AO_LISP_CONS:
- p = ao_lisp_cons_poly(ao_lisp_move(&ao_lisp_cons_type, ao_lisp_poly_cons(p)));
- break;
- case AO_LISP_STRING:
- p = ao_lisp_string_poly(ao_lisp_move(&ao_lisp_string_type, ao_lisp_poly_string(p)));
- break;
- case AO_LISP_ATOM:
- p = ao_lisp_atom_poly(ao_lisp_move(&ao_lisp_atom_type, ao_lisp_poly_atom(p)));
- break;
- }
- return p;
+ const struct ao_lisp_funcs *f = funcs(p);
+
+ if (f && f->patom)
+ f->patom(p);
}
+