rset/rsisamb.c

   1 /* $Id: rsisamb.c,v 1.13 2004-08-06 10:09:28 heikki Exp $
   2    Copyright (C) 1995,1996,1997,1998,1999,2000,2001,2002,2003,2004
   3    Index Data Aps
   4
   5 This file is part of the Zebra server.
   6
   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
  10 version.
  11
  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
  15 for more details.
  16
  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
  20 02111-1307, USA.
  21 */
  22
  23 #include <stdio.h>
  24 #include <assert.h>
  25 #include <zebrautl.h>
  26 #include <rsisamb.h>
  27 #include <string.h>
  28 #include <../index/index.h> /* for log_keydump. Debugging only */
  29
  30 #ifndef RSET_DEBUG
  31 #define RSET_DEBUG 0
  32 #endif
  33
  34 static void *r_create(RSET ct, const struct rset_control *sel, void *parms);
  35 static RSFD r_open (RSET ct, int flag);
  36 static void r_close (RSFD rfd);
  37 static void r_delete (RSET ct);
  38 static void r_rewind (RSFD rfd);
  39 static int r_forward(RSET ct, RSFD rfd, void *buf, int *term_index,
  40                      int (*cmpfunc)(const void *p1, const void *p2),
  41                      const void *untilbuf);
  42 static void r_pos (RSFD rfd, double *current, double *total);
  43 static int r_read (RSFD rfd, void *buf, int *term_index);
  44 static int r_write (RSFD rfd, const void *buf);
  45
  46 static const struct rset_control control =
  47 {
  48     "isamb",
  49     r_create,
  50     r_open,
  51     r_close,
  52     r_delete,
  53     r_rewind,
  54     r_forward, /* rset_default_forward, */
  55     r_pos,
  56     r_read,
  57     r_write,
  58 };
  59
  60 const struct rset_control *rset_kind_isamb = &control;
  61
  62 struct rset_pp_info {
  63     ISAMB_PP pt;
  64     struct rset_pp_info *next;
  65     struct rset_isamb_info *info;
  66     int *countp;
  67     void *buf;
  68 };
  69
  70 struct rset_isamb_info {
  71     ISAMB   is;
  72     ISAMB_P pos;
  73     int key_size;
  74     int (*cmp)(const void *p1, const void *p2);
  75     struct rset_pp_info *ispt_list;
  76 };
  77
  78 static void *r_create(RSET ct, const struct rset_control *sel, void *parms)
  79 {
  80     rset_isamb_parms *pt = (rset_isamb_parms *) parms;
  81     struct rset_isamb_info *info;
  82
  83     ct->flags |= RSET_FLAG_VOLATILE;
  84     info = (struct rset_isamb_info *) xmalloc (sizeof(*info));
  85     info->is = pt->is;
  86     info->pos = pt->pos;
  87     info->key_size = pt->key_size;
  88     info->cmp = pt->cmp;
  89     info->ispt_list = NULL;
  90     ct->no_rset_terms = 1;
  91     ct->rset_terms = (RSET_TERM *) xmalloc (sizeof(*ct->rset_terms));
  92     ct->rset_terms[0] = pt->rset_term;
  93     return info;
  94 }
  95
  96 RSFD r_open (RSET ct, int flag)
  97 {
  98     struct rset_isamb_info *info = (struct rset_isamb_info *) ct->buf;
  99     struct rset_pp_info *ptinfo;
 100
 101     logf (LOG_DEBUG, "risamb_open");
 102     if (flag & RSETF_WRITE)
 103     {
 104         logf (LOG_FATAL, "ISAMB set type is read-only");
 105         return NULL;
 106     }
 107     ptinfo = (struct rset_pp_info *) xmalloc (sizeof(*ptinfo));
 108     ptinfo->next = info->ispt_list;
 109     info->ispt_list = ptinfo;
 110     ptinfo->pt = isamb_pp_open (info->is, info->pos);
 111     ptinfo->info = info;
 112     if (ct->rset_terms[0]->nn < 0)
 113         ct->rset_terms[0]->nn = isamb_pp_num (ptinfo->pt);
 114     ct->rset_terms[0]->count = 0;
 115     ptinfo->countp = &ct->rset_terms[0]->count;
 116     ptinfo->buf = xmalloc (info->key_size);
 117     return ptinfo;
 118 }
 119
 120 static void r_close (RSFD rfd)
 121 {
 122     struct rset_isamb_info *info = ((struct rset_pp_info*) rfd)->info;
 123     struct rset_pp_info **ptinfop;
 124
 125     for (ptinfop = &info->ispt_list; *ptinfop; ptinfop = &(*ptinfop)->next)
 126         if (*ptinfop == rfd)
 127         {
 128             xfree ((*ptinfop)->buf);
 129             isamb_pp_close ((*ptinfop)->pt);
 130             *ptinfop = (*ptinfop)->next;
 131             xfree (rfd);
 132             return;
 133         }
 134     logf (LOG_FATAL, "r_close but no rfd match!");
 135     assert (0);
 136 }
 137
 138 static void r_delete (RSET ct)
 139 {
 140     struct rset_isamb_info *info = (struct rset_isamb_info *) ct->buf;
 141
 142     logf (LOG_DEBUG, "rsisamb_delete");
 143     assert (info->ispt_list == NULL);
 144     rset_term_destroy (ct->rset_terms[0]);
 145     xfree (ct->rset_terms);
 146     xfree (info);
 147 }
 148
 149 static void r_rewind (RSFD rfd)
 150 {
 151     logf (LOG_DEBUG, "rsisamb_rewind");
 152     abort ();
 153 }
 154
 155 static int r_forward(RSET ct, RSFD rfd, void *buf, int *term_index,
 156                      int (*cmpfunc)(const void *p1, const void *p2),
 157                      const void *untilbuf)
 158 {
 159     int i;
 160     struct rset_pp_info *pinfo = (struct rset_pp_info *) rfd;
 161 #if RSET_DEBUG
 162     logf (LOG_DEBUG, "rset_rsisamb_forward starting '%s' (ct=%p rfd=%p)",
 163                       ct->control->desc, ct,rfd);
 164     key_logdump(LOG_DEBUG, untilbuf);
 165     key_logdump(LOG_DEBUG, buf);
 166 #endif
 167
 168     i=isamb_pp_forward(pinfo->pt, buf, untilbuf);
 169 #if RSET_DEBUG
 170     logf (LOG_DEBUG, "rset_rsisamb_forward returning %d",i);
 171 #endif
 172     return i;
 173 }
 174
 175 static void r_pos (RSFD rfd, double *current, double *total)
 176 {
 177     struct rset_pp_info *pinfo = (struct rset_pp_info *) rfd;
 178     assert(rfd);
 179     isamb_pp_pos(pinfo->pt, current, total);
 180 #if RSET_DEBUG
 181     logf(LOG_DEBUG,"isamb.r_pos returning %0.1f/%0.1f",
 182               *current, *total);
 183 #endif
 184 }
 185
 186 static int r_read (RSFD rfd, void *buf, int *term_index)
 187 {
 188     struct rset_pp_info *pinfo = (struct rset_pp_info *) rfd;
 189     int r;
 190     *term_index = 0;
 191
 192     r = isamb_pp_read(pinfo->pt, buf);
 193     if (r > 0)
 194     {
 195         if (*pinfo->countp == 0 || (*pinfo->info->cmp)(buf, pinfo->buf) > 1)
 196         {
 197             memcpy (pinfo->buf, buf, pinfo->info->key_size);
 198             (*pinfo->countp)++;
 199         }
 200     }
 201     return r;
 202 }
 203
 204 static int r_write (RSFD rfd, const void *buf)
 205 {
 206     logf (LOG_FATAL, "ISAMB set type is read-only");
 207     return -1;
 208 }