assign.hpp

Go to the documentation of this file.

#ifndef STAN__MATH__MATRIX__ASSIGN_HPP
#define STAN__MATH__MATRIX__ASSIGN_HPP

#include <vector>
#include <sstream>
#include <stdexcept>

#include <stan/math/matrix/Eigen.hpp>
#include <stan/math/error_handling/matrix/check_matching_sizes.hpp>
#include <stan/math/error_handling/matrix/check_matching_dims.hpp>


#include <iostream>

namespace stan {

  namespace math {

    // Recursive assignment with size match checking and promotion

    template <typename LHS, typename RHS>
    inline void
    assign(LHS& lhs, const RHS& rhs) {
      lhs = rhs;
    }

    template <typename LHS, typename RHS, int R1, int C1, int R2, int C2>
    inline void 
    assign(Eigen::Matrix<LHS,R1,C1>& x, 
           const Eigen::Matrix<RHS,R2,C2>& y) {
      std::stringstream ss;
      ss << "error in call to assign(Eigen::Matrix,Eigen::Matrix)"
         << "; shapes must match, but found"
         << " R1=" << R1 
         << "; C1=" << C1
         << "; R2=" << R2
         << "; C2=" << C2;
      throw std::domain_error(ss.str());
    }

    template <typename LHS, typename RHS, int R, int C>
    inline void 
    assign(Eigen::Matrix<LHS,R,C>& x, 
           const Eigen::Matrix<RHS,R,C>& y) {
      stan::math::check_matching_dims("assign(%1%)",x,"x",
                                      y,"y",(double*)0);
      for (int i = 0; i < x.size(); ++i)
        assign(x(i),y(i));
    }

    template <typename LHS, typename RHS, int R, int C>
    inline void 
    assign(Eigen::Block<LHS> x,
           const Eigen::Matrix<RHS,R,C>& y) {
      stan::math::check_matching_sizes("assign(%1%)",x,"x",
                                       y,"y",(double*)0);
      for (int n = 0; n < y.cols(); ++n)
        for (int m = 0; m < y.rows(); ++m)
          assign(x(m,n),y(m,n));
    }


    template <typename LHS, typename RHS>
    inline void 
    assign(std::vector<LHS>& x, const std::vector<RHS>& y) {
      stan::math::check_matching_sizes("assign(%1%)",x,"x",
                                       y,"y",(double*)0);
      for (size_t i = 0; i < x.size(); ++i)
        assign(x[i],y[i]);
    }


  }
}
#endif