c++ 如何通过引用将一行boost::multi_array和std::vector传递给同一个模板函数?

ffscu2ro  于 11个月前  发布在  其他
关注(0)|答案(2)|浏览(139)

我对这段代码有一个问题:

#include <boost/multi_array.hpp>
#include <boost/array.hpp>
#include <vector>
#include <iostream>

template <typename Vec>
void foo(Vec& x, size_t N)
{
    for (size_t i = 0; i < N; ++i) {
        x[i] = i;
    }
}

int main()
{
    std::vector<double> v1(10);
    foo(v1, 5);
    std::cout << v1[4] << std::endl;

    boost::multi_array<double, 2> m1;
    boost::array<double, 2> shape;
    shape[0] = 10;
    shape[1] = 10;
    m1.resize(shape);
    foo(m1[0], 5);
    std::cout << m1[0][4] << std::endl;
    return 0;
}

字符串
尝试用gcc编译它,我得到错误:

boost_multi_array.cpp: In function 'int main()':
boost_multi_array.cpp:26: error: invalid initialization of non-const reference of type 'boost::detail::multi_array::sub_array<double, 1u>&' from a temporary of type 'boost::detail::multi_array::sub_array<double, 1u>'
boost_multi_array.cpp:7: error: in passing argument 1 of 'void foo(Vec&, size_t) [with Vec = boost::detail::multi_array::sub_array<double, 1u>]'


当我将函数foo的第一个参数的类型从Vec&更改为Vec时,它可以像boost::multi_array预期的那样工作,但是随后std::vector通过值传递,这不是我想要的。我如何才能在不编写两个模板的情况下实现我的目标?

cngwdvgl

cngwdvgl1#

问题是对于 NumDims > 1operator[]返回一个template subarray<NumDims-1>::type类型的临时对象。
一个(不太好的)解决方法是这样的:

typedef boost::multi_array<double, 2> MA;
MA m1;
MA::reference ref = m1[0];
foo(ref, 5); // ref is no temporary now

字符串
另一种方法是 Package 您的实现并为多数组情况提供重载.例如:

  • (注意:我没有看到如何让重载与boost::multi_array<T,N>::reference一起工作,请不要在这个detail::版本中投入生产使用;)*
template<class T>
void foo_impl(T x, size_t N) {
    for (size_t i = 0; i < N; ++i) {
        x[i] = i;
    }
}

template<class T>
void foo(T& t, size_t n) {
    foo_impl<T&>(t, n);
}

template<typename T, size_t size>
void foo(boost::detail::multi_array::sub_array<T, size> r, size_t n) {
    foo_impl(r, n);
}

0vvn1miw

0vvn1miw2#

解决方案(对于现代来说)应该很容易,可以普遍参考:

template <typename Vec>
void foo(Vec&& x, size_t N)
{
    for (size_t i = 0; i < N; ++i) {
        x[i] = i;
    }
}

字符串
既然你已经在做了,那就让实现更正确(向前),返回一些有用的东西,并给它起个好名字:

template <typename Vec>
void iota_first_n(Vec&& x, size_t N)
{
   auto const last = x.begin() + N;
   std::iota(std::forward<Vec>(x).begin(), last, std::size_t{0});
   return last;
}

相关问题