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