/*
* Copyright © 2006 Carl Worth
*
- * This program is free software; you can redistribute it and\/or modify
+ * This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2, or (at your option)
* any later version.
* Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA."
*/
-/* Portably, schmortability. I want ease of programming. */
-#define _GNU_SOURCE
-#include <stdio.h>
-#include <stdlib.h>
+#include "dict.h"
+
#include <string.h>
#include <errno.h>
#include <ctype.h>
-#include "dict.h"
-
typedef struct _string {
size_t size;
char *s;
size_t len;
} string_t;
-#define TRIE_FLAGS_IS_WORD (1<<0)
-
typedef bool_t
(*trie_predicate_t) (trie_t *trie);
free (trie);
}
-static void
+static trie_t *
trie_add (trie_t *trie,
const char *word_chunk)
{
char c = word_chunk[0];
int i;
- if (c == '\0') {
- trie->flags |= TRIE_FLAGS_IS_WORD;
- return;
- }
+ if (c == '\0')
+ return trie;
i = TRIE_CHAR_TO_INDEX (c);
if (trie->next[i] == NULL)
trie->next[i] = trie_create ();
- trie_add (trie->next[i], word_chunk + 1);
+ return trie_add (trie->next[i], word_chunk + 1);
}
static trie_t *
}
static int
-trie_count (trie_t *trie,
- trie_predicate_t predicate)
+trie_count_if (trie_t *trie,
+ trie_predicate_t predicate)
{
int i;
int count = 0;
- if ((predicate) (trie))
+ if (predicate == NULL || (predicate) (trie))
count = 1;
for (i = 0; i < 26; i++) {
if (trie->next[i] == NULL)
continue;
- count += trie_count (trie->next[i], predicate);
+ count += trie_count_if (trie->next[i], predicate);
}
return count;
}
static int
-trie_print (trie_t *trie,
- string_t *string,
- trie_predicate_t predicate,
- int length,
- int min_length,
- int max_length)
+trie_for_each_of_length_if (trie_t *trie,
+ dict_action_t action,
+ void *closure,
+ int min_length,
+ int max_length,
+ trie_predicate_t predicate,
+ string_t *string,
+ int length)
+
{
char c;
int i;
if (length >= min_length && (predicate) (trie))
{
count = 1;
- printf ("%s ", string->s);
+
+ if (action)
+ action (closure, string->s, &trie->flags);
}
- if (length == max_length)
+ if (max_length > 0 && length == max_length)
return count;
/* Loop over each element, appending the character and recursing. */
c = TRIE_INDEX_TO_CHAR (i);
string_append_char (string, c);
- count += trie_print (trie->next[i], string, predicate,
- length + 1, min_length, max_length);
+ count += trie_for_each_of_length_if (trie->next[i],
+ action, closure,
+ min_length, max_length,
+ predicate,
+ string, length + 1);
string_chop (string);
}
dict_add_word (dict_t *dict,
const char *word)
{
- trie_add (dict, word);
+ trie_t *trie;
+
+ trie = trie_add (dict, word);
+ trie->flags |= DICT_ENTRY_FLAG_IS_WORD;
}
void
}
int
-dict_count (dict_t *dict,
- dict_entry_predicate_t predicate)
+dict_count_if (dict_t *dict,
+ dict_entry_predicate_t predicate)
{
dict_entry_predicate = predicate;
- return trie_count (dict, dict_predicate);
+ return trie_count_if (dict, dict_predicate);
}
int
-dict_print (dict_t *dict)
+dict_count (dict_t *dict)
+{
+ return dict_count_if (dict, NULL);
+}
+
+int
+dict_for_each_of_length_if (dict_t *dict,
+ dict_action_t action,
+ void *closure,
+ int min_length,
+ int max_length,
+ dict_entry_predicate_t predicate)
{
int count;
string_t string;
string_init (&string);
- dict_entry_predicate = NULL;
- count = trie_print (dict, &string, dict_predicate,
- 0, 0, -1);
+ dict_entry_predicate = predicate;
- string_fini (&string);
+ count = trie_for_each_of_length_if (dict,
+ action, closure,
+ min_length, max_length,
+ dict_predicate,
+ &string, 0);
+ string_fini (&string);
+
return count;
}
int
-dict_print_if (dict_t *dict,
- dict_entry_predicate_t predicate)
+dict_for_each_of_length (dict_t *dict,
+ dict_action_t action,
+ void *closure,
+ int min_length,
+ int max_length)
{
- int count;
- string_t string;
+ return dict_for_each_of_length_if (dict,
+ action, closure,
+ min_length, max_length,
+ NULL);
+}
- string_init (&string);
+int
+dict_for_each_if (dict_t *dict,
+ dict_action_t action,
+ void *closure,
+ dict_entry_predicate_t predicate)
- dict_entry_predicate = predicate;
- count = trie_print (dict, &string, dict_predicate,
- 0, 0, -1);
+{
+ return dict_for_each_of_length_if (dict,
+ action, closure,
+ 0, 0,
+ predicate);
+}
- string_fini (&string);
+int
+dict_for_each (dict_t *dict,
+ dict_action_t action,
+ void *closure)
+{
+ return dict_for_each_if (dict, action, closure, NULL);
+}
- return count;
+static void
+dict_action_print (void *closure, char *word, dict_entry_t *entry)
+{
+ int *first = closure;
+
+ if (*first)
+ *first = 0;
+ else
+ printf(" ");
+
+ printf ("%s", word);
}
int
-dict_print_by_length_if (dict_t *dict,
+dict_print_of_length_if (dict_t *dict,
+ int min_length,
+ int max_length,
dict_entry_predicate_t predicate)
{
- int length, total, words, count = 0;
- string_t string;
+ int first = TRUE;
- string_init (&string);
+ return dict_for_each_of_length_if (dict,
+ dict_action_print, &first,
+ min_length, max_length,
+ predicate);
+}
- dict_entry_predicate = predicate;
+int
+dict_print_by_length_if (dict_t *dict,
+ dict_entry_predicate_t predicate)
+{
+ int length, total, words, count = 0;
- total = trie_count (dict, dict_predicate);
+ total = dict_count_if (dict, predicate);
length = 1;
do {
- words = trie_print (dict, &string, dict_predicate,
- 0, length, length);
+ words = dict_print_of_length_if (dict, length, length, predicate);
if (words) {
- printf ("\n");
count += words;
+ printf ("\n");
}
length++;
} while (count < total);
- string_fini (&string);
-
return count;
}
+
+int
+dict_print_of_length (dict_t *dict,
+ int min_length,
+ int max_length)
+{
+ return dict_print_of_length_if (dict, min_length, max_length, NULL);
+}
+
+int
+dict_print_if (dict_t *dict,
+ dict_entry_predicate_t predicate)
+{
+ return dict_print_of_length_if (dict, 0, 0, predicate);
+}
+
+int
+dict_print (dict_t *dict)
+{
+ return dict_print_if (dict, NULL);
+}