#include "perf_precomp.hpp"

using namespace std;
using namespace testing;

namespace {

//////////////////////////////////////////////////////////////////////
// SURF

DEF_PARAM_TEST_1(Image, string);

PERF_TEST_P(Image, Features2D_SURF, Values<string>("gpu/perf/aloe.jpg"))
{
    declare.time(2.0);

    cv::Mat img = readImage(GetParam(), cv::IMREAD_GRAYSCALE);
    ASSERT_FALSE(img.empty());

    cv::gpu::SURF_GPU d_surf;

    cv::gpu::GpuMat d_img(img);
    cv::gpu::GpuMat d_keypoints, d_descriptors;

    d_surf(d_img, cv::gpu::GpuMat(), d_keypoints, d_descriptors);

    TEST_CYCLE()
    {
        d_surf(d_img, cv::gpu::GpuMat(), d_keypoints, d_descriptors);
    }
}

//////////////////////////////////////////////////////////////////////
// FAST

PERF_TEST_P(Image, Features2D_FAST, Values<string>("gpu/perf/aloe.jpg"))
{
    cv::Mat img = readImage(GetParam(), cv::IMREAD_GRAYSCALE);
    ASSERT_FALSE(img.empty());

    cv::gpu::FAST_GPU d_fast(20);

    cv::gpu::GpuMat d_img(img);
    cv::gpu::GpuMat d_keypoints;

    d_fast(d_img, cv::gpu::GpuMat(), d_keypoints);

    TEST_CYCLE()
    {
        d_fast(d_img, cv::gpu::GpuMat(), d_keypoints);
    }
}

//////////////////////////////////////////////////////////////////////
// ORB

PERF_TEST_P(Image, Features2D_ORB, Values<string>("gpu/perf/aloe.jpg"))
{
    cv::Mat img = readImage(GetParam(), cv::IMREAD_GRAYSCALE);
    ASSERT_FALSE(img.empty());

    cv::gpu::ORB_GPU d_orb(4000);

    cv::gpu::GpuMat d_img(img);
    cv::gpu::GpuMat d_keypoints, d_descriptors;

    d_orb(d_img, cv::gpu::GpuMat(), d_keypoints, d_descriptors);

    TEST_CYCLE()
    {
        d_orb(d_img, cv::gpu::GpuMat(), d_keypoints, d_descriptors);
    }
}

//////////////////////////////////////////////////////////////////////
// BFMatch

DEF_PARAM_TEST(DescSize_Norm, int, NormType);

PERF_TEST_P(DescSize_Norm, Features2D_BFMatch, Combine(Values(64, 128, 256), Values(NormType(cv::NORM_L1), NormType(cv::NORM_L2), NormType(cv::NORM_HAMMING))))
{
    declare.time(3.0);

    int desc_size = GET_PARAM(0);
    int normType = GET_PARAM(1);

    int type = normType == cv::NORM_HAMMING ? CV_8U : CV_32F;

    cv::Mat query(3000, desc_size, type);
    fillRandom(query);

    cv::Mat train(3000, desc_size, type);
    fillRandom(train);

    cv::gpu::BFMatcher_GPU d_matcher(normType);

    cv::gpu::GpuMat d_query(query);
    cv::gpu::GpuMat d_train(train);
    cv::gpu::GpuMat d_trainIdx, d_distance;

    d_matcher.matchSingle(d_query, d_train, d_trainIdx, d_distance);

    TEST_CYCLE()
    {
        d_matcher.matchSingle(d_query, d_train, d_trainIdx, d_distance);
    }
}

//////////////////////////////////////////////////////////////////////
// BFKnnMatch

DEF_PARAM_TEST(DescSize_K_Norm, int, int, NormType);

PERF_TEST_P(DescSize_K_Norm, Features2D_BFKnnMatch, Combine(
    Values(64, 128, 256),
    Values(2, 3),
    Values(NormType(cv::NORM_L1), NormType(cv::NORM_L2), NormType(cv::NORM_HAMMING))))
{
    declare.time(3.0);

    int desc_size = GET_PARAM(0);
    int k = GET_PARAM(1);
    int normType = GET_PARAM(2);

    int type = normType == cv::NORM_HAMMING ? CV_8U : CV_32F;

    cv::Mat query(3000, desc_size, type);
    fillRandom(query);

    cv::Mat train(3000, desc_size, type);
    fillRandom(train);

    cv::gpu::BFMatcher_GPU d_matcher(normType);

    cv::gpu::GpuMat d_query(query);
    cv::gpu::GpuMat d_train(train);
    cv::gpu::GpuMat d_trainIdx, d_distance, d_allDist;

    d_matcher.knnMatchSingle(d_query, d_train, d_trainIdx, d_distance, d_allDist, k);

    TEST_CYCLE()
    {
        d_matcher.knnMatchSingle(d_query, d_train, d_trainIdx, d_distance, d_allDist, k);
    }
}

//////////////////////////////////////////////////////////////////////
// BFRadiusMatch

PERF_TEST_P(DescSize_Norm, Features2D_BFRadiusMatch, Combine(Values(64, 128, 256), Values(NormType(cv::NORM_L1), NormType(cv::NORM_L2), NormType(cv::NORM_HAMMING))))
{
    declare.time(3.0);

    int desc_size = GET_PARAM(0);
    int normType = GET_PARAM(1);

    int type = normType == cv::NORM_HAMMING ? CV_8U : CV_32F;

    cv::Mat query(3000, desc_size, type);
    fillRandom(query, 0.0, 1.0);

    cv::Mat train(3000, desc_size, type);
    fillRandom(train, 0.0, 1.0);

    cv::gpu::BFMatcher_GPU d_matcher(normType);

    cv::gpu::GpuMat d_query(query);
    cv::gpu::GpuMat d_train(train);
    cv::gpu::GpuMat d_trainIdx, d_nMatches, d_distance;

    d_matcher.radiusMatchSingle(d_query, d_train, d_trainIdx, d_distance, d_nMatches, 2.0);

    TEST_CYCLE()
    {
        d_matcher.radiusMatchSingle(d_query, d_train, d_trainIdx, d_distance, d_nMatches, 2.0);
    }
}

} // namespace