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

Side by Side Diff: webrtc/modules/audio_processing/test/audio_file_processor.cc

Issue 1800413002: Add IntelligibilityEnhancer support to audioproc_float (Closed) Base URL: https://chromium.googlesource.com/external/webrtc.git@reverse
Patch Set: Created 4 years, 9 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 unified diff | Download patch
OLDNEW
1 /* 1 /*
2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. 2 * Copyright (c) 2015 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
(...skipping 24 matching lines...) Expand all
35 ChannelBuffer<float> GetChannelBuffer(const WavFile& file) { 35 ChannelBuffer<float> GetChannelBuffer(const WavFile& file) {
36 return ChannelBuffer<float>( 36 return ChannelBuffer<float>(
37 CheckedDivExact(file.sample_rate(), AudioFileProcessor::kChunksPerSecond), 37 CheckedDivExact(file.sample_rate(), AudioFileProcessor::kChunksPerSecond),
38 file.num_channels()); 38 file.num_channels());
39 } 39 }
40 40
41 } // namespace 41 } // namespace
42 42
43 WavFileProcessor::WavFileProcessor(std::unique_ptr<AudioProcessing> ap, 43 WavFileProcessor::WavFileProcessor(std::unique_ptr<AudioProcessing> ap,
44 std::unique_ptr<WavReader> in_file, 44 std::unique_ptr<WavReader> in_file,
45 std::unique_ptr<WavWriter> out_file) 45 std::unique_ptr<WavWriter> out_file,
46 std::unique_ptr<WavReader> reverse_in_file,
47 std::unique_ptr<WavWriter> reverse_out_file)
46 : ap_(std::move(ap)), 48 : ap_(std::move(ap)),
47 in_buf_(GetChannelBuffer(*in_file)), 49 in_buf_(GetChannelBuffer(*in_file)),
48 out_buf_(GetChannelBuffer(*out_file)), 50 out_buf_(GetChannelBuffer(*out_file)),
49 input_config_(GetStreamConfig(*in_file)), 51 input_config_(GetStreamConfig(*in_file)),
50 output_config_(GetStreamConfig(*out_file)), 52 output_config_(GetStreamConfig(*out_file)),
51 buffer_reader_(std::move(in_file)), 53 buffer_reader_(std::move(in_file)),
52 buffer_writer_(std::move(out_file)) {} 54 buffer_writer_(std::move(out_file)),
55 process_reverse_stream_(reverse_in_file && reverse_out_file) {
peah-webrtc 2016/03/18 09:38:02 Why is this true only if both reverse_in_file reve
aluebs-webrtc 2016/03/18 18:36:54 In practice, the reverse_out_file is always specif
56 if (process_reverse_stream_) {
57 reverse_in_buf_.reset(
58 new ChannelBuffer<float>(GetChannelBuffer(*reverse_in_file)));
59 reverse_out_buf_.reset(
60 new ChannelBuffer<float>(GetChannelBuffer(*reverse_out_file)));
61 reverse_input_config_.reset(
62 new StreamConfig(GetStreamConfig(*reverse_in_file)));
63 reverse_output_config_.reset(
64 new StreamConfig(GetStreamConfig(*reverse_out_file)));
65 reverse_buffer_reader_.reset(
66 new ChannelBufferWavReader(std::move(reverse_in_file)));
67 reverse_buffer_writer_.reset(
68 new ChannelBufferWavWriter(std::move(reverse_out_file)));
69 }
70 }
53 71
54 bool WavFileProcessor::ProcessChunk() { 72 bool WavFileProcessor::ProcessChunk() {
55 if (!buffer_reader_.Read(&in_buf_)) { 73 if (!buffer_reader_.Read(&in_buf_)) {
56 return false; 74 return false;
57 } 75 }
58 { 76 {
59 const auto st = ScopedTimer(mutable_proc_time()); 77 const auto st = ScopedTimer(mutable_proc_time());
60 RTC_CHECK_EQ(kNoErr, 78 RTC_CHECK_EQ(kNoErr,
61 ap_->ProcessStream(in_buf_.channels(), input_config_, 79 ap_->ProcessStream(in_buf_.channels(), input_config_,
62 output_config_, out_buf_.channels())); 80 output_config_, out_buf_.channels()));
63 } 81 }
64 buffer_writer_.Write(out_buf_); 82 buffer_writer_.Write(out_buf_);
83 if (process_reverse_stream_) {
peah-webrtc 2016/03/18 09:38:02 This bool is a bit redundant: you should be able t
aluebs-webrtc 2016/03/18 18:36:53 I just though it was less bug prone to have one ce
84 if (!reverse_buffer_reader_->Read(reverse_in_buf_.get())) {
85 return false;
86 }
87 {
88 const auto st = ScopedTimer(mutable_proc_time());
89 RTC_CHECK_EQ(kNoErr,
90 ap_->ProcessReverseStream(reverse_in_buf_->channels(),
91 *reverse_input_config_.get(),
92 *reverse_output_config_.get(),
93 reverse_out_buf_->channels()));
94 }
95 reverse_buffer_writer_->Write(*reverse_out_buf_.get());
96 }
65 return true; 97 return true;
66 } 98 }
67 99
68 AecDumpFileProcessor::AecDumpFileProcessor(std::unique_ptr<AudioProcessing> ap, 100 AecDumpFileProcessor::AecDumpFileProcessor(std::unique_ptr<AudioProcessing> ap,
69 FILE* dump_file, 101 FILE* dump_file,
70 std::unique_ptr<WavWriter> out_file) 102 std::unique_ptr<WavWriter> out_file)
71 : ap_(std::move(ap)), 103 : ap_(std::move(ap)),
72 dump_file_(dump_file), 104 dump_file_(dump_file),
73 out_buf_(GetChannelBuffer(*out_file)), 105 out_buf_(GetChannelBuffer(*out_file)),
74 output_config_(GetStreamConfig(*out_file)), 106 output_config_(GetStreamConfig(*out_file)),
(...skipping 95 matching lines...) Expand 10 before | Expand all | Expand 10 after
170 const auto st = ScopedTimer(mutable_proc_time()); 202 const auto st = ScopedTimer(mutable_proc_time());
171 // TODO(ajm): This currently discards the processed output, which is needed 203 // TODO(ajm): This currently discards the processed output, which is needed
172 // for e.g. intelligibility enhancement. 204 // for e.g. intelligibility enhancement.
173 RTC_CHECK_EQ(kNoErr, ap_->ProcessReverseStream( 205 RTC_CHECK_EQ(kNoErr, ap_->ProcessReverseStream(
174 reverse_buf_->channels(), reverse_config_, 206 reverse_buf_->channels(), reverse_config_,
175 reverse_config_, reverse_buf_->channels())); 207 reverse_config_, reverse_buf_->channels()));
176 } 208 }
177 } 209 }
178 210
179 } // namespace webrtc 211 } // namespace webrtc
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698