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