utils/macros.h: Add <ctype.h> and `foocmp' helper macros.

[mLib] / struct / t / assoc-test.c

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

#include "assoc.h"
#include "atom.h"
#include "macros.h"

typedef struct word {
  assoc_base _b;
  int i;
} word;

static int cmp(const void *a, const void *b)
{
  const word *const *v = b;
  const word *const *w = a;
  return (strcmp(ATOM_NAME(ASSOC_ATOM(*w)), ATOM_NAME(ASSOC_ATOM(*v))));
}

int main(void)
{
  char buf[256];
  char *p;
  atom_table at;
  assoc_table t;
  size_t n = 0, j;

  atom_createtable(&at);
  assoc_create(&t);

  while (fgets(buf, sizeof(buf), stdin)) {
/*     printf("+++ %s", buf); */
    buf[strlen(buf) - 1] = 0;
    p = strtok(buf, " ");

    if (STRCMP(p, ==, "set")) {
      char *k = strtok(0, " ");
      int i = atoi(strtok(0, " "));
      unsigned f;
      word *w = assoc_find(&t, atom_intern(&at, k), sizeof(word), &f);
      w->i = i;
      if (!f)
	n++;
    } else if (STRCMP(p, ==, "get")) {
      char *k = strtok(0, " ");
      word *w = assoc_find(&t, atom_intern(&at, k), 0, 0);
      if (w)
	printf("%i\n", w->i);
      else
	puts("*MISSING*");
    } else if (STRCMP(p, ==, "del")) {
      char *k = strtok(0, " ");
      word *w = assoc_find(&t, atom_intern(&at, k), 0, 0);
      if (w) {
	assoc_remove(&t, w);
	n--;
      } else
	puts("*MISSING*");
    } else if (STRCMP(p, ==, "count")) {
      printf("%lu\n", (unsigned long)n);
    } else if (STRCMP(p, ==, "show")) {
      assoc_iter i;
      word *w;
      word **v, **vv;

      if (!n)
	puts("*EMPTY*");
      else {
	v = malloc(n * sizeof(*v));
	if (!v) {
	  puts("*NOMEM*");
	  continue;
	}
	for (vv = v, assoc_mkiter(&i, &t), j = 0;
	     (w = assoc_next(&i)) != 0;
	     vv++, j++) {
	  assert(j < n);
	  *vv = w;
	}
	assert(j == n);
	qsort(v, n, sizeof(*v), cmp);
	printf("%s:%i", ATOM_NAME(ASSOC_ATOM(*v)), (*v)->i);
	for (vv = v + 1; --j; vv++)
	  printf(" %s:%i", ATOM_NAME(ASSOC_ATOM(*vv)), (*vv)->i);
	free(v);
	putchar('\n');
      }
    } else
      puts("*BAD*");
/*     printf("--- %d\n", n); */
  }

  assoc_destroy(&t);
  atom_destroytable(&at);
  return (0);
}