compression

type BRQData struct {
	InputDim uint32
	Rotation FastRotation
	Rounding []float32
}

type Encoder byte

type FastRotation struct {
	OutputDim uint32      // The dimension of the output returned by Rotate().
	Rounds    uint32      // The number of rounds of random signs, swaps, and blocked transforms that the Rotate() function is going to apply.
	Swaps     [][]Swap    // Random swaps to apply each round prior to transforming.
	Signs     [][]float32 // Random signs to apply each round prior to transforming. We store these as float32 values for performance reasons, to avoid casts.
}

type MuveraData struct {
	KSim         uint32        // 4 bytes
	NumClusters  uint32        // 4 bytes
	Dimensions   uint32        // 4 bytes
	DProjections uint32        // 4 bytes
	Repetitions  uint32        // 4 bytes
	Gaussians    [][][]float32 // (repetitions, kSim, dimensions)
	S            [][][]float32 // (repetitions, dProjections, dimensions)
}

type PQData struct {
	Ks                  uint16
	M                   uint16
	Dimensions          uint16
	EncoderType         Encoder
	EncoderDistribution byte
	Encoders            []PQSegmentEncoder
	UseBitsEncoding     bool
	TrainingLimit       int
}

type PQSegmentEncoder interface {
	Encode(x []float32) byte
	Centroid(b byte) []float32
	Add(x []float32)
	Fit(data [][]float32) error
	ExposeDataForRestore() []byte
}

type RQData struct {
	InputDim uint32
	Bits     uint32
	Rotation FastRotation
	Rounding []float32
}

type SQData struct {
	A          float32
	B          float32
	Dimensions uint16
}

type State struct {
	// Compressed indicates if the index uses compression.
	Compressed bool

	// PQData holds Product Quantization compression data.
	PQData *PQData

	// SQData holds Scalar Quantization compression data.
	SQData *SQData

	// RQData holds Rotational Quantization compression data.
	RQData *RQData

	// BRQData holds Binary Rotational Quantization compression data.
	BRQData *BRQData

	// MuveraEnabled indicates if Muvera multi-vector encoding is enabled.
	MuveraEnabled bool

	// MuveraData holds Muvera multi-vector encoding data.
	MuveraData *MuveraData
}

type Swap struct {
	I, J uint16
}