Package org.apache.lucene.codecs.lucene94

Class Lucene94HnswVectorsWriter

java.lang.Object

org.apache.lucene.codecs.KnnVectorsWriter

org.apache.lucene.codecs.lucene94.Lucene94HnswVectorsWriter

All Implemented Interfaces:

Closeable, AutoCloseable, Accountable

public final class Lucene94HnswVectorsWriter
extends KnnVectorsWriter

Writes vector values and knn graphs to index segments.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
private static class	Lucene94HnswVectorsWriter.FieldWriter<T>
private static class	Lucene94HnswVectorsWriter.RAVectorValues<T>

Nested classes/interfaces inherited from class org.apache.lucene.codecs.KnnVectorsWriter

KnnVectorsWriter.MergedVectorValues

Field Summary

Fields

Modifier and Type	Field	Description
private final int	beamWidth
private final List<Lucene94HnswVectorsWriter.FieldWriter<?>>	fields
private boolean	finished
private final int	M
private final IndexOutput	meta
private final SegmentWriteState	segmentWriteState
private final IndexOutput	vectorData
private final IndexOutput	vectorIndex

Fields inherited from interface org.apache.lucene.util.Accountable

NULL_ACCOUNTABLE

Constructor Summary

Constructors

Constructor	Description
Lucene94HnswVectorsWriter(SegmentWriteState state, int M, int beamWidth)

Method Summary

Modifier and Type	Method	Description
KnnFieldVectorsWriter<?>	addField(FieldInfo fieldInfo)	Add new field for indexing
void	close()
void	finish()	Called once at the end before close
void	flush(int maxDoc, Sorter.DocMap sortMap)	Flush all buffered data on disk *
void	mergeOneField(FieldInfo fieldInfo, MergeState mergeState)	Write field for merging
long	ramBytesUsed()	Return the memory usage of this object in bytes.
private HnswGraph	reconstructAndWriteGraph(OnHeapHnswGraph graph, int[] newToOldMap, int[] oldToNewMap)
private void	reconstructAndWriteNeigbours(NeighborArray neighbors, int[] oldToNewMap, int maxConnOnLevel, int maxOrd)
private void	writeByteVectors(Lucene94HnswVectorsWriter.FieldWriter<?> fieldData)
private void	writeField(Lucene94HnswVectorsWriter.FieldWriter<?> fieldData, int maxDoc)
private void	writeFloat32Vectors(Lucene94HnswVectorsWriter.FieldWriter<?> fieldData)
private void	writeGraph(OnHeapHnswGraph graph)
private void	writeMeta(FieldInfo field, int maxDoc, long vectorDataOffset, long vectorDataLength, long vectorIndexOffset, long vectorIndexLength, DocsWithFieldSet docsWithField, HnswGraph graph)
private long	writeSortedByteVectors(Lucene94HnswVectorsWriter.FieldWriter<?> fieldData, int[] ordMap)
private long	writeSortedFloat32Vectors(Lucene94HnswVectorsWriter.FieldWriter<?> fieldData, int[] ordMap)
private void	writeSortingField(Lucene94HnswVectorsWriter.FieldWriter<?> fieldData, int maxDoc, Sorter.DocMap sortMap)
private static DocsWithFieldSet	writeVectorData(IndexOutput output, VectorValues vectors, int scalarSize)	Writes the vector values to the output and returns a set of documents that contains vectors.

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

merge

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.apache.lucene.util.Accountable

getChildResources

Field Details

segmentWriteState

private final SegmentWriteState segmentWriteState

meta

private final IndexOutput meta

vectorData

private final IndexOutput vectorData

vectorIndex

private final IndexOutput vectorIndex

M

private final int M

beamWidth

private final int beamWidth

fields

private final List<Lucene94HnswVectorsWriter.FieldWriter<?>> fields

finished

private boolean finished

Constructor Details

Lucene94HnswVectorsWriter

Lucene94HnswVectorsWriter(SegmentWriteState state,
 int M,
 int beamWidth)
                   throws IOException

Throws:

IOException

Method Details

addField

public KnnFieldVectorsWriter<?> addField(FieldInfo fieldInfo)
                                  throws IOException

Description copied from class: KnnVectorsWriter

Add new field for indexing

Specified by:

addField in class KnnVectorsWriter

Throws:

IOException

flush

public void flush(int maxDoc,
 Sorter.DocMap sortMap)
           throws IOException

Description copied from class: KnnVectorsWriter

Flush all buffered data on disk *

Specified by:

flush in class KnnVectorsWriter

Throws:

IOException

finish

public void finish()
            throws IOException

Description copied from class: KnnVectorsWriter

Called once at the end before close

Specified by:

finish in class KnnVectorsWriter

Throws:

IOException

ramBytesUsed

public long ramBytesUsed()

Description copied from interface: Accountable

Return the memory usage of this object in bytes. Negative values are illegal.

writeField

private void writeField(Lucene94HnswVectorsWriter.FieldWriter<?> fieldData,
 int maxDoc)
                 throws IOException

Throws:

IOException

writeFloat32Vectors

private void writeFloat32Vectors(Lucene94HnswVectorsWriter.FieldWriter<?> fieldData)
                          throws IOException

Throws:

IOException

writeByteVectors

private void writeByteVectors(Lucene94HnswVectorsWriter.FieldWriter<?> fieldData)
                       throws IOException

Throws:

IOException

writeSortingField

private void writeSortingField(Lucene94HnswVectorsWriter.FieldWriter<?> fieldData,
 int maxDoc,
 Sorter.DocMap sortMap)
                        throws IOException

Throws:

IOException

writeSortedFloat32Vectors

private long writeSortedFloat32Vectors(Lucene94HnswVectorsWriter.FieldWriter<?> fieldData,
 int[] ordMap)
                                throws IOException

Throws:

IOException

writeSortedByteVectors

private long writeSortedByteVectors(Lucene94HnswVectorsWriter.FieldWriter<?> fieldData,
 int[] ordMap)
                             throws IOException

Throws:

IOException

reconstructAndWriteGraph

private HnswGraph reconstructAndWriteGraph(OnHeapHnswGraph graph,
 int[] newToOldMap,
 int[] oldToNewMap)
                                    throws IOException

Throws:

IOException

reconstructAndWriteNeigbours

private void reconstructAndWriteNeigbours(NeighborArray neighbors,
 int[] oldToNewMap,
 int maxConnOnLevel,
 int maxOrd)
                                   throws IOException

Throws:

IOException

mergeOneField

public void mergeOneField(FieldInfo fieldInfo,
 MergeState mergeState)
                   throws IOException

Description copied from class: KnnVectorsWriter

Write field for merging

Overrides:

mergeOneField in class KnnVectorsWriter

Throws:

IOException

writeGraph

private void writeGraph(OnHeapHnswGraph graph)
                 throws IOException

Throws:

IOException

writeMeta

private void writeMeta(FieldInfo field,
 int maxDoc,
 long vectorDataOffset,
 long vectorDataLength,
 long vectorIndexOffset,
 long vectorIndexLength,
 DocsWithFieldSet docsWithField,
 HnswGraph graph)
                throws IOException

Throws:

IOException

writeVectorData

private static DocsWithFieldSet writeVectorData(IndexOutput output,
 VectorValues vectors,
 int scalarSize)
                                         throws IOException

Writes the vector values to the output and returns a set of documents that contains vectors.

Throws:

IOException

close

public void close()
           throws IOException

Throws:

IOException