/* Copyright 06/25/96 Sun Microsystems, Inc. All Rights Reserved */ /* Complex Number library Copyright 1991 by TauMetric Corporation ALL RIGHTS RESERVED @(#)complex.h 1.1 06/25/96 09:37:08 */ #ifndef _COMPLEX_H_ #define _COMPLEX_H_ #include #include class complex { public: complex(double _v, double _i=0.0); complex(); void operator+= (const complex); void operator+= (double); void operator-= (const complex); void operator-= (double); void operator*= (const complex); void operator*= (double); void operator/= (const complex); void operator/= (double); friend double real(const complex&); // real part friend double imag(const complex&); // imaginary part friend complex conj(const complex); // complex conjugate friend double norm(const complex); // square of the magnitude friend double arg(const complex); // angle in the complex plane // Create a complex object given polar coordinates friend complex polar(double _mag, double _arg=0.0); // Overloaded ANSI C math functions friend double abs(const complex); friend complex acos(const complex); friend complex asin(const complex); friend complex atan(const complex); friend complex cos(const complex); friend complex cosh(const complex); friend complex exp(const complex); friend complex log(const complex); friend complex log10(const complex); friend complex pow(double, const complex _exp); friend complex pow(const complex, int _exp); friend complex pow(const complex, double _exp); friend complex pow(const complex, const complex _exp); friend complex sin(const complex); friend complex sinh(const complex); friend complex sqrt(const complex); friend complex tan(const complex); friend complex tanh(const complex); // Unary Operator Functions friend complex operator- (const complex); // Binary Operator Functions friend complex operator+ (const complex, const complex); friend complex operator+ (double, const complex); friend complex operator+ (const complex, double); friend complex operator- (const complex, const complex); friend complex operator- (double, const complex); friend complex operator- (const complex, double); friend complex operator* (const complex, const complex); friend complex operator* (const complex, double); friend complex operator* (double, const complex); friend complex operator/ (const complex, const complex); friend complex operator/ (const complex, double); friend complex operator/ (double, const complex); friend int operator== (const complex, const complex); friend int operator!= (const complex, const complex); private: double re, im; }; inline complex::complex(double _v, double _i) : re(_v), im(_i) { } inline complex::complex() : re(0.0), im(0.0) { } inline void complex::operator+= (const complex _z) { re += _z.re; im += _z.im; } inline void complex::operator+= (double _v2) { re += _v2; } inline void complex::operator-= (const complex _z) { re -= _z.re; im -= _z.im; } inline void complex::operator-= (double _z) { re -= _z; } inline void complex::operator*= (double _v) { re *= _v; im *= _v; } inline double real(const complex& _z) { return _z.re; } inline double imag(const complex& _z) { return _z.im; } inline complex conj(const complex _z) { return complex(_z.re, -_z.im); } inline double norm(const complex _z) { return _z.re*_z.re + _z.im*_z.im; } inline double arg(const complex _z) { return atan2(_z.im, _z.re); } inline complex operator- (const complex _z) { return complex(-_z.re, -_z.im); } inline complex operator+ (const complex _z1, const complex _z2) { return complex(_z1.re + _z2.re, _z1.im + _z2.im); } inline complex operator+ (double _v1, const complex _z2) { return complex(_v1 + _z2.re, _z2.im); } inline complex operator+ (const complex _z1, double _v2) { return complex(_z1.re + _v2, _z1.im); } inline complex operator- (const complex _z1, const complex _z2) { return complex(_z1.re - _z2.re, _z1.im - _z2.im); } inline complex operator- (double _v1, const complex _z2) { return complex(_v1 - _z2.re, -_z2.im); } inline complex operator- (const complex _z1, double _v2) { return complex(_z1.re - _v2, _z1.im); } inline complex operator* (const complex _z1, double _v2) { return complex(_z1.re*_v2, _z1.im*_v2); } inline complex operator* (double _v1, const complex _z2) { return complex(_z2.re*_v1, _z2.im*_v1); } inline complex operator/ (const complex z1, double v2) { return complex(z1.re/v2, z1.im/v2); } inline int operator== (const complex _z1, const complex _z2) { return _z1.re == _z2.re && _z1.im == _z2.im; } inline int operator!= (const complex _z1, const complex _z2) { return _z1.re != _z2.re || _z1.im != _z2.im; } inline complex pow(const complex _b, int _exp) { return pow(_b, (double)_exp); } /* * Complex stream I/O */ #include ostream& operator<< (ostream&, const complex&); // user may override istream& operator>> (istream&, complex&); // user may override /* * Error Handling * * AT&T relies on its own errhand() function for reporting ordinary * (C library) math errors. This is not part of ANSI C, and is not * supported by the TauMetric C library. We rely instead on the signal * mechanism in the ANSI and TauMetric C libraries. */ static const complex complex_zero(0.0, 0.0); // types of error -- 0 means no error or unknown error #ifdef _STRICTANSI_ const int SING = 2; // argument singularity (e.g, div by 0) const int OVERFLOW = 3; // result overflowed const int UNDERFLOW = 4; // result underflowed #endif class c_exception { public: c_exception(char *n, const complex& a1, const complex& a2=complex_zero); c_exception(unsigned char *n, const complex& a1, const complex& a2=complex_zero); friend int complex_error(c_exception&); // user may override friend complex exp(const complex); friend complex log(const complex); friend complex log10(const complex); friend complex sinh(const complex); friend complex cosh(const complex); private: friend void do_error(c_exception&, int); friend void common_err(c_exception&, const double&, const double&); int type; char *name; complex arg1, arg2, retval; }; inline c_exception::c_exception(char *n, const complex& a1, const complex& a2) { type = 0; name = (char*)n; arg1 = a1; arg2 = a2; } inline c_exception::c_exception(unsigned char *n, const complex& a1, const complex& a2) { type = 0; name = (char*)n; arg1 = a1; arg2 = a2; } #endif /* _COMPLEX_H_ */