Add ParameterSpace.SetIndexParameter

faiss.go

@@ -8,6 +8,7 @@ package faiss
 #cgo LDFLAGS: -lfaiss_c
 
 #include <stdlib.h>
+#include <faiss/c_api/AutoTune_c.h>
 #include <faiss/c_api/IndexFlat_c.h>
 #include <faiss/c_api/Index_c.h>
 #include <faiss/c_api/error_c.h>
@@ -26,6 +27,40 @@ func getLastError() error {
 	return errors.New(C.GoString(C.faiss_get_last_error()))
 }
 
+//--------------------------------------------------
+// AutoTune
+//--------------------------------------------------
+
+type ParameterSpace struct {
+	ps *C.FaissParameterSpace
+}
+
+// NewParameterSpace creates a new ParameterSpace.
+func NewParameterSpace() (*ParameterSpace, error) {
+	var ps *C.FaissParameterSpace
+	if c := C.faiss_ParameterSpace_new(&ps); c != 0 {
+		return nil, getLastError()
+	}
+	return &ParameterSpace{ps}, nil
+}
+
+// SetIndexParameter sets one of the parameters.
+func (p *ParameterSpace) SetIndexParameter(idx *Index, name string, val float64) error {
+	cname := C.CString(name)
+	defer C.free(unsafe.Pointer(cname))
+
+	c := C.faiss_ParameterSpace_set_index_parameter(p.ps, idx.idx, cname, C.double(val))
+	if c != 0 {
+		getLastError()
+	}
+	return nil
+}
+
+// Delete frees the memory associated with p.
+func (p *ParameterSpace) Delete() {
+	C.faiss_ParameterSpace_free(p.ps)
+}
+
 //--------------------------------------------------
 // AuxIndexStructures
 //--------------------------------------------------