faiss-next 0.6.0

Rust bindings for Faiss (Facebook AI Similarity Search)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

use faiss_next_sys::{self, FaissDistanceComputer};

use crate::error::{check_return_code, Result};

#[allow(dead_code)]
pub struct DistanceComputer {
    ptr: *mut FaissDistanceComputer,
}

impl DistanceComputer {
    #[allow(dead_code)]
    pub(crate) fn new(ptr: *mut FaissDistanceComputer) -> Self {
        Self { ptr }
    }

    pub fn set_query(&mut self, x: &[f32]) -> Result<()> {
        check_return_code(unsafe {
            faiss_next_sys::faiss_DistanceComputer_set_query(self.ptr, x.as_ptr())
        })
    }

    pub fn symmetric_dis(&self, i: i64, j: i64) -> Result<f32> {
        let mut dis = 0.0f32;
        check_return_code(unsafe {
            faiss_next_sys::faiss_DistanceComputer_symmetric_dis(self.ptr, i, j, &mut dis)
        })?;
        Ok(dis)
    }

    pub fn vector_to_query_dis(&self, i: i64) -> Result<f32> {
        let mut dis = 0.0f32;
        check_return_code(unsafe {
            faiss_next_sys::faiss_DistanceComputer_vector_to_query_dis(self.ptr, i, &mut dis)
        })?;
        Ok(dis)
    }
}

impl Drop for DistanceComputer {
    fn drop(&mut self) {
        if !self.ptr.is_null() {
            unsafe {
                faiss_next_sys::faiss_DistanceComputer_free(self.ptr);
            }
        }
    }
}