Function Concatenate

cpp/src/arrow/array/concatenate.cc:921–952

namespace internal {

Result<std::shared_ptr<Array>> Concatenate(
    const ArrayVector& arrays, MemoryPool* pool,
    std::shared_ptr<DataType>* out_suggested_cast) {
  DCHECK(out_suggested_cast);
  *out_suggested_cast = nullptr;
  if (arrays.size() == 0) {
    return Status::Invalid("Must pass at least one array");
  }

  // gather ArrayData of input arrays
  ArrayDataVector data(arrays.size());
  for (size_t i = 0; i < arrays.size(); ++i) {
    if (!arrays[i]->type()->Equals(*arrays[0]->type())) {
      return Status::Invalid("arrays to be concatenated must be identically typed, but ",
                             *arrays[0]->type(), " and ", *arrays[i]->type(),
                             " were encountered.");
    }
    data[i] = arrays[i]->data();
  }

  std::shared_ptr<ArrayData> out_data;
  ErrorHints hints;
  auto status = ConcatenateImpl(data, pool).Concatenate(&out_data, &hints);
  if (!status.ok()) {
    if (hints.suggested_cast) {
      DCHECK(status.IsInvalid());
      *out_suggested_cast = std::move(hints.suggested_cast);
    }
    return status;
  }
  return MakeArray(std::move(out_data));
}

}  // namespace internal