mxlib
c++ tools for analyzing astronomical data and other tasks by Jared R. Males. [git repo]
|
Class to manage fitting a 1D Gaussian to data via the levmarInterface.
Fits the following function to the data: \( G(x) = G_0 + G\exp[-(0.5/\sigma^2)((x-x_0)^2)]\)
Can use a vector of x-coordinates, or use the array index as the position corresponding to each y-value. A mask can be provided, where any 0 entries cause that position to be ignored.
Any of the parameters can be fixed, and not included in the fit.
fitterT | a type meeting the above requirements. |
Definition at line 145 of file fitGaussian.hpp.
#include <math/fit/fitGaussian.hpp>
Public Member Functions | |
void | setFixed (bool G0, bool G, bool x0, bool sigma) |
Set whether each parameter is fixed. More... | |
void | setGuess (realT G0, realT G, realT x0, realT sigma) |
Set the initial guess for a symmetric Gaussian. More... | |
void | setArray (realT *data, int nx) |
Set the data aray. More... | |
void | setArray (realT *data, realT *coords, int nx) |
Set the data aray. More... | |
int | fit () |
Do the fit. More... | |
realT | G0 () |
Get the current value of G0, the constant. More... | |
realT | G () |
Get the peak scaling. More... | |
realT | x0 () |
Get the center x-coordinate. More... | |
realT | sigma () |
Return sigma. More... | |
Public Member Functions inherited from mx::math::fit::levmarInterface< gaussian1D_fitter< _realT > > | |
levmarInterface () | |
Default constructor. More... | |
levmarInterface (realT *i_p, realT *i_x, int i_m, int i_n, void *i_adata) | |
Setup constructor. More... | |
~levmarInterface () | |
Destructor. More... | |
void | nParams (int i_m) |
Set number of parameters, but don't allocate. More... | |
int | nParams () |
Get the current number of parameters. More... | |
void | allocate_params () |
Allocate parameters array based on previous call to nParams. More... | |
void | allocate_params (int i_m) |
Set number of parameters and allocate. More... | |
void | point_params (realT *i_p) |
Point the parameter pointer at an externally allocated array. More... | |
void | point_params (realT *i_p, int i_m) |
Point the parameter pointer at an externally allocated array. More... | |
void | set_params (realT *i_p) |
Copy parameters to the parameter array. More... | |
realT * | get_params () |
Get current pointer array address. More... | |
void | set_itmax (int i_itmax) |
Set the maximum number of iterations. More... | |
int | get_itmax () |
Get the maximum number of iterations. More... | |
void | allocate_work () |
Allocate the work and covar matrices. More... | |
void | set_opts (int n, realT val) |
Set one of the minimization options to val. More... | |
void | set_opts_default (int n=-1) |
Set one or all of the minimization options to the default. More... | |
realT | get_opts (int n) |
Get the current value of an option. More... | |
int | fit () |
Perform the fit. More... | |
realT | get_initial_norm () |
Returns the L2-norm before minimization occurs. More... | |
realT | get_final_norm () |
Returns the L2-norm at the end of the minimization. More... | |
int | get_iterations () |
Get the number of iterations taken during the minimization. More... | |
int | get_reason_code () |
Get a code specifying the reason minimization terminated. More... | |
std::string | get_reason_string () |
Get the descriptive string describing the reason minimization terminated. More... | |
int | get_fevals () |
Get the number of function evaluations during the minimization. More... | |
int | get_jevals () |
Get the number of jacobian evaluations during the minimization. More... | |
int | get_nlinsys () |
Get the number of linear system solutions during the minimization. More... | |
double | get_deltaT () |
Get the elapsed time of the fit. More... | |
iosT & | dumpParameters (iosT &ios) |
Output current parameters to a stream. More... | |
std::ostream & | dumpParameters () |
Dump the parameter vector to stdout. More... | |
iosT & | dumpReport (iosT &ios, bool dumpParams=true) |
Output current parameters to a stream. More... | |
std::ostream & | dumpReport (bool dumpParams=true) |
Dump a status report to stdout. More... | |
Additional Inherited Members | |
Public Attributes inherited from mx::math::fit::levmarInterface< gaussian1D_fitter< _realT > > | |
realT * | x |
I: measurement vector. NULL implies a zero vector. More... | |
int | n |
I: measurement vector dimension. More... | |
realT | opts [LM_OPTS_SZ] |
Options passed to the minimization routines. See set_opts for details. More... | |
realT | info [LM_INFO_SZ] |
Information regarding the minimization. More... | |
double | deltaT |
Elapsed time of the fitting procedure. More... | |
realT * | work |
Working memory passed to the levmar routines. More... | |
int | work_sz |
The current size of the work array. More... | |
realT * | covar |
Covariance matrix corresponding to LS solution; mxm. More... | |
int | covar_sz |
The current size of the covar array. More... | |
bool | getCovar |
Controls whether the covar array is allocated. More... | |
void * | adata |
Pointer to possibly additional data, passed uninterpreted to func & jacf. More... | |
Protected Attributes inherited from mx::math::fit::levmarInterface< gaussian1D_fitter< _realT > > | |
realT * | p |
Parameter array. On input is the initial estimates. On output has the estimated solution. More... | |
realT * | init_p |
Parameter array on input, saved for comparison. More... | |
int | m |
Parameter vector dimension (i.e. number of unknowns) More... | |
bool | own_p |
Flag indicating whether the p array is owned by this object (for de-allocation). More... | |
int | itmax |
Maximum number of iterations, default is 100. More... | |
|
inline |
Do the fit.
Definition at line 228 of file fitGaussian.hpp.
References mx::math::fit::levmarInterface< fitterT >::fit().
|
inline |
Get the peak scaling.
Definition at line 245 of file fitGaussian.hpp.
|
inline |
Get the current value of G0, the constant.
Definition at line 239 of file fitGaussian.hpp.
|
inline |
Set the data aray.
Definition at line 204 of file fitGaussian.hpp.
References mx::math::fit::array2FitGaussian1D< realT >::m_data, and mx::math::fit::array2FitGaussian1D< realT >::m_nx.
|
inline |
Set the data aray.
Definition at line 215 of file fitGaussian.hpp.
References mx::math::fit::array2FitGaussian1D< realT >::m_coords, mx::math::fit::array2FitGaussian1D< realT >::m_data, and mx::math::fit::array2FitGaussian1D< realT >::m_nx.
|
inline |
Set whether each parameter is fixed.
Sets the parameter indices appropriately.
[in] | G0 | if true, then G0 will be not be part of the fit |
[in] | G | if true, then G will be not be part of the fit |
[in] | x0 | if true, then x0 will be not be part of the fit |
[in] | sigma | if true, then sigma will be not be part of the fit |
Definition at line 178 of file fitGaussian.hpp.
References mx::astro::constants::G(), mx::math::fit::array2FitGaussian1D< realT >::setFixed(), and mx::astro::constants::sigma().
|
inline |
Set the initial guess for a symmetric Gaussian.
Also works for the general case, setting the same width in both directions.
[in] | G0 | the constant background level |
[in] | G | the peak scaling |
[in] | x0 | the center x-coordinate |
[in] | sigma | the width parameter |
Definition at line 191 of file fitGaussian.hpp.
References mx::astro::constants::G(), and mx::astro::constants::sigma().
|
inline |
Return sigma.
Definition at line 259 of file fitGaussian.hpp.
|
inline |
Get the center x-coordinate.
Definition at line 251 of file fitGaussian.hpp.