Index: webrtc/modules/audio_processing/intelligibility/test/intelligibility_proc.cc |
diff --git a/webrtc/modules/audio_processing/intelligibility/test/intelligibility_proc.cc b/webrtc/modules/audio_processing/intelligibility/test/intelligibility_proc.cc |
index abd10d851656d985a7f6c29984f748503639bd88..c9281245b7372b47cd44a4dbc10f4c56ab27f4dd 100644 |
--- a/webrtc/modules/audio_processing/intelligibility/test/intelligibility_proc.cc |
+++ b/webrtc/modules/audio_processing/intelligibility/test/intelligibility_proc.cc |
@@ -39,7 +39,7 @@ void void_main(int argc, char* argv[]) { |
in_file.num_channels()); |
rtc::CriticalSection crit; |
NoiseSuppressionImpl ns(&crit); |
- IntelligibilityEnhancer enh(in_file.sample_rate(), in_file.num_channels(), |
+ IntelligibilityEnhancer enh(in_file.sample_rate(), in_file.num_channels(), 1u, |
NoiseSuppressionImpl::num_noise_bins()); |
ns.Initialize(noise_file.num_channels(), noise_file.sample_rate()); |
ns.Enable(true); |
@@ -52,23 +52,29 @@ void void_main(int argc, char* argv[]) { |
AudioBuffer capture_audio(noise_samples, noise_file.num_channels(), |
noise_samples, noise_file.num_channels(), |
noise_samples); |
- StreamConfig stream_config(noise_file.sample_rate(), |
- noise_file.num_channels()); |
+ AudioBuffer render_audio(in_samples, in_file.num_channels(), in_samples, |
+ in_file.num_channels(), in_samples); |
+ StreamConfig noise_config(noise_file.sample_rate(), |
+ noise_file.num_channels()); |
+ StreamConfig in_config(in_file.sample_rate(), in_file.num_channels()); |
while (in_file.ReadSamples(in.size(), in.data()) == in.size() && |
noise_file.ReadSamples(noise.size(), noise.data()) == noise.size()) { |
FloatS16ToFloat(noise.data(), noise.size(), noise.data()); |
+ FloatS16ToFloat(in.data(), in.size(), in.data()); |
Deinterleave(in.data(), in_buf.num_frames(), in_buf.num_channels(), |
in_buf.channels()); |
Deinterleave(noise.data(), noise_buf.num_frames(), noise_buf.num_channels(), |
noise_buf.channels()); |
- capture_audio.CopyFrom(noise_buf.channels(), stream_config); |
+ capture_audio.CopyFrom(noise_buf.channels(), noise_config); |
+ render_audio.CopyFrom(in_buf.channels(), in_config); |
ns.AnalyzeCaptureAudio(&capture_audio); |
ns.ProcessCaptureAudio(&capture_audio); |
- enh.SetCaptureNoiseEstimate(ns.NoiseEstimate(), 0); |
- enh.ProcessRenderAudio(in_buf.channels(), in_file.sample_rate(), |
- in_file.num_channels()); |
+ enh.SetCaptureNoiseEstimate(ns.NoiseEstimate(), 1); |
+ enh.ProcessRenderAudio(&render_audio); |
+ render_audio.CopyTo(in_config, in_buf.channels()); |
Interleave(in_buf.channels(), in_buf.num_frames(), in_buf.num_channels(), |
in.data()); |
+ FloatToFloatS16(in.data(), in.size(), in.data()); |
out_file.WriteSamples(in.data(), in.size()); |
} |
} |