cohen_acceleration.hpp 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. // (C) Copyright Nick Thompson 2020.
  2. // Use, modification and distribution are subject to the
  3. // Boost Software License, Version 1.0. (See accompanying file
  4. // LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
  5. #ifndef BOOST_MATH_TOOLS_COHEN_ACCELERATION_HPP
  6. #define BOOST_MATH_TOOLS_COHEN_ACCELERATION_HPP
  7. #include <limits>
  8. #include <cmath>
  9. #include <cstdint>
  10. namespace boost::math::tools {
  11. // Algorithm 1 of https://people.mpim-bonn.mpg.de/zagier/files/exp-math-9/fulltext.pdf
  12. // Convergence Acceleration of Alternating Series: Henri Cohen, Fernando Rodriguez Villegas, and Don Zagier
  13. template<class G>
  14. auto cohen_acceleration(G& generator, std::int64_t n = -1)
  15. {
  16. using Real = decltype(generator());
  17. // This test doesn't pass for float128, sad!
  18. //static_assert(std::is_floating_point_v<Real>, "Real must be a floating point type.");
  19. using std::log;
  20. using std::pow;
  21. using std::ceil;
  22. using std::sqrt;
  23. auto n_ = static_cast<Real>(n);
  24. if (n < 0)
  25. {
  26. // relative error grows as 2*5.828^-n; take 5.828^-n < eps/4 => -nln(5.828) < ln(eps/4) => n > ln(4/eps)/ln(5.828).
  27. // Is there a way to do it rapidly with std::log2? (Yes, of course; but for primitive types it's computed at compile-time anyway.)
  28. n_ = static_cast<Real>(ceil(log(Real(4)/std::numeric_limits<Real>::epsilon())*Real(0.5672963285532555)));
  29. n = static_cast<std::int64_t>(n_);
  30. }
  31. // d can get huge and overflow if you pick n too large:
  32. auto d = static_cast<Real>(pow(Real(3 + sqrt(Real(8))), n_));
  33. d = (d + Real(1)/d)/2;
  34. Real b = -1;
  35. Real c = -d;
  36. Real s = 0;
  37. for (Real k = 0; k < n_; ++k) {
  38. c = b - c;
  39. s += c*generator();
  40. b = (k+n_)*(k-n_)*b/((k+Real(1)/Real(2))*(k+1));
  41. }
  42. return s/d;
  43. }
  44. }
  45. #endif