FFmpeg coverage

Line	Branch	Exec	Source
1			/*
2			* This file is part of FFmpeg.
3			*
4			* FFmpeg is free software; you can redistribute it and/or
5			* modify it under the terms of the GNU Lesser General Public
6			* License as published by the Free Software Foundation; either
7			* version 2.1 of the License, or (at your option) any later version.
8			*
9			* FFmpeg is distributed in the hope that it will be useful,
10			* but WITHOUT ANY WARRANTY; without even the implied warranty of
11			* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12			* Lesser General Public License for more details.
13			*
14			* You should have received a copy of the GNU Lesser General Public
15			* License along with FFmpeg; if not, write to the Free Software
16			* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17			*/
18
19			#include "hw_base_encode_h264.h"
20
21			#include "h2645data.h"
22			#include "h264_levels.h"
23
24			#include "libavutil/pixdesc.h"
25
26		✗	int ff_hw_base_encode_init_params_h264(FFHWBaseEncodeContext *base_ctx,
27			AVCodecContext *avctx,
28			FFHWBaseEncodeH264 *common,
29			FFHWBaseEncodeH264Opts *opts)
30			{
31		✗	H264RawSPS *sps = &common->raw_sps;
32		✗	H264RawPPS *pps = &common->raw_pps;
33			const AVPixFmtDescriptor *desc;
34			int bit_depth;
35
36		✗	memset(sps, 0, sizeof(*sps));
37		✗	memset(pps, 0, sizeof(*pps));
38
39		✗	desc = av_pix_fmt_desc_get(base_ctx->input_frames->sw_format);
40		✗	av_assert0(desc);
41		✗	if (desc->nb_components == 1 || desc->log2_chroma_w != 1 || desc->log2_chroma_h != 1) {
42		✗	av_log(avctx, AV_LOG_ERROR, "Chroma format of input pixel format "
43		✗	"%s is not supported.\n", desc->name);
44		✗	return AVERROR(EINVAL);
45			}
46		✗	bit_depth = desc->comp[0].depth;
47
48		✗	sps->nal_unit_header.nal_ref_idc = 3;
49		✗	sps->nal_unit_header.nal_unit_type = H264_NAL_SPS;
50
51		✗	sps->profile_idc = avctx->profile & 0xff;
52
53		✗	if (avctx->profile == AV_PROFILE_H264_CONSTRAINED_BASELINE ||
54		✗	avctx->profile == AV_PROFILE_H264_MAIN)
55		✗	sps->constraint_set1_flag = 1;
56
57		✗	if (avctx->profile == AV_PROFILE_H264_HIGH || avctx->profile == AV_PROFILE_H264_HIGH_10)
58		✗	sps->constraint_set3_flag = base_ctx->gop_size == 1;
59
60		✗	if (avctx->profile == AV_PROFILE_H264_MAIN ||
61		✗	avctx->profile == AV_PROFILE_H264_HIGH || avctx->profile == AV_PROFILE_H264_HIGH_10) {
62		✗	sps->constraint_set4_flag = 1;
63		✗	sps->constraint_set5_flag = base_ctx->b_per_p == 0;
64			}
65
66		✗	if (base_ctx->gop_size == 1)
67		✗	common->dpb_frames = 0;
68			else
69		✗	common->dpb_frames = 1 + base_ctx->max_b_depth;
70
71		✗	if (avctx->level != AV_LEVEL_UNKNOWN) {
72		✗	sps->level_idc = avctx->level;
73			} else {
74			const H264LevelDescriptor *level;
75			int framerate;
76
77		✗	if (avctx->framerate.num > 0 && avctx->framerate.den > 0)
78		✗	framerate = avctx->framerate.num / avctx->framerate.den;
79			else
80		✗	framerate = 0;
81
82		✗	level = ff_h264_guess_level(sps->profile_idc,
83			opts->bit_rate,
84			framerate,
85		✗	opts->mb_width * 16,
86		✗	opts->mb_height * 16,
87			common->dpb_frames);
88		✗	if (level) {
89		✗	av_log(avctx, AV_LOG_VERBOSE, "Using level %s.\n", level->name);
90		✗	if (level->constraint_set3_flag)
91		✗	sps->constraint_set3_flag = 1;
92		✗	sps->level_idc = level->level_idc;
93			} else {
94		✗	av_log(avctx, AV_LOG_WARNING, "Stream will not conform "
95			"to any level: using level 6.2.\n");
96		✗	sps->level_idc = 62;
97			}
98			}
99
100		✗	sps->seq_parameter_set_id = 0;
101		✗	sps->chroma_format_idc = 1;
102		✗	sps->bit_depth_luma_minus8 = bit_depth - 8;
103		✗	sps->bit_depth_chroma_minus8 = bit_depth - 8;
104
105		✗	sps->log2_max_frame_num_minus4 = 4;
106		✗	sps->pic_order_cnt_type = base_ctx->max_b_depth ? 0 : 2;
107		✗	if (sps->pic_order_cnt_type == 0) {
108		✗	sps->log2_max_pic_order_cnt_lsb_minus4 = 4;
109			}
110
111		✗	sps->max_num_ref_frames = common->dpb_frames;
112
113		✗	sps->pic_width_in_mbs_minus1 = opts->mb_width - 1;
114		✗	sps->pic_height_in_map_units_minus1 = opts->mb_height - 1;
115
116		✗	sps->frame_mbs_only_flag = 1;
117		✗	sps->direct_8x8_inference_flag = 1;
118
119		✗	if (avctx->width != 16 * opts->mb_width ||
120		✗	avctx->height != 16 * opts->mb_height) {
121		✗	sps->frame_cropping_flag = 1;
122
123		✗	sps->frame_crop_left_offset = 0;
124		✗	sps->frame_crop_right_offset =
125		✗	(16 * opts->mb_width - avctx->width) / 2;
126		✗	sps->frame_crop_top_offset = 0;
127		✗	sps->frame_crop_bottom_offset =
128		✗	(16 * opts->mb_height - avctx->height) / 2;
129			} else {
130		✗	sps->frame_cropping_flag = 0;
131			}
132
133		✗	sps->vui_parameters_present_flag = 1;
134
135		✗	if (avctx->sample_aspect_ratio.num != 0 &&
136		✗	avctx->sample_aspect_ratio.den != 0) {
137			int num, den, i;
138		✗	av_reduce(&num, &den, avctx->sample_aspect_ratio.num,
139		✗	avctx->sample_aspect_ratio.den, 65535);
140		✗	for (i = 0; i < FF_ARRAY_ELEMS(ff_h2645_pixel_aspect); i++) {
141		✗	if (num == ff_h2645_pixel_aspect[i].num &&
142		✗	den == ff_h2645_pixel_aspect[i].den) {
143		✗	sps->vui.aspect_ratio_idc = i;
144		✗	break;
145			}
146			}
147		✗	if (i >= FF_ARRAY_ELEMS(ff_h2645_pixel_aspect)) {
148		✗	sps->vui.aspect_ratio_idc = 255;
149		✗	sps->vui.sar_width = num;
150		✗	sps->vui.sar_height = den;
151			}
152		✗	sps->vui.aspect_ratio_info_present_flag = 1;
153			}
154
155			// Unspecified video format, from table E-2.
156		✗	sps->vui.video_format = 5;
157		✗	sps->vui.video_full_range_flag =
158		✗	avctx->color_range == AVCOL_RANGE_JPEG;
159		✗	sps->vui.colour_primaries = avctx->color_primaries;
160		✗	sps->vui.transfer_characteristics = avctx->color_trc;
161		✗	sps->vui.matrix_coefficients = avctx->colorspace;
162		✗	if (avctx->color_primaries != AVCOL_PRI_UNSPECIFIED ||
163		✗	avctx->color_trc != AVCOL_TRC_UNSPECIFIED ||
164		✗	avctx->colorspace != AVCOL_SPC_UNSPECIFIED)
165		✗	sps->vui.colour_description_present_flag = 1;
166		✗	if (avctx->color_range != AVCOL_RANGE_UNSPECIFIED ||
167		✗	sps->vui.colour_description_present_flag)
168		✗	sps->vui.video_signal_type_present_flag = 1;
169
170		✗	if (avctx->chroma_sample_location != AVCHROMA_LOC_UNSPECIFIED) {
171		✗	sps->vui.chroma_loc_info_present_flag = 1;
172		✗	sps->vui.chroma_sample_loc_type_top_field =
173		✗	sps->vui.chroma_sample_loc_type_bottom_field =
174		✗	avctx->chroma_sample_location - 1;
175			}
176
177		✗	sps->vui.timing_info_present_flag = 1;
178		✗	if (avctx->framerate.num > 0 && avctx->framerate.den > 0) {
179		✗	sps->vui.num_units_in_tick = avctx->framerate.den;
180		✗	sps->vui.time_scale = 2 * avctx->framerate.num;
181		✗	sps->vui.fixed_frame_rate_flag = 1;
182			} else {
183		✗	sps->vui.num_units_in_tick = avctx->time_base.num;
184		✗	sps->vui.time_scale = 2 * avctx->time_base.den;
185		✗	sps->vui.fixed_frame_rate_flag = 0;
186			}
187
188		✗	if (opts->flags & FF_HW_H264_SEI_TIMING) {
189		✗	H264RawHRD *hrd = &sps->vui.nal_hrd_parameters;
190		✗	H264RawSEIBufferingPeriod *bp = &common->sei_buffering_period;
191
192		✗	sps->vui.nal_hrd_parameters_present_flag = 1;
193
194		✗	hrd->cpb_cnt_minus1 = 0;
195
196			// Try to scale these to a sensible range so that the
197			// golomb encode of the value is not overlong.
198		✗	hrd->bit_rate_scale =
199		✗	av_clip_uintp2(av_log2(opts->bit_rate) - 15 - 6, 4);
200		✗	hrd->bit_rate_value_minus1[0] =
201		✗	(opts->bit_rate >> hrd->bit_rate_scale + 6) - 1;
202
203		✗	hrd->cpb_size_scale =
204		✗	av_clip_uintp2(av_log2(opts->hrd_buffer_size) - 15 - 4, 4);
205		✗	hrd->cpb_size_value_minus1[0] =
206		✗	(opts->hrd_buffer_size >> hrd->cpb_size_scale + 4) - 1;
207
208			// CBR mode as defined for the HRD cannot be achieved without filler
209			// data, so this flag cannot be set even with VAAPI CBR modes.
210		✗	hrd->cbr_flag[0] = 0;
211
212		✗	hrd->initial_cpb_removal_delay_length_minus1 = 23;
213		✗	hrd->cpb_removal_delay_length_minus1 = 23;
214		✗	hrd->dpb_output_delay_length_minus1 = 7;
215		✗	hrd->time_offset_length = 0;
216
217		✗	bp->seq_parameter_set_id = sps->seq_parameter_set_id;
218
219			// This calculation can easily overflow 32 bits.
220		✗	bp->nal.initial_cpb_removal_delay[0] = 90000 *
221		✗	(uint64_t)opts->initial_buffer_fullness /
222		✗	opts->hrd_buffer_size;
223		✗	bp->nal.initial_cpb_removal_delay_offset[0] = 0;
224			} else {
225		✗	sps->vui.nal_hrd_parameters_present_flag = 0;
226		✗	sps->vui.low_delay_hrd_flag = 1 - sps->vui.fixed_frame_rate_flag;
227			}
228
229		✗	sps->vui.bitstream_restriction_flag = 1;
230		✗	sps->vui.motion_vectors_over_pic_boundaries_flag = 1;
231		✗	sps->vui.log2_max_mv_length_horizontal = 15;
232		✗	sps->vui.log2_max_mv_length_vertical = 15;
233		✗	sps->vui.max_num_reorder_frames = base_ctx->max_b_depth;
234		✗	sps->vui.max_dec_frame_buffering = base_ctx->max_b_depth + 1;
235
236		✗	pps->nal_unit_header.nal_ref_idc = 3;
237		✗	pps->nal_unit_header.nal_unit_type = H264_NAL_PPS;
238
239		✗	pps->pic_parameter_set_id = 0;
240		✗	pps->seq_parameter_set_id = 0;
241
242		✗	pps->entropy_coding_mode_flag =
243		✗	!(sps->profile_idc == AV_PROFILE_H264_BASELINE ||
244		✗	sps->profile_idc == AV_PROFILE_H264_EXTENDED ||
245		✗	sps->profile_idc == AV_PROFILE_H264_CAVLC_444);
246		✗	if (!opts->cabac && pps->entropy_coding_mode_flag)
247		✗	pps->entropy_coding_mode_flag = 0;
248
249		✗	pps->num_ref_idx_l0_default_active_minus1 = 0;
250		✗	pps->num_ref_idx_l1_default_active_minus1 = 0;
251
252		✗	pps->pic_init_qp_minus26 = opts->fixed_qp_idr - 26;
253
254		✗	if (sps->profile_idc == AV_PROFILE_H264_BASELINE ||
255		✗	sps->profile_idc == AV_PROFILE_H264_EXTENDED ||
256		✗	sps->profile_idc == AV_PROFILE_H264_MAIN) {
257		✗	pps->more_rbsp_data = 0;
258			} else {
259		✗	pps->more_rbsp_data = 1;
260
261		✗	pps->transform_8x8_mode_flag = 1;
262			}
263
264		✗	return 0;
265			}
266