8d607372b7543a817f0998b1c246576f04363ecb
[fw/altos] / src / lisp / ao_lisp_cons.c
1 /*
2  * Copyright © 2016 Keith Packard <keithp@keithp.com>
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation, either version 2 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful, but
10  * WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * General Public License for more details.
13  */
14
15 #include "ao_lisp.h"
16
17 static void cons_mark(void *addr)
18 {
19         struct ao_lisp_cons     *cons = addr;
20
21         for (;;) {
22                 ao_poly cdr = cons->cdr;
23
24                 ao_lisp_poly_mark(cons->car, 1);
25                 if (!cdr)
26                         break;
27                 if (ao_lisp_poly_type(cdr) != AO_LISP_CONS) {
28                         ao_lisp_poly_mark(cdr, 1);
29                         break;
30                 }
31                 cons = ao_lisp_poly_cons(cdr);
32                 if (ao_lisp_mark_memory(&ao_lisp_cons_type, cons))
33                         break;
34         }
35 }
36
37 static int cons_size(void *addr)
38 {
39         (void) addr;
40         return sizeof (struct ao_lisp_cons);
41 }
42
43 static void cons_move(void *addr)
44 {
45         struct ao_lisp_cons     *cons = addr;
46
47         if (!cons)
48                 return;
49
50         for (;;) {
51                 ao_poly                 cdr;
52                 struct ao_lisp_cons     *c;
53                 int     ret;
54
55                 MDBG_MOVE("cons_move start %d (%d, %d)\n",
56                           MDBG_OFFSET(cons), MDBG_OFFSET(ao_lisp_ref(cons->car)), MDBG_OFFSET(ao_lisp_ref(cons->cdr)));
57                 (void) ao_lisp_poly_move(&cons->car, 1);
58                 cdr = cons->cdr;
59                 if (!cdr)
60                         break;
61                 if (ao_lisp_poly_type(cdr) != AO_LISP_CONS) {
62                         (void) ao_lisp_poly_move(&cons->cdr, 1);
63                         break;
64                 }
65                 c = ao_lisp_poly_cons(cdr);
66                 ret = ao_lisp_move_memory(&ao_lisp_cons_type, (void **) &c);
67                 if (c != ao_lisp_poly_cons(cons->cdr))
68                         cons->cdr = ao_lisp_cons_poly(c);
69                 MDBG_MOVE("cons_move end %d (%d, %d)\n",
70                           MDBG_OFFSET(cons), MDBG_OFFSET(ao_lisp_ref(cons->car)), MDBG_OFFSET(ao_lisp_ref(cons->cdr)));
71                 if (ret)
72                         break;
73                 cons = c;
74         }
75 }
76
77 const struct ao_lisp_type ao_lisp_cons_type = {
78         .mark = cons_mark,
79         .size = cons_size,
80         .move = cons_move,
81         .name = "cons",
82 };
83
84 struct ao_lisp_cons *ao_lisp_cons_free_list;
85
86 struct ao_lisp_cons *
87 ao_lisp_cons_cons(ao_poly car, ao_poly cdr)
88 {
89         struct ao_lisp_cons     *cons;
90
91         if (ao_lisp_cons_free_list) {
92                 cons = ao_lisp_cons_free_list;
93                 ao_lisp_cons_free_list = ao_lisp_poly_cons(cons->cdr);
94         } else {
95                 ao_lisp_poly_stash(0, car);
96                 ao_lisp_poly_stash(1, cdr);
97                 cons = ao_lisp_alloc(sizeof (struct ao_lisp_cons));
98                 car = ao_lisp_poly_fetch(0);
99                 cdr = ao_lisp_poly_fetch(1);
100                 if (!cons)
101                         return NULL;
102         }
103         cons->car = car;
104         cons->cdr = cdr;
105         return cons;
106 }
107
108 ao_poly
109 ao_lisp__cons(ao_poly car, ao_poly cdr)
110 {
111         return ao_lisp_cons_poly(ao_lisp_cons_cons(car, cdr));
112 }
113
114 void
115 ao_lisp_cons_free(struct ao_lisp_cons *cons)
116 {
117         while (cons) {
118                 ao_poly cdr = cons->cdr;
119                 cons->cdr = ao_lisp_cons_poly(ao_lisp_cons_free_list);
120                 ao_lisp_cons_free_list = cons;
121                 cons = ao_lisp_poly_cons(cdr);
122         }
123 }
124
125 void
126 ao_lisp_cons_print(ao_poly c)
127 {
128         struct ao_lisp_cons *cons = ao_lisp_poly_cons(c);
129         int     first = 1;
130         printf("(");
131         while (cons) {
132                 if (!first)
133                         printf(" ");
134                 ao_lisp_poly_print(cons->car);
135                 c = cons->cdr;
136                 if (ao_lisp_poly_type(c) == AO_LISP_CONS) {
137                         cons = ao_lisp_poly_cons(c);
138                         first = 0;
139                 } else {
140                         printf(" . ");
141                         ao_lisp_poly_print(c);
142                         cons = NULL;
143                 }
144         }
145         printf(")");
146 }
147
148 void
149 ao_lisp_cons_patom(ao_poly c)
150 {
151         struct ao_lisp_cons *cons = ao_lisp_poly_cons(c);
152
153         while (cons) {
154                 ao_lisp_poly_patom(cons->car);
155                 cons = ao_lisp_poly_cons(cons->cdr);
156         }
157 }
158
159 int
160 ao_lisp_cons_length(struct ao_lisp_cons *cons)
161 {
162         int     len = 0;
163         while (cons) {
164                 len++;
165                 cons = ao_lisp_poly_cons(cons->cdr);
166         }
167         return len;
168 }