X-Git-Url: http://git.indexdata.com/?a=blobdiff_plain;f=rset%2Frsbool.c;h=39f9ec13cb177a59d98927ff9321ec777fcf3f44;hb=d195e3229385349f842d2f82b65ad8118d64aec5;hp=c4aee9c165f72ae5f13e17ca00c898ddd0bb8a82;hpb=1d0ee69119ae254a033b5c9e25aad14b3e627f9e;p=idzebra-moved-to-github.git

diff --git a/rset/rsbool.c b/rset/rsbool.c
index c4aee9c..39f9ec1 100644
--- a/rset/rsbool.c
+++ b/rset/rsbool.c
@@ -1,101 +1,527 @@
-/*
- * Copyright (C) 1994-1995, Index Data I/S 
- * All rights reserved.
- * Sebastian Hammer, Adam Dickmeiss
- *
- * $Log: rsbool.c,v $
- * Revision 1.1  1995-09-06 13:27:15  adam
- * New set type: bool. Not finished yet.
- *
- */
+/* $Id: rsbool.c,v 1.51 2004-11-19 10:27:14 heikki Exp $
+   Copyright (C) 1995,1996,1997,1998,1999,2000,2001,2002,2003,2004
+   Index Data Aps
+
+This file is part of the Zebra server.
+
+Zebra is free software; you can redistribute it and/or modify it under
+the terms of the GNU General Public License as published by the Free
+Software Foundation; either version 2, or (at your option) any later
+version.
+
+Zebra is distributed in the hope that it will be useful, but WITHOUT ANY
+WARRANTY; without even the implied warranty of MERCHANTABILITY or
+FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+for more details.
+
+You should have received a copy of the GNU General Public License
+along with Zebra; see the file LICENSE.zebra.  If not, write to the
+Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
+02111-1307, USA.
+*/
 
 #include <stdio.h>
-#include <rsbool.h>
-#include <alexutil.h>
-
-static rset_control *r_create(const struct rset_control *sel, void *parms);
-static int r_open (rset_control *ct, int wflag);
-static void r_close (rset_control *ct);
-static void r_delete (rset_control *ct);
-static void r_rewind (rset_control *ct);
-static int r_count (rset_control *ct);
-static int r_read (rset_control *ct, void *buf);
-static int r_write (rset_control *ct, const void *buf);
-
-static const rset_control control = 
-{
-    "BOOL set type",
-    0,
-    r_create,
+#include <stdlib.h>
+#include <string.h>
+#include <assert.h>
+
+#include <zebrautl.h>
+#include <rset.h>
+
+#ifndef RSET_DEBUG
+#define RSET_DEBUG 0
+#endif
+
+static RSFD r_open (RSET ct, int flag);
+static void r_close (RSFD rfd);
+static void r_delete (RSET ct);
+static int r_forward(RSFD rfd, void *buf, TERMID *term, const void *untilbuf);
+static void r_pos (RSFD rfd, double *current, double *total); 
+static int r_read_and (RSFD rfd, void *buf, TERMID *term);
+static int r_read_or (RSFD rfd, void *buf, TERMID *term);
+static int r_read_not (RSFD rfd, void *buf, TERMID *term);
+static int r_write (RSFD rfd, const void *buf);
+static void r_get_terms(RSET ct, TERMID *terms, int maxterms, int *curterm);
+
+
+static const struct rset_control control_and = 
+{
+    "and",
+    r_delete,
+    r_get_terms,
     r_open,
     r_close,
+    r_forward, 
+    r_pos,    
+    r_read_and,
+    r_write,
+};
+
+static const struct rset_control control_or = 
+{
+    "or",
+    r_delete,
+    r_get_terms,
+    r_open,
+    r_close,
+    r_forward, 
+    r_pos,
+    r_read_or,
+    r_write,
+};
+
+static const struct rset_control control_not = 
+{
+    "not",
     r_delete,
-    r_rewind,
-    r_count,
-    r_read,
-    r_write
+    r_get_terms,
+    r_open,
+    r_close,
+    r_forward, 
+    r_pos,
+    r_read_not,
+    r_write,
 };
 
-const rset_control *rset_kind_bool = &control;
+
+const struct rset_control *rset_kind_and = &control_and;
+const struct rset_control *rset_kind_or  = &control_or;
+const struct rset_control *rset_kind_not = &control_not;
 
 struct rset_bool_info {
-    int key_size;
-    int op;
+    RSET rset_l;
+    RSET rset_r;
 };
 
-static rset_control *r_create(const struct rset_control *sel, void *parms)
+struct rset_bool_rfd {
+    zint hits;
+    RSFD rfd_l;
+    RSFD rfd_r;
+    int  more_l;
+    int  more_r;
+    void *buf_l;
+    void *buf_r;
+    TERMID term_l;
+    TERMID term_r;
+    int tail;
+};    
+
+static RSET rsbool_create_base( const struct rset_control *ctrl,
+            NMEM nmem, const struct key_control *kcontrol,
+            int scope, 
+            RSET rset_l, RSET rset_r)
 {
-    rset_control *newct;
-    rset_bool_parms *bool_parms = parms;
+    RSET rnew=rset_create_base(ctrl, nmem, kcontrol, scope,0);
     struct rset_bool_info *info;
+    info = (struct rset_bool_info *) nmem_malloc(rnew->nmem,sizeof(*info));
+    info->rset_l = rset_l;
+    info->rset_r = rset_r;
+    rnew->priv=info;
+    return rnew;
+}
+
 
-    logf (LOG_DEBUG, "rsbool_create(%s)", sel->desc);
-    newct = xmalloc(sizeof(*newct));
-    memcpy(newct, sel, sizeof(*sel));
-    newct->buf = xmalloc (sizeof(struct rset_bool_info));
-    info = (struct rset_bool_info*) newct->buf;
-    info->key_size = bool_parms->key_size;
-    info->op = bool_parms->op;
-    return newct;
+RSET rsbool_create_and( NMEM nmem, const struct key_control *kcontrol,
+                        int scope, 
+                        RSET rset_l, RSET rset_r)
+{
+    return rsbool_create_base(rset_kind_and, nmem, kcontrol,
+                              scope,
+                              rset_l, rset_r);
 }
 
-static int r_open(rset_control *ct, int wflag)
+RSET rsbool_create_or( NMEM nmem, const struct key_control *kcontrol,
+                        int scope, 
+                       RSET rset_l, RSET rset_r)
+{
+    return rsbool_create_base(rset_kind_or, nmem, kcontrol,
+                              scope,
+                              rset_l, rset_r);
+}
+
+RSET rsbool_create_not( NMEM nmem, const struct key_control *kcontrol,
+                        int scope, 
+                        RSET rset_l, RSET rset_r)
+{
+    return rsbool_create_base(rset_kind_not, nmem, kcontrol,
+                              scope, 
+                              rset_l, rset_r);
+}
+
+static void r_delete (RSET ct)
+{
+    struct rset_bool_info *info = (struct rset_bool_info *) ct->priv;
+    rset_delete (info->rset_l);
+    rset_delete (info->rset_r);
+}
+
+
+static RSFD r_open (RSET ct, int flag)
 {
-    if (wflag)
+    struct rset_bool_info *info = (struct rset_bool_info *) ct->priv;
+    RSFD rfd;
+    struct rset_bool_rfd *p;
+   
+
+    if (flag & RSETF_WRITE)
     {
-	logf (LOG_FATAL, "bool set type is read-only");
-	return -1;
+        yaz_log (YLOG_FATAL, "bool set type is read-only");
+        return NULL;
     }
-    return 0;
+    rfd = rfd_create_base(ct);
+    if (rfd->priv)
+        p=(struct rset_bool_rfd *)rfd->priv;
+    else {
+        p=nmem_malloc(ct->nmem,sizeof(*p));
+        rfd->priv=p;
+        p->buf_l = nmem_malloc(ct->nmem, ct->keycontrol->key_size);
+        p->buf_r = nmem_malloc(ct->nmem, ct->keycontrol->key_size);
+    }
+
+    yaz_log(YLOG_DEBUG,"rsbool (%s) open [%p]", ct->control->desc, rfd);
+    p->hits=0;
+
+    p->rfd_l = rset_open (info->rset_l, RSETF_READ);
+    p->rfd_r = rset_open (info->rset_r, RSETF_READ);
+    p->more_l = rset_read (p->rfd_l, p->buf_l, &p->term_l);
+    p->more_r = rset_read (p->rfd_r, p->buf_r, &p->term_r);
+    p->tail = 0;
+    return rfd;
 }
 
-static void r_close(rset_control *ct)
+static void r_close (RSFD rfd)
 {
-    /* NOP */
+ /* struct rset_bool_info *info = (struct rset_bool_info*)(rfd->rset->priv); */
+    struct rset_bool_rfd *prfd=(struct rset_bool_rfd *)rfd->priv;
+
+    rset_close (prfd->rfd_l);
+    rset_close (prfd->rfd_r);
+    rfd_delete_base(rfd);
 }
 
-static void r_delete(rset_control *ct)
+
+
+static int r_forward (RSFD rfd, void *buf, TERMID *term,
+                     const void *untilbuf)
 {
-    xfree(ct);
+    struct rset_bool_rfd *p=(struct rset_bool_rfd *)rfd->priv;
+    const struct key_control *kctrl=rfd->rset->keycontrol;
+
+    if ( p->more_l && ((kctrl->cmp)(untilbuf,p->buf_l)>=rfd->rset->scope) )
+        p->more_l = rset_forward(p->rfd_l, p->buf_l, &p->term_l, untilbuf);
+    if ( p->more_r && ((kctrl->cmp)(untilbuf,p->buf_r)>=rfd->rset->scope))
+        p->more_r = rset_forward(p->rfd_r, p->buf_r, &p->term_r, untilbuf);
+    p->tail=0; 
+    return rset_read(rfd,buf,term); 
 }
 
-static void r_rewind(rset_control *ct)
+
+/*
+    1,1         1,3
+    1,9         2,1
+    1,11        3,1
+    2,9
+
+  1,1     1,1
+  1,3     1,3
+          1,9
+          1,11
+  2,1     2,1
+          2,9
+          3,1
+*/
+
+static int r_read_and (RSFD rfd, void *buf, TERMID *term)
 {
-    logf (LOG_DEBUG, "rsbool_rewind");
+    struct rset_bool_rfd *p=(struct rset_bool_rfd *)rfd->priv;
+    const struct key_control *kctrl=rfd->rset->keycontrol;
+
+    while (p->more_l || p->more_r)
+    {
+        int cmp;
+
+        if (p->more_l && p->more_r)
+            cmp = (*kctrl->cmp)(p->buf_l, p->buf_r);
+        else if (p->more_l)
+            cmp = -rfd->rset->scope;
+        else
+            cmp = rfd->rset->scope;
+#if RSET_DEBUG
+        yaz_log (YLOG_DEBUG, "r_read_and [%p] looping: m=%d/%d c=%d t=%d",
+                        rfd, p->more_l, p->more_r, cmp, p->tail);
+        (*kctrl->log_item)(YLOG_DEBUG, p->buf_l, "left ");
+        (*kctrl->log_item)(YLOG_DEBUG, p->buf_r, "right ");
+#endif
+        if (!cmp)
+        {  /* cmp==0 */
+            memcpy (buf, p->buf_l, kctrl->key_size);
+            if (term)
+                *term=p->term_l;
+            p->more_l = rset_read (p->rfd_l, p->buf_l, &p->term_l);
+            p->tail = 1;
+        }
+        else if ( (cmp>0) && (cmp<rfd->rset->scope))
+        {  /* typically cmp == 1 */
+            memcpy (buf, p->buf_r, kctrl->key_size);
+            if (term)
+                *term=p->term_r;
+            p->more_r = rset_read (p->rfd_r, p->buf_r, &p->term_r);
+            p->tail = 1;
+#if RSET_DEBUG
+            yaz_log (YLOG_DEBUG, "r_read_and [%p] returning R m=%d/%d c=%d",
+                    rfd, p->more_l, p->more_r, cmp);
+            key_logdump(YLOG_DEBUG,buf);
+            (*kctrl->log_item)(YLOG_DEBUG, buf, "");
+#endif
+            p->hits++;
+            return 1;
+        }
+        else if ( (cmp<0) && (-cmp<rfd->rset->scope))
+        {  /* cmp == -1 */
+            memcpy (buf, p->buf_l, kctrl->key_size);
+            if (term)
+                *term=p->term_l;
+            p->more_l = rset_read (p->rfd_l, p->buf_l,&p->term_l);
+            p->tail = 1;
+#if RSET_DEBUG
+            yaz_log (YLOG_DEBUG, "r_read_and [%p] returning L m=%d/%d c=%d",
+                    rfd, p->more_l, p->more_r, cmp);
+            (*kctrl->log_item)(YLOG_DEBUG, buf, "");
+#endif
+            p->hits++;
+            return 1;
+        }
+        else if (cmp >= rfd->rset->scope )  
+        {  /* cmp == 2 */
+            if (p->tail)
+            {
+                memcpy (buf, p->buf_r, kctrl->key_size);
+                if (term)
+                    *term=p->term_r;
+                p->more_r = rset_read (p->rfd_r, p->buf_r, &p->term_r);
+                if (!p->more_r || (*kctrl->cmp)(p->buf_r, buf) > 1)
+                    p->tail = 0;
+#if RSET_DEBUG
+                yaz_log (YLOG_DEBUG, "r_read_and [%p] returning R tail m=%d/%d c=%d",
+                        rfd, p->more_l, p->more_r, cmp);
+                (*kctrl->log_item)(YLOG_DEBUG, buf, "");
+#endif
+                p->hits++;
+                return 1;
+            }
+            else
+            {
+#if RSET_DEBUG
+                yaz_log (YLOG_DEBUG, "r_read_and [%p] about to forward R "
+                                 "m=%d/%d c=%d",
+                        rfd, p->more_l, p->more_r, cmp);
+#endif
+                if (p->more_r && p->more_l)
+                    p->more_r = rset_forward( p->rfd_r, p->buf_r,
+                             &p->term_r, p->buf_l);
+                else 
+                    return 0; /* no point in reading further */
+            }
+        }
+        else  
+        { /* cmp == -2 */
+            if (p->tail)
+            {
+                memcpy (buf, p->buf_l, kctrl->key_size);
+                if (term)
+                    *term=p->term_l;
+                p->more_l = rset_read (p->rfd_l, p->buf_l, &p->term_l);
+                if (!p->more_l || (*kctrl->cmp)(p->buf_l, buf) > 1)
+                    p->tail = 0;
+#if RSET_DEBUG
+                yaz_log (YLOG_DEBUG, "r_read_and [%p] returning L tail m=%d/%d c=%d",
+                        rfd, p->more_l, p->more_r, cmp);
+                (*kctrl->log_item)(YLOG_DEBUG, buf, "");
+#endif
+                p->hits++;
+                return 1;
+            }
+            else
+            {
+#if RSET_DEBUG
+                yaz_log (YLOG_DEBUG, "r_read_and [%p] about to forward L "
+                                 "m=%d/%d c=%d",
+                        rfd, p->more_l, p->more_r, cmp);
+#endif
+                if (p->more_r && p->more_l)
+                    p->more_l = rset_forward(p->rfd_l, p->buf_l, 
+                                 &p->term_l, p->buf_r);
+                else 
+                    return 0; /* no point in reading further */
+            }
+        }
+    }
+#if RSET_DEBUG
+    yaz_log (YLOG_DEBUG, "r_read_and [%p] reached its end",rfd);
+#endif
+    return 0;
 }
 
-static int r_count (rset_control *ct)
+static int r_read_or (RSFD rfd, void *buf, TERMID *term)
 {
+    struct rset_bool_rfd *p=(struct rset_bool_rfd *)rfd->priv;
+    const struct key_control *kctrl=rfd->rset->keycontrol;
+
+    while (p->more_l || p->more_r)
+    {
+        int cmp;
+
+        if (p->more_l && p->more_r)
+            cmp = (*kctrl->cmp)(p->buf_l, p->buf_r);
+        else if (p->more_r)
+            cmp = rfd->rset->scope;
+        else
+            cmp = -rfd->rset->scope;
+        if (!cmp)
+        { /* cmp==0 */
+            memcpy (buf, p->buf_l, kctrl->key_size);
+            if (term)
+                *term=p->term_l;
+            p->more_l = rset_read (p->rfd_l, p->buf_l, &p->term_l);
+            /* FIXME - is this right, should we not leave _r as it is */
+            /* and return that in the next read, so that ranking etc */
+            /* get to see both? */
+            p->more_r = rset_read (p->rfd_r, p->buf_r, &p->term_r);
+#if RSET_DEBUG
+            yaz_log (YLOG_DEBUG, "r_read_or returning A m=%d/%d c=%d",
+                    p->more_l, p->more_r, cmp);
+            (*kctrl->log_item)(YLOG_DEBUG, buf, "");
+#endif
+            p->hits++;
+            return 1;
+        }
+        else if (cmp > 0)
+        {
+            memcpy (buf, p->buf_r, kctrl->key_size);
+            if (term)
+                *term=p->term_r;
+            p->more_r = rset_read (p->rfd_r, p->buf_r, &p->term_r);
+#if RSET_DEBUG
+            yaz_log (YLOG_DEBUG, "r_read_or returning B m=%d/%d c=%d",
+                    p->more_l, p->more_r, cmp);
+            (*kctrl->log_item)(YLOG_DEBUG, buf, "");
+#endif
+            p->hits++;
+            return 1;
+        }
+        else
+        {
+            memcpy (buf, p->buf_l, kctrl->key_size);
+            if (term)
+                *term=p->term_l;
+            p->more_l = rset_read ( p->rfd_l, p->buf_l, &p->term_l);
+#if RSET_DEBUG
+            yaz_log (YLOG_DEBUG, "r_read_or returning C m=%d/%d c=%d",
+                    p->more_l, p->more_r, cmp);
+            (*kctrl->log_item)(YLOG_DEBUG, buf, "");
+#endif
+            p->hits++;
+            return 1;
+        }
+    }
     return 0;
 }
 
-static int r_read (rset_control *ct, void *buf)
+static int r_read_not (RSFD rfd, void *buf, TERMID *term)
 {
+    struct rset_bool_rfd *p=(struct rset_bool_rfd *)rfd->priv;
+    const struct key_control *kctrl=rfd->rset->keycontrol;
+
+    while (p->more_l || p->more_r)
+    {
+        int cmp;
+
+        if (p->more_l && p->more_r)
+            cmp = (*kctrl->cmp)(p->buf_l, p->buf_r);
+        else if (p->more_r)
+            cmp = rfd->rset->scope;
+        else
+            cmp = -rfd->rset->scope;
+
+        if (cmp <= -rfd->rset->scope)
+        { /* cmp == -2 */
+            memcpy (buf, p->buf_l, kctrl->key_size);
+            if (term)
+                *term=p->term_l;
+            p->more_l = rset_read (p->rfd_l, p->buf_l, &p->term_l);
+            p->hits++;
+            return 1;
+        }
+        else if (cmp >= rfd->rset->scope)   /* cmp >1 */
+            p->more_r = rset_forward( p->rfd_r, p->buf_r, 
+                          &p->term_r, p->buf_l);
+        else
+        { /* cmp== -1, 0, or 1 */
+            memcpy (buf, p->buf_l, kctrl->key_size);
+            if (term)
+                *term=p->term_l;
+            do
+            { 
+                p->more_l = rset_read (p->rfd_l, p->buf_l, &p->term_l);
+                if (!p->more_l)
+                    break;
+                cmp = (*kctrl->cmp)(p->buf_l, buf);
+            } while (abs(cmp)<rfd->rset->scope);
+                /*  (cmp >= -1 && cmp <= 1) */
+            do
+            {
+                p->more_r = rset_read (p->rfd_r, p->buf_r, &p->term_r);
+                if (!p->more_r)
+                    break;
+                cmp = (*kctrl->cmp)(p->buf_r, buf);
+            } while (abs(cmp)<rfd->rset->scope);
+               /* (cmp >= -1 && cmp <= 1) */
+        }
+    }
     return 0;
 }
 
-static int r_write (rset_control *ct, const void *buf)
+
+static int r_write (RSFD rfd, const void *buf)
 {
-    logf (LOG_FATAL, "bool set type is read-only");
+    yaz_log (YLOG_FATAL, "bool set type is read-only");
     return -1;
 }
+
+static void r_pos (RSFD rfd, double *current, double *total)
+{
+    struct rset_bool_rfd *p=(struct rset_bool_rfd *)rfd->priv;
+    double lcur,ltot;
+    double rcur,rtot;
+    double r;
+    ltot=-1; rtot=-1;
+    rset_pos(p->rfd_l,  &lcur, &ltot);
+    rset_pos(p->rfd_r,  &rcur, &rtot);
+    if ( (rtot<0) && (ltot<0)) { /*no position */
+        *current=rcur;  /* return same as you got */
+        *total=rtot;    /* probably -1 for not available */
+    }
+    if ( rtot<0) { rtot=0; rcur=0;} /* if only one useful, use it */
+    if ( ltot<0) { ltot=0; lcur=0;}
+    if ( rtot+ltot < 1 ) { /* empty rset */
+        *current=0;
+        *total=0;
+        return;
+    }
+    r=1.0*(lcur+rcur)/(ltot+rtot); /* weighed average of l and r */
+    *current=(double) (p->hits);
+    *total=*current/r ; 
+#if RSET_DEBUG
+    yaz_log(YLOG_DEBUG,"bool_pos: (%s/%s) %0.1f/%0.1f= %0.4f ",
+                    info->rset_l->control->desc, info->rset_r->control->desc,
+                    *current, *total, r);
+#endif
+}
+
+static void r_get_terms(RSET ct, TERMID *terms, int maxterms, int *curterm)
+{
+    struct rset_bool_info *info = (struct rset_bool_info *) ct->priv;
+    rset_getterms(info->rset_l, terms, maxterms, curterm);
+    rset_getterms(info->rset_r, terms, maxterms, curterm);
+}
+