我正在寻找一些合适的2D元素容器。我想要的是能够使用,例如BOOST_FOREACH
迭代容器的每个元素,我也希望能够构建我的容器的子视图(切片/子范围),也可能遍历它们
现在我正在使用boost::numeric::ublas::matrix
用于这些目的,但是,它对我来说看起来不是一个好的解决方案,因为它是一个BLAS矩阵,尽管它表现得非常好2d元素容器(自定义unbounded
/ bounded
存储也非常甜蜜。)
另一个boost
替代,boost::multi_array
是不好的,因为你不能使用一个BOOST_FOREACH
语句迭代每个元素,因为构造视图的语法非常混乱。
有其他选择吗?
谢谢。
答案 0 :(得分:1)
定义自己的类型(平凡),给它一个迭代器和const_interator(平凡),BOOST_FOREACH也可以使用它。
答案 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_;
};