OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2013 The WebM project authors. All Rights Reserved. | 2 * Copyright (c) 2013 The WebM project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #include <string> | 11 #include <string> |
12 | 12 |
13 #include "third_party/googletest/src/include/gtest/gtest.h" | 13 #include "third_party/googletest/src/include/gtest/gtest.h" |
14 #include "./vpx_config.h" | 14 #include "./vpx_config.h" |
15 #include "test/codec_factory.h" | 15 #include "test/codec_factory.h" |
16 #include "test/decode_test_driver.h" | 16 #include "test/decode_test_driver.h" |
17 #include "test/md5_helper.h" | 17 #include "test/md5_helper.h" |
18 #if CONFIG_WEBM_IO | 18 #if CONFIG_WEBM_IO |
19 #include "test/webm_video_source.h" | 19 #include "test/webm_video_source.h" |
20 #endif | 20 #endif |
21 #include "vp9/decoder/vp9_thread.h" | 21 #include "vp9/common/vp9_thread.h" |
22 | 22 |
23 namespace { | 23 namespace { |
24 | 24 |
25 using std::string; | 25 using std::string; |
26 | 26 |
27 class VP9WorkerThreadTest : public ::testing::TestWithParam<bool> { | 27 class VP9WorkerThreadTest : public ::testing::TestWithParam<bool> { |
28 protected: | 28 protected: |
29 virtual ~VP9WorkerThreadTest() {} | 29 virtual ~VP9WorkerThreadTest() {} |
30 virtual void SetUp() { | 30 virtual void SetUp() { |
31 vp9_get_worker_interface()->init(&worker_); | 31 vp9_get_worker_interface()->init(&worker_); |
32 } | 32 } |
33 | 33 |
34 virtual void TearDown() { | 34 virtual void TearDown() { |
35 vp9_get_worker_interface()->end(&worker_); | 35 vp9_get_worker_interface()->end(&worker_); |
36 } | 36 } |
37 | 37 |
| 38 void Run(VP9Worker* worker) { |
| 39 const bool synchronous = GetParam(); |
| 40 if (synchronous) { |
| 41 vp9_get_worker_interface()->execute(worker); |
| 42 } else { |
| 43 vp9_get_worker_interface()->launch(worker); |
| 44 } |
| 45 } |
| 46 |
38 VP9Worker worker_; | 47 VP9Worker worker_; |
39 }; | 48 }; |
40 | 49 |
41 int ThreadHook(void* data, void* return_value) { | 50 int ThreadHook(void* data, void* return_value) { |
42 int* const hook_data = reinterpret_cast<int*>(data); | 51 int* const hook_data = reinterpret_cast<int*>(data); |
43 *hook_data = 5; | 52 *hook_data = 5; |
44 return *reinterpret_cast<int*>(return_value); | 53 return *reinterpret_cast<int*>(return_value); |
45 } | 54 } |
46 | 55 |
47 TEST_P(VP9WorkerThreadTest, HookSuccess) { | 56 TEST_P(VP9WorkerThreadTest, HookSuccess) { |
48 // should be a no-op. | 57 // should be a no-op. |
49 EXPECT_NE(vp9_get_worker_interface()->sync(&worker_), 0); | 58 EXPECT_NE(vp9_get_worker_interface()->sync(&worker_), 0); |
50 | 59 |
51 for (int i = 0; i < 2; ++i) { | 60 for (int i = 0; i < 2; ++i) { |
52 EXPECT_NE(vp9_get_worker_interface()->reset(&worker_), 0); | 61 EXPECT_NE(vp9_get_worker_interface()->reset(&worker_), 0); |
53 | 62 |
54 int hook_data = 0; | 63 int hook_data = 0; |
55 int return_value = 1; // return successfully from the hook | 64 int return_value = 1; // return successfully from the hook |
56 worker_.hook = ThreadHook; | 65 worker_.hook = ThreadHook; |
57 worker_.data1 = &hook_data; | 66 worker_.data1 = &hook_data; |
58 worker_.data2 = &return_value; | 67 worker_.data2 = &return_value; |
59 | 68 |
60 const bool synchronous = GetParam(); | 69 Run(&worker_); |
61 if (synchronous) { | |
62 vp9_get_worker_interface()->execute(&worker_); | |
63 } else { | |
64 vp9_get_worker_interface()->launch(&worker_); | |
65 } | |
66 EXPECT_NE(vp9_get_worker_interface()->sync(&worker_), 0); | 70 EXPECT_NE(vp9_get_worker_interface()->sync(&worker_), 0); |
67 EXPECT_FALSE(worker_.had_error); | 71 EXPECT_FALSE(worker_.had_error); |
68 EXPECT_EQ(5, hook_data); | 72 EXPECT_EQ(5, hook_data); |
69 | 73 |
70 // should be a no-op. | 74 // should be a no-op. |
71 EXPECT_NE(vp9_get_worker_interface()->sync(&worker_), 0); | 75 EXPECT_NE(vp9_get_worker_interface()->sync(&worker_), 0); |
72 } | 76 } |
73 } | 77 } |
74 | 78 |
75 TEST_P(VP9WorkerThreadTest, HookFailure) { | 79 TEST_P(VP9WorkerThreadTest, HookFailure) { |
76 EXPECT_NE(vp9_get_worker_interface()->reset(&worker_), 0); | 80 EXPECT_NE(vp9_get_worker_interface()->reset(&worker_), 0); |
77 | 81 |
78 int hook_data = 0; | 82 int hook_data = 0; |
79 int return_value = 0; // return failure from the hook | 83 int return_value = 0; // return failure from the hook |
80 worker_.hook = ThreadHook; | 84 worker_.hook = ThreadHook; |
81 worker_.data1 = &hook_data; | 85 worker_.data1 = &hook_data; |
82 worker_.data2 = &return_value; | 86 worker_.data2 = &return_value; |
83 | 87 |
84 const bool synchronous = GetParam(); | 88 Run(&worker_); |
85 if (synchronous) { | |
86 vp9_get_worker_interface()->execute(&worker_); | |
87 } else { | |
88 vp9_get_worker_interface()->launch(&worker_); | |
89 } | |
90 EXPECT_FALSE(vp9_get_worker_interface()->sync(&worker_)); | 89 EXPECT_FALSE(vp9_get_worker_interface()->sync(&worker_)); |
91 EXPECT_EQ(1, worker_.had_error); | 90 EXPECT_EQ(1, worker_.had_error); |
92 | 91 |
93 // Ensure _reset() clears the error and _launch() can be called again. | 92 // Ensure _reset() clears the error and _launch() can be called again. |
94 return_value = 1; | 93 return_value = 1; |
95 EXPECT_NE(vp9_get_worker_interface()->reset(&worker_), 0); | 94 EXPECT_NE(vp9_get_worker_interface()->reset(&worker_), 0); |
96 EXPECT_FALSE(worker_.had_error); | 95 EXPECT_FALSE(worker_.had_error); |
97 vp9_get_worker_interface()->launch(&worker_); | 96 vp9_get_worker_interface()->launch(&worker_); |
98 EXPECT_NE(vp9_get_worker_interface()->sync(&worker_), 0); | 97 EXPECT_NE(vp9_get_worker_interface()->sync(&worker_), 0); |
99 EXPECT_FALSE(worker_.had_error); | 98 EXPECT_FALSE(worker_.had_error); |
100 } | 99 } |
101 | 100 |
| 101 TEST_P(VP9WorkerThreadTest, EndWithoutSync) { |
| 102 // Create a large number of threads to increase the chances of detecting a |
| 103 // race. Doing more work in the hook is no guarantee as any race would occur |
| 104 // post hook execution in the main thread loop driver. |
| 105 static const int kNumWorkers = 64; |
| 106 VP9Worker workers[kNumWorkers]; |
| 107 int hook_data[kNumWorkers]; |
| 108 int return_value[kNumWorkers]; |
| 109 |
| 110 for (int n = 0; n < kNumWorkers; ++n) { |
| 111 vp9_get_worker_interface()->init(&workers[n]); |
| 112 return_value[n] = 1; // return successfully from the hook |
| 113 workers[n].hook = ThreadHook; |
| 114 workers[n].data1 = &hook_data[n]; |
| 115 workers[n].data2 = &return_value[n]; |
| 116 } |
| 117 |
| 118 for (int i = 0; i < 2; ++i) { |
| 119 for (int n = 0; n < kNumWorkers; ++n) { |
| 120 EXPECT_NE(vp9_get_worker_interface()->reset(&workers[n]), 0); |
| 121 hook_data[n] = 0; |
| 122 } |
| 123 |
| 124 for (int n = 0; n < kNumWorkers; ++n) { |
| 125 Run(&workers[n]); |
| 126 } |
| 127 |
| 128 for (int n = kNumWorkers - 1; n >= 0; --n) { |
| 129 vp9_get_worker_interface()->end(&workers[n]); |
| 130 } |
| 131 } |
| 132 } |
| 133 |
102 TEST(VP9WorkerThreadTest, TestInterfaceAPI) { | 134 TEST(VP9WorkerThreadTest, TestInterfaceAPI) { |
103 EXPECT_EQ(0, vp9_set_worker_interface(NULL)); | 135 EXPECT_EQ(0, vp9_set_worker_interface(NULL)); |
104 EXPECT_TRUE(vp9_get_worker_interface() != NULL); | 136 EXPECT_TRUE(vp9_get_worker_interface() != NULL); |
105 for (int i = 0; i < 6; ++i) { | 137 for (int i = 0; i < 6; ++i) { |
106 VP9WorkerInterface winterface = *vp9_get_worker_interface(); | 138 VP9WorkerInterface winterface = *vp9_get_worker_interface(); |
107 switch (i) { | 139 switch (i) { |
108 default: | 140 default: |
109 case 0: winterface.init = NULL; break; | 141 case 0: winterface.init = NULL; break; |
110 case 1: winterface.reset = NULL; break; | 142 case 1: winterface.reset = NULL; break; |
111 case 2: winterface.sync = NULL; break; | 143 case 2: winterface.sync = NULL; break; |
(...skipping 161 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
273 { NULL, NULL } | 305 { NULL, NULL } |
274 }; | 306 }; |
275 | 307 |
276 DecodeFiles(files); | 308 DecodeFiles(files); |
277 } | 309 } |
278 #endif // CONFIG_WEBM_IO | 310 #endif // CONFIG_WEBM_IO |
279 | 311 |
280 INSTANTIATE_TEST_CASE_P(Synchronous, VP9WorkerThreadTest, ::testing::Bool()); | 312 INSTANTIATE_TEST_CASE_P(Synchronous, VP9WorkerThreadTest, ::testing::Bool()); |
281 | 313 |
282 } // namespace | 314 } // namespace |
OLD | NEW |