pybind11_tests.cpp 1.61 KB
Newer Older
Wenzel Jakob's avatar
Wenzel Jakob committed
1
/*
Dean Moldovan's avatar
Dean Moldovan committed
2
    tests/pybind11_tests.cpp -- pybind example plugin
Wenzel Jakob's avatar
Wenzel Jakob committed
3

4
    Copyright (c) 2016 Wenzel Jakob <wenzel.jakob@epfl.ch>
Wenzel Jakob's avatar
Wenzel Jakob committed
5
6
7
8
9

    All rights reserved. Use of this source code is governed by a
    BSD-style license that can be found in the LICENSE file.
*/

Dean Moldovan's avatar
Dean Moldovan committed
10
11
#include "pybind11_tests.h"
#include "constructor_stats.h"
Wenzel Jakob's avatar
Wenzel Jakob committed
12

13
14
15
16
std::list<std::function<void(py::module &)>> &initializers() {
    static std::list<std::function<void(py::module &)>> inits;
    return inits;
}
Wenzel Jakob's avatar
Wenzel Jakob committed
17

18
19
20
test_initializer::test_initializer(std::function<void(py::module &)> initializer) {
    initializers().push_back(std::move(initializer));
}
21

22
23
24
25
26
27
28
29
30
void bind_ConstructorStats(py::module &m) {
    py::class_<ConstructorStats>(m, "ConstructorStats")
        .def("alive", &ConstructorStats::alive)
        .def("values", &ConstructorStats::values)
        .def_readwrite("default_constructions", &ConstructorStats::default_constructions)
        .def_readwrite("copy_assignments", &ConstructorStats::copy_assignments)
        .def_readwrite("move_assignments", &ConstructorStats::move_assignments)
        .def_readwrite("copy_constructions", &ConstructorStats::copy_constructions)
        .def_readwrite("move_constructions", &ConstructorStats::move_constructions)
31
        .def_static("get", (ConstructorStats &(*)(py::object)) &ConstructorStats::get, py::return_value_policy::reference_internal);
32
33
}

Dean Moldovan's avatar
Dean Moldovan committed
34
35
PYBIND11_PLUGIN(pybind11_tests) {
    py::module m("pybind11_tests", "pybind example plugin");
Wenzel Jakob's avatar
Wenzel Jakob committed
36

37
38
    bind_ConstructorStats(m);

39
40
    for (const auto &initializer : initializers())
        initializer(m);
Wenzel Jakob's avatar
Wenzel Jakob committed
41

42
    if (!m.attr("have_eigen")) m.attr("have_eigen") = py::cast(false);
43

Wenzel Jakob's avatar
Wenzel Jakob committed
44
45
    return m.ptr();
}