diff options
Diffstat (limited to 'tensorflow/python/platform/base.i')
| -rw-r--r-- | tensorflow/python/platform/base.i | 176 |
1 files changed, 176 insertions, 0 deletions
diff --git a/tensorflow/python/platform/base.i b/tensorflow/python/platform/base.i new file mode 100644 index 0000000000..85fa3968a1 --- /dev/null +++ b/tensorflow/python/platform/base.i @@ -0,0 +1,176 @@ +// Helper macros and typemaps for use in Tensorflow swig files. +// +%{ + #include <memory> + #include "tensorflow/core/platform/port.h" + using tensorflow::uint64; + using tensorflow::string; + + template<class T> + bool _PyObjAs(PyObject *pystr, T* cstr) { + T::undefined; // You need to define specialization _PyObjAs<T> + } + + template<class T> + PyObject *_PyObjFrom(const T& c) { + T::undefined; // You need to define specialization _PyObjFrom<T> + } + +#ifdef HAS_GLOBAL_STRING + template<> + bool _PyObjAs(PyObject *pystr, ::string* cstr) { + char *buf; + Py_ssize_t len; +#if PY_VERSION_HEX >= 0x03030000 + if (PyUnicode_Check(pystr)) { + buf = PyUnicode_AsUTF8AndSize(pystr, &len); + if (!buf) return false; + } else // NOLINT +#endif + if (PyBytes_AsStringAndSize(pystr, &buf, &len) == -1) return false; + if (cstr) cstr->assign(buf, len); + return true; + } +#endif + template<> + bool _PyObjAs(PyObject *pystr, std::string* cstr) { + char *buf; + Py_ssize_t len; +#if PY_VERSION_HEX >= 0x03030000 + if (PyUnicode_Check(pystr)) { + buf = PyUnicode_AsUTF8AndSize(pystr, &len); + if (!buf) return false; + } else // NOLINT +#endif + if (PyBytes_AsStringAndSize(pystr, &buf, &len) == -1) return false; + if (cstr) cstr->assign(buf, len); + return true; + } +#ifdef HAS_GLOBAL_STRING + template<> + PyObject* _PyObjFrom(const ::string& c) { + return PyString_FromStringAndSize(c.data(), c.size()); + } +#endif + template<> + PyObject* _PyObjFrom(const std::string& c) { + return PyString_FromStringAndSize(c.data(), c.size()); + } + + PyObject* _SwigString_FromString(const string& s) { + return PyUnicode_FromStringAndSize(s.data(), s.size()); + } +%} + +%typemap(in) string { + if (!_PyObjAs<string>($input, &$1)) return NULL; +} + +%typemap(in) const string& (string temp) { + if (!_PyObjAs<string>($input, &temp)) return NULL; + $1 = &temp; +} + +%typemap(out) string { + $result = PyString_FromStringAndSize($1.data(), $1.size()); +} + +%typemap(out) const string& { + $result = PyString_FromStringAndSize($1->data(), $1->size()); +} + +%typemap(in, numinputs = 0) string* OUTPUT (string temp) { + $1 = &temp; +} + +%typemap(argout) string * OUTPUT { + PyObject *str = PyString_FromStringAndSize($1->data(), $1->length()); + if (!str) SWIG_fail; + %append_output(str); +} + +%typemap(argout) string* INOUT = string* OUTPUT; + +%typemap(varout) string { + $result = PyString_FromStringAndSize($1.data(), $1.size()); +} + +%define _LIST_OUTPUT_TYPEMAP(type, py_converter) + %typemap(in) std::vector<type>(std::vector<type> temp) { + if (!vector_input_helper($input, &temp, _PyObjAs<type>)) { + if (!PyErr_Occurred()) + PyErr_SetString(PyExc_TypeError, "sequence(type) expected"); + return NULL; + } + $1 = temp; +} +%typemap(in) const std::vector<type>& (std::vector<type> temp), + const std::vector<type>* (std::vector<type> temp) { + if (!vector_input_helper($input, &temp, _PyObjAs<type>)) { + if (!PyErr_Occurred()) + PyErr_SetString(PyExc_TypeError, "sequence(type) expected"); + return NULL; + } + $1 = &temp; +} +%typemap(in,numinputs=0) +std::vector<type>* OUTPUT (std::vector<type> temp), + hash_set<type>* OUTPUT (hash_set<type> temp), + set<type>* OUTPUT (set<type> temp) { + $1 = &temp; +} +%typemap(argout) std::vector<type>* OUTPUT, set<type>* OUTPUT, hash_set<type>* OUTPUT { + %append_output(list_output_helper($1, &py_converter)); +} +%typemap(out) std::vector<type> { + $result = vector_output_helper(&$1, &py_converter); +} +%typemap(out) std::vector<type>*, const std::vector<type>& { + $result = vector_output_helper($1, &py_converter); +} +%enddef + +_LIST_OUTPUT_TYPEMAP(string, _SwigString_FromString); +_LIST_OUTPUT_TYPEMAP(unsigned long long, PyLong_FromUnsignedLongLong); + +%typemap(in) uint64 { + // TODO(gps): Check if another implementation + // from hosting/images/util/image-hosting-utils.swig is better. May be not. +%#if PY_MAJOR_VERSION < 3 + if (PyInt_Check($input)) { + $1 = static_cast<uint64>(PyInt_AsLong($input)); + } else +%#endif + if (PyLong_Check($input)) { + $1 = static_cast<uint64>(PyLong_AsUnsignedLongLong($input)); + } else { + PyErr_SetString(PyExc_TypeError, + "int or long value expected for argument \"$1_name\""); + } + // TODO(mrovner): Make consistent use of SWIG_fail vs. return NULL. + if (PyErr_Occurred()) return NULL; +} + +%define _COPY_TYPEMAPS(oldtype, newtype) + typedef oldtype newtype; +%apply oldtype * OUTPUT { newtype * OUTPUT }; +%apply oldtype & OUTPUT { newtype & OUTPUT }; +%apply oldtype * INPUT { newtype * INPUT }; +%apply oldtype & INPUT { newtype & INPUT }; +%apply oldtype * INOUT { newtype * INOUT }; +%apply oldtype & INOUT { newtype & INOUT }; +%apply std::vector<oldtype> * OUTPUT { std::vector<newtype> * OUTPUT }; +%enddef + +_COPY_TYPEMAPS(unsigned long long, uint64); + +// SWIG macros for explicit API declaration. +// Usage: +// +// %ignoreall +// %unignore SomeName; // namespace / class / method +// %include "somelib.h" +// %unignoreall // mandatory closing "bracket" +%define %ignoreall %ignore ""; %enddef +%define %unignore %rename("%s") %enddef +%define %unignoreall %rename("%s") ""; %enddef |