pywstrings.swg 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. /* ------------------------------------------------------------
  2. * utility methods for wchar_t strings
  3. * ------------------------------------------------------------ */
  4. %{
  5. #if PY_VERSION_HEX >= 0x03020000
  6. # define SWIGPY_UNICODE_ARG(obj) ((PyObject*) (obj))
  7. #else
  8. # define SWIGPY_UNICODE_ARG(obj) ((PyUnicodeObject*) (obj))
  9. #endif
  10. %}
  11. %fragment("SWIG_AsWCharPtrAndSize","header",fragment="<wchar.h>",fragment="SWIG_pwchar_descriptor") {
  12. SWIGINTERN int
  13. SWIG_AsWCharPtrAndSize(PyObject *obj, wchar_t **cptr, size_t *psize, int *alloc)
  14. {
  15. PyObject *tmp = 0;
  16. int isunicode = PyUnicode_Check(obj);
  17. %#if PY_VERSION_HEX < 0x03000000
  18. if (!isunicode && PyString_Check(obj)) {
  19. obj = tmp = PyUnicode_FromObject(obj);
  20. isunicode = 1;
  21. }
  22. %#endif
  23. if (isunicode) {
  24. Py_ssize_t len = PyUnicode_GetSize(obj);
  25. if (cptr) {
  26. *cptr = %new_array(len + 1, wchar_t);
  27. PyUnicode_AsWideChar(SWIGPY_UNICODE_ARG(obj), *cptr, len);
  28. (*cptr)[len] = 0;
  29. }
  30. if (psize) *psize = (size_t) len + 1;
  31. if (alloc) *alloc = cptr ? SWIG_NEWOBJ : 0;
  32. Py_XDECREF(tmp);
  33. return SWIG_OK;
  34. } else {
  35. swig_type_info* pwchar_descriptor = SWIG_pwchar_descriptor();
  36. if (pwchar_descriptor) {
  37. void * vptr = 0;
  38. if (SWIG_ConvertPtr(obj, &vptr, pwchar_descriptor, 0) == SWIG_OK) {
  39. if (cptr) *cptr = (wchar_t *)vptr;
  40. if (psize) *psize = vptr ? (wcslen((wchar_t *)vptr) + 1) : 0;
  41. return SWIG_OK;
  42. }
  43. }
  44. }
  45. return SWIG_TypeError;
  46. }
  47. }
  48. %fragment("SWIG_FromWCharPtrAndSize","header",fragment="<wchar.h>",fragment="SWIG_pwchar_descriptor") {
  49. SWIGINTERNINLINE PyObject *
  50. SWIG_FromWCharPtrAndSize(const wchar_t * carray, size_t size)
  51. {
  52. if (carray) {
  53. if (size > INT_MAX) {
  54. swig_type_info* pwchar_descriptor = SWIG_pwchar_descriptor();
  55. return pwchar_descriptor ?
  56. SWIG_InternalNewPointerObj(%const_cast(carray,wchar_t *), pwchar_descriptor, 0) : SWIG_Py_Void();
  57. } else {
  58. return PyUnicode_FromWideChar(carray, %numeric_cast(size, Py_ssize_t));
  59. }
  60. } else {
  61. return SWIG_Py_Void();
  62. }
  63. }
  64. }