使用bitset访问向量索引

时间:2015-01-31 21:57:41

标签: c++

我正在尝试编写自己的实体组件系统,我想要一个快速(尽可能的恒定时间),但也需要内存有效的方式来访问实体的组件。

使用全局映射(std::unordered_map)将每个实体映射到包含其组件的有序向量:

Key        Value
---        -----
EntityA -> Sorted vector of components 
EntityB -> Sorted vector of components
[..]    -> [..]

每个实体都有一个bitset,指示实体具有哪些组件即。 :

|C0 |C1 |C2 |C3 |C4 |C5
 0   0   1   0   1   1

and on the map:
Key        Value
---        -----
EntityA -> [C2,C4,C5]

由于组件很少添加,我可以承担分类插入的成本,但我绝对想要快速访问。

现在我从bitset知道C4是第二个元素集(从左边开始计数)所以它应该在第二个向量索引处。

如何将其写入一个方法,该方法将返回给定该组件类型ID的实体组件? 例如

Component* getComponent(ComponentID id){ // component id of C5 should be 6 since it occupies the 6th position of the bitset
return [..];
}

1 个答案:

答案 0 :(得分:1)

假设我们的成员是:

std::bitset<6> bits;
std::vector<Component> comps;

然后:

Component* getComponent(int id) {
    // we need to find how many bits are set for ids < id
    // first, make sure this one is set:
    if (!bits[id]) return nullptr;

    // then, let's count
    int idx = 0;
    for (int i = 0; i < id; ++i) {
        if (bits[i]) ++idx;
    }

    // now, just return the right pointer
    return &comps[idx];
}

如果要进行边界检查,也可以使用std::bitset::test而不是索引运算符。

更快的解决方案可能是这样的:

Component* getComponent(int id) {
    if (!bits[id]) return nullptr;

    // flip id and get a mask
    // if we have C0 .. C5, and we pass in 4
    // we want the mask 0x111100
    unsigned long mask = (1 << (bits.size() - id)) - 1;
    mask = ~mask;

    // apply the mask to the bitset
    // so from 0x001011, we get 0x001000
    unsigned long new_bits = bits.to_ulong() & mask;

    // count how many bits are set there
    unsigned int popcnt = __builtin_popcount(new_bits);

    // and there we have it
    return &comps[popcnt];
}