mirror of
https://github.com/slsdetectorgroup/aare.git
synced 2025-06-18 18:27:13 +02:00
92 lines
2.9 KiB
C++
92 lines
2.9 KiB
C++
#pragma once
|
|
|
|
#include <cmath>
|
|
#include <fmt/core.h>
|
|
#include <vector>
|
|
|
|
#include "aare/NDArray.hpp"
|
|
|
|
namespace aare {
|
|
|
|
namespace func {
|
|
double gaus(const double x, const double *par);
|
|
NDArray<double, 1> gaus(NDView<double, 1> x, NDView<double, 1> par);
|
|
|
|
double pol1(const double x, const double *par);
|
|
NDArray<double, 1> pol1(NDView<double, 1> x, NDView<double, 1> par);
|
|
|
|
} // namespace func
|
|
|
|
|
|
/**
|
|
* @brief Estimate the initial parameters for a Gaussian fit
|
|
*/
|
|
std::array<double, 3> gaus_init_par(const NDView<double, 1> x, const NDView<double, 1> y);
|
|
|
|
std::array<double, 2> pol1_init_par(const NDView<double, 1> x, const NDView<double, 1> y);
|
|
|
|
static constexpr int DEFAULT_NUM_THREADS = 4;
|
|
|
|
/**
|
|
* @brief Fit a 1D Gaussian to data.
|
|
* @param data data to fit
|
|
* @param x x values
|
|
*/
|
|
NDArray<double, 1> fit_gaus(NDView<double, 1> x, NDView<double, 1> y);
|
|
|
|
|
|
/**
|
|
* @brief Fit a 1D Gaussian to each pixel. Data layout [row, col, values]
|
|
* @param x x values
|
|
* @param y y vales, layout [row, col, values]
|
|
* @param n_threads number of threads to use
|
|
*/
|
|
|
|
NDArray<double, 3> fit_gaus(NDView<double, 1> x, NDView<double, 3> y,
|
|
int n_threads = DEFAULT_NUM_THREADS);
|
|
|
|
|
|
|
|
|
|
/**
|
|
* @brief Fit a 1D Gaussian with error estimates
|
|
* @param x x values
|
|
* @param y y vales, layout [row, col, values]
|
|
* @param y_err error in y, layout [row, col, values]
|
|
* @param par_out output parameters
|
|
* @param par_err_out output error parameters
|
|
*/
|
|
void fit_gaus(NDView<double, 1> x, NDView<double, 1> y, NDView<double, 1> y_err,
|
|
NDView<double, 1> par_out, NDView<double, 1> par_err_out,
|
|
double& chi2);
|
|
|
|
/**
|
|
* @brief Fit a 1D Gaussian to each pixel with error estimates. Data layout
|
|
* [row, col, values]
|
|
* @param x x values
|
|
* @param y y vales, layout [row, col, values]
|
|
* @param y_err error in y, layout [row, col, values]
|
|
* @param par_out output parameters, layout [row, col, values]
|
|
* @param par_err_out output parameter errors, layout [row, col, values]
|
|
* @param n_threads number of threads to use
|
|
*/
|
|
void fit_gaus(NDView<double, 1> x, NDView<double, 3> y, NDView<double, 3> y_err,
|
|
NDView<double, 3> par_out, NDView<double, 3> par_err_out, NDView<double, 2> chi2_out,
|
|
int n_threads = DEFAULT_NUM_THREADS
|
|
);
|
|
|
|
NDArray<double, 1> fit_pol1(NDView<double, 1> x, NDView<double, 1> y);
|
|
|
|
NDArray<double, 3> fit_pol1(NDView<double, 1> x, NDView<double, 3> y,
|
|
int n_threads = DEFAULT_NUM_THREADS);
|
|
|
|
void fit_pol1(NDView<double, 1> x, NDView<double, 1> y, NDView<double, 1> y_err,
|
|
NDView<double, 1> par_out, NDView<double, 1> par_err_out, double& chi2);
|
|
|
|
// TODO! not sure we need to offer the different version in C++
|
|
void fit_pol1(NDView<double, 1> x, NDView<double, 3> y, NDView<double, 3> y_err,
|
|
NDView<double, 3> par_out, NDView<double, 3> par_err_out,NDView<double, 2> chi2_out,
|
|
int n_threads = DEFAULT_NUM_THREADS);
|
|
|
|
|
|
} // namespace aare
|