1 /************************************************************************** 2 * 3 * Copyright 2010 Thomas Balling Srensen & Orasanu Lucian. 4 * Copyright 2014 Advanced Micro Devices, Inc. 5 * All Rights Reserved. 6 * 7 * Permission is hereby granted, free of charge, to any person obtaining a 8 * copy of this software and associated documentation files (the 9 * "Software"), to deal in the Software without restriction, including 10 * without limitation the rights to use, copy, modify, merge, publish, 11 * distribute, sub license, and/or sell copies of the Software, and to 12 * permit persons to whom the Software is furnished to do so, subject to 13 * the following conditions: 14 * 15 * The above copyright notice and this permission notice (including the 16 * next paragraph) shall be included in all copies or substantial portions 17 * of the Software. 18 * 19 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS 20 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF 21 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. 22 * IN NO EVENT SHALL THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR 23 * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, 24 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE 25 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. 26 * 27 **************************************************************************/ 28 29 #include "util/u_video.h" 30 #include "va_private.h" 31 32 static void resetReferencePictureDesc(struct pipe_h264_picture_desc *h264, 33 unsigned int i) 34 { 35 h264->ref[i] = NULL; 36 h264->frame_num_list[i] = 0; 37 h264->is_long_term[i] = 0; 38 h264->top_is_reference[i] = 0; 39 h264->bottom_is_reference[i] = 0; 40 h264->field_order_cnt_list[i][0] = 0; 41 h264->field_order_cnt_list[i][1] = 0; 42 } 43 44 void vlVaHandlePictureParameterBufferH264(vlVaDriver *drv, vlVaContext *context, vlVaBuffer *buf) 45 { 46 VAPictureParameterBufferH264 *h264 = buf->data; 47 unsigned int top_or_bottom_field; 48 unsigned i; 49 50 assert(buf->size >= sizeof(VAPictureParameterBufferH264) && buf->num_elements == 1); 51 context->desc.h264.slice_count = 0; 52 /*CurrPic*/ 53 context->desc.h264.field_order_cnt[0] = h264->CurrPic.TopFieldOrderCnt; 54 context->desc.h264.field_order_cnt[1] = h264->CurrPic.BottomFieldOrderCnt; 55 /*ReferenceFrames[16]*/ 56 /*picture_width_in_mbs_minus1*/ 57 /*picture_height_in_mbs_minus1*/ 58 /*bit_depth_luma_minus8*/ 59 /*bit_depth_chroma_minus8*/ 60 context->desc.h264.num_ref_frames = h264->num_ref_frames; 61 /*chroma_format_idc*/ 62 /*residual_colour_transform_flag*/ 63 /*gaps_in_frame_num_value_allowed_flag*/ 64 context->desc.h264.pps->sps->frame_mbs_only_flag = 65 h264->seq_fields.bits.frame_mbs_only_flag; 66 context->desc.h264.pps->sps->mb_adaptive_frame_field_flag = 67 h264->seq_fields.bits.mb_adaptive_frame_field_flag; 68 context->desc.h264.pps->sps->direct_8x8_inference_flag = 69 h264->seq_fields.bits.direct_8x8_inference_flag; 70 /*MinLumaBiPredSize8x8*/ 71 context->desc.h264.pps->sps->log2_max_frame_num_minus4 = 72 h264->seq_fields.bits.log2_max_frame_num_minus4; 73 context->desc.h264.pps->sps->pic_order_cnt_type = 74 h264->seq_fields.bits.pic_order_cnt_type; 75 context->desc.h264.pps->sps->log2_max_pic_order_cnt_lsb_minus4 = 76 h264->seq_fields.bits.log2_max_pic_order_cnt_lsb_minus4; 77 context->desc.h264.pps->sps->delta_pic_order_always_zero_flag = 78 h264->seq_fields.bits.delta_pic_order_always_zero_flag; 79 /*num_slice_groups_minus1*/ 80 /*slice_group_map_type*/ 81 /*slice_group_change_rate_minus1*/ 82 context->desc.h264.pps->pic_init_qp_minus26 = 83 h264->pic_init_qp_minus26; 84 /*pic_init_qs_minus26*/ 85 context->desc.h264.pps->chroma_qp_index_offset = 86 h264->chroma_qp_index_offset; 87 context->desc.h264.pps->second_chroma_qp_index_offset = 88 h264->second_chroma_qp_index_offset; 89 context->desc.h264.pps->entropy_coding_mode_flag = 90 h264->pic_fields.bits.entropy_coding_mode_flag; 91 context->desc.h264.pps->weighted_pred_flag = 92 h264->pic_fields.bits.weighted_pred_flag; 93 context->desc.h264.pps->weighted_bipred_idc = 94 h264->pic_fields.bits.weighted_bipred_idc; 95 context->desc.h264.pps->transform_8x8_mode_flag = 96 h264->pic_fields.bits.transform_8x8_mode_flag; 97 context->desc.h264.field_pic_flag = 98 h264->pic_fields.bits.field_pic_flag; 99 context->desc.h264.pps->constrained_intra_pred_flag = 100 h264->pic_fields.bits.constrained_intra_pred_flag; 101 context->desc.h264.pps->bottom_field_pic_order_in_frame_present_flag = 102 h264->pic_fields.bits.pic_order_present_flag; 103 context->desc.h264.pps->deblocking_filter_control_present_flag = 104 h264->pic_fields.bits.deblocking_filter_control_present_flag; 105 context->desc.h264.pps->redundant_pic_cnt_present_flag = 106 h264->pic_fields.bits.redundant_pic_cnt_present_flag; 107 /*reference_pic_flag*/ 108 context->desc.h264.frame_num = h264->frame_num; 109 context->desc.h264.is_reference = h264->pic_fields.bits.reference_pic_flag; 110 context->desc.h264.bottom_field_flag = 111 h264->pic_fields.bits.field_pic_flag && 112 (h264->CurrPic.flags & VA_PICTURE_H264_BOTTOM_FIELD) != 0; 113 114 if (!context->decoder && context->desc.h264.num_ref_frames > 0) 115 context->templat.max_references = MIN2(context->desc.h264.num_ref_frames, 16); 116 117 for (i = 0; i < context->templat.max_references; ++i) { 118 if ((h264->ReferenceFrames[i].flags & VA_PICTURE_H264_INVALID) || 119 (h264->ReferenceFrames[i].picture_id == VA_INVALID_SURFACE)) { 120 resetReferencePictureDesc(&context->desc.h264, i); 121 break; 122 } 123 124 vlVaGetReferenceFrame(drv, h264->ReferenceFrames[i].picture_id, &context->desc.h264.ref[i]); 125 context->desc.h264.frame_num_list[i] = h264->ReferenceFrames[i].frame_idx; 126 127 top_or_bottom_field = h264->ReferenceFrames[i].flags & 128 (VA_PICTURE_H264_TOP_FIELD | VA_PICTURE_H264_BOTTOM_FIELD); 129 context->desc.h264.is_long_term[i] = (h264->ReferenceFrames[i].flags & 130 (VA_PICTURE_H264_SHORT_TERM_REFERENCE | 131 VA_PICTURE_H264_LONG_TERM_REFERENCE)) != 132 VA_PICTURE_H264_SHORT_TERM_REFERENCE; 133 context->desc.h264.top_is_reference[i] = 134 !context->desc.h264.is_long_term[i] || 135 !!(h264->ReferenceFrames[i].flags & VA_PICTURE_H264_TOP_FIELD); 136 context->desc.h264.bottom_is_reference[i] = 137 !context->desc.h264.is_long_term[i] || 138 !!(h264->ReferenceFrames[i].flags & VA_PICTURE_H264_BOTTOM_FIELD); 139 context->desc.h264.field_order_cnt_list[i][0] = 140 top_or_bottom_field != VA_PICTURE_H264_BOTTOM_FIELD ? 141 h264->ReferenceFrames[i].TopFieldOrderCnt: INT_MAX; 142 context->desc.h264.field_order_cnt_list[i][1] = 143 top_or_bottom_field != VA_PICTURE_H264_TOP_FIELD ? 144 h264->ReferenceFrames[i].BottomFieldOrderCnt: INT_MAX; 145 } 146 147 /* Make sure remaining elements are clean */ 148 for (; i < 16; ++i) 149 resetReferencePictureDesc(&context->desc.h264, i); 150 } 151 152 void vlVaHandleIQMatrixBufferH264(vlVaContext *context, vlVaBuffer *buf) 153 { 154 VAIQMatrixBufferH264 *h264 = buf->data; 155 156 assert(buf->size >= sizeof(VAIQMatrixBufferH264) && buf->num_elements == 1); 157 memcpy(&context->desc.h264.pps->ScalingList4x4, h264->ScalingList4x4, 6 * 16); 158 memcpy(&context->desc.h264.pps->ScalingList8x8, h264->ScalingList8x8, 2 * 64); 159 } 160 161 void vlVaHandleSliceParameterBufferH264(vlVaContext *context, vlVaBuffer *buf) 162 { 163 VASliceParameterBufferH264 *h264 = buf->data; 164 165 assert(buf->size >= sizeof(VASliceParameterBufferH264) && buf->num_elements == 1); 166 context->desc.h264.slice_count += buf->num_elements; 167 context->desc.h264.num_ref_idx_l0_active_minus1 = 168 h264->num_ref_idx_l0_active_minus1; 169 context->desc.h264.num_ref_idx_l1_active_minus1 = 170 h264->num_ref_idx_l1_active_minus1; 171 } 172