binary param related funcs

Author: metonymic-smokey

c_api/IndexBinary_c.cpp

@@ -32,6 +32,9 @@ DEFINE_SETTER(IndexBinary, int, verbose);
 DEFINE_GETTER(IndexBinaryIVF, size_t, nprobe);
 DEFINE_SETTER(IndexBinaryIVF, size_t, nprobe);
 
+DEFINE_GETTER(IndexBinaryIVF, size_t, nlist);
+DEFINE_SETTER(IndexBinaryIVF, size_t, nlist);
+
 DEFINE_INDEXBINARY_DOWNCAST(IndexBinaryIVF)
 
 int faiss_IndexBinaryIVF_set_direct_map(
@@ -86,6 +89,26 @@ int faiss_IndexBinary_search(
     CATCH_AND_HANDLE
 }
 
+int faiss_IndexBinary_search_with_params(
+        const FaissIndexBinary* index,
+        idx_t n,
+        const uint8_t* x,
+        idx_t k,
+        const FaissSearchParameters* params,
+        int32_t* distances,
+        idx_t* labels) {
+    try {
+        reinterpret_cast<const faiss::IndexBinary*>(index)->search(
+                n,
+                x,
+                k,
+                distances,
+                labels,
+                reinterpret_cast<const faiss::SearchParameters*>(params));
+    }
+    CATCH_AND_HANDLE
+}
+
 int faiss_IndexBinary_range_search(
         const FaissIndexBinary* index,
         idx_t n,

c_api/IndexBinary_c.h

@@ -47,6 +47,8 @@ FAISS_DECLARE_GETTER(IndexBinary, FaissMetricType, metric_type)
 
 FAISS_DECLARE_GETTER_SETTER(IndexBinary, int, verbose)
 
+FAISS_DECLARE_GETTER_SETTER(IndexBinaryIVF, size_t, nlist)
+
 FAISS_DECLARE_GETTER_SETTER(IndexBinaryIVF, size_t, nprobe)
 
 /** Perform training on a representative set of vectors
@@ -99,6 +101,27 @@ int faiss_IndexBinary_search(
         int32_t* distances,
         idx_t* labels);
 
+/** query n vectors of dimension d to the index.
+ *
+ * return at most k vectors. If there are not enough results for a
+ * query, the result array is padded with -1s.
+ *
+ * @param index       opaque pointer to index object
+ * @param x           input vectors to search, size n * d
+ * @param k           number of results to return
+ * @param params      search parameters
+ * @param distances   output distances, size n*k
+ * @param labels      output labels of the NNs, size n*k
+ */
+int faiss_IndexBinary_search_with_params(
+        const FaissIndexBinary* index,
+        idx_t n,
+        const uint8_t* x,
+        idx_t k,
+        const FaissSearchParameters* params,
+        int32_t* distances,
+        idx_t* labels);
+
 /** query n vectors of dimension d to the index.
  *
  * return all vectors with distance < radius. Note that many