|
| 1 | +import random |
| 2 | +import os |
| 3 | +import time |
| 4 | +import datetime as dt |
| 5 | + |
| 6 | +nfns = 4 # Functions per class |
| 7 | +nargs = 4 # Arguments per function |
| 8 | + |
| 9 | + |
| 10 | +def generate_dummy_code_pybind11(nclasses=10): |
| 11 | + decl = "" |
| 12 | + bindings = "" |
| 13 | + |
| 14 | + for cl in range(nclasses): |
| 15 | + decl += "class cl%03i;\n" % cl |
| 16 | + decl += '\n' |
| 17 | + |
| 18 | + for cl in range(nclasses): |
| 19 | + decl += "class cl%03i {\n" % cl |
| 20 | + decl += "public:\n" |
| 21 | + bindings += ' py::class_<cl%03i>(m, "cl%03i")\n' % (cl, cl) |
| 22 | + for fn in range(nfns): |
| 23 | + ret = random.randint(0, nclasses - 1) |
| 24 | + params = [random.randint(0, nclasses - 1) for i in range(nargs)] |
| 25 | + decl += " cl%03i *fn_%03i(" % (ret, fn) |
| 26 | + decl += ", ".join("cl%03i *" % p for p in params) |
| 27 | + decl += ");\n" |
| 28 | + bindings += ' .def("fn_%03i", &cl%03i::fn_%03i)\n' % \ |
| 29 | + (fn, cl, fn) |
| 30 | + decl += "};\n\n" |
| 31 | + bindings += ' ;\n' |
| 32 | + |
| 33 | + result = "#include <pybind11/pybind11.h>\n\n" |
| 34 | + result += "namespace py = pybind11;\n\n" |
| 35 | + result += decl + '\n' |
| 36 | + result += "PYBIND11_PLUGIN(example) {\n" |
| 37 | + result += " py::module m(\"example\");" |
| 38 | + result += bindings |
| 39 | + result += " return m.ptr();" |
| 40 | + result += "}" |
| 41 | + return result |
| 42 | + |
| 43 | + |
| 44 | +def generate_dummy_code_boost(nclasses=10): |
| 45 | + decl = "" |
| 46 | + bindings = "" |
| 47 | + |
| 48 | + for cl in range(nclasses): |
| 49 | + decl += "class cl%03i;\n" % cl |
| 50 | + decl += '\n' |
| 51 | + |
| 52 | + for cl in range(nclasses): |
| 53 | + decl += "class cl%03i {\n" % cl |
| 54 | + decl += "public:\n" |
| 55 | + bindings += ' py::class_<cl%03i>("cl%03i")\n' % (cl, cl) |
| 56 | + for fn in range(nfns): |
| 57 | + ret = random.randint(0, nclasses - 1) |
| 58 | + params = [random.randint(0, nclasses - 1) for i in range(nargs)] |
| 59 | + decl += " cl%03i *fn_%03i(" % (ret, fn) |
| 60 | + decl += ", ".join("cl%03i *" % p for p in params) |
| 61 | + decl += ");\n" |
| 62 | + bindings += ' .def("fn_%03i", &cl%03i::fn_%03i, py::return_value_policy<py::manage_new_object>())\n' % \ |
| 63 | + (fn, cl, fn) |
| 64 | + decl += "};\n\n" |
| 65 | + bindings += ' ;\n' |
| 66 | + |
| 67 | + result = "#include <boost/python.hpp>\n\n" |
| 68 | + result += "namespace py = boost::python;\n\n" |
| 69 | + result += decl + '\n' |
| 70 | + result += "BOOST_PYTHON_MODULE(example) {\n" |
| 71 | + result += bindings |
| 72 | + result += "}" |
| 73 | + return result |
| 74 | + |
| 75 | + |
| 76 | +for codegen in [generate_dummy_code_pybind11, generate_dummy_code_boost]: |
| 77 | + print ("{") |
| 78 | + for i in range(0, 10): |
| 79 | + nclasses = 2 ** i |
| 80 | + with open("test.cpp", "w") as f: |
| 81 | + f.write(codegen(nclasses)) |
| 82 | + n1 = dt.datetime.now() |
| 83 | + os.system("g++ -Os -shared -rdynamic -undefined dynamic_lookup " |
| 84 | + "-fvisibility=hidden -std=c++11 test.cpp -I include " |
| 85 | + "-I /System/Library/Frameworks/Python.framework/Headers -o test.so") |
| 86 | + n2 = dt.datetime.now() |
| 87 | + elapsed = (n2 - n1).total_seconds() |
| 88 | + size = os.stat('test.so').st_size |
| 89 | + print(" {%i, %f, %i}," % (nclasses * nfns, elapsed, size)) |
| 90 | + print ("}") |
0 commit comments