std_complex.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_REV_CORE_STD_COMPLEX_HPP
#define STAN_MATH_REV_CORE_STD_COMPLEX_HPP
 
#include <stan/math/prim/meta.hpp>
#include <stan/math/prim/core/complex_base.hpp>
#include <stan/math/rev/core/var.hpp>
#include <cmath>
#include <complex>
 
namespace std {
 
template <>
class complex<stan::math::var>
    : public stan::math::complex_base<stan::math::var> {
 public:
  using base_t = stan::math::complex_base<stan::math::var>;
 
  complex() = default;
 
  template <typename U, typename V>
  complex(const U& re, const V& im) : base_t(re, im) {}
 
  template <typename U, typename = stan::require_stan_scalar_t<U>>
  complex(const U& re) : base_t(re) {}  // NOLINT(runtime/explicit)
 
  template <typename U>
  complex(const std::complex<U>& z)  // NOLINT(runtime/explicit)
      : base_t(z.real(), z.imag()) {}
 
  template <typename U, typename = stan::require_arithmetic_t<U>>
  auto& operator=(const std::complex<U>& x) {
    re_ = x.real();
    im_ = x.imag();
    return *this;
  }
};
 
}  // namespace std
 
#endif