reproducer.cpp

Ilya Lavrenov, 2013-10-22 08:34 pm

Download (2 kB)

 
1
PARAM_TEST_CASE(ImgprocTestBase, MatType,
2
                int, // blockSize
3
                int, // border type
4
                bool) // roi or not
5
{
6
    int type, borderType, blockSize;
7
    bool useRoi;
8
9
    Mat src, dst_whole, src_roi, dst_roi;
10
    ocl::oclMat gsrc_whole, gsrc_roi, gdst_whole, gdst_roi;
11
12
    virtual void SetUp()
13
    {
14
        type = GET_PARAM(0);
15
        blockSize = GET_PARAM(1);
16
        borderType = GET_PARAM(2);
17
        useRoi = GET_PARAM(3);
18
    }
19
20
    void random_roi()
21
    {
22
        Size roiSize = randomSize(1, MAX_VALUE);
23
        Border srcBorder = randomBorder(0, useRoi ? MAX_VALUE : 0);
24
        randomSubMat(src, src_roi, roiSize, srcBorder, type, 5, 256);
25
26
        Border dstBorder = randomBorder(0, useRoi ? MAX_VALUE : 0);
27
        randomSubMat(dst_whole, dst_roi, roiSize, dstBorder, type, 5, 16);
28
29
        generateOclMat(gsrc_whole, gsrc_roi, src, roiSize, srcBorder);
30
        generateOclMat(gdst_whole, gdst_roi, dst_whole, roiSize, dstBorder);
31
    }
32
33
    void Near(double threshold = 0.0)
34
    {
35
        Mat whole, roi;
36
        gdst_whole.download(whole);
37
        gdst_roi.download(roi);
38
39
        EXPECT_MAT_NEAR(dst_whole, whole, threshold);
40
        EXPECT_MAT_NEAR(dst_roi, roi, threshold);
41
    }
42
};
43
44
typedef ImgprocTestBase CornerHarris;
45
46
OCL_TEST_P(CornerHarris, Mat)
47
{
48
    for (int j = 0; j < LOOP_TIMES; j++)
49
    {
50
        random_roi();
51
52
        int apertureSize = 3;
53
        double k = 2.0;
54
55
        cornerHarris(src_roi, dst_roi, blockSize, apertureSize, k, borderType);
56
        ocl::cornerHarris(gsrc_roi, gdst_roi, blockSize, apertureSize, k, borderType);
57
58
        Near(1.0);
59
    }
60
}
61
62
INSTANTIATE_TEST_CASE_P(Imgproc, CornerHarris, Combine(
63
                            Values((MatType)CV_8UC1, CV_32FC1), // TODO does not work properly with CV_32FC1
64
                            Values(3, 5),
65
                            Values((int)BORDER_REFLECT101, (int)BORDER_REFLECT, (int)BORDER_CONSTANT, (int)BORDER_REPLICATE),
66
                            Bool()));