Module org.elasticsearch.server

Package org.elasticsearch.index.codec.vectors.diskbbq

Class ES920DiskBBQVectorsReader

java.lang.Object

org.apache.lucene.codecs.KnnVectorsReader

org.elasticsearch.index.codec.vectors.diskbbq.IVFVectorsReader<IVFVectorsReader.FieldEntry>

org.elasticsearch.index.codec.vectors.diskbbq.ES920DiskBBQVectorsReader

All Implemented Interfaces:

Closeable, AutoCloseable

public class ES920DiskBBQVectorsReader
extends IVFVectorsReader<IVFVectorsReader.FieldEntry>

Default implementation of IVFVectorsReader. It scores the posting lists centroids using brute force and then scores the top ones using the posting list.

Nested Class Summary

Nested classes/interfaces inherited from class org.elasticsearch.index.codec.vectors.diskbbq.IVFVectorsReader

IVFVectorsReader.FieldEntry, IVFVectorsReader.PostingVisitor

Field Summary

Fields inherited from class org.elasticsearch.index.codec.vectors.diskbbq.IVFVectorsReader

fields, ivfCentroids, ivfClusters, versionMeta

Method Summary

Modifier and Type	Method	Description
protected IVFVectorsReader.FieldEntry	doReadField(org.apache.lucene.store.IndexInput input, String rawVectorFormat, boolean useDirectIOReads, org.apache.lucene.index.VectorSimilarityFunction similarityFunction, org.apache.lucene.index.VectorEncoding vectorEncoding, int numCentroids, long centroidOffset, long centroidLength, long postingListOffset, long postingListLength, float[] globalCentroid, float globalCentroidDp)
CentroidIterator	getCentroidIterator(org.apache.lucene.index.FieldInfo fieldInfo, int numCentroids, org.apache.lucene.store.IndexInput centroids, float[] targetQuery, org.apache.lucene.store.IndexInput postingListSlice, org.apache.lucene.search.AcceptDocs acceptDocs, float approximateCost, org.apache.lucene.index.FloatVectorValues values, float visitRatio)
CentroidIterator	getPostingListPrefetchIterator(CentroidIterator centroidIterator, org.apache.lucene.store.IndexInput postingListSlice)
IVFVectorsReader.PostingVisitor	getPostingVisitor(org.apache.lucene.index.FieldInfo fieldInfo, int numVectors, org.apache.lucene.store.IndexInput indexInput, float[] target, org.apache.lucene.util.Bits acceptDocs, org.apache.lucene.store.IndexInput centroidSlice)

Methods inherited from class org.elasticsearch.index.codec.vectors.diskbbq.IVFVectorsReader

checkIntegrity, close, getByteVectorValues, getFloatVectorValues, getOffHeapByteSize, getReaderForField, openDataInput, search, search

Methods inherited from class org.apache.lucene.codecs.KnnVectorsReader

finishMerge, getMergeInstance, mergeOffHeapByteSizeMaps

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Details

getPostingListPrefetchIterator

public CentroidIterator getPostingListPrefetchIterator(CentroidIterator centroidIterator,
 org.apache.lucene.store.IndexInput postingListSlice)
                                                throws IOException

Throws:

IOException

getCentroidIterator

public CentroidIterator getCentroidIterator(org.apache.lucene.index.FieldInfo fieldInfo,
 int numCentroids,
 org.apache.lucene.store.IndexInput centroids,
 float[] targetQuery,
 org.apache.lucene.store.IndexInput postingListSlice,
 org.apache.lucene.search.AcceptDocs acceptDocs,
 float approximateCost,
 org.apache.lucene.index.FloatVectorValues values,
 float visitRatio)
                                     throws IOException

Specified by:

getCentroidIterator in class IVFVectorsReader<IVFVectorsReader.FieldEntry>

Throws:

IOException

doReadField

protected IVFVectorsReader.FieldEntry doReadField(org.apache.lucene.store.IndexInput input,
 String rawVectorFormat,
 boolean useDirectIOReads,
 org.apache.lucene.index.VectorSimilarityFunction similarityFunction,
 org.apache.lucene.index.VectorEncoding vectorEncoding,
 int numCentroids,
 long centroidOffset,
 long centroidLength,
 long postingListOffset,
 long postingListLength,
 float[] globalCentroid,
 float globalCentroidDp)

Specified by:

doReadField in class IVFVectorsReader<IVFVectorsReader.FieldEntry>

getPostingVisitor

public IVFVectorsReader.PostingVisitor getPostingVisitor(org.apache.lucene.index.FieldInfo fieldInfo,
 int numVectors,
 org.apache.lucene.store.IndexInput indexInput,
 float[] target,
 org.apache.lucene.util.Bits acceptDocs,
 org.apache.lucene.store.IndexInput centroidSlice)
                                                  throws IOException

Specified by:

getPostingVisitor in class IVFVectorsReader<IVFVectorsReader.FieldEntry>

Throws:

IOException