mxlib
c++ tools for analyzing astronomical data and other tasks by Jared R. Males. [git repo]
mx::math::fit::fitGaussian1D< _realT > Class Template Reference

template<typename _realT>
class mx::math::fit::fitGaussian1D< _realT >

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.

Template Parameters
fitterTa type meeting the above requirements.

Definition at line 145 of file fitGaussian.hpp.

#include <math/fit/fitGaussian.hpp>

+ Inheritance diagram for mx::math::fit::fitGaussian1D< _realT >:

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...
 

Member Function Documentation

◆ fit()

template<typename _realT >
int mx::math::fit::fitGaussian1D< _realT >::fit ( )
inline

Do the fit.

Definition at line 228 of file fitGaussian.hpp.

References mx::math::fit::levmarInterface< fitterT >::fit().

◆ G()

template<typename _realT >
realT mx::math::fit::fitGaussian1D< _realT >::G ( )
inline

Get the peak scaling.

Definition at line 245 of file fitGaussian.hpp.

◆ G0()

template<typename _realT >
realT mx::math::fit::fitGaussian1D< _realT >::G0 ( )
inline

Get the current value of G0, the constant.

Returns
the current value of G0, which is p[0].

Definition at line 239 of file fitGaussian.hpp.

◆ setArray() [1/2]

template<typename _realT >
void mx::math::fit::fitGaussian1D< _realT >::setArray ( realT *  data,
int  nx 
)
inline

◆ setArray() [2/2]

template<typename _realT >
void mx::math::fit::fitGaussian1D< _realT >::setArray ( realT *  data,
realT *  coords,
int  nx 
)
inline

◆ setFixed()

template<typename _realT >
void mx::math::fit::fitGaussian1D< _realT >::setFixed ( bool  G0,
bool  G,
bool  x0,
bool  sigma 
)
inline

Set whether each parameter is fixed.

Sets the parameter indices appropriately.

Parameters
[in]G0if true, then G0 will be not be part of the fit
[in]Gif true, then G will be not be part of the fit
[in]x0if true, then x0 will be not be part of the fit
[in]sigmaif 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().

◆ setGuess()

template<typename _realT >
void mx::math::fit::fitGaussian1D< _realT >::setGuess ( realT  G0,
realT  G,
realT  x0,
realT  sigma 
)
inline

Set the initial guess for a symmetric Gaussian.

Also works for the general case, setting the same width in both directions.

Parameters
[in]G0the constant background level
[in]Gthe peak scaling
[in]x0the center x-coordinate
[in]sigmathe width parameter

Definition at line 191 of file fitGaussian.hpp.

References mx::astro::constants::G(), and mx::astro::constants::sigma().

◆ sigma()

template<typename _realT >
realT mx::math::fit::fitGaussian1D< _realT >::sigma ( )
inline

Return sigma.

Definition at line 259 of file fitGaussian.hpp.

◆ x0()

template<typename _realT >
realT mx::math::fit::fitGaussian1D< _realT >::x0 ( )
inline

Get the center x-coordinate.

Definition at line 251 of file fitGaussian.hpp.


The documentation for this class was generated from the following file: