altos/lisp: re-use small frames

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

diff --git a/src/lisp/ao_lisp_atom.c b/src/lisp/ao_lisp_atom.c
index 41ba97f5d7f5b674948d9da00324cdb0d97f5f10..8c9e8ed11346ae5062c3f601ba38b6406f91233d 100644 (file)
--- a/src/lisp/ao_lisp_atom.c
+++ b/src/lisp/ao_lisp_atom.c
@@ -38,7 +38,7 @@ static void atom_mark(void *addr)
                atom = ao_lisp_poly_atom(atom->next);
                if (!atom)
                        break;
-               if (ao_lisp_mark_memory(atom, atom_size(atom)))
+               if (ao_lisp_mark_memory(&ao_lisp_atom_type, atom))
                        break;
        }
 }
@@ -46,11 +46,19 @@ static void atom_mark(void *addr)
 static void atom_move(void *addr)
 {
        struct ao_lisp_atom     *atom = addr;
+       int                     ret;
 
        for (;;) {
-               if (ao_lisp_poly_move(&atom->next, 0))
+               struct ao_lisp_atom *next = ao_lisp_poly_atom(atom->next);
+
+               if (!next)
                        break;
-               atom = ao_lisp_poly_atom(atom->next);
+               ret = ao_lisp_move_memory(&ao_lisp_atom_type, (void **) &next);
+               if (next != ao_lisp_poly_atom(atom->next))
+                       atom->next = ao_lisp_atom_poly(next);
+               if (ret)
+                       break;
+               atom = next;
        }
 }
 
@@ -58,6 +66,7 @@ const struct ao_lisp_type ao_lisp_atom_type = {
        .mark = atom_mark,
        .size = atom_size,
        .move = atom_move,
+       .name = "atom"
 };
 
 struct ao_lisp_atom    *ao_lisp_atoms;
@@ -77,32 +86,29 @@ ao_lisp_atom_intern(char *name)
                        return atom;
        }
 #endif
+       ao_lisp_string_stash(0, name);
        atom = ao_lisp_alloc(name_size(name));
+       name = ao_lisp_string_fetch(0);
        if (atom) {
                atom->type = AO_LISP_ATOM;
                atom->next = ao_lisp_atom_poly(ao_lisp_atoms);
-               if (!ao_lisp_atoms)
-                       ao_lisp_root_add(&ao_lisp_atom_type, &ao_lisp_atoms);
                ao_lisp_atoms = atom;
                strcpy(atom->name, name);
        }
        return atom;
 }
 
-static struct ao_lisp_frame    *ao_lisp_frame_global;
-struct ao_lisp_frame           *ao_lisp_frame_current;
+struct ao_lisp_frame   *ao_lisp_frame_global;
+struct ao_lisp_frame   *ao_lisp_frame_current;
 
 static void
 ao_lisp_atom_init(void)
 {
-       if (!ao_lisp_frame_global) {
+       if (!ao_lisp_frame_global)
                ao_lisp_frame_global = ao_lisp_frame_new(0);
-               ao_lisp_root_add(&ao_lisp_frame_type, &ao_lisp_frame_global);
-               ao_lisp_root_add(&ao_lisp_frame_type, &ao_lisp_frame_current);
-       }
 }
 
-static ao_poly *
+ao_poly *
 ao_lisp_atom_ref(struct ao_lisp_frame *frame, ao_poly atom)
 {
        ao_poly *ref;
@@ -111,7 +117,7 @@ ao_lisp_atom_ref(struct ao_lisp_frame *frame, ao_poly atom)
                ref = ao_lisp_frame_ref(frame, atom);
                if (ref)
                        return ref;
-               frame = ao_lisp_poly_frame(frame->next);
+               frame = ao_lisp_poly_frame(frame->prev);
        }
        if (ao_lisp_frame_global) {
                ref = ao_lisp_frame_ref(ao_lisp_frame_global, atom);
@@ -134,7 +140,7 @@ ao_lisp_atom_get(ao_poly atom)
 #endif
        if (ref)
                return *ref;
-       return AO_LISP_NIL;
+       return ao_lisp_error(AO_LISP_UNDEFINED, "undefined atom %s", ao_lisp_poly_atom(atom)->name);
 }
 
 ao_poly
@@ -147,7 +153,7 @@ ao_lisp_atom_set(ao_poly atom, ao_poly val)
        if (ref)
                *ref = val;
        else
-               ao_lisp_frame_global = ao_lisp_frame_add(ao_lisp_frame_global, atom, val);
+               ao_lisp_frame_add(&ao_lisp_frame_global, atom, val);
        return val;
 }