Function setValueAt

bolt/python/Aggregate.cpp:98–132

template <TypeKind TYPE>
void setValueAt(
    BaseVector& v,
    const vector_size_t i,
    const pybind11::object& pyVal) {
  bolt::python::pyTry<void>(
      [&]() {
        if constexpr (TYPE == TypeKind::ROW) {
          auto rhs = pyVal.cast<RowVectorPtr>();
          v.slice(i, 1)->copy(*rhs);
        }
        if constexpr (TYPE == TypeKind::MAP) {
          auto rhs = pyVal.cast<std::shared_ptr<MapVector>>();
          v.slice(i, 1)->copy(*rhs);
        }
        if constexpr (TYPE == TypeKind::ARRAY) {
          auto rhs = pyVal.cast<std::shared_ptr<ArrayVector>>();
          v.slice(i, 1)->copy(*rhs);
        }
        if constexpr (
            TYPE != TypeKind::ROW && TYPE != TypeKind::MAP &&
            TYPE != TypeKind::ARRAY) {
          auto cppVal = pyVal.cast<typename TypeTraits<TYPE>::DeepCopiedType>();
          auto* vec = v.as<FlatVector<decltype(cppVal)>>();
          BOLT_CHECK(vec != nullptr);
          vec->set(i, std::move(cppVal));
        }
      },
      [&]() {
        return fmt::format(
            "Invalid or unsupported cast of python aggregation function"
            "result type '{}' to the cpp type: '{}'.",
            bolt::python::pyInstanceTypeStr(pyVal),
            mapTypeKindToName(TYPE));
      });
}
} // namespace

namespace bolt::python {