modules/ocl/test/test_fft.cpp

   1 /*M///////////////////////////////////////////////////////////////////////////////////////
   2 //
   3 //  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
   4 //
   5 //  By downloading, copying, installing or using the software you agree to this license.
   6 //  If you do not agree to this license, do not download, install,
   7 //  copy or use the software.
   8 //
   9 //
  10 //                           License Agreement
  11 //                For Open Source Computer Vision Library
  12 //
  13 // Copyright (C) 2010-2012, Multicoreware, Inc., all rights reserved.
  14 // Copyright (C) 2010-2012, Advanced Micro Devices, Inc., all rights reserved.
  15 // Third party copyrights are property of their respective owners.
  16 //
  17 // @Authors
  18 //    Peng Xiao, pengxiao@multicorewareinc.com
  19 //
  20 // Redistribution and use in source and binary forms, with or without modification,
  21 // are permitted provided that the following conditions are met:
  22 //
  23 //   * Redistribution's of source code must retain the above copyright notice,
  24 //     this list of conditions and the following disclaimer.
  25 //
  26 //   * Redistribution's in binary form must reproduce the above copyright notice,
  27 //     this list of conditions and the following disclaimer in the documentation
  28 //     and/or other oclMaterials provided with the distribution.
  29 //
  30 //   * The name of the copyright holders may not be used to endorse or promote products
  31 //     derived from this software without specific prior written permission.
  32 //
  33 // This software is provided by the copyright holders and contributors as is and
  34 // any express or implied warranties, including, but not limited to, the implied
  35 // warranties of merchantability and fitness for a particular purpose are disclaimed.
  36 // In no event shall the Intel Corporation or contributors be liable for any direct,
  37 // indirect, incidental, special, exemplary, or consequential damages
  38 // (including, but not limited to, procurement of substitute goods or services;
  39 // loss of use, data, or profits; or business interruption) however caused
  40 // and on any theory of liability, whether in contract, strict liability,
  41 // or tort (including negligence or otherwise) arising in any way out of
  42 // the use of this software, even if advised of the possibility of such damage.
  43 //
  44 //M*/
  45
  46 #include "test_precomp.hpp"
  47
  48 using namespace std;
  49
  50 #ifdef HAVE_CLAMDFFT
  51
  52 ////////////////////////////////////////////////////////////////////////////
  53 // Dft
  54
  55 PARAM_TEST_CASE(Dft, cv::Size, int)
  56 {
  57     cv::Size dft_size;
  58     int  dft_flags;
  59     virtual void SetUp()
  60     {
  61         dft_size  = GET_PARAM(0);
  62         dft_flags = GET_PARAM(1);
  63     }
  64 };
  65
  66 OCL_TEST_P(Dft, C2C)
  67 {
  68     cv::Mat a = randomMat(dft_size, CV_32FC2, 0.0, 100.0);
  69     cv::Mat b_gold;
  70
  71     cv::ocl::oclMat d_b;
  72
  73     cv::dft(a, b_gold, dft_flags);
  74     cv::ocl::dft(cv::ocl::oclMat(a), d_b, a.size(), dft_flags);
  75     EXPECT_MAT_NEAR(b_gold, cv::Mat(d_b), a.size().area() * 1e-4);
  76 }
  77
  78 OCL_TEST_P(Dft, R2C)
  79 {
  80     cv::Mat a = randomMat(dft_size, CV_32FC1, 0.0, 100.0);
  81     cv::Mat b_gold, b_gold_roi;
  82
  83     cv::ocl::oclMat d_b, d_c;
  84     cv::ocl::dft(cv::ocl::oclMat(a), d_b, a.size(), dft_flags);
  85     cv::dft(a, b_gold, cv::DFT_COMPLEX_OUTPUT | dft_flags);
  86
  87     b_gold_roi = b_gold(cv::Rect(0, 0, d_b.cols, d_b.rows));
  88     EXPECT_MAT_NEAR(b_gold_roi, cv::Mat(d_b), a.size().area() * 1e-4);
  89
  90     cv::Mat c_gold;
  91     cv::dft(b_gold, c_gold, cv::DFT_INVERSE | cv::DFT_REAL_OUTPUT | cv::DFT_SCALE);
  92     EXPECT_MAT_NEAR(b_gold_roi, cv::Mat(d_b), a.size().area() * 1e-4);
  93 }
  94
  95 OCL_TEST_P(Dft, R2CthenC2R)
  96 {
  97     cv::Mat a = randomMat(dft_size, CV_32FC1, 0.0, 10.0);
  98
  99     cv::ocl::oclMat d_b, d_c;
 100     cv::ocl::dft(cv::ocl::oclMat(a), d_b, a.size(), 0);
 101     cv::ocl::dft(d_b, d_c, a.size(), cv::DFT_SCALE | cv::DFT_INVERSE | cv::DFT_REAL_OUTPUT);
 102     EXPECT_MAT_NEAR(a, d_c, a.size().area() * 1e-4);
 103 }
 104
 105
 106 INSTANTIATE_TEST_CASE_P(OCL_ImgProc, Dft, testing::Combine(
 107                             testing::Values(cv::Size(2, 3), cv::Size(5, 4), cv::Size(25, 20), cv::Size(512, 1), cv::Size(1024, 768)),
 108                             testing::Values(0, (int)cv::DFT_ROWS, (int)cv::DFT_SCALE) ));
 109
 110 #endif // HAVE_CLAMDFFT