gwm17/catima
1
0
Fork 0
mirror of https://github.com/gwm17/catima.git synced 2024-11-22 18:28:51 -05:00
Code Issues Projects Releases Wiki Activity

Merge pull request #87 from hrosiak/py

Browse Source 
repr
This commit is contained in:
Andrej Prochazka 2021-06-20 12:29:06 +02:00 committed by GitHub
parent 6708f6abc3 e68df0a428
commit dd1289e934
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
pymodule/pycatima.cpp
View File

@ -175,7 +175,10 @@ PYBIND11_MODULE(pycatima,m){
.def_readwrite("sigma_x", &Result::sigma_x) .def_readwrite("sigma_x", &Result::sigma_x)
.def_readwrite("tof", &Result::tof) .def_readwrite("tof", &Result::tof)
.def_readwrite("sp", &Result::sp) .def_readwrite("sp", &Result::sp)
.def("get_dict",&get_result_dict); .def("get_dict",&get_result_dict)
.def("__repr__",[](const Result &self){
return py::str(get_result_dict(self));
});
py::class_<MultiResult>(m,"MultiResult") py::class_<MultiResult>(m,"MultiResult")
.def(py::init<>(),"constructor") .def(py::init<>(),"constructor")
@ -211,6 +214,16 @@ PYBIND11_MODULE(pycatima,m){
} }
d["partial"] = p; d["partial"] = p;
return d; return d;
})
.def("__repr__",[](const MultiResult &r){
py::dict d;
py::list p;
d["result"] = get_result_dict(r.total_result);
for(auto& entry:r.results){
p.append(get_result_dict(entry));
}
d["partial"] = p;
return py::str(d);
}); });
py::enum_<z_eff_type>(m,"z_eff_type") py::enum_<z_eff_type>(m,"z_eff_type")