| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2014 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_coding/neteq/tools/neteq_performance_test.h" | 11 #include "webrtc/modules/audio_coding/neteq/tools/neteq_performance_test.h" |
| 12 | 12 |
| 13 #include "webrtc/api/audio_codecs/builtin_audio_decoder_factory.h" | 13 #include "webrtc/api/audio_codecs/builtin_audio_decoder_factory.h" |
| 14 #include "webrtc/base/checks.h" | 14 #include "webrtc/base/checks.h" |
| 15 #include "webrtc/common_types.h" |
| 15 #include "webrtc/modules/audio_coding/codecs/pcm16b/pcm16b.h" | 16 #include "webrtc/modules/audio_coding/codecs/pcm16b/pcm16b.h" |
| 16 #include "webrtc/modules/audio_coding/neteq/include/neteq.h" | 17 #include "webrtc/modules/audio_coding/neteq/include/neteq.h" |
| 17 #include "webrtc/modules/audio_coding/neteq/tools/audio_loop.h" | 18 #include "webrtc/modules/audio_coding/neteq/tools/audio_loop.h" |
| 18 #include "webrtc/modules/audio_coding/neteq/tools/rtp_generator.h" | 19 #include "webrtc/modules/audio_coding/neteq/tools/rtp_generator.h" |
| 19 #include "webrtc/modules/include/module_common_types.h" | 20 #include "webrtc/modules/include/module_common_types.h" |
| 20 #include "webrtc/system_wrappers/include/clock.h" | 21 #include "webrtc/system_wrappers/include/clock.h" |
| 21 #include "webrtc/test/testsupport/fileutils.h" | 22 #include "webrtc/test/testsupport/fileutils.h" |
| 22 #include "webrtc/typedefs.h" | 23 #include "webrtc/typedefs.h" |
| 23 | 24 |
| 24 using webrtc::NetEq; | 25 using webrtc::NetEq; |
| 25 using webrtc::test::AudioLoop; | 26 using webrtc::test::AudioLoop; |
| 26 using webrtc::test::RtpGenerator; | 27 using webrtc::test::RtpGenerator; |
| 27 using webrtc::WebRtcRTPHeader; | |
| 28 | 28 |
| 29 namespace webrtc { | 29 namespace webrtc { |
| 30 namespace test { | 30 namespace test { |
| 31 | 31 |
| 32 int64_t NetEqPerformanceTest::Run(int runtime_ms, | 32 int64_t NetEqPerformanceTest::Run(int runtime_ms, |
| 33 int lossrate, | 33 int lossrate, |
| 34 double drift_factor) { | 34 double drift_factor) { |
| 35 const std::string kInputFileName = | 35 const std::string kInputFileName = |
| 36 webrtc::test::ResourcePath("audio_coding/testfile32kHz", "pcm"); | 36 webrtc::test::ResourcePath("audio_coding/testfile32kHz", "pcm"); |
| 37 const int kSampRateHz = 32000; | 37 const int kSampRateHz = 32000; |
| (...skipping 14 matching lines...) Expand all Loading... |
| 52 AudioLoop audio_loop; | 52 AudioLoop audio_loop; |
| 53 const size_t kMaxLoopLengthSamples = kSampRateHz * 10; // 10 second loop. | 53 const size_t kMaxLoopLengthSamples = kSampRateHz * 10; // 10 second loop. |
| 54 const size_t kInputBlockSizeSamples = 60 * kSampRateHz / 1000; // 60 ms. | 54 const size_t kInputBlockSizeSamples = 60 * kSampRateHz / 1000; // 60 ms. |
| 55 if (!audio_loop.Init(kInputFileName, kMaxLoopLengthSamples, | 55 if (!audio_loop.Init(kInputFileName, kMaxLoopLengthSamples, |
| 56 kInputBlockSizeSamples)) | 56 kInputBlockSizeSamples)) |
| 57 return -1; | 57 return -1; |
| 58 | 58 |
| 59 int32_t time_now_ms = 0; | 59 int32_t time_now_ms = 0; |
| 60 | 60 |
| 61 // Get first input packet. | 61 // Get first input packet. |
| 62 WebRtcRTPHeader rtp_header; | 62 RTPHeader rtp_header; |
| 63 RtpGenerator rtp_gen(kSampRateHz / 1000); | 63 RtpGenerator rtp_gen(kSampRateHz / 1000); |
| 64 // Start with positive drift first half of simulation. | 64 // Start with positive drift first half of simulation. |
| 65 rtp_gen.set_drift_factor(drift_factor); | 65 rtp_gen.set_drift_factor(drift_factor); |
| 66 bool drift_flipped = false; | 66 bool drift_flipped = false; |
| 67 int32_t packet_input_time_ms = | 67 int32_t packet_input_time_ms = |
| 68 rtp_gen.GetRtpHeader(kPayloadType, kInputBlockSizeSamples, &rtp_header); | 68 rtp_gen.GetRtpHeader(kPayloadType, kInputBlockSizeSamples, &rtp_header); |
| 69 auto input_samples = audio_loop.GetNextBlock(); | 69 auto input_samples = audio_loop.GetNextBlock(); |
| 70 if (input_samples.empty()) | 70 if (input_samples.empty()) |
| 71 exit(1); | 71 exit(1); |
| 72 uint8_t input_payload[kInputBlockSizeSamples * sizeof(int16_t)]; | 72 uint8_t input_payload[kInputBlockSizeSamples * sizeof(int16_t)]; |
| 73 size_t payload_len = WebRtcPcm16b_Encode(input_samples.data(), | 73 size_t payload_len = WebRtcPcm16b_Encode(input_samples.data(), |
| 74 input_samples.size(), input_payload); | 74 input_samples.size(), input_payload); |
| 75 RTC_CHECK_EQ(sizeof(input_payload), payload_len); | 75 RTC_CHECK_EQ(sizeof(input_payload), payload_len); |
| 76 | 76 |
| 77 // Main loop. | 77 // Main loop. |
| 78 webrtc::Clock* clock = webrtc::Clock::GetRealTimeClock(); | 78 webrtc::Clock* clock = webrtc::Clock::GetRealTimeClock(); |
| 79 int64_t start_time_ms = clock->TimeInMilliseconds(); | 79 int64_t start_time_ms = clock->TimeInMilliseconds(); |
| 80 AudioFrame out_frame; | 80 AudioFrame out_frame; |
| 81 while (time_now_ms < runtime_ms) { | 81 while (time_now_ms < runtime_ms) { |
| 82 while (packet_input_time_ms <= time_now_ms) { | 82 while (packet_input_time_ms <= time_now_ms) { |
| 83 // Drop every N packets, where N = FLAGS_lossrate. | 83 // Drop every N packets, where N = FLAGS_lossrate. |
| 84 bool lost = false; | 84 bool lost = false; |
| 85 if (lossrate > 0) { | 85 if (lossrate > 0) { |
| 86 lost = ((rtp_header.header.sequenceNumber - 1) % lossrate) == 0; | 86 lost = ((rtp_header.sequenceNumber - 1) % lossrate) == 0; |
| 87 } | 87 } |
| 88 if (!lost) { | 88 if (!lost) { |
| 89 // Insert packet. | 89 // Insert packet. |
| 90 int error = | 90 int error = |
| 91 neteq->InsertPacket(rtp_header.header, input_payload, | 91 neteq->InsertPacket(rtp_header, input_payload, |
| 92 packet_input_time_ms * kSampRateHz / 1000); | 92 packet_input_time_ms * kSampRateHz / 1000); |
| 93 if (error != NetEq::kOK) | 93 if (error != NetEq::kOK) |
| 94 return -1; | 94 return -1; |
| 95 } | 95 } |
| 96 | 96 |
| 97 // Get next packet. | 97 // Get next packet. |
| 98 packet_input_time_ms = rtp_gen.GetRtpHeader(kPayloadType, | 98 packet_input_time_ms = rtp_gen.GetRtpHeader(kPayloadType, |
| 99 kInputBlockSizeSamples, | 99 kInputBlockSizeSamples, |
| 100 &rtp_header); | 100 &rtp_header); |
| 101 input_samples = audio_loop.GetNextBlock(); | 101 input_samples = audio_loop.GetNextBlock(); |
| (...skipping 22 matching lines...) Expand all Loading... |
| 124 drift_flipped = true; | 124 drift_flipped = true; |
| 125 } | 125 } |
| 126 } | 126 } |
| 127 int64_t end_time_ms = clock->TimeInMilliseconds(); | 127 int64_t end_time_ms = clock->TimeInMilliseconds(); |
| 128 delete neteq; | 128 delete neteq; |
| 129 return end_time_ms - start_time_ms; | 129 return end_time_ms - start_time_ms; |
| 130 } | 130 } |
| 131 | 131 |
| 132 } // namespace test | 132 } // namespace test |
| 133 } // namespace webrtc | 133 } // namespace webrtc |
| OLD | NEW |