1 /* $Id: rsbool.c,v 1.44 2004-08-26 11:11:59 heikki Exp $
2 Copyright (C) 1995,1996,1997,1998,1999,2000,2001,2002,2003,2004
5 This file is part of the Zebra server.
7 Zebra is free software; you can redistribute it and/or modify it under
8 the terms of the GNU General Public License as published by the Free
9 Software Foundation; either version 2, or (at your option) any later
12 Zebra is distributed in the hope that it will be useful, but WITHOUT ANY
13 WARRANTY; without even the implied warranty of MERCHANTABILITY or
14 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
17 You should have received a copy of the GNU General Public License
18 along with Zebra; see the file LICENSE.zebra. If not, write to the
19 Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
36 static RSFD r_open (RSET ct, int flag);
37 static void r_close (RSFD rfd);
38 static void r_delete (RSET ct);
39 static void r_rewind (RSFD rfd);
40 static int r_forward(RSET ct, RSFD rfd, void *buf,
41 int (*cmpfunc)(const void *p1, const void *p2),
42 const void *untilbuf);
43 static void r_pos (RSFD rfd, double *current, double *total);
44 static int r_read_and (RSFD rfd, void *buf);
45 static int r_read_or (RSFD rfd, void *buf);
46 static int r_read_not (RSFD rfd, void *buf);
47 static int r_write (RSFD rfd, const void *buf);
49 static const struct rset_control control_and =
62 static const struct rset_control control_or =
75 static const struct rset_control control_not =
89 const struct rset_control *rset_kind_and = &control_and;
90 const struct rset_control *rset_kind_or = &control_or;
91 const struct rset_control *rset_kind_not = &control_not;
93 struct rset_bool_info {
97 int (*cmp)(const void *p1, const void *p2);
98 void (*log_item)(int logmask, const void *p, const char *txt);
99 struct rset_bool_rfd *rfd_list; /* RSFDs in use */
100 struct rset_bool_rfd *free_list; /* RSFDs that are freed (for reuse) */
103 struct rset_bool_rfd {
112 struct rset_bool_rfd *next;
113 struct rset_bool_info *info;
116 static RSET rsbool_create_base( const struct rset_control *ctrl,
117 NMEM nmem, int key_size,
118 int (*cmp)(const void *p1, const void *p2),
119 RSET rset_l, RSET rset_r,
120 void (*log_item)(int logmask, const void *p, const char *txt) )
122 RSET rnew=rset_create_base(ctrl, nmem);
123 struct rset_bool_info *info;
124 info = (struct rset_bool_info *) nmem_malloc(rnew->nmem,sizeof(*info));
125 info->key_size = key_size;
126 info->rset_l = rset_l;
127 info->rset_r = rset_r;
129 info->log_item = log_item;
130 info->rfd_list = NULL;
131 info->free_list = NULL;
138 RSET rsbool_create_and( NMEM nmem, int key_size,
139 int (*cmp)(const void *p1, const void *p2),
140 RSET rset_l, RSET rset_r,
141 void (*log_item)(int logmask, const void *p, const char *txt) )
143 return rsbool_create_base(rset_kind_and, nmem, key_size, cmp,
144 rset_l, rset_r, log_item);
147 RSET rsbool_create_or( NMEM nmem, int key_size,
148 int (*cmp)(const void *p1, const void *p2),
149 RSET rset_l, RSET rset_r,
150 void (*log_item)(int logmask, const void *p, const char *txt) )
152 return rsbool_create_base(rset_kind_or, nmem, key_size, cmp,
153 rset_l, rset_r, log_item);
156 RSET rsbool_create_not( NMEM nmem, int key_size,
157 int (*cmp)(const void *p1, const void *p2),
158 RSET rset_l, RSET rset_r,
159 void (*log_item)(int logmask, const void *p, const char *txt) )
161 return rsbool_create_base(rset_kind_not, nmem, key_size, cmp,
162 rset_l, rset_r, log_item);
165 static void r_delete (RSET ct)
167 struct rset_bool_info *info = (struct rset_bool_info *) ct->priv;
169 assert (info->rfd_list == NULL);
170 rset_delete (info->rset_l);
171 rset_delete (info->rset_r);
175 static void *r_create (RSET ct, const struct rset_control *sel, void *parms)
177 rset_bool_parms *bool_parms = (rset_bool_parms *) parms;
178 struct rset_bool_info *info;
180 info = (struct rset_bool_info *) xmalloc (sizeof(*info));
181 info->key_size = bool_parms->key_size;
182 info->rset_l = bool_parms->rset_l;
183 info->rset_r = bool_parms->rset_r;
184 info->cmp = bool_parms->cmp;
185 info->log_item = bool_parms->log_item;
186 info->rfd_list = NULL;
192 static RSFD r_open (RSET ct, int flag)
194 struct rset_bool_info *info = (struct rset_bool_info *) ct->priv;
195 struct rset_bool_rfd *rfd;
197 if (flag & RSETF_WRITE)
199 logf (LOG_FATAL, "bool set type is read-only");
202 rfd = info->free_list;
204 info->free_list = rfd->next;
206 rfd = (struct rset_bool_rfd *) nmem_malloc(ct->nmem, sizeof(*rfd));
207 rfd->buf_l = nmem_malloc(ct->nmem, info->key_size);
208 rfd->buf_r = nmem_malloc(ct->nmem, info->key_size);
211 logf(LOG_DEBUG,"rsbool (%s) open [%p]", ct->control->desc, rfd);
212 rfd->next = info->rfd_list;
213 info->rfd_list = rfd;
217 rfd->rfd_l = rset_open (info->rset_l, RSETF_READ);
218 rfd->rfd_r = rset_open (info->rset_r, RSETF_READ);
219 rfd->more_l = rset_read (info->rset_l, rfd->rfd_l, rfd->buf_l);
220 rfd->more_r = rset_read (info->rset_r, rfd->rfd_r, rfd->buf_r);
225 static void r_close (RSFD rfd)
227 struct rset_bool_info *info = ((struct rset_bool_rfd*)rfd)->info;
228 struct rset_bool_rfd **rfdp;
230 for (rfdp = &info->rfd_list; *rfdp; rfdp = &(*rfdp)->next)
233 struct rset_bool_rfd *rfd_tmp = *rfdp;
235 rset_close (info->rset_l, (*rfdp)->rfd_l);
236 rset_close (info->rset_r, (*rfdp)->rfd_r);
237 *rfdp = (*rfdp)->next;
239 rfd_tmp->next = info->free_list;
240 info->free_list = rfd_tmp;
244 logf (LOG_FATAL, "r_close but no rfd match!");
249 static void r_rewind (RSFD rfd)
251 struct rset_bool_info *info = ((struct rset_bool_rfd*)rfd)->info;
252 struct rset_bool_rfd *p = (struct rset_bool_rfd *) rfd;
254 logf (LOG_DEBUG, "rsbool_rewind");
255 rset_rewind (info->rset_l, p->rfd_l);
256 rset_rewind (info->rset_r, p->rfd_r);
257 p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l);
258 p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r);
262 static int r_forward (RSET ct, RSFD rfd, void *buf,
263 int (*cmpfunc)(const void *p1, const void *p2),
264 const void *untilbuf)
266 struct rset_bool_info *info = ((struct rset_bool_rfd*)rfd)->info;
267 struct rset_bool_rfd *p = (struct rset_bool_rfd *) rfd;
271 logf (LOG_DEBUG, "rsbool_forward (L) [%p] '%s' (ct=%p rfd=%p m=%d,%d)",
272 rfd, ct->control->desc, ct, rfd, p->more_l, p->more_r);
274 if ( p->more_l && ((cmpfunc)(untilbuf,p->buf_l)==2) )
275 p->more_l = rset_forward(info->rset_l, p->rfd_l, p->buf_l,
276 info->cmp, untilbuf);
278 logf (LOG_DEBUG, "rsbool_forward (R) [%p] '%s' (ct=%p rfd=%p m=%d,%d)",
279 rfd, ct->control->desc, ct, rfd, p->more_l, p->more_r);
281 if ( p->more_r && ((cmpfunc)(untilbuf,p->buf_r)==2))
282 p->more_r = rset_forward(info->rset_r, p->rfd_r, p->buf_r,
283 info->cmp, untilbuf);
285 logf (LOG_DEBUG, "rsbool_forward [%p] calling read, m=%d,%d t=%d",
286 rfd, p->more_l, p->more_r, p->tail);
290 rc = rset_read(ct,rfd,buf);
292 logf (LOG_DEBUG, "rsbool_forward returning [%p] %d m=%d,%d",
293 rfd, rc, p->more_l, p->more_r);
314 static int r_read_and (RSFD rfd, void *buf)
316 struct rset_bool_rfd *p = (struct rset_bool_rfd *) rfd;
317 struct rset_bool_info *info = p->info;
319 while (p->more_l || p->more_r)
323 if (p->more_l && p->more_r)
324 cmp = (*info->cmp)(p->buf_l, p->buf_r);
330 logf (LOG_DEBUG, "r_read_and [%p] looping: m=%d/%d c=%d t=%d",
331 rfd, p->more_l, p->more_r, cmp, p->tail);
332 (*info->log_item)(LOG_DEBUG, p->buf_l, "left ");
333 (*info->log_item)(LOG_DEBUG, p->buf_r, "right ");
337 memcpy (buf, p->buf_l, info->key_size);
338 p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l);
343 memcpy (buf, p->buf_r, info->key_size);
344 p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r);
347 logf (LOG_DEBUG, "r_read_and [%p] returning R m=%d/%d c=%d",
348 rfd, p->more_l, p->more_r, cmp);
349 key_logdump(LOG_DEBUG,buf);
350 (*info->log_item)(LOG_DEBUG, buf, "");
357 memcpy (buf, p->buf_l, info->key_size);
358 p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l);
361 logf (LOG_DEBUG, "r_read_and [%p] returning L m=%d/%d c=%d",
362 rfd, p->more_l, p->more_r, cmp);
363 (*info->log_item)(LOG_DEBUG, buf, "");
368 else if (cmp > 1) /* cmp == 2 */
372 memcpy (buf, p->buf_r, info->key_size);
374 p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r);
377 if (!p->more_r || (*info->cmp)(p->buf_r, buf) > 1)
380 logf (LOG_DEBUG, "r_read_and returning C m=%d/%d c=%d",
381 p->more_l, p->more_r, cmp);
382 (*info->log_item)(LOG_DEBUG, buf, "");
391 memcpy (buf, p->buf_r, info->key_size);
392 p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r);
393 if (!p->more_r || (*info->cmp)(p->buf_r, buf) > 1)
396 logf (LOG_DEBUG, "r_read_and [%p] returning R tail m=%d/%d c=%d",
397 rfd, p->more_l, p->more_r, cmp);
398 (*info->log_item)(LOG_DEBUG, buf, "");
406 logf (LOG_DEBUG, "r_read_and [%p] about to forward R m=%d/%d c=%d",
407 rfd, p->more_l, p->more_r, cmp);
409 if (p->more_r && p->more_l)
410 p->more_r = rset_forward( info->rset_r, p->rfd_r,
411 p->buf_r, (info->cmp), p->buf_l);
413 return 0; /* no point in reading further */
420 memcpy (buf, p->buf_l, info->key_size);
421 p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l);
424 if (!p->more_l || (*info->cmp)(p->buf_l, buf) > 1)
427 logf (LOG_DEBUG, "r_read_and [%p] returning R tail m=%d/%d c=%d",
428 rfd, p->more_l, p->more_r, cmp);
429 (*info->log_item)(LOG_DEBUG, buf, "");
437 memcpy (buf, p->buf_l, info->key_size);
438 p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l);
439 if (!p->more_l || (*info->cmp)(p->buf_l, buf) > 1)
442 logf (LOG_DEBUG, "r_read_and [%p] returning L tail m=%d/%d c=%d",
443 rfd, p->more_l, p->more_r, cmp);
444 (*info->log_item)(LOG_DEBUG, buf, "");
452 logf (LOG_DEBUG, "r_read_and [%p] about to forward L m=%d/%d c=%d",
453 rfd, p->more_l, p->more_r, cmp);
455 if (p->more_r && p->more_l)
456 p->more_l = rset_forward(
457 info->rset_l, p->rfd_l,
458 p->buf_l, (info->cmp), p->buf_r);
460 return 0; /* no point in reading further */
466 logf (LOG_DEBUG, "r_read_and [%p] reached its end",rfd);
471 static int r_read_or (RSFD rfd, void *buf)
473 struct rset_bool_rfd *p = (struct rset_bool_rfd *) rfd;
474 struct rset_bool_info *info = p->info;
476 while (p->more_l || p->more_r)
480 if (p->more_l && p->more_r)
481 cmp = (*info->cmp)(p->buf_l, p->buf_r);
488 memcpy (buf, p->buf_l, info->key_size);
489 p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l);
490 p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r);
492 logf (LOG_DEBUG, "r_read_or returning A m=%d/%d c=%d",
493 p->more_l, p->more_r, cmp);
494 (*info->log_item)(LOG_DEBUG, buf, "");
501 memcpy (buf, p->buf_r, info->key_size);
502 p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r);
504 logf (LOG_DEBUG, "r_read_or returning B m=%d/%d c=%d",
505 p->more_l, p->more_r, cmp);
506 (*info->log_item)(LOG_DEBUG, buf, "");
513 memcpy (buf, p->buf_l, info->key_size);
514 p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l);
516 logf (LOG_DEBUG, "r_read_or returning C m=%d/%d c=%d",
517 p->more_l, p->more_r, cmp);
518 (*info->log_item)(LOG_DEBUG, buf, "");
527 static int r_read_not (RSFD rfd, void *buf)
529 struct rset_bool_rfd *p = (struct rset_bool_rfd *) rfd;
530 struct rset_bool_info *info = p->info;
532 while (p->more_l || p->more_r)
536 if (p->more_l && p->more_r)
537 cmp = (*info->cmp)(p->buf_l, p->buf_r);
544 memcpy (buf, p->buf_l, info->key_size);
545 p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l);
551 p->more_r = rset_forward(
552 info->rset_r, p->rfd_r,
553 p->buf_r, (info->cmp), p->buf_l);
557 memcpy (buf, p->buf_l, info->key_size);
560 p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l);
563 cmp = (*info->cmp)(p->buf_l, buf);
564 } while (cmp >= -1 && cmp <= 1);
567 p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r);
570 cmp = (*info->cmp)(p->buf_r, buf);
571 } while (cmp >= -1 && cmp <= 1);
578 static int r_write (RSFD rfd, const void *buf)
580 logf (LOG_FATAL, "bool set type is read-only");
584 static void r_pos (RSFD rfd, double *current, double *total)
586 struct rset_bool_rfd *p = (struct rset_bool_rfd *) rfd;
587 struct rset_bool_info *info = p->info;
592 rset_pos(info->rset_l, p->rfd_l, &lcur, <ot);
593 rset_pos(info->rset_r, p->rfd_r, &rcur, &rtot);
594 if ( (rtot<0) && (ltot<0)) { /*no position */
595 *current=rcur; /* return same as you got */
596 *total=rtot; /* probably -1 for not available */
598 if ( rtot<0) { rtot=0; rcur=0;} /* if only one useful, use it */
599 if ( ltot<0) { ltot=0; lcur=0;}
600 if ( rtot+ltot < 1 ) { /* empty rset */
605 r=1.0*(lcur+rcur)/(ltot+rtot); /* weighed average of l and r */
606 *current=(double) (p->hits);
609 yaz_log(LOG_DEBUG,"bool_pos: (%s/%s) %0.1f/%0.1f= %0.4f ",
610 info->rset_l->control->desc, info->rset_r->control->desc,
611 *current, *total, r);