mirror of
https://github.com/bolero-MURAKAMI/Sprout.git
synced 2024-11-14 10:39:05 +00:00
109 lines
2.6 KiB
C++
109 lines
2.6 KiB
C++
#ifndef SPROUT_COMPLEX_COMPLEX_HPP
|
|
#define SPROUT_COMPLEX_COMPLEX_HPP
|
|
|
|
#include <sprout/config.hpp>
|
|
|
|
namespace sprout {
|
|
template<typename T>
|
|
class complex;
|
|
|
|
namespace detail {
|
|
template<typename T>
|
|
inline SPROUT_CONSTEXPR T
|
|
complex_norm(sprout::complex<T> const& x) {
|
|
return x.real() * x.real() + x.imag() * x.imag();
|
|
}
|
|
} // namespace detail
|
|
|
|
//
|
|
// complex
|
|
//
|
|
template<typename T>
|
|
class complex {
|
|
public:
|
|
typedef T value_type;
|
|
private:
|
|
T re_;
|
|
T im_;
|
|
public:
|
|
SPROUT_CONSTEXPR complex(T const& re = T(), T const& im = T()) SPROUT_NOEXCEPT
|
|
: re_(re), im_(im)
|
|
{}
|
|
SPROUT_CONSTEXPR complex(complex const&) = default;
|
|
template<typename X>
|
|
SPROUT_CONSTEXPR complex(complex<X> const& other) SPROUT_NOEXCEPT
|
|
: re_(other.real()), im_(other.imag())
|
|
{}
|
|
SPROUT_CONSTEXPR T real() const SPROUT_NOEXCEPT {
|
|
return re_;
|
|
}
|
|
void real(T re) SPROUT_NOEXCEPT {
|
|
re_ = re;
|
|
}
|
|
SPROUT_CONSTEXPR T imag() const SPROUT_NOEXCEPT {
|
|
return im_;
|
|
}
|
|
void imag(T im) SPROUT_NOEXCEPT {
|
|
im_ = im;
|
|
}
|
|
complex& operator=(T const& rhs) SPROUT_NOEXCEPT {
|
|
re_ = rhs;
|
|
im_ = T();
|
|
return *this;
|
|
}
|
|
complex& operator+=(T const& rhs) SPROUT_NOEXCEPT {
|
|
re_ += rhs;
|
|
return *this;
|
|
}
|
|
complex& operator-=(T const& rhs) SPROUT_NOEXCEPT {
|
|
re_ -= rhs;
|
|
return *this;
|
|
}
|
|
complex& operator*=(T const& rhs) SPROUT_NOEXCEPT {
|
|
re_ *= rhs;
|
|
im_ *= rhs;
|
|
return *this;
|
|
}
|
|
complex& operator/=(T const& rhs) SPROUT_NOEXCEPT {
|
|
re_ /= rhs;
|
|
im_ /= rhs;
|
|
return *this;
|
|
}
|
|
complex& operator=(complex const&) = default;
|
|
template<typename X>
|
|
complex& operator=(complex<X> const& rhs) SPROUT_NOEXCEPT {
|
|
re_ = rhs.real();
|
|
im_ = rhs.imag();
|
|
return *this;
|
|
}
|
|
template<typename X>
|
|
complex& operator+=(complex<X> const& rhs) SPROUT_NOEXCEPT {
|
|
re_ += rhs.real();
|
|
im_ += rhs.imag();
|
|
return *this;
|
|
}
|
|
template<typename X>
|
|
complex& operator-=(complex<X> const& rhs) SPROUT_NOEXCEPT {
|
|
re_ -= rhs.real();
|
|
im_ -= rhs.imag();
|
|
return *this;
|
|
}
|
|
template<typename X>
|
|
complex& operator*=(complex<X> const& rhs) SPROUT_NOEXCEPT {
|
|
return *this = complex(
|
|
re_ * rhs.real() - im_ * rhs.imag(),
|
|
re_ * rhs.imag() + im_ * rhs.real()
|
|
);
|
|
}
|
|
template<typename X>
|
|
complex& operator/=(complex<X> const& rhs) SPROUT_NOEXCEPT {
|
|
T n = sprout::detail::complex_norm(rhs);
|
|
return *this = complex(
|
|
(re_ * rhs.real() + im_ * rhs.imag()) / n,
|
|
(im_ * rhs.real() - re_ * rhs.imag()) / n
|
|
);
|
|
}
|
|
};
|
|
} // namespace sprout
|
|
|
|
#endif // #ifndef SPROUT_COMPLEX_COMPLEX_HPP
|