Могу ли я объединить std :: tuple из Eigen :: Vectors? - PullRequest
Новая
       126

Могу ли я объединить std :: tuple из Eigen :: Vectors?

0 голосов
/ 08 мая 2020

Если у меня есть std::tuple статически выделенных Eigen::Vector s (из популярной библиотеки Eigen ), например 

std::tuple<Eigen::Vector2f, Eigen::Vector3f, Eigen::Vector2f>

Есть ли способ включить это в один Eigen::Vector7f (т.е. Eigen::Matrix<float, 7, 1>) из трех сцепленных векторов? Такое ощущение, что я смогу сделать это во время компиляции, учитывая, что размеры и типы всего известны.

1 Ответ

1 голос
/ 08 мая 2020

Вы можете написать: 

template<int R1, int R2, int R3>
auto foo(std::tuple< 
    Eigen::Matrix<float,R1,1>, 
    Eigen::Matrix<float,R2,1>, 
    Eigen::Matrix<float,R3,1> > t)
{
    Eigen::Matrix<float,R1+R2+R3,1> res;
    res.template block<R1,1>(0,0) = std::get<0>(t);
    res.template block<R2,1>(R1,0) = std::get<1>(t);
    res.template block<R3,1>(R1+R2,0) = std::get<2>(t);
    return res;
}

int main() {
    Eigen::Vector2f v1;
    v1 << 1,2;
    Eigen::Vector3f v2;
    v2 << 3,4,5;
    std::cout << foo(std::make_tuple(v1,v2,v1)) << std::endl;

в качестве вывода:

1,2,3,4,5,1,2

Live demo

Ниже представлена ​​более общая c версия, использующая несколько векторов в качестве компонентов кортежа: 

template<class RES, int ... R, size_t ... Indices>
void concatenateHelper(RES& res, 
    const std::tuple< Eigen::Matrix<float,R,1>... >& t,
    std::index_sequence<Indices...>)
{
    int idx = 0;
    int fakeArray [] = {(res.template block<R,1>(idx,0) = std::get<Indices>(t),idx += R,0)...};
    static_cast<void>(fakeArray);
}

template<int ... R>
auto concatenate(const std::tuple< Eigen::Matrix<float,R,1> ... >& t) 
{
    Eigen::Matrix<float, (R + ...),1> res;
    concatenateHelper(res,t,std::make_index_sequence<sizeof...(R)>{});
    return res;
}

Demo

...