onilink_

Bitfield Class

Feb 18th, 2015
351
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.30 KB | None | 0 0
  1. #include <iostream>
  2.  
  3. using namespace std;
  4.  
  5. template<int nbits>
  6. class BitField
  7. {
  8.     static constexpr unsigned int size = (nbits + 7)/8;
  9.     unsigned char data[ size ];
  10.    
  11.     struct setter
  12.     {
  13.         BitField & bf;
  14.         int i;
  15.        
  16.         setter(BitField & bf, int i):bf(bf),i(i){}
  17.        
  18.         setter operator<<(bool bit)
  19.         {
  20.             bf.setBit(i, bit);
  21.             setter bs(bf, i+1);
  22.             return bs;
  23.         }
  24.     };
  25.    
  26. public:
  27.     static const int bits = nbits;
  28.    
  29.     BitField():data{0} {
  30.         cout << size << endl;
  31.     }
  32.    
  33.     void setBit(int i, bool value)
  34.     {
  35.         const int p1 = i/8;
  36.         const int p2 = i%8;
  37.         // clear
  38.         data[p1] &= ~(1 << p2);
  39.         // set
  40.         data[p1] |= value << p2;
  41.     }
  42.     bool bit(int i) const
  43.     {
  44.         return data[i/8] & (1 << (i%8));
  45.     }
  46.    
  47.     setter operator<<(bool bit)
  48.     {
  49.         setter bs(*this, 0);
  50.         return bs << bit;
  51.     }
  52.     setter set(int i)
  53.     {
  54.         setter bs(*this, i);
  55.         return bs;
  56.     }
  57. };
  58.  
  59. int main()
  60. {
  61.     BitField<8> a, b;
  62.    
  63.     a << 1 << 0 << 1 << 0 << 1 << 1 << 0 << 0;
  64.     a.set(4) << 0 << 1 << 0 << 1;
  65.    
  66.     for(int i=0; i<a.bits; ++i)
  67.     {
  68.         cout << a.bit(i) << " ";
  69.     }
  70.     return 0;
  71. }
Advertisement
Add Comment
Please, Sign In to add comment