working calling python
This commit is contained in:
parent
2f3ea04010
commit
1c861466dc
2 changed files with 57 additions and 0 deletions
1
embedding.py
Normal file
1
embedding.py
Normal file
|
@ -0,0 +1 @@
|
||||||
|
print("hello llama.cpp")
|
56
plugin_python.cpp
Normal file
56
plugin_python.cpp
Normal file
|
@ -0,0 +1,56 @@
|
||||||
|
#include <boost/python.hpp>
|
||||||
|
#include <iostream>
|
||||||
|
#include <frameobject.h>
|
||||||
|
#include <string>
|
||||||
|
|
||||||
|
class Base {
|
||||||
|
public:
|
||||||
|
Base() : mName("Base") {}
|
||||||
|
Base(const std::string& name) : mName(name) {}
|
||||||
|
virtual ~Base() {}
|
||||||
|
std::string name() const
|
||||||
|
{ return mName; }
|
||||||
|
private:
|
||||||
|
std::string mName;
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
using namespace boost::python;
|
||||||
|
|
||||||
|
#if PY_MAJOR_VERSION >= 3
|
||||||
|
# define INIT_MODULE PyInit_mymodule
|
||||||
|
extern "C" PyObject* INIT_MODULE();
|
||||||
|
#else
|
||||||
|
# define INIT_MODULE initmymodule
|
||||||
|
extern "C" void INIT_MODULE();
|
||||||
|
#endif
|
||||||
|
|
||||||
|
|
||||||
|
int call_python()
|
||||||
|
{
|
||||||
|
try {
|
||||||
|
PyImport_AppendInittab((char*)"mymodule", INIT_MODULE);
|
||||||
|
Py_Initialize();
|
||||||
|
object main_module = import("__main__");
|
||||||
|
dict main_namespace = extract<dict>(main_module.attr("__dict__"));
|
||||||
|
object mymodule = import("mymodule");
|
||||||
|
|
||||||
|
main_namespace["precreated_object"] = Base("created on C++ side");
|
||||||
|
exec_file("embedding.py", main_namespace, main_namespace);
|
||||||
|
} catch (error_already_set& e) {
|
||||||
|
PyErr_PrintEx(0);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
using namespace boost::python;
|
||||||
|
|
||||||
|
BOOST_PYTHON_MODULE(mymodule)
|
||||||
|
{
|
||||||
|
class_<Base>("Base")
|
||||||
|
.def("__str__", &Base::name)
|
||||||
|
;
|
||||||
|
}
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue