add numerical recipes library
This commit is contained in:
64
lib/nr/ansi/recipes/lnsrch.c
Normal file
64
lib/nr/ansi/recipes/lnsrch.c
Normal file
@@ -0,0 +1,64 @@
|
||||
|
||||
#include <math.h>
|
||||
#define NRANSI
|
||||
#include "nrutil.h"
|
||||
#define ALF 1.0e-4
|
||||
#define TOLX 1.0e-7
|
||||
|
||||
void lnsrch(int n, float xold[], float fold, float g[], float p[], float x[],
|
||||
float *f, float stpmax, int *check, float (*func)(float []))
|
||||
{
|
||||
int i;
|
||||
float a,alam,alam2,alamin,b,disc,f2,rhs1,rhs2,slope,sum,temp,
|
||||
test,tmplam;
|
||||
|
||||
*check=0;
|
||||
for (sum=0.0,i=1;i<=n;i++) sum += p[i]*p[i];
|
||||
sum=sqrt(sum);
|
||||
if (sum > stpmax)
|
||||
for (i=1;i<=n;i++) p[i] *= stpmax/sum;
|
||||
for (slope=0.0,i=1;i<=n;i++)
|
||||
slope += g[i]*p[i];
|
||||
if (slope >= 0.0) nrerror("Roundoff problem in lnsrch.");
|
||||
test=0.0;
|
||||
for (i=1;i<=n;i++) {
|
||||
temp=fabs(p[i])/FMAX(fabs(xold[i]),1.0);
|
||||
if (temp > test) test=temp;
|
||||
}
|
||||
alamin=TOLX/test;
|
||||
alam=1.0;
|
||||
for (;;) {
|
||||
for (i=1;i<=n;i++) x[i]=xold[i]+alam*p[i];
|
||||
*f=(*func)(x);
|
||||
if (alam < alamin) {
|
||||
for (i=1;i<=n;i++) x[i]=xold[i];
|
||||
*check=1;
|
||||
return;
|
||||
} else if (*f <= fold+ALF*alam*slope) return;
|
||||
else {
|
||||
if (alam == 1.0)
|
||||
tmplam = -slope/(2.0*(*f-fold-slope));
|
||||
else {
|
||||
rhs1 = *f-fold-alam*slope;
|
||||
rhs2=f2-fold-alam2*slope;
|
||||
a=(rhs1/(alam*alam)-rhs2/(alam2*alam2))/(alam-alam2);
|
||||
b=(-alam2*rhs1/(alam*alam)+alam*rhs2/(alam2*alam2))/(alam-alam2);
|
||||
if (a == 0.0) tmplam = -slope/(2.0*b);
|
||||
else {
|
||||
disc=b*b-3.0*a*slope;
|
||||
if (disc < 0.0) tmplam=0.5*alam;
|
||||
else if (b <= 0.0) tmplam=(-b+sqrt(disc))/(3.0*a);
|
||||
else tmplam=-slope/(b+sqrt(disc));
|
||||
}
|
||||
if (tmplam > 0.5*alam)
|
||||
tmplam=0.5*alam;
|
||||
}
|
||||
}
|
||||
alam2=alam;
|
||||
f2 = *f;
|
||||
alam=FMAX(tmplam,0.1*alam);
|
||||
}
|
||||
}
|
||||
#undef ALF
|
||||
#undef TOLX
|
||||
#undef NRANSI
|
||||
Reference in New Issue
Block a user