vb: add seq and string?

[jackhill/mal.git] / c / types.h

#ifndef __MAL_TYPES__
#define __MAL_TYPES__

#include <glib.h>


struct MalVal; // pre-declare


// Env (implentation in env.c)

typedef struct Env {
    struct Env *outer;
    GHashTable *table;
} Env;

Env *new_env(Env *outer, struct MalVal* binds, struct MalVal *exprs);
Env *env_find(Env *env, struct MalVal *key);
struct MalVal *env_get(Env *env, struct MalVal *key);
Env *env_set(Env *env, struct MalVal *key, struct MalVal *val);


// Utility functiosn
void g_hash_table_print(GHashTable *hash_table);
GHashTable *g_hash_table_copy(GHashTable *src_table);


// Errors/exceptions

extern struct MalVal *mal_error;
void _error(const char *fmt, ...);

#define abort(format, ...) \
    { _error(format, ##__VA_ARGS__); return NULL; }

#define assert(test, format, ...) \
    if (!(test)) { \
        _error(format, ##__VA_ARGS__); \
        return NULL; \
    }

#define assert_type(mv, typ, format, ...) \
    if (!(mv->type & (typ))) { \
        _error(format, ##__VA_ARGS__); \
        return NULL; \
    }


typedef enum {
    MAL_NIL = 1,
    MAL_TRUE = 2,
    MAL_FALSE = 4,
    MAL_INTEGER = 8,
    MAL_FLOAT = 16,
    MAL_SYMBOL = 32,
    MAL_STRING = 64,
    MAL_LIST = 128,
    MAL_VECTOR = 256,
    MAL_HASH_MAP = 512,
    MAL_ATOM = 1024,
    MAL_FUNCTION_C = 2048,
    MAL_FUNCTION_MAL = 4096,
} MalType;

typedef struct MalVal {
    MalType type;
    struct MalVal *metadata;
    union {
        gint64 intnum;
        gdouble floatnum;
        char *string;
        GArray *array;
        GHashTable *hash_table;
        struct MalVal *atom_val;
        void *(*f0) ();
        void *(*f1) (void*);
        void *(*f2) (void*,void*);
        void *(*f3) (void*,void*,void*);
        void *(*f4) (void*,void*,void*,void*);
        void *(*f5) (void*,void*,void*,void*,void*);
        void *(*f6) (void*,void*,void*,void*,void*,void*);
        void *(*f7) (void*,void*,void*,void*,void*,void*,void*);
        void *(*f8) (void*,void*,void*,void*,void*,void*,void*,void*);
        void *(*f9) (void*,void*,void*,void*,void*,void*,void*,void*,void*);
        void *(*f10)(void*,void*,void*,void*,void*,void*,void*,void*,void*,void*);
        void *(*f11)(void*,void*,void*,void*,void*,void*,void*,void*,void*,void*,
                     void*);
        void *(*f12)(void*,void*,void*,void*,void*,void*,void*,void*,void*,void*,
                     void*,void*);
        void *(*f13)(void*,void*,void*,void*,void*,void*,void*,void*,void*,void*,
                     void*,void*,void*);
        void *(*f14)(void*,void*,void*,void*,void*,void*,void*,void*,void*,void*,
                     void*,void*,void*,void*);
        void *(*f15)(void*,void*,void*,void*,void*,void*,void*,void*,void*,void*,
                     void*,void*,void*,void*,void*);
        void *(*f16)(void*,void*,void*,void*,void*,void*,void*,void*,void*,void*,
                     void*,void*,void*,void*,void*,void*);
        void *(*f17)(void*,void*,void*,void*,void*,void*,void*,void*,void*,void*,
                     void*,void*,void*,void*,void*,void*,void*);
        void *(*f18)(void*,void*,void*,void*,void*,void*,void*,void*,void*,void*,
                     void*,void*,void*,void*,void*,void*,void*,void*);
        void *(*f19)(void*,void*,void*,void*,void*,void*,void*,void*,void*,void*,
                     void*,void*,void*,void*,void*,void*,void*,void*,void*);
        void *(*f20)(void*,void*,void*,void*,void*,void*,void*,void*,void*,void*,
                     void*,void*,void*,void*,void*,void*,void*,void*,void*,void*);
        struct {
            struct MalVal *(*evaluator)(struct MalVal *, Env *);
            struct MalVal *args;
            struct MalVal *body;
            struct Env    *env;
        } func;
    } val;
    int func_arg_cnt;
    int ismacro;
} MalVal;

// Constants

extern MalVal mal_nil;
extern MalVal mal_true;
extern MalVal mal_false;


// Declare functions used internally (by other C code).
// Mal visible functions are "exported" in types_ns

MalVal *malval_new(MalType type, MalVal *metadata);
void malval_free(MalVal *mv);
MalVal *malval_new_integer(gint64 val);
MalVal *malval_new_float(gdouble val);
MalVal *malval_new_string(char *val);
MalVal *malval_new_symbol(char *val);
MalVal *malval_new_keyword(char *val);
MalVal *malval_new_list(MalType type, GArray *val);
MalVal *malval_new_hash_map(GHashTable *val);
MalVal *malval_new_atom(MalVal *val);
MalVal *malval_new_function(void *(*func)(void *), int arg_cnt);

// Numbers
#define WRAP_INTEGER_OP(name, op) \
    static MalVal *int_ ## name(MalVal *a, MalVal *b)     { \
        return malval_new_integer(a->val.intnum op b->val.intnum); \
    }
#define WRAP_INTEGER_CMP_OP(name, op) \
    static MalVal *int_ ## name(MalVal *a, MalVal *b)     { \
        return a->val.intnum op b->val.intnum ? &mal_true : &mal_false; \
    }

// Collections
MalVal *_listX(int count, ...);
MalVal *_list(MalVal *args);
MalVal *_vector(MalVal *args);
MalVal *_hash_map(MalVal *args);
MalVal *_assoc_BANG(MalVal* hm, MalVal *args);
MalVal *_dissoc_BANG(MalVal* hm, MalVal *args);

MalVal *_apply(MalVal *f, MalVal *el);

char *_pr_str(MalVal *args, int print_readably);

MalVal *_slice(MalVal *seq, int start, int end);
MalVal *_nth(MalVal *seq, int idx);
MalVal *_first(MalVal *seq);
MalVal *_rest(MalVal *seq);
MalVal *_last(MalVal *seq);
int _count(MalVal *obj);

int _atom_Q(MalVal *exp);
int _sequential_Q(MalVal *seq);
int _list_Q(MalVal *seq);
int _vector_Q(MalVal *seq);
int _hash_map_Q(MalVal *seq);
int _equal_Q(MalVal *a, MalVal *b);

MalVal *_map2(MalVal *(*func)(void*, void*), MalVal *lst, void *arg2);

#endif