ref: 9f20c727d1e5c019269b56fb4d278b9fe182a049
dir: /codec/encoder/core/src/nal_encap.cpp/
/*! * \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© int32_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© 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