altos/scheme: Use memory manager mark code to note recursive print
[fw/altos] / src / scheme / ao_scheme.h
index db4417e5249ed3ffabac3622f0edaf0580d44648..cc7f8f1d5600d4a2c4565c70b057228261d8daa2 100644 (file)
@@ -23,6 +23,7 @@
 
 #include <stdint.h>
 #include <string.h>
+#include <stdbool.h>
 #define AO_SCHEME_BUILTIN_FEATURES
 #include "ao_scheme_builtin.h"
 #undef AO_SCHEME_BUILTIN_FEATURES
@@ -43,6 +44,10 @@ struct ao_scheme_os_save {
        uint16_t        const_checksum_inv;
 };
 
+#ifndef AO_SCHEME_POOL_TOTAL
+#error Must define AO_SCHEME_POOL_TOTAL for AO_SCHEME_SAVE
+#endif
+
 #define AO_SCHEME_POOL_EXTRA   (sizeof(struct ao_scheme_os_save))
 #define AO_SCHEME_POOL ((int) (AO_SCHEME_POOL_TOTAL - AO_SCHEME_POOL_EXTRA))
 
@@ -63,7 +68,7 @@ extern uint8_t ao_scheme_const[AO_SCHEME_POOL_CONST] __attribute__((aligned(4)))
 #define ao_scheme_pool ao_scheme_const
 #define AO_SCHEME_POOL AO_SCHEME_POOL_CONST
 
-#define _atom(n) ao_scheme_atom_poly(ao_scheme_atom_intern(n))
+#define _atom(n) ao_scheme_atom_poly(ao_scheme_atom_intern((char *) n))
 #define _bool(v) ao_scheme_bool_poly(ao_scheme_bool_get(v))
 
 #define _ao_scheme_bool_true   _bool(1)
@@ -78,7 +83,7 @@ extern uint8_t ao_scheme_const[AO_SCHEME_POOL_CONST] __attribute__((aligned(4)))
 #else
 #include "ao_scheme_const.h"
 #ifndef AO_SCHEME_POOL
-#define AO_SCHEME_POOL 3072
+#error Must define AO_SCHEME_POOL
 #endif
 #ifndef AO_SCHEME_POOL_EXTRA
 #define AO_SCHEME_POOL_EXTRA 0
@@ -89,7 +94,7 @@ extern uint8_t                ao_scheme_pool[AO_SCHEME_POOL + AO_SCHEME_POOL_EXTRA] __attribut
 /* Primitive types */
 #define AO_SCHEME_CONS         0
 #define AO_SCHEME_INT          1
-#define AO_SCHEME_STRING       2
+#define AO_SCHEME_BIGINT       2
 #define AO_SCHEME_OTHER                3
 
 #define AO_SCHEME_TYPE_MASK    0x0003
@@ -105,17 +110,12 @@ extern uint8_t            ao_scheme_pool[AO_SCHEME_POOL + AO_SCHEME_POOL_EXTRA] __attribut
 #define AO_SCHEME_LAMBDA       8
 #define AO_SCHEME_STACK                9
 #define AO_SCHEME_BOOL         10
-#ifdef AO_SCHEME_FEATURE_BIGINT
-#define AO_SCHEME_BIGINT       11
-#define _AO_SCHEME_BIGINT      AO_SCHEME_BIGINT
-#else
-#define _AO_SCHEME_BIGINT      AO_SCHEME_BOOL
-#endif
+#define AO_SCHEME_STRING       11
 #ifdef AO_SCHEME_FEATURE_FLOAT
-#define AO_SCHEME_FLOAT                (_AO_SCHEME_BIGINT + 1)
+#define AO_SCHEME_FLOAT                12
 #define _AO_SCHEME_FLOAT       AO_SCHEME_FLOAT
 #else
-#define _AO_SCHEME_FLOAT       _AO_SCHEME_BIGINT
+#define _AO_SCHEME_FLOAT       12
 #endif
 #ifdef AO_SCHEME_FEATURE_VECTOR
 #define AO_SCHEME_VECTOR       13
@@ -150,6 +150,7 @@ ao_scheme_is_const(ao_poly poly) {
 #define AO_SCHEME_IS_CONST(a)  (ao_scheme_const <= ((uint8_t *) (a)) && ((uint8_t *) (a)) < ao_scheme_const + AO_SCHEME_POOL_CONST)
 #define AO_SCHEME_IS_POOL(a)   (ao_scheme_pool <= ((uint8_t *) (a)) && ((uint8_t *) (a)) < ao_scheme_pool + AO_SCHEME_POOL)
 #define AO_SCHEME_IS_INT(p)    (ao_scheme_poly_base_type(p) == AO_SCHEME_INT)
+#define AO_SCHEME_IS_CONS(p)   (ao_scheme_poly_base_type(p) == AO_SCHEME_CONS)
 
 void *
 ao_scheme_ref(ao_poly poly);
@@ -176,6 +177,11 @@ struct ao_scheme_atom {
        char            name[];
 };
 
+struct ao_scheme_string {
+       uint8_t         type;
+       char            val[];
+};
+
 struct ao_scheme_val {
        ao_poly         atom;
        ao_poly         val;
@@ -223,38 +229,16 @@ struct ao_scheme_vector {
 #define AO_SCHEME_MAX_INT      ((1 << (15 - AO_SCHEME_TYPE_SHIFT)) - 1)
 
 #ifdef AO_SCHEME_FEATURE_BIGINT
+
 struct ao_scheme_bigint {
        uint32_t                value;
 };
 
-#define AO_SCHEME_MIN_BIGINT   (-(1 << 24))
-#define AO_SCHEME_MAX_BIGINT   ((1 << 24) - 1)
+#define AO_SCHEME_MIN_BIGINT   INT32_MIN
+#define AO_SCHEME_MAX_BIGINT   INT32_MAX
 
-#if __BYTE_ORDER == __LITTLE_ENDIAN
-
-static inline uint32_t
-ao_scheme_int_bigint(int32_t i) {
-       return AO_SCHEME_BIGINT | (i << 8);
-}
-static inline int32_t
-ao_scheme_bigint_int(uint32_t bi) {
-       return (int32_t) bi >> 8;
-}
-#else
-static inline uint32_t
-ao_scheme_int_bigint(int32_t i) {
-       return (uint32_t) (i & 0xffffff) | (AO_SCHEME_BIGINT << 24);
-}
-static inlint int32_t
-ao_scheme_bigint_int(uint32_t bi) {
-       return (int32_t) (bi << 8) >> 8;
-}
-
-#endif /* __BYTE_ORDER */
 #endif /* AO_SCHEME_FEATURE_BIGINT */
 
-#define AO_SCHEME_NOT_INTEGER  0x7fffffff
-
 /* Set on type when the frame escapes the lambda */
 #define AO_SCHEME_FRAME_MARK   0x80
 #define AO_SCHEME_FRAME_PRINT  0x40
@@ -471,20 +455,20 @@ ao_scheme_poly_bigint(ao_poly poly)
 static inline ao_poly
 ao_scheme_bigint_poly(struct ao_scheme_bigint *bi)
 {
-       return ao_scheme_poly(bi, AO_SCHEME_OTHER);
+       return ao_scheme_poly(bi, AO_SCHEME_BIGINT);
 }
 #endif /* AO_SCHEME_FEATURE_BIGINT */
 
-static inline char *
+static inline struct ao_scheme_string *
 ao_scheme_poly_string(ao_poly poly)
 {
        return ao_scheme_ref(poly);
 }
 
 static inline ao_poly
-ao_scheme_string_poly(char *s)
+ao_scheme_string_poly(struct ao_scheme_string *s)
 {
-       return ao_scheme_poly(s, AO_SCHEME_STRING);
+       return ao_scheme_poly(s, AO_SCHEME_OTHER);
 }
 
 static inline struct ao_scheme_atom *
@@ -562,25 +546,29 @@ extern uint64_t ao_scheme_loops[2];
 
 /* returns 1 if the object was already marked */
 int
-ao_scheme_mark(const struct ao_scheme_type *type, void *addr);
+ao_scheme_mark_memory(const struct ao_scheme_type *type, void *addr);
 
-/* returns 1 if the object was already marked */
+/* returns 1 if the object was already moved */
 int
-ao_scheme_mark_memory(const struct ao_scheme_type *type, void *addr);
+ao_scheme_move_memory(const struct ao_scheme_type *type, void **ref);
 
 void *
-ao_scheme_move_map(void *addr);
+ao_scheme_alloc(int size);
 
-/* returns 1 if the object was already moved */
+/* Marks an object as being printed, returns 1 if it was already marked */
 int
-ao_scheme_move(const struct ao_scheme_type *type, void **ref);
+ao_scheme_print_mark_addr(void *addr);
 
-/* returns 1 if the object was already moved */
 int
-ao_scheme_move_memory(const struct ao_scheme_type *type, void **ref);
+ao_scheme_print_mark_poly(ao_poly poly);
 
-void *
-ao_scheme_alloc(int size);
+/* Notes that printing has started */
+void
+ao_scheme_print_start(void);
+
+/* Notes that printing has ended */
+void
+ao_scheme_print_stop(void);
 
 #define AO_SCHEME_COLLECT_FULL         1
 #define AO_SCHEME_COLLECT_INCREMENTAL  0
@@ -606,9 +594,9 @@ ao_poly
 ao_scheme_poly_fetch(int id);
 
 void
-ao_scheme_string_stash(int id, char *string);
+ao_scheme_string_stash(int id, struct ao_scheme_string *string);
 
-char *
+struct ao_scheme_string *
 ao_scheme_string_fetch(int id);
 
 static inline void
@@ -635,7 +623,7 @@ void
 ao_scheme_bool_write(ao_poly v);
 
 #ifdef AO_SCHEME_MAKE_CONST
-struct ao_scheme_bool  *ao_scheme_true, *ao_scheme_false;
+extern struct ao_scheme_bool   *ao_scheme_true, *ao_scheme_false;
 
 struct ao_scheme_bool *
 ao_scheme_bool_get(uint8_t value);
@@ -674,17 +662,23 @@ ao_scheme_cons_copy(struct ao_scheme_cons *cons);
 /* string */
 extern const struct ao_scheme_type ao_scheme_string_type;
 
-char *
-ao_scheme_string_copy(char *a);
+struct ao_scheme_string *
+ao_scheme_string_copy(struct ao_scheme_string *a);
 
-char *
-ao_scheme_string_cat(char *a, char *b);
+struct ao_scheme_string *
+ao_scheme_string_make(char *a);
+
+struct ao_scheme_string *
+ao_scheme_atom_to_string(struct ao_scheme_atom *a);
+
+struct ao_scheme_string *
+ao_scheme_string_cat(struct ao_scheme_string *a, struct ao_scheme_string *b);
 
 ao_poly
 ao_scheme_string_pack(struct ao_scheme_cons *cons);
 
 ao_poly
-ao_scheme_string_unpack(char *a);
+ao_scheme_string_unpack(struct ao_scheme_string *a);
 
 void
 ao_scheme_string_write(ao_poly s);
@@ -702,6 +696,9 @@ extern struct ao_scheme_frame       *ao_scheme_frame_current;
 void
 ao_scheme_atom_write(ao_poly a);
 
+struct ao_scheme_atom *
+ao_scheme_string_to_atom(struct ao_scheme_string *string);
+
 struct ao_scheme_atom *
 ao_scheme_atom_intern(char *name);
 
@@ -723,7 +720,7 @@ ao_scheme_int_write(ao_poly i);
 
 #ifdef AO_SCHEME_FEATURE_BIGINT
 int32_t
-ao_scheme_poly_integer(ao_poly p);
+ao_scheme_poly_integer(ao_poly p, bool *fail);
 
 ao_poly
 ao_scheme_integer_poly(int32_t i);
@@ -741,7 +738,7 @@ extern const struct ao_scheme_type  ao_scheme_bigint_type;
 
 #else
 
-#define ao_scheme_poly_integer ao_scheme_poly_int
+#define ao_scheme_poly_integer(a,b) ao_scheme_poly_int(a)
 #define ao_scheme_integer_poly ao_scheme_int_poly
 
 static inline int
@@ -825,12 +822,8 @@ ao_scheme_number_typep(uint8_t t)
 {
        return ao_scheme_integer_typep(t) || (t == AO_SCHEME_FLOAT);
 }
-
-float
-ao_scheme_poly_number(ao_poly p);
 #else
 #define ao_scheme_number_typep ao_scheme_integer_typep
-#define ao_scheme_poly_number ao_scheme_poly_integer
 #endif
 
 /* builtin */
@@ -940,19 +933,19 @@ ao_scheme_stack_eval(void);
 /* error */
 
 void
-ao_scheme_vprintf(char *format, va_list args);
+ao_scheme_vprintf(const char *format, va_list args);
 
 void
-ao_scheme_printf(char *format, ...);
+ao_scheme_printf(const char *format, ...);
 
 void
-ao_scheme_error_poly(char *name, ao_poly poly, ao_poly last);
+ao_scheme_error_poly(const char *name, ao_poly poly, ao_poly last);
 
 void
-ao_scheme_error_frame(int indent, char *name, struct ao_scheme_frame *frame);
+ao_scheme_error_frame(int indent, const char *name, struct ao_scheme_frame *frame);
 
 ao_poly
-ao_scheme_error(int error, char *format, ...);
+ao_scheme_error(int error, const char *format, ...);
 
 /* builtins */