Chiebot-Mirror
/
opencv
mirror of https://github.com/opencv/opencv.git
8
0
Fork 0
Code Issues Projects Releases Wiki Activity

Extend test suite on GAPI_TRANSFORM macro

Browse Source
pull/15283/head
smirnov-alexey 5 years ago
parent 183fc43a67
commit 4c168e95de
1 changed files with 113 additions and 37 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 150
      modules/gapi/test/gapi_transform_tests.cpp

150
modules/gapi/test/gapi_transform_tests.cpp
Unescape View File

@ -23,6 +23,9 @@ using GMat3 = std::tuple<GMat, GMat, GMat>;
using GScalar = cv::GScalar;
template <typename T> using GArray = cv::GArray<T>;
using ArrayT = int;
using WrongArrayT = char;
GAPI_TRANSFORM(gmat_in_gmat_out, <GMat(GMat)>, "gmat_in_gmat_out")
{
static GMat pattern(GMat) { return {}; }
@ -59,22 +62,34 @@ GAPI_TRANSFORM(gmat_in_gsc_out, <GScalar(GMat)>, "gmat_in_gsc_out")
static GScalar substitute(GMat) { return {}; }
};
GAPI_TRANSFORM(garr_in_gmat_out, <GMat(GArray<int>)>, "garr_in_gmat_out")
GAPI_TRANSFORM(garr_in_gmat_out, <GMat(GArray<ArrayT>)>, "garr_in_gmat_out")
{
static GMat pattern(GArray<ArrayT>) { return {}; }
static GMat substitute(GArray<ArrayT>) { return {}; }
};
GAPI_TRANSFORM(gmat_in_garr_out, <GArray<ArrayT>(GMat)>, "gmat_in_garr_out")
{
static GMat pattern(GArray<int>) { return {}; }
static GMat substitute(GArray<int>) { return {}; }
static GArray<ArrayT> pattern(GMat) { return {}; }
static GArray<ArrayT> substitute(GMat) { return {}; }
};
GAPI_TRANSFORM(gmat_in_garr_out, <GArray<int>(GMat)>, "gmat_in_garr_out")
GAPI_TRANSFORM(garr_in_gscalar_out, <GScalar(GArray<ArrayT>)>, "garr_in_gscalar_out")
{
static GArray<int> pattern(GMat) { return {}; }
static GArray<int> substitute(GMat) { return {}; }
static GScalar pattern(GArray<ArrayT>) { return {}; }
static GScalar substitute(GArray<ArrayT>) { return {}; }
};
GAPI_TRANSFORM(gmat_gsc_garray_in_gmat2_out, <GMat2(GMat, GScalar, GArray<int>)>, "gmat_gsc_garray_in_gmat2_out")
GAPI_TRANSFORM(gscalar_in_garr_out, <GArray<ArrayT>(GScalar)>, "gscalar_in_garr_out")
{
static GMat2 pattern(GMat, GScalar, GArray<int>) { return {}; }
static GMat2 substitute(GMat, GScalar, GArray<int>) { return {}; }
static GArray<ArrayT> pattern(GScalar) { return {}; }
static GArray<ArrayT> substitute(GScalar) { return {}; }
};
GAPI_TRANSFORM(gmat_gsc_garray_in_gmat2_out, <GMat2(GMat, GScalar, GArray<ArrayT>)>, "gmat_gsc_garray_in_gmat2_out")
{
static GMat2 pattern(GMat, GScalar, GArray<ArrayT>) { return {}; }
static GMat2 substitute(GMat, GScalar, GArray<ArrayT>) { return {}; }
};
} // anonymous namespace
@ -90,11 +105,13 @@ TEST(KernelPackageTransform, CreatePackage)
, gmat_in_gsc_out
, garr_in_gmat_out
, gmat_in_garr_out
, garr_in_gscalar_out
, gscalar_in_garr_out
, gmat_gsc_garray_in_gmat2_out
>();
auto tr = pkg.get_transformations();
EXPECT_EQ(9u, tr.size());
EXPECT_EQ(11u, tr.size());
}
TEST(KernelPackageTransform, Include)
@ -116,14 +133,15 @@ TEST(KernelPackageTransform, Combine)
EXPECT_EQ(2u, tr.size());
}
namespace {
namespace
{
template <typename T>
inline bool ProtoContainsT(const cv::GProtoArg &arg) {
return cv::GProtoArg::index_of<T>() == arg.index();
}
} // anonymous namespace
TEST(KernelPackageTransform, gmat_gsc_in_gmat_out)
TEST(KernelPackageTransform, gmat_gsc_garray_in_gmat2_out)
{
auto tr = gmat_gsc_garray_in_gmat2_out::transformation();
@ -135,8 +153,8 @@ TEST(KernelPackageTransform, gmat_gsc_in_gmat_out)
EXPECT_TRUE(ProtoContainsT<GMat>(p.m_ins[0]));
EXPECT_TRUE(ProtoContainsT<GScalar>(p.m_ins[1]));
EXPECT_TRUE(ProtoContainsT<cv::detail::GArrayU>(p.m_ins[2]));
EXPECT_TRUE(cv::util::get<cv::detail::GArrayU>(p.m_ins[2]).holds<int>());
EXPECT_FALSE(cv::util::get<cv::detail::GArrayU>(p.m_ins[2]).holds<char>());
EXPECT_TRUE(cv::util::get<cv::detail::GArrayU>(p.m_ins[2]).holds<ArrayT>());
EXPECT_FALSE(cv::util::get<cv::detail::GArrayU>(p.m_ins[2]).holds<WrongArrayT>());
EXPECT_TRUE(ProtoContainsT<GMat>(p.m_outs[0]));
EXPECT_TRUE(ProtoContainsT<GMat>(p.m_outs[1]));
@ -146,44 +164,102 @@ TEST(KernelPackageTransform, gmat_gsc_in_gmat_out)
check(tr.substitute());
}
TEST(KernelPackageTransform, gmat_in_garr_out)
namespace
{
auto tr = gmat_in_garr_out::transformation();
template<typename InType, typename OutType>
typename std::enable_if<(cv::detail::GTypeTraits<OutType>::kind == cv::detail::ArgKind::GARRAY), void>::type
array_check(const cv::GComputation::Priv &p)
{
EXPECT_TRUE(ProtoContainsT<InType>(p.m_ins[0]));
auto check = [](const cv::GComputation &comp){
EXPECT_TRUE(ProtoContainsT<cv::detail::GArrayU>(p.m_outs[0]));
EXPECT_TRUE(cv::util::get<cv::detail::GArrayU>(p.m_outs[0]).holds<ArrayT>());
EXPECT_FALSE(cv::util::get<cv::detail::GArrayU>(p.m_outs[0]).holds<WrongArrayT>());
}
template<typename InType, typename OutType>
typename std::enable_if<(cv::detail::GTypeTraits<InType>::kind == cv::detail::ArgKind::GARRAY), void>::type
array_check(const cv::GComputation::Priv &p)
{
EXPECT_TRUE(ProtoContainsT<OutType>(p.m_outs[0]));
EXPECT_TRUE(ProtoContainsT<cv::detail::GArrayU>(p.m_ins[0]));
EXPECT_TRUE(cv::util::get<cv::detail::GArrayU>(p.m_ins[0]).holds<ArrayT>());
EXPECT_FALSE(cv::util::get<cv::detail::GArrayU>(p.m_ins[0]).holds<WrongArrayT>());
}
template<typename InType, typename OutType>
typename std::enable_if<(cv::detail::GTypeTraits<InType>::kind == cv::detail::ArgKind::GARRAY ||
cv::detail::GTypeTraits<OutType>::kind == cv::detail::ArgKind::GARRAY), void>::type
args_check(const cv::GComputation &comp)
{
const auto &p = comp.priv();
EXPECT_EQ(1u, p.m_ins.size());
EXPECT_EQ(1u, p.m_outs.size());
EXPECT_TRUE(ProtoContainsT<GMat>(p.m_ins[0]));
EXPECT_TRUE(ProtoContainsT<cv::detail::GArrayU>(p.m_outs[0]));
EXPECT_TRUE(cv::util::get<cv::detail::GArrayU>(p.m_outs[0]).holds<int>());
EXPECT_FALSE(cv::util::get<cv::detail::GArrayU>(p.m_outs[0]).holds<float>());
array_check<InType, OutType>(p);
};
check(tr.pattern());
check(tr.substitute());
template<typename InType, typename OutType>
typename std::enable_if<(cv::detail::GTypeTraits<InType>::kind != cv::detail::ArgKind::GARRAY &&
cv::detail::GTypeTraits<OutType>::kind != cv::detail::ArgKind::GARRAY), void>::type
args_check(const cv::GComputation &comp)
{
const auto &p = comp.priv();
EXPECT_EQ(1u, p.m_ins.size());
EXPECT_EQ(1u, p.m_outs.size());
EXPECT_TRUE(ProtoContainsT<InType>(p.m_ins[0]));
EXPECT_TRUE(ProtoContainsT<OutType>(p.m_outs[0]));
}
} // anonymous namespace
template <typename Transformation, typename InType, typename OutType>
static void transformTest()
{
auto tr = Transformation::transformation();
args_check<InType, OutType>(tr.pattern());
args_check<InType, OutType>(tr.substitute());
}
TEST(KernelPackageTransform, garr_in_gmat_out)
TEST(KernelPackageTransform, gmat_in_gmat_out)
{
auto tr = garr_in_gmat_out::transformation();
transformTest<gmat_in_gmat_out, GMat, GMat>();
}
auto check = [](const cv::GComputation &comp){
const auto &p = comp.priv();
EXPECT_EQ(1u, p.m_ins.size());
EXPECT_EQ(1u, p.m_outs.size());
TEST(KernelPackageTransform, gmatp_in_gmatp_out)
{
transformTest<gmatp_in_gmatp_out, GMatP, GMatP>();
}
EXPECT_TRUE(ProtoContainsT<cv::detail::GArrayU>(p.m_ins[0]));
EXPECT_TRUE(cv::util::get<cv::detail::GArrayU>(p.m_ins[0]).holds<int>());
EXPECT_FALSE(cv::util::get<cv::detail::GArrayU>(p.m_ins[0]).holds<bool>());
TEST(KernelPackageTransform, gsc_in_gmat_out)
{
transformTest<gsc_in_gmat_out, GScalar, GMat>();
}
EXPECT_TRUE(ProtoContainsT<GMat>(p.m_outs[0]));
};
TEST(KernelPackageTransform, gmat_in_gsc_out)
{
transformTest<gmat_in_gsc_out, GMat, GScalar>();
}
check(tr.pattern());
check(tr.substitute());
TEST(KernelPackageTransform, garr_in_gmat_out)
{
transformTest<garr_in_gmat_out, GArray<ArrayT>, GMat>();
}
TEST(KernelPackageTransform, gmat_in_garr_out)
{
transformTest<gmat_in_garr_out, GMat, GArray<ArrayT>>();
}
TEST(KernelPackageTransform, garr_in_gscalar_out)
{
transformTest<garr_in_gscalar_out, GArray<ArrayT>, GScalar>();
}
TEST(KernelPackageTransform, gscalar_in_garr_out)
{
transformTest<gscalar_in_garr_out, GScalar, GArray<ArrayT>>();
}
} // namespace opencv_test

Write Preview
Loading…
Cancel
Save