cds/algo/split_bitstring.h

   1 //$$CDS-header$$
   2
   3 #ifndef CDSLIB_ALGO_SPLIT_BITSTRING_H
   4 #define CDSLIB_ALGO_SPLIT_BITSTRING_H
   5
   6 #include <cds/algo/base.h>
   7
   8 namespace cds { namespace algo {
   9
  10     /// Cuts a bit sequence from fixed-size bit-string
  11     /**
  12         The splitter can be used as iterator over bit-string.
  13         Each call of \p cut() or \p safe_cut() cuts the bit count specified
  14         and keeps the position inside bit-string for the next call.
  15
  16         The splitter stores a const reference to bit-string, not a copy.
  17         The maximum count of bits that can be cut for single call is <tt> sizeof(UInt) * 8 </tt>
  18     */
  19     template <typename BitString, typename UInt = size_t >
  20     class split_bitstring
  21     {
  22     public:
  23         typedef BitString bitstring;    ///< Bit-string type
  24         typedef UInt      uint_type;    ///< Bit-string portion type
  25
  26         //@cond
  27         static CDS_CONSTEXPR size_t const c_nHashSize   = (sizeof(bitstring) + sizeof(uint_type) - 1) / sizeof(uint_type);
  28         static CDS_CONSTEXPR size_t const c_nBitPerByte = 8;
  29         static CDS_CONSTEXPR size_t const c_nBitPerHash = sizeof(bitstring) * c_nBitPerByte;
  30         static CDS_CONSTEXPR size_t const c_nBitPerInt  = sizeof(uint_type) * c_nBitPerByte;
  31         //@endcond
  32
  33     public:
  34         /// Initializises the splitter with reference to \p h and zero start bit offset
  35         explicit split_bitstring( bitstring const& h )
  36             : m_ptr(reinterpret_cast<uint_type const*>( &h ))
  37             , m_pos(0)
  38             , m_first( m_ptr )
  39 #   ifdef _DEBUG
  40             , m_last( m_ptr + c_nHashSize )
  41 #   endif
  42         {}
  43
  44         /// Initializises the splitter with reference to \p h and start bit offset \p nBitOffset
  45         split_bitstring( bitstring const& h, size_t nBitOffset )
  46             : m_ptr( reinterpret_cast<uint_type const*>( &h ) + nBitOffset / c_nBitPerInt )
  47             , m_pos( nBitOffset % c_nBitPerInt )
  48             , m_first( reinterpret_cast<uint_type const*>(&h))
  49 #   ifdef _DEBUG
  50             , m_last( m_first + c_nHashSize )
  51 #   endif
  52         {}
  53
  54
  55         /// Returns \p true if end-of-string is not reached yet
  56         explicit operator bool() const
  57         {
  58             return !eos();
  59         }
  60
  61         /// Returns \p true if end-of-stream encountered
  62         bool eos() const
  63         {
  64             return m_pos >= c_nBitPerHash;
  65         }
  66
  67         /// Cuts next \p nBits from bit-string
  68         /**
  69             Precondition: <tt>nBits <= sizeof(uint_type) * 8</tt>
  70
  71             This function does not manage out-of-bound condition.
  72             To control that use \p safe_cut().
  73         */
  74         uint_type cut( size_t nBits )
  75         {
  76             assert( !eos() );
  77             assert( nBits <= c_nBitPerInt );
  78             assert( m_pos + nBits <= c_nBitPerHash );
  79 #   ifdef _DEBUG
  80             assert( m_ptr < m_last );
  81 #   endif
  82             uint_type result;
  83
  84             uint_type const nRest = c_nBitPerInt - m_pos % c_nBitPerInt;
  85             m_pos += nBits;
  86             if ( nBits < nRest ) {
  87                 result = *m_ptr << ( nRest - nBits );
  88                 result = result >> ( c_nBitPerInt - nBits );
  89             }
  90             else if ( nBits == nRest ) {
  91                 result = *m_ptr >> ( c_nBitPerInt - nRest );
  92                 ++m_ptr;
  93             }
  94             else {
  95                 uint_type const lsb = *m_ptr >> ( c_nBitPerInt - nRest );
  96                 nBits -= nRest;
  97                 ++m_ptr;
  98
  99                 result = *m_ptr << ( c_nBitPerInt - nBits );
 100                 result = result >> ( c_nBitPerInt - nBits );
 101                 result = (result << nRest) + lsb;
 102             }
 103
 104             assert( m_pos <= c_nBitPerHash );
 105 #   ifdef _DEBUG
 106             assert( m_ptr <= m_last );
 107 #   endif
 108             return result;
 109         }
 110
 111         /// Cuts up to \p nBits from the bit-string
 112         /**
 113             Analog of \p cut() but if \p nBits is more than the rest of bit-string,
 114             only the rest is returned.
 115             If \p eos() is \p true the function returns 0.
 116         */
 117         uint_type safe_cut( size_t nBits )
 118         {
 119             if ( eos() )
 120                 return 0;
 121
 122             assert( nBits <= sizeof(uint_type) * c_nBitPerByte );
 123
 124             if ( m_pos + nBits > c_nBitPerHash )
 125                 nBits = c_nBitPerHash - m_pos;
 126             return nBits ? cut( nBits ) : 0;
 127         }
 128
 129         /// Resets the splitter
 130         void reset()
 131         {
 132             m_ptr = m_first;
 133             m_pos = 0;
 134         }
 135
 136         // Returns pointer to source bitstring
 137         bitstring const * source() const
 138         {
 139             return reinterpret_cast<bitstring const *>( m_first );
 140         }
 141
 142     private:
 143         //@cond
 144         uint_type const* m_ptr;  ///< current position in the hash
 145         size_t           m_pos;  ///< current position in bits
 146         uint_type const* m_first; ///< first position
 147 #   ifdef _DEBUG
 148         uint_type const* m_last;  ///< last position
 149 #   endif
 150         //@endcond
 151     };
 152
 153 }} // namespace cds::algo
 154
 155 #endif // #ifndef CDSLIB_ALGO_SPLIT_BITSTRING_H