altos/lisp: add length, pack, unpack and flush

[fw/altos] / src / lisp / ao_lisp_cons.c

diff --git a/src/lisp/ao_lisp_cons.c b/src/lisp/ao_lisp_cons.c
index 65908e30a610ea0f2b706eb17e314a9db1b2866d..b75ffaa0d611ffa5427e7b191d4d68d996e74534 100644 (file)
--- a/src/lisp/ao_lisp_cons.c
+++ b/src/lisp/ao_lisp_cons.c
@@ -14,12 +14,14 @@
 
 #include "ao_lisp.h"
 
+#define OFFSET(a)      ((int) ((uint8_t *) (a) - ao_lisp_const))
+
 static void cons_mark(void *addr)
 {
        struct ao_lisp_cons     *cons = addr;
 
        for (;;) {
-               ao_lisp_poly_mark(cons->car);
+               ao_lisp_poly_mark(cons->car, 1);
                cons = ao_lisp_poly_cons(cons->cdr);
                if (!cons)
                        break;
@@ -38,15 +40,20 @@ static void cons_move(void *addr)
 {
        struct ao_lisp_cons     *cons = addr;
 
+       if (!cons)
+               return;
+
        for (;;) {
                struct ao_lisp_cons     *cdr;
+               int                     ret;
 
-               cons->car = ao_lisp_poly_move(cons->car);
+               (void) ao_lisp_poly_move(&cons->car, 1);
                cdr = ao_lisp_poly_cons(cons->cdr);
-               cdr = ao_lisp_move_memory(cdr, sizeof (struct ao_lisp_cons));
-               if (!cdr)
+               ret = ao_lisp_move_memory((void **) &cdr, sizeof (struct ao_lisp_cons));
+               if (cdr != ao_lisp_poly_cons(cons->cdr))
+                       cons->cdr = ao_lisp_cons_poly(cdr);
+               if (ret)
                        break;
-               cons->cdr = ao_lisp_cons_poly(cdr);
                cons = cdr;
        }
 }
@@ -60,7 +67,13 @@ const struct ao_lisp_type ao_lisp_cons_type = {
 struct ao_lisp_cons *
 ao_lisp_cons_cons(ao_poly car, struct ao_lisp_cons *cdr)
 {
-       struct ao_lisp_cons     *cons = ao_lisp_alloc(sizeof (struct ao_lisp_cons));
+       struct ao_lisp_cons     *cons;
+
+       ao_lisp_root_add(&ao_lisp_cons_type, &cdr);
+       ao_lisp_root_poly_add(&car);
+       cons = ao_lisp_alloc(sizeof (struct ao_lisp_cons));
+       ao_lisp_root_clear(&car);
+       ao_lisp_root_clear(&cdr);
        if (!cons)
                return NULL;
        cons->car = car;
@@ -83,3 +96,25 @@ ao_lisp_cons_print(ao_poly c)
        }
        printf(")");
 }
+
+void
+ao_lisp_cons_patom(ao_poly c)
+{
+       struct ao_lisp_cons *cons = ao_lisp_poly_cons(c);
+
+       while (cons) {
+               ao_lisp_poly_patom(cons->car);
+               cons = ao_lisp_poly_cons(cons->cdr);
+       }
+}
+
+int
+ao_lisp_cons_length(struct ao_lisp_cons *cons)
+{
+       int     len = 0;
+       while (cons) {
+               len++;
+               cons = ao_lisp_poly_cons(cons->cdr);
+       }
+       return len;
+}