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
48
49

use faiss_next_sys::{self, FaissIndex};

use crate::error::{Error, Result};
use crate::index::native::InnerPtr;
use crate::index::traits::Index;

pub struct IndexImpl {
    inner: InnerPtr<FaissIndex>,
}

impl IndexImpl {
    pub(crate) fn from_raw(ptr: *mut FaissIndex) -> Result<Self> {
        Ok(Self {
            inner: InnerPtr::new(ptr)?,
        })
    }

    pub fn into_flat(self) -> Result<super::IndexFlat> {
        unsafe {
            let flat_ptr = faiss_next_sys::faiss_IndexFlat_cast(self.inner.as_ptr());
            if flat_ptr.is_null() {
                return Err(Error::invalid_cast(
                    "IndexFlat",
                    "index is not a flat index",
                ));
            }
            std::mem::forget(self);
            Ok(super::IndexFlat::from_raw(flat_ptr))
        }
    }
}

impl Index for IndexImpl {
    fn inner_ptr(&self) -> *mut FaissIndex {
        self.inner.as_ptr()
    }
}

impl Drop for IndexImpl {
    fn drop(&mut self) {
        tracing::trace!("dropping IndexImpl");
        unsafe {
            faiss_next_sys::faiss_Index_free(self.inner.as_ptr());
        }
    }
}

unsafe impl Send for IndexImpl {}
unsafe impl Sync for IndexImpl {}