File: cmath.hh

package info (click to toggle)
dune-common 2.11.0-1~exp2
  • links: PTS, VCS
  • area: main
  • in suites: experimental
  • size: 6,056 kB
  • sloc: cpp: 54,404; python: 4,136; sh: 1,657; makefile: 17
file content (68 lines) | stat: -rw-r--r-- 1,639 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
// SPDX-FileCopyrightInfo: Copyright © DUNE Project contributors, see file LICENSE.md in module root
// SPDX-License-Identifier: LicenseRef-GPL-2.0-only-with-DUNE-exception
#ifndef DUNE_COMMON_STD_CMATH_HH
#define DUNE_COMMON_STD_CMATH_HH

#include <cmath>

#if __has_include(<version>)
#include <version>
#if (__cpp_lib_constexpr_cmath < 202306L)
#include <limits>
#include <type_traits>
#include <utility>
#endif
#endif

namespace Dune {

namespace Std {

#if (__cpp_lib_constexpr_cmath < 202202L)
// backport for constexpr functions between C++20 and C++23
template<class T>
constexpr T
abs(T t)
{
  if (std::is_constant_evaluated()) {
    return (t < 0) ? -t : t;
  } else {
    using std::abs;
    return abs(t);
  }
}

template<class T>
constexpr auto
sqrt(T t)
  requires(std::is_floating_point_v<T> || std::is_integral_v<T>)
{
  if (std::is_constant_evaluated()) {
    using TT = std::conditional_t<std::is_floating_point_v<T>, T, double>;
    if (t >= TT{ 0 } and t < std::numeric_limits<TT>::infinity()) {
      // use Heron's method:
      // https://en.wikipedia.org/wiki/Methods_of_computing_square_roots#Heron's_method
      TT curr = t, prev = 0;
      while (curr != prev)
        prev = std::exchange(curr, TT{ 0.5 } * (curr + TT{ t } / curr));
      return curr;
    } else {
      return std::numeric_limits<TT>::quiet_NaN();
    }
  } else {
    using std::sqrt;
    return sqrt(t);
  }
};
#else
using std::abs;
using std::sqrt;
#endif

} // namespace Std

} // namespace Dune

#endif // #ifndef DUNE_COMMON_STD_CMATH_HH