daw_json_parse_unsigned_int.h

Go to the documentation of this file.

// Copyright (c) Darrell Wright
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE or copy at http://www.boost.org/LICENSE_1_0.txt)
//
// Official repository: https://github.com/beached/daw_json_link
//
 
#pragma once
 
#include "daw/json/impl/version.h"
 
#include "daw/json/impl/daw_json_assert.h"
#include "daw/json/impl/daw_json_exec_modes.h"
#include "daw/json/impl/daw_json_parse_digit.h"
#include "daw/json/impl/daw_json_parse_unsigned_int.h"
#include "daw/json/impl/daw_json_type_options.h"
 
#include <daw/daw_arith_traits.h>
#include <daw/daw_construct_a.h>
#include <daw/daw_cxmath.h>
#include <daw/daw_not_null.h>
#include <daw/daw_remove_cvref.h>
#include <daw/daw_restrict.h>
#include <daw/daw_uint_buffer.h>
 
#include <cstddef>
#include <daw/stdinc/data_access.h>
#include <limits>
 
#if defined( DAW_ALLOW_SSE42 )
#include <emmintrin.h>
#include <smmintrin.h>
#include <tmmintrin.h>
#include <xmmintrin.h>
#if defined( DAW_HAS_MSVC_LIKE )
#include <intrin.h>
#endif
#endif
 
namespace daw::json {
  inline namespace DAW_JSON_VER {
    namespace json_details {
      template<typename Signed, typename Unsigned>
      DAW_ATTRIB_INLINE constexpr Signed to_signed( Unsigned &&u,
                                                    Signed sign ) {
        if( sign <= Signed{ 0 } ) {
          return static_cast<Signed>( -u );
        }
        return static_cast<Signed>( u );
      }
 
      [[nodiscard]]
      constexpr bool
      is_made_of_eight_digits_cx( daw::not_null<char const *> const ptr ) {
        // The copy to local buffer is to get the compiler to treat it like a
        // reinterpret_cast
 
        std::byte const buff[8]{ static_cast<std::byte>( ptr[0] ),
                                 static_cast<std::byte>( ptr[1] ),
                                 static_cast<std::byte>( ptr[2] ),
                                 static_cast<std::byte>( ptr[3] ),
                                 static_cast<std::byte>( ptr[4] ),
                                 static_cast<std::byte>( ptr[5] ),
                                 static_cast<std::byte>( ptr[6] ),
                                 static_cast<std::byte>( ptr[7] ) };
 
        auto val = UInt64( );
        for( std::size_t n = 0; n < 8; ++n ) {
          val |= to_uint64( buff[n] ) << ( 8 * n );
        }
        return ( ( ( val & 0xF0F0'F0F0'F0F0'F0F0_u64 ) |
                   ( ( ( val + 0x0606'0606'0606'0606_u64 ) &
                       0xF0F0'F0F0'F0F0'F0F0_u64 ) >>
                     4U ) ) == 0x3333'3333'3333'3333_u64 );
      }
 
      template<options::JsonRangeCheck RangeCheck, typename Unsigned,
               typename MaxArithUnsigned>
      using max_unsigned_t = daw::conditional_t<
        daw::is_integral_v<Unsigned> or std::is_enum_v<Unsigned>,
        daw::conditional_t<( sizeof( Unsigned ) > sizeof( MaxArithUnsigned ) ),
                           Unsigned, MaxArithUnsigned>,
        Unsigned>;
 
      // Constexpr'ified version from
      // https://kholdstare.github.io/technical/2020/05/26/faster-integer-parsing.html
      constexpr UInt64 parse_8_digits( daw::not_null<char const *> const str ) {
        auto const chunk = daw::to_uint64_buffer( str.get( ) );
        // 1-byte mask trick (works on 4 pairs of single digits)
        auto const lower_digits =
          ( chunk & 0x0F'00'0F'00'0F'00'0F'00_u64 ) >> 8U;
        auto const upper_digits =
          ( chunk & 0x00'0F'00'0F'00'0F'00'0F_u64 ) * 10U;
        auto const chunk2 = lower_digits + upper_digits;
 
        // 2-byte mask trick (works on 2 pairs of two digits)
        auto const lower_digits2 =
          ( chunk2 & 0x00'FF'00'00'00'FF'00'00_u64 ) >> 16U;
        auto const upper_digits2 =
          ( chunk2 & 0x00'00'00'FF'00'00'00'FF_u64 ) * 100U;
        auto const chunk3 = lower_digits2 + upper_digits2;
 
        // 4-byte mask trick (works on pair of four digits)
        auto const lower_digits3 =
          ( chunk3 & 0x00'00'FF'FF'00'00'00'00_u64 ) >> 32U;
        auto const upper_digits3 =
          ( chunk3 & 0x00'00'00'00'00'00'FF'FF_u64 ) * 10000U;
        auto const chunk4 = lower_digits3 + upper_digits3;
 
        return chunk4 & 0xFFFF'FFFF_u64;
      }
 
      static_assert( parse_8_digits( "12345678" ) == 1234'5678_u64,
                     "8 digit parser does not work on this platform" );
 
      constexpr UInt64
      parse_16_digits( daw::not_null<char const *> const str ) {
        auto const upper = parse_8_digits( str );
        auto const lower = parse_8_digits( str + 8 );
        return upper * 100'000'000_u64 + lower;
      }
 
      static_assert( parse_16_digits( "1234567890123456" ) ==
                       1234567890123456_u64,
                     "16 digit parser does not work on this platform" );
 
      template<typename T>
      struct make_unsigned_with_bool : daw::make_unsigned<T> {};
 
      template<>
      struct make_unsigned_with_bool<bool> {
        using type = bool;
      };
 
      template<typename Integer, typename T, typename ParseState>
      [[nodiscard]] constexpr Integer
      narrow_cast( T value, ParseState const &parse_state ) {
        if constexpr( std::is_signed_v<T> ) {
          if constexpr( std::is_signed_v<Integer> ) {
            if constexpr( sizeof( T ) <= sizeof( Integer ) ) {
              return value;
            } else if( value <= static_cast<T>( daw::max_value<Integer> ) ) {
              return static_cast<Integer>( value );
            } else {
              daw_json_error( ErrorReason::NumberOutOfRange, parse_state );
            }
          } else if constexpr( sizeof( T ) <= sizeof( Integer ) ) {
            if( value >= 0 ) {
              return value;
            }
            daw_json_error( ErrorReason::NumberOutOfRange, parse_state );
          } else {
            if( value >= 0 and
                value <= static_cast<T>( daw::max_value<Integer> ) ) {
              return value;
            }
            daw_json_error( ErrorReason::NumberOutOfRange, parse_state );
          }
        } else if constexpr( std::is_signed_v<Integer> ) {
          if constexpr( sizeof( T ) < sizeof( Integer ) ) {
            return static_cast<Integer>( value );
          } else {
            if( value > static_cast<T>( daw::max_value<Integer> ) ) {
              daw_json_error( ErrorReason::NumberOutOfRange, parse_state );
            }
            return static_cast<Integer>( value );
          }
        } else if constexpr( sizeof( T ) <= sizeof( Integer ) ) {
          return static_cast<Integer>( value );
        } else {
          if( value <= static_cast<T>( daw::max_value<Integer> ) ) {
            return static_cast<Integer>( value );
          }
          daw_json_error( ErrorReason::NumberOutOfRange, parse_state );
        }
      }
 
      template<typename T>
      using make_unsigned_with_bool_t =
        typename make_unsigned_with_bool<T>::type;
 
      template<typename Unsigned, options::JsonRangeCheck RangeChecked,
               typename ParseState>
      [[nodiscard]] static constexpr Unsigned
      unsigned_parser_known( ParseState &parse_state ) {
        // We know how many digits are in the number
        using result_t = max_unsigned_t<RangeChecked, Unsigned, UInt64>;
        using uresult_t = max_unsigned_t<RangeChecked,
                                         make_unsigned_with_bool_t<Unsigned>,
                                         UInt64>;
        static_assert(
          not static_cast<bool>( RangeChecked ) or
            std::is_same_v<uresult_t, UInt64>,
          "Range checking is only supported for std integral types" );
 
        daw::not_null<char const *> first = parse_state.first;
        daw::not_null<char const *> const last = parse_state.last;
        uresult_t result = uresult_t( );
 
        while( last - first >= 16 ) {
          result *= static_cast<uresult_t>( 10'000'000'000'000'000ULL );
          result += static_cast<uresult_t>( parse_16_digits( first ) );
          first += 16;
        }
        if( last - first >= 8 ) {
          result *= static_cast<uresult_t>( 100'000'000ULL );
          result += static_cast<uresult_t>( parse_8_digits( first ) );
          first += 8;
        }
        if constexpr( ParseState::is_zero_terminated_string ) {
          auto dig = parse_digit( *first );
          while( dig < 10U ) {
            result *= 10U;
            result += dig;
            ++first;
            dig = parse_digit( *first );
          }
        } else {
          while( first < last ) {
            result *= 10U;
            result += parse_digit( *first );
            ++first;
          }
        }
        if constexpr( RangeChecked != options::JsonRangeCheck::Never ) {
          auto const count = ( daw::numeric_limits<result_t>::digits10 + 1U ) -
                             std::size( parse_state );
          daw_json_ensure(
            ( ( result <= static_cast<uresult_t>(
                            ( daw::numeric_limits<result_t>::max )( ) ) ) &
              ( count >= 0 ) ),
            ErrorReason::NumberOutOfRange,
            parse_state );
        }
        parse_state.first = first;
        if constexpr( RangeChecked == options::JsonRangeCheck::Never ) {
          return daw::construct_a<Unsigned>( static_cast<Unsigned>( result ) );
        } else {
          return daw::construct_a<Unsigned>(
            narrow_cast<Unsigned>( result, parse_state ) );
        }
      }
 
      //**************************
      template<typename Unsigned, options::JsonRangeCheck RangeChecked,
               typename ParseState>
      [[nodiscard]] static constexpr Unsigned
      unsigned_parser_not_known( ParseState &parse_state ) {
        // We do not know how long the string is
        using result_t = max_unsigned_t<RangeChecked, Unsigned, UInt64>;
        using uresult_t = max_unsigned_t<RangeChecked,
                                         make_unsigned_with_bool_t<Unsigned>,
                                         UInt64>;
        static_assert(
          not static_cast<bool>( RangeChecked ) or
            std::is_same_v<uresult_t, UInt64>,
          "Range checking is only supported for std integral types" );
        daw_json_assert_weak( parse_state.has_more( ),
                              ErrorReason::UnexpectedEndOfData,
                              parse_state );
        daw::not_null<char const *> first = parse_state.first;
        auto const orig_first = first;
        (void)orig_first; // only used inside if constexpr and gcc9 warns
        daw::not_null<char const *> const last = parse_state.last;
        uresult_t result = uresult_t( );
        bool has_eight =
          last - first >= 8 ? is_made_of_eight_digits_cx( first ) : false;
        if( has_eight & ( last - first >= 16 ) ) {
          bool has_sixteen = is_made_of_eight_digits_cx( first + 8 );
          while( has_sixteen ) {
            result *= static_cast<uresult_t>( 10'000'000'000'000'000ULL );
            result += static_cast<uresult_t>( parse_16_digits( first ) );
            first += 16;
            has_eight =
              last - first >= 8 ? is_made_of_eight_digits_cx( first ) : false;
            has_sixteen =
              has_eight and
              ( last - first >= 16 ? is_made_of_eight_digits_cx( first + 8 )
                                   : false );
          }
        }
        if( has_eight ) {
          result *= static_cast<uresult_t>( 100'000'000ULL );
          result += static_cast<uresult_t>( parse_8_digits( first ) );
          first += 8;
        }
        if constexpr( ParseState::is_zero_terminated_string ) {
          auto dig = parse_digit( *first );
          while( dig < 10U ) {
            result *= 10U;
            result += dig;
            ++first;
            dig = parse_digit( *first );
          }
        } else {
          while( first < last ) {
            auto const dig = parse_digit( *first );
            if( dig >= 10U ) {
              break;
            }
            result *= 10U;
            result += dig;
            ++first;
          }
        }
 
        if constexpr( RangeChecked != options::JsonRangeCheck::Never ) {
          auto const count = static_cast<std::ptrdiff_t>(
                               daw::numeric_limits<result_t>::digits10 + 1 ) -
                             ( first - orig_first );
          daw_json_ensure(
            count >= 0, ErrorReason::NumberOutOfRange, parse_state );
        }
 
        parse_state.first = first;
        if constexpr( RangeChecked == options::JsonRangeCheck::Never ) {
          return daw::construct_a<Unsigned>(
            static_cast<Unsigned>( static_cast<result_t>( result ) ) );
        } else {
          return daw::construct_a<Unsigned>(
            narrow_cast<Unsigned>( result, parse_state ) );
        }
      }
 
#if false and defined( DAW_ALLOW_SSE42 )
      /*
      // Adapted from
      //
      //
      https://github.com/lemire/simdjson/blob/102262c7abe64b517a36a6049b39d95f58bf4aea/src/haswell/numberparsing.h
      DAW_ATTRIB_NONNULL( )
      inline UInt64 parse_eight_digits_unrolled( char const *ptr ) {
        // this actually computes *16* values so we are being wasteful.
        static __m128i const ascii0 = _mm_set1_epi8( '0' );
 
        static __m128i const mul_1_10 =
          _mm_setr_epi8( 10, 1, 10, 1, 10, 1, 10, 1, 10, 1, 10, 1, 10, 1, 10, 1
      );
 
        static __m128i const mul_1_100 =
          _mm_setr_epi16( 100, 1, 100, 1, 100, 1, 100, 1 );
 
        static __m128i const mul_1_10000 =
          _mm_setr_epi16( 10000, 1, 10000, 1, 10000, 1, 10000, 1 );
 
        __m128i const input = _mm_sub_epi8(
          _mm_loadu_si128( reinterpret_cast<__m128i const *>( ptr ) ), ascii0 );
        __m128i const t1 = _mm_maddubs_epi16( input, mul_1_10 );
        __m128i const t2 = _mm_madd_epi16( t1, mul_1_100 );
        __m128i const t3 = _mm_packus_epi32( t2, t2 );
        __m128i const t4 = _mm_madd_epi16( t3, mul_1_10000 );
        return to_uint64( static_cast<std::uint32_t>( _mm_cvtsi128_si32(
          t4 ) ) ); // only captures the sum of the first 8 digits, drop the
      rest
      }
 
      inline UInt64 parse_sixteen_digits_unrolled( char const *ptr ) {
        static __m128i const ascii0 = _mm_set1_epi8( '0' );
 
        static __m128i const mul_1_10 =
          _mm_setr_epi8( 10, 1, 10, 1, 10, 1, 10, 1, 10, 1, 10, 1, 10, 1, 10, 1
      );
 
        static __m128i const mul_1_100 =
          _mm_setr_epi16( 100, 1, 100, 1, 100, 1, 100, 1 );
 
        static __m128i const mul_1_10000 =
          _mm_setr_epi16( 10000, 1, 10000, 1, 10000, 1, 10000, 1 );
 
        __m128i const input = _mm_sub_epi8(
          _mm_loadu_si128( reinterpret_cast<__m128i const *>( ptr ) ), ascii0 );
        __m128i const t1 = _mm_maddubs_epi16( input, mul_1_10 );
        __m128i const t2 = _mm_madd_epi16( t1, mul_1_100 );
        __m128i const t3 = _mm_packus_epi32( t2, t2 );
        __m128i const t4 = _mm_madd_epi16( t3, mul_1_10000 );
        return to_uint64( _mm_cvtsi128_si64( t4 ) );
      }
 
      [[nodiscard]] inline bool
      is_made_of_eight_digits_fast( char const *ptr ) {
        UInt64 val;
        memcpy( &val, ptr, sizeof( std::uint64_t ) );
        return ( ( ( val & 0xF0F0F0F0F0F0F0F0_u64 ) |
                   ( ( ( val + 0x0606060606060606_u64 ) & 0xF0F0F0F0F0F0F0F0_u64
      )
      >> 4_u64 ) ) == 0x3333333333333333_u64 );
      }
 
      template<typename Unsigned, options::JsonRangeCheck RangeChecked, bool, typename
      ParseState>
      [[nodiscard]] inline Unsigned
      unsigned_parser( sse42_exec_tag const &, ParseState &parse_state ) {
        daw_json_assert_weak( parse_state.has_more( ),
      ErrorRange::UnexpectedEndOfData, parse_state
      ); using result_t = max_unsigned_t<RangeChecked, Unsigned, UInt64>;
      result_t result = result_t( ); char const *first = parse_state.first; char const
       *const last = parse_state.last; char const *const orig_first =
      first;
        {
          auto sz = last - first;
          while( ( sz >= 8 ) & is_made_of_eight_digits_fast( first ) ) {
            if( ( sz < 16 ) | ( not is_made_of_eight_digits_fast( first + 8 ) )
      ) { result *= 100'000'000_u64; result += static_cast<result_t>(
      parse_eight_digits_unrolled( first ) ); first += 8; break;
            }
            result *= 10'000'000'000'000'000_u64;
            result +=
              static_cast<result_t>( parse_sixteen_digits_unrolled( first ) );
            sz -= 16;
            first += 16;
          }
        }
 
        auto dig = parse_digit( *first );
        while( dig < 10U ) {
          result *= 10U;
          result += dig;
          ++first;
          dig = parse_digit( *first );
        }
        if constexpr( RangeChecked != options::JsonRangeCheck::Never ) {
          auto const count =
            static_cast<intmax_t>( daw::numeric_limits<Unsigned>::digits10 + 1 )
      - ( first - orig_first ); daw_json_ensure( (count >= 0) & (result <=
      static_cast<result_t>( (daw::numeric_limits<Unsigned>::max)( ) )),
                                         ErrorReason::NumberOutOfRange,
      parse_state
      );
        }
        parse_state.first = first;
        if constexpr( RangeChecked == options::JsonRangeCheck::Never ) {
          return daw::construct_a<Unsigned>( static_cast<Unsigned>( result ) );
        } else {
          return daw::construct_a<Unsigned>( narrow_cast<Unsigned>( result, parse_state ) );
        }
      }
       */
#endif
 
      template<typename Unsigned, options::JsonRangeCheck RangeChecked,
               bool KnownBounds, typename ParseState>
      [[nodiscard]] DAW_ATTRIB_INLINE static constexpr Unsigned
      unsigned_parser( ParseState &parse_state ) {
        if constexpr( KnownBounds ) {
          return unsigned_parser_known<Unsigned, RangeChecked>( parse_state );
        } else {
          return unsigned_parser_not_known<Unsigned, RangeChecked>(
            parse_state );
        }
      }
    } // namespace json_details
  } // namespace DAW_JSON_VER
} // namespace daw::json