129 lines
2.8 KiB
C
129 lines
2.8 KiB
C
#include <stdint.h>
|
|
#include <string.h>
|
|
#include <assert.h>
|
|
#include "runtime.h"
|
|
#include "../utils/defs.h"
|
|
|
|
static Object *select(Object *self, Object *key, Heap *heap, Error *err);
|
|
static int count(Object *self);
|
|
|
|
static Object *bin_print (Runtime *runtime, Object **argv, unsigned int argc, Error *error);
|
|
static Object *bin_count (Runtime *runtime, Object **argv, unsigned int argc, Error *error);
|
|
static Object *bin_assert(Runtime *runtime, Object **argv, unsigned int argc, Error *error);
|
|
|
|
typedef struct {
|
|
Object base;
|
|
Runtime *runtime;
|
|
} BuiltinsMapOjbect;
|
|
|
|
static const Type t_builtins_map = {
|
|
.base = (Object) { .type = &t_type, .flags = Object_STATIC },
|
|
.name = "builtins map",
|
|
.size = sizeof(BuiltinsMapOjbect),
|
|
.select = select,
|
|
.count = count,
|
|
.print = NULL,
|
|
};
|
|
|
|
static Object *select(Object *self, Object *key, Heap *heap, Error *err)
|
|
{
|
|
BuiltinsMapOjbect *bm = (BuiltinsMapOjbect*) self;
|
|
|
|
if(!Object_IsString(key))
|
|
{
|
|
Error_Report(err, 0, "Non string key");
|
|
return NULL;
|
|
}
|
|
|
|
int n;
|
|
const char *s;
|
|
|
|
s = Object_ToString(key, &n, heap, err);
|
|
|
|
if(s == NULL)
|
|
return NULL;
|
|
|
|
#define PAIR(p, q) \
|
|
(((uint64_t) (p) << 32) | (uint32_t) (q))
|
|
|
|
switch(PAIR(n, s[0]))
|
|
{
|
|
case PAIR(sizeof("count")-1, 'c'):
|
|
{
|
|
if(!strcmp(s, "count"))
|
|
return Object_FromNativeFunction(bm->runtime, bin_count, 1, heap, err);
|
|
return NULL;
|
|
}
|
|
|
|
case PAIR(sizeof("print")-1, 'p'):
|
|
{
|
|
if(!strcmp(s, "print"))
|
|
return Object_FromNativeFunction(bm->runtime, bin_print, -1, heap, err);
|
|
return NULL;
|
|
}
|
|
|
|
case PAIR(sizeof("assert")-1, 'a'):
|
|
{
|
|
if(!strcmp(s, "assert"))
|
|
return Object_FromNativeFunction(bm->runtime, bin_assert, -1, heap, err);
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
// Not found.
|
|
return NULL;
|
|
}
|
|
|
|
static int count(Object *self)
|
|
{
|
|
(void) self;
|
|
return 3;
|
|
}
|
|
|
|
Object *Object_NewBuiltinsMap(Runtime *runtime, Heap *heap, Error *err)
|
|
{
|
|
BuiltinsMapOjbect *bm = (BuiltinsMapOjbect*) Heap_Malloc(heap, &t_builtins_map, err);
|
|
|
|
if(bm == NULL)
|
|
return NULL;
|
|
|
|
bm->runtime = runtime;
|
|
|
|
return (Object*) bm;
|
|
}
|
|
|
|
static Object *bin_print(Runtime *runtime, Object **argv, unsigned int argc, Error *error)
|
|
{
|
|
for(int i = 0; i < (int) argc; i += 1)
|
|
Object_Print(argv[i], stdout);
|
|
|
|
return Object_NewNone(Runtime_GetHeap(runtime), error);
|
|
}
|
|
|
|
static Object *bin_count(Runtime *runtime, Object **argv, unsigned int argc, Error *error)
|
|
{
|
|
assert(argc == 1);
|
|
|
|
int n = Object_Count(argv[0], error);
|
|
|
|
if(error->occurred)
|
|
return NULL;
|
|
|
|
return Object_FromInt(n, Runtime_GetHeap(runtime), error);
|
|
}
|
|
|
|
static Object *bin_assert(Runtime *runtime, Object **argv, unsigned int argc, Error *error)
|
|
{
|
|
assert(argc == 1);
|
|
|
|
if(Object_ToBool(argv[0], error))
|
|
{
|
|
return Object_NewNone(Runtime_GetHeap(runtime), error);
|
|
}
|
|
else
|
|
{
|
|
if(!error->occurred)
|
|
Error_Report(error, 0, "Assertion failed");
|
|
return NULL;
|
|
}
|
|
} |