src/yaz-z-query.cpp

   1 /*
   2  * Copyright (c) 1998-2005, Index Data.
   3  * See the file LICENSE for details.
   4  *
   5  * $Id: yaz-z-query.cpp,v 1.20 2006-06-19 13:12:07 adam Exp $
   6  */
   7
   8 #include <yaz/logrpn.h>
   9 #include <yazpp/z-query.h>
  10 #include <yaz/pquery.h>
  11 #include <assert.h>
  12
  13 using namespace yazpp_1;
  14
  15 Yaz_Z_Query::Yaz_Z_Query()
  16 {
  17     odr_encode = odr_createmem(ODR_ENCODE);
  18     odr_decode = odr_createmem(ODR_DECODE);
  19     odr_print = odr_createmem(ODR_PRINT);
  20 }
  21
  22
  23 Yaz_Z_Query::Yaz_Z_Query(const Yaz_Z_Query &q)
  24 {
  25     odr_encode = odr_createmem(ODR_ENCODE);
  26     odr_decode = odr_createmem(ODR_DECODE);
  27     odr_print = odr_createmem(ODR_PRINT);
  28
  29     m_len = q.m_len;
  30     m_buf = (char*) odr_malloc(odr_encode, m_len);
  31     memcpy(m_buf, q.m_buf, m_len);
  32 }
  33
  34 Yaz_Z_Query& Yaz_Z_Query::operator=(const Yaz_Z_Query &q)
  35 {
  36     if (this != &q)
  37     {
  38         odr_reset(odr_encode);
  39         if (!q.m_buf)
  40         {
  41             m_buf = 0;
  42             m_len = 0;
  43         }
  44         else
  45         {
  46             m_len = q.m_len;
  47             m_buf = (char*) odr_malloc(odr_encode, m_len);
  48             memcpy(m_buf, q.m_buf, m_len);
  49         }
  50     }
  51     return *this;
  52 }
  53
  54 Yaz_Z_Query& Yaz_Z_Query::operator=(const char *rpn)
  55 {
  56     set_rpn(rpn);
  57     return *this;
  58 }
  59
  60 int Yaz_Z_Query::set_rpn(const char *rpn)
  61 {
  62     m_buf = 0;
  63     odr_reset(odr_encode);
  64     Z_Query *query = (Z_Query*) odr_malloc(odr_encode, sizeof(*query));
  65     query->which = Z_Query_type_1;
  66     query->u.type_1 = p_query_rpn(odr_encode, PROTO_Z3950, rpn);
  67     if (!query->u.type_1)
  68         return -1;
  69     if (!z_Query(odr_encode, &query, 0, 0))
  70         return -1;
  71     // z_Query(odr_print, &query, 0, 0);
  72     m_buf = odr_getbuf(odr_encode, &m_len, 0);
  73     return m_len;
  74 }
  75
  76 void Yaz_Z_Query::set_Z_Query(Z_Query *z_query)
  77 {
  78     m_buf = 0;
  79     odr_reset(odr_encode);
  80     if (!z_Query(odr_encode, &z_query, 0, 0))
  81         return;
  82     m_buf = odr_getbuf(odr_encode, &m_len, 0);
  83 }
  84
  85 Yaz_Z_Query::~Yaz_Z_Query()
  86 {
  87     odr_destroy(odr_encode);
  88     odr_destroy(odr_decode);
  89     odr_destroy(odr_print);
  90 }
  91
  92 Z_Query *Yaz_Z_Query::get_Z_Query()
  93 {
  94     Z_Query *query;
  95     if (!m_buf)
  96         return 0;
  97     odr_reset(odr_decode);
  98     odr_setbuf(odr_decode, m_buf, m_len, 0);
  99     if (!z_Query(odr_decode, &query, 0, 0))
 100         return 0;
 101     return query;
 102 }
 103
 104 void Yaz_Z_Query::print(char *str, int len)
 105 {
 106     Z_Query *query;
 107     *str = 0;
 108     if (!m_buf)
 109         return;
 110     odr_setbuf(odr_decode, m_buf, m_len, 0);
 111     if (!z_Query(odr_decode, &query, 0, 0))
 112         return;
 113     WRBUF wbuf = zquery2pquery(query);
 114     if (wbuf)
 115     {
 116         if (wrbuf_len(wbuf) > len-1)
 117         {
 118             memcpy(str, wrbuf_buf(wbuf), len-1);
 119             str[len-1] = '\0';
 120         }
 121         else
 122             strcpy(str, wrbuf_buf(wbuf));
 123         wrbuf_free(wbuf,1);
 124     }
 125     odr_reset(odr_decode);
 126 }
 127
 128 int Yaz_Z_Query::match(const Yaz_Z_Query *other)
 129 {
 130     if (m_len != other->m_len)
 131         return 0;
 132     if (!m_buf || !other->m_buf)
 133         return 0;
 134     if (memcmp(m_buf, other->m_buf, m_len))
 135         return 0;
 136     return 1;
 137 }
 138
 139 WRBUF Yaz_Z_Query::zquery2pquery(Z_Query *q)
 140 {
 141     WRBUF w = wrbuf_alloc();
 142     wrbuf_put_zquery(w, q);
 143     return w;
 144 }
 145
 146 /*
 147  * Local variables:
 148  * c-basic-offset: 4
 149  * indent-tabs-mode: nil
 150  * End:
 151  * vim: shiftwidth=4 tabstop=8 expandtab
 152  */
 153