#include "vm/print.h"

#include <stdio.h>
#include <string.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_INTEGER:
		printf("INTEGER %zi\n", val->integer);
		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.namespace);
		break;

	case L2_VAL_TYPE_CFUNCTION:
		printf("C FUNCTION, %p\n", val->cfunc);
		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("Namespace Stack:\n");
	l2_vm_print_nstack(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_nstack(struct l2_vm *vm) {
	for (l2_word i = 0; i < vm->nsptr; ++i) {
		printf("  %i: %i\n", i, vm->nstack[i]);
	}
}

static void print_op_num(l2_word *ops, size_t opcount, size_t ptr) {
	if (ptr >= opcount) {
		printf("<EOF>");
	} else {
		printf("%i", ops[ptr]);
	}
}

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");
		break;

	case L2_OP_PUSH:
		printf("PUSH ");
		print_op_num(ops, opcount, (*ptr)++);
		printf("\n");
		break;

	case L2_OP_PUSH_2:
		printf("PUSH2 ");
		print_op_num(ops, opcount, (*ptr)++);
		printf(" ");
		print_op_num(ops, opcount, (*ptr)++);
		printf("\n");
		break;

	case L2_OP_POP:
		printf("POP\n");
		break;

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

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

	case L2_OP_FUNC_CALL:
		printf("FUNC_CALL\n");
		break;

	case L2_OP_RJMP:
		printf("RJMP\n");
		break;

	case L2_OP_STACK_FRAME_LOOKUP:
		printf("STACK_FRAME_LOOKUP\n");
		break;

	case L2_OP_STACK_FRAME_SET:
		printf("STACK_FRAME_SET\n");
		break;

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

	case L2_OP_ALLOC_INTEGER_32:
		printf("ALLOC_INTEGER_32\n");
		break;

	case L2_OP_ALLOC_INTEGER_64:
		printf("ALLOC_INTEGER_64\n");
		break;

	case L2_OP_ALLOC_REAL_32:
		printf("ALLOC_REAL_32\n");
		break;

	case L2_OP_ALLOC_REAL_64:
		printf("ALLOC_REAL_64\n");
		break;

	case L2_OP_ALLOC_BUFFER_STATIC:
		printf("ALLOC_BUFFER_STATIC\n");
		break;

	case L2_OP_ALLOC_BUFFER_ZERO:
		printf("ALLOC_BUFFER_ZERO\n");
		break;

	case L2_OP_ALLOC_ARRAY:
		printf("ALLOC_ARRAY\n");
		break;

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

	case L2_OP_ALLOC_FUNCTION:
		printf("ALLOC_FUNCTION\n");
		break;

	case L2_OP_NAMESPACE_SET:
		printf("NAMESPACE_SET\n");
		break;

	case L2_OP_HALT:
		printf("HALT\n");
		break;

	default:
		{
			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) {
		l2_vm_print_op(ops, opcount, &ptr);
	}
}