ref: 7f27198c6857096d755b592c7f55f7b1469c50ab
dir: /codec/decoder/core/src/memmgr_nal_unit.cpp/
/*! * \copy * Copyright (c) 2008-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. * * * memmgr_nal_unit.c * * Abstract * memory manager utils for NAL Unit list available * * History * 07/10/2008 Created * *****************************************************************************/ #include "memmgr_nal_unit.h" #include "mem_align.h" namespace WelsDec { int32_t MemInitNalList (PAccessUnit* ppAu, const uint32_t kuiSize) { uint32_t uiIdx = 0; uint8_t* pBase = NULL, *pPtr = NULL; const uint32_t kuiSizeAu = sizeof (SAccessUnit); const uint32_t kuiSizeNalUnitPtr = kuiSize * sizeof (PNalUnit); const uint32_t kuiSizeNalUnit = sizeof (SNalUnit); const uint32_t kuiCountSize = (kuiSizeAu + kuiSizeNalUnitPtr + kuiSize * kuiSizeNalUnit) * sizeof (uint8_t); if (kuiSize == 0) return 1; if (*ppAu != NULL) { MemFreeNalList (ppAu); } pBase = (uint8_t*)WelsMalloc (kuiCountSize, "Access Unit"); if (pBase == NULL) return 1; pPtr = pBase; *ppAu = (PAccessUnit)pPtr; pPtr += kuiSizeAu; (*ppAu)->pNalUnitsList = (PNalUnit*)pPtr; pPtr += kuiSizeNalUnitPtr; do { (*ppAu)->pNalUnitsList[uiIdx] = (PNalUnit)pPtr; pPtr += kuiSizeNalUnit; ++ uiIdx; } while (uiIdx < kuiSize); (*ppAu)->uiCountUnitsNum = kuiSize; (*ppAu)->uiAvailUnitsNum = 0; (*ppAu)->uiActualUnitsNum = 0; (*ppAu)->uiEndPos = 0; (*ppAu)->bCompletedAuFlag = false; return 0; } int32_t MemFreeNalList (PAccessUnit* ppAu) { if (ppAu != NULL) { PAccessUnit pAu = *ppAu; if (pAu != NULL) { WelsFree (pAu, "Access Unit"); *ppAu = NULL; } } return 0; } int32_t ExpandNalUnitList (PAccessUnit* ppAu, const int32_t kiOrgSize, const int32_t kiExpSize) { if (kiExpSize <= kiOrgSize) return 1; else { PAccessUnit pTmp = NULL; int32_t iIdx = 0; if (MemInitNalList (&pTmp, kiExpSize)) // request new list with expanding return 1; do { memcpy (pTmp->pNalUnitsList[iIdx], (*ppAu)->pNalUnitsList[iIdx], sizeof (SNalUnit)); //confirmed_safe_unsafe_usage ++ iIdx; } while (iIdx < kiOrgSize); pTmp->uiCountUnitsNum = kiExpSize; pTmp->uiAvailUnitsNum = (*ppAu)->uiAvailUnitsNum; pTmp->uiActualUnitsNum = (*ppAu)->uiActualUnitsNum; pTmp->uiEndPos = (*ppAu)->uiEndPos; pTmp->bCompletedAuFlag = (*ppAu)->bCompletedAuFlag; MemFreeNalList (ppAu); // free old list *ppAu = pTmp; return 0; } } /* * MemGetNextNal * Get next NAL Unit for using. * Need expand NAL Unit list if exceeding count number of available NAL Units withing an Access Unit */ PNalUnit MemGetNextNal (PAccessUnit* ppAu) { PAccessUnit pAu = *ppAu; PNalUnit pNu = NULL; if (pAu->uiAvailUnitsNum >= pAu->uiCountUnitsNum) { // need expand list const uint32_t kuiExpandingSize = pAu->uiCountUnitsNum + (MAX_NAL_UNIT_NUM_IN_AU >> 1); if (ExpandNalUnitList (ppAu, pAu->uiCountUnitsNum, kuiExpandingSize)) return NULL; // out of memory pAu = *ppAu; } pNu = pAu->pNalUnitsList[pAu->uiAvailUnitsNum++]; // ready for next nal position memset (pNu, 0, sizeof (SNalUnit)); // Please do not remove this for cache intend!! return pNu; } } // namespace WelsDec