michael@0: /********************************************************************
michael@0:  *                                                                  *
michael@0:  * THIS FILE IS PART OF THE OggVorbis SOFTWARE CODEC SOURCE CODE.   *
michael@0:  * USE, DISTRIBUTION AND REPRODUCTION OF THIS LIBRARY SOURCE IS     *
michael@0:  * GOVERNED BY A BSD-STYLE SOURCE LICENSE INCLUDED WITH THIS SOURCE *
michael@0:  * IN 'COPYING'. PLEASE READ THESE TERMS BEFORE DISTRIBUTING.       *
michael@0:  *                                                                  *
michael@0:  * THE OggVorbis SOURCE CODE IS (C) COPYRIGHT 1994-2010             *
michael@0:  * by the Xiph.Org Foundation http://www.xiph.org/                  *
michael@0:  *                                                                  *
michael@0:  ********************************************************************
michael@0: 
michael@0:  function: channel mapping 0 implementation
michael@0:  last mod: $Id: mapping0.c 17022 2010-03-25 03:45:42Z xiphmont $
michael@0: 
michael@0:  ********************************************************************/
michael@0: 
michael@0: #include <stdlib.h>
michael@0: #include <stdio.h>
michael@0: #include <string.h>
michael@0: #include <math.h>
michael@0: #include <ogg/ogg.h>
michael@0: #include "vorbis/codec.h"
michael@0: #include "codec_internal.h"
michael@0: #include "codebook.h"
michael@0: #include "window.h"
michael@0: #include "registry.h"
michael@0: #include "psy.h"
michael@0: #include "misc.h"
michael@0: 
michael@0: /* simplistic, wasteful way of doing this (unique lookup for each
michael@0:    mode/submapping); there should be a central repository for
michael@0:    identical lookups.  That will require minor work, so I'm putting it
michael@0:    off as low priority.
michael@0: 
michael@0:    Why a lookup for each backend in a given mode?  Because the
michael@0:    blocksize is set by the mode, and low backend lookups may require
michael@0:    parameters from other areas of the mode/mapping */
michael@0: 
michael@0: static void mapping0_free_info(vorbis_info_mapping *i){
michael@0:   vorbis_info_mapping0 *info=(vorbis_info_mapping0 *)i;
michael@0:   if(info){
michael@0:     memset(info,0,sizeof(*info));
michael@0:     _ogg_free(info);
michael@0:   }
michael@0: }
michael@0: 
michael@0: static int ilog(unsigned int v){
michael@0:   int ret=0;
michael@0:   if(v)--v;
michael@0:   while(v){
michael@0:     ret++;
michael@0:     v>>=1;
michael@0:   }
michael@0:   return(ret);
michael@0: }
michael@0: 
michael@0: static void mapping0_pack(vorbis_info *vi,vorbis_info_mapping *vm,
michael@0:                           oggpack_buffer *opb){
michael@0:   int i;
michael@0:   vorbis_info_mapping0 *info=(vorbis_info_mapping0 *)vm;
michael@0: 
michael@0:   /* another 'we meant to do it this way' hack...  up to beta 4, we
michael@0:      packed 4 binary zeros here to signify one submapping in use.  We
michael@0:      now redefine that to mean four bitflags that indicate use of
michael@0:      deeper features; bit0:submappings, bit1:coupling,
michael@0:      bit2,3:reserved. This is backward compatable with all actual uses
michael@0:      of the beta code. */
michael@0: 
michael@0:   if(info->submaps>1){
michael@0:     oggpack_write(opb,1,1);
michael@0:     oggpack_write(opb,info->submaps-1,4);
michael@0:   }else
michael@0:     oggpack_write(opb,0,1);
michael@0: 
michael@0:   if(info->coupling_steps>0){
michael@0:     oggpack_write(opb,1,1);
michael@0:     oggpack_write(opb,info->coupling_steps-1,8);
michael@0: 
michael@0:     for(i=0;i<info->coupling_steps;i++){
michael@0:       oggpack_write(opb,info->coupling_mag[i],ilog(vi->channels));
michael@0:       oggpack_write(opb,info->coupling_ang[i],ilog(vi->channels));
michael@0:     }
michael@0:   }else
michael@0:     oggpack_write(opb,0,1);
michael@0: 
michael@0:   oggpack_write(opb,0,2); /* 2,3:reserved */
michael@0: 
michael@0:   /* we don't write the channel submappings if we only have one... */
michael@0:   if(info->submaps>1){
michael@0:     for(i=0;i<vi->channels;i++)
michael@0:       oggpack_write(opb,info->chmuxlist[i],4);
michael@0:   }
michael@0:   for(i=0;i<info->submaps;i++){
michael@0:     oggpack_write(opb,0,8); /* time submap unused */
michael@0:     oggpack_write(opb,info->floorsubmap[i],8);
michael@0:     oggpack_write(opb,info->residuesubmap[i],8);
michael@0:   }
michael@0: }
michael@0: 
michael@0: /* also responsible for range checking */
michael@0: static vorbis_info_mapping *mapping0_unpack(vorbis_info *vi,oggpack_buffer *opb){
michael@0:   int i,b;
michael@0:   vorbis_info_mapping0 *info=_ogg_calloc(1,sizeof(*info));
michael@0:   codec_setup_info     *ci=vi->codec_setup;
michael@0:   memset(info,0,sizeof(*info));
michael@0: 
michael@0:   b=oggpack_read(opb,1);
michael@0:   if(b<0)goto err_out;
michael@0:   if(b){
michael@0:     info->submaps=oggpack_read(opb,4)+1;
michael@0:     if(info->submaps<=0)goto err_out;
michael@0:   }else
michael@0:     info->submaps=1;
michael@0: 
michael@0:   b=oggpack_read(opb,1);
michael@0:   if(b<0)goto err_out;
michael@0:   if(b){
michael@0:     info->coupling_steps=oggpack_read(opb,8)+1;
michael@0:     if(info->coupling_steps<=0)goto err_out;
michael@0:     for(i=0;i<info->coupling_steps;i++){
michael@0:       int testM=info->coupling_mag[i]=oggpack_read(opb,ilog(vi->channels));
michael@0:       int testA=info->coupling_ang[i]=oggpack_read(opb,ilog(vi->channels));
michael@0: 
michael@0:       if(testM<0 ||
michael@0:          testA<0 ||
michael@0:          testM==testA ||
michael@0:          testM>=vi->channels ||
michael@0:          testA>=vi->channels) goto err_out;
michael@0:     }
michael@0: 
michael@0:   }
michael@0: 
michael@0:   if(oggpack_read(opb,2)!=0)goto err_out; /* 2,3:reserved */
michael@0: 
michael@0:   if(info->submaps>1){
michael@0:     for(i=0;i<vi->channels;i++){
michael@0:       info->chmuxlist[i]=oggpack_read(opb,4);
michael@0:       if(info->chmuxlist[i]>=info->submaps || info->chmuxlist[i]<0)goto err_out;
michael@0:     }
michael@0:   }
michael@0:   for(i=0;i<info->submaps;i++){
michael@0:     oggpack_read(opb,8); /* time submap unused */
michael@0:     info->floorsubmap[i]=oggpack_read(opb,8);
michael@0:     if(info->floorsubmap[i]>=ci->floors || info->floorsubmap[i]<0)goto err_out;
michael@0:     info->residuesubmap[i]=oggpack_read(opb,8);
michael@0:     if(info->residuesubmap[i]>=ci->residues || info->residuesubmap[i]<0)goto err_out;
michael@0:   }
michael@0: 
michael@0:   return info;
michael@0: 
michael@0:  err_out:
michael@0:   mapping0_free_info(info);
michael@0:   return(NULL);
michael@0: }
michael@0: 
michael@0: #include "os.h"
michael@0: #include "lpc.h"
michael@0: #include "lsp.h"
michael@0: #include "envelope.h"
michael@0: #include "mdct.h"
michael@0: #include "psy.h"
michael@0: #include "scales.h"
michael@0: 
michael@0: #if 0
michael@0: static long seq=0;
michael@0: static ogg_int64_t total=0;
michael@0: static float FLOOR1_fromdB_LOOKUP[256]={
michael@0:   1.0649863e-07F, 1.1341951e-07F, 1.2079015e-07F, 1.2863978e-07F,
michael@0:   1.3699951e-07F, 1.4590251e-07F, 1.5538408e-07F, 1.6548181e-07F,
michael@0:   1.7623575e-07F, 1.8768855e-07F, 1.9988561e-07F, 2.128753e-07F,
michael@0:   2.2670913e-07F, 2.4144197e-07F, 2.5713223e-07F, 2.7384213e-07F,
michael@0:   2.9163793e-07F, 3.1059021e-07F, 3.3077411e-07F, 3.5226968e-07F,
michael@0:   3.7516214e-07F, 3.9954229e-07F, 4.2550680e-07F, 4.5315863e-07F,
michael@0:   4.8260743e-07F, 5.1396998e-07F, 5.4737065e-07F, 5.8294187e-07F,
michael@0:   6.2082472e-07F, 6.6116941e-07F, 7.0413592e-07F, 7.4989464e-07F,
michael@0:   7.9862701e-07F, 8.5052630e-07F, 9.0579828e-07F, 9.6466216e-07F,
michael@0:   1.0273513e-06F, 1.0941144e-06F, 1.1652161e-06F, 1.2409384e-06F,
michael@0:   1.3215816e-06F, 1.4074654e-06F, 1.4989305e-06F, 1.5963394e-06F,
michael@0:   1.7000785e-06F, 1.8105592e-06F, 1.9282195e-06F, 2.0535261e-06F,
michael@0:   2.1869758e-06F, 2.3290978e-06F, 2.4804557e-06F, 2.6416497e-06F,
michael@0:   2.8133190e-06F, 2.9961443e-06F, 3.1908506e-06F, 3.3982101e-06F,
michael@0:   3.6190449e-06F, 3.8542308e-06F, 4.1047004e-06F, 4.3714470e-06F,
michael@0:   4.6555282e-06F, 4.9580707e-06F, 5.2802740e-06F, 5.6234160e-06F,
michael@0:   5.9888572e-06F, 6.3780469e-06F, 6.7925283e-06F, 7.2339451e-06F,
michael@0:   7.7040476e-06F, 8.2047000e-06F, 8.7378876e-06F, 9.3057248e-06F,
michael@0:   9.9104632e-06F, 1.0554501e-05F, 1.1240392e-05F, 1.1970856e-05F,
michael@0:   1.2748789e-05F, 1.3577278e-05F, 1.4459606e-05F, 1.5399272e-05F,
michael@0:   1.6400004e-05F, 1.7465768e-05F, 1.8600792e-05F, 1.9809576e-05F,
michael@0:   2.1096914e-05F, 2.2467911e-05F, 2.3928002e-05F, 2.5482978e-05F,
michael@0:   2.7139006e-05F, 2.8902651e-05F, 3.0780908e-05F, 3.2781225e-05F,
michael@0:   3.4911534e-05F, 3.7180282e-05F, 3.9596466e-05F, 4.2169667e-05F,
michael@0:   4.4910090e-05F, 4.7828601e-05F, 5.0936773e-05F, 5.4246931e-05F,
michael@0:   5.7772202e-05F, 6.1526565e-05F, 6.5524908e-05F, 6.9783085e-05F,
michael@0:   7.4317983e-05F, 7.9147585e-05F, 8.4291040e-05F, 8.9768747e-05F,
michael@0:   9.5602426e-05F, 0.00010181521F, 0.00010843174F, 0.00011547824F,
michael@0:   0.00012298267F, 0.00013097477F, 0.00013948625F, 0.00014855085F,
michael@0:   0.00015820453F, 0.00016848555F, 0.00017943469F, 0.00019109536F,
michael@0:   0.00020351382F, 0.00021673929F, 0.00023082423F, 0.00024582449F,
michael@0:   0.00026179955F, 0.00027881276F, 0.00029693158F, 0.00031622787F,
michael@0:   0.00033677814F, 0.00035866388F, 0.00038197188F, 0.00040679456F,
michael@0:   0.00043323036F, 0.00046138411F, 0.00049136745F, 0.00052329927F,
michael@0:   0.00055730621F, 0.00059352311F, 0.00063209358F, 0.00067317058F,
michael@0:   0.00071691700F, 0.00076350630F, 0.00081312324F, 0.00086596457F,
michael@0:   0.00092223983F, 0.00098217216F, 0.0010459992F, 0.0011139742F,
michael@0:   0.0011863665F, 0.0012634633F, 0.0013455702F, 0.0014330129F,
michael@0:   0.0015261382F, 0.0016253153F, 0.0017309374F, 0.0018434235F,
michael@0:   0.0019632195F, 0.0020908006F, 0.0022266726F, 0.0023713743F,
michael@0:   0.0025254795F, 0.0026895994F, 0.0028643847F, 0.0030505286F,
michael@0:   0.0032487691F, 0.0034598925F, 0.0036847358F, 0.0039241906F,
michael@0:   0.0041792066F, 0.0044507950F, 0.0047400328F, 0.0050480668F,
michael@0:   0.0053761186F, 0.0057254891F, 0.0060975636F, 0.0064938176F,
michael@0:   0.0069158225F, 0.0073652516F, 0.0078438871F, 0.0083536271F,
michael@0:   0.0088964928F, 0.009474637F, 0.010090352F, 0.010746080F,
michael@0:   0.011444421F, 0.012188144F, 0.012980198F, 0.013823725F,
michael@0:   0.014722068F, 0.015678791F, 0.016697687F, 0.017782797F,
michael@0:   0.018938423F, 0.020169149F, 0.021479854F, 0.022875735F,
michael@0:   0.024362330F, 0.025945531F, 0.027631618F, 0.029427276F,
michael@0:   0.031339626F, 0.033376252F, 0.035545228F, 0.037855157F,
michael@0:   0.040315199F, 0.042935108F, 0.045725273F, 0.048696758F,
michael@0:   0.051861348F, 0.055231591F, 0.058820850F, 0.062643361F,
michael@0:   0.066714279F, 0.071049749F, 0.075666962F, 0.080584227F,
michael@0:   0.085821044F, 0.091398179F, 0.097337747F, 0.10366330F,
michael@0:   0.11039993F, 0.11757434F, 0.12521498F, 0.13335215F,
michael@0:   0.14201813F, 0.15124727F, 0.16107617F, 0.17154380F,
michael@0:   0.18269168F, 0.19456402F, 0.20720788F, 0.22067342F,
michael@0:   0.23501402F, 0.25028656F, 0.26655159F, 0.28387361F,
michael@0:   0.30232132F, 0.32196786F, 0.34289114F, 0.36517414F,
michael@0:   0.38890521F, 0.41417847F, 0.44109412F, 0.46975890F,
michael@0:   0.50028648F, 0.53279791F, 0.56742212F, 0.60429640F,
michael@0:   0.64356699F, 0.68538959F, 0.72993007F, 0.77736504F,
michael@0:   0.82788260F, 0.88168307F, 0.9389798F, 1.F,
michael@0: };
michael@0: 
michael@0: #endif
michael@0: 
michael@0: 
michael@0: static int mapping0_forward(vorbis_block *vb){
michael@0:   vorbis_dsp_state      *vd=vb->vd;
michael@0:   vorbis_info           *vi=vd->vi;
michael@0:   codec_setup_info      *ci=vi->codec_setup;
michael@0:   private_state         *b=vb->vd->backend_state;
michael@0:   vorbis_block_internal *vbi=(vorbis_block_internal *)vb->internal;
michael@0:   int                    n=vb->pcmend;
michael@0:   int i,j,k;
michael@0: 
michael@0:   int    *nonzero    = alloca(sizeof(*nonzero)*vi->channels);
michael@0:   float  **gmdct     = _vorbis_block_alloc(vb,vi->channels*sizeof(*gmdct));
michael@0:   int    **iwork      = _vorbis_block_alloc(vb,vi->channels*sizeof(*iwork));
michael@0:   int ***floor_posts = _vorbis_block_alloc(vb,vi->channels*sizeof(*floor_posts));
michael@0: 
michael@0:   float global_ampmax=vbi->ampmax;
michael@0:   float *local_ampmax=alloca(sizeof(*local_ampmax)*vi->channels);
michael@0:   int blocktype=vbi->blocktype;
michael@0: 
michael@0:   int modenumber=vb->W;
michael@0:   vorbis_info_mapping0 *info=ci->map_param[modenumber];
michael@0:   vorbis_look_psy *psy_look=b->psy+blocktype+(vb->W?2:0);
michael@0: 
michael@0:   vb->mode=modenumber;
michael@0: 
michael@0:   for(i=0;i<vi->channels;i++){
michael@0:     float scale=4.f/n;
michael@0:     float scale_dB;
michael@0: 
michael@0:     float *pcm     =vb->pcm[i];
michael@0:     float *logfft  =pcm;
michael@0: 
michael@0:     iwork[i]=_vorbis_block_alloc(vb,n/2*sizeof(**iwork));
michael@0:     gmdct[i]=_vorbis_block_alloc(vb,n/2*sizeof(**gmdct));
michael@0: 
michael@0:     scale_dB=todB(&scale) + .345; /* + .345 is a hack; the original
michael@0:                                      todB estimation used on IEEE 754
michael@0:                                      compliant machines had a bug that
michael@0:                                      returned dB values about a third
michael@0:                                      of a decibel too high.  The bug
michael@0:                                      was harmless because tunings
michael@0:                                      implicitly took that into
michael@0:                                      account.  However, fixing the bug
michael@0:                                      in the estimator requires
michael@0:                                      changing all the tunings as well.
michael@0:                                      For now, it's easier to sync
michael@0:                                      things back up here, and
michael@0:                                      recalibrate the tunings in the
michael@0:                                      next major model upgrade. */
michael@0: 
michael@0: #if 0
michael@0:     if(vi->channels==2){
michael@0:       if(i==0)
michael@0:         _analysis_output("pcmL",seq,pcm,n,0,0,total-n/2);
michael@0:       else
michael@0:         _analysis_output("pcmR",seq,pcm,n,0,0,total-n/2);
michael@0:     }else{
michael@0:       _analysis_output("pcm",seq,pcm,n,0,0,total-n/2);
michael@0:     }
michael@0: #endif
michael@0: 
michael@0:     /* window the PCM data */
michael@0:     _vorbis_apply_window(pcm,b->window,ci->blocksizes,vb->lW,vb->W,vb->nW);
michael@0: 
michael@0: #if 0
michael@0:     if(vi->channels==2){
michael@0:       if(i==0)
michael@0:         _analysis_output("windowedL",seq,pcm,n,0,0,total-n/2);
michael@0:       else
michael@0:         _analysis_output("windowedR",seq,pcm,n,0,0,total-n/2);
michael@0:     }else{
michael@0:       _analysis_output("windowed",seq,pcm,n,0,0,total-n/2);
michael@0:     }
michael@0: #endif
michael@0: 
michael@0:     /* transform the PCM data */
michael@0:     /* only MDCT right now.... */
michael@0:     mdct_forward(b->transform[vb->W][0],pcm,gmdct[i]);
michael@0: 
michael@0:     /* FFT yields more accurate tonal estimation (not phase sensitive) */
michael@0:     drft_forward(&b->fft_look[vb->W],pcm);
michael@0:     logfft[0]=scale_dB+todB(pcm)  + .345; /* + .345 is a hack; the
michael@0:                                      original todB estimation used on
michael@0:                                      IEEE 754 compliant machines had a
michael@0:                                      bug that returned dB values about
michael@0:                                      a third of a decibel too high.
michael@0:                                      The bug was harmless because
michael@0:                                      tunings implicitly took that into
michael@0:                                      account.  However, fixing the bug
michael@0:                                      in the estimator requires
michael@0:                                      changing all the tunings as well.
michael@0:                                      For now, it's easier to sync
michael@0:                                      things back up here, and
michael@0:                                      recalibrate the tunings in the
michael@0:                                      next major model upgrade. */
michael@0:     local_ampmax[i]=logfft[0];
michael@0:     for(j=1;j<n-1;j+=2){
michael@0:       float temp=pcm[j]*pcm[j]+pcm[j+1]*pcm[j+1];
michael@0:       temp=logfft[(j+1)>>1]=scale_dB+.5f*todB(&temp)  + .345; /* +
michael@0:                                      .345 is a hack; the original todB
michael@0:                                      estimation used on IEEE 754
michael@0:                                      compliant machines had a bug that
michael@0:                                      returned dB values about a third
michael@0:                                      of a decibel too high.  The bug
michael@0:                                      was harmless because tunings
michael@0:                                      implicitly took that into
michael@0:                                      account.  However, fixing the bug
michael@0:                                      in the estimator requires
michael@0:                                      changing all the tunings as well.
michael@0:                                      For now, it's easier to sync
michael@0:                                      things back up here, and
michael@0:                                      recalibrate the tunings in the
michael@0:                                      next major model upgrade. */
michael@0:       if(temp>local_ampmax[i])local_ampmax[i]=temp;
michael@0:     }
michael@0: 
michael@0:     if(local_ampmax[i]>0.f)local_ampmax[i]=0.f;
michael@0:     if(local_ampmax[i]>global_ampmax)global_ampmax=local_ampmax[i];
michael@0: 
michael@0: #if 0
michael@0:     if(vi->channels==2){
michael@0:       if(i==0){
michael@0:         _analysis_output("fftL",seq,logfft,n/2,1,0,0);
michael@0:       }else{
michael@0:         _analysis_output("fftR",seq,logfft,n/2,1,0,0);
michael@0:       }
michael@0:     }else{
michael@0:       _analysis_output("fft",seq,logfft,n/2,1,0,0);
michael@0:     }
michael@0: #endif
michael@0: 
michael@0:   }
michael@0: 
michael@0:   {
michael@0:     float   *noise        = _vorbis_block_alloc(vb,n/2*sizeof(*noise));
michael@0:     float   *tone         = _vorbis_block_alloc(vb,n/2*sizeof(*tone));
michael@0: 
michael@0:     for(i=0;i<vi->channels;i++){
michael@0:       /* the encoder setup assumes that all the modes used by any
michael@0:          specific bitrate tweaking use the same floor */
michael@0: 
michael@0:       int submap=info->chmuxlist[i];
michael@0: 
michael@0:       /* the following makes things clearer to *me* anyway */
michael@0:       float *mdct    =gmdct[i];
michael@0:       float *logfft  =vb->pcm[i];
michael@0: 
michael@0:       float *logmdct =logfft+n/2;
michael@0:       float *logmask =logfft;
michael@0: 
michael@0:       vb->mode=modenumber;
michael@0: 
michael@0:       floor_posts[i]=_vorbis_block_alloc(vb,PACKETBLOBS*sizeof(**floor_posts));
michael@0:       memset(floor_posts[i],0,sizeof(**floor_posts)*PACKETBLOBS);
michael@0: 
michael@0:       for(j=0;j<n/2;j++)
michael@0:         logmdct[j]=todB(mdct+j)  + .345; /* + .345 is a hack; the original
michael@0:                                      todB estimation used on IEEE 754
michael@0:                                      compliant machines had a bug that
michael@0:                                      returned dB values about a third
michael@0:                                      of a decibel too high.  The bug
michael@0:                                      was harmless because tunings
michael@0:                                      implicitly took that into
michael@0:                                      account.  However, fixing the bug
michael@0:                                      in the estimator requires
michael@0:                                      changing all the tunings as well.
michael@0:                                      For now, it's easier to sync
michael@0:                                      things back up here, and
michael@0:                                      recalibrate the tunings in the
michael@0:                                      next major model upgrade. */
michael@0: 
michael@0: #if 0
michael@0:       if(vi->channels==2){
michael@0:         if(i==0)
michael@0:           _analysis_output("mdctL",seq,logmdct,n/2,1,0,0);
michael@0:         else
michael@0:           _analysis_output("mdctR",seq,logmdct,n/2,1,0,0);
michael@0:       }else{
michael@0:         _analysis_output("mdct",seq,logmdct,n/2,1,0,0);
michael@0:       }
michael@0: #endif
michael@0: 
michael@0:       /* first step; noise masking.  Not only does 'noise masking'
michael@0:          give us curves from which we can decide how much resolution
michael@0:          to give noise parts of the spectrum, it also implicitly hands
michael@0:          us a tonality estimate (the larger the value in the
michael@0:          'noise_depth' vector, the more tonal that area is) */
michael@0: 
michael@0:       _vp_noisemask(psy_look,
michael@0:                     logmdct,
michael@0:                     noise); /* noise does not have by-frequency offset
michael@0:                                bias applied yet */
michael@0: #if 0
michael@0:       if(vi->channels==2){
michael@0:         if(i==0)
michael@0:           _analysis_output("noiseL",seq,noise,n/2,1,0,0);
michael@0:         else
michael@0:           _analysis_output("noiseR",seq,noise,n/2,1,0,0);
michael@0:       }else{
michael@0:         _analysis_output("noise",seq,noise,n/2,1,0,0);
michael@0:       }
michael@0: #endif
michael@0: 
michael@0:       /* second step: 'all the other crap'; all the stuff that isn't
michael@0:          computed/fit for bitrate management goes in the second psy
michael@0:          vector.  This includes tone masking, peak limiting and ATH */
michael@0: 
michael@0:       _vp_tonemask(psy_look,
michael@0:                    logfft,
michael@0:                    tone,
michael@0:                    global_ampmax,
michael@0:                    local_ampmax[i]);
michael@0: 
michael@0: #if 0
michael@0:       if(vi->channels==2){
michael@0:         if(i==0)
michael@0:           _analysis_output("toneL",seq,tone,n/2,1,0,0);
michael@0:         else
michael@0:           _analysis_output("toneR",seq,tone,n/2,1,0,0);
michael@0:       }else{
michael@0:         _analysis_output("tone",seq,tone,n/2,1,0,0);
michael@0:       }
michael@0: #endif
michael@0: 
michael@0:       /* third step; we offset the noise vectors, overlay tone
michael@0:          masking.  We then do a floor1-specific line fit.  If we're
michael@0:          performing bitrate management, the line fit is performed
michael@0:          multiple times for up/down tweakage on demand. */
michael@0: 
michael@0: #if 0
michael@0:       {
michael@0:       float aotuv[psy_look->n];
michael@0: #endif
michael@0: 
michael@0:         _vp_offset_and_mix(psy_look,
michael@0:                            noise,
michael@0:                            tone,
michael@0:                            1,
michael@0:                            logmask,
michael@0:                            mdct,
michael@0:                            logmdct);
michael@0: 
michael@0: #if 0
michael@0:         if(vi->channels==2){
michael@0:           if(i==0)
michael@0:             _analysis_output("aotuvM1_L",seq,aotuv,psy_look->n,1,1,0);
michael@0:           else
michael@0:             _analysis_output("aotuvM1_R",seq,aotuv,psy_look->n,1,1,0);
michael@0:         }else{
michael@0:           _analysis_output("aotuvM1",seq,aotuv,psy_look->n,1,1,0);
michael@0:         }
michael@0:       }
michael@0: #endif
michael@0: 
michael@0: 
michael@0: #if 0
michael@0:       if(vi->channels==2){
michael@0:         if(i==0)
michael@0:           _analysis_output("mask1L",seq,logmask,n/2,1,0,0);
michael@0:         else
michael@0:           _analysis_output("mask1R",seq,logmask,n/2,1,0,0);
michael@0:       }else{
michael@0:         _analysis_output("mask1",seq,logmask,n/2,1,0,0);
michael@0:       }
michael@0: #endif
michael@0: 
michael@0:       /* this algorithm is hardwired to floor 1 for now; abort out if
michael@0:          we're *not* floor1.  This won't happen unless someone has
michael@0:          broken the encode setup lib.  Guard it anyway. */
michael@0:       if(ci->floor_type[info->floorsubmap[submap]]!=1)return(-1);
michael@0: 
michael@0:       floor_posts[i][PACKETBLOBS/2]=
michael@0:         floor1_fit(vb,b->flr[info->floorsubmap[submap]],
michael@0:                    logmdct,
michael@0:                    logmask);
michael@0: 
michael@0:       /* are we managing bitrate?  If so, perform two more fits for
michael@0:          later rate tweaking (fits represent hi/lo) */
michael@0:       if(vorbis_bitrate_managed(vb) && floor_posts[i][PACKETBLOBS/2]){
michael@0:         /* higher rate by way of lower noise curve */
michael@0: 
michael@0:         _vp_offset_and_mix(psy_look,
michael@0:                            noise,
michael@0:                            tone,
michael@0:                            2,
michael@0:                            logmask,
michael@0:                            mdct,
michael@0:                            logmdct);
michael@0: 
michael@0: #if 0
michael@0:         if(vi->channels==2){
michael@0:           if(i==0)
michael@0:             _analysis_output("mask2L",seq,logmask,n/2,1,0,0);
michael@0:           else
michael@0:             _analysis_output("mask2R",seq,logmask,n/2,1,0,0);
michael@0:         }else{
michael@0:           _analysis_output("mask2",seq,logmask,n/2,1,0,0);
michael@0:         }
michael@0: #endif
michael@0: 
michael@0:         floor_posts[i][PACKETBLOBS-1]=
michael@0:           floor1_fit(vb,b->flr[info->floorsubmap[submap]],
michael@0:                      logmdct,
michael@0:                      logmask);
michael@0: 
michael@0:         /* lower rate by way of higher noise curve */
michael@0:         _vp_offset_and_mix(psy_look,
michael@0:                            noise,
michael@0:                            tone,
michael@0:                            0,
michael@0:                            logmask,
michael@0:                            mdct,
michael@0:                            logmdct);
michael@0: 
michael@0: #if 0
michael@0:         if(vi->channels==2){
michael@0:           if(i==0)
michael@0:             _analysis_output("mask0L",seq,logmask,n/2,1,0,0);
michael@0:           else
michael@0:             _analysis_output("mask0R",seq,logmask,n/2,1,0,0);
michael@0:         }else{
michael@0:           _analysis_output("mask0",seq,logmask,n/2,1,0,0);
michael@0:         }
michael@0: #endif
michael@0: 
michael@0:         floor_posts[i][0]=
michael@0:           floor1_fit(vb,b->flr[info->floorsubmap[submap]],
michael@0:                      logmdct,
michael@0:                      logmask);
michael@0: 
michael@0:         /* we also interpolate a range of intermediate curves for
michael@0:            intermediate rates */
michael@0:         for(k=1;k<PACKETBLOBS/2;k++)
michael@0:           floor_posts[i][k]=
michael@0:             floor1_interpolate_fit(vb,b->flr[info->floorsubmap[submap]],
michael@0:                                    floor_posts[i][0],
michael@0:                                    floor_posts[i][PACKETBLOBS/2],
michael@0:                                    k*65536/(PACKETBLOBS/2));
michael@0:         for(k=PACKETBLOBS/2+1;k<PACKETBLOBS-1;k++)
michael@0:           floor_posts[i][k]=
michael@0:             floor1_interpolate_fit(vb,b->flr[info->floorsubmap[submap]],
michael@0:                                    floor_posts[i][PACKETBLOBS/2],
michael@0:                                    floor_posts[i][PACKETBLOBS-1],
michael@0:                                    (k-PACKETBLOBS/2)*65536/(PACKETBLOBS/2));
michael@0:       }
michael@0:     }
michael@0:   }
michael@0:   vbi->ampmax=global_ampmax;
michael@0: 
michael@0:   /*
michael@0:     the next phases are performed once for vbr-only and PACKETBLOB
michael@0:     times for bitrate managed modes.
michael@0: 
michael@0:     1) encode actual mode being used
michael@0:     2) encode the floor for each channel, compute coded mask curve/res
michael@0:     3) normalize and couple.
michael@0:     4) encode residue
michael@0:     5) save packet bytes to the packetblob vector
michael@0: 
michael@0:   */
michael@0: 
michael@0:   /* iterate over the many masking curve fits we've created */
michael@0: 
michael@0:   {
michael@0:     int **couple_bundle=alloca(sizeof(*couple_bundle)*vi->channels);
michael@0:     int *zerobundle=alloca(sizeof(*zerobundle)*vi->channels);
michael@0: 
michael@0:     for(k=(vorbis_bitrate_managed(vb)?0:PACKETBLOBS/2);
michael@0:         k<=(vorbis_bitrate_managed(vb)?PACKETBLOBS-1:PACKETBLOBS/2);
michael@0:         k++){
michael@0:       oggpack_buffer *opb=vbi->packetblob[k];
michael@0: 
michael@0:       /* start out our new packet blob with packet type and mode */
michael@0:       /* Encode the packet type */
michael@0:       oggpack_write(opb,0,1);
michael@0:       /* Encode the modenumber */
michael@0:       /* Encode frame mode, pre,post windowsize, then dispatch */
michael@0:       oggpack_write(opb,modenumber,b->modebits);
michael@0:       if(vb->W){
michael@0:         oggpack_write(opb,vb->lW,1);
michael@0:         oggpack_write(opb,vb->nW,1);
michael@0:       }
michael@0: 
michael@0:       /* encode floor, compute masking curve, sep out residue */
michael@0:       for(i=0;i<vi->channels;i++){
michael@0:         int submap=info->chmuxlist[i];
michael@0:         int *ilogmask=iwork[i];
michael@0: 
michael@0:         nonzero[i]=floor1_encode(opb,vb,b->flr[info->floorsubmap[submap]],
michael@0:                                  floor_posts[i][k],
michael@0:                                  ilogmask);
michael@0: #if 0
michael@0:         {
michael@0:           char buf[80];
michael@0:           sprintf(buf,"maskI%c%d",i?'R':'L',k);
michael@0:           float work[n/2];
michael@0:           for(j=0;j<n/2;j++)
michael@0:             work[j]=FLOOR1_fromdB_LOOKUP[iwork[i][j]];
michael@0:           _analysis_output(buf,seq,work,n/2,1,1,0);
michael@0:         }
michael@0: #endif
michael@0:       }
michael@0: 
michael@0:       /* our iteration is now based on masking curve, not prequant and
michael@0:          coupling.  Only one prequant/coupling step */
michael@0: 
michael@0:       /* quantize/couple */
michael@0:       /* incomplete implementation that assumes the tree is all depth
michael@0:          one, or no tree at all */
michael@0:       _vp_couple_quantize_normalize(k,
michael@0:                                     &ci->psy_g_param,
michael@0:                                     psy_look,
michael@0:                                     info,
michael@0:                                     gmdct,
michael@0:                                     iwork,
michael@0:                                     nonzero,
michael@0:                                     ci->psy_g_param.sliding_lowpass[vb->W][k],
michael@0:                                     vi->channels);
michael@0: 
michael@0: #if 0
michael@0:       for(i=0;i<vi->channels;i++){
michael@0:         char buf[80];
michael@0:         sprintf(buf,"res%c%d",i?'R':'L',k);
michael@0:         float work[n/2];
michael@0:         for(j=0;j<n/2;j++)
michael@0:           work[j]=iwork[i][j];
michael@0:         _analysis_output(buf,seq,work,n/2,1,0,0);
michael@0:       }
michael@0: #endif
michael@0: 
michael@0:       /* classify and encode by submap */
michael@0:       for(i=0;i<info->submaps;i++){
michael@0:         int ch_in_bundle=0;
michael@0:         long **classifications;
michael@0:         int resnum=info->residuesubmap[i];
michael@0: 
michael@0:         for(j=0;j<vi->channels;j++){
michael@0:           if(info->chmuxlist[j]==i){
michael@0:             zerobundle[ch_in_bundle]=0;
michael@0:             if(nonzero[j])zerobundle[ch_in_bundle]=1;
michael@0:             couple_bundle[ch_in_bundle++]=iwork[j];
michael@0:           }
michael@0:         }
michael@0: 
michael@0:         classifications=_residue_P[ci->residue_type[resnum]]->
michael@0:           class(vb,b->residue[resnum],couple_bundle,zerobundle,ch_in_bundle);
michael@0: 
michael@0:         ch_in_bundle=0;
michael@0:         for(j=0;j<vi->channels;j++)
michael@0:           if(info->chmuxlist[j]==i)
michael@0:             couple_bundle[ch_in_bundle++]=iwork[j];
michael@0: 
michael@0:         _residue_P[ci->residue_type[resnum]]->
michael@0:           forward(opb,vb,b->residue[resnum],
michael@0:                   couple_bundle,zerobundle,ch_in_bundle,classifications,i);
michael@0:       }
michael@0: 
michael@0:       /* ok, done encoding.  Next protopacket. */
michael@0:     }
michael@0: 
michael@0:   }
michael@0: 
michael@0: #if 0
michael@0:   seq++;
michael@0:   total+=ci->blocksizes[vb->W]/4+ci->blocksizes[vb->nW]/4;
michael@0: #endif
michael@0:   return(0);
michael@0: }
michael@0: 
michael@0: static int mapping0_inverse(vorbis_block *vb,vorbis_info_mapping *l){
michael@0:   vorbis_dsp_state     *vd=vb->vd;
michael@0:   vorbis_info          *vi=vd->vi;
michael@0:   codec_setup_info     *ci=vi->codec_setup;
michael@0:   private_state        *b=vd->backend_state;
michael@0:   vorbis_info_mapping0 *info=(vorbis_info_mapping0 *)l;
michael@0: 
michael@0:   int                   i,j;
michael@0:   long                  n=vb->pcmend=ci->blocksizes[vb->W];
michael@0: 
michael@0:   float **pcmbundle=alloca(sizeof(*pcmbundle)*vi->channels);
michael@0:   int    *zerobundle=alloca(sizeof(*zerobundle)*vi->channels);
michael@0: 
michael@0:   int   *nonzero  =alloca(sizeof(*nonzero)*vi->channels);
michael@0:   void **floormemo=alloca(sizeof(*floormemo)*vi->channels);
michael@0: 
michael@0:   /* recover the spectral envelope; store it in the PCM vector for now */
michael@0:   for(i=0;i<vi->channels;i++){
michael@0:     int submap=info->chmuxlist[i];
michael@0:     floormemo[i]=_floor_P[ci->floor_type[info->floorsubmap[submap]]]->
michael@0:       inverse1(vb,b->flr[info->floorsubmap[submap]]);
michael@0:     if(floormemo[i])
michael@0:       nonzero[i]=1;
michael@0:     else
michael@0:       nonzero[i]=0;
michael@0:     memset(vb->pcm[i],0,sizeof(*vb->pcm[i])*n/2);
michael@0:   }
michael@0: 
michael@0:   /* channel coupling can 'dirty' the nonzero listing */
michael@0:   for(i=0;i<info->coupling_steps;i++){
michael@0:     if(nonzero[info->coupling_mag[i]] ||
michael@0:        nonzero[info->coupling_ang[i]]){
michael@0:       nonzero[info->coupling_mag[i]]=1;
michael@0:       nonzero[info->coupling_ang[i]]=1;
michael@0:     }
michael@0:   }
michael@0: 
michael@0:   /* recover the residue into our working vectors */
michael@0:   for(i=0;i<info->submaps;i++){
michael@0:     int ch_in_bundle=0;
michael@0:     for(j=0;j<vi->channels;j++){
michael@0:       if(info->chmuxlist[j]==i){
michael@0:         if(nonzero[j])
michael@0:           zerobundle[ch_in_bundle]=1;
michael@0:         else
michael@0:           zerobundle[ch_in_bundle]=0;
michael@0:         pcmbundle[ch_in_bundle++]=vb->pcm[j];
michael@0:       }
michael@0:     }
michael@0: 
michael@0:     _residue_P[ci->residue_type[info->residuesubmap[i]]]->
michael@0:       inverse(vb,b->residue[info->residuesubmap[i]],
michael@0:               pcmbundle,zerobundle,ch_in_bundle);
michael@0:   }
michael@0: 
michael@0:   /* channel coupling */
michael@0:   for(i=info->coupling_steps-1;i>=0;i--){
michael@0:     float *pcmM=vb->pcm[info->coupling_mag[i]];
michael@0:     float *pcmA=vb->pcm[info->coupling_ang[i]];
michael@0: 
michael@0:     for(j=0;j<n/2;j++){
michael@0:       float mag=pcmM[j];
michael@0:       float ang=pcmA[j];
michael@0: 
michael@0:       if(mag>0)
michael@0:         if(ang>0){
michael@0:           pcmM[j]=mag;
michael@0:           pcmA[j]=mag-ang;
michael@0:         }else{
michael@0:           pcmA[j]=mag;
michael@0:           pcmM[j]=mag+ang;
michael@0:         }
michael@0:       else
michael@0:         if(ang>0){
michael@0:           pcmM[j]=mag;
michael@0:           pcmA[j]=mag+ang;
michael@0:         }else{
michael@0:           pcmA[j]=mag;
michael@0:           pcmM[j]=mag-ang;
michael@0:         }
michael@0:     }
michael@0:   }
michael@0: 
michael@0:   /* compute and apply spectral envelope */
michael@0:   for(i=0;i<vi->channels;i++){
michael@0:     float *pcm=vb->pcm[i];
michael@0:     int submap=info->chmuxlist[i];
michael@0:     _floor_P[ci->floor_type[info->floorsubmap[submap]]]->
michael@0:       inverse2(vb,b->flr[info->floorsubmap[submap]],
michael@0:                floormemo[i],pcm);
michael@0:   }
michael@0: 
michael@0:   /* transform the PCM data; takes PCM vector, vb; modifies PCM vector */
michael@0:   /* only MDCT right now.... */
michael@0:   for(i=0;i<vi->channels;i++){
michael@0:     float *pcm=vb->pcm[i];
michael@0:     mdct_backward(b->transform[vb->W][0],pcm,pcm);
michael@0:   }
michael@0: 
michael@0:   /* all done! */
michael@0:   return(0);
michael@0: }
michael@0: 
michael@0: /* export hooks */
michael@0: const vorbis_func_mapping mapping0_exportbundle={
michael@0:   &mapping0_pack,
michael@0:   &mapping0_unpack,
michael@0:   &mapping0_free_info,
michael@0:   &mapping0_forward,
michael@0:   &mapping0_inverse
michael@0: };