static u32 DecodeBufferingPeriod( strmData_t *pStrmData, seiBufferingPeriod_t *pBufferingPeriod, u32 cpbCnt, u32 initialCpbRemovalDelayLength, u32 nalHrdBpPresentFlag, u32 vclHrdBpPresentFlag) { /* Variables */ u32 tmp, i; /* Code */ ASSERT(pStrmData); ASSERT(pBufferingPeriod); ASSERT(cpbCnt); ASSERT(initialCpbRemovalDelayLength); tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &pBufferingPeriod->seqParameterSetId); if (tmp != HANTRO_OK) return(tmp); if (pBufferingPeriod->seqParameterSetId > 31) return(HANTRO_NOK); if (nalHrdBpPresentFlag) { for (i = 0; i < cpbCnt; i++) { tmp = h264bsdGetBits(pStrmData, initialCpbRemovalDelayLength); if (tmp == END_OF_STREAM) return(HANTRO_NOK); if (tmp == 0) return(HANTRO_NOK); pBufferingPeriod->initialCpbRemovalDelay[i] = tmp; tmp = h264bsdGetBits(pStrmData, initialCpbRemovalDelayLength); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pBufferingPeriod->initialCpbRemovalDelayOffset[i] = tmp; } } if (vclHrdBpPresentFlag) { for (i = 0; i < cpbCnt; i++) { tmp = h264bsdGetBits(pStrmData, initialCpbRemovalDelayLength); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pBufferingPeriod->initialCpbRemovalDelay[i] = tmp; tmp = h264bsdGetBits(pStrmData, initialCpbRemovalDelayLength); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pBufferingPeriod->initialCpbRemovalDelayOffset[i] = tmp; } } return(HANTRO_OK); }
static u32 DecodePictureTiming( strmData_t *pStrmData, seiPicTiming_t *pPicTiming, u32 cpbRemovalDelayLength, u32 dpbOutputDelayLength, u32 timeOffsetLength, u32 cpbDpbDelaysPresentFlag, u32 picStructPresentFlag) { /* Variables */ u32 tmp, i; i32 itmp; /* Code */ ASSERT(pStrmData); ASSERT(pPicTiming); if (cpbDpbDelaysPresentFlag) { tmp = h264bsdGetBits(pStrmData, cpbRemovalDelayLength); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pPicTiming->cpbRemovalDelay = tmp; tmp = h264bsdGetBits(pStrmData, dpbOutputDelayLength); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pPicTiming->dpbOutputDelay = tmp; } if (picStructPresentFlag) { tmp = h264bsdGetBits(pStrmData, 4); if (tmp == END_OF_STREAM) return(HANTRO_NOK); if (tmp > 8) return(HANTRO_NOK); pPicTiming->picStruct = tmp; for (i = 0; i < numClockTS[pPicTiming->picStruct]; i++) { tmp = h264bsdGetBits(pStrmData, 1); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pPicTiming->clockTimeStampFlag[i] = tmp == 1 ? HANTRO_TRUE : HANTRO_FALSE; if (pPicTiming->clockTimeStampFlag[i]) { tmp = h264bsdGetBits(pStrmData, 2); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pPicTiming->ctType[i] = tmp; tmp = h264bsdGetBits(pStrmData, 1); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pPicTiming->nuitFieldBasedFlag[i] = tmp == 1 ? HANTRO_TRUE : HANTRO_FALSE; tmp = h264bsdGetBits(pStrmData, 5); if (tmp == END_OF_STREAM) return(HANTRO_NOK); if (tmp > 6) return(HANTRO_NOK); pPicTiming->countingType[i] = tmp; tmp = h264bsdGetBits(pStrmData, 1); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pPicTiming->fullTimeStampFlag[i] = tmp == 1 ? HANTRO_TRUE : HANTRO_FALSE; tmp = h264bsdGetBits(pStrmData, 1); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pPicTiming->discontinuityFlag[i] = tmp == 1 ? HANTRO_TRUE : HANTRO_FALSE; tmp = h264bsdGetBits(pStrmData, 1); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pPicTiming->cntDroppedFlag[i] = tmp == 1 ? HANTRO_TRUE : HANTRO_FALSE; tmp = h264bsdGetBits(pStrmData, 8); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pPicTiming->nFrames[i] = tmp; if (pPicTiming->fullTimeStampFlag[i]) { tmp = h264bsdGetBits(pStrmData, 6); if (tmp == END_OF_STREAM) return(HANTRO_NOK); if (tmp > 59) return(HANTRO_NOK); pPicTiming->secondsValue[i] = tmp; tmp = h264bsdGetBits(pStrmData, 6); if (tmp == END_OF_STREAM) return(HANTRO_NOK); if (tmp > 59) return(HANTRO_NOK); pPicTiming->minutesValue[i] = tmp; tmp = h264bsdGetBits(pStrmData, 5); if (tmp == END_OF_STREAM) return(HANTRO_NOK); if (tmp > 23) return(HANTRO_NOK); pPicTiming->hoursValue[i] = tmp; } else { tmp = h264bsdGetBits(pStrmData, 1); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pPicTiming->secondsFlag[i] = tmp == 1 ? HANTRO_TRUE : HANTRO_FALSE; if (pPicTiming->secondsFlag[i]) { tmp = h264bsdGetBits(pStrmData, 6); if (tmp == END_OF_STREAM) return(HANTRO_NOK); if (tmp > 59) return(HANTRO_NOK); pPicTiming->secondsValue[i] = tmp; tmp = h264bsdGetBits(pStrmData, 1); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pPicTiming->minutesFlag[i] = tmp == 1 ? HANTRO_TRUE : HANTRO_FALSE; if (pPicTiming->minutesFlag[i]) { tmp = h264bsdGetBits(pStrmData, 6); if (tmp == END_OF_STREAM) return(HANTRO_NOK); if (tmp > 59) return(HANTRO_NOK); pPicTiming->minutesValue[i] = tmp; tmp = h264bsdGetBits(pStrmData, 1); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pPicTiming->hoursFlag[i] = tmp == 1 ? HANTRO_TRUE : HANTRO_FALSE; if (pPicTiming->hoursFlag[i]) { tmp = h264bsdGetBits(pStrmData, 5); if (tmp == END_OF_STREAM) return(HANTRO_NOK); if (tmp > 23) return(HANTRO_NOK); pPicTiming->hoursValue[i] = tmp; } } } } if (timeOffsetLength) { tmp = h264bsdGetBits(pStrmData, timeOffsetLength); if (tmp == END_OF_STREAM) return(HANTRO_NOK); itmp = (i32)tmp; /* following "converts" timeOffsetLength-bit signed * integer into i32 */ /*lint -save -e701 -e702 */ itmp <<= (32 - timeOffsetLength); itmp >>= (32 - timeOffsetLength); /*lint -restore */ pPicTiming->timeOffset[i] = itmp; } else pPicTiming->timeOffset[i] = 0; } }
u32 DecodeMvcExtension(strmData_t *pStrmData, seqParamSet_t *pSeqParamSet) { /* Variables */ u32 tmp, i, j, k; u32 value, tmpCount, tmpCount1, tmpCount2; hrdParameters_t hrdParams; /* Code */ ASSERT(pStrmData); ASSERT(pSeqParamSet); tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); pSeqParamSet->mvc.numViews = value + 1; if (pSeqParamSet->mvc.numViews > MAX_NUM_VIEWS) return(HANTRO_NOK); /* view_id */ for (i = 0; i < pSeqParamSet->mvc.numViews; i++) { tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); if (i < MAX_NUM_VIEWS) pSeqParamSet->mvc.viewId[i] = value; } for (i = 1; i < pSeqParamSet->mvc.numViews; i++) { tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); /*pSeqParamSet->mvc.numAnchorRefsL0[i] = value;*/ tmpCount = value; for (j = 0; j < tmpCount; j++) { tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); /*pSeqParamSet->mvc.anchorRefL0[i][j] = value;*/ } tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); /*pSeqParamSet->mvc.numNonAnchorRefsL0[i] = value;*/ tmpCount = value; for (j = 0; j < tmpCount; j++) { tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); /*pSeqParamSet->mvc.nonAnchorRefL0[i][j] = value;*/ } } for (i = 1; i < pSeqParamSet->mvc.numViews; i++) { tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); /*pSeqParamSet->mvc.numAnchorRefsL1[i] = value;*/ tmpCount = value; for (j = 0; j < tmpCount; j++) { tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); /*pSeqParamSet->mvc.anchorRefL1[i][j] = value;*/ } tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); /*pSeqParamSet->mvc.numNonAnchorRefsL1[i] = value;*/ tmpCount = value; for (j = 0; j < tmpCount; j++) { tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); /*pSeqParamSet->mvc.nonAnchorRefL1[i][j] = value;*/ } } /* num_level_values_signalled_minus1 */ tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); tmpCount = value + 1; for (i = 0; i < tmpCount; i++) { /* level_idc */ tmp = h264bsdGetBits(pStrmData, 8); /* num_applicable_ops_minus1 */ tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); tmpCount1 = value + 1; for (j = 0; j < tmpCount1; j++) { /* applicable_op_temporal_id */ tmp = h264bsdGetBits(pStrmData, 3); /* applicable_op_num_target_views_minus1 */ tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); tmpCount2 = value + 1; for (k = 0; k < tmpCount2; k++) { /* applicable_op_target_view_id */ tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); } /* applicable_op_num_views_minus1 */ tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); } } /* mvc_vui_parameters_present_flag */ tmp = h264bsdGetBits(pStrmData, 1); if (tmp == 1) { /* vui_mvc_num_ops_minus1 */ tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); tmpCount = value + 1; for (i = 0; i < tmpCount; i++) { /* vui_mvc_temporal_id */ tmp = h264bsdGetBits(pStrmData, 3); /* vui_mvc_num_target_output_views_minus1 */ tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); tmpCount1 = value + 1; for (k = 0; k < tmpCount1; k++) { /* vui_mvc_view_id */ tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); } /* vui_mvc_timing_info_present_flag */ tmp = h264bsdGetBits(pStrmData, 1); if (tmp == 1) { /* vui_mvc_num_units_in_tick */ tmp = h264bsdShowBits(pStrmData,32); if (h264bsdFlushBits(pStrmData, 32) == END_OF_STREAM) return(END_OF_STREAM); /* vui_mvc_time_scale */ tmp = h264bsdShowBits(pStrmData,32); if (h264bsdFlushBits(pStrmData, 32) == END_OF_STREAM) return(END_OF_STREAM); /* vui_mvc_fixed_frame_rate_flag */ tmp = h264bsdGetBits(pStrmData, 1); } j = 0; /* vui_mvc_nal_hrd_parameters_present_flag */ tmp = h264bsdGetBits(pStrmData, 1); if (tmp == 1) { j = 1; tmp = h264bsdDecodeHrdParameters(pStrmData, &hrdParams); } /* vui_mvc_vcl_hrd_parameters_present_flag */ tmp = h264bsdGetBits(pStrmData, 1); if (tmp == 1) { j = 1; tmp = h264bsdDecodeHrdParameters(pStrmData, &hrdParams); } if (j) { /* vui_mvc_low_delay_hrd_flag */ tmp = h264bsdGetBits(pStrmData, 1); } /* vui_mvc_pic_struct_present_flag */ tmp = h264bsdGetBits(pStrmData, 1); } } return(HANTRO_OK); }
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 h264bsdDecodeSliceHeader(strmData_t *pStrmData, sliceHeader_t *pSliceHeader, seqParamSet_t *pSeqParamSet, picParamSet_t *pPicParamSet, nalUnit_t *pNalUnit) { /* Variables */ u32 tmp, i, value; i32 itmp; u32 picSizeInMbs; /* Code */ ASSERT(pStrmData); ASSERT(pSliceHeader); ASSERT(pSeqParamSet); ASSERT(pPicParamSet); ASSERT( pNalUnit->nalUnitType == NAL_CODED_SLICE || pNalUnit->nalUnitType == NAL_CODED_SLICE_IDR ); H264SwDecMemset(pSliceHeader, 0, sizeof(sliceHeader_t)); picSizeInMbs = pSeqParamSet->picWidthInMbs * pSeqParamSet->picHeightInMbs; tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); pSliceHeader->firstMbInSlice = value; if (value >= picSizeInMbs) { EPRINT("first_mb_in_slice"); return(HANTRO_NOK); } tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); pSliceHeader->sliceType = value; /* slice type has to be either I or P slice. P slice is not allowed when * current NAL unit is an IDR NAL unit or num_ref_frames is 0 */ if ( !IS_I_SLICE(pSliceHeader->sliceType) && ( !IS_P_SLICE(pSliceHeader->sliceType) || IS_IDR_NAL_UNIT(pNalUnit) || !pSeqParamSet->numRefFrames ) ) { EPRINT("slice_type"); return(HANTRO_NOK); } tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); pSliceHeader->picParameterSetId = value; if (pSliceHeader->picParameterSetId != pPicParamSet->picParameterSetId) { EPRINT("pic_parameter_set_id"); return(HANTRO_NOK); } /* log2(maxFrameNum) -> num bits to represent frame_num */ i = 0; while (pSeqParamSet->maxFrameNum >> i) i++; i--; tmp = h264bsdGetBits(pStrmData, i); if (tmp == END_OF_STREAM) return(HANTRO_NOK); if (IS_IDR_NAL_UNIT(pNalUnit) && tmp != 0) { EPRINT("frame_num"); return(HANTRO_NOK); } pSliceHeader->frameNum = tmp; if (IS_IDR_NAL_UNIT(pNalUnit)) { tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (tmp != HANTRO_OK) return(tmp); pSliceHeader->idrPicId = value; if (value > 65535) { EPRINT("idr_pic_id"); return(HANTRO_NOK); } } if (pSeqParamSet->picOrderCntType == 0) { /* log2(maxPicOrderCntLsb) -> num bits to represent pic_order_cnt_lsb */ i = 0; while (pSeqParamSet->maxPicOrderCntLsb >> i) i++; i--; tmp = h264bsdGetBits(pStrmData, i); if (tmp == END_OF_STREAM) return(HANTRO_NOK); pSliceHeader->picOrderCntLsb = tmp; if (pPicParamSet->picOrderPresentFlag) { tmp = h264bsdDecodeExpGolombSigned(pStrmData, &itmp); if (tmp != HANTRO_OK) return(tmp); pSliceHeader->deltaPicOrderCntBottom = itmp; } /* check that picOrderCnt for IDR picture will be zero. See * DecodePicOrderCnt function to understand the logic here */ if ( IS_IDR_NAL_UNIT(pNalUnit) && ( (pSliceHeader->picOrderCntLsb > pSeqParamSet->maxPicOrderCntLsb/2) || MIN((i32)pSliceHeader->picOrderCntLsb, (i32)pSliceHeader->picOrderCntLsb + pSliceHeader->deltaPicOrderCntBottom) != 0 ) ) { return(HANTRO_NOK); } }
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 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 h264bsdDecodeMacroblockLayer(strmData_t *pStrmData, macroblockLayer_t *pMbLayer, mbStorage_t *pMb, u32 sliceType, u32 numRefIdxActive) { /* Variables */ u32 tmp, i, value; i32 itmp; mbPartPredMode_e partMode; /* Code */ ASSERT(pStrmData); ASSERT(pMbLayer); #ifdef H264DEC_NEON h264bsdClearMbLayer(pMbLayer, ((sizeof(macroblockLayer_t) + 63) & ~0x3F)); #else H264SwDecMemset(pMbLayer, 0, sizeof(macroblockLayer_t)); #endif tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if (IS_I_SLICE(sliceType)) { if ((value + 6) > 31 || tmp != HANTRO_OK) return(HANTRO_NOK); pMbLayer->mbType = (mbType_e)(value + 6); } else { if ((value + 1) > 31 || tmp != HANTRO_OK) return(HANTRO_NOK); pMbLayer->mbType = (mbType_e)(value + 1); } if (pMbLayer->mbType == I_PCM) { i32 *level; while( !h264bsdIsByteAligned(pStrmData) ) { /* pcm_alignment_zero_bit */ tmp = h264bsdGetBits(pStrmData, 1); if (tmp) return(HANTRO_NOK); } level = pMbLayer->residual.level[0]; for (i = 0; i < 384; i++) { value = h264bsdGetBits(pStrmData, 8); if (value == END_OF_STREAM) return(HANTRO_NOK); *level++ = (i32)value; } } else { partMode = h264bsdMbPartPredMode(pMbLayer->mbType); if ( (partMode == PRED_MODE_INTER) && (h264bsdNumMbPart(pMbLayer->mbType) == 4) ) { tmp = DecodeSubMbPred(pStrmData, &pMbLayer->subMbPred, pMbLayer->mbType, numRefIdxActive); } else { tmp = DecodeMbPred(pStrmData, &pMbLayer->mbPred, pMbLayer->mbType, numRefIdxActive); } if (tmp != HANTRO_OK) return(tmp); if (partMode != PRED_MODE_INTRA16x16) { tmp = h264bsdDecodeExpGolombMapped(pStrmData, &value, (u32)(partMode == PRED_MODE_INTRA4x4)); if (tmp != HANTRO_OK) return(tmp); pMbLayer->codedBlockPattern = value; } else { pMbLayer->codedBlockPattern = CbpIntra16x16(pMbLayer->mbType); } if ( pMbLayer->codedBlockPattern || (partMode == PRED_MODE_INTRA16x16) ) { tmp = h264bsdDecodeExpGolombSigned(pStrmData, &itmp); if (tmp != HANTRO_OK || (itmp < -26) || (itmp > 25) ) return(HANTRO_NOK); pMbLayer->mbQpDelta = itmp; tmp = DecodeResidual(pStrmData, &pMbLayer->residual, pMb, pMbLayer->mbType, pMbLayer->codedBlockPattern); pStrmData->strmBuffReadBits = (u32)(pStrmData->pStrmCurrPos - pStrmData->pStrmBuffStart) * 8 + pStrmData->bitPosInWord; if (tmp != HANTRO_OK) return(tmp); } } 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); }
u32 h264bsdDecodeMacroblockLayerCavlc(strmData_t * pStrmData, macroblockLayer_t * pMbLayer, mbStorage_t * pMb, const sliceHeader_t * pSliceHdr ) { /* Variables */ u32 tmp, i, value; i32 itmp; mbPartPredMode_e partMode; /* Code */ ASSERT(pStrmData); ASSERT(pMbLayer); (void)DWLmemset(pMbLayer->residual.totalCoeff, 0, 24); tmp = h264bsdDecodeExpGolombUnsigned(pStrmData, &value); if(IS_I_SLICE(pSliceHdr->sliceType)) { if((value + 6) > 31 || tmp != HANTRO_OK) return (HANTRO_NOK); pMbLayer->mbType = (mbType_e) (value + 6); } else { if((value + 1) > 31 || tmp != HANTRO_OK) return (HANTRO_NOK); pMbLayer->mbType = (mbType_e) (value + 1); } if(pMbLayer->mbType == I_PCM) { u8 *level; while(!h264bsdIsByteAligned(pStrmData)) { /* pcm_alignment_zero_bit */ tmp = h264bsdGetBits(pStrmData, 1); if(tmp) return (HANTRO_NOK); } level = (u8 *) pMbLayer->residual.rlc; for(i = 384; i > 0; i--) { value = h264bsdGetBits(pStrmData, 8); if(value == END_OF_STREAM) return (HANTRO_NOK); *level++ = (u8) value; } } else { partMode = h264bsdMbPartPredMode(pMbLayer->mbType); if((partMode == PRED_MODE_INTER) && (h264bsdNumMbPart(pMbLayer->mbType) == 4)) { tmp = DecodeSubMbPred(pStrmData, &pMbLayer->subMbPred, pMbLayer->mbType, pSliceHdr->numRefIdxL0Active, pMb ); } else { tmp = DecodeMbPred(pStrmData, &pMbLayer->mbPred, pMbLayer->mbType, pSliceHdr->numRefIdxL0Active, pMb ); } if(tmp != HANTRO_OK) return (tmp); if(partMode != PRED_MODE_INTRA16x16) { tmp = h264bsdDecodeExpGolombMapped(pStrmData, &value, (u32) (partMode == PRED_MODE_INTRA4x4)); if(tmp != HANTRO_OK) return (tmp); pMbLayer->codedBlockPattern = value; } else { pMbLayer->codedBlockPattern = CbpIntra16x16(pMbLayer->mbType); } if(pMbLayer->codedBlockPattern || (partMode == PRED_MODE_INTRA16x16)) { tmp = h264bsdDecodeExpGolombSigned(pStrmData, &itmp); if(tmp != HANTRO_OK || (u32) (itmp + 26) > 51U /*(itmp >= -26) || (itmp < 26) */ ) return (HANTRO_NOK); pMbLayer->mbQpDelta = itmp; tmp = DecodeResidual(pStrmData, pMbLayer, pMb); if(tmp != HANTRO_OK) return (tmp); } } return (HANTRO_OK); }
u32 h264bsdDecodeNalUnit(strmData_t *pStrmData, nalUnit_t *pNalUnit) { /* Variables */ u32 tmp; /* Code */ ASSERT(pStrmData); ASSERT(pNalUnit); ASSERT(pStrmData->bitPosInWord == 0); memset(pNalUnit, 0, sizeof(nalUnit_t)); /* forbidden_zero_bit (not checked to be zero, errors ignored) */ tmp = h264bsdGetBits(pStrmData, 1); /* Assuming that NAL unit starts from byte boundary > don't have to check * following 7 bits for END_OF_STREAM */ if (tmp == END_OF_STREAM) return(HANTRO_NOK); tmp = h264bsdGetBits(pStrmData, 2); pNalUnit->nalRefIdc = tmp; tmp = h264bsdGetBits(pStrmData, 5); pNalUnit->nalUnitType = (nalUnitType_e)tmp; DEBUG_PRINT(("NAL TYPE %d\n", tmp)); /* data partitioning NAL units not supported */ if ( (tmp == NAL_CODED_SLICE_DP_A) || (tmp == NAL_CODED_SLICE_DP_B) || (tmp == NAL_CODED_SLICE_DP_C) ) { ERROR_PRINT(("DP slices not allowed!!!")); return(HANTRO_NOK); } /* nal_ref_idc shall not be zero for these nal_unit_types */ if ( ( (tmp == NAL_SEQ_PARAM_SET) || (tmp == NAL_PIC_PARAM_SET) || (tmp == NAL_CODED_SLICE_IDR) ) && (pNalUnit->nalRefIdc == 0) ) { ERROR_PRINT(("nal_ref_idc shall not be zero!!!")); return(HANTRO_NOK); } /* nal_ref_idc shall be zero for these nal_unit_types */ else if ( ( (tmp == NAL_SEI) || (tmp == NAL_ACCESS_UNIT_DELIMITER) || (tmp == NAL_END_OF_SEQUENCE) || (tmp == NAL_END_OF_STREAM) || (tmp == NAL_FILLER_DATA) ) && (pNalUnit->nalRefIdc != 0) ) { ERROR_PRINT(("nal_ref_idc shall be zero!!!")); return(HANTRO_NOK); } if (pNalUnit->nalUnitType == NAL_PREFIX || pNalUnit->nalUnitType == NAL_CODED_SLICE_EXT) { /* svc_extension_flag */ tmp = h264bsdGetBits(pStrmData, 1); tmp = h264bsdGetBits(pStrmData, 1); pNalUnit->nonIdrFlag = tmp; tmp = h264bsdGetBits(pStrmData, 6); pNalUnit->priorityId = tmp; tmp = h264bsdGetBits(pStrmData, 10); pNalUnit->viewId = tmp; tmp = h264bsdGetBits(pStrmData, 3); pNalUnit->temporalId = tmp; tmp = h264bsdGetBits(pStrmData, 1); pNalUnit->anchorPicFlag = tmp; tmp = h264bsdGetBits(pStrmData, 1); pNalUnit->interViewFlag = tmp; /* reserved_one_bit */ tmp = h264bsdGetBits(pStrmData, 1); if (tmp == END_OF_STREAM) return(HANTRO_NOK); } return(HANTRO_OK); }