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 1126bd1501SBarry Smith #if !defined(PETSCMATH_H) 1226bd1501SBarry Smith #define PETSCMATH_H 130a5f7794SBarry Smith #include <math.h> 14df4397b0SStefano Zampini #include <petscsystypes.h> 15df4397b0SStefano Zampini 165117d392SLisandro Dalcin /* 175117d392SLisandro Dalcin 185117d392SLisandro Dalcin Defines operations that are different for complex and real numbers. 195117d392SLisandro Dalcin All PETSc objects in one program are built around the object 205117d392SLisandro Dalcin PetscScalar which is either always a real or a complex. 215117d392SLisandro Dalcin 225117d392SLisandro Dalcin */ 235117d392SLisandro Dalcin 245117d392SLisandro Dalcin /* 255117d392SLisandro Dalcin Real number definitions 265117d392SLisandro Dalcin */ 275117d392SLisandro Dalcin #if defined(PETSC_USE_REAL_SINGLE) 285117d392SLisandro Dalcin #define PetscSqrtReal(a) sqrtf(a) 295117d392SLisandro Dalcin #define PetscCbrtReal(a) cbrtf(a) 305117d392SLisandro Dalcin #define PetscHypotReal(a,b) hypotf(a,b) 315117d392SLisandro Dalcin #define PetscAtan2Real(a,b) atan2f(a,b) 325117d392SLisandro Dalcin #define PetscPowReal(a,b) powf(a,b) 335117d392SLisandro Dalcin #define PetscExpReal(a) expf(a) 345117d392SLisandro Dalcin #define PetscLogReal(a) logf(a) 355117d392SLisandro Dalcin #define PetscLog10Real(a) log10f(a) 365117d392SLisandro Dalcin #define PetscLog2Real(a) log2f(a) 375117d392SLisandro Dalcin #define PetscSinReal(a) sinf(a) 385117d392SLisandro Dalcin #define PetscCosReal(a) cosf(a) 395117d392SLisandro Dalcin #define PetscTanReal(a) tanf(a) 405117d392SLisandro Dalcin #define PetscAsinReal(a) asinf(a) 415117d392SLisandro Dalcin #define PetscAcosReal(a) acosf(a) 425117d392SLisandro Dalcin #define PetscAtanReal(a) atanf(a) 435117d392SLisandro Dalcin #define PetscSinhReal(a) sinhf(a) 445117d392SLisandro Dalcin #define PetscCoshReal(a) coshf(a) 455117d392SLisandro Dalcin #define PetscTanhReal(a) tanhf(a) 465117d392SLisandro Dalcin #define PetscAsinhReal(a) asinhf(a) 475117d392SLisandro Dalcin #define PetscAcoshReal(a) acoshf(a) 485117d392SLisandro Dalcin #define PetscAtanhReal(a) atanhf(a) 495117d392SLisandro Dalcin #define PetscCeilReal(a) ceilf(a) 505117d392SLisandro Dalcin #define PetscFloorReal(a) floorf(a) 515117d392SLisandro Dalcin #define PetscFmodReal(a,b) fmodf(a,b) 525117d392SLisandro Dalcin #define PetscTGamma(a) tgammaf(a) 531f17fa70SToby Isaac #if defined(PETSC_HAVE_LGAMMA_IS_GAMMA) 541f17fa70SToby Isaac #define PetscLGamma(a) gammaf(a) 551f17fa70SToby Isaac #else 561f17fa70SToby Isaac #define PetscLGamma(a) lgammaf(a) 571f17fa70SToby Isaac #endif 585117d392SLisandro Dalcin 595117d392SLisandro Dalcin #elif defined(PETSC_USE_REAL_DOUBLE) 605117d392SLisandro Dalcin #define PetscSqrtReal(a) sqrt(a) 615117d392SLisandro Dalcin #define PetscCbrtReal(a) cbrt(a) 625117d392SLisandro Dalcin #define PetscHypotReal(a,b) hypot(a,b) 635117d392SLisandro Dalcin #define PetscAtan2Real(a,b) atan2(a,b) 645117d392SLisandro Dalcin #define PetscPowReal(a,b) pow(a,b) 655117d392SLisandro Dalcin #define PetscExpReal(a) exp(a) 665117d392SLisandro Dalcin #define PetscLogReal(a) log(a) 675117d392SLisandro Dalcin #define PetscLog10Real(a) log10(a) 685117d392SLisandro Dalcin #define PetscLog2Real(a) log2(a) 695117d392SLisandro Dalcin #define PetscSinReal(a) sin(a) 705117d392SLisandro Dalcin #define PetscCosReal(a) cos(a) 715117d392SLisandro Dalcin #define PetscTanReal(a) tan(a) 725117d392SLisandro Dalcin #define PetscAsinReal(a) asin(a) 735117d392SLisandro Dalcin #define PetscAcosReal(a) acos(a) 745117d392SLisandro Dalcin #define PetscAtanReal(a) atan(a) 755117d392SLisandro Dalcin #define PetscSinhReal(a) sinh(a) 765117d392SLisandro Dalcin #define PetscCoshReal(a) cosh(a) 775117d392SLisandro Dalcin #define PetscTanhReal(a) tanh(a) 785117d392SLisandro Dalcin #define PetscAsinhReal(a) asinh(a) 795117d392SLisandro Dalcin #define PetscAcoshReal(a) acosh(a) 805117d392SLisandro Dalcin #define PetscAtanhReal(a) atanh(a) 815117d392SLisandro Dalcin #define PetscCeilReal(a) ceil(a) 825117d392SLisandro Dalcin #define PetscFloorReal(a) floor(a) 835117d392SLisandro Dalcin #define PetscFmodReal(a,b) fmod(a,b) 845117d392SLisandro Dalcin #define PetscTGamma(a) tgamma(a) 851f17fa70SToby Isaac #if defined(PETSC_HAVE_LGAMMA_IS_GAMMA) 861f17fa70SToby Isaac #define PetscLGamma(a) gamma(a) 871f17fa70SToby Isaac #else 881f17fa70SToby Isaac #define PetscLGamma(a) lgamma(a) 891f17fa70SToby Isaac #endif 905117d392SLisandro Dalcin 915117d392SLisandro Dalcin #elif defined(PETSC_USE_REAL___FLOAT128) 925117d392SLisandro Dalcin #define PetscSqrtReal(a) sqrtq(a) 935117d392SLisandro Dalcin #define PetscCbrtReal(a) cbrtq(a) 945117d392SLisandro Dalcin #define PetscHypotReal(a,b) hypotq(a,b) 955117d392SLisandro Dalcin #define PetscAtan2Real(a,b) atan2q(a,b) 965117d392SLisandro Dalcin #define PetscPowReal(a,b) powq(a,b) 975117d392SLisandro Dalcin #define PetscExpReal(a) expq(a) 985117d392SLisandro Dalcin #define PetscLogReal(a) logq(a) 995117d392SLisandro Dalcin #define PetscLog10Real(a) log10q(a) 1005117d392SLisandro Dalcin #define PetscLog2Real(a) log2q(a) 1015117d392SLisandro Dalcin #define PetscSinReal(a) sinq(a) 1025117d392SLisandro Dalcin #define PetscCosReal(a) cosq(a) 1035117d392SLisandro Dalcin #define PetscTanReal(a) tanq(a) 1045117d392SLisandro Dalcin #define PetscAsinReal(a) asinq(a) 1055117d392SLisandro Dalcin #define PetscAcosReal(a) acosq(a) 1065117d392SLisandro Dalcin #define PetscAtanReal(a) atanq(a) 1075117d392SLisandro Dalcin #define PetscSinhReal(a) sinhq(a) 1085117d392SLisandro Dalcin #define PetscCoshReal(a) coshq(a) 1095117d392SLisandro Dalcin #define PetscTanhReal(a) tanhq(a) 1105117d392SLisandro Dalcin #define PetscAsinhReal(a) asinhq(a) 1115117d392SLisandro Dalcin #define PetscAcoshReal(a) acoshq(a) 1125117d392SLisandro Dalcin #define PetscAtanhReal(a) atanhq(a) 1135117d392SLisandro Dalcin #define PetscCeilReal(a) ceilq(a) 1145117d392SLisandro Dalcin #define PetscFloorReal(a) floorq(a) 1155117d392SLisandro Dalcin #define PetscFmodReal(a,b) fmodq(a,b) 1165117d392SLisandro Dalcin #define PetscTGamma(a) tgammaq(a) 1171f17fa70SToby Isaac #if defined(PETSC_HAVE_LGAMMA_IS_GAMMA) 1181f17fa70SToby Isaac #define PetscLGamma(a) gammaq(a) 1191f17fa70SToby Isaac #else 1201f17fa70SToby Isaac #define PetscLGamma(a) lgammaq(a) 1211f17fa70SToby Isaac #endif 1225117d392SLisandro Dalcin 1235117d392SLisandro Dalcin #elif defined(PETSC_USE_REAL___FP16) 1245117d392SLisandro Dalcin #define PetscSqrtReal(a) sqrtf(a) 1255117d392SLisandro Dalcin #define PetscCbrtReal(a) cbrtf(a) 1265117d392SLisandro Dalcin #define PetscHypotReal(a,b) hypotf(a,b) 1275117d392SLisandro Dalcin #define PetscAtan2Real(a,b) atan2f(a,b) 1285117d392SLisandro Dalcin #define PetscPowReal(a,b) powf(a,b) 1295117d392SLisandro Dalcin #define PetscExpReal(a) expf(a) 1305117d392SLisandro Dalcin #define PetscLogReal(a) logf(a) 1315117d392SLisandro Dalcin #define PetscLog10Real(a) log10f(a) 1325117d392SLisandro Dalcin #define PetscLog2Real(a) log2f(a) 1335117d392SLisandro Dalcin #define PetscSinReal(a) sinf(a) 1345117d392SLisandro Dalcin #define PetscCosReal(a) cosf(a) 1355117d392SLisandro Dalcin #define PetscTanReal(a) tanf(a) 1365117d392SLisandro Dalcin #define PetscAsinReal(a) asinf(a) 1375117d392SLisandro Dalcin #define PetscAcosReal(a) acosf(a) 1385117d392SLisandro Dalcin #define PetscAtanReal(a) atanf(a) 1395117d392SLisandro Dalcin #define PetscSinhReal(a) sinhf(a) 1405117d392SLisandro Dalcin #define PetscCoshReal(a) coshf(a) 1415117d392SLisandro Dalcin #define PetscTanhReal(a) tanhf(a) 1425117d392SLisandro Dalcin #define PetscAsinhReal(a) asinhf(a) 1435117d392SLisandro Dalcin #define PetscAcoshReal(a) acoshf(a) 1445117d392SLisandro Dalcin #define PetscAtanhReal(a) atanhf(a) 1455117d392SLisandro Dalcin #define PetscCeilReal(a) ceilf(a) 1465117d392SLisandro Dalcin #define PetscFloorReal(a) floorf(a) 1475117d392SLisandro Dalcin #define PetscFmodReal(a,b) fmodf(a,b) 1485117d392SLisandro Dalcin #define PetscTGamma(a) tgammaf(a) 1491f17fa70SToby Isaac #if defined(PETSC_HAVE_LGAMMA_IS_GAMMA) 1501f17fa70SToby Isaac #define PetscLGamma(a) gammaf(a) 1511f17fa70SToby Isaac #else 1521f17fa70SToby Isaac #define PetscLGamma(a) lgammaf(a) 1531f17fa70SToby Isaac #endif 1545117d392SLisandro Dalcin 1555117d392SLisandro Dalcin #endif /* PETSC_USE_REAL_* */ 1565117d392SLisandro Dalcin 1575117d392SLisandro Dalcin PETSC_STATIC_INLINE PetscReal PetscSignReal(PetscReal a) 1585117d392SLisandro Dalcin { 1595117d392SLisandro Dalcin return (PetscReal)((a < (PetscReal)0) ? -1 : ((a > (PetscReal)0) ? 1 : 0)); 1605117d392SLisandro Dalcin } 1615117d392SLisandro Dalcin 1625117d392SLisandro Dalcin #if !defined(PETSC_HAVE_LOG2) 1635117d392SLisandro Dalcin #undef PetscLog2Real 1645117d392SLisandro Dalcin PETSC_STATIC_INLINE PetscReal PetscLog2Real(PetscReal a) 1655117d392SLisandro Dalcin { 1665117d392SLisandro Dalcin return PetscLogReal(a)/PetscLogReal((PetscReal)2); 1675117d392SLisandro Dalcin } 1685117d392SLisandro Dalcin #endif 1695117d392SLisandro Dalcin 1705117d392SLisandro Dalcin #if defined(PETSC_USE_REAL___FLOAT128) 1715117d392SLisandro Dalcin PETSC_EXTERN MPI_Datatype MPIU___FLOAT128 PetscAttrMPITypeTag(__float128); 1725117d392SLisandro Dalcin #endif 1735117d392SLisandro Dalcin #if defined(PETSC_USE_REAL___FP16) 1745117d392SLisandro Dalcin PETSC_EXTERN MPI_Datatype MPIU___FP16 PetscAttrMPITypeTag(__fp16); 1755117d392SLisandro Dalcin #endif 1765117d392SLisandro Dalcin 177df4397b0SStefano Zampini /*MC 178df4397b0SStefano Zampini MPIU_REAL - MPI datatype corresponding to PetscReal 179df4397b0SStefano Zampini 180df4397b0SStefano Zampini Notes: 181df4397b0SStefano Zampini In MPI calls that require an MPI datatype that matches a PetscReal or array of PetscReal values, pass this value. 182df4397b0SStefano Zampini 183df4397b0SStefano Zampini Level: beginner 184df4397b0SStefano Zampini 185df4397b0SStefano Zampini .seealso: PetscReal, PetscScalar, PetscComplex, PetscInt, MPIU_SCALAR, MPIU_COMPLEX, MPIU_INT 186df4397b0SStefano Zampini M*/ 187c1d390e3SJed Brown #if defined(PETSC_USE_REAL_SINGLE) 188c1d390e3SJed Brown # define MPIU_REAL MPI_FLOAT 189c1d390e3SJed Brown #elif defined(PETSC_USE_REAL_DOUBLE) 190c1d390e3SJed Brown # define MPIU_REAL MPI_DOUBLE 191c1d390e3SJed Brown #elif defined(PETSC_USE_REAL___FLOAT128) 192c1d390e3SJed Brown # define MPIU_REAL MPIU___FLOAT128 193570b7f6dSBarry Smith #elif defined(PETSC_USE_REAL___FP16) 194570b7f6dSBarry Smith # define MPIU_REAL MPIU___FP16 195c1d390e3SJed Brown #endif /* PETSC_USE_REAL_* */ 19659cb5930SBarry Smith 1971093a601SBarry Smith /* 1981093a601SBarry Smith Complex number definitions 1991093a601SBarry Smith */ 200df4397b0SStefano Zampini #if defined(PETSC_HAVE_COMPLEX) 2017a19d461SSatish Balay #if defined(__cplusplus) 2021093a601SBarry Smith /* C++ support of complex number */ 203b7940d39SSatish Balay 20450f81f78SJed Brown #define PetscRealPartComplex(a) (a).real() 20550f81f78SJed Brown #define PetscImaginaryPartComplex(a) (a).imag() 206df4397b0SStefano Zampini #define PetscAbsComplex(a) petsccomplexlib::abs(a) 2075117d392SLisandro Dalcin #define PetscArgComplex(a) petsccomplexlib::arg(a) 208df4397b0SStefano Zampini #define PetscConjComplex(a) petsccomplexlib::conj(a) 209df4397b0SStefano Zampini #define PetscSqrtComplex(a) petsccomplexlib::sqrt(a) 210df4397b0SStefano Zampini #define PetscPowComplex(a,b) petsccomplexlib::pow(a,b) 211df4397b0SStefano Zampini #define PetscExpComplex(a) petsccomplexlib::exp(a) 212df4397b0SStefano Zampini #define PetscLogComplex(a) petsccomplexlib::log(a) 213df4397b0SStefano Zampini #define PetscSinComplex(a) petsccomplexlib::sin(a) 214df4397b0SStefano Zampini #define PetscCosComplex(a) petsccomplexlib::cos(a) 2155117d392SLisandro Dalcin #define PetscTanComplex(a) petsccomplexlib::tan(a) 216df4397b0SStefano Zampini #define PetscAsinComplex(a) petsccomplexlib::asin(a) 217df4397b0SStefano Zampini #define PetscAcosComplex(a) petsccomplexlib::acos(a) 2185117d392SLisandro Dalcin #define PetscAtanComplex(a) petsccomplexlib::atan(a) 219df4397b0SStefano Zampini #define PetscSinhComplex(a) petsccomplexlib::sinh(a) 220df4397b0SStefano Zampini #define PetscCoshComplex(a) petsccomplexlib::cosh(a) 221df4397b0SStefano Zampini #define PetscTanhComplex(a) petsccomplexlib::tanh(a) 2225117d392SLisandro Dalcin #define PetscAsinhComplex(a) petsccomplexlib::asinh(a) 2235117d392SLisandro Dalcin #define PetscAcoshComplex(a) petsccomplexlib::acosh(a) 2245117d392SLisandro Dalcin #define PetscAtanhComplex(a) petsccomplexlib::atanh(a) 2255117d392SLisandro Dalcin 2265117d392SLisandro Dalcin /* TODO: Add configure tests 2275117d392SLisandro Dalcin 2285117d392SLisandro Dalcin #if !defined(PETSC_HAVE_CXX_TAN_COMPLEX) 2295117d392SLisandro Dalcin #undef PetscTanComplex 2305117d392SLisandro Dalcin PETSC_STATIC_INLINE PetscComplex PetscTanComplex(PetscComplex z) 2315117d392SLisandro Dalcin { 2325117d392SLisandro Dalcin return PetscSinComplex(z)/PetscCosComplex(z); 2335117d392SLisandro Dalcin } 234027d9794SBarry Smith #endif 235debe9ee2SPaul Mullowney 2365117d392SLisandro Dalcin #if !defined(PETSC_HAVE_CXX_TANH_COMPLEX) 2375117d392SLisandro Dalcin #undef PetscTanhComplex 2385117d392SLisandro Dalcin PETSC_STATIC_INLINE PetscComplex PetscTanhComplex(PetscComplex z) 2395117d392SLisandro Dalcin { 2405117d392SLisandro Dalcin return PetscSinhComplex(z)/PetscCoshComplex(z); 2415117d392SLisandro Dalcin } 2425117d392SLisandro Dalcin #endif 2435117d392SLisandro Dalcin 2445117d392SLisandro Dalcin #if !defined(PETSC_HAVE_CXX_ASIN_COMPLEX) 2455117d392SLisandro Dalcin #undef PetscAsinComplex 2465117d392SLisandro Dalcin PETSC_STATIC_INLINE PetscComplex PetscAsinComplex(PetscComplex z) 2475117d392SLisandro Dalcin { 2485117d392SLisandro Dalcin const PetscComplex j(0,1); 2495117d392SLisandro Dalcin return -j*PetscLogComplex(j*z+PetscSqrtComplex(1.0f-z*z)); 2505117d392SLisandro Dalcin } 2515117d392SLisandro Dalcin #endif 2525117d392SLisandro Dalcin 2535117d392SLisandro Dalcin #if !defined(PETSC_HAVE_CXX_ACOS_COMPLEX) 2545117d392SLisandro Dalcin #undef PetscAcosComplex 2555117d392SLisandro Dalcin PETSC_STATIC_INLINE PetscComplex PetscAcosComplex(PetscComplex z) 2565117d392SLisandro Dalcin { 2575117d392SLisandro Dalcin const PetscComplex j(0,1); 2585117d392SLisandro Dalcin return j*PetscLogComplex(z-j*PetscSqrtComplex(1.0f-z*z)); 2595117d392SLisandro Dalcin } 2605117d392SLisandro Dalcin #endif 2615117d392SLisandro Dalcin 2625117d392SLisandro Dalcin #if !defined(PETSC_HAVE_CXX_ATAN_COMPLEX) 2635117d392SLisandro Dalcin #undef PetscAtanComplex 2645117d392SLisandro Dalcin PETSC_STATIC_INLINE PetscComplex PetscAtanComplex(PetscComplex z) 2655117d392SLisandro Dalcin { 2665117d392SLisandro Dalcin const PetscComplex j(0,1); 2675117d392SLisandro Dalcin return 0.5f*j*PetscLogComplex((1.0f-j*z)/(1.0f+j*z)); 2685117d392SLisandro Dalcin } 2695117d392SLisandro Dalcin #endif 2705117d392SLisandro Dalcin 2715117d392SLisandro Dalcin #if !defined(PETSC_HAVE_CXX_ASINH_COMPLEX) 2725117d392SLisandro Dalcin #undef PetscAsinhComplex 2735117d392SLisandro Dalcin PETSC_STATIC_INLINE PetscComplex PetscAsinhComplex(PetscComplex z) 2745117d392SLisandro Dalcin { 2755117d392SLisandro Dalcin return PetscLogComplex(z+PetscSqrtComplex(z*z+1.0f)); 2765117d392SLisandro Dalcin } 2775117d392SLisandro Dalcin #endif 2785117d392SLisandro Dalcin 2795117d392SLisandro Dalcin #if !defined(PETSC_HAVE_CXX_ACOSH_COMPLEX) 2805117d392SLisandro Dalcin #undef PetscAcoshComplex 2815117d392SLisandro Dalcin PETSC_STATIC_INLINE PetscComplex PetscAcoshComplex(PetscComplex z) 2825117d392SLisandro Dalcin { 2835117d392SLisandro Dalcin return PetscLogComplex(z+PetscSqrtComplex(z*z-1.0f)); 2845117d392SLisandro Dalcin } 2855117d392SLisandro Dalcin #endif 2865117d392SLisandro Dalcin 2875117d392SLisandro Dalcin #if !defined(PETSC_HAVE_CXX_ATANH_COMPLEX) 2885117d392SLisandro Dalcin #undef PetscAtanhComplex 2895117d392SLisandro Dalcin PETSC_STATIC_INLINE PetscComplex PetscAtanhComplex(PetscComplex z) 2905117d392SLisandro Dalcin { 2915117d392SLisandro Dalcin return 0.5f*PetscLogComplex((1.0f+z)/(1.0f-z)); 2925117d392SLisandro Dalcin } 2935117d392SLisandro Dalcin #endif 2945117d392SLisandro Dalcin 2955117d392SLisandro Dalcin */ 2965117d392SLisandro Dalcin 2977a19d461SSatish Balay #else /* C99 support of complex number */ 298519e2a1fSPaul Mullowney 2997a19d461SSatish Balay #if defined(PETSC_USE_REAL_SINGLE) 30050f81f78SJed Brown #define PetscRealPartComplex(a) crealf(a) 30150f81f78SJed Brown #define PetscImaginaryPartComplex(a) cimagf(a) 30250f81f78SJed Brown #define PetscAbsComplex(a) cabsf(a) 3035117d392SLisandro Dalcin #define PetscArgComplex(a) cargf(a) 30450f81f78SJed Brown #define PetscConjComplex(a) conjf(a) 30550f81f78SJed Brown #define PetscSqrtComplex(a) csqrtf(a) 30650f81f78SJed Brown #define PetscPowComplex(a,b) cpowf(a,b) 30750f81f78SJed Brown #define PetscExpComplex(a) cexpf(a) 30850f81f78SJed Brown #define PetscLogComplex(a) clogf(a) 30950f81f78SJed Brown #define PetscSinComplex(a) csinf(a) 31050f81f78SJed Brown #define PetscCosComplex(a) ccosf(a) 3115117d392SLisandro Dalcin #define PetscTanComplex(a) ctanf(a) 312255453a1SBarry Smith #define PetscAsinComplex(a) casinf(a) 313255453a1SBarry Smith #define PetscAcosComplex(a) cacosf(a) 3145117d392SLisandro Dalcin #define PetscAtanComplex(a) catanf(a) 315a4bea5a6SPeter Brune #define PetscSinhComplex(a) csinhf(a) 316a4bea5a6SPeter Brune #define PetscCoshComplex(a) ccoshf(a) 317a4bea5a6SPeter Brune #define PetscTanhComplex(a) ctanhf(a) 3185117d392SLisandro Dalcin #define PetscAsinhComplex(a) casinhf(a) 3195117d392SLisandro Dalcin #define PetscAcoshComplex(a) cacoshf(a) 3205117d392SLisandro Dalcin #define PetscAtanhComplex(a) catanhf(a) 3211093a601SBarry Smith 322ce63c4c1SBarry Smith #elif defined(PETSC_USE_REAL_DOUBLE) 32350f81f78SJed Brown #define PetscRealPartComplex(a) creal(a) 32450f81f78SJed Brown #define PetscImaginaryPartComplex(a) cimag(a) 32550f81f78SJed Brown #define PetscAbsComplex(a) cabs(a) 3265117d392SLisandro Dalcin #define PetscArgComplex(a) carg(a) 32750f81f78SJed Brown #define PetscConjComplex(a) conj(a) 32850f81f78SJed Brown #define PetscSqrtComplex(a) csqrt(a) 32950f81f78SJed Brown #define PetscPowComplex(a,b) cpow(a,b) 33050f81f78SJed Brown #define PetscExpComplex(a) cexp(a) 33150f81f78SJed Brown #define PetscLogComplex(a) clog(a) 33250f81f78SJed Brown #define PetscSinComplex(a) csin(a) 33350f81f78SJed Brown #define PetscCosComplex(a) ccos(a) 3345117d392SLisandro Dalcin #define PetscTanComplex(a) ctan(a) 335255453a1SBarry Smith #define PetscAsinComplex(a) casin(a) 336255453a1SBarry Smith #define PetscAcosComplex(a) cacos(a) 3375117d392SLisandro Dalcin #define PetscAtanComplex(a) catan(a) 338a4bea5a6SPeter Brune #define PetscSinhComplex(a) csinh(a) 339a4bea5a6SPeter Brune #define PetscCoshComplex(a) ccosh(a) 340a4bea5a6SPeter Brune #define PetscTanhComplex(a) ctanh(a) 3415117d392SLisandro Dalcin #define PetscAsinhComplex(a) casinh(a) 3425117d392SLisandro Dalcin #define PetscAcoshComplex(a) cacosh(a) 3435117d392SLisandro Dalcin #define PetscAtanhComplex(a) catanh(a) 3441093a601SBarry Smith 3458c764dc5SJose Roman #elif defined(PETSC_USE_REAL___FLOAT128) 34650f81f78SJed Brown #define PetscRealPartComplex(a) crealq(a) 34750f81f78SJed Brown #define PetscImaginaryPartComplex(a) cimagq(a) 34850f81f78SJed Brown #define PetscAbsComplex(a) cabsq(a) 3495117d392SLisandro Dalcin #define PetscArgComplex(a) cargq(a) 35050f81f78SJed Brown #define PetscConjComplex(a) conjq(a) 35150f81f78SJed Brown #define PetscSqrtComplex(a) csqrtq(a) 35250f81f78SJed Brown #define PetscPowComplex(a,b) cpowq(a,b) 35350f81f78SJed Brown #define PetscExpComplex(a) cexpq(a) 35450f81f78SJed Brown #define PetscLogComplex(a) clogq(a) 35550f81f78SJed Brown #define PetscSinComplex(a) csinq(a) 35650f81f78SJed Brown #define PetscCosComplex(a) ccosq(a) 3575117d392SLisandro Dalcin #define PetscTanComplex(a) ctanq(a) 358255453a1SBarry Smith #define PetscAsinComplex(a) casinq(a) 359255453a1SBarry Smith #define PetscAcosComplex(a) cacosq(a) 3605117d392SLisandro Dalcin #define PetscAtanComplex(a) catanq(a) 361a4bea5a6SPeter Brune #define PetscSinhComplex(a) csinhq(a) 362a4bea5a6SPeter Brune #define PetscCoshComplex(a) ccoshq(a) 363a4bea5a6SPeter Brune #define PetscTanhComplex(a) ctanhq(a) 3645117d392SLisandro Dalcin #define PetscAsinhComplex(a) casinhq(a) 3655117d392SLisandro Dalcin #define PetscAcoshComplex(a) cacoshq(a) 3665117d392SLisandro Dalcin #define PetscAtanhComplex(a) catanhq(a) 367a4bea5a6SPeter Brune 368ce63c4c1SBarry Smith #endif /* PETSC_USE_REAL_* */ 3697a19d461SSatish Balay #endif /* (__cplusplus) */ 370e489efc1SBarry Smith 3711093a601SBarry Smith /* 3725117d392SLisandro Dalcin PETSC_i is the imaginary number, i 3731093a601SBarry Smith */ 37450f81f78SJed Brown PETSC_EXTERN PetscComplex PETSC_i; 3758a351411SToby Isaac 3765117d392SLisandro Dalcin /* 3775117d392SLisandro Dalcin Try to do the right thing for complex number construction: see 3788a351411SToby Isaac http://www.open-std.org/jtc1/sc22/wg14/www/docs/n1464.htm 3798a351411SToby Isaac for details 3808a351411SToby Isaac */ 38119e222d7SToby Isaac PETSC_STATIC_INLINE PetscComplex PetscCMPLX(PetscReal x, PetscReal y) 3828a351411SToby Isaac { 3837a19d461SSatish Balay #if defined(__cplusplus) 3848a351411SToby Isaac return PetscComplex(x,y); 3858a351411SToby Isaac #elif defined(_Imaginary_I) 3868a351411SToby Isaac return x + y * _Imaginary_I; 3878a351411SToby Isaac #else 388616d7c5eSToby Isaac { /* In both C99 and C11 (ISO/IEC 9899, Section 6.2.5), 389616d7c5eSToby Isaac 390616d7c5eSToby Isaac "For each floating type there is a corresponding real type, which is always a real floating 391616d7c5eSToby Isaac type. For real floating types, it is the same type. For complex types, it is the type given 392616d7c5eSToby Isaac by deleting the keyword _Complex from the type name." 393616d7c5eSToby Isaac 394616d7c5eSToby Isaac So type punning should be portable. */ 395616d7c5eSToby Isaac union { PetscComplex z; PetscReal f[2]; } uz; 396616d7c5eSToby Isaac 397616d7c5eSToby Isaac uz.f[0] = x; 398616d7c5eSToby Isaac uz.f[1] = y; 399616d7c5eSToby Isaac return uz.z; 400616d7c5eSToby Isaac } 40150f81f78SJed Brown #endif 4028a351411SToby Isaac } 4038a351411SToby Isaac 404*de272c7aSSatish Balay #define MPIU_C_COMPLEX MPI_C_COMPLEX PETSC_DEPRECATED_MACRO("GCC warning \"MPIU_C_COMPLEX macro is deprecated use MPI_C_COMPLEX (since version 3.15)\"") 405*de272c7aSSatish Balay #define MPIU_C_DOUBLE_COMPLEX MPI_C_DOUBLE_COMPLEX PETSC_DEPRECATED_MACRO("GCC warning \"MPIU_C_DOUBLE_COMPLEX macro is deprecated use MPI_C_DOUBLE_COMPLEX (since version 3.15)\"") 406*de272c7aSSatish Balay 4075117d392SLisandro Dalcin #if defined(PETSC_USE_REAL___FLOAT128) 4085117d392SLisandro Dalcin PETSC_EXTERN MPI_Datatype MPIU___COMPLEX128 PetscAttrMPITypeTag(__complex128); 4095117d392SLisandro Dalcin #endif /* PETSC_USE_REAL___FLOAT128 */ 4105117d392SLisandro Dalcin 4115117d392SLisandro Dalcin /*MC 4125117d392SLisandro Dalcin MPIU_COMPLEX - MPI datatype corresponding to PetscComplex 4135117d392SLisandro Dalcin 4145117d392SLisandro Dalcin Notes: 4155117d392SLisandro Dalcin In MPI calls that require an MPI datatype that matches a PetscComplex or array of PetscComplex values, pass this value. 4165117d392SLisandro Dalcin 4175117d392SLisandro Dalcin Level: beginner 4185117d392SLisandro Dalcin 4195117d392SLisandro Dalcin .seealso: PetscReal, PetscScalar, PetscComplex, PetscInt, MPIU_REAL, MPIU_SCALAR, MPIU_COMPLEX, MPIU_INT, PETSC_i 4205117d392SLisandro Dalcin M*/ 4215117d392SLisandro Dalcin #if defined(PETSC_USE_REAL_SINGLE) 422*de272c7aSSatish Balay # define MPIU_COMPLEX MPI_C_COMPLEX 4235117d392SLisandro Dalcin #elif defined(PETSC_USE_REAL_DOUBLE) 424*de272c7aSSatish Balay # define MPIU_COMPLEX MPI_C_DOUBLE_COMPLEX 4255117d392SLisandro Dalcin #elif defined(PETSC_USE_REAL___FLOAT128) 4265117d392SLisandro Dalcin # define MPIU_COMPLEX MPIU___COMPLEX128 4275117d392SLisandro Dalcin #elif defined(PETSC_USE_REAL___FP16) 428*de272c7aSSatish Balay # define MPIU_COMPLEX MPI_C_COMPLEX 4295117d392SLisandro Dalcin #endif /* PETSC_USE_REAL_* */ 4305117d392SLisandro Dalcin 4315117d392SLisandro Dalcin #endif /* PETSC_HAVE_COMPLEX */ 4325117d392SLisandro Dalcin 4335117d392SLisandro Dalcin /* 4345117d392SLisandro Dalcin Scalar number definitions 4355117d392SLisandro Dalcin */ 4367a19d461SSatish Balay #if defined(PETSC_USE_COMPLEX) && defined(PETSC_HAVE_COMPLEX) 4375117d392SLisandro Dalcin /*MC 4385117d392SLisandro Dalcin MPIU_SCALAR - MPI datatype corresponding to PetscScalar 4395117d392SLisandro Dalcin 4405117d392SLisandro Dalcin Notes: 4415117d392SLisandro Dalcin In MPI calls that require an MPI datatype that matches a PetscScalar or array of PetscScalar values, pass this value. 4425117d392SLisandro Dalcin 4435117d392SLisandro Dalcin Level: beginner 4445117d392SLisandro Dalcin 4455117d392SLisandro Dalcin .seealso: PetscReal, PetscScalar, PetscComplex, PetscInt, MPIU_REAL, MPIU_COMPLEX, MPIU_INT 4465117d392SLisandro Dalcin M*/ 4475117d392SLisandro Dalcin #define MPIU_SCALAR MPIU_COMPLEX 4485117d392SLisandro Dalcin 4495117d392SLisandro Dalcin /*MC 4505117d392SLisandro Dalcin PetscRealPart - Returns the real part of a PetscScalar 4515117d392SLisandro Dalcin 4525117d392SLisandro Dalcin Synopsis: 4535117d392SLisandro Dalcin #include <petscmath.h> 4545117d392SLisandro Dalcin PetscReal PetscRealPart(PetscScalar v) 4555117d392SLisandro Dalcin 4565117d392SLisandro Dalcin Not Collective 4575117d392SLisandro Dalcin 4585117d392SLisandro Dalcin Input Parameter: 4595117d392SLisandro Dalcin . v - value to find the real part of 4605117d392SLisandro Dalcin 4615117d392SLisandro Dalcin Level: beginner 4625117d392SLisandro Dalcin 4635117d392SLisandro Dalcin .seealso: PetscScalar, PetscImaginaryPart(), PetscMax(), PetscClipInterval(), PetscAbsInt(), PetscAbsReal(), PetscSqr() 4645117d392SLisandro Dalcin 4655117d392SLisandro Dalcin M*/ 4665117d392SLisandro Dalcin #define PetscRealPart(a) PetscRealPartComplex(a) 4675117d392SLisandro Dalcin 4685117d392SLisandro Dalcin /*MC 4695117d392SLisandro Dalcin PetscImaginaryPart - Returns the imaginary part of a PetscScalar 4705117d392SLisandro Dalcin 4715117d392SLisandro Dalcin Synopsis: 4725117d392SLisandro Dalcin #include <petscmath.h> 4735117d392SLisandro Dalcin PetscReal PetscImaginaryPart(PetscScalar v) 4745117d392SLisandro Dalcin 4755117d392SLisandro Dalcin Not Collective 4765117d392SLisandro Dalcin 4775117d392SLisandro Dalcin Input Parameter: 4785117d392SLisandro Dalcin . v - value to find the imaginary part of 4795117d392SLisandro Dalcin 4805117d392SLisandro Dalcin Level: beginner 4815117d392SLisandro Dalcin 4825117d392SLisandro Dalcin Notes: 4835117d392SLisandro Dalcin If PETSc was configured for real numbers then this always returns the value 0 4845117d392SLisandro Dalcin 4855117d392SLisandro Dalcin .seealso: PetscScalar, PetscRealPart(), PetscMax(), PetscClipInterval(), PetscAbsInt(), PetscAbsReal(), PetscSqr() 4865117d392SLisandro Dalcin 4875117d392SLisandro Dalcin M*/ 4885117d392SLisandro Dalcin #define PetscImaginaryPart(a) PetscImaginaryPartComplex(a) 4895117d392SLisandro Dalcin 4905117d392SLisandro Dalcin #define PetscAbsScalar(a) PetscAbsComplex(a) 4915117d392SLisandro Dalcin #define PetscArgScalar(a) PetscArgComplex(a) 4925117d392SLisandro Dalcin #define PetscConj(a) PetscConjComplex(a) 4935117d392SLisandro Dalcin #define PetscSqrtScalar(a) PetscSqrtComplex(a) 4945117d392SLisandro Dalcin #define PetscPowScalar(a,b) PetscPowComplex(a,b) 4955117d392SLisandro Dalcin #define PetscExpScalar(a) PetscExpComplex(a) 4965117d392SLisandro Dalcin #define PetscLogScalar(a) PetscLogComplex(a) 4975117d392SLisandro Dalcin #define PetscSinScalar(a) PetscSinComplex(a) 4985117d392SLisandro Dalcin #define PetscCosScalar(a) PetscCosComplex(a) 4995117d392SLisandro Dalcin #define PetscTanScalar(a) PetscTanComplex(a) 5005117d392SLisandro Dalcin #define PetscAsinScalar(a) PetscAsinComplex(a) 5015117d392SLisandro Dalcin #define PetscAcosScalar(a) PetscAcosComplex(a) 5025117d392SLisandro Dalcin #define PetscAtanScalar(a) PetscAtanComplex(a) 5035117d392SLisandro Dalcin #define PetscSinhScalar(a) PetscSinhComplex(a) 5045117d392SLisandro Dalcin #define PetscCoshScalar(a) PetscCoshComplex(a) 5055117d392SLisandro Dalcin #define PetscTanhScalar(a) PetscTanhComplex(a) 5065117d392SLisandro Dalcin #define PetscAsinhScalar(a) PetscAsinhComplex(a) 5075117d392SLisandro Dalcin #define PetscAcoshScalar(a) PetscAcoshComplex(a) 5085117d392SLisandro Dalcin #define PetscAtanhScalar(a) PetscAtanhComplex(a) 5095117d392SLisandro Dalcin 5105117d392SLisandro Dalcin #else /* PETSC_USE_COMPLEX */ 5115117d392SLisandro Dalcin #define MPIU_SCALAR MPIU_REAL 5125117d392SLisandro Dalcin #define PetscRealPart(a) (a) 5135117d392SLisandro Dalcin #define PetscImaginaryPart(a) ((PetscReal)0) 5145117d392SLisandro Dalcin #define PetscAbsScalar(a) PetscAbsReal(a) 5155117d392SLisandro Dalcin #define PetscArgScalar(a) (((a) < (PetscReal)0) ? PETSC_PI : (PetscReal)0) 5165117d392SLisandro Dalcin #define PetscConj(a) (a) 5175117d392SLisandro Dalcin #define PetscSqrtScalar(a) PetscSqrtReal(a) 5185117d392SLisandro Dalcin #define PetscPowScalar(a,b) PetscPowReal(a,b) 5195117d392SLisandro Dalcin #define PetscExpScalar(a) PetscExpReal(a) 5205117d392SLisandro Dalcin #define PetscLogScalar(a) PetscLogReal(a) 5215117d392SLisandro Dalcin #define PetscSinScalar(a) PetscSinReal(a) 5225117d392SLisandro Dalcin #define PetscCosScalar(a) PetscCosReal(a) 5235117d392SLisandro Dalcin #define PetscTanScalar(a) PetscTanReal(a) 5245117d392SLisandro Dalcin #define PetscAsinScalar(a) PetscAsinReal(a) 5255117d392SLisandro Dalcin #define PetscAcosScalar(a) PetscAcosReal(a) 5265117d392SLisandro Dalcin #define PetscAtanScalar(a) PetscAtanReal(a) 5275117d392SLisandro Dalcin #define PetscSinhScalar(a) PetscSinhReal(a) 5285117d392SLisandro Dalcin #define PetscCoshScalar(a) PetscCoshReal(a) 5295117d392SLisandro Dalcin #define PetscTanhScalar(a) PetscTanhReal(a) 5305117d392SLisandro Dalcin #define PetscAsinhScalar(a) PetscAsinhReal(a) 5315117d392SLisandro Dalcin #define PetscAcoshScalar(a) PetscAcoshReal(a) 5325117d392SLisandro Dalcin #define PetscAtanhScalar(a) PetscAtanhReal(a) 5335117d392SLisandro Dalcin 5345117d392SLisandro Dalcin #endif /* PETSC_USE_COMPLEX */ 5355117d392SLisandro Dalcin 5365117d392SLisandro Dalcin /* 5375117d392SLisandro Dalcin Certain objects may be created using either single or double precision. 5385117d392SLisandro Dalcin This is currently not used. 5395117d392SLisandro Dalcin */ 5405117d392SLisandro Dalcin typedef enum { PETSC_SCALAR_DOUBLE, PETSC_SCALAR_SINGLE, PETSC_SCALAR_LONG_DOUBLE, PETSC_SCALAR_HALF } PetscScalarPrecision; 5415117d392SLisandro Dalcin 5425117d392SLisandro Dalcin /* --------------------------------------------------------------------------*/ 5435117d392SLisandro Dalcin 5445117d392SLisandro Dalcin /*MC 5455117d392SLisandro Dalcin PetscAbs - Returns the absolute value of a number 5465117d392SLisandro Dalcin 5475117d392SLisandro Dalcin Synopsis: 5485117d392SLisandro Dalcin #include <petscmath.h> 5495117d392SLisandro Dalcin type PetscAbs(type v) 5505117d392SLisandro Dalcin 5515117d392SLisandro Dalcin Not Collective 5525117d392SLisandro Dalcin 5535117d392SLisandro Dalcin Input Parameter: 5545117d392SLisandro Dalcin . v - the number 5555117d392SLisandro Dalcin 5565117d392SLisandro Dalcin Notes: 5575117d392SLisandro Dalcin type can be integer or real floating point value 5585117d392SLisandro Dalcin 5595117d392SLisandro Dalcin Level: beginner 5605117d392SLisandro Dalcin 5615117d392SLisandro Dalcin .seealso: PetscAbsInt(), PetscAbsReal(), PetscAbsScalar() 5625117d392SLisandro Dalcin 5635117d392SLisandro Dalcin M*/ 5645117d392SLisandro Dalcin #define PetscAbs(a) (((a) >= 0) ? (a) : (-(a))) 5655117d392SLisandro Dalcin 5665117d392SLisandro Dalcin /*MC 5675117d392SLisandro Dalcin PetscSign - Returns the sign of a number as an integer 5685117d392SLisandro Dalcin 5695117d392SLisandro Dalcin Synopsis: 5705117d392SLisandro Dalcin #include <petscmath.h> 5715117d392SLisandro Dalcin int PetscSign(type v) 5725117d392SLisandro Dalcin 5735117d392SLisandro Dalcin Not Collective 5745117d392SLisandro Dalcin 5755117d392SLisandro Dalcin Input Parameter: 5765117d392SLisandro Dalcin . v - the number 5775117d392SLisandro Dalcin 5785117d392SLisandro Dalcin Notes: 5795117d392SLisandro Dalcin type can be integer or real floating point value 5805117d392SLisandro Dalcin 5815117d392SLisandro Dalcin Level: beginner 5825117d392SLisandro Dalcin 5835117d392SLisandro Dalcin M*/ 5845117d392SLisandro Dalcin #define PetscSign(a) (((a) >= 0) ? ((a) == 0 ? 0 : 1) : -1) 585e489efc1SBarry Smith 586b6a5bde7SBarry Smith /*MC 587b6a5bde7SBarry Smith PetscMin - Returns minimum of two numbers 588b6a5bde7SBarry Smith 589eca87e8dSBarry Smith Synopsis: 590aaa7dc30SBarry Smith #include <petscmath.h> 591eca87e8dSBarry Smith type PetscMin(type v1,type v2) 592eca87e8dSBarry Smith 593eca87e8dSBarry Smith Not Collective 594eca87e8dSBarry Smith 595b6a5bde7SBarry Smith Input Parameter: 596b6a5bde7SBarry Smith + v1 - first value to find minimum of 597b6a5bde7SBarry Smith - v2 - second value to find minimum of 598b6a5bde7SBarry Smith 59995452b02SPatrick Sanan Notes: 60095452b02SPatrick Sanan type can be integer or floating point value 601b6a5bde7SBarry Smith 602b6a5bde7SBarry Smith Level: beginner 603b6a5bde7SBarry Smith 6041175f9beSHong Zhang .seealso: PetscMax(), PetscClipInterval(), PetscAbsInt(), PetscAbsReal(), PetscSqr() 605b6a5bde7SBarry Smith 606b6a5bde7SBarry Smith M*/ 607e489efc1SBarry Smith #define PetscMin(a,b) (((a)<(b)) ? (a) : (b)) 608b6a5bde7SBarry Smith 609b6a5bde7SBarry Smith /*MC 610b6a5bde7SBarry Smith PetscMax - Returns maxium of two numbers 611b6a5bde7SBarry Smith 612eca87e8dSBarry Smith Synopsis: 613aaa7dc30SBarry Smith #include <petscmath.h> 614eca87e8dSBarry Smith type max PetscMax(type v1,type v2) 615eca87e8dSBarry Smith 616eca87e8dSBarry Smith Not Collective 617eca87e8dSBarry Smith 618b6a5bde7SBarry Smith Input Parameter: 619b6a5bde7SBarry Smith + v1 - first value to find maximum of 620b6a5bde7SBarry Smith - v2 - second value to find maximum of 621b6a5bde7SBarry Smith 62295452b02SPatrick Sanan Notes: 62395452b02SPatrick Sanan type can be integer or floating point value 624b6a5bde7SBarry Smith 625b6a5bde7SBarry Smith Level: beginner 626b6a5bde7SBarry Smith 627d9a4bb16SJed Brown .seealso: PetscMin(), PetscClipInterval(), PetscAbsInt(), PetscAbsReal(), PetscSqr() 628b6a5bde7SBarry Smith 629b6a5bde7SBarry Smith M*/ 630e489efc1SBarry Smith #define PetscMax(a,b) (((a)<(b)) ? (b) : (a)) 631b6a5bde7SBarry Smith 632b6a5bde7SBarry Smith /*MC 633d9a4bb16SJed Brown PetscClipInterval - Returns a number clipped to be within an interval 634d9a4bb16SJed Brown 635d9a4bb16SJed Brown Synopsis: 636aaa7dc30SBarry Smith #include <petscmath.h> 637d9a4bb16SJed Brown type clip PetscClipInterval(type x,type a,type b) 638d9a4bb16SJed Brown 639d9a4bb16SJed Brown Not Collective 640d9a4bb16SJed Brown 641d9a4bb16SJed Brown Input Parameter: 6420d398bfeSStefano Zampini + x - value to use if within interval [a,b] 643d9a4bb16SJed Brown . a - lower end of interval 644d9a4bb16SJed Brown - b - upper end of interval 645d9a4bb16SJed Brown 64695452b02SPatrick Sanan Notes: 64795452b02SPatrick Sanan type can be integer or floating point value 648d9a4bb16SJed Brown 649d9a4bb16SJed Brown Level: beginner 650d9a4bb16SJed Brown 651d9a4bb16SJed Brown .seealso: PetscMin(), PetscMax(), PetscAbsInt(), PetscAbsReal(), PetscSqr() 652d9a4bb16SJed Brown 653d9a4bb16SJed Brown M*/ 654d9a4bb16SJed Brown #define PetscClipInterval(x,a,b) (PetscMax((a),PetscMin((x),(b)))) 655d9a4bb16SJed Brown 656d9a4bb16SJed Brown /*MC 657b6a5bde7SBarry Smith PetscAbsInt - Returns the absolute value of an integer 658b6a5bde7SBarry Smith 659b6a5bde7SBarry Smith Synopsis: 660aaa7dc30SBarry Smith #include <petscmath.h> 661b6a5bde7SBarry Smith int abs PetscAbsInt(int v1) 662b6a5bde7SBarry Smith 663eca87e8dSBarry Smith Not Collective 664eca87e8dSBarry Smith 665eca87e8dSBarry Smith Input Parameter: 666eca87e8dSBarry Smith . v1 - the integer 667b6a5bde7SBarry Smith 668b6a5bde7SBarry Smith Level: beginner 669b6a5bde7SBarry Smith 670b6a5bde7SBarry Smith .seealso: PetscMax(), PetscMin(), PetscAbsReal(), PetscSqr() 671b6a5bde7SBarry Smith 672b6a5bde7SBarry Smith M*/ 6739fa7d148SSatish Balay #define PetscAbsInt(a) (((a)<0) ? (-(a)) : (a)) 674b6a5bde7SBarry Smith 675b6a5bde7SBarry Smith /*MC 676b6a5bde7SBarry Smith PetscAbsReal - Returns the absolute value of an real number 677b6a5bde7SBarry Smith 678eca87e8dSBarry Smith Synopsis: 679aaa7dc30SBarry Smith #include <petscmath.h> 680eca87e8dSBarry Smith Real abs PetscAbsReal(PetscReal v1) 681eca87e8dSBarry Smith 682eca87e8dSBarry Smith Not Collective 683eca87e8dSBarry Smith 684b6a5bde7SBarry Smith Input Parameter: 685b6a5bde7SBarry Smith . v1 - the double 686b6a5bde7SBarry Smith 687b6a5bde7SBarry Smith 688b6a5bde7SBarry Smith Level: beginner 689b6a5bde7SBarry Smith 690b6a5bde7SBarry Smith .seealso: PetscMax(), PetscMin(), PetscAbsInt(), PetscSqr() 691b6a5bde7SBarry Smith 692b6a5bde7SBarry Smith M*/ 6931118d4bcSLisandro Dalcin #if defined(PETSC_USE_REAL_SINGLE) 6941118d4bcSLisandro Dalcin #define PetscAbsReal(a) fabsf(a) 6951118d4bcSLisandro Dalcin #elif defined(PETSC_USE_REAL_DOUBLE) 6961118d4bcSLisandro Dalcin #define PetscAbsReal(a) fabs(a) 6971118d4bcSLisandro Dalcin #elif defined(PETSC_USE_REAL___FLOAT128) 6981118d4bcSLisandro Dalcin #define PetscAbsReal(a) fabsq(a) 6991118d4bcSLisandro Dalcin #elif defined(PETSC_USE_REAL___FP16) 7001118d4bcSLisandro Dalcin #define PetscAbsReal(a) fabsf(a) 7011118d4bcSLisandro Dalcin #endif 702b6a5bde7SBarry Smith 703b6a5bde7SBarry Smith /*MC 704b6a5bde7SBarry Smith PetscSqr - Returns the square of a number 705b6a5bde7SBarry Smith 706b6a5bde7SBarry Smith Synopsis: 707aaa7dc30SBarry Smith #include <petscmath.h> 708b6a5bde7SBarry Smith type sqr PetscSqr(type v1) 709b6a5bde7SBarry Smith 710eca87e8dSBarry Smith Not Collective 711eca87e8dSBarry Smith 712eca87e8dSBarry Smith Input Parameter: 713eca87e8dSBarry Smith . v1 - the value 714eca87e8dSBarry Smith 71595452b02SPatrick Sanan Notes: 71695452b02SPatrick Sanan type can be integer or floating point value 717b6a5bde7SBarry Smith 718b6a5bde7SBarry Smith Level: beginner 719b6a5bde7SBarry Smith 720b6a5bde7SBarry Smith .seealso: PetscMax(), PetscMin(), PetscAbsInt(), PetscAbsReal() 721b6a5bde7SBarry Smith 722b6a5bde7SBarry Smith M*/ 7234ebda54eSMatthew Knepley #define PetscSqr(a) ((a)*(a)) 724e489efc1SBarry Smith 725314da920SBarry Smith /* ----------------------------------------------------------------------------*/ 726ee223c85SLisandro Dalcin 727ee223c85SLisandro Dalcin #if defined(PETSC_USE_REAL_SINGLE) 728ee223c85SLisandro Dalcin #define PetscRealConstant(constant) constant##F 7295117d392SLisandro Dalcin #elif defined(PETSC_USE_REAL_DOUBLE) 7305117d392SLisandro Dalcin #define PetscRealConstant(constant) constant 731ee223c85SLisandro Dalcin #elif defined(PETSC_USE_REAL___FLOAT128) 732ee223c85SLisandro Dalcin #define PetscRealConstant(constant) constant##Q 7335117d392SLisandro Dalcin #elif defined(PETSC_USE_REAL___FP16) 7345117d392SLisandro Dalcin #define PetscRealConstant(constant) constant##F 735ee223c85SLisandro Dalcin #endif 736ee223c85SLisandro Dalcin 737314da920SBarry Smith /* 738d34fcf5fSBarry Smith Basic constants 739314da920SBarry Smith */ 7402fab75feSLisandro Dalcin #define PETSC_PI PetscRealConstant(3.1415926535897932384626433832795029) 7412fab75feSLisandro Dalcin #define PETSC_PHI PetscRealConstant(1.6180339887498948482045868343656381) 7427b156302SMatthew G. Knepley #define PETSC_SQRT2 PetscRealConstant(1.4142135623730950488016887242096981) 743d34fcf5fSBarry Smith 744ab824b78SBarry Smith #if !defined(PETSC_USE_64BIT_INDICES) 74571fd2e92SBarry Smith #define PETSC_MAX_INT 2147483647 746ab824b78SBarry Smith #define PETSC_MIN_INT (-PETSC_MAX_INT - 1) 747ab824b78SBarry Smith #else 748ab824b78SBarry Smith #define PETSC_MAX_INT 9223372036854775807L 749ab824b78SBarry Smith #define PETSC_MIN_INT (-PETSC_MAX_INT - 1) 750ab824b78SBarry Smith #endif 751569ea7c4SPierre Jolivet #define PETSC_MAX_UINT16 65535 752e489efc1SBarry Smith 753ce63c4c1SBarry Smith #if defined(PETSC_USE_REAL_SINGLE) 754ab824b78SBarry Smith # define PETSC_MAX_REAL 3.40282346638528860e+38F 7559fa7d148SSatish Balay # define PETSC_MIN_REAL (-PETSC_MAX_REAL) 75682a7e548SBarry Smith # define PETSC_MACHINE_EPSILON 1.19209290e-07F 75782a7e548SBarry Smith # define PETSC_SQRT_MACHINE_EPSILON 3.45266983e-04F 758ee223c85SLisandro Dalcin # define PETSC_SMALL 1.e-5F 759ce63c4c1SBarry Smith #elif defined(PETSC_USE_REAL_DOUBLE) 760ab824b78SBarry Smith # define PETSC_MAX_REAL 1.7976931348623157e+308 7619fa7d148SSatish Balay # define PETSC_MIN_REAL (-PETSC_MAX_REAL) 76282a7e548SBarry Smith # define PETSC_MACHINE_EPSILON 2.2204460492503131e-16 76382a7e548SBarry Smith # define PETSC_SQRT_MACHINE_EPSILON 1.490116119384766e-08 764cf6e855fSSatish Balay # define PETSC_SMALL 1.e-10 765ce63c4c1SBarry Smith #elif defined(PETSC_USE_REAL___FLOAT128) 766ea345e14SBarry Smith # define PETSC_MAX_REAL FLT128_MAX 7679fa7d148SSatish Balay # define PETSC_MIN_REAL (-FLT128_MAX) 768d34fcf5fSBarry Smith # define PETSC_MACHINE_EPSILON FLT128_EPSILON 769ee223c85SLisandro Dalcin # define PETSC_SQRT_MACHINE_EPSILON 1.38777878078144567552953958511352539e-17Q 770ee223c85SLisandro Dalcin # define PETSC_SMALL 1.e-20Q 7715117d392SLisandro Dalcin #elif defined(PETSC_USE_REAL___FP16) 7725117d392SLisandro Dalcin # define PETSC_MAX_REAL 65504.0F 7739fa7d148SSatish Balay # define PETSC_MIN_REAL (-PETSC_MAX_REAL) 7745117d392SLisandro Dalcin # define PETSC_MACHINE_EPSILON .0009765625F 7755117d392SLisandro Dalcin # define PETSC_SQRT_MACHINE_EPSILON .03125F 7765117d392SLisandro Dalcin # define PETSC_SMALL 5.e-3F 7779cf09972SJed Brown #endif 7783e523bebSBarry Smith 77925d0f998SSatish Balay #define PETSC_INFINITY (PETSC_MAX_REAL/4) 7809fa7d148SSatish Balay #define PETSC_NINFINITY (-PETSC_INFINITY) 781e270355aSBarry Smith 7829f4f8022SLisandro Dalcin PETSC_EXTERN PetscBool PetscIsInfReal(PetscReal); 7833948c36eSLisandro Dalcin PETSC_EXTERN PetscBool PetscIsNanReal(PetscReal); 7848b49ba18SBarry Smith PETSC_EXTERN PetscBool PetscIsNormalReal(PetscReal); 7859f4f8022SLisandro Dalcin PETSC_STATIC_INLINE PetscBool PetscIsInfOrNanReal(PetscReal v) {return PetscIsInfReal(v) || PetscIsNanReal(v) ? PETSC_TRUE : PETSC_FALSE;} 7869f4f8022SLisandro Dalcin PETSC_STATIC_INLINE PetscBool PetscIsInfScalar(PetscScalar v) {return PetscIsInfReal(PetscAbsScalar(v));} 7873948c36eSLisandro Dalcin PETSC_STATIC_INLINE PetscBool PetscIsNanScalar(PetscScalar v) {return PetscIsNanReal(PetscAbsScalar(v));} 7889f4f8022SLisandro Dalcin PETSC_STATIC_INLINE PetscBool PetscIsInfOrNanScalar(PetscScalar v) {return PetscIsInfOrNanReal(PetscAbsScalar(v));} 7893948c36eSLisandro Dalcin PETSC_STATIC_INLINE PetscBool PetscIsNormalScalar(PetscScalar v) {return PetscIsNormalReal(PetscAbsScalar(v));} 7909a25a3ccSBarry Smith 791b10005b4SLisandro Dalcin PETSC_EXTERN PetscBool PetscIsCloseAtTol(PetscReal,PetscReal,PetscReal,PetscReal); 792ce4818fdSLisandro Dalcin PETSC_EXTERN PetscBool PetscEqualReal(PetscReal,PetscReal); 793ce4818fdSLisandro Dalcin PETSC_EXTERN PetscBool PetscEqualScalar(PetscScalar,PetscScalar); 794ce4818fdSLisandro Dalcin 79598725619SBarry Smith /* 79698725619SBarry Smith These macros are currently hardwired to match the regular data types, so there is no support for a different 79798725619SBarry Smith MatScalar from PetscScalar. We left the MatScalar in the source just in case we use it again. 79898725619SBarry Smith */ 79998725619SBarry Smith #define MPIU_MATSCALAR MPIU_SCALAR 80098725619SBarry Smith typedef PetscScalar MatScalar; 80198725619SBarry Smith typedef PetscReal MatReal; 80298725619SBarry Smith 8038ad47952SJed Brown struct petsc_mpiu_2scalar {PetscScalar a,b;}; 8048ad47952SJed Brown PETSC_EXTERN MPI_Datatype MPIU_2SCALAR PetscAttrMPITypeTagLayoutCompatible(struct petsc_mpiu_2scalar); 805df4397b0SStefano Zampini 806a616ada9SVaclav Hapla #if defined(PETSC_USE_64BIT_INDICES) 8078ad47952SJed Brown struct petsc_mpiu_2int {PetscInt a,b;}; 8088ad47952SJed Brown PETSC_EXTERN MPI_Datatype MPIU_2INT PetscAttrMPITypeTagLayoutCompatible(struct petsc_mpiu_2int); 8098ad47952SJed Brown #else 8108ad47952SJed Brown #define MPIU_2INT MPI_2INT 8118ad47952SJed Brown #endif 812e9fa29b7SSatish Balay 813b2fb0278SBarry Smith PETSC_STATIC_INLINE PetscInt PetscPowInt(PetscInt base,PetscInt power) 814b2fb0278SBarry Smith { 815fa711258SJed Brown PetscInt result = 1; 816fa711258SJed Brown while (power) { 817fa711258SJed Brown if (power & 1) result *= base; 818fa711258SJed Brown power >>= 1; 819fa711258SJed Brown base *= base; 820fa711258SJed Brown } 821fa711258SJed Brown return result; 822fa711258SJed Brown } 823b2fb0278SBarry Smith 824ad70a4c3SStefano Zampini PETSC_STATIC_INLINE PetscInt64 PetscPowInt64(PetscInt base,PetscInt power) 825ad70a4c3SStefano Zampini { 826ad70a4c3SStefano Zampini PetscInt64 result = 1; 827ad70a4c3SStefano Zampini while (power) { 828ad70a4c3SStefano Zampini if (power & 1) result *= base; 829ad70a4c3SStefano Zampini power >>= 1; 830ad70a4c3SStefano Zampini base *= base; 831ad70a4c3SStefano Zampini } 832ad70a4c3SStefano Zampini return result; 833ad70a4c3SStefano Zampini } 834ad70a4c3SStefano Zampini 835b2fb0278SBarry Smith PETSC_STATIC_INLINE PetscReal PetscPowRealInt(PetscReal base,PetscInt power) 836b2fb0278SBarry Smith { 837fa711258SJed Brown PetscReal result = 1; 838d98d5da7SBarry Smith if (power < 0) { 839d98d5da7SBarry Smith power = -power; 84010d40e53SLisandro Dalcin base = ((PetscReal)1)/base; 841d98d5da7SBarry Smith } 842fa711258SJed Brown while (power) { 843fa711258SJed Brown if (power & 1) result *= base; 844fa711258SJed Brown power >>= 1; 845fa711258SJed Brown base *= base; 846fa711258SJed Brown } 847fa711258SJed Brown return result; 848fa711258SJed Brown } 849fa711258SJed Brown 850b2fb0278SBarry Smith PETSC_STATIC_INLINE PetscScalar PetscPowScalarInt(PetscScalar base,PetscInt power) 851b2fb0278SBarry Smith { 8525117d392SLisandro Dalcin PetscScalar result = (PetscReal)1; 8538b49ba18SBarry Smith if (power < 0) { 8548b49ba18SBarry Smith power = -power; 85510d40e53SLisandro Dalcin base = ((PetscReal)1)/base; 8568b49ba18SBarry Smith } 8578b49ba18SBarry Smith while (power) { 8588b49ba18SBarry Smith if (power & 1) result *= base; 8598b49ba18SBarry Smith power >>= 1; 8608b49ba18SBarry Smith base *= base; 8618b49ba18SBarry Smith } 8628b49ba18SBarry Smith return result; 8638b49ba18SBarry Smith } 8648b49ba18SBarry Smith 865b2fb0278SBarry Smith PETSC_STATIC_INLINE PetscScalar PetscPowScalarReal(PetscScalar base,PetscReal power) 866b2fb0278SBarry Smith { 867b2fb0278SBarry Smith PetscScalar cpower = power; 868b2fb0278SBarry Smith return PetscPowScalar(base,cpower); 869b2fb0278SBarry Smith } 87078a59e97SMatthew G. Knepley 871bebf13c0SMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscLinearRegression(PetscInt,const PetscReal[],const PetscReal[],PetscReal*,PetscReal*); 872e489efc1SBarry Smith #endif 873