Expanded tabs in all source files. Added vim/emacs local variables
[yazpp-moved-to-github.git] / src / yaz-z-query.cpp
1 /*
2  * Copyright (c) 1998-2003, Index Data.
3  * See the file LICENSE for details.
4  * 
5  * $Id: yaz-z-query.cpp,v 1.17 2005-06-25 15:53:19 adam Exp $
6  */
7
8 #include <yaz++/z-query.h>
9 #include <yaz/pquery.h>
10
11 using namespace yazpp_1;
12
13 Yaz_Z_Query::Yaz_Z_Query()
14 {
15     odr_encode = odr_createmem (ODR_ENCODE);
16     odr_decode = odr_createmem (ODR_DECODE);
17     odr_print = odr_createmem (ODR_PRINT);
18 }
19
20 int Yaz_Z_Query::set_rpn (const char *rpn)
21 {
22     m_buf = 0;
23     odr_reset (odr_encode);
24     Z_Query *query = (Z_Query*) odr_malloc (odr_encode, sizeof(*query));
25     query->which = Z_Query_type_1;
26     query->u.type_1 = p_query_rpn (odr_encode, PROTO_Z3950, rpn);
27     if (!query->u.type_1)
28         return -1;
29     if (!z_Query (odr_encode, &query, 0, 0))
30         return -1;
31     // z_Query(odr_print, &query, 0, 0);
32     m_buf = odr_getbuf (odr_encode, &m_len, 0);
33     return m_len;
34 }
35
36 void Yaz_Z_Query::set_Z_Query(Z_Query *z_query)
37 {
38     m_buf = 0;
39     odr_reset (odr_encode);
40     if (!z_Query (odr_encode, &z_query, 0, 0))
41         return;
42     m_buf = odr_getbuf (odr_encode, &m_len, 0);
43 }
44
45 Yaz_Z_Query::~Yaz_Z_Query()
46 {
47     odr_destroy (odr_encode);
48     odr_destroy (odr_decode);
49     odr_destroy (odr_print);
50 }
51
52 Z_Query *Yaz_Z_Query::get_Z_Query ()
53 {
54     Z_Query *query;
55     if (!m_buf)
56         return 0;
57     odr_reset(odr_decode);
58     odr_setbuf(odr_decode, m_buf, m_len, 0);
59     if (!z_Query(odr_decode, &query, 0, 0))
60         return 0;
61     return query;
62 }
63
64 void Yaz_Z_Query::print(char *str, int len)
65 {
66     Z_Query *query;
67     *str = 0;
68     if (!m_buf)
69         return;
70     odr_setbuf (odr_decode, m_buf, m_len, 0);
71     if (!z_Query(odr_decode, &query, 0, 0))
72         return;
73     WRBUF wbuf = zquery2pquery(query);
74     if (wbuf)
75     {
76         if (wrbuf_len(wbuf) > len-1)
77         {
78             memcpy(str, wrbuf_buf(wbuf), len-1);
79             str[len-1] = '\0';
80         }
81         else
82             strcpy(str, wrbuf_buf(wbuf));
83         wrbuf_free(wbuf,1);
84     }
85     odr_reset(odr_decode);
86 }
87
88 int Yaz_Z_Query::match(Yaz_Z_Query *other)
89 {
90     if (m_len != other->m_len)
91         return 0;
92     if (!m_buf || !other->m_buf)
93         return 0;
94     if (memcmp(m_buf, other->m_buf, m_len))
95         return 0;
96     return 1;
97 }
98
99 void Yaz_Z_Query::oid2str(Odr_oid *o, WRBUF buf)
100 {
101     for (; *o >= 0; o++) {
102         char ibuf[16];
103         sprintf(ibuf, "%d", *o);
104         wrbuf_puts(buf, ibuf);
105         if (o[1] > 0)
106             wrbuf_putc(buf, '.');
107     }
108 }
109
110 void Yaz_Z_Query::pr_term(WRBUF wbuf, char *buf, int len)
111 {
112     int i;
113     wrbuf_putc(wbuf, '"');
114     for (i = 0; i<len; i++)
115     {
116         int ch = buf[i];
117         if (ch == '"')
118             wrbuf_putc(wbuf, '\\');
119         wrbuf_putc(wbuf, ch);
120     }
121     wrbuf_puts(wbuf, "\" ");
122 }
123
124 int Yaz_Z_Query::rpn2pquery(Z_RPNStructure *s, WRBUF buf)
125 {
126     if (s->which == Z_RPNStructure_simple)
127     {
128         Z_Operand *o = s->u.simple;
129         
130         if (o->which == Z_Operand_APT) 
131         {
132             Z_AttributesPlusTerm *at = s->u.simple->u.attributesPlusTerm;
133             if (at->attributes) {
134                 int i;
135                 for (i = 0; i < at->attributes->num_attributes; i++) {
136                     wrbuf_puts(buf, "@attr ");
137                     if (at->attributes->attributes[i]->attributeSet) {
138                         oid2str(at->attributes->attributes[i]->attributeSet, buf);
139                         wrbuf_putc(buf, ' ');
140                     }
141                     wrbuf_printf(buf, "%d=", *at->attributes->attributes[i]->attributeType);
142                     wrbuf_printf(buf, "%d ", *at->attributes->attributes[i]->value.numeric);
143                 }
144             }
145             if (at->term->which == Z_Term_general)
146             {
147                 pr_term(buf, (char*) at->term->u.general->buf,
148                         at->term->u.general->len);
149             }
150             else if (at->term->which == Z_Term_characterString)
151             {
152                 wrbuf_puts(buf, "@term string ");
153                 pr_term(buf, at->term->u.characterString,
154                         strlen(at->term->u.characterString));
155
156             }
157         }
158         else if (o->which == Z_Operand_resultSetId)
159         {
160             wrbuf_printf(buf, "@set %s ", o->u.resultSetId);
161         }
162     }
163     else if (s->which == Z_RPNStructure_complex)
164     {
165         Z_Complex *c = s->u.complex;
166         
167         switch (c->roperator->which) {
168         case Z_Operator_and: wrbuf_puts(buf, "@and "); break;
169         case Z_Operator_or: wrbuf_puts(buf, "@or "); break;
170         case Z_Operator_and_not: wrbuf_puts(buf, "@not "); break;
171         case Z_Operator_prox: wrbuf_puts(buf, "@prox "); break;
172         default: wrbuf_puts(buf, "@unknown ");
173         }
174         if (!rpn2pquery(c->s1, buf))
175             return 0;
176         if (!rpn2pquery(c->s2, buf))
177             return 0;
178     }
179     return 1;
180 }
181
182 WRBUF Yaz_Z_Query::zquery2pquery(Z_Query *q)
183 {
184     if (q->which != Z_Query_type_1 && q->which != Z_Query_type_101) 
185         return 0;
186     WRBUF buf = wrbuf_alloc();
187     if (q->u.type_1->attributeSetId) {
188         /* Output attribute set ID */
189         wrbuf_puts(buf, "@attrset ");
190         oid2str(q->u.type_1->attributeSetId, buf);
191         wrbuf_putc(buf, ' ');
192     }
193     return rpn2pquery(q->u.type_1->RPNStructure, buf) ? buf : 0;
194 }
195
196
197 /*
198  * Local variables:
199  * c-basic-offset: 4
200  * indent-tabs-mode: nil
201  * End:
202  * vim: shiftwidth=4 tabstop=8 expandtab
203  */
204