Branch data Line data Source code
1 : : /** @file smallvector.h
2 : : * @brief Custom vector implementations using small vector optimisation
3 : : */
4 : : /* Copyright (C) 2012,2013,2014,2017,2018 Olly Betts
5 : : *
6 : : * Permission is hereby granted, free of charge, to any person obtaining a copy
7 : : * of this software and associated documentation files (the "Software"), to
8 : : * deal in the Software without restriction, including without limitation the
9 : : * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10 : : * sell copies of the Software, and to permit persons to whom the Software is
11 : : * furnished to do so, subject to the following conditions:
12 : : *
13 : : * The above copyright notice and this permission notice shall be included in
14 : : * all copies or substantial portions of the Software.
15 : : *
16 : : * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 : : * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 : : * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 : : * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 : : * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
21 : : * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
22 : : * IN THE SOFTWARE.
23 : : */
24 : :
25 : : #ifndef XAPIAN_INCLUDED_SMALLVECTOR_H
26 : : #define XAPIAN_INCLUDED_SMALLVECTOR_H
27 : :
28 : : #ifndef PACKAGE
29 : : # error config.h must be included first in each C++ source file
30 : : #endif
31 : :
32 : : #include <algorithm>
33 : : #include <cstddef> // For std::size_t
34 : : #include <cstring> // For std::memcpy
35 : : #include <type_traits>
36 : :
37 : : namespace Xapian {
38 : :
39 : : /** Suitable for "simple" type T.
40 : : *
41 : : * If sizeof(T) > 2 * sizeof(T*) this isn't going to work, and it's not very
42 : : * useful when sizeof(T) == 2 * sizeof(T*) as you can only store a single
43 : : * element inline.
44 : : *
45 : : * Offers optional Copy-On-Write functionality - if COW is true, then copying
46 : : * a Vec with external data only makes a copy of that data if you attempt
47 : : * to modify it. Current COW is only supported for integral types T.
48 : : */
49 : : template<typename T,
50 : : bool COW = false,
51 : : typename = typename std::enable_if<
52 : : (COW ?
53 : : std::is_integral<T>::value :
54 : : #ifdef HAVE_STD_IS_TRIVIALLY_COPYABLE
55 : : std::is_trivially_copyable<T>::value
56 : : #else
57 : : // std::is_trivially_copyable<T> is C++11, but for example GCC
58 : : // didn't support it until version 6. Assume "true" if it's not
59 : : // available - we'll have to rely on newer compiler versions to
60 : : // catch attempts to use unsuitable types here.
61 : : true
62 : : #endif
63 : : )>::type>
64 : : class Vec {
65 : : std::size_t c;
66 : :
67 : : static constexpr std::size_t INTERNAL_CAPACITY = 2 * sizeof(T*) / sizeof(T);
68 : :
69 : : union {
70 : : T v[INTERNAL_CAPACITY];
71 : : struct {
72 : : T* b;
73 : : T* e;
74 : : };
75 : : } u;
76 : :
77 : : struct Vec_to_copy {
78 : : const Vec& ref;
79 : 1565245 : explicit Vec_to_copy(const Vec& o) : ref(o) {}
80 : : };
81 : :
82 : : public:
83 : : typedef std::size_t size_type;
84 : :
85 : : typedef const T* const_iterator;
86 : :
87 : : typedef T* iterator;
88 : :
89 : 14799694 : Vec() : c(0) { }
90 : :
91 : : // Prevent inadvertent copying.
92 : : Vec(const Vec&) = delete;
93 : :
94 : 1565245 : Vec(const Vec_to_copy& o) {
95 : 1565245 : do_copy_from(o.ref);
96 : 1565245 : }
97 : :
98 : : // Prevent inadvertent copying.
99 : : void operator=(const Vec&) = delete;
100 : :
101 : : void operator=(const Vec_to_copy& o) {
102 : : clear();
103 : : do_copy_from(o.ref);
104 : : }
105 : :
106 : 1565245 : Vec_to_copy copy() const {
107 : 1565245 : return Vec_to_copy(*this);
108 : : }
109 : :
110 : 17214422 : Vec(Vec&& o) noexcept : Vec() {
111 : 8607211 : std::memcpy(&u, &o.u, sizeof(u));
112 : 8607211 : std::swap(c, o.c);
113 : 8607211 : }
114 : :
115 : 837 : void operator=(Vec&& o) {
116 : 837 : clear();
117 : 837 : u = o.u;
118 : 837 : std::swap(c, o.c);
119 : 837 : }
120 : :
121 : 0 : explicit Vec(size_type n) : c(0) {
122 : 0 : reserve(n);
123 : 0 : }
124 : :
125 : 16364939 : ~Vec() {
126 : 16364939 : clear();
127 : 16364939 : }
128 : :
129 : 16095780 : size_type size() const {
130 [ + + ]: 16095780 : return is_external() ? u.e - u.b : c;
131 : : }
132 : :
133 : 8472342 : size_type capacity() const {
134 [ + + ]: 8472342 : return is_external() ? c : INTERNAL_CAPACITY;
135 : : }
136 : :
137 : 10045098 : bool empty() const {
138 [ + + ]: 10045098 : return is_external() ? u.b == u.e : c == 0;
139 : : }
140 : :
141 : 743280 : void reserve(size_type n) {
142 [ + + ]: 743280 : if (n > capacity()) {
143 : 189364 : do_reserve(n);
144 : : }
145 : 743280 : }
146 : :
147 : 5028275 : const_iterator cbegin() const {
148 [ + + ]: 5028275 : return is_external() ? u.b : u.v;
149 : : }
150 : :
151 : 3728247 : const_iterator cend() const {
152 [ + + ]: 3728247 : return is_external() ? u.e : u.v + c;
153 : : }
154 : :
155 : 5028201 : const_iterator begin() const {
156 : 5028201 : return cbegin();
157 : : }
158 : :
159 : 3728216 : const_iterator end() const {
160 : 3728216 : return cend();
161 : : }
162 : :
163 : 1487064 : iterator begin() {
164 : : // FIXME: This is a bit eager - non-const begin() is often invoked when
165 : : // no modification is needed, but doing it lazily is a bit tricky as
166 : : // the pointer will change when we COW.
167 [ + + ][ + + ]: 1487064 : if (COW && is_external() && u.b[-1] > 0) {
[ + + ]
168 : 12 : do_cow();
169 : : }
170 [ + + ]: 1487064 : return is_external() ? u.b : u.v;
171 : : }
172 : :
173 : 743788 : iterator end() {
174 [ + + ][ - + ]: 743788 : if (COW && is_external() && u.b[-1] > 0) {
[ - + ]
175 : 0 : do_cow();
176 : : }
177 [ + + ]: 743788 : return is_external() ? u.e : u.v + c;
178 : : }
179 : :
180 : 7729062 : void push_back(T elt) {
181 : 7729062 : auto cap = capacity();
182 [ + + ]: 7729062 : if (size() == cap) {
183 : 92285 : do_reserve(cap * 2);
184 : : }
185 [ + + ]: 7729062 : if (c >= INTERNAL_CAPACITY) {
186 [ - + ]: 567582 : if (COW && u.b[-1] > 0)
187 : 0 : do_cow();
188 : 567582 : *(u.e++) = elt;
189 : : } else {
190 : 7161480 : u.v[c++] = elt;
191 : : }
192 : 7729062 : }
193 : :
194 : 5 : void pop_back() {
195 [ - + ]: 5 : if (is_external()) {
196 : 0 : --u.e;
197 : : } else {
198 : 5 : --c;
199 : : }
200 : 5 : }
201 : :
202 : 16365863 : void clear() {
203 [ + + ]: 16365863 : if (is_external())
204 : 140649 : do_free();
205 : 16365863 : c = 0;
206 : 16365863 : }
207 : :
208 : 4 : void erase(const_iterator it) {
209 : 4 : T* p = const_cast<T*>(it);
210 : 4 : std::memmove(p, p + 1, (end() - it - 1) * sizeof(T));
211 [ - + ]: 4 : if (is_external()) {
212 : 0 : --u.e;
213 : : } else {
214 : 4 : --c;
215 : : }
216 : 4 : }
217 : :
218 : 3 : void erase(const_iterator b, const_iterator e) {
219 : 3 : auto n_erased = e - b;
220 [ - + ]: 3 : if (n_erased == 0) return;
221 : 3 : std::memmove(const_cast<T*>(b), const_cast<T*>(e),
222 : 3 : (end() - e) * sizeof(T));
223 [ + - ]: 3 : if (is_external()) {
224 : 3 : u.e -= n_erased;
225 : : } else {
226 : 0 : c -= n_erased;
227 : : }
228 : : }
229 : :
230 : : void insert(const_iterator pos, const T& elt) {
231 : : push_back(T());
232 : : T* p = const_cast<T*>(end());
233 : : while (--p != pos) {
234 : : *p = p[-1];
235 : : }
236 : : *(const_cast<T*>(pos)) = elt;
237 : : }
238 : :
239 : 4284923 : const T& operator[](size_type idx) const {
240 : 4284923 : return begin()[idx];
241 : : }
242 : :
243 : : T& operator[](size_type idx) {
244 : : if (COW && is_external() && u.b[-1] > 0) {
245 : : do_cow();
246 : : }
247 : : return const_cast<T&>(begin()[idx]);
248 : : }
249 : :
250 : : const T& front() const {
251 : : return *(begin());
252 : : }
253 : :
254 : 2984938 : const T& back() const {
255 : 2984938 : return end()[-1];
256 : : }
257 : :
258 : : protected:
259 : 306665 : void do_free() {
260 [ + + ]: 306665 : if (!COW || u.b[-1] == 0)
261 [ + - ]: 281661 : delete [] (u.b - COW);
262 : : else
263 : 25004 : --u.b[-1];
264 : 306665 : }
265 : :
266 : 281649 : void do_reserve(size_type n) {
267 : : // Logic error or size_t wrapping.
268 [ - + ]: 281649 : if (rare(COW ? n < c : n <= c))
269 : 0 : throw std::bad_alloc();
270 [ + - ]: 281649 : T* blk = new T[n + COW];
271 : : if (COW)
272 : 281649 : *blk++ = 0;
273 [ + + ]: 281649 : if (is_external()) {
274 : 166016 : u.e = std::copy(u.b, u.e, blk);
275 : 166016 : do_free();
276 : : } else {
277 : 115633 : u.e = std::copy(u.v, u.v + c, blk);
278 : : }
279 : 281649 : u.b = blk;
280 : 281649 : c = n;
281 : 281649 : }
282 : :
283 : 12 : void do_cow() {
284 [ + - ]: 12 : T* blk = new T[c + 1];
285 : 12 : *blk++ = 0;
286 : 12 : u.e = std::copy(u.b, u.e, blk);
287 : 12 : --u.b[-1];
288 : 12 : u.b = blk;
289 : 12 : }
290 : :
291 : 1565245 : void do_copy_from(const Vec& o) {
292 [ + + ]: 1565245 : if (!o.is_external()) {
293 : 1540229 : u = o.u;
294 : 1565245 : c = o.c;
295 : : } else if (COW) {
296 : 25016 : u = o.u;
297 : 25016 : c = o.c;
298 : 25016 : ++u.b[-1];
299 : : } else {
300 : : T* blk = new T[o.c];
301 : : u.e = std::copy(o.u.b, o.u.e, blk);
302 : : u.b = blk;
303 : : c = o.c;
304 : : }
305 : 1565245 : }
306 : :
307 : : /// Return true if storage is external to the object.
308 : 66044215 : bool is_external() const noexcept {
309 : 66044215 : return c > INTERNAL_CAPACITY;
310 : : }
311 : : };
312 : :
313 : : template<typename T>
314 : : using VecCOW = Vec<T, true>;
315 : :
316 : : class SmallVector_ {
317 : : std::size_t c;
318 : :
319 : : static constexpr std::size_t INTERNAL_CAPACITY = 2;
320 : :
321 : : void * p[INTERNAL_CAPACITY];
322 : :
323 : : public:
324 : : SmallVector_() : c(0) { }
325 : :
326 : : // Prevent inadvertent copying.
327 : : SmallVector_(const SmallVector_&) = delete;
328 : :
329 : : // Prevent inadvertent copying.
330 : : void operator=(const SmallVector_&) = delete;
331 : :
332 : : SmallVector_(SmallVector_&& o) noexcept : SmallVector_() {
333 : : std::memcpy(p, o.p, sizeof(p));
334 : : std::swap(c, o.c);
335 : : }
336 : :
337 : 123500 : explicit SmallVector_(std::size_t n) : c(0) {
338 : 123500 : reserve(n);
339 : 123500 : }
340 : :
341 : 10326547 : std::size_t size() const {
342 [ + + ]: 10326547 : if (!is_external())
343 : 10232049 : return c;
344 : 94498 : void * const * b = static_cast<void * const *>(p[0]);
345 : 94498 : void * const * e = static_cast<void * const *>(p[1]);
346 : 94498 : return e - b;
347 : : }
348 : :
349 : 302227 : std::size_t capacity() const {
350 [ + + ]: 302227 : return is_external() ? c : INTERNAL_CAPACITY;
351 : : }
352 : :
353 : 122859 : bool empty() const {
354 [ + + ]: 122859 : return is_external() ? p[0] == p[1] : c == 0;
355 : : }
356 : :
357 : 123538 : void reserve(std::size_t n) {
358 [ + + ][ + - ]: 123538 : if (n > INTERNAL_CAPACITY && n > c) {
359 : 1409 : do_reserve(n);
360 : : }
361 : 123538 : }
362 : :
363 : : protected:
364 : : void do_free();
365 : :
366 : : void do_reserve(std::size_t n);
367 : :
368 : 123554 : void do_clear() {
369 [ + + ]: 123554 : if (is_external())
370 : 1637 : do_free();
371 : 123554 : c = 0;
372 : 123554 : }
373 : :
374 : : /// Return true if storage is external to the object.
375 : 23632875 : bool is_external() const {
376 : 23632875 : return c > INTERNAL_CAPACITY;
377 : : }
378 : :
379 : 10749216 : void * const * do_begin() const {
380 [ + + ]: 10749216 : return is_external() ? static_cast<void * const *>(p[0]) : p;
381 : : }
382 : :
383 : 2006682 : void * const * do_end() const {
384 [ + + ]: 2006682 : return is_external() ? static_cast<void * const *>(p[1]) : p + c;
385 : : }
386 : :
387 : 302227 : void do_push_back(void* elt) {
388 : 302227 : std::size_t cap = capacity();
389 [ + + ]: 302227 : if (size() == cap) {
390 : 381 : do_reserve(cap * 2);
391 : : }
392 [ + + ]: 302227 : if (c >= INTERNAL_CAPACITY) {
393 : 89554 : void ** e = static_cast<void **>(p[1]);
394 : 89554 : *e++ = elt;
395 : 89554 : p[1] = static_cast<void*>(e);
396 : : } else {
397 : 212673 : p[c++] = elt;
398 : : }
399 : 302227 : }
400 : : };
401 : :
402 : : /** Vector of Xapian PIMPL internal objects.
403 : : *
404 : : * A notable feature is that if the vector holds <= 2 objects, there's no
405 : : * extra storage - the two internal pointers are held in the two
406 : : * pointers which otherwise point to the first and just after the last
407 : : * element.
408 : : *
409 : : * This means that for the fairly common cases of pair-wise Query operators
410 : : * and Database objects with one or two subdatabases, we use less space than
411 : : * std::vector<Xapian::Foo::Internal*> would.
412 : : */
413 : : template<typename TI>
414 : : class SmallVectorI : public SmallVector_ {
415 : : public:
416 : : typedef std::size_t size_type;
417 : :
418 : : typedef TI*const* const_iterator;
419 : :
420 : : // Create an empty SmallVectorI.
421 : : SmallVectorI() : SmallVector_() { }
422 : :
423 : : // Create an empty SmallVectorI with n elements reserved.
424 : 247000 : explicit SmallVectorI(size_type n) : SmallVector_(n) { }
425 : :
426 : : SmallVectorI(SmallVectorI&& o) noexcept : SmallVector_(o) { }
427 : :
428 : 123500 : ~SmallVectorI() {
429 : 123500 : clear();
430 : 123500 : }
431 : :
432 : 10749216 : const_iterator begin() const {
433 : 10749216 : return const_iterator(do_begin());
434 : : }
435 : :
436 : 2006682 : const_iterator end() const {
437 : 2006682 : return const_iterator(do_end());
438 : : }
439 : :
440 : 123554 : void clear() {
441 [ + + ]: 425781 : for (const_iterator i = begin(); i != end(); ++i)
442 [ + + ][ + + ]: 302227 : if ((*i) && --(*i)->_refs == 0)
[ + + ]
443 [ + - ]: 229951 : delete *i;
444 : :
445 : 123554 : do_clear();
446 : 123554 : }
447 : :
448 : 302227 : void push_back(TI* elt) {
449 : : // Cast away potential const-ness in TI. We can only try to modify an
450 : : // element after casting to TI*, so this is const-safe overall.
451 : 302227 : do_push_back(const_cast<void*>(static_cast<const void*>(elt)));
452 [ + + ]: 302227 : if (elt)
453 : 302167 : ++elt->_refs;
454 : 302227 : }
455 : :
456 : 9784098 : TI* operator[](size_type idx) const {
457 : 9784098 : return begin()[idx];
458 : : }
459 : :
460 : : TI* front() const {
461 : : return *(begin());
462 : : }
463 : :
464 : : TI* back() const {
465 : : return end()[-1];
466 : : }
467 : : };
468 : :
469 : : /** Vector of Xapian PIMPL objects.
470 : : *
471 : : * A notable feature is that if the vector holds <= 2 objects, there's no
472 : : * extra storage - the two internal pointers are held in the two
473 : : * pointers which otherwise point to the first and just after the last
474 : : * element.
475 : : *
476 : : * This means that for the fairly common cases of pair-wise Query operators
477 : : * and Database objects with one or two subdatabases, we use less space than
478 : : * std::vector<Xapian::Foo> would.
479 : : */
480 : : template<typename T>
481 : 245632 : class SmallVector : public SmallVectorI<typename T::Internal> {
482 : : typedef SmallVectorI<typename T::Internal> super;
483 : :
484 : : public:
485 : : typedef std::size_t size_type;
486 : :
487 : : class const_iterator {
488 : : typename super::const_iterator ptr;
489 : :
490 : : public:
491 : 386948 : const_iterator() : ptr(nullptr) { }
492 : :
493 : 1768544 : explicit const_iterator(typename super::const_iterator ptr_)
494 : 1768544 : : ptr(ptr_) { }
495 : :
496 : 1595744 : const_iterator & operator++() { ++ptr; return *this; }
497 : :
498 : : const_iterator operator++(int) { return const_iterator(ptr++); }
499 : :
500 : 800325 : T operator*() const {
501 : 800325 : return T(*ptr);
502 : : }
503 : :
504 : 55522 : T operator[](size_type idx) const {
505 : 55522 : return T(ptr[idx]);
506 : : }
507 : :
508 : 2372110 : bool operator==(const const_iterator& o) const { return ptr == o.ptr; }
509 : :
510 : 2372110 : bool operator!=(const const_iterator& o) const { return !(*this == o); }
511 : :
512 : 281150 : const_iterator operator+(int n) { return const_iterator(ptr + n); }
513 : :
514 : : const_iterator operator-(int n) { return const_iterator(ptr - n); }
515 : : };
516 : :
517 : : // Create an empty SmallVector.
518 : : SmallVector() { }
519 : :
520 : : // Create an empty SmallVector with n elements reserved.
521 : 245632 : explicit SmallVector(size_type n) : super(n) { }
522 : :
523 : 444316 : const_iterator begin() const {
524 : 444316 : return const_iterator(super::begin());
525 : : }
526 : :
527 : 1183653 : const_iterator end() const {
528 : 1183653 : return const_iterator(super::end());
529 : : }
530 : :
531 : : using super::push_back;
532 : :
533 : 300801 : void push_back(const T & elt) {
534 : 300801 : push_back(elt.internal.get());
535 : 300801 : }
536 : :
537 : 55522 : T operator[](size_type idx) const {
538 [ + - ]: 55522 : return begin()[idx];
539 : : }
540 : :
541 : 611 : T front() const {
542 [ + - ]: 611 : return *(begin());
543 : : }
544 : :
545 : : T back() const {
546 : : return end()[-1];
547 : : }
548 : : };
549 : :
550 : : }
551 : :
552 : : #endif // XAPIAN_INCLUDED_SMALLVECTOR_H
|
