shithub: openh264

ref: 063709c92e8bafe467d384f564afbdc6b66eb39a
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 "utils.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