Function GenerateNumericArray

cpp/src/arrow/testing/random.cc:176–196

template <typename ArrowType, typename OptionType>
std::shared_ptr<NumericArray<ArrowType>> GenerateNumericArray(int64_t size,
                                                              OptionType options,
                                                              int64_t alignment,
                                                              MemoryPool* memory_pool) {
  using CType = typename ArrowType::c_type;
  auto type = TypeTraits<ArrowType>::type_singleton();
  BufferVector buffers{2};

  int64_t null_count = 0;
  buffers[0] = *AllocateEmptyBitmap(size, alignment, memory_pool);
  options.GenerateBitmap(buffers[0]->mutable_data(), size, &null_count);

  buffers[1] = *AllocateBuffer(sizeof(CType) * size, alignment, memory_pool);
  options.GenerateData(buffers[1]->mutable_data(), size);
  if (std::is_same<ArrowType, Date64Type>::value) {
    GenerateFullDayMillisNoNan(buffers[1]->mutable_data(), size);
  }

  auto array_data = ArrayData::Make(type, size, buffers, null_count);
  return std::make_shared<NumericArray<ArrowType>>(array_data);
}

}  // namespace