#include "ao_lisp.h"
static inline int
-frame_num_size(int num)
+frame_vals_num_size(int num)
{
- return sizeof (struct ao_lisp_frame) + num * sizeof (struct ao_lisp_val);
+ return sizeof (struct ao_lisp_frame_vals) + num * sizeof (struct ao_lisp_val);
}
+static int
+frame_vals_size(void *addr)
+{
+ struct ao_lisp_frame_vals *vals = addr;
+ return frame_vals_num_size(vals->size);
+}
+
+static void
+frame_vals_mark(void *addr)
+{
+ struct ao_lisp_frame_vals *vals = addr;
+ int f;
+
+ for (f = 0; f < vals->size; f++) {
+ struct ao_lisp_val *v = &vals->vals[f];
+
+ ao_lisp_poly_mark(v->val, 0);
+ MDBG_MOVE("frame mark atom %s %d val %d at %d ",
+ ao_lisp_poly_atom(v->atom)->name,
+ MDBG_OFFSET(ao_lisp_ref(v->atom)),
+ MDBG_OFFSET(ao_lisp_ref(v->val)), f);
+ MDBG_DO(ao_lisp_poly_write(v->val));
+ MDBG_DO(printf("\n"));
+ }
+}
+
+static void
+frame_vals_move(void *addr)
+{
+ struct ao_lisp_frame_vals *vals = addr;
+ int f;
+
+ for (f = 0; f < vals->size; f++) {
+ struct ao_lisp_val *v = &vals->vals[f];
+
+ ao_lisp_poly_move(&v->atom, 0);
+ ao_lisp_poly_move(&v->val, 0);
+ MDBG_MOVE("frame move atom %s %d val %d at %d\n",
+ ao_lisp_poly_atom(v->atom)->name,
+ MDBG_OFFSET(ao_lisp_ref(v->atom)),
+ MDBG_OFFSET(ao_lisp_ref(v->val)), f);
+ }
+}
+
+const struct ao_lisp_type ao_lisp_frame_vals_type = {
+ .mark = frame_vals_mark,
+ .size = frame_vals_size,
+ .move = frame_vals_move,
+ .name = "frame_vals"
+};
+
static int
frame_size(void *addr)
{
- struct ao_lisp_frame *frame = addr;
- return frame_num_size(frame->num);
+ (void) addr;
+ return sizeof (struct ao_lisp_frame);
}
static void
frame_mark(void *addr)
{
struct ao_lisp_frame *frame = addr;
- int f;
for (;;) {
MDBG_MOVE("frame mark %d\n", MDBG_OFFSET(frame));
if (!AO_LISP_IS_POOL(frame))
break;
- for (f = 0; f < frame->num; f++) {
- struct ao_lisp_val *v = &frame->vals[f];
-
- ao_lisp_poly_mark(v->val, 0);
- MDBG_MOVE("frame mark atom %s %d val %d at %d\n",
- ao_lisp_poly_atom(v->atom)->name,
- MDBG_OFFSET(ao_lisp_ref(v->atom)),
- MDBG_OFFSET(ao_lisp_ref(v->val)), f);
- }
- frame = ao_lisp_poly_frame(frame->next);
+ ao_lisp_poly_mark(frame->vals, 0);
+ frame = ao_lisp_poly_frame(frame->prev);
MDBG_MOVE("frame next %d\n", MDBG_OFFSET(frame));
if (!frame)
break;
frame_move(void *addr)
{
struct ao_lisp_frame *frame = addr;
- int f;
for (;;) {
- struct ao_lisp_frame *next;
+ struct ao_lisp_frame *prev;
int ret;
MDBG_MOVE("frame move %d\n", MDBG_OFFSET(frame));
if (!AO_LISP_IS_POOL(frame))
break;
- for (f = 0; f < frame->num; f++) {
- struct ao_lisp_val *v = &frame->vals[f];
-
- ao_lisp_poly_move(&v->atom, 0);
- ao_lisp_poly_move(&v->val, 0);
- MDBG_MOVE("frame move atom %s %d val %d at %d\n",
- ao_lisp_poly_atom(v->atom)->name,
- MDBG_OFFSET(ao_lisp_ref(v->atom)),
- MDBG_OFFSET(ao_lisp_ref(v->val)), f);
- }
- next = ao_lisp_poly_frame(frame->next);
- if (!next)
+ ao_lisp_poly_move(&frame->vals, 0);
+ prev = ao_lisp_poly_frame(frame->prev);
+ if (!prev)
break;
- ret = ao_lisp_move_memory(&ao_lisp_frame_type, (void **) &next);
- if (next != ao_lisp_poly_frame(frame->next)) {
- MDBG_MOVE("frame next moved from %d to %d\n",
- MDBG_OFFSET(ao_lisp_poly_frame(frame->next)),
- MDBG_OFFSET(next));
- frame->next = ao_lisp_frame_poly(next);
+ ret = ao_lisp_move_memory(&ao_lisp_frame_type, (void **) &prev);
+ if (prev != ao_lisp_poly_frame(frame->prev)) {
+ MDBG_MOVE("frame prev moved from %d to %d\n",
+ MDBG_OFFSET(ao_lisp_poly_frame(frame->prev)),
+ MDBG_OFFSET(prev));
+ frame->prev = ao_lisp_frame_poly(prev);
}
if (ret)
break;
- frame = next;
+ frame = prev;
}
}
};
void
-ao_lisp_frame_print(ao_poly p)
+ao_lisp_frame_write(ao_poly p)
{
- struct ao_lisp_frame *frame = ao_lisp_poly_frame(p);
- int f;
+ struct ao_lisp_frame *frame = ao_lisp_poly_frame(p);
+ struct ao_lisp_frame_vals *vals = ao_lisp_poly_frame_vals(frame->vals);
+ int f;
printf ("{");
if (frame) {
- for (f = 0; f < frame->num; f++) {
- if (f != 0)
- printf(", ");
- ao_lisp_poly_print(frame->vals[f].atom);
- printf(" = ");
- ao_lisp_poly_print(frame->vals[f].val);
+ if (frame->type & AO_LISP_FRAME_PRINT)
+ printf("recurse...");
+ else {
+ frame->type |= AO_LISP_FRAME_PRINT;
+ for (f = 0; f < frame->num; f++) {
+ if (f != 0)
+ printf(", ");
+ ao_lisp_poly_write(vals->vals[f].atom);
+ printf(" = ");
+ ao_lisp_poly_write(vals->vals[f].val);
+ }
+ if (frame->prev)
+ ao_lisp_poly_write(frame->prev);
+ frame->type &= ~AO_LISP_FRAME_PRINT;
}
- if (frame->next)
- ao_lisp_poly_print(frame->next);
}
printf("}");
}
+static int
+ao_lisp_frame_find(struct ao_lisp_frame *frame, int top, ao_poly atom)
+{
+ struct ao_lisp_frame_vals *vals = ao_lisp_poly_frame_vals(frame->vals);
+ int l = 0;
+ int r = top - 1;
+
+ while (l <= r) {
+ int m = (l + r) >> 1;
+ if (vals->vals[m].atom < atom)
+ l = m + 1;
+ else
+ r = m - 1;
+ }
+ return l;
+}
+
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;
+ struct ao_lisp_frame_vals *vals = ao_lisp_poly_frame_vals(frame->vals);
+ int l = ao_lisp_frame_find(frame, frame->num, atom);
+
+ if (l >= frame->num)
+ return NULL;
+
+ if (vals->vals[l].atom != atom)
+ return NULL;
+ return &vals->vals[l].val;
}
-int
-ao_lisp_frame_set(struct ao_lisp_frame *frame, ao_poly atom, ao_poly val)
+struct ao_lisp_frame *ao_lisp_frame_free_list[AO_LISP_FRAME_FREE];
+
+static struct ao_lisp_frame_vals *
+ao_lisp_frame_vals_new(int num)
{
- while (frame) {
- if (!AO_LISP_IS_CONST(frame)) {
- ao_poly *ref = ao_lisp_frame_ref(frame, atom);
- if (ref) {
- *ref = val;
- return 1;
- }
- }
- frame = ao_lisp_poly_frame(frame->next);
+ struct ao_lisp_frame_vals *vals;
+
+ vals = ao_lisp_alloc(frame_vals_num_size(num));
+ if (!vals)
+ return NULL;
+ vals->type = AO_LISP_FRAME_VALS;
+ vals->size = num;
+ memset(vals->vals, '\0', num * sizeof (struct ao_lisp_val));
+ return vals;
+}
+
+struct ao_lisp_frame *
+ao_lisp_frame_new(int num)
+{
+ struct ao_lisp_frame *frame;
+ struct ao_lisp_frame_vals *vals;
+
+ if (num < AO_LISP_FRAME_FREE && (frame = ao_lisp_frame_free_list[num])) {
+ ao_lisp_frame_free_list[num] = ao_lisp_poly_frame(frame->prev);
+ vals = ao_lisp_poly_frame_vals(frame->vals);
+ } else {
+ frame = ao_lisp_alloc(sizeof (struct ao_lisp_frame));
+ if (!frame)
+ return NULL;
+ frame->type = AO_LISP_FRAME;
+ frame->num = 0;
+ frame->prev = AO_LISP_NIL;
+ frame->vals = AO_LISP_NIL;
+ ao_lisp_poly_stash(0, ao_lisp_frame_poly(frame));
+ vals = ao_lisp_frame_vals_new(num);
+ frame = ao_lisp_poly_frame(ao_lisp_poly_fetch(0));
+ frame->vals = ao_lisp_frame_vals_poly(vals);
+ frame->num = num;
}
- return 0;
+ frame->prev = AO_LISP_NIL;
+ return frame;
}
ao_poly
-ao_lisp_frame_get(struct ao_lisp_frame *frame, ao_poly atom)
+ao_lisp_frame_mark(struct ao_lisp_frame *frame)
{
- 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;
+ if (!frame)
+ return AO_LISP_NIL;
+ frame->type |= AO_LISP_FRAME_MARK;
+ return ao_lisp_frame_poly(frame);
}
-struct ao_lisp_frame *
-ao_lisp_frame_new(int num)
+void
+ao_lisp_frame_free(struct ao_lisp_frame *frame)
{
- struct ao_lisp_frame *frame = ao_lisp_alloc(frame_num_size(num));
+ if (frame && !ao_lisp_frame_marked(frame)) {
+ int num = frame->num;
+ if (num < AO_LISP_FRAME_FREE) {
+ struct ao_lisp_frame_vals *vals;
- if (!frame)
- return NULL;
- frame->type = AO_LISP_FRAME;
- frame->num = num;
- frame->next = AO_LISP_NIL;
- memset(frame->vals, '\0', num * sizeof (struct ao_lisp_val));
- return frame;
+ vals = ao_lisp_poly_frame_vals(frame->vals);
+ memset(vals->vals, '\0', vals->size * sizeof (struct ao_lisp_val));
+ frame->prev = ao_lisp_frame_poly(ao_lisp_frame_free_list[num]);
+ ao_lisp_frame_free_list[num] = frame;
+ }
+ }
}
static struct ao_lisp_frame *
-ao_lisp_frame_realloc(struct ao_lisp_frame **frame_ref, int new_num)
+ao_lisp_frame_realloc(struct ao_lisp_frame *frame, int new_num)
{
- struct ao_lisp_frame *frame = *frame_ref;
- struct ao_lisp_frame *new;
- int copy;
+ struct ao_lisp_frame_vals *vals;
+ struct ao_lisp_frame_vals *new_vals;
+ int copy;
if (new_num == frame->num)
return frame;
- new = ao_lisp_frame_new(new_num);
- if (!new)
+ ao_lisp_frame_stash(0, frame);
+ new_vals = ao_lisp_frame_vals_new(new_num);
+ if (!new_vals)
return NULL;
- /*
- * Re-fetch the frame as it may have moved
- * during the allocation
- */
- frame = *frame_ref;
+ frame = ao_lisp_frame_fetch(0);
+ vals = ao_lisp_poly_frame_vals(frame->vals);
copy = new_num;
if (copy > frame->num)
copy = frame->num;
- memcpy(new->vals, frame->vals, copy * sizeof (struct ao_lisp_val));
- new->next = frame->next;
- return new;
+ memcpy(new_vals->vals, vals->vals, copy * sizeof (struct ao_lisp_val));
+ frame->vals = ao_lisp_frame_vals_poly(new_vals);
+ frame->num = new_num;
+ return frame;
}
-int
-ao_lisp_frame_add(struct ao_lisp_frame **frame_ref, ao_poly atom, ao_poly val)
+void
+ao_lisp_frame_bind(struct ao_lisp_frame *frame, int num, ao_poly atom, ao_poly val)
+{
+ struct ao_lisp_frame_vals *vals = ao_lisp_poly_frame_vals(frame->vals);
+ int l = ao_lisp_frame_find(frame, num, atom);
+
+ memmove(&vals->vals[l+1],
+ &vals->vals[l],
+ (num - l) * sizeof (struct ao_lisp_val));
+ vals->vals[l].atom = atom;
+ vals->vals[l].val = val;
+}
+
+ao_poly
+ao_lisp_frame_add(struct ao_lisp_frame *frame, ao_poly atom, ao_poly val)
{
- struct ao_lisp_frame *frame = *frame_ref;
ao_poly *ref = frame ? ao_lisp_frame_ref(frame, atom) : NULL;
if (!ref) {
int f;
ao_lisp_poly_stash(0, atom);
ao_lisp_poly_stash(1, val);
- if (frame) {
- f = frame->num;
- frame = ao_lisp_frame_realloc(frame_ref, f + 1);
- } else {
- f = 0;
- frame = ao_lisp_frame_new(1);
- }
+ f = frame->num;
+ frame = ao_lisp_frame_realloc(frame, f + 1);
+ if (!frame)
+ return AO_LISP_NIL;
atom = ao_lisp_poly_fetch(0);
val = ao_lisp_poly_fetch(1);
- if (!frame)
- return 0;
- *frame_ref = frame;
- frame->vals[f].atom = atom;
- ref = &frame->vals[f].val;
- }
- *ref = val;
- return 1;
+ ao_lisp_frame_bind(frame, frame->num - 1, atom, val);
+ } else
+ *ref = val;
+ return val;
+}
+
+struct ao_lisp_frame *ao_lisp_frame_global;
+struct ao_lisp_frame *ao_lisp_frame_current;
+
+void
+ao_lisp_frame_init(void)
+{
+ if (!ao_lisp_frame_global)
+ ao_lisp_frame_global = ao_lisp_frame_new(0);
}