rset/rsbool.c

   1 /*
   2  * Copyright (C) 1994-1995, Index Data I/S
   3  * All rights reserved.
   4  * Sebastian Hammer, Adam Dickmeiss
   5  *
   6  * $Log: rsbool.c,v $
   7  * Revision 1.6  1995-10-06 14:38:05  adam
   8  * New result set method: r_score.
   9  * Local no (sysno) and score is transferred to retrieveCtrl.
  10  *
  11  * Revision 1.5  1995/09/08  14:52:41  adam
  12  * Work on relevance feedback.
  13  *
  14  * Revision 1.4  1995/09/08  08:54:04  adam
  15  * More efficient and operation.
  16  *
  17  * Revision 1.3  1995/09/07  13:58:43  adam
  18  * New parameter: result-set file descriptor (RSFD) to support multiple
  19  * positions within the same result-set.
  20  * Boolean operators: and, or, not implemented.
  21  *
  22  * Revision 1.2  1995/09/06  16:11:55  adam
  23  * More work on boolean sets.
  24  *
  25  * Revision 1.1  1995/09/06  13:27:15  adam
  26  * New set type: bool. Not finished yet.
  27  *
  28  */
  29
  30 #include <stdio.h>
  31 #include <assert.h>
  32
  33 #include <rsbool.h>
  34 #include <alexutil.h>
  35
  36 static rset_control *r_create(const struct rset_control *sel, void *parms);
  37 static RSFD r_open (rset_control *ct, int wflag);
  38 static void r_close (RSFD rfd);
  39 static void r_delete (rset_control *ct);
  40 static void r_rewind (RSFD rfd);
  41 static int r_count (rset_control *ct);
  42 static int r_read_and (RSFD rfd, void *buf);
  43 static int r_read_or (RSFD rfd, void *buf);
  44 static int r_read_not (RSFD rfd, void *buf);
  45 static int r_write (RSFD rfd, const void *buf);
  46 static int r_score (RSFD rfd, int *score);
  47
  48 static const rset_control control_and =
  49 {
  50     "AND set type",
  51     0,
  52     r_create,
  53     r_open,
  54     r_close,
  55     r_delete,
  56     r_rewind,
  57     r_count,
  58     r_read_and,
  59     r_write,
  60     r_score
  61 };
  62
  63 static const rset_control control_or =
  64 {
  65     "OR set type",
  66     0,
  67     r_create,
  68     r_open,
  69     r_close,
  70     r_delete,
  71     r_rewind,
  72     r_count,
  73     r_read_or,
  74     r_write,
  75     r_score
  76 };
  77
  78 static const rset_control control_not =
  79 {
  80     "NOT set type",
  81     0,
  82     r_create,
  83     r_open,
  84     r_close,
  85     r_delete,
  86     r_rewind,
  87     r_count,
  88     r_read_not,
  89     r_write,
  90     r_score
  91 };
  92
  93
  94 const rset_control *rset_kind_and = &control_and;
  95 const rset_control *rset_kind_or = &control_or;
  96 const rset_control *rset_kind_not = &control_not;
  97
  98 struct rset_bool_info {
  99     int key_size;
 100     RSET rset_l;
 101     RSET rset_r;
 102     int (*cmp)(const void *p1, const void *p2);
 103     struct rset_bool_rfd *rfd_list;
 104 };
 105
 106 struct rset_bool_rfd {
 107     RSFD rfd_l;
 108     RSFD rfd_r;
 109     int  more_l;
 110     int  more_r;
 111     void *buf_l;
 112     void *buf_r;
 113     struct rset_bool_rfd *next;
 114     struct rset_bool_info *info;
 115 };
 116
 117 static rset_control *r_create (const struct rset_control *sel, void *parms)
 118 {
 119     rset_control *newct;
 120     rset_bool_parms *bool_parms = parms;
 121     struct rset_bool_info *info;
 122
 123     newct = xmalloc(sizeof(*newct));
 124     memcpy (newct, sel, sizeof(*sel));
 125     newct->buf = xmalloc (sizeof(struct rset_bool_info));
 126     info = (struct rset_bool_info*) newct->buf;
 127     info->key_size = bool_parms->key_size;
 128     info->rset_l = bool_parms->rset_l;
 129     info->rset_r = bool_parms->rset_r;
 130     info->cmp = bool_parms->cmp;
 131     info->rfd_list = NULL;
 132     return newct;
 133 }
 134
 135 static RSFD r_open (rset_control *ct, int wflag)
 136 {
 137     struct rset_bool_info *info = ct->buf;
 138     struct rset_bool_rfd *rfd;
 139
 140     if (wflag)
 141     {
 142         logf (LOG_FATAL, "bool set type is read-only");
 143         return NULL;
 144     }
 145     rfd = xmalloc (sizeof(*rfd));
 146     rfd->next = info->rfd_list;
 147     info->rfd_list = rfd;
 148
 149     rfd->buf_l = xmalloc (info->key_size);
 150     rfd->buf_r = xmalloc (info->key_size);
 151     rfd->rfd_l = rset_open (info->rset_l, wflag);
 152     rfd->rfd_r = rset_open (info->rset_r, wflag);
 153     rfd->more_l = rset_read (info->rset_l, rfd->rfd_l, rfd->buf_l);
 154     rfd->more_r = rset_read (info->rset_r, rfd->rfd_r, rfd->buf_r);
 155     rfd->info = info;
 156     return rfd;
 157 }
 158
 159 static void r_close (RSFD rfd)
 160 {
 161     struct rset_bool_info *info = ((struct rset_bool_rfd*)rfd)->info;
 162     struct rset_bool_rfd **rfdp;
 163
 164     for (rfdp = &info->rfd_list; *rfdp; rfdp = &(*rfdp)->next)
 165         if (*rfdp == rfd)
 166         {
 167             xfree ((*rfdp)->buf_l);
 168             xfree ((*rfdp)->buf_r);
 169             rset_close (info->rset_l, (*rfdp)->rfd_l);
 170             rset_close (info->rset_r, (*rfdp)->rfd_r);
 171             *rfdp = (*rfdp)->next;
 172             free (rfd);
 173             return;
 174         }
 175     logf (LOG_FATAL, "r_close but no rfd match!");
 176     assert (0);
 177 }
 178
 179 static void r_delete (rset_control *ct)
 180 {
 181     struct rset_bool_info *info = ct->buf;
 182
 183     assert (info->rfd_list == NULL);
 184     xfree (info);
 185     xfree (ct);
 186 }
 187
 188 static void r_rewind (RSFD rfd)
 189 {
 190     struct rset_bool_info *info = ((struct rset_bool_rfd*)rfd)->info;
 191     struct rset_bool_rfd *p = rfd;
 192
 193     logf (LOG_DEBUG, "rsbool_rewind");
 194     rset_rewind (info->rset_l, p->rfd_l);
 195     rset_rewind (info->rset_r, p->rfd_r);
 196     p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l);
 197     p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r);
 198 }
 199
 200 static int r_count (rset_control *ct)
 201 {
 202     return 0;
 203 }
 204
 205 static int r_read_and (RSFD rfd, void *buf)
 206 {
 207     struct rset_bool_rfd *p = rfd;
 208     struct rset_bool_info *info = p->info;
 209
 210     while (p->more_l && p->more_r)
 211     {
 212         int cmp;
 213
 214         cmp = (*info->cmp)(p->buf_l, p->buf_r);
 215         if (!cmp)
 216         {
 217             memcpy (buf, p->buf_l, info->key_size);
 218             p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l);
 219             p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r);
 220             return 1;
 221         }
 222         else if (cmp == 1)
 223         {
 224             memcpy (buf, p->buf_r, info->key_size);
 225             p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r);
 226             return 1;
 227         }
 228         else if (cmp == -1)
 229         {
 230             memcpy (buf, p->buf_l, info->key_size);
 231             p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l);
 232             return 1;
 233         }
 234         else if (cmp > 1)
 235             p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r);
 236         else
 237             p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l);
 238     }
 239     return 0;
 240 }
 241
 242 static int r_read_or (RSFD rfd, void *buf)
 243 {
 244     struct rset_bool_rfd *p = rfd;
 245     struct rset_bool_info *info = p->info;
 246
 247     while (p->more_l || p->more_r)
 248     {
 249         int cmp;
 250
 251         if (p->more_l && p->more_r)
 252             cmp = (*info->cmp)(p->buf_l, p->buf_r);
 253         else if (p->more_r)
 254             cmp = 2;
 255         else
 256             cmp = -2;
 257         if (!cmp)
 258         {
 259             memcpy (buf, p->buf_l, info->key_size);
 260             p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l);
 261             p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r);
 262             return 1;
 263         }
 264         else if (cmp > 0)
 265         {
 266             memcpy (buf, p->buf_r, info->key_size);
 267             p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r);
 268             return 1;
 269         }
 270         else
 271         {
 272             memcpy (buf, p->buf_l, info->key_size);
 273             p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l);
 274             return 1;
 275         }
 276     }
 277     return 0;
 278 }
 279
 280 static int r_read_not (RSFD rfd, void *buf)
 281 {
 282     struct rset_bool_rfd *p = rfd;
 283     struct rset_bool_info *info = p->info;
 284
 285     while (p->more_l || p->more_r)
 286     {
 287         int cmp;
 288
 289         if (p->more_l && p->more_r)
 290             cmp = (*info->cmp)(p->buf_l, p->buf_r);
 291         else if (p->more_r)
 292             cmp = 2;
 293         else
 294             cmp = -2;
 295         if (cmp < -1)
 296         {
 297             memcpy (buf, p->buf_l, info->key_size);
 298             p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l);
 299             return 1;
 300         }
 301         else if (cmp > 1)
 302             p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r);
 303         else
 304         {
 305             memcpy (buf, p->buf_l, info->key_size);
 306             do
 307             {
 308                 p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l);
 309                 if (!p->more_l)
 310                     break;
 311                 cmp = (*info->cmp)(p->buf_l, buf);
 312             } while (cmp >= -1 && cmp <= 1);
 313             do
 314             {
 315                 p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r);
 316                 if (!p->more_r)
 317                     break;
 318                 cmp = (*info->cmp)(p->buf_r, buf);
 319             } while (cmp >= -1 && cmp <= 1);
 320         }
 321     }
 322     return 0;
 323 }
 324
 325
 326 static int r_write (RSFD rfd, const void *buf)
 327 {
 328     logf (LOG_FATAL, "bool set type is read-only");
 329     return -1;
 330 }
 331
 332 static int r_score (RSFD rfd, int *score)
 333 {
 334     *score = -1;
 335     return -1;
 336 }
 337