value_of_rec.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_PRIM_FUN_VALUE_OF_REC_HPP
#define STAN_MATH_PRIM_FUN_VALUE_OF_REC_HPP
 
#include <stan/math/prim/fun/Eigen.hpp>
#include <stan/math/prim/meta.hpp>
#include <stan/math/prim/functor/apply.hpp>
#include <stan/math/prim/functor/make_holder_tuple.hpp>
#include <complex>
#include <cstddef>
#include <vector>
 
namespace stan {
namespace math {
 
template <typename Tuple, require_tuple_t<Tuple>* = nullptr>
inline auto value_of_rec(Tuple&& tup);
 
template <typename T, require_not_same_t<double, T>* = nullptr>
inline std::vector<promote_scalar_t<double, T>> value_of_rec(
    const std::vector<T>& x);
 
template <typename T, require_stan_scalar_t<T> = nullptr>
inline double constexpr value_of_rec(const T x) noexcept {
  return static_cast<double>(x);
}
 
inline double constexpr value_of_rec(double x) noexcept { return x; }
 
template <typename T>
inline std::complex<double> value_of_rec(const std::complex<T>& x) {
  return {value_of_rec(x.real()), value_of_rec(x.imag())};
}
 
template <typename T, require_std_vector_t<T>* = nullptr,
          require_vt_same<double, T>* = nullptr>
inline T value_of_rec(T&& x) {
  return std::forward<T>(x);
}
 
template <typename T, require_not_st_same<T, double>* = nullptr,
          require_eigen_t<T>* = nullptr>
inline auto value_of_rec(T&& M) {
  return make_holder(
      [](auto&& m) {
        return m.unaryExpr([](auto x) { return value_of_rec(x); });
      },
      std::forward<T>(M));
}
 
template <typename T, require_st_same<T, double>* = nullptr,
          require_eigen_t<T>* = nullptr>
inline decltype(auto) value_of_rec(T&& x) {
  if constexpr (is_plain_type<T>::value || is_holder_v<T>) {
    if constexpr (std::is_rvalue_reference_v<T&&>) {
      return std::decay_t<T>(std::forward<T>(x));
    } else {
      return x;
    }
  } else {
    return make_holder([](auto&& m) { return m; }, std::forward<T>(x));
  }
}
 
template <typename T, require_not_same_t<double, T>*>
inline std::vector<promote_scalar_t<double, T>> value_of_rec(
    const std::vector<T>& x) {
  size_t x_size = x.size();
  std::vector<promote_scalar_t<double, plain_type_t<T>>> result(x_size);
  for (size_t i = 0; i < x_size; i++) {
    result[i] = value_of_rec(x[i]);
  }
  return result;
}
 
template <typename Tuple, require_tuple_t<Tuple>*>
inline auto value_of_rec(Tuple&& tup) {
  return stan::math::apply(
      [](auto&&... args) {
        return make_holder_tuple(
            value_of_rec(std::forward<decltype(args)>(args))...);
      },
      std::forward<Tuple>(tup));
}
 
}  // namespace math
}  // namespace stan
 
#endif