Limiting hits to INT_MAX, because yaz can't handle 64-bit hitcounts

[idzebra-moved-to-github.git] / rset / rsbetween.c

diff --git a/rset/rsbetween.c b/rset/rsbetween.c
index f631f33..bf63992 100644 (file)
--- a/rset/rsbetween.c
+++ b/rset/rsbetween.c
@@ -1,4 +1,4 @@
-/* $Id: rsbetween.c,v 1.12 2004-02-12 15:15:54 heikki Exp $
+/* $Id: rsbetween.c,v 1.17 2004-08-06 14:09:02 heikki Exp $
    Copyright (C) 1995,1996,1997,1998,1999,2000,2001,2002
    Index Data Aps
 
@@ -35,8 +35,10 @@ Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
 #include <string.h>
 #include <assert.h>
 
-#include <rsbetween.h>
 #include <zebrautl.h>
+#include <rsbetween.h>
+
+#define RSBETWEEN_DEBUG 0 
 
 static void *r_create_between(RSET ct, const struct rset_control *sel, void *parms);
 static RSFD r_open_between (RSET ct, int flag);
@@ -46,9 +48,9 @@ static void r_rewind_between (RSFD rfd);
 static int r_forward_between(RSET ct, RSFD rfd, void *buf, int *term_index,
                      int (*cmpfunc)(const void *p1, const void *p2),
                      const void *untilbuf);
-static int r_count_between (RSET ct);
 static int r_read_between (RSFD rfd, void *buf, int *term_index);
 static int r_write_between (RSFD rfd, const void *buf);
+static void r_pos_between (RSFD rfd, double *current, double *total);
 
 static const struct rset_control control_between = 
 {
@@ -59,7 +61,7 @@ static const struct rset_control control_between =
     r_delete_between,
     r_rewind_between,
     r_forward_between, /* rset_default_forward, */
-    r_count_between,
+    r_pos_between,
     r_read_between,
     r_write_between,
 };
@@ -98,8 +100,10 @@ struct rset_between_rfd {
     int level;
     struct rset_between_rfd *next;
     struct rset_between_info *info;
+    zint hits;
 };    
 
+#if RSBETWEEN_DEBUG
 static void log2 (struct rset_between_rfd *p, char *msg, int cmp_l, int cmp_r)
 {
     char buf_l[32];
@@ -112,6 +116,7 @@ static void log2 (struct rset_between_rfd *p, char *msg, int cmp_l, int cmp_r)
       (*p->info->printer)(p->buf_r, buf_r), p->more_r, cmp_r,
       p->level);
 }
+#endif
 
 static void *r_create_between (RSET ct, const struct rset_control *sel,
                                void *parms)
@@ -207,6 +212,7 @@ static RSFD r_open_between (RSET ct, int flag)
                                     rfd->buf_attr, &dummy);
     }
     rfd->level=0;
+    rfd->hits=0;
     return rfd;
 }
 
@@ -255,7 +261,9 @@ static void r_rewind_between (RSFD rfd)
     struct rset_between_info *info = ((struct rset_between_rfd*)rfd)->info;
     struct rset_between_rfd *p = (struct rset_between_rfd *) rfd;
 
+#if RSBETWEEN_DEBUG
     logf (LOG_DEBUG, "rsbetween_rewind");
+#endif
     rset_rewind (info->rset_l, p->rfd_l);
     rset_rewind (info->rset_m, p->rfd_m);
     rset_rewind (info->rset_r, p->rfd_r);
@@ -270,6 +278,7 @@ static void r_rewind_between (RSFD rfd)
                                   &dummy);
     }
     p->level=0;
+    p->hits=0;
 }
 
 
@@ -281,22 +290,24 @@ static int r_forward_between(RSET ct, RSFD rfd, void *buf, int *term_index,
     struct rset_between_info *info = ((struct rset_between_rfd*)rfd)->info;
     struct rset_between_rfd *p = (struct rset_between_rfd *) rfd;
     int rc;
+#if RSBETWEEN_DEBUG
     log2( p, "fwd: before forward", 0,0);
+#endif
     /* It is enough to forward the m pointer here, the read will */
     /* naturally forward the l, m, and attr pointers */
     if (p->more_m)
         p->more_m=rset_forward(info->rset_m,p->rfd_m, p->buf_m,
                         &p->term_index_m, info->cmp,untilbuf);
+#if RSBETWEEN_DEBUG
     log2( p, "fwd: after forward M", 0,0);
+#endif
     rc = r_read_between(rfd, buf, term_index);
+#if RSBETWEEN_DEBUG
     log2( p, "fwd: after forward", 0,0);
+#endif
     return rc;
 }
 
-static int r_count_between (RSET ct)
-{
-    return 0;
-}
 
 
 
@@ -310,7 +321,9 @@ static int r_read_between (RSFD rfd, void *buf, int *term_index)
 
     while (p->more_m)
     {
+#if RSBETWEEN_DEBUG
         log2( p, "start of loop", cmp_l, cmp_r);
+#endif
 
         /* forward L until past m, count levels, note rec boundaries */
         if (p->more_l)
@@ -320,7 +333,9 @@ static int r_read_between (RSFD rfd, void *buf, int *term_index)
             p->level = 0;
             cmp_l=2; /* past this record */
         }
+#if RSBETWEEN_DEBUG
         log2( p, "after first L", cmp_l, cmp_r);
+#endif
 
         while (cmp_l < 0)   /* l before m */
         {
@@ -359,7 +374,9 @@ static int r_read_between (RSFD rfd, void *buf, int *term_index)
                                       info->rset_attr, p->rfd_attr,
                                       p->buf_attr, &dummy_term,
                                       info->cmp, p->buf_l);
+#if RSBETWEEN_DEBUG
                             logf(LOG_DEBUG, "btw: after frowarding attr m=%d",p->more_attr);
+#endif
                         }
                     } /* while more_attr */
                 }
@@ -378,7 +395,9 @@ static int r_read_between (RSFD rfd, void *buf, int *term_index)
                         cmp_l= (*info->cmp)(p->buf_l, p->buf_m);
                     else
                         cmp_l=2;
+#if RSBETWEEN_DEBUG
                     log2( p, "after forwarding L", cmp_l, cmp_r);
+#endif
                 }
             } else
             {
@@ -395,17 +414,23 @@ static int r_read_between (RSFD rfd, void *buf, int *term_index)
             }
             else
                 cmp_l=2; 
+#if RSBETWEEN_DEBUG
             log2( p, "end of L loop", cmp_l, cmp_r);
+#endif
         } /* forward L */
 
             
         /* forward R until past m, count levels */
+#if RSBETWEEN_DEBUG
         log2( p, "Before moving R", cmp_l, cmp_r);
+#endif
         if (p->more_r)
             cmp_r= (*info->cmp)(p->buf_r, p->buf_m);
         else
             cmp_r=2; 
+#if RSBETWEEN_DEBUG
         log2( p, "after first R", cmp_l, cmp_r);
+#endif
         while (cmp_r < 0)   /* r before m */
         {
              /* -2, earlier record, don't count level */
@@ -436,7 +461,9 @@ static int r_read_between (RSFD rfd, void *buf, int *term_index)
             }
             else
                 cmp_r=2; 
+#if RSBETWEEN_DEBUG
         log2( p, "End of R loop", cmp_l, cmp_r);
+#endif
         } /* forward R */
         
         if ( ( p->level <= 0 ) && ! p->more_l)
@@ -446,16 +473,21 @@ static int r_read_between (RSFD rfd, void *buf, int *term_index)
         {
             memcpy (buf, p->buf_m, info->key_size);
             *term_index = p->term_index_m;
+#if RSBETWEEN_DEBUG
             log2( p, "Returning a hit (and forwarding m)", cmp_l, cmp_r);
+#endif
             p->more_m = rset_read (info->rset_m, p->rfd_m, p->buf_m,
                                    &p->term_index_m);
             if (cmp_l == 2)
                 p->level = 0;
+            p->hits++;
             return 1;
         }
         else if ( ! p->more_l )  /* not in data, no more starts */
         {
+#if RSBETWEEN_DEBUG
             log2( p, "no more starts, exiting without a hit", cmp_l, cmp_r);
+#endif
             return 0;  /* ergo, nothing can be found. stop scanning */
         }
 #if NEWCODE                
@@ -477,10 +509,14 @@ static int r_read_between (RSFD rfd, void *buf, int *term_index)
         p->more_m = rset_read (info->rset_m, p->rfd_m, p->buf_m,
                                &p->term_index_m);
 #endif
+#if RSBETWEEN_DEBUG
         log2( p, "End of M loop", cmp_l, cmp_r);
+#endif
     } /* while more_m */
     
+#if RSBETWEEN_DEBUG
     log2( p, "Exiting, nothing more in m", cmp_l, cmp_r);
+#endif
     return 0;  /* no more data possible */
 
 
@@ -493,3 +529,37 @@ static int r_write_between (RSFD rfd, const void *buf)
     return -1;
 }
 
+
+static void r_pos_between (RSFD rfd, double *current, double *total)
+{
+    struct rset_between_rfd *p = (struct rset_between_rfd *) rfd;
+    struct rset_between_info *info = p->info;
+    double lcur,ltot;
+    double mcur,mtot;
+    double rcur,rtot;
+    double r;
+    ltot=-1; rtot=-1;
+    rset_pos(info->rset_l, p->rfd_l,  &lcur, &ltot);
+    rset_pos(info->rset_m, p->rfd_m,  &mcur, &mtot);
+    rset_pos(info->rset_r, p->rfd_r,  &rcur, &rtot);
+    if ( (ltot<0) && (mtot<0) && (rtot<0) ) { /*no position */
+        *current=mcur;  /* return same as you got */
+        *total=mtot;    /* probably -1 for not available */
+    }
+    if ( ltot<0) { ltot=0; lcur=0;} /* if only one useful, use it */
+    if ( mtot<0) { mtot=0; mcur=0;}
+    if ( rtot<0) { rtot=0; rcur=0;}
+    if ( ltot+mtot+rtot < 1 ) { /* empty rset */
+        *current=0;
+        *total=0;
+        return;
+    }
+    r=1.0*(lcur+mcur+rcur)/(ltot+mtot+rtot); /* weighed average of l and r */
+    *current=p->hits;
+    *total=*current/r ; 
+#if RSBETWEEN_DEBUG
+    yaz_log(LOG_DEBUG,"betw_pos: (%s/%s) %0.1f/%0.1f= %0.4f ",
+                    info->rset_l->control->desc, info->rset_r->control->desc,
+                    *current, *total, r);
+#endif
+}