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 "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 std::string workdir;
  77 int main(int argc, char **argv)
  78 {
  79     TS::ptr()->init("ocl");
  80     InitGoogleTest(&argc, argv);
  81     const char *keys =
  82         "{ h | help     | false              | print help message }"
  83         "{ w | workdir  | ../../../samples/c/| set working directory }"
  84         "{ t | type     | gpu                | set device type:cpu or gpu}"
  85         "{ p | platform | 0                  | set platform id }"
  86         "{ d | device   | 0                  | set device id }";
  87
  88     CommandLineParser cmd(argc, argv, keys);
  89     if (cmd.get<bool>("help"))
  90     {
  91         cout << "Avaible options besides goole test option:" << endl;
  92         cmd.printParams();
  93         return 0;
  94     }
  95     workdir = cmd.get<string>("workdir");
  96     string type = cmd.get<string>("type");
  97     unsigned int pid = cmd.get<unsigned int>("platform");
  98     int device = cmd.get<int>("device");
  99
 100     print_info();
 101     int flag = CVCL_DEVICE_TYPE_GPU;
 102     if(type == "cpu")
 103     {
 104         flag = CVCL_DEVICE_TYPE_CPU;
 105     }
 106     std::vector<cv::ocl::Info> oclinfo;
 107     int devnums = getDevice(oclinfo, flag);
 108     if(devnums <= device || device < 0)
 109     {
 110         std::cout << "device invalid\n";
 111         return -1;
 112     }
 113     if(pid >= oclinfo.size())
 114     {
 115         std::cout << "platform invalid\n";
 116         return -1;
 117     }
 118
 119     setDevice(oclinfo[pid], device);
 120
 121     cout << "Device type:" << type << endl << "Device name:" << oclinfo[pid].DeviceName[device] << endl;
 122     return RUN_ALL_TESTS();
 123 }
 124
 125 #else // DON'T HAVE_OPENCL
 126
 127 int main()
 128 {
 129     printf("OpenCV was built without OpenCL support\n");
 130     return 0;
 131 }
 132
 133
 134 #endif // HAVE_OPENCL