Method prepareForFiltering

query/aql_processor.go:718–731

prepareForFiltering prepares the input and the index vectors for filtering.

(
	columns []deviceVectorPartySlice, firstColumn int, startRow int, stream unsafe.Pointer)

// prepareForFiltering prepares the input and the index vectors for filtering.
func (bc *oopkBatchContext) prepareForFiltering(
	columns []deviceVectorPartySlice, firstColumn int, startRow int, stream unsafe.Pointer) {
	bc.columns = columns
	bc.startRow = startRow

	if firstColumn >= 0 {
		bc.size = columns[firstColumn].length
		// Allocate twice of the size to save number of allocations of temporary index vector.
		bc.indexVectorD = deviceAllocate(bc.size*4, bc.device)
		bc.predicateVectorD = deviceAllocate(bc.size, bc.device)
		bc.baseCountD = columns[firstColumn].counts.offset(columns[firstColumn].countStartIndex * 4)
	}
	bc.stats.batchSize = bc.size
}

// prepareForDimAndMeasureEval ensures that dim/measure vectors have enough
// capacity for bc.resultSize+bc.size.