Fix signed overflow in _book_maptype1_quantvals().
[platform/upstream/libvorbis.git] / lib / floor0.c
index 3cf4158..d469be8 100644 (file)
@@ -1,18 +1,17 @@
 /********************************************************************
  *                                                                  *
  * THIS FILE IS PART OF THE OggVorbis SOFTWARE CODEC SOURCE CODE.   *
- * USE, DISTRIBUTION AND REPRODUCTION OF THIS SOURCE IS GOVERNED BY *
- * THE GNU LESSER/LIBRARY PUBLIC LICENSE, WHICH IS INCLUDED WITH    *
- * THIS SOURCE. PLEASE READ THESE TERMS BEFORE DISTRIBUTING.        *
+ * USE, DISTRIBUTION AND REPRODUCTION OF THIS LIBRARY SOURCE IS     *
+ * GOVERNED BY A BSD-STYLE SOURCE LICENSE INCLUDED WITH THIS SOURCE *
+ * IN 'COPYING'. PLEASE READ THESE TERMS BEFORE DISTRIBUTING.       *
  *                                                                  *
- * THE OggVorbis SOURCE CODE IS (C) COPYRIGHT 1994-2000             *
- * by Monty <monty@xiph.org> and the XIPHOPHORUS Company            *
- * http://www.xiph.org/                                             *
+ * THE OggVorbis SOURCE CODE IS (C) COPYRIGHT 1994-2015             *
+ * by the Xiph.Org Foundation http://www.xiph.org/                  *
  *                                                                  *
  ********************************************************************
 
  function: floor backend 0 implementation
- last mod: $Id: floor0.c,v 1.27 2000/11/07 09:51:42 xiphmont Exp $
+ last mod: $Id$
 
  ********************************************************************/
 
 #include <stdio.h>
 
 typedef struct {
-  long n;
   int ln;
   int  m;
-  int *linearmap;
+  int **linearmap;
+  int  n[2];
 
   vorbis_info_floor0 *vi;
-  lpc_lookup lpclook;
-  float *lsp_look;
 
+  long bits;
+  long frames;
 } vorbis_look_floor0;
 
-/* infrastructure for finding fit */
-static long _f0_fit(codebook *book,
-                   float *orig,
-                   float *workfit,
-                   int cursor){
-  int dim=book->dim;
-  float norm,base=0.;
-  int i,best=0;
-  float *lsp=workfit+cursor;
-
-  if(cursor)base=workfit[cursor-1];
-  norm=orig[cursor+dim-1]-base;
-
-  for(i=0;i<dim;i++)
-    lsp[i]=(orig[i+cursor]-base);
-  best=_best(book,lsp,1);
-
-  memcpy(lsp,book->valuelist+best*dim,dim*sizeof(float));
-  for(i=0;i<dim;i++)
-    lsp[i]+=base;
-  return(best);
-}
 
 /***********************************************/
 
-static vorbis_info_floor *floor0_copy_info (vorbis_info_floor *i){
-  vorbis_info_floor0 *info=(vorbis_info_floor0 *)i;
-  vorbis_info_floor0 *ret=_ogg_malloc(sizeof(vorbis_info_floor0));
-  memcpy(ret,info,sizeof(vorbis_info_floor0));
-  return(ret);
-}
-
 static void floor0_free_info(vorbis_info_floor *i){
-  if(i){
-    memset(i,0,sizeof(vorbis_info_floor0));
-    free(i);
+  vorbis_info_floor0 *info=(vorbis_info_floor0 *)i;
+  if(info){
+    memset(info,0,sizeof(*info));
+    _ogg_free(info);
   }
 }
 
 static void floor0_free_look(vorbis_look_floor *i){
   vorbis_look_floor0 *look=(vorbis_look_floor0 *)i;
-  if(i){
-    if(look->linearmap)free(look->linearmap);
-    if(look->lsp_look)free(look->lsp_look);
-    lpc_clear(&look->lpclook);
-    memset(look,0,sizeof(vorbis_look_floor0));
-    free(look);
-  }
-}
+  if(look){
 
-static void floor0_pack (vorbis_info_floor *i,oggpack_buffer *opb){
-  vorbis_info_floor0 *info=(vorbis_info_floor0 *)i;
-  int j;
-  oggpack_write(opb,info->order,8);
-  oggpack_write(opb,info->rate,16);
-  oggpack_write(opb,info->barkmap,16);
-  oggpack_write(opb,info->ampbits,6);
-  oggpack_write(opb,info->ampdB,8);
-  oggpack_write(opb,info->numbooks-1,4);
-  for(j=0;j<info->numbooks;j++)
-    oggpack_write(opb,info->books[j],8);
+    if(look->linearmap){
+
+      if(look->linearmap[0])_ogg_free(look->linearmap[0]);
+      if(look->linearmap[1])_ogg_free(look->linearmap[1]);
+
+      _ogg_free(look->linearmap);
+    }
+    memset(look,0,sizeof(*look));
+    _ogg_free(look);
+  }
 }
 
 static vorbis_info_floor *floor0_unpack (vorbis_info *vi,oggpack_buffer *opb){
   codec_setup_info     *ci=vi->codec_setup;
   int j;
 
-  vorbis_info_floor0 *info=_ogg_malloc(sizeof(vorbis_info_floor0));
+  vorbis_info_floor0 *info=_ogg_malloc(sizeof(*info));
   info->order=oggpack_read(opb,8);
   info->rate=oggpack_read(opb,16);
   info->barkmap=oggpack_read(opb,16);
   info->ampbits=oggpack_read(opb,6);
   info->ampdB=oggpack_read(opb,8);
   info->numbooks=oggpack_read(opb,4)+1;
-  
+
   if(info->order<1)goto err_out;
   if(info->rate<1)goto err_out;
   if(info->barkmap<1)goto err_out;
   if(info->numbooks<1)goto err_out;
-    
+
   for(j=0;j<info->numbooks;j++){
     info->books[j]=oggpack_read(opb,8);
     if(info->books[j]<0 || info->books[j]>=ci->books)goto err_out;
+    if(ci->book_param[info->books[j]]->maptype==0)goto err_out;
+    if(ci->book_param[info->books[j]]->dim<1)goto err_out;
   }
   return(info);
 
@@ -144,272 +109,116 @@ static vorbis_info_floor *floor0_unpack (vorbis_info *vi,oggpack_buffer *opb){
    Note that the scale depends on the sampling rate as well as the
    linear block and mapping sizes */
 
-static vorbis_look_floor *floor0_look (vorbis_dsp_state *vd,vorbis_info_mode *mi,
-                              vorbis_info_floor *i){
-  int j;
-  float scale;
-  vorbis_info        *vi=vd->vi;
-  codec_setup_info   *ci=vi->codec_setup;
+static void floor0_map_lazy_init(vorbis_block      *vb,
+                                 vorbis_info_floor *infoX,
+                                 vorbis_look_floor0 *look){
+  if(!look->linearmap[vb->W]){
+    vorbis_dsp_state   *vd=vb->vd;
+    vorbis_info        *vi=vd->vi;
+    codec_setup_info   *ci=vi->codec_setup;
+    vorbis_info_floor0 *info=(vorbis_info_floor0 *)infoX;
+    int W=vb->W;
+    int n=ci->blocksizes[W]/2,j;
+
+    /* we choose a scaling constant so that:
+       floor(bark(rate/2-1)*C)=mapped-1
+     floor(bark(rate/2)*C)=mapped */
+    float scale=look->ln/toBARK(info->rate/2.f);
+
+    /* the mapping from a linear scale to a smaller bark scale is
+       straightforward.  We do *not* make sure that the linear mapping
+       does not skip bark-scale bins; the decoder simply skips them and
+       the encoder may do what it wishes in filling them.  They're
+       necessary in some mapping combinations to keep the scale spacing
+       accurate */
+    look->linearmap[W]=_ogg_malloc((n+1)*sizeof(**look->linearmap));
+    for(j=0;j<n;j++){
+      int val=floor( toBARK((info->rate/2.f)/n*j)
+                     *scale); /* bark numbers represent band edges */
+      if(val>=look->ln)val=look->ln-1; /* guard against the approximation */
+      look->linearmap[W][j]=val;
+    }
+    look->linearmap[W][j]=-1;
+    look->n[W]=n;
+  }
+}
+
+static vorbis_look_floor *floor0_look(vorbis_dsp_state *vd,
+                                      vorbis_info_floor *i){
   vorbis_info_floor0 *info=(vorbis_info_floor0 *)i;
-  vorbis_look_floor0 *look=_ogg_calloc(1,sizeof(vorbis_look_floor0));
+  vorbis_look_floor0 *look=_ogg_calloc(1,sizeof(*look));
+
+  (void)vd;
+
   look->m=info->order;
-  look->n=ci->blocksizes[mi->blockflag]/2;
   look->ln=info->barkmap;
   look->vi=info;
 
-  if(vd->analysisp)
-    lpc_init(&look->lpclook,look->ln,look->m);
-
-  /* we choose a scaling constant so that:
-     floor(bark(rate/2-1)*C)=mapped-1
-     floor(bark(rate/2)*C)=mapped */
-  scale=look->ln/toBARK(info->rate/2.);
-
-  /* the mapping from a linear scale to a smaller bark scale is
-     straightforward.  We do *not* make sure that the linear mapping
-     does not skip bark-scale bins; the decoder simply skips them and
-     the encoder may do what it wishes in filling them.  They're
-     necessary in some mapping combinations to keep the scale spacing
-     accurate */
-  look->linearmap=_ogg_malloc((look->n+1)*sizeof(int));
-  for(j=0;j<look->n;j++){
-    int val=floor( toBARK((info->rate/2.)/look->n*j) 
-                  *scale); /* bark numbers represent band edges */
-    if(val>look->ln)val=look->ln; /* guard against the approximation */
-    look->linearmap[j]=val;
-  }
-  look->linearmap[j]=-1;
-
-  look->lsp_look=_ogg_malloc(look->ln*sizeof(float));
-  for(j=0;j<look->ln;j++)
-    look->lsp_look[j]=2*cos(M_PI/look->ln*j);
+  look->linearmap=_ogg_calloc(2,sizeof(*look->linearmap));
 
   return look;
 }
 
-/* less efficient than the decode side (written for clarity).  We're
-   not bottlenecked here anyway */
-
-float _curve_to_lpc(float *curve,float *lpc,
-                    vorbis_look_floor0 *l){
-  /* map the input curve to a bark-scale curve for encoding */
-  
-  int mapped=l->ln;
-  float *work=alloca(sizeof(float)*mapped);
-  int i,j,last=0;
-  int bark=0;
-
-  memset(work,0,sizeof(float)*mapped);
-  
-  /* Only the decode side is behavior-specced; for now in the encoder,
-     we select the maximum value of each band as representative (this
-     helps make sure peaks don't go out of range.  In error terms,
-     selecting min would make more sense, but the codebook is trained
-     numerically, so we don't actually lose.  We'd still want to
-     use the original curve for error and noise estimation */
-  
-  for(i=0;i<l->n;i++){
-    bark=l->linearmap[i];
-    if(work[bark]<curve[i])work[bark]=curve[i];
-    if(bark>last+1){
-      /* If the bark scale is climbing rapidly, some bins may end up
-         going unused.  This isn't a waste actually; it keeps the
-         scale resolution even so that the LPC generator has an easy
-         time.  However, if we leave the bins empty we lose energy.
-         So, fill 'em in.  The decoder does not do anything with  he
-         unused bins, so we can fill them anyway we like to end up
-         with a better spectral curve */
-
-      /* we'll always have a bin zero, so we don't need to guard init */
-      long span=bark-last;
-      for(j=1;j<span;j++){
-       float del=(float)j/span;
-       work[j+last]=work[bark]*del+work[last]*(1.-del);
-      }
-    }
-    last=bark;
-  }
-
-  /* If we're over-ranged to avoid edge effects, fill in the end of spectrum gap */
-  for(i=bark+1;i<mapped;i++)
-    work[i]=work[i-1];
-  
-  return vorbis_lpc_from_curve(work,lpc,&(l->lpclook));
-}
-
-/* generate the whole freq response curve of an LSP IIR filter */
-static int floor0_forward(vorbis_block *vb,vorbis_look_floor *i,
-                   float *in,float *out){
-  long j;
+static void *floor0_inverse1(vorbis_block *vb,vorbis_look_floor *i){
   vorbis_look_floor0 *look=(vorbis_look_floor0 *)i;
   vorbis_info_floor0 *info=look->vi;
-  float *work=alloca((look->ln+look->n)*sizeof(float));
-  float amp;
-  long bits=0;
-  static int seq=0;
-
-#ifdef TRAIN_LSP
-  FILE *of;
-  FILE *ef;
-  char buffer[80];
-
-#if 1
-  sprintf(buffer,"lsp0coeff_%d.vqd",vb->mode);
-  of=fopen(buffer,"a");
-#endif
-
-  sprintf(buffer,"lsp0ent_%d.vqd",vb->mode);
-  ef=fopen(buffer,"a");
-#endif
-
-  /* our floor comes in on a linear scale; go to a [-Inf...0] dB
-     scale.  The curve has to be positive, so we offset it. */
-
-  for(j=0;j<look->n;j++)
-    work[j]=todB(in[j])+info->ampdB;
-
-  /* use 'out' as temp storage */
-  /* Convert our floor to a set of lpc coefficients */ 
-  amp=sqrt(_curve_to_lpc(work,out,look));
-
-  /* amp is in the range (0. to ampdB].  Encode that range using
-     ampbits bits */
-  {
-    long maxval=(1L<<info->ampbits)-1;
-    
-    long val=rint(amp/info->ampdB*maxval);
-
-    if(val<0)val=0;           /* likely */
-    if(val>maxval)val=maxval; /* not bloody likely */
-
-    oggpack_write(&vb->opb,val,info->ampbits);
-    if(val>0)
-      amp=(float)val/maxval*info->ampdB;
-    else
-      amp=0;
-  }
-
-  if(amp>0){
-
-    /* the spec supports using one of a number of codebooks.  Right
-       now, encode using this lib supports only one */
-    backend_lookup_state *be=vb->vd->backend_state;
-    codebook *b=be->fullbooks+info->books[0];
-    oggpack_write(&vb->opb,0,_ilog(info->numbooks));
-
-    /* LSP <-> LPC is orthogonal and LSP quantizes more stably  */
-    vorbis_lpc_to_lsp(out,out,look->m);
-
-#ifdef ANALYSIS
-    {
-      float *lspwork=alloca(look->m*sizeof(float));
-      memcpy(lspwork,out,look->m*sizeof(float));
-      vorbis_lsp_to_curve(lspwork,look->linearmap,look->n,look->ln,
-                         work,look->m,amp,info->ampdB);
-      _analysis_output("prefit",seq,work,look->n,0,1);
+  int j,k;
 
-    }
+  int ampraw=oggpack_read(&vb->opb,info->ampbits);
+  if(ampraw>0){ /* also handles the -1 out of data case */
+    long maxval=(1<<info->ampbits)-1;
+    float amp=(float)ampraw/maxval*info->ampdB;
+    int booknum=oggpack_read(&vb->opb,ov_ilog(info->numbooks));
 
-#endif
+    if(booknum!=-1 && booknum<info->numbooks){ /* be paranoid */
+      codec_setup_info  *ci=vb->vd->vi->codec_setup;
+      codebook *b=ci->fullbooks+info->books[booknum];
+      float last=0.f;
 
+      /* the additional b->dim is a guard against any possible stack
+         smash; b->dim is provably more than we can overflow the
+         vector */
+      float *lsp=_vorbis_block_alloc(vb,sizeof(*lsp)*(look->m+b->dim+1));
 
-#if 1
-#ifdef TRAIN_LSP
-    {
-      float last=0.;
-      for(j=0;j<look->m;j++){
-       fprintf(of,"%.12g, ",out[j]-last);
-       last=out[j];
+      if(vorbis_book_decodev_set(b,lsp,&vb->opb,look->m)==-1)goto eop;
+      for(j=0;j<look->m;){
+        for(k=0;j<look->m && k<b->dim;k++,j++)lsp[j]+=last;
+        last=lsp[j-1];
       }
-    }
-    fprintf(of,"\n");
-    fclose(of);
-#endif
-#endif
-
-    /* code the spectral envelope, and keep track of the actual
-       quantized values; we don't want creeping error as each block is
-       nailed to the last quantized value of the previous block. */
-
-    for(j=0;j<look->m;j+=b->dim){
-      int entry=_f0_fit(b,out,work,j);
-      bits+=vorbis_book_encode(b,entry,&vb->opb);
-
-#ifdef TRAIN_LSP
-      fprintf(ef,"%d,\n",entry);
-#endif
 
+      lsp[look->m]=amp;
+      return(lsp);
     }
-
-#ifdef ANALYSIS
-    {
-      float last=0;
-      for(j=0;j<look->m;j++){
-       out[j]=work[j]-last;
-       last=work[j];
-      }
-    }
-       
-#endif
-
-#ifdef TRAIN_LSP
-    fclose(ef);
-#endif
-
-    /* take the coefficients back to a spectral envelope curve */
-    vorbis_lsp_to_curve(out,look->linearmap,look->n,look->ln,
-                       work,look->m,amp,info->ampdB);
-    return(1);
   }
-
-  memset(out,0,sizeof(float)*look->n);
-  seq++;
-  return(0);
+ eop:
+  return(NULL);
 }
 
-static int floor0_inverse(vorbis_block *vb,vorbis_look_floor *i,float *out){
+static int floor0_inverse2(vorbis_block *vb,vorbis_look_floor *i,
+                           void *memo,float *out){
   vorbis_look_floor0 *look=(vorbis_look_floor0 *)i;
   vorbis_info_floor0 *info=look->vi;
-  int j,k;
-  
-  int ampraw=oggpack_read(&vb->opb,info->ampbits);
-  if(ampraw>0){ /* also handles the -1 out of data case */
-    long maxval=(1<<info->ampbits)-1;
-    float amp=(float)ampraw/maxval*info->ampdB;
-    int booknum=oggpack_read(&vb->opb,_ilog(info->numbooks));
-    float *lsp=alloca(sizeof(float)*look->m);
-
-    if(booknum!=-1){
-      backend_lookup_state *be=vb->vd->backend_state;
-      codebook *b=be->fullbooks+info->books[booknum];
-      float last=0.;
-      
-      memset(out,0,sizeof(double)*look->m);    
-      
-      for(j=0;j<look->m;j+=b->dim)
-       if(vorbis_book_decodevs(b,lsp+j,&vb->opb,1,-1)==-1)goto eop;
-      for(j=0;j<look->m;){
-       for(k=0;k<b->dim;k++,j++)lsp[j]+=last;
-       last=lsp[j-1];
-      }
-      
-      /* take the coefficients back to a spectral envelope curve */
-      vorbis_lsp_to_curve(out,look->linearmap,look->n,look->ln,
-                         lsp,look->m,amp,info->ampdB);
-      return(1);
-    }
-  }
 
- eop:
-  memset(out,0,sizeof(float)*look->n);
+  floor0_map_lazy_init(vb,info,look);
+
+  if(memo){
+    float *lsp=(float *)memo;
+    float amp=lsp[look->m];
+
+    /* take the coefficients back to a spectral envelope curve */
+    vorbis_lsp_to_curve(out,
+                        look->linearmap[vb->W],
+                        look->n[vb->W],
+                        look->ln,
+                        lsp,look->m,amp,(float)info->ampdB);
+    return(1);
+  }
+  memset(out,0,sizeof(*out)*look->n[vb->W]);
   return(0);
 }
 
 /* export hooks */
-vorbis_func_floor floor0_exportbundle={
-  &floor0_pack,&floor0_unpack,&floor0_look,&floor0_copy_info,&floor0_free_info,
-  &floor0_free_look,&floor0_forward,&floor0_inverse
+const vorbis_func_floor floor0_exportbundle={
+  NULL,&floor0_unpack,&floor0_look,&floor0_free_info,
+  &floor0_free_look,&floor0_inverse1,&floor0_inverse2
 };
-
-