index/zsets.c

   1 /* $Id: zsets.c,v 1.101 2006-05-10 08:13:26 adam Exp $
   2    Copyright (C) 1995-2005
   3    Index Data ApS
   4
   5 This file is part of the Zebra server.
   6
   7 Zebra is free software; you can redistribute it and/or modify it under
   8 the terms of the GNU General Public License as published by the Free
   9 Software Foundation; either version 2, or (at your option) any later
  10 version.
  11
  12 Zebra is distributed in the hope that it will be useful, but WITHOUT ANY
  13 WARRANTY; without even the implied warranty of MERCHANTABILITY or
  14 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  15 for more details.
  16
  17 You should have received a copy of the GNU General Public License
  18 along with Zebra; see the file LICENSE.zebra.  If not, write to the
  19 Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
  20 02111-1307, USA.
  21 */
  22
  23
  24 #include <stdio.h>
  25 #include <assert.h>
  26 #ifdef WIN32
  27 #include <io.h>
  28 #else
  29 #include <unistd.h>
  30 #endif
  31
  32 #include "index.h"
  33 #include "rank.h"
  34 #include <yaz/diagbib1.h>
  35 #include <rset.h>
  36
  37 #define SORT_IDX_ENTRYSIZE 64
  38 #define ZSET_SORT_MAX_LEVEL 3
  39
  40 struct zebra_set_term_entry {
  41     int reg_type;
  42     char *db;
  43     int set;
  44     int use;
  45     char *term;
  46 };
  47
  48 struct zebra_set {
  49     char *name;
  50     RSET rset;
  51     NMEM nmem;
  52     NMEM rset_nmem; /* for creating the rsets in */
  53     zint hits;
  54     int num_bases;
  55     char **basenames;
  56     Z_RPNQuery *rpn;
  57     Z_SortKeySpecList *sortSpec;
  58     struct zset_sort_info *sort_info;
  59     struct zebra_set_term_entry *term_entries;
  60     int term_entries_max;
  61     struct zebra_set *next;
  62     int locked;
  63
  64     zint cache_position;  /* last position */
  65     RSFD cache_rfd;       /* rfd (NULL if not existing) */
  66     zint cache_psysno;    /* sysno for last position */
  67     zint approx_limit;    /* limit before we do approx */
  68 };
  69
  70 struct zset_sort_entry {
  71     zint sysno;
  72     int score;
  73     char buf[ZSET_SORT_MAX_LEVEL][SORT_IDX_ENTRYSIZE];
  74 };
  75
  76 struct zset_sort_info {
  77     int max_entries;
  78     int num_entries;
  79     struct zset_sort_entry *all_entries;
  80     struct zset_sort_entry **entries;
  81 };
  82
  83 static int log_level_set=0;
  84 static int log_level_sort=0;
  85 static int log_level_searchhits=0;
  86 static int log_level_searchterms=0;
  87 static int log_level_resultsets=0;
  88
  89 static void loglevels()
  90 {
  91     if (log_level_set)
  92         return;
  93     log_level_sort = yaz_log_module_level("sorting");
  94     log_level_searchhits = yaz_log_module_level("searchhits");
  95     log_level_searchterms = yaz_log_module_level("searchterms");
  96     log_level_resultsets = yaz_log_module_level("resultsets");
  97     log_level_set = 1;
  98 }
  99
 100 ZEBRA_RES resultSetSearch(ZebraHandle zh, NMEM nmem, NMEM rset_nmem,
 101                           Z_RPNQuery *rpn, ZebraSet sset)
 102 {
 103     RSET rset = 0;
 104     oident *attrset;
 105     Z_SortKeySpecList *sort_sequence;
 106     int sort_status, i;
 107     ZEBRA_RES res = ZEBRA_OK;
 108
 109     zh->hits = 0;
 110
 111     sort_sequence = (Z_SortKeySpecList *)
 112         nmem_malloc(nmem, sizeof(*sort_sequence));
 113     sort_sequence->num_specs = 10; /* FIXME - Hard-coded number */
 114     sort_sequence->specs = (Z_SortKeySpec **)
 115         nmem_malloc(nmem, sort_sequence->num_specs *
 116                      sizeof(*sort_sequence->specs));
 117     for (i = 0; i<sort_sequence->num_specs; i++)
 118         sort_sequence->specs[i] = 0;
 119
 120     attrset = oid_getentbyoid (rpn->attributeSetId);
 121     res = rpn_search_top(zh, rpn->RPNStructure, attrset->value,
 122                          nmem, rset_nmem,
 123                          sort_sequence,
 124                          sset->num_bases, sset->basenames,
 125                          &rset);
 126     if (res != ZEBRA_OK)
 127     {
 128         sset->rset = 0;
 129         return res;
 130     }
 131     for (i = 0; sort_sequence->specs[i]; i++)
 132         ;
 133     sort_sequence->num_specs = i;
 134     rset->hits_limit = sset->approx_limit;
 135     if (!i)
 136     {
 137         res = resultSetRank (zh, sset, rset, rset_nmem);
 138     }
 139     else
 140     {
 141         res = resultSetSortSingle (zh, nmem, sset, rset,
 142                                    sort_sequence, &sort_status);
 143     }
 144     sset->rset = rset;
 145     return res;
 146 }
 147
 148
 149 ZEBRA_RES resultSetAddRPN (ZebraHandle zh, NMEM m, Z_RPNQuery *rpn,
 150                            int num_bases, char **basenames,
 151                            const char *setname)
 152 {
 153     ZebraSet zebraSet;
 154     int i;
 155     ZEBRA_RES res;
 156
 157     zh->hits = 0;
 158
 159     zebraSet = resultSetAdd(zh, setname, 1);
 160     if (!zebraSet)
 161         return ZEBRA_FAIL;
 162     zebraSet->locked = 1;
 163     zebraSet->rpn = 0;
 164     zebraSet->nmem = m;
 165     zebraSet->rset_nmem = nmem_create();
 166
 167     zebraSet->num_bases = num_bases;
 168     zebraSet->basenames =
 169         nmem_malloc (zebraSet->nmem, num_bases * sizeof(*zebraSet->basenames));
 170     for (i = 0; i<num_bases; i++)
 171         zebraSet->basenames[i] = nmem_strdup(zebraSet->nmem, basenames[i]);
 172
 173     res = resultSetSearch(zh, zebraSet->nmem, zebraSet->rset_nmem,
 174                           rpn, zebraSet);
 175     zh->hits = zebraSet->hits;
 176     if (zebraSet->rset)
 177         zebraSet->rpn = rpn;
 178     zebraSet->locked = 0;
 179     if (!zebraSet->rset)
 180         return ZEBRA_FAIL;
 181     return res;
 182 }
 183
 184 void resultSetAddTerm (ZebraHandle zh, ZebraSet s, int reg_type,
 185                        const char *db, int set,
 186                        int use, const char *term)
 187 {
 188     assert(zh); /* compiler shut up */
 189     if (!s->nmem)
 190         s->nmem = nmem_create ();
 191     if (!s->term_entries)
 192     {
 193         int i;
 194         s->term_entries_max = 1000;
 195         s->term_entries =
 196             nmem_malloc (s->nmem, s->term_entries_max *
 197                          sizeof(*s->term_entries));
 198         for (i = 0; i < s->term_entries_max; i++)
 199             s->term_entries[i].term = 0;
 200     }
 201     if (s->hits < s->term_entries_max)
 202     {
 203         s->term_entries[s->hits].reg_type = reg_type;
 204         s->term_entries[s->hits].db = nmem_strdup (s->nmem, db);
 205         s->term_entries[s->hits].set = set;
 206         s->term_entries[s->hits].use = use;
 207         s->term_entries[s->hits].term = nmem_strdup (s->nmem, term);
 208     }
 209     (s->hits)++;
 210 }
 211
 212 ZebraSet resultSetAdd(ZebraHandle zh, const char *name, int ov)
 213 {
 214     ZebraSet s;
 215     int i;
 216
 217     for (s = zh->sets; s; s = s->next)
 218         if (!strcmp (s->name, name))
 219             break;
 220
 221     if (!log_level_set)
 222         loglevels();
 223     if (s)
 224     {
 225         yaz_log(log_level_resultsets, "updating result set %s", name);
 226         if (!ov || s->locked)
 227             return NULL;
 228         if (s->rset)
 229         {
 230             if (s->cache_rfd)
 231                 rset_close(s->cache_rfd);
 232             rset_delete (s->rset);
 233         }
 234         if (s->rset_nmem)
 235             nmem_destroy (s->rset_nmem);
 236         if (s->nmem)
 237             nmem_destroy (s->nmem);
 238     }
 239     else
 240     {
 241         const char *sort_max_str = zebra_get_resource(zh, "sortmax", "1000");
 242
 243         yaz_log(log_level_resultsets, "adding result set %s", name);
 244         s = (ZebraSet) xmalloc (sizeof(*s));
 245         s->next = zh->sets;
 246         zh->sets = s;
 247         s->name = (char *) xmalloc (strlen(name)+1);
 248         strcpy (s->name, name);
 249
 250         s->sort_info = (struct zset_sort_info *)
 251             xmalloc (sizeof(*s->sort_info));
 252         s->sort_info->max_entries = atoi(sort_max_str);
 253         if (s->sort_info->max_entries < 2)
 254             s->sort_info->max_entries = 2;
 255
 256         s->sort_info->entries = (struct zset_sort_entry **)
 257             xmalloc (sizeof(*s->sort_info->entries) *
 258                      s->sort_info->max_entries);
 259         s->sort_info->all_entries = (struct zset_sort_entry *)
 260             xmalloc (sizeof(*s->sort_info->all_entries) *
 261                      s->sort_info->max_entries);
 262         for (i = 0; i < s->sort_info->max_entries; i++)
 263             s->sort_info->entries[i] = s->sort_info->all_entries + i;
 264     }
 265     s->locked = 0;
 266     s->term_entries = 0;
 267     s->hits = 0;
 268     s->rset = 0;
 269     s->rset_nmem = 0;
 270     s->nmem = 0;
 271     s->rpn = 0;
 272     s->sortSpec = 0;
 273     s->cache_position = 0;
 274     s->cache_rfd = 0;
 275     s->approx_limit = zh->approx_limit;
 276     return s;
 277 }
 278
 279 ZebraSet resultSetGet(ZebraHandle zh, const char *name)
 280 {
 281     ZebraSet s;
 282
 283     for (s = zh->sets; s; s = s->next)
 284         if (!strcmp (s->name, name))
 285         {
 286             if (!s->term_entries && !s->rset && s->rpn)
 287             {
 288                 NMEM nmem = nmem_create ();
 289                 yaz_log(log_level_resultsets, "research %s", name);
 290                 if (!s->rset_nmem)
 291                     s->rset_nmem=nmem_create();
 292                 resultSetSearch(zh, nmem, s->rset_nmem, s->rpn, s);
 293                 if (s->rset && s->sortSpec)
 294                 {
 295                     int sort_status;
 296                     yaz_log(log_level_resultsets, "resort %s", name);
 297                     resultSetSortSingle (zh, nmem, s, s->rset, s->sortSpec,
 298                                          &sort_status);
 299                 }
 300                 nmem_destroy (nmem);
 301             }
 302             return s;
 303         }
 304     return NULL;
 305 }
 306
 307 void resultSetInvalidate (ZebraHandle zh)
 308 {
 309     ZebraSet s = zh->sets;
 310
 311     yaz_log(log_level_resultsets, "invalidating result sets");
 312     for (; s; s = s->next)
 313     {
 314         if (s->rset)
 315         {
 316             if (s->cache_rfd)
 317                 rset_close(s->cache_rfd);
 318             rset_delete (s->rset);
 319         }
 320         s->rset = 0;
 321         s->cache_rfd = 0;
 322         s->cache_position = 0;
 323         if (s->rset_nmem)
 324             nmem_destroy(s->rset_nmem);
 325         s->rset_nmem=0;
 326     }
 327 }
 328
 329 void resultSetDestroy(ZebraHandle zh, int num, char **names,int *statuses)
 330 {
 331     ZebraSet * ss = &zh->sets;
 332     int i;
 333
 334     if (statuses)
 335         for (i = 0; i<num; i++)
 336             statuses[i] = Z_DeleteStatus_resultSetDidNotExist;
 337     while (*ss)
 338     {
 339         int i = -1;
 340         ZebraSet s = *ss;
 341         if (num >= 0)
 342         {
 343             for (i = 0; i<num; i++)
 344                 if (!strcmp (s->name, names[i]))
 345                 {
 346                     if (statuses)
 347                         statuses[i] = Z_DeleteStatus_success;
 348                     i = -1;
 349                     break;
 350                 }
 351         }
 352         if (i < 0)
 353         {
 354             *ss = s->next;
 355
 356             xfree (s->sort_info->all_entries);
 357             xfree (s->sort_info->entries);
 358             xfree (s->sort_info);
 359
 360             if (s->nmem)
 361                 nmem_destroy (s->nmem);
 362             if (s->rset)
 363             {
 364                 if (s->cache_rfd)
 365                     rset_close(s->cache_rfd);
 366                 rset_delete (s->rset);
 367             }
 368             if (s->rset_nmem)
 369                 nmem_destroy(s->rset_nmem);
 370             xfree (s->name);
 371             xfree (s);
 372         }
 373         else
 374             ss = &s->next;
 375     }
 376 }
 377
 378 ZebraMetaRecord *zebra_meta_records_create_range(ZebraHandle zh,
 379                                                  const char *name,
 380                                                  zint start, int num)
 381 {
 382     zint pos_small[10];
 383     zint *pos = pos_small;
 384     ZebraMetaRecord *mr;
 385     int i;
 386
 387     if (num > 10000 || num <= 0)
 388         return 0;
 389
 390     if (num > 10)
 391         pos = xmalloc(sizeof(*pos) * num);
 392
 393     for (i = 0; i<num; i++)
 394         pos[i] = start+i;
 395
 396     mr = zebra_meta_records_create(zh, name, num, pos);
 397
 398     if (num > 10)
 399         xfree(pos);
 400     return mr;
 401 }
 402
 403 ZebraMetaRecord *zebra_meta_records_create(ZebraHandle zh, const char *name,
 404                                            int num, zint *positions)
 405 {
 406     ZebraSet sset;
 407     ZebraMetaRecord *sr = 0;
 408     RSET rset;
 409     int i;
 410     struct zset_sort_info *sort_info;
 411     size_t sysno_mem_index = 0;
 412
 413     if (zh->m_staticrank)
 414         sysno_mem_index = 1;
 415
 416     if (!log_level_set)
 417         loglevels();
 418     if (!(sset = resultSetGet (zh, name)))
 419         return NULL;
 420     if (!(rset = sset->rset))
 421     {
 422         if (!sset->term_entries)
 423             return 0;
 424         sr = (ZebraMetaRecord *) xmalloc (sizeof(*sr) * num);
 425         for (i = 0; i<num; i++)
 426         {
 427             sr[i].sysno = 0;
 428             sr[i].score = -1;
 429             sr[i].term = 0;
 430             sr[i].db = 0;
 431
 432             if (positions[i] <= sset->term_entries_max)
 433             {
 434                 sr[i].term = sset->term_entries[positions[i]-1].term;
 435                 sr[i].db = sset->term_entries[positions[i]-1].db;
 436             }
 437         }
 438     }
 439     else
 440     {
 441         sr = (ZebraMetaRecord *) xmalloc (sizeof(*sr) * num);
 442         for (i = 0; i<num; i++)
 443         {
 444             sr[i].sysno = 0;
 445             sr[i].score = -1;
 446             sr[i].term = 0;
 447             sr[i].db = 0;
 448         }
 449         sort_info = sset->sort_info;
 450         if (sort_info)
 451         {
 452             zint position;
 453
 454             for (i = 0; i<num; i++)
 455             {
 456                 position = positions[i];
 457                 if (position > 0 && position <= sort_info->num_entries)
 458                 {
 459                     yaz_log(log_level_sort, "got pos=" ZINT_FORMAT
 460                             " (sorted)", position);
 461                     sr[i].sysno = sort_info->entries[position-1]->sysno;
 462                     sr[i].score = sort_info->entries[position-1]->score;
 463                 }
 464             }
 465         }
 466         /* did we really get all entries using sort ? */
 467         for (i = 0; i<num; i++)
 468         {
 469             if (!sr[i].sysno)
 470                 break;
 471         }
 472         if (i < num) /* nope, get the rest, unsorted - sorry */
 473         {
 474             zint position = 0;
 475             int num_i = 0;
 476             zint psysno = 0;
 477             RSFD rfd;
 478             struct it_key key;
 479
 480             if (sort_info)
 481                 position = sort_info->num_entries;
 482             while (num_i < num && positions[num_i] <= position)
 483                 num_i++;
 484
 485             if (sset->cache_rfd &&
 486                 num_i < num && positions[num_i] > sset->cache_position)
 487             {
 488                 position = sset->cache_position;
 489                 rfd = sset->cache_rfd;
 490                 psysno = sset->cache_psysno;
 491             }
 492             else
 493             {
 494                 if (sset->cache_rfd)
 495                     rset_close(sset->cache_rfd);
 496                 rfd = rset_open (rset, RSETF_READ);
 497             }
 498             while (num_i < num && rset_read (rfd, &key, 0))
 499             {
 500                 zint this_sys = key.mem[sysno_mem_index];
 501                 if (this_sys != psysno)
 502                 {
 503                     psysno = this_sys;
 504                     if (sort_info)
 505                     {
 506                         /* determine we alreay have this in our set */
 507                         for (i = sort_info->num_entries; --i >= 0; )
 508                             if (psysno == sort_info->entries[i]->sysno)
 509                                 break;
 510                         if (i >= 0)
 511                             continue;
 512                     }
 513                     position++;
 514                     assert (num_i < num);
 515                     if (position == positions[num_i])
 516                     {
 517                         sr[num_i].sysno = psysno;
 518                         yaz_log(log_level_sort, "got pos=" ZINT_FORMAT " (unsorted)", position);
 519                         sr[num_i].score = -1;
 520                         num_i++;
 521                     }
 522                 }
 523             }
 524             sset->cache_position = position;
 525             sset->cache_psysno = psysno;
 526             sset->cache_rfd = rfd;
 527         }
 528     }
 529     return sr;
 530 }
 531
 532 void zebra_meta_records_destroy (ZebraHandle zh, ZebraMetaRecord *records,
 533                                  int num)
 534 {
 535     assert(zh); /* compiler shut up about unused arg */
 536     xfree (records);
 537 }
 538
 539 struct sortKeyInfo {
 540     int relation;
 541     int attrUse;
 542     int numerical;
 543 };
 544
 545 void resultSetInsertSort (ZebraHandle zh, ZebraSet sset,
 546                           struct sortKeyInfo *criteria, int num_criteria,
 547                           zint sysno)
 548 {
 549     struct zset_sort_entry this_entry;
 550     struct zset_sort_entry *new_entry = NULL;
 551     struct zset_sort_info *sort_info = sset->sort_info;
 552     int i, j;
 553
 554     sortIdx_sysno (zh->reg->sortIdx, sysno);
 555     for (i = 0; i<num_criteria; i++)
 556     {
 557         sortIdx_type (zh->reg->sortIdx, criteria[i].attrUse);
 558         sortIdx_read (zh->reg->sortIdx, this_entry.buf[i]);
 559     }
 560     i = sort_info->num_entries;
 561     while (--i >= 0)
 562     {
 563         int rel = 0;
 564         for (j = 0; j<num_criteria; j++)
 565         {
 566             if (criteria[j].numerical)
 567             {
 568                 double diff = atof(this_entry.buf[j]) -
 569                               atof(sort_info->entries[i]->buf[j]);
 570                 rel = 0;
 571                 if (diff > 0.0)
 572                     rel = 1;
 573                 else if (diff < 0.0)
 574                     rel = -1;
 575             }
 576             else
 577             {
 578                 rel = memcmp (this_entry.buf[j], sort_info->entries[i]->buf[j],
 579                           SORT_IDX_ENTRYSIZE);
 580             }
 581             if (rel)
 582                 break;
 583         }
 584         if (!rel)
 585             break;
 586         if (criteria[j].relation == 'A')
 587         {
 588             if (rel > 0)
 589                 break;
 590         }
 591         else if (criteria[j].relation == 'D')
 592         {
 593             if (rel < 0)
 594                 break;
 595         }
 596     }
 597     ++i;
 598     j = sort_info->max_entries;
 599     if (i == j)
 600         return;
 601
 602     if (sort_info->num_entries == j)
 603         --j;
 604     else
 605         j = (sort_info->num_entries)++;
 606     new_entry = sort_info->entries[j];
 607     while (j != i)
 608     {
 609         sort_info->entries[j] = sort_info->entries[j-1];
 610         --j;
 611     }
 612     sort_info->entries[i] = new_entry;
 613     assert (new_entry);
 614     for (i = 0; i<num_criteria; i++)
 615         memcpy (new_entry->buf[i], this_entry.buf[i], SORT_IDX_ENTRYSIZE);
 616     new_entry->sysno = sysno;
 617     new_entry->score = -1;
 618 }
 619
 620 void resultSetInsertRank (ZebraHandle zh, struct zset_sort_info *sort_info,
 621                           zint sysno, int score, int relation)
 622 {
 623     struct zset_sort_entry *new_entry = NULL;
 624     int i, j;
 625     assert(zh); /* compiler shut up about unused arg */
 626
 627     i = sort_info->num_entries;
 628     while (--i >= 0)
 629     {
 630         int rel = 0;
 631
 632         rel = score - sort_info->entries[i]->score;
 633
 634         if (relation == 'D')
 635         {
 636             if (rel >= 0)
 637                 break;
 638         }
 639         else if (relation == 'A')
 640         {
 641             if (rel <= 0)
 642                 break;
 643         }
 644     }
 645     ++i;
 646     j = sort_info->max_entries;
 647     if (i == j)
 648         return;
 649
 650     if (sort_info->num_entries == j)
 651         --j;
 652     else
 653         j = (sort_info->num_entries)++;
 654
 655     new_entry = sort_info->entries[j];
 656     while (j != i)
 657     {
 658         sort_info->entries[j] = sort_info->entries[j-1];
 659         --j;
 660     }
 661     sort_info->entries[i] = new_entry;
 662     assert (new_entry);
 663     new_entry->sysno = sysno;
 664     new_entry->score = score;
 665 }
 666
 667 static Z_RPNQuery *copy_RPNQuery(Z_RPNQuery *src, NMEM nmem)
 668 {
 669     Z_RPNQuery *dst = 0;
 670     ODR encode = odr_createmem(ODR_ENCODE);
 671     ODR decode = odr_createmem(ODR_DECODE);
 672
 673     if (z_RPNQuery(encode, &src, 0, 0))
 674     {
 675         int len;
 676         char *buf = odr_getbuf(encode, &len, 0);
 677
 678         if (buf)
 679         {
 680             odr_setbuf(decode, buf, len, 0);
 681             z_RPNQuery(decode, &dst, 0, 0);
 682         }
 683     }
 684     nmem_transfer(nmem, decode->mem);
 685     odr_destroy(encode);
 686     odr_destroy(decode);
 687     return dst;
 688 }
 689
 690 static Z_SortKeySpecList *copy_SortKeySpecList(Z_SortKeySpecList *src, NMEM nmem)
 691 {
 692     Z_SortKeySpecList *dst = 0;
 693     ODR encode = odr_createmem(ODR_ENCODE);
 694     ODR decode = odr_createmem(ODR_DECODE);
 695
 696     if (z_SortKeySpecList(encode, &src, 0, 0))
 697     {
 698         int len;
 699         char *buf = odr_getbuf(encode, &len, 0);
 700
 701         if (buf)
 702         {
 703             odr_setbuf(decode, buf, len, 0);
 704             z_SortKeySpecList(decode, &dst, 0, 0);
 705         }
 706     }
 707     nmem_transfer(nmem, decode->mem);
 708     odr_destroy(encode);
 709     odr_destroy(decode);
 710     return dst;
 711 }
 712
 713 ZebraSet resultSetClone(ZebraHandle zh, const char *setname,
 714                         ZebraSet rset)
 715 {
 716     ZebraSet nset;
 717     int i;
 718
 719     nset = resultSetAdd(zh, setname, 1);
 720     if (!nset)
 721         return 0;
 722
 723     nset->nmem = nmem_create();
 724
 725     nset->num_bases = rset->num_bases;
 726     nset->basenames =
 727         nmem_malloc (nset->nmem, nset->num_bases * sizeof(*rset->basenames));
 728     for (i = 0; i<rset->num_bases; i++)
 729         nset->basenames[i] = nmem_strdup(nset->nmem, rset->basenames[i]);
 730
 731     if (rset->rset)
 732         nset->rset = rset_dup(rset->rset);
 733     if (rset->rpn)
 734         nset->rpn = copy_RPNQuery(rset->rpn, nset->nmem);
 735     return nset;
 736 }
 737
 738 ZEBRA_RES resultSetSort(ZebraHandle zh, NMEM nmem,
 739                         int num_input_setnames, const char **input_setnames,
 740                         const char *output_setname,
 741                         Z_SortKeySpecList *sort_sequence, int *sort_status)
 742 {
 743     ZebraSet sset;
 744     RSET rset;
 745
 746     if (num_input_setnames == 0)
 747     {
 748         zebra_setError(zh, YAZ_BIB1_NO_RESULT_SET_NAME_SUPPLIED_ON_SORT, 0);
 749         return ZEBRA_FAIL;
 750     }
 751     if (num_input_setnames > 1)
 752     {
 753         zebra_setError(zh, YAZ_BIB1_SORT_TOO_MANY_INPUT_RESULTS, 0);
 754         return ZEBRA_FAIL;
 755     }
 756     if (!log_level_set)
 757         loglevels();
 758     yaz_log(log_level_sort, "result set sort input=%s output=%s",
 759           *input_setnames, output_setname);
 760     sset = resultSetGet (zh, input_setnames[0]);
 761     if (!sset)
 762     {
 763         zebra_setError(zh, YAZ_BIB1_SPECIFIED_RESULT_SET_DOES_NOT_EXIST,
 764                        input_setnames[0]);
 765         return ZEBRA_FAIL;
 766     }
 767     if (!(rset = sset->rset))
 768     {
 769         zebra_setError(zh, YAZ_BIB1_SPECIFIED_RESULT_SET_DOES_NOT_EXIST,
 770                        input_setnames[0]);
 771         return ZEBRA_FAIL;
 772     }
 773     if (strcmp (output_setname, input_setnames[0]))
 774         sset = resultSetClone(zh, output_setname, sset);
 775     sset->sortSpec = copy_SortKeySpecList(sort_sequence, sset->nmem);
 776     return resultSetSortSingle (zh, nmem, sset, rset, sort_sequence,
 777                                 sort_status);
 778 }
 779
 780 ZEBRA_RES resultSetSortSingle(ZebraHandle zh, NMEM nmem,
 781                               ZebraSet sset, RSET rset,
 782                               Z_SortKeySpecList *sort_sequence,
 783                               int *sort_status)
 784 {
 785     int i;
 786     int n = 0;
 787     zint kno = 0;
 788     zint psysno = 0;
 789     struct it_key key;
 790     struct sortKeyInfo sort_criteria[3];
 791     int num_criteria;
 792     RSFD rfd;
 793     TERMID termid;
 794     TERMID *terms;
 795     int numTerms = 0;
 796     size_t sysno_mem_index = 0;
 797
 798     if (zh->m_staticrank)
 799         sysno_mem_index = 1;
 800
 801
 802     assert(nmem); /* compiler shut up about unused param */
 803     sset->sort_info->num_entries = 0;
 804
 805     rset_getterms(rset, 0, 0, &n);
 806     terms = (TERMID *) nmem_malloc(nmem, sizeof(*terms)*n);
 807     rset_getterms(rset, terms, n, &numTerms);
 808
 809     sset->hits = 0;
 810     num_criteria = sort_sequence->num_specs;
 811     if (num_criteria > 3)
 812         num_criteria = 3;
 813     for (i = 0; i < num_criteria; i++)
 814     {
 815         Z_SortKeySpec *sks = sort_sequence->specs[i];
 816         Z_SortKey *sk;
 817
 818         if (*sks->sortRelation == Z_SortKeySpec_ascending)
 819             sort_criteria[i].relation = 'A';
 820         else if (*sks->sortRelation == Z_SortKeySpec_descending)
 821             sort_criteria[i].relation = 'D';
 822         else
 823         {
 824             zebra_setError(zh, YAZ_BIB1_ILLEGAL_SORT_RELATION, 0);
 825             return ZEBRA_FAIL;
 826         }
 827         if (sks->sortElement->which == Z_SortElement_databaseSpecific)
 828         {
 829             zebra_setError(zh, YAZ_BIB1_DATABASE_SPECIFIC_SORT_UNSUPP, 0);
 830             return ZEBRA_FAIL;
 831         }
 832         else if (sks->sortElement->which != Z_SortElement_generic)
 833         {
 834             zebra_setError(zh, YAZ_BIB1_SORT_ILLEGAL_SORT, 0);
 835             return ZEBRA_FAIL;
 836         }
 837         sk = sks->sortElement->u.generic;
 838         switch (sk->which)
 839         {
 840         case Z_SortKey_sortField:
 841             yaz_log(log_level_sort, "key %d is of type sortField",
 842                     i+1);
 843             zebra_setError(zh, YAZ_BIB1_CANNOT_SORT_ACCORDING_TO_SEQUENCE, 0);
 844             return ZEBRA_FAIL;
 845         case Z_SortKey_elementSpec:
 846             yaz_log(log_level_sort, "key %d is of type elementSpec",
 847                     i+1);
 848             zebra_setError(zh, YAZ_BIB1_CANNOT_SORT_ACCORDING_TO_SEQUENCE, 0);
 849             return ZEBRA_FAIL;
 850         case Z_SortKey_sortAttributes:
 851             yaz_log(log_level_sort, "key %d is of type sortAttributes", i+1);
 852             sort_criteria[i].attrUse =
 853                 zebra_maps_sort (zh->reg->zebra_maps,
 854                                  sk->u.sortAttributes,
 855                                  &sort_criteria[i].numerical);
 856             yaz_log(log_level_sort, "use value = %d", sort_criteria[i].attrUse);
 857             if (sort_criteria[i].attrUse == -1)
 858             {
 859                 zebra_setError(
 860                     zh, YAZ_BIB1_USE_ATTRIBUTE_REQUIRED_BUT_NOT_SUPPLIED, 0);
 861                 return ZEBRA_FAIL;
 862             }
 863             if (sortIdx_type (zh->reg->sortIdx, sort_criteria[i].attrUse))
 864             {
 865                 zebra_setError(
 866                     zh, YAZ_BIB1_CANNOT_SORT_ACCORDING_TO_SEQUENCE, 0);
 867                 return ZEBRA_FAIL;
 868             }
 869             break;
 870         }
 871     }
 872     rfd = rset_open (rset, RSETF_READ);
 873     while (rset_read (rfd, &key, &termid))
 874     {
 875         zint this_sys = key.mem[sysno_mem_index];
 876         if (log_level_searchhits)
 877             key_logdump_txt(log_level_searchhits, &key, termid->name);
 878         kno++;
 879         if (this_sys != psysno)
 880         {
 881             (sset->hits)++;
 882             psysno = this_sys;
 883             resultSetInsertSort (zh, sset,
 884                                  sort_criteria, num_criteria, psysno);
 885         }
 886     }
 887     rset_close (rfd);
 888     yaz_log(log_level_sort, ZINT_FORMAT " keys, " ZINT_FORMAT " sysnos, sort",
 889             kno, sset->hits);
 890     for (i = 0; i < numTerms; i++)
 891         yaz_log(log_level_sort, "term=\"%s\" type=%s count=" ZINT_FORMAT,
 892                  terms[i]->name, terms[i]->flags, terms[i]->rset->hits_count);
 893     *sort_status = Z_SortResponse_success;
 894     return ZEBRA_OK;
 895 }
 896
 897 RSET resultSetRef(ZebraHandle zh, const char *resultSetId)
 898 {
 899     ZebraSet s;
 900
 901     if ((s = resultSetGet (zh, resultSetId)))
 902         return s->rset;
 903     return NULL;
 904 }
 905
 906 ZEBRA_RES resultSetRank(ZebraHandle zh, ZebraSet zebraSet,
 907                         RSET rset, NMEM nmem)
 908 {
 909     struct it_key key;
 910     TERMID termid;
 911     TERMID *terms;
 912     zint kno = 0;
 913     int numTerms = 0;
 914     int n = 0;
 915     int i;
 916     ZebraRankClass rank_class;
 917     struct zset_sort_info *sort_info;
 918     const char *rank_handler_name = res_get_def(zh->res, "rank", "rank-1");
 919     size_t sysno_mem_index = 0;
 920
 921     if (zh->m_staticrank)
 922         sysno_mem_index = 1;
 923
 924     if (!log_level_set)
 925         loglevels();
 926     sort_info = zebraSet->sort_info;
 927     sort_info->num_entries = 0;
 928     zebraSet->hits = 0;
 929     rset_getterms(rset, 0, 0, &n);
 930     terms = (TERMID *) nmem_malloc(nmem, sizeof(*terms)*n);
 931     rset_getterms(rset, terms, n, &numTerms);
 932
 933
 934     rank_class = zebraRankLookup(zh, rank_handler_name);
 935     if (!rank_class)
 936     {
 937         yaz_log(YLOG_WARN, "No such rank handler: %s", rank_handler_name);
 938         zebra_setError(zh, YAZ_BIB1_UNSUPP_SEARCH, "Cannot find rank handler");
 939         return ZEBRA_FAIL;
 940     }
 941     else
 942     {
 943         RSFD rfd = rset_open(rset, RSETF_READ);
 944         struct rank_control *rc = rank_class->control;
 945         double score;
 946         zint count = 0;
 947
 948         void *handle =
 949             (*rc->begin) (zh->reg, rank_class->class_handle, rset, nmem,
 950                           terms, numTerms);
 951         zint psysno = 0;  /* previous doc id / sys no */
 952         zint pstaticrank = 0; /* previous static rank */
 953         int stop_flag = 0;
 954         while (rset_read(rfd, &key, &termid))
 955         {
 956             zint this_sys = key.mem[sysno_mem_index];
 957
 958             zint seqno = key.mem[key.len-1];
 959             kno++;
 960             if (log_level_searchhits)
 961                 key_logdump_txt(log_level_searchhits, &key, termid->name);
 962             if (this_sys != psysno)
 963             {   /* new record .. */
 964                 if (rfd->counted_items > rset->hits_limit)
 965                     break;
 966                 if (psysno)
 967                 {   /* only if we did have a previous record */
 968                     score = (*rc->calc) (handle, psysno, pstaticrank,
 969                                          &stop_flag);
 970                     /* insert the hit. A=Ascending */
 971                     resultSetInsertRank (zh, sort_info, psysno, score, 'A');
 972                     count++;
 973                     if (stop_flag)
 974                         break;
 975                 }
 976                 psysno = this_sys;
 977                 if (zh->m_staticrank)
 978                     pstaticrank = key.mem[0];
 979             }
 980             (*rc->add) (handle, CAST_ZINT_TO_INT(seqno), termid);
 981         }
 982         /* no more items */
 983         if (psysno)
 984         {   /* we had - at least - one record */
 985             score = (*rc->calc)(handle, psysno, pstaticrank, &stop_flag);
 986             /* insert the hit. A=Ascending */
 987             resultSetInsertRank(zh, sort_info, psysno, score, 'A');
 988             count++;
 989         }
 990         (*rc->end) (zh->reg, handle);
 991         rset_close (rfd);
 992     }
 993     zebraSet->hits = rset->hits_count;
 994
 995     yaz_log(log_level_searchterms, ZINT_FORMAT " keys, "
 996             ZINT_FORMAT " sysnos, rank",  kno, zebraSet->hits);
 997     for (i = 0; i < numTerms; i++)
 998     {
 999         yaz_log(log_level_searchterms, "term=\"%s\" type=%s count="
1000                 ZINT_FORMAT,
1001                 terms[i]->name, terms[i]->flags, terms[i]->rset->hits_count);
1002     }
1003     return ZEBRA_OK;
1004 }
1005
1006 ZebraRankClass zebraRankLookup(ZebraHandle zh, const char *name)
1007 {
1008     ZebraRankClass p = zh->reg->rank_classes;
1009     while (p && strcmp (p->control->name, name))
1010         p = p->next;
1011     if (p && !p->init_flag)
1012     {
1013         if (p->control->create)
1014             p->class_handle = (*p->control->create)(zh);
1015         p->init_flag = 1;
1016     }
1017     return p;
1018 }
1019
1020 void zebraRankInstall(struct zebra_register *reg, struct rank_control *ctrl)
1021 {
1022     ZebraRankClass p = (ZebraRankClass) xmalloc (sizeof(*p));
1023     p->control = (struct rank_control *) xmalloc (sizeof(*p->control));
1024     memcpy (p->control, ctrl, sizeof(*p->control));
1025     p->control->name = xstrdup (ctrl->name);
1026     p->init_flag = 0;
1027     p->next = reg->rank_classes;
1028     reg->rank_classes = p;
1029 }
1030
1031 void zebraRankDestroy(struct zebra_register *reg)
1032 {
1033     ZebraRankClass p = reg->rank_classes;
1034     while (p)
1035     {
1036         ZebraRankClass p_next = p->next;
1037         if (p->init_flag && p->control->destroy)
1038             (*p->control->destroy)(reg, p->class_handle);
1039         xfree(p->control->name);
1040         xfree(p->control);
1041         xfree(p);
1042         p = p_next;
1043     }
1044     reg->rank_classes = NULL;
1045 }
1046
1047 static int trav_rset_for_termids(RSET rset, TERMID *termid_array,
1048                                  zint *hits_array, int *approx_array)
1049 {
1050     int no = 0;
1051     int i;
1052     for (i = 0; i<rset->no_children; i++)
1053         no += trav_rset_for_termids(rset->children[i],
1054                                     (termid_array ? termid_array + no : 0),
1055                                     (hits_array ? hits_array + no : 0),
1056                                     (approx_array ? approx_array + no : 0));
1057     if (rset->term)
1058     {
1059         if (termid_array)
1060             termid_array[no] = rset->term;
1061         if (hits_array)
1062             hits_array[no] = rset->hits_count;
1063         if (approx_array)
1064             approx_array[no] = rset->hits_approx;
1065 #if 0
1066         yaz_log(YLOG_LOG, "rset=%p term=%s limit=" ZINT_FORMAT
1067                 " count=" ZINT_FORMAT,
1068                 rset, rset->term->name, rset->hits_limit, rset->hits_count);
1069 #endif
1070         no++;
1071     }
1072     return no;
1073 }
1074
1075 ZEBRA_RES zebra_result_set_term_no(ZebraHandle zh, const char *setname,
1076                                    int *num_terms)
1077 {
1078     ZebraSet sset = resultSetGet(zh, setname);
1079     *num_terms = 0;
1080     if (sset)
1081     {
1082         *num_terms = trav_rset_for_termids(sset->rset, 0, 0, 0);
1083         return ZEBRA_OK;
1084     }
1085     return ZEBRA_FAIL;
1086 }
1087
1088 ZEBRA_RES zebra_result_set_term_info(ZebraHandle zh, const char *setname,
1089                                      int no, zint *count, int *approx,
1090                                      char *termbuf, size_t *termlen,
1091                                      const char **term_ref_id)
1092 {
1093     ZebraSet sset = resultSetGet(zh, setname);
1094     if (sset)
1095     {
1096         int num_terms = trav_rset_for_termids(sset->rset, 0, 0, 0);
1097         if (no >= 0 && no < num_terms)
1098         {
1099             TERMID *term_array = xmalloc(num_terms * sizeof(*term_array));
1100             zint *hits_array = xmalloc(num_terms * sizeof(*hits_array));
1101             int *approx_array = xmalloc(num_terms * sizeof(*approx_array));
1102
1103             trav_rset_for_termids(sset->rset, term_array,
1104                                   hits_array, approx_array);
1105
1106             if (count)
1107                 *count = hits_array[no];
1108             if (approx)
1109                 *approx = approx_array[no];
1110             if (termbuf)
1111             {
1112                 char *inbuf = term_array[no]->name;
1113                 size_t inleft = strlen(inbuf);
1114                 size_t outleft = *termlen - 1;
1115
1116                 if (zh->iconv_from_utf8 != 0)
1117                 {
1118                     char *outbuf = termbuf;
1119                     size_t ret;
1120
1121                     ret = yaz_iconv(zh->iconv_from_utf8, &inbuf, &inleft,
1122                                     &outbuf, &outleft);
1123                     if (ret == (size_t)(-1))
1124                         *termlen = 0;
1125                     else
1126                         *termlen = outbuf - termbuf;
1127                 }
1128                 else
1129                 {
1130                     if (inleft > outleft)
1131                         inleft = outleft;
1132                     *termlen = inleft;
1133                     memcpy(termbuf, inbuf, *termlen);
1134                 }
1135                 termbuf[*termlen] = '\0';
1136             }
1137             if (term_ref_id)
1138                 *term_ref_id = term_array[no]->ref_id;
1139
1140             xfree(term_array);
1141             xfree(hits_array);
1142             xfree(approx_array);
1143             return ZEBRA_OK;
1144         }
1145     }
1146     return ZEBRA_FAIL;
1147 }
1148
1149 ZEBRA_RES zebra_snippets_hit_vector(ZebraHandle zh, const char *setname,
1150                                     zint sysno, zebra_snippets *snippets)
1151 {
1152     ZebraSet sset = resultSetGet(zh, setname);
1153     yaz_log(YLOG_DEBUG, "zebra_get_hit_vector setname=%s zysno=" ZINT_FORMAT,
1154             setname, sysno);
1155     if (!sset)
1156         return ZEBRA_FAIL;
1157     else
1158     {
1159         struct rset_key_control *kc = zebra_key_control_create(zh);
1160         NMEM nmem = nmem_create();
1161         struct it_key key;
1162         RSET rsets[2], rset_comb;
1163         RSET rset_temp = rstemp_create(nmem, kc, kc->scope,
1164                                        res_get (zh->res, "setTmpDir"),0 );
1165
1166         TERMID termid;
1167         RSFD rsfd = rset_open(rset_temp, RSETF_WRITE);
1168
1169         key.mem[0] = sysno;
1170         key.mem[1] = 0;
1171         key.mem[2] = 0;
1172         key.mem[3] = 0;
1173         key.len = 2;
1174         rset_write (rsfd, &key);
1175         rset_close (rsfd);
1176
1177         rsets[0] = rset_temp;
1178         rsets[1] = rset_dup(sset->rset);
1179
1180         rset_comb = rsmulti_and_create(nmem, kc, kc->scope, 2, rsets);
1181
1182         rsfd = rset_open(rset_comb, RSETF_READ);
1183
1184         while (rset_read(rsfd, &key, &termid))
1185         {
1186             if (termid)
1187             {
1188                 struct ord_list *ol;
1189                 for (ol = termid->ol; ol; ol = ol->next)
1190                 {
1191                     zebra_snippets_append(snippets, key.mem[key.len-1],
1192                                           ol->ord, termid->name);
1193                 }
1194             }
1195         }
1196         rset_close(rsfd);
1197
1198         rset_delete(rset_comb);
1199         nmem_destroy(nmem);
1200         kc->dec(kc);
1201     }
1202     return ZEBRA_OK;
1203 }
1204
1205 /*
1206  * Local variables:
1207  * c-basic-offset: 4
1208  * indent-tabs-mode: nil
1209  * End:
1210  * vim: shiftwidth=4 tabstop=8 expandtab
1211  */
1212