Package org.apache.cassandra.index.sai.disk.v1.vector

Class DiskAnn

java.lang.Object

org.apache.cassandra.index.sai.disk.v1.vector.DiskAnn

All Implemented Interfaces:

AutoCloseable

public class DiskAnn
extends Object
implements AutoCloseable

Constructor Summary

Constructors

Constructor	Description
DiskAnn(SegmentMetadata.ComponentMetadataMap componentMetadatas, PerColumnIndexFiles indexFiles, IndexWriterConfig config, SSTableId sstableId)

Method Summary

Modifier and Type	Method	Description
void	close()
io.github.jbellis.jvector.graph.NeighborSimilarity.ApproximateScoreFunction	getApproximateScoreFunction(float[] queryVector)
io.github.jbellis.jvector.pq.CompressedVectors	getCompressedVectors()
io.github.jbellis.jvector.graph.NeighborSimilarity.ExactScoreFunction	getExactScoreFunction(float[] queryVector, io.github.jbellis.jvector.graph.GraphIndex.View<float[]> view)
OnDiskOrdinalsMap.OrdinalsView	getOrdinalsView()
io.github.jbellis.jvector.graph.GraphIndex.View<float[]>	getView()	Get the graph view, callers must close the view.
long	ramBytesUsed()
CloseableIterator<RowIdWithScore>	search(float[] queryVector, int topK, int limit, io.github.jbellis.jvector.util.Bits acceptBits, IntConsumer nodesVisitedConsumer)
int	size()

Methods inherited from class java.lang.Object

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

Constructor Details

DiskAnn

public DiskAnn(SegmentMetadata.ComponentMetadataMap componentMetadatas,
 PerColumnIndexFiles indexFiles,
 IndexWriterConfig config,
 SSTableId sstableId)
        throws IOException

Throws:

IOException

Method Details

ramBytesUsed

public long ramBytesUsed()

size

public int size()

getCompressedVectors

public io.github.jbellis.jvector.pq.CompressedVectors getCompressedVectors()

search

public CloseableIterator<RowIdWithScore> search(float[] queryVector,
 int topK,
 int limit,
 io.github.jbellis.jvector.util.Bits acceptBits,
 IntConsumer nodesVisitedConsumer)

Returns:

Row IDs associated with the topK vectors near the query

getApproximateScoreFunction

public io.github.jbellis.jvector.graph.NeighborSimilarity.ApproximateScoreFunction getApproximateScoreFunction(float[] queryVector)

getExactScoreFunction

public io.github.jbellis.jvector.graph.NeighborSimilarity.ExactScoreFunction getExactScoreFunction(float[] queryVector,
 io.github.jbellis.jvector.graph.GraphIndex.View<float[]> view)

close

public void close()
           throws IOException

Specified by:

close in interface AutoCloseable

Throws:

IOException

getOrdinalsView

public OnDiskOrdinalsMap.OrdinalsView getOrdinalsView()

getView

public io.github.jbellis.jvector.graph.GraphIndex.View<float[]> getView()

Get the graph view, callers must close the view.

Returns: