/src/vorbis/lib/mapping0.c
Line | Count | Source (jump to first uncovered line) |
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 | 0 | static void mapping0_free_info(vorbis_info_mapping *i){ |
40 | 0 | vorbis_info_mapping0 *info=(vorbis_info_mapping0 *)i; |
41 | 0 | if(info){ |
42 | 0 | memset(info,0,sizeof(*info)); |
43 | 0 | _ogg_free(info); |
44 | 0 | } |
45 | 0 | } |
46 | | |
47 | | static void mapping0_pack(vorbis_info *vi,vorbis_info_mapping *vm, |
48 | 0 | oggpack_buffer *opb){ |
49 | 0 | int i; |
50 | 0 | 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 | 0 | if(info->submaps>1){ |
60 | 0 | oggpack_write(opb,1,1); |
61 | 0 | oggpack_write(opb,info->submaps-1,4); |
62 | 0 | }else |
63 | 0 | oggpack_write(opb,0,1); |
64 | |
|
65 | 0 | if(info->coupling_steps>0){ |
66 | 0 | oggpack_write(opb,1,1); |
67 | 0 | oggpack_write(opb,info->coupling_steps-1,8); |
68 | |
|
69 | 0 | for(i=0;i<info->coupling_steps;i++){ |
70 | 0 | oggpack_write(opb,info->coupling_mag[i],ov_ilog(vi->channels-1)); |
71 | 0 | oggpack_write(opb,info->coupling_ang[i],ov_ilog(vi->channels-1)); |
72 | 0 | } |
73 | 0 | }else |
74 | 0 | oggpack_write(opb,0,1); |
75 | |
|
76 | 0 | oggpack_write(opb,0,2); /* 2,3:reserved */ |
77 | | |
78 | | /* we don't write the channel submappings if we only have one... */ |
79 | 0 | if(info->submaps>1){ |
80 | 0 | for(i=0;i<vi->channels;i++) |
81 | 0 | oggpack_write(opb,info->chmuxlist[i],4); |
82 | 0 | } |
83 | 0 | for(i=0;i<info->submaps;i++){ |
84 | 0 | oggpack_write(opb,0,8); /* time submap unused */ |
85 | 0 | oggpack_write(opb,info->floorsubmap[i],8); |
86 | 0 | oggpack_write(opb,info->residuesubmap[i],8); |
87 | 0 | } |
88 | 0 | } |
89 | | |
90 | | /* also responsible for range checking */ |
91 | 0 | static vorbis_info_mapping *mapping0_unpack(vorbis_info *vi,oggpack_buffer *opb){ |
92 | 0 | int i,b; |
93 | 0 | vorbis_info_mapping0 *info=_ogg_calloc(1,sizeof(*info)); |
94 | 0 | codec_setup_info *ci=vi->codec_setup; |
95 | 0 | if(vi->channels<=0)goto err_out; |
96 | | |
97 | 0 | b=oggpack_read(opb,1); |
98 | 0 | if(b<0)goto err_out; |
99 | 0 | if(b){ |
100 | 0 | info->submaps=oggpack_read(opb,4)+1; |
101 | 0 | if(info->submaps<=0)goto err_out; |
102 | 0 | }else |
103 | 0 | info->submaps=1; |
104 | | |
105 | 0 | b=oggpack_read(opb,1); |
106 | 0 | if(b<0)goto err_out; |
107 | 0 | if(b){ |
108 | 0 | info->coupling_steps=oggpack_read(opb,8)+1; |
109 | 0 | if(info->coupling_steps<=0)goto err_out; |
110 | 0 | for(i=0;i<info->coupling_steps;i++){ |
111 | | /* vi->channels > 0 is enforced in the caller */ |
112 | 0 | int testM=info->coupling_mag[i]= |
113 | 0 | oggpack_read(opb,ov_ilog(vi->channels-1)); |
114 | 0 | int testA=info->coupling_ang[i]= |
115 | 0 | oggpack_read(opb,ov_ilog(vi->channels-1)); |
116 | |
|
117 | 0 | if(testM<0 || |
118 | 0 | testA<0 || |
119 | 0 | testM==testA || |
120 | 0 | testM>=vi->channels || |
121 | 0 | testA>=vi->channels) goto err_out; |
122 | 0 | } |
123 | |
|
124 | 0 | } |
125 | | |
126 | 0 | if(oggpack_read(opb,2)!=0)goto err_out; /* 2,3:reserved */ |
127 | | |
128 | 0 | if(info->submaps>1){ |
129 | 0 | for(i=0;i<vi->channels;i++){ |
130 | 0 | info->chmuxlist[i]=oggpack_read(opb,4); |
131 | 0 | if(info->chmuxlist[i]>=info->submaps || info->chmuxlist[i]<0)goto err_out; |
132 | 0 | } |
133 | 0 | } |
134 | 0 | for(i=0;i<info->submaps;i++){ |
135 | 0 | oggpack_read(opb,8); /* time submap unused */ |
136 | 0 | info->floorsubmap[i]=oggpack_read(opb,8); |
137 | 0 | if(info->floorsubmap[i]>=ci->floors || info->floorsubmap[i]<0)goto err_out; |
138 | 0 | info->residuesubmap[i]=oggpack_read(opb,8); |
139 | 0 | if(info->residuesubmap[i]>=ci->residues || info->residuesubmap[i]<0)goto err_out; |
140 | 0 | } |
141 | | |
142 | 0 | return info; |
143 | | |
144 | 0 | err_out: |
145 | 0 | mapping0_free_info(info); |
146 | 0 | return(NULL); |
147 | 0 | } |
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 | 0 | static int mapping0_forward(vorbis_block *vb){ |
231 | 0 | vorbis_dsp_state *vd=vb->vd; |
232 | 0 | vorbis_info *vi=vd->vi; |
233 | 0 | codec_setup_info *ci=vi->codec_setup; |
234 | 0 | private_state *b=vb->vd->backend_state; |
235 | 0 | vorbis_block_internal *vbi=(vorbis_block_internal *)vb->internal; |
236 | 0 | int n=vb->pcmend; |
237 | 0 | int i,j,k; |
238 | |
|
239 | 0 | int *nonzero = alloca(sizeof(*nonzero)*vi->channels); |
240 | 0 | float **gmdct = _vorbis_block_alloc(vb,vi->channels*sizeof(*gmdct)); |
241 | 0 | int **iwork = _vorbis_block_alloc(vb,vi->channels*sizeof(*iwork)); |
242 | 0 | int ***floor_posts = _vorbis_block_alloc(vb,vi->channels*sizeof(*floor_posts)); |
243 | |
|
244 | 0 | float global_ampmax=vbi->ampmax; |
245 | 0 | float *local_ampmax=alloca(sizeof(*local_ampmax)*vi->channels); |
246 | 0 | int blocktype=vbi->blocktype; |
247 | |
|
248 | 0 | int modenumber=vb->W; |
249 | 0 | vorbis_info_mapping0 *info=ci->map_param[modenumber]; |
250 | 0 | vorbis_look_psy *psy_look=b->psy+blocktype+(vb->W?2:0); |
251 | |
|
252 | 0 | vb->mode=modenumber; |
253 | |
|
254 | 0 | for(i=0;i<vi->channels;i++){ |
255 | 0 | float scale=4.f/n; |
256 | 0 | float scale_dB; |
257 | |
|
258 | 0 | float *pcm =vb->pcm[i]; |
259 | 0 | float *logfft =pcm; |
260 | |
|
261 | 0 | iwork[i]=_vorbis_block_alloc(vb,n/2*sizeof(**iwork)); |
262 | 0 | gmdct[i]=_vorbis_block_alloc(vb,n/2*sizeof(**gmdct)); |
263 | |
|
264 | 0 | 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 | 0 | _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 | 0 | mdct_forward(b->transform[vb->W][0],pcm,gmdct[i]); |
307 | | |
308 | | /* FFT yields more accurate tonal estimation (not phase sensitive) */ |
309 | 0 | drft_forward(&b->fft_look[vb->W],pcm); |
310 | 0 | 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 | 0 | local_ampmax[i]=logfft[0]; |
325 | 0 | for(j=1;j<n-1;j+=2){ |
326 | 0 | float temp=pcm[j]*pcm[j]+pcm[j+1]*pcm[j+1]; |
327 | 0 | 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 | 0 | if(temp>local_ampmax[i])local_ampmax[i]=temp; |
343 | 0 | } |
344 | |
|
345 | 0 | if(local_ampmax[i]>0.f)local_ampmax[i]=0.f; |
346 | 0 | 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 | 0 | } |
361 | |
|
362 | 0 | { |
363 | 0 | float *noise = _vorbis_block_alloc(vb,n/2*sizeof(*noise)); |
364 | 0 | float *tone = _vorbis_block_alloc(vb,n/2*sizeof(*tone)); |
365 | |
|
366 | 0 | 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 | 0 | int submap=info->chmuxlist[i]; |
371 | | |
372 | | /* the following makes things clearer to *me* anyway */ |
373 | 0 | float *mdct =gmdct[i]; |
374 | 0 | float *logfft =vb->pcm[i]; |
375 | |
|
376 | 0 | float *logmdct =logfft+n/2; |
377 | 0 | float *logmask =logfft; |
378 | |
|
379 | 0 | vb->mode=modenumber; |
380 | |
|
381 | 0 | floor_posts[i]=_vorbis_block_alloc(vb,PACKETBLOBS*sizeof(**floor_posts)); |
382 | 0 | memset(floor_posts[i],0,sizeof(**floor_posts)*PACKETBLOBS); |
383 | |
|
384 | 0 | for(j=0;j<n/2;j++) |
385 | 0 | 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 | 0 | _vp_noisemask(psy_look, |
418 | 0 | logmdct, |
419 | 0 | 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 | 0 | _vp_tonemask(psy_look, |
437 | 0 | logfft, |
438 | 0 | tone, |
439 | 0 | global_ampmax, |
440 | 0 | 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 | 0 | _vp_offset_and_mix(psy_look, |
464 | 0 | noise, |
465 | 0 | tone, |
466 | 0 | 1, |
467 | 0 | logmask, |
468 | 0 | mdct, |
469 | 0 | 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 | 0 | if(ci->floor_type[info->floorsubmap[submap]]!=1)return(-1); |
499 | | |
500 | 0 | floor_posts[i][PACKETBLOBS/2]= |
501 | 0 | floor1_fit(vb,b->flr[info->floorsubmap[submap]], |
502 | 0 | logmdct, |
503 | 0 | logmask); |
504 | | |
505 | | /* are we managing bitrate? If so, perform two more fits for |
506 | | later rate tweaking (fits represent hi/lo) */ |
507 | 0 | if(vorbis_bitrate_managed(vb) && floor_posts[i][PACKETBLOBS/2]){ |
508 | | /* higher rate by way of lower noise curve */ |
509 | |
|
510 | 0 | _vp_offset_and_mix(psy_look, |
511 | 0 | noise, |
512 | 0 | tone, |
513 | 0 | 2, |
514 | 0 | logmask, |
515 | 0 | mdct, |
516 | 0 | 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 | 0 | floor_posts[i][PACKETBLOBS-1]= |
530 | 0 | floor1_fit(vb,b->flr[info->floorsubmap[submap]], |
531 | 0 | logmdct, |
532 | 0 | logmask); |
533 | | |
534 | | /* lower rate by way of higher noise curve */ |
535 | 0 | _vp_offset_and_mix(psy_look, |
536 | 0 | noise, |
537 | 0 | tone, |
538 | 0 | 0, |
539 | 0 | logmask, |
540 | 0 | mdct, |
541 | 0 | 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 | 0 | floor_posts[i][0]= |
555 | 0 | floor1_fit(vb,b->flr[info->floorsubmap[submap]], |
556 | 0 | logmdct, |
557 | 0 | logmask); |
558 | | |
559 | | /* we also interpolate a range of intermediate curves for |
560 | | intermediate rates */ |
561 | 0 | for(k=1;k<PACKETBLOBS/2;k++) |
562 | 0 | floor_posts[i][k]= |
563 | 0 | floor1_interpolate_fit(vb,b->flr[info->floorsubmap[submap]], |
564 | 0 | floor_posts[i][0], |
565 | 0 | floor_posts[i][PACKETBLOBS/2], |
566 | 0 | k*65536/(PACKETBLOBS/2)); |
567 | 0 | for(k=PACKETBLOBS/2+1;k<PACKETBLOBS-1;k++) |
568 | 0 | floor_posts[i][k]= |
569 | 0 | floor1_interpolate_fit(vb,b->flr[info->floorsubmap[submap]], |
570 | 0 | floor_posts[i][PACKETBLOBS/2], |
571 | 0 | floor_posts[i][PACKETBLOBS-1], |
572 | 0 | (k-PACKETBLOBS/2)*65536/(PACKETBLOBS/2)); |
573 | 0 | } |
574 | 0 | } |
575 | 0 | } |
576 | 0 | 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 | 0 | { |
593 | 0 | int **couple_bundle=alloca(sizeof(*couple_bundle)*vi->channels); |
594 | 0 | int *zerobundle=alloca(sizeof(*zerobundle)*vi->channels); |
595 | |
|
596 | 0 | for(k=(vorbis_bitrate_managed(vb)?0:PACKETBLOBS/2); |
597 | 0 | k<=(vorbis_bitrate_managed(vb)?PACKETBLOBS-1:PACKETBLOBS/2); |
598 | 0 | k++){ |
599 | 0 | 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 | 0 | oggpack_write(opb,0,1); |
604 | | /* Encode the modenumber */ |
605 | | /* Encode frame mode, pre,post windowsize, then dispatch */ |
606 | 0 | oggpack_write(opb,modenumber,b->modebits); |
607 | 0 | if(vb->W){ |
608 | 0 | oggpack_write(opb,vb->lW,1); |
609 | 0 | oggpack_write(opb,vb->nW,1); |
610 | 0 | } |
611 | | |
612 | | /* encode floor, compute masking curve, sep out residue */ |
613 | 0 | for(i=0;i<vi->channels;i++){ |
614 | 0 | int submap=info->chmuxlist[i]; |
615 | 0 | int *ilogmask=iwork[i]; |
616 | |
|
617 | 0 | nonzero[i]=floor1_encode(opb,vb,b->flr[info->floorsubmap[submap]], |
618 | 0 | floor_posts[i][k], |
619 | 0 | 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 | 0 | } |
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 | 0 | _vp_couple_quantize_normalize(k, |
639 | 0 | &ci->psy_g_param, |
640 | 0 | psy_look, |
641 | 0 | info, |
642 | 0 | gmdct, |
643 | 0 | iwork, |
644 | 0 | nonzero, |
645 | 0 | ci->psy_g_param.sliding_lowpass[vb->W][k], |
646 | 0 | 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 | 0 | for(i=0;i<info->submaps;i++){ |
661 | 0 | int ch_in_bundle=0; |
662 | 0 | long **classifications; |
663 | 0 | int resnum=info->residuesubmap[i]; |
664 | |
|
665 | 0 | for(j=0;j<vi->channels;j++){ |
666 | 0 | if(info->chmuxlist[j]==i){ |
667 | 0 | zerobundle[ch_in_bundle]=0; |
668 | 0 | if(nonzero[j])zerobundle[ch_in_bundle]=1; |
669 | 0 | couple_bundle[ch_in_bundle++]=iwork[j]; |
670 | 0 | } |
671 | 0 | } |
672 | |
|
673 | 0 | classifications=_residue_P[ci->residue_type[resnum]]-> |
674 | 0 | class(vb,b->residue[resnum],couple_bundle,zerobundle,ch_in_bundle); |
675 | |
|
676 | 0 | ch_in_bundle=0; |
677 | 0 | for(j=0;j<vi->channels;j++) |
678 | 0 | if(info->chmuxlist[j]==i) |
679 | 0 | couple_bundle[ch_in_bundle++]=iwork[j]; |
680 | |
|
681 | 0 | _residue_P[ci->residue_type[resnum]]-> |
682 | 0 | forward(opb,vb,b->residue[resnum], |
683 | 0 | couple_bundle,zerobundle,ch_in_bundle,classifications,i); |
684 | 0 | } |
685 | | |
686 | | /* ok, done encoding. Next protopacket. */ |
687 | 0 | } |
688 | |
|
689 | 0 | } |
690 | |
|
691 | | #if 0 |
692 | | seq++; |
693 | | total+=ci->blocksizes[vb->W]/4+ci->blocksizes[vb->nW]/4; |
694 | | #endif |
695 | 0 | return(0); |
696 | 0 | } |
697 | | |
698 | 0 | static int mapping0_inverse(vorbis_block *vb,vorbis_info_mapping *l){ |
699 | 0 | vorbis_dsp_state *vd=vb->vd; |
700 | 0 | vorbis_info *vi=vd->vi; |
701 | 0 | codec_setup_info *ci=vi->codec_setup; |
702 | 0 | private_state *b=vd->backend_state; |
703 | 0 | vorbis_info_mapping0 *info=(vorbis_info_mapping0 *)l; |
704 | |
|
705 | 0 | int i,j; |
706 | 0 | long n=vb->pcmend=ci->blocksizes[vb->W]; |
707 | |
|
708 | 0 | float **pcmbundle=alloca(sizeof(*pcmbundle)*vi->channels); |
709 | 0 | int *zerobundle=alloca(sizeof(*zerobundle)*vi->channels); |
710 | |
|
711 | 0 | int *nonzero =alloca(sizeof(*nonzero)*vi->channels); |
712 | 0 | void **floormemo=alloca(sizeof(*floormemo)*vi->channels); |
713 | | |
714 | | /* recover the spectral envelope; store it in the PCM vector for now */ |
715 | 0 | for(i=0;i<vi->channels;i++){ |
716 | 0 | int submap=info->chmuxlist[i]; |
717 | 0 | floormemo[i]=_floor_P[ci->floor_type[info->floorsubmap[submap]]]-> |
718 | 0 | inverse1(vb,b->flr[info->floorsubmap[submap]]); |
719 | 0 | if(floormemo[i]) |
720 | 0 | nonzero[i]=1; |
721 | 0 | else |
722 | 0 | nonzero[i]=0; |
723 | 0 | memset(vb->pcm[i],0,sizeof(*vb->pcm[i])*n/2); |
724 | 0 | } |
725 | | |
726 | | /* channel coupling can 'dirty' the nonzero listing */ |
727 | 0 | for(i=0;i<info->coupling_steps;i++){ |
728 | 0 | if(nonzero[info->coupling_mag[i]] || |
729 | 0 | nonzero[info->coupling_ang[i]]){ |
730 | 0 | nonzero[info->coupling_mag[i]]=1; |
731 | 0 | nonzero[info->coupling_ang[i]]=1; |
732 | 0 | } |
733 | 0 | } |
734 | | |
735 | | /* recover the residue into our working vectors */ |
736 | 0 | for(i=0;i<info->submaps;i++){ |
737 | 0 | int ch_in_bundle=0; |
738 | 0 | for(j=0;j<vi->channels;j++){ |
739 | 0 | if(info->chmuxlist[j]==i){ |
740 | 0 | if(nonzero[j]) |
741 | 0 | zerobundle[ch_in_bundle]=1; |
742 | 0 | else |
743 | 0 | zerobundle[ch_in_bundle]=0; |
744 | 0 | pcmbundle[ch_in_bundle++]=vb->pcm[j]; |
745 | 0 | } |
746 | 0 | } |
747 | |
|
748 | 0 | _residue_P[ci->residue_type[info->residuesubmap[i]]]-> |
749 | 0 | inverse(vb,b->residue[info->residuesubmap[i]], |
750 | 0 | pcmbundle,zerobundle,ch_in_bundle); |
751 | 0 | } |
752 | | |
753 | | /* channel coupling */ |
754 | 0 | for(i=info->coupling_steps-1;i>=0;i--){ |
755 | 0 | float *pcmM=vb->pcm[info->coupling_mag[i]]; |
756 | 0 | float *pcmA=vb->pcm[info->coupling_ang[i]]; |
757 | |
|
758 | 0 | for(j=0;j<n/2;j++){ |
759 | 0 | float mag=pcmM[j]; |
760 | 0 | float ang=pcmA[j]; |
761 | |
|
762 | 0 | if(mag>0) |
763 | 0 | if(ang>0){ |
764 | 0 | pcmM[j]=mag; |
765 | 0 | pcmA[j]=mag-ang; |
766 | 0 | }else{ |
767 | 0 | pcmA[j]=mag; |
768 | 0 | pcmM[j]=mag+ang; |
769 | 0 | } |
770 | 0 | else |
771 | 0 | if(ang>0){ |
772 | 0 | pcmM[j]=mag; |
773 | 0 | pcmA[j]=mag+ang; |
774 | 0 | }else{ |
775 | 0 | pcmA[j]=mag; |
776 | 0 | pcmM[j]=mag-ang; |
777 | 0 | } |
778 | 0 | } |
779 | 0 | } |
780 | | |
781 | | /* compute and apply spectral envelope */ |
782 | 0 | for(i=0;i<vi->channels;i++){ |
783 | 0 | float *pcm=vb->pcm[i]; |
784 | 0 | int submap=info->chmuxlist[i]; |
785 | 0 | _floor_P[ci->floor_type[info->floorsubmap[submap]]]-> |
786 | 0 | inverse2(vb,b->flr[info->floorsubmap[submap]], |
787 | 0 | floormemo[i],pcm); |
788 | 0 | } |
789 | | |
790 | | /* transform the PCM data; takes PCM vector, vb; modifies PCM vector */ |
791 | | /* only MDCT right now.... */ |
792 | 0 | for(i=0;i<vi->channels;i++){ |
793 | 0 | float *pcm=vb->pcm[i]; |
794 | 0 | mdct_backward(b->transform[vb->W][0],pcm,pcm); |
795 | 0 | } |
796 | | |
797 | | /* all done! */ |
798 | 0 | return(0); |
799 | 0 | } |
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 | | }; |