1 /*
2  *  Copyright (c) 2014 The WebRTC project authors. All Rights Reserved.
3  *
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
6  *  tree. An additional intellectual property rights grant can be found
7  *  in the file PATENTS.  All contributing project authors may
8  *  be found in the AUTHORS file in the root of the source tree.
9  */
10 
11 #ifndef MODULES_AUDIO_CODING_NETEQ_TOOLS_OUTPUT_WAV_FILE_H_
12 #define MODULES_AUDIO_CODING_NETEQ_TOOLS_OUTPUT_WAV_FILE_H_
13 
14 #include <string>
15 
16 #include "common_audio/wav_file.h"
17 #include "modules/audio_coding/neteq/tools/audio_sink.h"
18 #include "rtc_base/constructor_magic.h"
19 
20 namespace webrtc {
21 namespace test {
22 
23 class OutputWavFile : public AudioSink {
24  public:
25   // Creates an OutputWavFile, opening a file named |file_name| for writing.
26   // The output file is a PCM encoded wav file.
27   OutputWavFile(const std::string& file_name,
28                 int sample_rate_hz,
29                 int num_channels = 1)
wav_writer_(file_name,sample_rate_hz,num_channels)30       : wav_writer_(file_name, sample_rate_hz, num_channels) {}
31 
WriteArray(const int16_t * audio,size_t num_samples)32   bool WriteArray(const int16_t* audio, size_t num_samples) override {
33     wav_writer_.WriteSamples(audio, num_samples);
34     return true;
35   }
36 
37  private:
38   WavWriter wav_writer_;
39 
40   RTC_DISALLOW_COPY_AND_ASSIGN(OutputWavFile);
41 };
42 
43 }  // namespace test
44 }  // namespace webrtc
45 #endif  // MODULES_AUDIO_CODING_NETEQ_TOOLS_OUTPUT_WAV_FILE_H_
46