Branch data Line data Source code
1 : : /* brass_termlist.cc: Termlists in a brass database
2 : : *
3 : : * Copyright 1999,2000,2001 BrightStation PLC
4 : : * Copyright 2002 Ananova Ltd
5 : : * Copyright 2002,2003,2004,2006,2007,2008,2009,2010 Olly Betts
6 : : *
7 : : * This program is free software; you can redistribute it and/or
8 : : * modify it under the terms of the GNU General Public License as
9 : : * published by the Free Software Foundation; either version 2 of the
10 : : * License, or (at your option) any later version.
11 : : *
12 : : * This program is distributed in the hope that it will be useful,
13 : : * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 : : * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 : : * GNU General Public License for more details.
16 : : *
17 : : * You should have received a copy of the GNU General Public License
18 : : * along with this program; if not, write to the Free Software
19 : : * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301
20 : : * USA
21 : : */
22 : :
23 : : #include <config.h>
24 : : #include "brass_termlist.h"
25 : :
26 : : #include "xapian/error.h"
27 : :
28 : : #include "expandweight.h"
29 : : #include "brass_positionlist.h"
30 : : #include "debuglog.h"
31 : : #include "omassert.h"
32 : : #include "pack.h"
33 : : #include "str.h"
34 : :
35 : : using namespace std;
36 : :
37 : 53688 : BrassTermList::BrassTermList(Xapian::Internal::RefCntPtr<const BrassDatabase> db_,
38 : : Xapian::docid did_)
39 : 53688 : : db(db_), did(did_), current_wdf(0), current_termfreq(0)
40 : : {
41 : : LOGCALL_CTOR(DB, "BrassTermList", db_ | did_);
42 : :
43 [ + + ][ # # ]: 53689 : if (!db->termlist_table.get_exact_entry(BrassTermListTable::make_key(did),
44 : : data))
45 : 3029 : throw Xapian::DocNotFoundError("No termlist for document " + str(did));
46 : :
47 : 50658 : pos = data.data();
48 : 50658 : end = pos + data.size();
49 : :
50 [ + + # # ]: 50658 : if (pos == end) {
51 : 9068 : doclen = 0;
52 : 9068 : termlist_size = 0;
53 : 9068 : return;
54 : : }
55 : :
56 : : // Read doclen
57 [ - + ][ # # ]: 41590 : if (!unpack_uint(&pos, end, &doclen)) {
58 : : const char *msg;
59 [ # # ][ # # ]: 0 : if (pos == 0) {
60 : 0 : msg = "Too little data for doclen in termlist";
61 : : } else {
62 : 0 : msg = "Overflowed value for doclen in termlist";
63 : : }
64 : 0 : throw Xapian::DatabaseCorruptError(msg);
65 : : }
66 : :
67 : : // Read termlist_size
68 [ - + ][ # # ]: 41590 : if (!unpack_uint(&pos, end, &termlist_size)) {
69 : : const char *msg;
70 [ # # ][ # # ]: 0 : if (pos == 0) {
71 : 0 : msg = "Too little data for list size in termlist";
72 : : } else {
73 : 0 : msg = "Overflowed value for list size in termlist";
74 : : }
75 : 0 : throw Xapian::DatabaseCorruptError(msg);
76 : : }
77 : 62778 : }
78 : :
79 : : brass_doclen_t
80 : 19094 : BrassTermList::get_doclength() const
81 : : {
82 : : LOGCALL(DB, brass_doclen_t, "BrassTermList::get_doclength", NO_ARGS);
83 : 19094 : RETURN(doclen);
84 : : }
85 : :
86 : : Xapian::termcount
87 : 46 : BrassTermList::get_approx_size() const
88 : : {
89 : : LOGCALL(DB, Xapian::termcount, "BrassTermList::get_approx_size", NO_ARGS);
90 : 46 : RETURN(termlist_size);
91 : : }
92 : :
93 : : void
94 : 1441 : BrassTermList::accumulate_stats(Xapian::Internal::ExpandStats & stats) const
95 : : {
96 : : LOGCALL_VOID(DB, "BrassTermList::accumulate_stats", stats);
97 : : Assert(!at_end());
98 : 1441 : stats.accumulate(current_wdf, doclen, get_termfreq(), db->get_doccount());
99 : 1441 : }
100 : :
101 : : string
102 : 108043 : BrassTermList::get_termname() const
103 : : {
104 : : LOGCALL(DB, string, "BrassTermList::get_termname", NO_ARGS);
105 : 108043 : RETURN(current_term);
106 : : }
107 : :
108 : : Xapian::termcount
109 : 81945 : BrassTermList::get_wdf() const
110 : : {
111 : : LOGCALL(DB, Xapian::termcount, "BrassTermList::get_wdf", NO_ARGS);
112 : 81945 : RETURN(current_wdf);
113 : : }
114 : :
115 : : Xapian::doccount
116 : 26897 : BrassTermList::get_termfreq() const
117 : : {
118 : : LOGCALL(DB, Xapian::doccount, "BrassTermList::get_termfreq", NO_ARGS);
119 [ + + ]: 26897 : if (current_termfreq == 0)
120 : 26895 : current_termfreq = db->get_termfreq(current_term);
121 : 26897 : RETURN(current_termfreq);
122 : : }
123 : :
124 : : TermList *
125 : 113033 : BrassTermList::next()
126 : : {
127 : : LOGCALL(DB, TermList *, "BrassTermList::next", NO_ARGS);
128 : : Assert(!at_end());
129 [ + + ]: 113033 : if (pos == end) {
130 : 50633 : pos = NULL;
131 : 50633 : RETURN(NULL);
132 : : }
133 : :
134 : : // Reset to 0 to indicate that the termfreq needs to be read.
135 : 62400 : current_termfreq = 0;
136 : :
137 : 62400 : bool wdf_in_reuse = false;
138 [ + + ]: 62400 : if (!current_term.empty()) {
139 : : // Find out how much of the previous term to reuse.
140 : 20810 : size_t len = static_cast<unsigned char>(*pos++);
141 [ + + ]: 20810 : if (len > current_term.size()) {
142 : : // The wdf is also stored in the "reuse" byte.
143 : 20798 : wdf_in_reuse = true;
144 : 20798 : size_t divisor = current_term.size() + 1;
145 : 20798 : current_wdf = len / divisor - 1;
146 : 20798 : len %= divisor;
147 : : }
148 : 20810 : current_term.resize(len);
149 : : }
150 : :
151 : : // Append the new tail to form the next term.
152 : 62400 : size_t append_len = static_cast<unsigned char>(*pos++);
153 : 62400 : current_term.append(pos, append_len);
154 : 62400 : pos += append_len;
155 : :
156 : : // Read the wdf if it wasn't packed into the reuse byte.
157 [ + + - + ]: 62400 : if (!wdf_in_reuse && !unpack_uint(&pos, end, ¤t_wdf)) {
[ - + ]
158 : : const char *msg;
159 [ # # ]: 0 : if (pos == 0) {
160 : 0 : msg = "Too little data for wdf in termlist";
161 : : } else {
162 : 0 : msg = "Overflowed value for wdf in termlist";
163 : : }
164 : 0 : throw Xapian::DatabaseCorruptError(msg);
165 : : }
166 : :
167 : 113033 : RETURN(NULL);
168 : : }
169 : :
170 : : TermList *
171 : 56 : BrassTermList::skip_to(const string & term)
172 : : {
173 : : LOGCALL(API, TermList *, "BrassTermList::skip_to", term);
174 [ + + ][ + + ]: 598 : while (pos != NULL && current_term < term) {
[ + + ]
175 : 542 : (void)BrassTermList::next();
176 : : }
177 : 56 : RETURN(NULL);
178 : : }
179 : :
180 : : bool
181 : 113297 : BrassTermList::at_end() const
182 : : {
183 : : LOGCALL(DB, bool, "BrassTermList::at_end", NO_ARGS);
184 : 113297 : RETURN(pos == NULL);
185 : : }
186 : :
187 : : Xapian::termcount
188 : 11210 : BrassTermList::positionlist_count() const
189 : : {
190 : : LOGCALL(DB, Xapian::termcount, "BrassTermList::positionlist_count", NO_ARGS);
191 : 11210 : RETURN(db->position_table.positionlist_count(did, current_term));
192 : : }
193 : :
194 : : Xapian::PositionIterator
195 : 33755 : BrassTermList::positionlist_begin() const
196 : : {
197 : : LOGCALL(DB, Xapian::PositionIterator, "BrassTermList::positionlist_begin", NO_ARGS);
198 : : return Xapian::PositionIterator(
199 : 33755 : new BrassPositionList(&db->position_table, did, current_term));
200 : : }
|
