872536ff72c2757a09de0312ac80d95c6eebaccd
[idzebra-moved-to-github.git] / rset / rsisamb.c
1 /* $Id: rsisamb.c,v 1.32 2005-04-26 10:09:38 adam Exp $
2    Copyright (C) 1995-2005
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 <idzebra/util.h>
26 #include <rset.h>
27 #include <string.h>
28
29
30 static RSFD r_open (RSET ct, int flag);
31 static void r_close (RSFD rfd);
32 static void r_delete (RSET ct);
33 static int r_forward(RSFD rfd, void *buf, TERMID *term, const void *untilbuf);
34 static void r_pos (RSFD rfd, double *current, double *total);
35 static int r_read (RSFD rfd, void *buf, TERMID *term);
36 static int r_write (RSFD rfd, const void *buf);
37
38 static const struct rset_control control = 
39 {
40     "isamb",
41     r_delete,
42     rset_get_one_term,
43     r_open,
44     r_close,
45     r_forward, 
46     r_pos,
47     r_read,
48     r_write,
49 };
50
51 struct rset_pp_info {
52     ISAMB_PP pt;
53     void *buf;
54 };
55
56 struct rset_isamb_info {
57     ISAMB   is;
58     ISAM_P pos;
59 };
60
61 static int log_level = 0;
62 static int log_level_initialized = 0;
63
64 RSET rsisamb_create(NMEM nmem, const struct key_control *kcontrol, int scope,
65                     ISAMB is, ISAM_P pos, TERMID term)
66 {
67     RSET rnew = rset_create_base(&control, nmem, kcontrol, scope, term);
68     struct rset_isamb_info *info;
69     if (!log_level_initialized)
70     {
71         log_level = yaz_log_module_level("rsisamb");
72         log_level_initialized = 1;
73     }
74     info = (struct rset_isamb_info *) nmem_malloc(rnew->nmem,sizeof(*info));
75     info->is = is;
76     info->pos = pos;
77     rnew->priv = info;
78     yaz_log(log_level,"rsisamb_create");
79     return rnew;
80 }
81
82 static void r_delete (RSET ct)
83 {
84     yaz_log(log_level, "rsisamb_delete");
85 }
86
87 RSFD r_open (RSET ct, int flag)
88 {
89     struct rset_isamb_info *info = (struct rset_isamb_info *) ct->priv;
90     RSFD rfd;
91     struct rset_pp_info *ptinfo;
92
93     if (flag & RSETF_WRITE)
94     {
95         yaz_log(YLOG_FATAL, "ISAMB set type is read-only");
96         return NULL;
97     }
98     rfd = rfd_create_base(ct);
99     if (rfd->priv)
100         ptinfo = (struct rset_pp_info *) (rfd->priv);
101     else {
102         ptinfo = (struct rset_pp_info *)nmem_malloc(ct->nmem,sizeof(*ptinfo));
103         ptinfo->buf = nmem_malloc (ct->nmem,ct->keycontrol->key_size);
104         rfd->priv = ptinfo;
105     }
106     ptinfo->pt = isamb_pp_open (info->is, info->pos, ct->scope );
107     yaz_log(log_level,"rsisamb_open");
108     return rfd;
109 }
110
111 static void r_close (RSFD rfd)
112 {
113     struct rset_pp_info *ptinfo = (struct rset_pp_info *)(rfd->priv);
114     isamb_pp_close (ptinfo->pt);
115     rfd_delete_base(rfd);
116     yaz_log(log_level,"rsisamb_close");
117 }
118
119
120 static int r_forward(RSFD rfd, void *buf, TERMID *term, const void *untilbuf)
121 {
122     struct rset_pp_info *pinfo = (struct rset_pp_info *)(rfd->priv);
123     int rc;
124     rc = isamb_pp_forward(pinfo->pt, buf, untilbuf);
125     if (rc && term)
126         *term = rfd->rset->term;
127     yaz_log(log_level,"rsisamb_forward");
128     return rc; 
129 }
130
131 static void r_pos (RSFD rfd, double *current, double *total)
132 {
133     struct rset_pp_info *pinfo = (struct rset_pp_info *)(rfd->priv);
134     assert(rfd);
135     isamb_pp_pos(pinfo->pt, current, total);
136     yaz_log(log_level,"isamb.r_pos returning %0.1f/%0.1f",
137               *current, *total);
138 }
139
140 static int r_read (RSFD rfd, void *buf, TERMID *term)
141 {
142     struct rset_pp_info *pinfo = (struct rset_pp_info *)(rfd->priv);
143     int rc;
144     rc = isamb_pp_read(pinfo->pt, buf);
145     if (rc && term)
146         *term = rfd->rset->term;
147     yaz_log(log_level,"isamb.r_read ");
148     return rc;
149 }
150
151 static int r_write (RSFD rfd, const void *buf)
152 {
153     yaz_log(YLOG_FATAL, "ISAMB set type is read-only");
154     return -1;
155 }