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