Think both encoder and decoder for SB-CELP work...
[speexdsp.git] / libspeex / sb_celp.c
1 /* Copyright (C) 2002 Jean-Marc Valin 
2    File: speex.c
3
4    This library is free software; you can redistribute it and/or
5    modify it under the terms of the GNU Lesser General Public
6    License as published by the Free Software Foundation; either
7    version 2.1 of the License, or (at your option) any later version.
8    
9    This library is distributed in the hope that it will be useful,
10    but WITHOUT ANY WARRANTY; without even the implied warranty of
11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12    Lesser General Public License for more details.
13    
14    You should have received a copy of the GNU Lesser General Public
15    License along with this library; if not, write to the Free Software
16    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
17 */
18
19
20 #include "speex.h"
21 #include "sb_celp.h"
22 #include "stdlib.h"
23 #include "filters.h"
24 #include <math.h>
25 #include "lpc.h"
26 #include "lsp.h"
27 #include <stdio.h>
28 #include "stack_alloc.h"
29 #include "cb_search.h"
30 #include "quant_lsp.h"
31
32 #ifndef M_PI
33 #define M_PI           3.14159265358979323846  /* pi */
34 #endif
35
36 extern float stoc[];
37
38 #define sqr(x) ((x)*(x))
39
40 float quant_high_gain[16]={
41    -2.387860,
42    -1.504710,
43    -0.988013,
44    -0.610249,
45    -0.310298,
46    -0.050495,
47    0.188963,
48    0.413744,
49    0.628971,
50    0.840555,
51    1.055630,
52    1.283410,
53    1.544990,
54    1.855790,
55    2.281910,
56    3.002660
57 };
58
59
60 float quant_high_gain2[8] = {
61    -1.51541,
62    -0.70324,
63    -0.17024,
64    0.26748,
65    0.67232,
66    1.08402,
67    1.56110,
68    2.25160,
69 };
70
71 #if 0
72 #define QMF_ORDER 32
73 static float h0[32] = {
74    0.0006910579, -0.001403793,
75    -0.001268303, 0.004234195,
76    0.001414246, -0.009458318,
77    -0.0001303859, 0.01798145,
78    -0.004187483, -0.03123862,
79    0.01456844, 0.05294745,
80    -0.03934878, -0.09980243,
81    0.1285579, 0.4664053,
82    0.4664053, 0.1285579,
83    -0.09980243, -0.03934878,
84    0.05294745, 0.01456844,
85    -0.03123862, -0.004187483,
86    0.01798145, -0.0001303859,
87    -0.009458318, 0.001414246,
88    0.004234195, -0.001268303,
89    -0.001403793, 0.0006910579
90 };
91
92 static float h1[32] = {
93    0.0006910579, 0.001403793,
94    -0.001268303, -0.004234195,
95    0.001414246, 0.009458318,
96    -0.0001303859, -0.01798145,
97    -0.004187483, 0.03123862,
98    0.01456844, -0.05294745,
99    -0.03934878, 0.09980243,
100    0.1285579, -0.4664053,
101    0.4664053, -0.1285579,
102    -0.09980243, 0.03934878,
103    0.05294745, -0.01456844,
104    -0.03123862, 0.004187483,
105    0.01798145, 0.0001303859,
106    -0.009458318, -0.001414246,
107    0.004234195, 0.001268303,
108    -0.001403793, -0.0006910579
109 };
110 #else 
111 #define QMF_ORDER 64
112 static float h0[64] = {
113    3.596189e-05, -0.0001123515,
114    -0.0001104587, 0.0002790277,
115    0.0002298438, -0.0005953563,
116    -0.0003823631, 0.00113826,
117    0.0005308539, -0.001986177,
118    -0.0006243724, 0.003235877,
119    0.0005743159, -0.004989147,
120    -0.0002584767, 0.007367171,
121    -0.0004857935, -0.01050689,
122    0.001894714, 0.01459396,
123    -0.004313674, -0.01994365,
124    0.00828756, 0.02716055,
125    -0.01485397, -0.03764973,
126    0.026447, 0.05543245,
127    -0.05095487, -0.09779096,
128    0.1382363, 0.4600981,
129    0.4600981, 0.1382363,
130    -0.09779096, -0.05095487,
131    0.05543245, 0.026447,
132    -0.03764973, -0.01485397,
133    0.02716055, 0.00828756,
134    -0.01994365, -0.004313674,
135    0.01459396, 0.001894714,
136    -0.01050689, -0.0004857935,
137    0.007367171, -0.0002584767,
138    -0.004989147, 0.0005743159,
139    0.003235877, -0.0006243724,
140    -0.001986177, 0.0005308539,
141    0.00113826, -0.0003823631,
142    -0.0005953563, 0.0002298438,
143    0.0002790277, -0.0001104587,
144    -0.0001123515, 3.596189e-05
145 };
146
147 static float h1[64] = {
148    3.596189e-05, 0.0001123515,
149    -0.0001104587, -0.0002790277,
150    0.0002298438, 0.0005953563,
151    -0.0003823631, -0.00113826,
152    0.0005308539, 0.001986177,
153    -0.0006243724, -0.003235877,
154    0.0005743159, 0.004989147,
155    -0.0002584767, -0.007367171,
156    -0.0004857935, 0.01050689,
157    0.001894714, -0.01459396,
158    -0.004313674, 0.01994365,
159    0.00828756, -0.02716055,
160    -0.01485397, 0.03764973,
161    0.026447, -0.05543245,
162    -0.05095487, 0.09779096,
163    0.1382363, -0.4600981,
164    0.4600981, -0.1382363,
165    -0.09779096, 0.05095487,
166    0.05543245, -0.026447,
167    -0.03764973, 0.01485397,
168    0.02716055, -0.00828756,
169    -0.01994365, 0.004313674,
170    0.01459396, -0.001894714,
171    -0.01050689, 0.0004857935,
172    0.007367171, 0.0002584767,
173    -0.004989147, -0.0005743159,
174    0.003235877, 0.0006243724,
175    -0.001986177, -0.0005308539,
176    0.00113826, 0.0003823631,
177    -0.0005953563, -0.0002298438,
178    0.0002790277, 0.0001104587,
179    -0.0001123515, -3.596189e-05
180 };
181 #endif
182
183 void sb_encoder_init(SBEncState *st, SpeexMode *mode)
184 {
185    int i;
186    encoder_init(&st->st_low, mode);
187    st->full_frame_size = 2*st->st_low.frameSize;
188    st->frame_size = st->st_low.frameSize;
189    st->subframeSize = 40;
190    st->nbSubframes = 4;
191    st->windowSize = mode->windowSize;
192    st->lpcSize=8;
193
194    st->lag_factor = .002;
195    st->lpc_floor = 1.0001;
196    st->gamma1=.9;
197    st->gamma2=.6;
198    st->first=1;
199    st->stack = calloc(10000, sizeof(float));
200
201    st->x0=calloc(st->full_frame_size, sizeof(float));
202    st->x1=calloc(st->full_frame_size, sizeof(float));
203    st->x0d=calloc(st->frame_size, sizeof(float));
204    st->x1d=calloc(st->frame_size, sizeof(float));
205    st->high=calloc(st->full_frame_size, sizeof(float));
206    st->y0=calloc(st->full_frame_size, sizeof(float));
207    st->y1=calloc(st->full_frame_size, sizeof(float));
208
209    st->h0_mem=calloc(QMF_ORDER, sizeof(float));
210    st->h1_mem=calloc(QMF_ORDER, sizeof(float));
211    st->g0_mem=calloc(QMF_ORDER, sizeof(float));
212    st->g1_mem=calloc(QMF_ORDER, sizeof(float));
213
214    st->buf=calloc(st->windowSize, sizeof(float));
215    st->excBuf=calloc(2*st->frame_size, sizeof(float));
216    st->exc=st->excBuf+st->frame_size;
217
218    st->res=calloc(st->frame_size, sizeof(float));
219    st->sw=calloc(st->frame_size, sizeof(float));
220    st->target=calloc(st->frame_size, sizeof(float));
221    st->window=calloc(st->windowSize, sizeof(float));
222    for (i=0;i<st->windowSize;i++)
223       st->window[i]=.5*(1-cos(2*M_PI*i/st->windowSize));
224
225    st->lagWindow = malloc((st->lpcSize+1)*sizeof(float));
226    for (i=0;i<st->lpcSize+1;i++)
227       st->lagWindow[i]=exp(-.5*sqr(2*M_PI*st->lag_factor*i));
228
229    st->rc = malloc(st->lpcSize*sizeof(float));
230    st->autocorr = malloc((st->lpcSize+1)*sizeof(float));
231    st->lpc = malloc((st->lpcSize+1)*sizeof(float));
232    st->bw_lpc1 = malloc((st->lpcSize+1)*sizeof(float));
233    st->bw_lpc2 = malloc((st->lpcSize+1)*sizeof(float));
234    st->lsp = malloc(st->lpcSize*sizeof(float));
235    st->qlsp = malloc(st->lpcSize*sizeof(float));
236    st->old_lsp = malloc(st->lpcSize*sizeof(float));
237    st->old_qlsp = malloc(st->lpcSize*sizeof(float));
238    st->interp_lsp = malloc(st->lpcSize*sizeof(float));
239    st->interp_qlsp = malloc(st->lpcSize*sizeof(float));
240    st->interp_lpc = malloc((st->lpcSize+1)*sizeof(float));
241    st->interp_qlpc = malloc((st->lpcSize+1)*sizeof(float));
242
243    st->mem_sp = calloc(st->lpcSize, sizeof(float));
244    st->mem_sp2 = calloc(st->lpcSize, sizeof(float));
245    st->mem_sw = calloc(st->lpcSize, sizeof(float));
246
247 }
248
249 void sb_encoder_destroy(SBEncState *st)
250 {
251    encoder_destroy(&st->st_low);
252    free(st->x0);
253    free(st->x0d);
254    free(st->x1);
255    free(st->x1d);
256    free(st->high);
257    free(st->y0);
258    free(st->y1);
259    free(st->h0_mem);
260    free(st->h1_mem);
261    free(st->g0_mem);
262    free(st->g1_mem);
263    
264    free(st->buf);
265    free(st->window);
266    free(st->excBuf);
267    free(st->sw);
268    free(st->res);
269    free(st->target);
270    free(st->lagWindow);
271    free(st->rc);
272    free(st->autocorr);
273    free(st->lpc);
274    free(st->bw_lpc1);
275    free(st->bw_lpc2);
276    free(st->lsp);
277    free(st->qlsp);
278    free(st->old_lsp);
279    free(st->old_qlsp);
280    free(st->interp_lsp);
281    free(st->interp_qlsp);
282    free(st->interp_lpc);
283    free(st->interp_qlpc);
284
285    free(st->mem_sp);
286    free(st->mem_sp2);
287    free(st->mem_sw);
288
289    free(st->stack);
290    
291 }
292
293
294 void sb_encode(SBEncState *st, float *in, FrameBits *bits)
295 {
296    int i, roots, sub;
297    /* Compute the two sub-bands by filtering with h0 and h1*/
298    fir_mem(in, h0, st->x0, st->full_frame_size, QMF_ORDER, st->h0_mem);
299    fir_mem(in, h1, st->x1, st->full_frame_size, QMF_ORDER, st->h1_mem);
300    /* Down-sample x0 and x1 */
301    for (i=0;i<st->frame_size;i++)
302    {
303       st->x0d[i]=st->x0[i<<1];
304       st->x1d[i]=st->x1[i<<1];
305    }
306    /* Encode the narrowband part*/
307    encode(&st->st_low, st->x0d, bits);
308
309    /* High-band buffering / sync with low band */
310    for (i=0;i<st->frame_size;i++)
311    {
312       st->excBuf[i]=st->exc[i];
313       st->high[i]=st->high[st->frame_size+i];
314       st->high[st->frame_size+i]=st->x1d[i];
315    }
316    
317    /* Start encoding the high-band */
318
319    for (i=0;i<st->windowSize;i++)
320       st->buf[i] = st->high[i] * st->window[i];
321
322    /* Compute auto-correlation */
323    autocorr(st->buf, st->autocorr, st->lpcSize+1, st->windowSize);
324
325    st->autocorr[0] += 1;        /* prevents NANs */
326    st->autocorr[0] *= st->lpc_floor; /* Noise floor in auto-correlation domain */
327    /* Lag windowing: equivalent to filtering in the power-spectrum domain */
328    for (i=0;i<st->lpcSize+1;i++)
329       st->autocorr[i] *= st->lagWindow[i];
330
331    /* Levinson-Durbin */
332    wld(st->lpc+1, st->autocorr, st->rc, st->lpcSize);
333    st->lpc[0]=1;
334
335    /* LPC to LSPs (x-domain) transform */
336    roots=lpc_to_lsp (st->lpc, st->lpcSize, st->lsp, 6, 0.002, st->stack);
337    if (roots!=st->lpcSize)
338    {
339       fprintf (stderr, "roots!=st->lpcSize (found only %d roots)\n", roots);
340       exit(1);
341    }
342
343    /* x-domain to angle domain*/
344    for (i=0;i<st->lpcSize;i++)
345       st->lsp[i] = acos(st->lsp[i]);
346
347    /* LSP quantization */
348    lsp_quant_high(st->lsp, st->qlsp, st->lpcSize, bits);
349    
350    /*printf ("high_lsp:");
351    for (i=0;i<st->lpcSize;i++)
352       printf (" %f", st->lsp[i]);
353       printf ("\n");
354    for (i=0;i<st->lpcSize;i++)
355    st->qlsp[i]=st->lsp[i];
356    */
357
358    if (st->first)
359    {
360       for (i=0;i<st->lpcSize;i++)
361          st->old_lsp[i] = st->lsp[i];
362       for (i=0;i<st->lpcSize;i++)
363          st->old_qlsp[i] = st->qlsp[i];
364    }
365    
366    for (sub=0;sub<st->nbSubframes;sub++)
367    {
368       float *exc, *sp, *mem, *res, *target, *sw, tmp, filter_ratio;
369       int offset;
370       
371       offset = st->subframeSize*sub;
372       sp=st->high+offset;
373       exc=st->excBuf+offset;
374       res=st->res+offset;
375       target=st->target+offset;
376       sw=st->sw+offset;
377
378       mem=PUSH(st->stack, st->lpcSize);
379       
380       /* LSP interpolation (quantized and unquantized) */
381       tmp = (.5 + sub)/st->nbSubframes;
382       for (i=0;i<st->lpcSize;i++)
383          st->interp_lsp[i] = (1-tmp)*st->old_lsp[i] + tmp*st->lsp[i];
384       for (i=0;i<st->lpcSize;i++)
385          st->interp_qlsp[i] = (1-tmp)*st->old_qlsp[i] + tmp*st->qlsp[i];
386
387       /* Compute interpolated LPCs (quantized and unquantized) */
388       for (i=0;i<st->lpcSize;i++)
389          st->interp_lsp[i] = cos(st->interp_lsp[i]);
390       lsp_to_lpc(st->interp_lsp, st->interp_lpc, st->lpcSize,st->stack);
391
392       for (i=0;i<st->lpcSize;i++)
393          st->interp_qlsp[i] = cos(st->interp_qlsp[i]);
394       lsp_to_lpc(st->interp_qlsp, st->interp_qlpc, st->lpcSize, st->stack);
395
396       bw_lpc(st->gamma1, st->interp_lpc, st->bw_lpc1, st->lpcSize);
397       bw_lpc(st->gamma2, st->interp_lpc, st->bw_lpc2, st->lpcSize);
398
399       {
400          float tmp, rl=0, rh=0;
401          tmp=1;
402          for (i=0;i<=st->lpcSize;i++)
403          {
404             rh += tmp*st->interp_qlpc[i];
405             tmp = -tmp;
406          }
407          rl = st->st_low.pi_gain[sub];
408          rl=1/(fabs(rl)+.001);
409          rh=1/(fabs(rh)+.001);
410          filter_ratio=fabs(.001+rh)/(.001+fabs(rl));
411          printf ("filter_ratio: %f\n", filter_ratio);
412       }
413 #if 0 /* 1 for spectral folding excitation, 0 for stochastic */
414       for (i=0;i<st->lpcSize;i++)
415          mem[i]=st->mem_sp[i];
416       residue_mem(sp, st->interp_qlpc, exc, st->subframeSize, st->lpcSize, st->mem_sp);
417       {
418          float el=0,eh=0,g;
419          for (i=0;i<st->subframeSize;i++)
420             eh+=sqr(exc[i]);
421          for (i=0;i<st->subframeSize;i++)
422            el+=sqr(st->st_low.exc[offset+i]);
423          g=eh/(.01+el);
424          printf ("ratio: %f\n", g);
425          g=sqrt(g);
426          for (i=0;i<st->subframeSize;i++)
427            exc[i]=g*st->st_low.exc[offset+i];
428       }
429       syn_filt_mem(exc, st->interp_qlpc, sp, st->subframeSize, st->lpcSize, mem);
430
431 #else
432       /* Reset excitation */
433       for (i=0;i<st->subframeSize;i++)
434          exc[i]=0;
435
436       /* Compute zero response of A(z/g1) / ( A(z/g2) * Aq(z) ) */
437       for (i=0;i<st->lpcSize;i++)
438          mem[i]=st->mem_sp[i];
439       syn_filt_mem(exc, st->interp_qlpc, exc, st->subframeSize, st->lpcSize, mem);
440       for (i=0;i<st->lpcSize;i++)
441          mem[i]=st->mem_sp[i];
442       residue_mem(exc, st->bw_lpc1, res, st->subframeSize, st->lpcSize, mem);
443       for (i=0;i<st->lpcSize;i++)
444          mem[i]=st->mem_sw[i];
445       syn_filt_mem(res, st->bw_lpc2, res, st->subframeSize, st->lpcSize, mem);
446
447       /* Compute weighted signal */
448       for (i=0;i<st->lpcSize;i++)
449          mem[i]=st->mem_sp[i];
450       residue_mem(sp, st->bw_lpc1, sw, st->subframeSize, st->lpcSize, mem);
451       for (i=0;i<st->lpcSize;i++)
452          mem[i]=st->mem_sw[i];
453       syn_filt_mem(sw, st->bw_lpc2, sw, st->subframeSize, st->lpcSize, mem);
454
455       /* Compute target signal */
456       for (i=0;i<st->subframeSize;i++)
457          target[i]=sw[i]-res[i];
458       {
459 #if 0
460          
461          float el=0,eh=0,g;
462          residue_mem(sp, st->interp_qlpc, exc, st->subframeSize, st->lpcSize, st->mem_sp2);
463          
464          for (i=0;i<st->subframeSize;i++)
465             eh+=sqr(exc[i]);
466          overlap_cb_search(target, st->interp_qlpc, st->bw_lpc1, st->bw_lpc2,
467                            &stoc[0], 512, &gain, &ind, st->lpcSize,
468                            st->subframeSize);
469          for (i=0;i<st->subframeSize;i++)
470             exc[i]=gain*stoc[ind+i];
471          for (i=0;i<st->subframeSize;i++)
472             el+=sqr(exc[i]);
473          g=sqrt(eh/(el+.001));
474          for (i=0;i<st->subframeSize;i++)
475             exc[i]*=g;
476          
477 #else
478          int k,N=4;
479          float el=0,eh=0,g;
480          int *index;
481          float *gains;
482          gains = PUSH(st->stack, N);
483          index = (int*) PUSH(st->stack, N);
484          residue_mem(sp, st->interp_qlpc, exc, st->subframeSize, st->lpcSize, st->mem_sp2);
485          
486          for (i=0;i<st->subframeSize;i++)
487             eh+=sqr(exc[i]);
488
489          for (i=0;i<st->subframeSize;i++)
490             exc[i]=0;
491          for (k=0;k<N;k++)
492          {
493             int of=k*st->subframeSize/N;
494          overlap_cb_search(target+of, st->interp_qlpc, st->bw_lpc1, st->bw_lpc2,
495                            &stoc[0], 64, &gains[k], &index[k], st->lpcSize,
496                            st->subframeSize/N);
497
498          frame_bits_pack(bits,index[k],6);
499
500          for (i=0;i<st->subframeSize;i++)
501             res[i]=0;
502          for (i=0;i<st->subframeSize/N;i++)
503             res[of+i]=gains[k]*stoc[index[k]+i];
504          residue_zero(res, st->bw_lpc1, res, st->subframeSize, st->lpcSize);
505          syn_filt_zero(res, st->interp_qlpc, res, st->subframeSize, st->lpcSize);
506          syn_filt_zero(res, st->bw_lpc2, res, st->subframeSize, st->lpcSize);
507          for (i=0;i<st->subframeSize;i++)
508             target[i]-=res[i];
509          for (i=0;i<st->subframeSize/N;i++)
510             exc[of+i]+=gains[k]*stoc[index[k]+i];
511          }
512          for (i=0;i<st->subframeSize;i++)
513             el+=sqr(exc[i]);
514          g=sqrt(eh/(el+.001));
515
516          for (i=0;i<st->subframeSize;i++)
517             el+=sqr(st->st_low.exc[offset+i]);
518
519          for (k=0;k<N;k++)
520          {
521             int sign=0;
522             float quant, dist, min_dist;
523             int best_ind;
524             int of=k*st->subframeSize/N;
525             gains[k]*=g;
526
527             if (gains[k]<0)
528             {
529                sign=1;
530                gains[k] = -gains[k];
531             }
532             quant = log((1+gains[k])*filter_ratio/(1+sqrt(el/st->subframeSize)));
533             min_dist = sqr(quant-quant_high_gain2[0]);
534             best_ind=0;
535             for (i=1;i<8;i++)
536             {
537                dist = sqr(quant-quant_high_gain2[i]);
538                if (dist<min_dist)
539                {
540                   best_ind=i;
541                   min_dist=dist;
542                }
543             }
544             quant=quant_high_gain2[best_ind];
545
546             frame_bits_pack(bits,sign,1);
547             frame_bits_pack(bits,best_ind,3);
548
549             gains[k]=exp(quant)*(1+sqrt(el/st->subframeSize))/filter_ratio;
550             if (sign)
551                gains[k] = -gains[k];
552
553             for (i=0;i<st->subframeSize/N;i++)
554                exc[of+i]=gains[k]*stoc[index[k]+i];
555          }
556          /*for (i=0;i<st->subframeSize;i++)
557            exc[i]*=g;*/
558          POP(st->stack);
559          POP(st->stack);
560 #endif
561       }
562
563       /*Keep the previous memory*/
564       for (i=0;i<st->lpcSize;i++)
565          mem[i]=st->mem_sp[i];
566       /* Final signal synthesis from excitation */
567       syn_filt_mem(exc, st->interp_qlpc, sp, st->subframeSize, st->lpcSize, st->mem_sp);
568        
569       /* Compute weighted signal again, from synthesized speech (not sure it's the right thing) */
570       residue_mem(sp, st->bw_lpc1, sw, st->subframeSize, st->lpcSize, mem);
571       syn_filt_mem(sw, st->bw_lpc2, sw, st->subframeSize, st->lpcSize, st->mem_sw);
572 #endif
573
574       POP(st->stack);
575    }
576
577    /* Up-sample coded low-band and high-band*/
578    for (i=0;i<st->frame_size;i++)
579    {
580       st->x0[(i<<1)]=st->x0d[i];
581       st->x1[(i<<1)]=st->high[i];
582       st->x0[(i<<1)+1]=0;
583       st->x1[(i<<1)+1]=0;
584    }
585    /* Reconstruct the original */
586    fir_mem(st->x0, h0, st->y0, st->full_frame_size, QMF_ORDER, st->g0_mem);
587    fir_mem(st->x1, h1, st->y1, st->full_frame_size, QMF_ORDER, st->g1_mem);
588    for (i=0;i<st->full_frame_size;i++)
589       in[i]=2*(st->y0[i]-st->y1[i]);
590
591    for (i=0;i<st->lpcSize;i++)
592       st->old_lsp[i] = st->lsp[i];
593    for (i=0;i<st->lpcSize;i++)
594       st->old_qlsp[i] = st->qlsp[i];
595
596    st->first=0;
597 }
598
599
600
601
602
603 void sb_decoder_init(SBDecState *st, SpeexMode *mode)
604 {
605    decoder_init(&st->st_low, mode);
606    st->full_frame_size = 2*st->st_low.frameSize;
607    st->frame_size = st->st_low.frameSize;
608    st->subframeSize = 40;
609    st->nbSubframes = 4;
610    st->lpcSize=8;
611
612    st->first=1;
613    st->stack = calloc(10000, sizeof(float));
614
615    st->x0=calloc(st->full_frame_size, sizeof(float));
616    st->x1=calloc(st->full_frame_size, sizeof(float));
617    st->x0d=calloc(st->frame_size, sizeof(float));
618    st->x1d=calloc(st->frame_size, sizeof(float));
619    st->high=calloc(st->full_frame_size, sizeof(float));
620    st->y0=calloc(st->full_frame_size, sizeof(float));
621    st->y1=calloc(st->full_frame_size, sizeof(float));
622
623    st->h0_mem=calloc(QMF_ORDER, sizeof(float));
624    st->h1_mem=calloc(QMF_ORDER, sizeof(float));
625    st->g0_mem=calloc(QMF_ORDER, sizeof(float));
626    st->g1_mem=calloc(QMF_ORDER, sizeof(float));
627
628    st->exc=calloc(st->frame_size, sizeof(float));
629
630    st->qlsp = malloc(st->lpcSize*sizeof(float));
631    st->old_qlsp = malloc(st->lpcSize*sizeof(float));
632    st->interp_qlsp = malloc(st->lpcSize*sizeof(float));
633    st->interp_qlpc = malloc((st->lpcSize+1)*sizeof(float));
634
635    st->mem_sp = calloc(st->lpcSize, sizeof(float));
636    st->mem_sw = calloc(st->lpcSize, sizeof(float));
637
638 }
639
640 void sb_decoder_destroy(SBDecState *st)
641 {
642    decoder_destroy(&st->st_low);
643    free(st->x0);
644    free(st->x0d);
645    free(st->x1);
646    free(st->x1d);
647    free(st->high);
648    free(st->y0);
649    free(st->y1);
650    free(st->h0_mem);
651    free(st->h1_mem);
652    free(st->g0_mem);
653    free(st->g1_mem);
654    
655    free(st->exc);
656    free(st->qlsp);
657    free(st->old_qlsp);
658    free(st->interp_qlsp);
659    free(st->interp_qlpc);
660
661    free(st->mem_sp);
662    free(st->mem_sw);
663
664    free(st->stack);
665    
666 }
667
668
669 void sb_decode(SBDecState *st, FrameBits *bits, float *out)
670 {
671    int i, sub;
672
673    /* Decode the low-band */
674    decode(&st->st_low, bits, st->x0d);
675
676    for (i=0;i<st->frame_size;i++)
677       st->exc[i]=0;
678
679    lsp_unquant_high(st->qlsp, st->lpcSize, bits);
680    
681    if (st->first)
682    {
683       for (i=0;i<st->lpcSize;i++)
684          st->old_qlsp[i] = st->qlsp[i];
685    }
686    
687    for (sub=0;sub<st->nbSubframes;sub++)
688    {
689       float *exc, *sp, tmp, filter_ratio;
690       int offset;
691       
692       offset = st->subframeSize*sub;
693       sp=st->high+offset;
694       exc=st->exc+offset;
695       
696       /* LSP interpolation */
697       tmp = (.5 + sub)/st->nbSubframes;
698       for (i=0;i<st->lpcSize;i++)
699          st->interp_qlsp[i] = (1-tmp)*st->old_qlsp[i] + tmp*st->qlsp[i];
700
701       /* LSPs to x-domain */
702       for (i=0;i<st->lpcSize;i++)
703          st->interp_qlsp[i] = cos(st->interp_qlsp[i]);
704
705       /* LSP to LPC */
706       lsp_to_lpc(st->interp_qlsp, st->interp_qlpc, st->lpcSize, st->stack);
707
708       {
709          float rl=0, rh=0;
710          tmp=1;
711          for (i=0;i<=st->lpcSize;i++)
712          {
713             rh += tmp*st->interp_qlpc[i];
714             tmp = -tmp;
715          }
716          rl = st->st_low.pi_gain[sub];
717          rl=1/(fabs(rl)+.001);
718          rh=1/(fabs(rh)+.001);
719          filter_ratio=fabs(.001+rh)/(.001+fabs(rl));
720          printf ("filter_ratio: %f\n", filter_ratio);
721       }
722
723       {
724          int k,N=4,el=0;
725          int *index;
726          float *gains;
727          int of;
728          gains = PUSH(st->stack, N);
729          index = (int*) PUSH(st->stack, N);
730
731          for (i=0;i<st->subframeSize;i++)
732            el+=sqr(st->st_low.exc[offset+i]);
733          
734          for (k=0;k<N;k++)
735             index[k] = frame_bits_unpack_unsigned(bits,6);
736
737          for (k=0;k<N;k++)
738          {
739             int sign, gain_ind;
740             sign = frame_bits_unpack_unsigned(bits,1);
741             gain_ind = frame_bits_unpack_unsigned(bits,3);
742             gains[k]=exp(quant_high_gain2[gain_ind])*(1+sqrt(el/st->subframeSize))/filter_ratio;
743             if (sign)
744                gains[k] =- gains[k];
745          }
746
747          for (k=0;k<N;k++)
748          {
749             of=k*st->subframeSize/N;
750             for (i=0;i<st->subframeSize/N;i++)
751                exc[of+i]=gains[k]*stoc[index[k]+i];
752          }
753
754          POP(st->stack);
755          POP(st->stack);
756       }
757
758       syn_filt_mem(exc, st->interp_qlpc, sp, st->subframeSize, st->lpcSize, st->mem_sp);
759
760    }
761
762    /* Up-sample coded low-band and high-band*/
763    for (i=0;i<st->frame_size;i++)
764    {
765       st->x0[(i<<1)]=st->x0d[i];
766       st->x1[(i<<1)]=st->high[i];
767       st->x0[(i<<1)+1]=0;
768       st->x1[(i<<1)+1]=0;
769    }
770    /* Reconstruct the original */
771    fir_mem(st->x0, h0, st->y0, st->full_frame_size, QMF_ORDER, st->g0_mem);
772    fir_mem(st->x1, h1, st->y1, st->full_frame_size, QMF_ORDER, st->g1_mem);
773    for (i=0;i<st->full_frame_size;i++)
774       out[i]=2*(st->y0[i]-st->y1[i]);
775
776    for (i=0;i<st->lpcSize;i++)
777       st->old_qlsp[i] = st->qlsp[i];
778
779    st->first=0;
780
781 }