ENCODE.C
上传用户:dshsh2009
上传日期:2007-01-07
资源大小:155k
文件大小:51k
源码类别:

mpeg/mp3

开发平台:

Unix_Linux

  1. /**********************************************************************
  2. Copyright (c) 1991 MPEG/audio software simulation group, All Rights Reserved
  3. encode.c
  4. **********************************************************************/
  5. /**********************************************************************
  6.  * MPEG/audio coding/decoding software, work in progress              *
  7.  *   NOT for public distribution until verified and approved by the   *
  8.  *   MPEG/audio committee.  For further information, please contact   *
  9.  *   Davis Pan, 508-493-2241, e-mail: pan@3d.enet.dec.com             *
  10.  *                                                                    *
  11.  * VERSION 3.9t                                                       *
  12.  *   changes made since last update:                                  *
  13.  *   date   programmers         comment                               *
  14.  * 3/01/91  Douglas Wong,       start of version 1.1 records          *
  15.  *          Davis Pan                                                 *
  16.  * 3/06/91  Douglas Wong        rename: setup.h to endef.h            *
  17.  *                                      efilter to enfilter           *
  18.  *                                      ewindow to enwindow           *
  19.  *                              integrated "quantizer", "scalefactor",*
  20.  *                              and "transmission" files              *
  21.  *                              update routine "window_subband"       *
  22.  * 3/31/91  Bill Aspromonte     replaced read_filter by               *
  23.  *                              create_an_filter                      *
  24.  * 5/10/91  W. Joseph Carter    Ported to Macintosh and Unix.         *
  25.  *                              Incorporated Jean-Georges Fritsch's   *
  26.  *                              "bitstream.c" package.                *
  27.  *                              Incorporated Bill Aspromonte's        *
  28.  *                              filterbank coefficient matrix         *
  29.  *                              calculation routines and added        *
  30.  *                              roundoff to coincide with specs.      *
  31.  *                              Modified to strictly adhere to        *
  32.  *                              encoded bitstream specs, including    *
  33.  *                              "Berlin changes".                     *
  34.  *                              Modified PCM sound file handling to   *
  35.  *                              process all incoming samples and fill *
  36.  *                              out last encoded frame with zeros     *
  37.  *                              (silence) if needed.                  *
  38.  *                              Located and fixed numerous software   *
  39.  *                              bugs and table data errors.           *
  40.  * 19jun91  dpwe (Aware)        moved "alloc_*" reader to common.c    *
  41.  *                              Globals sblimit, alloc replaced by new*
  42.  *                              struct 'frame_params' passed as arg.  *
  43.  *                              Added JOINT STEREO coding, layers I,II*
  44.  *                              Affects: *_bit_allocation,            *
  45.  *                              subband_quantization, encode_bit_alloc*
  46.  *                              sample_encoding                       *
  47.  * 6/10/91  Earle Jennings      modified II_subband_quantization to   *
  48.  *                              resolve type cast problem for MS_DOS  *
  49.  * 6/11/91  Earle Jennings      modified to avoid overflow on MS_DOS  *
  50.  *                              in routine filter_subband             *
  51.  * 7/10/91  Earle Jennings      port to MsDos from MacIntosh version  *
  52.  * 8/ 8/91  Jens Spille         Change for MS-C6.00                   *
  53.  *10/ 1/91  S.I. Sudharsanan,   Ported to IBM AIX platform.           *
  54.  *          Don H. Lee,                                               *
  55.  *          Peter W. Farrett                                          *
  56.  *10/ 3/91  Don H. Lee          implemented CRC-16 error protection   *
  57.  *                              newly introduced function encode_CRC  *
  58.  *11/ 8/91  Kathy Wang          Documentation of code                 *
  59.  *                              All variablenames are referred to     *
  60.  *                              with surrounding pound (#) signs      *
  61.  * 2/11/92  W. Joseph Carter    Ported new code to Macintosh.  Most   *
  62.  *                              important fixes involved changing     *
  63.  *                              16-bit ints to long or unsigned in    *
  64.  *                              bit alloc routines for quant of 65535 *
  65.  *                              and passing proper function args.     *
  66.  *                              Removed "Other Joint Stereo" option   *
  67.  *                              and made bitrate be total channel     *
  68.  *                              bitrate, irrespective of the mode.    *
  69.  *                              Fixed many small bugs & reorganized.  *
  70.  * 6/16/92  Shaun Astarabadi    Changed I_scale_factor_calc() and     *
  71.  *                              II_scale_factor_calc() to use scale   *
  72.  *                              factor 0 thru 62 only and not to      *
  73.  *                              encode index 63 into the bit stream.  *
  74.  * 7/27/92  Mike Li             (re-)Port to MS-DOS                   *
  75.  * 9/22/92  jddevine@aware.com  Fixed _scale_factor_calc() defs       *
  76.  * 3/31/93  Giogio Dimino       changed II_a_bit_allocation() from:   *
  77.  *                              if( ad > ...) to if(ad >= ...)        *
  78.  * 8/05/93  TEST                changed I_a_bit_allocation() from:    *
  79.  *                              if( ad > ...) to if(ad >= ...)        *
  80.  **********************************************************************/
  81.  
  82. #include "common.h"
  83. #include "encoder.h"
  84. #ifdef MS_DOS
  85. extern unsigned _stklen = 16384;
  86. #endif
  87. /*=======================================================================
  88. |                                                                       |
  89. | This segment contains all the core routines of the encoder,           |
  90. | except for the psychoacoustic models.                                 |
  91. |                                                                       |
  92. | The user can select either one of the two psychoacoustic              |
  93. | models. Model I is a simple tonal and noise masking threshold         |
  94. | generator, and Model II is a more sophisticated cochlear masking      |
  95. | threshold generator. Model I is recommended for lower complexity      |
  96. | applications whereas Model II gives better subjective quality at low  |
  97. | bit rates.                                                            |
  98. |                                                                       |
  99. | Layers I and II of mono, stereo, and joint stereo modes are supported.|
  100. | Routines associated with a given layer are prefixed by "I_" for layer |
  101. | 1 and "II_" for layer 2.                                              |
  102. =======================================================================*/
  103.  
  104. /************************************************************************/
  105. /*
  106. /* read_samples()
  107. /*
  108. /* PURPOSE:  reads the PCM samples from a file to the buffer
  109. /*
  110. /*  SEMANTICS:
  111. /* Reads #samples_read# number of shorts from #musicin# filepointer
  112. /* into #sample_buffer[]#.  Returns the number of samples read.
  113. /*
  114. /************************************************************************/
  115. unsigned long read_samples(musicin, sample_buffer, num_samples, frame_size)
  116. FILE *musicin;
  117. short sample_buffer[2304];
  118. unsigned long num_samples, frame_size;
  119. {
  120.     unsigned long samples_read;
  121.     static unsigned long samples_to_read;
  122.     static char init = TRUE;
  123.     if (init) {
  124.         samples_to_read = num_samples;
  125.         init = FALSE;
  126.     }
  127.     if (samples_to_read >= frame_size)
  128.         samples_read = frame_size;
  129.     else
  130.         samples_read = samples_to_read;
  131.     if ((samples_read =
  132.          fread(sample_buffer, sizeof(short), (int)samples_read, musicin)) == 0)
  133.         printf("Hit end of audio datan");
  134.     samples_to_read -= samples_read;
  135.     if (samples_read < frame_size && samples_read > 0) {
  136.         printf("Insufficient PCM input for one frame - fillout with zerosn");
  137.         for (; samples_read < frame_size; sample_buffer[samples_read++] = 0);
  138.         samples_to_read = 0;
  139.     }
  140.     return(samples_read);
  141. }
  142. /************************************************************************/
  143. /*
  144. /* get_audio()
  145. /*
  146. /* PURPOSE:  reads a frame of audio data from a file to the buffer,
  147. /*   aligns the data for future processing, and separates the
  148. /*   left and right channels
  149. /*
  150. /*  SEMANTICS:
  151. /* Calls read_samples() to read a frame of audio data from filepointer
  152. /* #musicin# to #insampl[]#.  The data is shifted to make sure the data
  153. /* is centered for the 1024pt window to be used by the psychoacoustic model,
  154. /* and to compensate for the 256 sample delay from the filter bank. For
  155. /* stereo, the channels are also demultiplexed into #buffer[0][]# and
  156. /* #buffer[1][]#
  157. /*
  158. /************************************************************************/
  159.  
  160. unsigned long get_audio(musicin, buffer, num_samples, stereo, lay)
  161. FILE *musicin;
  162. short FAR buffer[2][1152];
  163. unsigned long num_samples;
  164. int stereo, lay;
  165. {
  166.    int j;
  167.    short insamp[2304];
  168.    unsigned long samples_read;
  169.  
  170.    if (lay == 1){
  171.       if(stereo == 2){ /* layer 1, stereo */
  172.          samples_read = read_samples(musicin, insamp, num_samples,
  173.                                      (unsigned long) 768);
  174.          for(j=0;j<448;j++) {
  175.             if(j<64) {
  176.                buffer[0][j] = buffer[0][j+384];
  177.                buffer[1][j] = buffer[1][j+384];
  178.             }
  179.             else {
  180.                buffer[0][j] = insamp[2*j-128];
  181.                buffer[1][j] = insamp[2*j-127];
  182.             }
  183.          }
  184.       }
  185.       else { /* layer 1, mono */
  186.          samples_read = read_samples(musicin, insamp, num_samples,
  187.                                      (unsigned long) 384);
  188.          for(j=0;j<448;j++){
  189.             if(j<64) {
  190.                buffer[0][j] = buffer[0][j+384];
  191.                buffer[1][j] = 0;
  192.             }
  193.             else {
  194.                buffer[0][j] = insamp[j-64];
  195.                buffer[1][j] = 0;
  196.             }
  197.          }
  198.       }
  199.    }
  200.    else {
  201.       if(stereo == 2){ /* layer 2 (or 3), stereo */
  202.          samples_read = read_samples(musicin, insamp, num_samples,
  203.                                      (unsigned long) 2304);
  204.          for(j=0;j<1152;j++) {
  205.             buffer[0][j] = insamp[2*j];
  206.             buffer[1][j] = insamp[2*j+1];
  207.          }
  208.       }
  209.       else { /* layer 2 (or 3), mono */
  210.          samples_read = read_samples(musicin, insamp, num_samples,
  211.                                      (unsigned long) 1152);
  212.          for(j=0;j<1152;j++){
  213.             buffer[0][j] = insamp[j];
  214.             buffer[1][j] = 0;
  215.          }
  216.       }
  217.    }
  218.    return(samples_read);
  219. }
  220.  
  221. /************************************************************************/
  222. /*
  223. /* read_ana_window()
  224. /*
  225. /* PURPOSE:  Reads encoder window file "enwindow" into array #ana_win#
  226. /*
  227. /************************************************************************/
  228.  
  229. void read_ana_window(ana_win)
  230. double FAR ana_win[HAN_SIZE];
  231. {
  232.     int i,j[4];
  233.     FILE *fp;
  234.     double f[4];
  235.     char t[150];
  236.  
  237.     if (!(fp = OpenTableFile("enwindow") ) ) {
  238.        printf("Please check analysis window table 'enwindow'n");
  239.        exit(1);
  240.     }
  241.     for (i=0;i<512;i+=4) {
  242.        fgets(t, 150, fp);
  243.        sscanf(t,"C[%d] = %lf C[%d] = %lf C[%d] = %lf C[%d] = %lfn",
  244.               j, f,j+1,f+1,j+2,f+2,j+3,f+3);
  245.        if (i==j[0]) {
  246.           ana_win[i] = f[0];
  247.           ana_win[i+1] = f[1];
  248.           ana_win[i+2] = f[2];
  249.           ana_win[i+3] = f[3];
  250.        }
  251.        else {
  252.           printf("Check index in analysis window tablen");
  253.           exit(1);
  254.        }
  255.        fgets(t,150,fp);
  256.     }
  257.     fclose(fp);
  258. }
  259. /************************************************************************/
  260. /*
  261. /* window_subband()
  262. /*
  263. /* PURPOSE:  Overlapping window on PCM samples
  264. /*
  265. /* SEMANTICS:
  266. /* 32 16-bit pcm samples are scaled to fractional 2's complement and
  267. /* concatenated to the end of the window buffer #x#. The updated window
  268. /* buffer #x# is then windowed by the analysis window #c# to produce the
  269. /* windowed sample #z#
  270. /*
  271. /************************************************************************/
  272.  
  273. void window_subband(buffer, z, k)
  274. short FAR **buffer;
  275. double FAR z[HAN_SIZE];
  276. int k;
  277. {
  278.     typedef double FAR XX[2][HAN_SIZE];
  279.     static XX FAR *x;
  280.     int i, j;
  281.     static off[2] = {0,0};
  282.     static char init = 0;
  283.     static double FAR *c;
  284.     if (!init) {
  285.         c = (double FAR *) mem_alloc(sizeof(double) * HAN_SIZE, "window");
  286.         read_ana_window(c);
  287.         x = (XX FAR *) mem_alloc(sizeof(XX),"x");
  288.         for (i=0;i<2;i++)
  289.             for (j=0;j<HAN_SIZE;j++)
  290.                 (*x)[i][j] = 0;
  291.         init = 1;
  292.     }
  293.     /* replace 32 oldest samples with 32 new samples */
  294.     for (i=0;i<32;i++) (*x)[k][31-i+off[k]] = (double) *(*buffer)++/SCALE;
  295.     /* shift samples into proper window positions */
  296.     for (i=0;i<HAN_SIZE;i++) z[i] = (*x)[k][(i+off[k])&HAN_SIZE-1] * c[i];
  297.     off[k] += 480;              /*offset is modulo (HAN_SIZE-1)*/
  298.     off[k] &= HAN_SIZE-1;
  299. }
  300.  
  301. /************************************************************************/
  302. /*
  303. /* create_ana_filter()
  304. /*
  305. /* PURPOSE:  Calculates the analysis filter bank coefficients
  306. /*
  307. /* SEMANTICS:
  308. /* Calculates the analysis filterbank coefficients and rounds to the
  309. /* 9th decimal place accuracy of the filterbank tables in the ISO
  310. /* document.  The coefficients are stored in #filter#
  311. /*
  312. /************************************************************************/
  313.  
  314. void create_ana_filter(filter)
  315. double FAR filter[SBLIMIT][64];
  316. {
  317.    register int i,k;
  318.  
  319.    for (i=0; i<32; i++)
  320.       for (k=0; k<64; k++) {
  321.           if ((filter[i][k] = 1e9*cos((double)((2*i+1)*(16-k)*PI64))) >= 0)
  322.              modf(filter[i][k]+0.5, &filter[i][k]);
  323.           else
  324.              modf(filter[i][k]-0.5, &filter[i][k]);
  325.           filter[i][k] *= 1e-9;
  326.    }
  327. }
  328. /************************************************************************/
  329. /*
  330. /* filter_subband()
  331. /*
  332. /* PURPOSE:  Calculates the analysis filter bank coefficients
  333. /*
  334. /* SEMANTICS:
  335. /*      The windowed samples #z# is filtered by the digital filter matrix #m#
  336. /* to produce the subband samples #s#. This done by first selectively
  337. /* picking out values from the windowed samples, and then multiplying
  338. /* them by the filter matrix, producing 32 subband samples.
  339. /*
  340. /************************************************************************/
  341.  
  342. void filter_subband(z,s)
  343. double FAR z[HAN_SIZE], s[SBLIMIT];
  344. {
  345.    double y[64];
  346.    int i,j;
  347. static char init = 0;
  348.    typedef double MM[SBLIMIT][64];
  349. static MM FAR *m;
  350. #ifdef MS_DOS
  351.    long    SIZE_OF_MM;
  352.    SIZE_OF_MM      = SBLIMIT*64;
  353.    SIZE_OF_MM      *= 8;
  354.    if (!init) {
  355.        m = (MM FAR *) mem_alloc(SIZE_OF_MM, "filter");
  356.        create_ana_filter(*m);
  357.        init = 1;
  358.    }
  359. #else
  360.    if (!init) {
  361.        m = (MM FAR *) mem_alloc(sizeof(MM), "filter");
  362.        create_ana_filter(*m);
  363.        init = 1;
  364.    }
  365. #endif
  366.    for (i=0;i<64;i++) for (j=0, y[i] = 0;j<8;j++) y[i] += z[i+64*j];
  367.    for (i=0;i<SBLIMIT;i++)
  368.        for (j=0, s[i]= 0;j<64;j++) s[i] += (*m)[i][j] * y[j];
  369. }
  370. /************************************************************************/
  371. /*
  372. /* encode_info()
  373. /*
  374. /* PURPOSE:  Puts the syncword and header information on the output
  375. /* bitstream.
  376. /*
  377. /************************************************************************/
  378.  
  379. void encode_info(fr_ps,bs)
  380. frame_params *fr_ps;
  381. Bit_stream_struc *bs;
  382. {
  383.         layer *info = fr_ps->header;
  384.  
  385.         putbits(bs,0xfff,12);                    /* syncword 12 bits */
  386.         put1bit(bs,info->version);               /* ID        1 bit  */
  387.         putbits(bs,4-info->lay,2);               /* layer     2 bits */
  388.         put1bit(bs,!info->error_protection);     /* bit set => no err prot */
  389.         putbits(bs,info->bitrate_index,4);
  390.         putbits(bs,info->sampling_frequency,2);
  391.         put1bit(bs,info->padding);
  392.         put1bit(bs,info->extension);             /* private_bit */
  393.         putbits(bs,info->mode,2);
  394.         putbits(bs,info->mode_ext,2);
  395.         put1bit(bs,info->copyright);
  396.         put1bit(bs,info->original);
  397.         putbits(bs,info->emphasis,2);
  398. }
  399.  
  400. /************************************************************************/
  401. /*
  402. /* mod()
  403. /*
  404. /* PURPOSE:  Returns the absolute value of its argument
  405. /*
  406. /************************************************************************/
  407.  
  408. double mod(a)
  409. double a;
  410. {
  411.     return (a > 0) ? a : -a;
  412. }
  413.  
  414. /************************************************************************/
  415. /*
  416. /* I_combine_LR    (Layer I)
  417. /* II_combine_LR   (Layer II)
  418. /*
  419. /* PURPOSE:Combines left and right channels into a mono channel
  420. /*
  421. /* SEMANTICS:  The average of left and right subband samples is put into
  422. /* #joint_sample#
  423. /*
  424. /* Layer I and II differ in frame length and # subbands used
  425. /*
  426. /************************************************************************/
  427.  
  428. void I_combine_LR(sb_sample, joint_sample)
  429. double FAR sb_sample[2][3][SCALE_BLOCK][SBLIMIT];
  430. double FAR joint_sample[3][SCALE_BLOCK][SBLIMIT];
  431. {   /* make a filtered mono for joint stereo */
  432.     int sb, smp;
  433.  
  434.    for(sb = 0; sb<SBLIMIT; ++sb)
  435.       for(smp = 0; smp<SCALE_BLOCK; ++smp)
  436.         joint_sample[0][smp][sb] = .5 *
  437.                     (sb_sample[0][0][smp][sb] + sb_sample[1][0][smp][sb]);
  438. }
  439.  
  440. void II_combine_LR(sb_sample, joint_sample, sblimit)
  441. double FAR sb_sample[2][3][SCALE_BLOCK][SBLIMIT];
  442. double FAR joint_sample[3][SCALE_BLOCK][SBLIMIT];
  443. int sblimit;
  444. {  /* make a filtered mono for joint stereo */
  445.    int sb, smp, sufr;
  446.  
  447.    for(sb = 0; sb<sblimit; ++sb)
  448.       for(smp = 0; smp<SCALE_BLOCK; ++smp)
  449.          for(sufr = 0; sufr<3; ++sufr)
  450.             joint_sample[sufr][smp][sb] = .5 * (sb_sample[0][sufr][smp][sb]
  451.                                            + sb_sample[1][sufr][smp][sb]);
  452. }
  453.  
  454. /************************************************************************
  455. /*
  456. /* I_scale_factor_calc     (Layer I)
  457. /* II_scale_factor_calc    (Layer II)
  458. /*
  459. /* PURPOSE:For each subband, calculate the scale factor for each set
  460. /* of the 12 subband samples
  461. /*
  462. /* SEMANTICS:  Pick the scalefactor #multiple[]# just larger than the
  463. /* absolute value of the peak subband sample of 12 samples,
  464. /* and store the corresponding scalefactor index in #scalar#.
  465. /*
  466. /* Layer II has three sets of 12-subband samples for a given
  467. /* subband.
  468. /*
  469. /************************************************************************/
  470.  
  471. void I_scale_factor_calc(sb_sample,scalar,stereo)
  472. double FAR sb_sample[][3][SCALE_BLOCK][SBLIMIT];
  473. unsigned int scalar[][3][SBLIMIT];
  474. int stereo;
  475. {
  476.    int i,j, k;
  477.    double s[SBLIMIT];
  478.  
  479.    for (k=0;k<stereo;k++) {
  480.      for (i=0;i<SBLIMIT;i++)
  481.        for (j=1, s[i] = mod(sb_sample[k][0][0][i]);j<SCALE_BLOCK;j++)
  482.          if (mod(sb_sample[k][0][j][i]) > s[i])
  483.             s[i] = mod(sb_sample[k][0][j][i]);
  484.  
  485.      for (i=0;i<SBLIMIT;i++)
  486.        for (j=SCALE_RANGE-2,scalar[k][0][i]=0;j>=0;j--) /* $A 6/16/92 */
  487.          if (s[i] <= multiple[j]) {
  488.             scalar[k][0][i] = j;
  489.             break;
  490.          }
  491.    }
  492. }
  493. /******************************** Layer II ******************************/
  494.  
  495. void II_scale_factor_calc(sb_sample,scalar,stereo,sblimit)
  496. double FAR sb_sample[][3][SCALE_BLOCK][SBLIMIT];
  497. unsigned int scalar[][3][SBLIMIT];
  498. int stereo,sblimit;
  499. {
  500.   int i,j, k,t;
  501.   double s[SBLIMIT];
  502.  
  503.   for (k=0;k<stereo;k++) for (t=0;t<3;t++) {
  504.     for (i=0;i<sblimit;i++)
  505.       for (j=1, s[i] = mod(sb_sample[k][t][0][i]);j<SCALE_BLOCK;j++)
  506.         if (mod(sb_sample[k][t][j][i]) > s[i])
  507.              s[i] = mod(sb_sample[k][t][j][i]);
  508.  
  509.   for (i=0;i<sblimit;i++)
  510.     for (j=SCALE_RANGE-2,scalar[k][t][i]=0;j>=0;j--)    /* $A 6/16/92 */
  511.       if (s[i] <= multiple[j]) {
  512.          scalar[k][t][i] = j;
  513.          break;
  514.       }
  515.       for (i=sblimit;i<SBLIMIT;i++) scalar[k][t][i] = SCALE_RANGE-1;
  516.     }
  517. }
  518. /************************************************************************
  519. /*
  520. /* pick_scale  (Layer II)
  521. /*
  522. /* PURPOSE:For each subband, puts the smallest scalefactor of the 3
  523. /* associated with a frame into #max_sc#.  This is used
  524. /* used by Psychoacoustic Model I.
  525. /* (I would recommend changin max_sc to min_sc)
  526. /*
  527. /************************************************************************/
  528.  
  529. void pick_scale(scalar, fr_ps, max_sc)
  530. unsigned int scalar[2][3][SBLIMIT];
  531. frame_params *fr_ps;
  532. double FAR max_sc[2][SBLIMIT];
  533. {
  534.   int i,j,k,max;
  535.   int stereo  = fr_ps->stereo;
  536.   int sblimit = fr_ps->sblimit;
  537.  
  538.   for (k=0;k<stereo;k++)
  539.     for (i=0;i<sblimit;max_sc[k][i] = multiple[max],i++)
  540.       for (j=1, max = scalar[k][0][i];j<3;j++)
  541.          if (max > scalar[k][j][i]) max = scalar[k][j][i];
  542.   for (i=sblimit;i<SBLIMIT;i++) max_sc[0][i] = max_sc[1][i] = 1E-20;
  543. }
  544. /************************************************************************
  545. /*
  546. /* put_scale   (Layer I)
  547. /*
  548. /* PURPOSE:Sets #max_sc# to the scalefactor index in #scalar.
  549. /* This is used by Psychoacoustic Model I
  550. /*
  551. /************************************************************************/
  552.  
  553. void put_scale(scalar, fr_ps, max_sc)
  554. unsigned int scalar[2][3][SBLIMIT];
  555. frame_params *fr_ps;
  556. double FAR max_sc[2][SBLIMIT];
  557. {
  558.    int i,j,k, max;
  559.    int stereo  = fr_ps->stereo;
  560.    int sblimit = fr_ps->sblimit;
  561.  
  562.    for (k=0;k<stereo;k++) for (i=0;i<SBLIMIT;i++)
  563.         max_sc[k][i] = multiple[scalar[k][0][i]];
  564. }
  565.  
  566. /************************************************************************
  567. /*
  568. /* II_transmission_pattern (Layer II only)
  569. /*
  570. /* PURPOSE:For a given subband, determines whether to send 1, 2, or
  571. /* all 3 of the scalefactors, and fills in the scalefactor
  572. /* select information accordingly
  573. /*
  574. /* SEMANTICS:  The subbands and channels are classified based on how much
  575. /* the scalefactors changes over its three values (corresponding
  576. /* to the 3 sets of 12 samples per subband).  The classification
  577. /* will send 1 or 2 scalefactors instead of three if the scalefactors
  578. /* do not change much.  The scalefactor select information,
  579. /* #scfsi#, is filled in accordingly.
  580. /*
  581. /************************************************************************/
  582.  
  583. void II_transmission_pattern(scalar, scfsi, fr_ps)
  584. unsigned int scalar[2][3][SBLIMIT];
  585. unsigned int scfsi[2][SBLIMIT];
  586. frame_params *fr_ps;
  587. {
  588.    int stereo  = fr_ps->stereo;
  589.    int sblimit = fr_ps->sblimit;
  590.    int dscf[2];
  591.    int class[2],i,j,k;
  592. static int pattern[5][5] = {0x123, 0x122, 0x122, 0x133, 0x123,
  593.                             0x113, 0x111, 0x111, 0x444, 0x113,
  594.                             0x111, 0x111, 0x111, 0x333, 0x113,
  595.                             0x222, 0x222, 0x222, 0x333, 0x123,
  596.                             0x123, 0x122, 0x122, 0x133, 0x123};
  597.  
  598.    for (k=0;k<stereo;k++)
  599.      for (i=0;i<sblimit;i++) {
  600.        dscf[0] =  (scalar[k][0][i]-scalar[k][1][i]);
  601.        dscf[1] =  (scalar[k][1][i]-scalar[k][2][i]);
  602.        for (j=0;j<2;j++) {
  603.          if (dscf[j]<=-3) class[j] = 0;
  604.          else if (dscf[j] > -3 && dscf[j] <0) class[j] = 1;
  605.               else if (dscf[j] == 0) class[j] = 2;
  606.                    else if (dscf[j] > 0 && dscf[j] < 3) class[j] = 3;
  607.                         else class[j] = 4;
  608.        }
  609.        switch (pattern[class[0]][class[1]]) {
  610.          case 0x123 :    scfsi[k][i] = 0;
  611.                          break;
  612.          case 0x122 :    scfsi[k][i] = 3;
  613.                          scalar[k][2][i] = scalar[k][1][i];
  614.                          break;
  615.          case 0x133 :    scfsi[k][i] = 3;
  616.                          scalar[k][1][i] = scalar[k][2][i];
  617.                          break;
  618.          case 0x113 :    scfsi[k][i] = 1;
  619.                          scalar[k][1][i] = scalar[k][0][i];
  620.                          break;
  621.          case 0x111 :    scfsi[k][i] = 2;
  622.                          scalar[k][1][i] = scalar[k][2][i] = scalar[k][0][i];
  623.                          break;
  624.          case 0x222 :    scfsi[k][i] = 2;
  625.                          scalar[k][0][i] = scalar[k][2][i] = scalar[k][1][i];
  626.                          break;
  627.          case 0x333 :    scfsi[k][i] = 2;
  628.                          scalar[k][0][i] = scalar[k][1][i] = scalar[k][2][i];
  629.                          break;
  630.          case 0x444 :    scfsi[k][i] = 2;
  631.                          if (scalar[k][0][i] > scalar[k][2][i])
  632.                               scalar[k][0][i] = scalar[k][2][i];
  633.                          scalar[k][1][i] = scalar[k][2][i] = scalar[k][0][i];
  634.       }
  635.    }
  636. }
  637.  
  638. /************************************************************************
  639. /*
  640. /* I_encode_scale  (Layer I)
  641. /* II_encode_scale (Layer II)
  642. /*
  643. /* PURPOSE:The encoded scalar factor information is arranged and
  644. /* queued into the output fifo to be transmitted.
  645. /*
  646. /* For Layer II, the three scale factors associated with
  647. /* a given subband and channel are transmitted in accordance
  648. /* with the scfsi, which is transmitted first.
  649. /*
  650. /************************************************************************/
  651.  
  652. void I_encode_scale(scalar, bit_alloc, fr_ps, bs)
  653. unsigned int scalar[2][3][SBLIMIT];
  654. unsigned int bit_alloc[2][SBLIMIT];
  655. frame_params *fr_ps;
  656. Bit_stream_struc *bs;
  657. {
  658.    int stereo  = fr_ps->stereo;
  659.    int sblimit = fr_ps->sblimit;
  660.    int i,j;
  661.  
  662.    for (i=0;i<SBLIMIT;i++) for (j=0;j<stereo;j++)
  663.       if (bit_alloc[j][i]) putbits(bs,scalar[j][0][i],6);
  664. }
  665.  
  666. /***************************** Layer II  ********************************/
  667.  
  668. void II_encode_scale(bit_alloc, scfsi, scalar, fr_ps, bs)
  669. unsigned int bit_alloc[2][SBLIMIT], scfsi[2][SBLIMIT];
  670. unsigned int scalar[2][3][SBLIMIT];
  671. frame_params *fr_ps;
  672. Bit_stream_struc *bs;
  673. {
  674.    int stereo  = fr_ps->stereo;
  675.    int sblimit = fr_ps->sblimit;
  676.    int jsbound = fr_ps->jsbound;
  677.    int i,j,k;
  678.  
  679.    for (i=0;i<sblimit;i++) for (k=0;k<stereo;k++)
  680.      if (bit_alloc[k][i])  putbits(bs,scfsi[k][i],2);
  681.  
  682.    for (i=0;i<sblimit;i++) for (k=0;k<stereo;k++)
  683.      if (bit_alloc[k][i])  /* above jsbound, bit_alloc[0][i] == ba[1][i] */
  684.         switch (scfsi[k][i]) {
  685.            case 0: for (j=0;j<3;j++)
  686.                      putbits(bs,scalar[k][j][i],6);
  687.                    break;
  688.            case 1:
  689.            case 3: putbits(bs,scalar[k][0][i],6);
  690.                    putbits(bs,scalar[k][2][i],6);
  691.                    break;
  692.            case 2: putbits(bs,scalar[k][0][i],6);
  693.         }
  694. }
  695.  
  696. /*=======================================================================
  697. |                                                                        |
  698. |      The following routines are done after the masking threshold       |
  699. | has been calculated by the fft analysis routines in the Psychoacoustic |
  700. | model. Using the MNR calculated, the actual number of bits allocated   |
  701. | to each subband is found iteratively.                                  |
  702. |                                                                        |
  703. =======================================================================*/
  704.  
  705. /************************************************************************
  706. /*
  707. /* I_bits_for_nonoise  (Layer I)
  708. /* II_bits_for_nonoise (Layer II)
  709. /*
  710. /* PURPOSE:Returns the number of bits required to produce a
  711. /* mask-to-noise ratio better or equal to the noise/no_noise threshold.
  712. /*
  713. /* SEMANTICS:
  714. /* bbal = # bits needed for encoding bit allocation
  715. /* bsel = # bits needed for encoding scalefactor select information
  716. /* banc = # bits needed for ancillary data (header info included)
  717. /*
  718. /* For each subband and channel, will add bits until one of the
  719. /* following occurs:
  720. /* - Hit maximum number of bits we can allocate for that subband
  721. /* - MNR is better than or equal to the minimum masking level
  722. /*   (NOISY_MIN_MNR)
  723. /* Then the bits required for scalefactors, scfsi, bit allocation,
  724. /* and the subband samples are tallied (#req_bits#) and returned.
  725. /*
  726. /* (NOISY_MIN_MNR) is the smallest MNR a subband can have before it is
  727. /* counted as 'noisy' by the logic which chooses the number of JS
  728. /* subbands.
  729. /*
  730. /* Joint stereo is supported.
  731. /*
  732. /************************************************************************/
  733. static double snr[18] = {0.00, 7.00, 11.00, 16.00, 20.84,
  734.                          25.28, 31.59, 37.75, 43.84,
  735.                          49.89, 55.93, 61.96, 67.98, 74.01,
  736.                          80.03, 86.05, 92.01, 98.01};
  737. int I_bits_for_nonoise(perm_smr, fr_ps)
  738. double FAR perm_smr[2][SBLIMIT];
  739. frame_params *fr_ps;
  740. {
  741.    int i,j,k;
  742.    int stereo  = fr_ps->stereo;
  743.    int sblimit = fr_ps->sblimit;
  744.    int jsbound = fr_ps->jsbound;
  745.    int req_bits = 0;
  746.  
  747.    /* initial b_anc (header) allocation bits */
  748.    req_bits = 32 + 4 * ( (jsbound * stereo) + (SBLIMIT-jsbound) );
  749.  
  750.    for(i=0; i<SBLIMIT; ++i)
  751.      for(j=0; j<((i<jsbound)?stereo:1); ++j) {
  752.        for(k=0;k<14; ++k)
  753.          if( (-perm_smr[j][i] + snr[k]) >= NOISY_MIN_MNR)
  754.            break; /* we found enough bits */
  755.          if(stereo == 2 && i >= jsbound)     /* check other JS channel */
  756.            for(;k<14; ++k)
  757.              if( (-perm_smr[1-j][i] + snr[k]) >= NOISY_MIN_MNR) break;
  758.          if(k>0) req_bits += (k+1)*SCALE_BLOCK + 6*((i>=jsbound)?stereo:1);
  759.    }
  760.    return req_bits;
  761. }
  762.  
  763. /***************************** Layer II  ********************************/
  764.  
  765. int II_bits_for_nonoise(perm_smr, scfsi, fr_ps)
  766. double FAR perm_smr[2][SBLIMIT];
  767. unsigned int scfsi[2][SBLIMIT];
  768. frame_params *fr_ps;
  769. {
  770.    int sb,ch,ba;
  771.    int stereo  = fr_ps->stereo;
  772.    int sblimit = fr_ps->sblimit;
  773.    int jsbound = fr_ps->jsbound;
  774.    al_table *alloc = fr_ps->alloc;
  775.    int req_bits = 0, bbal = 0, berr = 0, banc = 32;
  776.    int maxAlloc, sel_bits, sc_bits, smp_bits;
  777. static int sfsPerScfsi[] = { 3,2,1,2 };    /* lookup # sfs per scfsi */
  778.    /* added 92-08-11 shn */
  779.    if (fr_ps->header->error_protection) berr=16; else berr=0; 
  780.  
  781.    for (sb=0; sb<jsbound; ++sb)
  782.      bbal += stereo * (*alloc)[sb][0].bits;
  783.    for (sb=jsbound; sb<sblimit; ++sb)
  784.      bbal += (*alloc)[sb][0].bits;
  785.    req_bits = banc + bbal + berr;
  786.  
  787.    for(sb=0; sb<sblimit; ++sb)
  788.      for(ch=0; ch<((sb<jsbound)?stereo:1); ++ch) {
  789.        maxAlloc = (1<<(*alloc)[sb][0].bits)-1;
  790.        sel_bits = sc_bits = smp_bits = 0;
  791.        for(ba=0;ba<maxAlloc-1; ++ba)
  792.          if( (-perm_smr[ch][sb] + snr[(*alloc)[sb][ba].quant+((ba>0)?1:0)])
  793.              >= NOISY_MIN_MNR)
  794.             break;      /* we found enough bits */
  795.        if(stereo == 2 && sb >= jsbound) /* check other JS channel */
  796.          for(;ba<maxAlloc-1; ++ba)
  797.            if( (-perm_smr[1-ch][sb]+ snr[(*alloc)[sb][ba].quant+((ba>0)?1:0)])
  798.                >= NOISY_MIN_MNR)
  799.              break;
  800.        if(ba>0) {
  801.          smp_bits = SCALE_BLOCK * ((*alloc)[sb][ba].group * (*alloc)[sb][ba].bits);
  802.          /* scale factor bits required for subband */
  803.          sel_bits = 2;
  804.          sc_bits  = 6 * sfsPerScfsi[scfsi[ch][sb]];
  805.          if(stereo == 2 && sb >= jsbound) {
  806.            /* each new js sb has L+R scfsis */
  807.            sel_bits += 2;
  808.            sc_bits  += 6 * sfsPerScfsi[scfsi[1-ch][sb]];
  809.          }
  810.          req_bits += smp_bits+sel_bits+sc_bits;
  811.        }
  812.    }
  813.    return req_bits;
  814. }
  815.  
  816. /************************************************************************
  817. /*
  818. /* I_main_bit_allocation   (Layer I)
  819. /* II_main_bit_allocation  (Layer II)
  820. /*
  821. /* PURPOSE:For joint stereo mode, determines which of the 4 joint
  822. /* stereo modes is needed.  Then calls *_a_bit_allocation(), which
  823. /* allocates bits for each of the subbands until there are no more bits
  824. /* left, or the MNR is at the noise/no_noise threshold.
  825. /*
  826. /* SEMANTICS:
  827. /*
  828. /* For joint stereo mode, joint stereo is changed to stereo if
  829. /* there are enough bits to encode stereo at or better than the
  830. /* no-noise threshold (NOISY_MIN_MNR).  Otherwise, the system
  831. /* iteratively allocates less bits by using joint stereo until one
  832. /* of the following occurs:
  833. /* - there are no more noisy subbands (MNR >= NOISY_MIN_MNR)
  834. /* - mode_ext has been reduced to 0, which means that all but the
  835. /*   lowest 4 subbands have been converted from stereo to joint
  836. /*   stereo, and no more subbands may be converted
  837. /*
  838. /*     This function calls *_bits_for_nonoise() and *_a_bit_allocation().
  839. /*
  840. /************************************************************************/
  841.  
  842. void I_main_bit_allocation(perm_smr, bit_alloc, adb, fr_ps)
  843. double FAR perm_smr[2][SBLIMIT];
  844. unsigned int bit_alloc[2][SBLIMIT];
  845. int *adb;
  846. frame_params *fr_ps;
  847. {
  848.    int  noisy_sbs;
  849.    int  mode, mode_ext, lay, i;
  850.    int  rq_db, av_db = *adb;
  851. static  int init = 0;
  852.  
  853.    if(init == 0) {
  854.      /* rearrange snr for layer I */
  855.      snr[2] = snr[3];
  856.      for (i=3;i<16;i++) snr[i] = snr[i+2];
  857.      init = 1;
  858.    }
  859.  
  860.    if((mode = fr_ps->actual_mode) == MPG_MD_JOINT_STEREO) {
  861.      fr_ps->header->mode = MPG_MD_STEREO;
  862.      fr_ps->header->mode_ext = 0;
  863.      fr_ps->jsbound = fr_ps->sblimit;
  864.      if(rq_db = I_bits_for_nonoise(perm_smr, fr_ps) > *adb) {
  865.        fr_ps->header->mode = MPG_MD_JOINT_STEREO;
  866.        mode_ext = 4;           /* 3 is least severe reduction */
  867.        lay = fr_ps->header->lay;
  868.        do {
  869.           --mode_ext;
  870.           fr_ps->jsbound = js_bound(lay, mode_ext);
  871.           rq_db = I_bits_for_nonoise(perm_smr, fr_ps);
  872.        } while( (rq_db > *adb) && (mode_ext > 0));
  873.        fr_ps->header->mode_ext = mode_ext;
  874.      }    /* well we either eliminated noisy sbs or mode_ext == 0 */
  875.    }
  876.    noisy_sbs = I_a_bit_allocation(perm_smr, bit_alloc, adb, fr_ps);
  877. }
  878.  
  879. /***************************** Layer II  ********************************/
  880.  
  881. void II_main_bit_allocation(perm_smr, scfsi, bit_alloc, adb, fr_ps)
  882. double FAR perm_smr[2][SBLIMIT];
  883. unsigned int scfsi[2][SBLIMIT];
  884. unsigned int bit_alloc[2][SBLIMIT];
  885. int *adb;
  886. frame_params *fr_ps;
  887. {
  888.    int  noisy_sbs, nn;
  889.    int  mode, mode_ext, lay;
  890.    int  rq_db, av_db = *adb;
  891.  
  892.    if((mode = fr_ps->actual_mode) == MPG_MD_JOINT_STEREO) {
  893.      fr_ps->header->mode = MPG_MD_STEREO;
  894.      fr_ps->header->mode_ext = 0;
  895.      fr_ps->jsbound = fr_ps->sblimit;
  896.      if((rq_db=II_bits_for_nonoise(perm_smr, scfsi, fr_ps)) > *adb) {
  897.        fr_ps->header->mode = MPG_MD_JOINT_STEREO;
  898.        mode_ext = 4;           /* 3 is least severe reduction */
  899.        lay = fr_ps->header->lay;
  900.        do {
  901.          --mode_ext;
  902.          fr_ps->jsbound = js_bound(lay, mode_ext);
  903.          rq_db = II_bits_for_nonoise(perm_smr, scfsi, fr_ps);
  904.        } while( (rq_db > *adb) && (mode_ext > 0));
  905.        fr_ps->header->mode_ext = mode_ext;
  906.      }    /* well we either eliminated noisy sbs or mode_ext == 0 */
  907.    }
  908.    noisy_sbs = II_a_bit_allocation(perm_smr, scfsi, bit_alloc, adb, fr_ps);
  909. }
  910.  
  911. /************************************************************************
  912. /*
  913. /* I_a_bit_allocation  (Layer I)
  914. /* II_a_bit_allocation (Layer II)
  915. /*
  916. /* PURPOSE:Adds bits to the subbands with the lowest mask-to-noise
  917. /* ratios, until the maximum number of bits for the subband has
  918. /* been allocated.
  919. /*
  920. /* SEMANTICS:
  921. /* 1. Find the subband and channel with the smallest MNR (#min_sb#,
  922. /*    and #min_ch#)
  923. /* 2. Calculate the increase in bits needed if we increase the bit
  924. /*    allocation to the next higher level
  925. /* 3. If there are enough bits available for increasing the resolution
  926. /*    in #min_sb#, #min_ch#, and the subband has not yet reached its
  927. /*    maximum allocation, update the bit allocation, MNR, and bits
  928. /*    available accordingly
  929. /* 4. Repeat until there are no more bits left, or no more available
  930. /*    subbands. (A subband is still available until the maximum
  931. /*    number of bits for the subband has been allocated, or there
  932. /*    aren't enough bits to go to the next higher resolution in the
  933. /*    subband.)
  934. /*
  935. /************************************************************************/
  936.  
  937. int I_a_bit_allocation(perm_smr, bit_alloc, adb, fr_ps) /* return noisy sbs */
  938. double FAR perm_smr[2][SBLIMIT];
  939. unsigned int bit_alloc[2][SBLIMIT];
  940. int *adb;
  941. frame_params *fr_ps;
  942. {
  943.    int i, k, smpl_bits, scale_bits, min_sb, min_ch, oth_ch;
  944.    int bspl, bscf, ad, noisy_sbs, done = 0, bbal ;
  945.    double mnr[2][SBLIMIT], small;
  946.    char used[2][SBLIMIT];
  947.    int stereo  = fr_ps->stereo;
  948.    int sblimit = fr_ps->sblimit;
  949.    int jsbound = fr_ps->jsbound;
  950.    al_table *alloc = fr_ps->alloc;
  951. static char init= 0;
  952. static int banc=32, berr=0;
  953.  
  954.    if (!init) {
  955.       init = 1;
  956.       if (fr_ps->header->error_protection) berr = 16;  /* added 92-08-11 shn */
  957.    }
  958.    bbal = 4 * ( (jsbound * stereo) + (SBLIMIT-jsbound) );
  959.    *adb -= bbal + berr + banc;
  960.    ad= *adb;
  961.  
  962.    for (i=0;i<SBLIMIT;i++) for (k=0;k<stereo;k++) {
  963.      mnr[k][i]=snr[0]-perm_smr[k][i];
  964.      bit_alloc[k][i] = 0;
  965.      used[k][i] = 0;
  966.    }
  967.    bspl = bscf = 0;
  968.  
  969.    do  {
  970.      /* locate the subband with minimum SMR */
  971.      small = mnr[0][0]+1;    min_sb = -1; min_ch = -1;
  972.      for (i=0;i<SBLIMIT;i++) for (k=0;k<stereo;k++)
  973.        /* go on only if there are bits left */
  974.        if (used[k][i] != 2 && small > mnr[k][i]) {
  975.          small = mnr[k][i];
  976.          min_sb = i;  min_ch = k;
  977.        }
  978.      if(min_sb > -1) {   /* there was something to find */
  979.        /* first step of bit allocation is biggest */
  980.        if (used[min_ch][min_sb])  { smpl_bits = SCALE_BLOCK; scale_bits = 0; }
  981.        else                       { smpl_bits = 24; scale_bits = 6; }
  982.        if(min_sb >= jsbound)        scale_bits *= stereo;
  983.  
  984.        /* check to see enough bits were available for */
  985.        /* increasing resolution in the minimum band */
  986.  
  987.        if (ad >= bspl + bscf + scale_bits + smpl_bits) {
  988.          bspl += smpl_bits; /* bit for subband sample */
  989.          bscf += scale_bits; /* bit for scale factor */
  990.          bit_alloc[min_ch][min_sb]++;
  991.          used[min_ch][min_sb] = 1; /* subband has bits */
  992.          mnr[min_ch][min_sb] = -perm_smr[min_ch][min_sb]
  993.                                + snr[bit_alloc[min_ch][min_sb]];
  994.          /* Check if subband has been fully allocated max bits */
  995.          if (bit_alloc[min_ch][min_sb] ==  14 ) used[min_ch][min_sb] = 2;
  996.        }
  997.        else            /* no room to improve this band */
  998.          used[min_ch][min_sb] = 2; /*   for allocation anymore */
  999.        if(stereo == 2 && min_sb >= jsbound) {
  1000.          oth_ch = 1-min_ch;  /* joint-st : fix other ch */
  1001.          bit_alloc[oth_ch][min_sb] = bit_alloc[min_ch][min_sb];
  1002.          used[oth_ch][min_sb] = used[min_ch][min_sb];
  1003.          mnr[oth_ch][min_sb] = -perm_smr[oth_ch][min_sb]
  1004.                                + snr[bit_alloc[oth_ch][min_sb]];
  1005.        }
  1006.      }
  1007.    } while(min_sb>-1);     /* i.e. still some sub-bands to find */
  1008.    /* Calculate the number of bits left, add on to pointed var */
  1009.    ad -= bspl+bscf;
  1010.    *adb = ad;
  1011.    /* see how many channels are noisy */
  1012.    noisy_sbs = 0; small = mnr[0][0];
  1013.    for(k=0; k<stereo; ++k) {
  1014.      for(i = 0; i< SBLIMIT; ++i) {
  1015.        if(mnr[k][i] < NOISY_MIN_MNR)   ++noisy_sbs;
  1016.        if(small > mnr[k][i])           small = mnr[k][i];
  1017.      }
  1018.    }
  1019.    return noisy_sbs;
  1020. }
  1021. /***************************** Layer II  ********************************/
  1022.  
  1023. int II_a_bit_allocation(perm_smr, scfsi, bit_alloc, adb, fr_ps)
  1024. double FAR perm_smr[2][SBLIMIT];
  1025. unsigned int scfsi[2][SBLIMIT];
  1026. unsigned int bit_alloc[2][SBLIMIT];
  1027. int *adb;
  1028. frame_params *fr_ps;
  1029. {
  1030.    int i, min_ch, min_sb, oth_ch, k, increment, scale, seli, ba;
  1031.    int bspl, bscf, bsel, ad, noisy_sbs, bbal=0;
  1032.    double mnr[2][SBLIMIT], small;
  1033.    char used[2][SBLIMIT];
  1034.    int stereo  = fr_ps->stereo;
  1035.    int sblimit = fr_ps->sblimit;
  1036.    int jsbound = fr_ps->jsbound;
  1037.    al_table *alloc = fr_ps->alloc;
  1038. static char init= 0;
  1039. static int banc=32, berr=0;
  1040. static int sfsPerScfsi[] = { 3,2,1,2 };    /* lookup # sfs per scfsi */
  1041.  
  1042.    if (!init) { 
  1043.        init = 1;  
  1044.        if (fr_ps->header->error_protection) berr=16; /* added 92-08-11 shn */
  1045.    }
  1046.    for (i=0; i<jsbound; ++i)
  1047.      bbal += stereo * (*alloc)[i][0].bits;
  1048.    for (i=jsbound; i<sblimit; ++i)
  1049.      bbal += (*alloc)[i][0].bits;
  1050.    *adb -= bbal + berr + banc;
  1051.    ad = *adb;
  1052.  
  1053.    for (i=0;i<sblimit;i++) for (k=0;k<stereo;k++) {
  1054.      mnr[k][i]=snr[0]-perm_smr[k][i];
  1055.      bit_alloc[k][i] = 0;
  1056.      used[k][i] = 0;
  1057.    }
  1058.    bspl = bscf = bsel = 0;
  1059.  
  1060.    do  {
  1061.      /* locate the subband with minimum SMR */
  1062.      small = 999999.0; min_sb = -1; min_ch = -1;
  1063.      for (i=0;i<sblimit;i++) for(k=0;k<stereo;++k)
  1064.        if (used[k][i]  != 2 && small > mnr[k][i]) {
  1065.          small = mnr[k][i];
  1066.          min_sb = i;  min_ch = k;
  1067.      }
  1068.      if(min_sb > -1) {   /* there was something to find */
  1069.        /* find increase in bit allocation in subband [min] */
  1070.        increment = SCALE_BLOCK * ((*alloc)[min_sb][bit_alloc[min_ch][min_sb]+1].group *
  1071.                         (*alloc)[min_sb][bit_alloc[min_ch][min_sb]+1].bits);
  1072.        if (used[min_ch][min_sb])
  1073.          increment -= SCALE_BLOCK * ((*alloc)[min_sb][bit_alloc[min_ch][min_sb]].group*
  1074.                            (*alloc)[min_sb][bit_alloc[min_ch][min_sb]].bits);
  1075.  
  1076.        /* scale factor bits required for subband [min] */
  1077.        oth_ch = 1 - min_ch;    /* above js bound, need both chans */
  1078.        if (used[min_ch][min_sb]) scale = seli = 0;
  1079.        else {          /* this channel had no bits or scfs before */
  1080.          seli = 2;
  1081.          scale = 6 * sfsPerScfsi[scfsi[min_ch][min_sb]];
  1082.          if(stereo == 2 && min_sb >= jsbound) {
  1083.            /* each new js sb has L+R scfsis */
  1084.            seli += 2;
  1085.            scale += 6 * sfsPerScfsi[scfsi[oth_ch][min_sb]];
  1086.          }
  1087.        }
  1088.        /* check to see enough bits were available for */
  1089.        /* increasing resolution in the minimum band */
  1090.        if (ad >= bspl + bscf + bsel + seli + scale + increment) {
  1091.          ba = ++bit_alloc[min_ch][min_sb]; /* next up alloc */
  1092.          bspl += increment;  /* bits for subband sample */
  1093.          bscf += scale;      /* bits for scale factor */
  1094.          bsel += seli;       /* bits for scfsi code */
  1095.          used[min_ch][min_sb] = 1; /* subband has bits */
  1096.          mnr[min_ch][min_sb] = -perm_smr[min_ch][min_sb] +
  1097.                                snr[(*alloc)[min_sb][ba].quant+1];
  1098.          /* Check if subband has been fully allocated max bits */
  1099.          if (ba >= (1<<(*alloc)[min_sb][0].bits)-1) used[min_ch][min_sb] = 2;
  1100.        }
  1101.        else used[min_ch][min_sb] = 2; /* can't increase this alloc */
  1102.        if(min_sb >= jsbound && stereo == 2) {
  1103.          /* above jsbound, alloc applies L+R */
  1104.          ba = bit_alloc[oth_ch][min_sb] = bit_alloc[min_ch][min_sb];
  1105.          used[oth_ch][min_sb] = used[min_ch][min_sb];
  1106.          mnr[oth_ch][min_sb] = -perm_smr[oth_ch][min_sb] +
  1107.                                snr[(*alloc)[min_sb][ba].quant+1];
  1108.        }
  1109.      }
  1110.    } while(min_sb > -1);   /* until could find no channel */
  1111.    /* Calculate the number of bits left */
  1112.    ad -= bspl+bscf+bsel;   *adb = ad;
  1113.    for (i=sblimit;i<SBLIMIT;i++) for (k=0;k<stereo;k++) bit_alloc[k][i]=0;
  1114.  
  1115.    noisy_sbs = 0;  small = mnr[0][0];      /* calc worst noise in case */
  1116.    for(k=0;k<stereo;++k) {
  1117.      for (i=0;i<sblimit;i++) {
  1118.        if (small > mnr[k][i]) small = mnr[k][i];
  1119.        if(mnr[k][i] < NOISY_MIN_MNR) ++noisy_sbs; /* noise is not masked */
  1120.      }
  1121.    }
  1122.    return noisy_sbs;
  1123. }
  1124.  
  1125. /************************************************************************
  1126. /*
  1127. /* I_subband_quantization  (Layer I)
  1128. /* II_subband_quantization (Layer II)
  1129. /*
  1130. /* PURPOSE:Quantizes subband samples to appropriate number of bits
  1131. /*
  1132. /* SEMANTICS:  Subband samples are divided by their scalefactors, which
  1133. /* makes the quantization more efficient. The scaled samples are
  1134. /* quantized by the function a*x+b, where a and b are functions of
  1135. /* the number of quantization levels. The result is then truncated
  1136. /* to the appropriate number of bits and the MSB is inverted.
  1137. /*
  1138. /* Note that for fractional 2's complement, inverting the MSB for a
  1139. /* negative number x is equivalent to adding 1 to it.
  1140. /*
  1141. /************************************************************************/
  1142.  
  1143. static double a[17] = {
  1144.   0.750000000, 0.625000000, 0.875000000, 0.562500000, 0.937500000,
  1145.   0.968750000, 0.984375000, 0.992187500, 0.996093750, 0.998046875,
  1146.   0.999023438, 0.999511719, 0.999755859, 0.999877930, 0.999938965,
  1147.   0.999969482, 0.999984741 };
  1148.  
  1149. static double b[17] = {
  1150.   -0.250000000, -0.375000000, -0.125000000, -0.437500000, -0.062500000,
  1151.   -0.031250000, -0.015625000, -0.007812500, -0.003906250, -0.001953125,
  1152.   -0.000976563, -0.000488281, -0.000244141, -0.000122070, -0.000061035,
  1153.   -0.000030518, -0.000015259 };
  1154.  
  1155. void I_subband_quantization(scalar, sb_samples, j_scale, j_samps,
  1156.                             bit_alloc, sbband, fr_ps)
  1157. unsigned int scalar[2][3][SBLIMIT];
  1158. double FAR sb_samples[2][3][SCALE_BLOCK][SBLIMIT];
  1159. unsigned int j_scale[3][SBLIMIT];
  1160. double FAR j_samps[3][SCALE_BLOCK][SBLIMIT]; /* L+R for j-stereo if necess */
  1161. unsigned int bit_alloc[2][SBLIMIT];
  1162. unsigned int FAR sbband[2][3][SCALE_BLOCK][SBLIMIT];
  1163. frame_params *fr_ps;
  1164. {
  1165.    int i, j, k, n, sig;
  1166.    int stereo  = fr_ps->stereo;
  1167.    int sblimit = fr_ps->sblimit;
  1168.    int jsbound = fr_ps->jsbound;
  1169.    double d;
  1170. static char init = 0;
  1171.    if (!init) {
  1172.      init = 1;
  1173.      /* rearrange quantization coef to correspond to layer I table */
  1174.      a[1] = a[2]; b[1] = b[2];
  1175.      for (i=2;i<15;i++) { a[i] = a[i+2]; b[i] = b[i+2]; }
  1176.    }
  1177.    for (j=0;j<SCALE_BLOCK;j++) for (i=0;i<SBLIMIT;i++)
  1178.      for (k=0;k<((i<jsbound)?stereo:1);k++)
  1179.        if (bit_alloc[k][i]) {
  1180.          /* for joint stereo mode, have to construct a single subband stream
  1181.             for the js channels.  At present, we calculate a set of mono
  1182.             subband samples and pass them through the scaling system to
  1183.             generate an alternate normalised sample stream.
  1184.  
  1185.             Could normalise both streams (divide by their scfs), then average
  1186.             them.  In bad conditions, this could give rise to spurious
  1187.             cancellations.  Instead, we could just select the sb stream from
  1188.             the larger channel (higher scf), in which case _that_ channel
  1189.             would be 'properly' reconstructed, and the mate would just be a
  1190.             scaled version.  Spec recommends averaging the two (unnormalised)
  1191.             subband channels, then normalising this new signal without
  1192.             actually sending this scale factor... This means looking ahead.
  1193.          */
  1194.          if(stereo == 2 && i>=jsbound)
  1195.            /* use the joint data passed in */
  1196.            d = j_samps[0][j][i] / multiple[j_scale[0][i]];
  1197.          else
  1198.            d = sb_samples[k][0][j][i] / multiple[scalar[k][0][i]];
  1199.          /* scale and quantize floating point sample */
  1200.          n = bit_alloc[k][i];
  1201.          d = d * a[n-1] + b[n-1];
  1202.          /* extract MSB N-1 bits from the floating point sample */
  1203.          if (d >= 0) sig = 1;
  1204.          else { sig = 0; d += 1.0; }
  1205.          sbband[k][0][j][i] = (unsigned int) (d * (double) (1L<<n));
  1206.          /* tag the inverted sign bit to sbband at position N */
  1207.          if (sig) sbband[k][0][j][i] |= 1<<n;
  1208.        }
  1209. }
  1210.  
  1211. /***************************** Layer II  ********************************/
  1212.  
  1213. void II_subband_quantization(scalar, sb_samples, j_scale, j_samps,
  1214.                              bit_alloc, sbband, fr_ps)
  1215. unsigned int scalar[2][3][SBLIMIT];
  1216. double FAR sb_samples[2][3][SCALE_BLOCK][SBLIMIT];
  1217. unsigned int j_scale[3][SBLIMIT];
  1218. double FAR j_samps[3][SCALE_BLOCK][SBLIMIT];
  1219. unsigned int bit_alloc[2][SBLIMIT];
  1220. unsigned int FAR sbband[2][3][SCALE_BLOCK][SBLIMIT];
  1221. frame_params *fr_ps;
  1222. {
  1223.    int i, j, k, s, n, qnt, sig;
  1224.    int stereo  = fr_ps->stereo;
  1225.    int sblimit = fr_ps->sblimit;
  1226.    int jsbound = fr_ps->jsbound;
  1227.    unsigned int stps;
  1228.    double d;
  1229.    al_table *alloc = fr_ps->alloc;
  1230.    for (s=0;s<3;s++)
  1231.      for (j=0;j<SCALE_BLOCK;j++)
  1232.        for (i=0;i<sblimit;i++)
  1233.          for (k=0;k<((i<jsbound)?stereo:1);k++)
  1234.            if (bit_alloc[k][i]) {
  1235.              /* scale and quantize floating point sample */
  1236.              if(stereo == 2 && i>=jsbound)       /* use j-stereo samples */
  1237.                d = j_samps[s][j][i] / multiple[j_scale[s][i]];
  1238.              else
  1239.                d = sb_samples[k][s][j][i] / multiple[scalar[k][s][i]];
  1240.              if (mod(d) > 1.0)
  1241.                printf("Not scaled properly %d %d %d %dn",k,s,j,i);
  1242.              qnt = (*alloc)[i][bit_alloc[k][i]].quant;
  1243.              d = d * a[qnt] + b[qnt];
  1244.              /* extract MSB N-1 bits from the floating point sample */
  1245.              if (d >= 0) sig = 1;
  1246.              else { sig = 0; d += 1.0; }
  1247.              n = 0;
  1248. #ifndef MS_DOS
  1249.              stps = (*alloc)[i][bit_alloc[k][i]].steps;
  1250.              while ((1L<<n) < stps) n++;
  1251. #else
  1252.              while  ( ( (unsigned long)(1L<<(long)n) <
  1253.                        ((unsigned long) ((*alloc)[i][bit_alloc[k][i]].steps)
  1254.                         & 0xffff
  1255.                         )
  1256.                        ) && ( n <16)
  1257.                      ) n++;
  1258. #endif
  1259.              n--;
  1260.              sbband[k][s][j][i] = (unsigned int) (d * (double) (1L<<n));
  1261.              /* tag the inverted sign bit to sbband at position N */
  1262.              /* The bit inversion is a must for grouping with 3,5,9 steps
  1263.                 so it is done for all subbands */
  1264.              if (sig) sbband[k][s][j][i] |= 1<<n;
  1265.            }
  1266.            for (s=0;s<3;s++)
  1267.              for (j=sblimit;j<SBLIMIT;j++)
  1268.                for (i=0;i<SCALE_BLOCK;i++) for (k=0;k<stereo;k++) sbband[k][s][i][j] = 0;
  1269. }
  1270.  
  1271. /************************************************************************
  1272. /*
  1273. /* I_encode_bit_alloc  (Layer I)
  1274. /* II_encode_bit_alloc (Layer II)
  1275. /*
  1276. /* PURPOSE:Writes bit allocation information onto bitstream
  1277. /*
  1278. /* Layer I uses 4 bits/subband for bit allocation information,
  1279. /* and Layer II uses 4,3,2, or 0 bits depending on the
  1280. /* quantization table used.
  1281. /*
  1282. /************************************************************************/
  1283.  
  1284. void I_encode_bit_alloc(bit_alloc, fr_ps, bs)
  1285. unsigned int bit_alloc[2][SBLIMIT];
  1286. frame_params *fr_ps;
  1287. Bit_stream_struc *bs;
  1288. {
  1289.    int i,k;
  1290.    int stereo  = fr_ps->stereo;
  1291.    int sblimit = fr_ps->sblimit;
  1292.    int jsbound = fr_ps->jsbound;
  1293.  
  1294.    for (i=0;i<SBLIMIT;i++)
  1295.      for (k=0;k<((i<jsbound)?stereo:1);k++) putbits(bs,bit_alloc[k][i],4);
  1296. }
  1297.  
  1298. /***************************** Layer II  ********************************/
  1299.  
  1300. void II_encode_bit_alloc(bit_alloc, fr_ps, bs)
  1301. unsigned int bit_alloc[2][SBLIMIT];
  1302. frame_params *fr_ps;
  1303. Bit_stream_struc *bs;
  1304. {
  1305.    int i,k;
  1306.    int stereo  = fr_ps->stereo;
  1307.    int sblimit = fr_ps->sblimit;
  1308.    int jsbound = fr_ps->jsbound;
  1309.    al_table *alloc = fr_ps->alloc;
  1310.  
  1311.    for (i=0;i<sblimit;i++)
  1312.      for (k=0;k<((i<jsbound)?stereo:1);k++)
  1313.        putbits(bs,bit_alloc[k][i],(*alloc)[i][0].bits);
  1314. }
  1315.  
  1316. /************************************************************************
  1317. /*
  1318. /* I_sample_encoding   (Layer I)
  1319. /* II_sample_encoding  (Layer II)
  1320. /*
  1321. /* PURPOSE:Put one frame of subband samples on to the bitstream
  1322. /*
  1323. /* SEMANTICS:  The number of bits allocated per sample is read from
  1324. /* the bit allocation information #bit_alloc#.  Layer 2
  1325. /* supports writing grouped samples for quantization steps
  1326. /* that are not a power of 2.
  1327. /*
  1328. /************************************************************************/
  1329.  
  1330. void I_sample_encoding(sbband, bit_alloc, fr_ps, bs)
  1331. unsigned int FAR sbband[2][3][SCALE_BLOCK][SBLIMIT];
  1332. unsigned int bit_alloc[2][SBLIMIT];
  1333. frame_params *fr_ps;
  1334. Bit_stream_struc *bs;
  1335. {
  1336.    int i,j,k;
  1337.    int stereo  = fr_ps->stereo;
  1338.    int sblimit = fr_ps->sblimit;
  1339.    int jsbound = fr_ps->jsbound;
  1340.  
  1341.    for(j=0;j<SCALE_BLOCK;j++) {
  1342.      for(i=0;i<SBLIMIT;i++)
  1343.        for(k=0;k<((i<jsbound)?stereo:1);k++)
  1344.          if(bit_alloc[k][i]) putbits(bs,sbband[k][0][j][i],bit_alloc[k][i]+1);
  1345.    }
  1346. }
  1347.  
  1348. /***************************** Layer II  ********************************/
  1349.  
  1350. void II_sample_encoding(sbband, bit_alloc, fr_ps, bs)
  1351. unsigned int FAR sbband[2][3][SCALE_BLOCK][SBLIMIT];
  1352. unsigned int bit_alloc[2][SBLIMIT];
  1353. frame_params *fr_ps;
  1354. Bit_stream_struc *bs;
  1355. {
  1356.    unsigned int temp;
  1357.    unsigned int i,j,k,s,x,y;
  1358.    int stereo  = fr_ps->stereo;
  1359.    int sblimit = fr_ps->sblimit;
  1360.    int jsbound = fr_ps->jsbound;
  1361.    al_table *alloc = fr_ps->alloc;
  1362.  
  1363.    for (s=0;s<3;s++)
  1364.      for (j=0;j<SCALE_BLOCK;j+=3)
  1365.        for (i=0;i<sblimit;i++)
  1366.          for (k=0;k<((i<jsbound)?stereo:1);k++)
  1367.            if (bit_alloc[k][i]) {
  1368.              if ((*alloc)[i][bit_alloc[k][i]].group == 3) {
  1369.                for (x=0;x<3;x++) putbits(bs,sbband[k][s][j+x][i],
  1370.                                          (*alloc)[i][bit_alloc[k][i]].bits);
  1371.              }
  1372.              else {
  1373.                y =(*alloc)[i][bit_alloc[k][i]].steps;
  1374.                temp = sbband[k][s][j][i] +
  1375.                       sbband[k][s][j+1][i] * y +
  1376.                       sbband[k][s][j+2][i] * y * y;
  1377.                putbits(bs,temp,(*alloc)[i][bit_alloc[k][i]].bits);
  1378.              }
  1379.            }
  1380. }
  1381.  
  1382. /************************************************************************
  1383. /*
  1384. /* encode_CRC
  1385. /*
  1386. /************************************************************************/
  1387.  
  1388. void encode_CRC(crc, bs)
  1389. unsigned int crc;
  1390. Bit_stream_struc *bs;
  1391. {
  1392.    putbits(bs, crc, 16);
  1393. }