Zengtudor/pybind11
1
0
Fork 0
mirror of https://github.com/pybind/pybind11.git synced 2024-11-26 07:02:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

add wstring caster

Browse Source
This commit is contained in:
hulucc 2016-03-02 13:59:39 +08:00
parent 937d646bab
commit c8f68b3d92
1 changed files with 23 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
include/pybind11/cast.h
View File

@ -348,6 +348,29 @@ public:
PYBIND11_TYPE_CASTER(std::string, _(PYBIND11_STRING_NAME)); PYBIND11_TYPE_CASTER(std::string, _(PYBIND11_STRING_NAME));
}; };
template <> class type_caster<std::wstring> {
public:
bool load(handle src, bool) {
object temp;
handle load_src = src;
if (!PyUnicode_Check(load_src.ptr())) {
temp = object(PyUnicode_FromObject(load_src.ptr()), false);
if (!temp) { PyErr_Clear(); return false; }
load_src = temp;
}
wchar_t *buffer = PyUnicode_AS_UNICODE(load_src.ptr());
size_t length = PyUnicode_GET_SIZE(load_src.ptr());
value = std::wstring(buffer, length);
return true;
}
static handle cast(const std::wstring &src, return_value_policy /* policy */, handle /* parent */) {
return PyUnicode_FromWideChar(src.c_str(), src.length());
}
PYBIND11_TYPE_CASTER(std::wstring, _(PYBIND11_STRING_NAME));
};
template <> class type_caster<char> { template <> class type_caster<char> {
public: public:
bool load(handle src, bool) { bool load(handle src, bool) {