blob: 0a4c34db963817f190d5eb58126b7859714ce51f (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include "field_length_scanner.h"
#include <vespa/searchlib/index/docidandfeatures.h>
namespace search::diskindex {
FieldLengthScanner::FieldLengthScanner(uint32_t doc_id_limit)
: _field_length_vector(doc_id_limit),
_scanned_elements_map()
{
}
FieldLengthScanner::~FieldLengthScanner() = default;
void
FieldLengthScanner::scan_features(const index::DocIdAndFeatures &features)
{
if (features.elements().empty()) {
return;
}
auto &entry = _field_length_vector[features.doc_id()];
if (features.elements().back().getElementId() < element_id_bias) {
for (const auto &element : features.elements()) {
entry.add_element_length(element.getElementLen(), element.getElementId());
}
} else {
auto element = features.elements().cbegin();
while (element->getElementId() < element_id_bias) {
entry.add_element_length(element->getElementLen(), element->getElementId());
++element;
}
auto &scanned_elements = _scanned_elements_map[features.doc_id()];
auto size_needed = features.elements().back().getElementId() + 1 - element_id_bias;
if (size_needed > scanned_elements.size()) {
if (size_needed > scanned_elements.capacity()) {
scanned_elements.reserve(std::max(size_needed + (size_needed / 4), 32u));
}
scanned_elements.resize(size_needed);
}
while (element != features.elements().cend()) {
if (!scanned_elements[element->getElementId() - element_id_bias]) {
scanned_elements[element->getElementId() - element_id_bias] = true;
entry.add_element_length(element->getElementLen());
}
++element;
}
}
}
}
|