Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(413)

Unified Diff: webrtc/test/fake_audio_device.cc

Issue 2652803002: Refactor FakeAudioDevice to have separate methods for starting recording and playout. (Closed)
Patch Set: Removed drifting clock. Created 3 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« webrtc/test/call_test.cc ('K') | « webrtc/test/fake_audio_device.h ('k') | no next file » | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: webrtc/test/fake_audio_device.cc
diff --git a/webrtc/test/fake_audio_device.cc b/webrtc/test/fake_audio_device.cc
index 9e5e95fb0dbdc229b61dceb7dd182b404c4edc91..4b7e0cf9223fb9e385a212b57461bb35a7f0786b 100644
--- a/webrtc/test/fake_audio_device.cc
+++ b/webrtc/test/fake_audio_device.cc
@@ -12,49 +12,99 @@
#include <algorithm>
-#include "webrtc/base/platform_thread.h"
-#include "webrtc/modules/media_file/media_file_utility.h"
-#include "webrtc/system_wrappers/include/clock.h"
+#include "webrtc/base/array_view.h"
+#include "webrtc/base/checks.h"
+#include "webrtc/base/random.h"
#include "webrtc/system_wrappers/include/event_wrapper.h"
-#include "webrtc/system_wrappers/include/file_wrapper.h"
-#include "webrtc/test/gtest.h"
namespace webrtc {
+
+namespace {
+
+constexpr int kFrameLengthMs = 10;
+constexpr int kFramesPerSecond = 1000 / kFrameLengthMs;
+
+} // namespace
namespace test {
-FakeAudioDevice::FakeAudioDevice(Clock* clock,
- const std::string& filename,
- float speed)
- : audio_callback_(NULL),
- capturing_(false),
- captured_audio_(),
- playout_buffer_(),
+// Assuming 10ms audio packets..
+class FakeAudioDevice::PulsedNoiseCapturer {
+ public:
+ PulsedNoiseCapturer(int num_samples_per_frame, int16_t max_amplitude)
+ : fill_with_zero_(false),
+ random_generator_(1),
+ max_amplitude_(max_amplitude),
+ random_audio_(num_samples_per_frame),
+ silent_audio_(num_samples_per_frame, 0) {
+ RTC_DCHECK_GT(max_amplitude, 0);
+ }
+
+ rtc::ArrayView<const int16_t> Capture() {
+ fill_with_zero_ = !fill_with_zero_;
+ if (!fill_with_zero_) {
+ std::generate(random_audio_.begin(), random_audio_.end(), [&]() {
+ return random_generator_.Rand(-max_amplitude_, max_amplitude_);
+ });
+ }
+ return fill_with_zero_ ? silent_audio_ : random_audio_;
+ }
+
+ private:
+ bool fill_with_zero_;
+ webrtc::Random random_generator_;
peah-webrtc 2017/01/30 06:45:08 Since you are in the namespace webrtc, I don't thi
perkj_webrtc 2017/01/30 12:02:12 Done.
+ const int16_t max_amplitude_;
+ std::vector<int16_t> random_audio_;
+ std::vector<int16_t> silent_audio_;
+};
+
+FakeAudioDevice::FakeAudioDevice(float speed, int sampling_frequency_in_hz)
+ : sampling_frequency_in_hz_(sampling_frequency_in_hz),
+ num_samples_per_frame_(
+ rtc::CheckedDivExact(sampling_frequency_in_hz_, kFramesPerSecond)),
+ audio_callback_(nullptr),
+ rendering_(false),
+ playout_buffer_(num_samples_per_frame_, 0),
speed_(speed),
- last_playout_ms_(-1),
- clock_(clock, speed),
tick_(EventTimerWrapper::Create()),
- thread_(FakeAudioDevice::Run, this, "FakeAudioDevice"),
- file_utility_(new ModuleFileUtility(0)),
- input_stream_(FileWrapper::Create()) {
- memset(captured_audio_, 0, sizeof(captured_audio_));
- memset(playout_buffer_, 0, sizeof(playout_buffer_));
- // Open audio input file as read-only and looping.
- EXPECT_TRUE(input_stream_->OpenFile(filename.c_str(), true)) << filename;
+ thread_(FakeAudioDevice::Run, this, "FakeAudioDevice") {
+ RTC_DCHECK(
+ sampling_frequency_in_hz == 8000 || sampling_frequency_in_hz == 16000 ||
+ sampling_frequency_in_hz == 32000 || sampling_frequency_in_hz == 44100 ||
+ sampling_frequency_in_hz == 48000);
}
FakeAudioDevice::~FakeAudioDevice() {
- Stop();
-
+ StopPlayout();
+ StopRecording();
thread_.Stop();
}
-int32_t FakeAudioDevice::Init() {
+int32_t FakeAudioDevice::StartPlayout() {
rtc::CritScope cs(&lock_);
- if (file_utility_->InitPCMReading(*input_stream_.get()) != 0)
- return -1;
+ rendering_ = true;
+ return 0;
+}
- if (!tick_->StartTimer(true, 10 / speed_))
- return -1;
+int32_t FakeAudioDevice::StopPlayout() {
+ rtc::CritScope cs(&lock_);
+ rendering_ = false;
+ return 0;
+}
+
+void FakeAudioDevice::StartRecordingPulsedNoise(int16_t max_amplitude) {
+ rtc::CritScope cs(&lock_);
+ capturer_.reset(new FakeAudioDevice::PulsedNoiseCapturer(
+ num_samples_per_frame_, max_amplitude));
+}
+
+int32_t FakeAudioDevice::StopRecording() {
+ rtc::CritScope cs(&lock_);
+ capturer_.reset();
+ return 0;
+}
+
+int32_t FakeAudioDevice::Init() {
+ RTC_CHECK(tick_->StartTimer(true, kFrameLengthMs / speed_));
thread_.Start();
thread_.SetPriority(rtc::kHighPriority);
return 0;
@@ -62,85 +112,57 @@ int32_t FakeAudioDevice::Init() {
int32_t FakeAudioDevice::RegisterAudioCallback(AudioTransport* callback) {
peah-webrtc 2017/01/30 06:45:08 Since this is not used anywhere in the class and i
perkj_webrtc 2017/01/30 12:02:12 ? audio_callback_ is used.
peah-webrtc 2017/01/30 17:06:39 I cannot see that but I'm probably missing somethi
perkj_webrtc 2017/01/31 06:50:35 audio_callback_ is used both for recording an play
peah-webrtc 2017/01/31 16:37:32 True, of course, my mistake. Thanks for the explan
rtc::CritScope cs(&lock_);
+ RTC_DCHECK(callback || audio_callback_ != nullptr);
peah-webrtc 2017/01/30 06:45:08 This DCHECK I don't follow. It basically checks th
perkj_webrtc 2017/01/30 12:02:12 No, it allows callback == nullptr if audio_callbac
peah-webrtc 2017/01/30 17:06:39 In my mind that is a too strict restriction as tha
perkj_webrtc 2017/01/31 06:50:35 that is not harmfull but is probably not what the
peah-webrtc 2017/01/31 16:37:32 Acknowledged.
audio_callback_ = callback;
return 0;
}
bool FakeAudioDevice::Playing() const {
rtc::CritScope cs(&lock_);
- return capturing_;
+ return rendering_;
}
int32_t FakeAudioDevice::PlayoutDelay(uint16_t* delay_ms) const {
peah-webrtc 2017/01/30 06:45:08 Since PlayoutDelay, and it is not used inside this
perkj_webrtc 2017/01/30 12:02:12 We could move the implementation to the base class
+ RTC_DCHECK(delay_ms);
*delay_ms = 0;
return 0;
}
bool FakeAudioDevice::Recording() const {
rtc::CritScope cs(&lock_);
- return capturing_;
+ return !!capturer_;
}
bool FakeAudioDevice::Run(void* obj) {
- static_cast<FakeAudioDevice*>(obj)->CaptureAudio();
+ static_cast<FakeAudioDevice*>(obj)->ProcessAudio();
return true;
}
-void FakeAudioDevice::CaptureAudio() {
+void FakeAudioDevice::ProcessAudio() {
{
rtc::CritScope cs(&lock_);
- if (capturing_) {
- int bytes_read = file_utility_->ReadPCMData(
- *input_stream_.get(), captured_audio_, kBufferSizeBytes);
- if (bytes_read <= 0)
- return;
- // 2 bytes per sample.
- size_t num_samples = static_cast<size_t>(bytes_read / 2);
- uint32_t new_mic_level;
- EXPECT_EQ(0,
- audio_callback_->RecordedDataIsAvailable(captured_audio_,
- num_samples,
- 2,
- 1,
- kFrequencyHz,
- 0,
- 0,
- 0,
- false,
- new_mic_level));
- size_t samples_needed = kFrequencyHz / 100;
- int64_t now_ms = clock_.TimeInMilliseconds();
- uint32_t time_since_last_playout_ms = now_ms - last_playout_ms_;
- if (last_playout_ms_ > 0 && time_since_last_playout_ms > 0) {
- samples_needed = std::min(
- static_cast<size_t>(kFrequencyHz / time_since_last_playout_ms),
- kBufferSizeBytes / 2);
- }
+ if (capturer_) {
+ // Capture 10ms of audio. 2 bytes per sample.
+ rtc::ArrayView<const int16_t> audio_data = capturer_->Capture();
+ uint32_t new_mic_level = 0;
+ RTC_CHECK_EQ(
peah-webrtc 2017/01/30 06:45:09 Do we really want to crash this code if the return
perkj_webrtc 2017/01/30 12:02:12 yes I think so. We can not use gtest.
+ 0, audio_callback_->RecordedDataIsAvailable(
+ audio_data.data(), audio_data.size(), 2, 1,
+ sampling_frequency_in_hz_, 0, 0, 0, false, new_mic_level));
+ }
+ if (rendering_) {
size_t samples_out = 0;
int64_t elapsed_time_ms = -1;
int64_t ntp_time_ms = -1;
- EXPECT_EQ(0,
- audio_callback_->NeedMorePlayData(samples_needed,
- 2,
- 1,
- kFrequencyHz,
- playout_buffer_,
- samples_out,
- &elapsed_time_ms,
- &ntp_time_ms));
+ RTC_CHECK_EQ(0, audio_callback_->NeedMorePlayData(
peah-webrtc 2017/01/30 06:45:08 Do we really want to crash this code if the return
perkj_webrtc 2017/01/30 12:02:12 dito
+ num_samples_per_frame_, 2, 1,
+ sampling_frequency_in_hz_, playout_buffer_.data(),
+ samples_out, &elapsed_time_ms, &ntp_time_ms));
}
}
tick_->Wait(WEBRTC_EVENT_INFINITE);
}
-void FakeAudioDevice::Start() {
- rtc::CritScope cs(&lock_);
- capturing_ = true;
-}
-void FakeAudioDevice::Stop() {
- rtc::CritScope cs(&lock_);
- capturing_ = false;
-}
} // namespace test
} // namespace webrtc
« webrtc/test/call_test.cc ('K') | « webrtc/test/fake_audio_device.h ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698