X-Git-Url: https://git.gag.com/?a=blobdiff_plain;ds=sidebyside;f=src%2Flisp%2Fao_lisp_prim.c;h=bfd75ae3b629a1ad9e235277418f3d528e73d6a1;hb=0ee44c8e4bf5dabe6a97bf76b366c8b767c387f8;hp=38dcb9616aa8f028fc3b35b4c361fdae697ad5c5;hpb=d2408e72d1e0d3459918601712b09860ab17e200;p=fw%2Faltos

diff --git a/src/lisp/ao_lisp_prim.c b/src/lisp/ao_lisp_prim.c
index 38dcb961..bfd75ae3 100644
--- a/src/lisp/ao_lisp_prim.c
+++ b/src/lisp/ao_lisp_prim.c
@@ -14,52 +14,73 @@
 
 #include "ao_lisp.h"
 
-static void (*const ao_lisp_print_funcs[AO_LISP_NUM_TYPE])(ao_poly) = {
-	[AO_LISP_CONS] = ao_lisp_cons_print,
-	[AO_LISP_STRING] = ao_lisp_string_print,
-	[AO_LISP_INT] = ao_lisp_int_print,
-	[AO_LISP_ATOM] = ao_lisp_atom_print,
-	[AO_LISP_BUILTIN] = ao_lisp_builtin_print
+#if 0
+#define DBG(...) printf (__VA_ARGS__)
+#else
+#define DBG(...)
+#endif
+
+struct ao_lisp_funcs {
+	void (*print)(ao_poly);
+	void (*patom)(ao_poly);
 };
 
-ao_poly
-ao_lisp_poly_print(ao_poly p)
+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)
 {
-	void (*print)(ao_poly) = ao_lisp_print_funcs[ao_lisp_poly_type(p)];
-	if (print)
-		print(p);
-	return p;
+	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_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_poly
-ao_lisp_poly_move(ao_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);
 }
+