* 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;
-void *
+typedef bool_t
+(*trie_predicate_t) (trie_t *trie);
+
+#define TRIE_CHAR_TO_INDEX(c) (tolower (c) - 'a')
+#define TRIE_INDEX_TO_CHAR(i) (i + 'a')
+
+static void *
xmalloc (size_t size)
{
void *res;
return res;
}
-void *
+static void *
xrealloc (void *ptr, size_t size)
{
void *res;
return res;
}
-void
+static void
chomp (char *s)
{
int len = strlen (s);
+ if (len == 0)
+ return;
if (s[len - 1] == '\n')
s[len - 1] = '\0';
}
-void
+static void
string_init (string_t *string)
{
string->size = 0;
string->len = 0;
}
-void
+static void
string_fini (string_t *string)
{
string->size = 0;
string->len = 0;
}
-void
+static void
string_append_char (string_t *string, char c)
{
if (string->size == 0) {
string->s[string->len] = '\0';
}
-void
+static void
string_chop (string_t *string)
{
if (string->len == 0)
trie_destroy (trie->next[i]);
}
-trie_t *
+static trie_t *
trie_create (void)
{
trie_t *trie;
free (trie);
}
-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);
}
-trie_t *
+static trie_t *
trie_find (trie_t *trie,
const char *word_chunk)
{
return trie;
}
-int
+static int
+trie_count (trie_t *trie,
+ trie_predicate_t predicate)
+{
+ int i;
+ int count = 0;
+
+ if ((predicate) (trie))
+ count = 1;
+
+ for (i = 0; i < 26; i++) {
+ if (trie->next[i] == NULL)
+ continue;
+
+ count += trie_count (trie->next[i], predicate);
+ }
+
+ return count;
+}
+
+static int
trie_print (trie_t *trie,
string_t *string,
- trie_predicate_t predicate)
+ trie_predicate_t predicate,
+ int length,
+ int min_length,
+ int max_length)
{
char c;
int i;
int count = 0;
- if (trie->flags & TRIE_FLAGS_IS_WORD
- && (predicate == NULL || predicate (trie)))
+ if (length >= min_length && (predicate) (trie))
{
count = 1;
printf ("%s ", string->s);
}
+ if (length == max_length)
+ return count;
+
/* Loop over each element, appending the character and recursing. */
for (i = 0; i < 26; i++) {
if (trie->next[i] == NULL)
c = TRIE_INDEX_TO_CHAR (i);
string_append_char (string, c);
- count += trie_print (trie->next[i], string, predicate);
+ count += trie_print (trie->next[i], string, predicate,
+ length + 1, min_length, max_length);
string_chop (string);
}
return count;
}
-bool_t
-trie_seen_predicate (trie_t *trie)
-{
- return (trie->flags & TRIE_FLAGS_SEEN);
-}
-
-int
-trie_print_seen (trie_t *trie, string_t *word)
-{
- return trie_print (trie, word, trie_seen_predicate);
-}
-
-bool_t
-trie_unseen_predicate (trie_t *trie)
-{
- return (! trie_seen_predicate (trie));
-}
-
-int
-trie_print_unseen (trie_t *trie, string_t *word)
-{
- return trie_print (trie, word, trie_unseen_predicate);
-}
-
void
dict_init (dict_t *dict)
{
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
return &trie->flags;
}
-void
+/* XXX: This static function pointer is really nasty and definitely
+ * un-thread safe. What I really want it a lambda so I can construct a
+ * composite predicate on the fly. Oh well... */
+static dict_entry_predicate_t dict_entry_predicate = NULL;
+
+static bool_t
+dict_predicate (trie_t *trie)
+{
+ dict_entry_t *entry;
+
+ entry = dict_cursor_resolve (trie);
+ if (! DICT_ENTRY_IS_WORD (entry))
+ return FALSE;
+
+ if (dict_entry_predicate)
+ return (dict_entry_predicate) (*entry);
+
+ return TRUE;
+}
+
+int
+dict_count (dict_t *dict,
+ dict_entry_predicate_t predicate)
+{
+ dict_entry_predicate = predicate;
+ return trie_count (dict, dict_predicate);
+}
+
+int
dict_print (dict_t *dict)
{
+ int count;
string_t string;
string_init (&string);
- trie_print (dict, &string, NULL);
+ dict_entry_predicate = NULL;
+ count = trie_print (dict, &string, dict_predicate,
+ 0, 0, -1);
string_fini (&string);
+
+ return count;
+}
+
+int
+dict_print_if (dict_t *dict,
+ dict_entry_predicate_t predicate)
+{
+ int count;
+ string_t string;
+
+ string_init (&string);
+
+ dict_entry_predicate = predicate;
+ count = trie_print (dict, &string, dict_predicate,
+ 0, 0, -1);
+
+ string_fini (&string);
+
+ return count;
+}
+
+int
+dict_print_by_length_if (dict_t *dict,
+ dict_entry_predicate_t predicate)
+{
+ int length, total, words, count = 0;
+ string_t string;
+
+ string_init (&string);
+
+ dict_entry_predicate = predicate;
+
+ total = trie_count (dict, dict_predicate);
+
+ length = 1;
+ do {
+ words = trie_print (dict, &string, dict_predicate,
+ 0, length, length);
+ if (words) {
+ printf ("\n");
+ count += words;
+ }
+ length++;
+ } while (count < total);
+
+ string_fini (&string);
+
+ return count;
}