From a4678029b3662920c53ab1fcb71a21894b6151a5 Mon Sep 17 00:00:00 2001 From: Justin Carpentier <justin.carpentier@inria.fr> Date: Fri, 15 Nov 2019 20:34:57 +0100 Subject: [PATCH] core: add getNumpy{Matrix,Array}Type + is{Matrix,Array} methods --- include/eigenpy/details.hpp | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/include/eigenpy/details.hpp b/include/eigenpy/details.hpp index 65b2103c..1cbe269f 100644 --- a/include/eigenpy/details.hpp +++ b/include/eigenpy/details.hpp @@ -119,6 +119,28 @@ namespace eigenpy { return getInstance().CurrentNumpyType; } + + static const PyTypeObject * getNumpyMatrixType() + { + return getInstance().NumpyMatrixType; + } + + static const PyTypeObject * getNumpyArrayType() + { + return getInstance().NumpyArrayType; + } + + static bool isMatrix() + { + return PyType_IsSubtype(reinterpret_cast<PyTypeObject*>(getInstance().CurrentNumpyType.ptr()), + getInstance().NumpyMatrixType); + } + + static bool isArray() + { + return PyType_IsSubtype(reinterpret_cast<PyTypeObject*>(getInstance().CurrentNumpyType.ptr()), + getInstance().NumpyArrayType); + } protected: NumpyType() -- GitLab