Function IfElseBench

cpp/src/arrow/compute/kernels/scalar_if_else_benchmark.cc:98–125

template <typename Type>
static void IfElseBench(benchmark::State& state, const std::shared_ptr<DataType>& type) {
  using ArrayType = typename TypeTraits<Type>::ArrayType;

  int64_t len = state.range(0);
  int64_t offset = state.range(1);

  random::RandomArrayGenerator rand(/*seed=*/0);

  auto cond = std::static_pointer_cast<BooleanArray>(
                  rand.ArrayOf(boolean(), len, /*null_probability=*/0.01))
                  ->Slice(offset);
  auto left = std::static_pointer_cast<ArrayType>(
                  rand.ArrayOf(type, len, /*null_probability=*/0.01))
                  ->Slice(offset);
  auto right = std::static_pointer_cast<ArrayType>(
                   rand.ArrayOf(type, len, /*null_probability=*/0.01))
                   ->Slice(offset);

  for (auto _ : state) {
    ABORT_NOT_OK(IfElse(cond, left, right));
  }

  state.SetBytesProcessed(
      state.iterations() *
      (GetBytesProcessed(*cond) + GetBytesProcessed(*left) + GetBytesProcessed(*right)));
  state.SetItemsProcessed(state.iterations() * (len - offset));
  state.counters["offset"] = static_cast<double>(offset);
}

template <typename Type>
static void IfElseBench(benchmark::State& state) {