diag_pre_multiply.hpp

Go to the documentation of this file.

#ifndef __STAN__MATH__MATRIX__DIAG_PRE_MULTIPLY_HPP__
#define __STAN__MATH__MATRIX__DIAG_PRE_MULTIPLY_HPP__

#include <stdexcept>
#include <boost/math/tools/promotion.hpp>
#include <stan/math/matrix/Eigen.hpp>

namespace stan {
  namespace math {

    template <typename T1, typename T2, int R1,int C1,int R2,int C2>
    Eigen::Matrix<typename boost::math::tools::promote_args<T1,T2>::type, R2, C2>
    diag_pre_multiply(const Eigen::Matrix<T1,R1,C1>& m1,
                  const Eigen::Matrix<T2,R2,C2>& m2) {
      if (m1.cols() != 1 && m1.rows() != 1)
        throw std::domain_error("m1 must be a vector");
      if (m1.size() != m2.rows())
        throw std::domain_error("m1 must have same length as m2 has rows");
      Eigen::Matrix<typename boost::math::tools::promote_args<T1,T2>::type, R2, C2>
        result(m2.rows(),m2.cols());
      for (int i = 0; i < m2.rows(); ++i)
        for (int j = 0; i < m2.cols(); ++j)
          result(i,j) = m1(i) * m2(i,j);
      return result;
    }

  }
}
#endif