summaryrefslogtreecommitdiff
path: root/gsl-1.9/poly/zsolve.c
diff options
context:
space:
mode:
Diffstat (limited to 'gsl-1.9/poly/zsolve.c')
-rw-r--r--gsl-1.9/poly/zsolve.c85
1 files changed, 85 insertions, 0 deletions
diff --git a/gsl-1.9/poly/zsolve.c b/gsl-1.9/poly/zsolve.c
new file mode 100644
index 0000000..ae2aa51
--- /dev/null
+++ b/gsl-1.9/poly/zsolve.c
@@ -0,0 +1,85 @@
+/* poly/zsolve.c
+ *
+ * Copyright (C) 1996, 1997, 1998, 1999, 2000 Brian Gough
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or (at
+ * your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ */
+
+/* zsolve.c - finds the complex roots of = 0 */
+
+#include <config.h>
+#include <math.h>
+#include <stdlib.h>
+#include <gsl/gsl_math.h>
+#include <gsl/gsl_errno.h>
+#include <gsl/gsl_complex.h>
+#include <gsl/gsl_poly.h>
+
+/* C-style matrix elements */
+#define MAT(m,i,j,n) ((m)[(i)*(n) + (j)])
+
+/* Fortran-style matrix elements */
+#define FMAT(m,i,j,n) ((m)[((i)-1)*(n) + ((j)-1)])
+
+#include "companion.c"
+#include "balance.c"
+#include "qr.c"
+
+int
+gsl_poly_complex_solve (const double *a, size_t n,
+ gsl_poly_complex_workspace * w,
+ gsl_complex_packed_ptr z)
+{
+ int status;
+ double *m;
+
+ if (n == 0)
+ {
+ GSL_ERROR ("number of terms must be a positive integer", GSL_EINVAL);
+ }
+
+ if (n == 1)
+ {
+ GSL_ERROR ("cannot solve for only one term", GSL_EINVAL);
+ }
+
+ if (a[n - 1] == 0)
+ {
+ GSL_ERROR ("leading term of polynomial must be non-zero", GSL_EINVAL) ;
+ }
+
+ if (w->nc != n - 1)
+ {
+ GSL_ERROR ("size of workspace does not match polynomial", GSL_EINVAL);
+ }
+
+ m = w->matrix;
+
+ set_companion_matrix (a, n - 1, m);
+
+ balance_companion_matrix (m, n - 1);
+
+ status = qr_companion (m, n - 1, z);
+
+ if (status)
+ {
+ GSL_ERROR("root solving qr method failed to converge", GSL_EFAILED);
+ }
+
+ return GSL_SUCCESS;
+}
+
+
+