summaryrefslogtreecommitdiffstats
path: root/searchlib
diff options
context:
space:
mode:
authorArne Juul <arnej@verizonmedia.com>2020-06-10 12:17:45 +0000
committerArne Juul <arnej@verizonmedia.com>2020-06-11 09:00:34 +0000
commitc939ee3192354a3fbd1114a04027d013218782c7 (patch)
treefef639bbb1eaeffbe568160c0019514640b83a0e /searchlib
parentc03d44aec3ca14517c8e5caba966fe9ec8c8c720 (diff)
rename internal prepare result
Diffstat (limited to 'searchlib')
-rw-r--r--searchlib/src/vespa/searchlib/tensor/hnsw_index.cpp14
-rw-r--r--searchlib/src/vespa/searchlib/tensor/hnsw_index.h12
2 files changed, 13 insertions, 13 deletions
diff --git a/searchlib/src/vespa/searchlib/tensor/hnsw_index.cpp b/searchlib/src/vespa/searchlib/tensor/hnsw_index.cpp
index 2b908dd2bd4..a03614a785e 100644
--- a/searchlib/src/vespa/searchlib/tensor/hnsw_index.cpp
+++ b/searchlib/src/vespa/searchlib/tensor/hnsw_index.cpp
@@ -275,16 +275,16 @@ HnswIndex::~HnswIndex() = default;
void
HnswIndex::add_document(uint32_t docid)
{
- AddDocOperation op = internal_prepare_add(docid, get_vector(docid));
+ PreparedAddDoc op = internal_prepare_add(docid, get_vector(docid));
internal_complete_add(docid, op);
}
-HnswIndex::AddDocOperation
+HnswIndex::PreparedAddDoc
HnswIndex::internal_prepare_add(uint32_t docid, TypedCells input_vector) const
{
// TODO: Add capping on num_levels
int level = _level_generator->max_level();
- AddDocOperation op(docid, level);
+ PreparedAddDoc op(docid, level);
auto entry = _graph.get_entry_node();
if (entry.docid == 0) {
return op;
@@ -327,7 +327,7 @@ HnswIndex::filter_valid_docids(const LinkArrayRef &docids)
}
void
-HnswIndex::internal_complete_add(uint32_t docid, AddDocOperation &op)
+HnswIndex::internal_complete_add(uint32_t docid, PreparedAddDoc &op)
{
_graph.make_node_for_document(docid, op.max_level + 1);
for (int level = 0; level <= op.max_level; ++level) {
@@ -344,15 +344,15 @@ HnswIndex::prepare_add_document(uint32_t docid,
TypedCells vector,
vespalib::GenerationHandler::Guard read_guard) const
{
- AddDocOperation op = internal_prepare_add(docid, vector);
+ PreparedAddDoc op = internal_prepare_add(docid, vector);
(void) read_guard; // must keep guard until this point
- return std::make_unique<AddDocOperation>(std::move(op));
+ return std::make_unique<PreparedAddDoc>(std::move(op));
}
void
HnswIndex::complete_add_document(uint32_t docid, std::unique_ptr<PrepareResult> prepare_result)
{
- auto prepared = dynamic_cast<AddDocOperation *>(prepare_result.get());
+ auto prepared = dynamic_cast<PreparedAddDoc *>(prepare_result.get());
if (prepared && (prepared->docid == docid)) {
internal_complete_add(docid, *prepared);
} else {
diff --git a/searchlib/src/vespa/searchlib/tensor/hnsw_index.h b/searchlib/src/vespa/searchlib/tensor/hnsw_index.h
index 47712f56c18..df7453023cf 100644
--- a/searchlib/src/vespa/searchlib/tensor/hnsw_index.h
+++ b/searchlib/src/vespa/searchlib/tensor/hnsw_index.h
@@ -120,19 +120,19 @@ protected:
std::vector<Neighbor> top_k_by_docid(uint32_t k, TypedCells vector,
const BitVector *filter, uint32_t explore_k) const;
- struct AddDocOperation : public PrepareResult {
+ struct PreparedAddDoc : public PrepareResult {
uint32_t docid;
int32_t max_level;
std::vector<LinkArray> connections;
- AddDocOperation(uint32_t docid_in, int32_t max_level_in)
+ PreparedAddDoc(uint32_t docid_in, int32_t max_level_in)
: docid(docid_in), max_level(max_level_in), connections(max_level+1)
{}
- ~AddDocOperation() = default;
- AddDocOperation(AddDocOperation&& other) = default;
+ ~PreparedAddDoc() = default;
+ PreparedAddDoc(PreparedAddDoc&& other) = default;
};
- AddDocOperation internal_prepare_add(uint32_t docid, TypedCells input_vector) const;
+ PreparedAddDoc internal_prepare_add(uint32_t docid, TypedCells input_vector) const;
LinkArray filter_valid_docids(const LinkArrayRef &docids);
- void internal_complete_add(uint32_t docid, AddDocOperation &op);
+ void internal_complete_add(uint32_t docid, PreparedAddDoc &op);
public:
HnswIndex(const DocVectorAccess& vectors, DistanceFunction::UP distance_func,
RandomLevelGenerator::UP level_generator, const Config& cfg);