可变大小类的分配

Allocation of variable-sized class

本文关键字:分配 小类      更新时间:2023-10-16

我有一个可变长度的数据结构,一个多维迭代器:

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'。有什么好的解决方案的建议吗?

不幸的是,鉴于dim是一个运行时值,除了使用宏之外,没有任何方法可以做到这一点

#define CREATE_ITERATOR(dim, sizes) 
    Iterator::init(dim, sizes, alloca(Iterator::alloc_size(dim)))

您可以将维度参数作为模板参数。

我的建议可能不是您想要的,但为什么不使用create|make_iterator函数来调用alloca呢?

我根本不建议使用alloca。如果dim值很小,那么类中的一些固定大小的缓冲区就足够了。如果dim很大,那么与迭代器上执行的其他操作的复杂性相比,堆分配成本可以忽略不计(请注意,对于非常大的dim值,alloca可能会导致堆栈溢出)。您可以在运行时在固定缓冲区和堆分配之间进行选择,具体取决于dim大小。

所以我建议使用类似于std::string中的小字符串优化的方法。

也许,某种COW(书面复印http://en.wikipedia.org/wiki/Copy-on-write)technique可能对迭代器也很有用。

请注意,此技术不能用于alloca,只能用于堆分配。此外,如果迭代器使用alloca(至少没有越来越多丑陋的宏),那么几乎不可能复制或复制初始化迭代器。

Alloca是邪恶的:)