add numerical recipes library
This commit is contained in:
42
lib/nr/k_and_r/recipes/rkqs.c
Normal file
42
lib/nr/k_and_r/recipes/rkqs.c
Normal file
@@ -0,0 +1,42 @@
|
||||
|
||||
#include <math.h>
|
||||
#include "nrutil.h"
|
||||
#define SAFETY 0.9
|
||||
#define PGROW -0.2
|
||||
#define PSHRNK -0.25
|
||||
#define ERRCON 1.89e-4
|
||||
|
||||
void rkqs(y,dydx,n,x,htry,eps,yscal,hdid,hnext,derivs)
|
||||
float *hdid,*hnext,*x,dydx[],eps,htry,y[],yscal[];
|
||||
int n;
|
||||
void (*derivs)();
|
||||
{
|
||||
void rkck();
|
||||
int i;
|
||||
float errmax,h,htemp,xnew,*yerr,*ytemp;
|
||||
|
||||
yerr=vector(1,n);
|
||||
ytemp=vector(1,n);
|
||||
h=htry;
|
||||
for (;;) {
|
||||
rkck(y,dydx,n,*x,h,ytemp,yerr,derivs);
|
||||
errmax=0.0;
|
||||
for (i=1;i<=n;i++) errmax=FMAX(errmax,fabs(yerr[i]/yscal[i]));
|
||||
errmax /= eps;
|
||||
if (errmax <= 1.0) break;
|
||||
htemp=SAFETY*h*pow(errmax,PSHRNK);
|
||||
h=(h >= 0.0 ? FMAX(htemp,0.1*h) : FMIN(htemp,0.1*h));
|
||||
xnew=(*x)+h;
|
||||
if (xnew == *x) nrerror("stepsize underflow in rkqs");
|
||||
}
|
||||
if (errmax > ERRCON) *hnext=SAFETY*h*pow(errmax,PGROW);
|
||||
else *hnext=5.0*h;
|
||||
*x += (*hdid=h);
|
||||
for (i=1;i<=n;i++) y[i]=ytemp[i];
|
||||
free_vector(ytemp,1,n);
|
||||
free_vector(yerr,1,n);
|
||||
}
|
||||
#undef SAFETY
|
||||
#undef PGROW
|
||||
#undef PSHRNK
|
||||
#undef ERRCON
|
||||
Reference in New Issue
Block a user