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 7d3ca68d6a7134914459791faaf283359264345a..b75ffaa0d611ffa5427e7b191d4d68d996e74534 100644 (file)
--- a/src/lisp/ao_lisp_cons.c
+++ b/src/lisp/ao_lisp_cons.c
@@ -21,7 +21,7 @@ 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;
@@ -44,10 +44,17 @@ static void cons_move(void *addr)
                return;
 
        for (;;) {
-               (void) ao_lisp_poly_move(&cons->car);
-               if (ao_lisp_poly_move(&cons->cdr))
+               struct ao_lisp_cons     *cdr;
+               int                     ret;
+
+               (void) ao_lisp_poly_move(&cons->car, 1);
+               cdr = ao_lisp_poly_cons(cons->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 = ao_lisp_poly_cons(cons->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;
@@ -94,3 +107,14 @@ ao_lisp_cons_patom(ao_poly c)
                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;
+}