diff options
author | Arne Juul <arnej@verizonmedia.com> | 2020-06-10 12:17:45 +0000 |
---|---|---|
committer | Arne Juul <arnej@verizonmedia.com> | 2020-06-11 09:00:34 +0000 |
commit | c939ee3192354a3fbd1114a04027d013218782c7 (patch) | |
tree | fef639bbb1eaeffbe568160c0019514640b83a0e /searchlib | |
parent | c03d44aec3ca14517c8e5caba966fe9ec8c8c720 (diff) |
rename internal prepare result
Diffstat (limited to 'searchlib')
-rw-r--r-- | searchlib/src/vespa/searchlib/tensor/hnsw_index.cpp | 14 | ||||
-rw-r--r-- | searchlib/src/vespa/searchlib/tensor/hnsw_index.h | 12 |
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); |