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-2010 * |
9 | * by the Xiph.Org Foundation https://xiph.org/ * |
10 | * * |
11 | ******************************************************************** |
12 | |
13 | function: channel mapping 0 implementation |
14 | |
15 | ********************************************************************/ |
16 | |
17 | #include <stdlib.h> |
18 | #include <stdio.h> |
19 | #include <string.h> |
20 | #include <math.h> |
21 | #include <ogg/ogg.h> |
22 | #include "vorbis/codec.h" |
23 | #include "codec_internal.h" |
24 | #include "codebook.h" |
25 | #include "window.h" |
26 | #include "registry.h" |
27 | #include "psy.h" |
28 | #include "misc.h" |
29 | |
30 | /* simplistic, wasteful way of doing this (unique lookup for each |
31 | mode/submapping); there should be a central repository for |
32 | identical lookups. That will require minor work, so I'm putting it |
33 | off as low priority. |
34 | |
35 | Why a lookup for each backend in a given mode? Because the |
36 | blocksize is set by the mode, and low backend lookups may require |
37 | parameters from other areas of the mode/mapping */ |
38 | |
39 | static void mapping0_free_info(vorbis_info_mapping *i){ |
40 | vorbis_info_mapping0 *info=(vorbis_info_mapping0 *)i; |
41 | if(info){ |
42 | memset(info,0,sizeof(*info)); |
43 | _ogg_free(info); |
44 | } |
45 | } |
46 | |
47 | static void mapping0_pack(vorbis_info *vi,vorbis_info_mapping *vm, |
48 | oggpack_buffer *opb){ |
49 | int i; |
50 | vorbis_info_mapping0 *info=(vorbis_info_mapping0 *)vm; |
51 | |
52 | /* another 'we meant to do it this way' hack... up to beta 4, we |
53 | packed 4 binary zeros here to signify one submapping in use. We |
54 | now redefine that to mean four bitflags that indicate use of |
55 | deeper features; bit0:submappings, bit1:coupling, |
56 | bit2,3:reserved. This is backward compatable with all actual uses |
57 | of the beta code. */ |
58 | |
59 | if(info->submaps>1){ |
60 | oggpack_write(opb,1,1); |
61 | oggpack_write(opb,info->submaps-1,4); |
62 | }else |
63 | oggpack_write(opb,0,1); |
64 | |
65 | if(info->coupling_steps>0){ |
66 | oggpack_write(opb,1,1); |
67 | oggpack_write(opb,info->coupling_steps-1,8); |
68 | |
69 | for(i=0;i<info->coupling_steps;i++){ |
70 | oggpack_write(opb,info->coupling_mag[i],ov_ilog(vi->channels-1)); |
71 | oggpack_write(opb,info->coupling_ang[i],ov_ilog(vi->channels-1)); |
72 | } |
73 | }else |
74 | oggpack_write(opb,0,1); |
75 | |
76 | oggpack_write(opb,0,2); /* 2,3:reserved */ |
77 | |
78 | /* we don't write the channel submappings if we only have one... */ |
79 | if(info->submaps>1){ |
80 | for(i=0;i<vi->channels;i++) |
81 | oggpack_write(opb,info->chmuxlist[i],4); |
82 | } |
83 | for(i=0;i<info->submaps;i++){ |
84 | oggpack_write(opb,0,8); /* time submap unused */ |
85 | oggpack_write(opb,info->floorsubmap[i],8); |
86 | oggpack_write(opb,info->residuesubmap[i],8); |
87 | } |
88 | } |
89 | |
90 | /* also responsible for range checking */ |
91 | static vorbis_info_mapping *mapping0_unpack(vorbis_info *vi,oggpack_buffer *opb){ |
92 | int i,b; |
93 | vorbis_info_mapping0 *info=_ogg_calloc(1,sizeof(*info)); |
94 | codec_setup_info *ci=vi->codec_setup; |
95 | if(vi->channels<=0)goto err_out; |
96 | |
97 | b=oggpack_read(opb,1); |
98 | if(b<0)goto err_out; |
99 | if(b){ |
100 | info->submaps=oggpack_read(opb,4)+1; |
101 | if(info->submaps<=0)goto err_out; |
102 | }else |
103 | info->submaps=1; |
104 | |
105 | b=oggpack_read(opb,1); |
106 | if(b<0)goto err_out; |
107 | if(b){ |
108 | info->coupling_steps=oggpack_read(opb,8)+1; |
109 | if(info->coupling_steps<=0)goto err_out; |
110 | for(i=0;i<info->coupling_steps;i++){ |
111 | /* vi->channels > 0 is enforced in the caller */ |
112 | int testM=info->coupling_mag[i]= |
113 | oggpack_read(opb,ov_ilog(vi->channels-1)); |
114 | int testA=info->coupling_ang[i]= |
115 | oggpack_read(opb,ov_ilog(vi->channels-1)); |
116 | |
117 | if(testM<0 || |
118 | testA<0 || |
119 | testM==testA || |
120 | testM>=vi->channels || |
121 | testA>=vi->channels) goto err_out; |
122 | } |
123 | |
124 | } |
125 | |
126 | if(oggpack_read(opb,2)!=0)goto err_out; /* 2,3:reserved */ |
127 | |
128 | if(info->submaps>1){ |
129 | for(i=0;i<vi->channels;i++){ |
130 | info->chmuxlist[i]=oggpack_read(opb,4); |
131 | if(info->chmuxlist[i]>=info->submaps || info->chmuxlist[i]<0)goto err_out; |
132 | } |
133 | } |
134 | for(i=0;i<info->submaps;i++){ |
135 | oggpack_read(opb,8); /* time submap unused */ |
136 | info->floorsubmap[i]=oggpack_read(opb,8); |
137 | if(info->floorsubmap[i]>=ci->floors || info->floorsubmap[i]<0)goto err_out; |
138 | info->residuesubmap[i]=oggpack_read(opb,8); |
139 | if(info->residuesubmap[i]>=ci->residues || info->residuesubmap[i]<0)goto err_out; |
140 | } |
141 | |
142 | return info; |
143 | |
144 | err_out: |
145 | mapping0_free_info(info); |
146 | return(NULL); |
147 | } |
148 | |
149 | #include "os.h" |
150 | #include "lpc.h" |
151 | #include "lsp.h" |
152 | #include "envelope.h" |
153 | #include "mdct.h" |
154 | #include "psy.h" |
155 | #include "scales.h" |
156 | |
157 | #if 0 |
158 | static long seq=0; |
159 | static ogg_int64_t total=0; |
160 | static float FLOOR1_fromdB_LOOKUP[256]={ |
161 | 1.0649863e-07F, 1.1341951e-07F, 1.2079015e-07F, 1.2863978e-07F, |
162 | 1.3699951e-07F, 1.4590251e-07F, 1.5538408e-07F, 1.6548181e-07F, |
163 | 1.7623575e-07F, 1.8768855e-07F, 1.9988561e-07F, 2.128753e-07F, |
164 | 2.2670913e-07F, 2.4144197e-07F, 2.5713223e-07F, 2.7384213e-07F, |
165 | 2.9163793e-07F, 3.1059021e-07F, 3.3077411e-07F, 3.5226968e-07F, |
166 | 3.7516214e-07F, 3.9954229e-07F, 4.2550680e-07F, 4.5315863e-07F, |
167 | 4.8260743e-07F, 5.1396998e-07F, 5.4737065e-07F, 5.8294187e-07F, |
168 | 6.2082472e-07F, 6.6116941e-07F, 7.0413592e-07F, 7.4989464e-07F, |
169 | 7.9862701e-07F, 8.5052630e-07F, 9.0579828e-07F, 9.6466216e-07F, |
170 | 1.0273513e-06F, 1.0941144e-06F, 1.1652161e-06F, 1.2409384e-06F, |
171 | 1.3215816e-06F, 1.4074654e-06F, 1.4989305e-06F, 1.5963394e-06F, |
172 | 1.7000785e-06F, 1.8105592e-06F, 1.9282195e-06F, 2.0535261e-06F, |
173 | 2.1869758e-06F, 2.3290978e-06F, 2.4804557e-06F, 2.6416497e-06F, |
174 | 2.8133190e-06F, 2.9961443e-06F, 3.1908506e-06F, 3.3982101e-06F, |
175 | 3.6190449e-06F, 3.8542308e-06F, 4.1047004e-06F, 4.3714470e-06F, |
176 | 4.6555282e-06F, 4.9580707e-06F, 5.2802740e-06F, 5.6234160e-06F, |
177 | 5.9888572e-06F, 6.3780469e-06F, 6.7925283e-06F, 7.2339451e-06F, |
178 | 7.7040476e-06F, 8.2047000e-06F, 8.7378876e-06F, 9.3057248e-06F, |
179 | 9.9104632e-06F, 1.0554501e-05F, 1.1240392e-05F, 1.1970856e-05F, |
180 | 1.2748789e-05F, 1.3577278e-05F, 1.4459606e-05F, 1.5399272e-05F, |
181 | 1.6400004e-05F, 1.7465768e-05F, 1.8600792e-05F, 1.9809576e-05F, |
182 | 2.1096914e-05F, 2.2467911e-05F, 2.3928002e-05F, 2.5482978e-05F, |
183 | 2.7139006e-05F, 2.8902651e-05F, 3.0780908e-05F, 3.2781225e-05F, |
184 | 3.4911534e-05F, 3.7180282e-05F, 3.9596466e-05F, 4.2169667e-05F, |
185 | 4.4910090e-05F, 4.7828601e-05F, 5.0936773e-05F, 5.4246931e-05F, |
186 | 5.7772202e-05F, 6.1526565e-05F, 6.5524908e-05F, 6.9783085e-05F, |
187 | 7.4317983e-05F, 7.9147585e-05F, 8.4291040e-05F, 8.9768747e-05F, |
188 | 9.5602426e-05F, 0.00010181521F, 0.00010843174F, 0.00011547824F, |
189 | 0.00012298267F, 0.00013097477F, 0.00013948625F, 0.00014855085F, |
190 | 0.00015820453F, 0.00016848555F, 0.00017943469F, 0.00019109536F, |
191 | 0.00020351382F, 0.00021673929F, 0.00023082423F, 0.00024582449F, |
192 | 0.00026179955F, 0.00027881276F, 0.00029693158F, 0.00031622787F, |
193 | 0.00033677814F, 0.00035866388F, 0.00038197188F, 0.00040679456F, |
194 | 0.00043323036F, 0.00046138411F, 0.00049136745F, 0.00052329927F, |
195 | 0.00055730621F, 0.00059352311F, 0.00063209358F, 0.00067317058F, |
196 | 0.00071691700F, 0.00076350630F, 0.00081312324F, 0.00086596457F, |
197 | 0.00092223983F, 0.00098217216F, 0.0010459992F, 0.0011139742F, |
198 | 0.0011863665F, 0.0012634633F, 0.0013455702F, 0.0014330129F, |
199 | 0.0015261382F, 0.0016253153F, 0.0017309374F, 0.0018434235F, |
200 | 0.0019632195F, 0.0020908006F, 0.0022266726F, 0.0023713743F, |
201 | 0.0025254795F, 0.0026895994F, 0.0028643847F, 0.0030505286F, |
202 | 0.0032487691F, 0.0034598925F, 0.0036847358F, 0.0039241906F, |
203 | 0.0041792066F, 0.0044507950F, 0.0047400328F, 0.0050480668F, |
204 | 0.0053761186F, 0.0057254891F, 0.0060975636F, 0.0064938176F, |
205 | 0.0069158225F, 0.0073652516F, 0.0078438871F, 0.0083536271F, |
206 | 0.0088964928F, 0.009474637F, 0.010090352F, 0.010746080F, |
207 | 0.011444421F, 0.012188144F, 0.012980198F, 0.013823725F, |
208 | 0.014722068F, 0.015678791F, 0.016697687F, 0.017782797F, |
209 | 0.018938423F, 0.020169149F, 0.021479854F, 0.022875735F, |
210 | 0.024362330F, 0.025945531F, 0.027631618F, 0.029427276F, |
211 | 0.031339626F, 0.033376252F, 0.035545228F, 0.037855157F, |
212 | 0.040315199F, 0.042935108F, 0.045725273F, 0.048696758F, |
213 | 0.051861348F, 0.055231591F, 0.058820850F, 0.062643361F, |
214 | 0.066714279F, 0.071049749F, 0.075666962F, 0.080584227F, |
215 | 0.085821044F, 0.091398179F, 0.097337747F, 0.10366330F, |
216 | 0.11039993F, 0.11757434F, 0.12521498F, 0.13335215F, |
217 | 0.14201813F, 0.15124727F, 0.16107617F, 0.17154380F, |
218 | 0.18269168F, 0.19456402F, 0.20720788F, 0.22067342F, |
219 | 0.23501402F, 0.25028656F, 0.26655159F, 0.28387361F, |
220 | 0.30232132F, 0.32196786F, 0.34289114F, 0.36517414F, |
221 | 0.38890521F, 0.41417847F, 0.44109412F, 0.46975890F, |
222 | 0.50028648F, 0.53279791F, 0.56742212F, 0.60429640F, |
223 | 0.64356699F, 0.68538959F, 0.72993007F, 0.77736504F, |
224 | 0.82788260F, 0.88168307F, 0.9389798F, 1.F, |
225 | }; |
226 | |
227 | #endif |
228 | |
229 | |
230 | static int mapping0_forward(vorbis_block *vb){ |
231 | vorbis_dsp_state *vd=vb->vd; |
232 | vorbis_info *vi=vd->vi; |
233 | codec_setup_info *ci=vi->codec_setup; |
234 | private_state *b=vb->vd->backend_state; |
235 | vorbis_block_internal *vbi=(vorbis_block_internal *)vb->internal; |
236 | int n=vb->pcmend; |
237 | int i,j,k; |
238 | |
239 | int *nonzero = alloca(sizeof(*nonzero)*vi->channels); |
240 | float **gmdct = _vorbis_block_alloc(vb,vi->channels*sizeof(*gmdct)); |
241 | int **iwork = _vorbis_block_alloc(vb,vi->channels*sizeof(*iwork)); |
242 | int ***floor_posts = _vorbis_block_alloc(vb,vi->channels*sizeof(*floor_posts)); |
243 | |
244 | float global_ampmax=vbi->ampmax; |
245 | float *local_ampmax=alloca(sizeof(*local_ampmax)*vi->channels); |
246 | int blocktype=vbi->blocktype; |
247 | |
248 | int modenumber=vb->W; |
249 | vorbis_info_mapping0 *info=ci->map_param[modenumber]; |
250 | vorbis_look_psy *psy_look=b->psy+blocktype+(vb->W?2:0); |
251 | |
252 | vb->mode=modenumber; |
253 | |
254 | for(i=0;i<vi->channels;i++){ |
255 | float scale=4.f/n; |
256 | float scale_dB; |
257 | |
258 | float *pcm =vb->pcm[i]; |
259 | float *logfft =pcm; |
260 | |
261 | iwork[i]=_vorbis_block_alloc(vb,n/2*sizeof(**iwork)); |
262 | gmdct[i]=_vorbis_block_alloc(vb,n/2*sizeof(**gmdct)); |
263 | |
264 | scale_dB=todB(&scale) + .345; /* + .345 is a hack; the original |
265 | todB estimation used on IEEE 754 |
266 | compliant machines had a bug that |
267 | returned dB values about a third |
268 | of a decibel too high. The bug |
269 | was harmless because tunings |
270 | implicitly took that into |
271 | account. However, fixing the bug |
272 | in the estimator requires |
273 | changing all the tunings as well. |
274 | For now, it's easier to sync |
275 | things back up here, and |
276 | recalibrate the tunings in the |
277 | next major model upgrade. */ |
278 | |
279 | #if 0 |
280 | if(vi->channels==2){ |
281 | if(i==0) |
282 | _analysis_output("pcmL" ,seq,pcm,n,0,0,total-n/2); |
283 | else |
284 | _analysis_output("pcmR" ,seq,pcm,n,0,0,total-n/2); |
285 | }else{ |
286 | _analysis_output("pcm" ,seq,pcm,n,0,0,total-n/2); |
287 | } |
288 | #endif |
289 | |
290 | /* window the PCM data */ |
291 | _vorbis_apply_window(pcm,b->window,ci->blocksizes,vb->lW,vb->W,vb->nW); |
292 | |
293 | #if 0 |
294 | if(vi->channels==2){ |
295 | if(i==0) |
296 | _analysis_output("windowedL" ,seq,pcm,n,0,0,total-n/2); |
297 | else |
298 | _analysis_output("windowedR" ,seq,pcm,n,0,0,total-n/2); |
299 | }else{ |
300 | _analysis_output("windowed" ,seq,pcm,n,0,0,total-n/2); |
301 | } |
302 | #endif |
303 | |
304 | /* transform the PCM data */ |
305 | /* only MDCT right now.... */ |
306 | mdct_forward(b->transform[vb->W][0],pcm,gmdct[i]); |
307 | |
308 | /* FFT yields more accurate tonal estimation (not phase sensitive) */ |
309 | drft_forward(&b->fft_look[vb->W],pcm); |
310 | logfft[0]=scale_dB+todB(pcm) + .345; /* + .345 is a hack; the |
311 | original todB estimation used on |
312 | IEEE 754 compliant machines had a |
313 | bug that returned dB values about |
314 | a third of a decibel too high. |
315 | The bug was harmless because |
316 | tunings implicitly took that into |
317 | account. However, fixing the bug |
318 | in the estimator requires |
319 | changing all the tunings as well. |
320 | For now, it's easier to sync |
321 | things back up here, and |
322 | recalibrate the tunings in the |
323 | next major model upgrade. */ |
324 | local_ampmax[i]=logfft[0]; |
325 | for(j=1;j<n-1;j+=2){ |
326 | float temp=pcm[j]*pcm[j]+pcm[j+1]*pcm[j+1]; |
327 | temp=logfft[(j+1)>>1]=scale_dB+.5f*todB(&temp) + .345; /* + |
328 | .345 is a hack; the original todB |
329 | estimation used on IEEE 754 |
330 | compliant machines had a bug that |
331 | returned dB values about a third |
332 | of a decibel too high. The bug |
333 | was harmless because tunings |
334 | implicitly took that into |
335 | account. However, fixing the bug |
336 | in the estimator requires |
337 | changing all the tunings as well. |
338 | For now, it's easier to sync |
339 | things back up here, and |
340 | recalibrate the tunings in the |
341 | next major model upgrade. */ |
342 | if(temp>local_ampmax[i])local_ampmax[i]=temp; |
343 | } |
344 | |
345 | if(local_ampmax[i]>0.f)local_ampmax[i]=0.f; |
346 | if(local_ampmax[i]>global_ampmax)global_ampmax=local_ampmax[i]; |
347 | |
348 | #if 0 |
349 | if(vi->channels==2){ |
350 | if(i==0){ |
351 | _analysis_output("fftL" ,seq,logfft,n/2,1,0,0); |
352 | }else{ |
353 | _analysis_output("fftR" ,seq,logfft,n/2,1,0,0); |
354 | } |
355 | }else{ |
356 | _analysis_output("fft" ,seq,logfft,n/2,1,0,0); |
357 | } |
358 | #endif |
359 | |
360 | } |
361 | |
362 | { |
363 | float *noise = _vorbis_block_alloc(vb,n/2*sizeof(*noise)); |
364 | float *tone = _vorbis_block_alloc(vb,n/2*sizeof(*tone)); |
365 | |
366 | for(i=0;i<vi->channels;i++){ |
367 | /* the encoder setup assumes that all the modes used by any |
368 | specific bitrate tweaking use the same floor */ |
369 | |
370 | int submap=info->chmuxlist[i]; |
371 | |
372 | /* the following makes things clearer to *me* anyway */ |
373 | float *mdct =gmdct[i]; |
374 | float *logfft =vb->pcm[i]; |
375 | |
376 | float *logmdct =logfft+n/2; |
377 | float *logmask =logfft; |
378 | |
379 | vb->mode=modenumber; |
380 | |
381 | floor_posts[i]=_vorbis_block_alloc(vb,PACKETBLOBS*sizeof(**floor_posts)); |
382 | memset(floor_posts[i],0,sizeof(**floor_posts)*PACKETBLOBS); |
383 | |
384 | for(j=0;j<n/2;j++) |
385 | logmdct[j]=todB(mdct+j) + .345; /* + .345 is a hack; the original |
386 | todB estimation used on IEEE 754 |
387 | compliant machines had a bug that |
388 | returned dB values about a third |
389 | of a decibel too high. The bug |
390 | was harmless because tunings |
391 | implicitly took that into |
392 | account. However, fixing the bug |
393 | in the estimator requires |
394 | changing all the tunings as well. |
395 | For now, it's easier to sync |
396 | things back up here, and |
397 | recalibrate the tunings in the |
398 | next major model upgrade. */ |
399 | |
400 | #if 0 |
401 | if(vi->channels==2){ |
402 | if(i==0) |
403 | _analysis_output("mdctL" ,seq,logmdct,n/2,1,0,0); |
404 | else |
405 | _analysis_output("mdctR" ,seq,logmdct,n/2,1,0,0); |
406 | }else{ |
407 | _analysis_output("mdct" ,seq,logmdct,n/2,1,0,0); |
408 | } |
409 | #endif |
410 | |
411 | /* first step; noise masking. Not only does 'noise masking' |
412 | give us curves from which we can decide how much resolution |
413 | to give noise parts of the spectrum, it also implicitly hands |
414 | us a tonality estimate (the larger the value in the |
415 | 'noise_depth' vector, the more tonal that area is) */ |
416 | |
417 | _vp_noisemask(psy_look, |
418 | logmdct, |
419 | noise); /* noise does not have by-frequency offset |
420 | bias applied yet */ |
421 | #if 0 |
422 | if(vi->channels==2){ |
423 | if(i==0) |
424 | _analysis_output("noiseL" ,seq,noise,n/2,1,0,0); |
425 | else |
426 | _analysis_output("noiseR" ,seq,noise,n/2,1,0,0); |
427 | }else{ |
428 | _analysis_output("noise" ,seq,noise,n/2,1,0,0); |
429 | } |
430 | #endif |
431 | |
432 | /* second step: 'all the other crap'; all the stuff that isn't |
433 | computed/fit for bitrate management goes in the second psy |
434 | vector. This includes tone masking, peak limiting and ATH */ |
435 | |
436 | _vp_tonemask(psy_look, |
437 | logfft, |
438 | tone, |
439 | global_ampmax, |
440 | local_ampmax[i]); |
441 | |
442 | #if 0 |
443 | if(vi->channels==2){ |
444 | if(i==0) |
445 | _analysis_output("toneL" ,seq,tone,n/2,1,0,0); |
446 | else |
447 | _analysis_output("toneR" ,seq,tone,n/2,1,0,0); |
448 | }else{ |
449 | _analysis_output("tone" ,seq,tone,n/2,1,0,0); |
450 | } |
451 | #endif |
452 | |
453 | /* third step; we offset the noise vectors, overlay tone |
454 | masking. We then do a floor1-specific line fit. If we're |
455 | performing bitrate management, the line fit is performed |
456 | multiple times for up/down tweakage on demand. */ |
457 | |
458 | #if 0 |
459 | { |
460 | float aotuv[psy_look->n]; |
461 | #endif |
462 | |
463 | _vp_offset_and_mix(psy_look, |
464 | noise, |
465 | tone, |
466 | 1, |
467 | logmask, |
468 | mdct, |
469 | logmdct); |
470 | |
471 | #if 0 |
472 | if(vi->channels==2){ |
473 | if(i==0) |
474 | _analysis_output("aotuvM1_L" ,seq,aotuv,psy_look->n,1,1,0); |
475 | else |
476 | _analysis_output("aotuvM1_R" ,seq,aotuv,psy_look->n,1,1,0); |
477 | }else{ |
478 | _analysis_output("aotuvM1" ,seq,aotuv,psy_look->n,1,1,0); |
479 | } |
480 | } |
481 | #endif |
482 | |
483 | |
484 | #if 0 |
485 | if(vi->channels==2){ |
486 | if(i==0) |
487 | _analysis_output("mask1L" ,seq,logmask,n/2,1,0,0); |
488 | else |
489 | _analysis_output("mask1R" ,seq,logmask,n/2,1,0,0); |
490 | }else{ |
491 | _analysis_output("mask1" ,seq,logmask,n/2,1,0,0); |
492 | } |
493 | #endif |
494 | |
495 | /* this algorithm is hardwired to floor 1 for now; abort out if |
496 | we're *not* floor1. This won't happen unless someone has |
497 | broken the encode setup lib. Guard it anyway. */ |
498 | if(ci->floor_type[info->floorsubmap[submap]]!=1)return(-1); |
499 | |
500 | floor_posts[i][PACKETBLOBS/2]= |
501 | floor1_fit(vb,b->flr[info->floorsubmap[submap]], |
502 | logmdct, |
503 | logmask); |
504 | |
505 | /* are we managing bitrate? If so, perform two more fits for |
506 | later rate tweaking (fits represent hi/lo) */ |
507 | if(vorbis_bitrate_managed(vb) && floor_posts[i][PACKETBLOBS/2]){ |
508 | /* higher rate by way of lower noise curve */ |
509 | |
510 | _vp_offset_and_mix(psy_look, |
511 | noise, |
512 | tone, |
513 | 2, |
514 | logmask, |
515 | mdct, |
516 | logmdct); |
517 | |
518 | #if 0 |
519 | if(vi->channels==2){ |
520 | if(i==0) |
521 | _analysis_output("mask2L" ,seq,logmask,n/2,1,0,0); |
522 | else |
523 | _analysis_output("mask2R" ,seq,logmask,n/2,1,0,0); |
524 | }else{ |
525 | _analysis_output("mask2" ,seq,logmask,n/2,1,0,0); |
526 | } |
527 | #endif |
528 | |
529 | floor_posts[i][PACKETBLOBS-1]= |
530 | floor1_fit(vb,b->flr[info->floorsubmap[submap]], |
531 | logmdct, |
532 | logmask); |
533 | |
534 | /* lower rate by way of higher noise curve */ |
535 | _vp_offset_and_mix(psy_look, |
536 | noise, |
537 | tone, |
538 | 0, |
539 | logmask, |
540 | mdct, |
541 | logmdct); |
542 | |
543 | #if 0 |
544 | if(vi->channels==2){ |
545 | if(i==0) |
546 | _analysis_output("mask0L" ,seq,logmask,n/2,1,0,0); |
547 | else |
548 | _analysis_output("mask0R" ,seq,logmask,n/2,1,0,0); |
549 | }else{ |
550 | _analysis_output("mask0" ,seq,logmask,n/2,1,0,0); |
551 | } |
552 | #endif |
553 | |
554 | floor_posts[i][0]= |
555 | floor1_fit(vb,b->flr[info->floorsubmap[submap]], |
556 | logmdct, |
557 | logmask); |
558 | |
559 | /* we also interpolate a range of intermediate curves for |
560 | intermediate rates */ |
561 | for(k=1;k<PACKETBLOBS/2;k++) |
562 | floor_posts[i][k]= |
563 | floor1_interpolate_fit(vb,b->flr[info->floorsubmap[submap]], |
564 | floor_posts[i][0], |
565 | floor_posts[i][PACKETBLOBS/2], |
566 | k*65536/(PACKETBLOBS/2)); |
567 | for(k=PACKETBLOBS/2+1;k<PACKETBLOBS-1;k++) |
568 | floor_posts[i][k]= |
569 | floor1_interpolate_fit(vb,b->flr[info->floorsubmap[submap]], |
570 | floor_posts[i][PACKETBLOBS/2], |
571 | floor_posts[i][PACKETBLOBS-1], |
572 | (k-PACKETBLOBS/2)*65536/(PACKETBLOBS/2)); |
573 | } |
574 | } |
575 | } |
576 | vbi->ampmax=global_ampmax; |
577 | |
578 | /* |
579 | the next phases are performed once for vbr-only and PACKETBLOB |
580 | times for bitrate managed modes. |
581 | |
582 | 1) encode actual mode being used |
583 | 2) encode the floor for each channel, compute coded mask curve/res |
584 | 3) normalize and couple. |
585 | 4) encode residue |
586 | 5) save packet bytes to the packetblob vector |
587 | |
588 | */ |
589 | |
590 | /* iterate over the many masking curve fits we've created */ |
591 | |
592 | { |
593 | int **couple_bundle=alloca(sizeof(*couple_bundle)*vi->channels); |
594 | int *zerobundle=alloca(sizeof(*zerobundle)*vi->channels); |
595 | |
596 | for(k=(vorbis_bitrate_managed(vb)?0:PACKETBLOBS/2); |
597 | k<=(vorbis_bitrate_managed(vb)?PACKETBLOBS-1:PACKETBLOBS/2); |
598 | k++){ |
599 | oggpack_buffer *opb=vbi->packetblob[k]; |
600 | |
601 | /* start out our new packet blob with packet type and mode */ |
602 | /* Encode the packet type */ |
603 | oggpack_write(opb,0,1); |
604 | /* Encode the modenumber */ |
605 | /* Encode frame mode, pre,post windowsize, then dispatch */ |
606 | oggpack_write(opb,modenumber,b->modebits); |
607 | if(vb->W){ |
608 | oggpack_write(opb,vb->lW,1); |
609 | oggpack_write(opb,vb->nW,1); |
610 | } |
611 | |
612 | /* encode floor, compute masking curve, sep out residue */ |
613 | for(i=0;i<vi->channels;i++){ |
614 | int submap=info->chmuxlist[i]; |
615 | int *ilogmask=iwork[i]; |
616 | |
617 | nonzero[i]=floor1_encode(opb,vb,b->flr[info->floorsubmap[submap]], |
618 | floor_posts[i][k], |
619 | ilogmask); |
620 | #if 0 |
621 | { |
622 | char buf[80]; |
623 | sprintf(buf,"maskI%c%d" ,i?'R':'L',k); |
624 | float work[n/2]; |
625 | for(j=0;j<n/2;j++) |
626 | work[j]=FLOOR1_fromdB_LOOKUP[iwork[i][j]]; |
627 | _analysis_output(buf,seq,work,n/2,1,1,0); |
628 | } |
629 | #endif |
630 | } |
631 | |
632 | /* our iteration is now based on masking curve, not prequant and |
633 | coupling. Only one prequant/coupling step */ |
634 | |
635 | /* quantize/couple */ |
636 | /* incomplete implementation that assumes the tree is all depth |
637 | one, or no tree at all */ |
638 | _vp_couple_quantize_normalize(k, |
639 | &ci->psy_g_param, |
640 | psy_look, |
641 | info, |
642 | gmdct, |
643 | iwork, |
644 | nonzero, |
645 | ci->psy_g_param.sliding_lowpass[vb->W][k], |
646 | vi->channels); |
647 | |
648 | #if 0 |
649 | for(i=0;i<vi->channels;i++){ |
650 | char buf[80]; |
651 | sprintf(buf,"res%c%d" ,i?'R':'L',k); |
652 | float work[n/2]; |
653 | for(j=0;j<n/2;j++) |
654 | work[j]=iwork[i][j]; |
655 | _analysis_output(buf,seq,work,n/2,1,0,0); |
656 | } |
657 | #endif |
658 | |
659 | /* classify and encode by submap */ |
660 | for(i=0;i<info->submaps;i++){ |
661 | int ch_in_bundle=0; |
662 | long **classifications; |
663 | int resnum=info->residuesubmap[i]; |
664 | |
665 | for(j=0;j<vi->channels;j++){ |
666 | if(info->chmuxlist[j]==i){ |
667 | zerobundle[ch_in_bundle]=0; |
668 | if(nonzero[j])zerobundle[ch_in_bundle]=1; |
669 | couple_bundle[ch_in_bundle++]=iwork[j]; |
670 | } |
671 | } |
672 | |
673 | classifications=_residue_P[ci->residue_type[resnum]]-> |
674 | class(vb,b->residue[resnum],couple_bundle,zerobundle,ch_in_bundle); |
675 | |
676 | ch_in_bundle=0; |
677 | for(j=0;j<vi->channels;j++) |
678 | if(info->chmuxlist[j]==i) |
679 | couple_bundle[ch_in_bundle++]=iwork[j]; |
680 | |
681 | _residue_P[ci->residue_type[resnum]]-> |
682 | forward(opb,vb,b->residue[resnum], |
683 | couple_bundle,zerobundle,ch_in_bundle,classifications,i); |
684 | } |
685 | |
686 | /* ok, done encoding. Next protopacket. */ |
687 | } |
688 | |
689 | } |
690 | |
691 | #if 0 |
692 | seq++; |
693 | total+=ci->blocksizes[vb->W]/4+ci->blocksizes[vb->nW]/4; |
694 | #endif |
695 | return(0); |
696 | } |
697 | |
698 | static int mapping0_inverse(vorbis_block *vb,vorbis_info_mapping *l){ |
699 | vorbis_dsp_state *vd=vb->vd; |
700 | vorbis_info *vi=vd->vi; |
701 | codec_setup_info *ci=vi->codec_setup; |
702 | private_state *b=vd->backend_state; |
703 | vorbis_info_mapping0 *info=(vorbis_info_mapping0 *)l; |
704 | |
705 | int i,j; |
706 | long n=vb->pcmend=ci->blocksizes[vb->W]; |
707 | |
708 | float **pcmbundle=alloca(sizeof(*pcmbundle)*vi->channels); |
709 | int *zerobundle=alloca(sizeof(*zerobundle)*vi->channels); |
710 | |
711 | int *nonzero =alloca(sizeof(*nonzero)*vi->channels); |
712 | void **floormemo=alloca(sizeof(*floormemo)*vi->channels); |
713 | |
714 | /* recover the spectral envelope; store it in the PCM vector for now */ |
715 | for(i=0;i<vi->channels;i++){ |
716 | int submap=info->chmuxlist[i]; |
717 | floormemo[i]=_floor_P[ci->floor_type[info->floorsubmap[submap]]]-> |
718 | inverse1(vb,b->flr[info->floorsubmap[submap]]); |
719 | if(floormemo[i]) |
720 | nonzero[i]=1; |
721 | else |
722 | nonzero[i]=0; |
723 | memset(vb->pcm[i],0,sizeof(*vb->pcm[i])*n/2); |
724 | } |
725 | |
726 | /* channel coupling can 'dirty' the nonzero listing */ |
727 | for(i=0;i<info->coupling_steps;i++){ |
728 | if(nonzero[info->coupling_mag[i]] || |
729 | nonzero[info->coupling_ang[i]]){ |
730 | nonzero[info->coupling_mag[i]]=1; |
731 | nonzero[info->coupling_ang[i]]=1; |
732 | } |
733 | } |
734 | |
735 | /* recover the residue into our working vectors */ |
736 | for(i=0;i<info->submaps;i++){ |
737 | int ch_in_bundle=0; |
738 | for(j=0;j<vi->channels;j++){ |
739 | if(info->chmuxlist[j]==i){ |
740 | if(nonzero[j]) |
741 | zerobundle[ch_in_bundle]=1; |
742 | else |
743 | zerobundle[ch_in_bundle]=0; |
744 | pcmbundle[ch_in_bundle++]=vb->pcm[j]; |
745 | } |
746 | } |
747 | |
748 | _residue_P[ci->residue_type[info->residuesubmap[i]]]-> |
749 | inverse(vb,b->residue[info->residuesubmap[i]], |
750 | pcmbundle,zerobundle,ch_in_bundle); |
751 | } |
752 | |
753 | /* channel coupling */ |
754 | for(i=info->coupling_steps-1;i>=0;i--){ |
755 | float *pcmM=vb->pcm[info->coupling_mag[i]]; |
756 | float *pcmA=vb->pcm[info->coupling_ang[i]]; |
757 | |
758 | for(j=0;j<n/2;j++){ |
759 | float mag=pcmM[j]; |
760 | float ang=pcmA[j]; |
761 | |
762 | if(mag>0) |
763 | if(ang>0){ |
764 | pcmM[j]=mag; |
765 | pcmA[j]=mag-ang; |
766 | }else{ |
767 | pcmA[j]=mag; |
768 | pcmM[j]=mag+ang; |
769 | } |
770 | else |
771 | if(ang>0){ |
772 | pcmM[j]=mag; |
773 | pcmA[j]=mag+ang; |
774 | }else{ |
775 | pcmA[j]=mag; |
776 | pcmM[j]=mag-ang; |
777 | } |
778 | } |
779 | } |
780 | |
781 | /* compute and apply spectral envelope */ |
782 | for(i=0;i<vi->channels;i++){ |
783 | float *pcm=vb->pcm[i]; |
784 | int submap=info->chmuxlist[i]; |
785 | _floor_P[ci->floor_type[info->floorsubmap[submap]]]-> |
786 | inverse2(vb,b->flr[info->floorsubmap[submap]], |
787 | floormemo[i],pcm); |
788 | } |
789 | |
790 | /* transform the PCM data; takes PCM vector, vb; modifies PCM vector */ |
791 | /* only MDCT right now.... */ |
792 | for(i=0;i<vi->channels;i++){ |
793 | float *pcm=vb->pcm[i]; |
794 | mdct_backward(b->transform[vb->W][0],pcm,pcm); |
795 | } |
796 | |
797 | /* all done! */ |
798 | return(0); |
799 | } |
800 | |
801 | /* export hooks */ |
802 | const vorbis_func_mapping mapping0_exportbundle={ |
803 | &mapping0_pack, |
804 | &mapping0_unpack, |
805 | &mapping0_free_info, |
806 | &mapping0_forward, |
807 | &mapping0_inverse |
808 | }; |
809 | |