Method getWindowFunctionResults

bolt/exec/WindowPartition.cpp:700–731

template <RowFormat R>
void SpilledWindowPartition<R>::getWindowFunctionResults(
    VectorPtr& result,
    int32_t numOutputRows,
    int32_t functionIndex,
    bool isAggregateWindowFunc) {
  if (isAggregateWindowFunc) {
    const auto* aggResult = aggregateResults_[functionIndex].back().get();
    for (auto i = 0; i < numOutputRows; ++i) {
      result->copy(aggResult, i, 0, 1);
    }
  } else {
    auto& aggResults = aggregateResults_[functionIndex];
    int32_t target_index = 0;
    while (numOutputRows && (aggResults.size() > 0)) {
      const auto* currentResult = aggResults.front().get();
      auto actualCopySize = std::min(numOutputRows, currentResult->size());
      result->copy(currentResult, target_index, 0, actualCopySize);
      numOutputRows = numOutputRows - actualCopySize;
      target_index = target_index + actualCopySize;
      if (actualCopySize == currentResult->size()) {
        // If we copied all rows from the first aggregate result, we can
        // prepare it for reuse and remove it from the list.
        aggResults.front()->prepareForReuse();
        aggResults.pop_front();
      } else {
        // If we copied only part of the first aggregate result, we slice it
        aggResults.front() = currentResult->slice(
            actualCopySize, currentResult->size() - actualCopySize);
      }
    }
  }
}

template class WindowPartitionImpl<RowFormat::kRowContainer>;
template class WindowPartitionImpl<RowFormat::kSerializedRows>;