ext/pybind11/docs/benchmark.py

   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_MODULE(example, m) {\n"
  37     result += bindings
  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
  48     decl += '\n'
  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)
  56             params  = [random.randint(0, nclasses - 1) for i in range(nargs)]
  57             decl += "    cl%03i *fn_%03i(" % (ret, fn)
  58             decl += ", ".join("cl%03i *" % p for p in params)
  59             decl += ");\n"
  60             bindings += '        .def("fn_%03i", &cl%03i::fn_%03i, py::return_value_policy<py::manage_new_object>())\n' % \
  61                 (fn, cl, fn)
  62         decl += "};\n\n"
  63         bindings += '        ;\n'
  64
  65     result = "#include <boost/python.hpp>\n\n"
  66     result += "namespace py = boost::python;\n\n"
  67     result += decl + '\n'
  68     result += "BOOST_PYTHON_MODULE(example) {\n"
  69     result += bindings
  70     result += "}"
  71     return result
  72
  73
  74 for codegen in [generate_dummy_code_pybind11, generate_dummy_code_boost]:
  75     print ("{")
  76     for i in range(0, 10):
  77         nclasses = 2 ** i
  78         with open("test.cpp", "w") as f:
  79             f.write(codegen(nclasses))
  80         n1 = dt.datetime.now()
  81         os.system("g++ -Os -shared -rdynamic -undefined dynamic_lookup "
  82             "-fvisibility=hidden -std=c++14 test.cpp -I include "
  83             "-I /System/Library/Frameworks/Python.framework/Headers -o test.so")
  84         n2 = dt.datetime.now()
  85         elapsed = (n2 - n1).total_seconds()
  86         size = os.stat('test.so').st_size
  87         print("   {%i, %f, %i}," % (nclasses * nfns, elapsed, size))
  88     print ("}")