henrik.lundin@webrtc.org | d94659d | 2013-01-29 12:09:21 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. |
| 3 | * |
| 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 |
| 6 | * tree. An additional intellectual property rights grant can be found |
| 7 | * in the file PATENTS. All contributing project authors may |
| 8 | * be found in the AUTHORS file in the root of the source tree. |
| 9 | */ |
| 10 | |
Mirko Bonadei | 92ea95e | 2017-09-15 04:47:31 | [diff] [blame] | 11 | #ifndef MODULES_AUDIO_CODING_NETEQ_ACCELERATE_H_ |
| 12 | #define MODULES_AUDIO_CODING_NETEQ_ACCELERATE_H_ |
henrik.lundin@webrtc.org | d94659d | 2013-01-29 12:09:21 | [diff] [blame] | 13 | |
Yves Gerey | 988cc08 | 2018-10-23 10:03:01 | [diff] [blame] | 14 | #include <stddef.h> |
| 15 | #include <stdint.h> |
| 16 | |
Mirko Bonadei | 92ea95e | 2017-09-15 04:47:31 | [diff] [blame] | 17 | #include "modules/audio_coding/neteq/time_stretch.h" |
henrik.lundin@webrtc.org | d94659d | 2013-01-29 12:09:21 | [diff] [blame] | 18 | |
| 19 | namespace webrtc { |
| 20 | |
Yves Gerey | 988cc08 | 2018-10-23 10:03:01 | [diff] [blame] | 21 | class AudioMultiVector; |
henrik.lundin@webrtc.org | d94659d | 2013-01-29 12:09:21 | [diff] [blame] | 22 | class BackgroundNoise; |
| 23 | |
| 24 | // This class implements the Accelerate operation. Most of the work is done |
| 25 | // in the base class TimeStretch, which is shared with the PreemptiveExpand |
| 26 | // operation. In the Accelerate class, the operations that are specific to |
| 27 | // Accelerate are implemented. |
| 28 | class Accelerate : public TimeStretch { |
| 29 | public: |
Yves Gerey | 665174f | 2018-06-19 13:03:05 | [diff] [blame] | 30 | Accelerate(int sample_rate_hz, |
| 31 | size_t num_channels, |
henrik.lundin@webrtc.org | d94659d | 2013-01-29 12:09:21 | [diff] [blame] | 32 | const BackgroundNoise& background_noise) |
Yves Gerey | 665174f | 2018-06-19 13:03:05 | [diff] [blame] | 33 | : TimeStretch(sample_rate_hz, num_channels, background_noise) {} |
henrik.lundin@webrtc.org | d94659d | 2013-01-29 12:09:21 | [diff] [blame] | 34 | |
Byoungchan Lee | 604fd2f | 2022-01-21 00:49:39 | [diff] [blame] | 35 | Accelerate(const Accelerate&) = delete; |
| 36 | Accelerate& operator=(const Accelerate&) = delete; |
| 37 | |
henrik.lundin@webrtc.org | d94659d | 2013-01-29 12:09:21 | [diff] [blame] | 38 | // This method performs the actual Accelerate operation. The samples are |
Artem Titov | d00ce74 | 2021-07-28 18:00:17 | [diff] [blame] | 39 | // read from `input`, of length `input_length` elements, and are written to |
| 40 | // `output`. The number of samples removed through time-stretching is |
| 41 | // is provided in the output `length_change_samples`. The method returns |
| 42 | // the outcome of the operation as an enumerator value. If `fast_accelerate` |
Henrik Lundin | cf808d2 | 2015-05-27 12:33:29 | [diff] [blame] | 43 | // is true, the algorithm will relax the requirements on finding strong |
| 44 | // correlations, and may remove multiple pitch periods if possible. |
henrik.lundin@webrtc.org | d94659d | 2013-01-29 12:09:21 | [diff] [blame] | 45 | ReturnCodes Process(const int16_t* input, |
turaj@webrtc.org | 362a55e | 2013-09-20 16:25:28 | [diff] [blame] | 46 | size_t input_length, |
Henrik Lundin | cf808d2 | 2015-05-27 12:33:29 | [diff] [blame] | 47 | bool fast_accelerate, |
henrik.lundin@webrtc.org | fd11bbf | 2013-09-30 20:38:44 | [diff] [blame] | 48 | AudioMultiVector* output, |
Peter Kasting | dce40cf | 2015-08-24 21:52:23 | [diff] [blame] | 49 | size_t* length_change_samples); |
henrik.lundin@webrtc.org | d94659d | 2013-01-29 12:09:21 | [diff] [blame] | 50 | |
| 51 | protected: |
Artem Titov | d00ce74 | 2021-07-28 18:00:17 | [diff] [blame] | 52 | // Sets the parameters `best_correlation` and `peak_index` to suitable |
henrik.lundin@webrtc.org | d94659d | 2013-01-29 12:09:21 | [diff] [blame] | 53 | // values when the signal contains no active speech. |
kjellander@webrtc.org | 14665ff | 2015-03-04 12:58:35 | [diff] [blame] | 54 | void SetParametersForPassiveSpeech(size_t len, |
| 55 | int16_t* best_correlation, |
Peter Kasting | dce40cf | 2015-08-24 21:52:23 | [diff] [blame] | 56 | size_t* peak_index) const override; |
henrik.lundin@webrtc.org | d94659d | 2013-01-29 12:09:21 | [diff] [blame] | 57 | |
| 58 | // Checks the criteria for performing the time-stretching operation and, |
| 59 | // if possible, performs the time-stretching. |
kjellander@webrtc.org | 14665ff | 2015-03-04 12:58:35 | [diff] [blame] | 60 | ReturnCodes CheckCriteriaAndStretch(const int16_t* input, |
| 61 | size_t input_length, |
| 62 | size_t peak_index, |
| 63 | int16_t best_correlation, |
| 64 | bool active_speech, |
Henrik Lundin | cf808d2 | 2015-05-27 12:33:29 | [diff] [blame] | 65 | bool fast_mode, |
kjellander@webrtc.org | 14665ff | 2015-03-04 12:58:35 | [diff] [blame] | 66 | AudioMultiVector* output) const override; |
henrik.lundin@webrtc.org | d94659d | 2013-01-29 12:09:21 | [diff] [blame] | 67 | }; |
| 68 | |
henrik.lundin@webrtc.org | d9faa46 | 2014-01-14 10:18:45 | [diff] [blame] | 69 | struct AccelerateFactory { |
| 70 | AccelerateFactory() {} |
| 71 | virtual ~AccelerateFactory() {} |
| 72 | |
| 73 | virtual Accelerate* Create(int sample_rate_hz, |
| 74 | size_t num_channels, |
| 75 | const BackgroundNoise& background_noise) const; |
| 76 | }; |
| 77 | |
henrik.lundin@webrtc.org | d94659d | 2013-01-29 12:09:21 | [diff] [blame] | 78 | } // namespace webrtc |
Mirko Bonadei | 92ea95e | 2017-09-15 04:47:31 | [diff] [blame] | 79 | #endif // MODULES_AUDIO_CODING_NETEQ_ACCELERATE_H_ |