]> git.xonotic.org Git - xonotic/gmqcc.git/blobdiff - correct.c
An evil optimization to the trie, now has a fixed amount of branches and uses a char...
[xonotic/gmqcc.git] / correct.c
index cd2ab2d9331bde1c6c1a8dd9d8743e599222a6b6..bd79c63cdf7ae12892aeb5e9e03e144e2bdc524c 100644 (file)
--- a/correct.c
+++ b/correct.c
  * and speeds us up a little.  It also saves us space in a way since each
  * allocation isn't wasting a little header space for when NOTRACK isn't
  * defined.
- */    
+ */
 static unsigned char **correct_pool_data = NULL;
 static unsigned char  *correct_pool_this = NULL;
 static size_t          correct_pool_addr = 0;
@@ -172,6 +172,26 @@ static GMQCC_INLINE char *correct_pool_claim(const char *data) {
     return claim;
 }
 
+/*
+ * _ is valid in identifiers. I've yet to implement numerics however
+ * because they're only valid after the first character is of a _, or
+ * alpha character.
+ */
+static const char correct_alpha[] = "abcdefghijklmnopqrstuvwxyz"
+                                    "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
+                                    "_"; /* TODO: Numbers ... */
+
+static const size_t correct_alpha_index[0x80] = {
+     0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
+     0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
+     0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
+     0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
+     0,  0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14,
+    15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25,  0,  0,  0,  0, 52,
+     0, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40,
+    41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51,  0,  0,  0,  0,  0
+};
+
 /*
  * A fast space efficent trie for a dictionary of identifiers.  This is
  * faster than a hashtable for one reason.  A hashtable itself may have
@@ -179,7 +199,7 @@ static GMQCC_INLINE char *correct_pool_claim(const char *data) {
  * have one of the fastest hash functions for strings, but if you do a
  * lost of hashing (which we do, almost 3 million hashes per identifier)
  * a hashtable becomes slow.
- */   
+ */
 correct_trie_t* correct_trie_new() {
     correct_trie_t *t = (correct_trie_t*)mem_a(sizeof(correct_trie_t));
     t->value   = NULL;
@@ -189,16 +209,21 @@ correct_trie_t* correct_trie_new() {
 
 void correct_trie_del_sub(correct_trie_t *t) {
     size_t i;
-    for (i = 0; i < vec_size(t->entries); ++i)
+    if (!t->entries)
+        return;
+    for (i = 0; i < sizeof(correct_alpha)-1; ++i) {
         correct_trie_del_sub(&t->entries[i]);
-    vec_free(t->entries);
+    }
+    mem_d(t->entries);
 }
 
 void correct_trie_del(correct_trie_t *t) {
     size_t i;
-    for (i = 0; i < vec_size(t->entries); ++i)
-        correct_trie_del_sub(&t->entries[i]);
-    vec_free(t->entries);
+    if (t->entries) {
+        for (i = 0; i < sizeof(correct_alpha)-1; ++i)
+            correct_trie_del_sub(&t->entries[i]);
+        mem_d(t->entries);
+    }
     mem_d(t);
 }
 
@@ -206,20 +231,10 @@ void* correct_trie_get(const correct_trie_t *t, const char *key) {
     const unsigned char *data = (const unsigned char*)key;
 
     while (*data) {
-        const correct_trie_t *entries = t->entries;
-        unsigned char         ch      = *data;
-        const size_t          vs      = vec_size(entries);
-        size_t                i;
-
-        for (i = 0; i < vs; ++i) {
-            if (entries[i].ch == ch) {
-                t = &entries[i];
-                ++data;
-                break;
-            }
-        }
-        if (i == vs)
+        if (!t->entries)
             return NULL;
+        t = t->entries + correct_alpha_index[*data];
+        ++data;
     }
     return t->value;
 }
@@ -227,25 +242,11 @@ void* correct_trie_get(const correct_trie_t *t, const char *key) {
 void correct_trie_set(correct_trie_t *t, const char *key, void * const value) {
     const unsigned char *data = (const unsigned char*)key;
     while (*data) {
-        correct_trie_t *entries = t->entries;
-        const size_t    vs      = vec_size(entries);
-        unsigned char   ch      = *data;
-        size_t          i;
-
-        for (i = 0; i < vs; ++i) {
-            if (entries[i].ch == ch) {
-                t = &entries[i];
-                break;
-            }
-        }
-        if (i == vs) {
-            correct_trie_t *elem  = (correct_trie_t*)vec_add(t->entries, 1);
-
-            elem->ch      = ch;
-            elem->value   = NULL;
-            elem->entries = NULL;
-            t             = elem;
+        if (!t->entries) {
+            t->entries = (correct_trie_t*)mem_a(sizeof(correct_trie_t)*(sizeof(correct_alpha)-1));
+            memset(t->entries, 0, sizeof(correct_trie_t)*(sizeof(correct_alpha)-1));
         }
+        t = t->entries + correct_alpha_index[*data];
         ++data;
     }
     t->value = value;
@@ -255,7 +256,7 @@ void correct_trie_set(correct_trie_t *t, const char *key, void * const value) {
 /*
  * Implementation of the corrector algorithm commences. A very efficent
  * brute-force attack (thanks to tries and mempool :-)).
- */  
+ */
 static GMQCC_INLINE size_t *correct_find(correct_trie_t *table, const char *word) {
     return (size_t*)correct_trie_get(table, word);
 }
@@ -272,7 +273,7 @@ static GMQCC_INLINE bool correct_update(correct_trie_t* *table, const char *word
 void correct_add(correct_trie_t* table, size_t ***size, const char *ident) {
     size_t     *data = NULL;
     const char *add  = ident;
-    
+
     if (!correct_update(&table, add)) {
         data  = (size_t*)mem_a(sizeof(size_t));
         *data = 1;
@@ -293,15 +294,6 @@ void correct_del(correct_trie_t* dictonary, size_t **data) {
     correct_trie_del(dictonary);
 }
 
-/*
- * _ is valid in identifiers. I've yet to implement numerics however
- * because they're only valid after the first character is of a _, or
- * alpha character.
- */
-static const char correct_alpha[] = "abcdefghijklmnopqrstuvwxyz"
-                                    "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
-                                    "_"; /* TODO: Numbers ... */
-
 /*
  * correcting logic for the following forms of transformations:
  *  1) deletion
@@ -318,7 +310,7 @@ static const char correct_alpha[] = "abcdefghijklmnopqrstuvwxyz"
  * need to take a size_t ** to carry it along (would all the argument
  * overhead be worth it?)  
  */
-static size_t correct_deletion(const char *ident, char **array, size_t index) {
+static size_t correct_deletion(const char *ident, char **array) {
     size_t       itr = 0;
     const size_t len = strlen(ident);
 
@@ -326,13 +318,13 @@ static size_t correct_deletion(const char *ident, char **array, size_t index) {
         char *a = (char*)correct_pool_alloc(len+1);
         memcpy(a, ident, itr);
         memcpy(a + itr, ident + itr + 1, len - itr);
-        array[index + itr] = a;
+        array[itr] = a;
     }
 
     return itr;
 }
 
-static size_t correct_transposition(const char *ident, char **array, size_t index) {
+static size_t correct_transposition(const char *ident, char **array) {
     size_t       itr = 0;
     const size_t len = strlen(ident);
 
@@ -343,13 +335,13 @@ static size_t correct_transposition(const char *ident, char **array, size_t inde
         tmp      = a[itr];
         a[itr  ] = a[itr+1];
         a[itr+1] = tmp;
-        array[index + itr] = a;
+        array[itr] = a;
     }
 
     return itr;
 }
 
-static size_t correct_alteration(const char *ident, char **array, size_t index) {
+static size_t correct_alteration(const char *ident, char **array) {
     size_t       itr = 0;
     size_t       jtr = 0;
     size_t       ktr = 0;
@@ -360,30 +352,29 @@ static size_t correct_alteration(const char *ident, char **array, size_t index)
             char *a = (char*)correct_pool_alloc(len+1);
             memcpy(a, ident, len+1);
             a[itr] = correct_alpha[jtr];
-            array[index + ktr] = a;
+            array[ktr] = a;
         }
     }
 
     return ktr;
 }
 
-static size_t correct_insertion(const char *ident, char **array, size_t index) {
+static size_t correct_insertion(const char *ident, char **array) {
     size_t       itr = 0;
     size_t       jtr = 0;
-    size_t       ktr = 0;
     const size_t len = strlen(ident);
 
     for (; itr <= len; itr++) {
-        for (jtr = 0; jtr < sizeof(correct_alpha)-1; jtr++, ktr++) {
+        for (jtr = 0; jtr < sizeof(correct_alpha)-1; jtr++) {
             char *a = (char*)correct_pool_alloc(len+2);
             memcpy(a, ident, itr);
             memcpy(a + itr + 1, ident + itr, len - itr + 1);
             a[itr] = correct_alpha[jtr];
-            array[index + ktr] = a;
+            array[itr * (sizeof(correct_alpha)-1) + jtr] = a;
         }
     }
 
-    return ktr;
+    return (len+1)*(sizeof(correct_alpha)-1);
 }
 
 static GMQCC_INLINE size_t correct_size(const char *ident) {
@@ -392,7 +383,7 @@ static GMQCC_INLINE size_t correct_size(const char *ident) {
      * transposition = len - 1
      * alteration    = len * sizeof(correct_alpha)
      * insertion     = (len + 1) * sizeof(correct_alpha)
-     */   
+     */
 
     register size_t len = strlen(ident);
     return (len) + (len - 1) + (len * (sizeof(correct_alpha)-1)) + ((len + 1) * (sizeof(correct_alpha)-1));
@@ -405,10 +396,10 @@ static char **correct_edit(const char *ident) {
     if (!find)
         return NULL;
 
-    next  = correct_deletion     (ident, find, 0);
-    next += correct_transposition(ident, findnext);
-    next += correct_alteration   (ident, findnext);
-    /*****/ correct_insertion    (ident, findnext);
+    next  = correct_deletion     (ident, find);
+    next += correct_transposition(ident, find+next);
+    next += correct_alteration   (ident, find+next);
+    /*****/ correct_insertion    (ident, find+next);
 
     return find;
 }
@@ -417,7 +408,7 @@ static char **correct_edit(const char *ident) {
  * We could use a hashtable but the space complexity isn't worth it
  * since we're only going to determine the "did you mean?" identifier
  * on error.
- */   
+ */
 static int correct_exist(char **array, size_t rows, char *ident) {
     size_t itr;
     /*
@@ -429,7 +420,7 @@ static int correct_exist(char **array, size_t rows, char *ident) {
      * cmpl %eax, %ebx      ; ebx = &LHS[END_POS]
      *
      * jbe correct_cmp_eq
-     * movb (%edx), %cl     ; micro-optimized on even atoms :-)
+     * movb (%edx), %cl     ; micro-optimized even on atoms :-)
      * cmpb %cl, (%eax)     ; ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
      * jg  correct_cmp_gt
      * jge correct_cmp_loop
@@ -457,12 +448,13 @@ static int correct_exist(char **array, size_t rows, char *ident) {
 static GMQCC_INLINE char **correct_known_resize(char **res, size_t *allocated, size_t size) {
     size_t oldallocated = *allocated;
     char **out;
-    if (size+1 < *allocated)
+    if (size < oldallocated)
         return res;
 
-    *allocated += 32;
-    out = correct_pool_alloc(sizeof(*res) * *allocated);
+    out = correct_pool_alloc(sizeof(*res) * oldallocated + 32);
     memcpy(out, res, sizeof(*res) * oldallocated);
+
+    *allocated += 32;
     return out;
 }
 
@@ -511,10 +503,7 @@ static char *correct_maximum(correct_trie_t* table, char **array, size_t rows) {
 /*
  * This is the exposed interface:
  * takes a table for the dictonary a vector of sizes (used for internal
- * probability calculation, and an identifier to "correct"
- *
- * the add function works the same.  Except the identifier is used to
- * add to the dictonary.  
+ * probability calculation), and an identifier to "correct".
  */
 char *correct_str(correct_trie_t* table, const char *ident) {
     char **e1      = NULL;