/* * Copyright (C) 2009 The Android Open Source Project * Modified for use by h264bsd standalone library * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * *      http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ /*------------------------------------------------------------------------------     Table of contents      1. Include headers     2. External compiler flags     3. Module defines     4. Local function prototypes     5. Functions          h264bsdDecodeSeqParamSet          GetDpbSize          h264bsdCompareSeqParamSets ------------------------------------------------------------------------------*/ /*------------------------------------------------------------------------------    1. Include headers------------------------------------------------------------------------------*/ #include "h264bsd_seq_param_set.h"#include "h264bsd_util.h"#include "h264bsd_vlc.h"#include "h264bsd_vui.h"#include "h264bsd_cfg.h" /*------------------------------------------------------------------------------    2. External compiler flags-------------------------------------------------------------------------------- --------------------------------------------------------------------------------    3. Module defines------------------------------------------------------------------------------*/ /* enumeration to indicate invalid return value from the GetDpbSize function */enum {INVALID_DPB_SIZE = 0x7FFFFFFF}; /*------------------------------------------------------------------------------    4. Local function prototypes------------------------------------------------------------------------------*/ static u32 GetDpbSize(u32 picSizeInMbs, u32 levelIdc); /*------------------------------------------------------------------------------     Function name: h264bsdDecodeSeqParamSet         Functional description:            Decode sequence parameter set information from the stream.             Function allocates memory for offsetForRefFrame array if            picture order count type is 1 and numRefFramesInPicOrderCntCycle            is greater than zero.         Inputs:            pStrmData       pointer to stream data structure         Outputs:            pSeqParamSet    decoded information is stored here         Returns:            HANTRO_OK       success            HANTRO_NOK      failure, invalid information or end of stream            MEMORY_ALLOCATION_ERROR for memory allocation failure ------------------------------------------------------------------------------*/ u32 h264bsdDecodeSeqParamSet(strmData_t *pStrmData, seqParamSet_t *pSeqParamSet){ /* Variables */     u32 tmp, i, value; /* Code */     ASSERT(pStrmData);    ASSERT(pSeqParamSet);     memset(pSeqParamSet, 0, sizeof(seqParamSet_t));     /* profile_idc */    tmp = h264bsdGetBits(pStrmData, 8);    if (tmp == END_OF_STREAM)        return(HANTRO_NOK);    if (tmp != 66) {        werrstr("not baseline profile: %d", tmp);        return(HANTRO_NOK);    }    pSeqParamSet->profileIdc = tmp;     /* constrained_set0_flag */    h264bsdGetBits(pStrmData, 1);    /* constrained_set1_flag */    h264bsdGetBits(pStrmData, 1);    /* constrained_set2_flag */    tmp = h264bsdGetBits(pStrmData, 1);     if (tmp == END_OF_STREAM)        return(HANTRO_NOK);     /* reserved_zero_5bits, values of these bits shall be ignored */    tmp = h264bsdGetBits(pStrmData, 5);    if (tmp == END_OF_STREAM)        return(HANTRO_NOK);     tmp = h264bsdGetBits(pStrmData, 8);    if (tmp == END_OF_STREAM)        return(HANTRO_NOK);    pSeqParamSet->levelIdc = tmp;     tmp = h264bsdDecodeExpGolombUnsigned(pStrmData,        &pSeqParamSet->seqParameterSetId);    if (tmp != HANTRO_OK)        return(tmp);    if (pSeqParamSet->seqParameterSetId >= MAX_NUM_SEQ_PARAM_SETS)    {        werrstr("seq_param_set_id out of range: %d", pSeqParamSet->seqParameterSetId);        return(HANTRO_NOK);    }     /* log2_max_frame_num_minus4 */    tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value);    if (tmp != HANTRO_OK)        return(tmp);    if (value > 12)    {        werrstr("log2_max_frame_num_minus4 out of range: %d", value);        return(HANTRO_NOK);    }    /* maxFrameNum = 2^(log2_max_frame_num_minus4 + 4) */    pSeqParamSet->maxFrameNum = 1 << (value+4);     /* valid POC types are 0, 1 and 2 */    tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value);    if (tmp != HANTRO_OK)        return(tmp);    if (value > 2)    {        werrstr("pic_order_cnt_type out of range: %d", value);        return(HANTRO_NOK);    }    pSeqParamSet->picOrderCntType = value;     if (pSeqParamSet->picOrderCntType == 0)    {        /* log2_max_pic_order_cnt_lsb_minus4 */        tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value);        if (tmp != HANTRO_OK)            return(tmp);        if (value > 12)        {            werrstr("log2_max_pic_order_cnt_lsb_minus4 out of range: %d", value);            return(HANTRO_NOK);        }        /* maxPicOrderCntLsb = 2^(log2_max_pic_order_cnt_lsb_minus4 + 4) */        pSeqParamSet->maxPicOrderCntLsb = 1 << (value+4);    }    else if (pSeqParamSet->picOrderCntType == 1)    {        tmp = h264bsdGetBits(pStrmData, 1);        if (tmp == END_OF_STREAM)            return(HANTRO_NOK);        pSeqParamSet->deltaPicOrderAlwaysZeroFlag = (tmp == 1) ?                                        HANTRO_TRUE : HANTRO_FALSE;         tmp = h264bsdDecodeExpGolombSigned(pStrmData,            &pSeqParamSet->offsetForNonRefPic);        if (tmp != HANTRO_OK)            return(tmp);         tmp = h264bsdDecodeExpGolombSigned(pStrmData,            &pSeqParamSet->offsetForTopToBottomField);        if (tmp != HANTRO_OK)            return(tmp);         tmp = h264bsdDecodeExpGolombUnsigned(pStrmData,            &pSeqParamSet->numRefFramesInPicOrderCntCycle);        if (tmp != HANTRO_OK)            return(tmp);        if (pSeqParamSet->numRefFramesInPicOrderCntCycle > 255)        {            werrstr("num_ref_frames_in_pic_order_cnt_cycle out of range: %d", pSeqParamSet->numRefFramesInPicOrderCntCycle);            return(HANTRO_NOK);        }         if (pSeqParamSet->numRefFramesInPicOrderCntCycle)        {            /* NOTE: This has to be freed somewhere! */            ALLOCATE(pSeqParamSet->offsetForRefFrame,                     pSeqParamSet->numRefFramesInPicOrderCntCycle, i32);            if (pSeqParamSet->offsetForRefFrame == NULL)                return(MEMORY_ALLOCATION_ERROR);             for (i = 0; i < pSeqParamSet->numRefFramesInPicOrderCntCycle; i++)            {                tmp =  h264bsdDecodeExpGolombSigned(pStrmData,                    pSeqParamSet->offsetForRefFrame + i);                if (tmp != HANTRO_OK)                    return(tmp);            }        }        else        {            pSeqParamSet->offsetForRefFrame = NULL;        }    }     tmp = h264bsdDecodeExpGolombUnsigned(pStrmData,        &pSeqParamSet->numRefFrames);    if (tmp != HANTRO_OK)        return(tmp);    if (pSeqParamSet->numRefFrames > MAX_NUM_REF_PICS)    {        werrstr("num_ref_frames out of range: %d", pSeqParamSet->numRefFrames);        return(HANTRO_NOK);    }     tmp = h264bsdGetBits(pStrmData, 1);    if (tmp == END_OF_STREAM)        return(HANTRO_NOK);    pSeqParamSet->gapsInFrameNumValueAllowedFlag = (tmp == 1) ?                                        HANTRO_TRUE : HANTRO_FALSE;     tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value);    if (tmp != HANTRO_OK)        return(tmp);    pSeqParamSet->picWidthInMbs = value + 1;     tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value);    if (tmp != HANTRO_OK)        return(tmp);    pSeqParamSet->picHeightInMbs = value + 1;     /* frame_mbs_only_flag, shall be 1 for baseline profile */    tmp = h264bsdGetBits(pStrmData, 1);    if (tmp == END_OF_STREAM)        return(HANTRO_NOK);    if (!tmp)    {        werrstr("frame_mbs_only_flag not set");        return(HANTRO_NOK);    }     /* direct_8x8_inference_flag */    tmp = h264bsdGetBits(pStrmData, 1);    if (tmp == END_OF_STREAM)        return(HANTRO_NOK);     tmp = h264bsdGetBits(pStrmData, 1);    if (tmp == END_OF_STREAM)        return(HANTRO_NOK);    pSeqParamSet->frameCroppingFlag = (tmp == 1) ? HANTRO_TRUE : HANTRO_FALSE;     if (pSeqParamSet->frameCroppingFlag)    {        tmp = h264bsdDecodeExpGolombUnsigned(pStrmData,            &pSeqParamSet->frameCropLeftOffset);        if (tmp != HANTRO_OK)            return(tmp);        tmp = h264bsdDecodeExpGolombUnsigned(pStrmData,            &pSeqParamSet->frameCropRightOffset);        if (tmp != HANTRO_OK)            return(tmp);        tmp = h264bsdDecodeExpGolombUnsigned(pStrmData,            &pSeqParamSet->frameCropTopOffset);        if (tmp != HANTRO_OK)            return(tmp);        tmp = h264bsdDecodeExpGolombUnsigned(pStrmData,            &pSeqParamSet->frameCropBottomOffset);        if (tmp != HANTRO_OK)            return(tmp);         /* check that frame cropping params are valid, parameters shall         * specify non-negative area within the original picture */        if ( ( (i32)pSeqParamSet->frameCropLeftOffset >               ( 8 * (i32)pSeqParamSet->picWidthInMbs -                 ((i32)pSeqParamSet->frameCropRightOffset + 1) ) ) ||             ( (i32)pSeqParamSet->frameCropTopOffset >               ( 8 * (i32)pSeqParamSet->picHeightInMbs -                 ((i32)pSeqParamSet->frameCropBottomOffset + 1) ) ) )        {            werrstr("frame_cropping invalid");            return(HANTRO_NOK);        }    }     /* check that image dimensions and levelIdc match */    tmp = pSeqParamSet->picWidthInMbs * pSeqParamSet->picHeightInMbs;    value = GetDpbSize(tmp, pSeqParamSet->levelIdc);    if (value == INVALID_DPB_SIZE || pSeqParamSet->numRefFrames > value)    {        DEBUG(("WARNING! Invalid DPB size based on SPS Level!\n"));        DEBUG(("WARNING! Using num_ref_frames =%d for DPB size!\n",                        pSeqParamSet->numRefFrames));        value = pSeqParamSet->numRefFrames;    }    pSeqParamSet->maxDpbSize = value;     tmp = h264bsdGetBits(pStrmData, 1);    if (tmp == END_OF_STREAM)        return(HANTRO_NOK);    pSeqParamSet->vuiParametersPresentFlag = (tmp == 1) ?                                HANTRO_TRUE : HANTRO_FALSE;     /* VUI */    if (pSeqParamSet->vuiParametersPresentFlag)    {        ALLOCATE(pSeqParamSet->vuiParameters, 1, vuiParameters_t);        if (pSeqParamSet->vuiParameters == NULL) {            werrstr("no memory");            return(MEMORY_ALLOCATION_ERROR);        }        tmp = h264bsdDecodeVuiParameters(pStrmData,            pSeqParamSet->vuiParameters);        if (tmp != HANTRO_OK)            return(tmp);        /* check numReorderFrames and maxDecFrameBuffering */        if (pSeqParamSet->vuiParameters->bitstreamRestrictionFlag)        {            if (pSeqParamSet->vuiParameters->numReorderFrames >                    pSeqParamSet->vuiParameters->maxDecFrameBuffering ||                pSeqParamSet->vuiParameters->maxDecFrameBuffering <                    pSeqParamSet->numRefFrames ||                pSeqParamSet->vuiParameters->maxDecFrameBuffering >                    pSeqParamSet->maxDpbSize)            {                return(HANTRO_NOK);            }             /* standard says that "the sequence shall not require a DPB with             * size of more than max(1, maxDecFrameBuffering) */            pSeqParamSet->maxDpbSize =                MAX(1, pSeqParamSet->vuiParameters->maxDecFrameBuffering);        }    }     h264bsdRbspTrailingBits(pStrmData);     /* ignore possible errors in trailing bits of parameters sets */    return(HANTRO_OK); } /*------------------------------------------------------------------------------     Function: GetDpbSize         Functional description:            Get size of the DPB in frames. Size is determined based on the            picture size and MaxDPB for the specified level. These determine            how many pictures may fit into to the buffer. However, the size            is also limited to a maximum of 16 frames and therefore function            returns the minimum of the determined size and 16.         Inputs:            picSizeInMbs    number of macroblocks in the picture            levelIdc        indicates the level         Outputs:            none         Returns:            size of the DPB in frames            INVALID_DPB_SIZE when invalid levelIdc specified or picSizeInMbs            is higher than supported by the level in question ------------------------------------------------------------------------------*/ static u32 GetDpbSize(u32 picSizeInMbs, u32 levelIdc){ /* Variables */     u32 tmp;    u32 maxPicSizeInMbs; /* Code */     ASSERT(picSizeInMbs);     /* use tmp as the size of the DPB in bytes, computes as 1024 * MaxDPB     * (from table A-1 in Annex A) */    switch (levelIdc)    {        case 10:            tmp = 152064;            maxPicSizeInMbs = 99;            break;         case 11:            tmp = 345600;            maxPicSizeInMbs = 396;            break;         case 12:            tmp = 912384;            maxPicSizeInMbs = 396;            break;         case 13:            tmp = 912384;            maxPicSizeInMbs = 396;            break;         case 20:            tmp = 912384;            maxPicSizeInMbs = 396;            break;         case 21:            tmp = 1824768;            maxPicSizeInMbs = 792;            break;         case 22:            tmp = 3110400;            maxPicSizeInMbs = 1620;            break;         case 30:            tmp = 3110400;            maxPicSizeInMbs = 1620;            break;         case 31:            tmp = 6912000;            maxPicSizeInMbs = 3600;            break;         case 32:            tmp = 7864320;            maxPicSizeInMbs = 5120;            break;         case 40:            tmp = 12582912;            maxPicSizeInMbs = 8192;            break;         case 41:            tmp = 12582912;            maxPicSizeInMbs = 8192;            break;         case 42:            tmp = 34816*384;            maxPicSizeInMbs = 8704;            break;         case 50:            /* standard says 42301440 here, but corrigendum "corrects" this to             * 42393600 */            tmp = 42393600;            maxPicSizeInMbs = 22080;            break;         case 51:            tmp = 70778880;            maxPicSizeInMbs = 36864;            break;         default:            return(INVALID_DPB_SIZE);    }     /* this is not "correct" return value! However, it results in error in     * decoding and this was easiest place to check picture size */    if (picSizeInMbs > maxPicSizeInMbs)        return(INVALID_DPB_SIZE);     tmp /= (picSizeInMbs*384);     return(MIN(tmp, 16)); } /*------------------------------------------------------------------------------     Function name: h264bsdCompareSeqParamSets         Functional description:            Compare two sequence parameter sets. 

Lines 1–500 of 580. Load more