allow RPN to CQL conversion with Yaz_cql2rpn class YPP-13
[yazpp-moved-to-github.git] / src / yaz-cql2rpn.cpp
1 /* This file is part of the yazpp toolkit.
2  * Copyright (C) Index Data 
3  * See the file LICENSE for details.
4  */
5
6 #if HAVE_CONFIG_H
7 #include <config.h>
8 #endif
9 #include <yaz/log.h>
10 #include <yaz/diagsrw.h>
11 #include <yaz/pquery.h>
12 #include <yaz/sortspec.h>
13 #include <yazpp/cql2rpn.h>
14 #include <yaz/rpn2cql.h>
15
16 using namespace yazpp_1;
17
18 Yaz_cql2rpn::Yaz_cql2rpn()
19 {
20     m_transform = 0;
21 }
22
23 Yaz_cql2rpn::~Yaz_cql2rpn()
24 {
25     if (m_transform)
26         cql_transform_close(m_transform);
27 }
28
29 void Yaz_cql2rpn::set_pqf_file(const char *fname)
30 {
31     if (!m_transform)
32         m_transform = cql_transform_open_fname(fname);
33 }
34
35
36 bool Yaz_cql2rpn::parse_spec_file(const char *fname, int *error)
37 {
38     *error = 0;
39     cql_transform_close(m_transform);
40     m_transform = cql_transform_open_fname(fname);
41     return m_transform ? true : false;
42 }
43
44 int Yaz_cql2rpn::rpn2cql_transform(Z_RPNQuery *q, WRBUF cql, ODR o,
45                                    char **addinfop)
46 {
47     int r = cql_transform_rpn2cql_wrbuf(m_transform, cql, q);
48     *addinfop = 0;
49     if (r)
50     {
51         const char *addinfo = 0;
52         r = cql_transform_error(m_transform, &addinfo);
53         *addinfop = odr_strdup_null(o, addinfo);
54     }
55     return r;
56 }
57
58 int Yaz_cql2rpn::query_transform(const char *cql_query,
59                                  Z_RPNQuery **rpnquery, ODR o,
60                                  char **addinfop)
61 {
62     const char *addinfo = 0;
63     if (!m_transform)
64         return -3;
65     CQL_parser cp = cql_parser_create();
66
67     int r = cql_parser_string(cp, cql_query);
68     if (r)
69     {
70         r = YAZ_SRW_QUERY_SYNTAX_ERROR;
71     }
72     else
73     {
74         WRBUF pqf = wrbuf_alloc();
75         r = cql_transform(m_transform, cql_parser_result(cp),
76                           wrbuf_vp_puts, pqf);
77         if (!r)
78         {
79             WRBUF sortkeys = wrbuf_alloc();
80             WRBUF sortspec = wrbuf_alloc();
81             if (cql_sortby_to_sortkeys(cql_parser_result(cp),
82                                        wrbuf_vp_puts, sortkeys))
83             {
84                 r = YAZ_SRW_UNSUPP_SORT_TYPE;
85             }
86             else
87             {
88                 yaz_srw_sortkeys_to_sort_spec(wrbuf_cstr(sortkeys), sortspec);
89                 Z_SortKeySpecList *sksl =
90                     yaz_sort_spec(o, wrbuf_cstr(sortspec));
91                 if (sksl)
92                     yaz_sort_spec_to_type7(sksl, pqf);
93             }
94             wrbuf_destroy(sortspec);
95             wrbuf_destroy(sortkeys);
96
97             YAZ_PQF_Parser pp = yaz_pqf_create();
98
99             *rpnquery = yaz_pqf_parse(pp, o, wrbuf_cstr(pqf));
100             if (!*rpnquery)
101             {
102                 size_t off;
103                 const char *pqf_msg;
104                 yaz_pqf_error(pp, &pqf_msg, &off);
105                 r = -1;
106             }
107             yaz_pqf_destroy(pp);
108         }
109         else
110         {
111             r = cql_transform_error(m_transform, &addinfo);
112         }
113         wrbuf_destroy(pqf);
114     }
115     cql_parser_destroy(cp);
116     *addinfop = odr_strdup_null(o, addinfo);
117     return r;
118 }
119 /*
120  * Local variables:
121  * c-basic-offset: 4
122  * c-file-style: "Stroustrup"
123  * indent-tabs-mode: nil
124  * End:
125  * vim: shiftwidth=4 tabstop=8 expandtab
126  */
127