annotate src/libvorbis-1.3.3/lib/psy.c @ 168:ceec0dd9ec9c

Replace these with versions built using an older toolset (so as to avoid ABI compatibilities when linking on Ubuntu 14.04 for packaging purposes)
author Chris Cannam <cannam@all-day-breakfast.com>
date Fri, 07 Feb 2020 11:51:13 +0000
parents 98c1576536ae
children
rev   line source
cannam@86 1 /********************************************************************
cannam@86 2 * *
cannam@86 3 * THIS FILE IS PART OF THE OggVorbis SOFTWARE CODEC SOURCE CODE. *
cannam@86 4 * USE, DISTRIBUTION AND REPRODUCTION OF THIS LIBRARY SOURCE IS *
cannam@86 5 * GOVERNED BY A BSD-STYLE SOURCE LICENSE INCLUDED WITH THIS SOURCE *
cannam@86 6 * IN 'COPYING'. PLEASE READ THESE TERMS BEFORE DISTRIBUTING. *
cannam@86 7 * *
cannam@86 8 * THE OggVorbis SOURCE CODE IS (C) COPYRIGHT 1994-2010 *
cannam@86 9 * by the Xiph.Org Foundation http://www.xiph.org/ *
cannam@86 10 * *
cannam@86 11 ********************************************************************
cannam@86 12
cannam@86 13 function: psychoacoustics not including preecho
cannam@86 14 last mod: $Id: psy.c 18077 2011-09-02 02:49:00Z giles $
cannam@86 15
cannam@86 16 ********************************************************************/
cannam@86 17
cannam@86 18 #include <stdlib.h>
cannam@86 19 #include <math.h>
cannam@86 20 #include <string.h>
cannam@86 21 #include "vorbis/codec.h"
cannam@86 22 #include "codec_internal.h"
cannam@86 23
cannam@86 24 #include "masking.h"
cannam@86 25 #include "psy.h"
cannam@86 26 #include "os.h"
cannam@86 27 #include "lpc.h"
cannam@86 28 #include "smallft.h"
cannam@86 29 #include "scales.h"
cannam@86 30 #include "misc.h"
cannam@86 31
cannam@86 32 #define NEGINF -9999.f
cannam@86 33 static const double stereo_threshholds[]={0.0, .5, 1.0, 1.5, 2.5, 4.5, 8.5, 16.5, 9e10};
cannam@86 34 static const double stereo_threshholds_limited[]={0.0, .5, 1.0, 1.5, 2.0, 2.5, 4.5, 8.5, 9e10};
cannam@86 35
cannam@86 36 vorbis_look_psy_global *_vp_global_look(vorbis_info *vi){
cannam@86 37 codec_setup_info *ci=vi->codec_setup;
cannam@86 38 vorbis_info_psy_global *gi=&ci->psy_g_param;
cannam@86 39 vorbis_look_psy_global *look=_ogg_calloc(1,sizeof(*look));
cannam@86 40
cannam@86 41 look->channels=vi->channels;
cannam@86 42
cannam@86 43 look->ampmax=-9999.;
cannam@86 44 look->gi=gi;
cannam@86 45 return(look);
cannam@86 46 }
cannam@86 47
cannam@86 48 void _vp_global_free(vorbis_look_psy_global *look){
cannam@86 49 if(look){
cannam@86 50 memset(look,0,sizeof(*look));
cannam@86 51 _ogg_free(look);
cannam@86 52 }
cannam@86 53 }
cannam@86 54
cannam@86 55 void _vi_gpsy_free(vorbis_info_psy_global *i){
cannam@86 56 if(i){
cannam@86 57 memset(i,0,sizeof(*i));
cannam@86 58 _ogg_free(i);
cannam@86 59 }
cannam@86 60 }
cannam@86 61
cannam@86 62 void _vi_psy_free(vorbis_info_psy *i){
cannam@86 63 if(i){
cannam@86 64 memset(i,0,sizeof(*i));
cannam@86 65 _ogg_free(i);
cannam@86 66 }
cannam@86 67 }
cannam@86 68
cannam@86 69 static void min_curve(float *c,
cannam@86 70 float *c2){
cannam@86 71 int i;
cannam@86 72 for(i=0;i<EHMER_MAX;i++)if(c2[i]<c[i])c[i]=c2[i];
cannam@86 73 }
cannam@86 74 static void max_curve(float *c,
cannam@86 75 float *c2){
cannam@86 76 int i;
cannam@86 77 for(i=0;i<EHMER_MAX;i++)if(c2[i]>c[i])c[i]=c2[i];
cannam@86 78 }
cannam@86 79
cannam@86 80 static void attenuate_curve(float *c,float att){
cannam@86 81 int i;
cannam@86 82 for(i=0;i<EHMER_MAX;i++)
cannam@86 83 c[i]+=att;
cannam@86 84 }
cannam@86 85
cannam@86 86 static float ***setup_tone_curves(float curveatt_dB[P_BANDS],float binHz,int n,
cannam@86 87 float center_boost, float center_decay_rate){
cannam@86 88 int i,j,k,m;
cannam@86 89 float ath[EHMER_MAX];
cannam@86 90 float workc[P_BANDS][P_LEVELS][EHMER_MAX];
cannam@86 91 float athc[P_LEVELS][EHMER_MAX];
cannam@86 92 float *brute_buffer=alloca(n*sizeof(*brute_buffer));
cannam@86 93
cannam@86 94 float ***ret=_ogg_malloc(sizeof(*ret)*P_BANDS);
cannam@86 95
cannam@86 96 memset(workc,0,sizeof(workc));
cannam@86 97
cannam@86 98 for(i=0;i<P_BANDS;i++){
cannam@86 99 /* we add back in the ATH to avoid low level curves falling off to
cannam@86 100 -infinity and unnecessarily cutting off high level curves in the
cannam@86 101 curve limiting (last step). */
cannam@86 102
cannam@86 103 /* A half-band's settings must be valid over the whole band, and
cannam@86 104 it's better to mask too little than too much */
cannam@86 105 int ath_offset=i*4;
cannam@86 106 for(j=0;j<EHMER_MAX;j++){
cannam@86 107 float min=999.;
cannam@86 108 for(k=0;k<4;k++)
cannam@86 109 if(j+k+ath_offset<MAX_ATH){
cannam@86 110 if(min>ATH[j+k+ath_offset])min=ATH[j+k+ath_offset];
cannam@86 111 }else{
cannam@86 112 if(min>ATH[MAX_ATH-1])min=ATH[MAX_ATH-1];
cannam@86 113 }
cannam@86 114 ath[j]=min;
cannam@86 115 }
cannam@86 116
cannam@86 117 /* copy curves into working space, replicate the 50dB curve to 30
cannam@86 118 and 40, replicate the 100dB curve to 110 */
cannam@86 119 for(j=0;j<6;j++)
cannam@86 120 memcpy(workc[i][j+2],tonemasks[i][j],EHMER_MAX*sizeof(*tonemasks[i][j]));
cannam@86 121 memcpy(workc[i][0],tonemasks[i][0],EHMER_MAX*sizeof(*tonemasks[i][0]));
cannam@86 122 memcpy(workc[i][1],tonemasks[i][0],EHMER_MAX*sizeof(*tonemasks[i][0]));
cannam@86 123
cannam@86 124 /* apply centered curve boost/decay */
cannam@86 125 for(j=0;j<P_LEVELS;j++){
cannam@86 126 for(k=0;k<EHMER_MAX;k++){
cannam@86 127 float adj=center_boost+abs(EHMER_OFFSET-k)*center_decay_rate;
cannam@86 128 if(adj<0. && center_boost>0)adj=0.;
cannam@86 129 if(adj>0. && center_boost<0)adj=0.;
cannam@86 130 workc[i][j][k]+=adj;
cannam@86 131 }
cannam@86 132 }
cannam@86 133
cannam@86 134 /* normalize curves so the driving amplitude is 0dB */
cannam@86 135 /* make temp curves with the ATH overlayed */
cannam@86 136 for(j=0;j<P_LEVELS;j++){
cannam@86 137 attenuate_curve(workc[i][j],curveatt_dB[i]+100.-(j<2?2:j)*10.-P_LEVEL_0);
cannam@86 138 memcpy(athc[j],ath,EHMER_MAX*sizeof(**athc));
cannam@86 139 attenuate_curve(athc[j],+100.-j*10.f-P_LEVEL_0);
cannam@86 140 max_curve(athc[j],workc[i][j]);
cannam@86 141 }
cannam@86 142
cannam@86 143 /* Now limit the louder curves.
cannam@86 144
cannam@86 145 the idea is this: We don't know what the playback attenuation
cannam@86 146 will be; 0dB SL moves every time the user twiddles the volume
cannam@86 147 knob. So that means we have to use a single 'most pessimal' curve
cannam@86 148 for all masking amplitudes, right? Wrong. The *loudest* sound
cannam@86 149 can be in (we assume) a range of ...+100dB] SL. However, sounds
cannam@86 150 20dB down will be in a range ...+80], 40dB down is from ...+60],
cannam@86 151 etc... */
cannam@86 152
cannam@86 153 for(j=1;j<P_LEVELS;j++){
cannam@86 154 min_curve(athc[j],athc[j-1]);
cannam@86 155 min_curve(workc[i][j],athc[j]);
cannam@86 156 }
cannam@86 157 }
cannam@86 158
cannam@86 159 for(i=0;i<P_BANDS;i++){
cannam@86 160 int hi_curve,lo_curve,bin;
cannam@86 161 ret[i]=_ogg_malloc(sizeof(**ret)*P_LEVELS);
cannam@86 162
cannam@86 163 /* low frequency curves are measured with greater resolution than
cannam@86 164 the MDCT/FFT will actually give us; we want the curve applied
cannam@86 165 to the tone data to be pessimistic and thus apply the minimum
cannam@86 166 masking possible for a given bin. That means that a single bin
cannam@86 167 could span more than one octave and that the curve will be a
cannam@86 168 composite of multiple octaves. It also may mean that a single
cannam@86 169 bin may span > an eighth of an octave and that the eighth
cannam@86 170 octave values may also be composited. */
cannam@86 171
cannam@86 172 /* which octave curves will we be compositing? */
cannam@86 173 bin=floor(fromOC(i*.5)/binHz);
cannam@86 174 lo_curve= ceil(toOC(bin*binHz+1)*2);
cannam@86 175 hi_curve= floor(toOC((bin+1)*binHz)*2);
cannam@86 176 if(lo_curve>i)lo_curve=i;
cannam@86 177 if(lo_curve<0)lo_curve=0;
cannam@86 178 if(hi_curve>=P_BANDS)hi_curve=P_BANDS-1;
cannam@86 179
cannam@86 180 for(m=0;m<P_LEVELS;m++){
cannam@86 181 ret[i][m]=_ogg_malloc(sizeof(***ret)*(EHMER_MAX+2));
cannam@86 182
cannam@86 183 for(j=0;j<n;j++)brute_buffer[j]=999.;
cannam@86 184
cannam@86 185 /* render the curve into bins, then pull values back into curve.
cannam@86 186 The point is that any inherent subsampling aliasing results in
cannam@86 187 a safe minimum */
cannam@86 188 for(k=lo_curve;k<=hi_curve;k++){
cannam@86 189 int l=0;
cannam@86 190
cannam@86 191 for(j=0;j<EHMER_MAX;j++){
cannam@86 192 int lo_bin= fromOC(j*.125+k*.5-2.0625)/binHz;
cannam@86 193 int hi_bin= fromOC(j*.125+k*.5-1.9375)/binHz+1;
cannam@86 194
cannam@86 195 if(lo_bin<0)lo_bin=0;
cannam@86 196 if(lo_bin>n)lo_bin=n;
cannam@86 197 if(lo_bin<l)l=lo_bin;
cannam@86 198 if(hi_bin<0)hi_bin=0;
cannam@86 199 if(hi_bin>n)hi_bin=n;
cannam@86 200
cannam@86 201 for(;l<hi_bin && l<n;l++)
cannam@86 202 if(brute_buffer[l]>workc[k][m][j])
cannam@86 203 brute_buffer[l]=workc[k][m][j];
cannam@86 204 }
cannam@86 205
cannam@86 206 for(;l<n;l++)
cannam@86 207 if(brute_buffer[l]>workc[k][m][EHMER_MAX-1])
cannam@86 208 brute_buffer[l]=workc[k][m][EHMER_MAX-1];
cannam@86 209
cannam@86 210 }
cannam@86 211
cannam@86 212 /* be equally paranoid about being valid up to next half ocatve */
cannam@86 213 if(i+1<P_BANDS){
cannam@86 214 int l=0;
cannam@86 215 k=i+1;
cannam@86 216 for(j=0;j<EHMER_MAX;j++){
cannam@86 217 int lo_bin= fromOC(j*.125+i*.5-2.0625)/binHz;
cannam@86 218 int hi_bin= fromOC(j*.125+i*.5-1.9375)/binHz+1;
cannam@86 219
cannam@86 220 if(lo_bin<0)lo_bin=0;
cannam@86 221 if(lo_bin>n)lo_bin=n;
cannam@86 222 if(lo_bin<l)l=lo_bin;
cannam@86 223 if(hi_bin<0)hi_bin=0;
cannam@86 224 if(hi_bin>n)hi_bin=n;
cannam@86 225
cannam@86 226 for(;l<hi_bin && l<n;l++)
cannam@86 227 if(brute_buffer[l]>workc[k][m][j])
cannam@86 228 brute_buffer[l]=workc[k][m][j];
cannam@86 229 }
cannam@86 230
cannam@86 231 for(;l<n;l++)
cannam@86 232 if(brute_buffer[l]>workc[k][m][EHMER_MAX-1])
cannam@86 233 brute_buffer[l]=workc[k][m][EHMER_MAX-1];
cannam@86 234
cannam@86 235 }
cannam@86 236
cannam@86 237
cannam@86 238 for(j=0;j<EHMER_MAX;j++){
cannam@86 239 int bin=fromOC(j*.125+i*.5-2.)/binHz;
cannam@86 240 if(bin<0){
cannam@86 241 ret[i][m][j+2]=-999.;
cannam@86 242 }else{
cannam@86 243 if(bin>=n){
cannam@86 244 ret[i][m][j+2]=-999.;
cannam@86 245 }else{
cannam@86 246 ret[i][m][j+2]=brute_buffer[bin];
cannam@86 247 }
cannam@86 248 }
cannam@86 249 }
cannam@86 250
cannam@86 251 /* add fenceposts */
cannam@86 252 for(j=0;j<EHMER_OFFSET;j++)
cannam@86 253 if(ret[i][m][j+2]>-200.f)break;
cannam@86 254 ret[i][m][0]=j;
cannam@86 255
cannam@86 256 for(j=EHMER_MAX-1;j>EHMER_OFFSET+1;j--)
cannam@86 257 if(ret[i][m][j+2]>-200.f)
cannam@86 258 break;
cannam@86 259 ret[i][m][1]=j;
cannam@86 260
cannam@86 261 }
cannam@86 262 }
cannam@86 263
cannam@86 264 return(ret);
cannam@86 265 }
cannam@86 266
cannam@86 267 void _vp_psy_init(vorbis_look_psy *p,vorbis_info_psy *vi,
cannam@86 268 vorbis_info_psy_global *gi,int n,long rate){
cannam@86 269 long i,j,lo=-99,hi=1;
cannam@86 270 long maxoc;
cannam@86 271 memset(p,0,sizeof(*p));
cannam@86 272
cannam@86 273 p->eighth_octave_lines=gi->eighth_octave_lines;
cannam@86 274 p->shiftoc=rint(log(gi->eighth_octave_lines*8.f)/log(2.f))-1;
cannam@86 275
cannam@86 276 p->firstoc=toOC(.25f*rate*.5/n)*(1<<(p->shiftoc+1))-gi->eighth_octave_lines;
cannam@86 277 maxoc=toOC((n+.25f)*rate*.5/n)*(1<<(p->shiftoc+1))+.5f;
cannam@86 278 p->total_octave_lines=maxoc-p->firstoc+1;
cannam@86 279 p->ath=_ogg_malloc(n*sizeof(*p->ath));
cannam@86 280
cannam@86 281 p->octave=_ogg_malloc(n*sizeof(*p->octave));
cannam@86 282 p->bark=_ogg_malloc(n*sizeof(*p->bark));
cannam@86 283 p->vi=vi;
cannam@86 284 p->n=n;
cannam@86 285 p->rate=rate;
cannam@86 286
cannam@86 287 /* AoTuV HF weighting */
cannam@86 288 p->m_val = 1.;
cannam@86 289 if(rate < 26000) p->m_val = 0;
cannam@86 290 else if(rate < 38000) p->m_val = .94; /* 32kHz */
cannam@86 291 else if(rate > 46000) p->m_val = 1.275; /* 48kHz */
cannam@86 292
cannam@86 293 /* set up the lookups for a given blocksize and sample rate */
cannam@86 294
cannam@86 295 for(i=0,j=0;i<MAX_ATH-1;i++){
cannam@86 296 int endpos=rint(fromOC((i+1)*.125-2.)*2*n/rate);
cannam@86 297 float base=ATH[i];
cannam@86 298 if(j<endpos){
cannam@86 299 float delta=(ATH[i+1]-base)/(endpos-j);
cannam@86 300 for(;j<endpos && j<n;j++){
cannam@86 301 p->ath[j]=base+100.;
cannam@86 302 base+=delta;
cannam@86 303 }
cannam@86 304 }
cannam@86 305 }
cannam@86 306
cannam@86 307 for(;j<n;j++){
cannam@86 308 p->ath[j]=p->ath[j-1];
cannam@86 309 }
cannam@86 310
cannam@86 311 for(i=0;i<n;i++){
cannam@86 312 float bark=toBARK(rate/(2*n)*i);
cannam@86 313
cannam@86 314 for(;lo+vi->noisewindowlomin<i &&
cannam@86 315 toBARK(rate/(2*n)*lo)<(bark-vi->noisewindowlo);lo++);
cannam@86 316
cannam@86 317 for(;hi<=n && (hi<i+vi->noisewindowhimin ||
cannam@86 318 toBARK(rate/(2*n)*hi)<(bark+vi->noisewindowhi));hi++);
cannam@86 319
cannam@86 320 p->bark[i]=((lo-1)<<16)+(hi-1);
cannam@86 321
cannam@86 322 }
cannam@86 323
cannam@86 324 for(i=0;i<n;i++)
cannam@86 325 p->octave[i]=toOC((i+.25f)*.5*rate/n)*(1<<(p->shiftoc+1))+.5f;
cannam@86 326
cannam@86 327 p->tonecurves=setup_tone_curves(vi->toneatt,rate*.5/n,n,
cannam@86 328 vi->tone_centerboost,vi->tone_decay);
cannam@86 329
cannam@86 330 /* set up rolling noise median */
cannam@86 331 p->noiseoffset=_ogg_malloc(P_NOISECURVES*sizeof(*p->noiseoffset));
cannam@86 332 for(i=0;i<P_NOISECURVES;i++)
cannam@86 333 p->noiseoffset[i]=_ogg_malloc(n*sizeof(**p->noiseoffset));
cannam@86 334
cannam@86 335 for(i=0;i<n;i++){
cannam@86 336 float halfoc=toOC((i+.5)*rate/(2.*n))*2.;
cannam@86 337 int inthalfoc;
cannam@86 338 float del;
cannam@86 339
cannam@86 340 if(halfoc<0)halfoc=0;
cannam@86 341 if(halfoc>=P_BANDS-1)halfoc=P_BANDS-1;
cannam@86 342 inthalfoc=(int)halfoc;
cannam@86 343 del=halfoc-inthalfoc;
cannam@86 344
cannam@86 345 for(j=0;j<P_NOISECURVES;j++)
cannam@86 346 p->noiseoffset[j][i]=
cannam@86 347 p->vi->noiseoff[j][inthalfoc]*(1.-del) +
cannam@86 348 p->vi->noiseoff[j][inthalfoc+1]*del;
cannam@86 349
cannam@86 350 }
cannam@86 351 #if 0
cannam@86 352 {
cannam@86 353 static int ls=0;
cannam@86 354 _analysis_output_always("noiseoff0",ls,p->noiseoffset[0],n,1,0,0);
cannam@86 355 _analysis_output_always("noiseoff1",ls,p->noiseoffset[1],n,1,0,0);
cannam@86 356 _analysis_output_always("noiseoff2",ls++,p->noiseoffset[2],n,1,0,0);
cannam@86 357 }
cannam@86 358 #endif
cannam@86 359 }
cannam@86 360
cannam@86 361 void _vp_psy_clear(vorbis_look_psy *p){
cannam@86 362 int i,j;
cannam@86 363 if(p){
cannam@86 364 if(p->ath)_ogg_free(p->ath);
cannam@86 365 if(p->octave)_ogg_free(p->octave);
cannam@86 366 if(p->bark)_ogg_free(p->bark);
cannam@86 367 if(p->tonecurves){
cannam@86 368 for(i=0;i<P_BANDS;i++){
cannam@86 369 for(j=0;j<P_LEVELS;j++){
cannam@86 370 _ogg_free(p->tonecurves[i][j]);
cannam@86 371 }
cannam@86 372 _ogg_free(p->tonecurves[i]);
cannam@86 373 }
cannam@86 374 _ogg_free(p->tonecurves);
cannam@86 375 }
cannam@86 376 if(p->noiseoffset){
cannam@86 377 for(i=0;i<P_NOISECURVES;i++){
cannam@86 378 _ogg_free(p->noiseoffset[i]);
cannam@86 379 }
cannam@86 380 _ogg_free(p->noiseoffset);
cannam@86 381 }
cannam@86 382 memset(p,0,sizeof(*p));
cannam@86 383 }
cannam@86 384 }
cannam@86 385
cannam@86 386 /* octave/(8*eighth_octave_lines) x scale and dB y scale */
cannam@86 387 static void seed_curve(float *seed,
cannam@86 388 const float **curves,
cannam@86 389 float amp,
cannam@86 390 int oc, int n,
cannam@86 391 int linesper,float dBoffset){
cannam@86 392 int i,post1;
cannam@86 393 int seedptr;
cannam@86 394 const float *posts,*curve;
cannam@86 395
cannam@86 396 int choice=(int)((amp+dBoffset-P_LEVEL_0)*.1f);
cannam@86 397 choice=max(choice,0);
cannam@86 398 choice=min(choice,P_LEVELS-1);
cannam@86 399 posts=curves[choice];
cannam@86 400 curve=posts+2;
cannam@86 401 post1=(int)posts[1];
cannam@86 402 seedptr=oc+(posts[0]-EHMER_OFFSET)*linesper-(linesper>>1);
cannam@86 403
cannam@86 404 for(i=posts[0];i<post1;i++){
cannam@86 405 if(seedptr>0){
cannam@86 406 float lin=amp+curve[i];
cannam@86 407 if(seed[seedptr]<lin)seed[seedptr]=lin;
cannam@86 408 }
cannam@86 409 seedptr+=linesper;
cannam@86 410 if(seedptr>=n)break;
cannam@86 411 }
cannam@86 412 }
cannam@86 413
cannam@86 414 static void seed_loop(vorbis_look_psy *p,
cannam@86 415 const float ***curves,
cannam@86 416 const float *f,
cannam@86 417 const float *flr,
cannam@86 418 float *seed,
cannam@86 419 float specmax){
cannam@86 420 vorbis_info_psy *vi=p->vi;
cannam@86 421 long n=p->n,i;
cannam@86 422 float dBoffset=vi->max_curve_dB-specmax;
cannam@86 423
cannam@86 424 /* prime the working vector with peak values */
cannam@86 425
cannam@86 426 for(i=0;i<n;i++){
cannam@86 427 float max=f[i];
cannam@86 428 long oc=p->octave[i];
cannam@86 429 while(i+1<n && p->octave[i+1]==oc){
cannam@86 430 i++;
cannam@86 431 if(f[i]>max)max=f[i];
cannam@86 432 }
cannam@86 433
cannam@86 434 if(max+6.f>flr[i]){
cannam@86 435 oc=oc>>p->shiftoc;
cannam@86 436
cannam@86 437 if(oc>=P_BANDS)oc=P_BANDS-1;
cannam@86 438 if(oc<0)oc=0;
cannam@86 439
cannam@86 440 seed_curve(seed,
cannam@86 441 curves[oc],
cannam@86 442 max,
cannam@86 443 p->octave[i]-p->firstoc,
cannam@86 444 p->total_octave_lines,
cannam@86 445 p->eighth_octave_lines,
cannam@86 446 dBoffset);
cannam@86 447 }
cannam@86 448 }
cannam@86 449 }
cannam@86 450
cannam@86 451 static void seed_chase(float *seeds, int linesper, long n){
cannam@86 452 long *posstack=alloca(n*sizeof(*posstack));
cannam@86 453 float *ampstack=alloca(n*sizeof(*ampstack));
cannam@86 454 long stack=0;
cannam@86 455 long pos=0;
cannam@86 456 long i;
cannam@86 457
cannam@86 458 for(i=0;i<n;i++){
cannam@86 459 if(stack<2){
cannam@86 460 posstack[stack]=i;
cannam@86 461 ampstack[stack++]=seeds[i];
cannam@86 462 }else{
cannam@86 463 while(1){
cannam@86 464 if(seeds[i]<ampstack[stack-1]){
cannam@86 465 posstack[stack]=i;
cannam@86 466 ampstack[stack++]=seeds[i];
cannam@86 467 break;
cannam@86 468 }else{
cannam@86 469 if(i<posstack[stack-1]+linesper){
cannam@86 470 if(stack>1 && ampstack[stack-1]<=ampstack[stack-2] &&
cannam@86 471 i<posstack[stack-2]+linesper){
cannam@86 472 /* we completely overlap, making stack-1 irrelevant. pop it */
cannam@86 473 stack--;
cannam@86 474 continue;
cannam@86 475 }
cannam@86 476 }
cannam@86 477 posstack[stack]=i;
cannam@86 478 ampstack[stack++]=seeds[i];
cannam@86 479 break;
cannam@86 480
cannam@86 481 }
cannam@86 482 }
cannam@86 483 }
cannam@86 484 }
cannam@86 485
cannam@86 486 /* the stack now contains only the positions that are relevant. Scan
cannam@86 487 'em straight through */
cannam@86 488
cannam@86 489 for(i=0;i<stack;i++){
cannam@86 490 long endpos;
cannam@86 491 if(i<stack-1 && ampstack[i+1]>ampstack[i]){
cannam@86 492 endpos=posstack[i+1];
cannam@86 493 }else{
cannam@86 494 endpos=posstack[i]+linesper+1; /* +1 is important, else bin 0 is
cannam@86 495 discarded in short frames */
cannam@86 496 }
cannam@86 497 if(endpos>n)endpos=n;
cannam@86 498 for(;pos<endpos;pos++)
cannam@86 499 seeds[pos]=ampstack[i];
cannam@86 500 }
cannam@86 501
cannam@86 502 /* there. Linear time. I now remember this was on a problem set I
cannam@86 503 had in Grad Skool... I didn't solve it at the time ;-) */
cannam@86 504
cannam@86 505 }
cannam@86 506
cannam@86 507 /* bleaugh, this is more complicated than it needs to be */
cannam@86 508 #include<stdio.h>
cannam@86 509 static void max_seeds(vorbis_look_psy *p,
cannam@86 510 float *seed,
cannam@86 511 float *flr){
cannam@86 512 long n=p->total_octave_lines;
cannam@86 513 int linesper=p->eighth_octave_lines;
cannam@86 514 long linpos=0;
cannam@86 515 long pos;
cannam@86 516
cannam@86 517 seed_chase(seed,linesper,n); /* for masking */
cannam@86 518
cannam@86 519 pos=p->octave[0]-p->firstoc-(linesper>>1);
cannam@86 520
cannam@86 521 while(linpos+1<p->n){
cannam@86 522 float minV=seed[pos];
cannam@86 523 long end=((p->octave[linpos]+p->octave[linpos+1])>>1)-p->firstoc;
cannam@86 524 if(minV>p->vi->tone_abs_limit)minV=p->vi->tone_abs_limit;
cannam@86 525 while(pos+1<=end){
cannam@86 526 pos++;
cannam@86 527 if((seed[pos]>NEGINF && seed[pos]<minV) || minV==NEGINF)
cannam@86 528 minV=seed[pos];
cannam@86 529 }
cannam@86 530
cannam@86 531 end=pos+p->firstoc;
cannam@86 532 for(;linpos<p->n && p->octave[linpos]<=end;linpos++)
cannam@86 533 if(flr[linpos]<minV)flr[linpos]=minV;
cannam@86 534 }
cannam@86 535
cannam@86 536 {
cannam@86 537 float minV=seed[p->total_octave_lines-1];
cannam@86 538 for(;linpos<p->n;linpos++)
cannam@86 539 if(flr[linpos]<minV)flr[linpos]=minV;
cannam@86 540 }
cannam@86 541
cannam@86 542 }
cannam@86 543
cannam@86 544 static void bark_noise_hybridmp(int n,const long *b,
cannam@86 545 const float *f,
cannam@86 546 float *noise,
cannam@86 547 const float offset,
cannam@86 548 const int fixed){
cannam@86 549
cannam@86 550 float *N=alloca(n*sizeof(*N));
cannam@86 551 float *X=alloca(n*sizeof(*N));
cannam@86 552 float *XX=alloca(n*sizeof(*N));
cannam@86 553 float *Y=alloca(n*sizeof(*N));
cannam@86 554 float *XY=alloca(n*sizeof(*N));
cannam@86 555
cannam@86 556 float tN, tX, tXX, tY, tXY;
cannam@86 557 int i;
cannam@86 558
cannam@86 559 int lo, hi;
cannam@86 560 float R=0.f;
cannam@86 561 float A=0.f;
cannam@86 562 float B=0.f;
cannam@86 563 float D=1.f;
cannam@86 564 float w, x, y;
cannam@86 565
cannam@86 566 tN = tX = tXX = tY = tXY = 0.f;
cannam@86 567
cannam@86 568 y = f[0] + offset;
cannam@86 569 if (y < 1.f) y = 1.f;
cannam@86 570
cannam@86 571 w = y * y * .5;
cannam@86 572
cannam@86 573 tN += w;
cannam@86 574 tX += w;
cannam@86 575 tY += w * y;
cannam@86 576
cannam@86 577 N[0] = tN;
cannam@86 578 X[0] = tX;
cannam@86 579 XX[0] = tXX;
cannam@86 580 Y[0] = tY;
cannam@86 581 XY[0] = tXY;
cannam@86 582
cannam@86 583 for (i = 1, x = 1.f; i < n; i++, x += 1.f) {
cannam@86 584
cannam@86 585 y = f[i] + offset;
cannam@86 586 if (y < 1.f) y = 1.f;
cannam@86 587
cannam@86 588 w = y * y;
cannam@86 589
cannam@86 590 tN += w;
cannam@86 591 tX += w * x;
cannam@86 592 tXX += w * x * x;
cannam@86 593 tY += w * y;
cannam@86 594 tXY += w * x * y;
cannam@86 595
cannam@86 596 N[i] = tN;
cannam@86 597 X[i] = tX;
cannam@86 598 XX[i] = tXX;
cannam@86 599 Y[i] = tY;
cannam@86 600 XY[i] = tXY;
cannam@86 601 }
cannam@86 602
cannam@86 603 for (i = 0, x = 0.f;; i++, x += 1.f) {
cannam@86 604
cannam@86 605 lo = b[i] >> 16;
cannam@86 606 if( lo>=0 ) break;
cannam@86 607 hi = b[i] & 0xffff;
cannam@86 608
cannam@86 609 tN = N[hi] + N[-lo];
cannam@86 610 tX = X[hi] - X[-lo];
cannam@86 611 tXX = XX[hi] + XX[-lo];
cannam@86 612 tY = Y[hi] + Y[-lo];
cannam@86 613 tXY = XY[hi] - XY[-lo];
cannam@86 614
cannam@86 615 A = tY * tXX - tX * tXY;
cannam@86 616 B = tN * tXY - tX * tY;
cannam@86 617 D = tN * tXX - tX * tX;
cannam@86 618 R = (A + x * B) / D;
cannam@86 619 if (R < 0.f)
cannam@86 620 R = 0.f;
cannam@86 621
cannam@86 622 noise[i] = R - offset;
cannam@86 623 }
cannam@86 624
cannam@86 625 for ( ;; i++, x += 1.f) {
cannam@86 626
cannam@86 627 lo = b[i] >> 16;
cannam@86 628 hi = b[i] & 0xffff;
cannam@86 629 if(hi>=n)break;
cannam@86 630
cannam@86 631 tN = N[hi] - N[lo];
cannam@86 632 tX = X[hi] - X[lo];
cannam@86 633 tXX = XX[hi] - XX[lo];
cannam@86 634 tY = Y[hi] - Y[lo];
cannam@86 635 tXY = XY[hi] - XY[lo];
cannam@86 636
cannam@86 637 A = tY * tXX - tX * tXY;
cannam@86 638 B = tN * tXY - tX * tY;
cannam@86 639 D = tN * tXX - tX * tX;
cannam@86 640 R = (A + x * B) / D;
cannam@86 641 if (R < 0.f) R = 0.f;
cannam@86 642
cannam@86 643 noise[i] = R - offset;
cannam@86 644 }
cannam@86 645 for ( ; i < n; i++, x += 1.f) {
cannam@86 646
cannam@86 647 R = (A + x * B) / D;
cannam@86 648 if (R < 0.f) R = 0.f;
cannam@86 649
cannam@86 650 noise[i] = R - offset;
cannam@86 651 }
cannam@86 652
cannam@86 653 if (fixed <= 0) return;
cannam@86 654
cannam@86 655 for (i = 0, x = 0.f;; i++, x += 1.f) {
cannam@86 656 hi = i + fixed / 2;
cannam@86 657 lo = hi - fixed;
cannam@86 658 if(lo>=0)break;
cannam@86 659
cannam@86 660 tN = N[hi] + N[-lo];
cannam@86 661 tX = X[hi] - X[-lo];
cannam@86 662 tXX = XX[hi] + XX[-lo];
cannam@86 663 tY = Y[hi] + Y[-lo];
cannam@86 664 tXY = XY[hi] - XY[-lo];
cannam@86 665
cannam@86 666
cannam@86 667 A = tY * tXX - tX * tXY;
cannam@86 668 B = tN * tXY - tX * tY;
cannam@86 669 D = tN * tXX - tX * tX;
cannam@86 670 R = (A + x * B) / D;
cannam@86 671
cannam@86 672 if (R - offset < noise[i]) noise[i] = R - offset;
cannam@86 673 }
cannam@86 674 for ( ;; i++, x += 1.f) {
cannam@86 675
cannam@86 676 hi = i + fixed / 2;
cannam@86 677 lo = hi - fixed;
cannam@86 678 if(hi>=n)break;
cannam@86 679
cannam@86 680 tN = N[hi] - N[lo];
cannam@86 681 tX = X[hi] - X[lo];
cannam@86 682 tXX = XX[hi] - XX[lo];
cannam@86 683 tY = Y[hi] - Y[lo];
cannam@86 684 tXY = XY[hi] - XY[lo];
cannam@86 685
cannam@86 686 A = tY * tXX - tX * tXY;
cannam@86 687 B = tN * tXY - tX * tY;
cannam@86 688 D = tN * tXX - tX * tX;
cannam@86 689 R = (A + x * B) / D;
cannam@86 690
cannam@86 691 if (R - offset < noise[i]) noise[i] = R - offset;
cannam@86 692 }
cannam@86 693 for ( ; i < n; i++, x += 1.f) {
cannam@86 694 R = (A + x * B) / D;
cannam@86 695 if (R - offset < noise[i]) noise[i] = R - offset;
cannam@86 696 }
cannam@86 697 }
cannam@86 698
cannam@86 699 void _vp_noisemask(vorbis_look_psy *p,
cannam@86 700 float *logmdct,
cannam@86 701 float *logmask){
cannam@86 702
cannam@86 703 int i,n=p->n;
cannam@86 704 float *work=alloca(n*sizeof(*work));
cannam@86 705
cannam@86 706 bark_noise_hybridmp(n,p->bark,logmdct,logmask,
cannam@86 707 140.,-1);
cannam@86 708
cannam@86 709 for(i=0;i<n;i++)work[i]=logmdct[i]-logmask[i];
cannam@86 710
cannam@86 711 bark_noise_hybridmp(n,p->bark,work,logmask,0.,
cannam@86 712 p->vi->noisewindowfixed);
cannam@86 713
cannam@86 714 for(i=0;i<n;i++)work[i]=logmdct[i]-work[i];
cannam@86 715
cannam@86 716 #if 0
cannam@86 717 {
cannam@86 718 static int seq=0;
cannam@86 719
cannam@86 720 float work2[n];
cannam@86 721 for(i=0;i<n;i++){
cannam@86 722 work2[i]=logmask[i]+work[i];
cannam@86 723 }
cannam@86 724
cannam@86 725 if(seq&1)
cannam@86 726 _analysis_output("median2R",seq/2,work,n,1,0,0);
cannam@86 727 else
cannam@86 728 _analysis_output("median2L",seq/2,work,n,1,0,0);
cannam@86 729
cannam@86 730 if(seq&1)
cannam@86 731 _analysis_output("envelope2R",seq/2,work2,n,1,0,0);
cannam@86 732 else
cannam@86 733 _analysis_output("envelope2L",seq/2,work2,n,1,0,0);
cannam@86 734 seq++;
cannam@86 735 }
cannam@86 736 #endif
cannam@86 737
cannam@86 738 for(i=0;i<n;i++){
cannam@86 739 int dB=logmask[i]+.5;
cannam@86 740 if(dB>=NOISE_COMPAND_LEVELS)dB=NOISE_COMPAND_LEVELS-1;
cannam@86 741 if(dB<0)dB=0;
cannam@86 742 logmask[i]= work[i]+p->vi->noisecompand[dB];
cannam@86 743 }
cannam@86 744
cannam@86 745 }
cannam@86 746
cannam@86 747 void _vp_tonemask(vorbis_look_psy *p,
cannam@86 748 float *logfft,
cannam@86 749 float *logmask,
cannam@86 750 float global_specmax,
cannam@86 751 float local_specmax){
cannam@86 752
cannam@86 753 int i,n=p->n;
cannam@86 754
cannam@86 755 float *seed=alloca(sizeof(*seed)*p->total_octave_lines);
cannam@86 756 float att=local_specmax+p->vi->ath_adjatt;
cannam@86 757 for(i=0;i<p->total_octave_lines;i++)seed[i]=NEGINF;
cannam@86 758
cannam@86 759 /* set the ATH (floating below localmax, not global max by a
cannam@86 760 specified att) */
cannam@86 761 if(att<p->vi->ath_maxatt)att=p->vi->ath_maxatt;
cannam@86 762
cannam@86 763 for(i=0;i<n;i++)
cannam@86 764 logmask[i]=p->ath[i]+att;
cannam@86 765
cannam@86 766 /* tone masking */
cannam@86 767 seed_loop(p,(const float ***)p->tonecurves,logfft,logmask,seed,global_specmax);
cannam@86 768 max_seeds(p,seed,logmask);
cannam@86 769
cannam@86 770 }
cannam@86 771
cannam@86 772 void _vp_offset_and_mix(vorbis_look_psy *p,
cannam@86 773 float *noise,
cannam@86 774 float *tone,
cannam@86 775 int offset_select,
cannam@86 776 float *logmask,
cannam@86 777 float *mdct,
cannam@86 778 float *logmdct){
cannam@86 779 int i,n=p->n;
cannam@86 780 float de, coeffi, cx;/* AoTuV */
cannam@86 781 float toneatt=p->vi->tone_masteratt[offset_select];
cannam@86 782
cannam@86 783 cx = p->m_val;
cannam@86 784
cannam@86 785 for(i=0;i<n;i++){
cannam@86 786 float val= noise[i]+p->noiseoffset[offset_select][i];
cannam@86 787 if(val>p->vi->noisemaxsupp)val=p->vi->noisemaxsupp;
cannam@86 788 logmask[i]=max(val,tone[i]+toneatt);
cannam@86 789
cannam@86 790
cannam@86 791 /* AoTuV */
cannam@86 792 /** @ M1 **
cannam@86 793 The following codes improve a noise problem.
cannam@86 794 A fundamental idea uses the value of masking and carries out
cannam@86 795 the relative compensation of the MDCT.
cannam@86 796 However, this code is not perfect and all noise problems cannot be solved.
cannam@86 797 by Aoyumi @ 2004/04/18
cannam@86 798 */
cannam@86 799
cannam@86 800 if(offset_select == 1) {
cannam@86 801 coeffi = -17.2; /* coeffi is a -17.2dB threshold */
cannam@86 802 val = val - logmdct[i]; /* val == mdct line value relative to floor in dB */
cannam@86 803
cannam@86 804 if(val > coeffi){
cannam@86 805 /* mdct value is > -17.2 dB below floor */
cannam@86 806
cannam@86 807 de = 1.0-((val-coeffi)*0.005*cx);
cannam@86 808 /* pro-rated attenuation:
cannam@86 809 -0.00 dB boost if mdct value is -17.2dB (relative to floor)
cannam@86 810 -0.77 dB boost if mdct value is 0dB (relative to floor)
cannam@86 811 -1.64 dB boost if mdct value is +17.2dB (relative to floor)
cannam@86 812 etc... */
cannam@86 813
cannam@86 814 if(de < 0) de = 0.0001;
cannam@86 815 }else
cannam@86 816 /* mdct value is <= -17.2 dB below floor */
cannam@86 817
cannam@86 818 de = 1.0-((val-coeffi)*0.0003*cx);
cannam@86 819 /* pro-rated attenuation:
cannam@86 820 +0.00 dB atten if mdct value is -17.2dB (relative to floor)
cannam@86 821 +0.45 dB atten if mdct value is -34.4dB (relative to floor)
cannam@86 822 etc... */
cannam@86 823
cannam@86 824 mdct[i] *= de;
cannam@86 825
cannam@86 826 }
cannam@86 827 }
cannam@86 828 }
cannam@86 829
cannam@86 830 float _vp_ampmax_decay(float amp,vorbis_dsp_state *vd){
cannam@86 831 vorbis_info *vi=vd->vi;
cannam@86 832 codec_setup_info *ci=vi->codec_setup;
cannam@86 833 vorbis_info_psy_global *gi=&ci->psy_g_param;
cannam@86 834
cannam@86 835 int n=ci->blocksizes[vd->W]/2;
cannam@86 836 float secs=(float)n/vi->rate;
cannam@86 837
cannam@86 838 amp+=secs*gi->ampmax_att_per_sec;
cannam@86 839 if(amp<-9999)amp=-9999;
cannam@86 840 return(amp);
cannam@86 841 }
cannam@86 842
cannam@86 843 static float FLOOR1_fromdB_LOOKUP[256]={
cannam@86 844 1.0649863e-07F, 1.1341951e-07F, 1.2079015e-07F, 1.2863978e-07F,
cannam@86 845 1.3699951e-07F, 1.4590251e-07F, 1.5538408e-07F, 1.6548181e-07F,
cannam@86 846 1.7623575e-07F, 1.8768855e-07F, 1.9988561e-07F, 2.128753e-07F,
cannam@86 847 2.2670913e-07F, 2.4144197e-07F, 2.5713223e-07F, 2.7384213e-07F,
cannam@86 848 2.9163793e-07F, 3.1059021e-07F, 3.3077411e-07F, 3.5226968e-07F,
cannam@86 849 3.7516214e-07F, 3.9954229e-07F, 4.2550680e-07F, 4.5315863e-07F,
cannam@86 850 4.8260743e-07F, 5.1396998e-07F, 5.4737065e-07F, 5.8294187e-07F,
cannam@86 851 6.2082472e-07F, 6.6116941e-07F, 7.0413592e-07F, 7.4989464e-07F,
cannam@86 852 7.9862701e-07F, 8.5052630e-07F, 9.0579828e-07F, 9.6466216e-07F,
cannam@86 853 1.0273513e-06F, 1.0941144e-06F, 1.1652161e-06F, 1.2409384e-06F,
cannam@86 854 1.3215816e-06F, 1.4074654e-06F, 1.4989305e-06F, 1.5963394e-06F,
cannam@86 855 1.7000785e-06F, 1.8105592e-06F, 1.9282195e-06F, 2.0535261e-06F,
cannam@86 856 2.1869758e-06F, 2.3290978e-06F, 2.4804557e-06F, 2.6416497e-06F,
cannam@86 857 2.8133190e-06F, 2.9961443e-06F, 3.1908506e-06F, 3.3982101e-06F,
cannam@86 858 3.6190449e-06F, 3.8542308e-06F, 4.1047004e-06F, 4.3714470e-06F,
cannam@86 859 4.6555282e-06F, 4.9580707e-06F, 5.2802740e-06F, 5.6234160e-06F,
cannam@86 860 5.9888572e-06F, 6.3780469e-06F, 6.7925283e-06F, 7.2339451e-06F,
cannam@86 861 7.7040476e-06F, 8.2047000e-06F, 8.7378876e-06F, 9.3057248e-06F,
cannam@86 862 9.9104632e-06F, 1.0554501e-05F, 1.1240392e-05F, 1.1970856e-05F,
cannam@86 863 1.2748789e-05F, 1.3577278e-05F, 1.4459606e-05F, 1.5399272e-05F,
cannam@86 864 1.6400004e-05F, 1.7465768e-05F, 1.8600792e-05F, 1.9809576e-05F,
cannam@86 865 2.1096914e-05F, 2.2467911e-05F, 2.3928002e-05F, 2.5482978e-05F,
cannam@86 866 2.7139006e-05F, 2.8902651e-05F, 3.0780908e-05F, 3.2781225e-05F,
cannam@86 867 3.4911534e-05F, 3.7180282e-05F, 3.9596466e-05F, 4.2169667e-05F,
cannam@86 868 4.4910090e-05F, 4.7828601e-05F, 5.0936773e-05F, 5.4246931e-05F,
cannam@86 869 5.7772202e-05F, 6.1526565e-05F, 6.5524908e-05F, 6.9783085e-05F,
cannam@86 870 7.4317983e-05F, 7.9147585e-05F, 8.4291040e-05F, 8.9768747e-05F,
cannam@86 871 9.5602426e-05F, 0.00010181521F, 0.00010843174F, 0.00011547824F,
cannam@86 872 0.00012298267F, 0.00013097477F, 0.00013948625F, 0.00014855085F,
cannam@86 873 0.00015820453F, 0.00016848555F, 0.00017943469F, 0.00019109536F,
cannam@86 874 0.00020351382F, 0.00021673929F, 0.00023082423F, 0.00024582449F,
cannam@86 875 0.00026179955F, 0.00027881276F, 0.00029693158F, 0.00031622787F,
cannam@86 876 0.00033677814F, 0.00035866388F, 0.00038197188F, 0.00040679456F,
cannam@86 877 0.00043323036F, 0.00046138411F, 0.00049136745F, 0.00052329927F,
cannam@86 878 0.00055730621F, 0.00059352311F, 0.00063209358F, 0.00067317058F,
cannam@86 879 0.00071691700F, 0.00076350630F, 0.00081312324F, 0.00086596457F,
cannam@86 880 0.00092223983F, 0.00098217216F, 0.0010459992F, 0.0011139742F,
cannam@86 881 0.0011863665F, 0.0012634633F, 0.0013455702F, 0.0014330129F,
cannam@86 882 0.0015261382F, 0.0016253153F, 0.0017309374F, 0.0018434235F,
cannam@86 883 0.0019632195F, 0.0020908006F, 0.0022266726F, 0.0023713743F,
cannam@86 884 0.0025254795F, 0.0026895994F, 0.0028643847F, 0.0030505286F,
cannam@86 885 0.0032487691F, 0.0034598925F, 0.0036847358F, 0.0039241906F,
cannam@86 886 0.0041792066F, 0.0044507950F, 0.0047400328F, 0.0050480668F,
cannam@86 887 0.0053761186F, 0.0057254891F, 0.0060975636F, 0.0064938176F,
cannam@86 888 0.0069158225F, 0.0073652516F, 0.0078438871F, 0.0083536271F,
cannam@86 889 0.0088964928F, 0.009474637F, 0.010090352F, 0.010746080F,
cannam@86 890 0.011444421F, 0.012188144F, 0.012980198F, 0.013823725F,
cannam@86 891 0.014722068F, 0.015678791F, 0.016697687F, 0.017782797F,
cannam@86 892 0.018938423F, 0.020169149F, 0.021479854F, 0.022875735F,
cannam@86 893 0.024362330F, 0.025945531F, 0.027631618F, 0.029427276F,
cannam@86 894 0.031339626F, 0.033376252F, 0.035545228F, 0.037855157F,
cannam@86 895 0.040315199F, 0.042935108F, 0.045725273F, 0.048696758F,
cannam@86 896 0.051861348F, 0.055231591F, 0.058820850F, 0.062643361F,
cannam@86 897 0.066714279F, 0.071049749F, 0.075666962F, 0.080584227F,
cannam@86 898 0.085821044F, 0.091398179F, 0.097337747F, 0.10366330F,
cannam@86 899 0.11039993F, 0.11757434F, 0.12521498F, 0.13335215F,
cannam@86 900 0.14201813F, 0.15124727F, 0.16107617F, 0.17154380F,
cannam@86 901 0.18269168F, 0.19456402F, 0.20720788F, 0.22067342F,
cannam@86 902 0.23501402F, 0.25028656F, 0.26655159F, 0.28387361F,
cannam@86 903 0.30232132F, 0.32196786F, 0.34289114F, 0.36517414F,
cannam@86 904 0.38890521F, 0.41417847F, 0.44109412F, 0.46975890F,
cannam@86 905 0.50028648F, 0.53279791F, 0.56742212F, 0.60429640F,
cannam@86 906 0.64356699F, 0.68538959F, 0.72993007F, 0.77736504F,
cannam@86 907 0.82788260F, 0.88168307F, 0.9389798F, 1.F,
cannam@86 908 };
cannam@86 909
cannam@86 910 /* this is for per-channel noise normalization */
cannam@86 911 static int apsort(const void *a, const void *b){
cannam@86 912 float f1=**(float**)a;
cannam@86 913 float f2=**(float**)b;
cannam@86 914 return (f1<f2)-(f1>f2);
cannam@86 915 }
cannam@86 916
cannam@86 917 static void flag_lossless(int limit, float prepoint, float postpoint, float *mdct,
cannam@86 918 float *floor, int *flag, int i, int jn){
cannam@86 919 int j;
cannam@86 920 for(j=0;j<jn;j++){
cannam@86 921 float point = j>=limit-i ? postpoint : prepoint;
cannam@86 922 float r = fabs(mdct[j])/floor[j];
cannam@86 923 if(r<point)
cannam@86 924 flag[j]=0;
cannam@86 925 else
cannam@86 926 flag[j]=1;
cannam@86 927 }
cannam@86 928 }
cannam@86 929
cannam@86 930 /* Overload/Side effect: On input, the *q vector holds either the
cannam@86 931 quantized energy (for elements with the flag set) or the absolute
cannam@86 932 values of the *r vector (for elements with flag unset). On output,
cannam@86 933 *q holds the quantized energy for all elements */
cannam@86 934 static float noise_normalize(vorbis_look_psy *p, int limit, float *r, float *q, float *f, int *flags, float acc, int i, int n, int *out){
cannam@86 935
cannam@86 936 vorbis_info_psy *vi=p->vi;
cannam@86 937 float **sort = alloca(n*sizeof(*sort));
cannam@86 938 int j,count=0;
cannam@86 939 int start = (vi->normal_p ? vi->normal_start-i : n);
cannam@86 940 if(start>n)start=n;
cannam@86 941
cannam@86 942 /* force classic behavior where only energy in the current band is considered */
cannam@86 943 acc=0.f;
cannam@86 944
cannam@86 945 /* still responsible for populating *out where noise norm not in
cannam@86 946 effect. There's no need to [re]populate *q in these areas */
cannam@86 947 for(j=0;j<start;j++){
cannam@86 948 if(!flags || !flags[j]){ /* lossless coupling already quantized.
cannam@86 949 Don't touch; requantizing based on
cannam@86 950 energy would be incorrect. */
cannam@86 951 float ve = q[j]/f[j];
cannam@86 952 if(r[j]<0)
cannam@86 953 out[j] = -rint(sqrt(ve));
cannam@86 954 else
cannam@86 955 out[j] = rint(sqrt(ve));
cannam@86 956 }
cannam@86 957 }
cannam@86 958
cannam@86 959 /* sort magnitudes for noise norm portion of partition */
cannam@86 960 for(;j<n;j++){
cannam@86 961 if(!flags || !flags[j]){ /* can't noise norm elements that have
cannam@86 962 already been loslessly coupled; we can
cannam@86 963 only account for their energy error */
cannam@86 964 float ve = q[j]/f[j];
cannam@86 965 /* Despite all the new, more capable coupling code, for now we
cannam@86 966 implement noise norm as it has been up to this point. Only
cannam@86 967 consider promotions to unit magnitude from 0. In addition
cannam@86 968 the only energy error counted is quantizations to zero. */
cannam@86 969 /* also-- the original point code only applied noise norm at > pointlimit */
cannam@86 970 if(ve<.25f && (!flags || j>=limit-i)){
cannam@86 971 acc += ve;
cannam@86 972 sort[count++]=q+j; /* q is fabs(r) for unflagged element */
cannam@86 973 }else{
cannam@86 974 /* For now: no acc adjustment for nonzero quantization. populate *out and q as this value is final. */
cannam@86 975 if(r[j]<0)
cannam@86 976 out[j] = -rint(sqrt(ve));
cannam@86 977 else
cannam@86 978 out[j] = rint(sqrt(ve));
cannam@86 979 q[j] = out[j]*out[j]*f[j];
cannam@86 980 }
cannam@86 981 }/* else{
cannam@86 982 again, no energy adjustment for error in nonzero quant-- for now
cannam@86 983 }*/
cannam@86 984 }
cannam@86 985
cannam@86 986 if(count){
cannam@86 987 /* noise norm to do */
cannam@86 988 qsort(sort,count,sizeof(*sort),apsort);
cannam@86 989 for(j=0;j<count;j++){
cannam@86 990 int k=sort[j]-q;
cannam@86 991 if(acc>=vi->normal_thresh){
cannam@86 992 out[k]=unitnorm(r[k]);
cannam@86 993 acc-=1.f;
cannam@86 994 q[k]=f[k];
cannam@86 995 }else{
cannam@86 996 out[k]=0;
cannam@86 997 q[k]=0.f;
cannam@86 998 }
cannam@86 999 }
cannam@86 1000 }
cannam@86 1001
cannam@86 1002 return acc;
cannam@86 1003 }
cannam@86 1004
cannam@86 1005 /* Noise normalization, quantization and coupling are not wholly
cannam@86 1006 seperable processes in depth>1 coupling. */
cannam@86 1007 void _vp_couple_quantize_normalize(int blobno,
cannam@86 1008 vorbis_info_psy_global *g,
cannam@86 1009 vorbis_look_psy *p,
cannam@86 1010 vorbis_info_mapping0 *vi,
cannam@86 1011 float **mdct,
cannam@86 1012 int **iwork,
cannam@86 1013 int *nonzero,
cannam@86 1014 int sliding_lowpass,
cannam@86 1015 int ch){
cannam@86 1016
cannam@86 1017 int i;
cannam@86 1018 int n = p->n;
cannam@86 1019 int partition=(p->vi->normal_p ? p->vi->normal_partition : 16);
cannam@86 1020 int limit = g->coupling_pointlimit[p->vi->blockflag][blobno];
cannam@86 1021 float prepoint=stereo_threshholds[g->coupling_prepointamp[blobno]];
cannam@86 1022 float postpoint=stereo_threshholds[g->coupling_postpointamp[blobno]];
cannam@86 1023 #if 0
cannam@86 1024 float de=0.1*p->m_val; /* a blend of the AoTuV M2 and M3 code here and below */
cannam@86 1025 #endif
cannam@86 1026
cannam@86 1027 /* mdct is our raw mdct output, floor not removed. */
cannam@86 1028 /* inout passes in the ifloor, passes back quantized result */
cannam@86 1029
cannam@86 1030 /* unquantized energy (negative indicates amplitude has negative sign) */
cannam@86 1031 float **raw = alloca(ch*sizeof(*raw));
cannam@86 1032
cannam@86 1033 /* dual pupose; quantized energy (if flag set), othersize fabs(raw) */
cannam@86 1034 float **quant = alloca(ch*sizeof(*quant));
cannam@86 1035
cannam@86 1036 /* floor energy */
cannam@86 1037 float **floor = alloca(ch*sizeof(*floor));
cannam@86 1038
cannam@86 1039 /* flags indicating raw/quantized status of elements in raw vector */
cannam@86 1040 int **flag = alloca(ch*sizeof(*flag));
cannam@86 1041
cannam@86 1042 /* non-zero flag working vector */
cannam@86 1043 int *nz = alloca(ch*sizeof(*nz));
cannam@86 1044
cannam@86 1045 /* energy surplus/defecit tracking */
cannam@86 1046 float *acc = alloca((ch+vi->coupling_steps)*sizeof(*acc));
cannam@86 1047
cannam@86 1048 /* The threshold of a stereo is changed with the size of n */
cannam@86 1049 if(n > 1000)
cannam@86 1050 postpoint=stereo_threshholds_limited[g->coupling_postpointamp[blobno]];
cannam@86 1051
cannam@86 1052 raw[0] = alloca(ch*partition*sizeof(**raw));
cannam@86 1053 quant[0] = alloca(ch*partition*sizeof(**quant));
cannam@86 1054 floor[0] = alloca(ch*partition*sizeof(**floor));
cannam@86 1055 flag[0] = alloca(ch*partition*sizeof(**flag));
cannam@86 1056
cannam@86 1057 for(i=1;i<ch;i++){
cannam@86 1058 raw[i] = &raw[0][partition*i];
cannam@86 1059 quant[i] = &quant[0][partition*i];
cannam@86 1060 floor[i] = &floor[0][partition*i];
cannam@86 1061 flag[i] = &flag[0][partition*i];
cannam@86 1062 }
cannam@86 1063 for(i=0;i<ch+vi->coupling_steps;i++)
cannam@86 1064 acc[i]=0.f;
cannam@86 1065
cannam@86 1066 for(i=0;i<n;i+=partition){
cannam@86 1067 int k,j,jn = partition > n-i ? n-i : partition;
cannam@86 1068 int step,track = 0;
cannam@86 1069
cannam@86 1070 memcpy(nz,nonzero,sizeof(*nz)*ch);
cannam@86 1071
cannam@86 1072 /* prefill */
cannam@86 1073 memset(flag[0],0,ch*partition*sizeof(**flag));
cannam@86 1074 for(k=0;k<ch;k++){
cannam@86 1075 int *iout = &iwork[k][i];
cannam@86 1076 if(nz[k]){
cannam@86 1077
cannam@86 1078 for(j=0;j<jn;j++)
cannam@86 1079 floor[k][j] = FLOOR1_fromdB_LOOKUP[iout[j]];
cannam@86 1080
cannam@86 1081 flag_lossless(limit,prepoint,postpoint,&mdct[k][i],floor[k],flag[k],i,jn);
cannam@86 1082
cannam@86 1083 for(j=0;j<jn;j++){
cannam@86 1084 quant[k][j] = raw[k][j] = mdct[k][i+j]*mdct[k][i+j];
cannam@86 1085 if(mdct[k][i+j]<0.f) raw[k][j]*=-1.f;
cannam@86 1086 floor[k][j]*=floor[k][j];
cannam@86 1087 }
cannam@86 1088
cannam@86 1089 acc[track]=noise_normalize(p,limit,raw[k],quant[k],floor[k],NULL,acc[track],i,jn,iout);
cannam@86 1090
cannam@86 1091 }else{
cannam@86 1092 for(j=0;j<jn;j++){
cannam@86 1093 floor[k][j] = 1e-10f;
cannam@86 1094 raw[k][j] = 0.f;
cannam@86 1095 quant[k][j] = 0.f;
cannam@86 1096 flag[k][j] = 0;
cannam@86 1097 iout[j]=0;
cannam@86 1098 }
cannam@86 1099 acc[track]=0.f;
cannam@86 1100 }
cannam@86 1101 track++;
cannam@86 1102 }
cannam@86 1103
cannam@86 1104 /* coupling */
cannam@86 1105 for(step=0;step<vi->coupling_steps;step++){
cannam@86 1106 int Mi = vi->coupling_mag[step];
cannam@86 1107 int Ai = vi->coupling_ang[step];
cannam@86 1108 int *iM = &iwork[Mi][i];
cannam@86 1109 int *iA = &iwork[Ai][i];
cannam@86 1110 float *reM = raw[Mi];
cannam@86 1111 float *reA = raw[Ai];
cannam@86 1112 float *qeM = quant[Mi];
cannam@86 1113 float *qeA = quant[Ai];
cannam@86 1114 float *floorM = floor[Mi];
cannam@86 1115 float *floorA = floor[Ai];
cannam@86 1116 int *fM = flag[Mi];
cannam@86 1117 int *fA = flag[Ai];
cannam@86 1118
cannam@86 1119 if(nz[Mi] || nz[Ai]){
cannam@86 1120 nz[Mi] = nz[Ai] = 1;
cannam@86 1121
cannam@86 1122 for(j=0;j<jn;j++){
cannam@86 1123
cannam@86 1124 if(j<sliding_lowpass-i){
cannam@86 1125 if(fM[j] || fA[j]){
cannam@86 1126 /* lossless coupling */
cannam@86 1127
cannam@86 1128 reM[j] = fabs(reM[j])+fabs(reA[j]);
cannam@86 1129 qeM[j] = qeM[j]+qeA[j];
cannam@86 1130 fM[j]=fA[j]=1;
cannam@86 1131
cannam@86 1132 /* couple iM/iA */
cannam@86 1133 {
cannam@86 1134 int A = iM[j];
cannam@86 1135 int B = iA[j];
cannam@86 1136
cannam@86 1137 if(abs(A)>abs(B)){
cannam@86 1138 iA[j]=(A>0?A-B:B-A);
cannam@86 1139 }else{
cannam@86 1140 iA[j]=(B>0?A-B:B-A);
cannam@86 1141 iM[j]=B;
cannam@86 1142 }
cannam@86 1143
cannam@86 1144 /* collapse two equivalent tuples to one */
cannam@86 1145 if(iA[j]>=abs(iM[j])*2){
cannam@86 1146 iA[j]= -iA[j];
cannam@86 1147 iM[j]= -iM[j];
cannam@86 1148 }
cannam@86 1149
cannam@86 1150 }
cannam@86 1151
cannam@86 1152 }else{
cannam@86 1153 /* lossy (point) coupling */
cannam@86 1154 if(j<limit-i){
cannam@86 1155 /* dipole */
cannam@86 1156 reM[j] += reA[j];
cannam@86 1157 qeM[j] = fabs(reM[j]);
cannam@86 1158 }else{
cannam@86 1159 #if 0
cannam@86 1160 /* AoTuV */
cannam@86 1161 /** @ M2 **
cannam@86 1162 The boost problem by the combination of noise normalization and point stereo is eased.
cannam@86 1163 However, this is a temporary patch.
cannam@86 1164 by Aoyumi @ 2004/04/18
cannam@86 1165 */
cannam@86 1166 float derate = (1.0 - de*((float)(j-limit+i) / (float)(n-limit)));
cannam@86 1167 /* elliptical */
cannam@86 1168 if(reM[j]+reA[j]<0){
cannam@86 1169 reM[j] = - (qeM[j] = (fabs(reM[j])+fabs(reA[j]))*derate*derate);
cannam@86 1170 }else{
cannam@86 1171 reM[j] = (qeM[j] = (fabs(reM[j])+fabs(reA[j]))*derate*derate);
cannam@86 1172 }
cannam@86 1173 #else
cannam@86 1174 /* elliptical */
cannam@86 1175 if(reM[j]+reA[j]<0){
cannam@86 1176 reM[j] = - (qeM[j] = fabs(reM[j])+fabs(reA[j]));
cannam@86 1177 }else{
cannam@86 1178 reM[j] = (qeM[j] = fabs(reM[j])+fabs(reA[j]));
cannam@86 1179 }
cannam@86 1180 #endif
cannam@86 1181
cannam@86 1182 }
cannam@86 1183 reA[j]=qeA[j]=0.f;
cannam@86 1184 fA[j]=1;
cannam@86 1185 iA[j]=0;
cannam@86 1186 }
cannam@86 1187 }
cannam@86 1188 floorM[j]=floorA[j]=floorM[j]+floorA[j];
cannam@86 1189 }
cannam@86 1190 /* normalize the resulting mag vector */
cannam@86 1191 acc[track]=noise_normalize(p,limit,raw[Mi],quant[Mi],floor[Mi],flag[Mi],acc[track],i,jn,iM);
cannam@86 1192 track++;
cannam@86 1193 }
cannam@86 1194 }
cannam@86 1195 }
cannam@86 1196
cannam@86 1197 for(i=0;i<vi->coupling_steps;i++){
cannam@86 1198 /* make sure coupling a zero and a nonzero channel results in two
cannam@86 1199 nonzero channels. */
cannam@86 1200 if(nonzero[vi->coupling_mag[i]] ||
cannam@86 1201 nonzero[vi->coupling_ang[i]]){
cannam@86 1202 nonzero[vi->coupling_mag[i]]=1;
cannam@86 1203 nonzero[vi->coupling_ang[i]]=1;
cannam@86 1204 }
cannam@86 1205 }
cannam@86 1206 }