/*! * \copy *     Copyright (c)  2009-2013, Cisco Systems *     All rights reserved. * *     Redistribution and use in source and binary forms, with or without *     modification, are permitted provided that the following conditions *     are met: * *        * Redistributions of source code must retain the above copyright *          notice, this list of conditions and the following disclaimer. * *        * Redistributions in binary form must reproduce the above copyright *          notice, this list of conditions and the following disclaimer in *          the documentation and/or other materials provided with the *          distribution. * *     THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS *     "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT *     LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS *     FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE *     COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, *     INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, *     BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; *     LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER *     CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT *     LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN *     ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE *     POSSIBILITY OF SUCH DAMAGE. * * * \file	nal_encap.c * * \brief	NAL pRawNal pData encapsulation * * \date	5/25/2009	Created * *************************************************************************************/#include "nal_encap.h"#include "svc_enc_golomb.h"#include "ls_defines.h"namespace WelsEnc {/*! * \brief	load an initialize NAL pRawNal pData */void WelsLoadNal (SWelsEncoderOutput* pEncoderOuput, const int32_t/*EWelsNalUnitType*/ kiType,                  const int32_t/*EWelsNalRefIdc*/ kiNalRefIdc) {  SWelsEncoderOutput* pWelsEncoderOuput	= pEncoderOuput;  SWelsNalRaw* pRawNal			= &pWelsEncoderOuput->sNalList[ pWelsEncoderOuput->iNalIndex ];  SNalUnitHeader* sNalHeader	= &pRawNal->sNalExt.sNalHeader;  const int32_t kiStartPos		= (BsGetBitsPos (&pWelsEncoderOuput->sBsWrite) >> 3);   sNalHeader->eNalUnitType	= (EWelsNalUnitType)kiType;  sNalHeader->uiNalRefIdc		= (EWelsNalRefIdc)kiNalRefIdc;  sNalHeader->uiForbiddenZeroBit	= 0;   pRawNal->pRawData		= &pWelsEncoderOuput->pBsBuffer[kiStartPos];  pRawNal->iStartPos	 = kiStartPos;  pRawNal->iPayloadSize	= 0;} /*! * \brief	unload pRawNal NAL */void WelsUnloadNal (SWelsEncoderOutput* pEncoderOuput) {  SWelsEncoderOutput*	pWelsEncoderOuput = pEncoderOuput;  int32_t*	pIdx			= &pWelsEncoderOuput->iNalIndex;  SWelsNalRaw* pRawNal		= &pWelsEncoderOuput->sNalList[ *pIdx ];  const int32_t kiEndPos		= (BsGetBitsPos (&pWelsEncoderOuput->sBsWrite) >> 3);   /* count payload size of pRawNal NAL */  pRawNal->iPayloadSize	= kiEndPos - pRawNal->iStartPos;   ++ (*pIdx);} /*! * \brief	load an initialize NAL pRawNal pData */void WelsLoadNalForSlice (SWelsSliceBs* pSliceBsIn, const int32_t/*EWelsNalUnitType*/ kiType,                          const int32_t/*EWelsNalRefIdc*/ kiNalRefIdc) {  SWelsSliceBs* pSliceBs		    = pSliceBsIn;  SWelsNalRaw* pRawNal		= &pSliceBs->sNalList[ pSliceBs->iNalIndex ];  SNalUnitHeader* sNalHeader	= &pRawNal->sNalExt.sNalHeader;  SBitStringAux* pBitStringAux	= &pSliceBs->sBsWrite;  const int32_t kiStartPos		    = (BsGetBitsPos (pBitStringAux) >> 3);   sNalHeader->eNalUnitType	= (EWelsNalUnitType)kiType;  sNalHeader->uiNalRefIdc		= (EWelsNalRefIdc)kiNalRefIdc;  sNalHeader->uiForbiddenZeroBit	= 0;   pRawNal->pRawData		= &pSliceBs->pBsBuffer[kiStartPos];  pRawNal->iStartPos	 = kiStartPos;  pRawNal->iPayloadSize	= 0;} /*! * \brief	unload pRawNal NAL */void WelsUnloadNalForSlice (SWelsSliceBs* pSliceBsIn) {  SWelsSliceBs* pSliceBs	        = pSliceBsIn;  int32_t*	pIdx			            = &pSliceBs->iNalIndex;  SWelsNalRaw* pRawNal		= &pSliceBs->sNalList[ *pIdx ];  SBitStringAux* pBitStringAux	= &pSliceBs->sBsWrite;  const int32_t kiEndPos		        = (BsGetBitsPos (pBitStringAux) >> 3);   /* count payload size of pRawNal NAL */  pRawNal->iPayloadSize	= kiEndPos - pRawNal->iStartPos;   ++ (*pIdx);} /*! * \brief	encode NAL with emulation forbidden three bytes checking * \param	pDst			pDst NAL pData * \param	pDstLen		length of pDst NAL output * \param	annexeb		annexeb flag * \param	pRawNal			pRawNal NAL pData * \return	ERRCODE *///TODO 1: refactor the calling of this func in multi-thread//TODO 2: complete the realloc&copyint32_t WelsEncodeNal (SWelsNalRaw* pRawNal, void* pNalHeaderExt, const int32_t kiDstBufferLen, void* pDst,                       int32_t* pDstLen) {  const bool kbNALExt = pRawNal->sNalExt.sNalHeader.eNalUnitType == NAL_UNIT_PREFIX                        || pRawNal->sNalExt.sNalHeader.eNalUnitType == NAL_UNIT_CODED_SLICE_EXT;  int32_t iAssumedNeededLength		= NAL_HEADER_SIZE + (kbNALExt ? 3 : 0) + pRawNal->iPayloadSize + 1;  WELS_VERIFY_RETURN_IF (ENC_RETURN_UNEXPECTED, (iAssumedNeededLength <= 0))   //since for each 0x000 need a 0x03, so the needed length will not exceed (iAssumeNeedLenth + iAssumeNeedLength/3), here adjust to >>1 to omit division  if (kiDstBufferLen < (iAssumedNeededLength + (iAssumedNeededLength >> 1))) {    return ENC_RETURN_MEMALLOCERR;    //TODO: call the realloc&copy instead  }  uint8_t* pDstStart	    = (uint8_t*)pDst;  uint8_t* pDstPointer	= pDstStart;  uint8_t* pSrcPointer	= pRawNal->pRawData;  uint8_t* pSrcEnd		= pRawNal->pRawData + pRawNal->iPayloadSize;  int32_t iZeroCount		= 0;  int32_t iNalLength		= 0;  *pDstLen = 0;   static const uint8_t kuiStartCodePrefix[NAL_HEADER_SIZE] = { 0, 0, 0, 1 };  ST32 (pDstPointer, LD32 (&kuiStartCodePrefix[0]));  pDstPointer += 4;   /* NAL Unit Header */  *pDstPointer++	= (pRawNal->sNalExt.sNalHeader.uiNalRefIdc << 5) | (pRawNal->sNalExt.sNalHeader.eNalUnitType & 0x1f);   if (kbNALExt) {    SNalUnitHeaderExt* sNalExt	= (SNalUnitHeaderExt*)pNalHeaderExt;     /* NAL UNIT Extension Header */    *pDstPointer++ =	(0x80) |                      (sNalExt->bIdrFlag << 6);     *pDstPointer++ =	(0x80) |                      (sNalExt->uiDependencyId << 4);     *pDstPointer++ =	(sNalExt->uiTemporalId << 5) |                      (sNalExt->bDiscardableFlag << 3) |                      (0x07);  }   while (pSrcPointer < pSrcEnd) {    if (iZeroCount == 2 && *pSrcPointer <= 3) {      //add the code 03      *pDstPointer++	= 3;      iZeroCount		= 0;    }    if (*pSrcPointer == 0) {      ++ iZeroCount;    } else {      iZeroCount		= 0;    }    *pDstPointer++ = *pSrcPointer++;  }   /* count length of NAL Unit */  iNalLength	= (int32_t) (pDstPointer - pDstStart);  if (NULL != pDstLen)    *pDstLen	= iNalLength;   return ENC_RETURN_SUCCESS;} /*! * \brief	write prefix nal */int32_t WelsWriteSVCPrefixNal (SBitStringAux* pBitStringAux, const int32_t kiNalRefIdc,                               const bool kbIdrFlag) {  if (0 < kiNalRefIdc) {    BsWriteOneBit (pBitStringAux, false/*bStoreRefBasePicFlag*/);    BsWriteOneBit (pBitStringAux, false);    BsRbspTrailingBits (pBitStringAux);  }  return 0;} } // namespace WelsEnc