#ifndef L2_VM_H
#define L2_VM_H

#include <stdlib.h>

#include "../bytecode.h"
#include "../bitset.h"
#include "../io.h"

struct l2_vm;
struct l2_vm_array;
typedef l2_word (*l2_vm_cfunction)(struct l2_vm *vm, l2_word argc, l2_word *argv);
typedef l2_word (*l2_vm_contcallback)(struct l2_vm *vm, l2_word retval, l2_word cont);
typedef void (*l2_vm_gcmarker)(
		struct l2_vm *vm, void *data, void (*mark)(struct l2_vm *vm, l2_word id));

enum l2_value_type {
	L2_VAL_TYPE_NONE,
	L2_VAL_TYPE_ATOM,
	L2_VAL_TYPE_REAL,
	L2_VAL_TYPE_BUFFER,
	L2_VAL_TYPE_ARRAY,
	L2_VAL_TYPE_NAMESPACE,
	L2_VAL_TYPE_FUNCTION,
	L2_VAL_TYPE_CFUNCTION,
	L2_VAL_TYPE_CONTINUATION,
	L2_VAL_TYPE_RETURN,
	L2_VAL_TYPE_ERROR,
};

const char *l2_value_type_name(enum l2_value_type typ);

enum l2_value_flags {
	L2_VAL_MARKED = 1 << 7,
	L2_VAL_CONST = 1 << 6,
	L2_VAL_SBO = 1 << 5,
};

struct l2_vm_contcontext {
	l2_vm_contcallback callback;
	l2_vm_gcmarker marker;
	l2_word args;
};

// The smallest size an l2_vm_value can be is 16 bytes on common platforms.
// Pointers are 8 bytes, and since we need to store _at least_ 1 pointer +
// 1 byte for flags, it's going to be padded up to 16 bytes anyways.
// Might as well store some useful extra info in here.
struct l2_vm_value {
	// Byte 0: 4 bytes
	union {
		l2_word ns_parent;
		l2_word cont_call;
		l2_word buf_length;
		l2_word arr_length;
	} extra;

	// Byte 4: 1 byte, 3 bytes padding
	uint8_t flags;

	// Byte 8: 8 bytes
	union {
		l2_word atom;
		double real;
		char *buffer;
		struct l2_vm_array *array;
		l2_word shortarray[2];
		struct l2_vm_namespace *ns;
		struct {
			l2_word pos;
			l2_word ns;
		} func;
		l2_vm_cfunction cfunc;
		struct l2_vm_contcontext *cont;
		l2_word ret;
		char *error;
	};
};

#define l2_value_get_type(val) ((enum l2_value_type)((val)->flags & 0x0f))

l2_word *l2_value_arr_data(struct l2_vm *vm, struct l2_vm_value *val);
l2_word l2_value_arr_get(struct l2_vm *vm, struct l2_vm_value *val, l2_word k);
l2_word l2_value_arr_set(struct l2_vm *vm, struct l2_vm_value *val, l2_word k, l2_word v);

struct l2_vm_array {
	size_t size;
	l2_word data[];
};

struct l2_vm_namespace {
	size_t len;
	size_t size;
	l2_word mask;
	l2_word data[];
};

l2_word l2_vm_namespace_get(struct l2_vm *vm, struct l2_vm_value *ns, l2_word key);
void l2_vm_namespace_set(struct l2_vm_value *ns, l2_word key, l2_word val);
int l2_vm_namespace_replace(struct l2_vm *vm, struct l2_vm_value *ns, l2_word key, l2_word val);

struct l2_vm_stack_frame {
	l2_word ns;
	l2_word sptr;
	l2_word retptr;
	l2_word args;
};

struct l2_vm {
	int halted;
	int need_gc;
	int need_check_retval;
	unsigned char *ops;
	size_t opslen;
	l2_word iptr;

	struct l2_io_writer *std_output;
	struct l2_io_writer *std_error;

	struct l2_vm_value *values;
	size_t valuessize;
	struct l2_bitset valueset;

	l2_word stack[1024];
	l2_word sptr;

	struct l2_vm_stack_frame fstack[1024];
	l2_word fsptr;

	l2_word knone, ktrue, kfalse, kstop;
	l2_word gc_start;
};

void l2_vm_init(struct l2_vm *vm, unsigned char *ops, size_t opslen);
l2_word l2_vm_alloc(struct l2_vm *vm, enum l2_value_type typ, enum l2_value_flags flags);
l2_word l2_vm_error(struct l2_vm *vm, const char *fmt, ...);
l2_word l2_vm_type_error(struct l2_vm *vm, struct l2_vm_value *val);
void l2_vm_free(struct l2_vm *vm);
void l2_vm_step(struct l2_vm *vm);
void l2_vm_run(struct l2_vm *vm);
size_t l2_vm_gc(struct l2_vm *vm);
int l2_vm_val_is_true(struct l2_vm *vm, struct l2_vm_value *val);

#endif