Tony Barbour | 92400bb | 2015-03-02 16:38:52 -0700 | [diff] [blame] | 1 | #include "test_common.h" |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 2 | #include "vktestbinding.h" |
Tony Barbour | 92400bb | 2015-03-02 16:38:52 -0700 | [diff] [blame] | 3 | #include "test_environment.h" |
| 4 | |
| 5 | #define ARRAY_SIZE(a) (sizeof(a) / sizeof(a[0])) |
| 6 | |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 7 | namespace vk_testing { |
Tony Barbour | 92400bb | 2015-03-02 16:38:52 -0700 | [diff] [blame] | 8 | |
| 9 | Environment::Environment() : |
Chia-I Wu | f869338 | 2015-04-16 22:02:10 +0800 | [diff] [blame] | 10 | default_dev_(0) |
Tony Barbour | 92400bb | 2015-03-02 16:38:52 -0700 | [diff] [blame] | 11 | { |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 12 | app_.sType = VK_STRUCTURE_TYPE_APPLICATION_INFO; |
| 13 | app_.pAppName = "vk_testing"; |
Tony Barbour | 92400bb | 2015-03-02 16:38:52 -0700 | [diff] [blame] | 14 | app_.appVersion = 1; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 15 | app_.pEngineName = "vk_testing"; |
Tony Barbour | 92400bb | 2015-03-02 16:38:52 -0700 | [diff] [blame] | 16 | app_.engineVersion = 1; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 17 | app_.apiVersion = VK_API_VERSION; |
Tony Barbour | 92400bb | 2015-03-02 16:38:52 -0700 | [diff] [blame] | 18 | } |
| 19 | |
| 20 | bool Environment::parse_args(int argc, char **argv) |
| 21 | { |
| 22 | int i; |
| 23 | |
| 24 | for (i = 1; i < argc; i++) { |
| 25 | #define ARG(name) (strcmp(argv[i], name) == 0) |
| 26 | #define ARG_P(name) (i < argc - 1 && ARG(name)) |
| 27 | if (ARG_P("--gpu")) { |
| 28 | default_dev_ = atoi(argv[++i]); |
| 29 | } else { |
| 30 | break; |
| 31 | } |
| 32 | #undef ARG |
| 33 | #undef ARG_P |
| 34 | } |
| 35 | |
| 36 | if (i < argc) { |
| 37 | std::cout << |
| 38 | "invalid argument: " << argv[i] << "\n\n" << |
| 39 | "Usage: " << argv[0] << " <options>\n\n" << |
| 40 | "Options:\n" |
| 41 | " --gpu <n> Use GPU<n> as the default GPU\n"; |
| 42 | |
| 43 | return false; |
| 44 | } |
| 45 | |
| 46 | return true; |
| 47 | } |
| 48 | |
| 49 | void Environment::SetUp() |
| 50 | { |
| 51 | |
| 52 | uint32_t count; |
Courtney Goeltzenleuchter | fb4efc6 | 2015-04-10 08:34:15 -0600 | [diff] [blame] | 53 | VkResult err; |
Courtney Goeltzenleuchter | 95487bc | 2015-04-14 18:48:46 -0600 | [diff] [blame] | 54 | VkInstanceCreateInfo inst_info = {}; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 55 | inst_info.sType = VK_STRUCTURE_TYPE_INSTANCE_CREATE_INFO; |
Jon Ashburn | b317fad | 2015-04-04 14:52:07 -0600 | [diff] [blame] | 56 | inst_info.pNext = NULL; |
| 57 | inst_info.pAppInfo = &app_; |
| 58 | inst_info.pAllocCb = NULL; |
| 59 | inst_info.extensionCount = 0; |
Courtney Goeltzenleuchter | f579fa6 | 2015-06-10 17:39:03 -0600 | [diff] [blame] | 60 | inst_info.pEnabledExtensions = NULL; |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 61 | err = vkCreateInstance(&inst_info, &inst); |
| 62 | ASSERT_EQ(VK_SUCCESS, err); |
Jon Ashburn | 83a6425 | 2015-04-15 11:31:12 -0600 | [diff] [blame] | 63 | err = vkEnumeratePhysicalDevices(inst, &count, NULL); |
| 64 | ASSERT_EQ(VK_SUCCESS, err); |
| 65 | ASSERT_LE(count, ARRAY_SIZE(gpus)); |
| 66 | err = vkEnumeratePhysicalDevices(inst, &count, gpus); |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 67 | ASSERT_EQ(VK_SUCCESS, err); |
Tony Barbour | 92400bb | 2015-03-02 16:38:52 -0700 | [diff] [blame] | 68 | ASSERT_GT(count, default_dev_); |
| 69 | |
| 70 | devs_.reserve(count); |
| 71 | for (uint32_t i = 0; i < count; i++) { |
| 72 | devs_.push_back(new Device(gpus[i])); |
| 73 | if (i == default_dev_) { |
| 74 | devs_[i]->init(); |
| 75 | ASSERT_NE(true, devs_[i]->graphics_queues().empty()); |
| 76 | } |
| 77 | } |
| 78 | } |
| 79 | |
Tony Barbour | 92400bb | 2015-03-02 16:38:52 -0700 | [diff] [blame] | 80 | void Environment::TearDown() |
| 81 | { |
| 82 | // destroy devices first |
| 83 | for (std::vector<Device *>::iterator it = devs_.begin(); it != devs_.end(); it++) |
| 84 | delete *it; |
| 85 | devs_.clear(); |
| 86 | |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 87 | vkDestroyInstance(inst); |
Tony Barbour | 92400bb | 2015-03-02 16:38:52 -0700 | [diff] [blame] | 88 | } |
Courtney Goeltzenleuchter | d8e229c | 2015-04-08 15:36:08 -0600 | [diff] [blame] | 89 | } // vk_testing namespace |