inf_nan.hpp 7.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197
  1. // Copyright Kevlin Henney, 2000-2005.
  2. // Copyright Alexander Nasonov, 2006-2010.
  3. // Copyright Antony Polukhin, 2011-2019.
  4. //
  5. // Distributed under the Boost Software License, Version 1.0. (See
  6. // accompanying file LICENSE_1_0.txt or copy at
  7. // http://www.boost.org/LICENSE_1_0.txt)
  8. //
  9. // what: lexical_cast custom keyword cast
  10. // who: contributed by Kevlin Henney,
  11. // enhanced with contributions from Terje Slettebo,
  12. // with additional fixes and suggestions from Gennaro Prota,
  13. // Beman Dawes, Dave Abrahams, Daryle Walker, Peter Dimov,
  14. // Alexander Nasonov, Antony Polukhin, Justin Viiret, Michael Hofmann,
  15. // Cheng Yang, Matthew Bradbury, David W. Birdsall, Pavel Korzh and other Boosters
  16. // when: November 2000, March 2003, June 2005, June 2006, March 2011 - 2014
  17. #ifndef BOOST_LEXICAL_CAST_DETAIL_INF_NAN_HPP
  18. #define BOOST_LEXICAL_CAST_DETAIL_INF_NAN_HPP
  19. #include <boost/config.hpp>
  20. #ifdef BOOST_HAS_PRAGMA_ONCE
  21. # pragma once
  22. #endif
  23. #if defined(BOOST_NO_STRINGSTREAM) || defined(BOOST_NO_STD_WSTRING)
  24. #define BOOST_LCAST_NO_WCHAR_T
  25. #endif
  26. #include <cstddef>
  27. #include <cstring>
  28. #include <boost/limits.hpp>
  29. #include <boost/detail/workaround.hpp>
  30. #include <boost/math/special_functions/sign.hpp>
  31. #include <boost/math/special_functions/fpclassify.hpp>
  32. #include <boost/lexical_cast/detail/lcast_char_constants.hpp>
  33. namespace boost {
  34. namespace detail
  35. {
  36. template <class CharT>
  37. bool lc_iequal(const CharT* val, const CharT* lcase, const CharT* ucase, unsigned int len) BOOST_NOEXCEPT {
  38. for( unsigned int i=0; i < len; ++i ) {
  39. if ( val[i] != lcase[i] && val[i] != ucase[i] ) return false;
  40. }
  41. return true;
  42. }
  43. /* Returns true and sets the correct value if found NaN or Inf. */
  44. template <class CharT, class T>
  45. inline bool parse_inf_nan_impl(const CharT* begin, const CharT* end, T& value
  46. , const CharT* lc_NAN, const CharT* lc_nan
  47. , const CharT* lc_INFINITY, const CharT* lc_infinity
  48. , const CharT opening_brace, const CharT closing_brace) BOOST_NOEXCEPT
  49. {
  50. using namespace std;
  51. if (begin == end) return false;
  52. const CharT minus = lcast_char_constants<CharT>::minus;
  53. const CharT plus = lcast_char_constants<CharT>::plus;
  54. const int inifinity_size = 8; // == sizeof("infinity") - 1
  55. /* Parsing +/- */
  56. bool const has_minus = (*begin == minus);
  57. if (has_minus || *begin == plus) {
  58. ++ begin;
  59. }
  60. if (end - begin < 3) return false;
  61. if (lc_iequal(begin, lc_nan, lc_NAN, 3)) {
  62. begin += 3;
  63. if (end != begin) {
  64. /* It is 'nan(...)' or some bad input*/
  65. if (end - begin < 2) return false; // bad input
  66. -- end;
  67. if (*begin != opening_brace || *end != closing_brace) return false; // bad input
  68. }
  69. if( !has_minus ) value = std::numeric_limits<T>::quiet_NaN();
  70. else value = (boost::math::changesign) (std::numeric_limits<T>::quiet_NaN());
  71. return true;
  72. } else if (
  73. ( /* 'INF' or 'inf' */
  74. end - begin == 3 // 3 == sizeof('inf') - 1
  75. && lc_iequal(begin, lc_infinity, lc_INFINITY, 3)
  76. )
  77. ||
  78. ( /* 'INFINITY' or 'infinity' */
  79. end - begin == inifinity_size
  80. && lc_iequal(begin, lc_infinity, lc_INFINITY, inifinity_size)
  81. )
  82. )
  83. {
  84. if( !has_minus ) value = std::numeric_limits<T>::infinity();
  85. else value = (boost::math::changesign) (std::numeric_limits<T>::infinity());
  86. return true;
  87. }
  88. return false;
  89. }
  90. template <class CharT, class T>
  91. bool put_inf_nan_impl(CharT* begin, CharT*& end, const T& value
  92. , const CharT* lc_nan
  93. , const CharT* lc_infinity) BOOST_NOEXCEPT
  94. {
  95. using namespace std;
  96. const CharT minus = lcast_char_constants<CharT>::minus;
  97. if ((boost::math::isnan)(value)) {
  98. if ((boost::math::signbit)(value)) {
  99. *begin = minus;
  100. ++ begin;
  101. }
  102. memcpy(begin, lc_nan, 3 * sizeof(CharT));
  103. end = begin + 3;
  104. return true;
  105. } else if ((boost::math::isinf)(value)) {
  106. if ((boost::math::signbit)(value)) {
  107. *begin = minus;
  108. ++ begin;
  109. }
  110. memcpy(begin, lc_infinity, 3 * sizeof(CharT));
  111. end = begin + 3;
  112. return true;
  113. }
  114. return false;
  115. }
  116. #ifndef BOOST_LCAST_NO_WCHAR_T
  117. template <class T>
  118. bool parse_inf_nan(const wchar_t* begin, const wchar_t* end, T& value) BOOST_NOEXCEPT {
  119. return parse_inf_nan_impl(begin, end, value
  120. , L"NAN", L"nan"
  121. , L"INFINITY", L"infinity"
  122. , L'(', L')');
  123. }
  124. template <class T>
  125. bool put_inf_nan(wchar_t* begin, wchar_t*& end, const T& value) BOOST_NOEXCEPT {
  126. return put_inf_nan_impl(begin, end, value, L"nan", L"infinity");
  127. }
  128. #endif
  129. #if !defined(BOOST_NO_CXX11_CHAR16_T) && !defined(BOOST_NO_CXX11_UNICODE_LITERALS)
  130. template <class T>
  131. bool parse_inf_nan(const char16_t* begin, const char16_t* end, T& value) BOOST_NOEXCEPT {
  132. return parse_inf_nan_impl(begin, end, value
  133. , u"NAN", u"nan"
  134. , u"INFINITY", u"infinity"
  135. , u'(', u')');
  136. }
  137. template <class T>
  138. bool put_inf_nan(char16_t* begin, char16_t*& end, const T& value) BOOST_NOEXCEPT {
  139. return put_inf_nan_impl(begin, end, value, u"nan", u"infinity");
  140. }
  141. #endif
  142. #if !defined(BOOST_NO_CXX11_CHAR32_T) && !defined(BOOST_NO_CXX11_UNICODE_LITERALS)
  143. template <class T>
  144. bool parse_inf_nan(const char32_t* begin, const char32_t* end, T& value) BOOST_NOEXCEPT {
  145. return parse_inf_nan_impl(begin, end, value
  146. , U"NAN", U"nan"
  147. , U"INFINITY", U"infinity"
  148. , U'(', U')');
  149. }
  150. template <class T>
  151. bool put_inf_nan(char32_t* begin, char32_t*& end, const T& value) BOOST_NOEXCEPT {
  152. return put_inf_nan_impl(begin, end, value, U"nan", U"infinity");
  153. }
  154. #endif
  155. template <class CharT, class T>
  156. bool parse_inf_nan(const CharT* begin, const CharT* end, T& value) BOOST_NOEXCEPT {
  157. return parse_inf_nan_impl(begin, end, value
  158. , "NAN", "nan"
  159. , "INFINITY", "infinity"
  160. , '(', ')');
  161. }
  162. template <class CharT, class T>
  163. bool put_inf_nan(CharT* begin, CharT*& end, const T& value) BOOST_NOEXCEPT {
  164. return put_inf_nan_impl(begin, end, value, "nan", "infinity");
  165. }
  166. }
  167. } // namespace boost
  168. #undef BOOST_LCAST_NO_WCHAR_T
  169. #endif // BOOST_LEXICAL_CAST_DETAIL_INF_NAN_HPP