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

use std::ffi::CString;
use std::ptr;

use crate::error::{check_return_code, Result};
use crate::index::{BinaryIndex, Index, IndexImpl};

pub fn write_index(index: &impl Index, path: &str) -> Result<()> {
    let c_path = CString::new(path)?;
    check_return_code(unsafe {
        faiss_next_sys::faiss_write_index_fname(index.inner_ptr(), c_path.as_ptr())
    })
}

pub fn read_index(path: &str) -> Result<IndexImpl> {
    let c_path = CString::new(path)?;
    let mut inner = ptr::null_mut();
    check_return_code(unsafe {
        faiss_next_sys::faiss_read_index_fname(c_path.as_ptr(), 0, &mut inner)
    })?;
    IndexImpl::from_raw(inner)
}

pub fn write_index_binary(index: &crate::index::IndexBinary, path: &str) -> Result<()> {
    let c_path = CString::new(path)?;
    check_return_code(unsafe {
        faiss_next_sys::faiss_write_index_binary_fname(index.inner_ptr(), c_path.as_ptr())
    })
}

pub fn read_index_binary(path: &str) -> Result<crate::index::IndexBinary> {
    let c_path = CString::new(path)?;
    let mut inner = ptr::null_mut();
    check_return_code(unsafe {
        faiss_next_sys::faiss_read_index_binary_fname(c_path.as_ptr(), 0, &mut inner)
    })?;
    crate::index::IndexBinary::from_raw(inner)
}