nok_bwp_enc.c

MPEG2/MPEG4编解码参考程序(实现了MPEG4的部分功能)

/**************************************************************************

This software module was originally developed by
Nokia in the course of development of the MPEG-2 AAC/MPEG-4 
Audio standard ISO/IEC13818-7, 14496-1, 2 and 3.
This software module is an implementation of a part
of one or more MPEG-2 AAC/MPEG-4 Audio tools as specified by the
MPEG-2 aac/MPEG-4 Audio standard. ISO/IEC  gives users of the
MPEG-2aac/MPEG-4 Audio standards free license to this software module
or modifications thereof for use in hardware or software products
claiming conformance to the MPEG-2 aac/MPEG-4 Audio  standards. Those
intending to use this software module in hardware or software products
are advised that this use may infringe existing patents. The original
developer of this software module, the subsequent
editors and their companies, and ISO/IEC have no liability for use of
this software module or modifications thereof in an
implementation. Copyright is not released for non MPEG-2 aac/MPEG-4
Audio conforming products. The original developer retains full right to
use the code for the developer's own purpose, assign or donate the code to a
third party and to inhibit third party from using the code for non
MPEG-2 aac/MPEG-4 Audio conforming products. This copyright notice
must be included in all copies or derivative works.
Copyright (c)1997.  

***************************************************************************/
/**************************************************************************
  nok_bwp_enc.c  -  Nokia BWP for encoding
  Author(s): Mikko Suonio, Lin Yin, Kalervo Kontola
  	     Nokia Research Center, Speech and Audio Systems.
  *************************************************************************/


/**************************************************************************
  Version Control Information			Method: CVS
  Identifiers:
  $Revision: 1.11 $
  $Date: 1999/04/22 16:19:21 $ (check in)
  $Author: purnhage $
  $State: Exp $
  $Source: /home/purnhage/cvsroot_vm/vm/src_tf_enc/nok_bwp_enc.c,v $
  *************************************************************************/
#include        <stdio.h>
#include        <string.h>
#include        <math.h>
#include        <stdlib.h>

#include "block.h"               /* handler, defines, enums */
#include "buffersHandle.h"       /* handler, defines, enums */
#include "concealmentHandle.h"   /* handler, defines, enums */
#include "interface.h"           /* handler, defines, enums */
#include "mod_bufHandle.h"       /* handler, defines, enums */
#include "reorderspecHandle.h"   /* handler, defines, enums */
#include "resilienceHandle.h"    /* handler, defines, enums */
#include "tf_mainHandle.h"       /* handler, defines, enums */

#include "nok_ltp_common.h"      /* structs */
#include "obj_descr.h"           /* structs */
#include "tf_mainStruct.h"       /* structs */

#include "bitstream.h"
#include "nok_bwp_enc.h"
/*#include "tf_main.h"*/

static float NOK_codebook[NOK_CODESIZE] =
{
  -2.9763780e+00, -2.9291339e+00, -2.8818898e+00, -2.8346457e+00,
  -2.7874016e+00, -2.7401575e+00, -2.6929134e+00, -2.6456693e+00,
  -2.5984252e+00, -2.5511811e+00, -2.5039370e+00, -2.4566929e+00,
  -2.4094488e+00, -2.3622047e+00, -2.3149606e+00, -2.2677165e+00,
  -2.2204724e+00, -2.1732283e+00, -2.1259843e+00, -2.0787402e+00,
  -2.0314961e+00, -1.9842520e+00, -1.9370079e+00, -1.8897638e+00,
  -1.8425197e+00, -1.7952756e+00, -1.7480315e+00, -1.7007874e+00,
  -1.6535433e+00, -1.6062992e+00, -1.5590551e+00, -1.5118110e+00,
  -1.4645669e+00, -1.4173228e+00, -1.3700787e+00, -1.3228346e+00,
  -1.2755906e+00, -1.2283465e+00, -1.1811024e+00, -1.1338583e+00,
  -1.0866142e+00, -1.0393701e+00, -9.9212598e-01, -9.4488189e-01,
  -8.9763780e-01, -8.5039370e-01, -8.0314961e-01, -7.5590551e-01,
  -7.0866142e-01, -6.6141732e-01, -6.1417323e-01, -5.6692913e-01,
  -5.1968504e-01, -4.7244094e-01, -4.2519685e-01, -3.7795276e-01,
  -3.3070866e-01, -2.8346457e-01, -2.3622047e-01, -1.8897638e-01,
  -1.4173228e-01, -9.4488189e-02, -4.7244094e-02, 0.0000000e+00,
  4.7244094e-02, 9.4488189e-02, 1.4173228e-01, 1.8897638e-01,
  2.3622047e-01, 2.8346457e-01, 3.3070866e-01, 3.7795276e-01,
  4.2519685e-01, 4.7244094e-01, 5.1968504e-01, 5.6692913e-01,
  6.1417323e-01, 6.6141732e-01, 7.0866142e-01, 7.5590551e-01,
  8.0314961e-01, 8.5039370e-01, 8.9763780e-01, 9.4488189e-01,
  9.9212598e-01, 1.0393701e+00, 1.0866142e+00, 1.1338583e+00,
  1.1811024e+00, 1.2283465e+00, 1.2755906e+00, 1.3228346e+00,
  1.3700787e+00, 1.4173228e+00, 1.4645669e+00, 1.5118110e+00,
  1.5590551e+00, 1.6062992e+00, 1.6535433e+00, 1.7007874e+00,
  1.7480315e+00, 1.7952756e+00, 1.8425197e+00, 1.8897638e+00,
  1.9370079e+00, 1.9842520e+00, 2.0314961e+00, 2.0787402e+00,
  2.1259843e+00, 2.1732283e+00, 2.2204724e+00, 2.2677165e+00,
  2.3149606e+00, 2.3622047e+00, 2.4094488e+00, 2.4566929e+00,
  2.5039370e+00, 2.5511811e+00, 2.5984252e+00, 2.6456693e+00,
  2.6929134e+00, 2.7401575e+00, 2.7874016e+00, 2.8346457e+00,
  2.8818898e+00, 2.9291339e+00, 2.9763780e+00, 2.9763780e+00,

  1.000000e+00, 9.921875e-01, 9.843750e-01, 9.765625e-01, 9.687500e-01, 9.609375e-01, 9.570312e-01, 9.492188e-01, 9.414062e-01, 9.335938e-01, 9.257812e-01, 9.218750e-01, 9.140625e-01, 9.062500e-01, 9.023438e-01, 8.945312e-01, 8.906250e-01, 8.828125e-01, 8.750000e-01, 8.710938e-01, 8.632812e-01, 8.593750e-01, 8.515625e-01, 8.476562e-01, 8.437500e-01, 8.359375e-01, 8.320312e-01, 8.242188e-01, 8.203125e-01, 8.164062e-01, 8.085938e-01, 8.046875e-01, 8.007812e-01, 7.968750e-01, 7.890625e-01, 7.851562e-01, 7.812500e-01, 7.773438e-01, 7.695312e-01, 7.656250e-01, 7.617188e-01, 7.578125e-01, 7.539062e-01, 7.500000e-01, 7.460938e-01, 7.382812e-01, 7.343750e-01, 7.304688e-01, 7.265625e-01, 7.226562e-01, 7.187500e-01, 7.148438e-01, 7.109375e-01, 7.070312e-01, 7.031250e-01, 6.992188e-01, 6.953125e-01, 6.914062e-01, 6.875000e-01, 6.835938e-01, 6.796875e-01, 6.757812e-01, 6.718750e-01, 6.718750e-01, 6.679688e-01, 6.640625e-01, 6.601562e-01, 6.562500e-01, 6.523438e-01, 6.484375e-01, 6.445312e-01, 6.445312e-01, 6.406250e-01, 6.367188e-01, 6.328125e-01, 6.289062e-01, 6.289062e-01, 6.250000e-01, 6.210938e-01, 6.171875e-01, 6.171875e-01, 6.132812e-01, 6.093750e-01, 6.054688e-01, 6.054688e-01, 6.015625e-01, 5.976562e-01, 5.937500e-01, 5.937500e-01, 5.898438e-01, 5.859375e-01, 5.859375e-01, 5.820312e-01, 5.781250e-01, 5.781250e-01, 5.742188e-01, 5.703125e-01, 5.703125e-01, 5.664062e-01, 5.625000e-01, 5.625000e-01, 5.585938e-01, 5.546875e-01, 5.546875e-01, 5.507812e-01, 5.507812e-01, 5.468750e-01, 5.429688e-01, 5.429688e-01, 5.390625e-01, 5.390625e-01, 5.351562e-01, 5.351562e-01, 5.312500e-01, 5.273438e-01, 5.273438e-01, 5.234375e-01, 5.234375e-01, 5.195312e-01, 5.195312e-01, 5.156250e-01, 5.156250e-01, 5.117188e-01, 5.117188e-01, 5.078125e-01, 5.078125e-01, 5.039062e-01, 5.039062e-01,

  9.99999e+37, 8.507059e+37, 4.253530e+37, 2.126765e+37, 1.063382e+37, 5.316912e+36, 2.658456e+36, 1.329228e+36, 6.646140e+35, 3.323070e+35, 1.661535e+35, 8.307675e+34, 4.153837e+34, 2.076919e+34, 1.038459e+34, 5.192297e+33, 2.596148e+33, 1.298074e+33, 6.490371e+32, 3.245186e+32, 1.622593e+32, 8.112964e+31, 4.056482e+31, 2.028241e+31, 1.014120e+31, 5.070602e+30, 2.535301e+30, 1.267651e+30, 6.338253e+29, 3.169127e+29, 1.584563e+29, 7.922816e+28, 3.961408e+28, 1.980704e+28, 9.903520e+27, 4.951760e+27, 2.475880e+27, 1.237940e+27, 6.189700e+26, 3.094850e+26, 1.547425e+26, 7.737125e+25, 3.868563e+25, 1.934281e+25, 9.671407e+24, 4.835703e+24, 2.417852e+24, 1.208926e+24, 6.044629e+23, 3.022315e+23, 1.511157e+23, 7.555786e+22, 3.777893e+22, 1.888947e+22, 9.444733e+21, 4.722366e+21, 2.361183e+21, 1.180592e+21, 5.902958e+20, 2.951479e+20, 1.475740e+20, 7.378698e+19, 3.689349e+19, 1.844674e+19, 9.223372e+18, 4.611686e+18, 2.305843e+18, 1.152922e+18, 5.764608e+17, 2.882304e+17, 1.441152e+17, 7.205759e+16, 3.602880e+16, 1.801440e+16, 9.007199e+15, 4.503600e+15, 2.251800e+15, 1.125900e+15, 5.629500e+14, 2.814750e+14, 1.407375e+14, 7.036874e+13, 3.518437e+13, 1.759219e+13, 8.796093e+12, 4.398047e+12, 2.199023e+12, 1.099512e+12, 5.497558e+11, 2.748779e+11, 1.374390e+11, 6.871948e+10, 3.435974e+10, 1.717987e+10, 8.589935e+09, 4.294967e+09, 2.147484e+09, 1.073742e+09, 5.368709e+08, 2.684355e+08, 1.342177e+08, 6.710886e+07, 3.355443e+07, 1.677722e+07, 8.388608e+06, 4.194304e+06, 2.097152e+06, 1.048576e+06, 5.242880e+05, 2.621440e+05, 1.310720e+05, 6.553600e+04, 3.276800e+04, 1.638400e+04, 8.192000e+03, 4.096000e+03, 2.048000e+03, 1.024000e+03, 5.120000e+02, 2.560000e+02, 1.280000e+02, 6.400000e+01, 3.200000e+01, 1.600000e+01, 8.000000e+00, 4.000000e+00, 2.000000e+00, 1.000000e+00, 5.000000e-01, 2.500000e-01, 1.250000e-01, 6.250000e-02, 3.125000e-02, 1.562500e-02, 7.812500e-03, 3.906250e-03, 1.953125e-03, 9.765625e-04, 4.882812e-04, 2.441406e-04, 1.220703e-04, 6.103516e-05, 3.051758e-05, 1.525879e-05, 7.629395e-06, 3.814697e-06, 1.907349e-06, 9.536743e-07, 4.768372e-07, 2.384186e-07, 1.192093e-07, 5.960464e-08, 2.980232e-08, 1.490116e-08, 7.450581e-09, 3.725290e-09, 1.862645e-09, 9.313226e-10, 4.656613e-10, 2.328306e-10, 1.164153e-10, 5.820766e-11, 2.910383e-11, 1.455192e-11, 7.275958e-12, 3.637979e-12, 1.818989e-12, 9.094947e-13, 4.547474e-13, 2.273737e-13, 1.136868e-13, 5.684342e-14, 2.842171e-14, 1.421085e-14, 7.105427e-15, 3.552714e-15, 1.776357e-15, 8.881784e-16, 4.440892e-16, 2.220446e-16, 1.110223e-16, 5.551115e-17, 2.775558e-17, 1.387779e-17, 6.938894e-18, 3.469447e-18, 1.734723e-18, 8.673617e-19, 4.336809e-19, 2.168404e-19, 1.084202e-19, 5.421011e-20, 2.710505e-20, 1.355253e-20, 6.776264e-21, 3.388132e-21, 1.694066e-21, 8.470329e-22, 4.235165e-22, 2.117582e-22, 1.058791e-22, 5.293956e-23, 2.646978e-23, 1.323489e-23, 6.617445e-24, 3.308722e-24, 1.654361e-24, 8.271806e-25, 4.135903e-25, 2.067952e-25, 1.033976e-25, 5.169879e-26, 2.584939e-26, 1.292470e-26, 6.462349e-27, 3.231174e-27, 1.615587e-27, 8.077936e-28, 4.038968e-28, 2.019484e-28, 1.009742e-28, 5.048710e-29, 2.524355e-29, 1.262177e-29, 6.310887e-30, 3.155444e-30, 1.577722e-30, 7.888609e-31, 3.944305e-31, 1.972152e-31, 9.860761e-32, 4.930381e-32, 2.465190e-32, 1.232595e-32, 6.162976e-33, 3.081488e-33, 1.540744e-33, 7.703720e-34, 3.851860e-34, 1.925930e-34, 9.629650e-35, 4.814825e-35, 2.407412e-35, 1.203706e-35, 6.018531e-36, 3.009266e-36, 1.504633e-36, 7.523164e-37, 3.761582e-37, 1.880791e-37, 9.403955e-38, 4.701977e-38, 2.350989e-38, 1.175494e-38, 5.877472e-39, 0.000000e+00};

#define         BWP_LPC                2
#define         BWP_BLN5              10
#define         BWP_BLEN             168

#define         SBMAX_L               49
#define         BLN5                  10
#define         CLEN                 672
#define         DATA_LEN               3
#define         NRCSTA                 4

#define         ONE_BIT                1
#define         BWP_RESET_BIT          1
#define         BWP_RESET_INDEX_BITS   5
#define         BWP_USED               1
#define         NOT_USED              -1
#define         MAX_SFB_SW             4
#define         MAX_SFB_LW             6
#define         NOK_BWP_PRED_SFB_MAX          40
#define SPECLEN 1024
#define LONG_BLOCK  0
#define START_BLOCK 1
#define SHORT_BLOCK 2
#define STOP_BLOCK 3
#define PRED_RESET_GROUPS 5
#define	RESET_PERIOD 8
#define	RESET_GROUPS 30
#define LEN_PREDICTOR_DATA_PRESENT 1
#define LEN_PREDICTION_USED 1
#define LEN_PREDICTOR_RESET 1
#define LEN_PREDICTOR_RESET_INDEX 5
#define DEBUG_THRESHOLD 4

static int indicate = 0;
static unsigned short x_buffer[BWP_BLN5][BWP_BLEN];
static unsigned char pred[2][CLEN];
static float sb_samples_pred[CLEN];
static int debuglevel;

static void autocorr (unsigned short x[BWP_BLN5][BWP_BLEN], double last_spec[CLEN], unsigned char pred[2][CLEN], float sb_samples_pred[CLEN], int m, int indicate);
static void flt_round_8 (float *pf);
static unsigned short fs (float x);
static float sf (unsigned short ps);
static void nok_quantize (float coef, unsigned char *findex);


void
nok_InitPrediction (int debug)
{
  int i, j;

  for (i = 0; i < BWP_BLN5; i++)
    for (j = 0; j < BWP_BLEN; j++)
      x_buffer[i][j] = 0;

  for (i = 0; i < BWP_LPC; i++)
    for (j = 0; j < CLEN; j++)
      pred[i][j] = 63;

  for (j = 0; j < CLEN; j++)
    sb_samples_pred[j] = 0;

  indicate = 0;

  debuglevel = debug;
}

/**************************************************************************
  Title:	PredCalcPrediction

  Purpose:	BWP for encoding

  Usage:	PredCalcPrediction (act_spec, last_spec, btype, nsfb,
  				    isfb_width, pred_global_flag,
				    pred_sfb_flag, reset_flag,
				    reset_group);

  Input:	act_spec	- output of MDCT
                last_spec       - dequantized values of the last predicted 
                                  spectrum; should be identically zero
				  on the first call
                btype           - window type (short/long)
                nsfb            - # of SFBs
                isfb_width      - table of SFB-widths
  Output:	act_spec	- prediction error if prediction is utilized
                pred_global_flag- TRUE if prediction used
                pred_sfb_flag   - a flag for each SFB indicating prediction
                reset_flag      - TRUE if reset used
                reset_group     - indicates which group (of 30 possibles) is 
                                  reset  (5 bit index)
  References:	

  Explanation:

  Author(s):	Mikko Suonio, Lin Yin, Kalervo Kontola
  *************************************************************************/

void
nok_PredCalcPrediction (double *act_spec, double *last_spec,
			int btype, int nsfb,
			int *isfb_width, short *pred_global_flag,
			int *pred_sfb_flag, short *reset_flag,
			int *reset_group, int *side_info)
{
  int i, k, j, cb_long, rg, last_band;
  float ftemp;
  double num_bit, snr[SBMAX_L];
  double energy[CLEN], snr_p[CLEN], temp1, temp2;

  static int do_reset = 0, resetting_group = 0, short_set = 0;
  static short pred_global_prev = 0;

  *reset_flag = 0;
  last_band = ( nsfb < NOK_BWP_PRED_SFB_MAX ) ? nsfb : NOK_BWP_PRED_SFB_MAX;
  if (debuglevel >= DEBUG_THRESHOLD)
    fprintf (stderr, "nok_PredCalcPrediction:  last_band %d\n", last_band);

  if (btype == EIGHT_SHORT_SEQUENCE)	/* was: ==2 */
    {
      for (i = 0; i < BWP_BLN5; i++)