Zengtudor
/
pybind11
mirror of https://github.com/pybind/pybind11.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

move check

This commit is contained in:
Francesco Rizzi 2023-06-28 18:20:52 +02:00
parent f52a31a004
commit 17912ba667
1 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
include/pybind11/numpy.h
View File

@ -1093,14 +1093,18 @@ public:
// Reference to element at a given index
template <typename... Ix>
const T &at(Ix... index) const {
check_access_precondition(index...);
if ((ssize_t) sizeof...(index) != ndim()) {
fail_dim_check(sizeof...(index), "index dimension mismatch");
}
return const_reference(index...);
}
// Mutable reference to element at a given index
template <typename... Ix>
T &mutable_at(Ix... index) {
check_access_precondition(index...);
if ((ssize_t) sizeof...(index) != ndim()) {
fail_dim_check(sizeof...(index), "index dimension mismatch");
}
return mutable_reference(index...);
}
@ -1108,7 +1112,9 @@ public:
template <typename... Ix>
const T &operator()(Ix... index) const {
#if !defined(NDEBUG)
check_access_precondition(index...);
if ((ssize_t) sizeof...(index) != ndim()) {
fail_dim_check(sizeof...(index), "index dimension mismatch");
}
#endif
return const_reference(index...);
}
@ -1117,7 +1123,9 @@ public:
template <typename... Ix>
T &operator()(Ix... index) {
#if !defined(NDEBUG)
check_access_precondition(index...);
if ((ssize_t) sizeof...(index) != ndim()) {
fail_dim_check(sizeof...(index), "index dimension mismatch");
}
#endif
return mutable_reference(index...);
}
@ -1191,13 +1199,6 @@ private:
return *(static_cast<T *>(array::mutable_data())
+ byte_offset(ssize_t(index)...) / itemsize());
}
template <typename... Ix>
void check_access_precondition(Ix... index) const {
if ((ssize_t) sizeof...(index) != ndim()) {
fail_dim_check(sizeof...(index), "index dimension mismatch");
}
}
};
template <typename T>