minyue | 3548357 | 2016-09-20 23:13:08 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2016 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 06:47:31 +0200 | [diff] [blame] | 11 | #ifndef COMMON_AUDIO_SMOOTHING_FILTER_H_ |
| 12 | #define COMMON_AUDIO_SMOOTHING_FILTER_H_ |
minyue | 3548357 | 2016-09-20 23:13:08 -0700 | [diff] [blame] | 13 | |
Yves Gerey | 988cc08 | 2018-10-23 12:03:01 +0200 | [diff] [blame] | 14 | #include <stdint.h> |
| 15 | |
Danil Chapovalov | 196100e | 2018-06-21 10:17:24 +0200 | [diff] [blame] | 16 | #include "absl/types/optional.h" |
Steve Anton | 10542f2 | 2019-01-11 09:11:00 -0800 | [diff] [blame] | 17 | #include "rtc_base/constructor_magic.h" |
minyue | 3548357 | 2016-09-20 23:13:08 -0700 | [diff] [blame] | 18 | |
| 19 | namespace webrtc { |
| 20 | |
| 21 | class SmoothingFilter { |
| 22 | public: |
| 23 | virtual ~SmoothingFilter() = default; |
| 24 | virtual void AddSample(float sample) = 0; |
Danil Chapovalov | 196100e | 2018-06-21 10:17:24 +0200 | [diff] [blame] | 25 | virtual absl::optional<float> GetAverage() = 0; |
minyue | 301fc4a | 2016-12-13 06:52:56 -0800 | [diff] [blame] | 26 | virtual bool SetTimeConstantMs(int time_constant_ms) = 0; |
minyue | 3548357 | 2016-09-20 23:13:08 -0700 | [diff] [blame] | 27 | }; |
| 28 | |
| 29 | // SmoothingFilterImpl applies an exponential filter |
minyue | 301fc4a | 2016-12-13 06:52:56 -0800 | [diff] [blame] | 30 | // alpha = exp(-1.0 / time_constant_ms); |
minyue | 3548357 | 2016-09-20 23:13:08 -0700 | [diff] [blame] | 31 | // y[t] = alpha * y[t-1] + (1 - alpha) * sample; |
minyue | 301fc4a | 2016-12-13 06:52:56 -0800 | [diff] [blame] | 32 | // This implies a sample rate of 1000 Hz, i.e., 1 sample / ms. |
| 33 | // But SmoothingFilterImpl allows sparse samples. All missing samples will be |
| 34 | // assumed to equal the last received sample. |
minyue | 3548357 | 2016-09-20 23:13:08 -0700 | [diff] [blame] | 35 | class SmoothingFilterImpl final : public SmoothingFilter { |
| 36 | public: |
minyue | 301fc4a | 2016-12-13 06:52:56 -0800 | [diff] [blame] | 37 | // |init_time_ms| is initialization time. It defines a period starting from |
| 38 | // the arriving time of the first sample. During this period, the exponential |
| 39 | // filter uses a varying time constant so that a smaller time constant will be |
| 40 | // applied to the earlier samples. This is to allow the the filter to adapt to |
| 41 | // earlier samples quickly. After the initialization period, the time constant |
| 42 | // will be set to |init_time_ms| first and can be changed through |
| 43 | // |SetTimeConstantMs|. |
michaelt | 92aef17 | 2017-04-18 00:11:48 -0700 | [diff] [blame] | 44 | explicit SmoothingFilterImpl(int init_time_ms); |
minyue | 301fc4a | 2016-12-13 06:52:56 -0800 | [diff] [blame] | 45 | ~SmoothingFilterImpl() override; |
minyue | 3548357 | 2016-09-20 23:13:08 -0700 | [diff] [blame] | 46 | |
| 47 | void AddSample(float sample) override; |
Danil Chapovalov | 196100e | 2018-06-21 10:17:24 +0200 | [diff] [blame] | 48 | absl::optional<float> GetAverage() override; |
minyue | 301fc4a | 2016-12-13 06:52:56 -0800 | [diff] [blame] | 49 | bool SetTimeConstantMs(int time_constant_ms) override; |
| 50 | |
| 51 | // Methods used for unittests. |
| 52 | float alpha() const { return alpha_; } |
minyue | 3548357 | 2016-09-20 23:13:08 -0700 | [diff] [blame] | 53 | |
| 54 | private: |
minyue | 301fc4a | 2016-12-13 06:52:56 -0800 | [diff] [blame] | 55 | void UpdateAlpha(int time_constant_ms); |
| 56 | void ExtrapolateLastSample(int64_t time_ms); |
| 57 | |
| 58 | const int init_time_ms_; |
| 59 | const float init_factor_; |
| 60 | const float init_const_; |
minyue | 3548357 | 2016-09-20 23:13:08 -0700 | [diff] [blame] | 61 | |
Danil Chapovalov | 196100e | 2018-06-21 10:17:24 +0200 | [diff] [blame] | 62 | absl::optional<int64_t> init_end_time_ms_; |
minyue | 301fc4a | 2016-12-13 06:52:56 -0800 | [diff] [blame] | 63 | float last_sample_; |
| 64 | float alpha_; |
| 65 | float state_; |
| 66 | int64_t last_state_time_ms_; |
minyue | 3548357 | 2016-09-20 23:13:08 -0700 | [diff] [blame] | 67 | |
minyue | 69b627d | 2016-11-24 11:01:09 -0800 | [diff] [blame] | 68 | RTC_DISALLOW_IMPLICIT_CONSTRUCTORS(SmoothingFilterImpl); |
minyue | 3548357 | 2016-09-20 23:13:08 -0700 | [diff] [blame] | 69 | }; |
| 70 | |
| 71 | } // namespace webrtc |
| 72 | |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 73 | #endif // COMMON_AUDIO_SMOOTHING_FILTER_H_ |