diff options
Diffstat (limited to 'src/include/cctk_Complex.h')
| -rw-r--r-- | src/include/cctk_Complex.h | 125 |
1 files changed, 106 insertions, 19 deletions
diff --git a/src/include/cctk_Complex.h b/src/include/cctk_Complex.h index 041884f3..977896d3 100644 --- a/src/include/cctk_Complex.h +++ b/src/include/cctk_Complex.h @@ -12,27 +12,92 @@ #define _CCTK_COMPLEX_H_ #ifdef __cplusplus -extern "C" -{ +# define EXTERN_C_BEGIN extern "C" { +# define EXTERN_C_END } +#else +# define EXTERN_C_BEGIN +# define EXTERN_C_END +#endif + + +#ifndef DEFINE_CCTK_COMPLEX_EXTERN_FUNCTIONS +# define PREFIX static inline +#else +# define PREFIX +#endif + + +#ifdef __cplusplus + /* declare C++ overloaded operators */ +# include <complex> +# include <ostream> +# define DECLARE_CMPLX_CXX_OPERATORS(CCTK_Cmplx, cctk_real, cctk_complex) \ +PREFIX cctk_complex operator+ (cctk_complex const & a); \ +PREFIX cctk_complex operator- (cctk_complex const & a); \ +PREFIX cctk_complex conj (cctk_complex const & a); \ +PREFIX cctk_real abs (cctk_complex const & a); \ +PREFIX cctk_real arg (cctk_complex const & a); \ +PREFIX cctk_real norm (cctk_complex const & a); \ +PREFIX cctk_complex operator+ (cctk_complex const & a, cctk_complex const & b); \ +PREFIX cctk_complex operator+ (cctk_real const & a, cctk_complex const & b); \ +PREFIX cctk_complex operator+ (cctk_complex const & a, cctk_real const & b); \ +PREFIX cctk_complex operator- (cctk_complex const & a, cctk_complex const & b); \ +PREFIX cctk_complex operator- (cctk_real const & a, cctk_complex const & b); \ +PREFIX cctk_complex operator- (cctk_complex const & a, cctk_real const & b); \ +PREFIX cctk_complex operator* (cctk_complex const & a, cctk_complex const & b); \ +PREFIX cctk_complex operator* (cctk_real const & a, cctk_complex const & b); \ +PREFIX cctk_complex operator* (cctk_complex const & a, cctk_real const & b); \ +PREFIX cctk_complex operator/ (cctk_complex const & a, cctk_complex const & b); \ +PREFIX cctk_complex operator/ (cctk_real const & a, cctk_complex const & b); \ +PREFIX cctk_complex operator/ (cctk_complex const & a, cctk_real const & b); \ +PREFIX cctk_complex operator+= (cctk_complex & a, cctk_complex const & b); \ +PREFIX cctk_complex operator+= (cctk_complex & a, cctk_real const & b); \ +PREFIX cctk_complex operator-= (cctk_complex & a, cctk_complex const & b); \ +PREFIX cctk_complex operator-= (cctk_complex & a, cctk_real const & b); \ +PREFIX cctk_complex operator*= (cctk_complex & a, cctk_complex const & b); \ +PREFIX cctk_complex operator*= (cctk_complex & a, cctk_real const & b); \ +PREFIX cctk_complex operator/= (cctk_complex & a, cctk_complex const & b); \ +PREFIX cctk_complex operator/= (cctk_complex & a, cctk_real const & b); \ +PREFIX cctk_complex pow (cctk_complex const & a, int i); \ +PREFIX cctk_complex pow (cctk_complex const & a, cctk_real const & r); \ +PREFIX cctk_complex pow (cctk_complex const & a, cctk_complex const & b); \ +PREFIX cctk_complex sin (cctk_complex const & a); \ +PREFIX cctk_complex cos (cctk_complex const & a); \ +PREFIX cctk_complex exp (cctk_complex const & a); \ +PREFIX cctk_complex log (cctk_complex const & a); \ +PREFIX cctk_complex sqrt (cctk_complex const & a); \ +PREFIX std::ostream & operator << (std::ostream & os, cctk_complex const & a); +#else + /* declare no C++ overloaded operators */ +# define DECLARE_CMPLX_CXX_OPERATORS(CCTK_Cmplx, cctk_real, cctk_complex) #endif /* macro to declare a set of complex functions for a given precision */ -#define DECLARE_CMPLX_FUNCTIONS(CCTK_Cmplx, cctk_real, cctk_complex) \ -cctk_complex CCTK_Cmplx (cctk_real Re, cctk_real Im); \ -cctk_real CCTK_Cmplx##Real (cctk_complex complex_number); \ -cctk_real CCTK_Cmplx##Imag (cctk_complex complex_number); \ -cctk_complex CCTK_Cmplx##Conjg (cctk_complex complex_number); \ -cctk_real CCTK_Cmplx##Abs (cctk_complex complex_number); \ -cctk_complex CCTK_Cmplx##Add (cctk_complex a, cctk_complex b); \ -cctk_complex CCTK_Cmplx##Sub (cctk_complex a, cctk_complex b); \ -cctk_complex CCTK_Cmplx##Mul (cctk_complex a, cctk_complex b); \ -cctk_complex CCTK_Cmplx##Div (cctk_complex a, cctk_complex b); \ -cctk_complex CCTK_Cmplx##Sin (cctk_complex complex_number); \ -cctk_complex CCTK_Cmplx##Cos (cctk_complex complex_number); \ -cctk_complex CCTK_Cmplx##Exp (cctk_complex complex_number); \ -cctk_complex CCTK_Cmplx##Sqrt (cctk_complex complex_number); \ -cctk_complex CCTK_Cmplx##Pow (cctk_complex complex_number, cctk_real w); +#define DECLARE_CMPLX_FUNCTIONS(CCTK_Cmplx, cctk_real, cctk_complex) \ +EXTERN_C_BEGIN \ +PREFIX cctk_complex CCTK_Cmplx (cctk_real Re, cctk_real Im); \ +PREFIX cctk_real CCTK_Cmplx##Real (cctk_complex a); \ +PREFIX cctk_real CCTK_Cmplx##Imag (cctk_complex a); \ +PREFIX cctk_complex CCTK_Cmplx##Neg (cctk_complex a); \ +PREFIX cctk_complex CCTK_Cmplx##Conjg (cctk_complex a); \ +PREFIX cctk_real CCTK_Cmplx##Abs (cctk_complex a); \ +PREFIX cctk_real CCTK_Cmplx##Arg (cctk_complex a); \ +PREFIX cctk_real CCTK_Cmplx##Norm (cctk_complex a); \ +PREFIX cctk_complex CCTK_Cmplx##Add (cctk_complex a, cctk_complex b); \ +PREFIX cctk_complex CCTK_Cmplx##Sub (cctk_complex a, cctk_complex b); \ +PREFIX cctk_complex CCTK_Cmplx##Mul (cctk_complex a, cctk_complex b); \ +PREFIX cctk_complex CCTK_Cmplx##Div (cctk_complex a, cctk_complex b); \ +PREFIX cctk_complex CCTK_Cmplx##CPow (cctk_complex a, cctk_complex b); \ +PREFIX cctk_complex CCTK_Cmplx##Sin (cctk_complex a); \ +PREFIX cctk_complex CCTK_Cmplx##Cos (cctk_complex a); \ +PREFIX cctk_complex CCTK_Cmplx##Exp (cctk_complex a); \ +PREFIX cctk_complex CCTK_Cmplx##Log (cctk_complex a); \ +PREFIX cctk_complex CCTK_Cmplx##Sqrt (cctk_complex a); \ +PREFIX cctk_complex CCTK_Cmplx##Pow (cctk_complex a, cctk_real b); \ +PREFIX cctk_complex CCTK_Cmplx##IPow (cctk_complex a, int b); \ +EXTERN_C_END \ +DECLARE_CMPLX_CXX_OPERATORS(CCTK_Cmplx, cctk_real, cctk_complex) /* declare complex functions for all available precisions */ @@ -48,57 +113,79 @@ DECLARE_CMPLX_FUNCTIONS (CCTK_Cmplx16, CCTK_REAL8, CCTK_COMPLEX16) DECLARE_CMPLX_FUNCTIONS (CCTK_Cmplx32, CCTK_REAL16, CCTK_COMPLEX32) #endif +#undef PREFIX + /* declare the default precision complex functions as #define'd macros */ #ifdef CCTK_REAL_PRECISION_4 #define CCTK_Cmplx CCTK_Cmplx8 #define CCTK_CmplxReal CCTK_Cmplx8Real #define CCTK_CmplxImag CCTK_Cmplx8Imag +#define CCTK_CmplxNeg CCTK_Cmplx8Neg #define CCTK_CmplxConjg CCTK_Cmplx8Conjg #define CCTK_CmplxAbs CCTK_Cmplx8Abs +#define CCTK_CmplxArg CCTK_Cmplx8Arg +#define CCTK_CmplxNorm CCTK_Cmplx8Norm #define CCTK_CmplxAdd CCTK_Cmplx8Add #define CCTK_CmplxSub CCTK_Cmplx8Sub #define CCTK_CmplxMul CCTK_Cmplx8Mul #define CCTK_CmplxDiv CCTK_Cmplx8Div +#define CCTK_CmplxCPow CCTK_Cmplx8CPow #define CCTK_CmplxSin CCTK_Cmplx8Sin #define CCTK_CmplxCos CCTK_Cmplx8Cos #define CCTK_CmplxExp CCTK_Cmplx8Exp +#define CCTK_CmplxLog CCTK_Cmplx8Log #define CCTK_CmplxSqrt CCTK_Cmplx8Sqrt #define CCTK_CmplxPow CCTK_Cmplx8Pow +#define CCTK_CmplxIPow CCTK_Cmplx8IPow #elif CCTK_REAL_PRECISION_8 #define CCTK_Cmplx CCTK_Cmplx16 #define CCTK_CmplxReal CCTK_Cmplx16Real #define CCTK_CmplxImag CCTK_Cmplx16Imag +#define CCTK_CmplxNeg CCTK_Cmplx16Neg #define CCTK_CmplxConjg CCTK_Cmplx16Conjg #define CCTK_CmplxAbs CCTK_Cmplx16Abs +#define CCTK_CmplxArg CCTK_Cmplx16Arg +#define CCTK_CmplxNorm CCTK_Cmplx16Norm #define CCTK_CmplxAdd CCTK_Cmplx16Add #define CCTK_CmplxSub CCTK_Cmplx16Sub #define CCTK_CmplxMul CCTK_Cmplx16Mul #define CCTK_CmplxDiv CCTK_Cmplx16Div +#define CCTK_CmplxCPow CCTK_Cmplx16CPow #define CCTK_CmplxSin CCTK_Cmplx16Sin #define CCTK_CmplxCos CCTK_Cmplx16Cos #define CCTK_CmplxExp CCTK_Cmplx16Exp +#define CCTK_CmplxLog CCTK_Cmplx16Log #define CCTK_CmplxSqrt CCTK_Cmplx16Sqrt #define CCTK_CmplxPow CCTK_Cmplx16Pow +#define CCTK_CmplxIPow CCTK_Cmplx16IPow #elif CCTK_REAL_PRECISION_16 #define CCTK_Cmplx CCTK_Cmplx32 #define CCTK_CmplxReal CCTK_Cmplx32Real #define CCTK_CmplxImag CCTK_Cmplx32Imag +#define CCTK_CmplxNeg CCTK_Cmplx32Neg #define CCTK_CmplxConjg CCTK_Cmplx32Conjg #define CCTK_CmplxAbs CCTK_Cmplx32Abs +#define CCTK_CmplxArg CCTK_Cmplx32Arg +#define CCTK_CmplxNorm CCTK_Cmplx32Norm #define CCTK_CmplxAdd CCTK_Cmplx32Add #define CCTK_CmplxSub CCTK_Cmplx32Sub #define CCTK_CmplxMul CCTK_Cmplx32Mul #define CCTK_CmplxDiv CCTK_Cmplx32Div +#define CCTK_CmplxCPow CCTK_Cmplx32CPow #define CCTK_CmplxSin CCTK_Cmplx32Sin #define CCTK_CmplxCos CCTK_Cmplx32Cos #define CCTK_CmplxExp CCTK_Cmplx32Exp +#define CCTK_CmplxLog CCTK_Cmplx32Log #define CCTK_CmplxSqrt CCTK_Cmplx32Sqrt #define CCTK_CmplxPow CCTK_Cmplx32Pow +#define CCTK_CmplxIPow CCTK_Cmplx32IPow #endif -#ifdef __cplusplus -} +#ifndef DEFINE_CCTK_COMPLEX_EXTERN_FUNCTIONS +# define DEFINE_CCTK_COMPLEX_INLINE_FUNCTIONS +# include "../main/Complex.c" +# undef DEFINE_CCTK_COMPLEX_INLINE_FUNCTIONS #endif #endif /* _CCTK_COMPLEX_H_ */ |