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