我有一个N字符的掩码大小,这是静态知道的(即可以在编译时计算,但它不是一个常量,所以我不能把它写下来),位设置为1表示“通缉”位。我有一个相同大小的值,只在运行时才知道。我想按顺序从该值中收集“想要的”位到新值的开头。为简单起见,我们假设所需位数是< = 32。
完全未经优化的参考代码,希望具有正确的行为:
template<int N, const char mask[N]>
unsigned gather_bits(const char* val)
{
unsigned result = 0;
char* result_p = (char*)&result;
int pos = 0;
for (int i = 0; i < N * CHAR_BIT; i++)
{
if (mask[i/CHAR_BIT] & (1 << (i % CHAR_BIT)))
{
if (val[i/CHAR_BIT] & (1 << (i % CHAR_BIT)))
{
if (pos < sizeof(unsigned) * CHAR_BIT)
{
result_p[pos/CHAR_BIT] |= 1 << (pos % CHAR_BIT);
}
else
{
abort();
}
}
pos += 1;
}
}
return result;
}
虽然我不确定该公式是否实际上允许在编译时访问掩码的内容。但无论如何,它可以使用,可能是一个constexpr
函数或者更好的主意。我不是在这里寻找必要的C ++魔法(我会想出来的),只是算法。
输入/输出的示例,为清晰起见,使用16位值和虚数二进制表示法:
mask = 0b0011011100100110
val = 0b0101000101110011
--
wanted = 0b__01_001__1__01_ // retain only those bits which are set in the mask
result = 0b0000000001001101 // bring them to the front
^ gathered bits begin here
我的问题是:
最有效的方法是什么? (有没有可以提供帮助的硬件说明?)
如果掩码和值都被限制为unsigned
,那么单个字,而不是无界字符数组会怎样?那么可以用固定的短序列指令来完成吗?
答案 0 :(得分:4)
将pext
(并行位提取)与您在英特尔Haswell中完全相同。我不知道该指令的性能如何,可能比其他选择更好。这个操作也称为“压缩权限”或简称“压缩”,Hacker's Delight的实现就是:
unsigned compress(unsigned x, unsigned m) {
unsigned mk, mp, mv, t;
int i;
x = x & m; // Clear irrelevant bits.
mk = ~m << 1; // We will count 0's to right.
for (i = 0; i < 5; i++) {
mp = mk ^ (mk << 1); // Parallel prefix.
mp = mp ^ (mp << 2);
mp = mp ^ (mp << 4);
mp = mp ^ (mp << 8);
mp = mp ^ (mp << 16);
mv = mp & m; // Bits to move.
m = m ^ mv | (mv >> (1 << i)); // Compress m.
t = x & mv;
x = x ^ t | (t >> (1 << i)); // Compress x.
mk = mk & ~mp;
}
return x;
}