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 |
(...skipping 10 matching lines...) Expand all Loading... |
21 | 21 |
22 namespace webrtc { | 22 namespace webrtc { |
23 | 23 |
24 // Interface to provide access to WAV file parameters. | 24 // Interface to provide access to WAV file parameters. |
25 class WavFile { | 25 class WavFile { |
26 public: | 26 public: |
27 virtual ~WavFile() {} | 27 virtual ~WavFile() {} |
28 | 28 |
29 virtual int sample_rate() const = 0; | 29 virtual int sample_rate() const = 0; |
30 virtual int num_channels() const = 0; | 30 virtual int num_channels() const = 0; |
31 virtual uint32_t num_samples() const = 0; | 31 virtual size_t num_samples() const = 0; |
32 | 32 |
33 // Returns a human-readable string containing the audio format. | 33 // Returns a human-readable string containing the audio format. |
34 std::string FormatAsString() const; | 34 std::string FormatAsString() const; |
35 }; | 35 }; |
36 | 36 |
37 // Simple C++ class for writing 16-bit PCM WAV files. All error handling is | 37 // Simple C++ class for writing 16-bit PCM WAV files. All error handling is |
38 // by calls to RTC_CHECK(), making it unsuitable for anything but debug code. | 38 // by calls to RTC_CHECK(), making it unsuitable for anything but debug code. |
39 class WavWriter final : public WavFile { | 39 class WavWriter final : public WavFile { |
40 public: | 40 public: |
41 // Open a new WAV file for writing. | 41 // Open a new WAV file for writing. |
42 WavWriter(const std::string& filename, int sample_rate, int num_channels); | 42 WavWriter(const std::string& filename, int sample_rate, int num_channels); |
43 | 43 |
44 // Close the WAV file, after writing its header. | 44 // Close the WAV file, after writing its header. |
45 ~WavWriter(); | 45 ~WavWriter(); |
46 | 46 |
47 // Write additional samples to the file. Each sample is in the range | 47 // Write additional samples to the file. Each sample is in the range |
48 // [-32768,32767], and there must be the previously specified number of | 48 // [-32768,32767], and there must be the previously specified number of |
49 // interleaved channels. | 49 // interleaved channels. |
50 void WriteSamples(const float* samples, size_t num_samples); | 50 void WriteSamples(const float* samples, size_t num_samples); |
51 void WriteSamples(const int16_t* samples, size_t num_samples); | 51 void WriteSamples(const int16_t* samples, size_t num_samples); |
52 | 52 |
53 int sample_rate() const override { return sample_rate_; } | 53 int sample_rate() const override { return sample_rate_; } |
54 int num_channels() const override { return num_channels_; } | 54 int num_channels() const override { return num_channels_; } |
55 uint32_t num_samples() const override { return num_samples_; } | 55 size_t num_samples() const override { return num_samples_; } |
56 | 56 |
57 private: | 57 private: |
58 void Close(); | 58 void Close(); |
59 const int sample_rate_; | 59 const int sample_rate_; |
60 const int num_channels_; | 60 const int num_channels_; |
61 uint32_t num_samples_; // Total number of samples written to file. | 61 size_t num_samples_; // Total number of samples written to file. |
62 FILE* file_handle_; // Output file, owned by this class | 62 FILE* file_handle_; // Output file, owned by this class |
63 | 63 |
64 RTC_DISALLOW_COPY_AND_ASSIGN(WavWriter); | 64 RTC_DISALLOW_COPY_AND_ASSIGN(WavWriter); |
65 }; | 65 }; |
66 | 66 |
67 // Follows the conventions of WavWriter. | 67 // Follows the conventions of WavWriter. |
68 class WavReader final : public WavFile { | 68 class WavReader final : public WavFile { |
69 public: | 69 public: |
70 // Opens an existing WAV file for reading. | 70 // Opens an existing WAV file for reading. |
71 explicit WavReader(const std::string& filename); | 71 explicit WavReader(const std::string& filename); |
72 | 72 |
73 // Close the WAV file. | 73 // Close the WAV file. |
74 ~WavReader(); | 74 ~WavReader(); |
75 | 75 |
76 // Returns the number of samples read. If this is less than requested, | 76 // Returns the number of samples read. If this is less than requested, |
77 // verifies that the end of the file was reached. | 77 // verifies that the end of the file was reached. |
78 size_t ReadSamples(size_t num_samples, float* samples); | 78 size_t ReadSamples(size_t num_samples, float* samples); |
79 size_t ReadSamples(size_t num_samples, int16_t* samples); | 79 size_t ReadSamples(size_t num_samples, int16_t* samples); |
80 | 80 |
81 int sample_rate() const override { return sample_rate_; } | 81 int sample_rate() const override { return sample_rate_; } |
82 int num_channels() const override { return num_channels_; } | 82 int num_channels() const override { return num_channels_; } |
83 uint32_t num_samples() const override { return num_samples_; } | 83 size_t num_samples() const override { return num_samples_; } |
84 | 84 |
85 private: | 85 private: |
86 void Close(); | 86 void Close(); |
87 int sample_rate_; | 87 int sample_rate_; |
88 int num_channels_; | 88 int num_channels_; |
89 uint32_t num_samples_; // Total number of samples in the file. | 89 size_t num_samples_; // Total number of samples in the file. |
90 uint32_t num_samples_remaining_; | 90 size_t num_samples_remaining_; |
91 FILE* file_handle_; // Input file, owned by this class. | 91 FILE* file_handle_; // Input file, owned by this class. |
92 | 92 |
93 RTC_DISALLOW_COPY_AND_ASSIGN(WavReader); | 93 RTC_DISALLOW_COPY_AND_ASSIGN(WavReader); |
94 }; | 94 }; |
95 | 95 |
96 } // namespace webrtc | 96 } // namespace webrtc |
97 | 97 |
98 extern "C" { | 98 extern "C" { |
99 #endif // __cplusplus | 99 #endif // __cplusplus |
100 | 100 |
101 // C wrappers for the WavWriter class. | 101 // C wrappers for the WavWriter class. |
102 typedef struct rtc_WavWriter rtc_WavWriter; | 102 typedef struct rtc_WavWriter rtc_WavWriter; |
103 rtc_WavWriter* rtc_WavOpen(const char* filename, | 103 rtc_WavWriter* rtc_WavOpen(const char* filename, |
104 int sample_rate, | 104 int sample_rate, |
105 int num_channels); | 105 int num_channels); |
106 void rtc_WavClose(rtc_WavWriter* wf); | 106 void rtc_WavClose(rtc_WavWriter* wf); |
107 void rtc_WavWriteSamples(rtc_WavWriter* wf, | 107 void rtc_WavWriteSamples(rtc_WavWriter* wf, |
108 const float* samples, | 108 const float* samples, |
109 size_t num_samples); | 109 size_t num_samples); |
110 int rtc_WavSampleRate(const rtc_WavWriter* wf); | 110 int rtc_WavSampleRate(const rtc_WavWriter* wf); |
111 int rtc_WavNumChannels(const rtc_WavWriter* wf); | 111 int rtc_WavNumChannels(const rtc_WavWriter* wf); |
112 uint32_t rtc_WavNumSamples(const rtc_WavWriter* wf); | 112 size_t rtc_WavNumSamples(const rtc_WavWriter* wf); |
113 | 113 |
114 #ifdef __cplusplus | 114 #ifdef __cplusplus |
115 } // extern "C" | 115 } // extern "C" |
116 #endif | 116 #endif |
117 | 117 |
118 #endif // WEBRTC_COMMON_AUDIO_WAV_FILE_H_ | 118 #endif // WEBRTC_COMMON_AUDIO_WAV_FILE_H_ |
OLD | NEW |