1 /**************************************************************************
2 *
3 * Copyright 2010 Thomas Balling Sørensen & 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
resetReferencePictureDesc(struct pipe_h264_picture_desc * h264,unsigned int i)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
vlVaHandlePictureParameterBufferH264(vlVaDriver * drv,vlVaContext * context,vlVaBuffer * buf)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
vlVaHandleIQMatrixBufferH264(vlVaContext * context,vlVaBuffer * buf)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
vlVaHandleSliceParameterBufferH264(vlVaContext * context,vlVaBuffer * buf)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