multi_normal_prec_rng.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_PRIM_PROB_MULTI_NORMAL_PREC_RNG_HPP
#define STAN_MATH_PRIM_PROB_MULTI_NORMAL_PREC_RNG_HPP
 
#include <stan/math/prim/fun/Eigen.hpp>
#include <stan/math/prim/meta.hpp>
#include <stan/math/prim/err.hpp>
#include <stan/math/prim/fun/as_column_vector_or_scalar.hpp>
#include <stan/math/prim/fun/to_ref.hpp>
#include <stan/math/prim/fun/vector_seq_view.hpp>
#include <boost/random/normal_distribution.hpp>
#include <boost/random/variate_generator.hpp>
 
namespace stan {
namespace math {
 
template <typename T_loc, class RNG>
inline typename StdVectorBuilder<true, Eigen::VectorXd, T_loc>::type
multi_normal_prec_rng(const T_loc &mu, const Eigen::MatrixXd &S, RNG &rng) {
  using boost::normal_distribution;
  using boost::variate_generator;
  static constexpr const char *function = "multi_normal_prec_rng";
  check_positive(function, "Precision matrix rows", S.rows());
 
  vector_seq_view<T_loc> mu_vec(mu);
  check_positive(function, "number of location parameter vectors",
                 mu_vec.size());
  size_t size_mu = mu_vec[0].size();
 
  size_t N = mu_vec.size();
 
  for (size_t i = 1; i < N; i++) {
    check_size_match(function,
                     "Size of one of the vectors of "
                     "the location variable",
                     mu_vec[i].size(),
                     "Size of the first vector of the "
                     "location variable",
                     size_mu);
  }
  check_size_match(function, "Rows of location parameter", size_mu, "Rows of S",
                   S.rows());
 
  for (size_t i = 0; i < N; i++) {
    check_finite(function, "Location parameter", mu_vec[i]);
  }
  const auto &S_ref = to_ref(S);
  check_finite(function, "Precision matrix", S_ref);
  check_symmetric(function, "Precision matrix", S_ref);
  Eigen::LLT<Eigen::MatrixXd> llt_of_S = S_ref.llt();
  check_pos_definite(function, "precision matrix argument", llt_of_S);
 
  StdVectorBuilder<true, Eigen::VectorXd, T_loc> output(N);
 
  variate_generator<RNG &, normal_distribution<>> std_normal_rng(
      rng, normal_distribution<>(0, 1));
 
  for (size_t n = 0; n < N; ++n) {
    Eigen::VectorXd z(S.cols());
    for (int i = 0; i < S.cols(); i++) {
      z(i) = std_normal_rng();
    }
 
    output[n]
        = as_column_vector_or_scalar(mu_vec[n]) + llt_of_S.matrixU().solve(z);
  }
 
  return output.data();
}
 
}  // namespace math
}  // namespace stan
#endif