Spectra
DenseGenMatProd.h
1 // Copyright (C) 2016-2018 Yixuan Qiu <yixuan.qiu@cos.name>
2 //
3 // This Source Code Form is subject to the terms of the Mozilla
4 // Public License v. 2.0. If a copy of the MPL was not distributed
5 // with this file, You can obtain one at https://mozilla.org/MPL/2.0/.
6 
7 #ifndef DENSE_GEN_MAT_PROD_H
8 #define DENSE_GEN_MAT_PROD_H
9 
10 #include <Eigen/Core>
11 
12 namespace Spectra {
13 
14 
20 
29 template <typename Scalar>
31 {
32 private:
33  typedef Eigen::Matrix<Scalar, Eigen::Dynamic, Eigen::Dynamic> Matrix;
34  typedef Eigen::Matrix<Scalar, Eigen::Dynamic, 1> Vector;
35  typedef Eigen::Map<const Vector> MapConstVec;
36  typedef Eigen::Map<Vector> MapVec;
37  typedef const Eigen::Ref<const Matrix> ConstGenericMatrix;
38 
39  ConstGenericMatrix m_mat;
40 
41 public:
50  DenseGenMatProd(ConstGenericMatrix& mat) :
51  m_mat(mat)
52  {}
53 
57  int rows() const { return m_mat.rows(); }
61  int cols() const { return m_mat.cols(); }
62 
69  // y_out = A * x_in
70  void perform_op(const Scalar* x_in, Scalar* y_out) const
71  {
72  MapConstVec x(x_in, m_mat.cols());
73  MapVec y(y_out, m_mat.rows());
74  y.noalias() = m_mat * x;
75  }
76 };
77 
78 
79 } // namespace Spectra
80 
81 #endif // DENSE_GEN_MAT_PROD_H
void perform_op(const Scalar *x_in, Scalar *y_out) const
DenseGenMatProd(ConstGenericMatrix &mat)