modules/ocl/test/main.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 //                        Intel License Agreement
  11 //                For Open Source Computer Vision Library
  12 //
  13 // Copyright (C) 2000, Intel Corporation, all rights reserved.
  14 // Third party copyrights are property of their respective owners.
  15 //
  16 // Redistribution and use in source and binary forms, with or without modification,
  17 // are permitted provided that the following conditions are met:
  18 //
  19 //   * Redistribution's of source code must retain the above copyright notice,
  20 //     this list of conditions and the following disclaimer.
  21 //
  22 //   * Redistribution's in binary form must reproduce the above copyright notice,
  23 //     this list of conditions and the following disclaimer in the documentation
  24 //     and/or other materials provided with the distribution.
  25 //
  26 //   * The name of Intel Corporation may not be used to endorse or promote products
  27 //     derived from this software without specific prior written permission.
  28 //
  29 // This software is provided by the copyright holders and contributors "as is" and
  30 // any express or implied warranties, including, but not limited to, the implied
  31 // warranties of merchantability and fitness for a particular purpose are disclaimed.
  32 // In no event shall the Intel Corporation or contributors be liable for any direct,
  33 // indirect, incidental, special, exemplary, or consequential damages
  34 // (including, but not limited to, procurement of substitute goods or services;
  35 // loss of use, data, or profits; or business interruption) however caused
  36 // and on any theory of liability, whether in contract, strict liability,
  37 // or tort (including negligence or otherwise) arising in any way out of
  38 // the use of this software, even if advised of the possibility of such damage.
  39 //
  40 //M*/
  41
  42 #include "test_precomp.hpp"
  43
  44 #ifdef HAVE_OPENCL
  45
  46 using namespace std;
  47 using namespace cv;
  48 using namespace cv::ocl;
  49 using namespace cvtest;
  50 using namespace testing;
  51
  52 void print_info()
  53 {
  54     printf("\n");
  55 #if defined _WIN32
  56 #   if defined _WIN64
  57     puts("OS: Windows 64");
  58 #   else
  59     puts("OS: Windows 32");
  60 #   endif
  61 #elif defined linux
  62 #   if defined _LP64
  63     puts("OS: Linux 64");
  64 #   else
  65     puts("OS: Linux 32");
  66 #   endif
  67 #elif defined __APPLE__
  68 #   if defined _LP64
  69     puts("OS: Apple 64");
  70 #   else
  71     puts("OS: Apple 32");
  72 #   endif
  73 #endif
  74
  75 }
  76 int main(int argc, char **argv)
  77 {
  78     TS::ptr()->init(".");
  79     InitGoogleTest(&argc, argv);
  80     const char *keys =
  81         "{ h | help     | false              | print help message }"
  82         "{ t | type     | gpu                | set device type:cpu or gpu}"
  83         "{ p | platform | 0                  | set platform id }"
  84         "{ d | device   | 0                  | set device id }";
  85
  86     CommandLineParser cmd(argc, argv, keys);
  87     if (cmd.get<bool>("help"))
  88     {
  89         cout << "Avaible options besides goole test option:" << endl;
  90         cmd.printParams();
  91         return 0;
  92     }
  93     string type = cmd.get<string>("type");
  94     unsigned int pid = cmd.get<unsigned int>("platform");
  95     int device = cmd.get<int>("device");
  96
  97     print_info();
  98     int flag = CVCL_DEVICE_TYPE_GPU;
  99     if(type == "cpu")
 100     {
 101         flag = CVCL_DEVICE_TYPE_CPU;
 102     }
 103     std::vector<cv::ocl::Info> oclinfo;
 104     int devnums = getDevice(oclinfo, flag);
 105     if(devnums <= device || device < 0)
 106     {
 107         std::cout << "device invalid\n";
 108         return -1;
 109     }
 110     if(pid >= oclinfo.size())
 111     {
 112         std::cout << "platform invalid\n";
 113         return -1;
 114     }
 115
 116     setDevice(oclinfo[pid], device);
 117
 118     setBinaryDiskCache(CACHE_UPDATE);
 119
 120     cout << "Device type:" << type << endl << "Device name:" << oclinfo[pid].DeviceName[device] << endl;
 121     return RUN_ALL_TESTS();
 122 }
 123
 124 #else // DON'T HAVE_OPENCL
 125
 126 int main()
 127 {
 128     printf("OpenCV was built without OpenCL support\n");
 129     return 0;
 130 }
 131
 132
 133 #endif // HAVE_OPENCL