Skip to content

Commit

Permalink
Update after c2py cleaning
Browse files Browse the repository at this point in the history
  • Loading branch information
@parcollet
parcollet committed Sep 29, 2023
1 parent 1fd7433 commit 6a92b30
Showing 1 changed file with 43 additions and 44 deletions.
87 changes: 43 additions & 44 deletions python/app4triqs/app4triqs_module.wrap.cxx
View file
Original file line number Diff line number Diff line change
@@ -1,9 +1,14 @@

// C.f. https://numpy.org/doc/1.21/reference/c-api/array.html#importing-the-api
#define PY_ARRAY_UNIQUE_SYMBOL _cpp2py_ARRAY_API
#ifdef __clang__
#pragma clang diagnostic ignored "-W#warnings"
#include "app4triqs_module.cpp"
#endif
#define C2PY_VERSION_MAJOR 0
#define C2PY_VERSION_MINOR 1
#include "c2py/c2py.hpp"
#include "app4triqs_module.cpp"
#include <c2py/serialization/h5.hpp>

using c2py::operator"" _a;

Expand All @@ -15,19 +20,17 @@ template <> inline constexpr bool c2py::is_wrapped<app4triqs::toto> = true;

// ==================== module classes =====================

template <> inline const std::string c2py::cpp_name<app4triqs::toto> = "app4triqs::toto";
template <> inline constexpr auto c2py::tp_name<app4triqs::toto> = "app4triqs.app4triqs_module.Toto";
template <>
inline constexpr const char *c2py::tp_doc<app4triqs::toto> = R"DOC( A very useful and important class)DOC";
template <> inline const std::string c2py::cpp_name<app4triqs::toto> = "app4triqs::toto";
template <> inline constexpr auto c2py::tp_name<app4triqs::toto> = "app4triqs.app4triqs_module.Toto";
template <> inline constexpr const char *c2py::tp_doc<app4triqs::toto> = R"DOC( A very useful and important class)DOC";

static auto init_0 =
c2py::dispatcher_c_kw_t{c2py::c_constructor<app4triqs::toto>(), c2py::c_constructor<app4triqs::toto, int>("i_")};
static auto init_0 = c2py::dispatcher_c_kw_t{c2py::c_constructor<app4triqs::toto>(), c2py::c_constructor<app4triqs::toto, int>("i_")};
template <> constexpr initproc c2py::tp_init<app4triqs::toto> = c2py::pyfkw_constructor<init_0>;
// f
static auto const fun_0 = c2py::dispatcher_f_kw_t{c2py::cfun(c2py::castm<int>(&app4triqs::toto::f), "u")};

// get_i
static auto const fun_1 = c2py::dispatcher_f_kw_t{c2py::cfun(c2py::castmc<>(&app4triqs::toto::get_i))};
static auto const fun_1 = c2py::dispatcher_f_kw_t{c2py::cfun(c2py::castmc<>(&app4triqs::toto::get_i))};
static const auto doc_d_0 = fun_0.doc({R"DOC( A simple function with :math:`G(
Parameters
Expand All @@ -39,35 +42,37 @@ static const auto doc_d_1 = fun_1.doc({R"DOC( Simple accessor)DOC"});
// ----- Method table ----
template <>
PyMethodDef c2py::tp_methods<app4triqs::toto>[] = {
{"f", (PyCFunction)c2py::pyfkw<fun_0>, METH_VARARGS | METH_KEYWORDS, doc_d_0.c_str()},
{"get_i", (PyCFunction)c2py::pyfkw<fun_1>, METH_VARARGS | METH_KEYWORDS, doc_d_1.c_str()},
{"__write_hdf5__", c2py::tpxx_write_h5<app4triqs::toto>, METH_VARARGS, " "},
{"__getstate__", c2py::getstate_tuple<app4triqs::toto>, METH_NOARGS, ""},
{"__setstate__", c2py::setstate_tuple<app4triqs::toto>, METH_O, ""},
{nullptr, nullptr, 0, nullptr} // Sentinel
{"f", (PyCFunction)c2py::pyfkw<fun_0>, METH_VARARGS | METH_KEYWORDS, doc_d_0.c_str()},
{"get_i", (PyCFunction)c2py::pyfkw<fun_1>, METH_VARARGS | METH_KEYWORDS, doc_d_1.c_str()},
{"__write_hdf5__", c2py::tpxx_write_h5<app4triqs::toto>, METH_VARARGS, " "},
{"__getstate__", c2py::getstate_tuple<app4triqs::toto>, METH_NOARGS, ""},
{"__setstate__", c2py::setstate_tuple<app4triqs::toto>, METH_O, ""},
{nullptr, nullptr, 0, nullptr} // Sentinel
};

// ----- Method table ----

template <>
constinit PyGetSetDef c2py::tp_getset<app4triqs::toto>[] = {

{nullptr, nullptr, nullptr, nullptr, nullptr}};
{nullptr, nullptr, nullptr, nullptr, nullptr}};

// ==================== module functions ====================

// chain
static auto const fun_2 = c2py::dispatcher_f_kw_t{c2py::cfun(c2py::cast<int, int>(&app4triqs::chain), "i", "j")};
static auto const fun_2 = c2py::dispatcher_f_kw_t{c2py::cfun([](int i, int j) { return app4triqs::chain(i, j); }, "i", "j")};

// make_array
static auto const fun_3 = c2py::dispatcher_f_kw_t{c2py::cfun(c2py::cast<double>(&app4triqs::make_array), "i")};
static auto const fun_3 = c2py::dispatcher_f_kw_t{c2py::cfun([](double i) { return app4triqs::make_array(i); }, "i")};

// make_gf
static auto const fun_4 = c2py::dispatcher_f_kw_t{c2py::cfun(c2py::cast<>(&app4triqs::make_gf))};
static auto const fun_4 = c2py::dispatcher_f_kw_t{c2py::cfun([]() { return app4triqs::make_gf(); })};

// print_gf
static auto const fun_5 = c2py::dispatcher_f_kw_t{c2py::cfun([](int i, const triqs::gfs::gf<triqs::mesh::imfreq> &g) { return app4triqs::print_gf(i, g); }, "i", "g")};

// sum_array
static auto const fun_5 =
c2py::dispatcher_f_kw_t{c2py::cfun(c2py::cast<const nda::array<double, 2> &>(&app4triqs::sum_array), "a")};
static auto const fun_6 = c2py::dispatcher_f_kw_t{c2py::cfun([](const nda::array<double, 2> &a) { return app4triqs::sum_array(a); }, "a")};

static const auto doc_d_2 = fun_2.doc({R"DOC( Chain digits of two integers
Expand All @@ -83,30 +88,28 @@ static const auto doc_d_2 = fun_2.doc({R"DOC( Chain digits of two integers
static const auto doc_d_3 = fun_3.doc({R"DOC( )DOC"});
static const auto doc_d_4 = fun_4.doc({R"DOC( )DOC"});
static const auto doc_d_5 = fun_5.doc({R"DOC( )DOC"});
static const auto doc_d_6 = fun_6.doc({R"DOC( )DOC"});
//--------------------- module function table -----------------------------

static PyMethodDef module_methods[] = {
{"chain", (PyCFunction)c2py::pyfkw<fun_2>, METH_VARARGS | METH_KEYWORDS, doc_d_2.c_str()},
{"make_array", (PyCFunction)c2py::pyfkw<fun_3>, METH_VARARGS | METH_KEYWORDS, doc_d_3.c_str()},
{"make_gf", (PyCFunction)c2py::pyfkw<fun_4>, METH_VARARGS | METH_KEYWORDS, doc_d_4.c_str()},
{"sum_array", (PyCFunction)c2py::pyfkw<fun_5>, METH_VARARGS | METH_KEYWORDS, doc_d_5.c_str()},
{nullptr, nullptr, 0, nullptr} // Sentinel
{"chain", (PyCFunction)c2py::pyfkw<fun_2>, METH_VARARGS | METH_KEYWORDS, doc_d_2.c_str()}, {"make_array", (PyCFunction)c2py::pyfkw<fun_3>, METH_VARARGS | METH_KEYWORDS, doc_d_3.c_str()},
{"make_gf", (PyCFunction)c2py::pyfkw<fun_4>, METH_VARARGS | METH_KEYWORDS, doc_d_4.c_str()}, {"print_gf", (PyCFunction)c2py::pyfkw<fun_5>, METH_VARARGS | METH_KEYWORDS, doc_d_5.c_str()},
{"sum_array", (PyCFunction)c2py::pyfkw<fun_6>, METH_VARARGS | METH_KEYWORDS, doc_d_6.c_str()}, {nullptr, nullptr, 0, nullptr} // Sentinel
};

//--------------------- module struct & init error definition ------------

//// module doc directly in the code or "" if not present...
/// Or mandatory ?
static struct PyModuleDef module_def = {
PyModuleDef_HEAD_INIT,
"app4triqs_module", /* name of module */
"DOC MODULE", /* module documentation, may be NULL */
-1, /* size of per-interpreter state of the module, or -1 if the module keeps state in global variables. */
module_methods,
NULL,
NULL,
NULL,
NULL};
static struct PyModuleDef module_def = {PyModuleDef_HEAD_INIT,
"app4triqs_module", /* name of module */
R"RAWDOC()RAWDOC", /* module documentation, may be NULL */
-1, /* size of per-interpreter state of the module, or -1 if the module keeps state in global variables. */
module_methods,
NULL,
NULL,
NULL,
NULL};

//--------------------- module init function -----------------------------

Expand All @@ -119,23 +122,19 @@ extern "C" __attribute__((visibility("default"))) PyObject *PyInit_app4triqs_mod

PyObject *m;

if (PyType_Ready(&c2py::py_type_object<c2py::py_range>) < 0)
return NULL;
if (PyType_Ready(&c2py::py_type_object<app4triqs::toto>) < 0)
return NULL;
if (PyType_Ready(&c2py::wrap_pytype<c2py::py_range>) < 0) return NULL;
if (PyType_Ready(&c2py::wrap_pytype<app4triqs::toto>) < 0) return NULL;

m = PyModule_Create(&module_def);
if (m == NULL)
return NULL;
if (m == NULL) return NULL;

auto &conv_table = *c2py::conv_table_sptr.get();

conv_table[std::type_index(typeid(c2py::py_range)).name()] = &c2py::py_type_object<c2py::py_range>;
conv_table[std::type_index(typeid(c2py::py_range)).name()] = &c2py::wrap_pytype<c2py::py_range>;
CLAIR_C2PY_ADD_TYPE_OBJECT(app4triqs::toto, "Toto");

c2py::pyref module = c2py::pyref::module("h5.formats");
if (not module)
return nullptr;
if (not module) return nullptr;
c2py::pyref register_class = module.attr("register_class");

register_h5_type<app4triqs::toto>(register_class);
Expand Down

0 comments on commit 6a92b30

Please sign in to comment.