Bug fixes in rpn_prox.
[idzebra-moved-to-github.git] / index / zrpn.c
index fb9d9ea..5fbaca4 100644 (file)
@@ -4,7 +4,16 @@
  * Sebastian Hammer, Adam Dickmeiss
  *
  * $Log: zrpn.c,v $
- * Revision 1.23  1995-10-06 16:33:37  adam
+ * Revision 1.26  1995-10-12 12:40:54  adam
+ * Bug fixes in rpn_prox.
+ *
+ * Revision 1.25  1995/10/10  13:59:24  adam
+ * Function rset_open changed its wflag parameter to general flags.
+ *
+ * Revision 1.24  1995/10/09  16:18:37  adam
+ * Function dict_lookup_grep got extra client data parameter.
+ *
+ * Revision 1.23  1995/10/06  16:33:37  adam
  * Use attribute mappings.
  *
  * Revision 1.22  1995/10/06  15:07:39  adam
@@ -97,10 +106,10 @@ int index_word_prefix_map (char *string, oid_value attrSet, int attrUse)
 {
     attent *attp;
 
-    logf (LOG_DEBUG, "oid_value attrSet = %d", attrSet);
-    logf (LOG_DEBUG, "int attrUse = %d", attrUse);
+    logf (LOG_DEBUG, "oid_value attrSet = %d, attrUse = %d", attrSet, attrUse);
     attp = att_getentbyatt (attrSet, attrUse);
-    assert (attp);
+    if (!attp)
+        return -1;
     logf (LOG_DEBUG, "ord=%d", attp->attset_ordinal);
     return index_word_prefix (string, attp->attset_ordinal,
                               attp->local_attribute);
@@ -318,7 +327,7 @@ static RSET rset_trunc (ISAM isam, ISAM_P *isam_p, int from, int to,
         ispt = xmalloc (sizeof(*ispt) * (to-from));
         parms.key_size = sizeof (struct it_key);
         result = rset_create (rset_kind_temp, &parms);
-        rsfd = rset_open (result, 1);
+        rsfd = rset_open (result, RSETF_WRITE|RSETF_SORT_SYSNO);
 
         ti = heap_init (to-from, sizeof(struct it_key),
                         key_compare);
@@ -346,42 +355,44 @@ static RSET rset_trunc (ISAM isam, ISAM_P *isam_p, int from, int to,
     }
 }
 
-static ISAM_P *isam_p_buf = NULL;
-static int isam_p_size = 0;
-static int isam_p_indx;
+struct grep_info {
+    ISAM_P *isam_p_buf;
+    int isam_p_size;
+    int isam_p_indx;
+};
 
-static void add_isam_p (const char *info)
+static void add_isam_p (const char *info, struct grep_info *p)
 {
-    if (isam_p_indx == isam_p_size)
+    if (p->isam_p_indx == p->isam_p_size)
     {
         ISAM_P *new_isam_p_buf;
         
-        isam_p_size = 2*isam_p_size + 100;
+        p->isam_p_size = 2*p->isam_p_size + 100;
         new_isam_p_buf = xmalloc (sizeof(*new_isam_p_buf) *
-                                  isam_p_size);
-        if (isam_p_buf)
+                                  p->isam_p_size);
+        if (p->isam_p_buf)
         {
-            memcpy (new_isam_p_buf, isam_p_buf,
-                    isam_p_indx * sizeof(*isam_p_buf));
-            xfree (isam_p_buf);
+            memcpy (new_isam_p_buf, p->isam_p_buf,
+                    p->isam_p_indx * sizeof(*p->isam_p_buf));
+            xfree (p->isam_p_buf);
         }
-        isam_p_buf = new_isam_p_buf;
+        p->isam_p_buf = new_isam_p_buf;
     }
-    assert (*info == sizeof(*isam_p_buf));
-    memcpy (isam_p_buf + isam_p_indx, info+1, sizeof(*isam_p_buf));
-    isam_p_indx++;
+    assert (*info == sizeof(*p->isam_p_buf));
+    memcpy (p->isam_p_buf + p->isam_p_indx, info+1, sizeof(*p->isam_p_buf));
+    (p->isam_p_indx)++;
 }
 
-static int grep_handle (Dict_char *name, const char *info)
+static int grep_handle (Dict_char *name, const char *info, void *p)
 {
     logf (LOG_DEBUG, "dict name: %s", name);
-    add_isam_p (info);
+    add_isam_p (info, p);
     return 0;
 }
 
 static int trunc_term (ZServerInfo *zi, Z_AttributesPlusTerm *zapt,
-                       const char *term_sub, ISAM_P **isam_ps,
-                       oid_value attributeSet)
+                       const char *term_sub,
+                       oid_value attributeSet, struct grep_info *grep_info)
 {
     char term_dict[2*IT_MAX_WORD+2];
     int i, j;
@@ -402,6 +413,11 @@ static int trunc_term (ZServerInfo *zi, Z_AttributesPlusTerm *zapt,
     if (use_value == -1)
         use_value = 1016;
     i = index_word_prefix_map (term_dict, curAttributeSet, use_value);
+    if (i < 0)
+    {
+        zi->errCode = 114;
+        return -1;
+    }
     
     switch (truncation_value)
     {
@@ -410,12 +426,12 @@ static int trunc_term (ZServerInfo *zi, Z_AttributesPlusTerm *zapt,
         strcat (term_dict, term_sub);
         logf (LOG_DEBUG, "dict_lookup: %s", term_dict);
         if ((info = dict_lookup (zi->wordDict, term_dict)))
-            add_isam_p (info);
+            add_isam_p (info, grep_info);
         break;
     case 1:          /* right truncation */
         strcat (term_dict, term_sub);
         strcat (term_dict, ".*");
-        dict_lookup_grep (zi->wordDict, term_dict, 0, grep_handle);
+        dict_lookup_grep (zi->wordDict, term_dict, 0, grep_info, grep_handle);
         break;
     case 2:          /* left truncation */
     case 3:          /* left&right truncation */
@@ -433,15 +449,14 @@ static int trunc_term (ZServerInfo *zi, Z_AttributesPlusTerm *zapt,
             else
                 term_dict[j++] = term_sub[i];
         term_dict[j] = '\0';
-        dict_lookup_grep (zi->wordDict, term_dict, 0, grep_handle);
+        dict_lookup_grep (zi->wordDict, term_dict, 0, grep_info, grep_handle);
         break;
     case 102:        /* regular expression */
         strcat (term_dict, term_sub);
-        dict_lookup_grep (zi->wordDict, term_dict, 0, grep_handle);
+        dict_lookup_grep (zi->wordDict, term_dict, 0, grep_info, grep_handle);
         break;
     }
-    *isam_ps = isam_p_buf;
-    logf (LOG_DEBUG, "%d positions", isam_p_indx);
+    logf (LOG_DEBUG, "%d positions", grep_info->isam_p_indx);
     return 0;
 }
 
@@ -466,7 +481,9 @@ static RSET rpn_search_APT_relevance (ZServerInfo *zi,
     rset_relevance_parms parms;
     char termz[IT_MAX_WORD+1];
     char term_sub[IT_MAX_WORD+1];
+    struct grep_info grep_info;
     char *p0 = termz, *p1 = NULL;
+    RSET result;
 
     parms.key_size = sizeof(struct it_key);
     parms.max_rec = 100;
@@ -479,7 +496,9 @@ static RSET rpn_search_APT_relevance (ZServerInfo *zi,
         return NULL;
     }
     trans_term (zi, zapt, termz);
-    isam_p_indx = 0;  /* global, set by trunc_term - see below */
+    grep_info.isam_p_indx = 0;
+    grep_info.isam_p_size = 0;
+    grep_info.isam_p_buf = NULL;
     while (1)
     {
         if ((p1 = strchr (p0, ' ')))
@@ -489,27 +508,32 @@ static RSET rpn_search_APT_relevance (ZServerInfo *zi,
         }
         else
             strcpy (term_sub, p0);
-        if (trunc_term (zi, zapt, term_sub, &parms.isam_positions,
-                        attributeSet))
+        if (trunc_term (zi, zapt, term_sub, attributeSet, &grep_info))
             return NULL;
         if (!p1)
             break;
-        p0 = p1+1;
+        p0 = p1;
+        while (*++p0 == ' ')
+            ;
     }
-    parms.no_isam_positions = isam_p_indx;
-    if (isam_p_indx > 0)
-        return rset_create (rset_kind_relevance, &parms);
+    parms.isam_positions = grep_info.isam_p_buf;
+    parms.no_isam_positions = grep_info.isam_p_indx;
+    if (grep_info.isam_p_indx > 0)
+        result = rset_create (rset_kind_relevance, &parms);
     else
-        return rset_create (rset_kind_null, NULL);
+        result = rset_create (rset_kind_null, NULL);
+    xfree (grep_info.isam_p_buf);
+    return result;
 }
 
 static RSET rpn_search_APT_word (ZServerInfo *zi,
                                  Z_AttributesPlusTerm *zapt,
                                  oid_value attributeSet)
 {
-    ISAM_P *isam_positions;
     rset_isam_parms parms;
     char termz[IT_MAX_WORD+1];
+    struct grep_info grep_info;
+    RSET result;
 
     if (zapt->term->which != Z_Term_general)
     {
@@ -517,28 +541,124 @@ static RSET rpn_search_APT_word (ZServerInfo *zi,
         return NULL;
     }
     trans_term (zi, zapt, termz);
-    isam_p_indx = 0;  /* global, set by trunc_term - see below */
-    if (trunc_term (zi, zapt, termz, &isam_positions, attributeSet))
+
+    grep_info.isam_p_indx = 0;
+    grep_info.isam_p_size = 0;
+    grep_info.isam_p_buf = NULL;
+
+    if (trunc_term (zi, zapt, termz, attributeSet, &grep_info))
         return NULL;
-    if (isam_p_indx < 1)
-        return rset_create (rset_kind_null, NULL);
-    else if (isam_p_indx == 1)
+    if (grep_info.isam_p_indx < 1)
+        result = rset_create (rset_kind_null, NULL);
+    else if (grep_info.isam_p_indx == 1)
     {
         parms.is = zi->wordIsam;
-        parms.pos = *isam_positions;
-        return rset_create (rset_kind_isam, &parms);
+        parms.pos = *grep_info.isam_p_buf;
+        result = rset_create (rset_kind_isam, &parms);
     }
     else
-        return rset_trunc (zi->wordIsam, isam_positions, 0, isam_p_indx, 400);
+        result = rset_trunc (zi->wordIsam, grep_info.isam_p_buf, 0,
+                             grep_info.isam_p_indx, 400);
+    xfree (grep_info.isam_p_buf);
+    return result;
+}
+
+static RSET rpn_prox (RSET *rset, int rset_no)
+{
+    int i;
+    RSFD *rsfd;
+    int  *more;
+    struct it_key **buf;
+    RSFD rsfd_result;
+    RSET result;
+    rset_temp_parms parms;
+    
+    rsfd = xmalloc (sizeof(*rsfd)*rset_no);
+    more = xmalloc (sizeof(*more)*rset_no);
+    buf = xmalloc (sizeof(*buf)*rset_no);
+
+    for (i = 0; i<rset_no; i++)
+    {
+        buf[i] = xmalloc (sizeof(**buf));
+        rsfd[i] = rset_open (rset[i], RSETF_READ|RSETF_SORT_SYSNO);
+        if (!(more[i] = rset_read (rset[i], rsfd[i], buf[i])))
+        {
+            while (i >= 0)
+            {
+                rset_close (rset[i], rsfd[i]);
+                xfree (buf[i]);
+                --i;
+            }
+            xfree (rsfd);
+            xfree (more);
+            xfree (buf);
+            return rset_create (rset_kind_null, NULL);
+        }
+    }
+    parms.key_size = sizeof (struct it_key);
+    result = rset_create (rset_kind_temp, &parms);
+    rsfd_result = rset_open (result, RSETF_WRITE|RSETF_SORT_SYSNO);
+    
+    while (*more)
+    {
+        for (i = 1; i<rset_no; i++)
+        {
+            int cmp;
+            
+            if (!more[i])
+            {
+                *more = 0;
+                break;
+            }
+            cmp = key_compare (buf[i], buf[i-1]);
+            if (cmp > 1)
+            {
+                more[i-1] = rset_read (rset[i-1], rsfd[i-1], buf[i-1]);
+                break;
+            }
+            else if (cmp == 1)
+            {
+                if (buf[i-1]->seqno+1 != buf[i]->seqno)
+                {
+                    more[i-1] = rset_read (rset[i-1], rsfd[i-1], buf[i-1]);
+                    break;
+                }
+            }
+            else
+            {
+                more[i] = rset_read (rset[i], rsfd[i], buf[i]);
+                break;
+            }
+        }
+        if (i == rset_no)
+        {
+            rset_write (result, rsfd_result, buf[0]);
+            more[0] = rset_read (*rset, *rsfd, *buf);
+        }
+    }
+    
+    for (i = 0; i<rset_no; i++)
+    {
+        rset_close (rset[i], rsfd[i]);
+        xfree (buf[i]);
+    }
+    rset_close (result, rsfd_result);
+    xfree (buf);
+    xfree (more);
+    xfree (rsfd);
+    return result;
 }
 
 static RSET rpn_search_APT_phrase (ZServerInfo *zi,
                                    Z_AttributesPlusTerm *zapt,
                                    oid_value attributeSet)
 {
-    ISAM_P *isam_positions;
-    rset_isam_parms parms;
     char termz[IT_MAX_WORD+1];
+    char term_sub[IT_MAX_WORD+1];
+    char *p0 = termz, *p1 = NULL;
+    RSET rset[60], result;
+    int i, rset_no = 0;
+    struct grep_info grep_info;
 
     if (zapt->term->which != Z_Term_general)
     {
@@ -546,14 +666,56 @@ static RSET rpn_search_APT_phrase (ZServerInfo *zi,
         return NULL;
     }
     trans_term (zi, zapt, termz);
-    isam_p_indx = 0;  /* global, set by trunc_term - see below */
-    if (trunc_term (zi, zapt, termz, &isam_positions, attributeSet))
-        return NULL;
-    if (isam_p_indx != 1)
+
+    grep_info.isam_p_size = 0;
+    grep_info.isam_p_buf = NULL;
+
+    while (1)
+    {
+        if ((p1 = strchr (p0, ' ')))
+        {
+            memcpy (term_sub, p0, p1-p0);
+            term_sub[p1-p0] = '\0';
+        }
+        else
+            strcpy (term_sub, p0);
+
+        grep_info.isam_p_indx = 0;
+        if (trunc_term (zi, zapt, term_sub, attributeSet, &grep_info))
+            return NULL;
+        if (grep_info.isam_p_indx == 0)
+            rset[rset_no] = rset_create (rset_kind_null, NULL);
+        else if (grep_info.isam_p_indx > 1)
+            rset[rset_no] = rset_trunc (zi->wordIsam,
+                                        grep_info.isam_p_buf, 0,
+                                        grep_info.isam_p_indx, 400);
+        else
+        {
+            rset_isam_parms parms;
+            
+            parms.is = zi->wordIsam;
+            parms.pos = *grep_info.isam_p_buf;
+            rset[rset_no] = rset_create (rset_kind_isam, &parms);
+        }
+        assert (rset[rset_no]);
+        if (++rset_no >= sizeof(rset)/sizeof(*rset))
+            break;
+        if (!p1)
+            break;
+        p0 = p1;
+        while (*++p0 == ' ')
+            ;
+    }
+    xfree (grep_info.isam_p_buf);
+    if (rset_no == 0)
         return rset_create (rset_kind_null, NULL);
-    parms.is = zi->wordIsam;
-    parms.pos = *isam_positions;
-    return rset_create (rset_kind_isam, &parms);
+    else if (rset_no == 1)
+        return (rset[0]);
+
+    result = rpn_prox (rset, rset_no);
+    for (i = 0; i<rset_no; i++)
+        rset_delete (rset[i]);
+    return result;
 }
 
 static RSET rpn_search_APT_local (ZServerInfo *zi, Z_AttributesPlusTerm *zapt,
@@ -572,7 +734,7 @@ static RSET rpn_search_APT_local (ZServerInfo *zi, Z_AttributesPlusTerm *zapt,
     }
     parms.key_size = sizeof (struct it_key);
     result = rset_create (rset_kind_temp, &parms);
-    rsfd = rset_open (result, 1);
+    rsfd = rset_open (result, RSETF_WRITE|RSETF_SORT_SYSNO);
 
     trans_term (zi, zapt, termz);
     key.sysno = atoi (termz);
@@ -601,7 +763,7 @@ static RSET rpn_search_APT (ZServerInfo *zi, Z_AttributesPlusTerm *zapt,
     case -1:
         if (relation_value == 102) /* relevance relation */
             return rpn_search_APT_relevance (zi, zapt, attributeSet);
-        return rpn_search_APT_word (zi, zapt, attributeSet);
+        return rpn_search_APT_phrase (zi, zapt, attributeSet);
     case 1: /* phrase */
         if (relation_value == 102) /* relevance relation */
             return rpn_search_APT_relevance (zi, zapt, attributeSet);
@@ -723,7 +885,7 @@ static void count_set (RSET r, int *count)
 
     logf (LOG_DEBUG, "rpn_save_set");
     *count = 0;
-    rfd = rset_open (r, 0);
+    rfd = rset_open (r, RSETF_READ|RSETF_SORT_SYSNO);
     while (rset_read (r, rfd, &key))
     {
         if (key.sysno != psysno)
@@ -760,46 +922,52 @@ int rpn_search (ZServerInfo *zi,
     return zi->errCode;
 }
 
-static struct scan_entry *scan_list;
-static ODR scan_odr;
-static int scan_before, scan_after;
-static ISAM scan_isam;
-static char scan_prefix[20];
+struct scan_info {
+    struct scan_entry *list;
+    ODR odr;
+    int before, after;
+    ISAM isam;
+    char prefix[20];
+};
 
-static int scan_handle (Dict_char *name, const char *info, int pos)
+static int scan_handle (Dict_char *name, const char *info, int pos, 
+                        void *client)
 {
     int len_prefix, idx;
     ISAM_P isam_p;
     RSET rset;
+    struct scan_info *scan_info = client;
 
     rset_isam_parms parms;
 
-    len_prefix = strlen(scan_prefix);
-    if (memcmp (name, scan_prefix, len_prefix))
+    len_prefix = strlen(scan_info->prefix);
+    if (memcmp (name, scan_info->prefix, len_prefix))
         return 1;
     if (pos > 0)
-        idx = scan_after - pos + scan_before;
+        idx = scan_info->after - pos + scan_info->before;
     else
         idx = - pos - 1;
-    scan_list[idx].term = odr_malloc (scan_odr, strlen(name + len_prefix)+1);
-    strcpy (scan_list[idx].term, name + len_prefix);
+    scan_info->list[idx].term = odr_malloc (scan_info->odr,
+                                            strlen(name + len_prefix)+1);
+    strcpy (scan_info->list[idx].term, name + len_prefix);
     assert (*info == sizeof(isam_p));
     memcpy (&isam_p, info+1, sizeof(isam_p));
-    parms.is = scan_isam;
+    parms.is = scan_info->isam;
     parms.pos = isam_p;
 #if 1
     rset = rset_create (rset_kind_isam, &parms);
-    count_set (rset, &scan_list[idx].occurrences);
+    count_set (rset, &scan_info->list[idx].occurrences);
     rset_delete (rset);
 #else
-    scan_list[idx].occurrences = 1;
+    scan_info->list[idx].occurrences = 1;
 #endif
     logf (LOG_DEBUG, "pos=%3d idx=%3d name=%s", pos, idx, name);
     return 0;
 }
 
 int rpn_scan (ZServerInfo *zi, ODR odr, Z_AttributesPlusTerm *zapt,
-              int *position, int *num_entries, struct scan_entry **list)
+              int *position, int *num_entries, struct scan_entry **list,
+              int *status)
 {
     int i, j, sizez;
     int pos = *position;
@@ -810,18 +978,19 @@ int rpn_scan (ZServerInfo *zi, ODR odr, Z_AttributesPlusTerm *zapt,
     AttrType use;
     int use_value;
     Z_Term *term = zapt->term;
+    struct scan_info scan_info;
 
     logf (LOG_DEBUG, "scan, position = %d, num = %d", pos, num);
-    scan_before = before = pos-1;
-    scan_after = after = 1+num-pos;
-    scan_odr = odr;
+    scan_info.before = before = pos-1;
+    scan_info.after = after = 1+num-pos;
+    scan_info.odr = odr;
 
     logf (LOG_DEBUG, "scan, before = %d, after = %d", before, after);
     
-    scan_isam = zi->wordIsam;
-    scan_list = *list = odr_malloc (odr, (before+after)*sizeof(**list));
+    scan_info.isam = zi->wordIsam;
+    scan_info.list = odr_malloc (odr, (before+after)*sizeof(*scan_info.list));
     for (j = 0; j<before+after; j++)
-        scan_list[j].term = "------";
+        scan_info.list[j].term = NULL;
     attr_init (&use, zapt, 1);
     use_value = attr_find (&use, NULL);
     logf (LOG_DEBUG, "use value %d", use_value);
@@ -829,7 +998,7 @@ int rpn_scan (ZServerInfo *zi, ODR odr, Z_AttributesPlusTerm *zapt,
     if (use_value == -1)
         use_value = 1016;
     i = index_word_prefix (termz, 1, use_value);
-    strcpy (scan_prefix, termz);
+    strcpy (scan_info.prefix, termz);
     sizez = term->u.general->len;
     if (sizez > IT_MAX_WORD)
         sizez = IT_MAX_WORD;
@@ -837,10 +1006,35 @@ int rpn_scan (ZServerInfo *zi, ODR odr, Z_AttributesPlusTerm *zapt,
         termz[j+i] = index_char_cvt (term->u.general->buf[j]);
     termz[j+i] = '\0';
     
-    dict_scan (zi->wordDict, termz, &before, &after, scan_handle);
+    dict_scan (zi->wordDict, termz, &before, &after, &scan_info, scan_handle);
+
+    *status = BEND_SCAN_SUCCESS;
+
+    for (i = 0; i<scan_info.after; i++)
+        if (scan_info.list[scan_info.before+scan_info.after-i-1].term)
+            break;
+    *num_entries -= i;
+    if (i)
+        *status = BEND_SCAN_PARTIAL;
+
+    for (i = 0; i<scan_info.before; i++)
+        if (scan_info.list[i].term)
+            break;
+    if (i)
+        *status = BEND_SCAN_PARTIAL;
+    *position -= i;
+    *num_entries -= i;
 
+    *list = scan_info.list+i;       /* list is set to first 'real' entry */
+
+    if (*num_entries == 0)          /* signal 'unsupported use-attribute' */
+        zi->errCode = 114;          /* if no entries was found */
+    logf (LOG_DEBUG, "position = %d, num_entries = %d",
+          *position, *num_entries);
     if (zi->errCode)
         logf (LOG_DEBUG, "scan error: %d", zi->errCode);
     return 0;
 }
               
+
+