具有保证原子性的单位操作

时间:2017-06-26 00:09:20

标签: c++ multithreading bit-manipulation atomic

有没有办法在c ++中设置,清除,测试和翻转单个位作为原子操作?例如," compare_and_swap"的按位变体。

2 个答案:

答案 0 :(得分:5)

在整数中翻转一个位只是比较和交换操作。您正在使用它来测试和翻转一个位并不能改变任何东西。所以一个简单的compare_exchange_weak循环就可以做到这一点。

答案 1 :(得分:4)

以原子方式操作位需要compare_exchange RMW以避免触及atomic变量中的其他位。 测试一点不是修改操作,因此load()就足够了。

您必须添加范围错误检查

template<typename T, typename OP>
T manipulate_bit(std::atomic<T> &a, unsigned n, OP bit_op)
{
    static_assert(std::is_integral<T>::value, "atomic type not integral");

    T val = a.load();
    while (!a.compare_exchange_weak(val, bit_op(val, n)));

    return val;
}

auto set_bit = [](auto val, unsigned n) { return val | (1 << n); };
auto clr_bit = [](auto val, unsigned n) { return val & ~(1 << n); };
auto tgl_bit = [](auto val, unsigned n) { return val ^ (1 << n); };


int main()
{
    std::atomic<int> a{0x2216};

    manipulate_bit(a, 3, set_bit);  // set bit 3

    manipulate_bit(a, 7, tgl_bit);  // toggle bit 7

    manipulate_bit(a, 13, clr_bit);  // clear bit 13

    bool isset = (a.load() >> 5) & 1;  // testing bit 5
}