gen_workspace *
gen_alloc(size_t n, int compute_schur)
{
gen_workspace *w;
w = (gen_workspace *) calloc(1, sizeof(gen_workspace));
w->gen_p = gsl_eigen_gen_alloc(n);
w->A = gsl_matrix_alloc(n, n);
w->B = gsl_matrix_alloc(n, n);
w->alpha = gsl_vector_complex_alloc(n);
w->beta = gsl_vector_alloc(n);
w->evals = gsl_vector_complex_alloc(n);
w->compute_schur = compute_schur;
if (compute_schur)
{
w->Q = gsl_matrix_alloc(n, n);
w->Z = gsl_matrix_alloc(n, n);
gsl_eigen_gen_params(1, 1, 0, w->gen_p);
}
return (w);
} /* gen_alloc() */
/**
* C++ version of gsl_eigen_gen_params().
* @param compute_s An integer
* @param compute_t An integer
* @param balance An integer
* @param w A workspace
*/
inline void gen_params( int const compute_s, int const compute_t, int const balance,
gen_workspace& w ){
gsl_eigen_gen_params( compute_s, compute_t, balance, w.get() ); }
void
test_eigen_gen_pencil(const gsl_matrix * A, const gsl_matrix * B,
size_t count, const char * desc, int test_schur,
test_eigen_gen_workspace *w)
{
const size_t N = A->size1;
size_t i;
gsl_matrix_memcpy(w->A, A);
gsl_matrix_memcpy(w->B, B);
if (test_schur)
{
gsl_eigen_genv_QZ(w->A, w->B, w->alphav, w->betav, w->evec, w->Q, w->Z, w->genv_p);
test_eigen_schur(A, w->A, w->Q, w->Z, count, "genv/A", desc);
test_eigen_schur(B, w->B, w->Q, w->Z, count, "genv/B", desc);
}
else
gsl_eigen_genv(w->A, w->B, w->alphav, w->betav, w->evec, w->genv_p);
test_eigen_gen_results(A, B, w->alphav, w->betav, w->evec, count, desc, "unsorted");
gsl_matrix_memcpy(w->A, A);
gsl_matrix_memcpy(w->B, B);
if (test_schur)
{
gsl_eigen_gen_params(1, 1, 0, w->gen_p);
gsl_eigen_gen_QZ(w->A, w->B, w->alpha, w->beta, w->Q, w->Z, w->gen_p);
test_eigen_schur(A, w->A, w->Q, w->Z, count, "gen/A", desc);
test_eigen_schur(B, w->B, w->Q, w->Z, count, "gen/B", desc);
}
else
{
gsl_eigen_gen_params(0, 0, 0, w->gen_p);
gsl_eigen_gen(w->A, w->B, w->alpha, w->beta, w->gen_p);
}
/* compute eval = alpha / beta values */
for (i = 0; i < N; ++i)
{
gsl_complex z, ai;
double bi;
ai = gsl_vector_complex_get(w->alpha, i);
bi = gsl_vector_get(w->beta, i);
GSL_SET_COMPLEX(&z, GSL_REAL(ai) / bi, GSL_IMAG(ai) / bi);
gsl_vector_complex_set(w->eval, i, z);
ai = gsl_vector_complex_get(w->alphav, i);
bi = gsl_vector_get(w->betav, i);
GSL_SET_COMPLEX(&z, GSL_REAL(ai) / bi, GSL_IMAG(ai) / bi);
gsl_vector_complex_set(w->evalv, i, z);
}
/* sort eval and evalv and test them */
gsl_eigen_nonsymmv_sort(w->eval, NULL, GSL_EIGEN_SORT_ABS_ASC);
gsl_eigen_nonsymmv_sort(w->evalv, NULL, GSL_EIGEN_SORT_ABS_ASC);
test_eigenvalues_complex(w->evalv, w->eval, "gen", desc);
gsl_eigen_genv_sort(w->alphav, w->betav, w->evec, GSL_EIGEN_SORT_ABS_ASC);
test_eigen_gen_results(A, B, w->alphav, w->betav, w->evec, count, desc, "abs/asc");
gsl_eigen_genv_sort(w->alphav, w->betav, w->evec, GSL_EIGEN_SORT_ABS_DESC);
test_eigen_gen_results(A, B, w->alphav, w->betav, w->evec, count, desc, "abs/desc");
} /* test_eigen_gen_pencil() */
