about summary refs log tree commit diff
path: root/src/eigenlib.cpp
diff options
context:
space:
mode:
authorPeter Carbonetto2017-08-02 13:07:23 -0500
committerGitHub2017-08-02 13:07:23 -0500
commitadaf9557f776ca274b51e921af0542ef1b84eb61 (patch)
tree8f7581bec20eb815848d02f926cb949ed038008b /src/eigenlib.cpp
parent84360c191f418bf8682b35e0c8235fcc3bd19a06 (diff)
parent1e49e7a27d0f4b811a87c64db1e875779766f6b0 (diff)
downloadpangemma-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.cpp103
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;
 }