| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2013 The WebRTC 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 "webrtc/modules/audio_processing/transient/wpd_tree.h" | 11 #include "webrtc/modules/audio_processing/transient/wpd_tree.h" |
| 12 | 12 |
| 13 #include <sstream> | 13 #include <sstream> |
| 14 #include <string> | 14 #include <string> |
| 15 | 15 |
| 16 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" |
| 17 #include "webrtc/base/scoped_ptr.h" | 17 #include "webrtc/base/scoped_ptr.h" |
| 18 #include "webrtc/modules/audio_processing/transient/daubechies_8_wavelet_coeffs.
h" | 18 #include "webrtc/modules/audio_processing/transient/daubechies_8_wavelet_coeffs.
h" |
| 19 #include "webrtc/modules/audio_processing/transient/file_utils.h" | 19 #include "webrtc/modules/audio_processing/transient/file_utils.h" |
| 20 #include "webrtc/system_wrappers/include/file_wrapper.h" | 20 #include "webrtc/system_wrappers/include/file_wrapper.h" |
| 21 #include "webrtc/test/testsupport/fileutils.h" | 21 #include "webrtc/test/testsupport/fileutils.h" |
| 22 #include "webrtc/test/testsupport/gtest_disable.h" | |
| 23 | 22 |
| 24 namespace webrtc { | 23 namespace webrtc { |
| 25 | 24 |
| 26 TEST(WPDTreeTest, Construction) { | 25 TEST(WPDTreeTest, Construction) { |
| 27 const size_t kTestBufferSize = 100; | 26 const size_t kTestBufferSize = 100; |
| 28 const int kLevels = 5; | 27 const int kLevels = 5; |
| 29 const int kExpectedNumberOfNodes = (1 << (kLevels + 1)) - 1; | 28 const int kExpectedNumberOfNodes = (1 << (kLevels + 1)) - 1; |
| 30 | 29 |
| 31 float test_buffer[kTestBufferSize]; | 30 float test_buffer[kTestBufferSize]; |
| 32 memset(test_buffer, 0.f, kTestBufferSize * sizeof(*test_buffer)); | 31 memset(test_buffer, 0.f, kTestBufferSize * sizeof(*test_buffer)); |
| (...skipping 29 matching lines...) Expand all Loading... |
| 62 EXPECT_EQ(-1, tree.Update(NULL, kTestBufferSize)); | 61 EXPECT_EQ(-1, tree.Update(NULL, kTestBufferSize)); |
| 63 EXPECT_EQ(-1, tree.Update(test_buffer, kTestBufferSize - 1)); | 62 EXPECT_EQ(-1, tree.Update(test_buffer, kTestBufferSize - 1)); |
| 64 } | 63 } |
| 65 | 64 |
| 66 // This test is for the correctness of the tree. | 65 // This test is for the correctness of the tree. |
| 67 // Checks the results from the Matlab equivalent, it is done comparing the | 66 // Checks the results from the Matlab equivalent, it is done comparing the |
| 68 // results that are stored in the output files from Matlab. | 67 // results that are stored in the output files from Matlab. |
| 69 // It also writes the results in its own set of files in the out directory. | 68 // It also writes the results in its own set of files in the out directory. |
| 70 // Matlab and output files contain all the results in double precision (Little | 69 // Matlab and output files contain all the results in double precision (Little |
| 71 // endian) appended. | 70 // endian) appended. |
| 72 TEST(WPDTreeTest, DISABLED_ON_IOS(CorrectnessBasedOnMatlabFiles)) { | 71 #if defined(WEBRTC_IOS) |
| 72 TEST(WPDTreeTest, DISABLED_CorrectnessBasedOnMatlabFiles) { |
| 73 #else |
| 74 TEST(WPDTreeTest, CorrectnessBasedOnMatlabFiles) { |
| 75 #endif |
| 73 // 10 ms at 16000 Hz. | 76 // 10 ms at 16000 Hz. |
| 74 const size_t kTestBufferSize = 160; | 77 const size_t kTestBufferSize = 160; |
| 75 const int kLevels = 3; | 78 const int kLevels = 3; |
| 76 const int kLeaves = 1 << kLevels; | 79 const int kLeaves = 1 << kLevels; |
| 77 const size_t kLeavesSamples = kTestBufferSize >> kLevels; | 80 const size_t kLeavesSamples = kTestBufferSize >> kLevels; |
| 78 // Create tree with Discrete Meyer Wavelet Coefficients. | 81 // Create tree with Discrete Meyer Wavelet Coefficients. |
| 79 WPDTree tree(kTestBufferSize, | 82 WPDTree tree(kTestBufferSize, |
| 80 kDaubechies8HighPassCoefficients, | 83 kDaubechies8HighPassCoefficients, |
| 81 kDaubechies8LowPassCoefficients, | 84 kDaubechies8LowPassCoefficients, |
| 82 kDaubechies8CoefficientsLength, | 85 kDaubechies8CoefficientsLength, |
| (...skipping 107 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 190 // Close all matlab and out files. | 193 // Close all matlab and out files. |
| 191 for (int i = 0; i < kLeaves; ++i) { | 194 for (int i = 0; i < kLeaves; ++i) { |
| 192 matlab_files_data[i]->CloseFile(); | 195 matlab_files_data[i]->CloseFile(); |
| 193 out_files_data[i]->CloseFile(); | 196 out_files_data[i]->CloseFile(); |
| 194 } | 197 } |
| 195 | 198 |
| 196 test_file->CloseFile(); | 199 test_file->CloseFile(); |
| 197 } | 200 } |
| 198 | 201 |
| 199 } // namespace webrtc | 202 } // namespace webrtc |
| OLD | NEW |