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
11 #include "webrtc/modules/audio_coding/codecs/isac/fix/source/pitch_estimator.h"
12
13 #include "webrtc/common_audio/signal_processing/include/signal_processing_library.h"
14 #include "webrtc/modules/audio_coding/codecs/isac/fix/source/settings.h"
15 #include "webrtc/modules/audio_coding/codecs/isac/fix/source/structs.h"
16 #include "webrtc/system_wrappers/include/compile_assert_c.h"
17
18 // Number of segments in a pitch subframe.
19 static const int kSegments = 5;
20
21 // A division factor of 1/5 in Q15.
22 static const int16_t kDivFactor = 6553;
23
24 // Interpolation coefficients; generated by design_pitch_filter.m.
25 // Coefficients are stored in Q14.
26 static const int16_t kIntrpCoef[PITCH_FRACS][PITCH_FRACORDER] = {
27 {-367, 1090, -2706, 9945, 10596, -3318, 1626, -781, 287},
28 {-325, 953, -2292, 7301, 12963, -3320, 1570, -743, 271},
29 {-240, 693, -1622, 4634, 14809, -2782, 1262, -587, 212},
30 {-125, 358, -817, 2144, 15982, -1668, 721, -329, 118},
31 { 0, 0, -1, 1, 16380, 1, -1, 0, 0},
32 { 118, -329, 721, -1668, 15982, 2144, -817, 358, -125},
33 { 212, -587, 1262, -2782, 14809, 4634, -1622, 693, -240},
34 { 271, -743, 1570, -3320, 12963, 7301, -2292, 953, -325}
35 };
36
CalcLrIntQ(int16_t fixVal,int16_t qDomain)37 static __inline size_t CalcLrIntQ(int16_t fixVal,
38 int16_t qDomain) {
39 int32_t roundVal = 1 << (qDomain - 1);
40
41 return (fixVal + roundVal) >> qDomain;
42 }
43
WebRtcIsacfix_PitchFilter(int16_t * indatQQ,int16_t * outdatQQ,PitchFiltstr * pfp,int16_t * lagsQ7,int16_t * gainsQ12,int16_t type)44 void WebRtcIsacfix_PitchFilter(int16_t* indatQQ, // Q10 if type is 1 or 4,
45 // Q0 if type is 2.
46 int16_t* outdatQQ,
47 PitchFiltstr* pfp,
48 int16_t* lagsQ7,
49 int16_t* gainsQ12,
50 int16_t type) {
51 int k, ind, cnt;
52 int16_t sign = 1;
53 int16_t inystateQQ[PITCH_DAMPORDER];
54 int16_t ubufQQ[PITCH_INTBUFFSIZE + QLOOKAHEAD];
55 const int16_t Gain = 21299; // 1.3 in Q14
56 int16_t oldLagQ7;
57 int16_t oldGainQ12, lagdeltaQ7, curLagQ7, gaindeltaQ12, curGainQ12;
58 size_t indW32 = 0, frcQQ = 0;
59 const int16_t* fracoeffQQ = NULL;
60
61 // Assumptions in ARM assembly for WebRtcIsacfix_PitchFilterCoreARM().
62 COMPILE_ASSERT(PITCH_FRACORDER == 9);
63 COMPILE_ASSERT(PITCH_DAMPORDER == 5);
64
65 // Set up buffer and states.
66 memcpy(ubufQQ, pfp->ubufQQ, sizeof(pfp->ubufQQ));
67 memcpy(inystateQQ, pfp->ystateQQ, sizeof(inystateQQ));
68
69 // Get old lag and gain value from memory.
70 oldLagQ7 = pfp->oldlagQ7;
71 oldGainQ12 = pfp->oldgainQ12;
72
73 if (type == 4) {
74 sign = -1;
75
76 // Make output more periodic.
77 for (k = 0; k < PITCH_SUBFRAMES; k++) {
78 gainsQ12[k] = (int16_t)(gainsQ12[k] * Gain >> 14);
79 }
80 }
81
82 // No interpolation if pitch lag step is big.
83 if (((lagsQ7[0] * 3 >> 1) < oldLagQ7) || (lagsQ7[0] > (oldLagQ7 * 3 >> 1))) {
84 oldLagQ7 = lagsQ7[0];
85 oldGainQ12 = gainsQ12[0];
86 }
87
88 ind = 0;
89
90 for (k = 0; k < PITCH_SUBFRAMES; k++) {
91 // Calculate interpolation steps.
92 lagdeltaQ7 = lagsQ7[k] - oldLagQ7;
93 lagdeltaQ7 = (int16_t)WEBRTC_SPL_MUL_16_16_RSFT_WITH_ROUND(
94 lagdeltaQ7, kDivFactor, 15);
95 curLagQ7 = oldLagQ7;
96 gaindeltaQ12 = gainsQ12[k] - oldGainQ12;
97 gaindeltaQ12 = (int16_t)(gaindeltaQ12 * kDivFactor >> 15);
98
99 curGainQ12 = oldGainQ12;
100 oldLagQ7 = lagsQ7[k];
101 oldGainQ12 = gainsQ12[k];
102
103 // Each frame has 4 60-sample pitch subframes, and each subframe has 5
104 // 12-sample segments. Each segment need to be processed with
105 // newly-updated parameters, so we break the pitch filtering into
106 // two for-loops (5 x 12) below. It's also why kDivFactor = 0.2 (in Q15).
107 for (cnt = 0; cnt < kSegments; cnt++) {
108 // Update parameters for each segment.
109 curGainQ12 += gaindeltaQ12;
110 curLagQ7 += lagdeltaQ7;
111 indW32 = CalcLrIntQ(curLagQ7, 7);
112 frcQQ = ((indW32 << 7) + 64 - curLagQ7) >> 4;
113
114 if (frcQQ == PITCH_FRACS) {
115 frcQQ = 0;
116 }
117 fracoeffQQ = kIntrpCoef[frcQQ];
118
119 // Pitch filtering.
120 WebRtcIsacfix_PitchFilterCore(PITCH_SUBFRAME_LEN / kSegments, curGainQ12,
121 indW32, sign, inystateQQ, ubufQQ, fracoeffQQ, indatQQ, outdatQQ, &ind);
122 }
123 }
124
125 // Export buffer and states.
126 memcpy(pfp->ubufQQ, ubufQQ + PITCH_FRAME_LEN, sizeof(pfp->ubufQQ));
127 memcpy(pfp->ystateQQ, inystateQQ, sizeof(pfp->ystateQQ));
128
129 pfp->oldlagQ7 = oldLagQ7;
130 pfp->oldgainQ12 = oldGainQ12;
131
132 if (type == 2) {
133 // Filter look-ahead segment.
134 WebRtcIsacfix_PitchFilterCore(QLOOKAHEAD, curGainQ12, indW32, 1, inystateQQ,
135 ubufQQ, fracoeffQQ, indatQQ, outdatQQ, &ind);
136 }
137 }
138
139
WebRtcIsacfix_PitchFilterGains(const int16_t * indatQ0,PitchFiltstr * pfp,int16_t * lagsQ7,int16_t * gainsQ12)140 void WebRtcIsacfix_PitchFilterGains(const int16_t* indatQ0,
141 PitchFiltstr* pfp,
142 int16_t* lagsQ7,
143 int16_t* gainsQ12) {
144 int k, n, m;
145 size_t ind, pos, pos3QQ;
146
147 int16_t ubufQQ[PITCH_INTBUFFSIZE];
148 int16_t oldLagQ7, lagdeltaQ7, curLagQ7;
149 const int16_t* fracoeffQQ = NULL;
150 int16_t scale;
151 int16_t cnt = 0, tmpW16;
152 size_t frcQQ, indW16 = 0;
153 int32_t tmpW32, tmp2W32, csum1QQ, esumxQQ;
154
155 // Set up buffer and states.
156 memcpy(ubufQQ, pfp->ubufQQ, sizeof(pfp->ubufQQ));
157 oldLagQ7 = pfp->oldlagQ7;
158
159 // No interpolation if pitch lag step is big.
160 if (((lagsQ7[0] * 3 >> 1) < oldLagQ7) || (lagsQ7[0] > (oldLagQ7 * 3 >> 1))) {
161 oldLagQ7 = lagsQ7[0];
162 }
163
164 ind = 0;
165 pos = ind + PITCH_BUFFSIZE;
166 scale = 0;
167 for (k = 0; k < PITCH_SUBFRAMES; k++) {
168
169 // Calculate interpolation steps.
170 lagdeltaQ7 = lagsQ7[k] - oldLagQ7;
171 lagdeltaQ7 = (int16_t)WEBRTC_SPL_MUL_16_16_RSFT_WITH_ROUND(
172 lagdeltaQ7, kDivFactor, 15);
173 curLagQ7 = oldLagQ7;
174 oldLagQ7 = lagsQ7[k];
175
176 csum1QQ = 1;
177 esumxQQ = 1;
178
179 // Same as function WebRtcIsacfix_PitchFilter(), we break the pitch
180 // filtering into two for-loops (5 x 12) below.
181 for (cnt = 0; cnt < kSegments; cnt++) {
182 // Update parameters for each segment.
183 curLagQ7 += lagdeltaQ7;
184 indW16 = CalcLrIntQ(curLagQ7, 7);
185 frcQQ = ((indW16 << 7) + 64 - curLagQ7) >> 4;
186
187 if (frcQQ == PITCH_FRACS) {
188 frcQQ = 0;
189 }
190 fracoeffQQ = kIntrpCoef[frcQQ];
191
192 pos3QQ = pos - (indW16 + 4);
193
194 for (n = 0; n < PITCH_SUBFRAME_LEN / kSegments; n++) {
195 // Filter to get fractional pitch.
196
197 tmpW32 = 0;
198 for (m = 0; m < PITCH_FRACORDER; m++) {
199 tmpW32 += ubufQQ[pos3QQ + m] * fracoeffQQ[m];
200 }
201
202 // Subtract from input and update buffer.
203 ubufQQ[pos] = indatQ0[ind];
204
205 tmp2W32 = WEBRTC_SPL_MUL_16_32_RSFT14(indatQ0[ind], tmpW32);
206 tmpW32 += 8192;
207 tmpW16 = tmpW32 >> 14;
208 tmpW32 = tmpW16 * tmpW16;
209
210 if ((tmp2W32 > 1073700000) || (csum1QQ > 1073700000) ||
211 (tmpW32 > 1073700000) || (esumxQQ > 1073700000)) { // 2^30
212 scale++;
213 csum1QQ >>= 1;
214 esumxQQ >>= 1;
215 }
216 csum1QQ += tmp2W32 >> scale;
217 esumxQQ += tmpW32 >> scale;
218
219 ind++;
220 pos++;
221 pos3QQ++;
222 }
223 }
224
225 if (csum1QQ < esumxQQ) {
226 tmp2W32 = WebRtcSpl_DivResultInQ31(csum1QQ, esumxQQ);
227
228 // Gain should be half the correlation.
229 tmpW32 = tmp2W32 >> 20;
230 } else {
231 tmpW32 = 4096;
232 }
233 gainsQ12[k] = (int16_t)WEBRTC_SPL_SAT(PITCH_MAX_GAIN_Q12, tmpW32, 0);
234 }
235
236 // Export buffer and states.
237 memcpy(pfp->ubufQQ, ubufQQ + PITCH_FRAME_LEN, sizeof(pfp->ubufQQ));
238 pfp->oldlagQ7 = lagsQ7[PITCH_SUBFRAMES - 1];
239 pfp->oldgainQ12 = gainsQ12[PITCH_SUBFRAMES - 1];
240
241 }
242