Mercurial > thymian
diff templates/pyembed.h @ 0:a4671277546c tip
created the repository for the thymian project
| author | ferencd |
|---|---|
| date | Tue, 17 Aug 2021 11:19:54 +0200 |
| parents | |
| children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/templates/pyembed.h Tue Aug 17 11:19:54 2021 +0200 @@ -0,0 +1,156 @@ +#ifndef PYEMBED_H +#define PYEMBED_H + +// +// Copyright (C) 2011 Mateusz Loskot <mateusz@loskot.net> +// Distributed under the Boost Software License, Version 1.0. +// (See accompanying file LICENSE_1_0.txt or copy at +// http://www.boost.org/LICENSE_1_0.txt) +// +// Blog article: http://mateusz.loskot.net/?p=2819 + +#include <functional> +#include <iostream> +#include <string> +#include <Python.h> + +namespace emb +{ + +typedef std::function<void(std::string)> stdout_write_type; + +struct Stdout +{ + PyObject_HEAD + stdout_write_type write; +}; + +PyObject* Stdout_write(PyObject* self, PyObject* args) +{ + std::size_t written(0); + Stdout* selfimpl = reinterpret_cast<Stdout*>(self); + if (selfimpl->write) + { + char* data; + if (!PyArg_ParseTuple(args, "s", &data)) + return 0; + + std::string str(data); + selfimpl->write(str); + written = str.size(); + } + return PyLong_FromSize_t(written); +} + +PyObject* Stdout_flush(PyObject* /*self*/, PyObject* /*args*/) +{ + // no-op + return Py_BuildValue(""); +} + +PyMethodDef Stdout_methods[] = +{ + {"write", Stdout_write, METH_VARARGS, "sys.stdout.write"}, + {"flush", Stdout_flush, METH_VARARGS, "sys.stdout.write"}, + {0, 0, 0, 0} // sentinel +}; + +PyTypeObject StdoutType = +{ + PyVarObject_HEAD_INIT(0, 0) + "emb.StdoutType", /* tp_name */ + sizeof(Stdout), /* tp_basicsize */ + 0, /* tp_itemsize */ + 0, /* tp_dealloc */ + 0, /* tp_print */ + 0, /* tp_getattr */ + 0, /* tp_setattr */ + 0, /* tp_reserved */ + 0, /* tp_repr */ + 0, /* tp_as_number */ + 0, /* tp_as_sequence */ + 0, /* tp_as_mapping */ + 0, /* tp_hash */ + 0, /* tp_call */ + 0, /* tp_str */ + 0, /* tp_getattro */ + 0, /* tp_setattro */ + 0, /* tp_as_buffer */ + Py_TPFLAGS_DEFAULT, /* tp_flags */ + "emb.Stdout objects", /* tp_doc */ + 0, /* tp_traverse */ + 0, /* tp_clear */ + 0, /* tp_richcompare */ + 0, /* tp_weaklistoffset */ + 0, /* tp_iter */ + 0, /* tp_iternext */ + Stdout_methods, /* tp_methods */ + 0, /* tp_members */ + 0, /* tp_getset */ + 0, /* tp_base */ + 0, /* tp_dict */ + 0, /* tp_descr_get */ + 0, /* tp_descr_set */ + 0, /* tp_dictoffset */ + 0, /* tp_init */ + 0, /* tp_alloc */ + 0, /* tp_new */ +}; + +#if PY_MAJOR_VERSION >= 3 +#define PY3K +#endif + +static PyModuleDef embmodule = +{ + PyModuleDef_HEAD_INIT, + "emb", 0, -1, 0, +}; + +// Internal state +PyObject* g_stdout; +PyObject* g_stdout_saved; + +PyMODINIT_FUNC PyInit_emb(void) +{ + g_stdout = 0; + g_stdout_saved = 0; + + StdoutType.tp_new = PyType_GenericNew; + if (PyType_Ready(&StdoutType) < 0) + return 0; + + PyObject* m = PyModule_Create(&embmodule); + if (m) + { + Py_INCREF(&StdoutType); + PyModule_AddObject(m, "Stdout", reinterpret_cast<PyObject*>(&StdoutType)); + } + return m; +} + +void set_stdout(stdout_write_type write) +{ + if (!g_stdout) + { + g_stdout_saved = PySys_GetObject("stdout"); // borrowed + g_stdout = StdoutType.tp_new(&StdoutType, 0, 0); + } + + Stdout* impl = reinterpret_cast<Stdout*>(g_stdout); + impl->write = write; + PySys_SetObject("stdout", g_stdout); +} + +void reset_stdout() +{ + if (g_stdout_saved) + PySys_SetObject("stdout", g_stdout_saved); + + Py_XDECREF(g_stdout); + g_stdout = 0; +} + +} // namespace emb + +#endif
