polar.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_PRIM_FUN_POLAR_HPP
#define STAN_MATH_PRIM_FUN_POLAR_HPP
 
#include <stan/math/prim/meta.hpp>
#include <stan/math/prim/fun/cos.hpp>
#include <stan/math/prim/fun/is_inf.hpp>
#include <stan/math/prim/fun/sin.hpp>
#include <cmath>
#include <complex>
#include <limits>
 
namespace stan {
namespace math {
namespace internal {
template <typename U, typename V>
inline complex_return_t<U, V> complex_polar(const U& r, const V& theta) {
  if (!(r >= 0) || is_inf(theta)) {
    return {std::numeric_limits<double>::quiet_NaN()};
  }
  return {r * cos(theta), r * sin(theta)};
}
}  // namespace internal
 
template <typename U, typename V, require_all_arithmetic_t<U, V>* = nullptr>
inline std::complex<double> polar(U r, V theta) {
  return internal::complex_polar(static_cast<double>(r),
                                 static_cast<double>(theta));
}
 
}  // namespace math
}  // namespace stan
 
#endif