1 | /******************************************************************** |
2 | * * |
3 | * THIS FILE IS PART OF THE OggVorbis SOFTWARE CODEC SOURCE CODE. * |
4 | * USE, DISTRIBUTION AND REPRODUCTION OF THIS LIBRARY SOURCE IS * |
5 | * GOVERNED BY A BSD-STYLE SOURCE LICENSE INCLUDED WITH THIS SOURCE * |
6 | * IN 'COPYING'. PLEASE READ THESE TERMS BEFORE DISTRIBUTING. * |
7 | * * |
8 | * THE OggVorbis SOURCE CODE IS (C) COPYRIGHT 1994-2015 * |
9 | * by the Xiph.Org Foundation https://xiph.org/ * |
10 | * * |
11 | ******************************************************************** |
12 | |
13 | function: floor backend 0 implementation |
14 | |
15 | ********************************************************************/ |
16 | |
17 | #include <stdlib.h> |
18 | #include <string.h> |
19 | #include <math.h> |
20 | #include <ogg/ogg.h> |
21 | #include "vorbis/codec.h" |
22 | #include "codec_internal.h" |
23 | #include "registry.h" |
24 | #include "lpc.h" |
25 | #include "lsp.h" |
26 | #include "codebook.h" |
27 | #include "scales.h" |
28 | #include "misc.h" |
29 | #include "os.h" |
30 | |
31 | #include "misc.h" |
32 | #include <stdio.h> |
33 | |
34 | typedef struct { |
35 | int ln; |
36 | int m; |
37 | int **linearmap; |
38 | int n[2]; |
39 | |
40 | vorbis_info_floor0 *vi; |
41 | |
42 | long bits; |
43 | long frames; |
44 | } vorbis_look_floor0; |
45 | |
46 | |
47 | /***********************************************/ |
48 | |
49 | static void floor0_free_info(vorbis_info_floor *i){ |
50 | vorbis_info_floor0 *info=(vorbis_info_floor0 *)i; |
51 | if(info){ |
52 | memset(info,0,sizeof(*info)); |
53 | _ogg_free(info); |
54 | } |
55 | } |
56 | |
57 | static void floor0_free_look(vorbis_look_floor *i){ |
58 | vorbis_look_floor0 *look=(vorbis_look_floor0 *)i; |
59 | if(look){ |
60 | |
61 | if(look->linearmap){ |
62 | |
63 | if(look->linearmap[0])_ogg_free(look->linearmap[0]); |
64 | if(look->linearmap[1])_ogg_free(look->linearmap[1]); |
65 | |
66 | _ogg_free(look->linearmap); |
67 | } |
68 | memset(look,0,sizeof(*look)); |
69 | _ogg_free(look); |
70 | } |
71 | } |
72 | |
73 | static vorbis_info_floor *floor0_unpack (vorbis_info *vi,oggpack_buffer *opb){ |
74 | codec_setup_info *ci=vi->codec_setup; |
75 | int j; |
76 | |
77 | vorbis_info_floor0 *info=_ogg_malloc(sizeof(*info)); |
78 | info->order=oggpack_read(opb,8); |
79 | info->rate=oggpack_read(opb,16); |
80 | info->barkmap=oggpack_read(opb,16); |
81 | info->ampbits=oggpack_read(opb,6); |
82 | info->ampdB=oggpack_read(opb,8); |
83 | info->numbooks=oggpack_read(opb,4)+1; |
84 | |
85 | if(info->order<1)goto err_out; |
86 | if(info->rate<1)goto err_out; |
87 | if(info->barkmap<1)goto err_out; |
88 | if(info->numbooks<1)goto err_out; |
89 | |
90 | for(j=0;j<info->numbooks;j++){ |
91 | info->books[j]=oggpack_read(opb,8); |
92 | if(info->books[j]<0 || info->books[j]>=ci->books)goto err_out; |
93 | if(ci->book_param[info->books[j]]->maptype==0)goto err_out; |
94 | if(ci->book_param[info->books[j]]->dim<1)goto err_out; |
95 | } |
96 | return(info); |
97 | |
98 | err_out: |
99 | floor0_free_info(info); |
100 | return(NULL); |
101 | } |
102 | |
103 | /* initialize Bark scale and normalization lookups. We could do this |
104 | with static tables, but Vorbis allows a number of possible |
105 | combinations, so it's best to do it computationally. |
106 | |
107 | The below is authoritative in terms of defining scale mapping. |
108 | Note that the scale depends on the sampling rate as well as the |
109 | linear block and mapping sizes */ |
110 | |
111 | static void floor0_map_lazy_init(vorbis_block *vb, |
112 | vorbis_info_floor *infoX, |
113 | vorbis_look_floor0 *look){ |
114 | if(!look->linearmap[vb->W]){ |
115 | vorbis_dsp_state *vd=vb->vd; |
116 | vorbis_info *vi=vd->vi; |
117 | codec_setup_info *ci=vi->codec_setup; |
118 | vorbis_info_floor0 *info=(vorbis_info_floor0 *)infoX; |
119 | int W=vb->W; |
120 | int n=ci->blocksizes[W]/2,j; |
121 | |
122 | /* we choose a scaling constant so that: |
123 | floor(bark(rate/2-1)*C)=mapped-1 |
124 | floor(bark(rate/2)*C)=mapped */ |
125 | float scale=look->ln/toBARK(info->rate/2.f); |
126 | |
127 | /* the mapping from a linear scale to a smaller bark scale is |
128 | straightforward. We do *not* make sure that the linear mapping |
129 | does not skip bark-scale bins; the decoder simply skips them and |
130 | the encoder may do what it wishes in filling them. They're |
131 | necessary in some mapping combinations to keep the scale spacing |
132 | accurate */ |
133 | look->linearmap[W]=_ogg_malloc((n+1)*sizeof(**look->linearmap)); |
134 | for(j=0;j<n;j++){ |
135 | int val=floor( toBARK((info->rate/2.f)/n*j) |
136 | *scale); /* bark numbers represent band edges */ |
137 | if(val>=look->ln)val=look->ln-1; /* guard against the approximation */ |
138 | look->linearmap[W][j]=val; |
139 | } |
140 | look->linearmap[W][j]=-1; |
141 | look->n[W]=n; |
142 | } |
143 | } |
144 | |
145 | static vorbis_look_floor *floor0_look(vorbis_dsp_state *vd, |
146 | vorbis_info_floor *i){ |
147 | vorbis_info_floor0 *info=(vorbis_info_floor0 *)i; |
148 | vorbis_look_floor0 *look=_ogg_calloc(1,sizeof(*look)); |
149 | |
150 | (void)vd; |
151 | |
152 | look->m=info->order; |
153 | look->ln=info->barkmap; |
154 | look->vi=info; |
155 | |
156 | look->linearmap=_ogg_calloc(2,sizeof(*look->linearmap)); |
157 | |
158 | return look; |
159 | } |
160 | |
161 | static void *floor0_inverse1(vorbis_block *vb,vorbis_look_floor *i){ |
162 | vorbis_look_floor0 *look=(vorbis_look_floor0 *)i; |
163 | vorbis_info_floor0 *info=look->vi; |
164 | int j,k; |
165 | |
166 | int ampraw=oggpack_read(&vb->opb,info->ampbits); |
167 | if(ampraw>0){ /* also handles the -1 out of data case */ |
168 | long maxval=(1<<info->ampbits)-1; |
169 | float amp=(float)ampraw/maxval*info->ampdB; |
170 | int booknum=oggpack_read(&vb->opb,ov_ilog(info->numbooks)); |
171 | |
172 | if(booknum!=-1 && booknum<info->numbooks){ /* be paranoid */ |
173 | codec_setup_info *ci=vb->vd->vi->codec_setup; |
174 | codebook *b=ci->fullbooks+info->books[booknum]; |
175 | float last=0.f; |
176 | |
177 | /* the additional b->dim is a guard against any possible stack |
178 | smash; b->dim is provably more than we can overflow the |
179 | vector */ |
180 | float *lsp=_vorbis_block_alloc(vb,sizeof(*lsp)*(look->m+b->dim+1)); |
181 | |
182 | if(vorbis_book_decodev_set(b,lsp,&vb->opb,look->m)==-1)goto eop; |
183 | for(j=0;j<look->m;){ |
184 | for(k=0;j<look->m && k<b->dim;k++,j++)lsp[j]+=last; |
185 | last=lsp[j-1]; |
186 | } |
187 | |
188 | lsp[look->m]=amp; |
189 | return(lsp); |
190 | } |
191 | } |
192 | eop: |
193 | return(NULL); |
194 | } |
195 | |
196 | static int floor0_inverse2(vorbis_block *vb,vorbis_look_floor *i, |
197 | void *memo,float *out){ |
198 | vorbis_look_floor0 *look=(vorbis_look_floor0 *)i; |
199 | vorbis_info_floor0 *info=look->vi; |
200 | |
201 | floor0_map_lazy_init(vb,info,look); |
202 | |
203 | if(memo){ |
204 | float *lsp=(float *)memo; |
205 | float amp=lsp[look->m]; |
206 | |
207 | /* take the coefficients back to a spectral envelope curve */ |
208 | vorbis_lsp_to_curve(out, |
209 | look->linearmap[vb->W], |
210 | look->n[vb->W], |
211 | look->ln, |
212 | lsp,look->m,amp,(float)info->ampdB); |
213 | return(1); |
214 | } |
215 | memset(out,0,sizeof(*out)*look->n[vb->W]); |
216 | return(0); |
217 | } |
218 | |
219 | /* export hooks */ |
220 | const vorbis_func_floor floor0_exportbundle={ |
221 | NULL,&floor0_unpack,&floor0_look,&floor0_free_info, |
222 | &floor0_free_look,&floor0_inverse1,&floor0_inverse2 |
223 | }; |
224 | |