mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-05 18:58:45 +02:00
102 lines
3.3 KiB
Java
102 lines
3.3 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 IndexIDMap extends Index {
|
||
|
private transient long swigCPtr;
|
||
|
|
||
|
protected IndexIDMap(long cPtr, boolean cMemoryOwn) {
|
||
|
super(swigfaissJNI.IndexIDMap_SWIGUpcast(cPtr), cMemoryOwn);
|
||
|
swigCPtr = cPtr;
|
||
|
}
|
||
|
|
||
|
protected static long getCPtr(IndexIDMap 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_IndexIDMap(swigCPtr);
|
||
|
}
|
||
|
swigCPtr = 0;
|
||
|
}
|
||
|
super.delete();
|
||
|
}
|
||
|
|
||
|
public void setIndex(Index value) {
|
||
|
swigfaissJNI.IndexIDMap_index_set(swigCPtr, this, Index.getCPtr(value), value);
|
||
|
}
|
||
|
|
||
|
public Index getIndex() {
|
||
|
long cPtr = swigfaissJNI.IndexIDMap_index_get(swigCPtr, this);
|
||
|
return (cPtr == 0) ? null : new Index(cPtr, false);
|
||
|
}
|
||
|
|
||
|
public void setOwn_fields(boolean value) {
|
||
|
swigfaissJNI.IndexIDMap_own_fields_set(swigCPtr, this, value);
|
||
|
}
|
||
|
|
||
|
public boolean getOwn_fields() {
|
||
|
return swigfaissJNI.IndexIDMap_own_fields_get(swigCPtr, this);
|
||
|
}
|
||
|
|
||
|
public void setId_map(SWIGTYPE_p_std__vectorT_int64_t_t value) {
|
||
|
swigfaissJNI.IndexIDMap_id_map_set(swigCPtr, this, SWIGTYPE_p_std__vectorT_int64_t_t.getCPtr(value));
|
||
|
}
|
||
|
|
||
|
public SWIGTYPE_p_std__vectorT_int64_t_t getId_map() {
|
||
|
long cPtr = swigfaissJNI.IndexIDMap_id_map_get(swigCPtr, this);
|
||
|
return (cPtr == 0) ? null : new SWIGTYPE_p_std__vectorT_int64_t_t(cPtr, false);
|
||
|
}
|
||
|
|
||
|
public IndexIDMap(Index index) {
|
||
|
this(swigfaissJNI.new_IndexIDMap__SWIG_0(Index.getCPtr(index), index), true);
|
||
|
}
|
||
|
|
||
|
public void add_with_ids(long n, SWIGTYPE_p_float x, LongVector xids) {
|
||
|
swigfaissJNI.IndexIDMap_add_with_ids(swigCPtr, this, n, SWIGTYPE_p_float.getCPtr(x), SWIGTYPE_p_long_long.getCPtr(xids.data()), xids);
|
||
|
}
|
||
|
|
||
|
public void add(long n, SWIGTYPE_p_float x) {
|
||
|
swigfaissJNI.IndexIDMap_add(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.IndexIDMap_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 train(long n, SWIGTYPE_p_float x) {
|
||
|
swigfaissJNI.IndexIDMap_train(swigCPtr, this, n, SWIGTYPE_p_float.getCPtr(x));
|
||
|
}
|
||
|
|
||
|
public void reset() {
|
||
|
swigfaissJNI.IndexIDMap_reset(swigCPtr, this);
|
||
|
}
|
||
|
|
||
|
public long remove_ids(IDSelector sel) {
|
||
|
return swigfaissJNI.IndexIDMap_remove_ids(swigCPtr, this, IDSelector.getCPtr(sel), sel);
|
||
|
}
|
||
|
|
||
|
public void range_search(long n, SWIGTYPE_p_float x, float radius, RangeSearchResult result) {
|
||
|
swigfaissJNI.IndexIDMap_range_search(swigCPtr, this, n, SWIGTYPE_p_float.getCPtr(x), radius, RangeSearchResult.getCPtr(result), result);
|
||
|
}
|
||
|
|
||
|
public IndexIDMap() {
|
||
|
this(swigfaissJNI.new_IndexIDMap__SWIG_1(), true);
|
||
|
}
|
||
|
|
||
|
}
|