lang2/lib/vm/print.c

#include "vm/print.h"

#include <stdio.h>
#include <string.h>
#include <stdint.h>

void l2_vm_print_val(struct l2_vm_value *val) {

	switch (l2_vm_value_type(val)) {
	case L2_VAL_TYPE_NONE:
		printf("NONE\n");
		break;

	case L2_VAL_TYPE_ATOM:
		printf("ATOM %u\n", val->atom);
		break;

	case L2_VAL_TYPE_REAL:
		printf("REAL %f\n", val->real);
		break;

	case L2_VAL_TYPE_ARRAY:
		{
			if (val->array == NULL) {
				printf("ARRAY, empty\n");
				return;
			}

			printf("ARRAY, len %zu\n", val->array->len);
			for (size_t i = 0; i < val->array->len; ++i) {
				printf("    %zu: %u\n", i, val->array->data[i]);
			}
		}
		break;

	case L2_VAL_TYPE_BUFFER:
		{
			if (val->buffer == NULL) {
				printf("BUFFER, empty\n");
				return;
			}

			printf("BUFFER, len %zu\n", val->buffer->len);
			for (size_t i = 0; i < val->buffer->len; ++i) {
				printf("    %zu: %c\n", i, val->buffer->data[i]);
			}
		}
		break;

	case L2_VAL_TYPE_NAMESPACE:
		{
			if (val->ns == NULL) {
				printf("NAMESPACE, empty, parent %u\n", val->extra.ns_parent);
				return;
			}

			printf("NAMESPACE, len %zu, parent %u\n", val->ns->len, val->extra.ns_parent);
			for (size_t i = 0; i < val->ns->size; ++i) {
				l2_word key = val->ns->data[i];
				l2_word v = val->ns->data[val->ns->size + i];
				if (key == 0 || key == ~(l2_word)0) continue;
				printf("    %u: %u\n", key, v);
			}
		}
		break;

	case L2_VAL_TYPE_FUNCTION:
		printf("FUNCTION, pos %u, ns %u\n", val->func.pos, val->func.ns);
		break;

	case L2_VAL_TYPE_CFUNCTION:
		// ISO C doesn't let you cast a function pointer to void*.
		printf("C FUNCTION, %jx\n", (uintmax_t)val->cfunc);
		break;

	case L2_VAL_TYPE_ERROR:
		printf("ERROR, %s\n", val->error);
		break;

	case L2_VAL_TYPE_CONTINUATION:
		printf("CONTINUATION, call %u, arg %u\n", val->cont.call, val->cont.arg);
		break;
	}
}

void l2_vm_print_state(struct l2_vm *vm) {
	printf("Stack:\n");
	l2_vm_print_stack(vm);
	printf("Heap:\n");
	l2_vm_print_heap(vm);
	printf("Frame Stack:\n");
	l2_vm_print_fstack(vm);
}

void l2_vm_print_heap(struct l2_vm *vm) {
	for (l2_word i = 0; i < vm->valuessize; ++i) {
		if (l2_bitset_get(&vm->valueset, i)) {
			printf("  %u: ", i);
			l2_vm_print_val(&vm->values[i]);
		}
	}
}

void l2_vm_print_stack(struct l2_vm *vm) {
	for (l2_word i = 0; i < vm->sptr; ++i) {
		printf("  %i: %i\n", i, vm->stack[i]);
	}
}

void l2_vm_print_fstack(struct l2_vm *vm) {
	for (l2_word i = 0; i < vm->fsptr; ++i) {
		printf("  %i: %i, ret %i, stack base %i\n",
				i, vm->fstack[i].ns, vm->fstack[i].retptr, vm->fstack[i].sptr);
	}
}

void l2_vm_print_op(l2_word *ops, size_t opcount, size_t *ptr) {
	enum l2_opcode opcode = (enum l2_opcode)ops[(*ptr)++];

	switch (opcode) {
	case L2_OP_NOP:
		printf("NOP\n");
		return;

	case L2_OP_DISCARD:
		printf("DISCARD\n");
		return;

	case L2_OP_SWAP_DISCARD:
		printf("SWAP_DISCARD\n");
		return;

	case L2_OP_DUP:
		printf("DUP\n");
		return;

	case L2_OP_ADD:
		printf("ADD\n");
		return;

	case L2_OP_FUNC_CALL:
		printf("FUNC_CALL %08x\n", ops[(*ptr)++]);
		return;

	case L2_OP_FUNC_CALL_INFIX:
		printf("FUNC_CALL_INFIX\n");
		return;

	case L2_OP_RJMP:
		printf("RJMP %08x\n", ops[(*ptr)++]);
		return;

	case L2_OP_STACK_FRAME_LOOKUP:
		printf("STACK_FRAME_LOOKUP %08x\n", ops[(*ptr)++]);
		return;

	case L2_OP_STACK_FRAME_SET:
		printf("STACK_FRAME_SET %08x\n", ops[(*ptr)++]);
		return;

	case L2_OP_STACK_FRAME_REPLACE:
		printf("STACK_FRAME_REPLACE %08x\n", ops[(*ptr)++]);
		return;

	case L2_OP_RET:
		printf("RET\n");
		return;

	case L2_OP_ALLOC_NONE:
		printf("ALLOC_NONE\n");
		return;

	case L2_OP_ALLOC_ATOM:
		printf("ALLOC_ATOM %08x\n", ops[(*ptr)++]);
		return;

	case L2_OP_ALLOC_REAL:
		{
			l2_word w1 = ops[(*ptr)++];
			l2_word w2 = ops[(*ptr)++];
			printf("ALLOC_REAL %08x %08x\n", w1, w2);
		}
		return;

	case L2_OP_ALLOC_BUFFER_STATIC:
		{
			l2_word w1 = ops[(*ptr)++];
			l2_word w2 = ops[(*ptr)++];
			printf("ALLOC_BUFFER_STATIC %08x %08x\n", w1, w2);
		}
		return;

	case L2_OP_ALLOC_ARRAY:
		printf("ALLOC_ARRAY %08x\n", ops[(*ptr)++]);
		return;

	case L2_OP_ALLOC_NAMESPACE:
		printf("ALLOC_NAMESPACE\n");
		return;

	case L2_OP_ALLOC_FUNCTION:
		printf("ALLOC_FUNCTION %08x\n", ops[(*ptr)++]);
		return;

	case L2_OP_NAMESPACE_SET:
		printf("NAMESPACE_SET %08x\n", ops[(*ptr)++]);
		return;

	case L2_OP_NAMESPACE_LOOKUP:
		printf("NAMESPACE_LOOKUP %08x\n", ops[(*ptr)++]);
		return;

	case L2_OP_ARRAY_LOOKUP:
		printf("ARRAY_LOOKUP %08x\n", ops[(*ptr)++]);
		return;

	case L2_OP_ARRAY_SET:
		printf("ARRAY_SET %08x\n", ops[(*ptr)++]);
		return;

	case L2_OP_DYNAMIC_LOOKUP:
		printf("DYNAMIC_LOOKUP\n");
		return;

	case L2_OP_DYNAMIC_SET:
		printf("DYNAMIC_SET\n");
		return;

	case L2_OP_HALT:
		printf("HALT\n");
		return;
	}

	l2_word word = (l2_word)opcode;
	char bytes[sizeof(word)];
	memcpy(&bytes, &word, sizeof(word));
	printf("?");
	for (size_t i = 0; i < sizeof(bytes); ++i) {
		printf(" %02x", bytes[i]);
	}
	printf("\n");
}

void l2_vm_print_bytecode(l2_word *ops, size_t opcount) {
	size_t ptr = 0;
	while (ptr < opcount) {
		printf("%04zu ", ptr);
		l2_vm_print_op(ops, opcount, &ptr);
	}
}