123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215 |
- #ifndef COMMON_AUDIO_CHANNEL_BUFFER_H_
- #define COMMON_AUDIO_CHANNEL_BUFFER_H_
- #include <string.h>
- #include <memory>
- #include <vector>
- #include "api/array_view.h"
- #include "common_audio/include/audio_util.h"
- #include "rtc_base/checks.h"
- #include "rtc_base/gtest_prod_util.h"
- namespace webrtc {
- template <typename T>
- class ChannelBuffer {
- public:
- ChannelBuffer(size_t num_frames, size_t num_channels, size_t num_bands = 1)
- : data_(new T[num_frames * num_channels]()),
- channels_(new T*[num_channels * num_bands]),
- bands_(new T*[num_channels * num_bands]),
- num_frames_(num_frames),
- num_frames_per_band_(num_frames / num_bands),
- num_allocated_channels_(num_channels),
- num_channels_(num_channels),
- num_bands_(num_bands),
- bands_view_(num_allocated_channels_,
- std::vector<rtc::ArrayView<T>>(num_bands_)),
- channels_view_(
- num_bands_,
- std::vector<rtc::ArrayView<T>>(num_allocated_channels_)) {
-
- auto* bands_view =
- const_cast<std::vector<std::vector<rtc::ArrayView<T>>>*>(&bands_view_);
- auto* channels_view =
- const_cast<std::vector<std::vector<rtc::ArrayView<T>>>*>(
- &channels_view_);
- for (size_t ch = 0; ch < num_allocated_channels_; ++ch) {
- for (size_t band = 0; band < num_bands_; ++band) {
- (*channels_view)[band][ch] = rtc::ArrayView<T>(
- &data_[ch * num_frames_ + band * num_frames_per_band_],
- num_frames_per_band_);
- (*bands_view)[ch][band] = channels_view_[band][ch];
- channels_[band * num_allocated_channels_ + ch] =
- channels_view_[band][ch].data();
- bands_[ch * num_bands_ + band] =
- channels_[band * num_allocated_channels_ + ch];
- }
- }
- }
-
-
-
-
-
-
-
-
-
-
-
-
- const T* const* channels(size_t band = 0) const {
- RTC_DCHECK_LT(band, num_bands_);
- return &channels_[band * num_allocated_channels_];
- }
- T* const* channels(size_t band = 0) {
- const ChannelBuffer<T>* t = this;
- return const_cast<T* const*>(t->channels(band));
- }
- rtc::ArrayView<const rtc::ArrayView<T>> channels_view(size_t band = 0) {
- return channels_view_[band];
- }
- rtc::ArrayView<const rtc::ArrayView<T>> channels_view(size_t band = 0) const {
- return channels_view_[band];
- }
-
-
-
-
-
-
-
- const T* const* bands(size_t channel) const {
- RTC_DCHECK_LT(channel, num_channels_);
- RTC_DCHECK_GE(channel, 0);
- return &bands_[channel * num_bands_];
- }
- T* const* bands(size_t channel) {
- const ChannelBuffer<T>* t = this;
- return const_cast<T* const*>(t->bands(channel));
- }
- rtc::ArrayView<const rtc::ArrayView<T>> bands_view(size_t channel) {
- return bands_view_[channel];
- }
- rtc::ArrayView<const rtc::ArrayView<T>> bands_view(size_t channel) const {
- return bands_view_[channel];
- }
-
-
- const T* const* Slice(T** slice, size_t start_frame) const {
- RTC_DCHECK_LT(start_frame, num_frames_);
- for (size_t i = 0; i < num_channels_; ++i)
- slice[i] = &channels_[i][start_frame];
- return slice;
- }
- T** Slice(T** slice, size_t start_frame) {
- const ChannelBuffer<T>* t = this;
- return const_cast<T**>(t->Slice(slice, start_frame));
- }
- size_t num_frames() const { return num_frames_; }
- size_t num_frames_per_band() const { return num_frames_per_band_; }
- size_t num_channels() const { return num_channels_; }
- size_t num_bands() const { return num_bands_; }
- size_t size() const { return num_frames_ * num_allocated_channels_; }
- void set_num_channels(size_t num_channels) {
- RTC_DCHECK_LE(num_channels, num_allocated_channels_);
- num_channels_ = num_channels;
- }
- void SetDataForTesting(const T* data, size_t size) {
- RTC_CHECK_EQ(size, this->size());
- memcpy(data_.get(), data, size * sizeof(*data));
- }
- private:
- std::unique_ptr<T[]> data_;
- std::unique_ptr<T*[]> channels_;
- std::unique_ptr<T*[]> bands_;
- const size_t num_frames_;
- const size_t num_frames_per_band_;
-
- const size_t num_allocated_channels_;
-
- size_t num_channels_;
- const size_t num_bands_;
- const std::vector<std::vector<rtc::ArrayView<T>>> bands_view_;
- const std::vector<std::vector<rtc::ArrayView<T>>> channels_view_;
- };
- class IFChannelBuffer {
- public:
- IFChannelBuffer(size_t num_frames, size_t num_channels, size_t num_bands = 1);
- ~IFChannelBuffer();
- ChannelBuffer<int16_t>* ibuf();
- ChannelBuffer<float>* fbuf();
- const ChannelBuffer<int16_t>* ibuf_const() const;
- const ChannelBuffer<float>* fbuf_const() const;
- size_t num_frames() const { return ibuf_.num_frames(); }
- size_t num_frames_per_band() const { return ibuf_.num_frames_per_band(); }
- size_t num_channels() const {
- return ivalid_ ? ibuf_.num_channels() : fbuf_.num_channels();
- }
- void set_num_channels(size_t num_channels) {
- ibuf_.set_num_channels(num_channels);
- fbuf_.set_num_channels(num_channels);
- }
- size_t num_bands() const { return ibuf_.num_bands(); }
- private:
- void RefreshF() const;
- void RefreshI() const;
- mutable bool ivalid_;
- mutable ChannelBuffer<int16_t> ibuf_;
- mutable bool fvalid_;
- mutable ChannelBuffer<float> fbuf_;
- };
- }
- #endif
|