dbf-halloween2015

annotate libs/vorbis/floor1.c @ 2:5ae5fd3626fa

find changesets by author, revision, files, or words in the commit message
macosx port works
author John Tsiombikas <nuclear@member.fsf.org>
date Sun, 01 Nov 2015 00:51:36 +0200
parents
children
rev   line source
nuclear@1 1 /********************************************************************
nuclear@1 2 * *
nuclear@1 3 * THIS FILE IS PART OF THE OggVorbis SOFTWARE CODEC SOURCE CODE. *
nuclear@1 4 * USE, DISTRIBUTION AND REPRODUCTION OF THIS LIBRARY SOURCE IS *
nuclear@1 5 * GOVERNED BY A BSD-STYLE SOURCE LICENSE INCLUDED WITH THIS SOURCE *
nuclear@1 6 * IN 'COPYING'. PLEASE READ THESE TERMS BEFORE DISTRIBUTING. *
nuclear@1 7 * *
nuclear@1 8 * THE OggVorbis SOURCE CODE IS (C) COPYRIGHT 1994-2009 *
nuclear@1 9 * by the Xiph.Org Foundation http://www.xiph.org/ *
nuclear@1 10 * *
nuclear@1 11 ********************************************************************
nuclear@1 12
nuclear@1 13 function: floor backend 1 implementation
nuclear@1 14 last mod: $Id: floor1.c 18151 2012-01-20 07:35:26Z xiphmont $
nuclear@1 15
nuclear@1 16 ********************************************************************/
nuclear@1 17
nuclear@1 18 #include <stdlib.h>
nuclear@1 19 #include <string.h>
nuclear@1 20 #include <math.h>
nuclear@1 21 #include <ogg/ogg.h>
nuclear@1 22 #include "vorbis/codec.h"
nuclear@1 23 #include "codec_internal.h"
nuclear@1 24 #include "registry.h"
nuclear@1 25 #include "codebook.h"
nuclear@1 26 #include "misc.h"
nuclear@1 27 #include "scales.h"
nuclear@1 28
nuclear@1 29 #include <stdio.h>
nuclear@1 30
nuclear@1 31 #define floor1_rangedB 140 /* floor 1 fixed at -140dB to 0dB range */
nuclear@1 32
nuclear@1 33 typedef struct lsfit_acc{
nuclear@1 34 int x0;
nuclear@1 35 int x1;
nuclear@1 36
nuclear@1 37 int xa;
nuclear@1 38 int ya;
nuclear@1 39 int x2a;
nuclear@1 40 int y2a;
nuclear@1 41 int xya;
nuclear@1 42 int an;
nuclear@1 43
nuclear@1 44 int xb;
nuclear@1 45 int yb;
nuclear@1 46 int x2b;
nuclear@1 47 int y2b;
nuclear@1 48 int xyb;
nuclear@1 49 int bn;
nuclear@1 50 } lsfit_acc;
nuclear@1 51
nuclear@1 52 /***********************************************/
nuclear@1 53
nuclear@1 54 static void floor1_free_info(vorbis_info_floor *i){
nuclear@1 55 vorbis_info_floor1 *info=(vorbis_info_floor1 *)i;
nuclear@1 56 if(info){
nuclear@1 57 memset(info,0,sizeof(*info));
nuclear@1 58 _ogg_free(info);
nuclear@1 59 }
nuclear@1 60 }
nuclear@1 61
nuclear@1 62 static void floor1_free_look(vorbis_look_floor *i){
nuclear@1 63 vorbis_look_floor1 *look=(vorbis_look_floor1 *)i;
nuclear@1 64 if(look){
nuclear@1 65 /*fprintf(stderr,"floor 1 bit usage %f:%f (%f total)\n",
nuclear@1 66 (float)look->phrasebits/look->frames,
nuclear@1 67 (float)look->postbits/look->frames,
nuclear@1 68 (float)(look->postbits+look->phrasebits)/look->frames);*/
nuclear@1 69
nuclear@1 70 memset(look,0,sizeof(*look));
nuclear@1 71 _ogg_free(look);
nuclear@1 72 }
nuclear@1 73 }
nuclear@1 74
nuclear@1 75 static int ilog(unsigned int v){
nuclear@1 76 int ret=0;
nuclear@1 77 while(v){
nuclear@1 78 ret++;
nuclear@1 79 v>>=1;
nuclear@1 80 }
nuclear@1 81 return(ret);
nuclear@1 82 }
nuclear@1 83
nuclear@1 84 static int ilog2(unsigned int v){
nuclear@1 85 int ret=0;
nuclear@1 86 if(v)--v;
nuclear@1 87 while(v){
nuclear@1 88 ret++;
nuclear@1 89 v>>=1;
nuclear@1 90 }
nuclear@1 91 return(ret);
nuclear@1 92 }
nuclear@1 93
nuclear@1 94 static void floor1_pack (vorbis_info_floor *i,oggpack_buffer *opb){
nuclear@1 95 vorbis_info_floor1 *info=(vorbis_info_floor1 *)i;
nuclear@1 96 int j,k;
nuclear@1 97 int count=0;
nuclear@1 98 int rangebits;
nuclear@1 99 int maxposit=info->postlist[1];
nuclear@1 100 int maxclass=-1;
nuclear@1 101
nuclear@1 102 /* save out partitions */
nuclear@1 103 oggpack_write(opb,info->partitions,5); /* only 0 to 31 legal */
nuclear@1 104 for(j=0;j<info->partitions;j++){
nuclear@1 105 oggpack_write(opb,info->partitionclass[j],4); /* only 0 to 15 legal */
nuclear@1 106 if(maxclass<info->partitionclass[j])maxclass=info->partitionclass[j];
nuclear@1 107 }
nuclear@1 108
nuclear@1 109 /* save out partition classes */
nuclear@1 110 for(j=0;j<maxclass+1;j++){
nuclear@1 111 oggpack_write(opb,info->class_dim[j]-1,3); /* 1 to 8 */
nuclear@1 112 oggpack_write(opb,info->class_subs[j],2); /* 0 to 3 */
nuclear@1 113 if(info->class_subs[j])oggpack_write(opb,info->class_book[j],8);
nuclear@1 114 for(k=0;k<(1<<info->class_subs[j]);k++)
nuclear@1 115 oggpack_write(opb,info->class_subbook[j][k]+1,8);
nuclear@1 116 }
nuclear@1 117
nuclear@1 118 /* save out the post list */
nuclear@1 119 oggpack_write(opb,info->mult-1,2); /* only 1,2,3,4 legal now */
nuclear@1 120 oggpack_write(opb,ilog2(maxposit),4);
nuclear@1 121 rangebits=ilog2(maxposit);
nuclear@1 122
nuclear@1 123 for(j=0,k=0;j<info->partitions;j++){
nuclear@1 124 count+=info->class_dim[info->partitionclass[j]];
nuclear@1 125 for(;k<count;k++)
nuclear@1 126 oggpack_write(opb,info->postlist[k+2],rangebits);
nuclear@1 127 }
nuclear@1 128 }
nuclear@1 129
nuclear@1 130 static int icomp(const void *a,const void *b){
nuclear@1 131 return(**(int **)a-**(int **)b);
nuclear@1 132 }
nuclear@1 133
nuclear@1 134 static vorbis_info_floor *floor1_unpack (vorbis_info *vi,oggpack_buffer *opb){
nuclear@1 135 codec_setup_info *ci=vi->codec_setup;
nuclear@1 136 int j,k,count=0,maxclass=-1,rangebits;
nuclear@1 137
nuclear@1 138 vorbis_info_floor1 *info=_ogg_calloc(1,sizeof(*info));
nuclear@1 139 /* read partitions */
nuclear@1 140 info->partitions=oggpack_read(opb,5); /* only 0 to 31 legal */
nuclear@1 141 for(j=0;j<info->partitions;j++){
nuclear@1 142 info->partitionclass[j]=oggpack_read(opb,4); /* only 0 to 15 legal */
nuclear@1 143 if(info->partitionclass[j]<0)goto err_out;
nuclear@1 144 if(maxclass<info->partitionclass[j])maxclass=info->partitionclass[j];
nuclear@1 145 }
nuclear@1 146
nuclear@1 147 /* read partition classes */
nuclear@1 148 for(j=0;j<maxclass+1;j++){
nuclear@1 149 info->class_dim[j]=oggpack_read(opb,3)+1; /* 1 to 8 */
nuclear@1 150 info->class_subs[j]=oggpack_read(opb,2); /* 0,1,2,3 bits */
nuclear@1 151 if(info->class_subs[j]<0)
nuclear@1 152 goto err_out;
nuclear@1 153 if(info->class_subs[j])info->class_book[j]=oggpack_read(opb,8);
nuclear@1 154 if(info->class_book[j]<0 || info->class_book[j]>=ci->books)
nuclear@1 155 goto err_out;
nuclear@1 156 for(k=0;k<(1<<info->class_subs[j]);k++){
nuclear@1 157 info->class_subbook[j][k]=oggpack_read(opb,8)-1;
nuclear@1 158 if(info->class_subbook[j][k]<-1 || info->class_subbook[j][k]>=ci->books)
nuclear@1 159 goto err_out;
nuclear@1 160 }
nuclear@1 161 }
nuclear@1 162
nuclear@1 163 /* read the post list */
nuclear@1 164 info->mult=oggpack_read(opb,2)+1; /* only 1,2,3,4 legal now */
nuclear@1 165 rangebits=oggpack_read(opb,4);
nuclear@1 166 if(rangebits<0)goto err_out;
nuclear@1 167
nuclear@1 168 for(j=0,k=0;j<info->partitions;j++){
nuclear@1 169 count+=info->class_dim[info->partitionclass[j]];
nuclear@1 170 if(count>VIF_POSIT) goto err_out;
nuclear@1 171 for(;k<count;k++){
nuclear@1 172 int t=info->postlist[k+2]=oggpack_read(opb,rangebits);
nuclear@1 173 if(t<0 || t>=(1<<rangebits))
nuclear@1 174 goto err_out;
nuclear@1 175 }
nuclear@1 176 }
nuclear@1 177 info->postlist[0]=0;
nuclear@1 178 info->postlist[1]=1<<rangebits;
nuclear@1 179
nuclear@1 180 /* don't allow repeated values in post list as they'd result in
nuclear@1 181 zero-length segments */
nuclear@1 182 {
nuclear@1 183 int *sortpointer[VIF_POSIT+2];
nuclear@1 184 for(j=0;j<count+2;j++)sortpointer[j]=info->postlist+j;
nuclear@1 185 qsort(sortpointer,count+2,sizeof(*sortpointer),icomp);
nuclear@1 186
nuclear@1 187 for(j=1;j<count+2;j++)
nuclear@1 188 if(*sortpointer[j-1]==*sortpointer[j])goto err_out;
nuclear@1 189 }
nuclear@1 190
nuclear@1 191 return(info);
nuclear@1 192
nuclear@1 193 err_out:
nuclear@1 194 floor1_free_info(info);
nuclear@1 195 return(NULL);
nuclear@1 196 }
nuclear@1 197
nuclear@1 198 static vorbis_look_floor *floor1_look(vorbis_dsp_state *vd,
nuclear@1 199 vorbis_info_floor *in){
nuclear@1 200
nuclear@1 201 int *sortpointer[VIF_POSIT+2];
nuclear@1 202 vorbis_info_floor1 *info=(vorbis_info_floor1 *)in;
nuclear@1 203 vorbis_look_floor1 *look=_ogg_calloc(1,sizeof(*look));
nuclear@1 204 int i,j,n=0;
nuclear@1 205
nuclear@1 206 look->vi=info;
nuclear@1 207 look->n=info->postlist[1];
nuclear@1 208
nuclear@1 209 /* we drop each position value in-between already decoded values,
nuclear@1 210 and use linear interpolation to predict each new value past the
nuclear@1 211 edges. The positions are read in the order of the position
nuclear@1 212 list... we precompute the bounding positions in the lookup. Of
nuclear@1 213 course, the neighbors can change (if a position is declined), but
nuclear@1 214 this is an initial mapping */
nuclear@1 215
nuclear@1 216 for(i=0;i<info->partitions;i++)n+=info->class_dim[info->partitionclass[i]];
nuclear@1 217 n+=2;
nuclear@1 218 look->posts=n;
nuclear@1 219
nuclear@1 220 /* also store a sorted position index */
nuclear@1 221 for(i=0;i<n;i++)sortpointer[i]=info->postlist+i;
nuclear@1 222 qsort(sortpointer,n,sizeof(*sortpointer),icomp);
nuclear@1 223
nuclear@1 224 /* points from sort order back to range number */
nuclear@1 225 for(i=0;i<n;i++)look->forward_index[i]=sortpointer[i]-info->postlist;
nuclear@1 226 /* points from range order to sorted position */
nuclear@1 227 for(i=0;i<n;i++)look->reverse_index[look->forward_index[i]]=i;
nuclear@1 228 /* we actually need the post values too */
nuclear@1 229 for(i=0;i<n;i++)look->sorted_index[i]=info->postlist[look->forward_index[i]];
nuclear@1 230
nuclear@1 231 /* quantize values to multiplier spec */
nuclear@1 232 switch(info->mult){
nuclear@1 233 case 1: /* 1024 -> 256 */
nuclear@1 234 look->quant_q=256;
nuclear@1 235 break;
nuclear@1 236 case 2: /* 1024 -> 128 */
nuclear@1 237 look->quant_q=128;
nuclear@1 238 break;
nuclear@1 239 case 3: /* 1024 -> 86 */
nuclear@1 240 look->quant_q=86;
nuclear@1 241 break;
nuclear@1 242 case 4: /* 1024 -> 64 */
nuclear@1 243 look->quant_q=64;
nuclear@1 244 break;
nuclear@1 245 }
nuclear@1 246
nuclear@1 247 /* discover our neighbors for decode where we don't use fit flags
nuclear@1 248 (that would push the neighbors outward) */
nuclear@1 249 for(i=0;i<n-2;i++){
nuclear@1 250 int lo=0;
nuclear@1 251 int hi=1;
nuclear@1 252 int lx=0;
nuclear@1 253 int hx=look->n;
nuclear@1 254 int currentx=info->postlist[i+2];
nuclear@1 255 for(j=0;j<i+2;j++){
nuclear@1 256 int x=info->postlist[j];
nuclear@1 257 if(x>lx && x<currentx){
nuclear@1 258 lo=j;
nuclear@1 259 lx=x;
nuclear@1 260 }
nuclear@1 261 if(x<hx && x>currentx){
nuclear@1 262 hi=j;
nuclear@1 263 hx=x;
nuclear@1 264 }
nuclear@1 265 }
nuclear@1 266 look->loneighbor[i]=lo;
nuclear@1 267 look->hineighbor[i]=hi;
nuclear@1 268 }
nuclear@1 269
nuclear@1 270 return(look);
nuclear@1 271 }
nuclear@1 272
nuclear@1 273 static int render_point(int x0,int x1,int y0,int y1,int x){
nuclear@1 274 y0&=0x7fff; /* mask off flag */
nuclear@1 275 y1&=0x7fff;
nuclear@1 276
nuclear@1 277 {
nuclear@1 278 int dy=y1-y0;
nuclear@1 279 int adx=x1-x0;
nuclear@1 280 int ady=abs(dy);
nuclear@1 281 int err=ady*(x-x0);
nuclear@1 282
nuclear@1 283 int off=err/adx;
nuclear@1 284 if(dy<0)return(y0-off);
nuclear@1 285 return(y0+off);
nuclear@1 286 }
nuclear@1 287 }
nuclear@1 288
nuclear@1 289 static int vorbis_dBquant(const float *x){
nuclear@1 290 int i= *x*7.3142857f+1023.5f;
nuclear@1 291 if(i>1023)return(1023);
nuclear@1 292 if(i<0)return(0);
nuclear@1 293 return i;
nuclear@1 294 }
nuclear@1 295
nuclear@1 296 static const float FLOOR1_fromdB_LOOKUP[256]={
nuclear@1 297 1.0649863e-07F, 1.1341951e-07F, 1.2079015e-07F, 1.2863978e-07F,
nuclear@1 298 1.3699951e-07F, 1.4590251e-07F, 1.5538408e-07F, 1.6548181e-07F,
nuclear@1 299 1.7623575e-07F, 1.8768855e-07F, 1.9988561e-07F, 2.128753e-07F,
nuclear@1 300 2.2670913e-07F, 2.4144197e-07F, 2.5713223e-07F, 2.7384213e-07F,
nuclear@1 301 2.9163793e-07F, 3.1059021e-07F, 3.3077411e-07F, 3.5226968e-07F,
nuclear@1 302 3.7516214e-07F, 3.9954229e-07F, 4.2550680e-07F, 4.5315863e-07F,
nuclear@1 303 4.8260743e-07F, 5.1396998e-07F, 5.4737065e-07F, 5.8294187e-07F,
nuclear@1 304 6.2082472e-07F, 6.6116941e-07F, 7.0413592e-07F, 7.4989464e-07F,
nuclear@1 305 7.9862701e-07F, 8.5052630e-07F, 9.0579828e-07F, 9.6466216e-07F,
nuclear@1 306 1.0273513e-06F, 1.0941144e-06F, 1.1652161e-06F, 1.2409384e-06F,
nuclear@1 307 1.3215816e-06F, 1.4074654e-06F, 1.4989305e-06F, 1.5963394e-06F,
nuclear@1 308 1.7000785e-06F, 1.8105592e-06F, 1.9282195e-06F, 2.0535261e-06F,
nuclear@1 309 2.1869758e-06F, 2.3290978e-06F, 2.4804557e-06F, 2.6416497e-06F,
nuclear@1 310 2.8133190e-06F, 2.9961443e-06F, 3.1908506e-06F, 3.3982101e-06F,
nuclear@1 311 3.6190449e-06F, 3.8542308e-06F, 4.1047004e-06F, 4.3714470e-06F,
nuclear@1 312 4.6555282e-06F, 4.9580707e-06F, 5.2802740e-06F, 5.6234160e-06F,
nuclear@1 313 5.9888572e-06F, 6.3780469e-06F, 6.7925283e-06F, 7.2339451e-06F,
nuclear@1 314 7.7040476e-06F, 8.2047000e-06F, 8.7378876e-06F, 9.3057248e-06F,
nuclear@1 315 9.9104632e-06F, 1.0554501e-05F, 1.1240392e-05F, 1.1970856e-05F,
nuclear@1 316 1.2748789e-05F, 1.3577278e-05F, 1.4459606e-05F, 1.5399272e-05F,
nuclear@1 317 1.6400004e-05F, 1.7465768e-05F, 1.8600792e-05F, 1.9809576e-05F,
nuclear@1 318 2.1096914e-05F, 2.2467911e-05F, 2.3928002e-05F, 2.5482978e-05F,
nuclear@1 319 2.7139006e-05F, 2.8902651e-05F, 3.0780908e-05F, 3.2781225e-05F,
nuclear@1 320 3.4911534e-05F, 3.7180282e-05F, 3.9596466e-05F, 4.2169667e-05F,
nuclear@1 321 4.4910090e-05F, 4.7828601e-05F, 5.0936773e-05F, 5.4246931e-05F,
nuclear@1 322 5.7772202e-05F, 6.1526565e-05F, 6.5524908e-05F, 6.9783085e-05F,
nuclear@1 323 7.4317983e-05F, 7.9147585e-05F, 8.4291040e-05F, 8.9768747e-05F,
nuclear@1 324 9.5602426e-05F, 0.00010181521F, 0.00010843174F, 0.00011547824F,
nuclear@1 325 0.00012298267F, 0.00013097477F, 0.00013948625F, 0.00014855085F,
nuclear@1 326 0.00015820453F, 0.00016848555F, 0.00017943469F, 0.00019109536F,
nuclear@1 327 0.00020351382F, 0.00021673929F, 0.00023082423F, 0.00024582449F,
nuclear@1 328 0.00026179955F, 0.00027881276F, 0.00029693158F, 0.00031622787F,
nuclear@1 329 0.00033677814F, 0.00035866388F, 0.00038197188F, 0.00040679456F,
nuclear@1 330 0.00043323036F, 0.00046138411F, 0.00049136745F, 0.00052329927F,
nuclear@1 331 0.00055730621F, 0.00059352311F, 0.00063209358F, 0.00067317058F,
nuclear@1 332 0.00071691700F, 0.00076350630F, 0.00081312324F, 0.00086596457F,
nuclear@1 333 0.00092223983F, 0.00098217216F, 0.0010459992F, 0.0011139742F,
nuclear@1 334 0.0011863665F, 0.0012634633F, 0.0013455702F, 0.0014330129F,
nuclear@1 335 0.0015261382F, 0.0016253153F, 0.0017309374F, 0.0018434235F,
nuclear@1 336 0.0019632195F, 0.0020908006F, 0.0022266726F, 0.0023713743F,
nuclear@1 337 0.0025254795F, 0.0026895994F, 0.0028643847F, 0.0030505286F,
nuclear@1 338 0.0032487691F, 0.0034598925F, 0.0036847358F, 0.0039241906F,
nuclear@1 339 0.0041792066F, 0.0044507950F, 0.0047400328F, 0.0050480668F,
nuclear@1 340 0.0053761186F, 0.0057254891F, 0.0060975636F, 0.0064938176F,
nuclear@1 341 0.0069158225F, 0.0073652516F, 0.0078438871F, 0.0083536271F,
nuclear@1 342 0.0088964928F, 0.009474637F, 0.010090352F, 0.010746080F,
nuclear@1 343 0.011444421F, 0.012188144F, 0.012980198F, 0.013823725F,
nuclear@1 344 0.014722068F, 0.015678791F, 0.016697687F, 0.017782797F,
nuclear@1 345 0.018938423F, 0.020169149F, 0.021479854F, 0.022875735F,
nuclear@1 346 0.024362330F, 0.025945531F, 0.027631618F, 0.029427276F,
nuclear@1 347 0.031339626F, 0.033376252F, 0.035545228F, 0.037855157F,
nuclear@1 348 0.040315199F, 0.042935108F, 0.045725273F, 0.048696758F,
nuclear@1 349 0.051861348F, 0.055231591F, 0.058820850F, 0.062643361F,
nuclear@1 350 0.066714279F, 0.071049749F, 0.075666962F, 0.080584227F,
nuclear@1 351 0.085821044F, 0.091398179F, 0.097337747F, 0.10366330F,
nuclear@1 352 0.11039993F, 0.11757434F, 0.12521498F, 0.13335215F,
nuclear@1 353 0.14201813F, 0.15124727F, 0.16107617F, 0.17154380F,
nuclear@1 354 0.18269168F, 0.19456402F, 0.20720788F, 0.22067342F,
nuclear@1 355 0.23501402F, 0.25028656F, 0.26655159F, 0.28387361F,
nuclear@1 356 0.30232132F, 0.32196786F, 0.34289114F, 0.36517414F,
nuclear@1 357 0.38890521F, 0.41417847F, 0.44109412F, 0.46975890F,
nuclear@1 358 0.50028648F, 0.53279791F, 0.56742212F, 0.60429640F,
nuclear@1 359 0.64356699F, 0.68538959F, 0.72993007F, 0.77736504F,
nuclear@1 360 0.82788260F, 0.88168307F, 0.9389798F, 1.F,
nuclear@1 361 };
nuclear@1 362
nuclear@1 363 static void render_line(int n, int x0,int x1,int y0,int y1,float *d){
nuclear@1 364 int dy=y1-y0;
nuclear@1 365 int adx=x1-x0;
nuclear@1 366 int ady=abs(dy);
nuclear@1 367 int base=dy/adx;
nuclear@1 368 int sy=(dy<0?base-1:base+1);
nuclear@1 369 int x=x0;
nuclear@1 370 int y=y0;
nuclear@1 371 int err=0;
nuclear@1 372
nuclear@1 373 ady-=abs(base*adx);
nuclear@1 374
nuclear@1 375 if(n>x1)n=x1;
nuclear@1 376
nuclear@1 377 if(x<n)
nuclear@1 378 d[x]*=FLOOR1_fromdB_LOOKUP[y];
nuclear@1 379
nuclear@1 380 while(++x<n){
nuclear@1 381 err=err+ady;
nuclear@1 382 if(err>=adx){
nuclear@1 383 err-=adx;
nuclear@1 384 y+=sy;
nuclear@1 385 }else{
nuclear@1 386 y+=base;
nuclear@1 387 }
nuclear@1 388 d[x]*=FLOOR1_fromdB_LOOKUP[y];
nuclear@1 389 }
nuclear@1 390 }
nuclear@1 391
nuclear@1 392 static void render_line0(int n, int x0,int x1,int y0,int y1,int *d){
nuclear@1 393 int dy=y1-y0;
nuclear@1 394 int adx=x1-x0;
nuclear@1 395 int ady=abs(dy);
nuclear@1 396 int base=dy/adx;
nuclear@1 397 int sy=(dy<0?base-1:base+1);
nuclear@1 398 int x=x0;
nuclear@1 399 int y=y0;
nuclear@1 400 int err=0;
nuclear@1 401
nuclear@1 402 ady-=abs(base*adx);
nuclear@1 403
nuclear@1 404 if(n>x1)n=x1;
nuclear@1 405
nuclear@1 406 if(x<n)
nuclear@1 407 d[x]=y;
nuclear@1 408
nuclear@1 409 while(++x<n){
nuclear@1 410 err=err+ady;
nuclear@1 411 if(err>=adx){
nuclear@1 412 err-=adx;
nuclear@1 413 y+=sy;
nuclear@1 414 }else{
nuclear@1 415 y+=base;
nuclear@1 416 }
nuclear@1 417 d[x]=y;
nuclear@1 418 }
nuclear@1 419 }
nuclear@1 420
nuclear@1 421 /* the floor has already been filtered to only include relevant sections */
nuclear@1 422 static int accumulate_fit(const float *flr,const float *mdct,
nuclear@1 423 int x0, int x1,lsfit_acc *a,
nuclear@1 424 int n,vorbis_info_floor1 *info){
nuclear@1 425 long i;
nuclear@1 426
nuclear@1 427 int xa=0,ya=0,x2a=0,y2a=0,xya=0,na=0, xb=0,yb=0,x2b=0,y2b=0,xyb=0,nb=0;
nuclear@1 428
nuclear@1 429 memset(a,0,sizeof(*a));
nuclear@1 430 a->x0=x0;
nuclear@1 431 a->x1=x1;
nuclear@1 432 if(x1>=n)x1=n-1;
nuclear@1 433
nuclear@1 434 for(i=x0;i<=x1;i++){
nuclear@1 435 int quantized=vorbis_dBquant(flr+i);
nuclear@1 436 if(quantized){
nuclear@1 437 if(mdct[i]+info->twofitatten>=flr[i]){
nuclear@1 438 xa += i;
nuclear@1 439 ya += quantized;
nuclear@1 440 x2a += i*i;
nuclear@1 441 y2a += quantized*quantized;
nuclear@1 442 xya += i*quantized;
nuclear@1 443 na++;
nuclear@1 444 }else{
nuclear@1 445 xb += i;
nuclear@1 446 yb += quantized;
nuclear@1 447 x2b += i*i;
nuclear@1 448 y2b += quantized*quantized;
nuclear@1 449 xyb += i*quantized;
nuclear@1 450 nb++;
nuclear@1 451 }
nuclear@1 452 }
nuclear@1 453 }
nuclear@1 454
nuclear@1 455 a->xa=xa;
nuclear@1 456 a->ya=ya;
nuclear@1 457 a->x2a=x2a;
nuclear@1 458 a->y2a=y2a;
nuclear@1 459 a->xya=xya;
nuclear@1 460 a->an=na;
nuclear@1 461
nuclear@1 462 a->xb=xb;
nuclear@1 463 a->yb=yb;
nuclear@1 464 a->x2b=x2b;
nuclear@1 465 a->y2b=y2b;
nuclear@1 466 a->xyb=xyb;
nuclear@1 467 a->bn=nb;
nuclear@1 468
nuclear@1 469 return(na);
nuclear@1 470 }
nuclear@1 471
nuclear@1 472 static int fit_line(lsfit_acc *a,int fits,int *y0,int *y1,
nuclear@1 473 vorbis_info_floor1 *info){
nuclear@1 474 double xb=0,yb=0,x2b=0,y2b=0,xyb=0,bn=0;
nuclear@1 475 int i;
nuclear@1 476 int x0=a[0].x0;
nuclear@1 477 int x1=a[fits-1].x1;
nuclear@1 478
nuclear@1 479 for(i=0;i<fits;i++){
nuclear@1 480 double weight = (a[i].bn+a[i].an)*info->twofitweight/(a[i].an+1)+1.;
nuclear@1 481
nuclear@1 482 xb+=a[i].xb + a[i].xa * weight;
nuclear@1 483 yb+=a[i].yb + a[i].ya * weight;
nuclear@1 484 x2b+=a[i].x2b + a[i].x2a * weight;
nuclear@1 485 y2b+=a[i].y2b + a[i].y2a * weight;
nuclear@1 486 xyb+=a[i].xyb + a[i].xya * weight;
nuclear@1 487 bn+=a[i].bn + a[i].an * weight;
nuclear@1 488 }
nuclear@1 489
nuclear@1 490 if(*y0>=0){
nuclear@1 491 xb+= x0;
nuclear@1 492 yb+= *y0;
nuclear@1 493 x2b+= x0 * x0;
nuclear@1 494 y2b+= *y0 * *y0;
nuclear@1 495 xyb+= *y0 * x0;
nuclear@1 496 bn++;
nuclear@1 497 }
nuclear@1 498
nuclear@1 499 if(*y1>=0){
nuclear@1 500 xb+= x1;
nuclear@1 501 yb+= *y1;
nuclear@1 502 x2b+= x1 * x1;
nuclear@1 503 y2b+= *y1 * *y1;
nuclear@1 504 xyb+= *y1 * x1;
nuclear@1 505 bn++;
nuclear@1 506 }
nuclear@1 507
nuclear@1 508 {
nuclear@1 509 double denom=(bn*x2b-xb*xb);
nuclear@1 510
nuclear@1 511 if(denom>0.){
nuclear@1 512 double a=(yb*x2b-xyb*xb)/denom;
nuclear@1 513 double b=(bn*xyb-xb*yb)/denom;
nuclear@1 514 *y0=rint(a+b*x0);
nuclear@1 515 *y1=rint(a+b*x1);
nuclear@1 516
nuclear@1 517 /* limit to our range! */
nuclear@1 518 if(*y0>1023)*y0=1023;
nuclear@1 519 if(*y1>1023)*y1=1023;
nuclear@1 520 if(*y0<0)*y0=0;
nuclear@1 521 if(*y1<0)*y1=0;
nuclear@1 522
nuclear@1 523 return 0;
nuclear@1 524 }else{
nuclear@1 525 *y0=0;
nuclear@1 526 *y1=0;
nuclear@1 527 return 1;
nuclear@1 528 }
nuclear@1 529 }
nuclear@1 530 }
nuclear@1 531
nuclear@1 532 static int inspect_error(int x0,int x1,int y0,int y1,const float *mask,
nuclear@1 533 const float *mdct,
nuclear@1 534 vorbis_info_floor1 *info){
nuclear@1 535 int dy=y1-y0;
nuclear@1 536 int adx=x1-x0;
nuclear@1 537 int ady=abs(dy);
nuclear@1 538 int base=dy/adx;
nuclear@1 539 int sy=(dy<0?base-1:base+1);
nuclear@1 540 int x=x0;
nuclear@1 541 int y=y0;
nuclear@1 542 int err=0;
nuclear@1 543 int val=vorbis_dBquant(mask+x);
nuclear@1 544 int mse=0;
nuclear@1 545 int n=0;
nuclear@1 546
nuclear@1 547 ady-=abs(base*adx);
nuclear@1 548
nuclear@1 549 mse=(y-val);
nuclear@1 550 mse*=mse;
nuclear@1 551 n++;
nuclear@1 552 if(mdct[x]+info->twofitatten>=mask[x]){
nuclear@1 553 if(y+info->maxover<val)return(1);
nuclear@1 554 if(y-info->maxunder>val)return(1);
nuclear@1 555 }
nuclear@1 556
nuclear@1 557 while(++x<x1){
nuclear@1 558 err=err+ady;
nuclear@1 559 if(err>=adx){
nuclear@1 560 err-=adx;
nuclear@1 561 y+=sy;
nuclear@1 562 }else{
nuclear@1 563 y+=base;
nuclear@1 564 }
nuclear@1 565
nuclear@1 566 val=vorbis_dBquant(mask+x);
nuclear@1 567 mse+=((y-val)*(y-val));
nuclear@1 568 n++;
nuclear@1 569 if(mdct[x]+info->twofitatten>=mask[x]){
nuclear@1 570 if(val){
nuclear@1 571 if(y+info->maxover<val)return(1);
nuclear@1 572 if(y-info->maxunder>val)return(1);
nuclear@1 573 }
nuclear@1 574 }
nuclear@1 575 }
nuclear@1 576
nuclear@1 577 if(info->maxover*info->maxover/n>info->maxerr)return(0);
nuclear@1 578 if(info->maxunder*info->maxunder/n>info->maxerr)return(0);
nuclear@1 579 if(mse/n>info->maxerr)return(1);
nuclear@1 580 return(0);
nuclear@1 581 }
nuclear@1 582
nuclear@1 583 static int post_Y(int *A,int *B,int pos){
nuclear@1 584 if(A[pos]<0)
nuclear@1 585 return B[pos];
nuclear@1 586 if(B[pos]<0)
nuclear@1 587 return A[pos];
nuclear@1 588
nuclear@1 589 return (A[pos]+B[pos])>>1;
nuclear@1 590 }
nuclear@1 591
nuclear@1 592 int *floor1_fit(vorbis_block *vb,vorbis_look_floor1 *look,
nuclear@1 593 const float *logmdct, /* in */
nuclear@1 594 const float *logmask){
nuclear@1 595 long i,j;
nuclear@1 596 vorbis_info_floor1 *info=look->vi;
nuclear@1 597 long n=look->n;
nuclear@1 598 long posts=look->posts;
nuclear@1 599 long nonzero=0;
nuclear@1 600 lsfit_acc fits[VIF_POSIT+1];
nuclear@1 601 int fit_valueA[VIF_POSIT+2]; /* index by range list position */
nuclear@1 602 int fit_valueB[VIF_POSIT+2]; /* index by range list position */
nuclear@1 603
nuclear@1 604 int loneighbor[VIF_POSIT+2]; /* sorted index of range list position (+2) */
nuclear@1 605 int hineighbor[VIF_POSIT+2];
nuclear@1 606 int *output=NULL;
nuclear@1 607 int memo[VIF_POSIT+2];
nuclear@1 608
nuclear@1 609 for(i=0;i<posts;i++)fit_valueA[i]=-200; /* mark all unused */
nuclear@1 610 for(i=0;i<posts;i++)fit_valueB[i]=-200; /* mark all unused */
nuclear@1 611 for(i=0;i<posts;i++)loneighbor[i]=0; /* 0 for the implicit 0 post */
nuclear@1 612 for(i=0;i<posts;i++)hineighbor[i]=1; /* 1 for the implicit post at n */
nuclear@1 613 for(i=0;i<posts;i++)memo[i]=-1; /* no neighbor yet */
nuclear@1 614
nuclear@1 615 /* quantize the relevant floor points and collect them into line fit
nuclear@1 616 structures (one per minimal division) at the same time */
nuclear@1 617 if(posts==0){
nuclear@1 618 nonzero+=accumulate_fit(logmask,logmdct,0,n,fits,n,info);
nuclear@1 619 }else{
nuclear@1 620 for(i=0;i<posts-1;i++)
nuclear@1 621 nonzero+=accumulate_fit(logmask,logmdct,look->sorted_index[i],
nuclear@1 622 look->sorted_index[i+1],fits+i,
nuclear@1 623 n,info);
nuclear@1 624 }
nuclear@1 625
nuclear@1 626 if(nonzero){
nuclear@1 627 /* start by fitting the implicit base case.... */
nuclear@1 628 int y0=-200;
nuclear@1 629 int y1=-200;
nuclear@1 630 fit_line(fits,posts-1,&y0,&y1,info);
nuclear@1 631
nuclear@1 632 fit_valueA[0]=y0;
nuclear@1 633 fit_valueB[0]=y0;
nuclear@1 634 fit_valueB[1]=y1;
nuclear@1 635 fit_valueA[1]=y1;
nuclear@1 636
nuclear@1 637 /* Non degenerate case */
nuclear@1 638 /* start progressive splitting. This is a greedy, non-optimal
nuclear@1 639 algorithm, but simple and close enough to the best
nuclear@1 640 answer. */
nuclear@1 641 for(i=2;i<posts;i++){
nuclear@1 642 int sortpos=look->reverse_index[i];
nuclear@1 643 int ln=loneighbor[sortpos];
nuclear@1 644 int hn=hineighbor[sortpos];
nuclear@1 645
nuclear@1 646 /* eliminate repeat searches of a particular range with a memo */
nuclear@1 647 if(memo[ln]!=hn){
nuclear@1 648 /* haven't performed this error search yet */
nuclear@1 649 int lsortpos=look->reverse_index[ln];
nuclear@1 650 int hsortpos=look->reverse_index[hn];
nuclear@1 651 memo[ln]=hn;
nuclear@1 652
nuclear@1 653 {
nuclear@1 654 /* A note: we want to bound/minimize *local*, not global, error */
nuclear@1 655 int lx=info->postlist[ln];
nuclear@1 656 int hx=info->postlist[hn];
nuclear@1 657 int ly=post_Y(fit_valueA,fit_valueB,ln);
nuclear@1 658 int hy=post_Y(fit_valueA,fit_valueB,hn);
nuclear@1 659
nuclear@1 660 if(ly==-1 || hy==-1){
nuclear@1 661 exit(1);
nuclear@1 662 }
nuclear@1 663
nuclear@1 664 if(inspect_error(lx,hx,ly,hy,logmask,logmdct,info)){
nuclear@1 665 /* outside error bounds/begin search area. Split it. */
nuclear@1 666 int ly0=-200;
nuclear@1 667 int ly1=-200;
nuclear@1 668 int hy0=-200;
nuclear@1 669 int hy1=-200;
nuclear@1 670 int ret0=fit_line(fits+lsortpos,sortpos-lsortpos,&ly0,&ly1,info);
nuclear@1 671 int ret1=fit_line(fits+sortpos,hsortpos-sortpos,&hy0,&hy1,info);
nuclear@1 672
nuclear@1 673 if(ret0){
nuclear@1 674 ly0=ly;
nuclear@1 675 ly1=hy0;
nuclear@1 676 }
nuclear@1 677 if(ret1){
nuclear@1 678 hy0=ly1;
nuclear@1 679 hy1=hy;
nuclear@1 680 }
nuclear@1 681
nuclear@1 682 if(ret0 && ret1){
nuclear@1 683 fit_valueA[i]=-200;
nuclear@1 684 fit_valueB[i]=-200;
nuclear@1 685 }else{
nuclear@1 686 /* store new edge values */
nuclear@1 687 fit_valueB[ln]=ly0;
nuclear@1 688 if(ln==0)fit_valueA[ln]=ly0;
nuclear@1 689 fit_valueA[i]=ly1;
nuclear@1 690 fit_valueB[i]=hy0;
nuclear@1 691 fit_valueA[hn]=hy1;
nuclear@1 692 if(hn==1)fit_valueB[hn]=hy1;
nuclear@1 693
nuclear@1 694 if(ly1>=0 || hy0>=0){
nuclear@1 695 /* store new neighbor values */
nuclear@1 696 for(j=sortpos-1;j>=0;j--)
nuclear@1 697 if(hineighbor[j]==hn)
nuclear@1 698 hineighbor[j]=i;
nuclear@1 699 else
nuclear@1 700 break;
nuclear@1 701 for(j=sortpos+1;j<posts;j++)
nuclear@1 702 if(loneighbor[j]==ln)
nuclear@1 703 loneighbor[j]=i;
nuclear@1 704 else
nuclear@1 705 break;
nuclear@1 706 }
nuclear@1 707 }
nuclear@1 708 }else{
nuclear@1 709 fit_valueA[i]=-200;
nuclear@1 710 fit_valueB[i]=-200;
nuclear@1 711 }
nuclear@1 712 }
nuclear@1 713 }
nuclear@1 714 }
nuclear@1 715
nuclear@1 716 output=_vorbis_block_alloc(vb,sizeof(*output)*posts);
nuclear@1 717
nuclear@1 718 output[0]=post_Y(fit_valueA,fit_valueB,0);
nuclear@1 719 output[1]=post_Y(fit_valueA,fit_valueB,1);
nuclear@1 720
nuclear@1 721 /* fill in posts marked as not using a fit; we will zero
nuclear@1 722 back out to 'unused' when encoding them so long as curve
nuclear@1 723 interpolation doesn't force them into use */
nuclear@1 724 for(i=2;i<posts;i++){
nuclear@1 725 int ln=look->loneighbor[i-2];
nuclear@1 726 int hn=look->hineighbor[i-2];
nuclear@1 727 int x0=info->postlist[ln];
nuclear@1 728 int x1=info->postlist[hn];
nuclear@1 729 int y0=output[ln];
nuclear@1 730 int y1=output[hn];
nuclear@1 731
nuclear@1 732 int predicted=render_point(x0,x1,y0,y1,info->postlist[i]);
nuclear@1 733 int vx=post_Y(fit_valueA,fit_valueB,i);
nuclear@1 734
nuclear@1 735 if(vx>=0 && predicted!=vx){
nuclear@1 736 output[i]=vx;
nuclear@1 737 }else{
nuclear@1 738 output[i]= predicted|0x8000;
nuclear@1 739 }
nuclear@1 740 }
nuclear@1 741 }
nuclear@1 742
nuclear@1 743 return(output);
nuclear@1 744
nuclear@1 745 }
nuclear@1 746
nuclear@1 747 int *floor1_interpolate_fit(vorbis_block *vb,vorbis_look_floor1 *look,
nuclear@1 748 int *A,int *B,
nuclear@1 749 int del){
nuclear@1 750
nuclear@1 751 long i;
nuclear@1 752 long posts=look->posts;
nuclear@1 753 int *output=NULL;
nuclear@1 754
nuclear@1 755 if(A && B){
nuclear@1 756 output=_vorbis_block_alloc(vb,sizeof(*output)*posts);
nuclear@1 757
nuclear@1 758 /* overly simpleminded--- look again post 1.2 */
nuclear@1 759 for(i=0;i<posts;i++){
nuclear@1 760 output[i]=((65536-del)*(A[i]&0x7fff)+del*(B[i]&0x7fff)+32768)>>16;
nuclear@1 761 if(A[i]&0x8000 && B[i]&0x8000)output[i]|=0x8000;
nuclear@1 762 }
nuclear@1 763 }
nuclear@1 764
nuclear@1 765 return(output);
nuclear@1 766 }
nuclear@1 767
nuclear@1 768
nuclear@1 769 int floor1_encode(oggpack_buffer *opb,vorbis_block *vb,
nuclear@1 770 vorbis_look_floor1 *look,
nuclear@1 771 int *post,int *ilogmask){
nuclear@1 772
nuclear@1 773 long i,j;
nuclear@1 774 vorbis_info_floor1 *info=look->vi;
nuclear@1 775 long posts=look->posts;
nuclear@1 776 codec_setup_info *ci=vb->vd->vi->codec_setup;
nuclear@1 777 int out[VIF_POSIT+2];
nuclear@1 778 static_codebook **sbooks=ci->book_param;
nuclear@1 779 codebook *books=ci->fullbooks;
nuclear@1 780
nuclear@1 781 /* quantize values to multiplier spec */
nuclear@1 782 if(post){
nuclear@1 783 for(i=0;i<posts;i++){
nuclear@1 784 int val=post[i]&0x7fff;
nuclear@1 785 switch(info->mult){
nuclear@1 786 case 1: /* 1024 -> 256 */
nuclear@1 787 val>>=2;
nuclear@1 788 break;
nuclear@1 789 case 2: /* 1024 -> 128 */
nuclear@1 790 val>>=3;
nuclear@1 791 break;
nuclear@1 792 case 3: /* 1024 -> 86 */
nuclear@1 793 val/=12;
nuclear@1 794 break;
nuclear@1 795 case 4: /* 1024 -> 64 */
nuclear@1 796 val>>=4;
nuclear@1 797 break;
nuclear@1 798 }
nuclear@1 799 post[i]=val | (post[i]&0x8000);
nuclear@1 800 }
nuclear@1 801
nuclear@1 802 out[0]=post[0];
nuclear@1 803 out[1]=post[1];
nuclear@1 804
nuclear@1 805 /* find prediction values for each post and subtract them */
nuclear@1 806 for(i=2;i<posts;i++){
nuclear@1 807 int ln=look->loneighbor[i-2];
nuclear@1 808 int hn=look->hineighbor[i-2];
nuclear@1 809 int x0=info->postlist[ln];
nuclear@1 810 int x1=info->postlist[hn];
nuclear@1 811 int y0=post[ln];
nuclear@1 812 int y1=post[hn];
nuclear@1 813
nuclear@1 814 int predicted=render_point(x0,x1,y0,y1,info->postlist[i]);
nuclear@1 815
nuclear@1 816 if((post[i]&0x8000) || (predicted==post[i])){
nuclear@1 817 post[i]=predicted|0x8000; /* in case there was roundoff jitter
nuclear@1 818 in interpolation */
nuclear@1 819 out[i]=0;
nuclear@1 820 }else{
nuclear@1 821 int headroom=(look->quant_q-predicted<predicted?
nuclear@1 822 look->quant_q-predicted:predicted);
nuclear@1 823
nuclear@1 824 int val=post[i]-predicted;
nuclear@1 825
nuclear@1 826 /* at this point the 'deviation' value is in the range +/- max
nuclear@1 827 range, but the real, unique range can always be mapped to
nuclear@1 828 only [0-maxrange). So we want to wrap the deviation into
nuclear@1 829 this limited range, but do it in the way that least screws
nuclear@1 830 an essentially gaussian probability distribution. */
nuclear@1 831
nuclear@1 832 if(val<0)
nuclear@1 833 if(val<-headroom)
nuclear@1 834 val=headroom-val-1;
nuclear@1 835 else
nuclear@1 836 val=-1-(val<<1);
nuclear@1 837 else
nuclear@1 838 if(val>=headroom)
nuclear@1 839 val= val+headroom;
nuclear@1 840 else
nuclear@1 841 val<<=1;
nuclear@1 842
nuclear@1 843 out[i]=val;
nuclear@1 844 post[ln]&=0x7fff;
nuclear@1 845 post[hn]&=0x7fff;
nuclear@1 846 }
nuclear@1 847 }
nuclear@1 848
nuclear@1 849 /* we have everything we need. pack it out */
nuclear@1 850 /* mark nontrivial floor */
nuclear@1 851 oggpack_write(opb,1,1);
nuclear@1 852
nuclear@1 853 /* beginning/end post */
nuclear@1 854 look->frames++;
nuclear@1 855 look->postbits+=ilog(look->quant_q-1)*2;
nuclear@1 856 oggpack_write(opb,out[0],ilog(look->quant_q-1));
nuclear@1 857 oggpack_write(opb,out[1],ilog(look->quant_q-1));
nuclear@1 858
nuclear@1 859
nuclear@1 860 /* partition by partition */
nuclear@1 861 for(i=0,j=2;i<info->partitions;i++){
nuclear@1 862 int class=info->partitionclass[i];
nuclear@1 863 int cdim=info->class_dim[class];
nuclear@1 864 int csubbits=info->class_subs[class];
nuclear@1 865 int csub=1<<csubbits;
nuclear@1 866 int bookas[8]={0,0,0,0,0,0,0,0};
nuclear@1 867 int cval=0;
nuclear@1 868 int cshift=0;
nuclear@1 869 int k,l;
nuclear@1 870
nuclear@1 871 /* generate the partition's first stage cascade value */
nuclear@1 872 if(csubbits){
nuclear@1 873 int maxval[8];
nuclear@1 874 for(k=0;k<csub;k++){
nuclear@1 875 int booknum=info->class_subbook[class][k];
nuclear@1 876 if(booknum<0){
nuclear@1 877 maxval[k]=1;
nuclear@1 878 }else{
nuclear@1 879 maxval[k]=sbooks[info->class_subbook[class][k]]->entries;
nuclear@1 880 }
nuclear@1 881 }
nuclear@1 882 for(k=0;k<cdim;k++){
nuclear@1 883 for(l=0;l<csub;l++){
nuclear@1 884 int val=out[j+k];
nuclear@1 885 if(val<maxval[l]){
nuclear@1 886 bookas[k]=l;
nuclear@1 887 break;
nuclear@1 888 }
nuclear@1 889 }
nuclear@1 890 cval|= bookas[k]<<cshift;
nuclear@1 891 cshift+=csubbits;
nuclear@1 892 }
nuclear@1 893 /* write it */
nuclear@1 894 look->phrasebits+=
nuclear@1 895 vorbis_book_encode(books+info->class_book[class],cval,opb);
nuclear@1 896
nuclear@1 897 #ifdef TRAIN_FLOOR1
nuclear@1 898 {
nuclear@1 899 FILE *of;
nuclear@1 900 char buffer[80];
nuclear@1 901 sprintf(buffer,"line_%dx%ld_class%d.vqd",
nuclear@1 902 vb->pcmend/2,posts-2,class);
nuclear@1 903 of=fopen(buffer,"a");
nuclear@1 904 fprintf(of,"%d\n",cval);
nuclear@1 905 fclose(of);
nuclear@1 906 }
nuclear@1 907 #endif
nuclear@1 908 }
nuclear@1 909
nuclear@1 910 /* write post values */
nuclear@1 911 for(k=0;k<cdim;k++){
nuclear@1 912 int book=info->class_subbook[class][bookas[k]];
nuclear@1 913 if(book>=0){
nuclear@1 914 /* hack to allow training with 'bad' books */
nuclear@1 915 if(out[j+k]<(books+book)->entries)
nuclear@1 916 look->postbits+=vorbis_book_encode(books+book,
nuclear@1 917 out[j+k],opb);
nuclear@1 918 /*else
nuclear@1 919 fprintf(stderr,"+!");*/
nuclear@1 920
nuclear@1 921 #ifdef TRAIN_FLOOR1
nuclear@1 922 {
nuclear@1 923 FILE *of;
nuclear@1 924 char buffer[80];
nuclear@1 925 sprintf(buffer,"line_%dx%ld_%dsub%d.vqd",
nuclear@1 926 vb->pcmend/2,posts-2,class,bookas[k]);
nuclear@1 927 of=fopen(buffer,"a");
nuclear@1 928 fprintf(of,"%d\n",out[j+k]);
nuclear@1 929 fclose(of);
nuclear@1 930 }
nuclear@1 931 #endif
nuclear@1 932 }
nuclear@1 933 }
nuclear@1 934 j+=cdim;
nuclear@1 935 }
nuclear@1 936
nuclear@1 937 {
nuclear@1 938 /* generate quantized floor equivalent to what we'd unpack in decode */
nuclear@1 939 /* render the lines */
nuclear@1 940 int hx=0;
nuclear@1 941 int lx=0;
nuclear@1 942 int ly=post[0]*info->mult;
nuclear@1 943 int n=ci->blocksizes[vb->W]/2;
nuclear@1 944
nuclear@1 945 for(j=1;j<look->posts;j++){
nuclear@1 946 int current=look->forward_index[j];
nuclear@1 947 int hy=post[current]&0x7fff;
nuclear@1 948 if(hy==post[current]){
nuclear@1 949
nuclear@1 950 hy*=info->mult;
nuclear@1 951 hx=info->postlist[current];
nuclear@1 952
nuclear@1 953 render_line0(n,lx,hx,ly,hy,ilogmask);
nuclear@1 954
nuclear@1 955 lx=hx;
nuclear@1 956 ly=hy;
nuclear@1 957 }
nuclear@1 958 }
nuclear@1 959 for(j=hx;j<vb->pcmend/2;j++)ilogmask[j]=ly; /* be certain */
nuclear@1 960 return(1);
nuclear@1 961 }
nuclear@1 962 }else{
nuclear@1 963 oggpack_write(opb,0,1);
nuclear@1 964 memset(ilogmask,0,vb->pcmend/2*sizeof(*ilogmask));
nuclear@1 965 return(0);
nuclear@1 966 }
nuclear@1 967 }
nuclear@1 968
nuclear@1 969 static void *floor1_inverse1(vorbis_block *vb,vorbis_look_floor *in){
nuclear@1 970 vorbis_look_floor1 *look=(vorbis_look_floor1 *)in;
nuclear@1 971 vorbis_info_floor1 *info=look->vi;
nuclear@1 972 codec_setup_info *ci=vb->vd->vi->codec_setup;
nuclear@1 973
nuclear@1 974 int i,j,k;
nuclear@1 975 codebook *books=ci->fullbooks;
nuclear@1 976
nuclear@1 977 /* unpack wrapped/predicted values from stream */
nuclear@1 978 if(oggpack_read(&vb->opb,1)==1){
nuclear@1 979 int *fit_value=_vorbis_block_alloc(vb,(look->posts)*sizeof(*fit_value));
nuclear@1 980
nuclear@1 981 fit_value[0]=oggpack_read(&vb->opb,ilog(look->quant_q-1));
nuclear@1 982 fit_value[1]=oggpack_read(&vb->opb,ilog(look->quant_q-1));
nuclear@1 983
nuclear@1 984 /* partition by partition */
nuclear@1 985 for(i=0,j=2;i<info->partitions;i++){
nuclear@1 986 int class=info->partitionclass[i];
nuclear@1 987 int cdim=info->class_dim[class];
nuclear@1 988 int csubbits=info->class_subs[class];
nuclear@1 989 int csub=1<<csubbits;
nuclear@1 990 int cval=0;
nuclear@1 991
nuclear@1 992 /* decode the partition's first stage cascade value */
nuclear@1 993 if(csubbits){
nuclear@1 994 cval=vorbis_book_decode(books+info->class_book[class],&vb->opb);
nuclear@1 995
nuclear@1 996 if(cval==-1)goto eop;
nuclear@1 997 }
nuclear@1 998
nuclear@1 999 for(k=0;k<cdim;k++){
nuclear@1 1000 int book=info->class_subbook[class][cval&(csub-1)];
nuclear@1 1001 cval>>=csubbits;
nuclear@1 1002 if(book>=0){
nuclear@1 1003 if((fit_value[j+k]=vorbis_book_decode(books+book,&vb->opb))==-1)
nuclear@1 1004 goto eop;
nuclear@1 1005 }else{
nuclear@1 1006 fit_value[j+k]=0;
nuclear@1 1007 }
nuclear@1 1008 }
nuclear@1 1009 j+=cdim;
nuclear@1 1010 }
nuclear@1 1011
nuclear@1 1012 /* unwrap positive values and reconsitute via linear interpolation */
nuclear@1 1013 for(i=2;i<look->posts;i++){
nuclear@1 1014 int predicted=render_point(info->postlist[look->loneighbor[i-2]],
nuclear@1 1015 info->postlist[look->hineighbor[i-2]],
nuclear@1 1016 fit_value[look->loneighbor[i-2]],
nuclear@1 1017 fit_value[look->hineighbor[i-2]],
nuclear@1 1018 info->postlist[i]);
nuclear@1 1019 int hiroom=look->quant_q-predicted;
nuclear@1 1020 int loroom=predicted;
nuclear@1 1021 int room=(hiroom<loroom?hiroom:loroom)<<1;
nuclear@1 1022 int val=fit_value[i];
nuclear@1 1023
nuclear@1 1024 if(val){
nuclear@1 1025 if(val>=room){
nuclear@1 1026 if(hiroom>loroom){
nuclear@1 1027 val = val-loroom;
nuclear@1 1028 }else{
nuclear@1 1029 val = -1-(val-hiroom);
nuclear@1 1030 }
nuclear@1 1031 }else{
nuclear@1 1032 if(val&1){
nuclear@1 1033 val= -((val+1)>>1);
nuclear@1 1034 }else{
nuclear@1 1035 val>>=1;
nuclear@1 1036 }
nuclear@1 1037 }
nuclear@1 1038
nuclear@1 1039 fit_value[i]=(val+predicted)&0x7fff;
nuclear@1 1040 fit_value[look->loneighbor[i-2]]&=0x7fff;
nuclear@1 1041 fit_value[look->hineighbor[i-2]]&=0x7fff;
nuclear@1 1042
nuclear@1 1043 }else{
nuclear@1 1044 fit_value[i]=predicted|0x8000;
nuclear@1 1045 }
nuclear@1 1046
nuclear@1 1047 }
nuclear@1 1048
nuclear@1 1049 return(fit_value);
nuclear@1 1050 }
nuclear@1 1051 eop:
nuclear@1 1052 return(NULL);
nuclear@1 1053 }
nuclear@1 1054
nuclear@1 1055 static int floor1_inverse2(vorbis_block *vb,vorbis_look_floor *in,void *memo,
nuclear@1 1056 float *out){
nuclear@1 1057 vorbis_look_floor1 *look=(vorbis_look_floor1 *)in;
nuclear@1 1058 vorbis_info_floor1 *info=look->vi;
nuclear@1 1059
nuclear@1 1060 codec_setup_info *ci=vb->vd->vi->codec_setup;
nuclear@1 1061 int n=ci->blocksizes[vb->W]/2;
nuclear@1 1062 int j;
nuclear@1 1063
nuclear@1 1064 if(memo){
nuclear@1 1065 /* render the lines */
nuclear@1 1066 int *fit_value=(int *)memo;
nuclear@1 1067 int hx=0;
nuclear@1 1068 int lx=0;
nuclear@1 1069 int ly=fit_value[0]*info->mult;
nuclear@1 1070 /* guard lookup against out-of-range values */
nuclear@1 1071 ly=(ly<0?0:ly>255?255:ly);
nuclear@1 1072
nuclear@1 1073 for(j=1;j<look->posts;j++){
nuclear@1 1074 int current=look->forward_index[j];
nuclear@1 1075 int hy=fit_value[current]&0x7fff;
nuclear@1 1076 if(hy==fit_value[current]){
nuclear@1 1077
nuclear@1 1078 hx=info->postlist[current];
nuclear@1 1079 hy*=info->mult;
nuclear@1 1080 /* guard lookup against out-of-range values */
nuclear@1 1081 hy=(hy<0?0:hy>255?255:hy);
nuclear@1 1082
nuclear@1 1083 render_line(n,lx,hx,ly,hy,out);
nuclear@1 1084
nuclear@1 1085 lx=hx;
nuclear@1 1086 ly=hy;
nuclear@1 1087 }
nuclear@1 1088 }
nuclear@1 1089 for(j=hx;j<n;j++)out[j]*=FLOOR1_fromdB_LOOKUP[ly]; /* be certain */
nuclear@1 1090 return(1);
nuclear@1 1091 }
nuclear@1 1092 memset(out,0,sizeof(*out)*n);
nuclear@1 1093 return(0);
nuclear@1 1094 }
nuclear@1 1095
nuclear@1 1096 /* export hooks */
nuclear@1 1097 const vorbis_func_floor floor1_exportbundle={
nuclear@1 1098 &floor1_pack,&floor1_unpack,&floor1_look,&floor1_free_info,
nuclear@1 1099 &floor1_free_look,&floor1_inverse1,&floor1_inverse2
nuclear@1 1100 };