mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-12 21:30:45 +01:00
5dca3f1336
This CL adds support for reading and writing floating point wav files in WebRTC. It also updates the former wav handling code as well as adds some simplifications. Beyond this, the CL also adds support in the APM data_dumper and in the audioproc_f tool for using the floating point wav format. Bug: webrtc:11307 Change-Id: I2ea33fd12f590b6031ac85f75708f6cc88a266b4 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/162902 Reviewed-by: Sam Zackrisson <saza@webrtc.org> Reviewed-by: Karl Wiberg <kwiberg@webrtc.org> Commit-Queue: Per Åhgren <peah@webrtc.org> Cr-Commit-Position: refs/heads/master@{#30423}
115 lines
3.5 KiB
C++
115 lines
3.5 KiB
C++
/*
|
|
* Copyright (c) 2014 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#ifndef COMMON_AUDIO_WAV_FILE_H_
|
|
#define COMMON_AUDIO_WAV_FILE_H_
|
|
|
|
#include <stdint.h>
|
|
|
|
#include <cstddef>
|
|
#include <string>
|
|
|
|
#include "common_audio/wav_header.h"
|
|
#include "rtc_base/system/file_wrapper.h"
|
|
|
|
namespace webrtc {
|
|
|
|
// Interface to provide access WAV file parameters.
|
|
class WavFile {
|
|
public:
|
|
enum class SampleFormat { kInt16, kFloat };
|
|
|
|
virtual ~WavFile() {}
|
|
|
|
virtual int sample_rate() const = 0;
|
|
virtual size_t num_channels() const = 0;
|
|
virtual size_t num_samples() const = 0;
|
|
};
|
|
|
|
// Simple C++ class for writing 16-bit integer and 32 bit floating point PCM WAV
|
|
// files. All error handling is by calls to RTC_CHECK(), making it unsuitable
|
|
// for anything but debug code.
|
|
class WavWriter final : public WavFile {
|
|
public:
|
|
// Opens a new WAV file for writing.
|
|
WavWriter(const std::string& filename,
|
|
int sample_rate,
|
|
size_t num_channels,
|
|
SampleFormat sample_format = SampleFormat::kInt16);
|
|
WavWriter(FileWrapper file,
|
|
int sample_rate,
|
|
size_t num_channels,
|
|
SampleFormat sample_format = SampleFormat::kInt16);
|
|
|
|
// Closes the WAV file, after writing its header.
|
|
~WavWriter() { Close(); }
|
|
|
|
WavWriter(const WavWriter&) = delete;
|
|
WavWriter& operator=(const WavWriter&) = delete;
|
|
|
|
// Write additional samples to the file. Each sample is in the range
|
|
// [-32768.0,32767.0], and there must be the previously specified number of
|
|
// interleaved channels.
|
|
void WriteSamples(const float* samples, size_t num_samples);
|
|
void WriteSamples(const int16_t* samples, size_t num_samples);
|
|
|
|
int sample_rate() const override { return sample_rate_; }
|
|
size_t num_channels() const override { return num_channels_; }
|
|
size_t num_samples() const override { return num_samples_written_; }
|
|
|
|
private:
|
|
void Close();
|
|
const int sample_rate_;
|
|
const size_t num_channels_;
|
|
size_t num_samples_written_;
|
|
WavFormat format_;
|
|
FileWrapper file_;
|
|
};
|
|
|
|
// Follows the conventions of WavWriter.
|
|
class WavReader final : public WavFile {
|
|
public:
|
|
// Opens an existing WAV file for reading.
|
|
explicit WavReader(const std::string& filename);
|
|
explicit WavReader(FileWrapper file);
|
|
|
|
// Close the WAV file.
|
|
~WavReader() { Close(); }
|
|
|
|
WavReader(const WavReader&) = delete;
|
|
WavReader& operator=(const WavReader&) = delete;
|
|
|
|
// Resets position to the beginning of the file.
|
|
void Reset();
|
|
|
|
// Returns the number of samples read. If this is less than requested,
|
|
// verifies that the end of the file was reached.
|
|
size_t ReadSamples(size_t num_samples, float* samples);
|
|
size_t ReadSamples(size_t num_samples, int16_t* samples);
|
|
|
|
int sample_rate() const override { return sample_rate_; }
|
|
size_t num_channels() const override { return num_channels_; }
|
|
size_t num_samples() const override { return num_samples_in_file_; }
|
|
|
|
private:
|
|
void Close();
|
|
int sample_rate_;
|
|
size_t num_channels_;
|
|
WavFormat format_;
|
|
size_t num_samples_in_file_;
|
|
size_t num_unread_samples_;
|
|
FileWrapper file_;
|
|
int64_t
|
|
data_start_pos_; // Position in the file immediately after WAV header.
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // COMMON_AUDIO_WAV_FILE_H_
|