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-2007 * |
---|
9 | * by the Xiph.Org Foundation http://www.xiph.org/ * |
---|
10 | * * |
---|
11 | ******************************************************************** |
---|
12 | |
---|
13 | function: psychoacoustics not including preecho |
---|
14 | last mod: $Id: psy.c 13293 2007-07-24 00:09:47Z xiphmont $ |
---|
15 | |
---|
16 | ********************************************************************/ |
---|
17 | |
---|
18 | #include <stdlib.h> |
---|
19 | #include <math.h> |
---|
20 | #include <string.h> |
---|
21 | #include "vorbis/codec.h" |
---|
22 | #include "codec_internal.h" |
---|
23 | |
---|
24 | #include "masking.h" |
---|
25 | #include "psy.h" |
---|
26 | #include "os.h" |
---|
27 | #include "lpc.h" |
---|
28 | #include "smallft.h" |
---|
29 | #include "scales.h" |
---|
30 | #include "misc.h" |
---|
31 | |
---|
32 | #define NEGINF -9999.f |
---|
33 | static double stereo_threshholds[]={0.0, .5, 1.0, 1.5, 2.5, 4.5, 8.5, 16.5, 9e10}; |
---|
34 | static double stereo_threshholds_limited[]={0.0, .5, 1.0, 1.5, 2.0, 2.5, 4.5, 8.5, 9e10}; |
---|
35 | |
---|
36 | vorbis_look_psy_global *_vp_global_look(vorbis_info *vi){ |
---|
37 | codec_setup_info *ci=vi->codec_setup; |
---|
38 | vorbis_info_psy_global *gi=&ci->psy_g_param; |
---|
39 | vorbis_look_psy_global *look=_ogg_calloc(1,sizeof(*look)); |
---|
40 | |
---|
41 | look->channels=vi->channels; |
---|
42 | |
---|
43 | look->ampmax=-9999.; |
---|
44 | look->gi=gi; |
---|
45 | return(look); |
---|
46 | } |
---|
47 | |
---|
48 | void _vp_global_free(vorbis_look_psy_global *look){ |
---|
49 | if(look){ |
---|
50 | memset(look,0,sizeof(*look)); |
---|
51 | _ogg_free(look); |
---|
52 | } |
---|
53 | } |
---|
54 | |
---|
55 | void _vi_gpsy_free(vorbis_info_psy_global *i){ |
---|
56 | if(i){ |
---|
57 | memset(i,0,sizeof(*i)); |
---|
58 | _ogg_free(i); |
---|
59 | } |
---|
60 | } |
---|
61 | |
---|
62 | void _vi_psy_free(vorbis_info_psy *i){ |
---|
63 | if(i){ |
---|
64 | memset(i,0,sizeof(*i)); |
---|
65 | _ogg_free(i); |
---|
66 | } |
---|
67 | } |
---|
68 | |
---|
69 | static void min_curve(float *c, |
---|
70 | float *c2){ |
---|
71 | int i; |
---|
72 | for(i=0;i<EHMER_MAX;i++)if(c2[i]<c[i])c[i]=c2[i]; |
---|
73 | } |
---|
74 | static void max_curve(float *c, |
---|
75 | float *c2){ |
---|
76 | int i; |
---|
77 | for(i=0;i<EHMER_MAX;i++)if(c2[i]>c[i])c[i]=c2[i]; |
---|
78 | } |
---|
79 | |
---|
80 | static void attenuate_curve(float *c,float att){ |
---|
81 | int i; |
---|
82 | for(i=0;i<EHMER_MAX;i++) |
---|
83 | c[i]+=att; |
---|
84 | } |
---|
85 | |
---|
86 | static float ***setup_tone_curves(float curveatt_dB[P_BANDS],float binHz,int n, |
---|
87 | float center_boost, float center_decay_rate){ |
---|
88 | int i,j,k,m; |
---|
89 | float ath[EHMER_MAX]; |
---|
90 | float workc[P_BANDS][P_LEVELS][EHMER_MAX]; |
---|
91 | float athc[P_LEVELS][EHMER_MAX]; |
---|
92 | float *brute_buffer=alloca(n*sizeof(*brute_buffer)); |
---|
93 | |
---|
94 | float ***ret=_ogg_malloc(sizeof(*ret)*P_BANDS); |
---|
95 | |
---|
96 | memset(workc,0,sizeof(workc)); |
---|
97 | |
---|
98 | for(i=0;i<P_BANDS;i++){ |
---|
99 | /* we add back in the ATH to avoid low level curves falling off to |
---|
100 | -infinity and unnecessarily cutting off high level curves in the |
---|
101 | curve limiting (last step). */ |
---|
102 | |
---|
103 | /* A half-band's settings must be valid over the whole band, and |
---|
104 | it's better to mask too little than too much */ |
---|
105 | int ath_offset=i*4; |
---|
106 | for(j=0;j<EHMER_MAX;j++){ |
---|
107 | float min=999.; |
---|
108 | for(k=0;k<4;k++) |
---|
109 | if(j+k+ath_offset<MAX_ATH){ |
---|
110 | if(min>ATH[j+k+ath_offset])min=ATH[j+k+ath_offset]; |
---|
111 | }else{ |
---|
112 | if(min>ATH[MAX_ATH-1])min=ATH[MAX_ATH-1]; |
---|
113 | } |
---|
114 | ath[j]=min; |
---|
115 | } |
---|
116 | |
---|
117 | /* copy curves into working space, replicate the 50dB curve to 30 |
---|
118 | and 40, replicate the 100dB curve to 110 */ |
---|
119 | for(j=0;j<6;j++) |
---|
120 | memcpy(workc[i][j+2],tonemasks[i][j],EHMER_MAX*sizeof(*tonemasks[i][j])); |
---|
121 | memcpy(workc[i][0],tonemasks[i][0],EHMER_MAX*sizeof(*tonemasks[i][0])); |
---|
122 | memcpy(workc[i][1],tonemasks[i][0],EHMER_MAX*sizeof(*tonemasks[i][0])); |
---|
123 | |
---|
124 | /* apply centered curve boost/decay */ |
---|
125 | for(j=0;j<P_LEVELS;j++){ |
---|
126 | for(k=0;k<EHMER_MAX;k++){ |
---|
127 | float adj=center_boost+abs(EHMER_OFFSET-k)*center_decay_rate; |
---|
128 | if(adj<0. && center_boost>0)adj=0.; |
---|
129 | if(adj>0. && center_boost<0)adj=0.; |
---|
130 | workc[i][j][k]+=adj; |
---|
131 | } |
---|
132 | } |
---|
133 | |
---|
134 | /* normalize curves so the driving amplitude is 0dB */ |
---|
135 | /* make temp curves with the ATH overlayed */ |
---|
136 | for(j=0;j<P_LEVELS;j++){ |
---|
137 | attenuate_curve(workc[i][j],curveatt_dB[i]+100.-(j<2?2:j)*10.-P_LEVEL_0); |
---|
138 | memcpy(athc[j],ath,EHMER_MAX*sizeof(**athc)); |
---|
139 | attenuate_curve(athc[j],+100.-j*10.f-P_LEVEL_0); |
---|
140 | max_curve(athc[j],workc[i][j]); |
---|
141 | } |
---|
142 | |
---|
143 | /* Now limit the louder curves. |
---|
144 | |
---|
145 | the idea is this: We don't know what the playback attenuation |
---|
146 | will be; 0dB SL moves every time the user twiddles the volume |
---|
147 | knob. So that means we have to use a single 'most pessimal' curve |
---|
148 | for all masking amplitudes, right? Wrong. The *loudest* sound |
---|
149 | can be in (we assume) a range of ...+100dB] SL. However, sounds |
---|
150 | 20dB down will be in a range ...+80], 40dB down is from ...+60], |
---|
151 | etc... */ |
---|
152 | |
---|
153 | for(j=1;j<P_LEVELS;j++){ |
---|
154 | min_curve(athc[j],athc[j-1]); |
---|
155 | min_curve(workc[i][j],athc[j]); |
---|
156 | } |
---|
157 | } |
---|
158 | |
---|
159 | for(i=0;i<P_BANDS;i++){ |
---|
160 | int hi_curve,lo_curve,bin; |
---|
161 | ret[i]=_ogg_malloc(sizeof(**ret)*P_LEVELS); |
---|
162 | |
---|
163 | /* low frequency curves are measured with greater resolution than |
---|
164 | the MDCT/FFT will actually give us; we want the curve applied |
---|
165 | to the tone data to be pessimistic and thus apply the minimum |
---|
166 | masking possible for a given bin. That means that a single bin |
---|
167 | could span more than one octave and that the curve will be a |
---|
168 | composite of multiple octaves. It also may mean that a single |
---|
169 | bin may span > an eighth of an octave and that the eighth |
---|
170 | octave values may also be composited. */ |
---|
171 | |
---|
172 | /* which octave curves will we be compositing? */ |
---|
173 | bin=floor(fromOC(i*.5)/binHz); |
---|
174 | lo_curve= ceil(toOC(bin*binHz+1)*2); |
---|
175 | hi_curve= floor(toOC((bin+1)*binHz)*2); |
---|
176 | if(lo_curve>i)lo_curve=i; |
---|
177 | if(lo_curve<0)lo_curve=0; |
---|
178 | if(hi_curve>=P_BANDS)hi_curve=P_BANDS-1; |
---|
179 | |
---|
180 | for(m=0;m<P_LEVELS;m++){ |
---|
181 | ret[i][m]=_ogg_malloc(sizeof(***ret)*(EHMER_MAX+2)); |
---|
182 | |
---|
183 | for(j=0;j<n;j++)brute_buffer[j]=999.; |
---|
184 | |
---|
185 | /* render the curve into bins, then pull values back into curve. |
---|
186 | The point is that any inherent subsampling aliasing results in |
---|
187 | a safe minimum */ |
---|
188 | for(k=lo_curve;k<=hi_curve;k++){ |
---|
189 | int l=0; |
---|
190 | |
---|
191 | for(j=0;j<EHMER_MAX;j++){ |
---|
192 | int lo_bin= fromOC(j*.125+k*.5-2.0625)/binHz; |
---|
193 | int hi_bin= fromOC(j*.125+k*.5-1.9375)/binHz+1; |
---|
194 | |
---|
195 | if(lo_bin<0)lo_bin=0; |
---|
196 | if(lo_bin>n)lo_bin=n; |
---|
197 | if(lo_bin<l)l=lo_bin; |
---|
198 | if(hi_bin<0)hi_bin=0; |
---|
199 | if(hi_bin>n)hi_bin=n; |
---|
200 | |
---|
201 | for(;l<hi_bin && l<n;l++) |
---|
202 | if(brute_buffer[l]>workc[k][m][j]) |
---|
203 | brute_buffer[l]=workc[k][m][j]; |
---|
204 | } |
---|
205 | |
---|
206 | for(;l<n;l++) |
---|
207 | if(brute_buffer[l]>workc[k][m][EHMER_MAX-1]) |
---|
208 | brute_buffer[l]=workc[k][m][EHMER_MAX-1]; |
---|
209 | |
---|
210 | } |
---|
211 | |
---|
212 | /* be equally paranoid about being valid up to next half ocatve */ |
---|
213 | if(i+1<P_BANDS){ |
---|
214 | int l=0; |
---|
215 | k=i+1; |
---|
216 | for(j=0;j<EHMER_MAX;j++){ |
---|
217 | int lo_bin= fromOC(j*.125+i*.5-2.0625)/binHz; |
---|
218 | int hi_bin= fromOC(j*.125+i*.5-1.9375)/binHz+1; |
---|
219 | |
---|
220 | if(lo_bin<0)lo_bin=0; |
---|
221 | if(lo_bin>n)lo_bin=n; |
---|
222 | if(lo_bin<l)l=lo_bin; |
---|
223 | if(hi_bin<0)hi_bin=0; |
---|
224 | if(hi_bin>n)hi_bin=n; |
---|
225 | |
---|
226 | for(;l<hi_bin && l<n;l++) |
---|
227 | if(brute_buffer[l]>workc[k][m][j]) |
---|
228 | brute_buffer[l]=workc[k][m][j]; |
---|
229 | } |
---|
230 | |
---|
231 | for(;l<n;l++) |
---|
232 | if(brute_buffer[l]>workc[k][m][EHMER_MAX-1]) |
---|
233 | brute_buffer[l]=workc[k][m][EHMER_MAX-1]; |
---|
234 | |
---|
235 | } |
---|
236 | |
---|
237 | |
---|
238 | for(j=0;j<EHMER_MAX;j++){ |
---|
239 | int bin=fromOC(j*.125+i*.5-2.)/binHz; |
---|
240 | if(bin<0){ |
---|
241 | ret[i][m][j+2]=-999.; |
---|
242 | }else{ |
---|
243 | if(bin>=n){ |
---|
244 | ret[i][m][j+2]=-999.; |
---|
245 | }else{ |
---|
246 | ret[i][m][j+2]=brute_buffer[bin]; |
---|
247 | } |
---|
248 | } |
---|
249 | } |
---|
250 | |
---|
251 | /* add fenceposts */ |
---|
252 | for(j=0;j<EHMER_OFFSET;j++) |
---|
253 | if(ret[i][m][j+2]>-200.f)break; |
---|
254 | ret[i][m][0]=j; |
---|
255 | |
---|
256 | for(j=EHMER_MAX-1;j>EHMER_OFFSET+1;j--) |
---|
257 | if(ret[i][m][j+2]>-200.f) |
---|
258 | break; |
---|
259 | ret[i][m][1]=j; |
---|
260 | |
---|
261 | } |
---|
262 | } |
---|
263 | |
---|
264 | return(ret); |
---|
265 | } |
---|
266 | |
---|
267 | void _vp_psy_init(vorbis_look_psy *p,vorbis_info_psy *vi, |
---|
268 | vorbis_info_psy_global *gi,int n,long rate){ |
---|
269 | long i,j,lo=-99,hi=1; |
---|
270 | long maxoc; |
---|
271 | memset(p,0,sizeof(*p)); |
---|
272 | |
---|
273 | p->eighth_octave_lines=gi->eighth_octave_lines; |
---|
274 | p->shiftoc=rint(log(gi->eighth_octave_lines*8.f)/log(2.f))-1; |
---|
275 | |
---|
276 | p->firstoc=toOC(.25f*rate*.5/n)*(1<<(p->shiftoc+1))-gi->eighth_octave_lines; |
---|
277 | maxoc=toOC((n+.25f)*rate*.5/n)*(1<<(p->shiftoc+1))+.5f; |
---|
278 | p->total_octave_lines=maxoc-p->firstoc+1; |
---|
279 | p->ath=_ogg_malloc(n*sizeof(*p->ath)); |
---|
280 | |
---|
281 | p->octave=_ogg_malloc(n*sizeof(*p->octave)); |
---|
282 | p->bark=_ogg_malloc(n*sizeof(*p->bark)); |
---|
283 | p->vi=vi; |
---|
284 | p->n=n; |
---|
285 | p->rate=rate; |
---|
286 | |
---|
287 | /* AoTuV HF weighting */ |
---|
288 | p->m_val = 1.; |
---|
289 | if(rate < 26000) p->m_val = 0; |
---|
290 | else if(rate < 38000) p->m_val = .94; /* 32kHz */ |
---|
291 | else if(rate > 46000) p->m_val = 1.275; /* 48kHz */ |
---|
292 | |
---|
293 | /* set up the lookups for a given blocksize and sample rate */ |
---|
294 | |
---|
295 | for(i=0,j=0;i<MAX_ATH-1;i++){ |
---|
296 | int endpos=rint(fromOC((i+1)*.125-2.)*2*n/rate); |
---|
297 | float base=ATH[i]; |
---|
298 | if(j<endpos){ |
---|
299 | float delta=(ATH[i+1]-base)/(endpos-j); |
---|
300 | for(;j<endpos && j<n;j++){ |
---|
301 | p->ath[j]=base+100.; |
---|
302 | base+=delta; |
---|
303 | } |
---|
304 | } |
---|
305 | } |
---|
306 | |
---|
307 | for(i=0;i<n;i++){ |
---|
308 | float bark=toBARK(rate/(2*n)*i); |
---|
309 | |
---|
310 | for(;lo+vi->noisewindowlomin<i && |
---|
311 | toBARK(rate/(2*n)*lo)<(bark-vi->noisewindowlo);lo++); |
---|
312 | |
---|
313 | for(;hi<=n && (hi<i+vi->noisewindowhimin || |
---|
314 | toBARK(rate/(2*n)*hi)<(bark+vi->noisewindowhi));hi++); |
---|
315 | |
---|
316 | p->bark[i]=((lo-1)<<16)+(hi-1); |
---|
317 | |
---|
318 | } |
---|
319 | |
---|
320 | for(i=0;i<n;i++) |
---|
321 | p->octave[i]=toOC((i+.25f)*.5*rate/n)*(1<<(p->shiftoc+1))+.5f; |
---|
322 | |
---|
323 | p->tonecurves=setup_tone_curves(vi->toneatt,rate*.5/n,n, |
---|
324 | vi->tone_centerboost,vi->tone_decay); |
---|
325 | |
---|
326 | /* set up rolling noise median */ |
---|
327 | p->noiseoffset=_ogg_malloc(P_NOISECURVES*sizeof(*p->noiseoffset)); |
---|
328 | for(i=0;i<P_NOISECURVES;i++) |
---|
329 | p->noiseoffset[i]=_ogg_malloc(n*sizeof(**p->noiseoffset)); |
---|
330 | |
---|
331 | for(i=0;i<n;i++){ |
---|
332 | float halfoc=toOC((i+.5)*rate/(2.*n))*2.; |
---|
333 | int inthalfoc; |
---|
334 | float del; |
---|
335 | |
---|
336 | if(halfoc<0)halfoc=0; |
---|
337 | if(halfoc>=P_BANDS-1)halfoc=P_BANDS-1; |
---|
338 | inthalfoc=(int)halfoc; |
---|
339 | del=halfoc-inthalfoc; |
---|
340 | |
---|
341 | for(j=0;j<P_NOISECURVES;j++) |
---|
342 | p->noiseoffset[j][i]= |
---|
343 | p->vi->noiseoff[j][inthalfoc]*(1.-del) + |
---|
344 | p->vi->noiseoff[j][inthalfoc+1]*del; |
---|
345 | |
---|
346 | } |
---|
347 | #if 0 |
---|
348 | { |
---|
349 | static int ls=0; |
---|
350 | _analysis_output_always("noiseoff0",ls,p->noiseoffset[0],n,1,0,0); |
---|
351 | _analysis_output_always("noiseoff1",ls,p->noiseoffset[1],n,1,0,0); |
---|
352 | _analysis_output_always("noiseoff2",ls++,p->noiseoffset[2],n,1,0,0); |
---|
353 | } |
---|
354 | #endif |
---|
355 | } |
---|
356 | |
---|
357 | void _vp_psy_clear(vorbis_look_psy *p){ |
---|
358 | int i,j; |
---|
359 | if(p){ |
---|
360 | if(p->ath)_ogg_free(p->ath); |
---|
361 | if(p->octave)_ogg_free(p->octave); |
---|
362 | if(p->bark)_ogg_free(p->bark); |
---|
363 | if(p->tonecurves){ |
---|
364 | for(i=0;i<P_BANDS;i++){ |
---|
365 | for(j=0;j<P_LEVELS;j++){ |
---|
366 | _ogg_free(p->tonecurves[i][j]); |
---|
367 | } |
---|
368 | _ogg_free(p->tonecurves[i]); |
---|
369 | } |
---|
370 | _ogg_free(p->tonecurves); |
---|
371 | } |
---|
372 | if(p->noiseoffset){ |
---|
373 | for(i=0;i<P_NOISECURVES;i++){ |
---|
374 | _ogg_free(p->noiseoffset[i]); |
---|
375 | } |
---|
376 | _ogg_free(p->noiseoffset); |
---|
377 | } |
---|
378 | memset(p,0,sizeof(*p)); |
---|
379 | } |
---|
380 | } |
---|
381 | |
---|
382 | /* octave/(8*eighth_octave_lines) x scale and dB y scale */ |
---|
383 | static void seed_curve(float *seed, |
---|
384 | const float **curves, |
---|
385 | float amp, |
---|
386 | int oc, int n, |
---|
387 | int linesper,float dBoffset){ |
---|
388 | int i,post1; |
---|
389 | int seedptr; |
---|
390 | const float *posts,*curve; |
---|
391 | |
---|
392 | int choice=(int)((amp+dBoffset-P_LEVEL_0)*.1f); |
---|
393 | choice=max(choice,0); |
---|
394 | choice=min(choice,P_LEVELS-1); |
---|
395 | posts=curves[choice]; |
---|
396 | curve=posts+2; |
---|
397 | post1=(int)posts[1]; |
---|
398 | seedptr=oc+(posts[0]-EHMER_OFFSET)*linesper-(linesper>>1); |
---|
399 | |
---|
400 | for(i=posts[0];i<post1;i++){ |
---|
401 | if(seedptr>0){ |
---|
402 | float lin=amp+curve[i]; |
---|
403 | if(seed[seedptr]<lin)seed[seedptr]=lin; |
---|
404 | } |
---|
405 | seedptr+=linesper; |
---|
406 | if(seedptr>=n)break; |
---|
407 | } |
---|
408 | } |
---|
409 | |
---|
410 | static void seed_loop(vorbis_look_psy *p, |
---|
411 | const float ***curves, |
---|
412 | const float *f, |
---|
413 | const float *flr, |
---|
414 | float *seed, |
---|
415 | float specmax){ |
---|
416 | vorbis_info_psy *vi=p->vi; |
---|
417 | long n=p->n,i; |
---|
418 | float dBoffset=vi->max_curve_dB-specmax; |
---|
419 | |
---|
420 | /* prime the working vector with peak values */ |
---|
421 | |
---|
422 | for(i=0;i<n;i++){ |
---|
423 | float max=f[i]; |
---|
424 | long oc=p->octave[i]; |
---|
425 | while(i+1<n && p->octave[i+1]==oc){ |
---|
426 | i++; |
---|
427 | if(f[i]>max)max=f[i]; |
---|
428 | } |
---|
429 | |
---|
430 | if(max+6.f>flr[i]){ |
---|
431 | oc=oc>>p->shiftoc; |
---|
432 | |
---|
433 | if(oc>=P_BANDS)oc=P_BANDS-1; |
---|
434 | if(oc<0)oc=0; |
---|
435 | |
---|
436 | seed_curve(seed, |
---|
437 | curves[oc], |
---|
438 | max, |
---|
439 | p->octave[i]-p->firstoc, |
---|
440 | p->total_octave_lines, |
---|
441 | p->eighth_octave_lines, |
---|
442 | dBoffset); |
---|
443 | } |
---|
444 | } |
---|
445 | } |
---|
446 | |
---|
447 | static void seed_chase(float *seeds, int linesper, long n){ |
---|
448 | long *posstack=alloca(n*sizeof(*posstack)); |
---|
449 | float *ampstack=alloca(n*sizeof(*ampstack)); |
---|
450 | long stack=0; |
---|
451 | long pos=0; |
---|
452 | long i; |
---|
453 | |
---|
454 | for(i=0;i<n;i++){ |
---|
455 | if(stack<2){ |
---|
456 | posstack[stack]=i; |
---|
457 | ampstack[stack++]=seeds[i]; |
---|
458 | }else{ |
---|
459 | while(1){ |
---|
460 | if(seeds[i]<ampstack[stack-1]){ |
---|
461 | posstack[stack]=i; |
---|
462 | ampstack[stack++]=seeds[i]; |
---|
463 | break; |
---|
464 | }else{ |
---|
465 | if(i<posstack[stack-1]+linesper){ |
---|
466 | if(stack>1 && ampstack[stack-1]<=ampstack[stack-2] && |
---|
467 | i<posstack[stack-2]+linesper){ |
---|
468 | /* we completely overlap, making stack-1 irrelevant. pop it */ |
---|
469 | stack--; |
---|
470 | continue; |
---|
471 | } |
---|
472 | } |
---|
473 | posstack[stack]=i; |
---|
474 | ampstack[stack++]=seeds[i]; |
---|
475 | break; |
---|
476 | |
---|
477 | } |
---|
478 | } |
---|
479 | } |
---|
480 | } |
---|
481 | |
---|
482 | /* the stack now contains only the positions that are relevant. Scan |
---|
483 | 'em straight through */ |
---|
484 | |
---|
485 | for(i=0;i<stack;i++){ |
---|
486 | long endpos; |
---|
487 | if(i<stack-1 && ampstack[i+1]>ampstack[i]){ |
---|
488 | endpos=posstack[i+1]; |
---|
489 | }else{ |
---|
490 | endpos=posstack[i]+linesper+1; /* +1 is important, else bin 0 is |
---|
491 | discarded in short frames */ |
---|
492 | } |
---|
493 | if(endpos>n)endpos=n; |
---|
494 | for(;pos<endpos;pos++) |
---|
495 | seeds[pos]=ampstack[i]; |
---|
496 | } |
---|
497 | |
---|
498 | /* there. Linear time. I now remember this was on a problem set I |
---|
499 | had in Grad Skool... I didn't solve it at the time ;-) */ |
---|
500 | |
---|
501 | } |
---|
502 | |
---|
503 | /* bleaugh, this is more complicated than it needs to be */ |
---|
504 | #include<stdio.h> |
---|
505 | static void max_seeds(vorbis_look_psy *p, |
---|
506 | float *seed, |
---|
507 | float *flr){ |
---|
508 | long n=p->total_octave_lines; |
---|
509 | int linesper=p->eighth_octave_lines; |
---|
510 | long linpos=0; |
---|
511 | long pos; |
---|
512 | |
---|
513 | seed_chase(seed,linesper,n); /* for masking */ |
---|
514 | |
---|
515 | pos=p->octave[0]-p->firstoc-(linesper>>1); |
---|
516 | |
---|
517 | while(linpos+1<p->n){ |
---|
518 | float minV=seed[pos]; |
---|
519 | long end=((p->octave[linpos]+p->octave[linpos+1])>>1)-p->firstoc; |
---|
520 | if(minV>p->vi->tone_abs_limit)minV=p->vi->tone_abs_limit; |
---|
521 | while(pos+1<=end){ |
---|
522 | pos++; |
---|
523 | if((seed[pos]>NEGINF && seed[pos]<minV) || minV==NEGINF) |
---|
524 | minV=seed[pos]; |
---|
525 | } |
---|
526 | |
---|
527 | end=pos+p->firstoc; |
---|
528 | for(;linpos<p->n && p->octave[linpos]<=end;linpos++) |
---|
529 | if(flr[linpos]<minV)flr[linpos]=minV; |
---|
530 | } |
---|
531 | |
---|
532 | { |
---|
533 | float minV=seed[p->total_octave_lines-1]; |
---|
534 | for(;linpos<p->n;linpos++) |
---|
535 | if(flr[linpos]<minV)flr[linpos]=minV; |
---|
536 | } |
---|
537 | |
---|
538 | } |
---|
539 | |
---|
540 | static void bark_noise_hybridmp(int n,const long *b, |
---|
541 | const float *f, |
---|
542 | float *noise, |
---|
543 | const float offset, |
---|
544 | const int fixed){ |
---|
545 | |
---|
546 | float *N=alloca(n*sizeof(*N)); |
---|
547 | float *X=alloca(n*sizeof(*N)); |
---|
548 | float *XX=alloca(n*sizeof(*N)); |
---|
549 | float *Y=alloca(n*sizeof(*N)); |
---|
550 | float *XY=alloca(n*sizeof(*N)); |
---|
551 | |
---|
552 | float tN, tX, tXX, tY, tXY; |
---|
553 | int i; |
---|
554 | |
---|
555 | int lo, hi; |
---|
556 | float R=0.f; |
---|
557 | float A=0.f; |
---|
558 | float B=0.f; |
---|
559 | float D=1.f; |
---|
560 | float w, x, y; |
---|
561 | |
---|
562 | tN = tX = tXX = tY = tXY = 0.f; |
---|
563 | |
---|
564 | y = f[0] + offset; |
---|
565 | if (y < 1.f) y = 1.f; |
---|
566 | |
---|
567 | w = y * y * .5; |
---|
568 | |
---|
569 | tN += w; |
---|
570 | tX += w; |
---|
571 | tY += w * y; |
---|
572 | |
---|
573 | N[0] = tN; |
---|
574 | X[0] = tX; |
---|
575 | XX[0] = tXX; |
---|
576 | Y[0] = tY; |
---|
577 | XY[0] = tXY; |
---|
578 | |
---|
579 | for (i = 1, x = 1.f; i < n; i++, x += 1.f) { |
---|
580 | |
---|
581 | y = f[i] + offset; |
---|
582 | if (y < 1.f) y = 1.f; |
---|
583 | |
---|
584 | w = y * y; |
---|
585 | |
---|
586 | tN += w; |
---|
587 | tX += w * x; |
---|
588 | tXX += w * x * x; |
---|
589 | tY += w * y; |
---|
590 | tXY += w * x * y; |
---|
591 | |
---|
592 | N[i] = tN; |
---|
593 | X[i] = tX; |
---|
594 | XX[i] = tXX; |
---|
595 | Y[i] = tY; |
---|
596 | XY[i] = tXY; |
---|
597 | } |
---|
598 | |
---|
599 | for (i = 0, x = 0.f;; i++, x += 1.f) { |
---|
600 | |
---|
601 | lo = b[i] >> 16; |
---|
602 | if( lo>=0 ) break; |
---|
603 | hi = b[i] & 0xffff; |
---|
604 | |
---|
605 | tN = N[hi] + N[-lo]; |
---|
606 | tX = X[hi] - X[-lo]; |
---|
607 | tXX = XX[hi] + XX[-lo]; |
---|
608 | tY = Y[hi] + Y[-lo]; |
---|
609 | tXY = XY[hi] - XY[-lo]; |
---|
610 | |
---|
611 | A = tY * tXX - tX * tXY; |
---|
612 | B = tN * tXY - tX * tY; |
---|
613 | D = tN * tXX - tX * tX; |
---|
614 | R = (A + x * B) / D; |
---|
615 | if (R < 0.f) |
---|
616 | R = 0.f; |
---|
617 | |
---|
618 | noise[i] = R - offset; |
---|
619 | } |
---|
620 | |
---|
621 | for ( ;; i++, x += 1.f) { |
---|
622 | |
---|
623 | lo = b[i] >> 16; |
---|
624 | hi = b[i] & 0xffff; |
---|
625 | if(hi>=n)break; |
---|
626 | |
---|
627 | tN = N[hi] - N[lo]; |
---|
628 | tX = X[hi] - X[lo]; |
---|
629 | tXX = XX[hi] - XX[lo]; |
---|
630 | tY = Y[hi] - Y[lo]; |
---|
631 | tXY = XY[hi] - XY[lo]; |
---|
632 | |
---|
633 | A = tY * tXX - tX * tXY; |
---|
634 | B = tN * tXY - tX * tY; |
---|
635 | D = tN * tXX - tX * tX; |
---|
636 | R = (A + x * B) / D; |
---|
637 | if (R < 0.f) R = 0.f; |
---|
638 | |
---|
639 | noise[i] = R - offset; |
---|
640 | } |
---|
641 | for ( ; i < n; i++, x += 1.f) { |
---|
642 | |
---|
643 | R = (A + x * B) / D; |
---|
644 | if (R < 0.f) R = 0.f; |
---|
645 | |
---|
646 | noise[i] = R - offset; |
---|
647 | } |
---|
648 | |
---|
649 | if (fixed <= 0) return; |
---|
650 | |
---|
651 | for (i = 0, x = 0.f;; i++, x += 1.f) { |
---|
652 | hi = i + fixed / 2; |
---|
653 | lo = hi - fixed; |
---|
654 | if(lo>=0)break; |
---|
655 | |
---|
656 | tN = N[hi] + N[-lo]; |
---|
657 | tX = X[hi] - X[-lo]; |
---|
658 | tXX = XX[hi] + XX[-lo]; |
---|
659 | tY = Y[hi] + Y[-lo]; |
---|
660 | tXY = XY[hi] - XY[-lo]; |
---|
661 | |
---|
662 | |
---|
663 | A = tY * tXX - tX * tXY; |
---|
664 | B = tN * tXY - tX * tY; |
---|
665 | D = tN * tXX - tX * tX; |
---|
666 | R = (A + x * B) / D; |
---|
667 | |
---|
668 | if (R - offset < noise[i]) noise[i] = R - offset; |
---|
669 | } |
---|
670 | for ( ;; i++, x += 1.f) { |
---|
671 | |
---|
672 | hi = i + fixed / 2; |
---|
673 | lo = hi - fixed; |
---|
674 | if(hi>=n)break; |
---|
675 | |
---|
676 | tN = N[hi] - N[lo]; |
---|
677 | tX = X[hi] - X[lo]; |
---|
678 | tXX = XX[hi] - XX[lo]; |
---|
679 | tY = Y[hi] - Y[lo]; |
---|
680 | tXY = XY[hi] - XY[lo]; |
---|
681 | |
---|
682 | A = tY * tXX - tX * tXY; |
---|
683 | B = tN * tXY - tX * tY; |
---|
684 | D = tN * tXX - tX * tX; |
---|
685 | R = (A + x * B) / D; |
---|
686 | |
---|
687 | if (R - offset < noise[i]) noise[i] = R - offset; |
---|
688 | } |
---|
689 | for ( ; i < n; i++, x += 1.f) { |
---|
690 | R = (A + x * B) / D; |
---|
691 | if (R - offset < noise[i]) noise[i] = R - offset; |
---|
692 | } |
---|
693 | } |
---|
694 | |
---|
695 | static float FLOOR1_fromdB_INV_LOOKUP[256]={ |
---|
696 | 0.F, 8.81683e+06F, 8.27882e+06F, 7.77365e+06F, |
---|
697 | 7.29930e+06F, 6.85389e+06F, 6.43567e+06F, 6.04296e+06F, |
---|
698 | 5.67422e+06F, 5.32798e+06F, 5.00286e+06F, 4.69759e+06F, |
---|
699 | 4.41094e+06F, 4.14178e+06F, 3.88905e+06F, 3.65174e+06F, |
---|
700 | 3.42891e+06F, 3.21968e+06F, 3.02321e+06F, 2.83873e+06F, |
---|
701 | 2.66551e+06F, 2.50286e+06F, 2.35014e+06F, 2.20673e+06F, |
---|
702 | 2.07208e+06F, 1.94564e+06F, 1.82692e+06F, 1.71544e+06F, |
---|
703 | 1.61076e+06F, 1.51247e+06F, 1.42018e+06F, 1.33352e+06F, |
---|
704 | 1.25215e+06F, 1.17574e+06F, 1.10400e+06F, 1.03663e+06F, |
---|
705 | 973377.F, 913981.F, 858210.F, 805842.F, |
---|
706 | 756669.F, 710497.F, 667142.F, 626433.F, |
---|
707 | 588208.F, 552316.F, 518613.F, 486967.F, |
---|
708 | 457252.F, 429351.F, 403152.F, 378551.F, |
---|
709 | 355452.F, 333762.F, 313396.F, 294273.F, |
---|
710 | 276316.F, 259455.F, 243623.F, 228757.F, |
---|
711 | 214798.F, 201691.F, 189384.F, 177828.F, |
---|
712 | 166977.F, 156788.F, 147221.F, 138237.F, |
---|
713 | 129802.F, 121881.F, 114444.F, 107461.F, |
---|
714 | 100903.F, 94746.3F, 88964.9F, 83536.2F, |
---|
715 | 78438.8F, 73652.5F, 69158.2F, 64938.1F, |
---|
716 | 60975.6F, 57254.9F, 53761.2F, 50480.6F, |
---|
717 | 47400.3F, 44507.9F, 41792.0F, 39241.9F, |
---|
718 | 36847.3F, 34598.9F, 32487.7F, 30505.3F, |
---|
719 | 28643.8F, 26896.0F, 25254.8F, 23713.7F, |
---|
720 | 22266.7F, 20908.0F, 19632.2F, 18434.2F, |
---|
721 | 17309.4F, 16253.1F, 15261.4F, 14330.1F, |
---|
722 | 13455.7F, 12634.6F, 11863.7F, 11139.7F, |
---|
723 | 10460.0F, 9821.72F, 9222.39F, 8659.64F, |
---|
724 | 8131.23F, 7635.06F, 7169.17F, 6731.70F, |
---|
725 | 6320.93F, 5935.23F, 5573.06F, 5232.99F, |
---|
726 | 4913.67F, 4613.84F, 4332.30F, 4067.94F, |
---|
727 | 3819.72F, 3586.64F, 3367.78F, 3162.28F, |
---|
728 | 2969.31F, 2788.13F, 2617.99F, 2458.24F, |
---|
729 | 2308.24F, 2167.39F, 2035.14F, 1910.95F, |
---|
730 | 1794.35F, 1684.85F, 1582.04F, 1485.51F, |
---|
731 | 1394.86F, 1309.75F, 1229.83F, 1154.78F, |
---|
732 | 1084.32F, 1018.15F, 956.024F, 897.687F, |
---|
733 | 842.910F, 791.475F, 743.179F, 697.830F, |
---|
734 | 655.249F, 615.265F, 577.722F, 542.469F, |
---|
735 | 509.367F, 478.286F, 449.101F, 421.696F, |
---|
736 | 395.964F, 371.803F, 349.115F, 327.812F, |
---|
737 | 307.809F, 289.026F, 271.390F, 254.830F, |
---|
738 | 239.280F, 224.679F, 210.969F, 198.096F, |
---|
739 | 186.008F, 174.658F, 164.000F, 153.993F, |
---|
740 | 144.596F, 135.773F, 127.488F, 119.708F, |
---|
741 | 112.404F, 105.545F, 99.1046F, 93.0572F, |
---|
742 | 87.3788F, 82.0469F, 77.0404F, 72.3394F, |
---|
743 | 67.9252F, 63.7804F, 59.8885F, 56.2341F, |
---|
744 | 52.8027F, 49.5807F, 46.5553F, 43.7144F, |
---|
745 | 41.0470F, 38.5423F, 36.1904F, 33.9821F, |
---|
746 | 31.9085F, 29.9614F, 28.1332F, 26.4165F, |
---|
747 | 24.8045F, 23.2910F, 21.8697F, 20.5352F, |
---|
748 | 19.2822F, 18.1056F, 17.0008F, 15.9634F, |
---|
749 | 14.9893F, 14.0746F, 13.2158F, 12.4094F, |
---|
750 | 11.6522F, 10.9411F, 10.2735F, 9.64662F, |
---|
751 | 9.05798F, 8.50526F, 7.98626F, 7.49894F, |
---|
752 | 7.04135F, 6.61169F, 6.20824F, 5.82941F, |
---|
753 | 5.47370F, 5.13970F, 4.82607F, 4.53158F, |
---|
754 | 4.25507F, 3.99542F, 3.75162F, 3.52269F, |
---|
755 | 3.30774F, 3.10590F, 2.91638F, 2.73842F, |
---|
756 | 2.57132F, 2.41442F, 2.26709F, 2.12875F, |
---|
757 | 1.99885F, 1.87688F, 1.76236F, 1.65482F, |
---|
758 | 1.55384F, 1.45902F, 1.36999F, 1.28640F, |
---|
759 | 1.20790F, 1.13419F, 1.06499F, 1.F |
---|
760 | }; |
---|
761 | |
---|
762 | void _vp_remove_floor(vorbis_look_psy *p, |
---|
763 | float *mdct, |
---|
764 | int *codedflr, |
---|
765 | float *residue, |
---|
766 | int sliding_lowpass){ |
---|
767 | |
---|
768 | int i,n=p->n; |
---|
769 | |
---|
770 | if(sliding_lowpass>n)sliding_lowpass=n; |
---|
771 | |
---|
772 | for(i=0;i<sliding_lowpass;i++){ |
---|
773 | residue[i]= |
---|
774 | mdct[i]*FLOOR1_fromdB_INV_LOOKUP[codedflr[i]]; |
---|
775 | } |
---|
776 | |
---|
777 | for(;i<n;i++) |
---|
778 | residue[i]=0.; |
---|
779 | } |
---|
780 | |
---|
781 | void _vp_noisemask(vorbis_look_psy *p, |
---|
782 | float *logmdct, |
---|
783 | float *logmask){ |
---|
784 | |
---|
785 | int i,n=p->n; |
---|
786 | float *work=alloca(n*sizeof(*work)); |
---|
787 | |
---|
788 | bark_noise_hybridmp(n,p->bark,logmdct,logmask, |
---|
789 | 140.,-1); |
---|
790 | |
---|
791 | for(i=0;i<n;i++)work[i]=logmdct[i]-logmask[i]; |
---|
792 | |
---|
793 | bark_noise_hybridmp(n,p->bark,work,logmask,0., |
---|
794 | p->vi->noisewindowfixed); |
---|
795 | |
---|
796 | for(i=0;i<n;i++)work[i]=logmdct[i]-work[i]; |
---|
797 | |
---|
798 | #if 0 |
---|
799 | { |
---|
800 | static int seq=0; |
---|
801 | |
---|
802 | float work2[n]; |
---|
803 | for(i=0;i<n;i++){ |
---|
804 | work2[i]=logmask[i]+work[i]; |
---|
805 | } |
---|
806 | |
---|
807 | if(seq&1) |
---|
808 | _analysis_output("median2R",seq/2,work,n,1,0,0); |
---|
809 | else |
---|
810 | _analysis_output("median2L",seq/2,work,n,1,0,0); |
---|
811 | |
---|
812 | if(seq&1) |
---|
813 | _analysis_output("envelope2R",seq/2,work2,n,1,0,0); |
---|
814 | else |
---|
815 | _analysis_output("envelope2L",seq/2,work2,n,1,0,0); |
---|
816 | seq++; |
---|
817 | } |
---|
818 | #endif |
---|
819 | |
---|
820 | for(i=0;i<n;i++){ |
---|
821 | int dB=logmask[i]+.5; |
---|
822 | if(dB>=NOISE_COMPAND_LEVELS)dB=NOISE_COMPAND_LEVELS-1; |
---|
823 | if(dB<0)dB=0; |
---|
824 | logmask[i]= work[i]+p->vi->noisecompand[dB]; |
---|
825 | } |
---|
826 | |
---|
827 | } |
---|
828 | |
---|
829 | void _vp_tonemask(vorbis_look_psy *p, |
---|
830 | float *logfft, |
---|
831 | float *logmask, |
---|
832 | float global_specmax, |
---|
833 | float local_specmax){ |
---|
834 | |
---|
835 | int i,n=p->n; |
---|
836 | |
---|
837 | float *seed=alloca(sizeof(*seed)*p->total_octave_lines); |
---|
838 | float att=local_specmax+p->vi->ath_adjatt; |
---|
839 | for(i=0;i<p->total_octave_lines;i++)seed[i]=NEGINF; |
---|
840 | |
---|
841 | /* set the ATH (floating below localmax, not global max by a |
---|
842 | specified att) */ |
---|
843 | if(att<p->vi->ath_maxatt)att=p->vi->ath_maxatt; |
---|
844 | |
---|
845 | for(i=0;i<n;i++) |
---|
846 | logmask[i]=p->ath[i]+att; |
---|
847 | |
---|
848 | /* tone masking */ |
---|
849 | seed_loop(p,(const float ***)p->tonecurves,logfft,logmask,seed,global_specmax); |
---|
850 | max_seeds(p,seed,logmask); |
---|
851 | |
---|
852 | } |
---|
853 | |
---|
854 | void _vp_offset_and_mix(vorbis_look_psy *p, |
---|
855 | float *noise, |
---|
856 | float *tone, |
---|
857 | int offset_select, |
---|
858 | float *logmask, |
---|
859 | float *mdct, |
---|
860 | float *logmdct){ |
---|
861 | int i,n=p->n; |
---|
862 | float de, coeffi, cx;/* AoTuV */ |
---|
863 | float toneatt=p->vi->tone_masteratt[offset_select]; |
---|
864 | |
---|
865 | cx = p->m_val; |
---|
866 | |
---|
867 | for(i=0;i<n;i++){ |
---|
868 | float val= noise[i]+p->noiseoffset[offset_select][i]; |
---|
869 | if(val>p->vi->noisemaxsupp)val=p->vi->noisemaxsupp; |
---|
870 | logmask[i]=max(val,tone[i]+toneatt); |
---|
871 | |
---|
872 | |
---|
873 | /* AoTuV */ |
---|
874 | /** @ M1 ** |
---|
875 | The following codes improve a noise problem. |
---|
876 | A fundamental idea uses the value of masking and carries out |
---|
877 | the relative compensation of the MDCT. |
---|
878 | However, this code is not perfect and all noise problems cannot be solved. |
---|
879 | by Aoyumi @ 2004/04/18 |
---|
880 | */ |
---|
881 | |
---|
882 | if(offset_select == 1) { |
---|
883 | coeffi = -17.2; /* coeffi is a -17.2dB threshold */ |
---|
884 | val = val - logmdct[i]; /* val == mdct line value relative to floor in dB */ |
---|
885 | |
---|
886 | if(val > coeffi){ |
---|
887 | /* mdct value is > -17.2 dB below floor */ |
---|
888 | |
---|
889 | de = 1.0-((val-coeffi)*0.005*cx); |
---|
890 | /* pro-rated attenuation: |
---|
891 | -0.00 dB boost if mdct value is -17.2dB (relative to floor) |
---|
892 | -0.77 dB boost if mdct value is 0dB (relative to floor) |
---|
893 | -1.64 dB boost if mdct value is +17.2dB (relative to floor) |
---|
894 | etc... */ |
---|
895 | |
---|
896 | if(de < 0) de = 0.0001; |
---|
897 | }else |
---|
898 | /* mdct value is <= -17.2 dB below floor */ |
---|
899 | |
---|
900 | de = 1.0-((val-coeffi)*0.0003*cx); |
---|
901 | /* pro-rated attenuation: |
---|
902 | +0.00 dB atten if mdct value is -17.2dB (relative to floor) |
---|
903 | +0.45 dB atten if mdct value is -34.4dB (relative to floor) |
---|
904 | etc... */ |
---|
905 | |
---|
906 | mdct[i] *= de; |
---|
907 | |
---|
908 | } |
---|
909 | } |
---|
910 | } |
---|
911 | |
---|
912 | float _vp_ampmax_decay(float amp,vorbis_dsp_state *vd){ |
---|
913 | vorbis_info *vi=vd->vi; |
---|
914 | codec_setup_info *ci=vi->codec_setup; |
---|
915 | vorbis_info_psy_global *gi=&ci->psy_g_param; |
---|
916 | |
---|
917 | int n=ci->blocksizes[vd->W]/2; |
---|
918 | float secs=(float)n/vi->rate; |
---|
919 | |
---|
920 | amp+=secs*gi->ampmax_att_per_sec; |
---|
921 | if(amp<-9999)amp=-9999; |
---|
922 | return(amp); |
---|
923 | } |
---|
924 | |
---|
925 | static void couple_lossless(float A, float B, |
---|
926 | float *qA, float *qB){ |
---|
927 | int test1=fabs(*qA)>fabs(*qB); |
---|
928 | test1-= fabs(*qA)<fabs(*qB); |
---|
929 | |
---|
930 | if(!test1)test1=((fabs(A)>fabs(B))<<1)-1; |
---|
931 | if(test1==1){ |
---|
932 | *qB=(*qA>0.f?*qA-*qB:*qB-*qA); |
---|
933 | }else{ |
---|
934 | float temp=*qB; |
---|
935 | *qB=(*qB>0.f?*qA-*qB:*qB-*qA); |
---|
936 | *qA=temp; |
---|
937 | } |
---|
938 | |
---|
939 | if(*qB>fabs(*qA)*1.9999f){ |
---|
940 | *qB= -fabs(*qA)*2.f; |
---|
941 | *qA= -*qA; |
---|
942 | } |
---|
943 | } |
---|
944 | |
---|
945 | static float hypot_lookup[32]={ |
---|
946 | -0.009935, -0.011245, -0.012726, -0.014397, |
---|
947 | -0.016282, -0.018407, -0.020800, -0.023494, |
---|
948 | -0.026522, -0.029923, -0.033737, -0.038010, |
---|
949 | -0.042787, -0.048121, -0.054064, -0.060671, |
---|
950 | -0.068000, -0.076109, -0.085054, -0.094892, |
---|
951 | -0.105675, -0.117451, -0.130260, -0.144134, |
---|
952 | -0.159093, -0.175146, -0.192286, -0.210490, |
---|
953 | -0.229718, -0.249913, -0.271001, -0.292893}; |
---|
954 | |
---|
955 | static void precomputed_couple_point(float premag, |
---|
956 | int floorA,int floorB, |
---|
957 | float *mag, float *ang){ |
---|
958 | |
---|
959 | int test=(floorA>floorB)-1; |
---|
960 | int offset=31-abs(floorA-floorB); |
---|
961 | float floormag=hypot_lookup[((offset<0)-1)&offset]+1.f; |
---|
962 | |
---|
963 | floormag*=FLOOR1_fromdB_INV_LOOKUP[(floorB&test)|(floorA&(~test))]; |
---|
964 | |
---|
965 | *mag=premag*floormag; |
---|
966 | *ang=0.f; |
---|
967 | } |
---|
968 | |
---|
969 | /* just like below, this is currently set up to only do |
---|
970 | single-step-depth coupling. Otherwise, we'd have to do more |
---|
971 | copying (which will be inevitable later) */ |
---|
972 | |
---|
973 | /* doing the real circular magnitude calculation is audibly superior |
---|
974 | to (A+B)/sqrt(2) */ |
---|
975 | static float dipole_hypot(float a, float b){ |
---|
976 | if(a>0.){ |
---|
977 | if(b>0.)return sqrt(a*a+b*b); |
---|
978 | if(a>-b)return sqrt(a*a-b*b); |
---|
979 | return -sqrt(b*b-a*a); |
---|
980 | } |
---|
981 | if(b<0.)return -sqrt(a*a+b*b); |
---|
982 | if(-a>b)return -sqrt(a*a-b*b); |
---|
983 | return sqrt(b*b-a*a); |
---|
984 | } |
---|
985 | static float round_hypot(float a, float b){ |
---|
986 | if(a>0.){ |
---|
987 | if(b>0.)return sqrt(a*a+b*b); |
---|
988 | if(a>-b)return sqrt(a*a+b*b); |
---|
989 | return -sqrt(b*b+a*a); |
---|
990 | } |
---|
991 | if(b<0.)return -sqrt(a*a+b*b); |
---|
992 | if(-a>b)return -sqrt(a*a+b*b); |
---|
993 | return sqrt(b*b+a*a); |
---|
994 | } |
---|
995 | |
---|
996 | /* revert to round hypot for now */ |
---|
997 | float **_vp_quantize_couple_memo(vorbis_block *vb, |
---|
998 | vorbis_info_psy_global *g, |
---|
999 | vorbis_look_psy *p, |
---|
1000 | vorbis_info_mapping0 *vi, |
---|
1001 | float **mdct){ |
---|
1002 | |
---|
1003 | int i,j,n=p->n; |
---|
1004 | float **ret=_vorbis_block_alloc(vb,vi->coupling_steps*sizeof(*ret)); |
---|
1005 | int limit=g->coupling_pointlimit[p->vi->blockflag][PACKETBLOBS/2]; |
---|
1006 | |
---|
1007 | for(i=0;i<vi->coupling_steps;i++){ |
---|
1008 | float *mdctM=mdct[vi->coupling_mag[i]]; |
---|
1009 | float *mdctA=mdct[vi->coupling_ang[i]]; |
---|
1010 | ret[i]=_vorbis_block_alloc(vb,n*sizeof(**ret)); |
---|
1011 | for(j=0;j<limit;j++) |
---|
1012 | ret[i][j]=dipole_hypot(mdctM[j],mdctA[j]); |
---|
1013 | for(;j<n;j++) |
---|
1014 | ret[i][j]=round_hypot(mdctM[j],mdctA[j]); |
---|
1015 | } |
---|
1016 | |
---|
1017 | return(ret); |
---|
1018 | } |
---|
1019 | |
---|
1020 | /* this is for per-channel noise normalization */ |
---|
1021 | static int apsort(const void *a, const void *b){ |
---|
1022 | float f1=fabs(**(float**)a); |
---|
1023 | float f2=fabs(**(float**)b); |
---|
1024 | return (f1<f2)-(f1>f2); |
---|
1025 | } |
---|
1026 | |
---|
1027 | int **_vp_quantize_couple_sort(vorbis_block *vb, |
---|
1028 | vorbis_look_psy *p, |
---|
1029 | vorbis_info_mapping0 *vi, |
---|
1030 | float **mags){ |
---|
1031 | |
---|
1032 | |
---|
1033 | if(p->vi->normal_point_p){ |
---|
1034 | int i,j,k,n=p->n; |
---|
1035 | int **ret=_vorbis_block_alloc(vb,vi->coupling_steps*sizeof(*ret)); |
---|
1036 | int partition=p->vi->normal_partition; |
---|
1037 | float **work=alloca(sizeof(*work)*partition); |
---|
1038 | |
---|
1039 | for(i=0;i<vi->coupling_steps;i++){ |
---|
1040 | ret[i]=_vorbis_block_alloc(vb,n*sizeof(**ret)); |
---|
1041 | |
---|
1042 | for(j=0;j<n;j+=partition){ |
---|
1043 | for(k=0;k<partition;k++)work[k]=mags[i]+k+j; |
---|
1044 | qsort(work,partition,sizeof(*work),apsort); |
---|
1045 | for(k=0;k<partition;k++)ret[i][k+j]=work[k]-mags[i]; |
---|
1046 | } |
---|
1047 | } |
---|
1048 | return(ret); |
---|
1049 | } |
---|
1050 | return(NULL); |
---|
1051 | } |
---|
1052 | |
---|
1053 | void _vp_noise_normalize_sort(vorbis_look_psy *p, |
---|
1054 | float *magnitudes,int *sortedindex){ |
---|
1055 | int i,j,n=p->n; |
---|
1056 | vorbis_info_psy *vi=p->vi; |
---|
1057 | int partition=vi->normal_partition; |
---|
1058 | float **work=alloca(sizeof(*work)*partition); |
---|
1059 | int start=vi->normal_start; |
---|
1060 | |
---|
1061 | for(j=start;j<n;j+=partition){ |
---|
1062 | if(j+partition>n)partition=n-j; |
---|
1063 | for(i=0;i<partition;i++)work[i]=magnitudes+i+j; |
---|
1064 | qsort(work,partition,sizeof(*work),apsort); |
---|
1065 | for(i=0;i<partition;i++){ |
---|
1066 | sortedindex[i+j-start]=work[i]-magnitudes; |
---|
1067 | } |
---|
1068 | } |
---|
1069 | } |
---|
1070 | |
---|
1071 | void _vp_noise_normalize(vorbis_look_psy *p, |
---|
1072 | float *in,float *out,int *sortedindex){ |
---|
1073 | int flag=0,i,j=0,n=p->n; |
---|
1074 | vorbis_info_psy *vi=p->vi; |
---|
1075 | int partition=vi->normal_partition; |
---|
1076 | int start=vi->normal_start; |
---|
1077 | |
---|
1078 | if(start>n)start=n; |
---|
1079 | |
---|
1080 | if(vi->normal_channel_p){ |
---|
1081 | for(;j<start;j++) |
---|
1082 | out[j]=rint(in[j]); |
---|
1083 | |
---|
1084 | for(;j+partition<=n;j+=partition){ |
---|
1085 | float acc=0.; |
---|
1086 | int k; |
---|
1087 | |
---|
1088 | for(i=j;i<j+partition;i++) |
---|
1089 | acc+=in[i]*in[i]; |
---|
1090 | |
---|
1091 | for(i=0;i<partition;i++){ |
---|
1092 | k=sortedindex[i+j-start]; |
---|
1093 | |
---|
1094 | if(in[k]*in[k]>=.25f){ |
---|
1095 | out[k]=rint(in[k]); |
---|
1096 | acc-=in[k]*in[k]; |
---|
1097 | flag=1; |
---|
1098 | }else{ |
---|
1099 | if(acc<vi->normal_thresh)break; |
---|
1100 | out[k]=unitnorm(in[k]); |
---|
1101 | acc-=1.; |
---|
1102 | } |
---|
1103 | } |
---|
1104 | |
---|
1105 | for(;i<partition;i++){ |
---|
1106 | k=sortedindex[i+j-start]; |
---|
1107 | out[k]=0.; |
---|
1108 | } |
---|
1109 | } |
---|
1110 | } |
---|
1111 | |
---|
1112 | for(;j<n;j++) |
---|
1113 | out[j]=rint(in[j]); |
---|
1114 | |
---|
1115 | } |
---|
1116 | |
---|
1117 | void _vp_couple(int blobno, |
---|
1118 | vorbis_info_psy_global *g, |
---|
1119 | vorbis_look_psy *p, |
---|
1120 | vorbis_info_mapping0 *vi, |
---|
1121 | float **res, |
---|
1122 | float **mag_memo, |
---|
1123 | int **mag_sort, |
---|
1124 | int **ifloor, |
---|
1125 | int *nonzero, |
---|
1126 | int sliding_lowpass){ |
---|
1127 | |
---|
1128 | int i,j,k,n=p->n; |
---|
1129 | |
---|
1130 | /* perform any requested channel coupling */ |
---|
1131 | /* point stereo can only be used in a first stage (in this encoder) |
---|
1132 | because of the dependency on floor lookups */ |
---|
1133 | for(i=0;i<vi->coupling_steps;i++){ |
---|
1134 | |
---|
1135 | /* once we're doing multistage coupling in which a channel goes |
---|
1136 | through more than one coupling step, the floor vector |
---|
1137 | magnitudes will also have to be recalculated an propogated |
---|
1138 | along with PCM. Right now, we're not (that will wait until 5.1 |
---|
1139 | most likely), so the code isn't here yet. The memory management |
---|
1140 | here is all assuming single depth couplings anyway. */ |
---|
1141 | |
---|
1142 | /* make sure coupling a zero and a nonzero channel results in two |
---|
1143 | nonzero channels. */ |
---|
1144 | if(nonzero[vi->coupling_mag[i]] || |
---|
1145 | nonzero[vi->coupling_ang[i]]){ |
---|
1146 | |
---|
1147 | |
---|
1148 | float *rM=res[vi->coupling_mag[i]]; |
---|
1149 | float *rA=res[vi->coupling_ang[i]]; |
---|
1150 | float *qM=rM+n; |
---|
1151 | float *qA=rA+n; |
---|
1152 | int *floorM=ifloor[vi->coupling_mag[i]]; |
---|
1153 | int *floorA=ifloor[vi->coupling_ang[i]]; |
---|
1154 | float prepoint=stereo_threshholds[g->coupling_prepointamp[blobno]]; |
---|
1155 | float postpoint=stereo_threshholds[g->coupling_postpointamp[blobno]]; |
---|
1156 | int partition=(p->vi->normal_point_p?p->vi->normal_partition:p->n); |
---|
1157 | int limit=g->coupling_pointlimit[p->vi->blockflag][blobno]; |
---|
1158 | int pointlimit=limit; |
---|
1159 | |
---|
1160 | nonzero[vi->coupling_mag[i]]=1; |
---|
1161 | nonzero[vi->coupling_ang[i]]=1; |
---|
1162 | |
---|
1163 | /* The threshold of a stereo is changed with the size of n */ |
---|
1164 | if(n > 1000) |
---|
1165 | postpoint=stereo_threshholds_limited[g->coupling_postpointamp[blobno]]; |
---|
1166 | |
---|
1167 | for(j=0;j<p->n;j+=partition){ |
---|
1168 | float acc=0.f; |
---|
1169 | |
---|
1170 | for(k=0;k<partition;k++){ |
---|
1171 | int l=k+j; |
---|
1172 | |
---|
1173 | if(l<sliding_lowpass){ |
---|
1174 | if((l>=limit && fabs(rM[l])<postpoint && fabs(rA[l])<postpoint) || |
---|
1175 | (fabs(rM[l])<prepoint && fabs(rA[l])<prepoint)){ |
---|
1176 | |
---|
1177 | |
---|
1178 | precomputed_couple_point(mag_memo[i][l], |
---|
1179 | floorM[l],floorA[l], |
---|
1180 | qM+l,qA+l); |
---|
1181 | |
---|
1182 | if(rint(qM[l])==0.f)acc+=qM[l]*qM[l]; |
---|
1183 | }else{ |
---|
1184 | couple_lossless(rM[l],rA[l],qM+l,qA+l); |
---|
1185 | } |
---|
1186 | }else{ |
---|
1187 | qM[l]=0.; |
---|
1188 | qA[l]=0.; |
---|
1189 | } |
---|
1190 | } |
---|
1191 | |
---|
1192 | if(p->vi->normal_point_p){ |
---|
1193 | for(k=0;k<partition && acc>=p->vi->normal_thresh;k++){ |
---|
1194 | int l=mag_sort[i][j+k]; |
---|
1195 | if(l<sliding_lowpass && l>=pointlimit && rint(qM[l])==0.f){ |
---|
1196 | qM[l]=unitnorm(qM[l]); |
---|
1197 | acc-=1.f; |
---|
1198 | } |
---|
1199 | } |
---|
1200 | } |
---|
1201 | } |
---|
1202 | } |
---|
1203 | } |
---|
1204 | } |
---|
1205 | |
---|
1206 | /* AoTuV */ |
---|
1207 | /** @ M2 ** |
---|
1208 | The boost problem by the combination of noise normalization and point stereo is eased. |
---|
1209 | However, this is a temporary patch. |
---|
1210 | by Aoyumi @ 2004/04/18 |
---|
1211 | */ |
---|
1212 | |
---|
1213 | void hf_reduction(vorbis_info_psy_global *g, |
---|
1214 | vorbis_look_psy *p, |
---|
1215 | vorbis_info_mapping0 *vi, |
---|
1216 | float **mdct){ |
---|
1217 | |
---|
1218 | int i,j,n=p->n, de=0.3*p->m_val; |
---|
1219 | int limit=g->coupling_pointlimit[p->vi->blockflag][PACKETBLOBS/2]; |
---|
1220 | int start=p->vi->normal_start; |
---|
1221 | |
---|
1222 | for(i=0; i<vi->coupling_steps; i++){ |
---|
1223 | /* for(j=start; j<limit; j++){} // ???*/ |
---|
1224 | for(j=limit; j<n; j++) |
---|
1225 | mdct[i][j] *= (1.0 - de*((float)(j-limit) / (float)(n-limit))); |
---|
1226 | } |
---|
1227 | } |
---|