mirror of
https://github.com/bolero-MURAKAMI/Sprout
synced 2024-11-12 21:09:01 +00:00
73 lines
1.9 KiB
C++
73 lines
1.9 KiB
C++
#ifndef SPROUT_COMPLEX_VALUES_HPP
|
|
#define SPROUT_COMPLEX_VALUES_HPP
|
|
|
|
#include <sprout/config.hpp>
|
|
#include <sprout/complex/complex.hpp>
|
|
#include <sprout/complex/operators.hpp>
|
|
#include <sprout/math/sin.hpp>
|
|
#include <sprout/math/cos.hpp>
|
|
#include <sprout/math/atan2.hpp>
|
|
#include <sprout/math/sqrt.hpp>
|
|
|
|
namespace sprout {
|
|
template<typename T>
|
|
SPROUT_CONSTEXPR T
|
|
norm(sprout::complex<T> const& x);
|
|
|
|
// 26.4.7, values:
|
|
template<typename T>
|
|
inline SPROUT_CONSTEXPR T
|
|
real(sprout::complex<T> const& x) {
|
|
return x.real();
|
|
}
|
|
template<typename T>
|
|
inline SPROUT_CONSTEXPR T
|
|
imag(sprout::complex<T> const& x) {
|
|
return x.imag();
|
|
}
|
|
template<typename T>
|
|
inline SPROUT_CONSTEXPR T
|
|
abs(sprout::complex<T> const& x) {
|
|
return sprout::sqrt(sprout::norm(x));
|
|
}
|
|
template<typename T>
|
|
inline SPROUT_CONSTEXPR T
|
|
arg(sprout::complex<T> const& x) {
|
|
return sprout::atan2(x.imag(), x.real());
|
|
}
|
|
template<typename T>
|
|
inline SPROUT_CONSTEXPR T
|
|
norm(sprout::complex<T> const& x) {
|
|
return x.real() * x.real() + x.imag() * x.imag();
|
|
}
|
|
template<typename T>
|
|
inline SPROUT_CONSTEXPR sprout::complex<T>
|
|
conj(sprout::complex<T> const& x) {
|
|
return sprout::complex<T>(x.real(), -x.imag());
|
|
}
|
|
namespace detail {
|
|
template<typename T>
|
|
inline SPROUT_CONSTEXPR sprout::complex<T>
|
|
proj_impl(sprout::complex<T> const& x, T const& den) {
|
|
return sprout::complex<T>(
|
|
T(2) * x.real() / den,
|
|
T(2) * x.imag() / den
|
|
);
|
|
}
|
|
} // detail
|
|
template<typename T>
|
|
inline SPROUT_CONSTEXPR sprout::complex<T>
|
|
proj(sprout::complex<T> const& x) {
|
|
return sprout::detail::proj_impl(
|
|
x,
|
|
sprout::norm(x) + T(1)
|
|
);
|
|
}
|
|
template<typename T>
|
|
inline SPROUT_CONSTEXPR sprout::complex<T>
|
|
polar(T const& rho, T const& theta = 0) {
|
|
return sprout::complex<T>(rho * sprout::cos(theta), rho * sprout::sin(theta));
|
|
}
|
|
} // namespace sprout
|
|
|
|
#endif // #ifndef SPROUT_COMPLEX_VALUES_HPP
|