| /* |
| * Copyright (c) 2018 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. |
| */ |
| |
| #include "modules/audio_processing/agc2/rnn_vad/pitch_search.h" |
| #include "modules/audio_processing/agc2/rnn_vad/pitch_search_internal.h" |
| |
| namespace webrtc { |
| namespace rnn_vad { |
| |
| PitchInfo PitchSearch(rtc::ArrayView<const float, kBufSize24kHz> pitch_buf, |
| PitchInfo prev_pitch_48kHz) { |
| // Perform the initial pitch search at 12 kHz. |
| std::array<float, kBufSize12kHz> pitch_buf_decimated; |
| Decimate2x(pitch_buf, |
| {pitch_buf_decimated.data(), pitch_buf_decimated.size()}); |
| // Compute auto-correlation terms. |
| std::array<float, kNumInvertedLags12kHz> auto_corr; |
| ComputePitchAutoCorrelation( |
| {pitch_buf_decimated.data(), pitch_buf_decimated.size()}, kMaxPitch12kHz, |
| {auto_corr.data(), auto_corr.size()}); |
| // Search for pitch at 12 kHz. |
| std::array<size_t, 2> pitch_candidates_inv_lags = FindBestPitchPeriods( |
| {auto_corr.data(), auto_corr.size()}, |
| {pitch_buf_decimated.data(), pitch_buf_decimated.size()}, kMaxPitch12kHz); |
| |
| // Refine the pitch period estimation. |
| // The refinement is done using the pitch buffer that contains 24 kHz samples. |
| // Therefore, adapt the inverted lags in |pitch_candidates_inv_lags| from 12 |
| // to 24 kHz. |
| for (size_t i = 0; i < pitch_candidates_inv_lags.size(); ++i) |
| pitch_candidates_inv_lags[i] *= 2; |
| size_t pitch_inv_lag_48kHz = RefinePitchPeriod48kHz( |
| pitch_buf, |
| {pitch_candidates_inv_lags.data(), pitch_candidates_inv_lags.size()}); |
| // Look for stronger harmonics to find the final pitch period and its gain. |
| RTC_DCHECK_LT(pitch_inv_lag_48kHz, kMaxPitch48kHz); |
| return CheckLowerPitchPeriodsAndComputePitchGain( |
| pitch_buf, kMaxPitch48kHz - pitch_inv_lag_48kHz, prev_pitch_48kHz); |
| } |
| |
| } // namespace rnn_vad |
| } // namespace webrtc |