#include "btree.h"

#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

/* Definitions */
typedef unsigned char byte;

struct node {
	size_t       n; /* number of items/keys/elements */
	size_t       c; /* number of children */
	byte        *items;
	struct node **children;
};

struct btree {
	/* Memory stuffs */
	void *(*alloc)(size_t);
	void  (*dealloc)(void*);

	/* Size stuffs */
	size_t elem_size;
	size_t degree;

	struct node *root;

	/* comparison */
	int (*cmp)(const void *a, const void *b);
};

/* Node functionality */
#define \
node_leaf(node) (node->children == NULL)

#define \
node_maxdegree(t) (2 * t - 1)

#define \
node_mindegree(t) (t - 1)

#define \
node_full(degree, t) (t->n == 2 * degree - 1)

/* Node memory */

/* `node_new` allocates a new leaf node, children should be added and allocated
 * when the node is no longer a leaf node */
struct node* node_new(const size_t degree, const size_t elem_size) {
	const size_t max_items = 2 * degree - 1;
	struct node *retval = malloc(sizeof(struct node));

	retval->n        = 0;
	retval->c        = 0;
	retval->items    = calloc(max_items, sizeof(elem_size));
	retval->children = NULL;

	return retval;
}

/* `node_transition` turns a leaf node into a non-leaf and allocates children
 * for it.
 * returnvalue: `false` if we we're unable to allocate space for the new
 * children. */
bool node_transition(const size_t degree, const size_t elem_size, struct node *node) {
	const int max_children = 2 * degree;
	int c;
	/* Allocate pointers for children */
	node->children = calloc(max_children, sizeof(struct node*));

	if (node->children == NULL) {
		perror("could not allocate space for children pointers");
		return false;
	}

	/* Allocate children */
	for (c = 0; c < max_children; c++) {
		node->children[c] = node_new(degree, elem_size);
		if (node->children[c] == NULL) {
			perror("could not allocate space for all children, freeing...");
			for (c = c - 1;c >= 0; c--) {
				free(node->children[c]);
			}
			free(node->children);
			return false;
		}
	}

	node->c = c;

	return true;
}

void node_free(struct node *node, size_t elem_size, void (*dealloc)(void*)) {
	size_t i;
	if (node == NULL) return;
	for (i = 0; i < node->c; i++) {
		node_free((node->children)[i], elem_size, dealloc);
	}

	if (!node_leaf(node))
		dealloc(node->items);

	free(node);
}

/* Split a child of `nonfull` of index `i` */
void node_tree_split_child(const size_t t, const size_t elem_size, struct node *nonfull, size_t i) {
	struct node *z = node_new(t, elem_size);
	struct node *y = nonfull->children[i];
	size_t j;

	printf("Splitting tree!\n");
	__asm__("int3");

	/* `z` should be a branching node if `y` is */
	if (node_leaf(y)) {
		node_transition(t, elem_size, z);
	}

	z->n = t - 1;

	for (j = 0; j < t - 1; j++) {
		const size_t offset_dst = elem_size * j;
		const size_t offset_src = offset_dst + elem_size * t;
		memcpy((z->items) + offset_dst, (y->items) + offset_src, elem_size);
	}

	if (!node_leaf(y)) {
		for (j = 0; j < t; j++) {
			z->children[j] = y->children[j + t];
		}
		y->c = t;
		z->c = t;
	}

	y->n = t - 1;

	for (j = nonfull->n + 1; j > i+1; j--) {
		nonfull->children[j + 1] = nonfull->children[j];
	}
	nonfull->children[i+1] = z;

	for (j = nonfull->n; j > i; j--) {
		const size_t offset = j * elem_size;
		memcpy((nonfull->items) + offset + elem_size,
		       (nonfull->items) + offset,
		        elem_size);
	}

	memcpy((nonfull->items) + i * elem_size,
			(y->items) + elem_size * t,
			elem_size);

	nonfull->n++;

	__asm__("int3");

}

/* `node_split` splits a _full_ node (c = 2t-1 items) into two nodes with t-1
 * items each.
 * The median key/item/element moves up to the original nodes parent, to signify
 * the split.
 * If the parent is full too, we need to split it before inserting the median
 * key.
 * This can potentially split full nodes all the way up throughout the tree. */
/* Instead of waiting to find out wether we should split the nodes, we split the
 * full nodes we encounter on the way down, including the leafs themselves.
 * By doing this, we are assured that whenever we split a node, its parent has
 * room for the median key. */
struct node *node_split() {
	/* TODO implement */
	return NULL;
}

struct node* node_insert_nonfull(
		struct node *root,
		void *elem,
		const size_t degree,
		const size_t elem_size,
		int (*cmp)(const void *a, const void *b)) {
	/* TODO check correctness */
	size_t i = root->n - 1;
	if (node_leaf(root)) {
		size_t offset = elem_size * i;
		while (i >= 0 && cmp(elem, root->items + offset) == BTREE_CMP_LT) {
			memcpy(root->items + offset + elem_size, root->items + offset, elem_size);

			i--;
			offset = elem_size * i;
		}
		offset = elem_size * (i++);
		memcpy(root->items + offset, elem, elem_size);

	} else {
		size_t offset = elem_size * i;
		while (i >= 0 && cmp(elem, root->items + offset) == BTREE_CMP_LT) {
			i--;
			offset = elem_size * i;
		}
		i++;
		struct node *nextchild = root->children[i];
		if (node_full(degree, nextchild)) {
			/* TODO Check if the root has changed */
			node_tree_split_child(degree, elem_size, root, i);
			if (cmp(elem, root->items + elem_size * i) == BTREE_CMP_GT) {
				nextchild = root->children[++i];
			}
		}
		return node_insert_nonfull(nextchild, elem, degree, elem_size, cmp);
	}
}

struct node* node_insert(
		struct node *root,
		void *elem,
		const size_t degree,
		const size_t elem_size,
		int (*cmp)(const void *a, const void *b)) {
	if (node_full(degree, root)) {
		struct node *s = node_new(degree, elem_size);
		node_transition(degree, elem_size, s);
		s->children[s->c++] = root;
		/* TODO Check if the root has changed */
		node_tree_split_child(degree, elem_size, s, 0);
		return node_insert_nonfull(s, elem, degree, elem_size, cmp);
	}
	else {
		return node_insert_nonfull(root, elem, degree, elem_size, cmp);
	}
	return NULL;
}

void* node_search(struct node *x,
                  void *key,
                  int(*cmp)(const void *a, const void *b),
                  const size_t elem_size) {
	size_t i            = 0;
	int    last_cmp_res = cmp(key, (const void*)x->items);

	while (i < x->n && last_cmp_res > 0) {
		i++;
		last_cmp_res = cmp(key, (const void*)(x->items + (i * elem_size)));
	}

	if (i < x->n && last_cmp_res == BTREE_CMP_EQ) {
		return (void*)(x->items + (i * elem_size));
	} else if (node_leaf(x)) {
		return NULL;
	}

	/* Assumption: ¬node_leaf(x) → x.children is allocated */
	return node_search(x->children[i], key, cmp, elem_size);
}


/* Btree functionality */
struct btree* btree_new(size_t elem_size,
                        size_t t,
                        int(*cmp)(const void *a, const void *b)) {
	return btree_new_with_allocator(elem_size, t, cmp, malloc, free);
}

struct btree* btree_new_with_allocator(size_t elem_size,
                        size_t t,
                        int(*cmp)(const void *a, const void *b),
                        void *(*alloc)(size_t),
                        void (*dealloc)(void*)) {
	struct btree *new_tree = malloc(sizeof(struct btree));

	new_tree->alloc     = alloc;
	new_tree->dealloc   = dealloc;

	new_tree->elem_size = elem_size;
	new_tree->degree    = t;

	new_tree->cmp       = cmp;

	return new_tree;
}

void btree_free(struct btree *btree) {
	node_free(btree->root, btree->elem_size, btree->dealloc);
	free(btree);
	btree = NULL;
}

void btree_insert(struct btree *btree, void *elem) {
	if (btree->root == NULL) {
		btree->root = node_new(btree->degree, btree->elem_size);
		node_insert(btree->root, elem, btree->degree, btree->elem_size, btree->cmp);
	}
}

void* btree_search(struct btree *btree, void *elem) {
	return node_search(btree->root, elem, btree->cmp, btree->elem_size);
}

void* btree_delete(struct btree *btree, void *elem) {}
void* btree_update(struct btree *btree, void *elem_key, void *elem) {}


void node_print(struct node *root, const size_t elem_size, const int indent, void (*print_elem)(const void*)) {
	ssize_t i;
	int t;

	for (t = 0; t < indent - 1; t++) { fputs(" ┃ ", stdout); }
	if (indent > 0) { fputs(" ┣┯", stdout); }
	printf("printing node %p, c:%ld n:%ld\n", (void*)root, root->c, root->n);

	if (node_leaf(root)) {
		for (i = 0; i < root->n - 1; i++) {
			const size_t ofst = i * elem_size;
			for (t = 0; t < indent; t++) { fputs(" ┃├", stdout); }
			print_elem(root->items + ofst);
		}
		for (t = 0; t < indent; t++) { fputs(" ┃└", stdout); }
		print_elem(root->items + i * elem_size);
	} else {
		size_t ofst = 0;
		for (i = 0; i < root->c - 1; i++) {
			node_print(root->children[i], elem_size, indent + 1, print_elem);
			for (t = 0; t < indent; t++) { fputs(" ┃ ", stdout); }
			print_elem(root->items + ofst);
			ofst += elem_size;
		}
		node_print(root->children[i], elem_size, indent + 1, print_elem);
	}

}

void btree_print(struct btree *btree, void (*print_elem)(const void*)) {
	printf("BTRee: degree:%ld\n", btree->degree);
	node_print(btree->root, btree->elem_size, 0, print_elem);
}