1 /* $Id: rsbool.c,v 1.34 2004-08-06 09:43:03 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 void *r_create(RSET ct, const struct rset_control *sel, void *parms);
37 static RSFD r_open (RSET ct, int flag);
38 static void r_close (RSFD rfd);
39 static void r_delete (RSET ct);
40 static void r_rewind (RSFD rfd);
41 static int r_forward(RSET ct, RSFD rfd, void *buf, int *term_index,
42 int (*cmpfunc)(const void *p1, const void *p2),
43 const void *untilbuf);
44 static void r_pos (RSFD rfd, zint *current, zint *total);
45 static int r_read_and (RSFD rfd, void *buf, int *term_index);
46 static int r_read_or (RSFD rfd, void *buf, int *term_index);
47 static int r_read_not (RSFD rfd, void *buf, int *term_index);
48 static int r_write (RSFD rfd, const void *buf);
50 static const struct rset_control control_and =
58 r_forward, /* rset_default_forward, */
59 r_pos, /* rset_default_pos */
64 static const struct rset_control control_or =
78 static const struct rset_control control_not =
93 const struct rset_control *rset_kind_and = &control_and;
94 const struct rset_control *rset_kind_or = &control_or;
95 const struct rset_control *rset_kind_not = &control_not;
97 struct rset_bool_info {
102 int (*cmp)(const void *p1, const void *p2);
103 void (*log_item)(int logmask, const void *p, const char *txt);
104 struct rset_bool_rfd *rfd_list;
107 struct rset_bool_rfd {
118 struct rset_bool_rfd *next;
119 struct rset_bool_info *info;
122 static void *r_create (RSET ct, const struct rset_control *sel, void *parms)
124 rset_bool_parms *bool_parms = (rset_bool_parms *) parms;
125 struct rset_bool_info *info;
127 info = (struct rset_bool_info *) xmalloc (sizeof(*info));
128 info->key_size = bool_parms->key_size;
129 info->rset_l = bool_parms->rset_l;
130 info->rset_r = bool_parms->rset_r;
131 if (rset_is_volatile(info->rset_l) || rset_is_volatile(info->rset_r))
132 ct->flags |= RSET_FLAG_VOLATILE;
133 info->cmp = bool_parms->cmp;
134 info->log_item = bool_parms->log_item;
135 info->rfd_list = NULL;
137 info->term_index_s = info->rset_l->no_rset_terms;
139 info->rset_l->no_rset_terms + info->rset_r->no_rset_terms;
140 ct->rset_terms = (RSET_TERM *)
141 xmalloc (sizeof (*ct->rset_terms) * ct->no_rset_terms);
143 memcpy (ct->rset_terms, info->rset_l->rset_terms,
144 info->rset_l->no_rset_terms * sizeof(*ct->rset_terms));
145 memcpy (ct->rset_terms + info->rset_l->no_rset_terms,
146 info->rset_r->rset_terms,
147 info->rset_r->no_rset_terms * sizeof(*ct->rset_terms));
151 static RSFD r_open (RSET ct, int flag)
153 struct rset_bool_info *info = (struct rset_bool_info *) ct->buf;
154 struct rset_bool_rfd *rfd;
156 if (flag & RSETF_WRITE)
158 logf (LOG_FATAL, "bool set type is read-only");
161 rfd = (struct rset_bool_rfd *) xmalloc (sizeof(*rfd));
162 logf(LOG_DEBUG,"rsbool (%s) open [%p]", ct->control->desc, rfd);
163 rfd->next = info->rfd_list;
164 info->rfd_list = rfd;
168 rfd->buf_l = xmalloc (info->key_size);
169 rfd->buf_r = xmalloc (info->key_size);
170 rfd->rfd_l = rset_open (info->rset_l, RSETF_READ);
171 rfd->rfd_r = rset_open (info->rset_r, RSETF_READ);
172 rfd->more_l = rset_read (info->rset_l, rfd->rfd_l, rfd->buf_l,
174 rfd->more_r = rset_read (info->rset_r, rfd->rfd_r, rfd->buf_r,
180 static void r_close (RSFD rfd)
182 struct rset_bool_info *info = ((struct rset_bool_rfd*)rfd)->info;
183 struct rset_bool_rfd **rfdp;
185 for (rfdp = &info->rfd_list; *rfdp; rfdp = &(*rfdp)->next)
188 xfree ((*rfdp)->buf_l);
189 xfree ((*rfdp)->buf_r);
190 rset_close (info->rset_l, (*rfdp)->rfd_l);
191 rset_close (info->rset_r, (*rfdp)->rfd_r);
192 *rfdp = (*rfdp)->next;
196 logf (LOG_FATAL, "r_close but no rfd match!");
200 static void r_delete (RSET ct)
202 struct rset_bool_info *info = (struct rset_bool_info *) ct->buf;
204 assert (info->rfd_list == NULL);
205 xfree (ct->rset_terms);
206 rset_delete (info->rset_l);
207 rset_delete (info->rset_r);
211 static void r_rewind (RSFD rfd)
213 struct rset_bool_info *info = ((struct rset_bool_rfd*)rfd)->info;
214 struct rset_bool_rfd *p = (struct rset_bool_rfd *) rfd;
216 logf (LOG_DEBUG, "rsbool_rewind");
217 rset_rewind (info->rset_l, p->rfd_l);
218 rset_rewind (info->rset_r, p->rfd_r);
219 p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l, &p->term_index_l);
220 p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r, &p->term_index_r);
224 static int r_forward (RSET ct, RSFD rfd, void *buf, int *term_index,
225 int (*cmpfunc)(const void *p1, const void *p2),
226 const void *untilbuf)
228 struct rset_bool_info *info = ((struct rset_bool_rfd*)rfd)->info;
229 struct rset_bool_rfd *p = (struct rset_bool_rfd *) rfd;
233 logf (LOG_DEBUG, "rsbool_forward (L) [%p] '%s' (ct=%p rfd=%p m=%d,%d)",
234 rfd, ct->control->desc, ct, rfd, p->more_l, p->more_r);
236 if ( p->more_l && ((cmpfunc)(untilbuf,p->buf_l)==2) )
237 p->more_l = rset_forward(info->rset_l, p->rfd_l, p->buf_l,
238 &p->term_index_l, info->cmp, untilbuf);
240 logf (LOG_DEBUG, "rsbool_forward (R) [%p] '%s' (ct=%p rfd=%p m=%d,%d)",
241 rfd, ct->control->desc, ct, rfd, p->more_l, p->more_r);
243 if ( p->more_r && ((cmpfunc)(untilbuf,p->buf_r)==2))
244 p->more_r = rset_forward(info->rset_r, p->rfd_r, p->buf_r,
245 &p->term_index_r, info->cmp, untilbuf);
247 logf (LOG_DEBUG, "rsbool_forward [%p] calling read, m=%d,%d t=%d",
248 rfd, p->more_l, p->more_r, p->tail);
252 rc = rset_read(ct,rfd,buf,term_index);
254 logf (LOG_DEBUG, "rsbool_forward returning [%p] %d m=%d,%d",
255 rfd, rc, p->more_l, p->more_r);
276 static int r_read_and (RSFD rfd, void *buf, int *term_index)
278 struct rset_bool_rfd *p = (struct rset_bool_rfd *) rfd;
279 struct rset_bool_info *info = p->info;
281 { zint cur,tot; r_pos(rfd, &cur, &tot); }
283 while (p->more_l || p->more_r)
287 if (p->more_l && p->more_r)
288 cmp = (*info->cmp)(p->buf_l, p->buf_r);
294 logf (LOG_DEBUG, "r_read_and [%p] looping: m=%d/%d c=%d t=%d",
295 rfd, p->more_l, p->more_r, cmp, p->tail);
296 (*info->log_item)(LOG_DEBUG, p->buf_l, "left ");
297 (*info->log_item)(LOG_DEBUG, p->buf_r, "right ");
301 memcpy (buf, p->buf_l, info->key_size);
302 *term_index = p->term_index_l;
303 p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l,
309 memcpy (buf, p->buf_r, info->key_size);
310 *term_index = p->term_index_r + info->term_index_s;
311 p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r,
315 logf (LOG_DEBUG, "r_read_and [%p] returning R m=%d/%d c=%d",
316 rfd, p->more_l, p->more_r, cmp);
317 key_logdump(LOG_DEBUG,buf);
318 (*info->log_item)(LOG_DEBUG, buf, "");
325 memcpy (buf, p->buf_l, info->key_size);
326 *term_index = p->term_index_l;
327 p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l,
331 logf (LOG_DEBUG, "r_read_and [%p] returning L m=%d/%d c=%d",
332 rfd, p->more_l, p->more_r, cmp);
333 (*info->log_item)(LOG_DEBUG, buf, "");
338 else if (cmp > 1) /* cmp == 2 */
342 memcpy (buf, p->buf_r, info->key_size);
343 *term_index = p->term_index_r + info->term_index_s;
345 p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r,
349 if (!p->more_r || (*info->cmp)(p->buf_r, buf) > 1)
352 logf (LOG_DEBUG, "r_read_and returning C m=%d/%d c=%d",
353 p->more_l, p->more_r, cmp);
354 (*info->log_item)(LOG_DEBUG, buf, "");
363 memcpy (buf, p->buf_r, info->key_size);
364 *term_index = p->term_index_r + info->term_index_s;
365 p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r,
367 if (!p->more_r || (*info->cmp)(p->buf_r, buf) > 1)
370 logf (LOG_DEBUG, "r_read_and [%p] returning R tail m=%d/%d c=%d",
371 rfd, p->more_l, p->more_r, cmp);
372 (*info->log_item)(LOG_DEBUG, buf, "");
380 logf (LOG_DEBUG, "r_read_and [%p] about to forward R m=%d/%d c=%d",
381 rfd, p->more_l, p->more_r, cmp);
383 if (p->more_r && p->more_l)
384 p->more_r = rset_forward(
385 info->rset_r, p->rfd_r,
386 p->buf_r, &p->term_index_r,
387 (info->cmp), p->buf_l);
389 return 0; /* no point in reading further */
396 memcpy (buf, p->buf_l, info->key_size);
397 *term_index = p->term_index_l;
398 p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l,
402 if (!p->more_l || (*info->cmp)(p->buf_l, buf) > 1)
405 logf (LOG_DEBUG, "r_read_and [%p] returning R tail m=%d/%d c=%d",
406 rfd, p->more_l, p->more_r, cmp);
407 (*info->log_item)(LOG_DEBUG, buf, "");
415 memcpy (buf, p->buf_l, info->key_size);
416 *term_index = p->term_index_l;
417 p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l,
419 if (!p->more_l || (*info->cmp)(p->buf_l, buf) > 1)
422 logf (LOG_DEBUG, "r_read_and [%p] returning L tail m=%d/%d c=%d",
423 rfd, p->more_l, p->more_r, cmp);
424 (*info->log_item)(LOG_DEBUG, buf, "");
432 logf (LOG_DEBUG, "r_read_and [%p] about to forward L m=%d/%d c=%d",
433 rfd, p->more_l, p->more_r, cmp);
435 if (p->more_r && p->more_l)
436 p->more_l = rset_forward(
437 /* p->more_l = rset_default_forward( */
438 info->rset_l, p->rfd_l,
439 p->buf_l, &p->term_index_l,
440 (info->cmp), p->buf_r);
442 return 0; /* no point in reading further */
448 logf (LOG_DEBUG, "r_read_and [%p] reached its end",rfd);
453 static int r_read_or (RSFD rfd, void *buf, int *term_index)
455 struct rset_bool_rfd *p = (struct rset_bool_rfd *) rfd;
456 struct rset_bool_info *info = p->info;
458 { zint cur,tot; r_pos(rfd, &cur, &tot); }
459 while (p->more_l || p->more_r)
463 if (p->more_l && p->more_r)
464 cmp = (*info->cmp)(p->buf_l, p->buf_r);
471 memcpy (buf, p->buf_l, info->key_size);
472 *term_index = p->term_index_l;
473 p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l,
475 p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r,
478 logf (LOG_DEBUG, "r_read_or returning A m=%d/%d c=%d",
479 p->more_l, p->more_r, cmp);
480 (*info->log_item)(LOG_DEBUG, buf, "");
487 memcpy (buf, p->buf_r, info->key_size);
488 *term_index = p->term_index_r + info->term_index_s;
489 p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r,
492 logf (LOG_DEBUG, "r_read_or returning B 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_l, info->key_size);
502 *term_index = p->term_index_l;
503 p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l,
506 logf (LOG_DEBUG, "r_read_or returning C m=%d/%d c=%d",
507 p->more_l, p->more_r, cmp);
508 (*info->log_item)(LOG_DEBUG, buf, "");
517 static int r_read_not (RSFD rfd, void *buf, int *term_index)
519 struct rset_bool_rfd *p = (struct rset_bool_rfd *) rfd;
520 struct rset_bool_info *info = p->info;
522 { zint cur,tot; r_pos(rfd, &cur, &tot); }
523 while (p->more_l || p->more_r)
527 if (p->more_l && p->more_r)
528 cmp = (*info->cmp)(p->buf_l, p->buf_r);
535 memcpy (buf, p->buf_l, info->key_size);
536 *term_index = p->term_index_l;
537 p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l,
544 p->more_r = rset_forward(
545 info->rset_r, p->rfd_r,
546 p->buf_r, &p->term_index_r,
547 (info->cmp), p->buf_l);
551 memcpy (buf, p->buf_l, info->key_size);
554 p->more_l = rset_read (info->rset_l, p->rfd_l, p->buf_l,
558 cmp = (*info->cmp)(p->buf_l, buf);
559 } while (cmp >= -1 && cmp <= 1);
562 p->more_r = rset_read (info->rset_r, p->rfd_r, p->buf_r,
566 cmp = (*info->cmp)(p->buf_r, buf);
567 } while (cmp >= -1 && cmp <= 1);
574 static int r_write (RSFD rfd, const void *buf)
576 logf (LOG_FATAL, "bool set type is read-only");
580 static void r_pos (RSFD rfd, zint *current, zint *total)
582 struct rset_bool_rfd *p = (struct rset_bool_rfd *) rfd;
583 struct rset_bool_info *info = p->info;
588 rset_pos(info->rset_l, p->rfd_l, &lcur, <ot);
589 rset_pos(info->rset_r, p->rfd_r, &rcur, &rtot);
590 if ( (rtot<0) && (ltot<0)) { /*no position */
591 *current=rcur; /* return same as you got */
592 *total=rtot; /* probably -1 for not available */
594 if ( rtot<0) { rtot=0; rcur=0;} /* if only one useful, use it */
595 if ( ltot<0) { ltot=0; lcur=0;}
596 if ( rtot+ltot == 0 ) { /* empty rset */
601 r=1.0*(lcur+rcur)/(ltot+rtot); /* weighed average of l and r */
603 *total=(zint)(0.5+*current/r);
605 yaz_log(LOG_DEBUG,"bool_pos: (%s/%s) "ZINT_FORMAT"/"ZINT_FORMAT"= %0.4f ",
606 info->rset_l->control->desc, info->rset_r->control->desc,
607 *current, *total, r);