OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2014 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 |
(...skipping 65 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
76 int in_channels, | 76 int in_channels, |
77 int frames, | 77 int frames, |
78 int out_channels, | 78 int out_channels, |
79 std::complex<float>* const* out_block); | 79 std::complex<float>* const* out_block); |
80 | 80 |
81 private: | 81 private: |
82 IntelligibilityEnhancer* parent_; | 82 IntelligibilityEnhancer* parent_; |
83 AudioSource source_; | 83 AudioSource source_; |
84 }; | 84 }; |
85 friend class TransformCallback; | 85 friend class TransformCallback; |
| 86 FRIEND_TEST_ALL_PREFIXES(EnhancerTest, TestErbCreation); |
| 87 FRIEND_TEST_ALL_PREFIXES(EnhancerTest, TestSolveForGains); |
86 | 88 |
87 // Sends streams to ProcessClearBlock or ProcessNoiseBlock based on source. | 89 // Sends streams to ProcessClearBlock or ProcessNoiseBlock based on source. |
88 void DispatchAudio(AudioSource source, | 90 void DispatchAudio(AudioSource source, |
89 const std::complex<float>* in_block, | 91 const std::complex<float>* in_block, |
90 std::complex<float>* out_block); | 92 std::complex<float>* out_block); |
91 | 93 |
92 // Updates variance computation and analysis with |in_block_|, | 94 // Updates variance computation and analysis with |in_block_|, |
93 // and writes modified speech to |out_block|. | 95 // and writes modified speech to |out_block|. |
94 void ProcessClearBlock(const std::complex<float>* in_block, | 96 void ProcessClearBlock(const std::complex<float>* in_block, |
95 std::complex<float>* out_block); | 97 std::complex<float>* out_block); |
96 | 98 |
97 // Computes and sets modified gains. | 99 // Computes and sets modified gains. |
98 void AnalyzeClearBlock(float power_target); | 100 void AnalyzeClearBlock(float power_target); |
99 | 101 |
| 102 // Bisection search for optimal |lambda|. |
| 103 void SolveForLambda(float power_target, float power_bot, float power_top); |
| 104 |
| 105 // Transforms freq gains to ERB gains. |
| 106 void UpdateErbGains(); |
| 107 |
100 // Updates variance calculation for noise input with |in_block|. | 108 // Updates variance calculation for noise input with |in_block|. |
101 void ProcessNoiseBlock(const std::complex<float>* in_block, | 109 void ProcessNoiseBlock(const std::complex<float>* in_block, |
102 std::complex<float>* out_block); | 110 std::complex<float>* out_block); |
103 | 111 |
104 // Returns number of ERB filters. | 112 // Returns number of ERB filters. |
105 static int GetBankSize(int sample_rate, int erb_resolution); | 113 static int GetBankSize(int sample_rate, int erb_resolution); |
106 | 114 |
107 // Initializes ERB filterbank. | 115 // Initializes ERB filterbank. |
108 void CreateErbBank(); | 116 void CreateErbBank(); |
109 | 117 |
(...skipping 10 matching lines...) Expand all Loading... |
120 | 128 |
121 static const int kErbResolution; | 129 static const int kErbResolution; |
122 static const int kWindowSizeMs; | 130 static const int kWindowSizeMs; |
123 static const int kChunkSizeMs; | 131 static const int kChunkSizeMs; |
124 static const int kAnalyzeRate; // Default for |analysis_rate_|. | 132 static const int kAnalyzeRate; // Default for |analysis_rate_|. |
125 static const int kVarianceRate; // Default for |variance_rate_|. | 133 static const int kVarianceRate; // Default for |variance_rate_|. |
126 static const float kClipFreq; | 134 static const float kClipFreq; |
127 static const float kConfigRho; // Default production and interpretation SNR. | 135 static const float kConfigRho; // Default production and interpretation SNR. |
128 static const float kKbdAlpha; | 136 static const float kKbdAlpha; |
129 static const float kGainChangeLimit; | 137 static const float kGainChangeLimit; |
| 138 static const float kLambdaBot; // Extreme values in bisection |
| 139 static const float kLambdaTop; // search for lambda. |
130 | 140 |
131 const int freqs_; // Num frequencies in frequency domain. | 141 const int freqs_; // Num frequencies in frequency domain. |
132 const int window_size_; // Window size in samples; also the block size. | 142 const int window_size_; // Window size in samples; also the block size. |
133 const int chunk_length_; // Chunk size in samples. | 143 const int chunk_length_; // Chunk size in samples. |
134 const int bank_size_; // Num ERB filters. | 144 const int bank_size_; // Num ERB filters. |
135 const int sample_rate_hz_; | 145 const int sample_rate_hz_; |
136 const int erb_resolution_; | 146 const int erb_resolution_; |
137 const int channels_; // Num channels. | 147 const int channels_; // Num channels. |
138 const int analysis_rate_; // Num blocks before gains recalculated. | 148 const int analysis_rate_; // Num blocks before gains recalculated. |
139 const int variance_rate_; // Num recalculations before history is cleared. | 149 const int variance_rate_; // Num recalculations before history is cleared. |
(...skipping 29 matching lines...) Expand all Loading... |
169 // Note: VAD currently does not affect anything in IntelligibilityEnhancer. | 179 // Note: VAD currently does not affect anything in IntelligibilityEnhancer. |
170 VadInst* vad_high_; | 180 VadInst* vad_high_; |
171 VadInst* vad_low_; | 181 VadInst* vad_low_; |
172 rtc::scoped_ptr<int16_t[]> vad_tmp_buffer_; | 182 rtc::scoped_ptr<int16_t[]> vad_tmp_buffer_; |
173 bool has_voice_low_; // Whether voice detected in speech stream. | 183 bool has_voice_low_; // Whether voice detected in speech stream. |
174 }; | 184 }; |
175 | 185 |
176 } // namespace webrtc | 186 } // namespace webrtc |
177 | 187 |
178 #endif // WEBRTC_MODULES_AUDIO_PROCESSING_INTELLIGIBILITY_INTELLIGIBILITY_ENHAN
CER_H_ | 188 #endif // WEBRTC_MODULES_AUDIO_PROCESSING_INTELLIGIBILITY_INTELLIGIBILITY_ENHAN
CER_H_ |
OLD | NEW |