Skip to content
Snippets Groups Projects
Commit 0ebd49e0 authored by Christoph Pohl's avatar Christoph Pohl
Browse files

Add more support for conversion from eigen types to NDArrays

parent 81313006
No related branches found
No related tags found
1 merge request!290Draft: Aron/feature/defaults ranges docs
......@@ -135,6 +135,16 @@ namespace armarx::aron::converter
}
}
template<typename T>
static data::NDArrayPtr ConvertFromQuaternion(
const Eigen::Quaternion<T>& quat)
{
data::NDArrayPtr ndArr(new data::NDArray);
ndArr->setShape({1, 4});
ndArr->setData(sizeof(T) * 4, reinterpret_cast <const unsigned char* >(quat.coeffs().data()));
return ndArr;
}
template<typename T>
static data::NDArrayPtr ConvertFromMatrix(
......@@ -148,6 +158,18 @@ namespace armarx::aron::converter
return ndArr;
}
template<typename T, int Rows = Eigen::Dynamic, int Cols = Eigen::Dynamic>
static data::NDArrayPtr ConvertFromMatrix(
const Eigen::Matrix < T, Rows, Cols >& mat)
{
data::NDArrayPtr ndArr(new data::NDArray);
ndArr->setShape({Rows, Cols});
ndArr->setData(sizeof(T) * mat.size(), reinterpret_cast <const unsigned char* >(mat.data()));
return ndArr;
}
// Eigen::Array
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment