wishart_cholesky_rng.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_PRIM_PROB_WISHART_CHOLESKY_RNG_HPP
#define STAN_MATH_PRIM_PROB_WISHART_CHOLESKY_RNG_HPP
 
#include <stan/math/prim/meta.hpp>
#include <stan/math/prim/err.hpp>
#include <stan/math/prim/fun/crossprod.hpp>
#include <stan/math/prim/prob/chi_square_rng.hpp>
#include <stan/math/prim/prob/normal_rng.hpp>
#include <cmath>
 
namespace stan {
namespace math {
 
template <class RNG>
inline Eigen::MatrixXd wishart_cholesky_rng(double nu,
                                            const Eigen::MatrixXd& L_S,
                                            RNG& rng) {
  using Eigen::MatrixXd;
  static constexpr const char* function = "wishart_cholesky_rng";
  index_type_t<MatrixXd> k = L_S.rows();
  check_greater(function, "degrees of freedom > dims - 1", nu, k - 1);
  check_square(function, "Cholesky Scale matrix", L_S);
  check_cholesky_factor(function, "Cholesky Scale matrix", L_S);
 
  MatrixXd B = MatrixXd::Zero(k, k);
  for (int j = 0; j < k; ++j) {
    for (int i = 0; i < j; ++i) {
      B(i, j) = normal_rng(0, 1, rng);
    }
    B(j, j) = std::sqrt(chi_square_rng(nu - j, rng));
  }
  return L_S.template triangularView<Eigen::Lower>() * B.transpose();
}
 
}  // namespace math
}  // namespace stan
#endif