std_complex.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_FWD_CORE_STD_COMPLEX_HPP
#define STAN_MATH_FWD_CORE_STD_COMPLEX_HPP
 
#include <stan/math/prim/core/complex_base.hpp>
#include <stan/math/fwd/core/fvar.hpp>
#include <stan/math/prim/meta.hpp>
#include <complex>
 
namespace std {
 
template <typename T>
class complex<stan::math::fvar<T>>
    : public stan::math::complex_base<stan::math::fvar<T>> {
 public:
  using base_t = stan::math::complex_base<stan::math::fvar<T>>;
 
  complex() = default;
 
  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 V>
  complex(const U& re, const V& im) : base_t(re, im) {}
 
  template <typename U, typename = stan::require_arithmetic_t<U>>
  auto& operator=(const std::complex<U>& x) {
    this->re_ = x.real();
    this->im_ = x.imag();
    return *this;
  }
};
 
}  // namespace std
 
#endif