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          h264bsdInit
270c1bc742181ded4930842b46e9507372f0b1b963James Dong          h264bsdDecode
280c1bc742181ded4930842b46e9507372f0b1b963James Dong          h264bsdShutdown
290c1bc742181ded4930842b46e9507372f0b1b963James Dong          h264bsdCurrentImage
300c1bc742181ded4930842b46e9507372f0b1b963James Dong          h264bsdNextOutputPicture
310c1bc742181ded4930842b46e9507372f0b1b963James Dong          h264bsdPicWidth
320c1bc742181ded4930842b46e9507372f0b1b963James Dong          h264bsdPicHeight
330c1bc742181ded4930842b46e9507372f0b1b963James Dong          h264bsdFlushBuffer
340c1bc742181ded4930842b46e9507372f0b1b963James Dong          h264bsdCheckValidParamSets
350c1bc742181ded4930842b46e9507372f0b1b963James Dong          h264bsdVideoRange
360c1bc742181ded4930842b46e9507372f0b1b963James Dong          h264bsdMatrixCoefficients
370c1bc742181ded4930842b46e9507372f0b1b963James Dong          h264bsdCroppingParams
380c1bc742181ded4930842b46e9507372f0b1b963James Dong
390c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/
400c1bc742181ded4930842b46e9507372f0b1b963James Dong
410c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------
420c1bc742181ded4930842b46e9507372f0b1b963James Dong    1. Include headers
430c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/
440c1bc742181ded4930842b46e9507372f0b1b963James Dong
450c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_decoder.h"
460c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_nal_unit.h"
470c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_byte_stream.h"
480c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_seq_param_set.h"
490c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_pic_param_set.h"
500c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_slice_header.h"
510c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_slice_data.h"
520c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_neighbour.h"
530c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_util.h"
540c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_dpb.h"
550c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_deblocking.h"
560c1bc742181ded4930842b46e9507372f0b1b963James Dong#include "h264bsd_conceal.h"
570c1bc742181ded4930842b46e9507372f0b1b963James Dong
580c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------
590c1bc742181ded4930842b46e9507372f0b1b963James Dong    2. External compiler flags
600c1bc742181ded4930842b46e9507372f0b1b963James Dong--------------------------------------------------------------------------------
610c1bc742181ded4930842b46e9507372f0b1b963James Dong
620c1bc742181ded4930842b46e9507372f0b1b963James Dong--------------------------------------------------------------------------------
630c1bc742181ded4930842b46e9507372f0b1b963James Dong    3. Module defines
640c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/
650c1bc742181ded4930842b46e9507372f0b1b963James Dong
660c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------
670c1bc742181ded4930842b46e9507372f0b1b963James Dong    4. Local function prototypes
680c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/
690c1bc742181ded4930842b46e9507372f0b1b963James Dong
700c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------
710c1bc742181ded4930842b46e9507372f0b1b963James Dong
720c1bc742181ded4930842b46e9507372f0b1b963James Dong    Function name: h264bsdInit
730c1bc742181ded4930842b46e9507372f0b1b963James Dong
740c1bc742181ded4930842b46e9507372f0b1b963James Dong        Functional description:
750c1bc742181ded4930842b46e9507372f0b1b963James Dong            Initialize the decoder.
760c1bc742181ded4930842b46e9507372f0b1b963James Dong
770c1bc742181ded4930842b46e9507372f0b1b963James Dong        Inputs:
780c1bc742181ded4930842b46e9507372f0b1b963James Dong            noOutputReordering  flag to indicate the decoder that it does not
790c1bc742181ded4930842b46e9507372f0b1b963James Dong                                have to perform reordering of display images.
800c1bc742181ded4930842b46e9507372f0b1b963James Dong
810c1bc742181ded4930842b46e9507372f0b1b963James Dong        Outputs:
820c1bc742181ded4930842b46e9507372f0b1b963James Dong            pStorage            pointer to initialized storage structure
830c1bc742181ded4930842b46e9507372f0b1b963James Dong
840c1bc742181ded4930842b46e9507372f0b1b963James Dong        Returns:
850c1bc742181ded4930842b46e9507372f0b1b963James Dong            none
860c1bc742181ded4930842b46e9507372f0b1b963James Dong
870c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/
880c1bc742181ded4930842b46e9507372f0b1b963James Dong
890c1bc742181ded4930842b46e9507372f0b1b963James Dongu32 h264bsdInit(storage_t *pStorage, u32 noOutputReordering)
900c1bc742181ded4930842b46e9507372f0b1b963James Dong{
910c1bc742181ded4930842b46e9507372f0b1b963James Dong
920c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */
930c1bc742181ded4930842b46e9507372f0b1b963James Dong    u32 size;
940c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */
950c1bc742181ded4930842b46e9507372f0b1b963James Dong
960c1bc742181ded4930842b46e9507372f0b1b963James Dong    ASSERT(pStorage);
970c1bc742181ded4930842b46e9507372f0b1b963James Dong
980c1bc742181ded4930842b46e9507372f0b1b963James Dong    h264bsdInitStorage(pStorage);
990c1bc742181ded4930842b46e9507372f0b1b963James Dong
1000c1bc742181ded4930842b46e9507372f0b1b963James Dong    /* allocate mbLayer to be next multiple of 64 to enable use of
1010c1bc742181ded4930842b46e9507372f0b1b963James Dong     * specific NEON optimized "memset" for clearing the structure */
1020c1bc742181ded4930842b46e9507372f0b1b963James Dong    size = (sizeof(macroblockLayer_t) + 63) & ~0x3F;
1030c1bc742181ded4930842b46e9507372f0b1b963James Dong
1040c1bc742181ded4930842b46e9507372f0b1b963James Dong    pStorage->mbLayer = (macroblockLayer_t*)H264SwDecMalloc(size);
1050c1bc742181ded4930842b46e9507372f0b1b963James Dong    if (!pStorage->mbLayer)
1060c1bc742181ded4930842b46e9507372f0b1b963James Dong        return HANTRO_NOK;
1070c1bc742181ded4930842b46e9507372f0b1b963James Dong
1080c1bc742181ded4930842b46e9507372f0b1b963James Dong    if (noOutputReordering)
1090c1bc742181ded4930842b46e9507372f0b1b963James Dong        pStorage->noReordering = HANTRO_TRUE;
1100c1bc742181ded4930842b46e9507372f0b1b963James Dong
1110c1bc742181ded4930842b46e9507372f0b1b963James Dong    return HANTRO_OK;
1120c1bc742181ded4930842b46e9507372f0b1b963James Dong}
1130c1bc742181ded4930842b46e9507372f0b1b963James Dong
1140c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------
1150c1bc742181ded4930842b46e9507372f0b1b963James Dong
1160c1bc742181ded4930842b46e9507372f0b1b963James Dong    Function: h264bsdDecode
1170c1bc742181ded4930842b46e9507372f0b1b963James Dong
1180c1bc742181ded4930842b46e9507372f0b1b963James Dong        Functional description:
1190c1bc742181ded4930842b46e9507372f0b1b963James Dong            Decode a NAL unit. This function calls other modules to perform
1200c1bc742181ded4930842b46e9507372f0b1b963James Dong            tasks like
1210c1bc742181ded4930842b46e9507372f0b1b963James Dong                * extract and decode NAL unit from the byte stream
1220c1bc742181ded4930842b46e9507372f0b1b963James Dong                * decode parameter sets
1230c1bc742181ded4930842b46e9507372f0b1b963James Dong                * decode slice header and slice data
1240c1bc742181ded4930842b46e9507372f0b1b963James Dong                * conceal errors in the picture
1250c1bc742181ded4930842b46e9507372f0b1b963James Dong                * perform deblocking filtering
1260c1bc742181ded4930842b46e9507372f0b1b963James Dong
1270c1bc742181ded4930842b46e9507372f0b1b963James Dong            This function contains top level control logic of the decoder.
1280c1bc742181ded4930842b46e9507372f0b1b963James Dong
1290c1bc742181ded4930842b46e9507372f0b1b963James Dong        Inputs:
1300c1bc742181ded4930842b46e9507372f0b1b963James Dong            pStorage        pointer to storage data structure
1310c1bc742181ded4930842b46e9507372f0b1b963James Dong            byteStrm        pointer to stream buffer given by application
1320c1bc742181ded4930842b46e9507372f0b1b963James Dong            len             length of the buffer in bytes
1330c1bc742181ded4930842b46e9507372f0b1b963James Dong            picId           identifier for a picture, assigned by the
1340c1bc742181ded4930842b46e9507372f0b1b963James Dong                            application
1350c1bc742181ded4930842b46e9507372f0b1b963James Dong
1360c1bc742181ded4930842b46e9507372f0b1b963James Dong        Outputs:
1370c1bc742181ded4930842b46e9507372f0b1b963James Dong            readBytes       number of bytes read from the stream is stored
1380c1bc742181ded4930842b46e9507372f0b1b963James Dong                            here
1390c1bc742181ded4930842b46e9507372f0b1b963James Dong
1400c1bc742181ded4930842b46e9507372f0b1b963James Dong        Returns:
1410c1bc742181ded4930842b46e9507372f0b1b963James Dong            H264BSD_RDY             decoding finished, nothing special
1420c1bc742181ded4930842b46e9507372f0b1b963James Dong            H264BSD_PIC_RDY         decoding of a picture finished
1430c1bc742181ded4930842b46e9507372f0b1b963James Dong            H264BSD_HDRS_RDY        param sets activated, information like
1440c1bc742181ded4930842b46e9507372f0b1b963James Dong                                    picture dimensions etc can be read
1450c1bc742181ded4930842b46e9507372f0b1b963James Dong            H264BSD_ERROR           error in decoding
1460c1bc742181ded4930842b46e9507372f0b1b963James Dong            H264BSD_PARAM_SET_ERROR serius error in decoding, failed to
1470c1bc742181ded4930842b46e9507372f0b1b963James Dong                                    activate param sets
1480c1bc742181ded4930842b46e9507372f0b1b963James Dong
1490c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/
1500c1bc742181ded4930842b46e9507372f0b1b963James Dong
1510c1bc742181ded4930842b46e9507372f0b1b963James Dongu32 h264bsdDecode(storage_t *pStorage, u8 *byteStrm, u32 len, u32 picId,
1520c1bc742181ded4930842b46e9507372f0b1b963James Dong    u32 *readBytes)
1530c1bc742181ded4930842b46e9507372f0b1b963James Dong{
1540c1bc742181ded4930842b46e9507372f0b1b963James Dong
1550c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */
1560c1bc742181ded4930842b46e9507372f0b1b963James Dong
1570c1bc742181ded4930842b46e9507372f0b1b963James Dong    u32 tmp, ppsId, spsId;
1580c1bc742181ded4930842b46e9507372f0b1b963James Dong    i32 picOrderCnt;
1590c1bc742181ded4930842b46e9507372f0b1b963James Dong    nalUnit_t nalUnit;
1600c1bc742181ded4930842b46e9507372f0b1b963James Dong    seqParamSet_t seqParamSet;
1610c1bc742181ded4930842b46e9507372f0b1b963James Dong    picParamSet_t picParamSet;
1620c1bc742181ded4930842b46e9507372f0b1b963James Dong    strmData_t strm;
1630c1bc742181ded4930842b46e9507372f0b1b963James Dong    u32 accessUnitBoundaryFlag = HANTRO_FALSE;
1640c1bc742181ded4930842b46e9507372f0b1b963James Dong    u32 picReady = HANTRO_FALSE;
1650c1bc742181ded4930842b46e9507372f0b1b963James Dong
1660c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */
1670c1bc742181ded4930842b46e9507372f0b1b963James Dong
1680c1bc742181ded4930842b46e9507372f0b1b963James Dong    ASSERT(pStorage);
1690c1bc742181ded4930842b46e9507372f0b1b963James Dong    ASSERT(byteStrm);
1700c1bc742181ded4930842b46e9507372f0b1b963James Dong    ASSERT(len);
1710c1bc742181ded4930842b46e9507372f0b1b963James Dong    ASSERT(readBytes);
1720c1bc742181ded4930842b46e9507372f0b1b963James Dong
1730c1bc742181ded4930842b46e9507372f0b1b963James Dong    /* if previous buffer was not finished and same pointer given -> skip NAL
1740c1bc742181ded4930842b46e9507372f0b1b963James Dong     * unit extraction */
1750c1bc742181ded4930842b46e9507372f0b1b963James Dong    if (pStorage->prevBufNotFinished && byteStrm == pStorage->prevBufPointer)
1760c1bc742181ded4930842b46e9507372f0b1b963James Dong    {
1770c1bc742181ded4930842b46e9507372f0b1b963James Dong        strm = pStorage->strm[0];
1780c1bc742181ded4930842b46e9507372f0b1b963James Dong        strm.pStrmCurrPos = strm.pStrmBuffStart;
1790c1bc742181ded4930842b46e9507372f0b1b963James Dong        strm.strmBuffReadBits = strm.bitPosInWord = 0;
1800c1bc742181ded4930842b46e9507372f0b1b963James Dong        *readBytes = pStorage->prevBytesConsumed;
1810c1bc742181ded4930842b46e9507372f0b1b963James Dong    }
1820c1bc742181ded4930842b46e9507372f0b1b963James Dong    else
1830c1bc742181ded4930842b46e9507372f0b1b963James Dong    {
1840c1bc742181ded4930842b46e9507372f0b1b963James Dong        tmp = h264bsdExtractNalUnit(byteStrm, len, &strm, readBytes);
1850c1bc742181ded4930842b46e9507372f0b1b963James Dong        if (tmp != HANTRO_OK)
1860c1bc742181ded4930842b46e9507372f0b1b963James Dong        {
1870c1bc742181ded4930842b46e9507372f0b1b963James Dong            EPRINT("BYTE_STREAM");
1880c1bc742181ded4930842b46e9507372f0b1b963James Dong            return(H264BSD_ERROR);
1890c1bc742181ded4930842b46e9507372f0b1b963James Dong        }
1900c1bc742181ded4930842b46e9507372f0b1b963James Dong        /* store stream */
1910c1bc742181ded4930842b46e9507372f0b1b963James Dong        pStorage->strm[0] = strm;
1920c1bc742181ded4930842b46e9507372f0b1b963James Dong        pStorage->prevBytesConsumed = *readBytes;
1930c1bc742181ded4930842b46e9507372f0b1b963James Dong        pStorage->prevBufPointer = byteStrm;
1940c1bc742181ded4930842b46e9507372f0b1b963James Dong    }
1950c1bc742181ded4930842b46e9507372f0b1b963James Dong    pStorage->prevBufNotFinished = HANTRO_FALSE;
1960c1bc742181ded4930842b46e9507372f0b1b963James Dong
1970c1bc742181ded4930842b46e9507372f0b1b963James Dong    tmp = h264bsdDecodeNalUnit(&strm, &nalUnit);
1980c1bc742181ded4930842b46e9507372f0b1b963James Dong    if (tmp != HANTRO_OK)
1990c1bc742181ded4930842b46e9507372f0b1b963James Dong    {
2000c1bc742181ded4930842b46e9507372f0b1b963James Dong        EPRINT("NAL_UNIT");
2010c1bc742181ded4930842b46e9507372f0b1b963James Dong        return(H264BSD_ERROR);
2020c1bc742181ded4930842b46e9507372f0b1b963James Dong    }
2030c1bc742181ded4930842b46e9507372f0b1b963James Dong
2040c1bc742181ded4930842b46e9507372f0b1b963James Dong    /* Discard unspecified, reserved, SPS extension and auxiliary picture slices */
2050c1bc742181ded4930842b46e9507372f0b1b963James Dong    if(nalUnit.nalUnitType == 0 || nalUnit.nalUnitType >= 13)
2060c1bc742181ded4930842b46e9507372f0b1b963James Dong    {
2070c1bc742181ded4930842b46e9507372f0b1b963James Dong        DEBUG(("DISCARDED NAL (UNSPECIFIED, REGISTERED, SPS ext or AUX slice)\n"));
2080c1bc742181ded4930842b46e9507372f0b1b963James Dong        return(H264BSD_RDY);
2090c1bc742181ded4930842b46e9507372f0b1b963James Dong    }
2100c1bc742181ded4930842b46e9507372f0b1b963James Dong
2110c1bc742181ded4930842b46e9507372f0b1b963James Dong    tmp = h264bsdCheckAccessUnitBoundary(
2120c1bc742181ded4930842b46e9507372f0b1b963James Dong      &strm,
2130c1bc742181ded4930842b46e9507372f0b1b963James Dong      &nalUnit,
2140c1bc742181ded4930842b46e9507372f0b1b963James Dong      pStorage,
2150c1bc742181ded4930842b46e9507372f0b1b963James Dong      &accessUnitBoundaryFlag);
2160c1bc742181ded4930842b46e9507372f0b1b963James Dong    if (tmp != HANTRO_OK)
2170c1bc742181ded4930842b46e9507372f0b1b963James Dong    {
2180c1bc742181ded4930842b46e9507372f0b1b963James Dong        EPRINT("ACCESS UNIT BOUNDARY CHECK");
2190c1bc742181ded4930842b46e9507372f0b1b963James Dong        if (tmp == PARAM_SET_ERROR)
2200c1bc742181ded4930842b46e9507372f0b1b963James Dong            return(H264BSD_PARAM_SET_ERROR);
2210c1bc742181ded4930842b46e9507372f0b1b963James Dong        else
2220c1bc742181ded4930842b46e9507372f0b1b963James Dong            return(H264BSD_ERROR);
2230c1bc742181ded4930842b46e9507372f0b1b963James Dong    }
2240c1bc742181ded4930842b46e9507372f0b1b963James Dong
2250c1bc742181ded4930842b46e9507372f0b1b963James Dong    if ( accessUnitBoundaryFlag )
2260c1bc742181ded4930842b46e9507372f0b1b963James Dong    {
2270c1bc742181ded4930842b46e9507372f0b1b963James Dong        DEBUG(("Access unit boundary\n"));
2280c1bc742181ded4930842b46e9507372f0b1b963James Dong        /* conceal if picture started and param sets activated */
2290c1bc742181ded4930842b46e9507372f0b1b963James Dong        if (pStorage->picStarted && pStorage->activeSps != NULL)
2300c1bc742181ded4930842b46e9507372f0b1b963James Dong        {
2310c1bc742181ded4930842b46e9507372f0b1b963James Dong            DEBUG(("CONCEALING..."));
2320c1bc742181ded4930842b46e9507372f0b1b963James Dong
2330c1bc742181ded4930842b46e9507372f0b1b963James Dong            /* return error if second phase of
2340c1bc742181ded4930842b46e9507372f0b1b963James Dong             * initialization is not completed */
2350c1bc742181ded4930842b46e9507372f0b1b963James Dong            if (pStorage->pendingActivation)
2360c1bc742181ded4930842b46e9507372f0b1b963James Dong            {
2370c1bc742181ded4930842b46e9507372f0b1b963James Dong                EPRINT("Pending activation not completed");
2380c1bc742181ded4930842b46e9507372f0b1b963James Dong                return (H264BSD_ERROR);
2390c1bc742181ded4930842b46e9507372f0b1b963James Dong            }
2400c1bc742181ded4930842b46e9507372f0b1b963James Dong
2410c1bc742181ded4930842b46e9507372f0b1b963James Dong            if (!pStorage->validSliceInAccessUnit)
2420c1bc742181ded4930842b46e9507372f0b1b963James Dong            {
2430c1bc742181ded4930842b46e9507372f0b1b963James Dong                pStorage->currImage->data =
2440c1bc742181ded4930842b46e9507372f0b1b963James Dong                    h264bsdAllocateDpbImage(pStorage->dpb);
2450c1bc742181ded4930842b46e9507372f0b1b963James Dong                h264bsdInitRefPicList(pStorage->dpb);
2460c1bc742181ded4930842b46e9507372f0b1b963James Dong                tmp = h264bsdConceal(pStorage, pStorage->currImage, P_SLICE);
2470c1bc742181ded4930842b46e9507372f0b1b963James Dong            }
2480c1bc742181ded4930842b46e9507372f0b1b963James Dong            else
2490c1bc742181ded4930842b46e9507372f0b1b963James Dong                tmp = h264bsdConceal(pStorage, pStorage->currImage,
2500c1bc742181ded4930842b46e9507372f0b1b963James Dong                    pStorage->sliceHeader->sliceType);
2510c1bc742181ded4930842b46e9507372f0b1b963James Dong
2520c1bc742181ded4930842b46e9507372f0b1b963James Dong            picReady = HANTRO_TRUE;
2530c1bc742181ded4930842b46e9507372f0b1b963James Dong
2540c1bc742181ded4930842b46e9507372f0b1b963James Dong            /* current NAL unit should be decoded on next activation -> set
2550c1bc742181ded4930842b46e9507372f0b1b963James Dong             * readBytes to 0 */
2560c1bc742181ded4930842b46e9507372f0b1b963James Dong            *readBytes = 0;
2570c1bc742181ded4930842b46e9507372f0b1b963James Dong            pStorage->prevBufNotFinished = HANTRO_TRUE;
2580c1bc742181ded4930842b46e9507372f0b1b963James Dong            DEBUG(("...DONE\n"));
2590c1bc742181ded4930842b46e9507372f0b1b963James Dong        }
2600c1bc742181ded4930842b46e9507372f0b1b963James Dong        else
2610c1bc742181ded4930842b46e9507372f0b1b963James Dong        {
2620c1bc742181ded4930842b46e9507372f0b1b963James Dong            pStorage->validSliceInAccessUnit = HANTRO_FALSE;
2630c1bc742181ded4930842b46e9507372f0b1b963James Dong        }
2640c1bc742181ded4930842b46e9507372f0b1b963James Dong        pStorage->skipRedundantSlices = HANTRO_FALSE;
2650c1bc742181ded4930842b46e9507372f0b1b963James Dong    }
2660c1bc742181ded4930842b46e9507372f0b1b963James Dong
2670c1bc742181ded4930842b46e9507372f0b1b963James Dong    if (!picReady)
2680c1bc742181ded4930842b46e9507372f0b1b963James Dong    {
2690c1bc742181ded4930842b46e9507372f0b1b963James Dong        switch (nalUnit.nalUnitType)
2700c1bc742181ded4930842b46e9507372f0b1b963James Dong        {
2710c1bc742181ded4930842b46e9507372f0b1b963James Dong            case NAL_SEQ_PARAM_SET:
2720c1bc742181ded4930842b46e9507372f0b1b963James Dong                DEBUG(("SEQ PARAM SET\n"));
2730c1bc742181ded4930842b46e9507372f0b1b963James Dong                tmp = h264bsdDecodeSeqParamSet(&strm, &seqParamSet);
2740c1bc742181ded4930842b46e9507372f0b1b963James Dong                if (tmp != HANTRO_OK)
2750c1bc742181ded4930842b46e9507372f0b1b963James Dong                {
2760c1bc742181ded4930842b46e9507372f0b1b963James Dong                    EPRINT("SEQ_PARAM_SET");
2770c1bc742181ded4930842b46e9507372f0b1b963James Dong                    FREE(seqParamSet.offsetForRefFrame);
2780c1bc742181ded4930842b46e9507372f0b1b963James Dong                    FREE(seqParamSet.vuiParameters);
2790c1bc742181ded4930842b46e9507372f0b1b963James Dong                    return(H264BSD_ERROR);
2800c1bc742181ded4930842b46e9507372f0b1b963James Dong                }
2810c1bc742181ded4930842b46e9507372f0b1b963James Dong                tmp = h264bsdStoreSeqParamSet(pStorage, &seqParamSet);
2820c1bc742181ded4930842b46e9507372f0b1b963James Dong                break;
2830c1bc742181ded4930842b46e9507372f0b1b963James Dong
2840c1bc742181ded4930842b46e9507372f0b1b963James Dong            case NAL_PIC_PARAM_SET:
2850c1bc742181ded4930842b46e9507372f0b1b963James Dong                DEBUG(("PIC PARAM SET\n"));
2860c1bc742181ded4930842b46e9507372f0b1b963James Dong                tmp = h264bsdDecodePicParamSet(&strm, &picParamSet);
2870c1bc742181ded4930842b46e9507372f0b1b963James Dong                if (tmp != HANTRO_OK)
2880c1bc742181ded4930842b46e9507372f0b1b963James Dong                {
2890c1bc742181ded4930842b46e9507372f0b1b963James Dong                    EPRINT("PIC_PARAM_SET");
2900c1bc742181ded4930842b46e9507372f0b1b963James Dong                    FREE(picParamSet.runLength);
2910c1bc742181ded4930842b46e9507372f0b1b963James Dong                    FREE(picParamSet.topLeft);
2920c1bc742181ded4930842b46e9507372f0b1b963James Dong                    FREE(picParamSet.bottomRight);
2930c1bc742181ded4930842b46e9507372f0b1b963James Dong                    FREE(picParamSet.sliceGroupId);
2940c1bc742181ded4930842b46e9507372f0b1b963James Dong                    return(H264BSD_ERROR);
2950c1bc742181ded4930842b46e9507372f0b1b963James Dong                }
2960c1bc742181ded4930842b46e9507372f0b1b963James Dong                tmp = h264bsdStorePicParamSet(pStorage, &picParamSet);
2970c1bc742181ded4930842b46e9507372f0b1b963James Dong                break;
2980c1bc742181ded4930842b46e9507372f0b1b963James Dong
2990c1bc742181ded4930842b46e9507372f0b1b963James Dong            case NAL_CODED_SLICE_IDR:
3000c1bc742181ded4930842b46e9507372f0b1b963James Dong                DEBUG(("IDR "));
3010c1bc742181ded4930842b46e9507372f0b1b963James Dong                /* fall through */
3020c1bc742181ded4930842b46e9507372f0b1b963James Dong            case NAL_CODED_SLICE:
3030c1bc742181ded4930842b46e9507372f0b1b963James Dong                DEBUG(("SLICE HEADER\n"));
3040c1bc742181ded4930842b46e9507372f0b1b963James Dong
3050c1bc742181ded4930842b46e9507372f0b1b963James Dong                /* picture successfully finished and still decoding same old
3060c1bc742181ded4930842b46e9507372f0b1b963James Dong                 * access unit -> no need to decode redundant slices */
3070c1bc742181ded4930842b46e9507372f0b1b963James Dong                if (pStorage->skipRedundantSlices)
3080c1bc742181ded4930842b46e9507372f0b1b963James Dong                    return(H264BSD_RDY);
3090c1bc742181ded4930842b46e9507372f0b1b963James Dong
3100c1bc742181ded4930842b46e9507372f0b1b963James Dong                pStorage->picStarted = HANTRO_TRUE;
3110c1bc742181ded4930842b46e9507372f0b1b963James Dong
3120c1bc742181ded4930842b46e9507372f0b1b963James Dong                if (h264bsdIsStartOfPicture(pStorage))
3130c1bc742181ded4930842b46e9507372f0b1b963James Dong                {
3140c1bc742181ded4930842b46e9507372f0b1b963James Dong                    pStorage->numConcealedMbs = 0;
3150c1bc742181ded4930842b46e9507372f0b1b963James Dong                    pStorage->currentPicId    = picId;
3160c1bc742181ded4930842b46e9507372f0b1b963James Dong
3170c1bc742181ded4930842b46e9507372f0b1b963James Dong                    tmp = h264bsdCheckPpsId(&strm, &ppsId);
3180c1bc742181ded4930842b46e9507372f0b1b963James Dong                    ASSERT(tmp == HANTRO_OK);
3190c1bc742181ded4930842b46e9507372f0b1b963James Dong                    /* store old activeSpsId and return headers ready
3200c1bc742181ded4930842b46e9507372f0b1b963James Dong                     * indication if activeSps changes */
3210c1bc742181ded4930842b46e9507372f0b1b963James Dong                    spsId = pStorage->activeSpsId;
3220c1bc742181ded4930842b46e9507372f0b1b963James Dong                    tmp = h264bsdActivateParamSets(pStorage, ppsId,
3230c1bc742181ded4930842b46e9507372f0b1b963James Dong                            IS_IDR_NAL_UNIT(&nalUnit) ?
3240c1bc742181ded4930842b46e9507372f0b1b963James Dong                            HANTRO_TRUE : HANTRO_FALSE);
3250c1bc742181ded4930842b46e9507372f0b1b963James Dong                    if (tmp != HANTRO_OK)
3260c1bc742181ded4930842b46e9507372f0b1b963James Dong                    {
3270c1bc742181ded4930842b46e9507372f0b1b963James Dong                        EPRINT("Param set activation");
3280c1bc742181ded4930842b46e9507372f0b1b963James Dong                        pStorage->activePpsId = MAX_NUM_PIC_PARAM_SETS;
3290c1bc742181ded4930842b46e9507372f0b1b963James Dong                        pStorage->activePps = NULL;
3300c1bc742181ded4930842b46e9507372f0b1b963James Dong                        pStorage->activeSpsId = MAX_NUM_SEQ_PARAM_SETS;
3310c1bc742181ded4930842b46e9507372f0b1b963James Dong                        pStorage->activeSps = NULL;
3320c1bc742181ded4930842b46e9507372f0b1b963James Dong                        pStorage->pendingActivation = HANTRO_FALSE;
3330c1bc742181ded4930842b46e9507372f0b1b963James Dong
3340c1bc742181ded4930842b46e9507372f0b1b963James Dong                        if(tmp == MEMORY_ALLOCATION_ERROR)
3350c1bc742181ded4930842b46e9507372f0b1b963James Dong                        {
3360c1bc742181ded4930842b46e9507372f0b1b963James Dong                            return H264BSD_MEMALLOC_ERROR;
3370c1bc742181ded4930842b46e9507372f0b1b963James Dong                        }
3380c1bc742181ded4930842b46e9507372f0b1b963James Dong                        else
3390c1bc742181ded4930842b46e9507372f0b1b963James Dong                            return(H264BSD_PARAM_SET_ERROR);
3400c1bc742181ded4930842b46e9507372f0b1b963James Dong                    }
3410c1bc742181ded4930842b46e9507372f0b1b963James Dong
3420c1bc742181ded4930842b46e9507372f0b1b963James Dong                    if (spsId != pStorage->activeSpsId)
3430c1bc742181ded4930842b46e9507372f0b1b963James Dong                    {
3440c1bc742181ded4930842b46e9507372f0b1b963James Dong                        seqParamSet_t *oldSPS = NULL;
3450c1bc742181ded4930842b46e9507372f0b1b963James Dong                        seqParamSet_t *newSPS = pStorage->activeSps;
3460c1bc742181ded4930842b46e9507372f0b1b963James Dong                        u32 noOutputOfPriorPicsFlag = 1;
3470c1bc742181ded4930842b46e9507372f0b1b963James Dong
3480c1bc742181ded4930842b46e9507372f0b1b963James Dong                        if(pStorage->oldSpsId < MAX_NUM_SEQ_PARAM_SETS)
3490c1bc742181ded4930842b46e9507372f0b1b963James Dong                        {
3500c1bc742181ded4930842b46e9507372f0b1b963James Dong                            oldSPS = pStorage->sps[pStorage->oldSpsId];
3510c1bc742181ded4930842b46e9507372f0b1b963James Dong                        }
3520c1bc742181ded4930842b46e9507372f0b1b963James Dong
3530c1bc742181ded4930842b46e9507372f0b1b963James Dong                        *readBytes = 0;
3540c1bc742181ded4930842b46e9507372f0b1b963James Dong                        pStorage->prevBufNotFinished = HANTRO_TRUE;
3550c1bc742181ded4930842b46e9507372f0b1b963James Dong
3560c1bc742181ded4930842b46e9507372f0b1b963James Dong
3570c1bc742181ded4930842b46e9507372f0b1b963James Dong                        if(nalUnit.nalUnitType == NAL_CODED_SLICE_IDR)
3580c1bc742181ded4930842b46e9507372f0b1b963James Dong                        {
3590c1bc742181ded4930842b46e9507372f0b1b963James Dong                            tmp =
3600c1bc742181ded4930842b46e9507372f0b1b963James Dong                            h264bsdCheckPriorPicsFlag(&noOutputOfPriorPicsFlag,
3610c1bc742181ded4930842b46e9507372f0b1b963James Dong                                                          &strm, newSPS,
3620c1bc742181ded4930842b46e9507372f0b1b963James Dong                                                          pStorage->activePps,
3630c1bc742181ded4930842b46e9507372f0b1b963James Dong                                                          nalUnit.nalUnitType);
3640c1bc742181ded4930842b46e9507372f0b1b963James Dong                        }
3650c1bc742181ded4930842b46e9507372f0b1b963James Dong                        else
3660c1bc742181ded4930842b46e9507372f0b1b963James Dong                        {
3670c1bc742181ded4930842b46e9507372f0b1b963James Dong                            tmp = HANTRO_NOK;
3680c1bc742181ded4930842b46e9507372f0b1b963James Dong                        }
3690c1bc742181ded4930842b46e9507372f0b1b963James Dong
3700c1bc742181ded4930842b46e9507372f0b1b963James Dong                        if((tmp != HANTRO_OK) ||
3710c1bc742181ded4930842b46e9507372f0b1b963James Dong                           (noOutputOfPriorPicsFlag != 0) ||
3720c1bc742181ded4930842b46e9507372f0b1b963James Dong                           (pStorage->dpb->noReordering) ||
3730c1bc742181ded4930842b46e9507372f0b1b963James Dong                           (oldSPS == NULL) ||
3740c1bc742181ded4930842b46e9507372f0b1b963James Dong                           (oldSPS->picWidthInMbs != newSPS->picWidthInMbs) ||
3750c1bc742181ded4930842b46e9507372f0b1b963James Dong                           (oldSPS->picHeightInMbs != newSPS->picHeightInMbs) ||
3760c1bc742181ded4930842b46e9507372f0b1b963James Dong                           (oldSPS->maxDpbSize != newSPS->maxDpbSize))
3770c1bc742181ded4930842b46e9507372f0b1b963James Dong                        {
3780c1bc742181ded4930842b46e9507372f0b1b963James Dong                            pStorage->dpb->flushed = 0;
3790c1bc742181ded4930842b46e9507372f0b1b963James Dong                        }
3800c1bc742181ded4930842b46e9507372f0b1b963James Dong                        else
3810c1bc742181ded4930842b46e9507372f0b1b963James Dong                        {
3820c1bc742181ded4930842b46e9507372f0b1b963James Dong                            h264bsdFlushDpb(pStorage->dpb);
3830c1bc742181ded4930842b46e9507372f0b1b963James Dong                        }
3840c1bc742181ded4930842b46e9507372f0b1b963James Dong
3850c1bc742181ded4930842b46e9507372f0b1b963James Dong                        pStorage->oldSpsId = pStorage->activeSpsId;
3860c1bc742181ded4930842b46e9507372f0b1b963James Dong
3870c1bc742181ded4930842b46e9507372f0b1b963James Dong                        return(H264BSD_HDRS_RDY);
3880c1bc742181ded4930842b46e9507372f0b1b963James Dong                    }
3890c1bc742181ded4930842b46e9507372f0b1b963James Dong                }
3900c1bc742181ded4930842b46e9507372f0b1b963James Dong
3910c1bc742181ded4930842b46e9507372f0b1b963James Dong                /* return error if second phase of
3920c1bc742181ded4930842b46e9507372f0b1b963James Dong                 * initialization is not completed */
3930c1bc742181ded4930842b46e9507372f0b1b963James Dong                if (pStorage->pendingActivation)
3940c1bc742181ded4930842b46e9507372f0b1b963James Dong                {
3950c1bc742181ded4930842b46e9507372f0b1b963James Dong                    EPRINT("Pending activation not completed");
3960c1bc742181ded4930842b46e9507372f0b1b963James Dong                    return (H264BSD_ERROR);
3970c1bc742181ded4930842b46e9507372f0b1b963James Dong                }
3980c1bc742181ded4930842b46e9507372f0b1b963James Dong                tmp = h264bsdDecodeSliceHeader(&strm, pStorage->sliceHeader + 1,
3990c1bc742181ded4930842b46e9507372f0b1b963James Dong                    pStorage->activeSps, pStorage->activePps, &nalUnit);
4000c1bc742181ded4930842b46e9507372f0b1b963James Dong                if (tmp != HANTRO_OK)
4010c1bc742181ded4930842b46e9507372f0b1b963James Dong                {
4020c1bc742181ded4930842b46e9507372f0b1b963James Dong                    EPRINT("SLICE_HEADER");
4030c1bc742181ded4930842b46e9507372f0b1b963James Dong                    return(H264BSD_ERROR);
4040c1bc742181ded4930842b46e9507372f0b1b963James Dong                }
4050c1bc742181ded4930842b46e9507372f0b1b963James Dong                if (h264bsdIsStartOfPicture(pStorage))
4060c1bc742181ded4930842b46e9507372f0b1b963James Dong                {
4070c1bc742181ded4930842b46e9507372f0b1b963James Dong                    if (!IS_IDR_NAL_UNIT(&nalUnit))
4080c1bc742181ded4930842b46e9507372f0b1b963James Dong                    {
4090c1bc742181ded4930842b46e9507372f0b1b963James Dong                        tmp = h264bsdCheckGapsInFrameNum(pStorage->dpb,
4100c1bc742181ded4930842b46e9507372f0b1b963James Dong                            pStorage->sliceHeader[1].frameNum,
4110c1bc742181ded4930842b46e9507372f0b1b963James Dong                            nalUnit.nalRefIdc != 0 ?
4120c1bc742181ded4930842b46e9507372f0b1b963James Dong                            HANTRO_TRUE : HANTRO_FALSE,
4130c1bc742181ded4930842b46e9507372f0b1b963James Dong                            pStorage->activeSps->
4140c1bc742181ded4930842b46e9507372f0b1b963James Dong                            gapsInFrameNumValueAllowedFlag);
4150c1bc742181ded4930842b46e9507372f0b1b963James Dong                        if (tmp != HANTRO_OK)
4160c1bc742181ded4930842b46e9507372f0b1b963James Dong                        {
4170c1bc742181ded4930842b46e9507372f0b1b963James Dong                            EPRINT("Gaps in frame num");
4180c1bc742181ded4930842b46e9507372f0b1b963James Dong                            return(H264BSD_ERROR);
4190c1bc742181ded4930842b46e9507372f0b1b963James Dong                        }
4200c1bc742181ded4930842b46e9507372f0b1b963James Dong                    }
4210c1bc742181ded4930842b46e9507372f0b1b963James Dong                    pStorage->currImage->data =
4220c1bc742181ded4930842b46e9507372f0b1b963James Dong                        h264bsdAllocateDpbImage(pStorage->dpb);
4230c1bc742181ded4930842b46e9507372f0b1b963James Dong                }
4240c1bc742181ded4930842b46e9507372f0b1b963James Dong
4250c1bc742181ded4930842b46e9507372f0b1b963James Dong                /* store slice header to storage if successfully decoded */
4260c1bc742181ded4930842b46e9507372f0b1b963James Dong                pStorage->sliceHeader[0] = pStorage->sliceHeader[1];
4270c1bc742181ded4930842b46e9507372f0b1b963James Dong                pStorage->validSliceInAccessUnit = HANTRO_TRUE;
4280c1bc742181ded4930842b46e9507372f0b1b963James Dong                pStorage->prevNalUnit[0] = nalUnit;
4290c1bc742181ded4930842b46e9507372f0b1b963James Dong
4300c1bc742181ded4930842b46e9507372f0b1b963James Dong                h264bsdComputeSliceGroupMap(pStorage,
4310c1bc742181ded4930842b46e9507372f0b1b963James Dong                    pStorage->sliceHeader->sliceGroupChangeCycle);
4320c1bc742181ded4930842b46e9507372f0b1b963James Dong
4330c1bc742181ded4930842b46e9507372f0b1b963James Dong                h264bsdInitRefPicList(pStorage->dpb);
4340c1bc742181ded4930842b46e9507372f0b1b963James Dong                tmp = h264bsdReorderRefPicList(pStorage->dpb,
4350c1bc742181ded4930842b46e9507372f0b1b963James Dong                    &pStorage->sliceHeader->refPicListReordering,
4360c1bc742181ded4930842b46e9507372f0b1b963James Dong                    pStorage->sliceHeader->frameNum,
4370c1bc742181ded4930842b46e9507372f0b1b963James Dong                    pStorage->sliceHeader->numRefIdxL0Active);
4380c1bc742181ded4930842b46e9507372f0b1b963James Dong                if (tmp != HANTRO_OK)
4390c1bc742181ded4930842b46e9507372f0b1b963James Dong                {
4400c1bc742181ded4930842b46e9507372f0b1b963James Dong                    EPRINT("Reordering");
4410c1bc742181ded4930842b46e9507372f0b1b963James Dong                    return(H264BSD_ERROR);
4420c1bc742181ded4930842b46e9507372f0b1b963James Dong                }
4430c1bc742181ded4930842b46e9507372f0b1b963James Dong
4440c1bc742181ded4930842b46e9507372f0b1b963James Dong                DEBUG(("SLICE DATA, FIRST %d\n",
4450c1bc742181ded4930842b46e9507372f0b1b963James Dong                        pStorage->sliceHeader->firstMbInSlice));
4460c1bc742181ded4930842b46e9507372f0b1b963James Dong                tmp = h264bsdDecodeSliceData(&strm, pStorage,
4470c1bc742181ded4930842b46e9507372f0b1b963James Dong                    pStorage->currImage, pStorage->sliceHeader);
4480c1bc742181ded4930842b46e9507372f0b1b963James Dong                if (tmp != HANTRO_OK)
4490c1bc742181ded4930842b46e9507372f0b1b963James Dong                {
4500c1bc742181ded4930842b46e9507372f0b1b963James Dong                    EPRINT("SLICE_DATA");
4510c1bc742181ded4930842b46e9507372f0b1b963James Dong                    h264bsdMarkSliceCorrupted(pStorage,
4520c1bc742181ded4930842b46e9507372f0b1b963James Dong                        pStorage->sliceHeader->firstMbInSlice);
4530c1bc742181ded4930842b46e9507372f0b1b963James Dong                    return(H264BSD_ERROR);
4540c1bc742181ded4930842b46e9507372f0b1b963James Dong                }
4550c1bc742181ded4930842b46e9507372f0b1b963James Dong
4560c1bc742181ded4930842b46e9507372f0b1b963James Dong                if (h264bsdIsEndOfPicture(pStorage))
4570c1bc742181ded4930842b46e9507372f0b1b963James Dong                {
4580c1bc742181ded4930842b46e9507372f0b1b963James Dong                    picReady = HANTRO_TRUE;
4590c1bc742181ded4930842b46e9507372f0b1b963James Dong                    pStorage->skipRedundantSlices = HANTRO_TRUE;
4600c1bc742181ded4930842b46e9507372f0b1b963James Dong                }
4610c1bc742181ded4930842b46e9507372f0b1b963James Dong                break;
4620c1bc742181ded4930842b46e9507372f0b1b963James Dong
4630c1bc742181ded4930842b46e9507372f0b1b963James Dong            case NAL_SEI:
4640c1bc742181ded4930842b46e9507372f0b1b963James Dong                DEBUG(("SEI MESSAGE, NOT DECODED"));
4650c1bc742181ded4930842b46e9507372f0b1b963James Dong                break;
4660c1bc742181ded4930842b46e9507372f0b1b963James Dong
4670c1bc742181ded4930842b46e9507372f0b1b963James Dong            default:
4680c1bc742181ded4930842b46e9507372f0b1b963James Dong                DEBUG(("NOT IMPLEMENTED YET %d\n",nalUnit.nalUnitType));
4690c1bc742181ded4930842b46e9507372f0b1b963James Dong        }
4700c1bc742181ded4930842b46e9507372f0b1b963James Dong    }
4710c1bc742181ded4930842b46e9507372f0b1b963James Dong
4720c1bc742181ded4930842b46e9507372f0b1b963James Dong    if (picReady)
4730c1bc742181ded4930842b46e9507372f0b1b963James Dong    {
4740c1bc742181ded4930842b46e9507372f0b1b963James Dong        h264bsdFilterPicture(pStorage->currImage, pStorage->mb);
4750c1bc742181ded4930842b46e9507372f0b1b963James Dong
4760c1bc742181ded4930842b46e9507372f0b1b963James Dong        h264bsdResetStorage(pStorage);
4770c1bc742181ded4930842b46e9507372f0b1b963James Dong
4780c1bc742181ded4930842b46e9507372f0b1b963James Dong        picOrderCnt = h264bsdDecodePicOrderCnt(pStorage->poc,
4790c1bc742181ded4930842b46e9507372f0b1b963James Dong            pStorage->activeSps, pStorage->sliceHeader, pStorage->prevNalUnit);
4800c1bc742181ded4930842b46e9507372f0b1b963James Dong
4810c1bc742181ded4930842b46e9507372f0b1b963James Dong        if (pStorage->validSliceInAccessUnit)
4820c1bc742181ded4930842b46e9507372f0b1b963James Dong        {
4830c1bc742181ded4930842b46e9507372f0b1b963James Dong            if (pStorage->prevNalUnit->nalRefIdc)
4840c1bc742181ded4930842b46e9507372f0b1b963James Dong            {
4850c1bc742181ded4930842b46e9507372f0b1b963James Dong                tmp = h264bsdMarkDecRefPic(pStorage->dpb,
4860c1bc742181ded4930842b46e9507372f0b1b963James Dong                    &pStorage->sliceHeader->decRefPicMarking,
4870c1bc742181ded4930842b46e9507372f0b1b963James Dong                    pStorage->currImage, pStorage->sliceHeader->frameNum,
4880c1bc742181ded4930842b46e9507372f0b1b963James Dong                    picOrderCnt,
4890c1bc742181ded4930842b46e9507372f0b1b963James Dong                    IS_IDR_NAL_UNIT(pStorage->prevNalUnit) ?
4900c1bc742181ded4930842b46e9507372f0b1b963James Dong                    HANTRO_TRUE : HANTRO_FALSE,
4910c1bc742181ded4930842b46e9507372f0b1b963James Dong                    pStorage->currentPicId, pStorage->numConcealedMbs);
4920c1bc742181ded4930842b46e9507372f0b1b963James Dong            }
4930c1bc742181ded4930842b46e9507372f0b1b963James Dong            /* non-reference picture, just store for possible display
4940c1bc742181ded4930842b46e9507372f0b1b963James Dong             * reordering */
4950c1bc742181ded4930842b46e9507372f0b1b963James Dong            else
4960c1bc742181ded4930842b46e9507372f0b1b963James Dong            {
4970c1bc742181ded4930842b46e9507372f0b1b963James Dong                tmp = h264bsdMarkDecRefPic(pStorage->dpb, NULL,
4980c1bc742181ded4930842b46e9507372f0b1b963James Dong                    pStorage->currImage, pStorage->sliceHeader->frameNum,
4990c1bc742181ded4930842b46e9507372f0b1b963James Dong                    picOrderCnt,
5000c1bc742181ded4930842b46e9507372f0b1b963James Dong                    IS_IDR_NAL_UNIT(pStorage->prevNalUnit) ?
5010c1bc742181ded4930842b46e9507372f0b1b963James Dong                    HANTRO_TRUE : HANTRO_FALSE,
5020c1bc742181ded4930842b46e9507372f0b1b963James Dong                    pStorage->currentPicId, pStorage->numConcealedMbs);
5030c1bc742181ded4930842b46e9507372f0b1b963James Dong            }
5040c1bc742181ded4930842b46e9507372f0b1b963James Dong        }
5050c1bc742181ded4930842b46e9507372f0b1b963James Dong
5060c1bc742181ded4930842b46e9507372f0b1b963James Dong        pStorage->picStarted = HANTRO_FALSE;
5070c1bc742181ded4930842b46e9507372f0b1b963James Dong        pStorage->validSliceInAccessUnit = HANTRO_FALSE;
5080c1bc742181ded4930842b46e9507372f0b1b963James Dong
5090c1bc742181ded4930842b46e9507372f0b1b963James Dong        return(H264BSD_PIC_RDY);
5100c1bc742181ded4930842b46e9507372f0b1b963James Dong    }
5110c1bc742181ded4930842b46e9507372f0b1b963James Dong    else
5120c1bc742181ded4930842b46e9507372f0b1b963James Dong        return(H264BSD_RDY);
5130c1bc742181ded4930842b46e9507372f0b1b963James Dong
5140c1bc742181ded4930842b46e9507372f0b1b963James Dong}
5150c1bc742181ded4930842b46e9507372f0b1b963James Dong
5160c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------
5170c1bc742181ded4930842b46e9507372f0b1b963James Dong
5180c1bc742181ded4930842b46e9507372f0b1b963James Dong    Function: h264bsdShutdown
5190c1bc742181ded4930842b46e9507372f0b1b963James Dong
5200c1bc742181ded4930842b46e9507372f0b1b963James Dong        Functional description:
5210c1bc742181ded4930842b46e9507372f0b1b963James Dong            Shutdown a decoder instance. Function frees all the memories
5220c1bc742181ded4930842b46e9507372f0b1b963James Dong            allocated for the decoder instance.
5230c1bc742181ded4930842b46e9507372f0b1b963James Dong
5240c1bc742181ded4930842b46e9507372f0b1b963James Dong        Inputs:
5250c1bc742181ded4930842b46e9507372f0b1b963James Dong            pStorage    pointer to storage data structure
5260c1bc742181ded4930842b46e9507372f0b1b963James Dong
5270c1bc742181ded4930842b46e9507372f0b1b963James Dong        Returns:
5280c1bc742181ded4930842b46e9507372f0b1b963James Dong            none
5290c1bc742181ded4930842b46e9507372f0b1b963James Dong
5300c1bc742181ded4930842b46e9507372f0b1b963James Dong
5310c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/
5320c1bc742181ded4930842b46e9507372f0b1b963James Dong
5330c1bc742181ded4930842b46e9507372f0b1b963James Dongvoid h264bsdShutdown(storage_t *pStorage)
5340c1bc742181ded4930842b46e9507372f0b1b963James Dong{
5350c1bc742181ded4930842b46e9507372f0b1b963James Dong
5360c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */
5370c1bc742181ded4930842b46e9507372f0b1b963James Dong
5380c1bc742181ded4930842b46e9507372f0b1b963James Dong    u32 i;
5390c1bc742181ded4930842b46e9507372f0b1b963James Dong
5400c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */
5410c1bc742181ded4930842b46e9507372f0b1b963James Dong
5420c1bc742181ded4930842b46e9507372f0b1b963James Dong    ASSERT(pStorage);
5430c1bc742181ded4930842b46e9507372f0b1b963James Dong
5440c1bc742181ded4930842b46e9507372f0b1b963James Dong    for (i = 0; i < MAX_NUM_SEQ_PARAM_SETS; i++)
5450c1bc742181ded4930842b46e9507372f0b1b963James Dong    {
5460c1bc742181ded4930842b46e9507372f0b1b963James Dong        if (pStorage->sps[i])
5470c1bc742181ded4930842b46e9507372f0b1b963James Dong        {
5480c1bc742181ded4930842b46e9507372f0b1b963James Dong            FREE(pStorage->sps[i]->offsetForRefFrame);
5490c1bc742181ded4930842b46e9507372f0b1b963James Dong            FREE(pStorage->sps[i]->vuiParameters);
5500c1bc742181ded4930842b46e9507372f0b1b963James Dong            FREE(pStorage->sps[i]);
5510c1bc742181ded4930842b46e9507372f0b1b963James Dong        }
5520c1bc742181ded4930842b46e9507372f0b1b963James Dong    }
5530c1bc742181ded4930842b46e9507372f0b1b963James Dong
5540c1bc742181ded4930842b46e9507372f0b1b963James Dong    for (i = 0; i < MAX_NUM_PIC_PARAM_SETS; i++)
5550c1bc742181ded4930842b46e9507372f0b1b963James Dong    {
5560c1bc742181ded4930842b46e9507372f0b1b963James Dong        if (pStorage->pps[i])
5570c1bc742181ded4930842b46e9507372f0b1b963James Dong        {
5580c1bc742181ded4930842b46e9507372f0b1b963James Dong            FREE(pStorage->pps[i]->runLength);
5590c1bc742181ded4930842b46e9507372f0b1b963James Dong            FREE(pStorage->pps[i]->topLeft);
5600c1bc742181ded4930842b46e9507372f0b1b963James Dong            FREE(pStorage->pps[i]->bottomRight);
5610c1bc742181ded4930842b46e9507372f0b1b963James Dong            FREE(pStorage->pps[i]->sliceGroupId);
5620c1bc742181ded4930842b46e9507372f0b1b963James Dong            FREE(pStorage->pps[i]);
5630c1bc742181ded4930842b46e9507372f0b1b963James Dong        }
5640c1bc742181ded4930842b46e9507372f0b1b963James Dong    }
5650c1bc742181ded4930842b46e9507372f0b1b963James Dong
5660c1bc742181ded4930842b46e9507372f0b1b963James Dong    FREE(pStorage->mbLayer);
5670c1bc742181ded4930842b46e9507372f0b1b963James Dong    FREE(pStorage->mb);
5680c1bc742181ded4930842b46e9507372f0b1b963James Dong    FREE(pStorage->sliceGroupMap);
5690c1bc742181ded4930842b46e9507372f0b1b963James Dong
5700c1bc742181ded4930842b46e9507372f0b1b963James Dong    h264bsdFreeDpb(pStorage->dpb);
5710c1bc742181ded4930842b46e9507372f0b1b963James Dong
5720c1bc742181ded4930842b46e9507372f0b1b963James Dong}
5730c1bc742181ded4930842b46e9507372f0b1b963James Dong
5740c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------
5750c1bc742181ded4930842b46e9507372f0b1b963James Dong
5760c1bc742181ded4930842b46e9507372f0b1b963James Dong    Function: h264bsdNextOutputPicture
5770c1bc742181ded4930842b46e9507372f0b1b963James Dong
5780c1bc742181ded4930842b46e9507372f0b1b963James Dong        Functional description:
5790c1bc742181ded4930842b46e9507372f0b1b963James Dong            Get next output picture in display order.
5800c1bc742181ded4930842b46e9507372f0b1b963James Dong
5810c1bc742181ded4930842b46e9507372f0b1b963James Dong        Inputs:
5820c1bc742181ded4930842b46e9507372f0b1b963James Dong            pStorage    pointer to storage data structure
5830c1bc742181ded4930842b46e9507372f0b1b963James Dong
5840c1bc742181ded4930842b46e9507372f0b1b963James Dong        Outputs:
5850c1bc742181ded4930842b46e9507372f0b1b963James Dong            picId       identifier of the picture will be stored here
5860c1bc742181ded4930842b46e9507372f0b1b963James Dong            isIdrPic    IDR flag of the picture will be stored here
5870c1bc742181ded4930842b46e9507372f0b1b963James Dong            numErrMbs   number of concealed macroblocks in the picture
5880c1bc742181ded4930842b46e9507372f0b1b963James Dong                        will be stored here
5890c1bc742181ded4930842b46e9507372f0b1b963James Dong
5900c1bc742181ded4930842b46e9507372f0b1b963James Dong        Returns:
5910c1bc742181ded4930842b46e9507372f0b1b963James Dong            pointer to the picture data
5920c1bc742181ded4930842b46e9507372f0b1b963James Dong            NULL if no pictures available for display
5930c1bc742181ded4930842b46e9507372f0b1b963James Dong
5940c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/
5950c1bc742181ded4930842b46e9507372f0b1b963James Dong
5960c1bc742181ded4930842b46e9507372f0b1b963James Dongu8* h264bsdNextOutputPicture(storage_t *pStorage, u32 *picId, u32 *isIdrPic,
5970c1bc742181ded4930842b46e9507372f0b1b963James Dong    u32 *numErrMbs)
5980c1bc742181ded4930842b46e9507372f0b1b963James Dong{
5990c1bc742181ded4930842b46e9507372f0b1b963James Dong
6000c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */
6010c1bc742181ded4930842b46e9507372f0b1b963James Dong
6020c1bc742181ded4930842b46e9507372f0b1b963James Dong    dpbOutPicture_t *pOut;
6030c1bc742181ded4930842b46e9507372f0b1b963James Dong
6040c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */
6050c1bc742181ded4930842b46e9507372f0b1b963James Dong
6060c1bc742181ded4930842b46e9507372f0b1b963James Dong    ASSERT(pStorage);
6070c1bc742181ded4930842b46e9507372f0b1b963James Dong
6080c1bc742181ded4930842b46e9507372f0b1b963James Dong    pOut = h264bsdDpbOutputPicture(pStorage->dpb);
6090c1bc742181ded4930842b46e9507372f0b1b963James Dong
6100c1bc742181ded4930842b46e9507372f0b1b963James Dong    if (pOut != NULL)
6110c1bc742181ded4930842b46e9507372f0b1b963James Dong    {
6120c1bc742181ded4930842b46e9507372f0b1b963James Dong        *picId = pOut->picId;
6130c1bc742181ded4930842b46e9507372f0b1b963James Dong        *isIdrPic = pOut->isIdr;
6140c1bc742181ded4930842b46e9507372f0b1b963James Dong        *numErrMbs = pOut->numErrMbs;
6150c1bc742181ded4930842b46e9507372f0b1b963James Dong        return (pOut->data);
6160c1bc742181ded4930842b46e9507372f0b1b963James Dong    }
6170c1bc742181ded4930842b46e9507372f0b1b963James Dong    else
6180c1bc742181ded4930842b46e9507372f0b1b963James Dong        return(NULL);
6190c1bc742181ded4930842b46e9507372f0b1b963James Dong
6200c1bc742181ded4930842b46e9507372f0b1b963James Dong}
6210c1bc742181ded4930842b46e9507372f0b1b963James Dong
6220c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------
6230c1bc742181ded4930842b46e9507372f0b1b963James Dong
6240c1bc742181ded4930842b46e9507372f0b1b963James Dong    Function: h264bsdPicWidth
6250c1bc742181ded4930842b46e9507372f0b1b963James Dong
6260c1bc742181ded4930842b46e9507372f0b1b963James Dong        Functional description:
6270c1bc742181ded4930842b46e9507372f0b1b963James Dong            Get width of the picture in macroblocks
6280c1bc742181ded4930842b46e9507372f0b1b963James Dong
6290c1bc742181ded4930842b46e9507372f0b1b963James Dong        Inputs:
6300c1bc742181ded4930842b46e9507372f0b1b963James Dong            pStorage    pointer to storage data structure
6310c1bc742181ded4930842b46e9507372f0b1b963James Dong
6320c1bc742181ded4930842b46e9507372f0b1b963James Dong        Outputs:
6330c1bc742181ded4930842b46e9507372f0b1b963James Dong            none
6340c1bc742181ded4930842b46e9507372f0b1b963James Dong
6350c1bc742181ded4930842b46e9507372f0b1b963James Dong        Returns:
6360c1bc742181ded4930842b46e9507372f0b1b963James Dong            picture width
6370c1bc742181ded4930842b46e9507372f0b1b963James Dong            0 if parameters sets not yet activated
6380c1bc742181ded4930842b46e9507372f0b1b963James Dong
6390c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/
6400c1bc742181ded4930842b46e9507372f0b1b963James Dong
6410c1bc742181ded4930842b46e9507372f0b1b963James Dongu32 h264bsdPicWidth(storage_t *pStorage)
6420c1bc742181ded4930842b46e9507372f0b1b963James Dong{
6430c1bc742181ded4930842b46e9507372f0b1b963James Dong
6440c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */
6450c1bc742181ded4930842b46e9507372f0b1b963James Dong
6460c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */
6470c1bc742181ded4930842b46e9507372f0b1b963James Dong
6480c1bc742181ded4930842b46e9507372f0b1b963James Dong    ASSERT(pStorage);
6490c1bc742181ded4930842b46e9507372f0b1b963James Dong
6500c1bc742181ded4930842b46e9507372f0b1b963James Dong    if (pStorage->activeSps)
6510c1bc742181ded4930842b46e9507372f0b1b963James Dong        return(pStorage->activeSps->picWidthInMbs);
6520c1bc742181ded4930842b46e9507372f0b1b963James Dong    else
6530c1bc742181ded4930842b46e9507372f0b1b963James Dong        return(0);
6540c1bc742181ded4930842b46e9507372f0b1b963James Dong
6550c1bc742181ded4930842b46e9507372f0b1b963James Dong}
6560c1bc742181ded4930842b46e9507372f0b1b963James Dong
6570c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------
6580c1bc742181ded4930842b46e9507372f0b1b963James Dong
6590c1bc742181ded4930842b46e9507372f0b1b963James Dong    Function: h264bsdPicHeight
6600c1bc742181ded4930842b46e9507372f0b1b963James Dong
6610c1bc742181ded4930842b46e9507372f0b1b963James Dong        Functional description:
6620c1bc742181ded4930842b46e9507372f0b1b963James Dong            Get height of the picture in macroblocks
6630c1bc742181ded4930842b46e9507372f0b1b963James Dong
6640c1bc742181ded4930842b46e9507372f0b1b963James Dong        Inputs:
6650c1bc742181ded4930842b46e9507372f0b1b963James Dong            pStorage    pointer to storage data structure
6660c1bc742181ded4930842b46e9507372f0b1b963James Dong
6670c1bc742181ded4930842b46e9507372f0b1b963James Dong        Outputs:
6680c1bc742181ded4930842b46e9507372f0b1b963James Dong            none
6690c1bc742181ded4930842b46e9507372f0b1b963James Dong
6700c1bc742181ded4930842b46e9507372f0b1b963James Dong        Returns:
6710c1bc742181ded4930842b46e9507372f0b1b963James Dong            picture width
6720c1bc742181ded4930842b46e9507372f0b1b963James Dong            0 if parameters sets not yet activated
6730c1bc742181ded4930842b46e9507372f0b1b963James Dong
6740c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/
6750c1bc742181ded4930842b46e9507372f0b1b963James Dong
6760c1bc742181ded4930842b46e9507372f0b1b963James Dongu32 h264bsdPicHeight(storage_t *pStorage)
6770c1bc742181ded4930842b46e9507372f0b1b963James Dong{
6780c1bc742181ded4930842b46e9507372f0b1b963James Dong
6790c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */
6800c1bc742181ded4930842b46e9507372f0b1b963James Dong
6810c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */
6820c1bc742181ded4930842b46e9507372f0b1b963James Dong
6830c1bc742181ded4930842b46e9507372f0b1b963James Dong    ASSERT(pStorage);
6840c1bc742181ded4930842b46e9507372f0b1b963James Dong
6850c1bc742181ded4930842b46e9507372f0b1b963James Dong    if (pStorage->activeSps)
6860c1bc742181ded4930842b46e9507372f0b1b963James Dong        return(pStorage->activeSps->picHeightInMbs);
6870c1bc742181ded4930842b46e9507372f0b1b963James Dong    else
6880c1bc742181ded4930842b46e9507372f0b1b963James Dong        return(0);
6890c1bc742181ded4930842b46e9507372f0b1b963James Dong
6900c1bc742181ded4930842b46e9507372f0b1b963James Dong}
6910c1bc742181ded4930842b46e9507372f0b1b963James Dong
6920c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------
6930c1bc742181ded4930842b46e9507372f0b1b963James Dong
6940c1bc742181ded4930842b46e9507372f0b1b963James Dong    Function: h264bsdFlushBuffer
6950c1bc742181ded4930842b46e9507372f0b1b963James Dong
6960c1bc742181ded4930842b46e9507372f0b1b963James Dong        Functional description:
6970c1bc742181ded4930842b46e9507372f0b1b963James Dong            Flush the decoded picture buffer, see dpb.c for details
6980c1bc742181ded4930842b46e9507372f0b1b963James Dong
6990c1bc742181ded4930842b46e9507372f0b1b963James Dong        Inputs:
7000c1bc742181ded4930842b46e9507372f0b1b963James Dong            pStorage    pointer to storage data structure
7010c1bc742181ded4930842b46e9507372f0b1b963James Dong
7020c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/
7030c1bc742181ded4930842b46e9507372f0b1b963James Dong
7040c1bc742181ded4930842b46e9507372f0b1b963James Dongvoid h264bsdFlushBuffer(storage_t *pStorage)
7050c1bc742181ded4930842b46e9507372f0b1b963James Dong{
7060c1bc742181ded4930842b46e9507372f0b1b963James Dong
7070c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */
7080c1bc742181ded4930842b46e9507372f0b1b963James Dong
7090c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */
7100c1bc742181ded4930842b46e9507372f0b1b963James Dong
7110c1bc742181ded4930842b46e9507372f0b1b963James Dong    ASSERT(pStorage);
7120c1bc742181ded4930842b46e9507372f0b1b963James Dong
7130c1bc742181ded4930842b46e9507372f0b1b963James Dong    h264bsdFlushDpb(pStorage->dpb);
7140c1bc742181ded4930842b46e9507372f0b1b963James Dong
7150c1bc742181ded4930842b46e9507372f0b1b963James Dong}
7160c1bc742181ded4930842b46e9507372f0b1b963James Dong
7170c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------
7180c1bc742181ded4930842b46e9507372f0b1b963James Dong
7190c1bc742181ded4930842b46e9507372f0b1b963James Dong    Function: h264bsdCheckValidParamSets
7200c1bc742181ded4930842b46e9507372f0b1b963James Dong
7210c1bc742181ded4930842b46e9507372f0b1b963James Dong        Functional description:
7220c1bc742181ded4930842b46e9507372f0b1b963James Dong            Check if any valid parameter set combinations (SPS/PPS) exists.
7230c1bc742181ded4930842b46e9507372f0b1b963James Dong
7240c1bc742181ded4930842b46e9507372f0b1b963James Dong        Inputs:
7250c1bc742181ded4930842b46e9507372f0b1b963James Dong            pStorage    pointer to storage structure
7260c1bc742181ded4930842b46e9507372f0b1b963James Dong
7270c1bc742181ded4930842b46e9507372f0b1b963James Dong        Returns:
7280c1bc742181ded4930842b46e9507372f0b1b963James Dong            1       at least one valid SPS/PPS combination found
7290c1bc742181ded4930842b46e9507372f0b1b963James Dong            0       no valid param set combinations found
7300c1bc742181ded4930842b46e9507372f0b1b963James Dong
7310c1bc742181ded4930842b46e9507372f0b1b963James Dong
7320c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/
7330c1bc742181ded4930842b46e9507372f0b1b963James Dong
7340c1bc742181ded4930842b46e9507372f0b1b963James Dongu32 h264bsdCheckValidParamSets(storage_t *pStorage)
7350c1bc742181ded4930842b46e9507372f0b1b963James Dong{
7360c1bc742181ded4930842b46e9507372f0b1b963James Dong
7370c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */
7380c1bc742181ded4930842b46e9507372f0b1b963James Dong
7390c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */
7400c1bc742181ded4930842b46e9507372f0b1b963James Dong
7410c1bc742181ded4930842b46e9507372f0b1b963James Dong    ASSERT(pStorage);
7420c1bc742181ded4930842b46e9507372f0b1b963James Dong
7430c1bc742181ded4930842b46e9507372f0b1b963James Dong    return(h264bsdValidParamSets(pStorage) == HANTRO_OK ? 1 : 0);
7440c1bc742181ded4930842b46e9507372f0b1b963James Dong
7450c1bc742181ded4930842b46e9507372f0b1b963James Dong}
7460c1bc742181ded4930842b46e9507372f0b1b963James Dong
7470c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------
7480c1bc742181ded4930842b46e9507372f0b1b963James Dong
7490c1bc742181ded4930842b46e9507372f0b1b963James Dong    Function: h264bsdVideoRange
7500c1bc742181ded4930842b46e9507372f0b1b963James Dong
7510c1bc742181ded4930842b46e9507372f0b1b963James Dong        Functional description:
7520c1bc742181ded4930842b46e9507372f0b1b963James Dong            Get value of video_full_range_flag received in the VUI data.
7530c1bc742181ded4930842b46e9507372f0b1b963James Dong
7540c1bc742181ded4930842b46e9507372f0b1b963James Dong        Inputs:
7550c1bc742181ded4930842b46e9507372f0b1b963James Dong            pStorage    pointer to storage structure
7560c1bc742181ded4930842b46e9507372f0b1b963James Dong
7570c1bc742181ded4930842b46e9507372f0b1b963James Dong        Returns:
7580c1bc742181ded4930842b46e9507372f0b1b963James Dong            1   video_full_range_flag received and value is 1
7590c1bc742181ded4930842b46e9507372f0b1b963James Dong            0   otherwise
7600c1bc742181ded4930842b46e9507372f0b1b963James Dong
7610c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/
7620c1bc742181ded4930842b46e9507372f0b1b963James Dong
7630c1bc742181ded4930842b46e9507372f0b1b963James Dongu32 h264bsdVideoRange(storage_t *pStorage)
7640c1bc742181ded4930842b46e9507372f0b1b963James Dong{
7650c1bc742181ded4930842b46e9507372f0b1b963James Dong
7660c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */
7670c1bc742181ded4930842b46e9507372f0b1b963James Dong
7680c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */
7690c1bc742181ded4930842b46e9507372f0b1b963James Dong
7700c1bc742181ded4930842b46e9507372f0b1b963James Dong    ASSERT(pStorage);
7710c1bc742181ded4930842b46e9507372f0b1b963James Dong
7720c1bc742181ded4930842b46e9507372f0b1b963James Dong    if (pStorage->activeSps && pStorage->activeSps->vuiParametersPresentFlag &&
7730c1bc742181ded4930842b46e9507372f0b1b963James Dong        pStorage->activeSps->vuiParameters &&
7740c1bc742181ded4930842b46e9507372f0b1b963James Dong        pStorage->activeSps->vuiParameters->videoSignalTypePresentFlag &&
7750c1bc742181ded4930842b46e9507372f0b1b963James Dong        pStorage->activeSps->vuiParameters->videoFullRangeFlag)
7760c1bc742181ded4930842b46e9507372f0b1b963James Dong        return(1);
7770c1bc742181ded4930842b46e9507372f0b1b963James Dong    else /* default value of video_full_range_flag is 0 */
7780c1bc742181ded4930842b46e9507372f0b1b963James Dong        return(0);
7790c1bc742181ded4930842b46e9507372f0b1b963James Dong
7800c1bc742181ded4930842b46e9507372f0b1b963James Dong}
7810c1bc742181ded4930842b46e9507372f0b1b963James Dong
7820c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------
7830c1bc742181ded4930842b46e9507372f0b1b963James Dong
7840c1bc742181ded4930842b46e9507372f0b1b963James Dong    Function: h264bsdMatrixCoefficients
7850c1bc742181ded4930842b46e9507372f0b1b963James Dong
7860c1bc742181ded4930842b46e9507372f0b1b963James Dong        Functional description:
7870c1bc742181ded4930842b46e9507372f0b1b963James Dong            Get value of matrix_coefficients received in the VUI data
7880c1bc742181ded4930842b46e9507372f0b1b963James Dong
7890c1bc742181ded4930842b46e9507372f0b1b963James Dong        Inputs:
7900c1bc742181ded4930842b46e9507372f0b1b963James Dong            pStorage    pointer to storage structure
7910c1bc742181ded4930842b46e9507372f0b1b963James Dong
7920c1bc742181ded4930842b46e9507372f0b1b963James Dong        Outputs:
7930c1bc742181ded4930842b46e9507372f0b1b963James Dong            value of matrix_coefficients if received
7940c1bc742181ded4930842b46e9507372f0b1b963James Dong            2   otherwise (this is the default value)
7950c1bc742181ded4930842b46e9507372f0b1b963James Dong
7960c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/
7970c1bc742181ded4930842b46e9507372f0b1b963James Dong
7980c1bc742181ded4930842b46e9507372f0b1b963James Dongu32 h264bsdMatrixCoefficients(storage_t *pStorage)
7990c1bc742181ded4930842b46e9507372f0b1b963James Dong{
8000c1bc742181ded4930842b46e9507372f0b1b963James Dong
8010c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */
8020c1bc742181ded4930842b46e9507372f0b1b963James Dong
8030c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */
8040c1bc742181ded4930842b46e9507372f0b1b963James Dong
8050c1bc742181ded4930842b46e9507372f0b1b963James Dong    ASSERT(pStorage);
8060c1bc742181ded4930842b46e9507372f0b1b963James Dong
8070c1bc742181ded4930842b46e9507372f0b1b963James Dong    if (pStorage->activeSps && pStorage->activeSps->vuiParametersPresentFlag &&
8080c1bc742181ded4930842b46e9507372f0b1b963James Dong        pStorage->activeSps->vuiParameters &&
8090c1bc742181ded4930842b46e9507372f0b1b963James Dong        pStorage->activeSps->vuiParameters->videoSignalTypePresentFlag &&
8100c1bc742181ded4930842b46e9507372f0b1b963James Dong        pStorage->activeSps->vuiParameters->colourDescriptionPresentFlag)
8110c1bc742181ded4930842b46e9507372f0b1b963James Dong        return(pStorage->activeSps->vuiParameters->matrixCoefficients);
8120c1bc742181ded4930842b46e9507372f0b1b963James Dong    else /* default unspecified */
8130c1bc742181ded4930842b46e9507372f0b1b963James Dong        return(2);
8140c1bc742181ded4930842b46e9507372f0b1b963James Dong
8150c1bc742181ded4930842b46e9507372f0b1b963James Dong}
8160c1bc742181ded4930842b46e9507372f0b1b963James Dong
8170c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------
8180c1bc742181ded4930842b46e9507372f0b1b963James Dong
8190c1bc742181ded4930842b46e9507372f0b1b963James Dong    Function: hh264bsdCroppingParams
8200c1bc742181ded4930842b46e9507372f0b1b963James Dong
8210c1bc742181ded4930842b46e9507372f0b1b963James Dong        Functional description:
8220c1bc742181ded4930842b46e9507372f0b1b963James Dong            Get cropping parameters of the active SPS
8230c1bc742181ded4930842b46e9507372f0b1b963James Dong
8240c1bc742181ded4930842b46e9507372f0b1b963James Dong        Inputs:
8250c1bc742181ded4930842b46e9507372f0b1b963James Dong            pStorage    pointer to storage structure
8260c1bc742181ded4930842b46e9507372f0b1b963James Dong
8270c1bc742181ded4930842b46e9507372f0b1b963James Dong        Outputs:
8280c1bc742181ded4930842b46e9507372f0b1b963James Dong            croppingFlag    flag indicating if cropping params present is
8290c1bc742181ded4930842b46e9507372f0b1b963James Dong                            stored here
8300c1bc742181ded4930842b46e9507372f0b1b963James Dong            leftOffset      cropping left offset in pixels is stored here
8310c1bc742181ded4930842b46e9507372f0b1b963James Dong            width           width of the image after cropping is stored here
8320c1bc742181ded4930842b46e9507372f0b1b963James Dong            topOffset       cropping top offset in pixels is stored here
8330c1bc742181ded4930842b46e9507372f0b1b963James Dong            height          height of the image after cropping is stored here
8340c1bc742181ded4930842b46e9507372f0b1b963James Dong
8350c1bc742181ded4930842b46e9507372f0b1b963James Dong        Returns:
8360c1bc742181ded4930842b46e9507372f0b1b963James Dong            none
8370c1bc742181ded4930842b46e9507372f0b1b963James Dong
8380c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/
8390c1bc742181ded4930842b46e9507372f0b1b963James Dong
8400c1bc742181ded4930842b46e9507372f0b1b963James Dongvoid h264bsdCroppingParams(storage_t *pStorage, u32 *croppingFlag,
8410c1bc742181ded4930842b46e9507372f0b1b963James Dong    u32 *leftOffset, u32 *width, u32 *topOffset, u32 *height)
8420c1bc742181ded4930842b46e9507372f0b1b963James Dong{
8430c1bc742181ded4930842b46e9507372f0b1b963James Dong
8440c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */
8450c1bc742181ded4930842b46e9507372f0b1b963James Dong
8460c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */
8470c1bc742181ded4930842b46e9507372f0b1b963James Dong
8480c1bc742181ded4930842b46e9507372f0b1b963James Dong    ASSERT(pStorage);
8490c1bc742181ded4930842b46e9507372f0b1b963James Dong
8500c1bc742181ded4930842b46e9507372f0b1b963James Dong    if (pStorage->activeSps && pStorage->activeSps->frameCroppingFlag)
8510c1bc742181ded4930842b46e9507372f0b1b963James Dong    {
8520c1bc742181ded4930842b46e9507372f0b1b963James Dong        *croppingFlag = 1;
8530c1bc742181ded4930842b46e9507372f0b1b963James Dong        *leftOffset = 2 * pStorage->activeSps->frameCropLeftOffset;
8540c1bc742181ded4930842b46e9507372f0b1b963James Dong        *width = 16 * pStorage->activeSps->picWidthInMbs -
8550c1bc742181ded4930842b46e9507372f0b1b963James Dong                 2 * (pStorage->activeSps->frameCropLeftOffset +
8560c1bc742181ded4930842b46e9507372f0b1b963James Dong                      pStorage->activeSps->frameCropRightOffset);
8570c1bc742181ded4930842b46e9507372f0b1b963James Dong        *topOffset = 2 * pStorage->activeSps->frameCropTopOffset;
8580c1bc742181ded4930842b46e9507372f0b1b963James Dong        *height = 16 * pStorage->activeSps->picHeightInMbs -
8590c1bc742181ded4930842b46e9507372f0b1b963James Dong                  2 * (pStorage->activeSps->frameCropTopOffset +
8600c1bc742181ded4930842b46e9507372f0b1b963James Dong                       pStorage->activeSps->frameCropBottomOffset);
8610c1bc742181ded4930842b46e9507372f0b1b963James Dong    }
8620c1bc742181ded4930842b46e9507372f0b1b963James Dong    else
8630c1bc742181ded4930842b46e9507372f0b1b963James Dong    {
8640c1bc742181ded4930842b46e9507372f0b1b963James Dong        *croppingFlag = 0;
8650c1bc742181ded4930842b46e9507372f0b1b963James Dong        *leftOffset = 0;
8660c1bc742181ded4930842b46e9507372f0b1b963James Dong        *width = 0;
8670c1bc742181ded4930842b46e9507372f0b1b963James Dong        *topOffset = 0;
8680c1bc742181ded4930842b46e9507372f0b1b963James Dong        *height = 0;
8690c1bc742181ded4930842b46e9507372f0b1b963James Dong    }
8700c1bc742181ded4930842b46e9507372f0b1b963James Dong
8710c1bc742181ded4930842b46e9507372f0b1b963James Dong}
8720c1bc742181ded4930842b46e9507372f0b1b963James Dong
8730c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------
8740c1bc742181ded4930842b46e9507372f0b1b963James Dong
8750c1bc742181ded4930842b46e9507372f0b1b963James Dong    Function: h264bsdSampleAspectRatio
8760c1bc742181ded4930842b46e9507372f0b1b963James Dong
8770c1bc742181ded4930842b46e9507372f0b1b963James Dong        Functional description:
8780c1bc742181ded4930842b46e9507372f0b1b963James Dong            Get aspect ratio received in the VUI data
8790c1bc742181ded4930842b46e9507372f0b1b963James Dong
8800c1bc742181ded4930842b46e9507372f0b1b963James Dong        Inputs:
8810c1bc742181ded4930842b46e9507372f0b1b963James Dong            pStorage    pointer to storage structure
8820c1bc742181ded4930842b46e9507372f0b1b963James Dong
8830c1bc742181ded4930842b46e9507372f0b1b963James Dong        Outputs:
8840c1bc742181ded4930842b46e9507372f0b1b963James Dong            sarWidth    sample aspect ratio height
8850c1bc742181ded4930842b46e9507372f0b1b963James Dong            sarHeight   sample aspect ratio width
8860c1bc742181ded4930842b46e9507372f0b1b963James Dong
8870c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/
8880c1bc742181ded4930842b46e9507372f0b1b963James Dong
8890c1bc742181ded4930842b46e9507372f0b1b963James Dongvoid h264bsdSampleAspectRatio(storage_t *pStorage, u32 *sarWidth, u32 *sarHeight)
8900c1bc742181ded4930842b46e9507372f0b1b963James Dong{
8910c1bc742181ded4930842b46e9507372f0b1b963James Dong
8920c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Variables */
8930c1bc742181ded4930842b46e9507372f0b1b963James Dong    u32 w = 1;
8940c1bc742181ded4930842b46e9507372f0b1b963James Dong    u32 h = 1;
8950c1bc742181ded4930842b46e9507372f0b1b963James Dong/* Code */
8960c1bc742181ded4930842b46e9507372f0b1b963James Dong
8970c1bc742181ded4930842b46e9507372f0b1b963James Dong    ASSERT(pStorage);
8980c1bc742181ded4930842b46e9507372f0b1b963James Dong
8990c1bc742181ded4930842b46e9507372f0b1b963James Dong
9000c1bc742181ded4930842b46e9507372f0b1b963James Dong    if (pStorage->activeSps &&
9010c1bc742181ded4930842b46e9507372f0b1b963James Dong        pStorage->activeSps->vuiParametersPresentFlag &&
9020c1bc742181ded4930842b46e9507372f0b1b963James Dong        pStorage->activeSps->vuiParameters &&
9030c1bc742181ded4930842b46e9507372f0b1b963James Dong        pStorage->activeSps->vuiParameters->aspectRatioPresentFlag )
9040c1bc742181ded4930842b46e9507372f0b1b963James Dong    {
9050c1bc742181ded4930842b46e9507372f0b1b963James Dong        switch (pStorage->activeSps->vuiParameters->aspectRatioIdc)
9060c1bc742181ded4930842b46e9507372f0b1b963James Dong        {
9070c1bc742181ded4930842b46e9507372f0b1b963James Dong            case ASPECT_RATIO_UNSPECIFIED:  w =   0; h =  0; break;
9080c1bc742181ded4930842b46e9507372f0b1b963James Dong            case ASPECT_RATIO_1_1:          w =   1; h =  1; break;
9090c1bc742181ded4930842b46e9507372f0b1b963James Dong            case ASPECT_RATIO_12_11:        w =  12; h = 11; break;
9100c1bc742181ded4930842b46e9507372f0b1b963James Dong            case ASPECT_RATIO_10_11:        w =  10; h = 11; break;
9110c1bc742181ded4930842b46e9507372f0b1b963James Dong            case ASPECT_RATIO_16_11:        w =  16; h = 11; break;
9120c1bc742181ded4930842b46e9507372f0b1b963James Dong            case ASPECT_RATIO_40_33:        w =  40; h = 33; break;
9130c1bc742181ded4930842b46e9507372f0b1b963James Dong            case ASPECT_RATIO_24_11:        w =  24; h = 11; break;
9140c1bc742181ded4930842b46e9507372f0b1b963James Dong            case ASPECT_RATIO_20_11:        w =  20; h = 11; break;
9150c1bc742181ded4930842b46e9507372f0b1b963James Dong            case ASPECT_RATIO_32_11:        w =  32; h = 11; break;
9160c1bc742181ded4930842b46e9507372f0b1b963James Dong            case ASPECT_RATIO_80_33:        w =  80; h = 33; break;
9170c1bc742181ded4930842b46e9507372f0b1b963James Dong            case ASPECT_RATIO_18_11:        w =  18; h = 11; break;
9180c1bc742181ded4930842b46e9507372f0b1b963James Dong            case ASPECT_RATIO_15_11:        w =  15; h = 11; break;
9190c1bc742181ded4930842b46e9507372f0b1b963James Dong            case ASPECT_RATIO_64_33:        w =  64; h = 33; break;
9200c1bc742181ded4930842b46e9507372f0b1b963James Dong            case ASPECT_RATIO_160_99:       w = 160; h = 99; break;
9210c1bc742181ded4930842b46e9507372f0b1b963James Dong            case ASPECT_RATIO_EXTENDED_SAR:
9220c1bc742181ded4930842b46e9507372f0b1b963James Dong                w = pStorage->activeSps->vuiParameters->sarWidth;
9230c1bc742181ded4930842b46e9507372f0b1b963James Dong                h = pStorage->activeSps->vuiParameters->sarHeight;
9240c1bc742181ded4930842b46e9507372f0b1b963James Dong                if ((w == 0) || (h == 0))
9250c1bc742181ded4930842b46e9507372f0b1b963James Dong                    w = h = 0;
9260c1bc742181ded4930842b46e9507372f0b1b963James Dong                break;
9270c1bc742181ded4930842b46e9507372f0b1b963James Dong            default:
9280c1bc742181ded4930842b46e9507372f0b1b963James Dong                w = 0;
9290c1bc742181ded4930842b46e9507372f0b1b963James Dong                h = 0;
9300c1bc742181ded4930842b46e9507372f0b1b963James Dong                break;
9310c1bc742181ded4930842b46e9507372f0b1b963James Dong        }
9320c1bc742181ded4930842b46e9507372f0b1b963James Dong    }
9330c1bc742181ded4930842b46e9507372f0b1b963James Dong
9340c1bc742181ded4930842b46e9507372f0b1b963James Dong    /* set aspect ratio*/
9350c1bc742181ded4930842b46e9507372f0b1b963James Dong    *sarWidth = w;
9360c1bc742181ded4930842b46e9507372f0b1b963James Dong    *sarHeight = h;
9370c1bc742181ded4930842b46e9507372f0b1b963James Dong
9380c1bc742181ded4930842b46e9507372f0b1b963James Dong}
9390c1bc742181ded4930842b46e9507372f0b1b963James Dong
9400c1bc742181ded4930842b46e9507372f0b1b963James Dong/*------------------------------------------------------------------------------
9410c1bc742181ded4930842b46e9507372f0b1b963James Dong
9420c1bc742181ded4930842b46e9507372f0b1b963James Dong    Function: h264bsdProfile
9430c1bc742181ded4930842b46e9507372f0b1b963James Dong
9440c1bc742181ded4930842b46e9507372f0b1b963James Dong        Functional description:
9450c1bc742181ded4930842b46e9507372f0b1b963James Dong            Get profile information from active SPS
9460c1bc742181ded4930842b46e9507372f0b1b963James Dong
9470c1bc742181ded4930842b46e9507372f0b1b963James Dong        Inputs:
9480c1bc742181ded4930842b46e9507372f0b1b963James Dong            pStorage    pointer to storage structure
9490c1bc742181ded4930842b46e9507372f0b1b963James Dong
9500c1bc742181ded4930842b46e9507372f0b1b963James Dong        Outputs:
9510c1bc742181ded4930842b46e9507372f0b1b963James Dong            profile   current profile
9520c1bc742181ded4930842b46e9507372f0b1b963James Dong
9530c1bc742181ded4930842b46e9507372f0b1b963James Dong------------------------------------------------------------------------------*/
9540c1bc742181ded4930842b46e9507372f0b1b963James Dongu32 h264bsdProfile(storage_t *pStorage)
9550c1bc742181ded4930842b46e9507372f0b1b963James Dong{
9560c1bc742181ded4930842b46e9507372f0b1b963James Dong    if (pStorage->activeSps)
9570c1bc742181ded4930842b46e9507372f0b1b963James Dong        return pStorage->activeSps->profileIdc;
9580c1bc742181ded4930842b46e9507372f0b1b963James Dong    else
9590c1bc742181ded4930842b46e9507372f0b1b963James Dong        return 0;
9600c1bc742181ded4930842b46e9507372f0b1b963James Dong}
9610c1bc742181ded4930842b46e9507372f0b1b963James Dong
962