Merge pull request #58 from ava57r/clustering-1

improve Clustering impl

src/cluster.rs

@@ -52,6 +52,12 @@ impl ClusteringParameters {
         self.inner.spherical != 0
     }
 
+    /// Getter for the `int_centroids` property.
+    /// Round centroids coordinates to integer
+    pub fn int_centroids(&self) -> bool {
+        self.inner.int_centroids != 0
+    }
+
     pub fn update_index(&self) -> bool {
         self.inner.update_index != 0
     }
@@ -64,6 +70,12 @@ impl ClusteringParameters {
         self.inner.seed as u32
     }
 
+    /// Getter for the `decode_block_size` property.
+    /// How many vectors at a time to decode
+    pub fn decode_block_size(&self) -> usize {
+        self.inner.decode_block_size
+    }
+
     pub fn set_niter(&mut self, niter: u32) {
         self.inner.niter = (niter & 0x7FFF_FFFF) as i32;
     }
@@ -92,13 +104,25 @@ impl ClusteringParameters {
         self.inner.spherical = if spherical { 1 } else { 0 };
     }
 
+    /// Setter for the `int_centroids` property.
+    /// Round centroids coordinates to integer
+    pub fn set_int_centroids(&mut self, int_centroids: bool) {
+        self.inner.int_centroids = if int_centroids { 1 } else { 0 }
+    }
+
     pub fn set_verbose(&mut self, verbose: bool) {
         self.inner.verbose = if verbose { 1 } else { 0 };
     }
 
     pub fn set_seed(&mut self, seed: u32) {
         self.inner.seed = seed as i32;
     }
+
+    /// Setter for the `decode_block_size` property.
+    /// How many vectors at a time to decode
+    pub fn set_decode_block_size(&mut self, decode_block_size: usize) {
+        self.inner.decode_block_size = decode_block_size;
+    }
 }
 
 #[repr(C)]
@@ -305,6 +329,12 @@ impl Clustering {
         unsafe { faiss_Clustering_spherical(self.inner) != 0 }
     }
 
+    /// Getter for the `int_centroids` property of `Clustering`.
+    /// Round centroids coordinates to integer
+    pub fn int_centroids(&self) -> bool {
+        unsafe { faiss_Clustering_int_centroids(self.inner) != 0 }
+    }
+
     /** Getter for the `update_index` property of `Clustering`. */
     pub fn update_index(&self) -> bool {
         unsafe { faiss_Clustering_update_index(self.inner) != 0 }
@@ -320,6 +350,12 @@ impl Clustering {
         unsafe { faiss_Clustering_seed(self.inner) as u32 }
     }
 
+    /// Getter for the `decode_block_size` property of `Clustering`.
+    /// How many vectors at a time to decode
+    pub fn decode_block_size(&self) -> usize {
+        unsafe { faiss_Clustering_decode_block_size(self.inner) }
+    }
+
     /** Getter for the minimum number of points per centroid. */
     pub fn min_points_per_centroid(&self) -> u32 {
         unsafe { faiss_Clustering_min_points_per_centroid(self.inner) as u32 }