Method ExecChunked

cpp/src/arrow/compute/kernels/vector_statistics.cc:80–100

  }

  static Status ExecChunked(KernelContext* ctx, const ExecBatch& batch, Datum* out) {
    const auto& options = WinsorizeState::Get(ctx);
    RETURN_NOT_OK(ValidateOptions(options));
    const auto& chunked_array = batch.values[0].chunked_array();
    ARROW_ASSIGN_OR_RAISE(auto maybe_quantiles,
                          GetQuantileValues(ctx, chunked_array, options));
    if (!maybe_quantiles.has_value()) {
      // Only nulls and NaNs => return input as-is
      *out = chunked_array;
      return Status::OK();
    }
    ArrayVector out_chunks;
    out_chunks.reserve(chunked_array->num_chunks());
    for (const auto& chunk : chunked_array->chunks()) {
      auto out_data = chunk->data()->Copy();
      RETURN_NOT_OK(
          ClipValues(*chunk->data(), maybe_quantiles.value(), out_data.get(), ctx));
      out_chunks.push_back(MakeArray(out_data));
    }
    return ChunkedArray::Make(std::move(out_chunks)).Value(out);
  }

  struct QuantileValues {
    CType lower_bound, upper_bound;