aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorJan Eilers <jan.eilers@arm.com>2020-04-10 12:40:24 +0100
committerJim Flynn <jim.flynn@arm.com>2020-04-10 18:17:02 +0000
commit76bc728bc1681ed216ffe6f7720f3f57b5137fab (patch)
tree0363480577a7e0dbb3759e66cad19debed62170f /src
parentdc0c6ed9f8b993e63f492f203d7d7080ab4c835c (diff)
downloadarmnn-76bc728bc1681ed216ffe6f7720f3f57b5137fab.tar.gz
IVGCVSW-4483 Introduce PolymorphicPointerDowncast
* as replacement for boost::polymorphic_pointer_downcast * added PolymorphicPointerDowncast * added related unit test * added description to PolymorphicDowncast Signed-off-by: Jan Eilers <jan.eilers@arm.com> Change-Id: I47e94344c1c21941865549a5632cfb7cad804d35
Diffstat (limited to 'src')
-rw-r--r--src/armnn/test/UtilityTests.cpp75
1 files changed, 74 insertions, 1 deletions
diff --git a/src/armnn/test/UtilityTests.cpp b/src/armnn/test/UtilityTests.cpp
index d5779c1a76..af56364b15 100644
--- a/src/armnn/test/UtilityTests.cpp
+++ b/src/armnn/test/UtilityTests.cpp
@@ -5,7 +5,6 @@
#include <boost/test/unit_test.hpp>
#include <boost/core/lightweight_test.hpp>
-#include <boost/polymorphic_cast.hpp>
#define ARMNN_POLYMORPHIC_CAST_TESTABLE
#define ARMNN_NUMERIC_CAST_TESTABLE
@@ -56,6 +55,80 @@ BOOST_AUTO_TEST_CASE(PolymorphicDowncast)
}
+BOOST_AUTO_TEST_CASE(PolymorphicPointerDowncast_SharedPointer)
+{
+ using namespace armnn;
+ class Base
+ {
+ public:
+ virtual ~Base(){}
+ float v;
+ };
+
+ class Child1 : public Base
+ {
+ public:
+ int j;
+ };
+
+ class Child2 : public Base
+ {
+ public:
+ char b;
+ };
+
+ std::shared_ptr<Base> base1 = std::make_shared<Child1>();
+
+ std::shared_ptr<Child1> ptr1 = std::static_pointer_cast<Child1>(base1);
+ BOOST_CHECK(ptr1);
+ BOOST_CHECK_NO_THROW(armnn::PolymorphicPointerDowncast<Child1>(base1));
+ BOOST_CHECK(armnn::PolymorphicPointerDowncast<Child1>(base1) == ptr1);
+
+ auto ptr2 = std::dynamic_pointer_cast<Child2>(base1);
+ BOOST_CHECK(!ptr2);
+ BOOST_CHECK_THROW(armnn::PolymorphicPointerDowncast<Child2>(base1), std::bad_cast);
+
+ armnn::IgnoreUnused(ptr1, ptr2);
+}
+
+
+BOOST_AUTO_TEST_CASE(PolymorphicPointerDowncast_BuildInPointer)
+{
+ using namespace armnn;
+ class Base
+ {
+ public:
+ virtual ~Base(){}
+ float v;
+ };
+
+ class Child1 : public Base
+ {
+ public:
+ int j;
+ };
+
+ class Child2 : public Base
+ {
+ public:
+ char b;
+ };
+
+ Child1 child1;
+ Base* base1 = &child1;
+ auto ptr1 = dynamic_cast<Child1*>(base1);
+ BOOST_CHECK(ptr1 != nullptr);
+ BOOST_CHECK_NO_THROW(armnn::PolymorphicPointerDowncast<Child1>(base1));
+ BOOST_CHECK(armnn::PolymorphicPointerDowncast<Child1>(base1) == ptr1);
+
+ auto ptr2 = dynamic_cast<Child2*>(base1);
+ BOOST_CHECK(ptr2 == nullptr);
+ BOOST_CHECK_THROW(armnn::PolymorphicPointerDowncast<Child2>(base1), std::bad_cast);
+
+ armnn::IgnoreUnused(ptr1, ptr2);
+}
+
+
BOOST_AUTO_TEST_CASE(NumericCast)
{
using namespace armnn;