1 /**************************************************************************
2 *
3 * Copyright 2018 Advanced Micro Devices, Inc.
4 * All Rights Reserved.
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a
7 * copy of this software and associated documentation files (the
8 * "Software"), to deal in the Software without restriction, including
9 * without limitation the rights to use, copy, modify, merge, publish,
10 * distribute, sub license, and/or sell copies of the Software, and to
11 * permit persons to whom the Software is furnished to do so, subject to
12 * the following conditions:
13 *
14 * The above copyright notice and this permission notice (including the
15 * next paragraph) shall be included in all copies or substantial portions
16 * of the Software.
17 *
18 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
19 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
20 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
21 * IN NO EVENT SHALL THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR
22 * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
23 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
24 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
25 *
26 **************************************************************************/
27
28 #include "util/u_handle_table.h"
29 #include "util/u_video.h"
30 #include "va_private.h"
31
32 #include "util/vl_rbsp.h"
33
34 enum H264NALUnitType {
35 H264_NAL_SPS = 7,
36 H264_NAL_PPS = 8,
37 };
38
39 VAStatus
vlVaHandleVAEncPictureParameterBufferTypeH264(vlVaDriver * drv,vlVaContext * context,vlVaBuffer * buf)40 vlVaHandleVAEncPictureParameterBufferTypeH264(vlVaDriver *drv, vlVaContext *context, vlVaBuffer *buf)
41 {
42 VAEncPictureParameterBufferH264 *h264;
43 vlVaBuffer *coded_buf;
44
45 h264 = buf->data;
46 if (h264->pic_fields.bits.idr_pic_flag == 1)
47 context->desc.h264enc.frame_num = 0;
48 context->desc.h264enc.not_referenced = !h264->pic_fields.bits.reference_pic_flag;
49 context->desc.h264enc.pic_order_cnt = h264->CurrPic.TopFieldOrderCnt;
50 context->desc.h264enc.is_ltr = h264->CurrPic.flags & VA_PICTURE_H264_LONG_TERM_REFERENCE;
51 if (context->desc.h264enc.is_ltr)
52 context->desc.h264enc.ltr_index = h264->CurrPic.frame_idx;
53 if (context->desc.h264enc.gop_cnt == 0)
54 context->desc.h264enc.i_remain = context->gop_coeff;
55 else if (context->desc.h264enc.frame_num == 1)
56 context->desc.h264enc.i_remain--;
57
58 context->desc.h264enc.p_remain = context->desc.h264enc.gop_size - context->desc.h264enc.gop_cnt - context->desc.h264enc.i_remain;
59
60 coded_buf = handle_table_get(drv->htab, h264->coded_buf);
61 if (!coded_buf)
62 return VA_STATUS_ERROR_INVALID_BUFFER;
63
64 if (!coded_buf->derived_surface.resource)
65 coded_buf->derived_surface.resource = pipe_buffer_create(drv->pipe->screen, PIPE_BIND_VERTEX_BUFFER,
66 PIPE_USAGE_STAGING, coded_buf->size);
67 context->coded_buf = coded_buf;
68
69 if (context->desc.h264enc.is_ltr)
70 _mesa_hash_table_insert(context->desc.h264enc.frame_idx,
71 UINT_TO_PTR(h264->CurrPic.picture_id + 1),
72 UINT_TO_PTR(context->desc.h264enc.ltr_index));
73 else
74 _mesa_hash_table_insert(context->desc.h264enc.frame_idx,
75 UINT_TO_PTR(h264->CurrPic.picture_id + 1),
76 UINT_TO_PTR(context->desc.h264enc.frame_num));
77
78 if (h264->pic_fields.bits.idr_pic_flag == 1)
79 context->desc.h264enc.picture_type = PIPE_H2645_ENC_PICTURE_TYPE_IDR;
80 else
81 context->desc.h264enc.picture_type = PIPE_H2645_ENC_PICTURE_TYPE_P;
82
83 /* Initialize slice descriptors for this picture */
84 context->desc.h264enc.num_slice_descriptors = 0;
85 memset(&context->desc.h264enc.slices_descriptors, 0, sizeof(context->desc.h264enc.slices_descriptors));
86
87 context->desc.h264enc.quant_i_frames = h264->pic_init_qp;
88 context->desc.h264enc.quant_b_frames = h264->pic_init_qp;
89 context->desc.h264enc.quant_p_frames = h264->pic_init_qp;
90 context->desc.h264enc.gop_cnt++;
91 if (context->desc.h264enc.gop_cnt == context->desc.h264enc.gop_size)
92 context->desc.h264enc.gop_cnt = 0;
93
94 context->desc.h264enc.pic_ctrl.enc_cabac_enable = h264->pic_fields.bits.entropy_coding_mode_flag;
95 context->desc.h264enc.num_ref_idx_l0_active_minus1 = h264->num_ref_idx_l0_active_minus1;
96 context->desc.h264enc.num_ref_idx_l1_active_minus1 = h264->num_ref_idx_l1_active_minus1;
97 context->desc.h264enc.pic_ctrl.deblocking_filter_control_present_flag
98 = h264->pic_fields.bits.deblocking_filter_control_present_flag;
99 context->desc.h264enc.pic_ctrl.redundant_pic_cnt_present_flag
100 = h264->pic_fields.bits.redundant_pic_cnt_present_flag;
101 context->desc.h264enc.pic_ctrl.chroma_qp_index_offset = h264->chroma_qp_index_offset;
102 context->desc.h264enc.pic_ctrl.second_chroma_qp_index_offset
103 = h264->second_chroma_qp_index_offset;
104
105 return VA_STATUS_SUCCESS;
106 }
107
108 VAStatus
vlVaHandleVAEncSliceParameterBufferTypeH264(vlVaDriver * drv,vlVaContext * context,vlVaBuffer * buf)109 vlVaHandleVAEncSliceParameterBufferTypeH264(vlVaDriver *drv, vlVaContext *context, vlVaBuffer *buf)
110 {
111 VAEncSliceParameterBufferH264 *h264;
112
113 h264 = buf->data;
114 memset(&context->desc.h264enc.ref_idx_l0_list, VA_INVALID_ID, sizeof(context->desc.h264enc.ref_idx_l0_list));
115 memset(&context->desc.h264enc.ref_idx_l1_list, VA_INVALID_ID, sizeof(context->desc.h264enc.ref_idx_l1_list));
116
117 if(h264->num_ref_idx_active_override_flag) {
118 context->desc.h264enc.num_ref_idx_l0_active_minus1 = h264->num_ref_idx_l0_active_minus1;
119 context->desc.h264enc.num_ref_idx_l1_active_minus1 = h264->num_ref_idx_l1_active_minus1;
120 }
121
122 for (int i = 0; i < 32; i++) {
123 if (h264->RefPicList0[i].picture_id != VA_INVALID_ID) {
124 context->desc.h264enc.ref_idx_l0_list[i] = PTR_TO_UINT(util_hash_table_get(context->desc.h264enc.frame_idx,
125 UINT_TO_PTR(h264->RefPicList0[i].picture_id + 1)));
126 context->desc.h264enc.l0_is_long_term[i] = h264->RefPicList0[i].flags &
127 VA_PICTURE_H264_LONG_TERM_REFERENCE;
128 }
129 if (h264->RefPicList1[i].picture_id != VA_INVALID_ID && h264->slice_type == 1) {
130 context->desc.h264enc.ref_idx_l1_list[i] = PTR_TO_UINT(util_hash_table_get(context->desc.h264enc.frame_idx,
131 UINT_TO_PTR(h264->RefPicList1[i].picture_id + 1)));
132 context->desc.h264enc.l1_is_long_term[i] = h264->RefPicList1[i].flags &
133 VA_PICTURE_H264_LONG_TERM_REFERENCE;
134 }
135 }
136
137 /**
138 * VAEncSliceParameterBufferH264.slice_type
139 * Slice type.
140 * Range: 0..2, 5..7, i.e. no switching slices.
141 */
142 struct h264_slice_descriptor slice_descriptor;
143 memset(&slice_descriptor, 0, sizeof(slice_descriptor));
144 slice_descriptor.macroblock_address = h264->macroblock_address;
145 slice_descriptor.num_macroblocks = h264->num_macroblocks;
146
147 if ((h264->slice_type == 1) || (h264->slice_type == 6)) {
148 context->desc.h264enc.picture_type = PIPE_H2645_ENC_PICTURE_TYPE_B;
149 slice_descriptor.slice_type = PIPE_H264_SLICE_TYPE_B;
150 } else if ((h264->slice_type == 0) || (h264->slice_type == 5)) {
151 context->desc.h264enc.picture_type = PIPE_H2645_ENC_PICTURE_TYPE_P;
152 slice_descriptor.slice_type = PIPE_H264_SLICE_TYPE_P;
153 } else if ((h264->slice_type == 2) || (h264->slice_type == 7)) {
154 if (context->desc.h264enc.picture_type == PIPE_H2645_ENC_PICTURE_TYPE_IDR) {
155 if (slice_descriptor.macroblock_address == 0) {
156 /* Increment it only for the first slice of the IDR frame */
157 context->desc.h264enc.idr_pic_id++;
158 }
159 slice_descriptor.slice_type = PIPE_H264_SLICE_TYPE_I;
160 } else {
161 context->desc.h264enc.picture_type = PIPE_H2645_ENC_PICTURE_TYPE_I;
162 slice_descriptor.slice_type = PIPE_H264_SLICE_TYPE_I;
163 }
164 } else {
165 context->desc.h264enc.picture_type = PIPE_H2645_ENC_PICTURE_TYPE_SKIP;
166 }
167
168 context->desc.h264enc.pic_ctrl.enc_cabac_init_idc = h264->cabac_init_idc;
169 context->desc.h264enc.dbk.disable_deblocking_filter_idc = h264->disable_deblocking_filter_idc;
170 context->desc.h264enc.dbk.alpha_c0_offset_div2 = h264->slice_alpha_c0_offset_div2;
171 context->desc.h264enc.dbk.beta_offset_div2 = h264->slice_beta_offset_div2;
172
173 /* Handle the slice control parameters */
174 if (context->desc.h264enc.num_slice_descriptors < ARRAY_SIZE(context->desc.h264enc.slices_descriptors)) {
175 context->desc.h264enc.slices_descriptors[context->desc.h264enc.num_slice_descriptors++] = slice_descriptor;
176 } else {
177 return VA_STATUS_ERROR_NOT_ENOUGH_BUFFER;
178 }
179
180 return VA_STATUS_SUCCESS;
181 }
182
183 VAStatus
vlVaHandleVAEncSequenceParameterBufferTypeH264(vlVaDriver * drv,vlVaContext * context,vlVaBuffer * buf)184 vlVaHandleVAEncSequenceParameterBufferTypeH264(vlVaDriver *drv, vlVaContext *context, vlVaBuffer *buf)
185 {
186 uint32_t num_units_in_tick = 0, time_scale = 0;
187
188 VAEncSequenceParameterBufferH264 *h264 = (VAEncSequenceParameterBufferH264 *)buf->data;
189 if (!context->decoder) {
190 context->templat.max_references = h264->max_num_ref_frames;
191 context->templat.level = h264->level_idc;
192 context->decoder = drv->pipe->create_video_codec(drv->pipe, &context->templat);
193 if (!context->decoder)
194 return VA_STATUS_ERROR_ALLOCATION_FAILED;
195
196 getEncParamPresetH264(context);
197 context->desc.h264enc.rate_ctrl[0].vbv_buffer_size = 20000000;
198 context->desc.h264enc.rate_ctrl[0].vbv_buf_lv = 48;
199 context->desc.h264enc.rate_ctrl[0].fill_data_enable = 1;
200 context->desc.h264enc.rate_ctrl[0].enforce_hrd = 1;
201 context->desc.h264enc.rate_ctrl[0].max_qp = 51;
202 context->desc.h264enc.rate_ctrl[0].min_qp = 0;
203 context->desc.h264enc.enable_vui = false;
204 context->desc.h264enc.intra_refresh.mode = INTRA_REFRESH_MODE_NONE;
205 context->desc.h264enc.intra_refresh.offset = 0;
206 context->desc.h264enc.intra_refresh.region_size = 0;
207 context->desc.h264enc.intra_refresh.need_sequence_header = 0;
208 }
209
210 context->desc.h264enc.ip_period = h264->ip_period;
211 context->desc.h264enc.intra_idr_period =
212 h264->intra_idr_period != 0 ? h264->intra_idr_period : PIPE_DEFAULT_INTRA_IDR_PERIOD;
213 context->gop_coeff = ((1024 + context->desc.h264enc.intra_idr_period - 1) /
214 context->desc.h264enc.intra_idr_period + 1) / 2 * 2;
215 if (context->gop_coeff > VL_VA_ENC_GOP_COEFF)
216 context->gop_coeff = VL_VA_ENC_GOP_COEFF;
217 context->desc.h264enc.gop_size = context->desc.h264enc.intra_idr_period * context->gop_coeff;
218 context->desc.h264enc.seq.pic_order_cnt_type = h264->seq_fields.bits.pic_order_cnt_type;
219 context->desc.h264enc.seq.log2_max_frame_num_minus4 = h264->seq_fields.bits.log2_max_frame_num_minus4;
220 context->desc.h264enc.seq.log2_max_pic_order_cnt_lsb_minus4 = h264->seq_fields.bits.log2_max_pic_order_cnt_lsb_minus4;
221 context->desc.h264enc.seq.vui_parameters_present_flag = h264->vui_parameters_present_flag;
222 if (h264->vui_parameters_present_flag) {
223 context->desc.h264enc.seq.vui_flags.aspect_ratio_info_present_flag =
224 h264->vui_fields.bits.aspect_ratio_info_present_flag;
225 context->desc.h264enc.seq.aspect_ratio_idc = h264->aspect_ratio_idc;
226 context->desc.h264enc.seq.sar_width = h264->sar_width;
227 context->desc.h264enc.seq.sar_height = h264->sar_height;
228 context->desc.h264enc.seq.vui_flags.timing_info_present_flag =
229 h264->vui_fields.bits.timing_info_present_flag;
230 num_units_in_tick = h264->num_units_in_tick;
231 time_scale = h264->time_scale;
232 context->desc.h264enc.seq.vui_flags.fixed_frame_rate_flag =
233 h264->vui_fields.bits.fixed_frame_rate_flag;
234 context->desc.h264enc.seq.vui_flags.low_delay_hrd_flag =
235 h264->vui_fields.bits.low_delay_hrd_flag;
236 context->desc.h264enc.seq.vui_flags.bitstream_restriction_flag =
237 h264->vui_fields.bits.bitstream_restriction_flag;
238 context->desc.h264enc.seq.vui_flags.motion_vectors_over_pic_boundaries_flag =
239 h264->vui_fields.bits.motion_vectors_over_pic_boundaries_flag;
240 context->desc.h264enc.seq.log2_max_mv_length_vertical =
241 h264->vui_fields.bits.log2_max_mv_length_vertical;
242 context->desc.h264enc.seq.log2_max_mv_length_horizontal =
243 h264->vui_fields.bits.log2_max_mv_length_horizontal;
244 } else {
245 context->desc.h264enc.seq.vui_flags.timing_info_present_flag = 0;
246 context->desc.h264enc.seq.vui_flags.fixed_frame_rate_flag = 0;
247 context->desc.h264enc.seq.vui_flags.low_delay_hrd_flag = 0;
248 context->desc.h264enc.seq.vui_flags.bitstream_restriction_flag = 0;
249 context->desc.h264enc.seq.vui_flags.motion_vectors_over_pic_boundaries_flag = 0;
250 context->desc.h264enc.seq.log2_max_mv_length_vertical = 0;
251 context->desc.h264enc.seq.log2_max_mv_length_horizontal = 0;
252 }
253
254 if (!context->desc.h264enc.seq.vui_flags.timing_info_present_flag) {
255 /* if not present, set default value */
256 num_units_in_tick = PIPE_DEFAULT_FRAME_RATE_DEN;
257 time_scale = PIPE_DEFAULT_FRAME_RATE_NUM * 2;
258 }
259
260 context->desc.h264enc.seq.num_units_in_tick = num_units_in_tick;
261 context->desc.h264enc.seq.time_scale = time_scale;
262 context->desc.h264enc.rate_ctrl[0].frame_rate_num = time_scale / 2;
263 context->desc.h264enc.rate_ctrl[0].frame_rate_den = num_units_in_tick;
264
265 if (h264->frame_cropping_flag) {
266 context->desc.h264enc.seq.enc_frame_cropping_flag = h264->frame_cropping_flag;
267 context->desc.h264enc.seq.enc_frame_crop_left_offset = h264->frame_crop_left_offset;
268 context->desc.h264enc.seq.enc_frame_crop_right_offset = h264->frame_crop_right_offset;
269 context->desc.h264enc.seq.enc_frame_crop_top_offset = h264->frame_crop_top_offset;
270 context->desc.h264enc.seq.enc_frame_crop_bottom_offset = h264->frame_crop_bottom_offset;
271 }
272 return VA_STATUS_SUCCESS;
273 }
274
275 VAStatus
vlVaHandleVAEncMiscParameterTypeRateControlH264(vlVaContext * context,VAEncMiscParameterBuffer * misc)276 vlVaHandleVAEncMiscParameterTypeRateControlH264(vlVaContext *context, VAEncMiscParameterBuffer *misc)
277 {
278 unsigned temporal_id;
279 VAEncMiscParameterRateControl *rc = (VAEncMiscParameterRateControl *)misc->data;
280
281 temporal_id = context->desc.h264enc.rate_ctrl[0].rate_ctrl_method !=
282 PIPE_H2645_ENC_RATE_CONTROL_METHOD_DISABLE ?
283 rc->rc_flags.bits.temporal_id :
284 0;
285
286 if (context->desc.h264enc.rate_ctrl[0].rate_ctrl_method ==
287 PIPE_H2645_ENC_RATE_CONTROL_METHOD_CONSTANT)
288 context->desc.h264enc.rate_ctrl[temporal_id].target_bitrate =
289 rc->bits_per_second;
290 else
291 context->desc.h264enc.rate_ctrl[temporal_id].target_bitrate =
292 rc->bits_per_second * (rc->target_percentage / 100.0);
293
294 if (context->desc.h264enc.seq.num_temporal_layers > 0 &&
295 temporal_id >= context->desc.h264enc.seq.num_temporal_layers)
296 return VA_STATUS_ERROR_INVALID_PARAMETER;
297
298 context->desc.h264enc.rate_ctrl[temporal_id].fill_data_enable = !(rc->rc_flags.bits.disable_bit_stuffing);
299 /* context->desc.h264enc.rate_ctrl[temporal_id].skip_frame_enable = !(rc->rc_flags.bits.disable_frame_skip); */
300 context->desc.h264enc.rate_ctrl[temporal_id].skip_frame_enable = 0;
301 context->desc.h264enc.rate_ctrl[temporal_id].peak_bitrate = rc->bits_per_second;
302
303 if ((context->desc.h264enc.rate_ctrl[0].rate_ctrl_method == PIPE_H2645_ENC_RATE_CONTROL_METHOD_CONSTANT) ||
304 (context->desc.h264enc.rate_ctrl[0].rate_ctrl_method == PIPE_H2645_ENC_RATE_CONTROL_METHOD_CONSTANT_SKIP))
305 context->desc.h264enc.rate_ctrl[temporal_id].vbv_buffer_size =
306 context->desc.h264enc.rate_ctrl[temporal_id].target_bitrate;
307 else if (context->desc.h264enc.rate_ctrl[temporal_id].target_bitrate < 2000000)
308 context->desc.h264enc.rate_ctrl[temporal_id].vbv_buffer_size =
309 MIN2((context->desc.h264enc.rate_ctrl[0].target_bitrate * 2.75), 2000000);
310 else
311 context->desc.h264enc.rate_ctrl[temporal_id].vbv_buffer_size =
312 context->desc.h264enc.rate_ctrl[temporal_id].target_bitrate;
313
314 context->desc.h264enc.rate_ctrl[temporal_id].max_qp = rc->max_qp;
315 context->desc.h264enc.rate_ctrl[temporal_id].min_qp = rc->min_qp;
316 /* Distinguishes from the default params set for these values in other
317 functions and app specific params passed down */
318 context->desc.h264enc.rate_ctrl[temporal_id].app_requested_qp_range = ((rc->max_qp > 0) || (rc->min_qp > 0));
319
320 if (context->desc.h264enc.rate_ctrl[0].rate_ctrl_method ==
321 PIPE_H2645_ENC_RATE_CONTROL_METHOD_QUALITY_VARIABLE)
322 context->desc.h264enc.rate_ctrl[temporal_id].vbr_quality_factor =
323 rc->quality_factor;
324
325 return VA_STATUS_SUCCESS;
326 }
327
328 VAStatus
vlVaHandleVAEncMiscParameterTypeFrameRateH264(vlVaContext * context,VAEncMiscParameterBuffer * misc)329 vlVaHandleVAEncMiscParameterTypeFrameRateH264(vlVaContext *context, VAEncMiscParameterBuffer *misc)
330 {
331 unsigned temporal_id;
332 VAEncMiscParameterFrameRate *fr = (VAEncMiscParameterFrameRate *)misc->data;
333
334 temporal_id = context->desc.h264enc.rate_ctrl[0].rate_ctrl_method !=
335 PIPE_H2645_ENC_RATE_CONTROL_METHOD_DISABLE ?
336 fr->framerate_flags.bits.temporal_id :
337 0;
338
339 if (context->desc.h264enc.seq.num_temporal_layers > 0 &&
340 temporal_id >= context->desc.h264enc.seq.num_temporal_layers)
341 return VA_STATUS_ERROR_INVALID_PARAMETER;
342
343 if (fr->framerate & 0xffff0000) {
344 context->desc.h264enc.rate_ctrl[temporal_id].frame_rate_num = fr->framerate & 0xffff;
345 context->desc.h264enc.rate_ctrl[temporal_id].frame_rate_den = fr->framerate >> 16 & 0xffff;
346 } else {
347 context->desc.h264enc.rate_ctrl[temporal_id].frame_rate_num = fr->framerate;
348 context->desc.h264enc.rate_ctrl[temporal_id].frame_rate_den = 1;
349 }
350
351 return VA_STATUS_SUCCESS;
352 }
353
parseEncHrdParamsH264(struct vl_rbsp * rbsp,pipe_h264_enc_hrd_params * hrd_params)354 static void parseEncHrdParamsH264(struct vl_rbsp *rbsp, pipe_h264_enc_hrd_params* hrd_params)
355 {
356 unsigned i;
357
358 hrd_params->cpb_cnt_minus1 = vl_rbsp_ue(rbsp);
359 hrd_params->bit_rate_scale = vl_rbsp_u(rbsp, 4);
360 hrd_params->cpb_size_scale = vl_rbsp_u(rbsp, 4);
361 for (i = 0; i <= hrd_params->cpb_cnt_minus1; ++i) {
362 hrd_params->bit_rate_value_minus1[i] = vl_rbsp_ue(rbsp);
363 hrd_params->cpb_size_value_minus1[i] = vl_rbsp_ue(rbsp);
364 hrd_params->cbr_flag[i] = vl_rbsp_u(rbsp, 1);
365 }
366 hrd_params->initial_cpb_removal_delay_length_minus1 = vl_rbsp_u(rbsp, 5);
367 hrd_params->cpb_removal_delay_length_minus1 = vl_rbsp_u(rbsp, 5);
368 hrd_params->dpb_output_delay_length_minus1 = vl_rbsp_u(rbsp, 5);
369 hrd_params->time_offset_length = vl_rbsp_u(rbsp, 5);
370 }
371
parseEncSpsParamsH264(vlVaContext * context,struct vl_rbsp * rbsp)372 static void parseEncSpsParamsH264(vlVaContext *context, struct vl_rbsp *rbsp)
373 {
374 unsigned i, profile_idc, num_ref_frames_in_pic_order_cnt_cycle;
375
376 profile_idc = vl_rbsp_u(rbsp, 8);
377
378 context->desc.h264enc.seq.enc_constraint_set_flags =
379 vl_rbsp_u(rbsp, 6); /* constraint_set_flags */
380 vl_rbsp_u(rbsp, 2); /* reserved_zero_2bits */
381 vl_rbsp_u(rbsp, 8); /* level_idc */
382
383 vl_rbsp_ue(rbsp); /* seq_parameter_set_id */
384
385 if (profile_idc == 100 || profile_idc == 110 ||
386 profile_idc == 122 || profile_idc == 244 || profile_idc == 44 ||
387 profile_idc == 83 || profile_idc == 86 || profile_idc == 118 ||
388 profile_idc == 128 || profile_idc == 138 || profile_idc == 139 ||
389 profile_idc == 134 || profile_idc == 135) {
390
391 if (vl_rbsp_ue(rbsp) == 3) /* chroma_format_idc */
392 vl_rbsp_u(rbsp, 1); /* separate_colour_plane_flag */
393
394 vl_rbsp_ue(rbsp); /* bit_depth_luma_minus8 */
395 vl_rbsp_ue(rbsp); /* bit_depth_chroma_minus8 */
396 vl_rbsp_u(rbsp, 1); /* qpprime_y_zero_transform_bypass_flag */
397
398 if (vl_rbsp_u(rbsp, 1)) /* seq_scaling_matrix_present_flag */
399 return; /* TODO */
400 }
401
402 context->desc.h264enc.seq.log2_max_frame_num_minus4 = vl_rbsp_ue(rbsp); /* log2_max_frame_num_minus4 */
403 context->desc.h264enc.seq.pic_order_cnt_type = vl_rbsp_ue(rbsp); /* pic_order_cnt_type */
404
405 if (context->desc.h264enc.seq.pic_order_cnt_type == 0)
406 context->desc.h264enc.seq.log2_max_pic_order_cnt_lsb_minus4 = vl_rbsp_ue(rbsp); /* log2_max_pic_order_cnt_lsb_minus4 */
407 else if (context->desc.h264enc.seq.pic_order_cnt_type == 1) {
408 vl_rbsp_u(rbsp, 1); /* delta_pic_order_always_zero_flag */
409 vl_rbsp_se(rbsp); /* offset_for_non_ref_pic */
410 vl_rbsp_se(rbsp); /* offset_for_top_to_bottom_field */
411 num_ref_frames_in_pic_order_cnt_cycle = vl_rbsp_ue(rbsp);
412 for (i = 0; i < num_ref_frames_in_pic_order_cnt_cycle; ++i)
413 vl_rbsp_se(rbsp); /* offset_for_ref_frame[i] */
414 }
415
416 vl_rbsp_ue(rbsp); /* max_num_ref_frames */
417 vl_rbsp_u(rbsp, 1); /* gaps_in_frame_num_value_allowed_flag */
418 vl_rbsp_ue(rbsp); /* pic_width_in_mbs_minus1 */
419 vl_rbsp_ue(rbsp); /* pic_height_in_map_units_minus1 */
420 if (!vl_rbsp_u(rbsp, 1)) /* frame_mbs_only_flag */
421 vl_rbsp_u(rbsp, 1); /* mb_adaptive_frame_field_flag */
422
423 vl_rbsp_u(rbsp, 1); /* direct_8x8_inference_flag */
424 if (vl_rbsp_u(rbsp, 1)) { /* frame_cropping_flag */
425 vl_rbsp_ue(rbsp); /* frame_crop_left_offset */
426 vl_rbsp_ue(rbsp); /* frame_crop_right_offset */
427 vl_rbsp_ue(rbsp); /* frame_crop_top_offset */
428 vl_rbsp_ue(rbsp); /* frame_crop_bottom_offset */
429 }
430
431 context->desc.h264enc.seq.vui_parameters_present_flag = vl_rbsp_u(rbsp, 1);
432 if (context->desc.h264enc.seq.vui_parameters_present_flag) {
433 context->desc.h264enc.seq.vui_flags.aspect_ratio_info_present_flag = vl_rbsp_u(rbsp, 1);
434 if (context->desc.h264enc.seq.vui_flags.aspect_ratio_info_present_flag) {
435 context->desc.h264enc.seq.aspect_ratio_idc = vl_rbsp_u(rbsp, 8);
436 if (context->desc.h264enc.seq.aspect_ratio_idc == 255 /* Extended_SAR */) {
437 context->desc.h264enc.seq.sar_width = vl_rbsp_u(rbsp, 16);
438 context->desc.h264enc.seq.sar_height = vl_rbsp_u(rbsp, 16);
439 }
440 }
441
442 context->desc.h264enc.seq.vui_flags.overscan_info_present_flag = vl_rbsp_u(rbsp, 1);
443 if (context->desc.h264enc.seq.vui_flags.overscan_info_present_flag)
444 context->desc.h264enc.seq.vui_flags.overscan_appropriate_flag = vl_rbsp_u(rbsp, 1);
445
446 context->desc.h264enc.seq.vui_flags.video_signal_type_present_flag = vl_rbsp_u(rbsp, 1);
447 if (context->desc.h264enc.seq.vui_flags.video_signal_type_present_flag) {
448 context->desc.h264enc.seq.video_format = vl_rbsp_u(rbsp, 3);
449 context->desc.h264enc.seq.video_full_range_flag = vl_rbsp_u(rbsp, 1);
450 context->desc.h264enc.seq.vui_flags.colour_description_present_flag = vl_rbsp_u(rbsp, 1);
451 if (context->desc.h264enc.seq.vui_flags.colour_description_present_flag) {
452 context->desc.h264enc.seq.colour_primaries = vl_rbsp_u(rbsp, 8);
453 context->desc.h264enc.seq.transfer_characteristics = vl_rbsp_u(rbsp, 8);
454 context->desc.h264enc.seq.matrix_coefficients = vl_rbsp_u(rbsp, 8);
455 }
456 }
457
458 context->desc.h264enc.seq.vui_flags.chroma_loc_info_present_flag = vl_rbsp_u(rbsp, 1);
459 if (context->desc.h264enc.seq.vui_flags.chroma_loc_info_present_flag) {
460 context->desc.h264enc.seq.chroma_sample_loc_type_top_field = vl_rbsp_ue(rbsp);
461 context->desc.h264enc.seq.chroma_sample_loc_type_bottom_field = vl_rbsp_ue(rbsp);
462 }
463
464 context->desc.h264enc.seq.vui_flags.timing_info_present_flag = vl_rbsp_u(rbsp, 1);
465 if (context->desc.h264enc.seq.vui_flags.timing_info_present_flag) {
466 context->desc.h264enc.seq.num_units_in_tick = vl_rbsp_u(rbsp, 32);
467 context->desc.h264enc.seq.time_scale = vl_rbsp_u(rbsp, 32);
468 context->desc.h264enc.seq.vui_flags.fixed_frame_rate_flag = vl_rbsp_u(rbsp, 1);
469 }
470
471 context->desc.h264enc.seq.vui_flags.nal_hrd_parameters_present_flag = vl_rbsp_u(rbsp, 1);
472 if (context->desc.h264enc.seq.vui_flags.nal_hrd_parameters_present_flag)
473 parseEncHrdParamsH264(rbsp, &context->desc.h264enc.seq.nal_hrd_parameters);
474
475 context->desc.h264enc.seq.vui_flags.vcl_hrd_parameters_present_flag = vl_rbsp_u(rbsp, 1);
476 if (context->desc.h264enc.seq.vui_flags.vcl_hrd_parameters_present_flag)
477 parseEncHrdParamsH264(rbsp, &context->desc.h264enc.seq.vcl_hrd_parameters);
478
479 if (context->desc.h264enc.seq.vui_flags.nal_hrd_parameters_present_flag ||
480 context->desc.h264enc.seq.vui_flags.vcl_hrd_parameters_present_flag)
481 context->desc.h264enc.seq.vui_flags.low_delay_hrd_flag = vl_rbsp_u(rbsp, 1);
482
483 context->desc.h264enc.seq.vui_flags.pic_struct_present_flag = vl_rbsp_u(rbsp, 1);
484
485 context->desc.h264enc.seq.vui_flags.bitstream_restriction_flag = vl_rbsp_u(rbsp, 1);
486 if (context->desc.h264enc.seq.vui_flags.bitstream_restriction_flag) {
487 context->desc.h264enc.seq.vui_flags.motion_vectors_over_pic_boundaries_flag = vl_rbsp_u(rbsp, 1);
488 context->desc.h264enc.seq.max_bytes_per_pic_denom = vl_rbsp_ue(rbsp);
489 context->desc.h264enc.seq.max_bits_per_mb_denom = vl_rbsp_ue(rbsp);
490 context->desc.h264enc.seq.log2_max_mv_length_horizontal = vl_rbsp_ue(rbsp);
491 context->desc.h264enc.seq.log2_max_mv_length_vertical = vl_rbsp_ue(rbsp);
492 context->desc.h264enc.seq.max_num_reorder_frames = vl_rbsp_ue(rbsp);
493 context->desc.h264enc.seq.max_dec_frame_buffering = vl_rbsp_ue(rbsp);
494 }
495 }
496 }
497
498 VAStatus
vlVaHandleVAEncPackedHeaderDataBufferTypeH264(vlVaContext * context,vlVaBuffer * buf)499 vlVaHandleVAEncPackedHeaderDataBufferTypeH264(vlVaContext *context, vlVaBuffer *buf)
500 {
501 struct vl_vlc vlc = {0};
502 vl_vlc_init(&vlc, 1, (const void * const*)&buf->data, &buf->size);
503
504 while (vl_vlc_bits_left(&vlc) > 0) {
505 /* search the first 64 bytes for a startcode */
506 for (int i = 0; i < 64 && vl_vlc_bits_left(&vlc) >= 24; ++i) {
507 if (vl_vlc_peekbits(&vlc, 24) == 0x000001)
508 break;
509 vl_vlc_eatbits(&vlc, 8);
510 vl_vlc_fillbits(&vlc);
511 }
512 vl_vlc_eatbits(&vlc, 24); /* eat the startcode */
513
514 if (vl_vlc_valid_bits(&vlc) < 15)
515 vl_vlc_fillbits(&vlc);
516
517 vl_vlc_eatbits(&vlc, 3);
518 unsigned nal_unit_type = vl_vlc_get_uimsbf(&vlc, 5);
519
520 struct vl_rbsp rbsp;
521 vl_rbsp_init(&rbsp, &vlc, ~0, context->packed_header_emulation_bytes);
522
523 switch(nal_unit_type) {
524 case H264_NAL_SPS:
525 parseEncSpsParamsH264(context, &rbsp);
526 break;
527 case H264_NAL_PPS:
528 default:
529 break;
530 }
531
532 if (!context->packed_header_emulation_bytes)
533 break;
534 }
535
536 return VA_STATUS_SUCCESS;
537 }
538
539 VAStatus
vlVaHandleVAEncMiscParameterTypeTemporalLayerH264(vlVaContext * context,VAEncMiscParameterBuffer * misc)540 vlVaHandleVAEncMiscParameterTypeTemporalLayerH264(vlVaContext *context, VAEncMiscParameterBuffer *misc)
541 {
542 VAEncMiscParameterTemporalLayerStructure *tl = (VAEncMiscParameterTemporalLayerStructure *)misc->data;
543
544 context->desc.h264enc.seq.num_temporal_layers = tl->number_of_layers;
545
546 return VA_STATUS_SUCCESS;
547 }
548
549 VAStatus
vlVaHandleVAEncMiscParameterTypeQualityLevelH264(vlVaContext * context,VAEncMiscParameterBuffer * misc)550 vlVaHandleVAEncMiscParameterTypeQualityLevelH264(vlVaContext *context, VAEncMiscParameterBuffer *misc)
551 {
552 VAEncMiscParameterBufferQualityLevel *ql = (VAEncMiscParameterBufferQualityLevel *)misc->data;
553 vlVaHandleVAEncMiscParameterTypeQualityLevel(&context->desc.h264enc.quality_modes,
554 (vlVaQualityBits *)&ql->quality_level);
555
556 return VA_STATUS_SUCCESS;
557 }
558
559 VAStatus
vlVaHandleVAEncMiscParameterTypeMaxFrameSizeH264(vlVaContext * context,VAEncMiscParameterBuffer * misc)560 vlVaHandleVAEncMiscParameterTypeMaxFrameSizeH264(vlVaContext *context, VAEncMiscParameterBuffer *misc)
561 {
562 VAEncMiscParameterBufferMaxFrameSize *ms = (VAEncMiscParameterBufferMaxFrameSize *)misc->data;
563 context->desc.h264enc.rate_ctrl[0].max_au_size = ms->max_frame_size;
564 return VA_STATUS_SUCCESS;
565 }
566
567 VAStatus
vlVaHandleVAEncMiscParameterTypeHRDH264(vlVaContext * context,VAEncMiscParameterBuffer * misc)568 vlVaHandleVAEncMiscParameterTypeHRDH264(vlVaContext *context, VAEncMiscParameterBuffer *misc)
569 {
570 VAEncMiscParameterHRD *ms = (VAEncMiscParameterHRD *)misc->data;
571
572 if (ms->buffer_size) {
573 context->desc.h264enc.rate_ctrl[0].vbv_buffer_size = ms->buffer_size;
574 context->desc.h264enc.rate_ctrl[0].vbv_buf_lv = (ms->initial_buffer_fullness << 6 ) / ms->buffer_size;
575 context->desc.h264enc.rate_ctrl[0].vbv_buf_initial_size = ms->initial_buffer_fullness;
576 /* Distinguishes from the default params set for these values in other
577 functions and app specific params passed down via HRD buffer */
578 context->desc.h264enc.rate_ctrl[0].app_requested_hrd_buffer = true;
579 }
580
581 return VA_STATUS_SUCCESS;
582 }
583
getEncParamPresetH264(vlVaContext * context)584 void getEncParamPresetH264(vlVaContext *context)
585 {
586 //rate control
587 if (context->desc.h264enc.rate_ctrl[0].frame_rate_num == 0 ||
588 context->desc.h264enc.rate_ctrl[0].frame_rate_den == 0) {
589 context->desc.h264enc.rate_ctrl[0].frame_rate_num = 30;
590 context->desc.h264enc.rate_ctrl[0].frame_rate_den = 1;
591 }
592 context->desc.h264enc.rate_ctrl[0].target_bits_picture =
593 context->desc.h264enc.rate_ctrl[0].target_bitrate *
594 ((float)context->desc.h264enc.rate_ctrl[0].frame_rate_den /
595 context->desc.h264enc.rate_ctrl[0].frame_rate_num);
596 context->desc.h264enc.rate_ctrl[0].peak_bits_picture_integer =
597 context->desc.h264enc.rate_ctrl[0].peak_bitrate *
598 ((float)context->desc.h264enc.rate_ctrl[0].frame_rate_den /
599 context->desc.h264enc.rate_ctrl[0].frame_rate_num);
600
601 context->desc.h264enc.rate_ctrl[0].peak_bits_picture_fraction = 0;
602 }
603