promote_elements.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_PRIM_FUN_PROMOTE_ELEMENTS_HPP
#define STAN_MATH_PRIM_FUN_PROMOTE_ELEMENTS_HPP
 
#include <stan/math/prim/meta.hpp>
#include <stan/math/prim/fun/Eigen.hpp>
#include <cstddef>
#include <vector>
 
namespace stan {
namespace math {
 
template <typename T, typename S>
struct promote_elements {
  inline static T promote(const S& u) { return u; }
};
 
template <typename T>
struct promote_elements<T, T> {
  inline static const T& promote(const T& u) { return u; }
};
 
template <typename T, typename S>
struct promote_elements<std::vector<T>, std::vector<S> > {
  inline static std::vector<T> promote(const std::vector<S>& u) {
    std::vector<T> t;
    t.reserve(u.size());
    for (size_t i = 0; i < u.size(); ++i) {
      t.push_back(promote_elements<T, S>::promote(u[i]));
    }
    return t;
  }
};
 
template <typename T>
struct promote_elements<std::vector<T>, std::vector<T> > {
  inline static const std::vector<T>& promote(const std::vector<T>& u) {
    return u;
  }
};
 
template <typename T, typename S, int R, int C>
struct promote_elements<Eigen::Matrix<T, R, C>, Eigen::Matrix<S, R, C> > {
  inline static Eigen::Matrix<T, R, C> promote(
      const Eigen::Matrix<S, R, C>& u) {
    Eigen::Matrix<T, Eigen::Dynamic, Eigen::Dynamic> t(u.rows(), u.cols());
    for (int i = 0; i < u.size(); ++i) {
      t(i) = promote_elements<T, S>::promote(u(i));
    }
    return t;
  }
};
 
template <typename T, int R, int C>
struct promote_elements<Eigen::Matrix<T, R, C>, Eigen::Matrix<T, R, C> > {
  inline static const Eigen::Matrix<T, R, C>& promote(
      const Eigen::Matrix<T, R, C>& u) {
    return u;
  }
};
 
}  // namespace math
}  // namespace stan
 
#endif