1*5ddc57e5SXin Li /*
2*5ddc57e5SXin Li * Library: lmfit (Levenberg-Marquardt least squares fitting)
3*5ddc57e5SXin Li *
4*5ddc57e5SXin Li * File: demo/curve1.c
5*5ddc57e5SXin Li *
6*5ddc57e5SXin Li * Contents: Example for curve fitting with lmcurve():
7*5ddc57e5SXin Li * fit a data set y(x) by a curve f(x;p).
8*5ddc57e5SXin Li *
9*5ddc57e5SXin Li * Note: Any modification of this example should be copied to
10*5ddc57e5SXin Li * the manual page source lmcurve.pod and to the wiki.
11*5ddc57e5SXin Li *
12*5ddc57e5SXin Li * Author: Joachim Wuttke <[email protected]> 2004-2013
13*5ddc57e5SXin Li *
14*5ddc57e5SXin Li * Licence: see ../COPYING (FreeBSD)
15*5ddc57e5SXin Li *
16*5ddc57e5SXin Li * Homepage: apps.jcns.fz-juelich.de/lmfit
17*5ddc57e5SXin Li */
18*5ddc57e5SXin Li
19*5ddc57e5SXin Li #include "lmcurve.h"
20*5ddc57e5SXin Li #include <stdio.h>
21*5ddc57e5SXin Li
22*5ddc57e5SXin Li /* model function: a parabola */
23*5ddc57e5SXin Li
f(double t,const double * p)24*5ddc57e5SXin Li double f( double t, const double *p )
25*5ddc57e5SXin Li {
26*5ddc57e5SXin Li return p[0] + p[1]*t + p[2]*t*t;
27*5ddc57e5SXin Li }
28*5ddc57e5SXin Li
main()29*5ddc57e5SXin Li int main()
30*5ddc57e5SXin Li {
31*5ddc57e5SXin Li int n = 3; /* number of parameters in model function f */
32*5ddc57e5SXin Li double par[3] = { 100, 0, -10 }; /* really bad starting value */
33*5ddc57e5SXin Li
34*5ddc57e5SXin Li /* data points: a slightly distorted standard parabola */
35*5ddc57e5SXin Li int m = 9;
36*5ddc57e5SXin Li int i;
37*5ddc57e5SXin Li double t[9] = { -4., -3., -2., -1., 0., 1., 2., 3., 4. };
38*5ddc57e5SXin Li double y[9] = { 16.6, 9.9, 4.4, 1.1, 0., 1.1, 4.2, 9.3, 16.4 };
39*5ddc57e5SXin Li
40*5ddc57e5SXin Li lm_control_struct control = lm_control_double;
41*5ddc57e5SXin Li lm_status_struct status;
42*5ddc57e5SXin Li control.verbosity = 9;
43*5ddc57e5SXin Li
44*5ddc57e5SXin Li printf( "Fitting ...\n" );
45*5ddc57e5SXin Li /* now the call to lmfit */
46*5ddc57e5SXin Li lmcurve( n, par, m, t, y, f, &control, &status );
47*5ddc57e5SXin Li
48*5ddc57e5SXin Li printf( "Results:\n" );
49*5ddc57e5SXin Li printf( "status after %d function evaluations:\n %s\n",
50*5ddc57e5SXin Li status.nfev, lm_infmsg[status.outcome] );
51*5ddc57e5SXin Li
52*5ddc57e5SXin Li printf("obtained parameters:\n");
53*5ddc57e5SXin Li for ( i = 0; i < n; ++i)
54*5ddc57e5SXin Li printf(" par[%i] = %12g\n", i, par[i]);
55*5ddc57e5SXin Li printf("obtained norm:\n %12g\n", status.fnorm );
56*5ddc57e5SXin Li
57*5ddc57e5SXin Li printf("fitting data as follows:\n");
58*5ddc57e5SXin Li for ( i = 0; i < m; ++i)
59*5ddc57e5SXin Li printf( " t[%2d]=%4g y=%6g fit=%10g residue=%12g\n",
60*5ddc57e5SXin Li i, t[i], y[i], f(t[i],par), y[i] - f(t[i],par) );
61*5ddc57e5SXin Li
62*5ddc57e5SXin Li return 0;
63*5ddc57e5SXin Li }
64