我具有可变长度数据结构,多维迭代:
class Iterator
{
public:
static Iterator& init(int dim, int* sizes, void* mem)
{
return *(new (mem) Iterator(dim, sizes));
}
static size_t alloc_size(int dim)
{
return sizeof(Iterator) + sizeof(int) * 2 * dim;
}
void operator++()
{
// increment counters, update pos_ and done_
}
bool done() const { return done_; }
bool pos() const { return pos_; }
private:
Iterator(int dim, int* sizes) : dim_(dim), pos_(0), done_(false)
{
for (int i=0; i<dim_; ++i) size(i) = sizes[i];
for (int i=0; i<dim_; ++i) counter(i) = 0;
}
int dim_;
int pos_;
bool done_;
int size (int i) { return reinterpret_cast<int*>(this+1)[i]; }
int& counter(int i) { return reinterpret_cast<int*>(this+1)[dim_+i]; }
};
迭代器的维度在编译时未知但可能很小,所以我用迭代分配内存alloca
:
void* mem = alloca(Iterator::alloc_size(dim));
for (Iterator& i = Iterator::create(dim, sizes, mem); !i.done(); ++i)
{
// do something with i.pos()
}
是否有迭代分配内存的更优雅的方式? 我知道的一个事实,即在从函数返回时,它的栈被松开,从而alloca
必须在调用者的堆栈帧被使用(例如参见这里 )。 这个回答表明,分配的默认参数来进行:
static Iterator& init(int dim, int* sizes, void* mem = alloca(alloc_size(dim)));
然而优雅,这种解决方案并不能帮助我: Default argument references parameter 'dim'
。 任何建议一个不错的解决方案吗?