Adding Python to/from string wrappers for internal IDs
This commit is contained in:
parent
0f6ebd6384
commit
3769b20580
@ -76,6 +76,54 @@ struct range_wrapper {
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
/*
|
||||||
|
A wrapper to enable custom type/ID to/from string conversions
|
||||||
|
*/
|
||||||
|
template <typename T> struct string_wrapper {
|
||||||
|
template<typename F>
|
||||||
|
struct from_pystring_converter {
|
||||||
|
from_pystring_converter() {
|
||||||
|
converter::registry::push_back(
|
||||||
|
&convertible,
|
||||||
|
&construct,
|
||||||
|
boost::python::type_id<T>());
|
||||||
|
};
|
||||||
|
|
||||||
|
static void* convertible(PyObject* object) {
|
||||||
|
return PyUnicode_Check(object) ? object : 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void construct(
|
||||||
|
PyObject* object,
|
||||||
|
converter::rvalue_from_python_stage1_data* data) {
|
||||||
|
const wchar_t* value = PyUnicode_AsUnicode(object);
|
||||||
|
const std::wstring value_ws(value);
|
||||||
|
if (value == 0) throw_error_already_set();
|
||||||
|
void* storage = (
|
||||||
|
(boost::python::converter::rvalue_from_python_storage<T>*)
|
||||||
|
data)->storage.bytes;
|
||||||
|
new (storage) T(fn(std::string(value_ws.begin(), value_ws.end())));
|
||||||
|
data->convertible = storage;
|
||||||
|
}
|
||||||
|
|
||||||
|
static F fn;
|
||||||
|
};
|
||||||
|
|
||||||
|
template<typename F> struct to_str_wrapper {
|
||||||
|
static F fn;
|
||||||
|
std::string str(T& x) {
|
||||||
|
return fn(x);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template<typename F1, typename F2> static void wrap(const char *type_name, F1 to_str_fn, F2 from_str_fn) {
|
||||||
|
from_pystring_converter<F2>::fn = from_str_fn;
|
||||||
|
from_pystring_converter<F2>();
|
||||||
|
to_str_wrapper<F1>::fn = to_str_fn;
|
||||||
|
class_<T>(type_name, no_init).def("__str__", to_str_wrapper<F1>::str);
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
#define WRAP_RANGE(t) range_wrapper<t##Range>().wrap(#t "Range", #t "Iterator")
|
#define WRAP_RANGE(t) range_wrapper<t##Range>().wrap(#t "Range", #t "Iterator")
|
||||||
|
|
||||||
void init_python(const char *executable);
|
void init_python(const char *executable);
|
||||||
|
Loading…
Reference in New Issue
Block a user