X-Git-Url: https://git.gag.com/?a=blobdiff_plain;f=src%2Flisp%2Fao_lisp_prim.c;h=7f02505df7d3c7b2f95b5a780c2a2df1824fa8ce;hb=286d07d83bd7ff361e5a904c151a75e5a9c8b071;hp=ccfd2be4e6dfb88fd9b6098eb9471c73cb904600;hpb=56d46ceaa1413415f25e47e81036426132f99924;p=fw%2Faltos

diff --git a/src/lisp/ao_lisp_prim.c b/src/lisp/ao_lisp_prim.c
index ccfd2be4..7f02505d 100644
--- a/src/lisp/ao_lisp_prim.c
+++ b/src/lisp/ao_lisp_prim.c
@@ -14,58 +14,77 @@
 
 #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
+
+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
+};
+
+ao_poly
+ao_lisp_poly_print(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;
+	void (*print)(ao_poly) = ao_lisp_print_funcs[ao_lisp_poly_type(p)];
+	if (print)
+		print(p);
+	return p;
 }
 
-void
-ao_lisp_poly_mark(ao_lisp_poly p)
+static const struct ao_lisp_type const *ao_lisp_types[AO_LISP_NUM_TYPE] = {
+	[AO_LISP_CONS] = &ao_lisp_cons_type,
+	[AO_LISP_INT] = NULL,
+	[AO_LISP_STRING] = &ao_lisp_string_type,
+	[AO_LISP_OTHER] = (void *) 0x1,
+	[AO_LISP_ATOM] = &ao_lisp_atom_type,
+	[AO_LISP_BUILTIN] = &ao_lisp_builtin_type,
+	[AO_LISP_FRAME] = &ao_lisp_frame_type,
+};
+
+int
+ao_lisp_poly_mark(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_type *lisp_type = ao_lisp_types[ao_lisp_poly_type(p)];
+	if (lisp_type)
+		return ao_lisp_mark(lisp_type, ao_lisp_ref(p));
+	return 1;
 }
 
-ao_lisp_poly
-ao_lisp_poly_move(ao_lisp_poly p)
+int
+ao_lisp_poly_move(ao_poly *ref)
 {
-	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;
+	uint8_t				type;
+	ao_poly				p = *ref;
+	const struct ao_lisp_type	*lisp_type;
+	int				ret;
+	void				*addr;
+
+	if (!p)
+		return 1;
+
+	type = p & AO_LISP_TYPE_MASK;
+	if (type == AO_LISP_OTHER)
+		type = ao_lisp_other_type(ao_lisp_move_map(ao_lisp_poly_other(p)));
+
+	if (type >= AO_LISP_NUM_TYPE)
+		abort();
+
+	lisp_type = ao_lisp_types[type];
+	if (!lisp_type)
+		return 1;
+	addr = ao_lisp_ref(p);
+	ret = ao_lisp_move(lisp_type, &addr);
+	if (addr != ao_lisp_ref(p)) {
+		ao_poly np = ao_lisp_poly(addr, p & AO_LISP_TYPE_MASK);
+		DBG("poly %d moved %04x -> %04x\n",
+		    type, p, np);
+		*ref = np;
 	}
-	return p;
+	return ret;
 }