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

#include "rkdtree.h"
#include "rkdnode.h"

#define ELCOUNT 2000
#define ELDIM 4

int main()
{
  unsigned int i, j;

  /* Create an rkdtree */
  rkdtree_t *T = rkdtree_init(ELDIM);

  /* Generate keys to insert */
  int **keys = malloc(sizeof(int*)*ELCOUNT);

  for(i=0;i<ELCOUNT;i++)
    {
      keys[i] = malloc(sizeof(int)*ELDIM);

      for(j=0;j<ELDIM;j++)
	keys[i][j] = random() % 200;
    }
  
  /* Insert keys */
  printf("Inserting keys...");
  for(i=0;i<ELCOUNT;i++)
    rkdtree_insert(T, keys[i], NULL);
  printf("done\n");

  printf("Node count after insert : %d\n", T->root->elem_count);

  /* Show rkdtree */
  rkdtree_dump(T);

  /* Insert keys */
  printf("Removing keys...");
  for(i=0;i<ELCOUNT;i++)
    rkdtree_remove(T, keys[i]);
  printf("done\n");

  printf("Checking if root is now NULL...");
  if ( T->root == NULL )
    printf("yes (good)\n");
  else
    {
      printf("no (bad)\n");
      printf("Elem count : %u\n", T->root->elem_count);
      /* Show rkdtree */
      rkdtree_dump(T);
    }

  /* Free keys */
  for(i=0;i<ELCOUNT;i++)
    free(keys[i]);
  free(keys);

  return EXIT_SUCCESS;
}