dynamic hashtables optimized
[henge/apc.git] / src / testston.c
index 71c0110..0f0f1f9 100644 (file)
-#include "../ston/ston.h"
 #include <stdlib.h> //malloc
 #include <stdio.h> //print
+#include <string.h> //memcpy
+#include "../ston/ston.h"
+
+/* ansi terminal colors */
+#define RED     "\x1b[31m"
+#define GREEN   "\x1b[32m"
+#define YELLOW  "\x1b[33m"
+#define BLUE    "\x1b[34m"
+#define MAGENTA "\x1b[35m"
+#define CYAN    "\x1b[36m"
+#define CLRC    "\x1b[0m" //clear current color
+#define CLRCN   CLRC"\n"
+
+#define print_val(val) do {                    \
+    for (i = 0; i < (size_t)columns; i++)      \
+      printf("["YELLOW"%x"CLRC"]",val[i]);     \
+    printf("\n");                              \
+  } while(0)
+
+#define check_ht(_HT)                                                  \
+  if ((_HT) == NULL)                                                   \
+    { fprintf(stderr,RED"Could not allocate ht32"CLRCN);               \
+      return -1;                                                       \
+    }                                                                  \
+  else                                                                 \
+    printf("ht_size: [units:%i][bytes:%li]\n",ston_ht_size(ht),ston_ht32_size(ht)); \
+
+#define check_dht(_HT)                                                 \
+  if ((_HT) == NULL)                                                   \
+    { fprintf(stderr,RED"Could not allocate dht32"CLRCN);              \
+      return -1;                                                       \
+    }                                                                  \
+  else                                                                 \
+    printf("ht_size: [units:%i][bytes:%i]\n",ston_dht_units(_HT,_HT->header.start_depth),ston_dht_bytes(_HT,_HT->header.start_depth)); \
+
 
 int main(int argc, char* argv[])
-{ ston_ht ht;
-  uint32_t* htval;
-
-  if ((ht = ston_ht32_new(2,1,0,malloc)) == NULL)
-    fprintf(stderr,"Could not allocate ht32\n");
-  ston_ht32_insertx(ht,50,1,200);
-  htval = ston_ht32_entry(ht,50,1);
-  printf("[50][1] = %i\n",*htval);
+{ static ston_ht   ht;
+  uint32_t* htval, previous_val;
+  uint32_t  val[255], key;
+  size_t    idx;
+  size_t    i;
+  size_t    elements;
+  uint16_t  columns;
+  int       fail;
+
+  printf("up2pow [5] = [%i]\n",ston_up2pow(5));
+  for (i = 0; i < 77; i += 7)
+    printf("trailing0 [%X] = [%x]\n",(uint32_t)i,ston_trailing0(i));
+
+  columns = 2;
+  elements = 1;
+  ht = ston_ht32_new(columns,elements,0,malloc);
+  check_ht(ht);
+  key = 0xFF;
+  val[0] = key;
+  val[1] = 0x5;
+  printf("ht32_insertx: [%x] = ", key);
+  print_val(val);
+  ston_ht32_insertx(ht,key,val,0,columns);
+  idx = 1;
+  htval = ston_ht32_row(ht,key);
+  printf("Read value [%x] matches: %s"CLRCN, htval[idx],
+        (htval[idx] == val[idx]) ? GREEN"PASS" : RED"FAIL");
+
+  val[1] = 0x2;
+  previous_val = ston_ht32_insert(ht,key,1,val[1]);
+  printf("ht32_insert: [%x] = ", key);
+  print_val(val);
+  printf("Previous value [%x] matches [5]: %s"CLRCN, previous_val,
+        (previous_val == 0x5) ? GREEN"PASS" : RED"FAIL");
+
   free(ht);
+
+  columns = 4;
+  elements = 20;
+  ht = ston_ht32_new(columns,elements,0,malloc);
+  check_ht(ht);
+  printf("Filling hashtable with %i entries\n", (int)elements);
+  for(key = 0xCEED; elements--; key *= 7)
+    { val[0] = key;
+      for(i = 1; i < columns; i++)
+       val[i] = i * key;
+      ston_ht32_insertx(ht,key,val,0,columns);
+    }
+  elements = 20;
+  printf("Reading entries: ");
+  fail = 0;
+  for(key = 0xCEED; elements--; key *= 7)
+    { htval = ston_ht32_row(ht,key);
+      if (*htval != key)
+       fail++;
+      for(i = 1; i < columns; i++)
+       if (htval[i] != (uint32_t)(i * key))
+         fail++;
+    }
+  if (fail)
+    printf(RED"FAIL"CLRC"(%i)\n", fail);
+  else
+    printf(GREEN"PASS"CLRCN);
+  int max_capacity = ston_up2pow(20 << 1) - 20;
+  int cap = max_capacity;
+  printf("Overfilling hashtable with %i entries\n", max_capacity);
+  for(key = 0xCEED2; cap--; key *= 13)
+    { val[0] = key;
+      for(i = 1; i < columns; i++)
+       val[i] = key * -i;
+      ston_ht32_insertx(ht,key,val,0,columns);
+    }
+  printf("Reading entries: ");
+  cap = max_capacity;
+  fail = 0;
+  for(key = 0xCEED2; cap--; key *= 13)
+    { htval = ston_ht32_row(ht,key);
+      if (*htval != key)
+       fail++;
+      for(i = 1; i < columns; i++)
+       if (htval[i] != (uint32_t)(key * -i))
+         fail++;
+    }
+  if (fail)
+    printf(RED"FAIL"CLRC"(%i)\n", fail);
+  else
+    printf(GREEN"PASS"CLRCN);
+
+  cap = 20;
+  printf("Post-capacity insertion of %i\n",cap);
+  for (key = 0xCEED3; cap--; key *= 23)
+    { val[0] = key;
+      for(i = 1; i < columns; i++)
+       val[i] = key * -i;
+      size_t count = ston_ht32_insertx(ht,key,val,0,columns);
+      printf("Insertion %2i wrote %i bytes: %s"CLRCN, (int)cap, (int) count,
+            (count == 0) ? GREEN"PASS" : RED"FAIL");
+    }
+  
+
+  printf("Refilling hashtable with %i entries\n", max_capacity);
+  cap = max_capacity;
+  for(key = 0xCEED2; cap--; key *= 13)
+    { val[0] = key;
+      for(i = 1; i < columns; i++)
+       val[i] = key * i;
+      ston_ht32_insertx(ht,key,val,0,columns);
+    }
+  printf("Reading entries: ");
+  cap = max_capacity;
+  fail = 0;
+  for(key = 0xCEED2; cap--; key *= 13)
+    { htval = ston_ht32_row(ht,key);
+      if (*htval != key)
+       fail ++;
+      for (i = 1; i < columns; i++)
+       if (htval[i] != (uint32_t)(key * i))
+         fail++;
+    }
+  if (fail)
+    printf(RED"FAIL"CLRC"(%i)\n", fail);
+  else
+    printf(GREEN"PASS"CLRCN);
+
+  free(ht);
+
+
+  printf("\n--------- DHT ----------\n\n");
+
+  ston_dht dht;
+  elements = 500;
+  columns = 6;
+  dht = ston_dht32_new(columns, malloc, free);
+  check_dht(dht);
+  elements = 50000;
+  printf("Filling Dynamic hashtable with %i entries\n", (int)elements);
+  for(key = 0xCEED; elements--; key *= 7)
+    { val[0] = key;
+      for(i = 1; i < columns; i++)
+       val[i] = i * key;
+      ston_dht32_insertx(dht,key,val,0,columns);
+    }
+  elements = 50000;
+  printf("Reading entries: ");
+  fail = 0;
+  for(key = 0xCEED; elements--; key *= 7)
+    { htval = ston_dht32_row(dht,key);
+      if (*htval != key)
+       fail++;
+      for(i = 1; i < columns; i++)
+       if (htval[i] != (uint32_t)(i * key))
+         fail++;
+    }
+  if (fail)
+    printf(RED"FAIL"CLRC"(%i)\n", fail);
+  else
+    printf(GREEN"PASS"CLRCN);
+  max_capacity = 100000;
+  cap = max_capacity;
+  printf("Overfilling hashtable with %i entries\n", max_capacity);
+  for(key = 0xCEED2; cap--; key *= 13)
+    { val[0] = key;
+      for(i = 1; i < columns; i++)
+       val[i] = key * -i;
+      ston_dht32_insertx(dht,key,val,0,columns);
+    }
+  printf("Reading entries: ");
+  cap = max_capacity;
+  fail = 0;
+  for(key = 0xCEED2; cap--; key *= 13)
+    { htval = ston_dht32_row(dht,key);
+      if (*htval != key)
+       fail++;
+      for(i = 1; i < columns; i++)
+       if (htval[i] != (uint32_t)(key * -i))
+         fail++;
+    }
+  if (fail)
+    printf(RED"FAIL"CLRC"(%i)\n", fail);
+  else
+    printf(GREEN"PASS"CLRCN);
+
+  max_capacity = 5000000;
+  printf("Refilling hashtable with %i entries\n", max_capacity);
+  cap = max_capacity;
+  for(key = 0xCEED2; cap--; key *= 13)
+    { val[0] = key;
+      for(i = 1; i < columns; i++)
+       val[i] = key * i;
+      ston_dht32_insertx(dht,key,val,0,columns);
+    }
+  printf("Reading entries: ");
+  cap = max_capacity;
+  fail = 0;
+  for(key = 0xCEED2; cap--; key *= 13)
+    { htval = ston_dht32_row(dht,key);
+      if (*htval != key)
+       fail ++;
+      for (i = 1; i < columns; i++)
+       if (htval[i] != (uint32_t)(key * i))
+         fail++;
+    }
+  if (fail)
+    printf(RED"FAIL"CLRC"(%i)\n", fail);
+  else
+    printf(GREEN"PASS"CLRCN);
+
+  ston_dht_free(dht);
+  
   return 0;
 }