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
