Zengtudor/pybind11
1
0
Fork 0
mirror of https://github.com/pybind/pybind11.git synced 2025-02-18 06:30:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

numpy in a new world

Browse Source
This commit is contained in:
Francis Williams 2023-01-11 15:27:09 -05:00
parent a34596bfe1
commit 5cf262c762
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
include/pybind11/numpy.h
View File

@ -25,6 +25,15 @@
#include <typeindex> #include <typeindex>
#include <utility> #include <utility>
#include <vector> #include <vector>
#include <iostream>
#if defined(_MSC_VER)
# pragma warning(push)
# pragma warning(disable: 4127) // warning C4127: Conditional expression is constant
typedef SSIZE_T ssize_t;
#endif
/* This will be true on all flat address space platforms and allows us to reduce the /* This will be true on all flat address space platforms and allows us to reduce the
whole npy_intp / ssize_t / Py_intptr_t business down to just ssize_t for all size whole npy_intp / ssize_t / Py_intptr_t business down to just ssize_t for all size
@ -622,6 +631,15 @@ public:
/// Flags for the array descriptor /// Flags for the array descriptor
char flags() const { return detail::array_descriptor_proxy(m_ptr)->flags; } char flags() const { return detail::array_descriptor_proxy(m_ptr)->flags; }
/// NumPy array type char
char type() const { return detail::array_descriptor_proxy(m_ptr)->type; }
/// NumPy array type num
int type_num() const { return detail::array_descriptor_proxy(m_ptr)->type_num; }
/// NumPy array element size
int elsize() const { return detail::array_descriptor_proxy(m_ptr)->elsize; }
private: private:
static object _dtype_from_pep3118() { static object _dtype_from_pep3118() {
static PyObject *obj = module_::import("numpy.core._internal") static PyObject *obj = module_::import("numpy.core._internal")
@ -824,6 +842,9 @@ public:
/// Return the NumPy array flags /// Return the NumPy array flags
int flags() const { return detail::array_proxy(m_ptr)->flags; } int flags() const { return detail::array_proxy(m_ptr)->flags; }
/// Mutable NumPy array flags
int& flags() { return detail::array_proxy(m_ptr)->flags; }
/// If set, the array is writeable (otherwise the buffer is read-only) /// If set, the array is writeable (otherwise the buffer is read-only)
bool writeable() const { bool writeable() const {
return detail::check_flags(m_ptr, detail::npy_api::NPY_ARRAY_WRITEABLE_); return detail::check_flags(m_ptr, detail::npy_api::NPY_ARRAY_WRITEABLE_);