mirror of
https://github.com/pybind/pybind11.git
synced 2024-11-14 17:43:53 +00:00
1419 lines
57 KiB
C++
1419 lines
57 KiB
C++
// clang-format off
|
|
/*
|
|
pybind11/cast.h: Partial template specializations to cast between
|
|
C++ and Python types
|
|
|
|
Copyright (c) 2016 Wenzel Jakob <wenzel.jakob@epfl.ch>
|
|
|
|
All rights reserved. Use of this source code is governed by a
|
|
BSD-style license that can be found in the LICENSE file.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "pytypes.h"
|
|
#include "detail/common.h"
|
|
#include "detail/descr.h"
|
|
#include "detail/smart_holder_sfinae_hooks_only.h"
|
|
#include "detail/type_caster_base.h"
|
|
#include "detail/typeid.h"
|
|
#include <array>
|
|
#include <cstring>
|
|
#include <functional>
|
|
#include <iosfwd>
|
|
#include <iterator>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <tuple>
|
|
#include <type_traits>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
#ifdef PYBIND11_USE_SMART_HOLDER_AS_DEFAULT
|
|
#include "detail/smart_holder_type_casters.h"
|
|
#endif
|
|
|
|
#if defined(PYBIND11_CPP17)
|
|
# if defined(__has_include)
|
|
# if __has_include(<string_view>)
|
|
# define PYBIND11_HAS_STRING_VIEW
|
|
# endif
|
|
# elif defined(_MSC_VER)
|
|
# define PYBIND11_HAS_STRING_VIEW
|
|
# endif
|
|
#endif
|
|
#ifdef PYBIND11_HAS_STRING_VIEW
|
|
#include <string_view>
|
|
#endif
|
|
|
|
#if defined(__cpp_lib_char8_t) && __cpp_lib_char8_t >= 201811L
|
|
# define PYBIND11_HAS_U8STRING
|
|
#endif
|
|
|
|
PYBIND11_NAMESPACE_BEGIN(PYBIND11_NAMESPACE)
|
|
PYBIND11_NAMESPACE_BEGIN(detail)
|
|
|
|
// clang-format on
|
|
#ifndef PYBIND11_USE_SMART_HOLDER_AS_DEFAULT
|
|
template <typename T>
|
|
class type_caster_for_class_ : public type_caster_base<T> {};
|
|
#endif
|
|
|
|
template <typename type, typename SFINAE = void>
|
|
class type_caster : public type_caster_for_class_<type> {};
|
|
|
|
template <typename type>
|
|
using make_caster = type_caster<intrinsic_t<type>>;
|
|
|
|
template <typename T>
|
|
struct type_uses_smart_holder_type_caster {
|
|
static constexpr bool value
|
|
= std::is_base_of<smart_holder_type_caster_base_tag, make_caster<T>>::value;
|
|
};
|
|
// clang-format off
|
|
|
|
// Shortcut for calling a caster's `cast_op_type` cast operator for casting a type_caster to a T
|
|
template <typename T> typename make_caster<T>::template cast_op_type<T> cast_op(make_caster<T> &caster) {
|
|
return caster.operator typename make_caster<T>::template cast_op_type<T>();
|
|
}
|
|
template <typename T> typename make_caster<T>::template cast_op_type<typename std::add_rvalue_reference<T>::type>
|
|
cast_op(make_caster<T> &&caster) {
|
|
return std::move(caster).operator
|
|
typename make_caster<T>::template cast_op_type<typename std::add_rvalue_reference<T>::type>();
|
|
}
|
|
|
|
template <typename type> class type_caster<std::reference_wrapper<type>> {
|
|
private:
|
|
using caster_t = make_caster<type>;
|
|
caster_t subcaster;
|
|
using reference_t = type&;
|
|
using subcaster_cast_op_type =
|
|
typename caster_t::template cast_op_type<reference_t>;
|
|
|
|
static_assert(std::is_same<typename std::remove_const<type>::type &, subcaster_cast_op_type>::value ||
|
|
std::is_same<reference_t, subcaster_cast_op_type>::value,
|
|
"std::reference_wrapper<T> caster requires T to have a caster with an "
|
|
"`operator T &()` or `operator const T &()`");
|
|
public:
|
|
bool load(handle src, bool convert) { return subcaster.load(src, convert); }
|
|
static constexpr auto name = caster_t::name;
|
|
static handle cast(const std::reference_wrapper<type> &src, return_value_policy policy, handle parent) {
|
|
// It is definitely wrong to take ownership of this pointer, so mask that rvp
|
|
if (policy == return_value_policy::take_ownership || policy == return_value_policy::automatic)
|
|
policy = return_value_policy::automatic_reference;
|
|
return caster_t::cast(&src.get(), policy, parent);
|
|
}
|
|
template <typename T> using cast_op_type = std::reference_wrapper<type>;
|
|
operator std::reference_wrapper<type>() { return cast_op<type &>(subcaster); }
|
|
};
|
|
|
|
#define PYBIND11_TYPE_CASTER(type, py_name) \
|
|
protected: \
|
|
type value; \
|
|
public: \
|
|
static constexpr auto name = py_name; \
|
|
template <typename T_, enable_if_t<std::is_same<type, remove_cv_t<T_>>::value, int> = 0> \
|
|
static handle cast(T_ *src, return_value_policy policy, handle parent) { \
|
|
if (!src) return none().release(); \
|
|
if (policy == return_value_policy::take_ownership) { \
|
|
auto h = cast(std::move(*src), policy, parent); delete src; return h; \
|
|
} else { \
|
|
return cast(*src, policy, parent); \
|
|
} \
|
|
} \
|
|
operator type*() { return &value; } \
|
|
operator type&() { return value; } \
|
|
operator type&&() && { return std::move(value); } \
|
|
template <typename T_> using cast_op_type = pybind11::detail::movable_cast_op_type<T_>
|
|
|
|
|
|
template <typename CharT> using is_std_char_type = any_of<
|
|
std::is_same<CharT, char>, /* std::string */
|
|
#if defined(PYBIND11_HAS_U8STRING)
|
|
std::is_same<CharT, char8_t>, /* std::u8string */
|
|
#endif
|
|
std::is_same<CharT, char16_t>, /* std::u16string */
|
|
std::is_same<CharT, char32_t>, /* std::u32string */
|
|
std::is_same<CharT, wchar_t> /* std::wstring */
|
|
>;
|
|
|
|
|
|
template <typename T>
|
|
struct type_caster<T, enable_if_t<std::is_arithmetic<T>::value && !is_std_char_type<T>::value>> {
|
|
using _py_type_0 = conditional_t<sizeof(T) <= sizeof(long), long, long long>;
|
|
using _py_type_1 = conditional_t<std::is_signed<T>::value, _py_type_0, typename std::make_unsigned<_py_type_0>::type>;
|
|
using py_type = conditional_t<std::is_floating_point<T>::value, double, _py_type_1>;
|
|
public:
|
|
|
|
bool load(handle src, bool convert) {
|
|
py_type py_value;
|
|
|
|
if (!src)
|
|
return false;
|
|
|
|
#if !defined(PYPY_VERSION)
|
|
auto index_check = [](PyObject *o) { return PyIndex_Check(o); };
|
|
#else
|
|
// In PyPy 7.3.3, `PyIndex_Check` is implemented by calling `__index__`,
|
|
// while CPython only considers the existence of `nb_index`/`__index__`.
|
|
auto index_check = [](PyObject *o) { return hasattr(o, "__index__"); };
|
|
#endif
|
|
|
|
if (std::is_floating_point<T>::value) {
|
|
if (convert || PyFloat_Check(src.ptr()))
|
|
py_value = (py_type) PyFloat_AsDouble(src.ptr());
|
|
else
|
|
return false;
|
|
} else if (PyFloat_Check(src.ptr())) {
|
|
return false;
|
|
} else if (!convert && !PYBIND11_LONG_CHECK(src.ptr()) && !index_check(src.ptr())) {
|
|
return false;
|
|
} else {
|
|
handle src_or_index = src;
|
|
#if PY_VERSION_HEX < 0x03080000
|
|
object index;
|
|
if (!PYBIND11_LONG_CHECK(src.ptr())) { // So: index_check(src.ptr())
|
|
index = reinterpret_steal<object>(PyNumber_Index(src.ptr()));
|
|
if (!index) {
|
|
PyErr_Clear();
|
|
if (!convert)
|
|
return false;
|
|
}
|
|
else {
|
|
src_or_index = index;
|
|
}
|
|
}
|
|
#endif
|
|
if (std::is_unsigned<py_type>::value) {
|
|
py_value = as_unsigned<py_type>(src_or_index.ptr());
|
|
} else { // signed integer:
|
|
py_value = sizeof(T) <= sizeof(long)
|
|
? (py_type) PyLong_AsLong(src_or_index.ptr())
|
|
: (py_type) PYBIND11_LONG_AS_LONGLONG(src_or_index.ptr());
|
|
}
|
|
}
|
|
|
|
// Python API reported an error
|
|
bool py_err = py_value == (py_type) -1 && PyErr_Occurred();
|
|
|
|
// Check to see if the conversion is valid (integers should match exactly)
|
|
// Signed/unsigned checks happen elsewhere
|
|
if (py_err || (std::is_integral<T>::value && sizeof(py_type) != sizeof(T) && py_value != (py_type) (T) py_value)) {
|
|
PyErr_Clear();
|
|
if (py_err && convert && PyNumber_Check(src.ptr())) {
|
|
auto tmp = reinterpret_steal<object>(std::is_floating_point<T>::value
|
|
? PyNumber_Float(src.ptr())
|
|
: PyNumber_Long(src.ptr()));
|
|
PyErr_Clear();
|
|
return load(tmp, false);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
value = (T) py_value;
|
|
return true;
|
|
}
|
|
|
|
template<typename U = T>
|
|
static typename std::enable_if<std::is_floating_point<U>::value, handle>::type
|
|
cast(U src, return_value_policy /* policy */, handle /* parent */) {
|
|
return PyFloat_FromDouble((double) src);
|
|
}
|
|
|
|
template<typename U = T>
|
|
static typename std::enable_if<!std::is_floating_point<U>::value && std::is_signed<U>::value && (sizeof(U) <= sizeof(long)), handle>::type
|
|
cast(U src, return_value_policy /* policy */, handle /* parent */) {
|
|
return PYBIND11_LONG_FROM_SIGNED((long) src);
|
|
}
|
|
|
|
template<typename U = T>
|
|
static typename std::enable_if<!std::is_floating_point<U>::value && std::is_unsigned<U>::value && (sizeof(U) <= sizeof(unsigned long)), handle>::type
|
|
cast(U src, return_value_policy /* policy */, handle /* parent */) {
|
|
return PYBIND11_LONG_FROM_UNSIGNED((unsigned long) src);
|
|
}
|
|
|
|
template<typename U = T>
|
|
static typename std::enable_if<!std::is_floating_point<U>::value && std::is_signed<U>::value && (sizeof(U) > sizeof(long)), handle>::type
|
|
cast(U src, return_value_policy /* policy */, handle /* parent */) {
|
|
return PyLong_FromLongLong((long long) src);
|
|
}
|
|
|
|
template<typename U = T>
|
|
static typename std::enable_if<!std::is_floating_point<U>::value && std::is_unsigned<U>::value && (sizeof(U) > sizeof(unsigned long)), handle>::type
|
|
cast(U src, return_value_policy /* policy */, handle /* parent */) {
|
|
return PyLong_FromUnsignedLongLong((unsigned long long) src);
|
|
}
|
|
|
|
PYBIND11_TYPE_CASTER(T, _<std::is_integral<T>::value>("int", "float"));
|
|
};
|
|
|
|
template<typename T> struct void_caster {
|
|
public:
|
|
bool load(handle src, bool) {
|
|
if (src && src.is_none())
|
|
return true;
|
|
return false;
|
|
}
|
|
static handle cast(T, return_value_policy /* policy */, handle /* parent */) {
|
|
return none().inc_ref();
|
|
}
|
|
PYBIND11_TYPE_CASTER(T, _("None"));
|
|
};
|
|
|
|
template <> class type_caster<void_type> : public void_caster<void_type> {};
|
|
|
|
template <> class type_caster<void> : public type_caster<void_type> {
|
|
public:
|
|
using type_caster<void_type>::cast;
|
|
|
|
bool load(handle h, bool) {
|
|
if (!h) {
|
|
return false;
|
|
} else if (h.is_none()) {
|
|
value = nullptr;
|
|
return true;
|
|
}
|
|
|
|
/* Check if this is a capsule */
|
|
if (isinstance<capsule>(h)) {
|
|
value = reinterpret_borrow<capsule>(h);
|
|
return true;
|
|
}
|
|
|
|
/* Check if this is a C++ type */
|
|
auto &bases = all_type_info((PyTypeObject *) type::handle_of(h).ptr());
|
|
if (bases.size() == 1) { // Only allowing loading from a single-value type
|
|
value = values_and_holders(reinterpret_cast<instance *>(h.ptr())).begin()->value_ptr();
|
|
return true;
|
|
}
|
|
|
|
/* Fail */
|
|
return false;
|
|
}
|
|
|
|
static handle cast(const void *ptr, return_value_policy /* policy */, handle /* parent */) {
|
|
if (ptr)
|
|
return capsule(ptr).release();
|
|
else
|
|
return none().inc_ref();
|
|
}
|
|
|
|
template <typename T> using cast_op_type = void*&;
|
|
operator void *&() { return value; }
|
|
static constexpr auto name = _("capsule");
|
|
private:
|
|
void *value = nullptr;
|
|
};
|
|
|
|
template <> class type_caster<std::nullptr_t> : public void_caster<std::nullptr_t> { };
|
|
|
|
template <> class type_caster<bool> {
|
|
public:
|
|
bool load(handle src, bool convert) {
|
|
if (!src) return false;
|
|
else if (src.ptr() == Py_True) { value = true; return true; }
|
|
else if (src.ptr() == Py_False) { value = false; return true; }
|
|
else if (convert || !std::strcmp("numpy.bool_", Py_TYPE(src.ptr())->tp_name)) {
|
|
// (allow non-implicit conversion for numpy booleans)
|
|
|
|
Py_ssize_t res = -1;
|
|
if (src.is_none()) {
|
|
res = 0; // None is implicitly converted to False
|
|
}
|
|
#if defined(PYPY_VERSION)
|
|
// On PyPy, check that "__bool__" (or "__nonzero__" on Python 2.7) attr exists
|
|
else if (hasattr(src, PYBIND11_BOOL_ATTR)) {
|
|
res = PyObject_IsTrue(src.ptr());
|
|
}
|
|
#else
|
|
// Alternate approach for CPython: this does the same as the above, but optimized
|
|
// using the CPython API so as to avoid an unneeded attribute lookup.
|
|
else if (auto tp_as_number = src.ptr()->ob_type->tp_as_number) {
|
|
if (PYBIND11_NB_BOOL(tp_as_number)) {
|
|
res = (*PYBIND11_NB_BOOL(tp_as_number))(src.ptr());
|
|
}
|
|
}
|
|
#endif
|
|
if (res == 0 || res == 1) {
|
|
value = (bool) res;
|
|
return true;
|
|
} else {
|
|
PyErr_Clear();
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
static handle cast(bool src, return_value_policy /* policy */, handle /* parent */) {
|
|
return handle(src ? Py_True : Py_False).inc_ref();
|
|
}
|
|
PYBIND11_TYPE_CASTER(bool, _("bool"));
|
|
};
|
|
|
|
// Helper class for UTF-{8,16,32} C++ stl strings:
|
|
template <typename StringType, bool IsView = false> struct string_caster {
|
|
using CharT = typename StringType::value_type;
|
|
|
|
// Simplify life by being able to assume standard char sizes (the standard only guarantees
|
|
// minimums, but Python requires exact sizes)
|
|
static_assert(!std::is_same<CharT, char>::value || sizeof(CharT) == 1, "Unsupported char size != 1");
|
|
#if defined(PYBIND11_HAS_U8STRING)
|
|
static_assert(!std::is_same<CharT, char8_t>::value || sizeof(CharT) == 1, "Unsupported char8_t size != 1");
|
|
#endif
|
|
static_assert(!std::is_same<CharT, char16_t>::value || sizeof(CharT) == 2, "Unsupported char16_t size != 2");
|
|
static_assert(!std::is_same<CharT, char32_t>::value || sizeof(CharT) == 4, "Unsupported char32_t size != 4");
|
|
// wchar_t can be either 16 bits (Windows) or 32 (everywhere else)
|
|
static_assert(!std::is_same<CharT, wchar_t>::value || sizeof(CharT) == 2 || sizeof(CharT) == 4,
|
|
"Unsupported wchar_t size != 2/4");
|
|
static constexpr size_t UTF_N = 8 * sizeof(CharT);
|
|
|
|
bool load(handle src, bool) {
|
|
#if PY_MAJOR_VERSION < 3
|
|
object temp;
|
|
#endif
|
|
handle load_src = src;
|
|
if (!src) {
|
|
return false;
|
|
} else if (!PyUnicode_Check(load_src.ptr())) {
|
|
#if PY_MAJOR_VERSION >= 3
|
|
return load_bytes(load_src);
|
|
#else
|
|
if (std::is_same<CharT, char>::value) {
|
|
return load_bytes(load_src);
|
|
}
|
|
|
|
// The below is a guaranteed failure in Python 3 when PyUnicode_Check returns false
|
|
if (!PYBIND11_BYTES_CHECK(load_src.ptr()))
|
|
return false;
|
|
|
|
temp = reinterpret_steal<object>(PyUnicode_FromObject(load_src.ptr()));
|
|
if (!temp) { PyErr_Clear(); return false; }
|
|
load_src = temp;
|
|
#endif
|
|
}
|
|
|
|
auto utfNbytes = reinterpret_steal<object>(PyUnicode_AsEncodedString(
|
|
load_src.ptr(), UTF_N == 8 ? "utf-8" : UTF_N == 16 ? "utf-16" : "utf-32", nullptr));
|
|
if (!utfNbytes) { PyErr_Clear(); return false; }
|
|
|
|
const auto *buffer = reinterpret_cast<const CharT *>(PYBIND11_BYTES_AS_STRING(utfNbytes.ptr()));
|
|
size_t length = (size_t) PYBIND11_BYTES_SIZE(utfNbytes.ptr()) / sizeof(CharT);
|
|
if (UTF_N > 8) { buffer++; length--; } // Skip BOM for UTF-16/32
|
|
value = StringType(buffer, length);
|
|
|
|
// If we're loading a string_view we need to keep the encoded Python object alive:
|
|
if (IsView)
|
|
loader_life_support::add_patient(utfNbytes);
|
|
|
|
return true;
|
|
}
|
|
|
|
static handle cast(const StringType &src, return_value_policy /* policy */, handle /* parent */) {
|
|
const char *buffer = reinterpret_cast<const char *>(src.data());
|
|
auto nbytes = ssize_t(src.size() * sizeof(CharT));
|
|
handle s = decode_utfN(buffer, nbytes);
|
|
if (!s) throw error_already_set();
|
|
return s;
|
|
}
|
|
|
|
PYBIND11_TYPE_CASTER(StringType, _(PYBIND11_STRING_NAME));
|
|
|
|
private:
|
|
static handle decode_utfN(const char *buffer, ssize_t nbytes) {
|
|
#if !defined(PYPY_VERSION)
|
|
return
|
|
UTF_N == 8 ? PyUnicode_DecodeUTF8(buffer, nbytes, nullptr) :
|
|
UTF_N == 16 ? PyUnicode_DecodeUTF16(buffer, nbytes, nullptr, nullptr) :
|
|
PyUnicode_DecodeUTF32(buffer, nbytes, nullptr, nullptr);
|
|
#else
|
|
// PyPy segfaults when on PyUnicode_DecodeUTF16 (and possibly on PyUnicode_DecodeUTF32 as well),
|
|
// so bypass the whole thing by just passing the encoding as a string value, which works properly:
|
|
return PyUnicode_Decode(buffer, nbytes, UTF_N == 8 ? "utf-8" : UTF_N == 16 ? "utf-16" : "utf-32", nullptr);
|
|
#endif
|
|
}
|
|
|
|
// When loading into a std::string or char*, accept a bytes object as-is (i.e.
|
|
// without any encoding/decoding attempt). For other C++ char sizes this is a no-op.
|
|
// which supports loading a unicode from a str, doesn't take this path.
|
|
template <typename C = CharT>
|
|
bool load_bytes(enable_if_t<std::is_same<C, char>::value, handle> src) {
|
|
if (PYBIND11_BYTES_CHECK(src.ptr())) {
|
|
// We were passed a Python 3 raw bytes; accept it into a std::string or char*
|
|
// without any encoding attempt.
|
|
const char *bytes = PYBIND11_BYTES_AS_STRING(src.ptr());
|
|
if (bytes) {
|
|
value = StringType(bytes, (size_t) PYBIND11_BYTES_SIZE(src.ptr()));
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
template <typename C = CharT>
|
|
bool load_bytes(enable_if_t<!std::is_same<C, char>::value, handle>) { return false; }
|
|
};
|
|
|
|
template <typename CharT, class Traits, class Allocator>
|
|
struct type_caster<std::basic_string<CharT, Traits, Allocator>, enable_if_t<is_std_char_type<CharT>::value>>
|
|
: string_caster<std::basic_string<CharT, Traits, Allocator>> {};
|
|
|
|
#ifdef PYBIND11_HAS_STRING_VIEW
|
|
template <typename CharT, class Traits>
|
|
struct type_caster<std::basic_string_view<CharT, Traits>, enable_if_t<is_std_char_type<CharT>::value>>
|
|
: string_caster<std::basic_string_view<CharT, Traits>, true> {};
|
|
#endif
|
|
|
|
// Type caster for C-style strings. We basically use a std::string type caster, but also add the
|
|
// ability to use None as a nullptr char* (which the string caster doesn't allow).
|
|
template <typename CharT> struct type_caster<CharT, enable_if_t<is_std_char_type<CharT>::value>> {
|
|
using StringType = std::basic_string<CharT>;
|
|
using StringCaster = type_caster<StringType>;
|
|
StringCaster str_caster;
|
|
bool none = false;
|
|
CharT one_char = 0;
|
|
public:
|
|
bool load(handle src, bool convert) {
|
|
if (!src) return false;
|
|
if (src.is_none()) {
|
|
// Defer accepting None to other overloads (if we aren't in convert mode):
|
|
if (!convert) return false;
|
|
none = true;
|
|
return true;
|
|
}
|
|
return str_caster.load(src, convert);
|
|
}
|
|
|
|
static handle cast(const CharT *src, return_value_policy policy, handle parent) {
|
|
if (src == nullptr) return pybind11::none().inc_ref();
|
|
return StringCaster::cast(StringType(src), policy, parent);
|
|
}
|
|
|
|
static handle cast(CharT src, return_value_policy policy, handle parent) {
|
|
if (std::is_same<char, CharT>::value) {
|
|
handle s = PyUnicode_DecodeLatin1((const char *) &src, 1, nullptr);
|
|
if (!s) throw error_already_set();
|
|
return s;
|
|
}
|
|
return StringCaster::cast(StringType(1, src), policy, parent);
|
|
}
|
|
|
|
operator CharT*() { return none ? nullptr : const_cast<CharT *>(static_cast<StringType &>(str_caster).c_str()); }
|
|
operator CharT&() {
|
|
if (none)
|
|
throw value_error("Cannot convert None to a character");
|
|
|
|
auto &value = static_cast<StringType &>(str_caster);
|
|
size_t str_len = value.size();
|
|
if (str_len == 0)
|
|
throw value_error("Cannot convert empty string to a character");
|
|
|
|
// If we're in UTF-8 mode, we have two possible failures: one for a unicode character that
|
|
// is too high, and one for multiple unicode characters (caught later), so we need to figure
|
|
// out how long the first encoded character is in bytes to distinguish between these two
|
|
// errors. We also allow want to allow unicode characters U+0080 through U+00FF, as those
|
|
// can fit into a single char value.
|
|
if (StringCaster::UTF_N == 8 && str_len > 1 && str_len <= 4) {
|
|
auto v0 = static_cast<unsigned char>(value[0]);
|
|
size_t char0_bytes = !(v0 & 0x80) ? 1 : // low bits only: 0-127
|
|
(v0 & 0xE0) == 0xC0 ? 2 : // 0b110xxxxx - start of 2-byte sequence
|
|
(v0 & 0xF0) == 0xE0 ? 3 : // 0b1110xxxx - start of 3-byte sequence
|
|
4; // 0b11110xxx - start of 4-byte sequence
|
|
|
|
if (char0_bytes == str_len) {
|
|
// If we have a 128-255 value, we can decode it into a single char:
|
|
if (char0_bytes == 2 && (v0 & 0xFC) == 0xC0) { // 0x110000xx 0x10xxxxxx
|
|
one_char = static_cast<CharT>(((v0 & 3) << 6) + (static_cast<unsigned char>(value[1]) & 0x3F));
|
|
return one_char;
|
|
}
|
|
// Otherwise we have a single character, but it's > U+00FF
|
|
throw value_error("Character code point not in range(0x100)");
|
|
}
|
|
}
|
|
|
|
// UTF-16 is much easier: we can only have a surrogate pair for values above U+FFFF, thus a
|
|
// surrogate pair with total length 2 instantly indicates a range error (but not a "your
|
|
// string was too long" error).
|
|
else if (StringCaster::UTF_N == 16 && str_len == 2) {
|
|
one_char = static_cast<CharT>(value[0]);
|
|
if (one_char >= 0xD800 && one_char < 0xE000)
|
|
throw value_error("Character code point not in range(0x10000)");
|
|
}
|
|
|
|
if (str_len != 1)
|
|
throw value_error("Expected a character, but multi-character string found");
|
|
|
|
one_char = value[0];
|
|
return one_char;
|
|
}
|
|
|
|
static constexpr auto name = _(PYBIND11_STRING_NAME);
|
|
template <typename _T> using cast_op_type = pybind11::detail::cast_op_type<_T>;
|
|
};
|
|
|
|
// Base implementation for std::tuple and std::pair
|
|
template <template<typename...> class Tuple, typename... Ts> class tuple_caster {
|
|
using type = Tuple<Ts...>;
|
|
static constexpr auto size = sizeof...(Ts);
|
|
using indices = make_index_sequence<size>;
|
|
public:
|
|
|
|
bool load(handle src, bool convert) {
|
|
if (!isinstance<sequence>(src))
|
|
return false;
|
|
const auto seq = reinterpret_borrow<sequence>(src);
|
|
if (seq.size() != size)
|
|
return false;
|
|
return load_impl(seq, convert, indices{});
|
|
}
|
|
|
|
template <typename T>
|
|
static handle cast(T &&src, return_value_policy policy, handle parent) {
|
|
return cast_impl(std::forward<T>(src), policy, parent, indices{});
|
|
}
|
|
|
|
// copied from the PYBIND11_TYPE_CASTER macro
|
|
template <typename T>
|
|
static handle cast(T *src, return_value_policy policy, handle parent) {
|
|
if (!src) return none().release();
|
|
if (policy == return_value_policy::take_ownership) {
|
|
auto h = cast(std::move(*src), policy, parent); delete src; return h;
|
|
} else {
|
|
return cast(*src, policy, parent);
|
|
}
|
|
}
|
|
|
|
static constexpr auto name = _("Tuple[") + concat(make_caster<Ts>::name...) + _("]");
|
|
|
|
template <typename T> using cast_op_type = type;
|
|
|
|
operator type() & { return implicit_cast(indices{}); }
|
|
operator type() && { return std::move(*this).implicit_cast(indices{}); }
|
|
|
|
protected:
|
|
template <size_t... Is>
|
|
type implicit_cast(index_sequence<Is...>) & { return type(cast_op<Ts>(std::get<Is>(subcasters))...); }
|
|
template <size_t... Is>
|
|
type implicit_cast(index_sequence<Is...>) && { return type(cast_op<Ts>(std::move(std::get<Is>(subcasters)))...); }
|
|
|
|
static constexpr bool load_impl(const sequence &, bool, index_sequence<>) { return true; }
|
|
|
|
template <size_t... Is>
|
|
bool load_impl(const sequence &seq, bool convert, index_sequence<Is...>) {
|
|
#ifdef __cpp_fold_expressions
|
|
if ((... || !std::get<Is>(subcasters).load(seq[Is], convert)))
|
|
return false;
|
|
#else
|
|
for (bool r : {std::get<Is>(subcasters).load(seq[Is], convert)...})
|
|
if (!r)
|
|
return false;
|
|
#endif
|
|
return true;
|
|
}
|
|
|
|
/* Implementation: Convert a C++ tuple into a Python tuple */
|
|
template <typename T, size_t... Is>
|
|
static handle cast_impl(T &&src, return_value_policy policy, handle parent, index_sequence<Is...>) {
|
|
std::array<object, size> entries{{
|
|
reinterpret_steal<object>(make_caster<Ts>::cast(std::get<Is>(std::forward<T>(src)), policy, parent))...
|
|
}};
|
|
for (const auto &entry: entries)
|
|
if (!entry)
|
|
return handle();
|
|
tuple result(size);
|
|
int counter = 0;
|
|
for (auto & entry: entries)
|
|
PyTuple_SET_ITEM(result.ptr(), counter++, entry.release().ptr());
|
|
return result.release();
|
|
}
|
|
|
|
Tuple<make_caster<Ts>...> subcasters;
|
|
};
|
|
|
|
template <typename T1, typename T2> class type_caster<std::pair<T1, T2>>
|
|
: public tuple_caster<std::pair, T1, T2> {};
|
|
|
|
template <typename... Ts> class type_caster<std::tuple<Ts...>>
|
|
: public tuple_caster<std::tuple, Ts...> {};
|
|
|
|
/// Helper class which abstracts away certain actions. Users can provide specializations for
|
|
/// custom holders, but it's only necessary if the type has a non-standard interface.
|
|
template <typename T>
|
|
struct holder_helper {
|
|
static auto get(const T &p) -> decltype(p.get()) { return p.get(); }
|
|
};
|
|
|
|
/// Type caster for holder types like std::shared_ptr, etc.
|
|
/// The SFINAE hook is provided to help work around the current lack of support
|
|
/// for smart-pointer interoperability. Please consider it an implementation
|
|
/// detail that may change in the future, as formal support for smart-pointer
|
|
/// interoperability is added into pybind11.
|
|
template <typename type, typename holder_type, typename SFINAE = void>
|
|
struct copyable_holder_caster : public type_caster_base<type> {
|
|
public:
|
|
using base = type_caster_base<type>;
|
|
static_assert(std::is_base_of<base, type_caster<type>>::value,
|
|
"Holder classes are only supported for custom types");
|
|
using base::base;
|
|
using base::cast;
|
|
using base::typeinfo;
|
|
using base::value;
|
|
|
|
bool load(handle src, bool convert) {
|
|
return base::template load_impl<copyable_holder_caster<type, holder_type>>(src, convert);
|
|
}
|
|
|
|
explicit operator type*() { return this->value; }
|
|
// static_cast works around compiler error with MSVC 17 and CUDA 10.2
|
|
// see issue #2180
|
|
explicit operator type&() { return *(static_cast<type *>(this->value)); }
|
|
explicit operator holder_type*() { return std::addressof(holder); }
|
|
explicit operator holder_type&() { return holder; }
|
|
|
|
static handle cast(const holder_type &src, return_value_policy, handle) {
|
|
const auto *ptr = holder_helper<holder_type>::get(src);
|
|
return type_caster_base<type>::cast_holder(ptr, &src);
|
|
}
|
|
|
|
protected:
|
|
friend class type_caster_generic;
|
|
void check_holder_compat() {
|
|
if (typeinfo->default_holder)
|
|
throw cast_error("Unable to load a custom holder type from a default-holder instance");
|
|
}
|
|
|
|
bool load_value(value_and_holder &&v_h) {
|
|
if (v_h.holder_constructed()) {
|
|
value = v_h.value_ptr();
|
|
holder = v_h.template holder<holder_type>();
|
|
return true;
|
|
} else {
|
|
throw cast_error("Unable to cast from non-held to held instance (T& to Holder<T>) "
|
|
#if defined(NDEBUG)
|
|
"(compile in debug mode for type information)");
|
|
#else
|
|
"of type '" + type_id<holder_type>() + "''");
|
|
#endif
|
|
}
|
|
}
|
|
|
|
template <typename T = holder_type, detail::enable_if_t<!std::is_constructible<T, const T &, type*>::value, int> = 0>
|
|
bool try_implicit_casts(handle, bool) { return false; }
|
|
|
|
template <typename T = holder_type, detail::enable_if_t<std::is_constructible<T, const T &, type*>::value, int> = 0>
|
|
bool try_implicit_casts(handle src, bool convert) {
|
|
for (auto &cast : typeinfo->implicit_casts) {
|
|
copyable_holder_caster sub_caster(*cast.first);
|
|
if (sub_caster.load(src, convert)) {
|
|
value = cast.second(sub_caster.value);
|
|
holder = holder_type(sub_caster.holder, (type *) value);
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static bool try_direct_conversions(handle) { return false; }
|
|
|
|
|
|
holder_type holder;
|
|
};
|
|
|
|
#ifndef PYBIND11_USE_SMART_HOLDER_AS_DEFAULT
|
|
/// Specialize for the common std::shared_ptr, so users don't need to
|
|
template <typename T>
|
|
class type_caster<std::shared_ptr<T>> : public copyable_holder_caster<T, std::shared_ptr<T>> { };
|
|
#endif
|
|
|
|
/// Type caster for holder types like std::unique_ptr.
|
|
/// Please consider the SFINAE hook an implementation detail, as explained
|
|
/// in the comment for the copyable_holder_caster.
|
|
template <typename type, typename holder_type, typename SFINAE = void>
|
|
struct move_only_holder_caster {
|
|
static_assert(std::is_base_of<type_caster_base<type>, type_caster<type>>::value,
|
|
"Holder classes are only supported for custom types");
|
|
|
|
static handle cast(holder_type &&src, return_value_policy, handle) {
|
|
auto *ptr = holder_helper<holder_type>::get(src);
|
|
return type_caster_base<type>::cast_holder(ptr, std::addressof(src));
|
|
}
|
|
static constexpr auto name = type_caster_base<type>::name;
|
|
};
|
|
|
|
#ifndef PYBIND11_USE_SMART_HOLDER_AS_DEFAULT
|
|
template <typename type, typename deleter>
|
|
class type_caster<std::unique_ptr<type, deleter>>
|
|
: public move_only_holder_caster<type, std::unique_ptr<type, deleter>> { };
|
|
#endif
|
|
|
|
template <typename type, typename holder_type>
|
|
using type_caster_holder = conditional_t<is_copy_constructible<holder_type>::value,
|
|
copyable_holder_caster<type, holder_type>,
|
|
move_only_holder_caster<type, holder_type>>;
|
|
|
|
template <typename T, bool Value = false> struct always_construct_holder { static constexpr bool value = Value; };
|
|
|
|
/// Create a specialization for custom holder types (silently ignores std::shared_ptr)
|
|
#define PYBIND11_DECLARE_HOLDER_TYPE(type, holder_type, ...) \
|
|
namespace pybind11 { namespace detail { \
|
|
template <typename type> \
|
|
struct always_construct_holder<holder_type> : always_construct_holder<void, ##__VA_ARGS__> { }; \
|
|
template <typename type> \
|
|
class type_caster<holder_type, enable_if_t<!is_shared_ptr<holder_type>::value>> \
|
|
: public type_caster_holder<type, holder_type> { }; \
|
|
}}
|
|
|
|
// PYBIND11_DECLARE_HOLDER_TYPE holder types:
|
|
template <typename base, typename holder> struct is_holder_type :
|
|
std::is_base_of<detail::type_caster_holder<base, holder>, detail::type_caster<holder>> {};
|
|
// Specialization for always-supported unique_ptr holders:
|
|
template <typename base, typename deleter> struct is_holder_type<base, std::unique_ptr<base, deleter>> :
|
|
std::true_type {};
|
|
|
|
template <typename T> struct handle_type_name { static constexpr auto name = _<T>(); };
|
|
template <> struct handle_type_name<bytes> { static constexpr auto name = _(PYBIND11_BYTES_NAME); };
|
|
template <> struct handle_type_name<int_> { static constexpr auto name = _("int"); };
|
|
template <> struct handle_type_name<iterable> { static constexpr auto name = _("Iterable"); };
|
|
template <> struct handle_type_name<iterator> { static constexpr auto name = _("Iterator"); };
|
|
template <> struct handle_type_name<none> { static constexpr auto name = _("None"); };
|
|
template <> struct handle_type_name<args> { static constexpr auto name = _("*args"); };
|
|
template <> struct handle_type_name<kwargs> { static constexpr auto name = _("**kwargs"); };
|
|
|
|
template <typename type>
|
|
struct pyobject_caster {
|
|
template <typename T = type, enable_if_t<std::is_same<T, handle>::value, int> = 0>
|
|
bool load(handle src, bool /* convert */) { value = src; return static_cast<bool>(value); }
|
|
|
|
template <typename T = type, enable_if_t<std::is_base_of<object, T>::value, int> = 0>
|
|
bool load(handle src, bool /* convert */) {
|
|
#if PY_MAJOR_VERSION < 3 && !defined(PYBIND11_STR_LEGACY_PERMISSIVE)
|
|
// For Python 2, without this implicit conversion, Python code would
|
|
// need to be cluttered with six.ensure_text() or similar, only to be
|
|
// un-cluttered later after Python 2 support is dropped.
|
|
if (std::is_same<T, str>::value && isinstance<bytes>(src)) {
|
|
PyObject *str_from_bytes = PyUnicode_FromEncodedObject(src.ptr(), "utf-8", nullptr);
|
|
if (!str_from_bytes) throw error_already_set();
|
|
value = reinterpret_steal<type>(str_from_bytes);
|
|
return true;
|
|
}
|
|
#endif
|
|
if (!isinstance<type>(src))
|
|
return false;
|
|
value = reinterpret_borrow<type>(src);
|
|
return true;
|
|
}
|
|
|
|
static handle cast(const handle &src, return_value_policy /* policy */, handle /* parent */) {
|
|
return src.inc_ref();
|
|
}
|
|
PYBIND11_TYPE_CASTER(type, handle_type_name<type>::name);
|
|
};
|
|
|
|
template <typename T>
|
|
class type_caster<T, enable_if_t<is_pyobject<T>::value>> : public pyobject_caster<T> { };
|
|
|
|
// Our conditions for enabling moving are quite restrictive:
|
|
// At compile time:
|
|
// - T needs to be a non-const, non-pointer, non-reference type
|
|
// - type_caster<T>::operator T&() must exist
|
|
// - the type must be move constructible (obviously)
|
|
// At run-time:
|
|
// - if the type is non-copy-constructible, the object must be the sole owner of the type (i.e. it
|
|
// must have ref_count() == 1)h
|
|
// If any of the above are not satisfied, we fall back to copying.
|
|
template <typename T> using move_is_plain_type = satisfies_none_of<T,
|
|
std::is_void, std::is_pointer, std::is_reference, std::is_const
|
|
>;
|
|
template <typename T, typename SFINAE = void> struct move_always : std::false_type {};
|
|
template <typename T> struct move_always<T, enable_if_t<all_of<
|
|
move_is_plain_type<T>,
|
|
negation<is_copy_constructible<T>>,
|
|
std::is_move_constructible<T>,
|
|
std::is_same<decltype(std::declval<make_caster<T>>().operator T&()), T&>
|
|
>::value>> : std::true_type {};
|
|
template <typename T, typename SFINAE = void> struct move_if_unreferenced : std::false_type {};
|
|
template <typename T> struct move_if_unreferenced<T, enable_if_t<all_of<
|
|
move_is_plain_type<T>,
|
|
negation<move_always<T>>,
|
|
std::is_move_constructible<T>,
|
|
std::is_same<decltype(std::declval<make_caster<T>>().operator T&()), T&>
|
|
>::value>> : std::true_type {};
|
|
template <typename T> using move_never = none_of<move_always<T>, move_if_unreferenced<T>>;
|
|
|
|
// Detect whether returning a `type` from a cast on type's type_caster is going to result in a
|
|
// reference or pointer to a local variable of the type_caster. Basically, only
|
|
// non-reference/pointer `type`s and reference/pointers from a type_caster_generic are safe;
|
|
// everything else returns a reference/pointer to a local variable.
|
|
template <typename type> using cast_is_temporary_value_reference = bool_constant<
|
|
(std::is_reference<type>::value || std::is_pointer<type>::value) &&
|
|
!std::is_base_of<type_caster_generic, make_caster<type>>::value &&
|
|
!type_uses_smart_holder_type_caster<intrinsic_t<type>>::value &&
|
|
!std::is_same<intrinsic_t<type>, void>::value
|
|
>;
|
|
|
|
// When a value returned from a C++ function is being cast back to Python, we almost always want to
|
|
// force `policy = move`, regardless of the return value policy the function/method was declared
|
|
// with.
|
|
template <typename Return, typename SFINAE = void> struct return_value_policy_override {
|
|
static return_value_policy policy(return_value_policy p) { return p; }
|
|
};
|
|
|
|
template <typename Return> struct return_value_policy_override<Return,
|
|
detail::enable_if_t<std::is_base_of<type_caster_generic, make_caster<Return>>::value, void>> {
|
|
static return_value_policy policy(return_value_policy p) {
|
|
return !std::is_lvalue_reference<Return>::value &&
|
|
!std::is_pointer<Return>::value
|
|
? return_value_policy::move : p;
|
|
}
|
|
};
|
|
|
|
// Basic python -> C++ casting; throws if casting fails
|
|
template <typename T, typename SFINAE> type_caster<T, SFINAE> &load_type(type_caster<T, SFINAE> &conv, const handle &handle) {
|
|
if (!conv.load(handle, true)) {
|
|
#if defined(NDEBUG)
|
|
throw cast_error("Unable to cast Python instance to C++ type (compile in debug mode for details)");
|
|
#else
|
|
throw cast_error("Unable to cast Python instance of type " +
|
|
(std::string) str(type::handle_of(handle)) + " to C++ type '" + type_id<T>() + "'");
|
|
#endif
|
|
}
|
|
return conv;
|
|
}
|
|
// Wrapper around the above that also constructs and returns a type_caster
|
|
template <typename T> make_caster<T> load_type(const handle &handle) {
|
|
make_caster<T> conv;
|
|
load_type(conv, handle);
|
|
return conv;
|
|
}
|
|
|
|
PYBIND11_NAMESPACE_END(detail)
|
|
|
|
// pytype -> C++ type
|
|
template <typename T, detail::enable_if_t<!detail::is_pyobject<T>::value, int> = 0>
|
|
T cast(const handle &handle) {
|
|
using namespace detail;
|
|
static_assert(!cast_is_temporary_value_reference<T>::value,
|
|
"Unable to cast type to reference: value is local to type caster");
|
|
return cast_op<T>(load_type<T>(handle));
|
|
}
|
|
|
|
// pytype -> pytype (calls converting constructor)
|
|
template <typename T, detail::enable_if_t<detail::is_pyobject<T>::value, int> = 0>
|
|
T cast(const handle &handle) { return T(reinterpret_borrow<object>(handle)); }
|
|
|
|
// C++ type -> py::object
|
|
template <typename T, detail::enable_if_t<!detail::is_pyobject<T>::value, int> = 0>
|
|
object cast(T &&value, return_value_policy policy = return_value_policy::automatic_reference,
|
|
handle parent = handle()) {
|
|
using no_ref_T = typename std::remove_reference<T>::type;
|
|
if (policy == return_value_policy::automatic)
|
|
policy = std::is_pointer<no_ref_T>::value ? return_value_policy::take_ownership :
|
|
std::is_lvalue_reference<T>::value ? return_value_policy::copy : return_value_policy::move;
|
|
else if (policy == return_value_policy::automatic_reference)
|
|
policy = std::is_pointer<no_ref_T>::value ? return_value_policy::reference :
|
|
std::is_lvalue_reference<T>::value ? return_value_policy::copy : return_value_policy::move;
|
|
return reinterpret_steal<object>(detail::make_caster<T>::cast(std::forward<T>(value), policy, parent));
|
|
}
|
|
|
|
template <typename T> T handle::cast() const { return pybind11::cast<T>(*this); }
|
|
template <> inline void handle::cast() const { return; }
|
|
|
|
template <typename T>
|
|
detail::enable_if_t<!detail::move_never<T>::value, T> move(object &&obj) {
|
|
if (obj.ref_count() > 1)
|
|
#if defined(NDEBUG)
|
|
throw cast_error("Unable to cast Python instance to C++ rvalue: instance has multiple references"
|
|
" (compile in debug mode for details)");
|
|
#else
|
|
throw cast_error("Unable to move from Python " + (std::string) str(type::handle_of(obj)) +
|
|
" instance to C++ " + type_id<T>() + " instance: instance has multiple references");
|
|
#endif
|
|
|
|
// Move into a temporary and return that, because the reference may be a local value of `conv`
|
|
T ret = std::move(detail::load_type<T>(obj).operator T&());
|
|
return ret;
|
|
}
|
|
|
|
// Calling cast() on an rvalue calls pybind11::cast with the object rvalue, which does:
|
|
// - If we have to move (because T has no copy constructor), do it. This will fail if the moved
|
|
// object has multiple references, but trying to copy will fail to compile.
|
|
// - If both movable and copyable, check ref count: if 1, move; otherwise copy
|
|
// - Otherwise (not movable), copy.
|
|
template <typename T> detail::enable_if_t<detail::move_always<T>::value, T> cast(object &&object) {
|
|
return move<T>(std::move(object));
|
|
}
|
|
template <typename T> detail::enable_if_t<detail::move_if_unreferenced<T>::value, T> cast(object &&object) {
|
|
if (object.ref_count() > 1)
|
|
return cast<T>(object);
|
|
else
|
|
return move<T>(std::move(object));
|
|
}
|
|
template <typename T> detail::enable_if_t<detail::move_never<T>::value, T> cast(object &&object) {
|
|
return cast<T>(object);
|
|
}
|
|
|
|
template <typename T> T object::cast() const & { return pybind11::cast<T>(*this); }
|
|
template <typename T> T object::cast() && { return pybind11::cast<T>(std::move(*this)); }
|
|
template <> inline void object::cast() const & { return; }
|
|
template <> inline void object::cast() && { return; }
|
|
|
|
PYBIND11_NAMESPACE_BEGIN(detail)
|
|
|
|
// Declared in pytypes.h:
|
|
template <typename T, enable_if_t<!is_pyobject<T>::value, int>>
|
|
object object_or_cast(T &&o) { return pybind11::cast(std::forward<T>(o)); }
|
|
|
|
struct override_unused {}; // Placeholder type for the unneeded (and dead code) static variable in the PYBIND11_OVERRIDE_OVERRIDE macro
|
|
template <typename ret_type> using override_caster_t = conditional_t<
|
|
cast_is_temporary_value_reference<ret_type>::value, make_caster<ret_type>, override_unused>;
|
|
|
|
// Trampoline use: for reference/pointer types to value-converted values, we do a value cast, then
|
|
// store the result in the given variable. For other types, this is a no-op.
|
|
template <typename T> enable_if_t<cast_is_temporary_value_reference<T>::value, T> cast_ref(object &&o, make_caster<T> &caster) {
|
|
return cast_op<T>(load_type(caster, o));
|
|
}
|
|
template <typename T> enable_if_t<!cast_is_temporary_value_reference<T>::value, T> cast_ref(object &&, override_unused &) {
|
|
pybind11_fail("Internal error: cast_ref fallback invoked"); }
|
|
|
|
// Trampoline use: Having a pybind11::cast with an invalid reference type is going to static_assert, even
|
|
// though if it's in dead code, so we provide a "trampoline" to pybind11::cast that only does anything in
|
|
// cases where pybind11::cast is valid.
|
|
template <typename T> enable_if_t<!cast_is_temporary_value_reference<T>::value, T> cast_safe(object &&o) {
|
|
return pybind11::cast<T>(std::move(o)); }
|
|
template <typename T> enable_if_t<cast_is_temporary_value_reference<T>::value, T> cast_safe(object &&) {
|
|
pybind11_fail("Internal error: cast_safe fallback invoked"); }
|
|
template <> inline void cast_safe<void>(object &&) {}
|
|
|
|
PYBIND11_NAMESPACE_END(detail)
|
|
|
|
template <return_value_policy policy = return_value_policy::automatic_reference>
|
|
tuple make_tuple() { return tuple(0); }
|
|
|
|
template <return_value_policy policy = return_value_policy::automatic_reference,
|
|
typename... Args> tuple make_tuple(Args&&... args_) {
|
|
constexpr size_t size = sizeof...(Args);
|
|
std::array<object, size> args {
|
|
{ reinterpret_steal<object>(detail::make_caster<Args>::cast(
|
|
std::forward<Args>(args_), policy, nullptr))... }
|
|
};
|
|
for (size_t i = 0; i < args.size(); i++) {
|
|
if (!args[i]) {
|
|
#if defined(NDEBUG)
|
|
throw cast_error("make_tuple(): unable to convert arguments to Python object (compile in debug mode for details)");
|
|
#else
|
|
std::array<std::string, size> argtypes { {type_id<Args>()...} };
|
|
throw cast_error("make_tuple(): unable to convert argument of type '" +
|
|
argtypes[i] + "' to Python object");
|
|
#endif
|
|
}
|
|
}
|
|
tuple result(size);
|
|
int counter = 0;
|
|
for (auto &arg_value : args)
|
|
PyTuple_SET_ITEM(result.ptr(), counter++, arg_value.release().ptr());
|
|
return result;
|
|
}
|
|
|
|
/// \ingroup annotations
|
|
/// Annotation for arguments
|
|
struct arg {
|
|
/// Constructs an argument with the name of the argument; if null or omitted, this is a positional argument.
|
|
constexpr explicit arg(const char *name = nullptr) : name(name), flag_noconvert(false), flag_none(true) { }
|
|
/// Assign a value to this argument
|
|
template <typename T> arg_v operator=(T &&value) const;
|
|
/// Indicate that the type should not be converted in the type caster
|
|
arg &noconvert(bool flag = true) { flag_noconvert = flag; return *this; }
|
|
/// Indicates that the argument should/shouldn't allow None (e.g. for nullable pointer args)
|
|
arg &none(bool flag = true) { flag_none = flag; return *this; }
|
|
|
|
const char *name; ///< If non-null, this is a named kwargs argument
|
|
bool flag_noconvert : 1; ///< If set, do not allow conversion (requires a supporting type caster!)
|
|
bool flag_none : 1; ///< If set (the default), allow None to be passed to this argument
|
|
};
|
|
|
|
/// \ingroup annotations
|
|
/// Annotation for arguments with values
|
|
struct arg_v : arg {
|
|
private:
|
|
template <typename T>
|
|
arg_v(arg &&base, T &&x, const char *descr = nullptr)
|
|
: arg(base),
|
|
value(reinterpret_steal<object>(
|
|
detail::make_caster<T>::cast(x, return_value_policy::automatic, {})
|
|
)),
|
|
descr(descr)
|
|
#if !defined(NDEBUG)
|
|
, type(type_id<T>())
|
|
#endif
|
|
{
|
|
// Workaround! See:
|
|
// https://github.com/pybind/pybind11/issues/2336
|
|
// https://github.com/pybind/pybind11/pull/2685#issuecomment-731286700
|
|
if (PyErr_Occurred()) {
|
|
PyErr_Clear();
|
|
}
|
|
}
|
|
|
|
public:
|
|
/// Direct construction with name, default, and description
|
|
template <typename T>
|
|
arg_v(const char *name, T &&x, const char *descr = nullptr)
|
|
: arg_v(arg(name), std::forward<T>(x), descr) { }
|
|
|
|
/// Called internally when invoking `py::arg("a") = value`
|
|
template <typename T>
|
|
arg_v(const arg &base, T &&x, const char *descr = nullptr)
|
|
: arg_v(arg(base), std::forward<T>(x), descr) { }
|
|
|
|
/// Same as `arg::noconvert()`, but returns *this as arg_v&, not arg&
|
|
arg_v &noconvert(bool flag = true) { arg::noconvert(flag); return *this; }
|
|
|
|
/// Same as `arg::nonone()`, but returns *this as arg_v&, not arg&
|
|
arg_v &none(bool flag = true) { arg::none(flag); return *this; }
|
|
|
|
/// The default value
|
|
object value;
|
|
/// The (optional) description of the default value
|
|
const char *descr;
|
|
#if !defined(NDEBUG)
|
|
/// The C++ type name of the default value (only available when compiled in debug mode)
|
|
std::string type;
|
|
#endif
|
|
};
|
|
|
|
/// \ingroup annotations
|
|
/// Annotation indicating that all following arguments are keyword-only; the is the equivalent of an
|
|
/// unnamed '*' argument (in Python 3)
|
|
struct kw_only {};
|
|
|
|
/// \ingroup annotations
|
|
/// Annotation indicating that all previous arguments are positional-only; the is the equivalent of an
|
|
/// unnamed '/' argument (in Python 3.8)
|
|
struct pos_only {};
|
|
|
|
template <typename T>
|
|
arg_v arg::operator=(T &&value) const {
|
|
return {*this, std::forward<T>(value)};
|
|
}
|
|
|
|
/// Alias for backward compatibility -- to be removed in version 2.0
|
|
template <typename /*unused*/> using arg_t = arg_v;
|
|
|
|
inline namespace literals {
|
|
/** \rst
|
|
String literal version of `arg`
|
|
\endrst */
|
|
constexpr arg operator"" _a(const char *name, size_t) { return arg(name); }
|
|
} // namespace literals
|
|
|
|
PYBIND11_NAMESPACE_BEGIN(detail)
|
|
|
|
// forward declaration (definition in attr.h)
|
|
struct function_record;
|
|
|
|
/// Internal data associated with a single function call
|
|
struct function_call {
|
|
function_call(const function_record &f, handle p); // Implementation in attr.h
|
|
|
|
/// The function data:
|
|
const function_record &func;
|
|
|
|
/// Arguments passed to the function:
|
|
std::vector<handle> args;
|
|
|
|
/// The `convert` value the arguments should be loaded with
|
|
std::vector<bool> args_convert;
|
|
|
|
/// Extra references for the optional `py::args` and/or `py::kwargs` arguments (which, if
|
|
/// present, are also in `args` but without a reference).
|
|
object args_ref, kwargs_ref;
|
|
|
|
/// The parent, if any
|
|
handle parent;
|
|
|
|
/// If this is a call to an initializer, this argument contains `self`
|
|
handle init_self;
|
|
};
|
|
|
|
|
|
/// Helper class which loads arguments for C++ functions called from Python
|
|
template <typename... Args>
|
|
class argument_loader {
|
|
using indices = make_index_sequence<sizeof...(Args)>;
|
|
|
|
template <typename Arg> using argument_is_args = std::is_same<intrinsic_t<Arg>, args>;
|
|
template <typename Arg> using argument_is_kwargs = std::is_same<intrinsic_t<Arg>, kwargs>;
|
|
// Get args/kwargs argument positions relative to the end of the argument list:
|
|
static constexpr auto args_pos = constexpr_first<argument_is_args, Args...>() - (int) sizeof...(Args),
|
|
kwargs_pos = constexpr_first<argument_is_kwargs, Args...>() - (int) sizeof...(Args);
|
|
|
|
static constexpr bool args_kwargs_are_last = kwargs_pos >= - 1 && args_pos >= kwargs_pos - 1;
|
|
|
|
static_assert(args_kwargs_are_last, "py::args/py::kwargs are only permitted as the last argument(s) of a function");
|
|
|
|
public:
|
|
static constexpr bool has_kwargs = kwargs_pos < 0;
|
|
static constexpr bool has_args = args_pos < 0;
|
|
|
|
static constexpr auto arg_names = concat(type_descr(make_caster<Args>::name)...);
|
|
|
|
bool load_args(function_call &call) {
|
|
return load_impl_sequence(call, indices{});
|
|
}
|
|
|
|
template <typename Return, typename Guard, typename Func>
|
|
enable_if_t<!std::is_void<Return>::value, Return> call(Func &&f) && {
|
|
return std::move(*this).template call_impl<Return>(std::forward<Func>(f), indices{}, Guard{});
|
|
}
|
|
|
|
template <typename Return, typename Guard, typename Func>
|
|
enable_if_t<std::is_void<Return>::value, void_type> call(Func &&f) && {
|
|
std::move(*this).template call_impl<Return>(std::forward<Func>(f), indices{}, Guard{});
|
|
return void_type();
|
|
}
|
|
|
|
private:
|
|
|
|
static bool load_impl_sequence(function_call &, index_sequence<>) { return true; }
|
|
|
|
template <size_t... Is>
|
|
bool load_impl_sequence(function_call &call, index_sequence<Is...>) {
|
|
#ifdef __cpp_fold_expressions
|
|
if ((... || !std::get<Is>(argcasters).load(call.args[Is], call.args_convert[Is])))
|
|
return false;
|
|
#else
|
|
for (bool r : {std::get<Is>(argcasters).load(call.args[Is], call.args_convert[Is])...})
|
|
if (!r)
|
|
return false;
|
|
#endif
|
|
return true;
|
|
}
|
|
|
|
template <typename Return, typename Func, size_t... Is, typename Guard>
|
|
Return call_impl(Func &&f, index_sequence<Is...>, Guard &&) && {
|
|
return std::forward<Func>(f)(cast_op<Args>(std::move(std::get<Is>(argcasters)))...);
|
|
}
|
|
|
|
std::tuple<make_caster<Args>...> argcasters;
|
|
};
|
|
|
|
/// Helper class which collects only positional arguments for a Python function call.
|
|
/// A fancier version below can collect any argument, but this one is optimal for simple calls.
|
|
template <return_value_policy policy>
|
|
class simple_collector {
|
|
public:
|
|
template <typename... Ts>
|
|
explicit simple_collector(Ts &&...values)
|
|
: m_args(pybind11::make_tuple<policy>(std::forward<Ts>(values)...)) { }
|
|
|
|
const tuple &args() const & { return m_args; }
|
|
dict kwargs() const { return {}; }
|
|
|
|
tuple args() && { return std::move(m_args); }
|
|
|
|
/// Call a Python function and pass the collected arguments
|
|
object call(PyObject *ptr) const {
|
|
PyObject *result = PyObject_CallObject(ptr, m_args.ptr());
|
|
if (!result)
|
|
throw error_already_set();
|
|
return reinterpret_steal<object>(result);
|
|
}
|
|
|
|
private:
|
|
tuple m_args;
|
|
};
|
|
|
|
/// Helper class which collects positional, keyword, * and ** arguments for a Python function call
|
|
template <return_value_policy policy>
|
|
class unpacking_collector {
|
|
public:
|
|
template <typename... Ts>
|
|
explicit unpacking_collector(Ts &&...values) {
|
|
// Tuples aren't (easily) resizable so a list is needed for collection,
|
|
// but the actual function call strictly requires a tuple.
|
|
auto args_list = list();
|
|
int _[] = { 0, (process(args_list, std::forward<Ts>(values)), 0)... };
|
|
ignore_unused(_);
|
|
|
|
m_args = std::move(args_list);
|
|
}
|
|
|
|
const tuple &args() const & { return m_args; }
|
|
const dict &kwargs() const & { return m_kwargs; }
|
|
|
|
tuple args() && { return std::move(m_args); }
|
|
dict kwargs() && { return std::move(m_kwargs); }
|
|
|
|
/// Call a Python function and pass the collected arguments
|
|
object call(PyObject *ptr) const {
|
|
PyObject *result = PyObject_Call(ptr, m_args.ptr(), m_kwargs.ptr());
|
|
if (!result)
|
|
throw error_already_set();
|
|
return reinterpret_steal<object>(result);
|
|
}
|
|
|
|
private:
|
|
template <typename T>
|
|
void process(list &args_list, T &&x) {
|
|
auto o = reinterpret_steal<object>(detail::make_caster<T>::cast(std::forward<T>(x), policy, {}));
|
|
if (!o) {
|
|
#if defined(NDEBUG)
|
|
argument_cast_error();
|
|
#else
|
|
argument_cast_error(std::to_string(args_list.size()), type_id<T>());
|
|
#endif
|
|
}
|
|
args_list.append(o);
|
|
}
|
|
|
|
void process(list &args_list, detail::args_proxy ap) {
|
|
for (auto a : ap)
|
|
args_list.append(a);
|
|
}
|
|
|
|
void process(list &/*args_list*/, arg_v a) {
|
|
if (!a.name)
|
|
#if defined(NDEBUG)
|
|
nameless_argument_error();
|
|
#else
|
|
nameless_argument_error(a.type);
|
|
#endif
|
|
|
|
if (m_kwargs.contains(a.name)) {
|
|
#if defined(NDEBUG)
|
|
multiple_values_error();
|
|
#else
|
|
multiple_values_error(a.name);
|
|
#endif
|
|
}
|
|
if (!a.value) {
|
|
#if defined(NDEBUG)
|
|
argument_cast_error();
|
|
#else
|
|
argument_cast_error(a.name, a.type);
|
|
#endif
|
|
}
|
|
m_kwargs[a.name] = a.value;
|
|
}
|
|
|
|
void process(list &/*args_list*/, detail::kwargs_proxy kp) {
|
|
if (!kp)
|
|
return;
|
|
for (auto k : reinterpret_borrow<dict>(kp)) {
|
|
if (m_kwargs.contains(k.first)) {
|
|
#if defined(NDEBUG)
|
|
multiple_values_error();
|
|
#else
|
|
multiple_values_error(str(k.first));
|
|
#endif
|
|
}
|
|
m_kwargs[k.first] = k.second;
|
|
}
|
|
}
|
|
|
|
[[noreturn]] static void nameless_argument_error() {
|
|
throw type_error("Got kwargs without a name; only named arguments "
|
|
"may be passed via py::arg() to a python function call. "
|
|
"(compile in debug mode for details)");
|
|
}
|
|
[[noreturn]] static void nameless_argument_error(const std::string &type) {
|
|
throw type_error("Got kwargs without a name of type '" + type + "'; only named "
|
|
"arguments may be passed via py::arg() to a python function call. ");
|
|
}
|
|
[[noreturn]] static void multiple_values_error() {
|
|
throw type_error("Got multiple values for keyword argument "
|
|
"(compile in debug mode for details)");
|
|
}
|
|
|
|
[[noreturn]] static void multiple_values_error(const std::string &name) {
|
|
throw type_error("Got multiple values for keyword argument '" + name + "'");
|
|
}
|
|
|
|
[[noreturn]] static void argument_cast_error() {
|
|
throw cast_error("Unable to convert call argument to Python object "
|
|
"(compile in debug mode for details)");
|
|
}
|
|
|
|
[[noreturn]] static void argument_cast_error(const std::string &name,
|
|
const std::string &type) {
|
|
throw cast_error("Unable to convert call argument '" + name
|
|
+ "' of type '" + type + "' to Python object");
|
|
}
|
|
|
|
private:
|
|
tuple m_args;
|
|
dict m_kwargs;
|
|
};
|
|
|
|
// [workaround(intel)] Separate function required here
|
|
// We need to put this into a separate function because the Intel compiler
|
|
// fails to compile enable_if_t<!all_of<is_positional<Args>...>::value>
|
|
// (tested with ICC 2021.1 Beta 20200827).
|
|
template <typename... Args>
|
|
constexpr bool args_are_all_positional()
|
|
{
|
|
return all_of<is_positional<Args>...>::value;
|
|
}
|
|
|
|
/// Collect only positional arguments for a Python function call
|
|
template <return_value_policy policy, typename... Args,
|
|
typename = enable_if_t<args_are_all_positional<Args...>()>>
|
|
simple_collector<policy> collect_arguments(Args &&...args) {
|
|
return simple_collector<policy>(std::forward<Args>(args)...);
|
|
}
|
|
|
|
/// Collect all arguments, including keywords and unpacking (only instantiated when needed)
|
|
template <return_value_policy policy, typename... Args,
|
|
typename = enable_if_t<!args_are_all_positional<Args...>()>>
|
|
unpacking_collector<policy> collect_arguments(Args &&...args) {
|
|
// Following argument order rules for generalized unpacking according to PEP 448
|
|
static_assert(
|
|
constexpr_last<is_positional, Args...>() < constexpr_first<is_keyword_or_ds, Args...>()
|
|
&& constexpr_last<is_s_unpacking, Args...>() < constexpr_first<is_ds_unpacking, Args...>(),
|
|
"Invalid function call: positional args must precede keywords and ** unpacking; "
|
|
"* unpacking must precede ** unpacking"
|
|
);
|
|
return unpacking_collector<policy>(std::forward<Args>(args)...);
|
|
}
|
|
|
|
template <typename Derived>
|
|
template <return_value_policy policy, typename... Args>
|
|
object object_api<Derived>::operator()(Args &&...args) const {
|
|
#if !defined(NDEBUG) && PY_VERSION_HEX >= 0x03060000
|
|
if (!PyGILState_Check()) {
|
|
pybind11_fail("pybind11::object_api<>::operator() PyGILState_Check() failure.");
|
|
}
|
|
#endif
|
|
return detail::collect_arguments<policy>(std::forward<Args>(args)...).call(derived().ptr());
|
|
}
|
|
|
|
template <typename Derived>
|
|
template <return_value_policy policy, typename... Args>
|
|
object object_api<Derived>::call(Args &&...args) const {
|
|
return operator()<policy>(std::forward<Args>(args)...);
|
|
}
|
|
|
|
PYBIND11_NAMESPACE_END(detail)
|
|
|
|
|
|
template<typename T>
|
|
handle type::handle_of() {
|
|
static_assert(
|
|
detail::any_of<std::is_base_of<detail::type_caster_generic, detail::make_caster<T>>,
|
|
detail::type_uses_smart_holder_type_caster<T>>::value,
|
|
"py::type::of<T> only supports the case where T is a registered C++ types.");
|
|
|
|
return detail::get_type_handle(typeid(T), true);
|
|
}
|
|
|
|
|
|
#define PYBIND11_MAKE_OPAQUE(...) \
|
|
namespace pybind11 { namespace detail { \
|
|
template<> class type_caster<__VA_ARGS__> : public type_caster_for_class_<__VA_ARGS__> { }; \
|
|
}}
|
|
|
|
/// Lets you pass a type containing a `,` through a macro parameter without needing a separate
|
|
/// typedef, e.g.: `PYBIND11_OVERRIDE(PYBIND11_TYPE(ReturnType<A, B>), PYBIND11_TYPE(Parent<C, D>), f, arg)`
|
|
#define PYBIND11_TYPE(...) __VA_ARGS__
|
|
|
|
PYBIND11_NAMESPACE_END(PYBIND11_NAMESPACE)
|