| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2013 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 |
| 11 #include "webrtc/common_audio/resampler/push_sinc_resampler.h" | 11 #include "webrtc/common_audio/resampler/push_sinc_resampler.h" |
| 12 | 12 |
| 13 #include <cstring> | 13 #include <cstring> |
| 14 | 14 |
| 15 #include "webrtc/base/checks.h" | 15 #include "webrtc/base/checks.h" |
| 16 #include "webrtc/common_audio/include/audio_util.h" | 16 #include "webrtc/common_audio/include/audio_util.h" |
| 17 | 17 |
| 18 namespace webrtc { | 18 namespace webrtc { |
| 19 | 19 |
| 20 PushSincResampler::PushSincResampler(int source_frames, int destination_frames) | 20 PushSincResampler::PushSincResampler(size_t source_frames, |
| 21 size_t destination_frames) |
| 21 : resampler_(new SincResampler(source_frames * 1.0 / destination_frames, | 22 : resampler_(new SincResampler(source_frames * 1.0 / destination_frames, |
| 22 source_frames, | 23 source_frames, |
| 23 this)), | 24 this)), |
| 24 source_ptr_(nullptr), | 25 source_ptr_(nullptr), |
| 25 source_ptr_int_(nullptr), | 26 source_ptr_int_(nullptr), |
| 26 destination_frames_(destination_frames), | 27 destination_frames_(destination_frames), |
| 27 first_pass_(true), | 28 first_pass_(true), |
| 28 source_available_(0) {} | 29 source_available_(0) {} |
| 29 | 30 |
| 30 PushSincResampler::~PushSincResampler() { | 31 PushSincResampler::~PushSincResampler() { |
| 31 } | 32 } |
| 32 | 33 |
| 33 int PushSincResampler::Resample(const int16_t* source, | 34 size_t PushSincResampler::Resample(const int16_t* source, |
| 34 int source_length, | 35 size_t source_length, |
| 35 int16_t* destination, | 36 int16_t* destination, |
| 36 int destination_capacity) { | 37 size_t destination_capacity) { |
| 37 if (!float_buffer_.get()) | 38 if (!float_buffer_.get()) |
| 38 float_buffer_.reset(new float[destination_frames_]); | 39 float_buffer_.reset(new float[destination_frames_]); |
| 39 | 40 |
| 40 source_ptr_int_ = source; | 41 source_ptr_int_ = source; |
| 41 // Pass nullptr as the float source to have Run() read from the int16 source. | 42 // Pass nullptr as the float source to have Run() read from the int16 source. |
| 42 Resample(nullptr, source_length, float_buffer_.get(), destination_frames_); | 43 Resample(nullptr, source_length, float_buffer_.get(), destination_frames_); |
| 43 FloatS16ToS16(float_buffer_.get(), destination_frames_, destination); | 44 FloatS16ToS16(float_buffer_.get(), destination_frames_, destination); |
| 44 source_ptr_int_ = nullptr; | 45 source_ptr_int_ = nullptr; |
| 45 return destination_frames_; | 46 return destination_frames_; |
| 46 } | 47 } |
| 47 | 48 |
| 48 int PushSincResampler::Resample(const float* source, | 49 size_t PushSincResampler::Resample(const float* source, |
| 49 int source_length, | 50 size_t source_length, |
| 50 float* destination, | 51 float* destination, |
| 51 int destination_capacity) { | 52 size_t destination_capacity) { |
| 52 CHECK_EQ(source_length, resampler_->request_frames()); | 53 CHECK_EQ(source_length, resampler_->request_frames()); |
| 53 CHECK_GE(destination_capacity, destination_frames_); | 54 CHECK_GE(destination_capacity, destination_frames_); |
| 54 // Cache the source pointer. Calling Resample() will immediately trigger | 55 // Cache the source pointer. Calling Resample() will immediately trigger |
| 55 // the Run() callback whereupon we provide the cached value. | 56 // the Run() callback whereupon we provide the cached value. |
| 56 source_ptr_ = source; | 57 source_ptr_ = source; |
| 57 source_available_ = source_length; | 58 source_available_ = source_length; |
| 58 | 59 |
| 59 // On the first pass, we call Resample() twice. During the first call, we | 60 // On the first pass, we call Resample() twice. During the first call, we |
| 60 // provide dummy input and discard the output. This is done to prime the | 61 // provide dummy input and discard the output. This is done to prime the |
| 61 // SincResampler buffer with the correct delay (half the kernel size), thereby | 62 // SincResampler buffer with the correct delay (half the kernel size), thereby |
| 62 // ensuring that all later Resample() calls will only result in one input | 63 // ensuring that all later Resample() calls will only result in one input |
| 63 // request through Run(). | 64 // request through Run(). |
| 64 // | 65 // |
| 65 // If this wasn't done, SincResampler would call Run() twice on the first | 66 // If this wasn't done, SincResampler would call Run() twice on the first |
| 66 // pass, and we'd have to introduce an entire |source_frames| of delay, rather | 67 // pass, and we'd have to introduce an entire |source_frames| of delay, rather |
| 67 // than the minimum half kernel. | 68 // than the minimum half kernel. |
| 68 // | 69 // |
| 69 // It works out that ChunkSize() is exactly the amount of output we need to | 70 // It works out that ChunkSize() is exactly the amount of output we need to |
| 70 // request in order to prime the buffer with a single Run() request for | 71 // request in order to prime the buffer with a single Run() request for |
| 71 // |source_frames|. | 72 // |source_frames|. |
| 72 if (first_pass_) | 73 if (first_pass_) |
| 73 resampler_->Resample(resampler_->ChunkSize(), destination); | 74 resampler_->Resample(resampler_->ChunkSize(), destination); |
| 74 | 75 |
| 75 resampler_->Resample(destination_frames_, destination); | 76 resampler_->Resample(destination_frames_, destination); |
| 76 source_ptr_ = nullptr; | 77 source_ptr_ = nullptr; |
| 77 return destination_frames_; | 78 return destination_frames_; |
| 78 } | 79 } |
| 79 | 80 |
| 80 void PushSincResampler::Run(int frames, float* destination) { | 81 void PushSincResampler::Run(size_t frames, float* destination) { |
| 81 // Ensure we are only asked for the available samples. This would fail if | 82 // Ensure we are only asked for the available samples. This would fail if |
| 82 // Run() was triggered more than once per Resample() call. | 83 // Run() was triggered more than once per Resample() call. |
| 83 CHECK_EQ(source_available_, frames); | 84 CHECK_EQ(source_available_, frames); |
| 84 | 85 |
| 85 if (first_pass_) { | 86 if (first_pass_) { |
| 86 // Provide dummy input on the first pass, the output of which will be | 87 // Provide dummy input on the first pass, the output of which will be |
| 87 // discarded, as described in Resample(). | 88 // discarded, as described in Resample(). |
| 88 std::memset(destination, 0, frames * sizeof(*destination)); | 89 std::memset(destination, 0, frames * sizeof(*destination)); |
| 89 first_pass_ = false; | 90 first_pass_ = false; |
| 90 return; | 91 return; |
| 91 } | 92 } |
| 92 | 93 |
| 93 if (source_ptr_) { | 94 if (source_ptr_) { |
| 94 std::memcpy(destination, source_ptr_, frames * sizeof(*destination)); | 95 std::memcpy(destination, source_ptr_, frames * sizeof(*destination)); |
| 95 } else { | 96 } else { |
| 96 for (int i = 0; i < frames; ++i) | 97 for (size_t i = 0; i < frames; ++i) |
| 97 destination[i] = static_cast<float>(source_ptr_int_[i]); | 98 destination[i] = static_cast<float>(source_ptr_int_[i]); |
| 98 } | 99 } |
| 99 source_available_ -= frames; | 100 source_available_ -= frames; |
| 100 } | 101 } |
| 101 | 102 |
| 102 } // namespace webrtc | 103 } // namespace webrtc |
| OLD | NEW |