summaryrefslogtreecommitdiffstats
path: root/libMpegTPEnc/src/tpenc_adts.cpp
diff options
context:
space:
mode:
authorDave Burke <daveburke@google.com>2012-04-17 09:51:45 -0700
committerDave Burke <daveburke@google.com>2012-04-17 23:04:43 -0700
commit9bf37cc9712506b2483650c82d3c41152337ef7e (patch)
tree77db44e2bae06e3d144b255628be2b7a55c581d3 /libMpegTPEnc/src/tpenc_adts.cpp
parenta37315fe10ee143d6d0b28c19d41a476a23e63ea (diff)
downloadfdk-aac-9bf37cc9712506b2483650c82d3c41152337ef7e.tar.gz
fdk-aac-9bf37cc9712506b2483650c82d3c41152337ef7e.tar.bz2
fdk-aac-9bf37cc9712506b2483650c82d3c41152337ef7e.zip
Fraunhofer AAC codec.
License boilerplate update to follow. Change-Id: I2810460c11a58b6d148d84673cc031f3685e79b5
Diffstat (limited to 'libMpegTPEnc/src/tpenc_adts.cpp')
-rw-r--r--libMpegTPEnc/src/tpenc_adts.cpp253
1 files changed, 253 insertions, 0 deletions
diff --git a/libMpegTPEnc/src/tpenc_adts.cpp b/libMpegTPEnc/src/tpenc_adts.cpp
new file mode 100644
index 0000000..86b11c1
--- /dev/null
+++ b/libMpegTPEnc/src/tpenc_adts.cpp
@@ -0,0 +1,253 @@
+/******************************** MPEG Audio Encoder **************************
+
+ (C) copyright Fraunhofer-IIS (1999)
+ All Rights Reserved
+
+ Please be advised that this software and/or program delivery is
+ Confidential Information of Fraunhofer and subject to and covered by the
+
+ Fraunhofer IIS Software Evaluation Agreement
+ between Google Inc. and Fraunhofer
+ effective and in full force since March 1, 2012.
+
+ You may use this software and/or program only under the terms and
+ conditions described in the above mentioned Fraunhofer IIS Software
+ Evaluation Agreement. Any other and/or further use requires a separate agreement.
+
+
+ This software and/or program is protected by copyright law and international
+ treaties. Any reproduction or distribution of this software and/or program,
+ or any portion of it, may result in severe civil and criminal penalties, and
+ will be prosecuted to the maximum extent possible under law.
+
+ $Id$
+ Initial author: Alex Groeschel
+ contents/description: ADTS Transport Headers support
+
+******************************************************************************/
+
+#include "tpenc_adts.h"
+
+
+#include "tpenc_lib.h"
+#include "tpenc_asc.h"
+
+
+int adtsWrite_CrcStartReg(
+ HANDLE_ADTS pAdts, /*!< pointer to adts stucture */
+ HANDLE_FDK_BITSTREAM hBs, /*!< handle to current bit buffer structure */
+ int mBits /*!< number of bits in crc region */
+ )
+{
+ if (pAdts->protection_absent) {
+ return 0;
+ }
+ return ( FDKcrcStartReg(&pAdts->crcInfo, hBs, mBits) );
+}
+
+void adtsWrite_CrcEndReg(
+ HANDLE_ADTS pAdts, /*!< pointer to adts crc info stucture */
+ HANDLE_FDK_BITSTREAM hBs, /*!< handle to current bit buffer structure */
+ int reg /*!< crc region */
+ )
+{
+ if (pAdts->protection_absent == 0)
+ {
+ FDKcrcEndReg(&pAdts->crcInfo, hBs, reg);
+ }
+}
+
+int adtsWrite_GetHeaderBits( HANDLE_ADTS hAdts )
+{
+ int bits = 0;
+
+ if (hAdts->currentBlock == 0) {
+ /* Static and variable header bits */
+ bits = 56;
+ if (!hAdts->protection_absent) {
+ /* Add header/ single raw data block CRC bits */
+ bits += 16;
+ if (hAdts->num_raw_blocks>0) {
+ /* Add bits of raw data block position markers */
+ bits += (hAdts->num_raw_blocks)*16;
+ }
+ }
+ }
+ if (!hAdts->protection_absent && hAdts->num_raw_blocks>0) {
+ /* Add raw data block CRC bits. Not really part of the header, put they cause bit overhead to be accounted. */
+ bits += 16;
+ }
+
+ hAdts->headerBits = bits;
+
+ return bits;
+}
+
+INT adtsWrite_Init(HANDLE_ADTS hAdts, CODER_CONFIG *config)
+{
+ /* Sanity checks */
+ if ( config->nSubFrames < 1
+ || config->nSubFrames > 4
+ || (int)config->aot > 4
+ || (int)config->aot < 1 ) {
+ return -1;
+ }
+
+ /* fixed header */
+ if (config->flags & CC_MPEG_ID) {
+ hAdts->mpeg_id = 0; /* MPEG 4 */
+ } else {
+ hAdts->mpeg_id = 1; /* MPEG 2 */
+ }
+ hAdts->layer=0;
+ hAdts->protection_absent = ! (config->flags & CC_PROTECTION);
+ hAdts->profile = ((int)config->aot) - 1;
+ hAdts->sample_freq_index = getSamplingRateIndex(config->samplingRate);
+ hAdts->sample_freq = config->samplingRate;
+ hAdts->private_bit=0;
+ hAdts->channel_mode = config->channelMode;
+ hAdts->original=0;
+ hAdts->home=0;
+ /* variable header */
+ hAdts->copyright_id=0;
+ hAdts->copyright_start=0;
+
+ hAdts->num_raw_blocks=config->nSubFrames-1; /* 0 means 1 raw data block */
+
+ FDKcrcInit(&hAdts->crcInfo, 0x8005, 0xFFFF, 16);
+
+ hAdts->currentBlock = 0;
+
+
+ return 0;
+}
+
+int adtsWrite_EncodeHeader(HANDLE_ADTS hAdts,
+ HANDLE_FDK_BITSTREAM hBitStream,
+ int buffer_fullness,
+ int frame_length)
+{
+ INT crcIndex = 0;
+
+
+ hAdts->headerBits = adtsWrite_GetHeaderBits(hAdts);
+
+ FDK_ASSERT(((frame_length+hAdts->headerBits)/8)<0x2000); /*13 bit*/
+ FDK_ASSERT(buffer_fullness<0x800); /* 11 bit */
+
+ if (!hAdts->protection_absent) {
+ FDKcrcReset(&hAdts->crcInfo);
+ }
+
+ if (hAdts->currentBlock == 0) {
+ FDKresetBitbuffer(hBitStream, BS_WRITER);
+ }
+
+ hAdts->subFrameStartBit = FDKgetValidBits(hBitStream);
+
+ /* Skip new header if this is raw data block 1..n */
+ if (hAdts->currentBlock == 0)
+ {
+ FDKresetBitbuffer(hBitStream, BS_WRITER);
+
+ if (hAdts->num_raw_blocks == 0) {
+ crcIndex = adtsWrite_CrcStartReg(hAdts, hBitStream, 0);
+ }
+
+ /* fixed header */
+ FDKwriteBits(hBitStream, 0xFFF, 12);
+ FDKwriteBits(hBitStream, hAdts->mpeg_id, 1);
+ FDKwriteBits(hBitStream, hAdts->layer, 2);
+ FDKwriteBits(hBitStream, hAdts->protection_absent, 1);
+ FDKwriteBits(hBitStream, hAdts->profile, 2);
+ FDKwriteBits(hBitStream, hAdts->sample_freq_index, 4);
+ FDKwriteBits(hBitStream, hAdts->private_bit, 1);
+ FDKwriteBits(hBitStream, getChannelConfig(hAdts->channel_mode), 3);
+ FDKwriteBits(hBitStream, hAdts->original, 1);
+ FDKwriteBits(hBitStream, hAdts->home, 1);
+ /* variable header */
+ FDKwriteBits(hBitStream, hAdts->copyright_id, 1);
+ FDKwriteBits(hBitStream, hAdts->copyright_start, 1);
+ FDKwriteBits(hBitStream, (frame_length + hAdts->headerBits)>>3, 13);
+ FDKwriteBits(hBitStream, buffer_fullness, 11);
+ FDKwriteBits(hBitStream, hAdts->num_raw_blocks, 2);
+
+ if (!hAdts->protection_absent) {
+ int i;
+
+ /* End header CRC portion for single raw data block and write dummy zero values for unknown fields. */
+ if (hAdts->num_raw_blocks == 0) {
+ adtsWrite_CrcEndReg(hAdts, hBitStream, crcIndex);
+ } else {
+ for (i=0; i<hAdts->num_raw_blocks; i++) {
+ FDKwriteBits(hBitStream, 0, 16);
+ }
+ }
+ FDKwriteBits(hBitStream, 0, 16);
+ }
+ } /* End of ADTS header */
+
+ return 0;
+}
+
+void adtsWrite_EndRawDataBlock(HANDLE_ADTS hAdts,
+ HANDLE_FDK_BITSTREAM hBs,
+ int *pBits)
+{
+ if (!hAdts->protection_absent) {
+ FDK_BITSTREAM bsWriter;
+
+ FDKinitBitStream(&bsWriter, hBs->hBitBuf.Buffer, hBs->hBitBuf.bufSize, 0, BS_WRITER);
+ FDKpushFor(&bsWriter, 56);
+
+ if (hAdts->num_raw_blocks == 0) {
+ FDKwriteBits(&bsWriter, FDKcrcGetCRC(&hAdts->crcInfo), 16);
+ } else {
+ int distance;
+
+ /* Write CRC of current raw data block */
+ FDKwriteBits(hBs, FDKcrcGetCRC(&hAdts->crcInfo), 16);
+
+ /* Write distance to current data block */
+ if (hAdts->currentBlock < hAdts->num_raw_blocks) {
+ FDKpushFor(&bsWriter, hAdts->currentBlock*16);
+ distance = FDKgetValidBits(hBs) - (56 + (hAdts->num_raw_blocks)*16 + 16);
+ FDKwriteBits(&bsWriter, distance>>3, 16);
+ }
+ }
+ FDKsyncCache(&bsWriter);
+ }
+
+ /* Write total frame lenth for multiple raw data blocks and header CRC */
+ if (hAdts->num_raw_blocks > 0 && hAdts->currentBlock == hAdts->num_raw_blocks) {
+ FDK_BITSTREAM bsWriter;
+ int crcIndex = 0;
+
+ FDKinitBitStream(&bsWriter, hBs->hBitBuf.Buffer, hBs->hBitBuf.bufSize, 0, BS_WRITER);
+
+ if (!hAdts->protection_absent) {
+ FDKcrcReset(&hAdts->crcInfo);
+ crcIndex = FDKcrcStartReg(&hAdts->crcInfo, &bsWriter, 0);
+ }
+ /* Write total frame length */
+ FDKpushFor(&bsWriter, 56-28+2);
+ FDKwriteBits(&bsWriter, FDKgetValidBits(hBs)>>3, 13);
+
+ /* Write header CRC */
+ if (!hAdts->protection_absent) {
+ FDKpushFor(&bsWriter, 11+2 + (hAdts->num_raw_blocks)*16);
+ FDKcrcEndReg(&hAdts->crcInfo, &bsWriter, crcIndex);
+ FDKwriteBits(&bsWriter, FDKcrcGetCRC(&hAdts->crcInfo), 16);
+ }
+ FDKsyncCache(&bsWriter);
+ }
+
+ /* Correct *pBits to reflect the amount of bits of the current subframe */
+ *pBits -= hAdts->subFrameStartBit;
+ if (!hAdts->protection_absent && hAdts->num_raw_blocks > 0) {
+ /* Fixup CRC bits, since they come after each raw data block */
+ *pBits += 16;
+ }
+ hAdts->currentBlock++;
+}
+