xref: /petsc/include/petscmath.h (revision a97cf724bc6ebfa2ca07523e495468f44ca078ed)
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;
27*a97cf724SLisandro Dalcin #define PetscRoundReal(a)   roundf(a)
28*a97cf724SLisandro Dalcin #define PetscSqrtReal(a)    sqrtf(a)
29*a97cf724SLisandro Dalcin #define PetscExpReal(a)     expf(a)
30*a97cf724SLisandro Dalcin #define PetscLogReal(a)     logf(a)
31*a97cf724SLisandro Dalcin #define PetscLog10Real(a)   log10f(a)
3278a59e97SMatthew G. Knepley #ifdef PETSC_HAVE_LOG2
33*a97cf724SLisandro Dalcin #define PetscLog2Real(a)    log2f(a)
3478a59e97SMatthew G. Knepley #endif
35*a97cf724SLisandro Dalcin #define PetscSinReal(a)     sinf(a)
36*a97cf724SLisandro Dalcin #define PetscCosReal(a)     cosf(a)
37*a97cf724SLisandro Dalcin #define PetscTanReal(a)     tanf(a)
38*a97cf724SLisandro Dalcin #define PetscAsinReal(a)    asinf(a)
39*a97cf724SLisandro Dalcin #define PetscAcosReal(a)    acosf(a)
40*a97cf724SLisandro Dalcin #define PetscAtanReal(a)    atanf(a)
41*a97cf724SLisandro Dalcin #define PetscAtan2Real(a,b) atan2f(a,b)
42*a97cf724SLisandro Dalcin #define PetscSinhReal(a)    sinhf(a)
43*a97cf724SLisandro Dalcin #define PetscCoshReal(a)    coshf(a)
44*a97cf724SLisandro Dalcin #define PetscTanhReal(a)    tanhf(a)
45*a97cf724SLisandro Dalcin #define PetscPowReal(a,b)   powf(a,b)
46*a97cf724SLisandro Dalcin #define PetscCeilReal(a)    ceilf(a)
47*a97cf724SLisandro Dalcin #define PetscFloorReal(a)   floorf(a)
48*a97cf724SLisandro Dalcin #define PetscFmodReal(a,b)  fmodf(a,b)
4943f0d3baSJed Brown #define PetscTGamma(a)      tgammaf(a)
50c1d390e3SJed Brown #elif defined(PETSC_USE_REAL_DOUBLE)
51c1d390e3SJed Brown #define MPIU_REAL   MPI_DOUBLE
52c1d390e3SJed Brown typedef double PetscReal;
53814a3092SSatish Balay #define PetscRoundReal(a)   round(a)
548f1a2a5eSBarry Smith #define PetscSqrtReal(a)    sqrt(a)
559a07f4dfSJed Brown #define PetscExpReal(a)     exp(a)
569a07f4dfSJed Brown #define PetscLogReal(a)     log(a)
5777b4d14cSPeter Brune #define PetscLog10Real(a)   log10(a)
5878a59e97SMatthew G. Knepley #ifdef PETSC_HAVE_LOG2
5978a59e97SMatthew G. Knepley #define PetscLog2Real(a)    log2(a)
6078a59e97SMatthew G. Knepley #endif
619a07f4dfSJed Brown #define PetscSinReal(a)     sin(a)
629a07f4dfSJed Brown #define PetscCosReal(a)     cos(a)
63a4bea5a6SPeter Brune #define PetscTanReal(a)     tan(a)
64255453a1SBarry Smith #define PetscAsinReal(a)    asin(a)
65255453a1SBarry Smith #define PetscAcosReal(a)    acos(a)
66a4bea5a6SPeter Brune #define PetscAtanReal(a)    atan(a)
671d5abddaSDmitry Karpeev #define PetscAtan2Real(a,b) atan2(a,b)
688d1b9be0SJed Brown #define PetscSinhReal(a)    sinh(a)
698d1b9be0SJed Brown #define PetscCoshReal(a)    cosh(a)
708d1b9be0SJed Brown #define PetscTanhReal(a)    tanh(a)
71369cc0aeSBarry Smith #define PetscPowReal(a,b)   pow(a,b)
7277b4d14cSPeter Brune #define PetscCeilReal(a)    ceil(a)
7377b4d14cSPeter Brune #define PetscFloorReal(a)   floor(a)
74a4bea5a6SPeter Brune #define PetscFmodReal(a,b)  fmod(a,b)
750646a658SBarry Smith #define PetscTGamma(a)      tgamma(a)
76c1d390e3SJed Brown #elif defined(PETSC_USE_REAL___FLOAT128)
77574fde7bSSatish Balay #if defined(__cplusplus)
78574fde7bSSatish Balay extern "C" {
79574fde7bSSatish Balay #endif
80574fde7bSSatish Balay #include <quadmath.h>
81574fde7bSSatish Balay #if defined(__cplusplus)
82574fde7bSSatish Balay }
83574fde7bSSatish Balay #endif
848ad47952SJed Brown PETSC_EXTERN MPI_Datatype MPIU___FLOAT128 PetscAttrMPITypeTag(__float128);
85c1d390e3SJed Brown #define MPIU_REAL MPIU___FLOAT128
86c1d390e3SJed Brown typedef __float128 PetscReal;
87814a3092SSatish Balay #define PetscRoundReal(a)   roundq(a)
888f1a2a5eSBarry Smith #define PetscSqrtReal(a)    sqrtq(a)
899a07f4dfSJed Brown #define PetscExpReal(a)     expq(a)
909a07f4dfSJed Brown #define PetscLogReal(a)     logq(a)
9177b4d14cSPeter Brune #define PetscLog10Real(a)   log10q(a)
9278a59e97SMatthew G. Knepley #ifdef PETSC_HAVE_LOG2
9378a59e97SMatthew G. Knepley #define PetscLog2Real(a)    log2q(a)
9478a59e97SMatthew G. Knepley #endif
959a07f4dfSJed Brown #define PetscSinReal(a)     sinq(a)
969a07f4dfSJed Brown #define PetscCosReal(a)     cosq(a)
97a4bea5a6SPeter Brune #define PetscTanReal(a)     tanq(a)
98255453a1SBarry Smith #define PetscAsinReal(a)    asinq(a)
99255453a1SBarry Smith #define PetscAcosReal(a)    acosq(a)
100a4bea5a6SPeter Brune #define PetscAtanReal(a)    atanq(a)
1011d5abddaSDmitry Karpeev #define PetscAtan2Real(a,b) atan2q(a,b)
1028d1b9be0SJed Brown #define PetscSinhReal(a)    sinhq(a)
1038d1b9be0SJed Brown #define PetscCoshReal(a)    coshq(a)
1048d1b9be0SJed Brown #define PetscTanhReal(a)    tanhq(a)
105369cc0aeSBarry Smith #define PetscPowReal(a,b)   powq(a,b)
10677b4d14cSPeter Brune #define PetscCeilReal(a)    ceilq(a)
10777b4d14cSPeter Brune #define PetscFloorReal(a)   floorq(a)
108a4bea5a6SPeter Brune #define PetscFmodReal(a,b)  fmodq(a,b)
1090646a658SBarry Smith #define PetscTGamma(a)      tgammaq(a)
110570b7f6dSBarry Smith #elif defined(PETSC_USE_REAL___FP16)
111570b7f6dSBarry Smith PETSC_EXTERN MPI_Datatype MPIU___FP16 PetscAttrMPITypeTag(__fp16);
112570b7f6dSBarry Smith #define MPIU_REAL MPIU___FP16
113570b7f6dSBarry Smith typedef __fp16 PetscReal;
114814a3092SSatish Balay #define PetscRound(a)       roundf(a)
115570b7f6dSBarry Smith #define PetscSqrtReal(a)    sqrtf(a)
116570b7f6dSBarry Smith #define PetscExpReal(a)     expf(a)
117570b7f6dSBarry Smith #define PetscLogReal(a)     logf(a)
118570b7f6dSBarry Smith #define PetscLog10Real(a)   log10f(a)
119570b7f6dSBarry Smith #ifdef PETSC_HAVE_LOG2
120570b7f6dSBarry Smith #define PetscLog2Real(a)    log2f(a)
121570b7f6dSBarry Smith #endif
122570b7f6dSBarry Smith #define PetscSinReal(a)     sinf(a)
123570b7f6dSBarry Smith #define PetscCosReal(a)     cosf(a)
124570b7f6dSBarry Smith #define PetscTanReal(a)     tanf(a)
125570b7f6dSBarry Smith #define PetscAsinReal(a)    asinf(a)
126570b7f6dSBarry Smith #define PetscAcosReal(a)    acosf(a)
127570b7f6dSBarry Smith #define PetscAtanReal(a)    atanf(a)
128570b7f6dSBarry Smith #define PetscAtan2Real(a,b) atan2f(a,b)
129570b7f6dSBarry Smith #define PetscSinhReal(a)    sinhf(a)
130570b7f6dSBarry Smith #define PetscCoshReal(a)    coshf(a)
131570b7f6dSBarry Smith #define PetscTanhReal(a)    tanhf(a)
132570b7f6dSBarry Smith #define PetscPowReal(a,b)   powf(a,b)
133570b7f6dSBarry Smith #define PetscCeilReal(a)    ceilf(a)
134570b7f6dSBarry Smith #define PetscFloorReal(a)   floorf(a)
135570b7f6dSBarry Smith #define PetscFmodReal(a,b)  fmodf(a,b)
136570b7f6dSBarry Smith #define PetscTGamma(a)      tgammaf(a)
137c1d390e3SJed Brown #endif /* PETSC_USE_REAL_* */
13859cb5930SBarry Smith 
1391093a601SBarry Smith /*
1401093a601SBarry Smith     Complex number definitions
1411093a601SBarry Smith  */
1422f217381SBarry Smith #if defined(__cplusplus) && defined(PETSC_HAVE_CXX_COMPLEX) && !defined(PETSC_USE_REAL___FLOAT128)
1438cd53115SBarry Smith #if !defined(PETSC_SKIP_COMPLEX)
14450f81f78SJed Brown #define PETSC_HAVE_COMPLEX 1
1451093a601SBarry Smith /* C++ support of complex number */
146debe9ee2SPaul Mullowney #if defined(PETSC_HAVE_CUSP)
147debe9ee2SPaul Mullowney #define complexlib cusp
1489ae82921SPaul Mullowney #include <cusp/complex.h>
149fda01a78SKarl Rupp #elif defined(PETSC_HAVE_VECCUDA) && __CUDACC_VER_MAJOR__ > 6
150fda01a78SKarl Rupp /* complex headers in thrust only available in CUDA 7.0 and above */
151ec42abe4SAlejandro Lamas Daviña #define complexlib thrust
152ec42abe4SAlejandro Lamas Daviña #include <thrust/complex.h>
153debe9ee2SPaul Mullowney #else
154debe9ee2SPaul Mullowney #define complexlib std
155debe9ee2SPaul Mullowney #include <complex>
1569ae82921SPaul Mullowney #endif
157b7940d39SSatish Balay 
15850f81f78SJed Brown #define PetscRealPartComplex(a)      (a).real()
15950f81f78SJed Brown #define PetscImaginaryPartComplex(a) (a).imag()
16050f81f78SJed Brown #define PetscAbsComplex(a)           complexlib::abs(a)
16150f81f78SJed Brown #define PetscConjComplex(a)          complexlib::conj(a)
16250f81f78SJed Brown #define PetscSqrtComplex(a)          complexlib::sqrt(a)
16350f81f78SJed Brown #define PetscPowComplex(a,b)         complexlib::pow(a,b)
16450f81f78SJed Brown #define PetscExpComplex(a)           complexlib::exp(a)
16550f81f78SJed Brown #define PetscLogComplex(a)           complexlib::log(a)
16650f81f78SJed Brown #define PetscSinComplex(a)           complexlib::sin(a)
16750f81f78SJed Brown #define PetscCosComplex(a)           complexlib::cos(a)
168255453a1SBarry Smith #define PetscAsinComplex(a)          complexlib::asin(a)
169255453a1SBarry Smith #define PetscAcosComplex(a)          complexlib::acos(a)
170027d9794SBarry Smith #if defined(PETSC_HAVE_TANCOMPLEX)
171a4bea5a6SPeter Brune #define PetscTanComplex(a)           complexlib::tan(a)
172027d9794SBarry Smith #else
173027d9794SBarry Smith #define PetscTanComplex(a)           PetscSinComplex(a)/PetscCosComplex(a)
174027d9794SBarry Smith #endif
175a4bea5a6SPeter Brune #define PetscSinhComplex(a)          complexlib::sinh(a)
176a4bea5a6SPeter Brune #define PetscCoshComplex(a)          complexlib::cosh(a)
177027d9794SBarry Smith #if defined(PETSC_HAVE_TANHCOMPLEX)
178a4bea5a6SPeter Brune #define PetscTanhComplex(a)          complexlib::tanh(a)
179027d9794SBarry Smith #else
180027d9794SBarry Smith #define PetscTanhComplex(a)          PetscSinhComplex(a)/PetscCoshComplex(a)
181027d9794SBarry Smith #endif
182debe9ee2SPaul Mullowney 
183debe9ee2SPaul Mullowney #if defined(PETSC_USE_REAL_SINGLE)
18450f81f78SJed Brown typedef complexlib::complex<float> PetscComplex;
1853be776deSLisandro Dalcin #if defined(PETSC_USE_CXX_COMPLEX_FLOAT_WORKAROUND)
1863be776deSLisandro Dalcin static inline PetscComplex operator+(const PetscComplex& lhs, const double& rhs) { return lhs + float(rhs); }
1873be776deSLisandro Dalcin static inline PetscComplex operator+(const double& lhs, const PetscComplex& rhs) { return float(lhs) + rhs; }
1883be776deSLisandro Dalcin static inline PetscComplex operator-(const PetscComplex& lhs, const double& rhs) { return lhs - float(rhs); }
1893be776deSLisandro Dalcin static inline PetscComplex operator-(const double& lhs, const PetscComplex& rhs) { return float(lhs) - rhs; }
1903be776deSLisandro Dalcin static inline PetscComplex operator*(const PetscComplex& lhs, const double& rhs) { return lhs * float(rhs); }
1913be776deSLisandro Dalcin static inline PetscComplex operator*(const double& lhs, const PetscComplex& rhs) { return float(lhs) * rhs; }
1923be776deSLisandro Dalcin static inline PetscComplex operator/(const PetscComplex& lhs, const double& rhs) { return lhs / float(rhs); }
1933be776deSLisandro Dalcin static inline PetscComplex operator/(const double& lhs, const PetscComplex& rhs) { return float(lhs) / rhs; }
1943be776deSLisandro Dalcin static inline bool operator==(const PetscComplex& lhs, const double& rhs) { return lhs.imag() == float(0) && lhs.real() == float(rhs); }
1953be776deSLisandro Dalcin static inline bool operator==(const double& lhs, const PetscComplex& rhs) { return rhs.imag() == float(0) && rhs.real() == float(lhs); }
1963be776deSLisandro Dalcin static inline bool operator!=(const PetscComplex& lhs, const double& rhs) { return lhs.imag() != float(0) || lhs.real() != float(rhs); }
1973be776deSLisandro Dalcin static inline bool operator!=(const double& lhs, const PetscComplex& rhs) { return rhs.imag() != float(0) || rhs.real() != float(lhs); }
1983be776deSLisandro Dalcin #endif  /* PETSC_USE_CXX_COMPLEX_FLOAT_WORKAROUND */
199debe9ee2SPaul Mullowney #elif defined(PETSC_USE_REAL_DOUBLE)
20050f81f78SJed Brown typedef complexlib::complex<double> PetscComplex;
2011a9c6b96SLisandro Dalcin #if defined(PETSC_USE_CXX_COMPLEX_FLOAT_WORKAROUND)
2021a9c6b96SLisandro Dalcin static inline PetscComplex operator+(const PetscComplex& lhs, const PetscInt& rhs) { return lhs + double(rhs); }
2031a9c6b96SLisandro Dalcin static inline PetscComplex operator+(const PetscInt& lhs, const PetscComplex& rhs) { return double(lhs) + rhs; }
2041a9c6b96SLisandro Dalcin static inline PetscComplex operator-(const PetscComplex& lhs, const PetscInt& rhs) { return lhs - double(rhs); }
2051a9c6b96SLisandro Dalcin static inline PetscComplex operator-(const PetscInt& lhs, const PetscComplex& rhs) { return double(lhs) - rhs; }
2061a9c6b96SLisandro Dalcin static inline PetscComplex operator*(const PetscComplex& lhs, const PetscInt& rhs) { return lhs * double(rhs); }
2071a9c6b96SLisandro Dalcin static inline PetscComplex operator*(const PetscInt& lhs, const PetscComplex& rhs) { return double(lhs) * rhs; }
2081a9c6b96SLisandro Dalcin static inline PetscComplex operator/(const PetscComplex& lhs, const PetscInt& rhs) { return lhs / double(rhs); }
2091a9c6b96SLisandro Dalcin static inline PetscComplex operator/(const PetscInt& lhs, const PetscComplex& rhs) { return double(lhs) / rhs; }
2101a9c6b96SLisandro Dalcin static inline bool operator==(const PetscComplex& lhs, const PetscInt& rhs) { return lhs.imag() == double(0) && lhs.real() == double(rhs); }
2111a9c6b96SLisandro Dalcin static inline bool operator==(const PetscInt& lhs, const PetscComplex& rhs) { return rhs.imag() == double(0) && rhs.real() == double(lhs); }
2121a9c6b96SLisandro Dalcin static inline bool operator!=(const PetscComplex& lhs, const PetscInt& rhs) { return lhs.imag() != double(0) || lhs.real() != double(rhs); }
2131a9c6b96SLisandro Dalcin static inline bool operator!=(const PetscInt& lhs, const PetscComplex& rhs) { return rhs.imag() != double(0) || rhs.real() != double(lhs); }
2141a9c6b96SLisandro Dalcin #endif  /* PETSC_USE_CXX_COMPLEX_FLOAT_WORKAROUND */
2158c764dc5SJose Roman #elif defined(PETSC_USE_REAL___FLOAT128)
21650f81f78SJed Brown typedef complexlib::complex<__float128> PetscComplex; /* Notstandard and not expected to work, use __complex128 */
21722b3908eSBarry Smith PETSC_EXTERN MPI_Datatype MPIU___COMPLEX128;
218debe9ee2SPaul Mullowney #endif  /* PETSC_USE_REAL_ */
2198cd53115SBarry Smith #endif  /* ! PETSC_SKIP_COMPLEX */
220debe9ee2SPaul Mullowney 
221546cf897SSatish Balay #elif defined(PETSC_HAVE_C99_COMPLEX) && !defined(PETSC_USE_REAL___FP16)
2228cd53115SBarry Smith #if !defined(PETSC_SKIP_COMPLEX)
2239f20b660SSatish Balay #define PETSC_HAVE_COMPLEX 1
224519e2a1fSPaul Mullowney #include <complex.h>
225519e2a1fSPaul Mullowney 
226570b7f6dSBarry Smith #if defined(PETSC_USE_REAL_SINGLE) || defined(PETSC_USE_REAL___FP16)
22750f81f78SJed Brown typedef float _Complex PetscComplex;
22885b47369SMatthew Knepley 
22950f81f78SJed Brown #define PetscRealPartComplex(a)      crealf(a)
23050f81f78SJed Brown #define PetscImaginaryPartComplex(a) cimagf(a)
23150f81f78SJed Brown #define PetscAbsComplex(a)           cabsf(a)
23250f81f78SJed Brown #define PetscConjComplex(a)          conjf(a)
23350f81f78SJed Brown #define PetscSqrtComplex(a)          csqrtf(a)
23450f81f78SJed Brown #define PetscPowComplex(a,b)         cpowf(a,b)
23550f81f78SJed Brown #define PetscExpComplex(a)           cexpf(a)
23650f81f78SJed Brown #define PetscLogComplex(a)           clogf(a)
23750f81f78SJed Brown #define PetscSinComplex(a)           csinf(a)
23850f81f78SJed Brown #define PetscCosComplex(a)           ccosf(a)
239255453a1SBarry Smith #define PetscAsinComplex(a)          casinf(a)
240255453a1SBarry Smith #define PetscAcosComplex(a)          cacosf(a)
241a4bea5a6SPeter Brune #define PetscTanComplex(a)           ctanf(a)
242a4bea5a6SPeter Brune #define PetscSinhComplex(a)          csinhf(a)
243a4bea5a6SPeter Brune #define PetscCoshComplex(a)          ccoshf(a)
244a4bea5a6SPeter Brune #define PetscTanhComplex(a)          ctanhf(a)
2451093a601SBarry Smith 
246ce63c4c1SBarry Smith #elif defined(PETSC_USE_REAL_DOUBLE)
24750f81f78SJed Brown typedef double _Complex PetscComplex;
2481093a601SBarry Smith 
24950f81f78SJed Brown #define PetscRealPartComplex(a)      creal(a)
25050f81f78SJed Brown #define PetscImaginaryPartComplex(a) cimag(a)
25150f81f78SJed Brown #define PetscAbsComplex(a)           cabs(a)
25250f81f78SJed Brown #define PetscConjComplex(a)          conj(a)
25350f81f78SJed Brown #define PetscSqrtComplex(a)          csqrt(a)
25450f81f78SJed Brown #define PetscPowComplex(a,b)         cpow(a,b)
25550f81f78SJed Brown #define PetscExpComplex(a)           cexp(a)
25650f81f78SJed Brown #define PetscLogComplex(a)           clog(a)
25750f81f78SJed Brown #define PetscSinComplex(a)           csin(a)
25850f81f78SJed Brown #define PetscCosComplex(a)           ccos(a)
259255453a1SBarry Smith #define PetscAsinComplex(a)          casin(a)
260255453a1SBarry Smith #define PetscAcosComplex(a)          cacos(a)
261a4bea5a6SPeter Brune #define PetscTanComplex(a)           ctan(a)
262a4bea5a6SPeter Brune #define PetscSinhComplex(a)          csinh(a)
263a4bea5a6SPeter Brune #define PetscCoshComplex(a)          ccosh(a)
264a4bea5a6SPeter Brune #define PetscTanhComplex(a)          ctanh(a)
2651093a601SBarry Smith 
2668c764dc5SJose Roman #elif defined(PETSC_USE_REAL___FLOAT128)
26750f81f78SJed Brown typedef __complex128 PetscComplex;
2688ad47952SJed Brown PETSC_EXTERN MPI_Datatype MPIU___COMPLEX128 PetscAttrMPITypeTag(__complex128);
2698c764dc5SJose Roman 
27050f81f78SJed Brown #define PetscRealPartComplex(a)      crealq(a)
27150f81f78SJed Brown #define PetscImaginaryPartComplex(a) cimagq(a)
27250f81f78SJed Brown #define PetscAbsComplex(a)           cabsq(a)
27350f81f78SJed Brown #define PetscConjComplex(a)          conjq(a)
27450f81f78SJed Brown #define PetscSqrtComplex(a)          csqrtq(a)
27550f81f78SJed Brown #define PetscPowComplex(a,b)         cpowq(a,b)
27650f81f78SJed Brown #define PetscExpComplex(a)           cexpq(a)
27750f81f78SJed Brown #define PetscLogComplex(a)           clogq(a)
27850f81f78SJed Brown #define PetscSinComplex(a)           csinq(a)
27950f81f78SJed Brown #define PetscCosComplex(a)           ccosq(a)
280255453a1SBarry Smith #define PetscAsinComplex(a)          casinq(a)
281255453a1SBarry Smith #define PetscAcosComplex(a)          cacosq(a)
282a4bea5a6SPeter Brune #define PetscTanComplex(a)           ctanq(a)
283a4bea5a6SPeter Brune #define PetscSinhComplex(a)          csinhq(a)
284a4bea5a6SPeter Brune #define PetscCoshComplex(a)          ccoshq(a)
285a4bea5a6SPeter Brune #define PetscTanhComplex(a)          ctanhq(a)
286a4bea5a6SPeter Brune 
287ce63c4c1SBarry Smith #endif /* PETSC_USE_REAL_* */
288d4161b4aSBarry Smith #elif (defined(PETSC_USE_COMPLEX) && !defined(PETSC_SKIP_COMPLEX))
28950f81f78SJed Brown #error "PETSc was configured --with-scalar-type=complex, but a language-appropriate complex library is not available"
2908cd53115SBarry Smith #endif /* !PETSC_SKIP_COMPLEX */
2912f217381SBarry Smith #endif /* (__cplusplus && PETSC_HAVE_CXX_COMPLEX) else-if (!__cplusplus && PETSC_HAVE_C99_COMPLEX) */
292e489efc1SBarry Smith 
2938dc6f2c2SJed Brown #if defined(PETSC_HAVE_COMPLEX)
29470da9c3bSJed Brown #if defined(PETSC_HAVE_MPI_C_DOUBLE_COMPLEX)
295500d8756SSatish Balay #define MPIU_C_DOUBLE_COMPLEX MPI_C_DOUBLE_COMPLEX
296500d8756SSatish Balay #define MPIU_C_COMPLEX MPI_C_COMPLEX
29770da9c3bSJed Brown #else
298252ecd31SSatish Balay # if defined(__cplusplus) && defined(PETSC_HAVE_CXX_COMPLEX) && !defined(PETSC_USE_REAL___FLOAT128)
2998ad47952SJed Brown   typedef complexlib::complex<double> petsc_mpiu_c_double_complex;
3008ad47952SJed Brown   typedef complexlib::complex<float> petsc_mpiu_c_complex;
3012f217381SBarry Smith # elif !defined(__cplusplus) && defined(PETSC_HAVE_C99_COMPLEX)
3028ad47952SJed Brown   typedef double _Complex petsc_mpiu_c_double_complex;
3038ad47952SJed Brown   typedef float _Complex petsc_mpiu_c_complex;
3048ad47952SJed Brown # else
3058ad47952SJed Brown   typedef struct {double real,imag;} petsc_mpiu_c_double_complex;
3068ad47952SJed Brown   typedef struct {float real,imag;} petsc_mpiu_c_complex;
3078ad47952SJed Brown # endif
3088ad47952SJed Brown PETSC_EXTERN MPI_Datatype MPIU_C_DOUBLE_COMPLEX PetscAttrMPITypeTagLayoutCompatible(petsc_mpiu_c_double_complex);
3098ad47952SJed Brown PETSC_EXTERN MPI_Datatype MPIU_C_COMPLEX PetscAttrMPITypeTagLayoutCompatible(petsc_mpiu_c_complex);
3101b65fc54SMatthew G Knepley #endif /* PETSC_HAVE_MPI_C_DOUBLE_COMPLEX */
3118dc6f2c2SJed Brown #endif /* PETSC_HAVE_COMPLEX */
3122c876bd9SBarry Smith 
3137c2de775SJed Brown #if defined(PETSC_HAVE_COMPLEX)
314570b7f6dSBarry Smith #  if defined(PETSC_USE_REAL_SINGLE) || defined(PETSC_USE_REAL___FP16)
3157c2de775SJed Brown #    define MPIU_COMPLEX MPIU_C_COMPLEX
3167c2de775SJed Brown #  elif defined(PETSC_USE_REAL_DOUBLE)
3177c2de775SJed Brown #    define MPIU_COMPLEX MPIU_C_DOUBLE_COMPLEX
3187c2de775SJed Brown #  elif defined(PETSC_USE_REAL___FLOAT128)
3197c2de775SJed Brown #    define MPIU_COMPLEX MPIU___COMPLEX128
3207c2de775SJed Brown #  endif /* PETSC_USE_REAL_* */
3217c2de775SJed Brown #endif
3227c2de775SJed Brown 
323d4161b4aSBarry Smith #if (defined(PETSC_USE_COMPLEX) && !defined(PETSC_SKIP_COMPLEX))
32450f81f78SJed Brown typedef PetscComplex PetscScalar;
32550f81f78SJed Brown #define PetscRealPart(a)      PetscRealPartComplex(a)
32650f81f78SJed Brown #define PetscImaginaryPart(a) PetscImaginaryPartComplex(a)
32750f81f78SJed Brown #define PetscAbsScalar(a)     PetscAbsComplex(a)
32850f81f78SJed Brown #define PetscConj(a)          PetscConjComplex(a)
32950f81f78SJed Brown #define PetscSqrtScalar(a)    PetscSqrtComplex(a)
33050f81f78SJed Brown #define PetscPowScalar(a,b)   PetscPowComplex(a,b)
33150f81f78SJed Brown #define PetscExpScalar(a)     PetscExpComplex(a)
33250f81f78SJed Brown #define PetscLogScalar(a)     PetscLogComplex(a)
33350f81f78SJed Brown #define PetscSinScalar(a)     PetscSinComplex(a)
33450f81f78SJed Brown #define PetscCosScalar(a)     PetscCosComplex(a)
335255453a1SBarry Smith #define PetscAsinScalar(a)    PetscAsinComplex(a)
336255453a1SBarry Smith #define PetscAcosScalar(a)    PetscAcosComplex(a)
337a4bea5a6SPeter Brune #define PetscTanScalar(a)     PetscTanComplex(a)
338a4bea5a6SPeter Brune #define PetscSinhScalar(a)    PetscSinhComplex(a)
339a4bea5a6SPeter Brune #define PetscCoshScalar(a)    PetscCoshComplex(a)
340a4bea5a6SPeter Brune #define PetscTanhScalar(a)    PetscTanhComplex(a)
3417c2de775SJed Brown #define MPIU_SCALAR MPIU_COMPLEX
34275567043SBarry Smith 
3431093a601SBarry Smith /*
3441093a601SBarry Smith     real number definitions
3451093a601SBarry Smith  */
3461b65fc54SMatthew G Knepley #else /* PETSC_USE_COMPLEX */
3477c2de775SJed Brown typedef PetscReal PetscScalar;
3487c2de775SJed Brown #define MPIU_SCALAR MPIU_REAL
3497c2de775SJed Brown 
350329f5518SBarry Smith #define PetscRealPart(a)      (a)
351c1d390e3SJed Brown #define PetscImaginaryPart(a) ((PetscReal)0.)
352c1d390e3SJed Brown PETSC_STATIC_INLINE PetscReal PetscAbsScalar(PetscScalar a) {return a < 0.0 ? -a : a;}
353e489efc1SBarry Smith #define PetscConj(a)          (a)
354570b7f6dSBarry Smith #if !defined(PETSC_USE_REAL___FLOAT128)  && !defined(PETSC_USE_REAL___FP16)
35518a7d68fSSatish Balay #define PetscSqrtScalar(a)    sqrt(a)
356184914b5SBarry Smith #define PetscPowScalar(a,b)   pow(a,b)
357184914b5SBarry Smith #define PetscExpScalar(a)     exp(a)
35806c1185fSBarry Smith #define PetscLogScalar(a)     log(a)
359184914b5SBarry Smith #define PetscSinScalar(a)     sin(a)
360184914b5SBarry Smith #define PetscCosScalar(a)     cos(a)
361255453a1SBarry Smith #define PetscAsinScalar(a)    asin(a)
362255453a1SBarry Smith #define PetscAcosScalar(a)    acos(a)
363a4bea5a6SPeter Brune #define PetscTanScalar(a)     tan(a)
364a4bea5a6SPeter Brune #define PetscSinhScalar(a)    sinh(a)
365a4bea5a6SPeter Brune #define PetscCoshScalar(a)    cosh(a)
366a4bea5a6SPeter Brune #define PetscTanhScalar(a)    tanh(a)
367570b7f6dSBarry Smith #elif defined(PETSC_USE_REAL___FP16)
368570b7f6dSBarry Smith #define PetscSqrtScalar(a)    sqrtf(a)
369570b7f6dSBarry Smith #define PetscPowScalar(a,b)   powf(a,b)
370570b7f6dSBarry Smith #define PetscExpScalar(a)     expf(a)
371570b7f6dSBarry Smith #define PetscLogScalar(a)     logf(a)
372570b7f6dSBarry Smith #define PetscSinScalar(a)     sinf(a)
373570b7f6dSBarry Smith #define PetscCosScalar(a)     cosf(a)
374570b7f6dSBarry Smith #define PetscAsinScalar(a)    asinf(a)
375570b7f6dSBarry Smith #define PetscAcosScalar(a)    acosf(a)
376570b7f6dSBarry Smith #define PetscTanScalar(a)     tanf(a)
377570b7f6dSBarry Smith #define PetscSinhScalar(a)    sinhf(a)
378570b7f6dSBarry Smith #define PetscCoshScalar(a)    coshf(a)
379570b7f6dSBarry Smith #define PetscTanhScalar(a)    tanhf(a)
380ce63c4c1SBarry Smith #else /* PETSC_USE_REAL___FLOAT128 */
3810d0cc1b5SBarry Smith #define PetscSqrtScalar(a)    sqrtq(a)
3820d0cc1b5SBarry Smith #define PetscPowScalar(a,b)   powq(a,b)
3830d0cc1b5SBarry Smith #define PetscExpScalar(a)     expq(a)
3840d0cc1b5SBarry Smith #define PetscLogScalar(a)     logq(a)
3850d0cc1b5SBarry Smith #define PetscSinScalar(a)     sinq(a)
3860d0cc1b5SBarry Smith #define PetscCosScalar(a)     cosq(a)
387255453a1SBarry Smith #define PetscAsinScalar(a)    asinq(a)
388255453a1SBarry Smith #define PetscAcosScalar(a)    acosq(a)
389a4bea5a6SPeter Brune #define PetscTanScalar(a)     tanq(a)
390a4bea5a6SPeter Brune #define PetscSinhScalar(a)    sinhq(a)
391a4bea5a6SPeter Brune #define PetscCoshScalar(a)    coshq(a)
392a4bea5a6SPeter Brune #define PetscTanhScalar(a)    tanhq(a)
393ce63c4c1SBarry Smith #endif /* PETSC_USE_REAL___FLOAT128 */
394b0a32e0cSBarry Smith 
3951b65fc54SMatthew G Knepley #endif /* PETSC_USE_COMPLEX */
396e489efc1SBarry Smith 
397da9b6338SBarry Smith #define PetscSign(a) (((a) >= 0) ? ((a) == 0 ? 0 : 1) : -1)
39836850ab7SBarry Smith #define PetscSignReal(a) (((a) >= 0.0) ? ((a) == 0.0 ? 0.0 : 1.0) : -1.0)
3999fa7d148SSatish Balay #define PetscAbs(a)  (((a) >= 0) ? (a) : (-(a)))
4003f1db9ecSBarry Smith 
401314da920SBarry Smith /* --------------------------------------------------------------------------*/
402314da920SBarry Smith 
403e489efc1SBarry Smith /*
404f22f69f0SBarry Smith    Certain objects may be created using either single or double precision.
405f22f69f0SBarry Smith    This is currently not used.
406e489efc1SBarry Smith */
407570b7f6dSBarry Smith typedef enum { PETSC_SCALAR_DOUBLE,PETSC_SCALAR_SINGLE, PETSC_SCALAR_LONG_DOUBLE, PETSC_SCALAR_HALF } PetscScalarPrecision;
408e489efc1SBarry Smith 
40950f81f78SJed Brown #if defined(PETSC_HAVE_COMPLEX)
410e489efc1SBarry Smith /* PETSC_i is the imaginary number, i */
41150f81f78SJed Brown PETSC_EXTERN PetscComplex PETSC_i;
4128a351411SToby Isaac 
4138a351411SToby Isaac /* Try to do the right thing for complex number construction: see
4148a351411SToby Isaac 
4158a351411SToby Isaac   http://www.open-std.org/jtc1/sc22/wg14/www/docs/n1464.htm
4168a351411SToby Isaac 
4178a351411SToby Isaac   for details
4188a351411SToby Isaac */
41919e222d7SToby Isaac PETSC_STATIC_INLINE PetscComplex PetscCMPLX(PetscReal x, PetscReal y)
4208a351411SToby Isaac {
421546cf897SSatish Balay #if   defined(__cplusplus) && !defined(PETSC_USE_REAL___FLOAT128)
4228a351411SToby Isaac   return PetscComplex(x,y);
4238a351411SToby Isaac #elif defined(_Imaginary_I)
4248a351411SToby Isaac   return x + y * _Imaginary_I;
4258a351411SToby Isaac #else
426616d7c5eSToby Isaac   { /* In both C99 and C11 (ISO/IEC 9899, Section 6.2.5),
427616d7c5eSToby Isaac 
428616d7c5eSToby Isaac        "For each floating type there is a corresponding real type, which is always a real floating
429616d7c5eSToby Isaac        type. For real floating types, it is the same type. For complex types, it is the type given
430616d7c5eSToby Isaac        by deleting the keyword _Complex from the type name."
431616d7c5eSToby Isaac 
432616d7c5eSToby Isaac        So type punning should be portable. */
433616d7c5eSToby Isaac     union { PetscComplex z; PetscReal f[2]; } uz;
434616d7c5eSToby Isaac 
435616d7c5eSToby Isaac     uz.f[0] = x;
436616d7c5eSToby Isaac     uz.f[1] = y;
437616d7c5eSToby Isaac     return uz.z;
438616d7c5eSToby Isaac   }
43950f81f78SJed Brown #endif
4408a351411SToby Isaac }
4418a351411SToby Isaac #endif
4428a351411SToby Isaac 
443e489efc1SBarry Smith 
444b6a5bde7SBarry Smith /*MC
445b6a5bde7SBarry Smith    PetscMin - Returns minimum of two numbers
446b6a5bde7SBarry Smith 
447eca87e8dSBarry Smith    Synopsis:
448aaa7dc30SBarry Smith    #include <petscmath.h>
449eca87e8dSBarry Smith    type PetscMin(type v1,type v2)
450eca87e8dSBarry Smith 
451eca87e8dSBarry Smith    Not Collective
452eca87e8dSBarry Smith 
453b6a5bde7SBarry Smith    Input Parameter:
454b6a5bde7SBarry Smith +  v1 - first value to find minimum of
455b6a5bde7SBarry Smith -  v2 - second value to find minimum of
456b6a5bde7SBarry Smith 
457b6a5bde7SBarry Smith    Notes: type can be integer or floating point value
458b6a5bde7SBarry Smith 
459b6a5bde7SBarry Smith    Level: beginner
460b6a5bde7SBarry Smith 
4611175f9beSHong Zhang .seealso: PetscMax(), PetscClipInterval(), PetscAbsInt(), PetscAbsReal(), PetscSqr()
462b6a5bde7SBarry Smith 
463b6a5bde7SBarry Smith M*/
464e489efc1SBarry Smith #define PetscMin(a,b)   (((a)<(b)) ?  (a) : (b))
465b6a5bde7SBarry Smith 
466b6a5bde7SBarry Smith /*MC
467b6a5bde7SBarry Smith    PetscMax - Returns maxium of two numbers
468b6a5bde7SBarry Smith 
469eca87e8dSBarry Smith    Synopsis:
470aaa7dc30SBarry Smith    #include <petscmath.h>
471eca87e8dSBarry Smith    type max PetscMax(type v1,type v2)
472eca87e8dSBarry Smith 
473eca87e8dSBarry Smith    Not Collective
474eca87e8dSBarry Smith 
475b6a5bde7SBarry Smith    Input Parameter:
476b6a5bde7SBarry Smith +  v1 - first value to find maximum of
477b6a5bde7SBarry Smith -  v2 - second value to find maximum of
478b6a5bde7SBarry Smith 
479b6a5bde7SBarry Smith    Notes: type can be integer or floating point value
480b6a5bde7SBarry Smith 
481b6a5bde7SBarry Smith    Level: beginner
482b6a5bde7SBarry Smith 
483d9a4bb16SJed Brown .seealso: PetscMin(), PetscClipInterval(), PetscAbsInt(), PetscAbsReal(), PetscSqr()
484b6a5bde7SBarry Smith 
485b6a5bde7SBarry Smith M*/
486e489efc1SBarry Smith #define PetscMax(a,b)   (((a)<(b)) ?  (b) : (a))
487b6a5bde7SBarry Smith 
488b6a5bde7SBarry Smith /*MC
489d9a4bb16SJed Brown    PetscClipInterval - Returns a number clipped to be within an interval
490d9a4bb16SJed Brown 
491d9a4bb16SJed Brown    Synopsis:
492aaa7dc30SBarry Smith    #include <petscmath.h>
493d9a4bb16SJed Brown    type clip PetscClipInterval(type x,type a,type b)
494d9a4bb16SJed Brown 
495d9a4bb16SJed Brown    Not Collective
496d9a4bb16SJed Brown 
497d9a4bb16SJed Brown    Input Parameter:
498d9a4bb16SJed Brown +  x - value to use if within interval (a,b)
499d9a4bb16SJed Brown .  a - lower end of interval
500d9a4bb16SJed Brown -  b - upper end of interval
501d9a4bb16SJed Brown 
502d9a4bb16SJed Brown    Notes: type can be integer or floating point value
503d9a4bb16SJed Brown 
504d9a4bb16SJed Brown    Level: beginner
505d9a4bb16SJed Brown 
506d9a4bb16SJed Brown .seealso: PetscMin(), PetscMax(), PetscAbsInt(), PetscAbsReal(), PetscSqr()
507d9a4bb16SJed Brown 
508d9a4bb16SJed Brown M*/
509d9a4bb16SJed Brown #define PetscClipInterval(x,a,b)   (PetscMax((a),PetscMin((x),(b))))
510d9a4bb16SJed Brown 
511d9a4bb16SJed Brown /*MC
512b6a5bde7SBarry Smith    PetscAbsInt - Returns the absolute value of an integer
513b6a5bde7SBarry Smith 
514b6a5bde7SBarry Smith    Synopsis:
515aaa7dc30SBarry Smith    #include <petscmath.h>
516b6a5bde7SBarry Smith    int abs PetscAbsInt(int v1)
517b6a5bde7SBarry Smith 
518eca87e8dSBarry Smith    Not Collective
519eca87e8dSBarry Smith 
520eca87e8dSBarry Smith    Input Parameter:
521eca87e8dSBarry Smith .   v1 - the integer
522b6a5bde7SBarry Smith 
523b6a5bde7SBarry Smith    Level: beginner
524b6a5bde7SBarry Smith 
525b6a5bde7SBarry Smith .seealso: PetscMax(), PetscMin(), PetscAbsReal(), PetscSqr()
526b6a5bde7SBarry Smith 
527b6a5bde7SBarry Smith M*/
5289fa7d148SSatish Balay #define PetscAbsInt(a)  (((a)<0)   ? (-(a)) : (a))
529b6a5bde7SBarry Smith 
530b6a5bde7SBarry Smith /*MC
531b6a5bde7SBarry Smith    PetscAbsReal - Returns the absolute value of an real number
532b6a5bde7SBarry Smith 
533eca87e8dSBarry Smith    Synopsis:
534aaa7dc30SBarry Smith    #include <petscmath.h>
535eca87e8dSBarry Smith    Real abs PetscAbsReal(PetscReal v1)
536eca87e8dSBarry Smith 
537eca87e8dSBarry Smith    Not Collective
538eca87e8dSBarry Smith 
539b6a5bde7SBarry Smith    Input Parameter:
540b6a5bde7SBarry Smith .   v1 - the double
541b6a5bde7SBarry Smith 
542b6a5bde7SBarry Smith 
543b6a5bde7SBarry Smith    Level: beginner
544b6a5bde7SBarry Smith 
545b6a5bde7SBarry Smith .seealso: PetscMax(), PetscMin(), PetscAbsInt(), PetscSqr()
546b6a5bde7SBarry Smith 
547b6a5bde7SBarry Smith M*/
5481118d4bcSLisandro Dalcin #if defined(PETSC_USE_REAL_SINGLE)
5491118d4bcSLisandro Dalcin #define PetscAbsReal(a) fabsf(a)
5501118d4bcSLisandro Dalcin #elif defined(PETSC_USE_REAL_DOUBLE)
5511118d4bcSLisandro Dalcin #define PetscAbsReal(a) fabs(a)
5521118d4bcSLisandro Dalcin #elif defined(PETSC_USE_REAL___FLOAT128)
5531118d4bcSLisandro Dalcin #define PetscAbsReal(a) fabsq(a)
5541118d4bcSLisandro Dalcin #elif defined(PETSC_USE_REAL___FP16)
5551118d4bcSLisandro Dalcin #define PetscAbsReal(a) fabsf(a)
5561118d4bcSLisandro Dalcin #endif
557b6a5bde7SBarry Smith 
558b6a5bde7SBarry Smith /*MC
559b6a5bde7SBarry Smith    PetscSqr - Returns the square of a number
560b6a5bde7SBarry Smith 
561b6a5bde7SBarry Smith    Synopsis:
562aaa7dc30SBarry Smith    #include <petscmath.h>
563b6a5bde7SBarry Smith    type sqr PetscSqr(type v1)
564b6a5bde7SBarry Smith 
565eca87e8dSBarry Smith    Not Collective
566eca87e8dSBarry Smith 
567eca87e8dSBarry Smith    Input Parameter:
568eca87e8dSBarry Smith .   v1 - the value
569eca87e8dSBarry Smith 
570b6a5bde7SBarry Smith    Notes: type can be integer or floating point value
571b6a5bde7SBarry Smith 
572b6a5bde7SBarry Smith    Level: beginner
573b6a5bde7SBarry Smith 
574b6a5bde7SBarry Smith .seealso: PetscMax(), PetscMin(), PetscAbsInt(), PetscAbsReal()
575b6a5bde7SBarry Smith 
576b6a5bde7SBarry Smith M*/
5774ebda54eSMatthew Knepley #define PetscSqr(a)     ((a)*(a))
578e489efc1SBarry Smith 
579314da920SBarry Smith /* ----------------------------------------------------------------------------*/
580ee223c85SLisandro Dalcin 
581ee223c85SLisandro Dalcin #if defined(PETSC_USE_REAL_SINGLE)
582ee223c85SLisandro Dalcin #define PetscRealConstant(constant) constant##F
583ee223c85SLisandro Dalcin #elif defined(PETSC_USE_REAL___FLOAT128)
584ee223c85SLisandro Dalcin #define PetscRealConstant(constant) constant##Q
585ee223c85SLisandro Dalcin #else
586ee223c85SLisandro Dalcin #define PetscRealConstant(constant) constant
587ee223c85SLisandro Dalcin #endif
588ee223c85SLisandro Dalcin 
589314da920SBarry Smith /*
590d34fcf5fSBarry Smith      Basic constants
591314da920SBarry Smith */
5922fab75feSLisandro Dalcin #define PETSC_PI   PetscRealConstant(3.1415926535897932384626433832795029)
5932fab75feSLisandro Dalcin #define PETSC_PHI  PetscRealConstant(1.6180339887498948482045868343656381)
594d34fcf5fSBarry Smith 
595ab824b78SBarry Smith #if !defined(PETSC_USE_64BIT_INDICES)
59671fd2e92SBarry Smith #define PETSC_MAX_INT            2147483647
597ab824b78SBarry Smith #define PETSC_MIN_INT            (-PETSC_MAX_INT - 1)
598ab824b78SBarry Smith #else
599ab824b78SBarry Smith #define PETSC_MAX_INT            9223372036854775807L
600ab824b78SBarry Smith #define PETSC_MIN_INT            (-PETSC_MAX_INT - 1)
601ab824b78SBarry Smith #endif
602e489efc1SBarry Smith 
603ce63c4c1SBarry Smith #if defined(PETSC_USE_REAL_SINGLE)
604ab824b78SBarry Smith #  define PETSC_MAX_REAL                3.40282346638528860e+38F
6059fa7d148SSatish Balay #  define PETSC_MIN_REAL                (-PETSC_MAX_REAL)
60682a7e548SBarry Smith #  define PETSC_MACHINE_EPSILON         1.19209290e-07F
60782a7e548SBarry Smith #  define PETSC_SQRT_MACHINE_EPSILON    3.45266983e-04F
608ee223c85SLisandro Dalcin #  define PETSC_SMALL                   1.e-5F
609ce63c4c1SBarry Smith #elif defined(PETSC_USE_REAL_DOUBLE)
610ab824b78SBarry Smith #  define PETSC_MAX_REAL                1.7976931348623157e+308
6119fa7d148SSatish Balay #  define PETSC_MIN_REAL                (-PETSC_MAX_REAL)
61282a7e548SBarry Smith #  define PETSC_MACHINE_EPSILON         2.2204460492503131e-16
61382a7e548SBarry Smith #  define PETSC_SQRT_MACHINE_EPSILON    1.490116119384766e-08
614cf6e855fSSatish Balay #  define PETSC_SMALL                   1.e-10
615ce63c4c1SBarry Smith #elif defined(PETSC_USE_REAL___FLOAT128)
616ea345e14SBarry Smith #  define PETSC_MAX_REAL                FLT128_MAX
6179fa7d148SSatish Balay #  define PETSC_MIN_REAL                (-FLT128_MAX)
618d34fcf5fSBarry Smith #  define PETSC_MACHINE_EPSILON         FLT128_EPSILON
619ee223c85SLisandro Dalcin #  define PETSC_SQRT_MACHINE_EPSILON    1.38777878078144567552953958511352539e-17Q
620ee223c85SLisandro Dalcin #  define PETSC_SMALL                   1.e-20Q
621570b7f6dSBarry Smith #elif defined(PETSC_USE_REAL___FP16)  /* maybe should use single precision values for these? */
622570b7f6dSBarry Smith #  define PETSC_MAX_REAL                65504.
6239fa7d148SSatish Balay #  define PETSC_MIN_REAL                (-PETSC_MAX_REAL)
624570b7f6dSBarry Smith #  define PETSC_MACHINE_EPSILON         .00097656
625570b7f6dSBarry Smith #  define PETSC_SQRT_MACHINE_EPSILON    .0312
626570b7f6dSBarry Smith #  define PETSC_SMALL                   5.e-3
6279cf09972SJed Brown #endif
6283e523bebSBarry Smith 
62925d0f998SSatish Balay #define PETSC_INFINITY               (PETSC_MAX_REAL/4)
6309fa7d148SSatish Balay #define PETSC_NINFINITY              (-PETSC_INFINITY)
631e270355aSBarry Smith 
6329f4f8022SLisandro Dalcin PETSC_EXTERN PetscBool PetscIsInfReal(PetscReal);
6333948c36eSLisandro Dalcin PETSC_EXTERN PetscBool PetscIsNanReal(PetscReal);
6348b49ba18SBarry Smith PETSC_EXTERN PetscBool PetscIsNormalReal(PetscReal);
6359f4f8022SLisandro Dalcin PETSC_STATIC_INLINE PetscBool PetscIsInfOrNanReal(PetscReal v) {return PetscIsInfReal(v) || PetscIsNanReal(v) ? PETSC_TRUE : PETSC_FALSE;}
6369f4f8022SLisandro Dalcin PETSC_STATIC_INLINE PetscBool PetscIsInfScalar(PetscScalar v) {return PetscIsInfReal(PetscAbsScalar(v));}
6373948c36eSLisandro Dalcin PETSC_STATIC_INLINE PetscBool PetscIsNanScalar(PetscScalar v) {return PetscIsNanReal(PetscAbsScalar(v));}
6389f4f8022SLisandro Dalcin PETSC_STATIC_INLINE PetscBool PetscIsInfOrNanScalar(PetscScalar v) {return PetscIsInfOrNanReal(PetscAbsScalar(v));}
6393948c36eSLisandro Dalcin PETSC_STATIC_INLINE PetscBool PetscIsNormalScalar(PetscScalar v) {return PetscIsNormalReal(PetscAbsScalar(v));}
6409a25a3ccSBarry Smith 
641ce4818fdSLisandro Dalcin PETSC_EXTERN PetscBool PetscEqualReal(PetscReal,PetscReal);
642ce4818fdSLisandro Dalcin PETSC_EXTERN PetscBool PetscEqualScalar(PetscScalar,PetscScalar);
643ce4818fdSLisandro Dalcin 
64498725619SBarry Smith /*
64598725619SBarry Smith     These macros are currently hardwired to match the regular data types, so there is no support for a different
64698725619SBarry Smith     MatScalar from PetscScalar. We left the MatScalar in the source just in case we use it again.
64798725619SBarry Smith  */
64898725619SBarry Smith #define MPIU_MATSCALAR MPIU_SCALAR
64998725619SBarry Smith typedef PetscScalar MatScalar;
65098725619SBarry Smith typedef PetscReal MatReal;
65198725619SBarry Smith 
6528ad47952SJed Brown struct petsc_mpiu_2scalar {PetscScalar a,b;};
6538ad47952SJed Brown PETSC_EXTERN MPI_Datatype MPIU_2SCALAR PetscAttrMPITypeTagLayoutCompatible(struct petsc_mpiu_2scalar);
6548ad47952SJed Brown #if defined(PETSC_USE_64BIT_INDICES) || !defined(MPI_2INT)
6558ad47952SJed Brown struct petsc_mpiu_2int {PetscInt a,b;};
6568ad47952SJed Brown PETSC_EXTERN MPI_Datatype MPIU_2INT PetscAttrMPITypeTagLayoutCompatible(struct petsc_mpiu_2int);
6578ad47952SJed Brown #else
6588ad47952SJed Brown #define MPIU_2INT MPI_2INT
6598ad47952SJed Brown #endif
660e9fa29b7SSatish Balay 
661b2fb0278SBarry Smith PETSC_STATIC_INLINE PetscInt PetscPowInt(PetscInt base,PetscInt power)
662b2fb0278SBarry Smith {
663fa711258SJed Brown   PetscInt result = 1;
664fa711258SJed Brown   while (power) {
665fa711258SJed Brown     if (power & 1) result *= base;
666fa711258SJed Brown     power >>= 1;
667fa711258SJed Brown     base *= base;
668fa711258SJed Brown   }
669fa711258SJed Brown   return result;
670fa711258SJed Brown }
671b2fb0278SBarry Smith 
672b2fb0278SBarry Smith PETSC_STATIC_INLINE PetscReal PetscPowRealInt(PetscReal base,PetscInt power)
673b2fb0278SBarry Smith {
674fa711258SJed Brown   PetscReal result = 1;
675d98d5da7SBarry Smith   if (power < 0) {
676d98d5da7SBarry Smith     power = -power;
67710d40e53SLisandro Dalcin     base  = ((PetscReal)1)/base;
678d98d5da7SBarry Smith   }
679fa711258SJed Brown   while (power) {
680fa711258SJed Brown     if (power & 1) result *= base;
681fa711258SJed Brown     power >>= 1;
682fa711258SJed Brown     base *= base;
683fa711258SJed Brown   }
684fa711258SJed Brown   return result;
685fa711258SJed Brown }
686fa711258SJed Brown 
687b2fb0278SBarry Smith PETSC_STATIC_INLINE PetscScalar PetscPowScalarInt(PetscScalar base,PetscInt power)
688b2fb0278SBarry Smith {
6898b49ba18SBarry Smith   PetscScalar result = 1;
6908b49ba18SBarry Smith   if (power < 0) {
6918b49ba18SBarry Smith     power = -power;
69210d40e53SLisandro Dalcin     base  = ((PetscReal)1)/base;
6938b49ba18SBarry Smith   }
6948b49ba18SBarry Smith   while (power) {
6958b49ba18SBarry Smith     if (power & 1) result *= base;
6968b49ba18SBarry Smith     power >>= 1;
6978b49ba18SBarry Smith     base *= base;
6988b49ba18SBarry Smith   }
6998b49ba18SBarry Smith   return result;
7008b49ba18SBarry Smith }
7018b49ba18SBarry Smith 
702b2fb0278SBarry Smith PETSC_STATIC_INLINE PetscScalar PetscPowScalarReal(PetscScalar base,PetscReal power)
703b2fb0278SBarry Smith {
704b2fb0278SBarry Smith   PetscScalar cpower = power;
705b2fb0278SBarry Smith   return PetscPowScalar(base,cpower);
706b2fb0278SBarry Smith }
70778a59e97SMatthew G. Knepley 
70878a59e97SMatthew G. Knepley #ifndef PETSC_HAVE_LOG2
70978a59e97SMatthew G. Knepley PETSC_STATIC_INLINE PetscReal PetscLog2Real(PetscReal n)
71078a59e97SMatthew G. Knepley {
71191954be4SBarry Smith   return PetscLogReal(n)/PetscLogReal(2.0);
71278a59e97SMatthew G. Knepley }
71378a59e97SMatthew G. Knepley #endif
714e489efc1SBarry Smith #endif
715