add fn to get total number of points

microsoft · Aug 15, 2023 · 008f5fc · 008f5fc
1 parent c72d7fb
commit 008f5fc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/include/pq_flash_index.h b/include/pq_flash_index.h
@@ -97,6 +97,7 @@ template <typename T, typename LabelT = uint32_t> class PQFlashIndex
     DISKANN_DLLEXPORT std::vector<T> get_vector(std::uint64_t VID);
     DISKANN_DLLEXPORT std::vector<std::uint8_t> get_pq_vector(std::uint64_t VID);
     DISKANN_DLLEXPORT std::vector<std::uint64_t> get_neighbors(std::uint64_t VID);
+    DISKANN_DLLEXPORT uint64_t get_num_points();
 
   protected:
     DISKANN_DLLEXPORT void use_medoids_data_as_centroids();

diff --git a/src/pq_flash_index.cpp b/src/pq_flash_index.cpp
@@ -1615,6 +1615,12 @@ std::vector<std::uint64_t> PQFlashIndex<T, LabelT>::get_neighbors(std::uint64_t
     return out;
 }
 
+template <typename T, typename LabelT>
+std::uint64_t PQFlashIndex<T, LabelT>::get_num_points()
+{
+    return num_points;
+}
+
 #ifdef EXEC_ENV_OLS
 template <typename T, typename LabelT> char *PQFlashIndex<T, LabelT>::getHeaderBytes()
 {