Henry Schreiner | d8c7ee0 | 2020-07-20 13:35:21 -0400 | [diff] [blame] | 1 | # -*- coding: utf-8 -*- |
Wenzel Jakob | 5cd3311 | 2015-10-20 00:58:59 +0200 | [diff] [blame] | 2 | import random |
| 3 | import os |
| 4 | import time |
| 5 | import datetime as dt |
| 6 | |
| 7 | nfns = 4 # Functions per class |
| 8 | nargs = 4 # Arguments per function |
| 9 | |
| 10 | |
| 11 | def generate_dummy_code_pybind11(nclasses=10): |
| 12 | decl = "" |
| 13 | bindings = "" |
| 14 | |
| 15 | for cl in range(nclasses): |
| 16 | decl += "class cl%03i;\n" % cl |
Henry Schreiner | c50f90e | 2020-10-16 16:38:13 -0400 | [diff] [blame] | 17 | decl += "\n" |
Wenzel Jakob | 5cd3311 | 2015-10-20 00:58:59 +0200 | [diff] [blame] | 18 | |
| 19 | for cl in range(nclasses): |
| 20 | decl += "class cl%03i {\n" % cl |
| 21 | decl += "public:\n" |
| 22 | bindings += ' py::class_<cl%03i>(m, "cl%03i")\n' % (cl, cl) |
| 23 | for fn in range(nfns): |
| 24 | ret = random.randint(0, nclasses - 1) |
Henry Schreiner | c50f90e | 2020-10-16 16:38:13 -0400 | [diff] [blame] | 25 | params = [random.randint(0, nclasses - 1) for i in range(nargs)] |
Wenzel Jakob | 5cd3311 | 2015-10-20 00:58:59 +0200 | [diff] [blame] | 26 | decl += " cl%03i *fn_%03i(" % (ret, fn) |
| 27 | decl += ", ".join("cl%03i *" % p for p in params) |
| 28 | decl += ");\n" |
Henry Schreiner | c50f90e | 2020-10-16 16:38:13 -0400 | [diff] [blame] | 29 | bindings += ' .def("fn_%03i", &cl%03i::fn_%03i)\n' % (fn, cl, fn) |
Wenzel Jakob | 5cd3311 | 2015-10-20 00:58:59 +0200 | [diff] [blame] | 30 | decl += "};\n\n" |
Henry Schreiner | c50f90e | 2020-10-16 16:38:13 -0400 | [diff] [blame] | 31 | bindings += " ;\n" |
Wenzel Jakob | 5cd3311 | 2015-10-20 00:58:59 +0200 | [diff] [blame] | 32 | |
| 33 | result = "#include <pybind11/pybind11.h>\n\n" |
| 34 | result += "namespace py = pybind11;\n\n" |
Henry Schreiner | c50f90e | 2020-10-16 16:38:13 -0400 | [diff] [blame] | 35 | result += decl + "\n" |
Dean Moldovan | 443ab59 | 2017-04-24 01:51:44 +0200 | [diff] [blame] | 36 | result += "PYBIND11_MODULE(example, m) {\n" |
Wenzel Jakob | 5cd3311 | 2015-10-20 00:58:59 +0200 | [diff] [blame] | 37 | result += bindings |
Wenzel Jakob | 5cd3311 | 2015-10-20 00:58:59 +0200 | [diff] [blame] | 38 | result += "}" |
| 39 | return result |
| 40 | |
| 41 | |
| 42 | def generate_dummy_code_boost(nclasses=10): |
| 43 | decl = "" |
| 44 | bindings = "" |
| 45 | |
| 46 | for cl in range(nclasses): |
| 47 | decl += "class cl%03i;\n" % cl |
Henry Schreiner | c50f90e | 2020-10-16 16:38:13 -0400 | [diff] [blame] | 48 | decl += "\n" |
Wenzel Jakob | 5cd3311 | 2015-10-20 00:58:59 +0200 | [diff] [blame] | 49 | |
| 50 | for cl in range(nclasses): |
| 51 | decl += "class cl%03i {\n" % cl |
| 52 | decl += "public:\n" |
| 53 | bindings += ' py::class_<cl%03i>("cl%03i")\n' % (cl, cl) |
| 54 | for fn in range(nfns): |
| 55 | ret = random.randint(0, nclasses - 1) |
Henry Schreiner | c50f90e | 2020-10-16 16:38:13 -0400 | [diff] [blame] | 56 | params = [random.randint(0, nclasses - 1) for i in range(nargs)] |
Wenzel Jakob | 5cd3311 | 2015-10-20 00:58:59 +0200 | [diff] [blame] | 57 | decl += " cl%03i *fn_%03i(" % (ret, fn) |
| 58 | decl += ", ".join("cl%03i *" % p for p in params) |
| 59 | decl += ");\n" |
Henry Schreiner | c50f90e | 2020-10-16 16:38:13 -0400 | [diff] [blame] | 60 | bindings += ( |
| 61 | ' .def("fn_%03i", &cl%03i::fn_%03i, py::return_value_policy<py::manage_new_object>())\n' |
| 62 | % (fn, cl, fn) |
| 63 | ) |
Wenzel Jakob | 5cd3311 | 2015-10-20 00:58:59 +0200 | [diff] [blame] | 64 | decl += "};\n\n" |
Henry Schreiner | c50f90e | 2020-10-16 16:38:13 -0400 | [diff] [blame] | 65 | bindings += " ;\n" |
Wenzel Jakob | 5cd3311 | 2015-10-20 00:58:59 +0200 | [diff] [blame] | 66 | |
| 67 | result = "#include <boost/python.hpp>\n\n" |
| 68 | result += "namespace py = boost::python;\n\n" |
Henry Schreiner | c50f90e | 2020-10-16 16:38:13 -0400 | [diff] [blame] | 69 | result += decl + "\n" |
Wenzel Jakob | 5cd3311 | 2015-10-20 00:58:59 +0200 | [diff] [blame] | 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]: |
Henry Schreiner | c50f90e | 2020-10-16 16:38:13 -0400 | [diff] [blame] | 77 | print("{") |
Wenzel Jakob | 5cd3311 | 2015-10-20 00:58:59 +0200 | [diff] [blame] | 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() |
Henry Schreiner | c50f90e | 2020-10-16 16:38:13 -0400 | [diff] [blame] | 83 | os.system( |
| 84 | "g++ -Os -shared -rdynamic -undefined dynamic_lookup " |
Wenzel Jakob | 66c9a40 | 2016-01-17 22:36:36 +0100 | [diff] [blame] | 85 | "-fvisibility=hidden -std=c++14 test.cpp -I include " |
Henry Schreiner | c50f90e | 2020-10-16 16:38:13 -0400 | [diff] [blame] | 86 | "-I /System/Library/Frameworks/Python.framework/Headers -o test.so" |
| 87 | ) |
Wenzel Jakob | 5cd3311 | 2015-10-20 00:58:59 +0200 | [diff] [blame] | 88 | n2 = dt.datetime.now() |
| 89 | elapsed = (n2 - n1).total_seconds() |
Henry Schreiner | c50f90e | 2020-10-16 16:38:13 -0400 | [diff] [blame] | 90 | size = os.stat("test.so").st_size |
Wenzel Jakob | 5cd3311 | 2015-10-20 00:58:59 +0200 | [diff] [blame] | 91 | print(" {%i, %f, %i}," % (nclasses * nfns, elapsed, size)) |
Henry Schreiner | c50f90e | 2020-10-16 16:38:13 -0400 | [diff] [blame] | 92 | print("}") |