add numerical recipes library
This commit is contained in:
41
lib/nr/k_and_r/recipes/zroots.c
Normal file
41
lib/nr/k_and_r/recipes/zroots.c
Normal file
@@ -0,0 +1,41 @@
|
||||
|
||||
#include <math.h>
|
||||
#include "complex.h"
|
||||
#define EPS 2.0e-6
|
||||
#define MAXM 100
|
||||
|
||||
void zroots(a,m,roots,polish)
|
||||
fcomplex a[],roots[];
|
||||
int m,polish;
|
||||
{
|
||||
void laguer();
|
||||
int i,its,j,jj;
|
||||
fcomplex x,b,c,ad[MAXM];
|
||||
|
||||
for (j=0;j<=m;j++) ad[j]=a[j];
|
||||
for (j=m;j>=1;j--) {
|
||||
x=Complex(0.0,0.0);
|
||||
laguer(ad,j,&x,&its);
|
||||
if (fabs(x.i) <= 2.0*EPS*fabs(x.r)) x.i=0.0;
|
||||
roots[j]=x;
|
||||
b=ad[j];
|
||||
for (jj=j-1;jj>=0;jj--) {
|
||||
c=ad[jj];
|
||||
ad[jj]=b;
|
||||
b=Cadd(Cmul(x,b),c);
|
||||
}
|
||||
}
|
||||
if (polish)
|
||||
for (j=1;j<=m;j++)
|
||||
laguer(a,m,&roots[j],&its);
|
||||
for (j=2;j<=m;j++) {
|
||||
x=roots[j];
|
||||
for (i=j-1;i>=1;i--) {
|
||||
if (roots[i].r <= x.r) break;
|
||||
roots[i+1]=roots[i];
|
||||
}
|
||||
roots[i+1]=x;
|
||||
}
|
||||
}
|
||||
#undef EPS
|
||||
#undef MAXM
|
||||
Reference in New Issue
Block a user