X-Git-Url: https://git.gag.com/?a=blobdiff_plain;f=src%2Flisp%2Fao_lisp_poly.c;h=d14f41513c1320837c70813b0c043adf3688996a;hb=8c19778d8b56aafa048ddf9654c40b32bd8c64b0;hp=1855d945d6a920abbe35fabc0de74ddd17a5747d;hpb=56d46ceaa1413415f25e47e81036426132f99924;p=fw%2Faltos

diff --git a/src/lisp/ao_lisp_poly.c b/src/lisp/ao_lisp_poly.c
index 1855d945..d14f4151 100644
--- a/src/lisp/ao_lisp_poly.c
+++ b/src/lisp/ao_lisp_poly.c
@@ -14,119 +14,105 @@
 
 #include "ao_lisp.h"
 
-enum math_op { math_plus, math_minus, math_times, math_divide, math_mod };
-
-ao_lisp_poly
-ao_lisp_math(struct ao_lisp_cons *cons, enum math_op op)
-{
-	ao_lisp_poly	ret = AO_LISP_NIL;
-
-	while (cons) {
-		ao_lisp_poly	car = cons->car;
-		uint8_t		rt = ao_lisp_poly_type(ret);
-		uint8_t		ct = ao_lisp_poly_type(car);
-
-		cons = cons->cdr;
-
-		if (rt == AO_LISP_NIL)
-			ret = car;
+struct ao_lisp_funcs {
+	void (*write)(ao_poly);
+	void (*display)(ao_poly);
+};
 
-		else if (rt == AO_LISP_INT && ct == AO_LISP_INT) {
-			int	r = ao_lisp_poly_int(ret);
-			int	c = ao_lisp_poly_int(car);
+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,
+	},
+};
 
-			switch(op) {
-			case math_plus:
-				r += c;
-				break;
-			case math_minus:
-				r -= c;
-				break;
-			case math_times:
-				r *= c;
-				break;
-			case math_divide:
-				if (c == 0)
-					return AO_LISP_NIL;
-				r /= c;
-				break;
-			case math_mod:
-				if (c == 0)
-					return AO_LISP_NIL;
-				r %= c;
-				break;
-			}
-			ret = ao_lisp_int_poly(r);
-		}
+static const struct ao_lisp_funcs *
+funcs(ao_poly p)
+{
+	uint8_t	type = ao_lisp_poly_type(p);
 
-		else if (rt == AO_LISP_STRING && ct == AO_LISP_STRING && op == math_plus)
-			ret = ao_lisp_string_poly(ao_lisp_string_cat(ao_lisp_poly_string(ret),
-								     ao_lisp_poly_string(car)));
-		else {
-			/* XXX exception */
-			return AO_LISP_NIL;
-		}
-	}
-	return ret;
+	if (type < AO_LISP_NUM_TYPE)
+		return &ao_lisp_funcs[type];
+	return NULL;
 }
 
-ao_lisp_poly
-ao_lisp_plus(struct ao_lisp_cons *cons)
+void
+ao_lisp_poly_write(ao_poly p)
 {
-	return ao_lisp_math(cons, math_plus);
-}
+	const struct ao_lisp_funcs *f = funcs(p);
 
-ao_lisp_poly
-ao_lisp_minus(struct ao_lisp_cons *cons)
-{
-	return ao_lisp_math(cons, math_minus);
+	if (f && f->write)
+		f->write(p);
 }
 
-ao_lisp_poly
-ao_lisp_times(struct ao_lisp_cons *cons)
+void
+ao_lisp_poly_display(ao_poly p)
 {
-	return ao_lisp_math(cons, math_times);
-}
+	const struct ao_lisp_funcs *f = funcs(p);
 
-ao_lisp_poly
-ao_lisp_divide(struct ao_lisp_cons *cons)
-{
-	return ao_lisp_math(cons, math_divide);
+	if (f && f->display)
+		f->display(p);
 }
 
-ao_lisp_poly
-ao_lisp_mod(struct ao_lisp_cons *cons)
-{
-	return ao_lisp_math(cons, math_mod);
+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);
 }
 
-static const struct ao_lisp_builtin builtin_plus = {
-	.type = AO_LISP_BUILTIN,
-	.func = ao_lisp_plus,
-	.name = "+"
-};
-
-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_builtin builtin_minus = {
-	.type = AO_LISP_BUILTIN,
-	.func = ao_lisp_minus
-};
-
-static const struct ao_lisp_builtin builtin_times = {
-	.type = AO_LISP_BUILTIN,
-	.func = ao_lisp_times
-};
-
-*/
-
-const struct ao_lisp_atom const *ao_lisp_builtins[] = {
-	&atom_plus,
-	0
-};
+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;
+}