detectors/aare
25
0
Fork 0
mirror of https://github.com/slsdetectorgroup/aare.git synced 2025-06-12 15:27:13 +02:00
Code Issues Actions Packages Releases Activity

Merge branch 'main' into developer

Browse Source
This commit is contained in:
Erik Fröjdh
2025-02-18 21:52:20 +01:00
committed by GitHub
parent 5d2f25a6e9 dadf5f4869
commit fc1c9f35d6
8 changed files with 55 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
python/src/fit.hpp
View File

@ -9,6 +9,7 @@
namespace py = pybind11;
using namespace pybind11::literals;
void define_fit_bindings(py::module &m) {
// TODO! Evaluate without converting to double
@ -54,6 +55,7 @@ void define_fit_bindings(py::module &m) {
)",
py::arg("x"), py::arg("par"));
m.def(
"fit_gaus",
[](py::array_t<double, py::array::c_style | py::array::forcecast> x,
@ -76,6 +78,7 @@ void define_fit_bindings(py::module &m) {
}
},
R"(
Fit a 1D Gaussian to data.
Parameters
@ -95,6 +98,7 @@ n_threads : int, optional
py::array_t<double, py::array::c_style | py::array::forcecast> y,
py::array_t<double, py::array::c_style | py::array::forcecast> y_err,
int n_threads) {
if (y.ndim() == 3) {
// Allocate memory for the output
// Need to have pointers to allow python to manage
@ -142,6 +146,7 @@ n_threads : int, optional
}
},
R"(
Fit a 1D Gaussian to data with error estimates.
Parameters
@ -189,6 +194,7 @@ n_threads : int, optional
int n_threads) {
if (y.ndim() == 3) {
auto par = new NDArray<double, 3>({y.shape(0), y.shape(1), 2});
auto par_err =
new NDArray<double, 3>({y.shape(0), y.shape(1), 2});
@ -221,6 +227,7 @@ n_threads : int, optional
return py::dict("par"_a = return_image_data(par),
"par_err"_a = return_image_data(par_err),
"chi2"_a = chi2, "Ndf"_a = y.size() - 2);
} else {
throw std::runtime_error("Data must be 1D or 3D");
}