benchmark.py revision 11986:c12e4625ab56 
1import random
2import os
3import time
4import datetime as dt
5
6nfns = 4  # Functions per class
7nargs = 4  # Arguments per function
8
9
10def 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
44def 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
76for 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++14 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 ("}")
91