diff options
Diffstat (limited to 'src/pj_zpoly1.c')
| -rw-r--r-- | src/pj_zpoly1.c | 49 |
1 files changed, 49 insertions, 0 deletions
diff --git a/src/pj_zpoly1.c b/src/pj_zpoly1.c new file mode 100644 index 00000000..80d85469 --- /dev/null +++ b/src/pj_zpoly1.c @@ -0,0 +1,49 @@ +/* evaluate complex polynomial */ +#ifndef lint +static const char SCCSID[]="@(#)pj_zpoly1.c 4.3 93/06/12 GIE REL"; +#endif +#include <projects.h> +/* note: coefficients are always from C_1 to C_n +** i.e. C_0 == (0., 0) +** n should always be >= 1 though no checks are made +*/ + COMPLEX +pj_zpoly1(COMPLEX z, COMPLEX *C, int n) { + COMPLEX a; + double t; + + a = *(C += n); + while (n-- > 0) { + a.r = (--C)->r + z.r * (t = a.r) - z.i * a.i; + a.i = C->i + z.r * a.i + z.i * t; + } + a.r = z.r * (t = a.r) - z.i * a.i; + a.i = z.r * a.i + z.i * t; + return a; +} +/* evaluate complex polynomial and derivative */ + COMPLEX +pj_zpolyd1(COMPLEX z, COMPLEX *C, int n, COMPLEX *der) { + COMPLEX a, b; + double t; + int first = 1; + + a = *(C += n); + while (n-- > 0) { + if (first) { + first = 0; + b = a; + } else { + b.r = a.r + z.r * (t = b.r) - z.i * b.i; + b.i = a.i + z.r * b.i + z.i * t; + } + a.r = (--C)->r + z.r * (t = a.r) - z.i * a.i; + a.i = C->i + z.r * a.i + z.i * t; + } + b.r = a.r + z.r * (t = b.r) - z.i * b.i; + b.i = a.i + z.r * b.i + z.i * t; + a.r = z.r * (t = a.r) - z.i * a.i; + a.i = z.r * a.i + z.i * t; + *der = b; + return a; +} |
