L3PSY.C
上传用户:njqiyou
上传日期:2007-01-08
资源大小:574k
文件大小:36k
源码类别:

mpeg/mp3

开发平台:

C/C++

  1. /**********************************************************************
  2.  * ISO MPEG Audio Subgroup Software Simulation Group (1996)
  3.  * ISO 13818-3 MPEG-2 Audio Encoder - Lower Sampling Frequency Extension
  4.  *
  5.  * $Id: l3psy.c,v 1.2 1997/01/19 22:28:29 rowlands Exp $
  6.  *
  7.  * $Log: l3psy.c,v $
  8.  * Revision 1.2  1997/01/19 22:28:29  rowlands
  9.  * Layer 3 bug fixes from Seymour Shlien
  10.  *
  11.  * Revision 1.1  1996/02/14 04:04:23  rowlands
  12.  * Initial revision
  13.  *
  14.  * Received from Mike Coleman
  15.  **********************************************************************/
  16. /**********************************************************************
  17.  *   date   programmers         comment                               *
  18.  * 2/25/91  Davis Pan           start of version 1.0 records          *
  19.  * 5/10/91  W. Joseph Carter    Ported to Macintosh and Unix.         *
  20.  * 7/10/91  Earle Jennings      Ported to MsDos.                      *
  21.  *                              replace of floats with FLOAT          *
  22.  * 2/11/92  W. Joseph Carter    Fixed mem_alloc() arg for "absthr".   *
  23.  * 3/16/92  Masahiro Iwadare Modification for Layer III            *
  24.  * 17/4/93  Masahiro Iwadare    Updated for IS Modification           *
  25.  **********************************************************************/
  26. #include "common.h"
  27. #include "encoder.h"
  28. #include "l3psy.h"
  29. #include "l3side.h"
  30. #include <assert.h>
  31. #define maximum(x,y) ( (x>y) ? x : y )
  32. #define minimum(x,y) ( (x<y) ? x : y )
  33. void L3para_read( double sfreq, int numlines[CBANDS], int partition_l[HBLKSIZE],
  34.   double minval[CBANDS], double qthr_l[CBANDS], double norm_l[CBANDS],
  35.   double s3_l[CBANDS][CBANDS], int partition_s[HBLKSIZE_s], double qthr_s[CBANDS_s],
  36.   double norm_s[CBANDS_s], double SNR_s[CBANDS_s],
  37.   int cbw_l[SBMAX_l], int bu_l[SBMAX_l], int bo_l[SBMAX_l],
  38.   double w1_l[SBMAX_l], double w2_l[SBMAX_l],
  39.   int cbw_s[SBMAX_s], int bu_s[SBMAX_s], int bo_s[SBMAX_s],
  40.   double w1_s[SBMAX_s], double w2_s[SBMAX_s] );
  41. void L3psycho_anal( short int *buffer, short int savebuf[1344], int chn, int lay, FLOAT snr32[32],
  42.     double sfreq, double ratio_d[21], double ratio_ds[12][3],
  43.     double *pe, gr_info *cod_info )
  44. {
  45.     static double ratio[2][21];
  46.     static double ratio_s[2][12][3];
  47.     int blocktype;
  48.     unsigned int   b, i, j, k;
  49.     double         r_prime, phi_prime; /* not FLOAT */
  50.     FLOAT          freq_mult, bval_lo, min_thres, sum_energy;
  51.     double         tb, temp1,temp2,temp3;
  52.     /*         nint(); Layer III */
  53.     double   thr[CBANDS];
  54. /* The static variables "r", "phi_sav", "new", "old" and "oldest" have    */
  55. /* to be remembered for the unpredictability measure.  For "r" and        */
  56. /* "phi_sav", the first index from the left is the channel select and     */
  57. /* the second index is the "age" of the data.                             */
  58.    static FLOAT window_s[BLKSIZE_s] ;
  59.  static int     new = 0, old = 1, oldest = 0;
  60.  static int     init = 0, flush, sync_flush, syncsize, sfreq_idx;
  61.  static double  cw[HBLKSIZE], eb[CBANDS];
  62.  static double  ctb[CBANDS];
  63.  static double SNR_l[CBANDS], SNR_s[CBANDS_s];
  64.  static int init_L3;
  65.  static double minval[CBANDS],qthr_l[CBANDS],norm_l[CBANDS];
  66.  static double qthr_s[CBANDS_s],norm_s[CBANDS_s];
  67.  static double nb_1[2][CBANDS], nb_2[2][CBANDS];
  68.  static double s3_l[CBANDS][CBANDS]; /* s3_s[CBANDS_s][CBANDS_s]; */
  69. /* Scale Factor Bands */
  70.  static int cbw_l[SBMAX_l],bu_l[SBMAX_l],bo_l[SBMAX_l] ;
  71.  static int cbw_s[SBMAX_s],bu_s[SBMAX_s],bo_s[SBMAX_s] ;
  72.  static double w1_l[SBMAX_l], w2_l[SBMAX_l];
  73.  static double w1_s[SBMAX_s], w2_s[SBMAX_s];
  74.  static double en[SBMAX_l],   thm[SBMAX_l] ;
  75.  static int blocktype_old[2] ;
  76.  int sb,sblock;
  77.  static int partition_l[HBLKSIZE],partition_s[HBLKSIZE_s];
  78. /* The following static variables are constants.                           */
  79.  static double  nmt = 5.5;
  80.  static FLOAT   crit_band[27] = {0,  100,  200, 300, 400, 510, 630,  770,
  81.                                920, 1080, 1270,1480,1720,2000,2320, 2700,
  82.                               3150, 3700, 4400,5300,6400,7700,9500,12000,
  83.                              15500,25000,30000};
  84.  static FLOAT   bmax[27] = {20.0, 20.0, 20.0, 20.0, 20.0, 17.0, 15.0,
  85.                             10.0,  7.0,  4.4,  4.5,  4.5,  4.5,  4.5,
  86.                              4.5,  4.5,  4.5,  4.5,  4.5,  4.5,  4.5,
  87.                              4.5,  4.5,  4.5,  3.5,  3.5,  3.5};
  88. /* The following pointer variables point to large areas of memory         */
  89. /* dynamically allocated by the mem_alloc() function.  Dynamic memory     */
  90. /* allocation is used in order to avoid stack frame or data area          */
  91. /* overflow errors that otherwise would have occurred at compile time     */
  92. /* on the Macintosh computer.                                             */
  93.  FLOAT          *grouped_c, *grouped_e, *nb, *cb, *ecb, *bc;
  94.  FLOAT          *wsamp_r, *wsamp_i, *phi, *energy;
  95.  static FLOAT energy_s[3][256];
  96.  static FLOAT phi_s[3][256] ; /* 256 samples not 129 */
  97.  FLOAT          *c, *fthr;
  98.  F32            *snrtmp;
  99.  static int *numlines ;
  100.  static int     *partition;
  101.  static FLOAT   *cbval, *rnorm;
  102.  static FLOAT   *window;
  103.  static FLOAT   *absthr;
  104.  static double  *tmn;
  105.  static FCB     *s;
  106.  static FHBLK   *lthr;
  107.  static F2HBLK  *r, *phi_sav;
  108. /* These dynamic memory allocations simulate "automatic" variables        */
  109. /* placed on the stack.  For each mem_alloc() call here, there must be    */
  110. /* a corresponding mem_free() call at the end of this function.           */
  111.  grouped_c = (FLOAT *) mem_alloc(sizeof(FCB), "grouped_c");
  112.  grouped_e = (FLOAT *) mem_alloc(sizeof(FCB), "grouped_e");
  113.  nb = (FLOAT *) mem_alloc(sizeof(FCB), "nb");
  114.  cb = (FLOAT *) mem_alloc(sizeof(FCB), "cb");
  115.  ecb = (FLOAT *) mem_alloc(sizeof(FCB), "ecb");
  116.  bc = (FLOAT *) mem_alloc(sizeof(FCB), "bc");
  117.  wsamp_r = (FLOAT *) mem_alloc(sizeof(FBLK), "wsamp_r");
  118.  wsamp_i = (FLOAT *) mem_alloc(sizeof(FBLK), "wsamp_i");
  119.  phi = (FLOAT *) mem_alloc(sizeof(FBLK), "phi");
  120.  energy = (FLOAT *) mem_alloc(sizeof(FBLK), "energy");
  121.  c = (FLOAT *) mem_alloc(sizeof(FHBLK), "c");
  122.  fthr = (FLOAT *) mem_alloc(sizeof(FHBLK), "fthr");
  123.  snrtmp = (F32 *) mem_alloc(sizeof(F2_32), "snrtmp");
  124.     assert( lay == 3 );
  125.  if(init==0){
  126. /* These dynamic memory allocations simulate "static" variables placed    */
  127. /* in the data space.  Each mem_alloc() call here occurs only once at     */
  128. /* initialization time.  The mem_free() function must not be called.      */
  129.      numlines = (int *) mem_alloc(sizeof(ICB), "numlines");
  130.      partition = (int *) mem_alloc(sizeof(IHBLK), "partition");
  131.      cbval = (FLOAT *) mem_alloc(sizeof(FCB), "cbval");
  132.      rnorm = (FLOAT *) mem_alloc(sizeof(FCB), "rnorm");
  133.      window = (FLOAT *) mem_alloc(sizeof(FBLK), "window");
  134.      absthr = (FLOAT *) mem_alloc(sizeof(FHBLK), "absthr"); 
  135.      tmn = (double *) mem_alloc(sizeof(DCB), "tmn");
  136.      s = (FCB *) mem_alloc(sizeof(FCBCB), "s");
  137.      lthr = (FHBLK *) mem_alloc(sizeof(F2HBLK), "lthr");
  138.      r = (F2HBLK *) mem_alloc(sizeof(F22HBLK), "r");
  139.      phi_sav = (F2HBLK *) mem_alloc(sizeof(F22HBLK), "phi_sav");
  140. /*#if 0 */
  141.      i = sfreq + 0.5;
  142.      switch(i){
  143.         case 32000: sfreq_idx = 0; break;
  144.         case 44100: sfreq_idx = 1; break;
  145.         case 48000: sfreq_idx = 2; break;
  146.         default:    printf("error, invalid sampling frequency: %d Hzn",i);
  147.         exit(-1);
  148.      }
  149.      printf("absthr[][] sampling frequency index: %dn",sfreq_idx);
  150.      read_absthr(absthr, sfreq_idx);
  151.      switch(lay){
  152. case 1: sync_flush=576; flush=384; syncsize=1024; break;
  153. case 2: sync_flush=480; flush=576; syncsize=1056; break;
  154. case 3: sync_flush=768; flush=576; syncsize=1344; break;
  155.        default: printf("Bad lay value:(%d)",lay); exit(-1); break;
  156.      }
  157. /* #endif */
  158. /* calculate HANN window coefficients */
  159. /*   for(i=0;i<BLKSIZE;i++)  window[i]  =0.5*(1-cos(2.0*PI*i/(BLKSIZE-1.0)));*/
  160.      for(i=0;i<BLKSIZE;i++)  window[i]  =0.5*(1-cos(2.0*PI*(i-0.5)/BLKSIZE));
  161.      for(i=0;i<BLKSIZE_s;i++)window_s[i]=0.5*(1-cos(2.0*PI*(i-0.5)/BLKSIZE_s));
  162. /* reset states used in unpredictability measure */
  163.      for(i=0;i<HBLKSIZE;i++){
  164.         r[0][0][i]=r[1][0][i]=r[0][1][i]=r[1][1][i]=0;
  165.         phi_sav[0][0][i]=phi_sav[1][0][i]=0;
  166.         phi_sav[0][1][i]=phi_sav[1][1][i]=0;
  167.         lthr[0][i] = 60802371420160.0;
  168.         lthr[1][i] = 60802371420160.0;
  169.      }
  170. /*****************************************************************************
  171.  * Initialization: Compute the following constants for use later             *
  172.  *    partition[HBLKSIZE] = the partition number associated with each        *
  173.  *                          frequency line                                   *
  174.  *    cbval[CBANDS]       = the center (average) bark value of each          *
  175.  *                          partition                                        *
  176.  *    numlines[CBANDS]    = the number of frequency lines in each partition  *
  177.  *    tmn[CBANDS]         = tone masking noise                               *
  178.  *****************************************************************************/
  179. /* compute fft frequency multiplicand */
  180.      freq_mult = sfreq/BLKSIZE;
  181.  
  182. /* calculate fft frequency, then bval of each line (use fthr[] as tmp storage)*/
  183.      for(i=0;i<HBLKSIZE;i++){
  184.         temp1 = i*freq_mult;
  185.         j = 1;
  186.         while(temp1>crit_band[j])j++;
  187.         fthr[i]=j-1+(temp1-crit_band[j-1])/(crit_band[j]-crit_band[j-1]);
  188.      }
  189.      partition[0] = 0;
  190. /* temp2 is the counter of the number of frequency lines in each partition */
  191.      temp2 = 1;
  192.      cbval[0]=fthr[0];
  193.      bval_lo=fthr[0];
  194.      for(i=1;i<HBLKSIZE;i++){
  195.         if((fthr[i]-bval_lo)>0.33){
  196.            partition[i]=partition[i-1]+1;
  197.            cbval[partition[i-1]] = cbval[partition[i-1]]/temp2;
  198.            cbval[partition[i]] = fthr[i];
  199.            bval_lo = fthr[i];
  200.            numlines[partition[i-1]] = temp2;
  201.            temp2 = 1;
  202.         }
  203.         else {
  204.            partition[i]=partition[i-1];
  205.            cbval[partition[i]] += fthr[i];
  206.            temp2++;
  207.         }
  208.      }
  209.      numlines[partition[i-1]] = temp2;
  210.      cbval[partition[i-1]] = cbval[partition[i-1]]/temp2;
  211.  
  212. /************************************************************************
  213.  * Now compute the spreading function, s[j][i], the value of the spread-*
  214.  * ing function, centered at band j, for band i, store for later use    *
  215.  ************************************************************************/
  216.      for(j=0;j<CBANDS;j++){
  217.         for(i=0;i<CBANDS;i++){
  218.            temp1 = (cbval[i] - cbval[j])*1.05;
  219.            if(temp1>=0.5 && temp1<=2.5){
  220.               temp2 = temp1 - 0.5;
  221.               temp2 = 8.0 * (temp2*temp2 - 2.0 * temp2);
  222.            }
  223.            else temp2 = 0;
  224.            temp1 += 0.474;
  225.            temp3 = 15.811389+7.5*temp1-17.5*sqrt((double) (1.0+temp1*temp1));
  226.            if(temp3 <= -100) s[i][j] = 0;
  227.            else {
  228.               temp3 = (temp2 + temp3)*LN_TO_LOG10;
  229.               s[i][j] = exp(temp3);
  230.            }
  231.         }
  232.      }
  233.   /* Calculate Tone Masking Noise values */
  234.      for(j=0;j<CBANDS;j++){
  235.         temp1 = 15.5 + cbval[j];
  236.         tmn[j] = (temp1>24.5) ? temp1 : 24.5;
  237.   /* Calculate normalization factors for the net spreading functions */
  238.         rnorm[j] = 0;
  239.         for(i=0;i<CBANDS;i++){
  240.            rnorm[j] += s[j][i];
  241.         }
  242.      }
  243.      init++;
  244.  }
  245.  
  246. /************************* End of Initialization *****************************/
  247.  switch(lay) {
  248.   case 1:
  249.   case 2:
  250. for ( i=0; i<lay; i++)
  251.   {
  252. /*****************************************************************************
  253.  * Net offset is 480 samples (1056-576) for layer 2; this is because one must*
  254.  * stagger input data by 256 samples to synchronize psychoacoustic model with*
  255.  * filter bank outputs, then stagger so that center of 1024 FFT window lines *
  256.  * up with center of 576 "new" audio samples.                                *
  257.  *                                                                           *
  258.  * For layer 1, the input data still needs to be staggered by 256 samples,   *
  259.  * then it must be staggered again so that the 384 "new" samples are centered*
  260.  * in the 1024 FFT window.  The net offset is then 576 and you need 448 "new"*
  261.  * samples for each iteration to keep the 384 samples of interest centered   *
  262.  *****************************************************************************/
  263.   for (j=0; j<syncsize; j++)
  264.   {
  265.     if (j < (sync_flush) )
  266.       savebuf[j] = savebuf[j+flush];
  267.     else
  268.       savebuf[j] = *buffer++;
  269. /**window data with HANN window***********************************************/
  270.     if (j<BLKSIZE)
  271.     {
  272.       wsamp_r[j] = window[j]*((FLOAT) savebuf[j]); 
  273.       wsamp_i[j] = 0;
  274.     }
  275.   }
  276. /**Compute FFT****************************************************************/
  277.         fft(wsamp_r,wsamp_i,energy,phi,1024);
  278. /*****************************************************************************
  279.  * calculate the unpredictability measure, given energy[f] and phi[f]        *
  280.  *****************************************************************************/
  281.         for(j=0; j<HBLKSIZE; j++){
  282.            r_prime = 2.0 * r[chn][old][j] - r[chn][oldest][j];
  283.            phi_prime = 2.0 * phi_sav[chn][old][j] - phi_sav[chn][oldest][j];
  284.            r[chn][new][j] = sqrt((double) energy[j]);
  285.            phi_sav[chn][new][j] = phi[j];
  286.    temp1 = r[chn][new][j] * cos((double) phi[j])
  287.    - r_prime * cos(phi_prime);
  288.    temp2=r[chn][new][j] * sin((double) phi[j])
  289.    - r_prime * sin(phi_prime);
  290.            temp3=r[chn][new][j] + fabs(r_prime);
  291.            if(temp3 != 0)c[j]=sqrt(temp1*temp1+temp2*temp2)/temp3;
  292.            else c[j] = 0;
  293.         }
  294. /*only update data "age" pointers after you are done with the second channel */
  295. /*for layer 1 computations, for the layer 2 double computations, the pointers*/
  296. /*are reset automatically on the second pass                                 */
  297.         if(lay==2 || chn==1){
  298.            if(new==0){new = 1; oldest = 1;}
  299.            else {new = 0; oldest = 0;}
  300.            if(old==0)old = 1; else old = 0;
  301.         }
  302. /*****************************************************************************
  303.  * Calculate the grouped, energy-weighted, unpredictability measure,         *
  304.  * grouped_c[], and the grouped energy. grouped_e[]                          *
  305.  *****************************************************************************/
  306.         for(j=1;j<CBANDS;j++){
  307.            grouped_e[j] = 0;
  308.            grouped_c[j] = 0;
  309.         }
  310.         grouped_e[0] = energy[0];
  311.         grouped_c[0] = energy[0]*c[0];
  312.         for(j=1;j<HBLKSIZE;j++){
  313.            grouped_e[partition[j]] += energy[j];
  314.            grouped_c[partition[j]] += energy[j]*c[j];
  315.         }
  316. /*****************************************************************************
  317.  * convolve the grouped energy-weighted unpredictability measure             *
  318.  * and the grouped energy with the spreading function, s[j][k]               *
  319.  *****************************************************************************/
  320.         for(j=0;j<CBANDS;j++){
  321.            ecb[j] = 0;
  322.            cb[j] = 0;
  323.            for(k=0;k<CBANDS;k++){
  324.               if(s[j][k] != 0.0){
  325.                  ecb[j] += s[j][k]*grouped_e[k];
  326.                  cb[j] += s[j][k]*grouped_c[k];
  327.               }
  328.            }
  329.            if(ecb[j] !=0)cb[j] = cb[j]/ecb[j];
  330.            else cb[j] = 0;
  331.         }
  332. /*****************************************************************************
  333.  * Calculate the required SNR for each of the frequency partitions           *
  334.  *         this whole section can be accomplished by a table lookup          *
  335.  *****************************************************************************/
  336.         for(j=0;j<CBANDS;j++){
  337.            if(cb[j]<.05)cb[j]=0.05;
  338.            else if(cb[j]>.5)cb[j]=0.5;
  339. /*         tb = -0.434294482*log((double) cb[j])-0.301029996; */
  340.            tb = -0.43 *log((double) cb[j]) - 0.29 ;
  341.            if(tb<0.0) tb=0.0; else if(tb>1.0) tb=1.0;
  342.            bc[j] = tmn[j]*tb + nmt*(1.0-tb);
  343.            k = cbval[j] + 0.5;
  344.            bc[j] = (bc[j] > bmax[k]) ? bc[j] : bmax[k];
  345.            bc[j] = exp((double) -bc[j]*LN_TO_LOG10);
  346.         }
  347. /*****************************************************************************
  348.  * Calculate the permissible noise energy level in each of the frequency     *
  349.  * partitions. Include absolute threshold and pre-echo controls              *
  350.  *         this whole section can be accomplished by a table lookup          *
  351.  *****************************************************************************/
  352.         for(j=0;j<CBANDS;j++)
  353.            if(rnorm[j] && numlines[j])
  354.               nb[j] = ecb[j]*bc[j]/(rnorm[j]*numlines[j]);
  355.            else nb[j] = 0;
  356.         for(j=0;j<HBLKSIZE;j++){
  357.            temp1=nb[partition[j]];  /* preliminary threshold */
  358.            temp1=(temp1>absthr[j])?temp1:absthr[j];
  359. /*do not use pre-echo control for layer 2 because it may do bad things to the*/
  360. /*  MUSICAM bit allocation algorithm                                         */
  361.            if(lay==1){
  362.               fthr[j] = (temp1 < lthr[chn][j]) ? temp1 : lthr[chn][j];
  363.               temp2 = temp1 * 0.00316;
  364.               fthr[j] = (temp2 > fthr[j]) ? temp2 : fthr[j];
  365.            }
  366.            else fthr[j] = temp1;
  367.            lthr[chn][j] = LXMIN*temp1;
  368.         }
  369. /*****************************************************************************
  370.  * Translate the 512 threshold values to the 32 filter bands of the coder    *
  371.  *****************************************************************************/
  372.         for(j=0;j<193;j += 16){
  373.            min_thres = fthr[j];
  374.            sum_energy = energy[j];
  375.            for(k=1;k<17;k++){
  376.               if(min_thres>fthr[j+k]) min_thres = fthr[j+k];
  377.               sum_energy += energy[j+k];
  378.            }
  379.            snrtmp[i][j/16] = sum_energy/(min_thres * 17.0);
  380.            snrtmp[i][j/16] = 4.342944819 * log((double)snrtmp[i][j/16]);
  381.         }
  382.         for(j=208;j<(HBLKSIZE-1);j += 16){
  383.            min_thres = 0.0;
  384.            sum_energy = 0.0;
  385.            for(k=0;k<17;k++){
  386.               min_thres += fthr[j+k];
  387.               sum_energy += energy[j+k];
  388.            }
  389.            snrtmp[i][j/16] = sum_energy/min_thres;
  390.            snrtmp[i][j/16] = 4.342944819 * log((double)snrtmp[i][j/16]);
  391.         }
  392. /*****************************************************************************
  393.  * End of Psychoacuostic calculation loop                                    *
  394.  *****************************************************************************/
  395.      }
  396.      for(i=0; i<32; i++){
  397.         if(lay==2) /* if(lay==2 && chn==2) MI */
  398.            snr32[i]=(snrtmp[0][i]>snrtmp[1][i])?snrtmp[0][i]:snrtmp[1][i];
  399.         else snr32[i]=snrtmp[0][i];
  400.      }
  401.      break;
  402. /*************************************************************************/
  403. /**       LAYER 3                                                        */
  404. /*************************************************************************/
  405.    case 3:
  406. if ( init_L3 == 0 )
  407. {
  408.     L3para_read( sfreq,numlines,partition_l,minval,qthr_l,norm_l,s3_l,
  409.  partition_s,qthr_s,norm_s,SNR_s,
  410.  cbw_l,bu_l,bo_l,w1_l,w2_l, cbw_s,bu_s,bo_s,w1_s,w2_s );
  411.     init_L3 ++ ;
  412. }
  413. for ( j = 0; j < 21; j++ )
  414.     ratio_d[j] = ratio[chn][j];
  415. for ( j = 0; j < 12; j++ )
  416.     for ( i = 0; i < 3; i++ )
  417. ratio_ds[j][i] = ratio_s[chn][j][i];
  418. if ( chn == 0 )
  419.     if ( new == 0 )
  420.     {
  421. new = 1;
  422. old = 0;
  423. oldest = 1;
  424.     }
  425.     else
  426.     {
  427. new = 0;
  428. old = 1;
  429. oldest = 0;
  430.     }
  431. /**********************************************************************
  432. *  Delay signal by sync_flush=768 samples                             *
  433. **********************************************************************/
  434. for ( j = 0; j < sync_flush; j++ ) /* for long window samples */
  435.     savebuf[j] = savebuf[j+flush];
  436. for ( j = sync_flush; j < syncsize; j++ )
  437.     savebuf[j] = *buffer++;
  438. for ( j = 0; j < BLKSIZE; j++ )
  439. { /**window data with HANN window**/
  440.     wsamp_r[j] = window[j] * savebuf[j];  
  441.     wsamp_i[j] = 0.0;
  442. }
  443. /**********************************************************************
  444. *    compute unpredicatability of first six spectral lines            * 
  445. **********************************************************************/
  446. fft( wsamp_r, wsamp_i, energy, phi, 1024 ); /**long FFT**/
  447. for ( j = 0; j < 6; j++ )
  448. {  /* calculate unpredictability measure cw */
  449.     r_prime = 2.0 * r[chn][old][j] - r[chn][oldest][j];
  450.     phi_prime = 2.0 * phi_sav[chn][old][j]-phi_sav[chn][oldest][j];
  451.     r[chn][new][j] = sqrt((double) energy[j]);
  452.     phi_sav[chn][new][j] = phi[j];
  453.     temp1 = r[chn][new][j] * cos((double) phi[j])
  454. - r_prime * cos(phi_prime);
  455.     temp2 = r[chn][new][j] * sin((double) phi[j])
  456. - r_prime * sin(phi_prime);
  457.     temp3=r[chn][new][j] + fabs(r_prime);
  458.     
  459.     if ( temp3 != 0.0 )
  460. cw[j] = sqrt( temp1*temp1+temp2*temp2 ) / temp3;
  461.     else
  462. cw[j] = 0;
  463. }
  464. /**********************************************************************
  465. *     compute unpredicatibility of next 200 spectral lines            *
  466. **********************************************************************/ 
  467. for ( sblock = 0; sblock < 3; sblock++ )
  468. { /**window data with HANN window**/
  469.     for ( j = 0, k = 128 * (2 + sblock); j < 256; j++, k++ )
  470.     {
  471. wsamp_r[j] = window_s[j]* savebuf[k]; 
  472. wsamp_i[j] = 0.0;
  473.     } /* short FFT*/
  474.     
  475.     fft( wsamp_r, wsamp_i, &energy_s[sblock][0], &phi_s[sblock][0], 256 );
  476.         }
  477.  
  478.         sblock = 1;
  479. for ( j = 6; j < 206; j += 4 )
  480. {/* calculate unpredictability measure cw */
  481.     double r2, phi2, temp1, temp2, temp3;
  482.     
  483.     k = (j+2) / 4; 
  484.     r_prime = 2.0 * sqrt((double) energy_s[0][k])
  485. - sqrt((double) energy_s[2][k]);
  486.     phi_prime = 2.0 * phi_s[0][k] - phi_s[2][k];
  487.     r2 = sqrt((double) energy_s[1][k]);
  488.     phi2 = phi_s[1][k];
  489.     temp1 = r2 * cos( phi2 ) - r_prime * cos( phi_prime );
  490.     temp2 = r2 * sin( phi2 ) - r_prime * sin( phi_prime );
  491.     temp3 = r2 + fabs( r_prime );
  492.     if ( temp3 != 0.0 )
  493. cw[j] = sqrt( temp1 * temp1 + temp2 * temp2 ) / temp3;
  494.     else
  495. cw[j] = 0.0;
  496.     cw[j+1] = cw[j+2] = cw[j+3] = cw[j];
  497. }
  498. /**********************************************************************
  499. *    Set unpredicatiblility of remaining spectral lines to 0.4        *
  500. **********************************************************************/
  501. for ( j = 206; j < HBLKSIZE; j++ )
  502.     cw[j] = 0.4;
  503. /**********************************************************************
  504. *    Calculate the energy and the unpredictability in the threshold   *
  505. *    calculation partitions                                           *
  506. **********************************************************************/
  507. for ( b = 0; b < CBANDS; b++ )
  508. {
  509.     eb[b] = 0.0;
  510.     cb[b] = 0.0;
  511. }
  512. for ( j = 0; j < HBLKSIZE; j++ )
  513. {
  514.     int tp = partition_l[j];
  515.     if ( tp >= 0 )
  516.     {
  517. eb[tp] += energy[j];
  518. cb[tp] += cw[j] * energy[j];
  519.     }
  520. }
  521. /**********************************************************************
  522. *      convolve the partitioned energy and unpredictability           *
  523. *      with the spreading function, s3_l[b][k]                        *
  524. ******************************************************************** */
  525. for ( b = 0; b < CBANDS; b++ )
  526. {
  527.     ecb[b] = 0.0;
  528.     ctb[b] = 0.0;
  529. }
  530. for ( b = 0;b < CBANDS; b++ )
  531. {
  532.     for ( k = 0; k < CBANDS; k++ )
  533.     {
  534. ecb[b] += s3_l[b][k] * eb[k]; /* sprdngf for Layer III */
  535. ctb[b] += s3_l[b][k] * cb[k];
  536.     }
  537. }
  538. /* calculate the tonality of each threshold calculation partition */
  539. /* calculate the SNR in each threshhold calculation partition */
  540. for ( b = 0; b < CBANDS; b++ )
  541. {
  542.     double cbb,tbb;
  543.     if (ecb[b] != 0.0 )
  544.                 {
  545. cbb = ctb[b]/ecb[b];
  546.                 if (cbb <0.01) cbb = 0.01;
  547. cbb = log( cbb);
  548.                 }
  549.     else
  550. cbb = 0.0 ;
  551.     tbb = -0.299 - 0.43*cbb;  /* conv1=-0.299, conv2=-0.43 */
  552.     tbb = minimum( 1.0, maximum( 0.0, tbb) ) ;  /* 0<tbb<1 */
  553.     SNR_l[b] = maximum( minval[b], 29.0*tbb+6.0*(1.0-tbb) );
  554. } /* TMN=29.0,NMT=6.0 for all calculation partitions */
  555. for ( b = 0; b < CBANDS; b++ ) /* calculate the threshold for each partition */
  556.     nb[b] = ecb[b] * norm_l[b] * exp( -SNR_l[b] * LN_TO_LOG10 );
  557. for ( b = 0; b < CBANDS; b++ )
  558. { /* pre-echo control */
  559.     double temp_1; /* BUG of IS */
  560.     temp_1 = minimum( nb[b], minimum(2.0*nb_1[chn][b],16.0*nb_2[chn][b]) );
  561.     thr[b] = maximum( qthr_l[b], temp_1 );/* rpelev=2.0, rpelev2=16.0 */
  562.     nb_2[chn][b] = nb_1[chn][b];
  563.     nb_1[chn][b] = nb[b];
  564. }
  565. *pe = 0.0; /*  calculate percetual entropy */
  566. for ( b = 0; b < CBANDS; b++ )
  567. {
  568.     double tp ;
  569.     tp = minimum( 0.0, log((thr[b]+1.0) / (eb[b]+1.0) ) ) ; /*not log*/
  570.     *pe -= numlines[b] * tp ;
  571. } /* thr[b] -> thr[b]+1.0 : for non sound portition */
  572. #define switch_pe  1800
  573.         blocktype = NORM_TYPE;
  574. if ( *pe < switch_pe )
  575. { /* no attack : use long blocks */
  576.     switch( blocktype_old[chn] ) 
  577.     {
  578.       case NORM_TYPE:
  579.       case STOP_TYPE:
  580. blocktype = NORM_TYPE;
  581. break;
  582.     
  583.       case SHORT_TYPE:
  584. blocktype = STOP_TYPE;
  585. break;
  586.     
  587.       case START_TYPE:
  588. fprintf( stderr, "Error in block selectingn" );
  589. abort();
  590. break; /* problem */
  591.     }
  592.     /* threshold calculation (part 2) */
  593.     for ( sb = 0; sb < SBMAX_l; sb++ )
  594.     {
  595. en[sb] = w1_l[sb] * eb[bu_l[sb]] + w2_l[sb] * eb[bo_l[sb]];
  596. thm[sb] = w1_l[sb] *thr[bu_l[sb]] + w2_l[sb] * thr[bo_l[sb]];
  597. for ( b = bu_l[sb]+1; b < bo_l[sb]; b++ )
  598. {
  599.     en[sb]  += eb[b];
  600.     thm[sb] += thr[b];
  601. }
  602. if ( en[sb] != 0.0 )
  603.     ratio[chn][sb] = thm[sb]/en[sb];
  604. else
  605.     ratio[chn][sb] = 0.0;
  606.     }
  607. }
  608. else 
  609. {
  610.     /* attack : use short blocks */
  611.     blocktype = SHORT_TYPE;
  612.     
  613.     if ( blocktype_old[chn] == NORM_TYPE ) 
  614. blocktype_old[chn] = START_TYPE;
  615.     if ( blocktype_old[chn] == STOP_TYPE )
  616. blocktype_old[chn] = SHORT_TYPE ;
  617.     
  618.     /* threshold calculation for short blocks */
  619.     
  620.     for ( sblock = 0; sblock < 3; sblock++ )
  621.     {
  622. for ( b = 0; b < CBANDS_s; b++ )
  623. {
  624.     eb[b] = 0.0;
  625.     ecb[b] = 0.0;
  626. }
  627. for ( j = 0; j < HBLKSIZE_s; j++ )
  628.     eb[partition_s[j]] += energy_s[sblock][j];
  629. for ( b = 0; b < CBANDS_s; b++ )
  630.     for ( k = 0; k < CBANDS_s; k++ )
  631. ecb[b] += s3_l[b][k] * eb[k];
  632. for ( b = 0; b < CBANDS_s; b++ )
  633. {
  634.     nb[b] = ecb[b] * norm_l[b] * exp( (double) SNR_s[b] * LN_TO_LOG10 );
  635.     thr[b] = maximum (qthr_s[b],nb[b]);
  636. }
  637. for ( sb = 0; sb < SBMAX_s; sb++ )
  638. {
  639.     en[sb] = w1_s[sb] * eb[bu_s[sb]] + w2_s[sb] * eb[bo_s[sb]];
  640.     thm[sb] = w1_s[sb] *thr[bu_s[sb]] + w2_s[sb] * thr[bo_s[sb]];
  641.     for ( b = bu_s[sb]+1; b < bo_s[sb]; b++ )
  642.     {
  643. en[sb] += eb[b];
  644. thm[sb] += thr[b];
  645.     }
  646.     if ( en[sb] != 0.0 )
  647. ratio_s[chn][sb][sblock] = thm[sb]/en[sb];
  648.     else
  649. ratio_s[chn][sb][sblock] = 0.0;
  650. }
  651.     }
  652. cod_info->block_type = blocktype_old[chn];
  653. blocktype_old[chn] = blocktype;
  654. if ( cod_info->block_type == NORM_TYPE )
  655.     cod_info->window_switching_flag = 0;
  656. else
  657.     cod_info->window_switching_flag = 1;
  658. cod_info->mixed_block_flag = 0;
  659. break;
  660.   default:
  661.      printf("error, invalid MPEG/audio coding layer: %dn",lay);
  662.  }
  663. /* These mem_free() calls must correspond with the mem_alloc() calls     */
  664. /* used at the beginning of this function to simulate "automatic"        */
  665. /* variables placed on the stack.                                        */
  666.  mem_free((void **) &grouped_c);
  667.  mem_free((void **) &grouped_e);
  668.  mem_free((void **) &nb);
  669.  mem_free((void **) &cb);
  670.  mem_free((void **) &ecb);
  671.  mem_free((void **) &bc);
  672.  mem_free((void **) &wsamp_r);
  673.  mem_free((void **) &wsamp_i);
  674.  mem_free((void **) &phi);
  675.  mem_free((void **) &energy);
  676.  mem_free((void **) &c);
  677.  mem_free((void **) &fthr);
  678.  mem_free((void **) &snrtmp);
  679. }
  680. #ifdef DEBUG
  681. #undef DEBUG
  682. #endif
  683. void L3para_read(double sfreq, int *numlines, int *partition_l, double *minval, double *qthr_l, double *norm_l, double (*s3_l)[63], int *partition_s, double *qthr_s, double *norm_s, double *SNR, int *cbw_l, int *bu_l, int *bo_l, double *w1_l, double *w2_l, int *cbw_s, int *bu_s, int *bo_s, double *w1_s, double *w2_s)
  684. {
  685.    double freq_tp;
  686.    static double bval_l[CBANDS], bval_s[CBANDS];
  687.    int   cbmax, cbmax_tp;
  688.    static double s3_s[CBANDS][CBANDS];
  689.    FILE *fin;
  690.    char tp[256];
  691.    int  sbmax ;
  692.    int  i,j,k,k2,loop, part_max ;
  693.    fin = OpenTableFile( "psy_data" );
  694.    if (fin == NULL)
  695.        exit( 1 );
  696. /* Read long block data */
  697.       for(loop=0;loop<6;loop++)
  698.       {
  699. fscanf(fin,"freq = %lf partition = %dn",&freq_tp,&cbmax_tp);
  700. cbmax_tp++;
  701. #ifdef DEBUG
  702. printf("freq = %f partition = %dn",freq_tp,cbmax);
  703. #endif
  704. if (sfreq == freq_tp )
  705.   {
  706.      cbmax = cbmax_tp;
  707.      for(i=0,k2=0;i<cbmax_tp;i++)
  708.        {
  709. fscanf(fin,
  710.   "No=%d #lines=%d minval=%lf qthr=%lf norm=%lf bval=%lfn",
  711.   &j,&numlines[i],&minval[i],&qthr_l[i],&norm_l[i],&bval_l[i]);
  712.         if (j!=i)
  713.          { printf("please check "psy_data"");
  714.    exit(-1);
  715.          }
  716. for(k=0;k<numlines[i];k++)
  717.   partition_l[k2++] = i ;
  718. #ifdef DEBUG
  719.      printf("No=%2d #lines=%2d minval=%4.1f qthr=%8.3f norm=%5.3f bval=%8.3fn",
  720.      i,numlines[i],minval[i],qthr_l[i],norm_l[i],bval_l[i] );
  721. #endif
  722. }
  723.    }
  724.    else
  725.    {
  726.      for(j=0;j<cbmax_tp;j++)
  727.        {
  728. fgets(tp,255,fin);
  729.         sscanf(tp,"No=%d %sn",&i,tp);
  730.         if (j!=i)
  731.          { printf("please check "psy_data."n");
  732.    exit(-1);
  733.          }
  734.        }
  735.    }
  736.        }
  737. /************************************************************************
  738.  * Now compute the spreading function, s[j][i], the value of the spread-*
  739.  * ing function, centered at band j, for band i, store for later use    *
  740.  ************************************************************************/
  741. #ifdef DEBUG
  742. printf("freq = %fn",sfreq);
  743. #endif
  744.   part_max = cbmax ;
  745.           for(i=0;i<part_max;i++)
  746.   {
  747.   double tempx,x,tempy,temp;
  748.             for(j=0;j<part_max;j++)
  749.     {
  750.              tempx = (bval_l[i] - bval_l[j])*1.05;
  751.              if (j>=i) tempx = (bval_l[i] - bval_l[j])*3.0;
  752.                else    tempx = (bval_l[i] - bval_l[j])*1.5;
  753. /*             if (j>=i) tempx = (bval_l[j] - bval_l[i])*3.0;
  754.                else    tempx = (bval_l[j] - bval_l[i])*1.5; */
  755.              if(tempx>=0.5 && tempx<=2.5)
  756.      {
  757.                temp = tempx - 0.5;
  758.                x = 8.0 * (temp*temp - 2.0 * temp);
  759.              }
  760.              else x = 0.0;
  761.              tempx += 0.474;
  762.              tempy = 15.811389 + 7.5*tempx - 17.5*sqrt(1.0+tempx*tempx);
  763.              if (tempy <= -60.0) s3_l[i][j] = 0.0;
  764.              else                s3_l[i][j] = exp( (x + tempy)*LN_TO_LOG10 );
  765. #ifdef DEBUG_S3
  766.      printf("s[%d][%d]=%fn",i,j,s3_l[i][j]);
  767. #endif
  768. #ifdef DEBUGP
  769.      printf("j=%d i=%d tempy=%f s[i][j]=%f n",i,j,tempy,s[i][j]);
  770.    minval[i] = bmax[j-1];
  771.    printf("minval[%d] = %f, j-1=%d %fn",i, minval[i] , j,fthr[i]) ;
  772. #endif
  773.             }
  774.           }
  775. /* Read short block data */
  776.       for(loop=0;loop<6;loop++)
  777.       {
  778. fscanf(fin,"freq = %lf partition = %dn",&freq_tp,&cbmax_tp);
  779. cbmax_tp++;
  780. #ifdef DEBUG
  781. printf("freq = %f partition = %dn",freq_tp,cbmax);
  782. #endif
  783. if (sfreq == freq_tp )
  784.   {
  785.      cbmax = cbmax_tp;
  786.      for(i=0,k2=0;i<cbmax_tp;i++)
  787.        {
  788. fscanf(fin,
  789.   "No=%d #lines=%d qthr=%lf norm=%lf SNR=%lf bval=%lfn",
  790.    &j,&numlines[i],&qthr_s[i],&norm_s[i],&SNR[i],&bval_s[i]);
  791.         if (j!=i)
  792.          { printf("please check "psy_data"");
  793.    exit(-1);
  794.          }
  795. for(k=0;k<numlines[i];k++)
  796.   partition_s[k2++] = i ;
  797. #ifdef DEBUG
  798.       printf("No=%2d #lines=%2d qthr=%8.3f norm=%5.3f SNR=%6.3f bval=%8.3fn",
  799.       i,numlines[i],qthr_s[i],norm_s[i],SNR[i],bval_s[i] );
  800. #endif
  801. }
  802.    }
  803.    else
  804.    {
  805.      for(j=0;j<cbmax_tp;j++)
  806.        {
  807. fgets(tp,255,fin);
  808.         sscanf(tp,"No=%d %sn",&i,tp);
  809.         if (j!=i)
  810.          { printf("please check "psy_data."n");
  811.    exit(-1);
  812.          }
  813.        }
  814.    }
  815.        }
  816. /************************************************************************
  817.  * Now compute the spreading function, s[j][i], the value of the spread-*
  818.  * ing function, centered at band j, for band i, store for later use    *
  819.  ************************************************************************/
  820. #ifdef DEBUG_S3
  821. fpp=fopen("s3_s","w");
  822. #endif
  823.   part_max = cbmax ;
  824.           for(i=0;i<part_max;i++)
  825.   {
  826.   double tempx,x,tempy,temp;
  827.             for(j=0;j<part_max;j++)
  828.     {
  829.              tempx = (bval_s[i] - bval_s[j])*1.05;
  830.              if (j>=i) tempx = (bval_s[i] - bval_s[j])*3.0;
  831.                else    tempx = (bval_s[i] - bval_s[j])*1.5;
  832.              if(tempx>=0.5 && tempx<=2.5)
  833.      {
  834.                temp = tempx - 0.5;
  835.                x = 8.0 * (temp*temp - 2.0 * temp);
  836.              }
  837.              else x = 0.0;
  838.              tempx += 0.474;
  839.              tempy = 15.811389 + 7.5*tempx - 17.5*sqrt(1.0+tempx*tempx);
  840.              if (tempy <= -60.0) s3_s[i][j] = 0.0;
  841.              else                s3_s[i][j] = exp( (x + tempy)*LN_TO_LOG10 );
  842. #ifdef DEBUG_S3
  843.      fprintf(fpp,"s3_s[%d][%d]=%fn",i,j,s3_s[i][j]);
  844. #endif
  845. #ifdef DEBUGP
  846.      printf("j=%d i=%d tempy=%f s[i][j]=%f n",i,j,tempy,s[i][j]);
  847.    minval[i] = bmax[j-1];
  848.    printf("minval[%d] = %f, j-1=%d %fn",i, minval[i] , j,fthr[i]) ;
  849. #endif
  850.             }
  851.           }
  852. #ifdef DEBUG_S3
  853. fclose(fpp);
  854. #endif
  855. /* Read long block data for converting threshold calculation 
  856.    partitions to scale factor bands */
  857.       for(loop=0;loop<6;loop++)
  858.       {
  859. fscanf(fin,"freq=%lf sb=%dn",&freq_tp,&sbmax);
  860. sbmax++;
  861. #ifdef DEBUG
  862. printf("freq = %f sb = %dn",freq_tp,sbmax);
  863. #endif
  864. if (sfreq == freq_tp)
  865.   {
  866.      for(i=0;i<sbmax;i++)
  867.       {
  868. fscanf(fin,
  869.   "sb=%d cbw=%d bu=%d bo=%d w1=%lf w2=%lfn",
  870.   &j,&cbw_l[i],&bu_l[i],&bo_l[i],&w1_l[i],&w2_l[i]);
  871.         if (j!=i)
  872.          { printf("30:please check "psy_data"n");
  873.    exit(-1);
  874.          }
  875. #ifdef DEBUG
  876. printf(
  877.   "sb=%2d cbw=%1d bu=%2d bo=%2d w1=%5.3f w2=%5.3fn",
  878.   j,cbw_l[i],bu_l[i],bo_l[i],w1_l[i],w2_l[i]);
  879. #endif
  880.         if (i!=0)
  881.  if ( (bo_l[i] != (bu_l[i]+cbw_l[i])) ||
  882.  (fabs(1.0-w1_l[i]-w2_l[i-1]) > 0.01 ) )
  883.          { printf("31:please check "psy_data."n");
  884.    exit(-1);
  885.          }
  886.       }
  887.    }
  888.    else
  889.    {
  890.      for(j=0;j<sbmax;j++)
  891.        {
  892. fgets(tp,255,fin);
  893.         sscanf(tp,"sb=%d %sn",&i,tp);
  894.         if (j!=i)
  895.          { printf("please check "psy_data."n");
  896.    exit(-1);
  897.          }
  898.        }
  899.    }
  900.        }
  901. /* Read short block data for converting threshold calculation 
  902.    partitions to scale factor bands */
  903.       for(loop=0;loop<6;loop++)
  904.       {
  905. fscanf(fin,"freq=%lf sb=%dn",&freq_tp,&sbmax);
  906. sbmax++;
  907. #ifdef DEBUG
  908. printf("freq = %f sb = %dn",freq_tp,sbmax);
  909. #endif
  910. if (sfreq == freq_tp)
  911.   {
  912.      for(i=0;i<sbmax;i++)
  913.       {
  914. fscanf(fin,
  915.   "sb=%d cbw=%d bu=%d bo=%d w1=%lf w2=%lfn",
  916.   &j,&cbw_s[i],&bu_s[i],&bo_s[i],&w1_s[i],&w2_s[i]);
  917.         if (j!=i)
  918.          { printf("30:please check "psy_data"n");
  919.    exit(-1);
  920.          }
  921. #ifdef DEBUG
  922. printf(
  923.   "sb=%2d cbw=%1d bu=%2d bo=%2d w1=%5.3f w2=%5.3fn",
  924.   j,cbw_s[i],bu_s[i],bo_s[i],w1_s[i],w2_s[i]);
  925. #endif
  926.         if (i!=0)
  927.  if ( (bo_s[i] != (bu_s[i]+cbw_s[i])) ||
  928.  (fabs(1.0-w1_s[i]-w2_s[i-1]) > 0.01 ) )
  929.          { printf("31:please check "psy_data."n");
  930.    exit(-1);
  931.          }
  932.       }
  933.    }
  934.    else
  935.    {
  936.      for(j=0;j<sbmax;j++)
  937.        {
  938. fgets(tp,255,fin);
  939.         sscanf(tp,"sb=%d %sn",&i,tp);
  940.         if (j!=i)
  941.          { printf("please check "psy_data."n");
  942.    exit(-1);
  943.          }
  944.        }
  945.    }
  946.        }
  947. }