2 * Copyright © 2011 Intel Corporation
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the
6 * "Software"), to deal in the Software without restriction, including
7 * without limitation the rights to use, copy, modify, merge, publish,
8 * distribute, sub license, and/or sell copies of the Software, and to
9 * permit persons to whom the Software is furnished to do so, subject to
10 * the following conditions:
12 * The above copyright notice and this permission notice (including the
13 * next paragraph) shall be included in all copies or substantial portions
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
17 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
18 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
19 * IN NO EVENT SHALL PRECISION INSIGHT AND/OR ITS SUPPLIERS BE LIABLE FOR
20 * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
21 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
22 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
30 #include <va/va_enc_h264.h>
31 #include <va/va_enc_mpeg2.h>
33 #include "i965_encoder_utils.h"
35 #define BITSTREAM_ALLOCATE_STEPPING 4096
37 #define NAL_REF_IDC_NONE 0
38 #define NAL_REF_IDC_LOW 1
39 #define NAL_REF_IDC_MEDIUM 2
40 #define NAL_REF_IDC_HIGH 3
48 #define SLICE_TYPE_P 0
49 #define SLICE_TYPE_B 1
50 #define SLICE_TYPE_I 2
52 #define IS_I_SLICE(type) (SLICE_TYPE_I == (type) || SLICE_TYPE_I == (type - 5))
53 #define IS_P_SLICE(type) (SLICE_TYPE_P == (type) || SLICE_TYPE_P == (type - 5))
54 #define IS_B_SLICE(type) (SLICE_TYPE_B == (type) || SLICE_TYPE_B == (type - 5))
56 #define ENTROPY_MODE_CAVLC 0
57 #define ENTROPY_MODE_CABAC 1
59 #define PROFILE_IDC_BASELINE 66
60 #define PROFILE_IDC_MAIN 77
61 #define PROFILE_IDC_HIGH 100
63 struct __avc_bitstream {
66 int max_size_in_dword;
69 typedef struct __avc_bitstream avc_bitstream;
72 swap32(unsigned int val)
74 unsigned char *pval = (unsigned char *)&val;
76 return ((pval[0] << 24) |
83 avc_bitstream_start(avc_bitstream *bs)
85 bs->max_size_in_dword = BITSTREAM_ALLOCATE_STEPPING;
86 bs->buffer = calloc(bs->max_size_in_dword * sizeof(int), 1);
91 avc_bitstream_end(avc_bitstream *bs)
93 int pos = (bs->bit_offset >> 5);
94 int bit_offset = (bs->bit_offset & 0x1f);
95 int bit_left = 32 - bit_offset;
98 bs->buffer[pos] = swap32((bs->buffer[pos] << bit_left));
105 avc_bitstream_put_ui(avc_bitstream *bs, unsigned int val, int size_in_bits)
107 int pos = (bs->bit_offset >> 5);
108 int bit_offset = (bs->bit_offset & 0x1f);
109 int bit_left = 32 - bit_offset;
114 if (size_in_bits < 32)
115 val &= (( 1 << size_in_bits) - 1);
117 bs->bit_offset += size_in_bits;
119 if (bit_left > size_in_bits) {
120 bs->buffer[pos] = (bs->buffer[pos] << size_in_bits | val);
122 size_in_bits -= bit_left;
123 bs->buffer[pos] = (bs->buffer[pos] << bit_left) | (val >> size_in_bits);
124 bs->buffer[pos] = swap32(bs->buffer[pos]);
126 if (pos + 1 == bs->max_size_in_dword) {
127 bs->max_size_in_dword += BITSTREAM_ALLOCATE_STEPPING;
128 bs->buffer = realloc(bs->buffer, bs->max_size_in_dword * sizeof(unsigned int));
134 bs->buffer[pos + 1] = val;
139 avc_bitstream_put_ue(avc_bitstream *bs, unsigned int val)
141 int size_in_bits = 0;
149 avc_bitstream_put_ui(bs, 0, size_in_bits - 1); // leading zero
150 avc_bitstream_put_ui(bs, val, size_in_bits);
154 avc_bitstream_put_se(avc_bitstream *bs, int val)
156 unsigned int new_val;
161 new_val = 2 * val - 1;
163 avc_bitstream_put_ue(bs, new_val);
167 avc_bitstream_byte_aligning(avc_bitstream *bs, int bit)
169 int bit_offset = (bs->bit_offset & 0x7);
170 int bit_left = 8 - bit_offset;
176 assert(bit == 0 || bit == 1);
179 new_val = (1 << bit_left) - 1;
183 avc_bitstream_put_ui(bs, new_val, bit_left);
185 static void avc_rbsp_trailing_bits(avc_bitstream *bs)
187 avc_bitstream_put_ui(bs, 1, 1);
188 avc_bitstream_byte_aligning(bs, 0);
190 static void nal_start_code_prefix(avc_bitstream *bs)
192 avc_bitstream_put_ui(bs, 0x00000001, 32);
195 static void nal_header(avc_bitstream *bs, int nal_ref_idc, int nal_unit_type)
197 avc_bitstream_put_ui(bs, 0, 1); /* forbidden_zero_bit: 0 */
198 avc_bitstream_put_ui(bs, nal_ref_idc, 2);
199 avc_bitstream_put_ui(bs, nal_unit_type, 5);
203 slice_header(avc_bitstream *bs,
204 VAEncSequenceParameterBufferH264 *sps_param,
205 VAEncPictureParameterBufferH264 *pic_param,
206 VAEncSliceParameterBufferH264 *slice_param)
208 int first_mb_in_slice = slice_param->macroblock_address;
210 avc_bitstream_put_ue(bs, first_mb_in_slice); /* first_mb_in_slice: 0 */
211 avc_bitstream_put_ue(bs, slice_param->slice_type); /* slice_type */
212 avc_bitstream_put_ue(bs, slice_param->pic_parameter_set_id); /* pic_parameter_set_id: 0 */
213 avc_bitstream_put_ui(bs, pic_param->frame_num, sps_param->seq_fields.bits.log2_max_frame_num_minus4 + 4); /* frame_num */
215 /* frame_mbs_only_flag == 1 */
216 if (!sps_param->seq_fields.bits.frame_mbs_only_flag) {
221 if (pic_param->pic_fields.bits.idr_pic_flag)
222 avc_bitstream_put_ue(bs, slice_param->idr_pic_id); /* idr_pic_id: 0 */
224 if (sps_param->seq_fields.bits.pic_order_cnt_type == 0) {
225 avc_bitstream_put_ui(bs, pic_param->CurrPic.TopFieldOrderCnt, sps_param->seq_fields.bits.log2_max_pic_order_cnt_lsb_minus4 + 4);
226 /* pic_order_present_flag == 0 */
232 /* redundant_pic_cnt_present_flag == 0 */
235 if (IS_P_SLICE(slice_param->slice_type)) {
236 avc_bitstream_put_ui(bs, 0, 1); /* num_ref_idx_active_override_flag: 0 */
238 /* ref_pic_list_reordering */
239 avc_bitstream_put_ui(bs, 0, 1); /* ref_pic_list_reordering_flag_l0: 0 */
240 } else if (IS_B_SLICE(slice_param->slice_type)) {
241 avc_bitstream_put_ui(bs, slice_param->direct_spatial_mv_pred_flag, 1); /* direct_spatial_mv_pred: 1 */
242 avc_bitstream_put_ui(bs, 0, 1); /* num_ref_idx_active_override_flag: 0 */
244 /* ref_pic_list_reordering */
245 avc_bitstream_put_ui(bs, 0, 1); /* ref_pic_list_reordering_flag_l0: 0 */
246 avc_bitstream_put_ui(bs, 0, 1); /* ref_pic_list_reordering_flag_l1: 0 */
249 if ((pic_param->pic_fields.bits.weighted_pred_flag &&
250 IS_P_SLICE(slice_param->slice_type)) ||
251 ((pic_param->pic_fields.bits.weighted_bipred_idc == 1) &&
252 IS_B_SLICE(slice_param->slice_type))) {
253 /* FIXME: fill weight/offset table */
257 /* dec_ref_pic_marking */
258 if (pic_param->pic_fields.bits.reference_pic_flag) { /* nal_ref_idc != 0 */
259 unsigned char no_output_of_prior_pics_flag = 0;
260 unsigned char long_term_reference_flag = 0;
261 unsigned char adaptive_ref_pic_marking_mode_flag = 0;
263 if (pic_param->pic_fields.bits.idr_pic_flag) {
264 avc_bitstream_put_ui(bs, no_output_of_prior_pics_flag, 1); /* no_output_of_prior_pics_flag: 0 */
265 avc_bitstream_put_ui(bs, long_term_reference_flag, 1); /* long_term_reference_flag: 0 */
267 avc_bitstream_put_ui(bs, adaptive_ref_pic_marking_mode_flag, 1); /* adaptive_ref_pic_marking_mode_flag: 0 */
271 if (pic_param->pic_fields.bits.entropy_coding_mode_flag &&
272 !IS_I_SLICE(slice_param->slice_type))
273 avc_bitstream_put_ue(bs, slice_param->cabac_init_idc); /* cabac_init_idc: 0 */
275 avc_bitstream_put_se(bs, slice_param->slice_qp_delta); /* slice_qp_delta: 0 */
277 /* ignore for SP/SI */
279 if (pic_param->pic_fields.bits.deblocking_filter_control_present_flag) {
280 avc_bitstream_put_ue(bs, slice_param->disable_deblocking_filter_idc); /* disable_deblocking_filter_idc: 0 */
282 if (slice_param->disable_deblocking_filter_idc != 1) {
283 avc_bitstream_put_se(bs, slice_param->slice_alpha_c0_offset_div2); /* slice_alpha_c0_offset_div2: 2 */
284 avc_bitstream_put_se(bs, slice_param->slice_beta_offset_div2); /* slice_beta_offset_div2: 2 */
288 if (pic_param->pic_fields.bits.entropy_coding_mode_flag) {
289 avc_bitstream_byte_aligning(bs, 1);
294 build_avc_slice_header(VAEncSequenceParameterBufferH264 *sps_param,
295 VAEncPictureParameterBufferH264 *pic_param,
296 VAEncSliceParameterBufferH264 *slice_param,
297 unsigned char **slice_header_buffer)
300 int is_idr = !!pic_param->pic_fields.bits.idr_pic_flag;
302 avc_bitstream_start(&bs);
303 nal_start_code_prefix(&bs);
305 if (IS_I_SLICE(slice_param->slice_type)) {
306 nal_header(&bs, NAL_REF_IDC_HIGH, is_idr ? NAL_IDR : NAL_NON_IDR);
307 } else if (IS_P_SLICE(slice_param->slice_type)) {
308 nal_header(&bs, NAL_REF_IDC_MEDIUM, is_idr ? NAL_IDR : NAL_NON_IDR);
310 assert(IS_B_SLICE(slice_param->slice_type));
311 nal_header(&bs, NAL_REF_IDC_NONE, is_idr ? NAL_IDR : NAL_NON_IDR);
314 slice_header(&bs, sps_param, pic_param, slice_param);
316 avc_bitstream_end(&bs);
317 *slice_header_buffer = (unsigned char *)bs.buffer;
319 return bs.bit_offset;
323 build_avc_sei_buffering_period(int cpb_removal_length,
324 unsigned int init_cpb_removal_delay,
325 unsigned int init_cpb_removal_delay_offset,
326 unsigned char **sei_buffer)
328 unsigned char *byte_buf;
331 avc_bitstream nal_bs;
332 avc_bitstream sei_bs;
334 avc_bitstream_start(&sei_bs);
335 avc_bitstream_put_ue(&sei_bs, 0); /*seq_parameter_set_id*/
336 avc_bitstream_put_ui(&sei_bs, init_cpb_removal_delay, cpb_removal_length);
337 avc_bitstream_put_ui(&sei_bs, init_cpb_removal_delay_offset, cpb_removal_length);
338 if ( sei_bs.bit_offset & 0x7) {
339 avc_bitstream_put_ui(&sei_bs, 1, 1);
341 avc_bitstream_end(&sei_bs);
342 byte_size = (sei_bs.bit_offset + 7) / 8;
344 avc_bitstream_start(&nal_bs);
345 nal_start_code_prefix(&nal_bs);
346 nal_header(&nal_bs, NAL_REF_IDC_NONE, NAL_SEI);
348 avc_bitstream_put_ui(&nal_bs, 0, 8);
349 avc_bitstream_put_ui(&nal_bs, byte_size, 8);
351 byte_buf = (unsigned char *)sei_bs.buffer;
352 for(i = 0; i < byte_size; i++) {
353 avc_bitstream_put_ui(&nal_bs, byte_buf[i], 8);
357 avc_rbsp_trailing_bits(&nal_bs);
358 avc_bitstream_end(&nal_bs);
360 *sei_buffer = (unsigned char *)nal_bs.buffer;
362 return nal_bs.bit_offset;
366 build_avc_sei_pic_timing(unsigned int cpb_removal_length, unsigned int cpb_removal_delay,
367 unsigned int dpb_output_length, unsigned int dpb_output_delay,
368 unsigned char **sei_buffer)
370 unsigned char *byte_buf;
373 avc_bitstream nal_bs;
374 avc_bitstream sei_bs;
376 avc_bitstream_start(&sei_bs);
377 avc_bitstream_put_ui(&sei_bs, cpb_removal_delay, cpb_removal_length);
378 avc_bitstream_put_ui(&sei_bs, dpb_output_delay, dpb_output_length);
379 if ( sei_bs.bit_offset & 0x7) {
380 avc_bitstream_put_ui(&sei_bs, 1, 1);
382 avc_bitstream_end(&sei_bs);
383 byte_size = (sei_bs.bit_offset + 7) / 8;
385 avc_bitstream_start(&nal_bs);
386 nal_start_code_prefix(&nal_bs);
387 nal_header(&nal_bs, NAL_REF_IDC_NONE, NAL_SEI);
389 avc_bitstream_put_ui(&nal_bs, 0x01, 8);
390 avc_bitstream_put_ui(&nal_bs, byte_size, 8);
392 byte_buf = (unsigned char *)sei_bs.buffer;
393 for(i = 0; i < byte_size; i++) {
394 avc_bitstream_put_ui(&nal_bs, byte_buf[i], 8);
398 avc_rbsp_trailing_bits(&nal_bs);
399 avc_bitstream_end(&nal_bs);
401 *sei_buffer = (unsigned char *)nal_bs.buffer;
403 return nal_bs.bit_offset;
408 build_avc_sei_buffer_timing(unsigned int init_cpb_removal_length,
409 unsigned int init_cpb_removal_delay,
410 unsigned int init_cpb_removal_delay_offset,
411 unsigned int cpb_removal_length,
412 unsigned int cpb_removal_delay,
413 unsigned int dpb_output_length,
414 unsigned int dpb_output_delay,
415 unsigned char **sei_buffer)
417 unsigned char *byte_buf;
418 int bp_byte_size, i, pic_byte_size;
420 avc_bitstream nal_bs;
421 avc_bitstream sei_bp_bs, sei_pic_bs;
423 avc_bitstream_start(&sei_bp_bs);
424 avc_bitstream_put_ue(&sei_bp_bs, 0); /*seq_parameter_set_id*/
425 avc_bitstream_put_ui(&sei_bp_bs, init_cpb_removal_delay, cpb_removal_length);
426 avc_bitstream_put_ui(&sei_bp_bs, init_cpb_removal_delay_offset, cpb_removal_length);
427 if ( sei_bp_bs.bit_offset & 0x7) {
428 avc_bitstream_put_ui(&sei_bp_bs, 1, 1);
430 avc_bitstream_end(&sei_bp_bs);
431 bp_byte_size = (sei_bp_bs.bit_offset + 7) / 8;
433 avc_bitstream_start(&sei_pic_bs);
434 avc_bitstream_put_ui(&sei_pic_bs, cpb_removal_delay, cpb_removal_length);
435 avc_bitstream_put_ui(&sei_pic_bs, dpb_output_delay, dpb_output_length);
436 if ( sei_pic_bs.bit_offset & 0x7) {
437 avc_bitstream_put_ui(&sei_pic_bs, 1, 1);
439 avc_bitstream_end(&sei_pic_bs);
440 pic_byte_size = (sei_pic_bs.bit_offset + 7) / 8;
442 avc_bitstream_start(&nal_bs);
443 nal_start_code_prefix(&nal_bs);
444 nal_header(&nal_bs, NAL_REF_IDC_NONE, NAL_SEI);
446 /* Write the SEI buffer period data */
447 avc_bitstream_put_ui(&nal_bs, 0, 8);
448 avc_bitstream_put_ui(&nal_bs, bp_byte_size, 8);
450 byte_buf = (unsigned char *)sei_bp_bs.buffer;
451 for(i = 0; i < bp_byte_size; i++) {
452 avc_bitstream_put_ui(&nal_bs, byte_buf[i], 8);
455 /* write the SEI timing data */
456 avc_bitstream_put_ui(&nal_bs, 0x01, 8);
457 avc_bitstream_put_ui(&nal_bs, pic_byte_size, 8);
459 byte_buf = (unsigned char *)sei_pic_bs.buffer;
460 for(i = 0; i < pic_byte_size; i++) {
461 avc_bitstream_put_ui(&nal_bs, byte_buf[i], 8);
465 avc_rbsp_trailing_bits(&nal_bs);
466 avc_bitstream_end(&nal_bs);
468 *sei_buffer = (unsigned char *)nal_bs.buffer;
470 return nal_bs.bit_offset;
474 build_mpeg2_slice_header(VAEncSequenceParameterBufferMPEG2 *sps_param,
475 VAEncPictureParameterBufferMPEG2 *pic_param,
476 VAEncSliceParameterBufferMPEG2 *slice_param,
477 unsigned char **slice_header_buffer)
481 avc_bitstream_start(&bs);
482 avc_bitstream_end(&bs);
483 *slice_header_buffer = (unsigned char *)bs.buffer;
485 return bs.bit_offset;