C ++容器问题

时间:2010-07-16 15:36:00

标签: c++ boost containers

我正在寻找一些合适的2D元素容器。我想要的是能够使用,例如BOOST_FOREACH迭代容器的每个元素,我也希望能够构建我的容器的子视图(切片/子范围),也可能遍历它们

现在我正在使用boost::numeric::ublas::matrix用于这些目的,但是,它对我来说看起来不是一个好的解决方案,因为它是一个BLAS矩阵,尽管它表现得非常好2d元素容器(自定义unbounded / bounded存储也非常甜蜜。)

另一个boost替代,boost::multi_array是不好的,因为你不能使用一个BOOST_FOREACH语句迭代每个元素,因为构造视图的语法非常混乱。

有其他选择吗?

谢谢。

2 个答案:

答案 0 :(得分:1)

定义自己的类型(平凡),给它一个迭代器和const_interator(平凡),BOOST_FOREACH也可以使用它。

http://beta.boost.org/doc/libs/1_39_0/doc/html/foreach.html

答案 1 :(得分:1)

我执行以下操作(数组类型是容器/迭代器范围概念):

ublas::matrix<douple> A;
foreach (double & element, A.data())
{
}

但是,这对切片不起作用:最好的解决方案是为它们编写迭代器。

以下是使用multi_array提供自定义类存储的示例。 也许你也可以这样做:

template<size_t N, typename T>
struct tensor_array : boost::multi_array_ref<T,N> {
    typedef boost::multi_array_ref<T,N> base_type;

    typedef T value_type;
    typedef T& reference;
    typedef const T& const_reference;

    tensor_array() : base_type(NULL, extents())
    {
        // std::cout << "create" << std::endl;
    }
    template<class A>
    tensor_array(const A &dims)
        : base_type(NULL, extents())
    {
        //std::cout << "create" << std::endl;
        resize(dims);
    }

    template<typename U>
    void resize(const U (&dims)[N]) {
        boost::array<U,N> dims_;
        std::copy(dims, dims + N, dims_.begin());
        resize(dims_);
    }

    template<typename U>
    void resize(const boost::array<U,N> &dims) {
        size_t size = 1;
        boost::array<size_t,N> shape;
        for (size_t i = 0; i < N; ++i)  {
            size *= dims[i];
            shape[N-(i+1)] = dims[i];
        }
        data_.clear();
        data_.resize(size, 0);
        // update base_type parent
        set_base_ptr(&data_[0]);
        this->num_elements_ = size;
        reshape(shape);
    }

    size_t size() const { return data_.size(); }
    size_t size(size_t i) const { return this->shape()[N-(i+1)]; }
    tensor_array& fill(const T &value) {
        std::fill(data_.begin(), data_.end(), value);
        return *this;
    }
private:
    typedef boost::detail::multi_array::extent_gen<N> extents;
    std::vector<T> data_;
};
相关问题