u32 h264bsdDecodeSeiMessage( strmData_t *pStrmData, seqParamSet_t *pSeqParamSet, seiMessage_t *pSeiMessage, u32 numSliceGroups) { /* Variables */ u32 tmp, payloadType, payloadSize, status; /* Code */ ASSERT(pStrmData); ASSERT(pSeiMessage); H264SwDecMemset(pSeiMessage, 0, sizeof(seiMessage_t)); do { payloadType = 0; while((tmp = h264bsdGetBits(pStrmData, 8)) == 0xFF) { payloadType += 255; } if (tmp == END_OF_STREAM) return(HANTRO_NOK); payloadType += tmp; payloadSize = 0; while((tmp = h264bsdGetBits(pStrmData, 8)) == 0xFF) { payloadSize += 255; } if (tmp == END_OF_STREAM) return(HANTRO_NOK); payloadSize += tmp; pSeiMessage->payloadType = payloadType; switch (payloadType) { case 0: ASSERT(pSeqParamSet); status = DecodeBufferingPeriod( pStrmData, &pSeiMessage->bufferingPeriod, pSeqParamSet->vuiParameters->vclHrdParameters.cpbCnt, pSeqParamSet->vuiParameters->vclHrdParameters. initialCpbRemovalDelayLength, pSeqParamSet->vuiParameters->nalHrdParametersPresentFlag, pSeqParamSet->vuiParameters->vclHrdParametersPresentFlag); break; case 1: ASSERT(pSeqParamSet->vuiParametersPresentFlag); status = DecodePictureTiming( pStrmData, &pSeiMessage->picTiming, pSeqParamSet->vuiParameters->vclHrdParameters. cpbRemovalDelayLength, pSeqParamSet->vuiParameters->vclHrdParameters. dpbOutputDelayLength, pSeqParamSet->vuiParameters->vclHrdParameters. timeOffsetLength, pSeqParamSet->vuiParameters->nalHrdParametersPresentFlag || pSeqParamSet->vuiParameters->vclHrdParametersPresentFlag ? HANTRO_TRUE : HANTRO_FALSE, pSeqParamSet->vuiParameters->picStructPresentFlag); break; case 2: status = DecodePanScanRectangle( pStrmData, &pSeiMessage->panScanRect); break; case 3: status = DecodeFillerPayload(pStrmData, payloadSize); break; case 4: status = DecodeUserDataRegisteredITuTT35( pStrmData, &pSeiMessage->userDataRegisteredItuTT35, payloadSize); break; case 5: status = DecodeUserDataUnregistered( pStrmData, &pSeiMessage->userDataUnregistered, payloadSize); break; case 6: status = DecodeRecoveryPoint( pStrmData, &pSeiMessage->recoveryPoint); break; case 7: status = DecodeDecRefPicMarkingRepetition( pStrmData, &pSeiMessage->decRefPicMarkingRepetition, pSeqParamSet->numRefFrames); break; case 8: ASSERT(pSeqParamSet); status = DecodeSparePic( pStrmData, &pSeiMessage->sparePic, pSeqParamSet->picWidthInMbs * pSeqParamSet->picHeightInMbs); break; case 9: status = DecodeSceneInfo( pStrmData, &pSeiMessage->sceneInfo); break; case 10: status = DecodeSubSeqInfo( pStrmData, &pSeiMessage->subSeqInfo); break; case 11: status = DecodeSubSeqLayerCharacteristics( pStrmData, &pSeiMessage->subSeqLayerCharacteristics); break; case 12: status = DecodeSubSeqCharacteristics( pStrmData, &pSeiMessage->subSeqCharacteristics); break; case 13: status = DecodeFullFrameFreeze( pStrmData, &pSeiMessage->fullFrameFreeze); break; case 14: /* This SEI does not contain data, what to do ??? */ status = HANTRO_OK; break; case 15: status = DecodeFullFrameSnapshot( pStrmData, &pSeiMessage->fullFrameSnapshot); break; case 16: status = DecodeProgressiveRefinementSegmentStart( pStrmData, &pSeiMessage->progressiveRefinementSegmentStart); break; case 17: status = DecodeProgressiveRefinementSegmentEnd( pStrmData, &pSeiMessage->progressiveRefinementSegmentEnd); break; case 18: ASSERT(numSliceGroups); status = DecodeMotionConstrainedSliceGroupSet( pStrmData, &pSeiMessage->motionConstrainedSliceGroupSet, numSliceGroups); break; default: status = DecodeReservedSeiMessage( pStrmData, &pSeiMessage->reservedSeiMessage, payloadSize); break; } if (status != HANTRO_OK) return(status); while (!h264bsdIsByteAligned(pStrmData)) { if (h264bsdGetBits(pStrmData, 1) != 1) return(HANTRO_NOK); while (!h264bsdIsByteAligned(pStrmData)) { if (h264bsdGetBits(pStrmData, 1) != 0) return(HANTRO_NOK); } } } while (h264bsdMoreRbspData(pStrmData)); return(h264bsdRbspTrailingBits(pStrmData)); }
u32 h264bsdDecodePicParamSet(strmData_t *pStrmData, picParamSet_t *pPicParamSet) { /* Variables */ u32 tmp, i, value; i32 itmp; /* Code */ ASSERT(pStrmData); ASSERT(pPicParamSet); H264SwDecMemset(pPicParamSet, 0, sizeof(picParamSet_t)); tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &pPicParamSet->picParameterSetId); if (tmp != HANTRO_OK) return(tmp); if (pPicParamSet->picParameterSetId >= MAX_NUM_PIC_PARAM_SETS) { EPRINT("pic_parameter_set_id"); return(HANTRO_NOK); } tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &pPicParamSet->seqParameterSetId); if (tmp != HANTRO_OK) return(tmp); if (pPicParamSet->seqParameterSetId >= MAX_NUM_SEQ_PARAM_SETS) { EPRINT("seq_param_set_id"); return(HANTRO_NOK); } /* entropy_coding_mode_flag, shall be 0 for baseline profile */ tmp = h264bsdGetBits(pStrmData, 1); if (tmp) { EPRINT("entropy_coding_mode_flag"); return(HANTRO_NOK); } tmp = h264bsdGetBits(pStrmData, 1); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pPicParamSet->picOrderPresentFlag = (tmp == 1) ? HANTRO_TRUE : HANTRO_FALSE; /* num_slice_groups_minus1 */ tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); pPicParamSet->numSliceGroups = value + 1; if (pPicParamSet->numSliceGroups > MAX_NUM_SLICE_GROUPS) { EPRINT("num_slice_groups_minus1"); return(HANTRO_NOK); } /* decode slice group mapping information if more than one slice groups */ if (pPicParamSet->numSliceGroups > 1) { tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &pPicParamSet->sliceGroupMapType); if (tmp != HANTRO_OK) return(tmp); if (pPicParamSet->sliceGroupMapType > 6) { EPRINT("slice_group_map_type"); return(HANTRO_NOK); } if (pPicParamSet->sliceGroupMapType == 0) { ALLOCATE(pPicParamSet->runLength, pPicParamSet->numSliceGroups, u32); if (pPicParamSet->runLength == NULL) return(MEMORY_ALLOCATION_ERROR); for (i = 0; i < pPicParamSet->numSliceGroups; i++) { tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); pPicParamSet->runLength[i] = value+1; /* param values checked in CheckPps() */ } } else if (pPicParamSet->sliceGroupMapType == 2) { ALLOCATE(pPicParamSet->topLeft, pPicParamSet->numSliceGroups - 1, u32); ALLOCATE(pPicParamSet->bottomRight, pPicParamSet->numSliceGroups - 1, u32); if (pPicParamSet->topLeft == NULL || pPicParamSet->bottomRight == NULL) return(MEMORY_ALLOCATION_ERROR); for (i = 0; i < pPicParamSet->numSliceGroups - 1; i++) { tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); pPicParamSet->topLeft[i] = value; tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); pPicParamSet->bottomRight[i] = value; /* param values checked in CheckPps() */ } } else if ( (pPicParamSet->sliceGroupMapType == 3) || (pPicParamSet->sliceGroupMapType == 4) || (pPicParamSet->sliceGroupMapType == 5) ) { tmp = h264bsdGetBits(pStrmData, 1); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pPicParamSet->sliceGroupChangeDirectionFlag = (tmp == 1) ? HANTRO_TRUE : HANTRO_FALSE; tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); pPicParamSet->sliceGroupChangeRate = value + 1; /* param value checked in CheckPps() */ } else if (pPicParamSet->sliceGroupMapType == 6) { tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); pPicParamSet->picSizeInMapUnits = value + 1; ALLOCATE(pPicParamSet->sliceGroupId, pPicParamSet->picSizeInMapUnits, u32); if (pPicParamSet->sliceGroupId == NULL) return(MEMORY_ALLOCATION_ERROR); /* determine number of bits needed to represent range * [0, numSliceGroups) */ tmp = CeilLog2NumSliceGroups[pPicParamSet->numSliceGroups-1]; for (i = 0; i < pPicParamSet->picSizeInMapUnits; i++) { pPicParamSet->sliceGroupId[i] = h264bsdGetBits(pStrmData, tmp); if ( pPicParamSet->sliceGroupId[i] >= pPicParamSet->numSliceGroups ) { EPRINT("slice_group_id"); return(HANTRO_NOK); } } } } /* num_ref_idx_l0_active_minus1 */ tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); if (value > 31) { EPRINT("num_ref_idx_l0_active_minus1"); return(HANTRO_NOK); } pPicParamSet->numRefIdxL0Active = value + 1; /* num_ref_idx_l1_active_minus1 */ tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); if (value > 31) { EPRINT("num_ref_idx_l1_active_minus1"); return(HANTRO_NOK); } /* weighted_pred_flag, this shall be 0 for baseline profile */ tmp = h264bsdGetBits(pStrmData, 1); if (tmp) { EPRINT("weighted_pred_flag"); return(HANTRO_NOK); } /* weighted_bipred_idc */ tmp = h264bsdGetBits(pStrmData, 2); if (tmp > 2) { EPRINT("weighted_bipred_idc"); return(HANTRO_NOK); } /* pic_init_qp_minus26 */ tmp = h264bsdDecodeExpGolombSigned(pStrmData, &itmp); if (tmp != HANTRO_OK) return(tmp); if ((itmp < -26) || (itmp > 25)) { EPRINT("pic_init_qp_minus26"); return(HANTRO_NOK); } pPicParamSet->picInitQp = (u32)(itmp + 26); /* pic_init_qs_minus26 */ tmp = h264bsdDecodeExpGolombSigned(pStrmData, &itmp); if (tmp != HANTRO_OK) return(tmp); if ((itmp < -26) || (itmp > 25)) { EPRINT("pic_init_qs_minus26"); return(HANTRO_NOK); } tmp = h264bsdDecodeExpGolombSigned(pStrmData, &itmp); if (tmp != HANTRO_OK) return(tmp); if ((itmp < -12) || (itmp > 12)) { EPRINT("chroma_qp_index_offset"); return(HANTRO_NOK); } pPicParamSet->chromaQpIndexOffset = itmp; tmp = h264bsdGetBits(pStrmData, 1); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pPicParamSet->deblockingFilterControlPresentFlag = (tmp == 1) ? HANTRO_TRUE : HANTRO_FALSE; tmp = h264bsdGetBits(pStrmData, 1); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pPicParamSet->constrainedIntraPredFlag = (tmp == 1) ? HANTRO_TRUE : HANTRO_FALSE; tmp = h264bsdGetBits(pStrmData, 1); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pPicParamSet->redundantPicCntPresentFlag = (tmp == 1) ? HANTRO_TRUE : HANTRO_FALSE; tmp = h264bsdRbspTrailingBits(pStrmData); /* ignore possible errors in trailing bits of parameters sets */ return(HANTRO_OK); }
u32 h264bsdDecodeSeqParamSet(strmData_t *pStrmData, seqParamSet_t *pSeqParamSet, u32 mvcFlag) { /* Variables */ u32 tmp, i, value; /* Code */ ASSERT(pStrmData); ASSERT(pSeqParamSet); (void)DWLmemset(pSeqParamSet, 0, sizeof(seqParamSet_t)); /* profile_idc */ tmp = h264bsdGetBits(pStrmData, 8); if (tmp == END_OF_STREAM) return(HANTRO_NOK); if (tmp != 66) { DEBUG_PRINT(("NOT BASELINE PROFILE %d\n", tmp)); } #ifdef ASIC_TRACE_SUPPORT if (tmp==66) trace_h264DecodingTools.profileType.baseline = 1; if (tmp==77) trace_h264DecodingTools.profileType.main = 1; if (tmp == 100) trace_h264DecodingTools.profileType.high = 1; #endif pSeqParamSet->profileIdc = tmp; /* constrained_set0_flag */ tmp = h264bsdGetBits(pStrmData, 1); pSeqParamSet->constrained_set0_flag = tmp; /* constrained_set1_flag */ tmp = h264bsdGetBits(pStrmData, 1); pSeqParamSet->constrained_set1_flag = tmp; /* constrained_set2_flag */ tmp = h264bsdGetBits(pStrmData, 1); pSeqParamSet->constrained_set2_flag = tmp; /* constrained_set3_flag */ tmp = h264bsdGetBits(pStrmData, 1); pSeqParamSet->constrained_set3_flag = tmp; if (tmp == END_OF_STREAM) return(HANTRO_NOK); /* reserved_zero_4bits, values of these bits shall be ignored */ tmp = h264bsdGetBits(pStrmData, 4); if (tmp == END_OF_STREAM) { ERROR_PRINT("reserved_zero_4bits"); return(HANTRO_NOK); } tmp = h264bsdGetBits(pStrmData, 8); if (tmp == END_OF_STREAM) { ERROR_PRINT("level_idc"); 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) { ERROR_PRINT("seq_param_set_id"); return(HANTRO_NOK); } if( pSeqParamSet->profileIdc >= 100 ) { /* chroma_format_idc */ tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); if( tmp > 1 ) return(HANTRO_NOK); pSeqParamSet->chromaFormatIdc = value; if (pSeqParamSet->chromaFormatIdc == 0) pSeqParamSet->monoChrome = 1; /* residual_colour_transform_flag (skipped) */ /* bit_depth_luma_minus8 */ tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); /* bit_depth_chroma_minus8 */ tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); /* qpprime_y_zero_transform_bypass_flag */ tmp = h264bsdGetBits(pStrmData, 1); if (tmp == END_OF_STREAM) return(HANTRO_NOK); /* seq_scaling_matrix_present_flag */ tmp = h264bsdGetBits(pStrmData, 1); #ifdef ASIC_TRACE_SUPPORT if (tmp) trace_h264DecodingTools.scalingMatrixPresentType.seq = 1; #endif if (tmp == END_OF_STREAM) return(HANTRO_NOK); pSeqParamSet->scalingMatrixPresentFlag = tmp; if (tmp) { for (i = 0; i < 8; i++) { tmp = h264bsdGetBits(pStrmData, 1); pSeqParamSet->scalingListPresent[i] = tmp; if (tmp) { ScalingList(pSeqParamSet->scalingList, pStrmData, i); } else FallbackScaling(pSeqParamSet->scalingList, i); } } } else { pSeqParamSet->chromaFormatIdc = 1; /* 4:2:0 */ pSeqParamSet->scalingMatrixPresentFlag = 0; } /* log2_max_frame_num_minus4 */ tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) { ERROR_PRINT("log2_max_frame_num_minus4"); return(tmp); } if (value > 12) { ERROR_PRINT("log2_max_frame_num_minus4"); 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) { ERROR_PRINT("pic_order_cnt_type"); return(tmp); } if (value > 2) { ERROR_PRINT("pic_order_cnt_type"); 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) { ERROR_PRINT("log2_max_pic_order_cnt_lsb_minus4"); 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) { ERROR_PRINT("num_ref_frames_in_pic_order_cnt_cycle"); 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 || /* max num ref frames in mvc stereo high profile is actually 8, but * here we just check that it is less than 15 (base 15 used for * inter view reference picture) */ (mvcFlag && (pSeqParamSet->numRefFrames > 15))) { ERROR_PRINT("num_ref_frames"); 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); pSeqParamSet->frameMbsOnlyFlag = tmp; if (!pSeqParamSet->frameMbsOnlyFlag) { pSeqParamSet->mbAdaptiveFrameFieldFlag = h264bsdGetBits(pStrmData, 1); pSeqParamSet->picHeightInMbs *= 2; } /* direct_8x8_inference_flag */ tmp = h264bsdGetBits(pStrmData, 1); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pSeqParamSet->direct8x8InferenceFlag = tmp; tmp = h264bsdGetBits(pStrmData, 1); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pSeqParamSet->frameCroppingFlag = (tmp == 1) ? HANTRO_TRUE : HANTRO_FALSE; #ifdef ASIC_TRACE_SUPPORT if (tmp) trace_h264DecodingTools.imageCropping = 1; #endif 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) ) ) ) { ERROR_PRINT("frame_cropping"); 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_PRINT(("WARNING! Invalid DPB size based on SPS Level!\n")); DEBUG_PRINT(("WARNING! Using num_ref_frames =%d for DPB size!\n", pSeqParamSet->numRefFrames)); /* set maxDpbSize to 1 if numRefFrames is zero */ value = pSeqParamSet->numRefFrames ? pSeqParamSet->numRefFrames : 1; } 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) return(MEMORY_ALLOCATION_ERROR); tmp = h264bsdDecodeVuiParameters(pStrmData, pSeqParamSet->vuiParameters); if (tmp == END_OF_STREAM) { pSeqParamSet->vuiParameters->bitstreamRestrictionFlag |= 1; pSeqParamSet->vuiParameters->maxDecFrameBuffering = pSeqParamSet->maxDpbSize; } else 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) { ERROR_PRINT("Not valid vuiParameters->bitstreamRestriction"); 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); } } if (mvcFlag) { if (pSeqParamSet->profileIdc == 118 || pSeqParamSet->profileIdc == 128) { /* bit_equal_to_one */ tmp = h264bsdGetBits(pStrmData, 1); tmp = DecodeMvcExtension(pStrmData, pSeqParamSet); if (tmp != HANTRO_OK) return tmp; } /* additional_extension2_flag, shall be zero */ tmp = h264bsdGetBits(pStrmData, 1); /* TODO: skip rest of the stuff if equal to 1 */ } tmp = h264bsdRbspTrailingBits(pStrmData); /* ignore possible errors in trailing bits of parameters sets */ return(HANTRO_OK); }
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) { DEBUG(("NOT BASELINE PROFILE %d\n", tmp)); } pSeqParamSet->profileIdc = tmp; /* constrained_set0_flag */ tmp = h264bsdGetBits(pStrmData, 1); /* constrained_set1_flag */ tmp = 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) { EPRINT("seq_param_set_id"); return(HANTRO_NOK); } /* log2_max_frame_num_minus4 */ tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); if (value > 12) { EPRINT("log2_max_frame_num_minus4"); 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) { EPRINT("pic_order_cnt_type"); 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) { EPRINT("log2_max_pic_order_cnt_lsb_minus4"); 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) { EPRINT("num_ref_frames_in_pic_order_cnt_cycle"); 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) { EPRINT("num_ref_frames"); 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) { EPRINT("frame_mbs_only_flag"); 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) ) ) ) { EPRINT("frame_cropping"); 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) 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); } } tmp = h264bsdRbspTrailingBits(pStrmData); /* ignore possible errors in trailing bits of parameters sets */ return(HANTRO_OK); }