Как передать строку boost :: multi_array и std :: vector по ссылке на одну и ту же функцию шаблона? - PullRequest
Новая
       21

Как передать строку boost :: multi_array и std :: vector по ссылке на одну и ту же функцию шаблона?

5 голосов
/ 25 января 2010

У меня проблема с этим битом кода: 

#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>]'

Он работает как положено для boost :: multi_array, когда я изменяю тип первого аргумента функции foo с Vec& на Vec, но тогда std :: vector передается по значению, которое не что я хочу. Как мне достичь своей цели без написания двух шаблонов?

1 Ответ

1 голос
/ 25 января 2010

Проблема в том, что для NumDims> 1 , operator[] возвращает временный объект типа 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

Альтернативой может быть обернуть вашу реализацию и обеспечить перегрузку для случая с несколькими массивами .... Например: 1011

(примечание: я не видел, как заставить перегрузку работать с 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);
}
...