save_varis.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_REV_CORE_SAVE_VARIS_HPP
#define STAN_MATH_REV_CORE_SAVE_VARIS_HPP
 
#include <stan/math/prim/fun/Eigen.hpp>
#include <stan/math/prim/functor/apply.hpp>
#include <stan/math/prim/meta.hpp>
#include <stan/math/rev/meta.hpp>
#include <stan/math/rev/core/var.hpp>
 
#include <utility>
#include <vector>
 
namespace stan {
namespace math {
 
template <typename... Pargs>
inline vari** save_varis(vari** dest, const var& x, Pargs&&... args);
 
template <typename VarVec, require_std_vector_vt<is_var, VarVec>* = nullptr,
          typename... Pargs>
inline vari** save_varis(vari** dest, VarVec&& x, Pargs&&... args);
 
template <typename VecContainer,
          require_std_vector_st<is_var, VecContainer>* = nullptr,
          require_std_vector_vt<is_container, VecContainer>* = nullptr,
          typename... Pargs>
inline vari** save_varis(vari** dest, VecContainer&& x, Pargs&&... args);
 
template <typename EigT, require_eigen_vt<is_var, EigT>* = nullptr,
          typename... Pargs>
inline vari** save_varis(vari** dest, EigT&& x, Pargs&&... args);
 
template <typename Arith, require_st_arithmetic<Arith>* = nullptr,
          typename... Pargs>
inline vari** save_varis(vari** dest, Arith&& x, Pargs&&... args);
 
template <typename Tuple, require_tuple_t<Tuple>* = nullptr, typename... Pargs>
inline vari** save_varis(vari** dest, Tuple&& x, Pargs&&... args);
 
inline vari** save_varis(vari** dest);
 
template <typename... Pargs>
inline vari** save_varis(vari** dest, const var& x, Pargs&&... args) {
  *dest = x.vi_;
  return save_varis(dest + 1, std::forward<Pargs>(args)...);
}
 
template <typename VarVec, require_std_vector_vt<is_var, VarVec>*,
          typename... Pargs>
inline vari** save_varis(vari** dest, VarVec&& x, Pargs&&... args) {
  for (int i = 0; i < x.size(); ++i) {
    dest[i] = x[i].vi_;
  }
  return save_varis(dest + x.size(), std::forward<Pargs>(args)...);
}
 
template <typename VecContainer, require_std_vector_st<is_var, VecContainer>*,
          require_std_vector_vt<is_container, VecContainer>*, typename... Pargs>
inline vari** save_varis(vari** dest, VecContainer&& x, Pargs&&... args) {
  for (size_t i = 0; i < x.size(); ++i) {
    dest = save_varis(dest, x[i]);
  }
  return save_varis(dest, std::forward<Pargs>(args)...);
}
 
template <typename EigT, require_eigen_vt<is_var, EigT>*, typename... Pargs>
inline vari** save_varis(vari** dest, EigT&& x, Pargs&&... args) {
  for (int i = 0; i < x.size(); ++i) {
    dest[i] = x.coeff(i).vi_;
  }
  return save_varis(dest + x.size(), std::forward<Pargs>(args)...);
}
 
template <typename Arith, require_st_arithmetic<Arith>*, typename... Pargs>
inline vari** save_varis(vari** dest, Arith&& x, Pargs&&... args) {
  return save_varis(dest, std::forward<Pargs>(args)...);
}
 
template <typename Tuple, require_tuple_t<Tuple>*, typename... Pargs>
inline vari** save_varis(vari** dest, Tuple&& x, Pargs&&... args) {
  dest = stan::math::apply(
      [dest](auto&&... tuple_args) { return save_varis(dest, tuple_args...); },
      std::forward<Tuple>(x));
  return save_varis(dest, std::forward<Pargs>(args)...);
}
 
inline vari** save_varis(vari** dest) { return dest; }
 
}  // namespace math
}  // namespace stan
 
#endif