index/zrpn.c

   1 /*
   2  * Copyright (C) 1994-1995, Index Data I/S
   3  * All rights reserved.
   4  * Sebastian Hammer, Adam Dickmeiss
   5  *
   6  * $Log: zrpn.c,v $
   7  * Revision 1.29  1995-10-13 16:01:49  adam
   8  * Work on relations.
   9  *
  10  * Revision 1.28  1995/10/13  12:26:43  adam
  11  * Optimization of truncation.
  12  *
  13  * Revision 1.27  1995/10/12  17:07:22  adam
  14  * Truncation works.
  15  *
  16  * Revision 1.26  1995/10/12  12:40:54  adam
  17  * Bug fixes in rpn_prox.
  18  *
  19  * Revision 1.25  1995/10/10  13:59:24  adam
  20  * Function rset_open changed its wflag parameter to general flags.
  21  *
  22  * Revision 1.24  1995/10/09  16:18:37  adam
  23  * Function dict_lookup_grep got extra client data parameter.
  24  *
  25  * Revision 1.23  1995/10/06  16:33:37  adam
  26  * Use attribute mappings.
  27  *
  28  * Revision 1.22  1995/10/06  15:07:39  adam
  29  * Structure 'local-number' handled.
  30  *
  31  * Revision 1.21  1995/10/06  13:52:06  adam
  32  * Bug fixes. Handler may abort further scanning.
  33  *
  34  * Revision 1.20  1995/10/06  11:06:33  adam
  35  * Scan entries include 'occurrences' now.
  36  *
  37  * Revision 1.19  1995/10/06  10:43:56  adam
  38  * Scan added. 'occurrences' in scan entries not set yet.
  39  *
  40  * Revision 1.18  1995/10/04  16:57:20  adam
  41  * Key input and merge sort in one pass.
  42  *
  43  * Revision 1.17  1995/10/04  12:55:17  adam
  44  * Bug fix in ranked search. Use=Any keys inserted.
  45  *
  46  * Revision 1.16  1995/10/02  16:24:40  adam
  47  * Use attribute actually used in search requests.
  48  *
  49  * Revision 1.15  1995/10/02  15:18:52  adam
  50  * New member in recRetrieveCtrl: diagnostic.
  51  *
  52  * Revision 1.14  1995/09/28  12:10:32  adam
  53  * Bug fixes. Field prefix used in queries.
  54  *
  55  * Revision 1.13  1995/09/18  14:17:50  adam
  56  * Minor changes.
  57  *
  58  * Revision 1.12  1995/09/15  14:45:21  adam
  59  * Retrieve control.
  60  * Work on truncation.
  61  *
  62  * Revision 1.11  1995/09/14  11:53:27  adam
  63  * First work on regular expressions/truncations.
  64  *
  65  * Revision 1.10  1995/09/11  15:23:26  adam
  66  * More work on relevance search.
  67  *
  68  * Revision 1.9  1995/09/11  13:09:35  adam
  69  * More work on relevance feedback.
  70  *
  71  * Revision 1.8  1995/09/08  14:52:27  adam
  72  * Minor changes. Dictionary is lower case now.
  73  *
  74  * Revision 1.7  1995/09/07  13:58:36  adam
  75  * New parameter: result-set file descriptor (RSFD) to support multiple
  76  * positions within the same result-set.
  77  * Boolean operators: and, or, not implemented.
  78  * Result-set references.
  79  *
  80  * Revision 1.6  1995/09/06  16:11:18  adam
  81  * Option: only one word key per file.
  82  *
  83  * Revision 1.5  1995/09/06  10:33:04  adam
  84  * More work on present. Some log messages removed.
  85  *
  86  * Revision 1.4  1995/09/05  15:28:40  adam
  87  * More work on search engine.
  88  *
  89  * Revision 1.3  1995/09/04  15:20:22  adam
  90  * Minor changes.
  91  *
  92  * Revision 1.2  1995/09/04  12:33:43  adam
  93  * Various cleanup. YAZ util used instead.
  94  *
  95  * Revision 1.1  1995/09/04  09:10:40  adam
  96  * More work on index add/del/update.
  97  * Merge sort implemented.
  98  * Initial work on z39 server.
  99  *
 100  */
 101 #include <stdio.h>
 102 #include <assert.h>
 103 #include <unistd.h>
 104
 105 #include "zserver.h"
 106 #include <attribute.h>
 107
 108 #include <rsisam.h>
 109 #include <rstemp.h>
 110 #include <rsnull.h>
 111 #include <rsbool.h>
 112 #include <rsrel.h>
 113
 114 int index_word_prefix_map (char *string, oid_value attrSet, int attrUse)
 115 {
 116     attent *attp;
 117
 118     logf (LOG_DEBUG, "oid_value attrSet = %d, attrUse = %d", attrSet, attrUse);
 119     attp = att_getentbyatt (attrSet, attrUse);
 120     if (!attp)
 121         return -1;
 122     logf (LOG_DEBUG, "ord=%d", attp->attset_ordinal);
 123     return index_word_prefix (string, attp->attset_ordinal,
 124                               attp->local_attribute);
 125 }
 126
 127 /*
 128  * attr_print: log attributes
 129  */
 130 static void attr_print (Z_AttributesPlusTerm *t)
 131 {
 132     int of, i;
 133     for (of = 0; of < t->num_attributes; of++)
 134     {
 135         Z_AttributeElement *element;
 136         element = t->attributeList[of];
 137
 138         switch (element->which)
 139         {
 140         case Z_AttributeValue_numeric:
 141             logf (LOG_DEBUG, "attributeType=%d value=%d",
 142                   *element->attributeType,
 143                   *element->value.numeric);
 144             break;
 145         case Z_AttributeValue_complex:
 146             logf (LOG_DEBUG, "attributeType=%d complex",
 147                   *element->attributeType);
 148             for (i = 0; i<element->value.complex->num_list; i++)
 149             {
 150                 if (element->value.complex->list[i]->which ==
 151                     Z_StringOrNumeric_string)
 152                     logf (LOG_DEBUG, "   string: '%s'",
 153                           element->value.complex->list[i]->u.string);
 154                 else if (element->value.complex->list[i]->which ==
 155                          Z_StringOrNumeric_numeric)
 156                     logf (LOG_DEBUG, "   numeric: '%d'",
 157                           *element->value.complex->list[i]->u.numeric);
 158             }
 159             break;
 160         default:
 161             assert (0);
 162         }
 163     }
 164 }
 165
 166 typedef struct {
 167     int type;
 168     int major;
 169     int minor;
 170     Z_AttributesPlusTerm *zapt;
 171 } AttrType;
 172
 173 static int attr_find (AttrType *src, oid_value *attributeSetP)
 174 {
 175     while (src->major < src->zapt->num_attributes)
 176     {
 177         Z_AttributeElement *element;
 178
 179         element = src->zapt->attributeList[src->major];
 180         if (src->type == *element->attributeType)
 181         {
 182             switch (element->which)
 183             {
 184             case Z_AttributeValue_numeric:
 185                 ++(src->major);
 186                 if (element->attributeSet && attributeSetP)
 187                 {
 188                     oident *attrset;
 189
 190                     attrset = oid_getentbyoid (element->attributeSet);
 191                     *attributeSetP = attrset->value;
 192                 }
 193                 return *element->value.numeric;
 194                 break;
 195             case Z_AttributeValue_complex:
 196                 if (src->minor >= element->value.complex->num_list ||
 197                     element->value.complex->list[src->minor]->which !=
 198                     Z_StringOrNumeric_numeric)
 199                     break;
 200                 ++(src->minor);
 201                 if (element->attributeSet && attributeSetP)
 202                 {
 203                     oident *attrset;
 204
 205                     attrset = oid_getentbyoid (element->attributeSet);
 206                     *attributeSetP = attrset->value;
 207                 }
 208                 return *element->value.complex->list[src->minor-1]->u.numeric;
 209             default:
 210                 assert (0);
 211             }
 212         }
 213         ++(src->major);
 214     }
 215     return -1;
 216 }
 217
 218 static void attr_init (AttrType *src, Z_AttributesPlusTerm *zapt,
 219                        int type)
 220 {
 221     src->zapt = zapt;
 222     src->type = type;
 223     src->major = 0;
 224     src->minor = 0;
 225 }
 226
 227 struct trunc_info {
 228     int  *ptr;
 229     int  *indx;
 230     char **heap;
 231     int  heapnum;
 232     int  (*cmp)(const void *p1, const void *p2);
 233     int  keysize;
 234     char *swapbuf;
 235     char *tmpbuf;
 236     char *buf;
 237 };
 238
 239 static void heap_swap (struct trunc_info *ti, int i1, int i2)
 240 {
 241     int swap;
 242
 243     swap = ti->ptr[i1];
 244     ti->ptr[i1] = ti->ptr[i2];
 245     ti->ptr[i2] = swap;
 246 }
 247
 248 static void heap_delete (struct trunc_info *ti)
 249 {
 250     int cur = 1, child = 2;
 251
 252     heap_swap (ti, 1, ti->heapnum--);
 253     while (child <= ti->heapnum) {
 254         if (child < ti->heapnum &&
 255             (*ti->cmp)(ti->heap[ti->ptr[child]],
 256                        ti->heap[ti->ptr[1+child]]) > 0)
 257             child++;
 258         if ((*ti->cmp)(ti->heap[ti->ptr[cur]],
 259                        ti->heap[ti->ptr[child]]) > 0)
 260         {
 261             heap_swap (ti, cur, child);
 262             cur = child;
 263             child = 2*cur;
 264         }
 265         else
 266             break;
 267     }
 268 }
 269
 270 static void heap_insert (struct trunc_info *ti, const char *buf, int indx)
 271 {
 272     int cur, parent;
 273
 274     cur = ++(ti->heapnum);
 275     memcpy (ti->heap[ti->ptr[cur]], buf, ti->keysize);
 276     ti->indx[ti->ptr[cur]] = indx;
 277     parent = cur/2;
 278     while (parent && (*ti->cmp)(ti->heap[ti->ptr[parent]],
 279                                 ti->heap[ti->ptr[cur]]) > 0)
 280     {
 281         heap_swap (ti, cur, parent);
 282         cur = parent;
 283         parent = cur/2;
 284     }
 285 }
 286
 287 static
 288 struct trunc_info *heap_init (int size, int key_size,
 289                               int (*cmp)(const void *p1, const void *p2))
 290 {
 291     struct trunc_info *ti = xmalloc (sizeof(*ti));
 292     int i;
 293
 294     ++size;
 295     ti->heapnum = 0;
 296     ti->keysize = key_size;
 297     ti->cmp = cmp;
 298     ti->indx = xmalloc (size * sizeof(*ti->indx));
 299     ti->heap = xmalloc (size * sizeof(*ti->heap));
 300     ti->ptr = xmalloc (size * sizeof(*ti->ptr));
 301     ti->swapbuf = xmalloc (ti->keysize);
 302     ti->tmpbuf = xmalloc (ti->keysize);
 303     ti->buf = xmalloc (size * ti->keysize);
 304     for (i = size; --i >= 0; )
 305     {
 306         ti->ptr[i] = i;
 307         ti->heap[i] = ti->buf + ti->keysize * i;
 308     }
 309     return ti;
 310 }
 311
 312 static void heap_close (struct trunc_info *ti)
 313 {
 314     xfree (ti->ptr);
 315     xfree (ti->indx);
 316     xfree (ti->heap);
 317     xfree (ti->swapbuf);
 318     xfree (ti->tmpbuf);
 319     xfree (ti);
 320 }
 321
 322 static RSET rset_trunc_r (ISAM isam, ISAM_P *isam_p, int from, int to,
 323                          int merge_chunk)
 324 {
 325     RSET result;
 326     RSFD result_rsfd;
 327     rset_temp_parms parms;
 328
 329     parms.key_size = sizeof(struct it_key);
 330     result = rset_create (rset_kind_temp, &parms);
 331     result_rsfd = rset_open (result, RSETF_WRITE|RSETF_SORT_SYSNO);
 332
 333     if (to - from > merge_chunk)
 334     {
 335         RSFD *rsfd;
 336         RSET *rset;
 337         int i, i_add = (to-from)/merge_chunk + 1;
 338         struct trunc_info *ti;
 339         int rscur = 0;
 340         int rsmax = (to-from)/i_add + 1;
 341
 342         rset = xmalloc (sizeof(*rset) * rsmax);
 343         rsfd = xmalloc (sizeof(*rsfd) * rsmax);
 344
 345         for (i = from; i < to; i += i_add)
 346         {
 347             if (i_add <= to - i)
 348                 rset[rscur] = rset_trunc_r (isam, isam_p, i, i+i_add,
 349                                             merge_chunk);
 350             else
 351                 rset[rscur] = rset_trunc_r (isam, isam_p, i, to,
 352                                             merge_chunk);
 353             rscur++;
 354         }
 355         ti = heap_init (rscur, sizeof(struct it_key), key_compare);
 356         for (i = rscur; --i >= 0; )
 357         {
 358             rsfd[i] = rset_open (rset[i], RSETF_READ|RSETF_SORT_SYSNO);
 359             if (rset_read (rset[i], rsfd[i], ti->tmpbuf))
 360                 heap_insert (ti, ti->tmpbuf, i);
 361             else
 362             {
 363                 rset_close (rset[i], rsfd[i]);
 364                 rset_delete (rset[i]);
 365             }
 366         }
 367         while (ti->heapnum)
 368         {
 369             int n = ti->indx[ti->ptr[1]];
 370
 371             rset_write (result, result_rsfd, ti->heap[ti->ptr[1]]);
 372
 373             while (1)
 374             {
 375                 if (!rset_read (rset[n], rsfd[n], ti->tmpbuf))
 376                 {
 377                     heap_delete (ti);
 378                     rset_close (rset[n], rsfd[n]);
 379                     rset_delete (rset[n]);
 380                     break;
 381                 }
 382                 if ((*ti->cmp)(ti->tmpbuf, ti->heap[ti->ptr[1]]) > 1)
 383                 {
 384                     heap_delete (ti);
 385                     heap_insert (ti, ti->tmpbuf, n);
 386                     break;
 387                 }
 388             }
 389         }
 390         xfree (rset);
 391         xfree (rsfd);
 392         heap_close (ti);
 393     }
 394     else
 395     {
 396         ISPT *ispt;
 397         int i;
 398         struct trunc_info *ti;
 399
 400         ispt = xmalloc (sizeof(*ispt) * (to-from));
 401
 402         ti = heap_init (to-from, sizeof(struct it_key),
 403                         key_compare);
 404         for (i = to-from; --i >= 0; )
 405         {
 406             ispt[i] = is_position (isam, isam_p[from+i]);
 407             if (is_readkey (ispt[i], ti->tmpbuf))
 408                 heap_insert (ti, ti->tmpbuf, i);
 409             else
 410                 is_pt_free (ispt[i]);
 411         }
 412         while (ti->heapnum)
 413         {
 414             int n = ti->indx[ti->ptr[1]];
 415
 416             rset_write (result, result_rsfd, ti->heap[ti->ptr[1]]);
 417 #if 0
 418             heap_delete (ti);
 419             if (is_readkey (ispt[n], ti->tmpbuf))
 420                 heap_insert (ti, ti->tmpbuf, n);
 421             else
 422                 is_pt_free (ispt[n]);
 423 #else
 424             while (1)
 425             {
 426                 if (!is_readkey (ispt[n], ti->tmpbuf))
 427                 {
 428                     heap_delete (ti);
 429                     is_pt_free (ispt[n]);
 430                     break;
 431                 }
 432                 if ((*ti->cmp)(ti->tmpbuf, ti->heap[ti->ptr[1]]) > 1)
 433                 {
 434                     heap_delete (ti);
 435                     heap_insert (ti, ti->tmpbuf, n);
 436                     break;
 437                 }
 438             }
 439 #endif
 440         }
 441         heap_close (ti);
 442         xfree (ispt);
 443     }
 444     rset_close (result, result_rsfd);
 445     return result;
 446 }
 447
 448 static int isam_trunc_cmp (const void *p1, const void *p2)
 449 {
 450     ISAM_P i1 = *(ISAM_P*) p1;
 451     ISAM_P i2 = *(ISAM_P*) p2;
 452     int d;
 453
 454     d = is_type (i1) - is_type (i2);
 455     if (d)
 456         return d;
 457     return is_block (i1) - is_block (i2);
 458 }
 459
 460 static RSET rset_trunc (ISAM isam, ISAM_P *isam_p, int no)
 461 {
 462
 463     qsort (isam_p, no, sizeof(*isam_p), isam_trunc_cmp);
 464     return rset_trunc_r (isam, isam_p, 0, no, 100);
 465 }
 466
 467 struct grep_info {
 468     ISAM_P *isam_p_buf;
 469     int isam_p_size;
 470     int isam_p_indx;
 471 };
 472
 473 static void add_isam_p (const char *info, struct grep_info *p)
 474 {
 475     if (p->isam_p_indx == p->isam_p_size)
 476     {
 477         ISAM_P *new_isam_p_buf;
 478
 479         p->isam_p_size = 2*p->isam_p_size + 100;
 480         new_isam_p_buf = xmalloc (sizeof(*new_isam_p_buf) *
 481                                   p->isam_p_size);
 482         if (p->isam_p_buf)
 483         {
 484             memcpy (new_isam_p_buf, p->isam_p_buf,
 485                     p->isam_p_indx * sizeof(*p->isam_p_buf));
 486             xfree (p->isam_p_buf);
 487         }
 488         p->isam_p_buf = new_isam_p_buf;
 489     }
 490     assert (*info == sizeof(*p->isam_p_buf));
 491     memcpy (p->isam_p_buf + p->isam_p_indx, info+1, sizeof(*p->isam_p_buf));
 492     (p->isam_p_indx)++;
 493 }
 494
 495 static int grep_handle (Dict_char *name, const char *info, void *p)
 496 {
 497     logf (LOG_DEBUG, "dict name: %s", name);
 498     add_isam_p (info, p);
 499     return 0;
 500 }
 501
 502 static void gen_regular_ge (char *dst, int val)
 503 {
 504     int dst_p = 0;
 505     int w = 1;
 506     int d;
 507     int pos = 0;
 508     int i;
 509
 510     if (val < 0)
 511         val = 0;
 512     while ((d=(val % (w*10))/w))
 513     {
 514         sprintf (dst + dst_p, "%d", val);
 515
 516         dst_p = strlen(dst) - pos - 1;
 517
 518         dst[dst_p++] = '[';
 519         dst[dst_p++] = d +'1';
 520         dst[dst_p++] = '-';
 521         dst[dst_p++] = '9';
 522         dst[dst_p++] = ']';
 523
 524         for (i = 0; i<pos; i++)
 525         {
 526             dst[dst_p++] = '[';
 527             dst[dst_p++] = '0';
 528             dst[dst_p++] = '-';
 529             dst[dst_p++] = '9';
 530             dst[dst_p++] = ']';
 531         }
 532         dst[dst_p++] = '|';
 533
 534         w = w * 10;
 535         pos++;
 536     }
 537     dst[dst_p] = '\0';
 538     for (i = 0; i<pos; i++)
 539         strcat (dst, "[0-9]");
 540     strcat (dst, "[0-9]*");
 541 }
 542
 543 static int trunc_term (ZServerInfo *zi, Z_AttributesPlusTerm *zapt,
 544                        const char *term_sub,
 545                        oid_value attributeSet, struct grep_info *grep_info)
 546 {
 547     char term_dict[10*IT_MAX_WORD+2];
 548     int i, j;
 549     const char *info;
 550     AttrType truncation;
 551     int truncation_value;
 552     AttrType use;
 553     int use_value;
 554     AttrType relation;
 555     int relation_value;
 556     oid_value curAttributeSet = attributeSet;
 557
 558     attr_init (&use, zapt, 1);
 559     use_value = attr_find (&use, &curAttributeSet);
 560     logf (LOG_DEBUG, "use value %d", use_value);
 561     attr_init (&truncation, zapt, 5);
 562     truncation_value = attr_find (&truncation, NULL);
 563     logf (LOG_DEBUG, "truncation value %d", truncation_value);
 564
 565     attr_init (&relation, zapt, 2);
 566     relation_value = attr_find (&relation, NULL);
 567
 568     if (use_value == -1)
 569         use_value = 1016;
 570     i = index_word_prefix_map (term_dict, curAttributeSet, use_value);
 571     if (i < 0)
 572     {
 573         zi->errCode = 114;
 574         return -1;
 575     }
 576     switch (relation_value)
 577     {
 578     case 1:
 579     case 2:
 580         break;
 581     case 4:
 582         logf (LOG_LOG, "Relation ge");
 583         gen_regular_ge (term_dict + strlen(term_dict), atoi(term_sub));
 584         logf (LOG_LOG, "dict_lookup_grep: %s", term_dict);
 585         dict_lookup_grep (zi->wordDict, term_dict, 0, grep_info, grep_handle);
 586         logf (LOG_LOG, "%d positions", grep_info->isam_p_indx);
 587         return 0;
 588     case 5:
 589         logf (LOG_LOG, "Relation gt");
 590         gen_regular_ge (term_dict + strlen(term_dict), atoi(term_sub)+1);
 591         logf (LOG_LOG, "dict_lookup_grep: %s", term_dict);
 592         dict_lookup_grep (zi->wordDict, term_dict, 0, grep_info, grep_handle);
 593         logf (LOG_LOG, "%d positions", grep_info->isam_p_indx);
 594         return 0;
 595     }
 596     switch (truncation_value)
 597     {
 598     case -1:         /* not specified */
 599     case 100:        /* do not truncate */
 600         strcat (term_dict, term_sub);
 601         logf (LOG_DEBUG, "dict_lookup: %s", term_dict);
 602         if ((info = dict_lookup (zi->wordDict, term_dict)))
 603             add_isam_p (info, grep_info);
 604         break;
 605     case 1:          /* right truncation */
 606         strcat (term_dict, term_sub);
 607         strcat (term_dict, ".*");
 608         dict_lookup_grep (zi->wordDict, term_dict, 0, grep_info, grep_handle);
 609         break;
 610     case 2:          /* left truncation */
 611     case 3:          /* left&right truncation */
 612         zi->errCode = 120;
 613         return -1;
 614     case 101:        /* process # in term */
 615         for (j = strlen(term_dict), i = 0; term_sub[i] && i < 2; i++)
 616             term_dict[j++] = term_sub[i];
 617         for (; term_sub[i]; i++)
 618             if (term_sub[i] == '#')
 619             {
 620                 term_dict[j++] = '.';
 621                 term_dict[j++] = '*';
 622             }
 623             else
 624                 term_dict[j++] = term_sub[i];
 625         term_dict[j] = '\0';
 626         dict_lookup_grep (zi->wordDict, term_dict, 0, grep_info, grep_handle);
 627         break;
 628     case 102:        /* regular expression */
 629         strcat (term_dict, term_sub);
 630         dict_lookup_grep (zi->wordDict, term_dict, 0, grep_info, grep_handle);
 631         break;
 632     }
 633     logf (LOG_LOG, "%d positions", grep_info->isam_p_indx);
 634     return 0;
 635 }
 636
 637 static void trans_term (ZServerInfo *zi, Z_AttributesPlusTerm *zapt,
 638                         char *termz)
 639 {
 640     size_t i, sizez;
 641     Z_Term *term = zapt->term;
 642
 643     sizez = term->u.general->len;
 644     if (sizez > IT_MAX_WORD)
 645         sizez = IT_MAX_WORD;
 646     for (i = 0; i < sizez; i++)
 647         termz[i] = index_char_cvt (term->u.general->buf[i]);
 648     termz[i] = '\0';
 649 }
 650
 651 static RSET rpn_search_APT_relevance (ZServerInfo *zi,
 652                                       Z_AttributesPlusTerm *zapt,
 653                                       oid_value attributeSet)
 654 {
 655     rset_relevance_parms parms;
 656     char termz[IT_MAX_WORD+1];
 657     char term_sub[IT_MAX_WORD+1];
 658     struct grep_info grep_info;
 659     char *p0 = termz, *p1 = NULL;
 660     RSET result;
 661
 662     parms.key_size = sizeof(struct it_key);
 663     parms.max_rec = 100;
 664     parms.cmp = key_compare;
 665     parms.is = zi->wordIsam;
 666
 667     if (zapt->term->which != Z_Term_general)
 668     {
 669         zi->errCode = 124;
 670         return NULL;
 671     }
 672     trans_term (zi, zapt, termz);
 673     grep_info.isam_p_indx = 0;
 674     grep_info.isam_p_size = 0;
 675     grep_info.isam_p_buf = NULL;
 676     while (1)
 677     {
 678         if ((p1 = strchr (p0, ' ')))
 679         {
 680             memcpy (term_sub, p0, p1-p0);
 681             term_sub[p1-p0] = '\0';
 682         }
 683         else
 684             strcpy (term_sub, p0);
 685         if (trunc_term (zi, zapt, term_sub, attributeSet, &grep_info))
 686             return NULL;
 687         if (!p1)
 688             break;
 689         p0 = p1;
 690         while (*++p0 == ' ')
 691             ;
 692     }
 693     parms.isam_positions = grep_info.isam_p_buf;
 694     parms.no_isam_positions = grep_info.isam_p_indx;
 695     if (grep_info.isam_p_indx > 0)
 696         result = rset_create (rset_kind_relevance, &parms);
 697     else
 698         result = rset_create (rset_kind_null, NULL);
 699     xfree (grep_info.isam_p_buf);
 700     return result;
 701 }
 702
 703 static RSET rpn_search_APT_word (ZServerInfo *zi,
 704                                  Z_AttributesPlusTerm *zapt,
 705                                  oid_value attributeSet)
 706 {
 707     rset_isam_parms parms;
 708     char termz[IT_MAX_WORD+1];
 709     struct grep_info grep_info;
 710     RSET result;
 711
 712     if (zapt->term->which != Z_Term_general)
 713     {
 714         zi->errCode = 124;
 715         return NULL;
 716     }
 717     trans_term (zi, zapt, termz);
 718
 719     grep_info.isam_p_indx = 0;
 720     grep_info.isam_p_size = 0;
 721     grep_info.isam_p_buf = NULL;
 722
 723     if (trunc_term (zi, zapt, termz, attributeSet, &grep_info))
 724         return NULL;
 725     if (grep_info.isam_p_indx < 1)
 726         result = rset_create (rset_kind_null, NULL);
 727     else if (grep_info.isam_p_indx == 1)
 728     {
 729         parms.is = zi->wordIsam;
 730         parms.pos = *grep_info.isam_p_buf;
 731         result = rset_create (rset_kind_isam, &parms);
 732     }
 733     else
 734         result = rset_trunc (zi->wordIsam, grep_info.isam_p_buf,
 735                              grep_info.isam_p_indx);
 736     xfree (grep_info.isam_p_buf);
 737     return result;
 738 }
 739
 740 static RSET rpn_prox (RSET *rset, int rset_no)
 741 {
 742     int i;
 743     RSFD *rsfd;
 744     int  *more;
 745     struct it_key **buf;
 746     RSFD rsfd_result;
 747     RSET result;
 748     rset_temp_parms parms;
 749
 750     rsfd = xmalloc (sizeof(*rsfd)*rset_no);
 751     more = xmalloc (sizeof(*more)*rset_no);
 752     buf = xmalloc (sizeof(*buf)*rset_no);
 753
 754     for (i = 0; i<rset_no; i++)
 755     {
 756         buf[i] = xmalloc (sizeof(**buf));
 757         rsfd[i] = rset_open (rset[i], RSETF_READ|RSETF_SORT_SYSNO);
 758         if (!(more[i] = rset_read (rset[i], rsfd[i], buf[i])))
 759         {
 760             while (i >= 0)
 761             {
 762                 rset_close (rset[i], rsfd[i]);
 763                 xfree (buf[i]);
 764                 --i;
 765             }
 766             xfree (rsfd);
 767             xfree (more);
 768             xfree (buf);
 769             return rset_create (rset_kind_null, NULL);
 770         }
 771     }
 772     parms.key_size = sizeof (struct it_key);
 773     result = rset_create (rset_kind_temp, &parms);
 774     rsfd_result = rset_open (result, RSETF_WRITE|RSETF_SORT_SYSNO);
 775
 776     while (*more)
 777     {
 778         for (i = 1; i<rset_no; i++)
 779         {
 780             int cmp;
 781
 782             if (!more[i])
 783             {
 784                 *more = 0;
 785                 break;
 786             }
 787             cmp = key_compare (buf[i], buf[i-1]);
 788             if (cmp > 1)
 789             {
 790                 more[i-1] = rset_read (rset[i-1], rsfd[i-1], buf[i-1]);
 791                 break;
 792             }
 793             else if (cmp == 1)
 794             {
 795                 if (buf[i-1]->seqno+1 != buf[i]->seqno)
 796                 {
 797                     more[i-1] = rset_read (rset[i-1], rsfd[i-1], buf[i-1]);
 798                     break;
 799                 }
 800             }
 801             else
 802             {
 803                 more[i] = rset_read (rset[i], rsfd[i], buf[i]);
 804                 break;
 805             }
 806         }
 807         if (i == rset_no)
 808         {
 809             rset_write (result, rsfd_result, buf[0]);
 810             more[0] = rset_read (*rset, *rsfd, *buf);
 811         }
 812     }
 813
 814     for (i = 0; i<rset_no; i++)
 815     {
 816         rset_close (rset[i], rsfd[i]);
 817         xfree (buf[i]);
 818     }
 819     rset_close (result, rsfd_result);
 820     xfree (buf);
 821     xfree (more);
 822     xfree (rsfd);
 823     return result;
 824 }
 825
 826 static RSET rpn_search_APT_phrase (ZServerInfo *zi,
 827                                    Z_AttributesPlusTerm *zapt,
 828                                    oid_value attributeSet)
 829 {
 830     char termz[IT_MAX_WORD+1];
 831     char term_sub[IT_MAX_WORD+1];
 832     char *p0 = termz, *p1 = NULL;
 833     RSET rset[60], result;
 834     int i, rset_no = 0;
 835     struct grep_info grep_info;
 836
 837     if (zapt->term->which != Z_Term_general)
 838     {
 839         zi->errCode = 124;
 840         return NULL;
 841     }
 842     trans_term (zi, zapt, termz);
 843
 844     grep_info.isam_p_size = 0;
 845     grep_info.isam_p_buf = NULL;
 846
 847     while (1)
 848     {
 849         if ((p1 = strchr (p0, ' ')))
 850         {
 851             memcpy (term_sub, p0, p1-p0);
 852             term_sub[p1-p0] = '\0';
 853         }
 854         else
 855             strcpy (term_sub, p0);
 856
 857         grep_info.isam_p_indx = 0;
 858         if (trunc_term (zi, zapt, term_sub, attributeSet, &grep_info))
 859             return NULL;
 860         if (grep_info.isam_p_indx == 0)
 861             rset[rset_no] = rset_create (rset_kind_null, NULL);
 862         else if (grep_info.isam_p_indx > 1)
 863             rset[rset_no] = rset_trunc (zi->wordIsam,
 864                                         grep_info.isam_p_buf,
 865                                         grep_info.isam_p_indx);
 866         else
 867         {
 868             rset_isam_parms parms;
 869
 870             parms.is = zi->wordIsam;
 871             parms.pos = *grep_info.isam_p_buf;
 872             rset[rset_no] = rset_create (rset_kind_isam, &parms);
 873         }
 874         assert (rset[rset_no]);
 875         if (++rset_no >= sizeof(rset)/sizeof(*rset))
 876             break;
 877         if (!p1)
 878             break;
 879         p0 = p1;
 880         while (*++p0 == ' ')
 881             ;
 882     }
 883     xfree (grep_info.isam_p_buf);
 884     if (rset_no == 0)
 885         return rset_create (rset_kind_null, NULL);
 886     else if (rset_no == 1)
 887         return (rset[0]);
 888
 889     result = rpn_prox (rset, rset_no);
 890     for (i = 0; i<rset_no; i++)
 891         rset_delete (rset[i]);
 892     return result;
 893 }
 894
 895 static RSET rpn_search_APT_local (ZServerInfo *zi, Z_AttributesPlusTerm *zapt,
 896                                   oid_value attributeSet)
 897 {
 898     RSET result;
 899     RSFD rsfd;
 900     struct it_key key;
 901     rset_temp_parms parms;
 902     char termz[IT_MAX_WORD+1];
 903
 904     if (zapt->term->which != Z_Term_general)
 905     {
 906         zi->errCode = 124;
 907         return NULL;
 908     }
 909     parms.key_size = sizeof (struct it_key);
 910     result = rset_create (rset_kind_temp, &parms);
 911     rsfd = rset_open (result, RSETF_WRITE|RSETF_SORT_SYSNO);
 912
 913     trans_term (zi, zapt, termz);
 914     key.sysno = atoi (termz);
 915     if (key.sysno <= 0)
 916         key.sysno = 1;
 917     rset_write (result, rsfd, &key);
 918     rset_close (result, rsfd);
 919     return result;
 920 }
 921
 922 static RSET rpn_search_APT (ZServerInfo *zi, Z_AttributesPlusTerm *zapt,
 923                             oid_value attributeSet)
 924 {
 925     AttrType relation;
 926     AttrType structure;
 927     int relation_value, structure_value;
 928
 929     attr_init (&relation, zapt, 2);
 930     attr_init (&structure, zapt, 4);
 931
 932     relation_value = attr_find (&relation, NULL);
 933     structure_value = attr_find (&structure, NULL);
 934     switch (structure_value)
 935     {
 936     case -1:
 937         if (relation_value == 102) /* relevance relation */
 938             return rpn_search_APT_relevance (zi, zapt, attributeSet);
 939         return rpn_search_APT_phrase (zi, zapt, attributeSet);
 940     case 1: /* phrase */
 941         if (relation_value == 102) /* relevance relation */
 942             return rpn_search_APT_relevance (zi, zapt, attributeSet);
 943         return rpn_search_APT_phrase (zi, zapt, attributeSet);
 944         break;
 945     case 2: /* word */
 946         if (relation_value == 102) /* relevance relation */
 947             return rpn_search_APT_relevance (zi, zapt, attributeSet);
 948         return rpn_search_APT_word (zi, zapt, attributeSet);
 949     case 3: /* key */
 950         break;
 951     case 4: /* year */
 952         break;
 953     case 5: /* date - normalized */
 954         break;
 955     case 6: /* word list */
 956         return rpn_search_APT_relevance (zi, zapt, attributeSet);
 957     case 100: /* date - un-normalized */
 958         break;
 959     case 101: /* name - normalized */
 960         break;
 961     case 102: /* date - un-normalized */
 962         break;
 963     case 103: /* structure */
 964         break;
 965     case 104: /* urx */
 966         break;
 967     case 105: /* free-form-text */
 968         return rpn_search_APT_relevance (zi, zapt, attributeSet);
 969     case 106: /* document-text */
 970         return rpn_search_APT_relevance (zi, zapt, attributeSet);
 971     case 107: /* local-number */
 972         return rpn_search_APT_local (zi, zapt, attributeSet);
 973     case 108: /* string */
 974         return rpn_search_APT_word (zi, zapt, attributeSet);
 975     case 109: /* numeric string */
 976         break;
 977     }
 978     zi->errCode = 118;
 979     return NULL;
 980 }
 981
 982 static RSET rpn_search_ref (ZServerInfo *zi, Z_ResultSetId *resultSetId)
 983 {
 984     ZServerSet *s;
 985
 986     if (!(s = resultSetGet (zi, resultSetId)))
 987         return rset_create (rset_kind_null, NULL);
 988     return s->rset;
 989 }
 990
 991 static RSET rpn_search_structure (ZServerInfo *zi, Z_RPNStructure *zs,
 992                                   oid_value attributeSet)
 993 {
 994     RSET r = NULL;
 995     if (zs->which == Z_RPNStructure_complex)
 996     {
 997         rset_bool_parms bool_parms;
 998
 999         bool_parms.rset_l = rpn_search_structure (zi, zs->u.complex->s1,
1000                                                   attributeSet);
1001         if (bool_parms.rset_l == NULL)
1002             return NULL;
1003         bool_parms.rset_r = rpn_search_structure (zi, zs->u.complex->s2,
1004                                                   attributeSet);
1005         if (bool_parms.rset_r == NULL)
1006         {
1007             rset_delete (bool_parms.rset_l);
1008             return NULL;
1009         }
1010         bool_parms.key_size = sizeof(struct it_key);
1011         bool_parms.cmp = key_compare;
1012
1013         switch (zs->u.complex->operator->which)
1014         {
1015         case Z_Operator_and:
1016             r = rset_create (rset_kind_and, &bool_parms);
1017             break;
1018         case Z_Operator_or:
1019             r = rset_create (rset_kind_or, &bool_parms);
1020             break;
1021         case Z_Operator_and_not:
1022             r = rset_create (rset_kind_not, &bool_parms);
1023             break;
1024         default:
1025             assert (0);
1026         }
1027     }
1028     else if (zs->which == Z_RPNStructure_simple)
1029     {
1030         if (zs->u.simple->which == Z_Operand_APT)
1031         {
1032             logf (LOG_DEBUG, "rpn_search_APT");
1033             r = rpn_search_APT (zi, zs->u.simple->u.attributesPlusTerm,
1034                                 attributeSet);
1035         }
1036         else if (zs->u.simple->which == Z_Operand_resultSetId)
1037         {
1038             logf (LOG_DEBUG, "rpn_search_ref");
1039             r = rpn_search_ref (zi, zs->u.simple->u.resultSetId);
1040         }
1041         else
1042         {
1043             assert (0);
1044         }
1045     }
1046     else
1047     {
1048         assert (0);
1049     }
1050     return r;
1051 }
1052
1053 static void count_set (RSET r, int *count)
1054 {
1055     int psysno = 0;
1056     struct it_key key;
1057     RSFD rfd;
1058
1059     logf (LOG_DEBUG, "rpn_save_set");
1060     *count = 0;
1061     rfd = rset_open (r, RSETF_READ|RSETF_SORT_SYSNO);
1062     while (rset_read (r, rfd, &key))
1063     {
1064         if (key.sysno != psysno)
1065         {
1066             psysno = key.sysno;
1067             (*count)++;
1068         }
1069     }
1070     rset_close (r, rfd);
1071     logf (LOG_DEBUG, "%d distinct sysnos", *count);
1072 }
1073
1074 int rpn_search (ZServerInfo *zi,
1075                 Z_RPNQuery *rpn, int num_bases, char **basenames,
1076                 const char *setname, int *hits)
1077 {
1078     RSET rset;
1079     oident *attrset;
1080     oid_value attributeSet;
1081
1082     zi->errCode = 0;
1083     zi->errString = NULL;
1084
1085     attrset = oid_getentbyoid (rpn->attributeSetId);
1086     attributeSet = attrset->value;
1087
1088     rset = rpn_search_structure (zi, rpn->RPNStructure, attributeSet);
1089     if (!rset)
1090         return zi->errCode;
1091     count_set (rset, hits);
1092     resultSetAdd (zi, setname, 1, rset);
1093     if (zi->errCode)
1094         logf (LOG_DEBUG, "search error: %d", zi->errCode);
1095     return zi->errCode;
1096 }
1097
1098 struct scan_info {
1099     struct scan_entry *list;
1100     ODR odr;
1101     int before, after;
1102     ISAM isam;
1103     char prefix[20];
1104 };
1105
1106 static int scan_handle (Dict_char *name, const char *info, int pos,
1107                         void *client)
1108 {
1109     int len_prefix, idx;
1110     ISAM_P isam_p;
1111     RSET rset;
1112     struct scan_info *scan_info = client;
1113
1114     rset_isam_parms parms;
1115
1116     len_prefix = strlen(scan_info->prefix);
1117     if (memcmp (name, scan_info->prefix, len_prefix))
1118         return 1;
1119     if (pos > 0)
1120         idx = scan_info->after - pos + scan_info->before;
1121     else
1122         idx = - pos - 1;
1123     scan_info->list[idx].term = odr_malloc (scan_info->odr,
1124                                             strlen(name + len_prefix)+1);
1125     strcpy (scan_info->list[idx].term, name + len_prefix);
1126     assert (*info == sizeof(isam_p));
1127     memcpy (&isam_p, info+1, sizeof(isam_p));
1128     parms.is = scan_info->isam;
1129     parms.pos = isam_p;
1130 #if 1
1131     rset = rset_create (rset_kind_isam, &parms);
1132     count_set (rset, &scan_info->list[idx].occurrences);
1133     rset_delete (rset);
1134 #else
1135     scan_info->list[idx].occurrences = 1;
1136 #endif
1137     logf (LOG_DEBUG, "pos=%3d idx=%3d name=%s", pos, idx, name);
1138     return 0;
1139 }
1140
1141 int rpn_scan (ZServerInfo *zi, ODR odr, Z_AttributesPlusTerm *zapt,
1142               int *position, int *num_entries, struct scan_entry **list,
1143               int *status)
1144 {
1145     int i, j, sizez;
1146     int pos = *position;
1147     int num = *num_entries;
1148     int before;
1149     int after;
1150     char termz[IT_MAX_WORD+20];
1151     AttrType use;
1152     int use_value;
1153     Z_Term *term = zapt->term;
1154     struct scan_info scan_info;
1155
1156     logf (LOG_DEBUG, "scan, position = %d, num = %d", pos, num);
1157     scan_info.before = before = pos-1;
1158     scan_info.after = after = 1+num-pos;
1159     scan_info.odr = odr;
1160
1161     logf (LOG_DEBUG, "scan, before = %d, after = %d", before, after);
1162
1163     scan_info.isam = zi->wordIsam;
1164     scan_info.list = odr_malloc (odr, (before+after)*sizeof(*scan_info.list));
1165     for (j = 0; j<before+after; j++)
1166         scan_info.list[j].term = NULL;
1167     attr_init (&use, zapt, 1);
1168     use_value = attr_find (&use, NULL);
1169     logf (LOG_DEBUG, "use value %d", use_value);
1170
1171     if (use_value == -1)
1172         use_value = 1016;
1173     i = index_word_prefix (termz, 1, use_value);
1174     strcpy (scan_info.prefix, termz);
1175     sizez = term->u.general->len;
1176     if (sizez > IT_MAX_WORD)
1177         sizez = IT_MAX_WORD;
1178     for (j = 0; j<sizez; j++)
1179         termz[j+i] = index_char_cvt (term->u.general->buf[j]);
1180     termz[j+i] = '\0';
1181
1182     dict_scan (zi->wordDict, termz, &before, &after, &scan_info, scan_handle);
1183
1184     *status = BEND_SCAN_SUCCESS;
1185
1186     for (i = 0; i<scan_info.after; i++)
1187         if (scan_info.list[scan_info.before+scan_info.after-i-1].term)
1188             break;
1189     *num_entries -= i;
1190     if (i)
1191         *status = BEND_SCAN_PARTIAL;
1192
1193     for (i = 0; i<scan_info.before; i++)
1194         if (scan_info.list[i].term)
1195             break;
1196     if (i)
1197         *status = BEND_SCAN_PARTIAL;
1198     *position -= i;
1199     *num_entries -= i;
1200
1201     *list = scan_info.list+i;       /* list is set to first 'real' entry */
1202
1203     if (*num_entries == 0)          /* signal 'unsupported use-attribute' */
1204         zi->errCode = 114;          /* if no entries was found */
1205     logf (LOG_DEBUG, "position = %d, num_entries = %d",
1206           *position, *num_entries);
1207     if (zi->errCode)
1208         logf (LOG_DEBUG, "scan error: %d", zi->errCode);
1209     return 0;
1210 }
1211
1212
1213