Home | History | Annotate | Download | only in src 
      1 /*
      2  ** Copyright 2003-2010, VisualOn, Inc.
      3  **
      4  ** Licensed under the Apache License, Version 2.0 (the "License");
      5  ** you may not use this file except in compliance with the License.
      6  ** You may obtain a copy of the License at
      7  **
      8  **     http://www.apache.org/licenses/LICENSE-2.0
      9  **
     10  ** Unless required by applicable law or agreed to in writing, software
     11  ** distributed under the License is distributed on an "AS IS" BASIS,
     12  ** WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  ** See the License for the specific language governing permissions and
     14  ** limitations under the License.
     15  */
     16 /*******************************************************************************
     17 	File:		aacenc_core.c
     18 
     19 	Content:	aac encoder core functions
     20 
     21 *******************************************************************************/
     22 
     23 #include "typedef.h"
     24 #include "aacenc_core.h"
     25 #include "bitenc.h"
     26 
     27 #include "psy_configuration.h"
     28 #include "psy_main.h"
     29 #include "qc_main.h"
     30 #include "psy_main.h"
     31 #include "channel_map.h"
     32 #include "aac_rom.h"
     33 
     34 /********************************************************************************
     35 *
     36 * function name: AacInitDefaultConfig
     37 * description:  gives reasonable default configuration
     38 *
     39 **********************************************************************************/
     40 void AacInitDefaultConfig(AACENC_CONFIG *config)
     41 {
     42   /* default configurations */
     43   config->adtsUsed        = 1;
     44   config->nChannelsIn     = 2;
     45   config->nChannelsOut    = 2;
     46   config->bitRate         = 128000;
     47   config->bandWidth       = 0;
     48 }
     49 
     50 /********************************************************************************
     51 *
     52 * function name: AacEncOpen
     53 * description:  allocate and initialize a new encoder instance
     54 * returns:      0 if success
     55 *
     56 **********************************************************************************/
     57 Word16  AacEncOpen(  AAC_ENCODER*      hAacEnc,        /* pointer to an encoder handle, initialized on return */
     58                      const  AACENC_CONFIG     config   /* pre-initialized config struct */
     59                      )
     60 {
     61   Word32 i;
     62   Word32 error = 0;
     63   Word16 profile = 1;
     64 
     65   ELEMENT_INFO *elInfo = NULL;
     66 
     67   if (hAacEnc==0) {
     68     error=1;
     69   }
     70 
     71   if (!error) {
     72     hAacEnc->config = config;
     73   }
     74 
     75   if (!error) {
     76     error = InitElementInfo (config.nChannelsOut,
     77                              &hAacEnc->elInfo);
     78   }
     79 
     80   if (!error) {
     81     elInfo = &hAacEnc->elInfo;
     82   }
     83 
     84   if (!error) {
     85     /* use or not tns tool for long and short block */
     86 	 Word16 tnsMask=3;
     87 
     88 	/* init encoder psychoacoustic */
     89     error = psyMainInit(&hAacEnc->psyKernel,
     90                         config.sampleRate,
     91                         config.bitRate,
     92                         elInfo->nChannelsInEl,
     93                         tnsMask,
     94                         hAacEnc->config.bandWidth);
     95   }
     96 
     97  /* use or not adts header */
     98   if(!error) {
     99 	  hAacEnc->qcOut.qcElement.adtsUsed = config.adtsUsed;
    100   }
    101 
    102   /* init encoder quantization */
    103   if (!error) {
    104     struct QC_INIT qcInit;
    105 
    106     /*qcInit.channelMapping = &hAacEnc->channelMapping;*/
    107     qcInit.elInfo = &hAacEnc->elInfo;
    108 
    109     qcInit.maxBits = (Word16) (MAXBITS_COEF*elInfo->nChannelsInEl);
    110     qcInit.bitRes = qcInit.maxBits;
    111     qcInit.averageBits = (Word16) ((config.bitRate * FRAME_LEN_LONG) / config.sampleRate);
    112 
    113     qcInit.padding.paddingRest = config.sampleRate;
    114 
    115     qcInit.meanPe = (Word16) ((10 * FRAME_LEN_LONG * hAacEnc->config.bandWidth) /
    116                                               (config.sampleRate>>1));
    117 
    118     qcInit.maxBitFac = (Word16) ((100 * (MAXBITS_COEF-MINBITS_COEF)* elInfo->nChannelsInEl)/
    119                                                  (qcInit.averageBits?qcInit.averageBits:1));
    120 
    121     qcInit.bitrate = config.bitRate;
    122 
    123     error = QCInit(&hAacEnc->qcKernel, &qcInit);
    124   }
    125 
    126   /* init bitstream encoder */
    127   if (!error) {
    128     hAacEnc->bseInit.nChannels   = elInfo->nChannelsInEl;
    129     hAacEnc->bseInit.bitrate     = config.bitRate;
    130     hAacEnc->bseInit.sampleRate  = config.sampleRate;
    131     hAacEnc->bseInit.profile     = profile;
    132   }
    133 
    134   return error;
    135 }
    136 
    137 /********************************************************************************
    138 *
    139 * function name: AacEncEncode
    140 * description:  encode pcm to aac data core function
    141 * returns:      0 if success
    142 *
    143 **********************************************************************************/
    144 Word16 AacEncEncode(AAC_ENCODER *aacEnc,		/*!< an encoder handle */
    145                     Word16 *timeSignal,         /*!< BLOCKSIZE*nChannels audio samples, interleaved */
    146                     const UWord8 *ancBytes,     /*!< pointer to ancillary data bytes */
    147                     Word16 *numAncBytes,		/*!< number of ancillary Data Bytes */
    148                     UWord8 *outBytes,           /*!< pointer to output buffer (must be large MINBITS_COEF/8*MAX_CHANNELS bytes) */
    149                     VO_U32 *numOutBytes         /*!< number of bytes in output buffer after processing */
    150                     )
    151 {
    152   ELEMENT_INFO *elInfo = &aacEnc->elInfo;
    153   Word16 globUsedBits;
    154   Word16 ancDataBytes, ancDataBytesLeft;
    155 
    156   ancDataBytes = ancDataBytesLeft = *numAncBytes;
    157 
    158   /* init output aac data buffer and length */
    159   aacEnc->hBitStream = CreateBitBuffer(&aacEnc->bitStream, outBytes, *numOutBytes);
    160 
    161   /* psychoacoustic process */
    162   psyMain(aacEnc->config.nChannelsOut,
    163           elInfo,
    164           timeSignal,
    165           &aacEnc->psyKernel.psyData[elInfo->ChannelIndex[0]],
    166           &aacEnc->psyKernel.tnsData[elInfo->ChannelIndex[0]],
    167           &aacEnc->psyKernel.psyConfLong,
    168           &aacEnc->psyKernel.psyConfShort,
    169           &aacEnc->psyOut.psyOutChannel[elInfo->ChannelIndex[0]],
    170           &aacEnc->psyOut.psyOutElement,
    171           aacEnc->psyKernel.pScratchTns,
    172 		  aacEnc->config.sampleRate);
    173 
    174   /* adjust bitrate and frame length */
    175   AdjustBitrate(&aacEnc->qcKernel,
    176                 aacEnc->config.bitRate,
    177                 aacEnc->config.sampleRate);
    178 
    179   /* quantization and coding process */
    180   QCMain(&aacEnc->qcKernel,
    181          &aacEnc->qcKernel.elementBits,
    182          &aacEnc->qcKernel.adjThr.adjThrStateElem,
    183          &aacEnc->psyOut.psyOutChannel[elInfo->ChannelIndex[0]],
    184          &aacEnc->psyOut.psyOutElement,
    185          &aacEnc->qcOut.qcChannel[elInfo->ChannelIndex[0]],
    186          &aacEnc->qcOut.qcElement,
    187          elInfo->nChannelsInEl,
    188 		 min(ancDataBytesLeft,ancDataBytes));
    189 
    190   ancDataBytesLeft = ancDataBytesLeft - ancDataBytes;
    191 
    192   globUsedBits = FinalizeBitConsumption(&aacEnc->qcKernel,
    193                          &aacEnc->qcOut);
    194 
    195   /* write bitstream process */
    196   WriteBitstream(aacEnc->hBitStream,
    197                  *elInfo,
    198                  &aacEnc->qcOut,
    199                  &aacEnc->psyOut,
    200                  &globUsedBits,
    201                  ancBytes,
    202 				 aacEnc->psyKernel.sampleRateIdx);
    203 
    204   updateBitres(&aacEnc->qcKernel,
    205                &aacEnc->qcOut);
    206 
    207   /* write out the bitstream */
    208   *numOutBytes = GetBitsAvail(aacEnc->hBitStream) >> 3;
    209 
    210   return 0;
    211 }
    212 
    213 
    214 /********************************************************************************
    215 *
    216 * function name:AacEncClose
    217 * description: deallocate an encoder instance
    218 *
    219 **********************************************************************************/
    220 void AacEncClose (AAC_ENCODER* hAacEnc, VO_MEM_OPERATOR *pMemOP)
    221 {
    222   if (hAacEnc) {
    223     QCDelete(&hAacEnc->qcKernel, pMemOP);
    224 
    225     QCOutDelete(&hAacEnc->qcOut, pMemOP);
    226 
    227     PsyDelete(&hAacEnc->psyKernel, pMemOP);
    228 
    229     PsyOutDelete(&hAacEnc->psyOut, pMemOP);
    230 
    231     DeleteBitBuffer(&hAacEnc->hBitStream);
    232 
    233 	if(hAacEnc->intbuf)
    234 	{
    235 		mem_free(pMemOP, hAacEnc->intbuf, VO_INDEX_ENC_AAC);
    236 		hAacEnc->intbuf = NULL;
    237 	}
    238   }
    239 }
    240