Method WindowBuild

bolt/exec/WindowBuild.cpp:99–144

} // namespace

WindowBuild::WindowBuild(
    const std::shared_ptr<const core::WindowNode>& windowNode,
    bolt::memory::MemoryPool* pool,
    const common::SpillConfig* spillConfig,
    tsan_atomic<bool>* nonReclaimableSection,
    uint64_t spillMemoryThreshold,
    bool enableJit)
    : spillConfig_{spillConfig},
      nonReclaimableSection_{nonReclaimableSection},
      decodedInputVectors_(windowNode->inputType()->size()),
      sortSpillCompareFlags_{makeSpillCompareFlags(
          windowNode->partitionKeys().size(),
          windowNode->sortingOrders())},
      pool_(pool),
      spillMemoryThreshold_(spillMemoryThreshold),
      enableJit_(enableJit),
      numPartitionKeys_{windowNode->partitionKeys().size()} {
  BOLT_CHECK_NOT_NULL(pool_);
  std::tie(inputChannels_, inversedInputChannels_, inputType_) =
      reorderInputChannels(
          windowNode->inputType(),
          windowNode->partitionKeys(),
          windowNode->sortingKeys());
  followedTopNum_ = windowNode->followedTopNum();
  const auto numPartitionKeys = windowNode->partitionKeys().size();
  const auto numSortingKeys = windowNode->sortingKeys().size();
  const auto numKeys = numPartitionKeys + numSortingKeys;
  data_ = std::make_unique<RowContainer>(
      slice(inputType_->children(), 0, numKeys),
      slice(inputType_->children(), numKeys, inputType_->size()),
      pool);

  for (auto i = 0; i < numPartitionKeys; ++i) {
    partitionKeyInfo_.push_back(std::make_pair(i, core::SortOrder{true, true}));
  }

  for (auto i = 0; i < numSortingKeys; ++i) {
    sortKeyInfo_.push_back(
        std::make_pair(numPartitionKeys + i, windowNode->sortingOrders()[i]));
  }
  allKeysInfo_.reserve(partitionKeyInfo_.size() + sortKeyInfo_.size());
  allKeysInfo_.insert(
      allKeysInfo_.cend(), partitionKeyInfo_.begin(), partitionKeyInfo_.end());
  allKeysInfo_.insert(
      allKeysInfo_.cend(), sortKeyInfo_.begin(), sortKeyInfo_.end());
}

void WindowBuild::updateEstimatedOutputRowSize() {
  const auto optionalRowSize = data_->estimateRowSize();