altos/lisp: Separate out values from atoms

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

diff --git a/src/lisp/ao_lisp_frame.c b/src/lisp/ao_lisp_frame.c
new file mode 100644 (file)
index 0000000..5aa50f6
--- /dev/null
+++ b/src/lisp/ao_lisp_frame.c
@@ -0,0 +1,191 @@
+/*
+ * Copyright © 2016 Keith Packard <keithp@keithp.com>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * General Public License for more details.
+ */
+
+#include "ao_lisp.h"
+
+#if 0
+#define DBG(...)       printf(__VA_ARGS__)
+#else
+#define DBG(...)
+#endif
+
+static inline int
+frame_num_size(int num)
+{
+       return sizeof (struct ao_lisp_frame) + num * sizeof (struct ao_lisp_val);
+}
+
+static int
+frame_size(void *addr)
+{
+       struct ao_lisp_frame    *frame = addr;
+       return frame_num_size(frame->num);
+}
+
+#define OFFSET(a)      ((uint8_t *) (ao_lisp_ref(a)) - ao_lisp_const)
+
+static void
+frame_mark(void *addr)
+{
+       struct ao_lisp_frame    *frame = addr;
+       int                     f;
+
+       for (;;) {
+               if (frame->readonly)
+                       break;
+               for (f = 0; f < frame->num; f++) {
+                       struct ao_lisp_val      *v = &frame->vals[f];
+
+                       ao_lisp_poly_mark(v->atom);
+                       ao_lisp_poly_mark(v->val);
+                       DBG ("\tframe mark atom %s %d val %d at %d\n", ao_lisp_poly_atom(v->atom)->name, OFFSET(v->atom), OFFSET(v->val), f);
+               }
+               frame = ao_lisp_poly_frame(frame->next);
+               if (!frame)
+                       break;
+               if (ao_lisp_mark_memory(frame, frame_size(frame)))
+                       break;
+       }
+}
+
+static void
+frame_move(void *addr)
+{
+       struct ao_lisp_frame    *frame = addr;
+       int                     f;
+
+       for (;;) {
+               struct ao_lisp_frame    *next;
+               if (frame->readonly)
+                       break;
+               for (f = 0; f < frame->num; f++) {
+                       struct ao_lisp_val      *v = &frame->vals[f];
+                       ao_poly                 t;
+
+                       t = ao_lisp_poly_move(v->atom);
+                       DBG("\t\tatom %s %d -> %d\n", ao_lisp_poly_atom(t)->name, OFFSET(v->atom), OFFSET(t));
+                       v->atom = t;
+                       t = ao_lisp_poly_move(v->val);
+                       DBG("\t\tval %d -> %d\n", OFFSET(v->val), OFFSET(t));
+                       v->val = t;
+               }
+               next = ao_lisp_poly_frame(frame->next);
+               if (!next)
+                       break;
+               next = ao_lisp_move_memory(next, frame_size(next));
+               frame->next = ao_lisp_frame_poly(next);
+               frame = next;
+       }
+}
+
+const struct ao_lisp_type ao_lisp_frame_type = {
+       .mark = frame_mark,
+       .size = frame_size,
+       .move = frame_move
+};
+
+static ao_poly *
+ao_lisp_frame_ref(struct ao_lisp_frame *frame, ao_poly atom)
+{
+       int f;
+       for (f = 0; f < frame->num; f++)
+               if (frame->vals[f].atom == atom)
+                       return &frame->vals[f].val;
+       return NULL;
+}
+
+int
+ao_lisp_frame_set(struct ao_lisp_frame *frame, ao_poly atom, ao_poly val)
+{
+       while (frame) {
+               if (!frame->readonly) {
+                       ao_poly *ref = ao_lisp_frame_ref(frame, atom);
+                       if (ref) {
+                               *ref = val;
+                               return 1;
+                       }
+               }
+               frame = ao_lisp_poly_frame(frame->next);
+       }
+       return 0;
+}
+
+ao_poly
+ao_lisp_frame_get(struct ao_lisp_frame *frame, ao_poly atom)
+{
+       while (frame) {
+               ao_poly *ref = ao_lisp_frame_ref(frame, atom);
+               if (ref)
+                       return *ref;
+               frame = ao_lisp_poly_frame(frame->next);
+       }
+       return AO_LISP_NIL;
+}
+
+struct ao_lisp_frame *
+ao_lisp_frame_new(int num, int readonly)
+{
+       struct ao_lisp_frame *frame = ao_lisp_alloc(frame_num_size(num));
+
+       if (!frame)
+               return NULL;
+       frame->num = num;
+       frame->readonly = readonly;
+       frame->next = AO_LISP_NIL;
+       memset(frame->vals, '\0', num * sizeof (struct ao_lisp_val));
+       return frame;
+}
+
+static struct ao_lisp_frame *
+ao_lisp_frame_realloc(struct ao_lisp_frame *frame, int new_num, int readonly)
+{
+       struct ao_lisp_frame    *new;
+       int                     copy;
+
+       if (new_num == frame->num)
+               return frame;
+       new = ao_lisp_frame_new(new_num, readonly);
+       if (!new)
+               return NULL;
+       copy = new_num;
+       if (copy > frame->num)
+               copy = frame->num;
+       memcpy(new->vals, frame->vals, copy * sizeof (struct ao_lisp_val));
+       if (frame)
+               new->next = frame->next;
+       return new;
+}
+
+struct ao_lisp_frame *
+ao_lisp_frame_add(struct ao_lisp_frame *frame, ao_poly atom, ao_poly val)
+{
+       ao_poly *ref = frame ? ao_lisp_frame_ref(frame, atom) : NULL;
+       if (!ref) {
+               int f;
+               if (frame) {
+                       f = frame->num;
+                       frame = ao_lisp_frame_realloc(frame, f + 1, frame->readonly);
+               } else {
+                       f = 0;
+                       frame = ao_lisp_frame_new(1, 0);
+               }
+               if (!frame)
+                       return NULL;
+               DBG ("add atom %s %d, val %d at %d\n", ao_lisp_poly_atom(atom)->name, OFFSET(atom), OFFSET(val), f);
+               frame->vals[f].atom = atom;
+               ref = &frame->vals[f].val;
+       }
+       *ref = val;
+       return frame;
+}