151 lines
5.8 KiB
Java
151 lines
5.8 KiB
Java
/* ----------------------------------------------------------------------------
|
|
* This file was automatically generated by SWIG (http://www.swig.org).
|
|
* Version 4.0.2
|
|
*
|
|
* Do not make changes to this file unless you know what you are doing--modify
|
|
* the SWIG interface file instead.
|
|
* ----------------------------------------------------------------------------- */
|
|
|
|
package com.twitter.ann.faiss;
|
|
|
|
public class IndexHNSW extends Index {
|
|
private transient long swigCPtr;
|
|
|
|
protected IndexHNSW(long cPtr, boolean cMemoryOwn) {
|
|
super(swigfaissJNI.IndexHNSW_SWIGUpcast(cPtr), cMemoryOwn);
|
|
swigCPtr = cPtr;
|
|
}
|
|
|
|
protected static long getCPtr(IndexHNSW obj) {
|
|
return (obj == null) ? 0 : obj.swigCPtr;
|
|
}
|
|
|
|
@SuppressWarnings("deprecation")
|
|
protected void finalize() {
|
|
delete();
|
|
}
|
|
|
|
public synchronized void delete() {
|
|
if (swigCPtr != 0) {
|
|
if (swigCMemOwn) {
|
|
swigCMemOwn = false;
|
|
swigfaissJNI.delete_IndexHNSW(swigCPtr);
|
|
}
|
|
swigCPtr = 0;
|
|
}
|
|
super.delete();
|
|
}
|
|
|
|
public void setHnsw(HNSW value) {
|
|
swigfaissJNI.IndexHNSW_hnsw_set(swigCPtr, this, HNSW.getCPtr(value), value);
|
|
}
|
|
|
|
public HNSW getHnsw() {
|
|
long cPtr = swigfaissJNI.IndexHNSW_hnsw_get(swigCPtr, this);
|
|
return (cPtr == 0) ? null : new HNSW(cPtr, false);
|
|
}
|
|
|
|
public void setOwn_fields(boolean value) {
|
|
swigfaissJNI.IndexHNSW_own_fields_set(swigCPtr, this, value);
|
|
}
|
|
|
|
public boolean getOwn_fields() {
|
|
return swigfaissJNI.IndexHNSW_own_fields_get(swigCPtr, this);
|
|
}
|
|
|
|
public void setStorage(Index value) {
|
|
swigfaissJNI.IndexHNSW_storage_set(swigCPtr, this, Index.getCPtr(value), value);
|
|
}
|
|
|
|
public Index getStorage() {
|
|
long cPtr = swigfaissJNI.IndexHNSW_storage_get(swigCPtr, this);
|
|
return (cPtr == 0) ? null : new Index(cPtr, false);
|
|
}
|
|
|
|
public void setReconstruct_from_neighbors(ReconstructFromNeighbors value) {
|
|
swigfaissJNI.IndexHNSW_reconstruct_from_neighbors_set(swigCPtr, this, ReconstructFromNeighbors.getCPtr(value), value);
|
|
}
|
|
|
|
public ReconstructFromNeighbors getReconstruct_from_neighbors() {
|
|
long cPtr = swigfaissJNI.IndexHNSW_reconstruct_from_neighbors_get(swigCPtr, this);
|
|
return (cPtr == 0) ? null : new ReconstructFromNeighbors(cPtr, false);
|
|
}
|
|
|
|
public IndexHNSW(int d, int M, MetricType metric) {
|
|
this(swigfaissJNI.new_IndexHNSW__SWIG_0(d, M, metric.swigValue()), true);
|
|
}
|
|
|
|
public IndexHNSW(int d, int M) {
|
|
this(swigfaissJNI.new_IndexHNSW__SWIG_1(d, M), true);
|
|
}
|
|
|
|
public IndexHNSW(int d) {
|
|
this(swigfaissJNI.new_IndexHNSW__SWIG_2(d), true);
|
|
}
|
|
|
|
public IndexHNSW() {
|
|
this(swigfaissJNI.new_IndexHNSW__SWIG_3(), true);
|
|
}
|
|
|
|
public IndexHNSW(Index storage, int M) {
|
|
this(swigfaissJNI.new_IndexHNSW__SWIG_4(Index.getCPtr(storage), storage, M), true);
|
|
}
|
|
|
|
public IndexHNSW(Index storage) {
|
|
this(swigfaissJNI.new_IndexHNSW__SWIG_5(Index.getCPtr(storage), storage), true);
|
|
}
|
|
|
|
public void add(long n, SWIGTYPE_p_float x) {
|
|
swigfaissJNI.IndexHNSW_add(swigCPtr, this, n, SWIGTYPE_p_float.getCPtr(x));
|
|
}
|
|
|
|
public void train(long n, SWIGTYPE_p_float x) {
|
|
swigfaissJNI.IndexHNSW_train(swigCPtr, this, n, SWIGTYPE_p_float.getCPtr(x));
|
|
}
|
|
|
|
public void search(long n, SWIGTYPE_p_float x, long k, SWIGTYPE_p_float distances, LongVector labels) {
|
|
swigfaissJNI.IndexHNSW_search(swigCPtr, this, n, SWIGTYPE_p_float.getCPtr(x), k, SWIGTYPE_p_float.getCPtr(distances), SWIGTYPE_p_long_long.getCPtr(labels.data()), labels);
|
|
}
|
|
|
|
public void reconstruct(long key, SWIGTYPE_p_float recons) {
|
|
swigfaissJNI.IndexHNSW_reconstruct(swigCPtr, this, key, SWIGTYPE_p_float.getCPtr(recons));
|
|
}
|
|
|
|
public void reset() {
|
|
swigfaissJNI.IndexHNSW_reset(swigCPtr, this);
|
|
}
|
|
|
|
public void shrink_level_0_neighbors(int size) {
|
|
swigfaissJNI.IndexHNSW_shrink_level_0_neighbors(swigCPtr, this, size);
|
|
}
|
|
|
|
public void search_level_0(long n, SWIGTYPE_p_float x, long k, SWIGTYPE_p_int nearest, SWIGTYPE_p_float nearest_d, SWIGTYPE_p_float distances, LongVector labels, int nprobe, int search_type) {
|
|
swigfaissJNI.IndexHNSW_search_level_0__SWIG_0(swigCPtr, this, n, SWIGTYPE_p_float.getCPtr(x), k, SWIGTYPE_p_int.getCPtr(nearest), SWIGTYPE_p_float.getCPtr(nearest_d), SWIGTYPE_p_float.getCPtr(distances), SWIGTYPE_p_long_long.getCPtr(labels.data()), labels, nprobe, search_type);
|
|
}
|
|
|
|
public void search_level_0(long n, SWIGTYPE_p_float x, long k, SWIGTYPE_p_int nearest, SWIGTYPE_p_float nearest_d, SWIGTYPE_p_float distances, LongVector labels, int nprobe) {
|
|
swigfaissJNI.IndexHNSW_search_level_0__SWIG_1(swigCPtr, this, n, SWIGTYPE_p_float.getCPtr(x), k, SWIGTYPE_p_int.getCPtr(nearest), SWIGTYPE_p_float.getCPtr(nearest_d), SWIGTYPE_p_float.getCPtr(distances), SWIGTYPE_p_long_long.getCPtr(labels.data()), labels, nprobe);
|
|
}
|
|
|
|
public void search_level_0(long n, SWIGTYPE_p_float x, long k, SWIGTYPE_p_int nearest, SWIGTYPE_p_float nearest_d, SWIGTYPE_p_float distances, LongVector labels) {
|
|
swigfaissJNI.IndexHNSW_search_level_0__SWIG_2(swigCPtr, this, n, SWIGTYPE_p_float.getCPtr(x), k, SWIGTYPE_p_int.getCPtr(nearest), SWIGTYPE_p_float.getCPtr(nearest_d), SWIGTYPE_p_float.getCPtr(distances), SWIGTYPE_p_long_long.getCPtr(labels.data()), labels);
|
|
}
|
|
|
|
public void init_level_0_from_knngraph(int k, SWIGTYPE_p_float D, LongVector I) {
|
|
swigfaissJNI.IndexHNSW_init_level_0_from_knngraph(swigCPtr, this, k, SWIGTYPE_p_float.getCPtr(D), SWIGTYPE_p_long_long.getCPtr(I.data()), I);
|
|
}
|
|
|
|
public void init_level_0_from_entry_points(int npt, SWIGTYPE_p_int points, SWIGTYPE_p_int nearests) {
|
|
swigfaissJNI.IndexHNSW_init_level_0_from_entry_points(swigCPtr, this, npt, SWIGTYPE_p_int.getCPtr(points), SWIGTYPE_p_int.getCPtr(nearests));
|
|
}
|
|
|
|
public void reorder_links() {
|
|
swigfaissJNI.IndexHNSW_reorder_links(swigCPtr, this);
|
|
}
|
|
|
|
public void link_singletons() {
|
|
swigfaissJNI.IndexHNSW_link_singletons(swigCPtr, this);
|
|
}
|
|
|
|
}
|