X-Git-Url: http://git.indexdata.com/?a=blobdiff_plain;f=src%2Ftermlists.c;h=8f06a47dbbbfa8282068a02cfb1d098503d09bf1;hb=817e3ec506c4095bc4fcc1923cee36153ef4ee43;hp=9741eb253103fb31373ce8c4f6a17dd4f3841dc4;hpb=2de4cab9b87f848767078447142668fc3c30e5c9;p=pazpar2-moved-to-github.git diff --git a/src/termlists.c b/src/termlists.c index 9741eb2..8f06a47 100644 --- a/src/termlists.c +++ b/src/termlists.c @@ -1,5 +1,5 @@ /* This file is part of Pazpar2. - Copyright (C) 2006-2011 Index Data + Copyright (C) Index Data Pazpar2 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 @@ -21,6 +21,7 @@ Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA #include #endif +#include #include #include #include @@ -44,15 +45,11 @@ struct termlist struct termlist_bucket **hashtable; unsigned hash_size; - struct termlist_score **highscore; - int highscore_size; - int highscore_num; - int highscore_min; - + int no_entries; NMEM nmem; }; -struct termlist *termlist_create(NMEM nmem, int highscore_size) +struct termlist *termlist_create(NMEM nmem) { struct termlist *res = nmem_malloc(nmem, sizeof(struct termlist)); res->hash_size = 399; @@ -60,74 +57,26 @@ struct termlist *termlist_create(NMEM nmem, int highscore_size) nmem_malloc(nmem, res->hash_size * sizeof(struct termlist_bucket*)); memset(res->hashtable, 0, res->hash_size * sizeof(struct termlist_bucket*)); res->nmem = nmem; - - res->highscore = nmem_malloc(nmem, highscore_size * sizeof(struct termlist_score *)); - res->highscore_size = highscore_size; - res->highscore_num = 0; - res->highscore_min = 0; - + res->no_entries = 0; return res; } -static void update_highscore(struct termlist *tl, struct termlist_score *t) -{ - int i; - int smallest; - int me = -1; - - if (tl->highscore_num > tl->highscore_size && t->frequency < tl->highscore_min) - return; - - smallest = 0; - for (i = 0; i < tl->highscore_num; i++) - { - if (tl->highscore[i]->frequency < tl->highscore[smallest]->frequency) - smallest = i; - if (tl->highscore[i] == t) - me = i; - } - if (tl->highscore_num) - tl->highscore_min = tl->highscore[smallest]->frequency; - if (t->frequency < tl->highscore_min) - tl->highscore_min = t->frequency; - if (me >= 0) - return; - if (tl->highscore_num < tl->highscore_size) - { - tl->highscore[tl->highscore_num++] = t; - if (t->frequency < tl->highscore_min) - tl->highscore_min = t->frequency; - } - else - { - if (t->frequency > tl->highscore[smallest]->frequency) - { - tl->highscore[smallest] = t; - } - } -} - void termlist_insert(struct termlist *tl, const char *display_term, const char *norm_term, int freq) { unsigned int bucket; struct termlist_bucket **p; - char buf[256], *cp; + char buf[256]; if (strlen(norm_term) > 255) return; strcpy(buf, norm_term); - /* chop right */ - for (cp = buf + strlen(buf); cp != buf && strchr(",. -", cp[-1]); cp--) - cp[-1] = '\0'; - bucket = jenkins_hash((unsigned char *)buf) % tl->hash_size; for (p = &tl->hashtable[bucket]; *p; p = &(*p)->next) { if (!strcmp(buf, (*p)->term.norm_term)) { (*p)->term.frequency += freq; - update_highscore(tl, &((*p)->term)); break; } } @@ -141,21 +90,39 @@ void termlist_insert(struct termlist *tl, const char *display_term, new->term.frequency = freq; new->next = 0; *p = new; - update_highscore(tl, &new->term); + tl->no_entries++; } } static int compare(const void *s1, const void *s2) { - struct termlist_score **p1 = (struct termlist_score**) s1, **p2 = (struct termlist_score **) s2; - return (*p2)->frequency - (*p1)->frequency; + struct termlist_score **p1 = (struct termlist_score **) s1; + struct termlist_score **p2 = (struct termlist_score **) s2; + int d = (*p2)->frequency - (*p1)->frequency; + if (d) + return d; + return strcmp((*p1)->display_term, (*p2)->display_term); } -struct termlist_score **termlist_highscore(struct termlist *tl, int *len) +struct termlist_score **termlist_highscore(struct termlist *tl, int *len, + NMEM nmem) { - qsort(tl->highscore, tl->highscore_num, sizeof(struct termlist_score*), compare); - *len = tl->highscore_num; - return tl->highscore; + struct termlist_score **highscore = + (struct termlist_score **) + nmem_malloc(nmem, tl->no_entries * sizeof(*highscore)); + + int no = 0; + unsigned bucket; + for (bucket = 0; bucket < tl->hash_size; bucket++) + { + struct termlist_bucket *p; + for (p = tl->hashtable[bucket]; p; p = p->next) + highscore[no++] = &p->term; + } + assert(no == tl->no_entries); + qsort(highscore, tl->no_entries, sizeof(struct termlist_score*), compare); + *len = tl->no_entries; + return highscore; } /*