mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-19 08:37:54 +01:00
7ba3b81ff5
Add seek methods to FileWrapper, and refactor WavReader to use that class instead. Bug: webrtc:6463 Change-Id: Ifbb1989a072da6280ea5fc04b4beff991614dd53 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/147265 Commit-Queue: Niels Moller <nisse@webrtc.org> Reviewed-by: Karl Wiberg <kwiberg@webrtc.org> Cr-Commit-Position: refs/heads/master@{#28770}
106 lines
3.2 KiB
C++
106 lines
3.2 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 "rtc_base/constructor_magic.h"
|
|
#include "rtc_base/system/file_wrapper.h"
|
|
|
|
namespace webrtc {
|
|
|
|
// Interface to provide access to WAV file parameters.
|
|
class WavFile {
|
|
public:
|
|
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 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:
|
|
// Open a new WAV file for writing.
|
|
WavWriter(const std::string& filename, int sample_rate, size_t num_channels);
|
|
|
|
// Open a new WAV file for writing.
|
|
WavWriter(FileWrapper file, int sample_rate, size_t num_channels);
|
|
|
|
// Close the WAV file, after writing its header.
|
|
~WavWriter() override;
|
|
|
|
// Write additional samples to the file. Each sample is in the range
|
|
// [-32768,32767], 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;
|
|
size_t num_channels() const override;
|
|
size_t num_samples() const override;
|
|
|
|
private:
|
|
void Close();
|
|
const int sample_rate_;
|
|
const size_t num_channels_;
|
|
size_t num_samples_; // Total number of samples written to file.
|
|
FileWrapper file_; // Output file, owned by this class
|
|
|
|
RTC_DISALLOW_COPY_AND_ASSIGN(WavWriter);
|
|
};
|
|
|
|
// Follows the conventions of WavWriter.
|
|
class WavReader final : public WavFile {
|
|
public:
|
|
// Opens an existing WAV file for reading.
|
|
explicit WavReader(const std::string& filename);
|
|
|
|
// Use an existing WAV file for reading.
|
|
explicit WavReader(FileWrapper file);
|
|
|
|
// Close the WAV file.
|
|
~WavReader() override;
|
|
|
|
// 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;
|
|
size_t num_channels() const override;
|
|
size_t num_samples() const override;
|
|
|
|
private:
|
|
void Close();
|
|
int sample_rate_;
|
|
size_t num_channels_;
|
|
size_t num_samples_; // Total number of samples in the file.
|
|
size_t num_samples_remaining_;
|
|
FileWrapper file_; // Input file, owned by this class.
|
|
int64_t
|
|
data_start_pos_; // Position in the file immediately after WAV header.
|
|
|
|
RTC_DISALLOW_COPY_AND_ASSIGN(WavReader);
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // COMMON_AUDIO_WAV_FILE_H_
|