zawz/ztd
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Added mod() and arg()

Browse source
This commit is contained in:
zawz 2019-08-27 17:00:09 +02:00
parent 5fda20a607
commit 829e95ad8d
2 changed files with 20 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
include/complex.hpp
View file

@ -3,6 +3,8 @@
#include <iostream> #include <iostream>
#include <math.h>
/*! @file complex.hpp /*! @file complex.hpp
* @brief Complex numbers and related computation * @brief Complex numbers and related computation
*/ */
@ -18,30 +20,36 @@ namespace ztd
//! @brief Contructor z //! @brief Contructor z
complex(const complex& z); complex(const complex& z);
inline complex& operator=(const complex& z) { real=z.real; im=z.im; return *this;}
//! @brief Complex Addition //! @brief Complex Addition
complex& operator+=(const complex& z); complex& operator+=(complex z);
//! @brief Complex Substraction //! @brief Complex Substraction
complex& operator-=(const complex& z); complex& operator-=(complex z);
//! @brief Complex Multiplication //! @brief Complex Multiplication
complex& operator*=(const complex& z); complex& operator*=(complex z);
//! @brief Complex Division //! @brief Complex Division
complex& operator/=(const complex& z); complex& operator/=(complex z);
//! @brief Real part //! @brief Real part
double real; double real;
//! @brief Imaginary part //! @brief Imaginary part
double im; double im;
inline double mod() { return sqrt(real*real + im*im); }
double arg() { return atan2(real, im); }
}; };
//! @brief Complex Addition //! @brief Complex Addition
inline complex operator+(const complex& z1, const complex& z2) { complex z3(z1); return z3 += z1; } inline complex operator+(complex z1, const complex& z2) { return z1 += z2; }
//! @brief Complex Substraction //! @brief Complex Substraction
inline complex operator-(const complex& z1, const complex& z2) { complex z3(z1); return z3 -= z1; } inline complex operator-(complex z1, const complex& z2) { return z1 -= z2; }
//! @brief Complex Multiplication //! @brief Complex Multiplication
inline complex operator*(const complex& z1, const complex& z2) { complex z3(z1); return z3 *= z1; } inline complex operator*(complex z1, const complex& z2) { return z1 *= z2; }
//! @brief Complex Division //! @brief Complex Division
inline complex operator/(const complex& z1, const complex& z2) { complex z3(z1); return z3 /= z1; } inline complex operator/(complex z1, const complex& z2) { return z1 /= z2; }
//! @brief Output a+bi //! @brief Output a+bi
inline std::ostream& operator<<(std::ostream& stream, complex z) { stream << z.real << '+' << z.im << 'i'; return stream; } inline std::ostream& operator<<(std::ostream& stream, complex z) { stream << z.real << '+' << z.im << 'i'; return stream; }

8
src/complex.cpp
View file

@ -13,26 +13,26 @@ ztd::complex::complex(const ztd::complex& z)
im=z.im; im=z.im;
} }
ztd::complex& ztd::complex::operator+=(const ztd::complex& z) ztd::complex& ztd::complex::operator+=(ztd::complex z)
{ {
real += z.real; real += z.real;
im += z.im; im += z.im;
return *this; return *this;
} }
ztd::complex& ztd::complex::operator-=(const ztd::complex& z) ztd::complex& ztd::complex::operator-=(ztd::complex z)
{ {
real -= z.real; real -= z.real;
im -= z.im; im -= z.im;
return *this; return *this;
} }
ztd::complex& ztd::complex::operator*=(const ztd::complex& z) ztd::complex& ztd::complex::operator*=(ztd::complex z)
{ {
double r=real; double r=real;
real = real*z.real - im*z.im; real = real*z.real - im*z.im;
im = r*z.im + im*z.real; im = r*z.im + im*z.real;
return *this; return *this;
} }
ztd::complex& ztd::complex::operator/=(const ztd::complex& z) ztd::complex& ztd::complex::operator/=(ztd::complex z)
{ {
double r=real; double r=real;
double div = z.real*z.real + z.im*z.im; double div = z.real*z.real + z.im*z.im;