You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

54 lines
1.4KB

  1. /*
  2. Implementation of the Fukushima method for the Lambert W function
  3. Copyright (C) 2015 Darko Veberic, darko.veberic@ijs.si
  4. This program is free software: you can redistribute it and/or modify
  5. it under the terms of the GNU General Public License as published by
  6. the Free Software Foundation, either version 3 of the License, or
  7. (at your option) any later version.
  8. This program is distributed in the hope that it will be useful,
  9. but WITHOUT ANY WARRANTY; without even the implied warranty of
  10. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  11. GNU General Public License for more details.
  12. You should have received a copy of the GNU General Public License
  13. along with this program. If not, see <http://www.gnu.org/licenses/>.
  14. */
  15. /*
  16. This code is based on the following publication and its author's fortran code:
  17. Toshio Fukushima, "Precise and fast computation of Lambert W-functions without
  18. transcendental function evaluations", J. Comp. Appl. Math. 244 (2013) 77-89.
  19. */
  20. #ifndef _FukushimaLambertW_h_
  21. #define _FukushimaLambertW_h_
  22. #include <limits>
  23. namespace Fukushima {
  24. double LambertW0(const double x);
  25. double LambertWm1(const double x);
  26. inline
  27. double
  28. LambertW(const int branch, const double x)
  29. {
  30. switch (branch) {
  31. case -1: return LambertWm1(x);
  32. case 0: return LambertW0(x);
  33. default: return std::numeric_limits<double>::quiet_NaN();
  34. }
  35. }
  36. }
  37. #endif