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