genstructs/htable/main.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <string.h>
   4 #include <time.h>    /* for time() in srand() */
   5
   6 #include "htable.h"
   7
   8 unsigned int myhashf(const void *key)
   9 {
  10     unsigned int i, hash = 5381;
  11     char *str = (char *)key;
  12
  13     for (i = 0; i < strlen(str); i++)
  14         hash = ((hash << 5) + hash) + str[i];
  15
  16     return hash;
  17 }
  18
  19 int mystrcmp(const void *arg1, const void *arg2)
  20 {
  21     return (strcmp((char *)arg1, (char *)arg2));
  22 }
  23
  24 void myprintf(const void *key, const void *data)
  25 {
  26     printf("%s(%s) ", (char *)key, (char *)data);
  27 }
  28
  29 void print_elm(void *data)
  30 {
  31     printf("%s\n", (char *)data);
  32 }
  33
  34 void get_rand_string(char *str, size_t len)
  35 {
  36     size_t i;
  37
  38     for (i = 0; i < len; i++)
  39         str[i] = 32 + 65 + (rand() / (RAND_MAX / 26 + 1));
  40     str[i] = '\0';
  41 }
  42
  43 int main(void)
  44 {
  45     htable_t htable;
  46     size_t i;
  47     char str[20][10+1];    /* +1 for '\0' */
  48
  49     /* Initialize random number generator */
  50     srand(time(NULL));
  51
  52     for (i = 0; i < sizeof str / sizeof *str; i++)
  53         get_rand_string(str[i], 10);
  54
  55     /* Initialize table */
  56     if (htable_init(&htable, 32, 2, myhashf, mystrcmp, myprintf) == HT_NOMEM) {
  57         perror("malloc");
  58         exit(EXIT_FAILURE);
  59     }
  60
  61     /* Add random entries in hash table */
  62     for (i = 0; i < sizeof str / sizeof *str; i++)
  63         htable_insert(&htable, str[i], str[i]);
  64     /* htable_print(&htable); */
  65
  66     /* Traverse all elemets and print each one of them */
  67     htable_traverse(&htable, print_elm);
  68
  69     /* Print the length of every chain */
  70     for (i = 0; i < htable_get_size(&htable); i++)
  71         printf("chain[%lu] = %lu\n",
  72                (unsigned long)i,
  73                (unsigned long)htable_stat_get_chain_len(&htable, i));
  74
  75     /* Print number of automatic resizes */
  76     printf("Automatic resizes: %lu\n", (unsigned long)htable_stat_get_grows(&htable));
  77
  78     /* Free memory */
  79     htable_free(&htable);
  80
  81     return EXIT_SUCCESS;
  82 }