10c1bc742181ded4930842b46e9507372f0b1b963James Dong/* 20c1bc742181ded4930842b46e9507372f0b1b963James Dong * Copyright (C) 2009 The Android Open Source Project 30c1bc742181ded4930842b46e9507372f0b1b963James Dong * 40c1bc742181ded4930842b46e9507372f0b1b963James Dong * Licensed under the Apache License, Version 2.0 (the "License"); 50c1bc742181ded4930842b46e9507372f0b1b963James Dong * you may not use this file except in compliance with the License. 60c1bc742181ded4930842b46e9507372f0b1b963James Dong * You may obtain a copy of the License at 70c1bc742181ded4930842b46e9507372f0b1b963James Dong * 80c1bc742181ded4930842b46e9507372f0b1b963James Dong * http://www.apache.org/licenses/LICENSE-2.0 90c1bc742181ded4930842b46e9507372f0b1b963James Dong * 100c1bc742181ded4930842b46e9507372f0b1b963James Dong * Unless required by applicable law or agreed to in writing, software 110c1bc742181ded4930842b46e9507372f0b1b963James Dong * distributed under the License is distributed on an "AS IS" BASIS, 120c1bc742181ded4930842b46e9507372f0b1b963James Dong * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 130c1bc742181ded4930842b46e9507372f0b1b963James Dong * See the License for the specific language governing permissions and 140c1bc742181ded4930842b46e9507372f0b1b963James Dong * limitations under the License. 150c1bc742181ded4930842b46e9507372f0b1b963James Dong */ 160c1bc742181ded4930842b46e9507372f0b1b963James Dong 170c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------ 180c1bc742181ded4930842b46e9507372f0b1b963James Dong 190c1bc742181ded4930842b46e9507372f0b1b963James Dong Table of contents 200c1bc742181ded4930842b46e9507372f0b1b963James Dong 210c1bc742181ded4930842b46e9507372f0b1b963James Dong 1. Include headers 220c1bc742181ded4930842b46e9507372f0b1b963James Dong 2. External compiler flags 230c1bc742181ded4930842b46e9507372f0b1b963James Dong 3. Module defines 240c1bc742181ded4930842b46e9507372f0b1b963James Dong 4. Local function prototypes 250c1bc742181ded4930842b46e9507372f0b1b963James Dong 5. Functions 260c1bc742181ded4930842b46e9507372f0b1b963James Dong h264bsdInitStorage 270c1bc742181ded4930842b46e9507372f0b1b963James Dong h264bsdStoreSeqParamSet 280c1bc742181ded4930842b46e9507372f0b1b963James Dong h264bsdStorePicParamSet 290c1bc742181ded4930842b46e9507372f0b1b963James Dong h264bsdActivateParamSets 300c1bc742181ded4930842b46e9507372f0b1b963James Dong h264bsdResetStorage 310c1bc742181ded4930842b46e9507372f0b1b963James Dong h264bsdIsStartOfPicture 320c1bc742181ded4930842b46e9507372f0b1b963James Dong h264bsdIsEndOfPicture 330c1bc742181ded4930842b46e9507372f0b1b963James Dong h264bsdComputeSliceGroupMap 340c1bc742181ded4930842b46e9507372f0b1b963James Dong h264bsdCheckAccessUnitBoundary 350c1bc742181ded4930842b46e9507372f0b1b963James Dong CheckPps 360c1bc742181ded4930842b46e9507372f0b1b963James Dong h264bsdValidParamSets 370c1bc742181ded4930842b46e9507372f0b1b963James Dong 380c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/ 390c1bc742181ded4930842b46e9507372f0b1b963James Dong 400c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------ 410c1bc742181ded4930842b46e9507372f0b1b963James Dong 1. Include headers 420c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/ 430c1bc742181ded4930842b46e9507372f0b1b963James Dong 440c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_storage.h" 450c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_util.h" 460c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_neighbour.h" 470c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_slice_group_map.h" 480c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_dpb.h" 490c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_nal_unit.h" 500c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_slice_header.h" 510c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_seq_param_set.h" 520c1bc742181ded4930842b46e9507372f0b1b963James Dong 530c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------ 540c1bc742181ded4930842b46e9507372f0b1b963James Dong 2. External compiler flags 550c1bc742181ded4930842b46e9507372f0b1b963James Dong-------------------------------------------------------------------------------- 560c1bc742181ded4930842b46e9507372f0b1b963James Dong 570c1bc742181ded4930842b46e9507372f0b1b963James Dong-------------------------------------------------------------------------------- 580c1bc742181ded4930842b46e9507372f0b1b963James Dong 3. Module defines 590c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/ 600c1bc742181ded4930842b46e9507372f0b1b963James Dong 6187277aac64a21461b657298dcdcb809737347980Pawin Vongmasa#ifndef UINT32_MAX 6287277aac64a21461b657298dcdcb809737347980Pawin Vongmasa#define UINT32_MAX (4294967295U) 6387277aac64a21461b657298dcdcb809737347980Pawin Vongmasa#endif 6487277aac64a21461b657298dcdcb809737347980Pawin Vongmasa 650c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------ 660c1bc742181ded4930842b46e9507372f0b1b963James Dong 4. Local function prototypes 670c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/ 680c1bc742181ded4930842b46e9507372f0b1b963James Dong 690c1bc742181ded4930842b46e9507372f0b1b963James Dongstatic u32 CheckPps(picParamSet_t *pps, seqParamSet_t *sps); 700c1bc742181ded4930842b46e9507372f0b1b963James Dong 710c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------ 720c1bc742181ded4930842b46e9507372f0b1b963James Dong 730c1bc742181ded4930842b46e9507372f0b1b963James Dong Function name: h264bsdInitStorage 740c1bc742181ded4930842b46e9507372f0b1b963James Dong 750c1bc742181ded4930842b46e9507372f0b1b963James Dong Functional description: 760c1bc742181ded4930842b46e9507372f0b1b963James Dong Initialize storage structure. Sets contents of the storage to '0' 770c1bc742181ded4930842b46e9507372f0b1b963James Dong except for the active parameter set ids, which are initialized 780c1bc742181ded4930842b46e9507372f0b1b963James Dong to invalid values. 790c1bc742181ded4930842b46e9507372f0b1b963James Dong 800c1bc742181ded4930842b46e9507372f0b1b963James Dong Inputs: 810c1bc742181ded4930842b46e9507372f0b1b963James Dong 820c1bc742181ded4930842b46e9507372f0b1b963James Dong Outputs: 830c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage initialized data stored here 840c1bc742181ded4930842b46e9507372f0b1b963James Dong 850c1bc742181ded4930842b46e9507372f0b1b963James Dong Returns: 860c1bc742181ded4930842b46e9507372f0b1b963James Dong none 870c1bc742181ded4930842b46e9507372f0b1b963James Dong 880c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/ 890c1bc742181ded4930842b46e9507372f0b1b963James Dong 900c1bc742181ded4930842b46e9507372f0b1b963James Dongvoid h264bsdInitStorage(storage_t *pStorage) 910c1bc742181ded4930842b46e9507372f0b1b963James Dong{ 920c1bc742181ded4930842b46e9507372f0b1b963James Dong 930c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */ 940c1bc742181ded4930842b46e9507372f0b1b963James Dong 950c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */ 960c1bc742181ded4930842b46e9507372f0b1b963James Dong 970c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(pStorage); 980c1bc742181ded4930842b46e9507372f0b1b963James Dong 990c1bc742181ded4930842b46e9507372f0b1b963James Dong H264SwDecMemset(pStorage, 0, sizeof(storage_t)); 1000c1bc742181ded4930842b46e9507372f0b1b963James Dong 1010c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSpsId = MAX_NUM_SEQ_PARAM_SETS; 1020c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activePpsId = MAX_NUM_PIC_PARAM_SETS; 1030c1bc742181ded4930842b46e9507372f0b1b963James Dong 1040c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->aub->firstCallFlag = HANTRO_TRUE; 1050c1bc742181ded4930842b46e9507372f0b1b963James Dong} 1060c1bc742181ded4930842b46e9507372f0b1b963James Dong 1070c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------ 1080c1bc742181ded4930842b46e9507372f0b1b963James Dong 1090c1bc742181ded4930842b46e9507372f0b1b963James Dong Function: h264bsdStoreSeqParamSet 1100c1bc742181ded4930842b46e9507372f0b1b963James Dong 1110c1bc742181ded4930842b46e9507372f0b1b963James Dong Functional description: 1120c1bc742181ded4930842b46e9507372f0b1b963James Dong Store sequence parameter set into the storage. If active SPS is 1130c1bc742181ded4930842b46e9507372f0b1b963James Dong overwritten -> check if contents changes and if it does, set 1140c1bc742181ded4930842b46e9507372f0b1b963James Dong parameters to force reactivation of parameter sets 1150c1bc742181ded4930842b46e9507372f0b1b963James Dong 1160c1bc742181ded4930842b46e9507372f0b1b963James Dong Inputs: 1170c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage pointer to storage structure 1180c1bc742181ded4930842b46e9507372f0b1b963James Dong pSeqParamSet pointer to param set to be stored 1190c1bc742181ded4930842b46e9507372f0b1b963James Dong 1200c1bc742181ded4930842b46e9507372f0b1b963James Dong Outputs: 1210c1bc742181ded4930842b46e9507372f0b1b963James Dong none 1220c1bc742181ded4930842b46e9507372f0b1b963James Dong 1230c1bc742181ded4930842b46e9507372f0b1b963James Dong Returns: 1240c1bc742181ded4930842b46e9507372f0b1b963James Dong HANTRO_OK success 1250c1bc742181ded4930842b46e9507372f0b1b963James Dong MEMORY_ALLOCATION_ERROR failure in memory allocation 1260c1bc742181ded4930842b46e9507372f0b1b963James Dong 1270c1bc742181ded4930842b46e9507372f0b1b963James Dong 1280c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/ 1290c1bc742181ded4930842b46e9507372f0b1b963James Dong 1300c1bc742181ded4930842b46e9507372f0b1b963James Dongu32 h264bsdStoreSeqParamSet(storage_t *pStorage, seqParamSet_t *pSeqParamSet) 1310c1bc742181ded4930842b46e9507372f0b1b963James Dong{ 1320c1bc742181ded4930842b46e9507372f0b1b963James Dong 1330c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */ 1340c1bc742181ded4930842b46e9507372f0b1b963James Dong 1350c1bc742181ded4930842b46e9507372f0b1b963James Dong u32 id; 1360c1bc742181ded4930842b46e9507372f0b1b963James Dong 1370c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */ 1380c1bc742181ded4930842b46e9507372f0b1b963James Dong 1390c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(pStorage); 1400c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(pSeqParamSet); 1410c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(pSeqParamSet->seqParameterSetId < MAX_NUM_SEQ_PARAM_SETS); 1420c1bc742181ded4930842b46e9507372f0b1b963James Dong 1430c1bc742181ded4930842b46e9507372f0b1b963James Dong id = pSeqParamSet->seqParameterSetId; 1440c1bc742181ded4930842b46e9507372f0b1b963James Dong 1450c1bc742181ded4930842b46e9507372f0b1b963James Dong /* seq parameter set with id not used before -> allocate memory */ 1460c1bc742181ded4930842b46e9507372f0b1b963James Dong if (pStorage->sps[id] == NULL) 1470c1bc742181ded4930842b46e9507372f0b1b963James Dong { 1480c1bc742181ded4930842b46e9507372f0b1b963James Dong ALLOCATE(pStorage->sps[id], 1, seqParamSet_t); 1490c1bc742181ded4930842b46e9507372f0b1b963James Dong if (pStorage->sps[id] == NULL) 1500c1bc742181ded4930842b46e9507372f0b1b963James Dong return(MEMORY_ALLOCATION_ERROR); 1510c1bc742181ded4930842b46e9507372f0b1b963James Dong } 1520c1bc742181ded4930842b46e9507372f0b1b963James Dong /* sequence parameter set with id equal to id of active sps */ 1530c1bc742181ded4930842b46e9507372f0b1b963James Dong else if (id == pStorage->activeSpsId) 1540c1bc742181ded4930842b46e9507372f0b1b963James Dong { 1550c1bc742181ded4930842b46e9507372f0b1b963James Dong /* if seq parameter set contents changes 1560c1bc742181ded4930842b46e9507372f0b1b963James Dong * -> overwrite and re-activate when next IDR picture decoded 1570c1bc742181ded4930842b46e9507372f0b1b963James Dong * ids of active param sets set to invalid values to force 1580c1bc742181ded4930842b46e9507372f0b1b963James Dong * re-activation. Memories allocated for old sps freed 1590c1bc742181ded4930842b46e9507372f0b1b963James Dong * otherwise free memeries allocated for just decoded sps and 1600c1bc742181ded4930842b46e9507372f0b1b963James Dong * continue */ 1610c1bc742181ded4930842b46e9507372f0b1b963James Dong if (h264bsdCompareSeqParamSets(pSeqParamSet, pStorage->activeSps) != 0) 1620c1bc742181ded4930842b46e9507372f0b1b963James Dong { 1630c1bc742181ded4930842b46e9507372f0b1b963James Dong FREE(pStorage->sps[id]->offsetForRefFrame); 1640c1bc742181ded4930842b46e9507372f0b1b963James Dong FREE(pStorage->sps[id]->vuiParameters); 1650c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSpsId = MAX_NUM_SEQ_PARAM_SETS + 1; 1660c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activePpsId = MAX_NUM_PIC_PARAM_SETS + 1; 1670c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSps = NULL; 1680c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activePps = NULL; 1690c1bc742181ded4930842b46e9507372f0b1b963James Dong } 1700c1bc742181ded4930842b46e9507372f0b1b963James Dong else 1710c1bc742181ded4930842b46e9507372f0b1b963James Dong { 1720c1bc742181ded4930842b46e9507372f0b1b963James Dong FREE(pSeqParamSet->offsetForRefFrame); 1730c1bc742181ded4930842b46e9507372f0b1b963James Dong FREE(pSeqParamSet->vuiParameters); 1740c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_OK); 1750c1bc742181ded4930842b46e9507372f0b1b963James Dong } 1760c1bc742181ded4930842b46e9507372f0b1b963James Dong } 1770c1bc742181ded4930842b46e9507372f0b1b963James Dong /* overwrite seq param set other than active one -> free memories 1780c1bc742181ded4930842b46e9507372f0b1b963James Dong * allocated for old param set */ 1790c1bc742181ded4930842b46e9507372f0b1b963James Dong else 1800c1bc742181ded4930842b46e9507372f0b1b963James Dong { 1810c1bc742181ded4930842b46e9507372f0b1b963James Dong FREE(pStorage->sps[id]->offsetForRefFrame); 1820c1bc742181ded4930842b46e9507372f0b1b963James Dong FREE(pStorage->sps[id]->vuiParameters); 1830c1bc742181ded4930842b46e9507372f0b1b963James Dong } 1840c1bc742181ded4930842b46e9507372f0b1b963James Dong 1850c1bc742181ded4930842b46e9507372f0b1b963James Dong *pStorage->sps[id] = *pSeqParamSet; 1860c1bc742181ded4930842b46e9507372f0b1b963James Dong 1870c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_OK); 1880c1bc742181ded4930842b46e9507372f0b1b963James Dong 1890c1bc742181ded4930842b46e9507372f0b1b963James Dong} 1900c1bc742181ded4930842b46e9507372f0b1b963James Dong 1910c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------ 1920c1bc742181ded4930842b46e9507372f0b1b963James Dong 1930c1bc742181ded4930842b46e9507372f0b1b963James Dong Function: h264bsdStorePicParamSet 1940c1bc742181ded4930842b46e9507372f0b1b963James Dong 1950c1bc742181ded4930842b46e9507372f0b1b963James Dong Functional description: 1960c1bc742181ded4930842b46e9507372f0b1b963James Dong Store picture parameter set into the storage. If active PPS is 1970c1bc742181ded4930842b46e9507372f0b1b963James Dong overwritten -> check if active SPS changes and if it does -> set 1980c1bc742181ded4930842b46e9507372f0b1b963James Dong parameters to force reactivation of parameter sets 1990c1bc742181ded4930842b46e9507372f0b1b963James Dong 2000c1bc742181ded4930842b46e9507372f0b1b963James Dong Inputs: 2010c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage pointer to storage structure 2020c1bc742181ded4930842b46e9507372f0b1b963James Dong pPicParamSet pointer to param set to be stored 2030c1bc742181ded4930842b46e9507372f0b1b963James Dong 2040c1bc742181ded4930842b46e9507372f0b1b963James Dong Outputs: 2050c1bc742181ded4930842b46e9507372f0b1b963James Dong none 2060c1bc742181ded4930842b46e9507372f0b1b963James Dong 2070c1bc742181ded4930842b46e9507372f0b1b963James Dong Returns: 2080c1bc742181ded4930842b46e9507372f0b1b963James Dong HANTRO_OK success 2090c1bc742181ded4930842b46e9507372f0b1b963James Dong MEMORY_ALLOCATION_ERROR failure in memory allocation 2100c1bc742181ded4930842b46e9507372f0b1b963James Dong 2110c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/ 2120c1bc742181ded4930842b46e9507372f0b1b963James Dong 2130c1bc742181ded4930842b46e9507372f0b1b963James Dongu32 h264bsdStorePicParamSet(storage_t *pStorage, picParamSet_t *pPicParamSet) 2140c1bc742181ded4930842b46e9507372f0b1b963James Dong{ 2150c1bc742181ded4930842b46e9507372f0b1b963James Dong 2160c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */ 2170c1bc742181ded4930842b46e9507372f0b1b963James Dong 2180c1bc742181ded4930842b46e9507372f0b1b963James Dong u32 id; 2190c1bc742181ded4930842b46e9507372f0b1b963James Dong 2200c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */ 2210c1bc742181ded4930842b46e9507372f0b1b963James Dong 2220c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(pStorage); 2230c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(pPicParamSet); 2240c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(pPicParamSet->picParameterSetId < MAX_NUM_PIC_PARAM_SETS); 2250c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(pPicParamSet->seqParameterSetId < MAX_NUM_SEQ_PARAM_SETS); 2260c1bc742181ded4930842b46e9507372f0b1b963James Dong 2270c1bc742181ded4930842b46e9507372f0b1b963James Dong id = pPicParamSet->picParameterSetId; 2280c1bc742181ded4930842b46e9507372f0b1b963James Dong 2290c1bc742181ded4930842b46e9507372f0b1b963James Dong /* pic parameter set with id not used before -> allocate memory */ 2300c1bc742181ded4930842b46e9507372f0b1b963James Dong if (pStorage->pps[id] == NULL) 2310c1bc742181ded4930842b46e9507372f0b1b963James Dong { 2320c1bc742181ded4930842b46e9507372f0b1b963James Dong ALLOCATE(pStorage->pps[id], 1, picParamSet_t); 2330c1bc742181ded4930842b46e9507372f0b1b963James Dong if (pStorage->pps[id] == NULL) 2340c1bc742181ded4930842b46e9507372f0b1b963James Dong return(MEMORY_ALLOCATION_ERROR); 2350c1bc742181ded4930842b46e9507372f0b1b963James Dong } 2360c1bc742181ded4930842b46e9507372f0b1b963James Dong /* picture parameter set with id equal to id of active pps */ 2370c1bc742181ded4930842b46e9507372f0b1b963James Dong else if (id == pStorage->activePpsId) 2380c1bc742181ded4930842b46e9507372f0b1b963James Dong { 2390c1bc742181ded4930842b46e9507372f0b1b963James Dong /* check whether seq param set changes, force re-activation of 2400c1bc742181ded4930842b46e9507372f0b1b963James Dong * param set if it does. Set activeSpsId to invalid value to 2410c1bc742181ded4930842b46e9507372f0b1b963James Dong * accomplish this */ 2420c1bc742181ded4930842b46e9507372f0b1b963James Dong if (pPicParamSet->seqParameterSetId != pStorage->activeSpsId) 2430c1bc742181ded4930842b46e9507372f0b1b963James Dong { 2440c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activePpsId = MAX_NUM_PIC_PARAM_SETS + 1; 2450c1bc742181ded4930842b46e9507372f0b1b963James Dong } 2460c1bc742181ded4930842b46e9507372f0b1b963James Dong /* free memories allocated for old param set */ 2470c1bc742181ded4930842b46e9507372f0b1b963James Dong FREE(pStorage->pps[id]->runLength); 2480c1bc742181ded4930842b46e9507372f0b1b963James Dong FREE(pStorage->pps[id]->topLeft); 2490c1bc742181ded4930842b46e9507372f0b1b963James Dong FREE(pStorage->pps[id]->bottomRight); 2500c1bc742181ded4930842b46e9507372f0b1b963James Dong FREE(pStorage->pps[id]->sliceGroupId); 2510c1bc742181ded4930842b46e9507372f0b1b963James Dong } 2520c1bc742181ded4930842b46e9507372f0b1b963James Dong /* overwrite pic param set other than active one -> free memories 2530c1bc742181ded4930842b46e9507372f0b1b963James Dong * allocated for old param set */ 2540c1bc742181ded4930842b46e9507372f0b1b963James Dong else 2550c1bc742181ded4930842b46e9507372f0b1b963James Dong { 2560c1bc742181ded4930842b46e9507372f0b1b963James Dong FREE(pStorage->pps[id]->runLength); 2570c1bc742181ded4930842b46e9507372f0b1b963James Dong FREE(pStorage->pps[id]->topLeft); 2580c1bc742181ded4930842b46e9507372f0b1b963James Dong FREE(pStorage->pps[id]->bottomRight); 2590c1bc742181ded4930842b46e9507372f0b1b963James Dong FREE(pStorage->pps[id]->sliceGroupId); 2600c1bc742181ded4930842b46e9507372f0b1b963James Dong } 2610c1bc742181ded4930842b46e9507372f0b1b963James Dong 2620c1bc742181ded4930842b46e9507372f0b1b963James Dong *pStorage->pps[id] = *pPicParamSet; 2630c1bc742181ded4930842b46e9507372f0b1b963James Dong 2640c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_OK); 2650c1bc742181ded4930842b46e9507372f0b1b963James Dong 2660c1bc742181ded4930842b46e9507372f0b1b963James Dong} 2670c1bc742181ded4930842b46e9507372f0b1b963James Dong 2680c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------ 2690c1bc742181ded4930842b46e9507372f0b1b963James Dong 2700c1bc742181ded4930842b46e9507372f0b1b963James Dong Function: h264bsdActivateParamSets 2710c1bc742181ded4930842b46e9507372f0b1b963James Dong 2720c1bc742181ded4930842b46e9507372f0b1b963James Dong Functional description: 2730c1bc742181ded4930842b46e9507372f0b1b963James Dong Activate certain SPS/PPS combination. This function shall be 2740c1bc742181ded4930842b46e9507372f0b1b963James Dong called in the beginning of each picture. Picture parameter set 2750c1bc742181ded4930842b46e9507372f0b1b963James Dong can be changed as wanted, but sequence parameter set may only be 2760c1bc742181ded4930842b46e9507372f0b1b963James Dong changed when the starting picture is an IDR picture. 2770c1bc742181ded4930842b46e9507372f0b1b963James Dong 2780c1bc742181ded4930842b46e9507372f0b1b963James Dong When new SPS is activated the function allocates memory for 2790c1bc742181ded4930842b46e9507372f0b1b963James Dong macroblock storages and slice group map and (re-)initializes the 2800c1bc742181ded4930842b46e9507372f0b1b963James Dong decoded picture buffer. If this is not the first activation the old 2810c1bc742181ded4930842b46e9507372f0b1b963James Dong allocations are freed and FreeDpb called before new allocations. 2820c1bc742181ded4930842b46e9507372f0b1b963James Dong 2830c1bc742181ded4930842b46e9507372f0b1b963James Dong Inputs: 2840c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage pointer to storage data structure 2850c1bc742181ded4930842b46e9507372f0b1b963James Dong ppsId identifies the PPS to be activated, SPS id obtained 2860c1bc742181ded4930842b46e9507372f0b1b963James Dong from the PPS 2870c1bc742181ded4930842b46e9507372f0b1b963James Dong isIdr flag to indicate if the picture is an IDR picture 2880c1bc742181ded4930842b46e9507372f0b1b963James Dong 2890c1bc742181ded4930842b46e9507372f0b1b963James Dong Outputs: 2900c1bc742181ded4930842b46e9507372f0b1b963James Dong none 2910c1bc742181ded4930842b46e9507372f0b1b963James Dong 2920c1bc742181ded4930842b46e9507372f0b1b963James Dong Returns: 2930c1bc742181ded4930842b46e9507372f0b1b963James Dong HANTRO_OK success 2940c1bc742181ded4930842b46e9507372f0b1b963James Dong HANTRO_NOK non-existing or invalid param set combination, 2950c1bc742181ded4930842b46e9507372f0b1b963James Dong trying to change SPS with non-IDR picture 2960c1bc742181ded4930842b46e9507372f0b1b963James Dong MEMORY_ALLOCATION_ERROR failure in memory allocation 2970c1bc742181ded4930842b46e9507372f0b1b963James Dong 2980c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/ 2990c1bc742181ded4930842b46e9507372f0b1b963James Dong 3000c1bc742181ded4930842b46e9507372f0b1b963James Dongu32 h264bsdActivateParamSets(storage_t *pStorage, u32 ppsId, u32 isIdr) 3010c1bc742181ded4930842b46e9507372f0b1b963James Dong{ 3020c1bc742181ded4930842b46e9507372f0b1b963James Dong 3030c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */ 3040c1bc742181ded4930842b46e9507372f0b1b963James Dong 3050c1bc742181ded4930842b46e9507372f0b1b963James Dong u32 tmp; 3060c1bc742181ded4930842b46e9507372f0b1b963James Dong u32 flag; 3070c1bc742181ded4930842b46e9507372f0b1b963James Dong 3080c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */ 3090c1bc742181ded4930842b46e9507372f0b1b963James Dong 3100c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(pStorage); 3110c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(ppsId < MAX_NUM_PIC_PARAM_SETS); 3120c1bc742181ded4930842b46e9507372f0b1b963James Dong 3130c1bc742181ded4930842b46e9507372f0b1b963James Dong /* check that pps and corresponding sps exist */ 3140c1bc742181ded4930842b46e9507372f0b1b963James Dong if ( (pStorage->pps[ppsId] == NULL) || 3150c1bc742181ded4930842b46e9507372f0b1b963James Dong (pStorage->sps[pStorage->pps[ppsId]->seqParameterSetId] == NULL) ) 3160c1bc742181ded4930842b46e9507372f0b1b963James Dong { 3170c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_NOK); 3180c1bc742181ded4930842b46e9507372f0b1b963James Dong } 3190c1bc742181ded4930842b46e9507372f0b1b963James Dong 3200c1bc742181ded4930842b46e9507372f0b1b963James Dong /* check that pps parameters do not violate picture size constraints */ 3210c1bc742181ded4930842b46e9507372f0b1b963James Dong tmp = CheckPps(pStorage->pps[ppsId], 3220c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->sps[pStorage->pps[ppsId]->seqParameterSetId]); 3230c1bc742181ded4930842b46e9507372f0b1b963James Dong if (tmp != HANTRO_OK) 3240c1bc742181ded4930842b46e9507372f0b1b963James Dong return(tmp); 3250c1bc742181ded4930842b46e9507372f0b1b963James Dong 3260c1bc742181ded4930842b46e9507372f0b1b963James Dong /* first activation part1 */ 3270c1bc742181ded4930842b46e9507372f0b1b963James Dong if (pStorage->activePpsId == MAX_NUM_PIC_PARAM_SETS) 3280c1bc742181ded4930842b46e9507372f0b1b963James Dong { 3290c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activePpsId = ppsId; 3300c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activePps = pStorage->pps[ppsId]; 3310c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSpsId = pStorage->activePps->seqParameterSetId; 3320c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSps = pStorage->sps[pStorage->activeSpsId]; 33387277aac64a21461b657298dcdcb809737347980Pawin Vongmasa 33487277aac64a21461b657298dcdcb809737347980Pawin Vongmasa /* report error before multiplication to prevent integer overflow */ 33587277aac64a21461b657298dcdcb809737347980Pawin Vongmasa if (pStorage->activeSps->picWidthInMbs == 0) 33687277aac64a21461b657298dcdcb809737347980Pawin Vongmasa { 33787277aac64a21461b657298dcdcb809737347980Pawin Vongmasa pStorage->picSizeInMbs = 0; 33887277aac64a21461b657298dcdcb809737347980Pawin Vongmasa } 33987277aac64a21461b657298dcdcb809737347980Pawin Vongmasa else if (pStorage->activeSps->picHeightInMbs > 34087277aac64a21461b657298dcdcb809737347980Pawin Vongmasa UINT32_MAX / pStorage->activeSps->picWidthInMbs) 34187277aac64a21461b657298dcdcb809737347980Pawin Vongmasa { 34287277aac64a21461b657298dcdcb809737347980Pawin Vongmasa return(MEMORY_ALLOCATION_ERROR); 34387277aac64a21461b657298dcdcb809737347980Pawin Vongmasa } 34487277aac64a21461b657298dcdcb809737347980Pawin Vongmasa else 34587277aac64a21461b657298dcdcb809737347980Pawin Vongmasa { 34687277aac64a21461b657298dcdcb809737347980Pawin Vongmasa pStorage->picSizeInMbs = 34787277aac64a21461b657298dcdcb809737347980Pawin Vongmasa pStorage->activeSps->picWidthInMbs * 34887277aac64a21461b657298dcdcb809737347980Pawin Vongmasa pStorage->activeSps->picHeightInMbs; 34987277aac64a21461b657298dcdcb809737347980Pawin Vongmasa } 3500c1bc742181ded4930842b46e9507372f0b1b963James Dong 3510c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->currImage->width = pStorage->activeSps->picWidthInMbs; 3520c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->currImage->height = pStorage->activeSps->picHeightInMbs; 3530c1bc742181ded4930842b46e9507372f0b1b963James Dong 3540c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->pendingActivation = HANTRO_TRUE; 3550c1bc742181ded4930842b46e9507372f0b1b963James Dong } 3560c1bc742181ded4930842b46e9507372f0b1b963James Dong /* first activation part2 */ 3570c1bc742181ded4930842b46e9507372f0b1b963James Dong else if (pStorage->pendingActivation) 3580c1bc742181ded4930842b46e9507372f0b1b963James Dong { 3590c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->pendingActivation = HANTRO_FALSE; 3600c1bc742181ded4930842b46e9507372f0b1b963James Dong 3610c1bc742181ded4930842b46e9507372f0b1b963James Dong FREE(pStorage->mb); 3620c1bc742181ded4930842b46e9507372f0b1b963James Dong FREE(pStorage->sliceGroupMap); 3630c1bc742181ded4930842b46e9507372f0b1b963James Dong 3640c1bc742181ded4930842b46e9507372f0b1b963James Dong ALLOCATE(pStorage->mb, pStorage->picSizeInMbs, mbStorage_t); 3650c1bc742181ded4930842b46e9507372f0b1b963James Dong ALLOCATE(pStorage->sliceGroupMap, pStorage->picSizeInMbs, u32); 3660c1bc742181ded4930842b46e9507372f0b1b963James Dong if (pStorage->mb == NULL || pStorage->sliceGroupMap == NULL) 3670c1bc742181ded4930842b46e9507372f0b1b963James Dong return(MEMORY_ALLOCATION_ERROR); 3680c1bc742181ded4930842b46e9507372f0b1b963James Dong 3690c1bc742181ded4930842b46e9507372f0b1b963James Dong H264SwDecMemset(pStorage->mb, 0, 3700c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->picSizeInMbs * sizeof(mbStorage_t)); 3710c1bc742181ded4930842b46e9507372f0b1b963James Dong 3720c1bc742181ded4930842b46e9507372f0b1b963James Dong h264bsdInitMbNeighbours(pStorage->mb, 3730c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSps->picWidthInMbs, 3740c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->picSizeInMbs); 3750c1bc742181ded4930842b46e9507372f0b1b963James Dong 3760c1bc742181ded4930842b46e9507372f0b1b963James Dong /* dpb output reordering disabled if 3770c1bc742181ded4930842b46e9507372f0b1b963James Dong * 1) application set noReordering flag 3780c1bc742181ded4930842b46e9507372f0b1b963James Dong * 2) POC type equal to 2 3790c1bc742181ded4930842b46e9507372f0b1b963James Dong * 3) num_reorder_frames in vui equal to 0 */ 3800c1bc742181ded4930842b46e9507372f0b1b963James Dong if ( pStorage->noReordering || 3810c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSps->picOrderCntType == 2 || 3820c1bc742181ded4930842b46e9507372f0b1b963James Dong (pStorage->activeSps->vuiParametersPresentFlag && 3830c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSps->vuiParameters->bitstreamRestrictionFlag && 3840c1bc742181ded4930842b46e9507372f0b1b963James Dong !pStorage->activeSps->vuiParameters->numReorderFrames) ) 3850c1bc742181ded4930842b46e9507372f0b1b963James Dong flag = HANTRO_TRUE; 3860c1bc742181ded4930842b46e9507372f0b1b963James Dong else 3870c1bc742181ded4930842b46e9507372f0b1b963James Dong flag = HANTRO_FALSE; 3880c1bc742181ded4930842b46e9507372f0b1b963James Dong 3890c1bc742181ded4930842b46e9507372f0b1b963James Dong tmp = h264bsdResetDpb(pStorage->dpb, 3900c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSps->picWidthInMbs * 3910c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSps->picHeightInMbs, 3920c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSps->maxDpbSize, 3930c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSps->numRefFrames, 3940c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSps->maxFrameNum, 3950c1bc742181ded4930842b46e9507372f0b1b963James Dong flag); 3960c1bc742181ded4930842b46e9507372f0b1b963James Dong if (tmp != HANTRO_OK) 3970c1bc742181ded4930842b46e9507372f0b1b963James Dong return(tmp); 3980c1bc742181ded4930842b46e9507372f0b1b963James Dong } 3990c1bc742181ded4930842b46e9507372f0b1b963James Dong else if (ppsId != pStorage->activePpsId) 4000c1bc742181ded4930842b46e9507372f0b1b963James Dong { 4010c1bc742181ded4930842b46e9507372f0b1b963James Dong /* sequence parameter set shall not change but before an IDR picture */ 4020c1bc742181ded4930842b46e9507372f0b1b963James Dong if (pStorage->pps[ppsId]->seqParameterSetId != pStorage->activeSpsId) 4030c1bc742181ded4930842b46e9507372f0b1b963James Dong { 4040c1bc742181ded4930842b46e9507372f0b1b963James Dong DEBUG(("SEQ PARAM SET CHANGING...\n")); 4050c1bc742181ded4930842b46e9507372f0b1b963James Dong if (isIdr) 4060c1bc742181ded4930842b46e9507372f0b1b963James Dong { 4070c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activePpsId = ppsId; 4080c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activePps = pStorage->pps[ppsId]; 4090c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSpsId = pStorage->activePps->seqParameterSetId; 4100c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSps = pStorage->sps[pStorage->activeSpsId]; 4110c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->picSizeInMbs = 4120c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSps->picWidthInMbs * 4130c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSps->picHeightInMbs; 4140c1bc742181ded4930842b46e9507372f0b1b963James Dong 4150c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->currImage->width = pStorage->activeSps->picWidthInMbs; 4160c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->currImage->height = 4170c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSps->picHeightInMbs; 4180c1bc742181ded4930842b46e9507372f0b1b963James Dong 4190c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->pendingActivation = HANTRO_TRUE; 4200c1bc742181ded4930842b46e9507372f0b1b963James Dong } 4210c1bc742181ded4930842b46e9507372f0b1b963James Dong else 4220c1bc742181ded4930842b46e9507372f0b1b963James Dong { 4230c1bc742181ded4930842b46e9507372f0b1b963James Dong DEBUG(("TRYING TO CHANGE SPS IN NON-IDR SLICE\n")); 4240c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_NOK); 4250c1bc742181ded4930842b46e9507372f0b1b963James Dong } 4260c1bc742181ded4930842b46e9507372f0b1b963James Dong } 4270c1bc742181ded4930842b46e9507372f0b1b963James Dong else 4280c1bc742181ded4930842b46e9507372f0b1b963James Dong { 4290c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activePpsId = ppsId; 4300c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activePps = pStorage->pps[ppsId]; 4310c1bc742181ded4930842b46e9507372f0b1b963James Dong } 4320c1bc742181ded4930842b46e9507372f0b1b963James Dong } 4330c1bc742181ded4930842b46e9507372f0b1b963James Dong 4340c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_OK); 4350c1bc742181ded4930842b46e9507372f0b1b963James Dong 4360c1bc742181ded4930842b46e9507372f0b1b963James Dong} 4370c1bc742181ded4930842b46e9507372f0b1b963James Dong 4380c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------ 4390c1bc742181ded4930842b46e9507372f0b1b963James Dong 4400c1bc742181ded4930842b46e9507372f0b1b963James Dong Function: h264bsdResetStorage 4410c1bc742181ded4930842b46e9507372f0b1b963James Dong 4420c1bc742181ded4930842b46e9507372f0b1b963James Dong Functional description: 4430c1bc742181ded4930842b46e9507372f0b1b963James Dong Reset contents of the storage. This should be called before 4440c1bc742181ded4930842b46e9507372f0b1b963James Dong processing of new image is started. 4450c1bc742181ded4930842b46e9507372f0b1b963James Dong 4460c1bc742181ded4930842b46e9507372f0b1b963James Dong Inputs: 4470c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage pointer to storage structure 4480c1bc742181ded4930842b46e9507372f0b1b963James Dong 4490c1bc742181ded4930842b46e9507372f0b1b963James Dong Outputs: 4500c1bc742181ded4930842b46e9507372f0b1b963James Dong none 4510c1bc742181ded4930842b46e9507372f0b1b963James Dong 4520c1bc742181ded4930842b46e9507372f0b1b963James Dong Returns: 4530c1bc742181ded4930842b46e9507372f0b1b963James Dong none 4540c1bc742181ded4930842b46e9507372f0b1b963James Dong 4550c1bc742181ded4930842b46e9507372f0b1b963James Dong 4560c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/ 4570c1bc742181ded4930842b46e9507372f0b1b963James Dong 4580c1bc742181ded4930842b46e9507372f0b1b963James Dongvoid h264bsdResetStorage(storage_t *pStorage) 4590c1bc742181ded4930842b46e9507372f0b1b963James Dong{ 4600c1bc742181ded4930842b46e9507372f0b1b963James Dong 4610c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */ 4620c1bc742181ded4930842b46e9507372f0b1b963James Dong 4630c1bc742181ded4930842b46e9507372f0b1b963James Dong u32 i; 4640c1bc742181ded4930842b46e9507372f0b1b963James Dong 4650c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */ 4660c1bc742181ded4930842b46e9507372f0b1b963James Dong 4670c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(pStorage); 4680c1bc742181ded4930842b46e9507372f0b1b963James Dong 4690c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->slice->numDecodedMbs = 0; 4700c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->slice->sliceId = 0; 4710c1bc742181ded4930842b46e9507372f0b1b963James Dong 4720c1bc742181ded4930842b46e9507372f0b1b963James Dong for (i = 0; i < pStorage->picSizeInMbs; i++) 4730c1bc742181ded4930842b46e9507372f0b1b963James Dong { 4740c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->mb[i].sliceId = 0; 4750c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->mb[i].decoded = 0; 4760c1bc742181ded4930842b46e9507372f0b1b963James Dong } 4770c1bc742181ded4930842b46e9507372f0b1b963James Dong 4780c1bc742181ded4930842b46e9507372f0b1b963James Dong} 4790c1bc742181ded4930842b46e9507372f0b1b963James Dong 4800c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------ 4810c1bc742181ded4930842b46e9507372f0b1b963James Dong 4820c1bc742181ded4930842b46e9507372f0b1b963James Dong Function: h264bsdIsStartOfPicture 4830c1bc742181ded4930842b46e9507372f0b1b963James Dong 4840c1bc742181ded4930842b46e9507372f0b1b963James Dong Functional description: 4850c1bc742181ded4930842b46e9507372f0b1b963James Dong Determine if the decoder is in the start of a picture. This 4860c1bc742181ded4930842b46e9507372f0b1b963James Dong information is needed to decide if h264bsdActivateParamSets and 4870c1bc742181ded4930842b46e9507372f0b1b963James Dong h264bsdCheckGapsInFrameNum functions should be called. Function 4880c1bc742181ded4930842b46e9507372f0b1b963James Dong considers that new picture is starting if no slice headers 4890c1bc742181ded4930842b46e9507372f0b1b963James Dong have been successfully decoded for the current access unit. 4900c1bc742181ded4930842b46e9507372f0b1b963James Dong 4910c1bc742181ded4930842b46e9507372f0b1b963James Dong Inputs: 4920c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage pointer to storage structure 4930c1bc742181ded4930842b46e9507372f0b1b963James Dong 4940c1bc742181ded4930842b46e9507372f0b1b963James Dong Outputs: 4950c1bc742181ded4930842b46e9507372f0b1b963James Dong none 4960c1bc742181ded4930842b46e9507372f0b1b963James Dong 4970c1bc742181ded4930842b46e9507372f0b1b963James Dong Returns: 4980c1bc742181ded4930842b46e9507372f0b1b963James Dong HANTRO_TRUE new picture is starting 4990c1bc742181ded4930842b46e9507372f0b1b963James Dong HANTRO_FALSE not starting 5000c1bc742181ded4930842b46e9507372f0b1b963James Dong 5010c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/ 5020c1bc742181ded4930842b46e9507372f0b1b963James Dong 5030c1bc742181ded4930842b46e9507372f0b1b963James Dongu32 h264bsdIsStartOfPicture(storage_t *pStorage) 5040c1bc742181ded4930842b46e9507372f0b1b963James Dong{ 5050c1bc742181ded4930842b46e9507372f0b1b963James Dong 5060c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */ 5070c1bc742181ded4930842b46e9507372f0b1b963James Dong 5080c1bc742181ded4930842b46e9507372f0b1b963James Dong 5090c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */ 5100c1bc742181ded4930842b46e9507372f0b1b963James Dong 5110c1bc742181ded4930842b46e9507372f0b1b963James Dong if (pStorage->validSliceInAccessUnit == HANTRO_FALSE) 5120c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_TRUE); 5130c1bc742181ded4930842b46e9507372f0b1b963James Dong else 5140c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_FALSE); 5150c1bc742181ded4930842b46e9507372f0b1b963James Dong 5160c1bc742181ded4930842b46e9507372f0b1b963James Dong} 5170c1bc742181ded4930842b46e9507372f0b1b963James Dong 5180c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------ 5190c1bc742181ded4930842b46e9507372f0b1b963James Dong 5200c1bc742181ded4930842b46e9507372f0b1b963James Dong Function: h264bsdIsEndOfPicture 5210c1bc742181ded4930842b46e9507372f0b1b963James Dong 5220c1bc742181ded4930842b46e9507372f0b1b963James Dong Functional description: 5230c1bc742181ded4930842b46e9507372f0b1b963James Dong Determine if the decoder is in the end of a picture. This 5240c1bc742181ded4930842b46e9507372f0b1b963James Dong information is needed to determine when deblocking filtering 5250c1bc742181ded4930842b46e9507372f0b1b963James Dong and reference picture marking processes should be performed. 5260c1bc742181ded4930842b46e9507372f0b1b963James Dong 5270c1bc742181ded4930842b46e9507372f0b1b963James Dong If the decoder is processing primary slices the return value 5280c1bc742181ded4930842b46e9507372f0b1b963James Dong is determined by checking the value of numDecodedMbs in the 5290c1bc742181ded4930842b46e9507372f0b1b963James Dong storage. On the other hand, if the decoder is processing 5300c1bc742181ded4930842b46e9507372f0b1b963James Dong redundant slices the numDecodedMbs may not contain valid 5310c1bc742181ded4930842b46e9507372f0b1b963James Dong informationa and each macroblock has to be checked separately. 5320c1bc742181ded4930842b46e9507372f0b1b963James Dong 5330c1bc742181ded4930842b46e9507372f0b1b963James Dong Inputs: 5340c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage pointer to storage structure 5350c1bc742181ded4930842b46e9507372f0b1b963James Dong 5360c1bc742181ded4930842b46e9507372f0b1b963James Dong Outputs: 5370c1bc742181ded4930842b46e9507372f0b1b963James Dong none 5380c1bc742181ded4930842b46e9507372f0b1b963James Dong 5390c1bc742181ded4930842b46e9507372f0b1b963James Dong Returns: 5400c1bc742181ded4930842b46e9507372f0b1b963James Dong HANTRO_TRUE end of picture 5410c1bc742181ded4930842b46e9507372f0b1b963James Dong HANTRO_FALSE noup 5420c1bc742181ded4930842b46e9507372f0b1b963James Dong 5430c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/ 5440c1bc742181ded4930842b46e9507372f0b1b963James Dong 5450c1bc742181ded4930842b46e9507372f0b1b963James Dongu32 h264bsdIsEndOfPicture(storage_t *pStorage) 5460c1bc742181ded4930842b46e9507372f0b1b963James Dong{ 5470c1bc742181ded4930842b46e9507372f0b1b963James Dong 5480c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */ 5490c1bc742181ded4930842b46e9507372f0b1b963James Dong 5500c1bc742181ded4930842b46e9507372f0b1b963James Dong u32 i, tmp; 5510c1bc742181ded4930842b46e9507372f0b1b963James Dong 5520c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */ 5530c1bc742181ded4930842b46e9507372f0b1b963James Dong 5540c1bc742181ded4930842b46e9507372f0b1b963James Dong /* primary picture */ 5550c1bc742181ded4930842b46e9507372f0b1b963James Dong if (!pStorage->sliceHeader[0].redundantPicCnt) 5560c1bc742181ded4930842b46e9507372f0b1b963James Dong { 5570c1bc742181ded4930842b46e9507372f0b1b963James Dong if (pStorage->slice->numDecodedMbs == pStorage->picSizeInMbs) 5580c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_TRUE); 5590c1bc742181ded4930842b46e9507372f0b1b963James Dong } 5600c1bc742181ded4930842b46e9507372f0b1b963James Dong else 5610c1bc742181ded4930842b46e9507372f0b1b963James Dong { 5620c1bc742181ded4930842b46e9507372f0b1b963James Dong for (i = 0, tmp = 0; i < pStorage->picSizeInMbs; i++) 5630c1bc742181ded4930842b46e9507372f0b1b963James Dong tmp += pStorage->mb[i].decoded ? 1 : 0; 5640c1bc742181ded4930842b46e9507372f0b1b963James Dong 5650c1bc742181ded4930842b46e9507372f0b1b963James Dong if (tmp == pStorage->picSizeInMbs) 5660c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_TRUE); 5670c1bc742181ded4930842b46e9507372f0b1b963James Dong } 5680c1bc742181ded4930842b46e9507372f0b1b963James Dong 5690c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_FALSE); 5700c1bc742181ded4930842b46e9507372f0b1b963James Dong 5710c1bc742181ded4930842b46e9507372f0b1b963James Dong} 5720c1bc742181ded4930842b46e9507372f0b1b963James Dong 5730c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------ 5740c1bc742181ded4930842b46e9507372f0b1b963James Dong 5750c1bc742181ded4930842b46e9507372f0b1b963James Dong Function: h264bsdComputeSliceGroupMap 5760c1bc742181ded4930842b46e9507372f0b1b963James Dong 5770c1bc742181ded4930842b46e9507372f0b1b963James Dong Functional description: 5780c1bc742181ded4930842b46e9507372f0b1b963James Dong Compute slice group map. Just call h264bsdDecodeSliceGroupMap with 5790c1bc742181ded4930842b46e9507372f0b1b963James Dong appropriate parameters. 5800c1bc742181ded4930842b46e9507372f0b1b963James Dong 5810c1bc742181ded4930842b46e9507372f0b1b963James Dong Inputs: 5820c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage pointer to storage structure 5830c1bc742181ded4930842b46e9507372f0b1b963James Dong sliceGroupChangeCycle 5840c1bc742181ded4930842b46e9507372f0b1b963James Dong 5850c1bc742181ded4930842b46e9507372f0b1b963James Dong Outputs: 5860c1bc742181ded4930842b46e9507372f0b1b963James Dong none 5870c1bc742181ded4930842b46e9507372f0b1b963James Dong 5880c1bc742181ded4930842b46e9507372f0b1b963James Dong Returns: 5890c1bc742181ded4930842b46e9507372f0b1b963James Dong none 5900c1bc742181ded4930842b46e9507372f0b1b963James Dong 5910c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/ 5920c1bc742181ded4930842b46e9507372f0b1b963James Dong 5930c1bc742181ded4930842b46e9507372f0b1b963James Dongvoid h264bsdComputeSliceGroupMap(storage_t *pStorage, u32 sliceGroupChangeCycle) 5940c1bc742181ded4930842b46e9507372f0b1b963James Dong{ 5950c1bc742181ded4930842b46e9507372f0b1b963James Dong 5960c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */ 5970c1bc742181ded4930842b46e9507372f0b1b963James Dong 5980c1bc742181ded4930842b46e9507372f0b1b963James Dong 5990c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */ 6000c1bc742181ded4930842b46e9507372f0b1b963James Dong 6010c1bc742181ded4930842b46e9507372f0b1b963James Dong h264bsdDecodeSliceGroupMap(pStorage->sliceGroupMap, 6020c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activePps, sliceGroupChangeCycle, 6030c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSps->picWidthInMbs, 6040c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->activeSps->picHeightInMbs); 6050c1bc742181ded4930842b46e9507372f0b1b963James Dong 6060c1bc742181ded4930842b46e9507372f0b1b963James Dong} 6070c1bc742181ded4930842b46e9507372f0b1b963James Dong 6080c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------ 6090c1bc742181ded4930842b46e9507372f0b1b963James Dong 6100c1bc742181ded4930842b46e9507372f0b1b963James Dong Function: h264bsdCheckAccessUnitBoundary 6110c1bc742181ded4930842b46e9507372f0b1b963James Dong 6120c1bc742181ded4930842b46e9507372f0b1b963James Dong Functional description: 6130c1bc742181ded4930842b46e9507372f0b1b963James Dong Check if next NAL unit starts a new access unit. Following 6140c1bc742181ded4930842b46e9507372f0b1b963James Dong conditions specify start of a new access unit: 6150c1bc742181ded4930842b46e9507372f0b1b963James Dong 6160c1bc742181ded4930842b46e9507372f0b1b963James Dong -NAL unit types 6-11, 13-18 (e.g. SPS, PPS) 6170c1bc742181ded4930842b46e9507372f0b1b963James Dong 6180c1bc742181ded4930842b46e9507372f0b1b963James Dong following conditions checked only for slice NAL units, values 6190c1bc742181ded4930842b46e9507372f0b1b963James Dong compared to ones obtained from previous slice: 6200c1bc742181ded4930842b46e9507372f0b1b963James Dong 6210c1bc742181ded4930842b46e9507372f0b1b963James Dong -NAL unit type differs (slice / IDR slice) 6220c1bc742181ded4930842b46e9507372f0b1b963James Dong -frame_num differs 6230c1bc742181ded4930842b46e9507372f0b1b963James Dong -nal_ref_idc differs and one of the values is 0 6240c1bc742181ded4930842b46e9507372f0b1b963James Dong -POC information differs 6250c1bc742181ded4930842b46e9507372f0b1b963James Dong -both are IDR slices and idr_pic_id differs 6260c1bc742181ded4930842b46e9507372f0b1b963James Dong 6270c1bc742181ded4930842b46e9507372f0b1b963James Dong Inputs: 6280c1bc742181ded4930842b46e9507372f0b1b963James Dong strm pointer to stream data structure 6290c1bc742181ded4930842b46e9507372f0b1b963James Dong nuNext pointer to NAL unit structure 6300c1bc742181ded4930842b46e9507372f0b1b963James Dong storage pointer to storage structure 6310c1bc742181ded4930842b46e9507372f0b1b963James Dong 6320c1bc742181ded4930842b46e9507372f0b1b963James Dong Outputs: 6330c1bc742181ded4930842b46e9507372f0b1b963James Dong accessUnitBoundaryFlag the result is stored here, TRUE for 6340c1bc742181ded4930842b46e9507372f0b1b963James Dong access unit boundary, FALSE otherwise 6350c1bc742181ded4930842b46e9507372f0b1b963James Dong 6360c1bc742181ded4930842b46e9507372f0b1b963James Dong Returns: 6370c1bc742181ded4930842b46e9507372f0b1b963James Dong HANTRO_OK success 6380c1bc742181ded4930842b46e9507372f0b1b963James Dong HANTRO_NOK failure, invalid stream data 6390c1bc742181ded4930842b46e9507372f0b1b963James Dong PARAM_SET_ERROR invalid param set usage 6400c1bc742181ded4930842b46e9507372f0b1b963James Dong 6410c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/ 6420c1bc742181ded4930842b46e9507372f0b1b963James Dong 6430c1bc742181ded4930842b46e9507372f0b1b963James Dongu32 h264bsdCheckAccessUnitBoundary( 6440c1bc742181ded4930842b46e9507372f0b1b963James Dong strmData_t *strm, 6450c1bc742181ded4930842b46e9507372f0b1b963James Dong nalUnit_t *nuNext, 6460c1bc742181ded4930842b46e9507372f0b1b963James Dong storage_t *storage, 6470c1bc742181ded4930842b46e9507372f0b1b963James Dong u32 *accessUnitBoundaryFlag) 6480c1bc742181ded4930842b46e9507372f0b1b963James Dong{ 6490c1bc742181ded4930842b46e9507372f0b1b963James Dong 6500c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */ 6510c1bc742181ded4930842b46e9507372f0b1b963James Dong 6520c1bc742181ded4930842b46e9507372f0b1b963James Dong u32 tmp, ppsId, frameNum, idrPicId, picOrderCntLsb; 6530c1bc742181ded4930842b46e9507372f0b1b963James Dong i32 deltaPicOrderCntBottom, deltaPicOrderCnt[2]; 6540c1bc742181ded4930842b46e9507372f0b1b963James Dong seqParamSet_t *sps; 6550c1bc742181ded4930842b46e9507372f0b1b963James Dong picParamSet_t *pps; 6560c1bc742181ded4930842b46e9507372f0b1b963James Dong 6570c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */ 6580c1bc742181ded4930842b46e9507372f0b1b963James Dong 6590c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(strm); 6600c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(nuNext); 6610c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(storage); 6620c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(storage->sps); 6630c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(storage->pps); 6640c1bc742181ded4930842b46e9507372f0b1b963James Dong 6650c1bc742181ded4930842b46e9507372f0b1b963James Dong /* initialize default output to FALSE */ 6660c1bc742181ded4930842b46e9507372f0b1b963James Dong *accessUnitBoundaryFlag = HANTRO_FALSE; 6670c1bc742181ded4930842b46e9507372f0b1b963James Dong 6680c1bc742181ded4930842b46e9507372f0b1b963James Dong if ( ( (nuNext->nalUnitType > 5) && (nuNext->nalUnitType < 12) ) || 6690c1bc742181ded4930842b46e9507372f0b1b963James Dong ( (nuNext->nalUnitType > 12) && (nuNext->nalUnitType <= 18) ) ) 6700c1bc742181ded4930842b46e9507372f0b1b963James Dong { 6710c1bc742181ded4930842b46e9507372f0b1b963James Dong *accessUnitBoundaryFlag = HANTRO_TRUE; 6720c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_OK); 6730c1bc742181ded4930842b46e9507372f0b1b963James Dong } 6740c1bc742181ded4930842b46e9507372f0b1b963James Dong else if ( nuNext->nalUnitType != NAL_CODED_SLICE && 6750c1bc742181ded4930842b46e9507372f0b1b963James Dong nuNext->nalUnitType != NAL_CODED_SLICE_IDR ) 6760c1bc742181ded4930842b46e9507372f0b1b963James Dong { 6770c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_OK); 6780c1bc742181ded4930842b46e9507372f0b1b963James Dong } 6790c1bc742181ded4930842b46e9507372f0b1b963James Dong 6800c1bc742181ded4930842b46e9507372f0b1b963James Dong /* check if this is the very first call to this function */ 6810c1bc742181ded4930842b46e9507372f0b1b963James Dong if (storage->aub->firstCallFlag) 6820c1bc742181ded4930842b46e9507372f0b1b963James Dong { 6830c1bc742181ded4930842b46e9507372f0b1b963James Dong *accessUnitBoundaryFlag = HANTRO_TRUE; 6840c1bc742181ded4930842b46e9507372f0b1b963James Dong storage->aub->firstCallFlag = HANTRO_FALSE; 6850c1bc742181ded4930842b46e9507372f0b1b963James Dong } 6860c1bc742181ded4930842b46e9507372f0b1b963James Dong 6870c1bc742181ded4930842b46e9507372f0b1b963James Dong /* get picture parameter set id */ 6880c1bc742181ded4930842b46e9507372f0b1b963James Dong tmp = h264bsdCheckPpsId(strm, &ppsId); 6890c1bc742181ded4930842b46e9507372f0b1b963James Dong if (tmp != HANTRO_OK) 6900c1bc742181ded4930842b46e9507372f0b1b963James Dong return(tmp); 6910c1bc742181ded4930842b46e9507372f0b1b963James Dong 6920c1bc742181ded4930842b46e9507372f0b1b963James Dong /* store sps and pps in separate pointers just to make names shorter */ 6930c1bc742181ded4930842b46e9507372f0b1b963James Dong pps = storage->pps[ppsId]; 6940c1bc742181ded4930842b46e9507372f0b1b963James Dong if ( pps == NULL || storage->sps[pps->seqParameterSetId] == NULL || 6950c1bc742181ded4930842b46e9507372f0b1b963James Dong (storage->activeSpsId != MAX_NUM_SEQ_PARAM_SETS && 6960c1bc742181ded4930842b46e9507372f0b1b963James Dong pps->seqParameterSetId != storage->activeSpsId && 6970c1bc742181ded4930842b46e9507372f0b1b963James Dong nuNext->nalUnitType != NAL_CODED_SLICE_IDR) ) 6980c1bc742181ded4930842b46e9507372f0b1b963James Dong return(PARAM_SET_ERROR); 6990c1bc742181ded4930842b46e9507372f0b1b963James Dong sps = storage->sps[pps->seqParameterSetId]; 7000c1bc742181ded4930842b46e9507372f0b1b963James Dong 7010c1bc742181ded4930842b46e9507372f0b1b963James Dong if (storage->aub->nuPrev->nalRefIdc != nuNext->nalRefIdc && 7020c1bc742181ded4930842b46e9507372f0b1b963James Dong (storage->aub->nuPrev->nalRefIdc == 0 || nuNext->nalRefIdc == 0)) 7030c1bc742181ded4930842b46e9507372f0b1b963James Dong *accessUnitBoundaryFlag = HANTRO_TRUE; 7040c1bc742181ded4930842b46e9507372f0b1b963James Dong 7050c1bc742181ded4930842b46e9507372f0b1b963James Dong if ((storage->aub->nuPrev->nalUnitType == NAL_CODED_SLICE_IDR && 7060c1bc742181ded4930842b46e9507372f0b1b963James Dong nuNext->nalUnitType != NAL_CODED_SLICE_IDR) || 7070c1bc742181ded4930842b46e9507372f0b1b963James Dong (storage->aub->nuPrev->nalUnitType != NAL_CODED_SLICE_IDR && 7080c1bc742181ded4930842b46e9507372f0b1b963James Dong nuNext->nalUnitType == NAL_CODED_SLICE_IDR)) 7090c1bc742181ded4930842b46e9507372f0b1b963James Dong *accessUnitBoundaryFlag = HANTRO_TRUE; 7100c1bc742181ded4930842b46e9507372f0b1b963James Dong 7110c1bc742181ded4930842b46e9507372f0b1b963James Dong tmp = h264bsdCheckFrameNum(strm, sps->maxFrameNum, &frameNum); 7120c1bc742181ded4930842b46e9507372f0b1b963James Dong if (tmp != HANTRO_OK) 7130c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_NOK); 7140c1bc742181ded4930842b46e9507372f0b1b963James Dong 7150c1bc742181ded4930842b46e9507372f0b1b963James Dong if (storage->aub->prevFrameNum != frameNum) 7160c1bc742181ded4930842b46e9507372f0b1b963James Dong { 7170c1bc742181ded4930842b46e9507372f0b1b963James Dong storage->aub->prevFrameNum = frameNum; 7180c1bc742181ded4930842b46e9507372f0b1b963James Dong *accessUnitBoundaryFlag = HANTRO_TRUE; 7190c1bc742181ded4930842b46e9507372f0b1b963James Dong } 7200c1bc742181ded4930842b46e9507372f0b1b963James Dong 7210c1bc742181ded4930842b46e9507372f0b1b963James Dong if (nuNext->nalUnitType == NAL_CODED_SLICE_IDR) 7220c1bc742181ded4930842b46e9507372f0b1b963James Dong { 7230c1bc742181ded4930842b46e9507372f0b1b963James Dong tmp = h264bsdCheckIdrPicId(strm, sps->maxFrameNum, nuNext->nalUnitType, 7240c1bc742181ded4930842b46e9507372f0b1b963James Dong &idrPicId); 7250c1bc742181ded4930842b46e9507372f0b1b963James Dong if (tmp != HANTRO_OK) 7260c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_NOK); 7270c1bc742181ded4930842b46e9507372f0b1b963James Dong 7280c1bc742181ded4930842b46e9507372f0b1b963James Dong if (storage->aub->nuPrev->nalUnitType == NAL_CODED_SLICE_IDR && 7290c1bc742181ded4930842b46e9507372f0b1b963James Dong storage->aub->prevIdrPicId != idrPicId) 7300c1bc742181ded4930842b46e9507372f0b1b963James Dong *accessUnitBoundaryFlag = HANTRO_TRUE; 7310c1bc742181ded4930842b46e9507372f0b1b963James Dong 7320c1bc742181ded4930842b46e9507372f0b1b963James Dong storage->aub->prevIdrPicId = idrPicId; 7330c1bc742181ded4930842b46e9507372f0b1b963James Dong } 7340c1bc742181ded4930842b46e9507372f0b1b963James Dong 7350c1bc742181ded4930842b46e9507372f0b1b963James Dong if (sps->picOrderCntType == 0) 7360c1bc742181ded4930842b46e9507372f0b1b963James Dong { 7370c1bc742181ded4930842b46e9507372f0b1b963James Dong tmp = h264bsdCheckPicOrderCntLsb(strm, sps, nuNext->nalUnitType, 7380c1bc742181ded4930842b46e9507372f0b1b963James Dong &picOrderCntLsb); 7390c1bc742181ded4930842b46e9507372f0b1b963James Dong if (tmp != HANTRO_OK) 7400c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_NOK); 7410c1bc742181ded4930842b46e9507372f0b1b963James Dong 7420c1bc742181ded4930842b46e9507372f0b1b963James Dong if (storage->aub->prevPicOrderCntLsb != picOrderCntLsb) 7430c1bc742181ded4930842b46e9507372f0b1b963James Dong { 7440c1bc742181ded4930842b46e9507372f0b1b963James Dong storage->aub->prevPicOrderCntLsb = picOrderCntLsb; 7450c1bc742181ded4930842b46e9507372f0b1b963James Dong *accessUnitBoundaryFlag = HANTRO_TRUE; 7460c1bc742181ded4930842b46e9507372f0b1b963James Dong } 7470c1bc742181ded4930842b46e9507372f0b1b963James Dong 7480c1bc742181ded4930842b46e9507372f0b1b963James Dong if (pps->picOrderPresentFlag) 7490c1bc742181ded4930842b46e9507372f0b1b963James Dong { 7500c1bc742181ded4930842b46e9507372f0b1b963James Dong tmp = h264bsdCheckDeltaPicOrderCntBottom(strm, sps, 7510c1bc742181ded4930842b46e9507372f0b1b963James Dong nuNext->nalUnitType, &deltaPicOrderCntBottom); 7520c1bc742181ded4930842b46e9507372f0b1b963James Dong if (tmp != HANTRO_OK) 7530c1bc742181ded4930842b46e9507372f0b1b963James Dong return(tmp); 7540c1bc742181ded4930842b46e9507372f0b1b963James Dong 7550c1bc742181ded4930842b46e9507372f0b1b963James Dong if (storage->aub->prevDeltaPicOrderCntBottom != 7560c1bc742181ded4930842b46e9507372f0b1b963James Dong deltaPicOrderCntBottom) 7570c1bc742181ded4930842b46e9507372f0b1b963James Dong { 7580c1bc742181ded4930842b46e9507372f0b1b963James Dong storage->aub->prevDeltaPicOrderCntBottom = 7590c1bc742181ded4930842b46e9507372f0b1b963James Dong deltaPicOrderCntBottom; 7600c1bc742181ded4930842b46e9507372f0b1b963James Dong *accessUnitBoundaryFlag = HANTRO_TRUE; 7610c1bc742181ded4930842b46e9507372f0b1b963James Dong } 7620c1bc742181ded4930842b46e9507372f0b1b963James Dong } 7630c1bc742181ded4930842b46e9507372f0b1b963James Dong } 7640c1bc742181ded4930842b46e9507372f0b1b963James Dong else if (sps->picOrderCntType == 1 && !sps->deltaPicOrderAlwaysZeroFlag) 7650c1bc742181ded4930842b46e9507372f0b1b963James Dong { 7660c1bc742181ded4930842b46e9507372f0b1b963James Dong tmp = h264bsdCheckDeltaPicOrderCnt(strm, sps, nuNext->nalUnitType, 7670c1bc742181ded4930842b46e9507372f0b1b963James Dong pps->picOrderPresentFlag, deltaPicOrderCnt); 7680c1bc742181ded4930842b46e9507372f0b1b963James Dong if (tmp != HANTRO_OK) 7690c1bc742181ded4930842b46e9507372f0b1b963James Dong return(tmp); 7700c1bc742181ded4930842b46e9507372f0b1b963James Dong 7710c1bc742181ded4930842b46e9507372f0b1b963James Dong if (storage->aub->prevDeltaPicOrderCnt[0] != deltaPicOrderCnt[0]) 7720c1bc742181ded4930842b46e9507372f0b1b963James Dong { 7730c1bc742181ded4930842b46e9507372f0b1b963James Dong storage->aub->prevDeltaPicOrderCnt[0] = deltaPicOrderCnt[0]; 7740c1bc742181ded4930842b46e9507372f0b1b963James Dong *accessUnitBoundaryFlag = HANTRO_TRUE; 7750c1bc742181ded4930842b46e9507372f0b1b963James Dong } 7760c1bc742181ded4930842b46e9507372f0b1b963James Dong 7770c1bc742181ded4930842b46e9507372f0b1b963James Dong if (pps->picOrderPresentFlag) 7780c1bc742181ded4930842b46e9507372f0b1b963James Dong if (storage->aub->prevDeltaPicOrderCnt[1] != deltaPicOrderCnt[1]) 7790c1bc742181ded4930842b46e9507372f0b1b963James Dong { 7800c1bc742181ded4930842b46e9507372f0b1b963James Dong storage->aub->prevDeltaPicOrderCnt[1] = deltaPicOrderCnt[1]; 7810c1bc742181ded4930842b46e9507372f0b1b963James Dong *accessUnitBoundaryFlag = HANTRO_TRUE; 7820c1bc742181ded4930842b46e9507372f0b1b963James Dong } 7830c1bc742181ded4930842b46e9507372f0b1b963James Dong } 7840c1bc742181ded4930842b46e9507372f0b1b963James Dong 7850c1bc742181ded4930842b46e9507372f0b1b963James Dong *storage->aub->nuPrev = *nuNext; 7860c1bc742181ded4930842b46e9507372f0b1b963James Dong 7870c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_OK); 7880c1bc742181ded4930842b46e9507372f0b1b963James Dong 7890c1bc742181ded4930842b46e9507372f0b1b963James Dong} 7900c1bc742181ded4930842b46e9507372f0b1b963James Dong 7910c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------ 7920c1bc742181ded4930842b46e9507372f0b1b963James Dong 7930c1bc742181ded4930842b46e9507372f0b1b963James Dong Function: CheckPps 7940c1bc742181ded4930842b46e9507372f0b1b963James Dong 7950c1bc742181ded4930842b46e9507372f0b1b963James Dong Functional description: 7960c1bc742181ded4930842b46e9507372f0b1b963James Dong Check picture parameter set. Contents of the picture parameter 7970c1bc742181ded4930842b46e9507372f0b1b963James Dong set information that depends on the image dimensions is checked 7980c1bc742181ded4930842b46e9507372f0b1b963James Dong against the dimensions in the sps. 7990c1bc742181ded4930842b46e9507372f0b1b963James Dong 8000c1bc742181ded4930842b46e9507372f0b1b963James Dong Inputs: 8010c1bc742181ded4930842b46e9507372f0b1b963James Dong pps pointer to picture paramter set 8020c1bc742181ded4930842b46e9507372f0b1b963James Dong sps pointer to sequence parameter set 8030c1bc742181ded4930842b46e9507372f0b1b963James Dong 8040c1bc742181ded4930842b46e9507372f0b1b963James Dong Outputs: 8050c1bc742181ded4930842b46e9507372f0b1b963James Dong none 8060c1bc742181ded4930842b46e9507372f0b1b963James Dong 8070c1bc742181ded4930842b46e9507372f0b1b963James Dong Returns: 8080c1bc742181ded4930842b46e9507372f0b1b963James Dong HANTRO_OK everything ok 8090c1bc742181ded4930842b46e9507372f0b1b963James Dong HANTRO_NOK invalid data in picture parameter set 8100c1bc742181ded4930842b46e9507372f0b1b963James Dong 8110c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/ 8120c1bc742181ded4930842b46e9507372f0b1b963James Dongu32 CheckPps(picParamSet_t *pps, seqParamSet_t *sps) 8130c1bc742181ded4930842b46e9507372f0b1b963James Dong{ 8140c1bc742181ded4930842b46e9507372f0b1b963James Dong 8150c1bc742181ded4930842b46e9507372f0b1b963James Dong u32 i; 8160c1bc742181ded4930842b46e9507372f0b1b963James Dong u32 picSize; 8170c1bc742181ded4930842b46e9507372f0b1b963James Dong 8180c1bc742181ded4930842b46e9507372f0b1b963James Dong picSize = sps->picWidthInMbs * sps->picHeightInMbs; 8190c1bc742181ded4930842b46e9507372f0b1b963James Dong 8200c1bc742181ded4930842b46e9507372f0b1b963James Dong /* check slice group params */ 8210c1bc742181ded4930842b46e9507372f0b1b963James Dong if (pps->numSliceGroups > 1) 8220c1bc742181ded4930842b46e9507372f0b1b963James Dong { 8230c1bc742181ded4930842b46e9507372f0b1b963James Dong if (pps->sliceGroupMapType == 0) 8240c1bc742181ded4930842b46e9507372f0b1b963James Dong { 8250c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(pps->runLength); 8260c1bc742181ded4930842b46e9507372f0b1b963James Dong for (i = 0; i < pps->numSliceGroups; i++) 8270c1bc742181ded4930842b46e9507372f0b1b963James Dong { 8280c1bc742181ded4930842b46e9507372f0b1b963James Dong if (pps->runLength[i] > picSize) 8290c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_NOK); 8300c1bc742181ded4930842b46e9507372f0b1b963James Dong } 8310c1bc742181ded4930842b46e9507372f0b1b963James Dong } 8320c1bc742181ded4930842b46e9507372f0b1b963James Dong else if (pps->sliceGroupMapType == 2) 8330c1bc742181ded4930842b46e9507372f0b1b963James Dong { 8340c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(pps->topLeft); 8350c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(pps->bottomRight); 8360c1bc742181ded4930842b46e9507372f0b1b963James Dong for (i = 0; i < pps->numSliceGroups-1; i++) 8370c1bc742181ded4930842b46e9507372f0b1b963James Dong { 8380c1bc742181ded4930842b46e9507372f0b1b963James Dong if (pps->topLeft[i] > pps->bottomRight[i] || 8390c1bc742181ded4930842b46e9507372f0b1b963James Dong pps->bottomRight[i] >= picSize) 8400c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_NOK); 8410c1bc742181ded4930842b46e9507372f0b1b963James Dong 8420c1bc742181ded4930842b46e9507372f0b1b963James Dong if ( (pps->topLeft[i] % sps->picWidthInMbs) > 8430c1bc742181ded4930842b46e9507372f0b1b963James Dong (pps->bottomRight[i] % sps->picWidthInMbs) ) 8440c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_NOK); 8450c1bc742181ded4930842b46e9507372f0b1b963James Dong } 8460c1bc742181ded4930842b46e9507372f0b1b963James Dong } 8470c1bc742181ded4930842b46e9507372f0b1b963James Dong else if (pps->sliceGroupMapType > 2 && pps->sliceGroupMapType < 6) 8480c1bc742181ded4930842b46e9507372f0b1b963James Dong { 8490c1bc742181ded4930842b46e9507372f0b1b963James Dong if (pps->sliceGroupChangeRate > picSize) 8500c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_NOK); 8510c1bc742181ded4930842b46e9507372f0b1b963James Dong } 8520c1bc742181ded4930842b46e9507372f0b1b963James Dong else if (pps->sliceGroupMapType == 6 && 8530c1bc742181ded4930842b46e9507372f0b1b963James Dong pps->picSizeInMapUnits < picSize) 8540c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_NOK); 8550c1bc742181ded4930842b46e9507372f0b1b963James Dong } 8560c1bc742181ded4930842b46e9507372f0b1b963James Dong 8570c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_OK); 8580c1bc742181ded4930842b46e9507372f0b1b963James Dong} 8590c1bc742181ded4930842b46e9507372f0b1b963James Dong 8600c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------ 8610c1bc742181ded4930842b46e9507372f0b1b963James Dong 8620c1bc742181ded4930842b46e9507372f0b1b963James Dong Function: h264bsdValidParamSets 8630c1bc742181ded4930842b46e9507372f0b1b963James Dong 8640c1bc742181ded4930842b46e9507372f0b1b963James Dong Functional description: 8650c1bc742181ded4930842b46e9507372f0b1b963James Dong Check if any valid SPS/PPS combination exists in the storage. 8660c1bc742181ded4930842b46e9507372f0b1b963James Dong Function tries each PPS in the buffer and checks if corresponding 8670c1bc742181ded4930842b46e9507372f0b1b963James Dong SPS exists and calls CheckPps to determine if the PPS conforms 8680c1bc742181ded4930842b46e9507372f0b1b963James Dong to image dimensions of the SPS. 8690c1bc742181ded4930842b46e9507372f0b1b963James Dong 8700c1bc742181ded4930842b46e9507372f0b1b963James Dong Inputs: 8710c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage pointer to storage structure 8720c1bc742181ded4930842b46e9507372f0b1b963James Dong 8730c1bc742181ded4930842b46e9507372f0b1b963James Dong Outputs: 8740c1bc742181ded4930842b46e9507372f0b1b963James Dong HANTRO_OK there is at least one valid combination 8750c1bc742181ded4930842b46e9507372f0b1b963James Dong HANTRO_NOK no valid combinations found 8760c1bc742181ded4930842b46e9507372f0b1b963James Dong 8770c1bc742181ded4930842b46e9507372f0b1b963James Dong 8780c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/ 8790c1bc742181ded4930842b46e9507372f0b1b963James Dong 8800c1bc742181ded4930842b46e9507372f0b1b963James Dongu32 h264bsdValidParamSets(storage_t *pStorage) 8810c1bc742181ded4930842b46e9507372f0b1b963James Dong{ 8820c1bc742181ded4930842b46e9507372f0b1b963James Dong 8830c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */ 8840c1bc742181ded4930842b46e9507372f0b1b963James Dong 8850c1bc742181ded4930842b46e9507372f0b1b963James Dong u32 i; 8860c1bc742181ded4930842b46e9507372f0b1b963James Dong 8870c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */ 8880c1bc742181ded4930842b46e9507372f0b1b963James Dong 8890c1bc742181ded4930842b46e9507372f0b1b963James Dong ASSERT(pStorage); 8900c1bc742181ded4930842b46e9507372f0b1b963James Dong 8910c1bc742181ded4930842b46e9507372f0b1b963James Dong for (i = 0; i < MAX_NUM_PIC_PARAM_SETS; i++) 8920c1bc742181ded4930842b46e9507372f0b1b963James Dong { 8930c1bc742181ded4930842b46e9507372f0b1b963James Dong if ( pStorage->pps[i] && 8940c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->sps[pStorage->pps[i]->seqParameterSetId] && 8950c1bc742181ded4930842b46e9507372f0b1b963James Dong CheckPps(pStorage->pps[i], 8960c1bc742181ded4930842b46e9507372f0b1b963James Dong pStorage->sps[pStorage->pps[i]->seqParameterSetId]) == 8970c1bc742181ded4930842b46e9507372f0b1b963James Dong HANTRO_OK) 8980c1bc742181ded4930842b46e9507372f0b1b963James Dong { 8990c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_OK); 9000c1bc742181ded4930842b46e9507372f0b1b963James Dong } 9010c1bc742181ded4930842b46e9507372f0b1b963James Dong } 9020c1bc742181ded4930842b46e9507372f0b1b963James Dong 9030c1bc742181ded4930842b46e9507372f0b1b963James Dong return(HANTRO_NOK); 9040c1bc742181ded4930842b46e9507372f0b1b963James Dong 9050c1bc742181ded4930842b46e9507372f0b1b963James Dong} 9060c1bc742181ded4930842b46e9507372f0b1b963James Dong 907