qpms/tests/tbeyn.c

41 lines
1.1 KiB
C

#include <qpms/beyn.h>
#include <stdio.h>
#include <string.h>
// Matrix as in Beyn, section 4.11
int M_function(complex double *target, const size_t m, const complex double z, void *no_params) {
complex double d = 2*m - 4*z / (6*m);
complex double od = -((double)m) - z / (6*m);
memset(target, 0, m*m*sizeof(complex double));
for (int i = 0; i < m; ++i) {
target[i*m + i] = d;
if(i > 0) target[i*m + i-1] = od;
if(i < m - 1) target[i*m + i+1] = od;
}
target[m*(m-1) + m-1] = d/2 + z/(z-1);
return 0;
}
int main() {
complex double z0 = 150+2*I;
double Rx = 148, Ry = 148;
int L = 10, N = 50, dim = 400;
beyn_contour_t *contour = beyn_contour_ellipse(z0, Rx, Ry, N);
beyn_result_t *result;
int K = beyn_solve(&result, dim, L, M_function, NULL /*M_inv_Vhat_function*/, NULL /*params*/,
contour, 1e-4, 0);
printf("Found %d eigenvalues:\n", K);
for (int i = 0; i < K; ++i) {
complex double eig = result->eigval[i];
printf("%d: %g%+gj\n", i, creal(eig), cimag(eig));
}
free(contour);
beyn_result_free(result);
return 0;
}