我有一个自定义容器,我想在基于范围的for循环中使用它。容器有点基于向量,如下所示:
template<typename T>
class IDMap
{
private:
struct Item {
uint16_t mVersion;
T mItem;
template <typename... Arguments>
Item(uint16_t version, Arguments&&... args) : mVersion(version), mItem(args...)
{
}
};
public:
typedef uint32_t ItemID;
template <typename... Arguments>
ItemID AddItem(Arguments&&... args);
void MarkAsFree(const ItemID id);
T& GetItem(const ItemID id);
T* TryGetItem(const ItemID id);
void Clear();
private:
std::vector<Item> mItems;
std::vector<uint16_t> mFreeIndices;
};
我想迭代mItems
向量,但只返回mItem
成员而不是整个Item
结构。有没有简单/优雅的方法来做到这一点?
答案 0 :(得分:5)
您必须提供begin
和end
函数,两者都返回相应的迭代器,迭代器本身实现运算符++
,!=
和*
。 begin
和end
函数可以是独立的,也可以是成员。
从实现具有所需行为的迭代器开始。您可以将其实现为std::vector::iterator
的包装,以保存大部分&#34;核心&#34;工作
以下是未经测试的代码
基本上,在类IDMap中,添加:
class ItemIterator {
// based on vector iterator
std::vector<Item>::iterator i;
public:
ItemIterator(std::vector<Item>::iterator i) : i(i) {}
// incrementing
ItemIterator & operator ++() { ++i; return *this; }
ItemIterator operator ++(int) { const_iterator old(*this); ++(*this); return old; }
// comparison
bool operator!=(const ItemIterator &o) const { return i != o.i; }
// dereferencing
const T & operator*() const { return i->mItem; }
};
using iterator = ItemIterator;
using value_type = T;
ItemIterator begin() const { return ItemIterator(mItems.begin()); }
ItemIterator end() const { return ItemIterator(mItems.end() ); }
如果你想支持多种&#34;特殊迭代&#34;在您的IDMap上,例如在索引上,或在&#34;整个&#34; Item
s,你应该将上面的所有内容包装在另一个适配器中。然后可以使用成员方法访问此适配器,如.items()
。
简要示例:
class IDMap {
// (your code)
public:
struct ItemsAdaptor {
// (insert above iterator definition + usings)
ItemsAdaptor(std::vector<Item>::iterator b,
std::vector<Item>::iterator e)
: b{b}, e{e}
{}
ItemIterator begin() const { return b; }
ItemIterator end() const { return e; }
private:
ItemIterator b, e;
};
ItemsAdaptor items() const {
return ItemsAdaptor(mItems.begin(), mItems.end());
}
};
然后,你可以写:
IDMap<int> map = ...;
for (int i : map.items()) {
...
}
答案 1 :(得分:0)
如果您希望基于范围的for适用于您的容器,则必须提供返回前向迭代器的 begin 和 end 函数。
typedef std::vector<Item>::iterator iterator;
typedef std::vector<Item>::const_iterator const_iterator;
iterator begin()
{
return mItems.begin();
}
const_iterator begin() const;
{
return mItems.begin();
}
//also add end functions, and viola.
这将返回整个项目结构。如果你只需要返回mItem,你必须编写自己的迭代器适配器并使用它而不是vector。