mirror of
https://github.com/slsdetectorgroup/aare.git
synced 2026-01-01 18:01:24 +01:00
Bug Fixed for Chunked Pedestal
This commit is contained in:
@@ -30,14 +30,29 @@ void define_ClusterFinder(py::module &m, const std::string &typestr) {
|
||||
|
||||
py::class_<ClusterFinder<ClusterType, uint16_t, pd_type>>(
|
||||
m, class_name.c_str())
|
||||
.def(py::init<Shape<2>, pd_type, size_t>(), py::arg("image_size"),
|
||||
py::arg("n_sigma") = 5.0, py::arg("capacity") = 1'000'000)
|
||||
.def(py::init<Shape<2>, pd_type, size_t, uint32_t, uint32_t>(), py::arg("image_size"),
|
||||
py::arg("n_sigma") = 5.0, py::arg("capacity") = 1'000'000,
|
||||
py::arg("chunk_size") = 50'000, py::arg("n_chunks") = 10)
|
||||
.def("push_pedestal_frame",
|
||||
[](ClusterFinder<ClusterType, uint16_t, pd_type> &self,
|
||||
py::array_t<uint16_t> frame) {
|
||||
auto view = make_view_2d(frame);
|
||||
self.push_pedestal_frame(view);
|
||||
})
|
||||
|
||||
.def("push_pedestal_mean",
|
||||
[](ClusterFinder<ClusterType, uint16_t, pd_type> &self,
|
||||
py::array_t<uint16_t> frame, uint32_t chunk_number) {
|
||||
auto view = make_view_2d(frame);
|
||||
self.push_pedestal_mean(view, chunk_number);
|
||||
})
|
||||
.def("push_pedestal_std",
|
||||
[](ClusterFinder<ClusterType, uint16_t, pd_type> &self,
|
||||
py::array_t<uint16_t> frame, uint32_t chunk_number) {
|
||||
auto view = make_view_2d(frame);
|
||||
self.push_pedestal_std(view, chunk_number);
|
||||
})
|
||||
|
||||
.def("clear_pedestal",
|
||||
&ClusterFinder<ClusterType, uint16_t, pd_type>::clear_pedestal)
|
||||
.def_property_readonly(
|
||||
|
||||
Reference in New Issue
Block a user