xref: /aosp_15_r20/external/protobuf/benchmarks/python/python_benchmark_messages.cc (revision 1b3f573f81763fcece89efc2b6a5209149e44ab8)
1 #include <Python.h>
2 
3 #include "benchmarks.pb.h"
4 #include "datasets/google_message1/proto2/benchmark_message1_proto2.pb.h"
5 #include "datasets/google_message1/proto3/benchmark_message1_proto3.pb.h"
6 #include "datasets/google_message2/benchmark_message2.pb.h"
7 #include "datasets/google_message3/benchmark_message3.pb.h"
8 #include "datasets/google_message4/benchmark_message4.pb.h"
9 
10 static struct PyModuleDef _module = {PyModuleDef_HEAD_INIT,
11                                      "libbenchmark_messages",
12                                      "Benchmark messages Python module",
13                                      -1,
14                                      NULL,
15                                      NULL,
16                                      NULL,
17                                      NULL,
18                                      NULL};
19 
20 extern "C" {
PyInit_libbenchmark_messages()21 PyMODINIT_FUNC PyInit_libbenchmark_messages() {
22   benchmarks::BenchmarkDataset().descriptor();
23   benchmarks::proto3::GoogleMessage1().descriptor();
24   benchmarks::proto2::GoogleMessage1().descriptor();
25   benchmarks::proto2::GoogleMessage2().descriptor();
26   benchmarks::google_message3::GoogleMessage3().descriptor();
27   benchmarks::google_message4::GoogleMessage4().descriptor();
28 
29   return PyModule_Create(&_module);
30 }
31 }
32