diff options
author | Peter Carbonetto | 2017-08-02 13:07:23 -0500 |
---|---|---|
committer | GitHub | 2017-08-02 13:07:23 -0500 |
commit | adaf9557f776ca274b51e921af0542ef1b84eb61 (patch) | |
tree | 8f7581bec20eb815848d02f926cb949ed038008b /src/eigenlib.cpp | |
parent | 84360c191f418bf8682b35e0c8235fcc3bd19a06 (diff) | |
parent | 1e49e7a27d0f4b811a87c64db1e875779766f6b0 (diff) | |
download | pangemma-adaf9557f776ca274b51e921af0542ef1b84eb61.tar.gz |
Merge pull request #62 from genenetwork/spacing
Spacing and coding style.
Diffstat (limited to 'src/eigenlib.cpp')
-rw-r--r-- | src/eigenlib.cpp | 103 |
1 files changed, 50 insertions, 53 deletions
diff --git a/src/eigenlib.cpp b/src/eigenlib.cpp index 733dae1..a8c545c 100644 --- a/src/eigenlib.cpp +++ b/src/eigenlib.cpp @@ -16,13 +16,13 @@ along with this program. If not, see <http://www.gnu.org/licenses/>. */ -#include <iostream> +#include "Eigen/Dense" +#include "gsl/gsl_linalg.h" +#include "gsl/gsl_matrix.h" +#include "gsl/gsl_vector.h" #include <cmath> +#include <iostream> #include <vector> -#include "gsl/gsl_vector.h" -#include "gsl/gsl_matrix.h" -#include "gsl/gsl_linalg.h" -#include "Eigen/Dense" using namespace std; using namespace Eigen; @@ -34,82 +34,79 @@ using namespace Eigen; // eigen, 1x or 0.3x slower than lapack // invert, 20x or 10x faster than lapack // -void eigenlib_dgemm (const char *TransA, const char *TransB, - const double alpha, const gsl_matrix *A, - const gsl_matrix *B, const double beta, - gsl_matrix *C) { - Map<Matrix<double, Dynamic, Dynamic, RowMajor>, 0, OuterStride<Dynamic> > - A_mat(A->data, A->size1, A->size2, OuterStride<Dynamic>(A->tda) ); - Map<Matrix<double, Dynamic, Dynamic, RowMajor>, 0, OuterStride<Dynamic> > - B_mat(B->data, B->size1, B->size2, OuterStride<Dynamic>(B->tda) ); - Map<Matrix<double, Dynamic, Dynamic, RowMajor>, 0, OuterStride<Dynamic> > - C_mat(C->data, C->size1, C->size2, OuterStride<Dynamic>(C->tda) ); +void eigenlib_dgemm(const char *TransA, const char *TransB, const double alpha, + const gsl_matrix *A, const gsl_matrix *B, const double beta, + gsl_matrix *C) { + Map<Matrix<double, Dynamic, Dynamic, RowMajor>, 0, OuterStride<Dynamic>> + A_mat(A->data, A->size1, A->size2, OuterStride<Dynamic>(A->tda)); + Map<Matrix<double, Dynamic, Dynamic, RowMajor>, 0, OuterStride<Dynamic>> + B_mat(B->data, B->size1, B->size2, OuterStride<Dynamic>(B->tda)); + Map<Matrix<double, Dynamic, Dynamic, RowMajor>, 0, OuterStride<Dynamic>> + C_mat(C->data, C->size1, C->size2, OuterStride<Dynamic>(C->tda)); - if (*TransA=='N' || *TransA=='n') { - if (*TransB=='N' || *TransB=='n') { - C_mat=alpha*A_mat*B_mat+beta*C_mat; + if (*TransA == 'N' || *TransA == 'n') { + if (*TransB == 'N' || *TransB == 'n') { + C_mat = alpha * A_mat * B_mat + beta * C_mat; } else { - C_mat=alpha*A_mat*B_mat.transpose()+beta*C_mat; + C_mat = alpha * A_mat * B_mat.transpose() + beta * C_mat; } } else { - if (*TransB=='N' || *TransB=='n') { - C_mat=alpha*A_mat.transpose()*B_mat+beta*C_mat; + if (*TransB == 'N' || *TransB == 'n') { + C_mat = alpha * A_mat.transpose() * B_mat + beta * C_mat; } else { - C_mat=alpha*A_mat.transpose()*B_mat.transpose()+beta*C_mat; + C_mat = alpha * A_mat.transpose() * B_mat.transpose() + beta * C_mat; } } return; } -void eigenlib_dgemv (const char *TransA, const double alpha, - const gsl_matrix *A, const gsl_vector *x, - const double beta, gsl_vector *y) { - Map<Matrix<double, Dynamic, Dynamic, RowMajor>, 0, OuterStride<Dynamic> > - A_mat(A->data, A->size1, A->size2, OuterStride<Dynamic>(A->tda) ); - Map<Matrix<double, Dynamic, 1>, 0, InnerStride<Dynamic> > - x_vec(x->data, x->size, InnerStride<Dynamic>(x->stride) ); - Map<Matrix<double, Dynamic, 1>, 0, InnerStride<Dynamic> > - y_vec(y->data, y->size, InnerStride<Dynamic>(y->stride) ); +void eigenlib_dgemv(const char *TransA, const double alpha, const gsl_matrix *A, + const gsl_vector *x, const double beta, gsl_vector *y) { + Map<Matrix<double, Dynamic, Dynamic, RowMajor>, 0, OuterStride<Dynamic>> + A_mat(A->data, A->size1, A->size2, OuterStride<Dynamic>(A->tda)); + Map<Matrix<double, Dynamic, 1>, 0, InnerStride<Dynamic>> x_vec( + x->data, x->size, InnerStride<Dynamic>(x->stride)); + Map<Matrix<double, Dynamic, 1>, 0, InnerStride<Dynamic>> y_vec( + y->data, y->size, InnerStride<Dynamic>(y->stride)); - if (*TransA=='N' || *TransA=='n') { - y_vec=alpha*A_mat*x_vec+beta*y_vec; + if (*TransA == 'N' || *TransA == 'n') { + y_vec = alpha * A_mat * x_vec + beta * y_vec; } else { - y_vec=alpha*A_mat.transpose()*x_vec+beta*y_vec; + y_vec = alpha * A_mat.transpose() * x_vec + beta * y_vec; } return; } void eigenlib_invert(gsl_matrix *A) { - Map<Matrix<double, Dynamic, Dynamic, RowMajor> > - A_mat(A->data, A->size1, A->size2); - A_mat=A_mat.inverse(); + Map<Matrix<double, Dynamic, Dynamic, RowMajor>> A_mat(A->data, A->size1, + A->size2); + A_mat = A_mat.inverse(); return; } -void eigenlib_dsyr (const double alpha, const gsl_vector *b, gsl_matrix *A) { - Map<Matrix<double, Dynamic, Dynamic, RowMajor> > - A_mat(A->data, A->size1, A->size2); - Map<Matrix<double, Dynamic, 1>, 0, OuterStride<Dynamic> > - b_vec(b->data, b->size, OuterStride<Dynamic>(b->stride) ); - A_mat=alpha*b_vec*b_vec.transpose()+A_mat; +void eigenlib_dsyr(const double alpha, const gsl_vector *b, gsl_matrix *A) { + Map<Matrix<double, Dynamic, Dynamic, RowMajor>> A_mat(A->data, A->size1, + A->size2); + Map<Matrix<double, Dynamic, 1>, 0, OuterStride<Dynamic>> b_vec( + b->data, b->size, OuterStride<Dynamic>(b->stride)); + A_mat = alpha * b_vec * b_vec.transpose() + A_mat; return; } -void eigenlib_eigensymm (const gsl_matrix *G, gsl_matrix *U, - gsl_vector *eval) { - Map<Matrix<double, Dynamic, Dynamic, RowMajor>, 0, OuterStride<Dynamic> > - G_mat(G->data, G->size1, G->size2, OuterStride<Dynamic>(G->tda) ); - Map<Matrix<double, Dynamic, Dynamic, RowMajor>, 0, OuterStride<Dynamic> > - U_mat(U->data, U->size1, U->size2, OuterStride<Dynamic>(U->tda) ); - Map<Matrix<double, Dynamic, 1>, 0, OuterStride<Dynamic> > - eval_vec(eval->data, eval->size, OuterStride<Dynamic>(eval->stride) ); +void eigenlib_eigensymm(const gsl_matrix *G, gsl_matrix *U, gsl_vector *eval) { + Map<Matrix<double, Dynamic, Dynamic, RowMajor>, 0, OuterStride<Dynamic>> + G_mat(G->data, G->size1, G->size2, OuterStride<Dynamic>(G->tda)); + Map<Matrix<double, Dynamic, Dynamic, RowMajor>, 0, OuterStride<Dynamic>> + U_mat(U->data, U->size1, U->size2, OuterStride<Dynamic>(U->tda)); + Map<Matrix<double, Dynamic, 1>, 0, OuterStride<Dynamic>> eval_vec( + eval->data, eval->size, OuterStride<Dynamic>(eval->stride)); SelfAdjointEigenSolver<MatrixXd> es(G_mat); if (es.info() != Success) abort(); - eval_vec=es.eigenvalues(); - U_mat=es.eigenvectors(); + eval_vec = es.eigenvalues(); + U_mat = es.eigenvectors(); return; } |