add.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_PRIM_FUN_ADD_HPP
#define STAN_MATH_PRIM_FUN_ADD_HPP
 
#include <stan/math/prim/meta.hpp>
#include <stan/math/prim/err.hpp>
#include <stan/math/prim/fun/Eigen.hpp>
 
namespace stan {
namespace math {
 
template <typename ScalarA, typename ScalarB,
          require_all_stan_scalar_t<ScalarA, ScalarB>* = nullptr,
          require_all_not_var_t<ScalarA, ScalarB>* = nullptr>
inline return_type_t<ScalarA, ScalarB> add(const ScalarA& a, const ScalarB& b) {
  return a + b;
}
 
template <typename Mat1, typename Mat2,
          require_all_eigen_t<Mat1, Mat2>* = nullptr,
          require_all_not_st_var<Mat1, Mat2>* = nullptr>
inline auto add(Mat1&& m1, Mat2&& m2) {
  check_matching_dims("add", "m1", m1, "m2", m2);
  return make_holder([](auto&& m1_, auto&& m2_) { return m1_ + m2_; },
                     std::forward<Mat1>(m1), std::forward<Mat2>(m2));
}
 
template <typename Mat, typename Scal, require_eigen_t<Mat>* = nullptr,
          require_stan_scalar_t<Scal>* = nullptr,
          require_all_not_st_var<Mat, Scal>* = nullptr>
inline auto add(Mat&& m, const Scal c) {
  return make_holder([c](auto&& m_) { return (m_.array() + c).matrix(); },
                     std::forward<Mat>(m));
}
 
template <typename Scal, typename Mat, require_stan_scalar_t<Scal>* = nullptr,
          require_eigen_t<Mat>* = nullptr,
          require_all_not_st_var<Scal, Mat>* = nullptr>
inline auto add(const Scal c, Mat&& m) {
  return make_holder([c](auto&& m_) { return (c + m_.array()).matrix(); },
                     std::forward<Mat>(m));
}
 
}  // namespace math
}  // namespace stan
 
#endif