annotate src/libvorbis-1.3.3/lib/res0.c @ 141:1b5b6dfd0d0e

Add updated build of PortAudio for OSX
author Chris Cannam <cannam@all-day-breakfast.com>
date Tue, 03 Jan 2017 15:10:52 +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: residue backend 0, 1 and 2 implementation
cannam@86 14 last mod: $Id: res0.c 17556 2010-10-21 18:25:19Z tterribe $
cannam@86 15
cannam@86 16 ********************************************************************/
cannam@86 17
cannam@86 18 /* Slow, slow, slow, simpleminded and did I mention it was slow? The
cannam@86 19 encode/decode loops are coded for clarity and performance is not
cannam@86 20 yet even a nagging little idea lurking in the shadows. Oh and BTW,
cannam@86 21 it's slow. */
cannam@86 22
cannam@86 23 #include <stdlib.h>
cannam@86 24 #include <string.h>
cannam@86 25 #include <math.h>
cannam@86 26 #include <ogg/ogg.h>
cannam@86 27 #include "vorbis/codec.h"
cannam@86 28 #include "codec_internal.h"
cannam@86 29 #include "registry.h"
cannam@86 30 #include "codebook.h"
cannam@86 31 #include "misc.h"
cannam@86 32 #include "os.h"
cannam@86 33
cannam@86 34 //#define TRAIN_RES 1
cannam@86 35 //#define TRAIN_RESAUX 1
cannam@86 36
cannam@86 37 #if defined(TRAIN_RES) || defined (TRAIN_RESAUX)
cannam@86 38 #include <stdio.h>
cannam@86 39 #endif
cannam@86 40
cannam@86 41 typedef struct {
cannam@86 42 vorbis_info_residue0 *info;
cannam@86 43
cannam@86 44 int parts;
cannam@86 45 int stages;
cannam@86 46 codebook *fullbooks;
cannam@86 47 codebook *phrasebook;
cannam@86 48 codebook ***partbooks;
cannam@86 49
cannam@86 50 int partvals;
cannam@86 51 int **decodemap;
cannam@86 52
cannam@86 53 long postbits;
cannam@86 54 long phrasebits;
cannam@86 55 long frames;
cannam@86 56
cannam@86 57 #if defined(TRAIN_RES) || defined(TRAIN_RESAUX)
cannam@86 58 int train_seq;
cannam@86 59 long *training_data[8][64];
cannam@86 60 float training_max[8][64];
cannam@86 61 float training_min[8][64];
cannam@86 62 float tmin;
cannam@86 63 float tmax;
cannam@86 64 int submap;
cannam@86 65 #endif
cannam@86 66
cannam@86 67 } vorbis_look_residue0;
cannam@86 68
cannam@86 69 void res0_free_info(vorbis_info_residue *i){
cannam@86 70 vorbis_info_residue0 *info=(vorbis_info_residue0 *)i;
cannam@86 71 if(info){
cannam@86 72 memset(info,0,sizeof(*info));
cannam@86 73 _ogg_free(info);
cannam@86 74 }
cannam@86 75 }
cannam@86 76
cannam@86 77 void res0_free_look(vorbis_look_residue *i){
cannam@86 78 int j;
cannam@86 79 if(i){
cannam@86 80
cannam@86 81 vorbis_look_residue0 *look=(vorbis_look_residue0 *)i;
cannam@86 82
cannam@86 83 #ifdef TRAIN_RES
cannam@86 84 {
cannam@86 85 int j,k,l;
cannam@86 86 for(j=0;j<look->parts;j++){
cannam@86 87 /*fprintf(stderr,"partition %d: ",j);*/
cannam@86 88 for(k=0;k<8;k++)
cannam@86 89 if(look->training_data[k][j]){
cannam@86 90 char buffer[80];
cannam@86 91 FILE *of;
cannam@86 92 codebook *statebook=look->partbooks[j][k];
cannam@86 93
cannam@86 94 /* long and short into the same bucket by current convention */
cannam@86 95 sprintf(buffer,"res_sub%d_part%d_pass%d.vqd",look->submap,j,k);
cannam@86 96 of=fopen(buffer,"a");
cannam@86 97
cannam@86 98 for(l=0;l<statebook->entries;l++)
cannam@86 99 fprintf(of,"%d:%ld\n",l,look->training_data[k][j][l]);
cannam@86 100
cannam@86 101 fclose(of);
cannam@86 102
cannam@86 103 /*fprintf(stderr,"%d(%.2f|%.2f) ",k,
cannam@86 104 look->training_min[k][j],look->training_max[k][j]);*/
cannam@86 105
cannam@86 106 _ogg_free(look->training_data[k][j]);
cannam@86 107 look->training_data[k][j]=NULL;
cannam@86 108 }
cannam@86 109 /*fprintf(stderr,"\n");*/
cannam@86 110 }
cannam@86 111 }
cannam@86 112 fprintf(stderr,"min/max residue: %g::%g\n",look->tmin,look->tmax);
cannam@86 113
cannam@86 114 /*fprintf(stderr,"residue bit usage %f:%f (%f total)\n",
cannam@86 115 (float)look->phrasebits/look->frames,
cannam@86 116 (float)look->postbits/look->frames,
cannam@86 117 (float)(look->postbits+look->phrasebits)/look->frames);*/
cannam@86 118 #endif
cannam@86 119
cannam@86 120
cannam@86 121 /*vorbis_info_residue0 *info=look->info;
cannam@86 122
cannam@86 123 fprintf(stderr,
cannam@86 124 "%ld frames encoded in %ld phrasebits and %ld residue bits "
cannam@86 125 "(%g/frame) \n",look->frames,look->phrasebits,
cannam@86 126 look->resbitsflat,
cannam@86 127 (look->phrasebits+look->resbitsflat)/(float)look->frames);
cannam@86 128
cannam@86 129 for(j=0;j<look->parts;j++){
cannam@86 130 long acc=0;
cannam@86 131 fprintf(stderr,"\t[%d] == ",j);
cannam@86 132 for(k=0;k<look->stages;k++)
cannam@86 133 if((info->secondstages[j]>>k)&1){
cannam@86 134 fprintf(stderr,"%ld,",look->resbits[j][k]);
cannam@86 135 acc+=look->resbits[j][k];
cannam@86 136 }
cannam@86 137
cannam@86 138 fprintf(stderr,":: (%ld vals) %1.2fbits/sample\n",look->resvals[j],
cannam@86 139 acc?(float)acc/(look->resvals[j]*info->grouping):0);
cannam@86 140 }
cannam@86 141 fprintf(stderr,"\n");*/
cannam@86 142
cannam@86 143 for(j=0;j<look->parts;j++)
cannam@86 144 if(look->partbooks[j])_ogg_free(look->partbooks[j]);
cannam@86 145 _ogg_free(look->partbooks);
cannam@86 146 for(j=0;j<look->partvals;j++)
cannam@86 147 _ogg_free(look->decodemap[j]);
cannam@86 148 _ogg_free(look->decodemap);
cannam@86 149
cannam@86 150 memset(look,0,sizeof(*look));
cannam@86 151 _ogg_free(look);
cannam@86 152 }
cannam@86 153 }
cannam@86 154
cannam@86 155 static int ilog(unsigned int v){
cannam@86 156 int ret=0;
cannam@86 157 while(v){
cannam@86 158 ret++;
cannam@86 159 v>>=1;
cannam@86 160 }
cannam@86 161 return(ret);
cannam@86 162 }
cannam@86 163
cannam@86 164 static int icount(unsigned int v){
cannam@86 165 int ret=0;
cannam@86 166 while(v){
cannam@86 167 ret+=v&1;
cannam@86 168 v>>=1;
cannam@86 169 }
cannam@86 170 return(ret);
cannam@86 171 }
cannam@86 172
cannam@86 173
cannam@86 174 void res0_pack(vorbis_info_residue *vr,oggpack_buffer *opb){
cannam@86 175 vorbis_info_residue0 *info=(vorbis_info_residue0 *)vr;
cannam@86 176 int j,acc=0;
cannam@86 177 oggpack_write(opb,info->begin,24);
cannam@86 178 oggpack_write(opb,info->end,24);
cannam@86 179
cannam@86 180 oggpack_write(opb,info->grouping-1,24); /* residue vectors to group and
cannam@86 181 code with a partitioned book */
cannam@86 182 oggpack_write(opb,info->partitions-1,6); /* possible partition choices */
cannam@86 183 oggpack_write(opb,info->groupbook,8); /* group huffman book */
cannam@86 184
cannam@86 185 /* secondstages is a bitmask; as encoding progresses pass by pass, a
cannam@86 186 bitmask of one indicates this partition class has bits to write
cannam@86 187 this pass */
cannam@86 188 for(j=0;j<info->partitions;j++){
cannam@86 189 if(ilog(info->secondstages[j])>3){
cannam@86 190 /* yes, this is a minor hack due to not thinking ahead */
cannam@86 191 oggpack_write(opb,info->secondstages[j],3);
cannam@86 192 oggpack_write(opb,1,1);
cannam@86 193 oggpack_write(opb,info->secondstages[j]>>3,5);
cannam@86 194 }else
cannam@86 195 oggpack_write(opb,info->secondstages[j],4); /* trailing zero */
cannam@86 196 acc+=icount(info->secondstages[j]);
cannam@86 197 }
cannam@86 198 for(j=0;j<acc;j++)
cannam@86 199 oggpack_write(opb,info->booklist[j],8);
cannam@86 200
cannam@86 201 }
cannam@86 202
cannam@86 203 /* vorbis_info is for range checking */
cannam@86 204 vorbis_info_residue *res0_unpack(vorbis_info *vi,oggpack_buffer *opb){
cannam@86 205 int j,acc=0;
cannam@86 206 vorbis_info_residue0 *info=_ogg_calloc(1,sizeof(*info));
cannam@86 207 codec_setup_info *ci=vi->codec_setup;
cannam@86 208
cannam@86 209 info->begin=oggpack_read(opb,24);
cannam@86 210 info->end=oggpack_read(opb,24);
cannam@86 211 info->grouping=oggpack_read(opb,24)+1;
cannam@86 212 info->partitions=oggpack_read(opb,6)+1;
cannam@86 213 info->groupbook=oggpack_read(opb,8);
cannam@86 214
cannam@86 215 /* check for premature EOP */
cannam@86 216 if(info->groupbook<0)goto errout;
cannam@86 217
cannam@86 218 for(j=0;j<info->partitions;j++){
cannam@86 219 int cascade=oggpack_read(opb,3);
cannam@86 220 int cflag=oggpack_read(opb,1);
cannam@86 221 if(cflag<0) goto errout;
cannam@86 222 if(cflag){
cannam@86 223 int c=oggpack_read(opb,5);
cannam@86 224 if(c<0) goto errout;
cannam@86 225 cascade|=(c<<3);
cannam@86 226 }
cannam@86 227 info->secondstages[j]=cascade;
cannam@86 228
cannam@86 229 acc+=icount(cascade);
cannam@86 230 }
cannam@86 231 for(j=0;j<acc;j++){
cannam@86 232 int book=oggpack_read(opb,8);
cannam@86 233 if(book<0) goto errout;
cannam@86 234 info->booklist[j]=book;
cannam@86 235 }
cannam@86 236
cannam@86 237 if(info->groupbook>=ci->books)goto errout;
cannam@86 238 for(j=0;j<acc;j++){
cannam@86 239 if(info->booklist[j]>=ci->books)goto errout;
cannam@86 240 if(ci->book_param[info->booklist[j]]->maptype==0)goto errout;
cannam@86 241 }
cannam@86 242
cannam@86 243 /* verify the phrasebook is not specifying an impossible or
cannam@86 244 inconsistent partitioning scheme. */
cannam@86 245 /* modify the phrasebook ranging check from r16327; an early beta
cannam@86 246 encoder had a bug where it used an oversized phrasebook by
cannam@86 247 accident. These files should continue to be playable, but don't
cannam@86 248 allow an exploit */
cannam@86 249 {
cannam@86 250 int entries = ci->book_param[info->groupbook]->entries;
cannam@86 251 int dim = ci->book_param[info->groupbook]->dim;
cannam@86 252 int partvals = 1;
cannam@86 253 if (dim<1) goto errout;
cannam@86 254 while(dim>0){
cannam@86 255 partvals *= info->partitions;
cannam@86 256 if(partvals > entries) goto errout;
cannam@86 257 dim--;
cannam@86 258 }
cannam@86 259 info->partvals = partvals;
cannam@86 260 }
cannam@86 261
cannam@86 262 return(info);
cannam@86 263 errout:
cannam@86 264 res0_free_info(info);
cannam@86 265 return(NULL);
cannam@86 266 }
cannam@86 267
cannam@86 268 vorbis_look_residue *res0_look(vorbis_dsp_state *vd,
cannam@86 269 vorbis_info_residue *vr){
cannam@86 270 vorbis_info_residue0 *info=(vorbis_info_residue0 *)vr;
cannam@86 271 vorbis_look_residue0 *look=_ogg_calloc(1,sizeof(*look));
cannam@86 272 codec_setup_info *ci=vd->vi->codec_setup;
cannam@86 273
cannam@86 274 int j,k,acc=0;
cannam@86 275 int dim;
cannam@86 276 int maxstage=0;
cannam@86 277 look->info=info;
cannam@86 278
cannam@86 279 look->parts=info->partitions;
cannam@86 280 look->fullbooks=ci->fullbooks;
cannam@86 281 look->phrasebook=ci->fullbooks+info->groupbook;
cannam@86 282 dim=look->phrasebook->dim;
cannam@86 283
cannam@86 284 look->partbooks=_ogg_calloc(look->parts,sizeof(*look->partbooks));
cannam@86 285
cannam@86 286 for(j=0;j<look->parts;j++){
cannam@86 287 int stages=ilog(info->secondstages[j]);
cannam@86 288 if(stages){
cannam@86 289 if(stages>maxstage)maxstage=stages;
cannam@86 290 look->partbooks[j]=_ogg_calloc(stages,sizeof(*look->partbooks[j]));
cannam@86 291 for(k=0;k<stages;k++)
cannam@86 292 if(info->secondstages[j]&(1<<k)){
cannam@86 293 look->partbooks[j][k]=ci->fullbooks+info->booklist[acc++];
cannam@86 294 #ifdef TRAIN_RES
cannam@86 295 look->training_data[k][j]=_ogg_calloc(look->partbooks[j][k]->entries,
cannam@86 296 sizeof(***look->training_data));
cannam@86 297 #endif
cannam@86 298 }
cannam@86 299 }
cannam@86 300 }
cannam@86 301
cannam@86 302 look->partvals=1;
cannam@86 303 for(j=0;j<dim;j++)
cannam@86 304 look->partvals*=look->parts;
cannam@86 305
cannam@86 306 look->stages=maxstage;
cannam@86 307 look->decodemap=_ogg_malloc(look->partvals*sizeof(*look->decodemap));
cannam@86 308 for(j=0;j<look->partvals;j++){
cannam@86 309 long val=j;
cannam@86 310 long mult=look->partvals/look->parts;
cannam@86 311 look->decodemap[j]=_ogg_malloc(dim*sizeof(*look->decodemap[j]));
cannam@86 312 for(k=0;k<dim;k++){
cannam@86 313 long deco=val/mult;
cannam@86 314 val-=deco*mult;
cannam@86 315 mult/=look->parts;
cannam@86 316 look->decodemap[j][k]=deco;
cannam@86 317 }
cannam@86 318 }
cannam@86 319 #if defined(TRAIN_RES) || defined (TRAIN_RESAUX)
cannam@86 320 {
cannam@86 321 static int train_seq=0;
cannam@86 322 look->train_seq=train_seq++;
cannam@86 323 }
cannam@86 324 #endif
cannam@86 325 return(look);
cannam@86 326 }
cannam@86 327
cannam@86 328 /* break an abstraction and copy some code for performance purposes */
cannam@86 329 static int local_book_besterror(codebook *book,int *a){
cannam@86 330 int dim=book->dim;
cannam@86 331 int i,j,o;
cannam@86 332 int minval=book->minval;
cannam@86 333 int del=book->delta;
cannam@86 334 int qv=book->quantvals;
cannam@86 335 int ze=(qv>>1);
cannam@86 336 int index=0;
cannam@86 337 /* assumes integer/centered encoder codebook maptype 1 no more than dim 8 */
cannam@86 338 int p[8]={0,0,0,0,0,0,0,0};
cannam@86 339
cannam@86 340 if(del!=1){
cannam@86 341 for(i=0,o=dim;i<dim;i++){
cannam@86 342 int v = (a[--o]-minval+(del>>1))/del;
cannam@86 343 int m = (v<ze ? ((ze-v)<<1)-1 : ((v-ze)<<1));
cannam@86 344 index = index*qv+ (m<0?0:(m>=qv?qv-1:m));
cannam@86 345 p[o]=v*del+minval;
cannam@86 346 }
cannam@86 347 }else{
cannam@86 348 for(i=0,o=dim;i<dim;i++){
cannam@86 349 int v = a[--o]-minval;
cannam@86 350 int m = (v<ze ? ((ze-v)<<1)-1 : ((v-ze)<<1));
cannam@86 351 index = index*qv+ (m<0?0:(m>=qv?qv-1:m));
cannam@86 352 p[o]=v*del+minval;
cannam@86 353 }
cannam@86 354 }
cannam@86 355
cannam@86 356 if(book->c->lengthlist[index]<=0){
cannam@86 357 const static_codebook *c=book->c;
cannam@86 358 int best=-1;
cannam@86 359 /* assumes integer/centered encoder codebook maptype 1 no more than dim 8 */
cannam@86 360 int e[8]={0,0,0,0,0,0,0,0};
cannam@86 361 int maxval = book->minval + book->delta*(book->quantvals-1);
cannam@86 362 for(i=0;i<book->entries;i++){
cannam@86 363 if(c->lengthlist[i]>0){
cannam@86 364 int this=0;
cannam@86 365 for(j=0;j<dim;j++){
cannam@86 366 int val=(e[j]-a[j]);
cannam@86 367 this+=val*val;
cannam@86 368 }
cannam@86 369 if(best==-1 || this<best){
cannam@86 370 memcpy(p,e,sizeof(p));
cannam@86 371 best=this;
cannam@86 372 index=i;
cannam@86 373 }
cannam@86 374 }
cannam@86 375 /* assumes the value patterning created by the tools in vq/ */
cannam@86 376 j=0;
cannam@86 377 while(e[j]>=maxval)
cannam@86 378 e[j++]=0;
cannam@86 379 if(e[j]>=0)
cannam@86 380 e[j]+=book->delta;
cannam@86 381 e[j]= -e[j];
cannam@86 382 }
cannam@86 383 }
cannam@86 384
cannam@86 385 if(index>-1){
cannam@86 386 for(i=0;i<dim;i++)
cannam@86 387 *a++ -= p[i];
cannam@86 388 }
cannam@86 389
cannam@86 390 return(index);
cannam@86 391 }
cannam@86 392
cannam@86 393 static int _encodepart(oggpack_buffer *opb,int *vec, int n,
cannam@86 394 codebook *book,long *acc){
cannam@86 395 int i,bits=0;
cannam@86 396 int dim=book->dim;
cannam@86 397 int step=n/dim;
cannam@86 398
cannam@86 399 for(i=0;i<step;i++){
cannam@86 400 int entry=local_book_besterror(book,vec+i*dim);
cannam@86 401
cannam@86 402 #ifdef TRAIN_RES
cannam@86 403 if(entry>=0)
cannam@86 404 acc[entry]++;
cannam@86 405 #endif
cannam@86 406
cannam@86 407 bits+=vorbis_book_encode(book,entry,opb);
cannam@86 408
cannam@86 409 }
cannam@86 410
cannam@86 411 return(bits);
cannam@86 412 }
cannam@86 413
cannam@86 414 static long **_01class(vorbis_block *vb,vorbis_look_residue *vl,
cannam@86 415 int **in,int ch){
cannam@86 416 long i,j,k;
cannam@86 417 vorbis_look_residue0 *look=(vorbis_look_residue0 *)vl;
cannam@86 418 vorbis_info_residue0 *info=look->info;
cannam@86 419
cannam@86 420 /* move all this setup out later */
cannam@86 421 int samples_per_partition=info->grouping;
cannam@86 422 int possible_partitions=info->partitions;
cannam@86 423 int n=info->end-info->begin;
cannam@86 424
cannam@86 425 int partvals=n/samples_per_partition;
cannam@86 426 long **partword=_vorbis_block_alloc(vb,ch*sizeof(*partword));
cannam@86 427 float scale=100./samples_per_partition;
cannam@86 428
cannam@86 429 /* we find the partition type for each partition of each
cannam@86 430 channel. We'll go back and do the interleaved encoding in a
cannam@86 431 bit. For now, clarity */
cannam@86 432
cannam@86 433 for(i=0;i<ch;i++){
cannam@86 434 partword[i]=_vorbis_block_alloc(vb,n/samples_per_partition*sizeof(*partword[i]));
cannam@86 435 memset(partword[i],0,n/samples_per_partition*sizeof(*partword[i]));
cannam@86 436 }
cannam@86 437
cannam@86 438 for(i=0;i<partvals;i++){
cannam@86 439 int offset=i*samples_per_partition+info->begin;
cannam@86 440 for(j=0;j<ch;j++){
cannam@86 441 int max=0;
cannam@86 442 int ent=0;
cannam@86 443 for(k=0;k<samples_per_partition;k++){
cannam@86 444 if(abs(in[j][offset+k])>max)max=abs(in[j][offset+k]);
cannam@86 445 ent+=abs(in[j][offset+k]);
cannam@86 446 }
cannam@86 447 ent*=scale;
cannam@86 448
cannam@86 449 for(k=0;k<possible_partitions-1;k++)
cannam@86 450 if(max<=info->classmetric1[k] &&
cannam@86 451 (info->classmetric2[k]<0 || ent<info->classmetric2[k]))
cannam@86 452 break;
cannam@86 453
cannam@86 454 partword[j][i]=k;
cannam@86 455 }
cannam@86 456 }
cannam@86 457
cannam@86 458 #ifdef TRAIN_RESAUX
cannam@86 459 {
cannam@86 460 FILE *of;
cannam@86 461 char buffer[80];
cannam@86 462
cannam@86 463 for(i=0;i<ch;i++){
cannam@86 464 sprintf(buffer,"resaux_%d.vqd",look->train_seq);
cannam@86 465 of=fopen(buffer,"a");
cannam@86 466 for(j=0;j<partvals;j++)
cannam@86 467 fprintf(of,"%ld, ",partword[i][j]);
cannam@86 468 fprintf(of,"\n");
cannam@86 469 fclose(of);
cannam@86 470 }
cannam@86 471 }
cannam@86 472 #endif
cannam@86 473 look->frames++;
cannam@86 474
cannam@86 475 return(partword);
cannam@86 476 }
cannam@86 477
cannam@86 478 /* designed for stereo or other modes where the partition size is an
cannam@86 479 integer multiple of the number of channels encoded in the current
cannam@86 480 submap */
cannam@86 481 static long **_2class(vorbis_block *vb,vorbis_look_residue *vl,int **in,
cannam@86 482 int ch){
cannam@86 483 long i,j,k,l;
cannam@86 484 vorbis_look_residue0 *look=(vorbis_look_residue0 *)vl;
cannam@86 485 vorbis_info_residue0 *info=look->info;
cannam@86 486
cannam@86 487 /* move all this setup out later */
cannam@86 488 int samples_per_partition=info->grouping;
cannam@86 489 int possible_partitions=info->partitions;
cannam@86 490 int n=info->end-info->begin;
cannam@86 491
cannam@86 492 int partvals=n/samples_per_partition;
cannam@86 493 long **partword=_vorbis_block_alloc(vb,sizeof(*partword));
cannam@86 494
cannam@86 495 #if defined(TRAIN_RES) || defined (TRAIN_RESAUX)
cannam@86 496 FILE *of;
cannam@86 497 char buffer[80];
cannam@86 498 #endif
cannam@86 499
cannam@86 500 partword[0]=_vorbis_block_alloc(vb,partvals*sizeof(*partword[0]));
cannam@86 501 memset(partword[0],0,partvals*sizeof(*partword[0]));
cannam@86 502
cannam@86 503 for(i=0,l=info->begin/ch;i<partvals;i++){
cannam@86 504 int magmax=0;
cannam@86 505 int angmax=0;
cannam@86 506 for(j=0;j<samples_per_partition;j+=ch){
cannam@86 507 if(abs(in[0][l])>magmax)magmax=abs(in[0][l]);
cannam@86 508 for(k=1;k<ch;k++)
cannam@86 509 if(abs(in[k][l])>angmax)angmax=abs(in[k][l]);
cannam@86 510 l++;
cannam@86 511 }
cannam@86 512
cannam@86 513 for(j=0;j<possible_partitions-1;j++)
cannam@86 514 if(magmax<=info->classmetric1[j] &&
cannam@86 515 angmax<=info->classmetric2[j])
cannam@86 516 break;
cannam@86 517
cannam@86 518 partword[0][i]=j;
cannam@86 519
cannam@86 520 }
cannam@86 521
cannam@86 522 #ifdef TRAIN_RESAUX
cannam@86 523 sprintf(buffer,"resaux_%d.vqd",look->train_seq);
cannam@86 524 of=fopen(buffer,"a");
cannam@86 525 for(i=0;i<partvals;i++)
cannam@86 526 fprintf(of,"%ld, ",partword[0][i]);
cannam@86 527 fprintf(of,"\n");
cannam@86 528 fclose(of);
cannam@86 529 #endif
cannam@86 530
cannam@86 531 look->frames++;
cannam@86 532
cannam@86 533 return(partword);
cannam@86 534 }
cannam@86 535
cannam@86 536 static int _01forward(oggpack_buffer *opb,
cannam@86 537 vorbis_block *vb,vorbis_look_residue *vl,
cannam@86 538 int **in,int ch,
cannam@86 539 long **partword,
cannam@86 540 int (*encode)(oggpack_buffer *,int *,int,
cannam@86 541 codebook *,long *),
cannam@86 542 int submap){
cannam@86 543 long i,j,k,s;
cannam@86 544 vorbis_look_residue0 *look=(vorbis_look_residue0 *)vl;
cannam@86 545 vorbis_info_residue0 *info=look->info;
cannam@86 546
cannam@86 547 #ifdef TRAIN_RES
cannam@86 548 look->submap=submap;
cannam@86 549 #endif
cannam@86 550
cannam@86 551 /* move all this setup out later */
cannam@86 552 int samples_per_partition=info->grouping;
cannam@86 553 int possible_partitions=info->partitions;
cannam@86 554 int partitions_per_word=look->phrasebook->dim;
cannam@86 555 int n=info->end-info->begin;
cannam@86 556
cannam@86 557 int partvals=n/samples_per_partition;
cannam@86 558 long resbits[128];
cannam@86 559 long resvals[128];
cannam@86 560
cannam@86 561 #ifdef TRAIN_RES
cannam@86 562 for(i=0;i<ch;i++)
cannam@86 563 for(j=info->begin;j<info->end;j++){
cannam@86 564 if(in[i][j]>look->tmax)look->tmax=in[i][j];
cannam@86 565 if(in[i][j]<look->tmin)look->tmin=in[i][j];
cannam@86 566 }
cannam@86 567 #endif
cannam@86 568
cannam@86 569 memset(resbits,0,sizeof(resbits));
cannam@86 570 memset(resvals,0,sizeof(resvals));
cannam@86 571
cannam@86 572 /* we code the partition words for each channel, then the residual
cannam@86 573 words for a partition per channel until we've written all the
cannam@86 574 residual words for that partition word. Then write the next
cannam@86 575 partition channel words... */
cannam@86 576
cannam@86 577 for(s=0;s<look->stages;s++){
cannam@86 578
cannam@86 579 for(i=0;i<partvals;){
cannam@86 580
cannam@86 581 /* first we encode a partition codeword for each channel */
cannam@86 582 if(s==0){
cannam@86 583 for(j=0;j<ch;j++){
cannam@86 584 long val=partword[j][i];
cannam@86 585 for(k=1;k<partitions_per_word;k++){
cannam@86 586 val*=possible_partitions;
cannam@86 587 if(i+k<partvals)
cannam@86 588 val+=partword[j][i+k];
cannam@86 589 }
cannam@86 590
cannam@86 591 /* training hack */
cannam@86 592 if(val<look->phrasebook->entries)
cannam@86 593 look->phrasebits+=vorbis_book_encode(look->phrasebook,val,opb);
cannam@86 594 #if 0 /*def TRAIN_RES*/
cannam@86 595 else
cannam@86 596 fprintf(stderr,"!");
cannam@86 597 #endif
cannam@86 598
cannam@86 599 }
cannam@86 600 }
cannam@86 601
cannam@86 602 /* now we encode interleaved residual values for the partitions */
cannam@86 603 for(k=0;k<partitions_per_word && i<partvals;k++,i++){
cannam@86 604 long offset=i*samples_per_partition+info->begin;
cannam@86 605
cannam@86 606 for(j=0;j<ch;j++){
cannam@86 607 if(s==0)resvals[partword[j][i]]+=samples_per_partition;
cannam@86 608 if(info->secondstages[partword[j][i]]&(1<<s)){
cannam@86 609 codebook *statebook=look->partbooks[partword[j][i]][s];
cannam@86 610 if(statebook){
cannam@86 611 int ret;
cannam@86 612 long *accumulator=NULL;
cannam@86 613
cannam@86 614 #ifdef TRAIN_RES
cannam@86 615 accumulator=look->training_data[s][partword[j][i]];
cannam@86 616 {
cannam@86 617 int l;
cannam@86 618 int *samples=in[j]+offset;
cannam@86 619 for(l=0;l<samples_per_partition;l++){
cannam@86 620 if(samples[l]<look->training_min[s][partword[j][i]])
cannam@86 621 look->training_min[s][partword[j][i]]=samples[l];
cannam@86 622 if(samples[l]>look->training_max[s][partword[j][i]])
cannam@86 623 look->training_max[s][partword[j][i]]=samples[l];
cannam@86 624 }
cannam@86 625 }
cannam@86 626 #endif
cannam@86 627
cannam@86 628 ret=encode(opb,in[j]+offset,samples_per_partition,
cannam@86 629 statebook,accumulator);
cannam@86 630
cannam@86 631 look->postbits+=ret;
cannam@86 632 resbits[partword[j][i]]+=ret;
cannam@86 633 }
cannam@86 634 }
cannam@86 635 }
cannam@86 636 }
cannam@86 637 }
cannam@86 638 }
cannam@86 639
cannam@86 640 /*{
cannam@86 641 long total=0;
cannam@86 642 long totalbits=0;
cannam@86 643 fprintf(stderr,"%d :: ",vb->mode);
cannam@86 644 for(k=0;k<possible_partitions;k++){
cannam@86 645 fprintf(stderr,"%ld/%1.2g, ",resvals[k],(float)resbits[k]/resvals[k]);
cannam@86 646 total+=resvals[k];
cannam@86 647 totalbits+=resbits[k];
cannam@86 648 }
cannam@86 649
cannam@86 650 fprintf(stderr,":: %ld:%1.2g\n",total,(double)totalbits/total);
cannam@86 651 }*/
cannam@86 652
cannam@86 653 return(0);
cannam@86 654 }
cannam@86 655
cannam@86 656 /* a truncated packet here just means 'stop working'; it's not an error */
cannam@86 657 static int _01inverse(vorbis_block *vb,vorbis_look_residue *vl,
cannam@86 658 float **in,int ch,
cannam@86 659 long (*decodepart)(codebook *, float *,
cannam@86 660 oggpack_buffer *,int)){
cannam@86 661
cannam@86 662 long i,j,k,l,s;
cannam@86 663 vorbis_look_residue0 *look=(vorbis_look_residue0 *)vl;
cannam@86 664 vorbis_info_residue0 *info=look->info;
cannam@86 665
cannam@86 666 /* move all this setup out later */
cannam@86 667 int samples_per_partition=info->grouping;
cannam@86 668 int partitions_per_word=look->phrasebook->dim;
cannam@86 669 int max=vb->pcmend>>1;
cannam@86 670 int end=(info->end<max?info->end:max);
cannam@86 671 int n=end-info->begin;
cannam@86 672
cannam@86 673 if(n>0){
cannam@86 674 int partvals=n/samples_per_partition;
cannam@86 675 int partwords=(partvals+partitions_per_word-1)/partitions_per_word;
cannam@86 676 int ***partword=alloca(ch*sizeof(*partword));
cannam@86 677
cannam@86 678 for(j=0;j<ch;j++)
cannam@86 679 partword[j]=_vorbis_block_alloc(vb,partwords*sizeof(*partword[j]));
cannam@86 680
cannam@86 681 for(s=0;s<look->stages;s++){
cannam@86 682
cannam@86 683 /* each loop decodes on partition codeword containing
cannam@86 684 partitions_per_word partitions */
cannam@86 685 for(i=0,l=0;i<partvals;l++){
cannam@86 686 if(s==0){
cannam@86 687 /* fetch the partition word for each channel */
cannam@86 688 for(j=0;j<ch;j++){
cannam@86 689 int temp=vorbis_book_decode(look->phrasebook,&vb->opb);
cannam@86 690
cannam@86 691 if(temp==-1 || temp>=info->partvals)goto eopbreak;
cannam@86 692 partword[j][l]=look->decodemap[temp];
cannam@86 693 if(partword[j][l]==NULL)goto errout;
cannam@86 694 }
cannam@86 695 }
cannam@86 696
cannam@86 697 /* now we decode residual values for the partitions */
cannam@86 698 for(k=0;k<partitions_per_word && i<partvals;k++,i++)
cannam@86 699 for(j=0;j<ch;j++){
cannam@86 700 long offset=info->begin+i*samples_per_partition;
cannam@86 701 if(info->secondstages[partword[j][l][k]]&(1<<s)){
cannam@86 702 codebook *stagebook=look->partbooks[partword[j][l][k]][s];
cannam@86 703 if(stagebook){
cannam@86 704 if(decodepart(stagebook,in[j]+offset,&vb->opb,
cannam@86 705 samples_per_partition)==-1)goto eopbreak;
cannam@86 706 }
cannam@86 707 }
cannam@86 708 }
cannam@86 709 }
cannam@86 710 }
cannam@86 711 }
cannam@86 712 errout:
cannam@86 713 eopbreak:
cannam@86 714 return(0);
cannam@86 715 }
cannam@86 716
cannam@86 717 int res0_inverse(vorbis_block *vb,vorbis_look_residue *vl,
cannam@86 718 float **in,int *nonzero,int ch){
cannam@86 719 int i,used=0;
cannam@86 720 for(i=0;i<ch;i++)
cannam@86 721 if(nonzero[i])
cannam@86 722 in[used++]=in[i];
cannam@86 723 if(used)
cannam@86 724 return(_01inverse(vb,vl,in,used,vorbis_book_decodevs_add));
cannam@86 725 else
cannam@86 726 return(0);
cannam@86 727 }
cannam@86 728
cannam@86 729 int res1_forward(oggpack_buffer *opb,vorbis_block *vb,vorbis_look_residue *vl,
cannam@86 730 int **in,int *nonzero,int ch, long **partword, int submap){
cannam@86 731 int i,used=0;
cannam@86 732 for(i=0;i<ch;i++)
cannam@86 733 if(nonzero[i])
cannam@86 734 in[used++]=in[i];
cannam@86 735
cannam@86 736 if(used){
cannam@86 737 return _01forward(opb,vb,vl,in,used,partword,_encodepart,submap);
cannam@86 738 }else{
cannam@86 739 return(0);
cannam@86 740 }
cannam@86 741 }
cannam@86 742
cannam@86 743 long **res1_class(vorbis_block *vb,vorbis_look_residue *vl,
cannam@86 744 int **in,int *nonzero,int ch){
cannam@86 745 int i,used=0;
cannam@86 746 for(i=0;i<ch;i++)
cannam@86 747 if(nonzero[i])
cannam@86 748 in[used++]=in[i];
cannam@86 749 if(used)
cannam@86 750 return(_01class(vb,vl,in,used));
cannam@86 751 else
cannam@86 752 return(0);
cannam@86 753 }
cannam@86 754
cannam@86 755 int res1_inverse(vorbis_block *vb,vorbis_look_residue *vl,
cannam@86 756 float **in,int *nonzero,int ch){
cannam@86 757 int i,used=0;
cannam@86 758 for(i=0;i<ch;i++)
cannam@86 759 if(nonzero[i])
cannam@86 760 in[used++]=in[i];
cannam@86 761 if(used)
cannam@86 762 return(_01inverse(vb,vl,in,used,vorbis_book_decodev_add));
cannam@86 763 else
cannam@86 764 return(0);
cannam@86 765 }
cannam@86 766
cannam@86 767 long **res2_class(vorbis_block *vb,vorbis_look_residue *vl,
cannam@86 768 int **in,int *nonzero,int ch){
cannam@86 769 int i,used=0;
cannam@86 770 for(i=0;i<ch;i++)
cannam@86 771 if(nonzero[i])used++;
cannam@86 772 if(used)
cannam@86 773 return(_2class(vb,vl,in,ch));
cannam@86 774 else
cannam@86 775 return(0);
cannam@86 776 }
cannam@86 777
cannam@86 778 /* res2 is slightly more different; all the channels are interleaved
cannam@86 779 into a single vector and encoded. */
cannam@86 780
cannam@86 781 int res2_forward(oggpack_buffer *opb,
cannam@86 782 vorbis_block *vb,vorbis_look_residue *vl,
cannam@86 783 int **in,int *nonzero,int ch, long **partword,int submap){
cannam@86 784 long i,j,k,n=vb->pcmend/2,used=0;
cannam@86 785
cannam@86 786 /* don't duplicate the code; use a working vector hack for now and
cannam@86 787 reshape ourselves into a single channel res1 */
cannam@86 788 /* ugly; reallocs for each coupling pass :-( */
cannam@86 789 int *work=_vorbis_block_alloc(vb,ch*n*sizeof(*work));
cannam@86 790 for(i=0;i<ch;i++){
cannam@86 791 int *pcm=in[i];
cannam@86 792 if(nonzero[i])used++;
cannam@86 793 for(j=0,k=i;j<n;j++,k+=ch)
cannam@86 794 work[k]=pcm[j];
cannam@86 795 }
cannam@86 796
cannam@86 797 if(used){
cannam@86 798 return _01forward(opb,vb,vl,&work,1,partword,_encodepart,submap);
cannam@86 799 }else{
cannam@86 800 return(0);
cannam@86 801 }
cannam@86 802 }
cannam@86 803
cannam@86 804 /* duplicate code here as speed is somewhat more important */
cannam@86 805 int res2_inverse(vorbis_block *vb,vorbis_look_residue *vl,
cannam@86 806 float **in,int *nonzero,int ch){
cannam@86 807 long i,k,l,s;
cannam@86 808 vorbis_look_residue0 *look=(vorbis_look_residue0 *)vl;
cannam@86 809 vorbis_info_residue0 *info=look->info;
cannam@86 810
cannam@86 811 /* move all this setup out later */
cannam@86 812 int samples_per_partition=info->grouping;
cannam@86 813 int partitions_per_word=look->phrasebook->dim;
cannam@86 814 int max=(vb->pcmend*ch)>>1;
cannam@86 815 int end=(info->end<max?info->end:max);
cannam@86 816 int n=end-info->begin;
cannam@86 817
cannam@86 818 if(n>0){
cannam@86 819 int partvals=n/samples_per_partition;
cannam@86 820 int partwords=(partvals+partitions_per_word-1)/partitions_per_word;
cannam@86 821 int **partword=_vorbis_block_alloc(vb,partwords*sizeof(*partword));
cannam@86 822
cannam@86 823 for(i=0;i<ch;i++)if(nonzero[i])break;
cannam@86 824 if(i==ch)return(0); /* no nonzero vectors */
cannam@86 825
cannam@86 826 for(s=0;s<look->stages;s++){
cannam@86 827 for(i=0,l=0;i<partvals;l++){
cannam@86 828
cannam@86 829 if(s==0){
cannam@86 830 /* fetch the partition word */
cannam@86 831 int temp=vorbis_book_decode(look->phrasebook,&vb->opb);
cannam@86 832 if(temp==-1 || temp>=info->partvals)goto eopbreak;
cannam@86 833 partword[l]=look->decodemap[temp];
cannam@86 834 if(partword[l]==NULL)goto errout;
cannam@86 835 }
cannam@86 836
cannam@86 837 /* now we decode residual values for the partitions */
cannam@86 838 for(k=0;k<partitions_per_word && i<partvals;k++,i++)
cannam@86 839 if(info->secondstages[partword[l][k]]&(1<<s)){
cannam@86 840 codebook *stagebook=look->partbooks[partword[l][k]][s];
cannam@86 841
cannam@86 842 if(stagebook){
cannam@86 843 if(vorbis_book_decodevv_add(stagebook,in,
cannam@86 844 i*samples_per_partition+info->begin,ch,
cannam@86 845 &vb->opb,samples_per_partition)==-1)
cannam@86 846 goto eopbreak;
cannam@86 847 }
cannam@86 848 }
cannam@86 849 }
cannam@86 850 }
cannam@86 851 }
cannam@86 852 errout:
cannam@86 853 eopbreak:
cannam@86 854 return(0);
cannam@86 855 }
cannam@86 856
cannam@86 857
cannam@86 858 const vorbis_func_residue residue0_exportbundle={
cannam@86 859 NULL,
cannam@86 860 &res0_unpack,
cannam@86 861 &res0_look,
cannam@86 862 &res0_free_info,
cannam@86 863 &res0_free_look,
cannam@86 864 NULL,
cannam@86 865 NULL,
cannam@86 866 &res0_inverse
cannam@86 867 };
cannam@86 868
cannam@86 869 const vorbis_func_residue residue1_exportbundle={
cannam@86 870 &res0_pack,
cannam@86 871 &res0_unpack,
cannam@86 872 &res0_look,
cannam@86 873 &res0_free_info,
cannam@86 874 &res0_free_look,
cannam@86 875 &res1_class,
cannam@86 876 &res1_forward,
cannam@86 877 &res1_inverse
cannam@86 878 };
cannam@86 879
cannam@86 880 const vorbis_func_residue residue2_exportbundle={
cannam@86 881 &res0_pack,
cannam@86 882 &res0_unpack,
cannam@86 883 &res0_look,
cannam@86 884 &res0_free_info,
cannam@86 885 &res0_free_look,
cannam@86 886 &res2_class,
cannam@86 887 &res2_forward,
cannam@86 888 &res2_inverse
cannam@86 889 };