Comment passer une ligne de boost :: multi_array et std :: vecteur par référence à la même fonction de modèle?
-
22-09-2019 - |
Question
J'ai un problème avec ce morceau de code:
#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;
}
Essayer de le compiler avec gcc, je reçois l'erreur:
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>]'
Il fonctionne comme prévu pour boost :: multi_array quand je change le type du premier argument de la fonction foo
de Vec&
à Vec
, mais le std :: vecteur est passé par valeur, ce qui est pas ce que je veux. Comment puis-je atteindre mon objectif sans écrire deux modèles?
La solution
Le problème est que pour NumDims> 1 , operator[]
retourne un objet temporaire de type template subarray<NumDims-1>::type
.
A (pas si beau) travail autour serait quelque chose comme ce qui suit:
typedef boost::multi_array<double, 2> MA;
MA m1;
MA::reference ref = m1[0];
foo(ref, 5); // ref is no temporary now
Une alternative serait d'envelopper votre mise en œuvre et de fournir une surcharge pour le cas multi-gamme .... par exemple:.
(note: je ne vois pas comment la surcharge de travailler avec boost::multi_array<T,N>::reference
, s'il vous plaît ne mettez pas en usage productif avec cette version 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);
}