xref: /petsc/include/petscmath.h (revision 1a9c6b968e125f576ba8970d5d0335f59ca115b3)
1e489efc1SBarry Smith /*
2314da920SBarry Smith 
3314da920SBarry Smith       PETSc mathematics include file. Defines certain basic mathematical
4a5057860SBarry Smith     constants and functions for working with single, double, and quad precision
5a5057860SBarry Smith     floating point numbers as well as complex single and double.
6314da920SBarry Smith 
7d382aafbSBarry Smith     This file is included by petscsys.h and should not be used directly.
8e7029fe1SSatish Balay 
9e489efc1SBarry Smith */
10e489efc1SBarry Smith 
11488ecbafSBarry Smith #if !defined(__PETSCMATH_H)
12488ecbafSBarry Smith #define __PETSCMATH_H
130a5f7794SBarry Smith #include <math.h>
140a5f7794SBarry Smith 
15314da920SBarry Smith /*
16f4ccad53SBarry Smith 
17f4ccad53SBarry Smith      Defines operations that are different for complex and real numbers;
18a5057860SBarry Smith    note that one cannot mix the use of complex and real in the same
19f4ccad53SBarry Smith    PETSc program. All PETSc objects in one program are built around the object
2098725619SBarry Smith    PetscScalar which is either always a real or a complex.
21f4ccad53SBarry Smith 
22e489efc1SBarry Smith */
23b36a9721SBarry Smith 
24c1d390e3SJed Brown #if defined(PETSC_USE_REAL_SINGLE)
25c1d390e3SJed Brown #define MPIU_REAL   MPI_FLOAT
26c1d390e3SJed Brown typedef float PetscReal;
279cf33046SSatish Balay #define PetscSqrtReal(a)    sqrt(a)
289a07f4dfSJed Brown #define PetscExpReal(a)     exp(a)
299a07f4dfSJed Brown #define PetscLogReal(a)     log(a)
3077b4d14cSPeter Brune #define PetscLog10Real(a)   log10(a)
3178a59e97SMatthew G. Knepley #ifdef PETSC_HAVE_LOG2
3278a59e97SMatthew G. Knepley #define PetscLog2Real(a)    log2(a)
3378a59e97SMatthew G. Knepley #endif
349a07f4dfSJed Brown #define PetscSinReal(a)     sin(a)
359a07f4dfSJed Brown #define PetscCosReal(a)     cos(a)
36a4bea5a6SPeter Brune #define PetscTanReal(a)     tan(a)
37255453a1SBarry Smith #define PetscAsinReal(a)    asin(a)
38255453a1SBarry Smith #define PetscAcosReal(a)    acos(a)
39a4bea5a6SPeter Brune #define PetscAtanReal(a)    atan(a)
401d5abddaSDmitry Karpeev #define PetscAtan2Real(a,b) atan2(a,b)
41a4bea5a6SPeter Brune #define PetscSinhReal(a)    sinh(a)
42a4bea5a6SPeter Brune #define PetscCoshReal(a)    cosh(a)
43a4bea5a6SPeter Brune #define PetscTanhReal(a)    tanh(a)
44369cc0aeSBarry Smith #define PetscPowReal(a,b)   pow(a,b)
4577b4d14cSPeter Brune #define PetscCeilReal(a)    ceil(a)
4677b4d14cSPeter Brune #define PetscFloorReal(a)   floor(a)
47a4bea5a6SPeter Brune #define PetscFmodReal(a,b)  fmod(a,b)
4843f0d3baSJed Brown #define PetscTGamma(a)      tgammaf(a)
49c1d390e3SJed Brown #elif defined(PETSC_USE_REAL_DOUBLE)
50c1d390e3SJed Brown #define MPIU_REAL   MPI_DOUBLE
51c1d390e3SJed Brown typedef double PetscReal;
528f1a2a5eSBarry Smith #define PetscSqrtReal(a)    sqrt(a)
539a07f4dfSJed Brown #define PetscExpReal(a)     exp(a)
549a07f4dfSJed Brown #define PetscLogReal(a)     log(a)
5577b4d14cSPeter Brune #define PetscLog10Real(a)   log10(a)
5678a59e97SMatthew G. Knepley #ifdef PETSC_HAVE_LOG2
5778a59e97SMatthew G. Knepley #define PetscLog2Real(a)    log2(a)
5878a59e97SMatthew G. Knepley #endif
599a07f4dfSJed Brown #define PetscSinReal(a)     sin(a)
609a07f4dfSJed Brown #define PetscCosReal(a)     cos(a)
61a4bea5a6SPeter Brune #define PetscTanReal(a)     tan(a)
62255453a1SBarry Smith #define PetscAsinReal(a)    asin(a)
63255453a1SBarry Smith #define PetscAcosReal(a)    acos(a)
64a4bea5a6SPeter Brune #define PetscAtanReal(a)    atan(a)
651d5abddaSDmitry Karpeev #define PetscAtan2Real(a,b) atan2(a,b)
668d1b9be0SJed Brown #define PetscSinhReal(a)    sinh(a)
678d1b9be0SJed Brown #define PetscCoshReal(a)    cosh(a)
688d1b9be0SJed Brown #define PetscTanhReal(a)    tanh(a)
69369cc0aeSBarry Smith #define PetscPowReal(a,b)   pow(a,b)
7077b4d14cSPeter Brune #define PetscCeilReal(a)    ceil(a)
7177b4d14cSPeter Brune #define PetscFloorReal(a)   floor(a)
72a4bea5a6SPeter Brune #define PetscFmodReal(a,b)  fmod(a,b)
730646a658SBarry Smith #define PetscTGamma(a)      tgamma(a)
74c1d390e3SJed Brown #elif defined(PETSC_USE_REAL___FLOAT128)
75574fde7bSSatish Balay #if defined(__cplusplus)
76574fde7bSSatish Balay extern "C" {
77574fde7bSSatish Balay #endif
78574fde7bSSatish Balay #include <quadmath.h>
79574fde7bSSatish Balay #if defined(__cplusplus)
80574fde7bSSatish Balay }
81574fde7bSSatish Balay #endif
828ad47952SJed Brown PETSC_EXTERN MPI_Datatype MPIU___FLOAT128 PetscAttrMPITypeTag(__float128);
83c1d390e3SJed Brown #define MPIU_REAL MPIU___FLOAT128
84c1d390e3SJed Brown typedef __float128 PetscReal;
858f1a2a5eSBarry Smith #define PetscSqrtReal(a)    sqrtq(a)
869a07f4dfSJed Brown #define PetscExpReal(a)     expq(a)
879a07f4dfSJed Brown #define PetscLogReal(a)     logq(a)
8877b4d14cSPeter Brune #define PetscLog10Real(a)   log10q(a)
8978a59e97SMatthew G. Knepley #ifdef PETSC_HAVE_LOG2
9078a59e97SMatthew G. Knepley #define PetscLog2Real(a)    log2q(a)
9178a59e97SMatthew G. Knepley #endif
929a07f4dfSJed Brown #define PetscSinReal(a)     sinq(a)
939a07f4dfSJed Brown #define PetscCosReal(a)     cosq(a)
94a4bea5a6SPeter Brune #define PetscTanReal(a)     tanq(a)
95255453a1SBarry Smith #define PetscAsinReal(a)    asinq(a)
96255453a1SBarry Smith #define PetscAcosReal(a)    acosq(a)
97a4bea5a6SPeter Brune #define PetscAtanReal(a)    atanq(a)
981d5abddaSDmitry Karpeev #define PetscAtan2Real(a,b) atan2q(a,b)
998d1b9be0SJed Brown #define PetscSinhReal(a)    sinhq(a)
1008d1b9be0SJed Brown #define PetscCoshReal(a)    coshq(a)
1018d1b9be0SJed Brown #define PetscTanhReal(a)    tanhq(a)
102369cc0aeSBarry Smith #define PetscPowReal(a,b)   powq(a,b)
10377b4d14cSPeter Brune #define PetscCeilReal(a)    ceilq(a)
10477b4d14cSPeter Brune #define PetscFloorReal(a)   floorq(a)
105a4bea5a6SPeter Brune #define PetscFmodReal(a,b)  fmodq(a,b)
1060646a658SBarry Smith #define PetscTGamma(a)      tgammaq(a)
107c1d390e3SJed Brown #endif /* PETSC_USE_REAL_* */
10859cb5930SBarry Smith 
1091093a601SBarry Smith /*
1101093a601SBarry Smith     Complex number definitions
1111093a601SBarry Smith  */
1122f217381SBarry Smith #if defined(__cplusplus) && defined(PETSC_HAVE_CXX_COMPLEX) && !defined(PETSC_USE_REAL___FLOAT128)
1138cd53115SBarry Smith #if !defined(PETSC_SKIP_COMPLEX)
11450f81f78SJed Brown #define PETSC_HAVE_COMPLEX 1
1151093a601SBarry Smith /* C++ support of complex number */
116debe9ee2SPaul Mullowney #if defined(PETSC_HAVE_CUSP)
117debe9ee2SPaul Mullowney #define complexlib cusp
1189ae82921SPaul Mullowney #include <cusp/complex.h>
119fda01a78SKarl Rupp #elif defined(PETSC_HAVE_VECCUDA) && __CUDACC_VER_MAJOR__ > 6
120fda01a78SKarl Rupp /* complex headers in thrust only available in CUDA 7.0 and above */
121ec42abe4SAlejandro Lamas Daviña #define complexlib thrust
122ec42abe4SAlejandro Lamas Daviña #include <thrust/complex.h>
123debe9ee2SPaul Mullowney #else
124debe9ee2SPaul Mullowney #define complexlib std
125debe9ee2SPaul Mullowney #include <complex>
1269ae82921SPaul Mullowney #endif
127b7940d39SSatish Balay 
12850f81f78SJed Brown #define PetscRealPartComplex(a)      (a).real()
12950f81f78SJed Brown #define PetscImaginaryPartComplex(a) (a).imag()
13050f81f78SJed Brown #define PetscAbsComplex(a)           complexlib::abs(a)
13150f81f78SJed Brown #define PetscConjComplex(a)          complexlib::conj(a)
13250f81f78SJed Brown #define PetscSqrtComplex(a)          complexlib::sqrt(a)
13350f81f78SJed Brown #define PetscPowComplex(a,b)         complexlib::pow(a,b)
13450f81f78SJed Brown #define PetscExpComplex(a)           complexlib::exp(a)
13550f81f78SJed Brown #define PetscLogComplex(a)           complexlib::log(a)
13650f81f78SJed Brown #define PetscSinComplex(a)           complexlib::sin(a)
13750f81f78SJed Brown #define PetscCosComplex(a)           complexlib::cos(a)
138255453a1SBarry Smith #define PetscAsinComplex(a)          complexlib::asin(a)
139255453a1SBarry Smith #define PetscAcosComplex(a)          complexlib::acos(a)
140027d9794SBarry Smith #if defined(PETSC_HAVE_TANCOMPLEX)
141a4bea5a6SPeter Brune #define PetscTanComplex(a)           complexlib::tan(a)
142027d9794SBarry Smith #else
143027d9794SBarry Smith #define PetscTanComplex(a)           PetscSinComplex(a)/PetscCosComplex(a)
144027d9794SBarry Smith #endif
145a4bea5a6SPeter Brune #define PetscSinhComplex(a)          complexlib::sinh(a)
146a4bea5a6SPeter Brune #define PetscCoshComplex(a)          complexlib::cosh(a)
147027d9794SBarry Smith #if defined(PETSC_HAVE_TANHCOMPLEX)
148a4bea5a6SPeter Brune #define PetscTanhComplex(a)          complexlib::tanh(a)
149027d9794SBarry Smith #else
150027d9794SBarry Smith #define PetscTanhComplex(a)          PetscSinhComplex(a)/PetscCoshComplex(a)
151027d9794SBarry Smith #endif
152debe9ee2SPaul Mullowney 
153debe9ee2SPaul Mullowney #if defined(PETSC_USE_REAL_SINGLE)
15450f81f78SJed Brown typedef complexlib::complex<float> PetscComplex;
1553be776deSLisandro Dalcin #if defined(PETSC_USE_CXX_COMPLEX_FLOAT_WORKAROUND)
1563be776deSLisandro Dalcin static inline PetscComplex operator+(const PetscComplex& lhs, const double& rhs) { return lhs + float(rhs); }
1573be776deSLisandro Dalcin static inline PetscComplex operator+(const double& lhs, const PetscComplex& rhs) { return float(lhs) + rhs; }
1583be776deSLisandro Dalcin static inline PetscComplex operator-(const PetscComplex& lhs, const double& rhs) { return lhs - float(rhs); }
1593be776deSLisandro Dalcin static inline PetscComplex operator-(const double& lhs, const PetscComplex& rhs) { return float(lhs) - rhs; }
1603be776deSLisandro Dalcin static inline PetscComplex operator*(const PetscComplex& lhs, const double& rhs) { return lhs * float(rhs); }
1613be776deSLisandro Dalcin static inline PetscComplex operator*(const double& lhs, const PetscComplex& rhs) { return float(lhs) * rhs; }
1623be776deSLisandro Dalcin static inline PetscComplex operator/(const PetscComplex& lhs, const double& rhs) { return lhs / float(rhs); }
1633be776deSLisandro Dalcin static inline PetscComplex operator/(const double& lhs, const PetscComplex& rhs) { return float(lhs) / rhs; }
1643be776deSLisandro Dalcin static inline bool operator==(const PetscComplex& lhs, const double& rhs) { return lhs.imag() == float(0) && lhs.real() == float(rhs); }
1653be776deSLisandro Dalcin static inline bool operator==(const double& lhs, const PetscComplex& rhs) { return rhs.imag() == float(0) && rhs.real() == float(lhs); }
1663be776deSLisandro Dalcin static inline bool operator!=(const PetscComplex& lhs, const double& rhs) { return lhs.imag() != float(0) || lhs.real() != float(rhs); }
1673be776deSLisandro Dalcin static inline bool operator!=(const double& lhs, const PetscComplex& rhs) { return rhs.imag() != float(0) || rhs.real() != float(lhs); }
1683be776deSLisandro Dalcin #endif  /* PETSC_USE_CXX_COMPLEX_FLOAT_WORKAROUND */
169debe9ee2SPaul Mullowney #elif defined(PETSC_USE_REAL_DOUBLE)
17050f81f78SJed Brown typedef complexlib::complex<double> PetscComplex;
171*1a9c6b96SLisandro Dalcin #if defined(PETSC_USE_CXX_COMPLEX_FLOAT_WORKAROUND)
172*1a9c6b96SLisandro Dalcin static inline PetscComplex operator+(const PetscComplex& lhs, const PetscInt& rhs) { return lhs + double(rhs); }
173*1a9c6b96SLisandro Dalcin static inline PetscComplex operator+(const PetscInt& lhs, const PetscComplex& rhs) { return double(lhs) + rhs; }
174*1a9c6b96SLisandro Dalcin static inline PetscComplex operator-(const PetscComplex& lhs, const PetscInt& rhs) { return lhs - double(rhs); }
175*1a9c6b96SLisandro Dalcin static inline PetscComplex operator-(const PetscInt& lhs, const PetscComplex& rhs) { return double(lhs) - rhs; }
176*1a9c6b96SLisandro Dalcin static inline PetscComplex operator*(const PetscComplex& lhs, const PetscInt& rhs) { return lhs * double(rhs); }
177*1a9c6b96SLisandro Dalcin static inline PetscComplex operator*(const PetscInt& lhs, const PetscComplex& rhs) { return double(lhs) * rhs; }
178*1a9c6b96SLisandro Dalcin static inline PetscComplex operator/(const PetscComplex& lhs, const PetscInt& rhs) { return lhs / double(rhs); }
179*1a9c6b96SLisandro Dalcin static inline PetscComplex operator/(const PetscInt& lhs, const PetscComplex& rhs) { return double(lhs) / rhs; }
180*1a9c6b96SLisandro Dalcin static inline bool operator==(const PetscComplex& lhs, const PetscInt& rhs) { return lhs.imag() == double(0) && lhs.real() == double(rhs); }
181*1a9c6b96SLisandro Dalcin static inline bool operator==(const PetscInt& lhs, const PetscComplex& rhs) { return rhs.imag() == double(0) && rhs.real() == double(lhs); }
182*1a9c6b96SLisandro Dalcin static inline bool operator!=(const PetscComplex& lhs, const PetscInt& rhs) { return lhs.imag() != double(0) || lhs.real() != double(rhs); }
183*1a9c6b96SLisandro Dalcin static inline bool operator!=(const PetscInt& lhs, const PetscComplex& rhs) { return rhs.imag() != double(0) || rhs.real() != double(lhs); }
184*1a9c6b96SLisandro Dalcin #endif  /* PETSC_USE_CXX_COMPLEX_FLOAT_WORKAROUND */
1858c764dc5SJose Roman #elif defined(PETSC_USE_REAL___FLOAT128)
18650f81f78SJed Brown typedef complexlib::complex<__float128> PetscComplex; /* Notstandard and not expected to work, use __complex128 */
18722b3908eSBarry Smith PETSC_EXTERN MPI_Datatype MPIU___COMPLEX128;
188debe9ee2SPaul Mullowney #endif  /* PETSC_USE_REAL_ */
1898cd53115SBarry Smith #endif  /* ! PETSC_SKIP_COMPLEX */
190debe9ee2SPaul Mullowney 
1912f217381SBarry Smith #elif !defined(__cplusplus) && defined(PETSC_HAVE_C99_COMPLEX)
1928cd53115SBarry Smith #if !defined(PETSC_SKIP_COMPLEX)
1939f20b660SSatish Balay #define PETSC_HAVE_COMPLEX 1
194519e2a1fSPaul Mullowney #include <complex.h>
195519e2a1fSPaul Mullowney 
196ce63c4c1SBarry Smith #if defined(PETSC_USE_REAL_SINGLE)
19750f81f78SJed Brown typedef float _Complex PetscComplex;
19885b47369SMatthew Knepley 
19950f81f78SJed Brown #define PetscRealPartComplex(a)      crealf(a)
20050f81f78SJed Brown #define PetscImaginaryPartComplex(a) cimagf(a)
20150f81f78SJed Brown #define PetscAbsComplex(a)           cabsf(a)
20250f81f78SJed Brown #define PetscConjComplex(a)          conjf(a)
20350f81f78SJed Brown #define PetscSqrtComplex(a)          csqrtf(a)
20450f81f78SJed Brown #define PetscPowComplex(a,b)         cpowf(a,b)
20550f81f78SJed Brown #define PetscExpComplex(a)           cexpf(a)
20650f81f78SJed Brown #define PetscLogComplex(a)           clogf(a)
20750f81f78SJed Brown #define PetscSinComplex(a)           csinf(a)
20850f81f78SJed Brown #define PetscCosComplex(a)           ccosf(a)
209255453a1SBarry Smith #define PetscAsinComplex(a)          casinf(a)
210255453a1SBarry Smith #define PetscAcosComplex(a)          cacosf(a)
211a4bea5a6SPeter Brune #define PetscTanComplex(a)           ctanf(a)
212a4bea5a6SPeter Brune #define PetscSinhComplex(a)          csinhf(a)
213a4bea5a6SPeter Brune #define PetscCoshComplex(a)          ccoshf(a)
214a4bea5a6SPeter Brune #define PetscTanhComplex(a)          ctanhf(a)
2151093a601SBarry Smith 
216ce63c4c1SBarry Smith #elif defined(PETSC_USE_REAL_DOUBLE)
21750f81f78SJed Brown typedef double _Complex PetscComplex;
2181093a601SBarry Smith 
21950f81f78SJed Brown #define PetscRealPartComplex(a)      creal(a)
22050f81f78SJed Brown #define PetscImaginaryPartComplex(a) cimag(a)
22150f81f78SJed Brown #define PetscAbsComplex(a)           cabs(a)
22250f81f78SJed Brown #define PetscConjComplex(a)          conj(a)
22350f81f78SJed Brown #define PetscSqrtComplex(a)          csqrt(a)
22450f81f78SJed Brown #define PetscPowComplex(a,b)         cpow(a,b)
22550f81f78SJed Brown #define PetscExpComplex(a)           cexp(a)
22650f81f78SJed Brown #define PetscLogComplex(a)           clog(a)
22750f81f78SJed Brown #define PetscSinComplex(a)           csin(a)
22850f81f78SJed Brown #define PetscCosComplex(a)           ccos(a)
229255453a1SBarry Smith #define PetscAsinComplex(a)          casin(a)
230255453a1SBarry Smith #define PetscAcosComplex(a)          cacos(a)
231a4bea5a6SPeter Brune #define PetscTanComplex(a)           ctan(a)
232a4bea5a6SPeter Brune #define PetscSinhComplex(a)          csinh(a)
233a4bea5a6SPeter Brune #define PetscCoshComplex(a)          ccosh(a)
234a4bea5a6SPeter Brune #define PetscTanhComplex(a)          ctanh(a)
2351093a601SBarry Smith 
2368c764dc5SJose Roman #elif defined(PETSC_USE_REAL___FLOAT128)
23750f81f78SJed Brown typedef __complex128 PetscComplex;
2388ad47952SJed Brown PETSC_EXTERN MPI_Datatype MPIU___COMPLEX128 PetscAttrMPITypeTag(__complex128);
2398c764dc5SJose Roman 
24050f81f78SJed Brown #define PetscRealPartComplex(a)      crealq(a)
24150f81f78SJed Brown #define PetscImaginaryPartComplex(a) cimagq(a)
24250f81f78SJed Brown #define PetscAbsComplex(a)           cabsq(a)
24350f81f78SJed Brown #define PetscConjComplex(a)          conjq(a)
24450f81f78SJed Brown #define PetscSqrtComplex(a)          csqrtq(a)
24550f81f78SJed Brown #define PetscPowComplex(a,b)         cpowq(a,b)
24650f81f78SJed Brown #define PetscExpComplex(a)           cexpq(a)
24750f81f78SJed Brown #define PetscLogComplex(a)           clogq(a)
24850f81f78SJed Brown #define PetscSinComplex(a)           csinq(a)
24950f81f78SJed Brown #define PetscCosComplex(a)           ccosq(a)
250255453a1SBarry Smith #define PetscAsinComplex(a)          casinq(a)
251255453a1SBarry Smith #define PetscAcosComplex(a)          cacosq(a)
252a4bea5a6SPeter Brune #define PetscTanComplex(a)           ctanq(a)
253a4bea5a6SPeter Brune #define PetscSinhComplex(a)          csinhq(a)
254a4bea5a6SPeter Brune #define PetscCoshComplex(a)          ccoshq(a)
255a4bea5a6SPeter Brune #define PetscTanhComplex(a)          ctanhq(a)
256a4bea5a6SPeter Brune 
257ce63c4c1SBarry Smith #endif /* PETSC_USE_REAL_* */
258d4161b4aSBarry Smith #elif (defined(PETSC_USE_COMPLEX) && !defined(PETSC_SKIP_COMPLEX))
25950f81f78SJed Brown #error "PETSc was configured --with-scalar-type=complex, but a language-appropriate complex library is not available"
2608cd53115SBarry Smith #endif /* !PETSC_SKIP_COMPLEX */
2612f217381SBarry Smith #endif /* (__cplusplus && PETSC_HAVE_CXX_COMPLEX) else-if (!__cplusplus && PETSC_HAVE_C99_COMPLEX) */
262e489efc1SBarry Smith 
2638dc6f2c2SJed Brown #if defined(PETSC_HAVE_COMPLEX)
26470da9c3bSJed Brown #if defined(PETSC_HAVE_MPI_C_DOUBLE_COMPLEX)
265500d8756SSatish Balay #define MPIU_C_DOUBLE_COMPLEX MPI_C_DOUBLE_COMPLEX
266500d8756SSatish Balay #define MPIU_C_COMPLEX MPI_C_COMPLEX
26770da9c3bSJed Brown #else
2682f217381SBarry Smith # if defined(__cplusplus) && defined(PETSC_HAVE_CXX_COMPLEX)
2698ad47952SJed Brown   typedef complexlib::complex<double> petsc_mpiu_c_double_complex;
2708ad47952SJed Brown   typedef complexlib::complex<float> petsc_mpiu_c_complex;
2712f217381SBarry Smith # elif !defined(__cplusplus) && defined(PETSC_HAVE_C99_COMPLEX)
2728ad47952SJed Brown   typedef double _Complex petsc_mpiu_c_double_complex;
2738ad47952SJed Brown   typedef float _Complex petsc_mpiu_c_complex;
2748ad47952SJed Brown # else
2758ad47952SJed Brown   typedef struct {double real,imag;} petsc_mpiu_c_double_complex;
2768ad47952SJed Brown   typedef struct {float real,imag;} petsc_mpiu_c_complex;
2778ad47952SJed Brown # endif
2788ad47952SJed Brown PETSC_EXTERN MPI_Datatype MPIU_C_DOUBLE_COMPLEX PetscAttrMPITypeTagLayoutCompatible(petsc_mpiu_c_double_complex);
2798ad47952SJed Brown PETSC_EXTERN MPI_Datatype MPIU_C_COMPLEX PetscAttrMPITypeTagLayoutCompatible(petsc_mpiu_c_complex);
2801b65fc54SMatthew G Knepley #endif /* PETSC_HAVE_MPI_C_DOUBLE_COMPLEX */
2818dc6f2c2SJed Brown #endif /* PETSC_HAVE_COMPLEX */
2822c876bd9SBarry Smith 
2837c2de775SJed Brown #if defined(PETSC_HAVE_COMPLEX)
2847c2de775SJed Brown #  if defined(PETSC_USE_REAL_SINGLE)
2857c2de775SJed Brown #    define MPIU_COMPLEX MPIU_C_COMPLEX
2867c2de775SJed Brown #  elif defined(PETSC_USE_REAL_DOUBLE)
2877c2de775SJed Brown #    define MPIU_COMPLEX MPIU_C_DOUBLE_COMPLEX
2887c2de775SJed Brown #  elif defined(PETSC_USE_REAL___FLOAT128)
2897c2de775SJed Brown #    define MPIU_COMPLEX MPIU___COMPLEX128
2907c2de775SJed Brown #  endif /* PETSC_USE_REAL_* */
2917c2de775SJed Brown #endif
2927c2de775SJed Brown 
293d4161b4aSBarry Smith #if (defined(PETSC_USE_COMPLEX) && !defined(PETSC_SKIP_COMPLEX))
29450f81f78SJed Brown typedef PetscComplex PetscScalar;
29550f81f78SJed Brown #define PetscRealPart(a)      PetscRealPartComplex(a)
29650f81f78SJed Brown #define PetscImaginaryPart(a) PetscImaginaryPartComplex(a)
29750f81f78SJed Brown #define PetscAbsScalar(a)     PetscAbsComplex(a)
29850f81f78SJed Brown #define PetscConj(a)          PetscConjComplex(a)
29950f81f78SJed Brown #define PetscSqrtScalar(a)    PetscSqrtComplex(a)
30050f81f78SJed Brown #define PetscPowScalar(a,b)   PetscPowComplex(a,b)
30150f81f78SJed Brown #define PetscExpScalar(a)     PetscExpComplex(a)
30250f81f78SJed Brown #define PetscLogScalar(a)     PetscLogComplex(a)
30350f81f78SJed Brown #define PetscSinScalar(a)     PetscSinComplex(a)
30450f81f78SJed Brown #define PetscCosScalar(a)     PetscCosComplex(a)
305255453a1SBarry Smith #define PetscAsinScalar(a)    PetscAsinComplex(a)
306255453a1SBarry Smith #define PetscAcosScalar(a)    PetscAcosComplex(a)
307a4bea5a6SPeter Brune #define PetscTanScalar(a)     PetscTanComplex(a)
308a4bea5a6SPeter Brune #define PetscSinhScalar(a)    PetscSinhComplex(a)
309a4bea5a6SPeter Brune #define PetscCoshScalar(a)    PetscCoshComplex(a)
310a4bea5a6SPeter Brune #define PetscTanhScalar(a)    PetscTanhComplex(a)
3117c2de775SJed Brown #define MPIU_SCALAR MPIU_COMPLEX
31275567043SBarry Smith 
3131093a601SBarry Smith /*
3141093a601SBarry Smith     real number definitions
3151093a601SBarry Smith  */
3161b65fc54SMatthew G Knepley #else /* PETSC_USE_COMPLEX */
3177c2de775SJed Brown typedef PetscReal PetscScalar;
3187c2de775SJed Brown #define MPIU_SCALAR MPIU_REAL
3197c2de775SJed Brown 
320329f5518SBarry Smith #define PetscRealPart(a)      (a)
321c1d390e3SJed Brown #define PetscImaginaryPart(a) ((PetscReal)0.)
322c1d390e3SJed Brown PETSC_STATIC_INLINE PetscReal PetscAbsScalar(PetscScalar a) {return a < 0.0 ? -a : a;}
323e489efc1SBarry Smith #define PetscConj(a)          (a)
324ce63c4c1SBarry Smith #if !defined(PETSC_USE_REAL___FLOAT128)
32518a7d68fSSatish Balay #define PetscSqrtScalar(a)    sqrt(a)
326184914b5SBarry Smith #define PetscPowScalar(a,b)   pow(a,b)
327184914b5SBarry Smith #define PetscExpScalar(a)     exp(a)
32806c1185fSBarry Smith #define PetscLogScalar(a)     log(a)
329184914b5SBarry Smith #define PetscSinScalar(a)     sin(a)
330184914b5SBarry Smith #define PetscCosScalar(a)     cos(a)
331255453a1SBarry Smith #define PetscAsinScalar(a)    asin(a)
332255453a1SBarry Smith #define PetscAcosScalar(a)    acos(a)
333a4bea5a6SPeter Brune #define PetscTanScalar(a)     tan(a)
334a4bea5a6SPeter Brune #define PetscSinhScalar(a)    sinh(a)
335a4bea5a6SPeter Brune #define PetscCoshScalar(a)    cosh(a)
336a4bea5a6SPeter Brune #define PetscTanhScalar(a)    tanh(a)
337ce63c4c1SBarry Smith #else /* PETSC_USE_REAL___FLOAT128 */
3380d0cc1b5SBarry Smith #define PetscSqrtScalar(a)    sqrtq(a)
3390d0cc1b5SBarry Smith #define PetscPowScalar(a,b)   powq(a,b)
3400d0cc1b5SBarry Smith #define PetscExpScalar(a)     expq(a)
3410d0cc1b5SBarry Smith #define PetscLogScalar(a)     logq(a)
3420d0cc1b5SBarry Smith #define PetscSinScalar(a)     sinq(a)
3430d0cc1b5SBarry Smith #define PetscCosScalar(a)     cosq(a)
344255453a1SBarry Smith #define PetscAsinScalar(a)    asinq(a)
345255453a1SBarry Smith #define PetscAcosScalar(a)    acosq(a)
346a4bea5a6SPeter Brune #define PetscTanScalar(a)     tanq(a)
347a4bea5a6SPeter Brune #define PetscSinhScalar(a)    sinhq(a)
348a4bea5a6SPeter Brune #define PetscCoshScalar(a)    coshq(a)
349a4bea5a6SPeter Brune #define PetscTanhScalar(a)    tanhq(a)
350ce63c4c1SBarry Smith #endif /* PETSC_USE_REAL___FLOAT128 */
351b0a32e0cSBarry Smith 
3521b65fc54SMatthew G Knepley #endif /* PETSC_USE_COMPLEX */
353e489efc1SBarry Smith 
354da9b6338SBarry Smith #define PetscSign(a) (((a) >= 0) ? ((a) == 0 ? 0 : 1) : -1)
35536850ab7SBarry Smith #define PetscSignReal(a) (((a) >= 0.0) ? ((a) == 0.0 ? 0.0 : 1.0) : -1.0)
35626aa1773SMatthew Knepley #define PetscAbs(a)  (((a) >= 0) ? (a) : -(a))
3573f1db9ecSBarry Smith 
358314da920SBarry Smith /* --------------------------------------------------------------------------*/
359314da920SBarry Smith 
360e489efc1SBarry Smith /*
361f22f69f0SBarry Smith    Certain objects may be created using either single or double precision.
362f22f69f0SBarry Smith    This is currently not used.
363e489efc1SBarry Smith */
364557d4da8SBarry Smith typedef enum { PETSC_SCALAR_DOUBLE,PETSC_SCALAR_SINGLE, PETSC_SCALAR_LONG_DOUBLE } PetscScalarPrecision;
365e489efc1SBarry Smith 
36650f81f78SJed Brown #if defined(PETSC_HAVE_COMPLEX)
367e489efc1SBarry Smith /* PETSC_i is the imaginary number, i */
36850f81f78SJed Brown PETSC_EXTERN PetscComplex PETSC_i;
36950f81f78SJed Brown #endif
370e489efc1SBarry Smith 
371b6a5bde7SBarry Smith /*MC
372b6a5bde7SBarry Smith    PetscMin - Returns minimum of two numbers
373b6a5bde7SBarry Smith 
374eca87e8dSBarry Smith    Synopsis:
375aaa7dc30SBarry Smith    #include <petscmath.h>
376eca87e8dSBarry Smith    type PetscMin(type v1,type v2)
377eca87e8dSBarry Smith 
378eca87e8dSBarry Smith    Not Collective
379eca87e8dSBarry Smith 
380b6a5bde7SBarry Smith    Input Parameter:
381b6a5bde7SBarry Smith +  v1 - first value to find minimum of
382b6a5bde7SBarry Smith -  v2 - second value to find minimum of
383b6a5bde7SBarry Smith 
384b6a5bde7SBarry Smith    Notes: type can be integer or floating point value
385b6a5bde7SBarry Smith 
386b6a5bde7SBarry Smith    Level: beginner
387b6a5bde7SBarry Smith 
388d9a4bb16SJed Brown .seealso: PetscMin(), PetscClipInterval(), PetscAbsInt(), PetscAbsReal(), PetscSqr()
389b6a5bde7SBarry Smith 
390b6a5bde7SBarry Smith M*/
391e489efc1SBarry Smith #define PetscMin(a,b)   (((a)<(b)) ?  (a) : (b))
392b6a5bde7SBarry Smith 
393b6a5bde7SBarry Smith /*MC
394b6a5bde7SBarry Smith    PetscMax - Returns maxium of two numbers
395b6a5bde7SBarry Smith 
396eca87e8dSBarry Smith    Synopsis:
397aaa7dc30SBarry Smith    #include <petscmath.h>
398eca87e8dSBarry Smith    type max PetscMax(type v1,type v2)
399eca87e8dSBarry Smith 
400eca87e8dSBarry Smith    Not Collective
401eca87e8dSBarry Smith 
402b6a5bde7SBarry Smith    Input Parameter:
403b6a5bde7SBarry Smith +  v1 - first value to find maximum of
404b6a5bde7SBarry Smith -  v2 - second value to find maximum of
405b6a5bde7SBarry Smith 
406b6a5bde7SBarry Smith    Notes: type can be integer or floating point value
407b6a5bde7SBarry Smith 
408b6a5bde7SBarry Smith    Level: beginner
409b6a5bde7SBarry Smith 
410d9a4bb16SJed Brown .seealso: PetscMin(), PetscClipInterval(), PetscAbsInt(), PetscAbsReal(), PetscSqr()
411b6a5bde7SBarry Smith 
412b6a5bde7SBarry Smith M*/
413e489efc1SBarry Smith #define PetscMax(a,b)   (((a)<(b)) ?  (b) : (a))
414b6a5bde7SBarry Smith 
415b6a5bde7SBarry Smith /*MC
416d9a4bb16SJed Brown    PetscClipInterval - Returns a number clipped to be within an interval
417d9a4bb16SJed Brown 
418d9a4bb16SJed Brown    Synopsis:
419aaa7dc30SBarry Smith    #include <petscmath.h>
420d9a4bb16SJed Brown    type clip PetscClipInterval(type x,type a,type b)
421d9a4bb16SJed Brown 
422d9a4bb16SJed Brown    Not Collective
423d9a4bb16SJed Brown 
424d9a4bb16SJed Brown    Input Parameter:
425d9a4bb16SJed Brown +  x - value to use if within interval (a,b)
426d9a4bb16SJed Brown .  a - lower end of interval
427d9a4bb16SJed Brown -  b - upper end of interval
428d9a4bb16SJed Brown 
429d9a4bb16SJed Brown    Notes: type can be integer or floating point value
430d9a4bb16SJed Brown 
431d9a4bb16SJed Brown    Level: beginner
432d9a4bb16SJed Brown 
433d9a4bb16SJed Brown .seealso: PetscMin(), PetscMax(), PetscAbsInt(), PetscAbsReal(), PetscSqr()
434d9a4bb16SJed Brown 
435d9a4bb16SJed Brown M*/
436d9a4bb16SJed Brown #define PetscClipInterval(x,a,b)   (PetscMax((a),PetscMin((x),(b))))
437d9a4bb16SJed Brown 
438d9a4bb16SJed Brown /*MC
439b6a5bde7SBarry Smith    PetscAbsInt - Returns the absolute value of an integer
440b6a5bde7SBarry Smith 
441b6a5bde7SBarry Smith    Synopsis:
442aaa7dc30SBarry Smith    #include <petscmath.h>
443b6a5bde7SBarry Smith    int abs PetscAbsInt(int v1)
444b6a5bde7SBarry Smith 
445eca87e8dSBarry Smith    Not Collective
446eca87e8dSBarry Smith 
447eca87e8dSBarry Smith    Input Parameter:
448eca87e8dSBarry Smith .   v1 - the integer
449b6a5bde7SBarry Smith 
450b6a5bde7SBarry Smith    Level: beginner
451b6a5bde7SBarry Smith 
452b6a5bde7SBarry Smith .seealso: PetscMax(), PetscMin(), PetscAbsReal(), PetscSqr()
453b6a5bde7SBarry Smith 
454b6a5bde7SBarry Smith M*/
455e489efc1SBarry Smith #define PetscAbsInt(a)  (((a)<0)   ? -(a) : (a))
456b6a5bde7SBarry Smith 
457b6a5bde7SBarry Smith /*MC
458b6a5bde7SBarry Smith    PetscAbsReal - Returns the absolute value of an real number
459b6a5bde7SBarry Smith 
460eca87e8dSBarry Smith    Synopsis:
461aaa7dc30SBarry Smith    #include <petscmath.h>
462eca87e8dSBarry Smith    Real abs PetscAbsReal(PetscReal v1)
463eca87e8dSBarry Smith 
464eca87e8dSBarry Smith    Not Collective
465eca87e8dSBarry Smith 
466b6a5bde7SBarry Smith    Input Parameter:
467b6a5bde7SBarry Smith .   v1 - the double
468b6a5bde7SBarry Smith 
469b6a5bde7SBarry Smith 
470b6a5bde7SBarry Smith    Level: beginner
471b6a5bde7SBarry Smith 
472b6a5bde7SBarry Smith .seealso: PetscMax(), PetscMin(), PetscAbsInt(), PetscSqr()
473b6a5bde7SBarry Smith 
474b6a5bde7SBarry Smith M*/
475f6275e2eSBarry Smith #define PetscAbsReal(a) (((a)<0)   ? -(a) : (a))
476b6a5bde7SBarry Smith 
477b6a5bde7SBarry Smith /*MC
478b6a5bde7SBarry Smith    PetscSqr - Returns the square of a number
479b6a5bde7SBarry Smith 
480b6a5bde7SBarry Smith    Synopsis:
481aaa7dc30SBarry Smith    #include <petscmath.h>
482b6a5bde7SBarry Smith    type sqr PetscSqr(type v1)
483b6a5bde7SBarry Smith 
484eca87e8dSBarry Smith    Not Collective
485eca87e8dSBarry Smith 
486eca87e8dSBarry Smith    Input Parameter:
487eca87e8dSBarry Smith .   v1 - the value
488eca87e8dSBarry Smith 
489b6a5bde7SBarry Smith    Notes: type can be integer or floating point value
490b6a5bde7SBarry Smith 
491b6a5bde7SBarry Smith    Level: beginner
492b6a5bde7SBarry Smith 
493b6a5bde7SBarry Smith .seealso: PetscMax(), PetscMin(), PetscAbsInt(), PetscAbsReal()
494b6a5bde7SBarry Smith 
495b6a5bde7SBarry Smith M*/
4964ebda54eSMatthew Knepley #define PetscSqr(a)     ((a)*(a))
497e489efc1SBarry Smith 
498314da920SBarry Smith /* ----------------------------------------------------------------------------*/
499314da920SBarry Smith /*
500d34fcf5fSBarry Smith      Basic constants
501314da920SBarry Smith */
502ce63c4c1SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128)
503d34fcf5fSBarry Smith #define PETSC_PI                 M_PIq
504d34fcf5fSBarry Smith #elif defined(M_PI)
505d34fcf5fSBarry Smith #define PETSC_PI                 M_PI
506d34fcf5fSBarry Smith #else
507faa6e9b0SMatthew G Knepley #define PETSC_PI                 3.14159265358979323846264338327950288419716939937510582
508d34fcf5fSBarry Smith #endif
509d34fcf5fSBarry Smith 
510ab824b78SBarry Smith #if !defined(PETSC_USE_64BIT_INDICES)
51171fd2e92SBarry Smith #define PETSC_MAX_INT            2147483647
512ab824b78SBarry Smith #define PETSC_MIN_INT            (-PETSC_MAX_INT - 1)
513ab824b78SBarry Smith #else
514ab824b78SBarry Smith #define PETSC_MAX_INT            9223372036854775807L
515ab824b78SBarry Smith #define PETSC_MIN_INT            (-PETSC_MAX_INT - 1)
516ab824b78SBarry Smith #endif
517e489efc1SBarry Smith 
518ce63c4c1SBarry Smith #if defined(PETSC_USE_REAL_SINGLE)
519ab824b78SBarry Smith #  define PETSC_MAX_REAL                3.40282346638528860e+38F
520ea345e14SBarry Smith #  define PETSC_MIN_REAL                -PETSC_MAX_REAL
52182a7e548SBarry Smith #  define PETSC_MACHINE_EPSILON         1.19209290e-07F
52282a7e548SBarry Smith #  define PETSC_SQRT_MACHINE_EPSILON    3.45266983e-04F
523cf6e855fSSatish Balay #  define PETSC_SMALL                   1.e-5
524ce63c4c1SBarry Smith #elif defined(PETSC_USE_REAL_DOUBLE)
525ab824b78SBarry Smith #  define PETSC_MAX_REAL                1.7976931348623157e+308
526ea345e14SBarry Smith #  define PETSC_MIN_REAL                -PETSC_MAX_REAL
52782a7e548SBarry Smith #  define PETSC_MACHINE_EPSILON         2.2204460492503131e-16
52882a7e548SBarry Smith #  define PETSC_SQRT_MACHINE_EPSILON    1.490116119384766e-08
529cf6e855fSSatish Balay #  define PETSC_SMALL                   1.e-10
530ce63c4c1SBarry Smith #elif defined(PETSC_USE_REAL___FLOAT128)
531ea345e14SBarry Smith #  define PETSC_MAX_REAL                FLT128_MAX
532ce63c4c1SBarry Smith #  define PETSC_MIN_REAL                -FLT128_MAX
533d34fcf5fSBarry Smith #  define PETSC_MACHINE_EPSILON         FLT128_EPSILON
53446d881e9SBarry Smith #  define PETSC_SQRT_MACHINE_EPSILON    1.38777878078e-17q
53546d881e9SBarry Smith #  define PETSC_SMALL                   1.e-20q
5369cf09972SJed Brown #endif
5373e523bebSBarry Smith 
538e270355aSBarry Smith #define PETSC_INFINITY                PETSC_MAX_REAL/4.0
539e270355aSBarry Smith #define PETSC_NINFINITY              -PETSC_INFINITY
540e270355aSBarry Smith 
541014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscIsInfOrNanReal(PetscReal);
542bae46576SBarry Smith PETSC_EXTERN PetscErrorCode PetscIsNanReal(PetscReal);
5438b49ba18SBarry Smith PETSC_EXTERN PetscBool PetscIsNormalReal(PetscReal);
544834b1445SSatish Balay PETSC_STATIC_INLINE PetscErrorCode PetscIsInfOrNanScalar(PetscScalar v) {return PetscIsInfOrNanReal(PetscAbsScalar(v));}
545bae46576SBarry Smith PETSC_STATIC_INLINE PetscErrorCode PetscIsNanScalar(PetscScalar v) {return PetscIsNanReal(PetscAbsScalar(v));}
546834b1445SSatish Balay PETSC_STATIC_INLINE PetscErrorCode PetscIsNormalScalar(PetscScalar v) {return PetscIsNormalReal(PetscAbsScalar(v));}
5479a25a3ccSBarry Smith 
54898725619SBarry Smith /*
54998725619SBarry Smith     These macros are currently hardwired to match the regular data types, so there is no support for a different
55098725619SBarry Smith     MatScalar from PetscScalar. We left the MatScalar in the source just in case we use it again.
55198725619SBarry Smith  */
55298725619SBarry Smith #define MPIU_MATSCALAR MPIU_SCALAR
55398725619SBarry Smith typedef PetscScalar MatScalar;
55498725619SBarry Smith typedef PetscReal MatReal;
55598725619SBarry Smith 
5568ad47952SJed Brown struct petsc_mpiu_2scalar {PetscScalar a,b;};
5578ad47952SJed Brown PETSC_EXTERN MPI_Datatype MPIU_2SCALAR PetscAttrMPITypeTagLayoutCompatible(struct petsc_mpiu_2scalar);
5588ad47952SJed Brown #if defined(PETSC_USE_64BIT_INDICES) || !defined(MPI_2INT)
5598ad47952SJed Brown struct petsc_mpiu_2int {PetscInt a,b;};
5608ad47952SJed Brown PETSC_EXTERN MPI_Datatype MPIU_2INT PetscAttrMPITypeTagLayoutCompatible(struct petsc_mpiu_2int);
5618ad47952SJed Brown #else
5628ad47952SJed Brown #define MPIU_2INT MPI_2INT
5638ad47952SJed Brown #endif
564e9fa29b7SSatish Balay 
565b2fb0278SBarry Smith PETSC_STATIC_INLINE PetscInt PetscPowInt(PetscInt base,PetscInt power)
566b2fb0278SBarry Smith {
567fa711258SJed Brown   PetscInt result = 1;
568fa711258SJed Brown   while (power) {
569fa711258SJed Brown     if (power & 1) result *= base;
570fa711258SJed Brown     power >>= 1;
571fa711258SJed Brown     base *= base;
572fa711258SJed Brown   }
573fa711258SJed Brown   return result;
574fa711258SJed Brown }
575b2fb0278SBarry Smith 
576b2fb0278SBarry Smith PETSC_STATIC_INLINE PetscReal PetscPowRealInt(PetscReal base,PetscInt power)
577b2fb0278SBarry Smith {
578fa711258SJed Brown   PetscReal result = 1;
579d98d5da7SBarry Smith   if (power < 0) {
580d98d5da7SBarry Smith     power = -power;
58110d40e53SLisandro Dalcin     base  = ((PetscReal)1)/base;
582d98d5da7SBarry Smith   }
583fa711258SJed Brown   while (power) {
584fa711258SJed Brown     if (power & 1) result *= base;
585fa711258SJed Brown     power >>= 1;
586fa711258SJed Brown     base *= base;
587fa711258SJed Brown   }
588fa711258SJed Brown   return result;
589fa711258SJed Brown }
590fa711258SJed Brown 
591b2fb0278SBarry Smith PETSC_STATIC_INLINE PetscScalar PetscPowScalarInt(PetscScalar base,PetscInt power)
592b2fb0278SBarry Smith {
5938b49ba18SBarry Smith   PetscScalar result = 1;
5948b49ba18SBarry Smith   if (power < 0) {
5958b49ba18SBarry Smith     power = -power;
59610d40e53SLisandro Dalcin     base  = ((PetscReal)1)/base;
5978b49ba18SBarry Smith   }
5988b49ba18SBarry Smith   while (power) {
5998b49ba18SBarry Smith     if (power & 1) result *= base;
6008b49ba18SBarry Smith     power >>= 1;
6018b49ba18SBarry Smith     base *= base;
6028b49ba18SBarry Smith   }
6038b49ba18SBarry Smith   return result;
6048b49ba18SBarry Smith }
6058b49ba18SBarry Smith 
606b2fb0278SBarry Smith PETSC_STATIC_INLINE PetscScalar PetscPowScalarReal(PetscScalar base,PetscReal power)
607b2fb0278SBarry Smith {
608b2fb0278SBarry Smith   PetscScalar cpower = power;
609b2fb0278SBarry Smith   return PetscPowScalar(base,cpower);
610b2fb0278SBarry Smith }
61178a59e97SMatthew G. Knepley 
61278a59e97SMatthew G. Knepley #ifndef PETSC_HAVE_LOG2
61378a59e97SMatthew G. Knepley PETSC_STATIC_INLINE PetscReal PetscLog2Real(PetscReal n)
61478a59e97SMatthew G. Knepley {
61591954be4SBarry Smith   return PetscLogReal(n)/PetscLogReal(2.0);
61678a59e97SMatthew G. Knepley }
61778a59e97SMatthew G. Knepley #endif
618e489efc1SBarry Smith #endif
619