任意数据长度的bitset，以及按位读写数据的函数

 如果没有定义过byte_t类型：

#include <stdint.h>
typedef uint8_t byte_t;
//typedef unsigned char byte_t; //1个byte字节，8位，一般系统中能表示的最小字节数据。

uint32_t bit_set(byte_t* data, size_t x, size_t bits, uint32_t value)
{
    uint32_t mask;
    if(bits == 32){
        reinterpret_cast<uint32_t*>(data)[x] = value;
    }
    else if(bits == 24){
        mask = 0x00FFFFFF;
        data += x*3;
        uint32_t* p = reinterpret_cast<uint32_t*>(data);
        *p &= ~mask;
        value &= mask;
        *p |= value;
    }
    else if(bits == 16){
        reinterpret_cast<uint16_t*>(data)[x] = value;
    }
    else if(bits == 8){
        data[x] = value;
    }
    else if(bits < 8){
        mask = (1<<bits)-1;
        x *= bits;
        data += x/8;
        value &= mask;
        x &= 7;
        value <<= x;
        mask <<= x;
        mask = ~mask;
        byte_t n = data[0];
        n &= mask;
        n |= value;
        data[0] = n;
    }
    return 0;
}

 

uint32_t bit_get(const byte_t* data, int x, int bits)
{
    uint32_t mask;
    if(bits == 32){
        return reinterpret_cast<const uint32_t*>(data)[x];
    }
    else if(bits == 24){
        mask = 0x00FFFFFF;
        data += x*3;
        const uint32_t* p = reinterpret_cast<const uint32_t*>(data);
        return *p & mask;
    }
    else if(bits == 16){
        return reinterpret_cast<const uint16_t*>(data)[x];
    }
    else if(bits == 8){
        return data[x];
    }
    else if(bits < 8){
        mask = (1<<bits)-1;
        x *= bits;
        data += x/8;
        x &= 7;
        return (data[0]>>x) & mask;
    }
}

这个bitset<>是c++bitset的扩展版，支持任意长度的数据

template<std::size_t BITSIZE>
struct bitset
{
public:
    const static std::size_t byte_size = BITSIZE/8 + (BITSIZE&0x07 ? 1 : 0);
public:
    bitset()
    {
    	this->reset();
    }

    size_t size()const
    {
    	return BITSIZE;
    }

    bool bit(size_t id)const
    {
        return (data[id/8]>>(id&0x07))&0x01;
    };
    bool operator[](size_t id)const
    {
    	return this->bit(id);
    }

    void set()
    {
    	memset(data, 0xFF, byte_size);
    }
    bool set(size_t id)
    {
    	byte_t& b = data[id/8];
        b |= 1<<(id&0x07);
        return true;
    }
    void reset()
    {
    	memset(data, 0, byte_size);
    }
    bool reset(size_t id)
    {
    	byte_t& b = data[id/8];
       	b &= ~(1<<(id&0x07));
        return false;
    }
public:
	byte_t data[byte_size];
};