| 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/interface/file_wrapper.h" | 20 #include "webrtc/system_wrappers/interface/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" | 
| 22 | 23 | 
| 23 namespace webrtc { | 24 namespace webrtc { | 
| 24 | 25 | 
| 25 TEST(WPDTreeTest, Construction) { | 26 TEST(WPDTreeTest, Construction) { | 
| 26   const size_t kTestBufferSize = 100; | 27   const size_t kTestBufferSize = 100; | 
| 27   const int kLevels = 5; | 28   const int kLevels = 5; | 
| 28   const int kExpectedNumberOfNodes = (1 << (kLevels + 1)) - 1; | 29   const int kExpectedNumberOfNodes = (1 << (kLevels + 1)) - 1; | 
| 29 | 30 | 
| 30   float test_buffer[kTestBufferSize]; | 31   float test_buffer[kTestBufferSize]; | 
| 31   memset(test_buffer, 0.f, kTestBufferSize * sizeof(*test_buffer)); | 32   memset(test_buffer, 0.f, kTestBufferSize * sizeof(*test_buffer)); | 
| (...skipping 29 matching lines...) Expand all  Loading... | 
| 61   EXPECT_EQ(-1, tree.Update(NULL, kTestBufferSize)); | 62   EXPECT_EQ(-1, tree.Update(NULL, kTestBufferSize)); | 
| 62   EXPECT_EQ(-1, tree.Update(test_buffer, kTestBufferSize - 1)); | 63   EXPECT_EQ(-1, tree.Update(test_buffer, kTestBufferSize - 1)); | 
| 63 } | 64 } | 
| 64 | 65 | 
| 65 // This test is for the correctness of the tree. | 66 // This test is for the correctness of the tree. | 
| 66 // Checks the results from the Matlab equivalent, it is done comparing the | 67 // Checks the results from the Matlab equivalent, it is done comparing the | 
| 67 // results that are stored in the output files from Matlab. | 68 // results that are stored in the output files from Matlab. | 
| 68 // It also writes the results in its own set of files in the out directory. | 69 // It also writes the results in its own set of files in the out directory. | 
| 69 // Matlab and output files contain all the results in double precision (Little | 70 // Matlab and output files contain all the results in double precision (Little | 
| 70 // endian) appended. | 71 // endian) appended. | 
| 71 TEST(WPDTreeTest, CorrectnessBasedOnMatlabFiles) { | 72 TEST(WPDTreeTest, DISABLED_ON_IOS(CorrectnessBasedOnMatlabFiles)) { | 
| 72   // 10 ms at 16000 Hz. | 73   // 10 ms at 16000 Hz. | 
| 73   const size_t kTestBufferSize = 160; | 74   const size_t kTestBufferSize = 160; | 
| 74   const int kLevels = 3; | 75   const int kLevels = 3; | 
| 75   const int kLeaves = 1 << kLevels; | 76   const int kLeaves = 1 << kLevels; | 
| 76   const size_t kLeavesSamples = kTestBufferSize >> kLevels; | 77   const size_t kLeavesSamples = kTestBufferSize >> kLevels; | 
| 77   // Create tree with Discrete Meyer Wavelet Coefficients. | 78   // Create tree with Discrete Meyer Wavelet Coefficients. | 
| 78   WPDTree tree(kTestBufferSize, | 79   WPDTree tree(kTestBufferSize, | 
| 79                kDaubechies8HighPassCoefficients, | 80                kDaubechies8HighPassCoefficients, | 
| 80                kDaubechies8LowPassCoefficients, | 81                kDaubechies8LowPassCoefficients, | 
| 81                kDaubechies8CoefficientsLength, | 82                kDaubechies8CoefficientsLength, | 
| (...skipping 107 matching lines...) Expand 10 before | Expand all | Expand 10 after  Loading... | 
| 189   // Close all matlab and out files. | 190   // Close all matlab and out files. | 
| 190   for (int i = 0; i < kLeaves; ++i) { | 191   for (int i = 0; i < kLeaves; ++i) { | 
| 191     matlab_files_data[i]->CloseFile(); | 192     matlab_files_data[i]->CloseFile(); | 
| 192     out_files_data[i]->CloseFile(); | 193     out_files_data[i]->CloseFile(); | 
| 193   } | 194   } | 
| 194 | 195 | 
| 195   test_file->CloseFile(); | 196   test_file->CloseFile(); | 
| 196 } | 197 } | 
| 197 | 198 | 
| 198 }  // namespace webrtc | 199 }  // namespace webrtc | 
| OLD | NEW | 
|---|