3

我编写了一个模板类，用于将多个布尔值存储在一个整数中。现在，设置和获取每个布尔值都是通过显式函数完成的

    bool isBitSet(int index)
    {
        return static_cast<bool>((block_ >> index) % 2)
    }

    void setBitOn(int index)
    {
        block_ |= 1 << index;
    }

我相信下面的方法可以用来获取一个值，但是设置如何工作，因为我们不能直接返回一个引用？

    const bool operator [] (int index) const 
    {
        return static_cast<bool>((block_ >> index) % 2);
    }
4

1 回答 1

6

std::vector<bool>在std::bitset标准库中也是如此。如参考中所述，std::vector<bool>它返回一个代理类，其运算符重载以充当向量的元素。

你也可以这样做。

对于用户友好的示例，请再次参阅公共接口的参考，它是这样的：

template <class Allocator>
class vector<bool, Allocator> {
  // ...
  public:
    class reference {
        friend class vector;
        reference();
      public:
        ~reference();
        operator bool() const;
        reference& operator=(bool x);
        reference& operator=(const reference&);
        void flip();
    };
  // ...
};

要实现这个类，您应该存储一个指向实际数据块的成员指针和一个要操作的掩码。

举一个真实的例子，在 g++ 头文件中查找在文件中std::vector<bool>调用的成员类。std::vector<bool>::_Bit_referencebits/stl_bvector.h

用一个例子来阐明 OP：

假设您有一个包含 320 个布尔值的类。你可以把它写成：

class boolcontainer {
  uint32_t data[10];
public:
  //default ctor. to initialize the elements with zeros
  boolcontainer() { for (int i = 0; i < 10; ++i) { data[i] = 0; } }
}

您想添加一个运算符[]。添加一个 const 很简单：

class boolcontainer {
  uint32_t data[10];
public:
  bool operator[](int i) const { return data[i/32] & (1 << (i%32)); }
}

要拥有一个非常量，您需要更多。首先，您需要创建一个代表对您的值的引用的类。您必须有某种指向值存储位置的指针，并且（在这种情况下）您需要一个位掩码来指定一个具体位。为了能够将其作为 bool& 处理，您需要添加一些运算符，即转换为 bool 和 operator=：

class reference {
  uint32_t *dataptr;
  uint32_t mask;
public:
  //constructor just initializing members
  reference(uint32_t *dataptr_, uint32_t mask_) : dataptr(dataptr_), mask(mask_) {}

  //conversion to bool
  operator bool() const {
    //just like in the getter, but the bitmask is stored now locally
    return *dataptr & mask;
  }

  //sets one single bit represented by mask to b
  reference& operator=(bool b) {
    if (b) {
      *dataptr |= mask;
    } else {
      *dataptr &= ~mask;
    }
    return *this;
  }

  //TODO copy ctor., operator==, operator<
};

请注意，上面的结构将表现为 bool& —— 读取它会从指针和掩码表示的数据点读取值，类似地，写入它会覆盖表示位置的位。我还编写了一个初始化成员的构造函数。

现在你需要的是你的 boolcontainer 的 operator[] 应该返回一个上述类的对象：

class boolcontainer {
  uint32_t data[10];
public:

  boolcontainer() { for (int i = 0; i < 10; ++i) { data[i] = 0; } }

  class reference {
     ... //see above
  }

  //keep the const version for efficiency
  bool operator[](int i) const { return data[i/32] & (1 << (i%32)); }

  //non-const version returns our reference object.
  reference operator[](int i) { return reference(&data[i/32], 1 << (i%32)); }
};

现在有一些代码来测试它（只打印前 40 个值）：

#include <iostream>
#include "boolcontainer.h"

void printboolcontainer(const boolcontainer &bc)
{
    //note that this is the constant version
    for (int i = 0; i < 40; ++i) {
        std::cout << bc[i];
    }
    std::cout << std::endl;
}

int main()
{
    boolcontainer bc;
    printboolcontainer(bc);
    bc[0] = true;
    bc[3] = true;
    bc[39] = true;
    printboolcontainer(bc);
}
于 2013-02-11T07:58:08.057 回答