| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2012 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/vad/standalone_vad.h" | 11 #include "webrtc/modules/audio_processing/vad/standalone_vad.h" |
| 12 | 12 |
| 13 #include <string.h> | 13 #include <string.h> |
| 14 | 14 |
| 15 #include "testing/gtest/include/gtest/gtest.h" | 15 #include "testing/gtest/include/gtest/gtest.h" |
| 16 #include "webrtc/base/scoped_ptr.h" | 16 #include "webrtc/base/scoped_ptr.h" |
| 17 #include "webrtc/modules/interface/module_common_types.h" | 17 #include "webrtc/modules/interface/module_common_types.h" |
| 18 #include "webrtc/test/testsupport/fileutils.h" | 18 #include "webrtc/test/testsupport/fileutils.h" |
| 19 #include "webrtc/test/testsupport/gtest_disable.h" | 19 #include "webrtc/test/testsupport/gtest_disable.h" |
| 20 | 20 |
| 21 namespace webrtc { | 21 namespace webrtc { |
| 22 | 22 |
| 23 TEST(StandaloneVadTest, Api) { | 23 TEST(StandaloneVadTest, Api) { |
| 24 rtc::scoped_ptr<StandaloneVad> vad(StandaloneVad::Create()); | 24 rtc::scoped_ptr<StandaloneVad> vad(StandaloneVad::Create()); |
| 25 int16_t data[kLength10Ms] = {0}; | 25 int16_t data[kLength10Ms] = {0}; |
| 26 | 26 |
| 27 // Valid frame length (for 32 kHz rate), but not what the VAD is expecting. | 27 // Valid frame length (for 32 kHz rate), but not what the VAD is expecting. |
| 28 EXPECT_EQ(-1, vad->AddAudio(data, 320)); | 28 EXPECT_EQ(-1, vad->AddAudio(data, 320)); |
| 29 | 29 |
| 30 const int kMaxNumFrames = 3; | 30 const size_t kMaxNumFrames = 3; |
| 31 double p[kMaxNumFrames]; | 31 double p[kMaxNumFrames]; |
| 32 for (int n = 0; n < kMaxNumFrames; n++) | 32 for (size_t n = 0; n < kMaxNumFrames; n++) |
| 33 EXPECT_EQ(0, vad->AddAudio(data, kLength10Ms)); | 33 EXPECT_EQ(0, vad->AddAudio(data, kLength10Ms)); |
| 34 | 34 |
| 35 // Pretend |p| is shorter that it should be. | 35 // Pretend |p| is shorter that it should be. |
| 36 EXPECT_EQ(-1, vad->GetActivity(p, kMaxNumFrames - 1)); | 36 EXPECT_EQ(-1, vad->GetActivity(p, kMaxNumFrames - 1)); |
| 37 | 37 |
| 38 EXPECT_EQ(0, vad->GetActivity(p, kMaxNumFrames)); | 38 EXPECT_EQ(0, vad->GetActivity(p, kMaxNumFrames)); |
| 39 | 39 |
| 40 // Ask for activity when buffer is empty. | 40 // Ask for activity when buffer is empty. |
| 41 EXPECT_EQ(-1, vad->GetActivity(p, kMaxNumFrames)); | 41 EXPECT_EQ(-1, vad->GetActivity(p, kMaxNumFrames)); |
| 42 | 42 |
| 43 // Should reset and result in one buffer. | 43 // Should reset and result in one buffer. |
| 44 for (int n = 0; n < kMaxNumFrames + 1; n++) | 44 for (size_t n = 0; n < kMaxNumFrames + 1; n++) |
| 45 EXPECT_EQ(0, vad->AddAudio(data, kLength10Ms)); | 45 EXPECT_EQ(0, vad->AddAudio(data, kLength10Ms)); |
| 46 EXPECT_EQ(0, vad->GetActivity(p, 1)); | 46 EXPECT_EQ(0, vad->GetActivity(p, 1)); |
| 47 | 47 |
| 48 // Wrong modes | 48 // Wrong modes |
| 49 EXPECT_EQ(-1, vad->set_mode(-1)); | 49 EXPECT_EQ(-1, vad->set_mode(-1)); |
| 50 EXPECT_EQ(-1, vad->set_mode(4)); | 50 EXPECT_EQ(-1, vad->set_mode(4)); |
| 51 | 51 |
| 52 // Valid mode. | 52 // Valid mode. |
| 53 const int kMode = 2; | 53 const int kMode = 2; |
| 54 EXPECT_EQ(0, vad->set_mode(kMode)); | 54 EXPECT_EQ(0, vad->set_mode(kMode)); |
| (...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 95 // When inactive, probabilities are set to 0.01. | 95 // When inactive, probabilities are set to 0.01. |
| 96 for (int n = 0; n < kNumVadFramesToProcess; n++) | 96 for (int n = 0; n < kNumVadFramesToProcess; n++) |
| 97 EXPECT_EQ(0.01, p[n]); | 97 EXPECT_EQ(0.01, p[n]); |
| 98 } | 98 } |
| 99 } | 99 } |
| 100 } | 100 } |
| 101 fclose(reference_file); | 101 fclose(reference_file); |
| 102 fclose(pcm_file); | 102 fclose(pcm_file); |
| 103 } | 103 } |
| 104 } // namespace webrtc | 104 } // namespace webrtc |
| OLD | NEW |