include/pybind11/numpy.h

42 NAMESPACE_BEGIN(detail)
442 NAMESPACE_END(detail)
446     PYBIND11_OBJECT_DEFAULT(dtype, object, detail::npy_api::get().PyArrayDescr_Check_);
472         if (!detail::npy_api::get().PyArray_DescrConverter_(args.ptr(), &ptr) || !ptr)
479         return detail::npy_format_descriptor<typename std::remove_cv<T>::type>::dtype();
484         return detail::array_descriptor_proxy(m_ptr)->elsize;
489         return detail::array_descriptor_proxy(m_ptr)->names != nullptr;
494         return detail::array_descriptor_proxy(m_ptr)->kind;
540     PYBIND11_OBJECT_CVT(array, buffer, detail::npy_api::get().PyArray_Check_, raw_array)
543         c_style = detail::npy_api::NPY_ARRAY_C_CONTIGUOUS_,
544         f_style = detail::npy_api::NPY_ARRAY_F_CONTIGUOUS_,
545         forcecast = detail::npy_api::NPY_ARRAY_FORCECAST_
550     using ShapeContainer = detail::any_container<ssize_t>;
551     using StridesContainer = detail::any_container<ssize_t>;
569                 flags = reinterpret_borrow<array>(base).flags() & ~detail::npy_api::NPY_ARRAY_OWNDATA_;
572                 flags = detail::npy_api::NPY_ARRAY_WRITEABLE_;
575         auto &api = detail::npy_api::get();
594     template <typename T, typename = detail::enable_if_t<std::is_integral<T>::value && !std::is_same<bool, T>::value>>
614         return reinterpret_borrow<pybind11::dtype>(detail::array_proxy(m_ptr)->descr);
624         return detail::array_descriptor_proxy(detail::array_proxy(m_ptr)->descr)->elsize;
634         return detail::array_proxy(m_ptr)->nd;
639         return reinterpret_borrow<object>(detail::array_proxy(m_ptr)->base);
644         return detail::array_proxy(m_ptr)->dimensions;
656         return detail::array_proxy(m_ptr)->strides;
668         return detail::array_proxy(m_ptr)->flags;
673         return detail::check_flags(m_ptr, detail::npy_api::NPY_ARRAY_WRITEABLE_);
678         return detail::check_flags(m_ptr, detail::npy_api::NPY_ARRAY_OWNDATA_);
684         return static_cast<const void *>(detail::array_proxy(m_ptr)->data + offset_at(index...));
692         return static_cast<void *>(detail::array_proxy(m_ptr)->data + offset_at(index...));
717     template <typename T, ssize_t Dims = -1> detail::unchecked_mutable_reference<T, Dims> mutable_unchecked() & {
721         return detail::unchecked_mutable_reference<T, Dims>(mutable_data(), shape(), strides(), ndim());
731     template <typename T, ssize_t Dims = -1> detail::unchecked_reference<T, Dims> unchecked() const & {
735         return detail::unchecked_reference<T, Dims>(data(), shape(), strides(), ndim());
740         auto& api = detail::npy_api::get();
748         detail::npy_api::PyArray_Dims d = {
753             detail::npy_api::get().PyArray_Resize_(m_ptr, &d, int(refcheck), -1)
769     template<typename, typename> friend struct detail::npy_format_descriptor;
778         return detail::byte_offset_unsafe(strides(), ssize_t(index)...);
826         return detail::npy_api::get().PyArray_FromAny_(
827             ptr, nullptr, 0, 0, detail::npy_api::NPY_ARRAY_ENSUREARRAY_ | ExtraFlags, nullptr);
838     static_assert(!detail::array_info<T>::is_array, "Array types cannot be used with array_t");
905     template <ssize_t Dims = -1> detail::unchecked_mutable_reference<T, Dims> mutable_unchecked() & {
916     template <ssize_t Dims = -1> detail::unchecked_reference<T, Dims> unchecked() const & {
930         const auto &api = detail::npy_api::get();
932                && api.PyArray_EquivTypes_(detail::array_proxy(h.ptr())->descr, dtype::of<T>().ptr());
942         return detail::npy_api::get().PyArray_FromAny_(
944             detail::npy_api::NPY_ARRAY_ENSUREARRAY_ | ExtraFlags, nullptr);
949 struct format_descriptor<T, detail::enable_if_t<detail::is_pod_struct<T>::value>> {
951         return detail::npy_format_descriptor<typename std::remove_cv<T>::type>::format();
963 struct format_descriptor<T, detail::enable_if_t<std::is_enum<T>::value>> {
971 struct format_descriptor<T, detail::enable_if_t<detail::array_info<T>::is_array>> {
973         using namespace detail;
979 NAMESPACE_BEGIN(detail)
998 struct compare_buffer_info<T, detail::enable_if_t<detail::is_pod_struct<T>::value>> {
1043     static constexpr int value = values[detail::is_fmt_numeric<T>::index];
1197     ::pybind11::detail::field_descriptor {                                                    \
1200         ::pybind11::detail::npy_format_descriptor<decltype(std::declval<T>().Field)>::dtype() \
1239     ::pybind11::detail::npy_format_descriptor<Type>::register_dtype \
1240         (::std::vector<::pybind11::detail::field_descriptor> \
1261     ::pybind11::detail::npy_format_descriptor<Type>::register_dtype \
1262         (::std::vector<::pybind11::detail::field_descriptor> \
1601     return detail::vectorize_helper<Func, Return, Args...>(f);
1608 NAMESPACE_END(detail)
1612 detail::vectorize_helper<Return (*)(Args...), Return, Args...>
1614     return detail::vectorize_helper<Return (*)(Args...), Return, Args...>(f);
1618 template <typename Func, detail::enable_if_t<detail::is_lambda<Func>::value, int> = 0>
1620         detail::vectorize_extractor(std::forward<Func>(f), (detail::function_signature_t<Func> *) nullptr)) {
1621     return detail::vectorize_extractor(std::forward<Func>(f), (detail::function_signature_t<Func> *) nullptr);
1626           typename Helper = detail::vectorize_helper<decltype(std::mem_fn(std::declval<Return (Class::*)(Args...)>())), Return, Class *, Args...>>
1633           typename Helper = detail::vectorize_helper<decltype(std::mem_fn(std::declval<Return (Class::*)(Args...) const>())), Return, const Class *, Args...>>