31 return VA_INVALID_SURFACE;
46 "Couldn't get bit depth from profile:%d.\n", seq->
seq_profile);
49 return bit_depth == 8 ? 0 : bit_depth == 10 ? 1 : 2;
61 VADecPictureParameterBufferAV1 pic_param;
70 if (bit_depth_idx < 0)
73 memset(&pic_param, 0,
sizeof(VADecPictureParameterBufferAV1));
74 pic_param = (VADecPictureParameterBufferAV1) {
75 .profile = seq->seq_profile,
76 .order_hint_bits_minus_1 = seq->order_hint_bits_minus_1,
77 .bit_depth_idx = bit_depth_idx,
78 .current_frame = pic->output_surface,
79 .current_display_picture = pic->output_surface,
81 .frame_height_minus1 =
frame_header->frame_height_minus_1,
86 .context_update_tile_id =
frame_header->context_update_tile_id,
93 .cdef_damping_minus_3 =
frame_header->cdef_damping_minus_3,
95 .seq_info_fields.fields = {
96 .still_picture = seq->still_picture,
97 .use_128x128_superblock = seq->use_128x128_superblock,
98 .enable_filter_intra = seq->enable_filter_intra,
99 .enable_intra_edge_filter = seq->enable_intra_edge_filter,
100 .enable_interintra_compound = seq->enable_interintra_compound,
101 .enable_masked_compound = seq->enable_masked_compound,
102 .enable_dual_filter = seq->enable_dual_filter,
103 .enable_order_hint = seq->enable_order_hint,
104 .enable_jnt_comp = seq->enable_jnt_comp,
105 .enable_cdef = seq->enable_cdef,
106 .mono_chrome = seq->color_config.mono_chrome,
107 .color_range = seq->color_config.color_range,
108 .subsampling_x = seq->color_config.subsampling_x,
109 .subsampling_y = seq->color_config.subsampling_y,
110 .chroma_sample_position = seq->color_config.chroma_sample_position,
111 .film_grain_params_present = seq->film_grain_params_present &&
114 .seg_info.segment_info_fields.bits = {
117 .temporal_update =
frame_header->segmentation_temporal_update,
121 .film_grain_info_fields.bits = {
122 .apply_grain = apply_grain,
123 .chroma_scaling_from_luma = film_grain->chroma_scaling_from_luma,
124 .grain_scaling_minus_8 = film_grain->grain_scaling_minus_8,
125 .ar_coeff_lag = film_grain->ar_coeff_lag,
126 .ar_coeff_shift_minus_6 = film_grain->ar_coeff_shift_minus_6,
127 .grain_scale_shift = film_grain->grain_scale_shift,
128 .overlap_flag = film_grain->overlap_flag,
129 .clip_to_restricted_range = film_grain->clip_to_restricted_range,
131 .grain_seed = film_grain->grain_seed,
132 .num_y_points = film_grain->num_y_points,
133 .num_cb_points = film_grain->num_cb_points,
134 .num_cr_points = film_grain->num_cr_points,
135 .cb_mult = film_grain->cb_mult,
136 .cb_luma_mult = film_grain->cb_luma_mult,
137 .cb_offset = film_grain->cb_offset,
138 .cr_mult = film_grain->cr_mult,
139 .cr_luma_mult = film_grain->cr_luma_mult,
140 .cr_offset = film_grain->cr_offset,
142 .pic_info_fields.bits = {
146 .error_resilient_mode =
frame_header->error_resilient_mode,
148 .allow_screen_content_tools =
frame_header->allow_screen_content_tools,
152 .allow_high_precision_mv =
frame_header->allow_high_precision_mv,
153 .is_motion_mode_switchable =
frame_header->is_motion_mode_switchable,
155 .disable_frame_end_update_cdf =
frame_header->disable_frame_end_update_cdf,
156 .uniform_tile_spacing_flag =
frame_header->uniform_tile_spacing_flag,
157 .allow_warped_motion =
frame_header->allow_warped_motion,
159 .loop_filter_info_fields.bits = {
161 .mode_ref_delta_enabled =
frame_header->loop_filter_delta_enabled,
162 .mode_ref_delta_update =
frame_header->loop_filter_delta_update,
164 .mode_control_fields.bits = {
172 .loop_restoration_fields.bits = {
173 .yframe_restoration_type = remap_lr_type[
frame_header->lr_type[0]],
174 .cbframe_restoration_type = remap_lr_type[
frame_header->lr_type[1]],
175 .crframe_restoration_type = remap_lr_type[
frame_header->lr_type[2]],
179 .qmatrix_fields.bits = {
185 if (pic_param.pic_info_fields.bits.frame_type ==
AV1_FRAME_KEY)
186 pic_param.ref_frame_map[
i] = VA_INVALID_ID;
194 pic_param.ref_deltas[
i] =
frame_header->loop_filter_ref_deltas[
i];
196 for (
int i = 0;
i < 2;
i++) {
197 pic_param.mode_deltas[
i] =
frame_header->loop_filter_mode_deltas[
i];
200 pic_param.cdef_y_strengths[
i] =
203 pic_param.cdef_uv_strengths[
i] =
208 pic_param.width_in_sbs_minus_1[
i] =
212 pic_param.height_in_sbs_minus_1[
i] =
216 pic_param.wm[
i - 1].wmtype =
s->cur_frame.gm_type[
i];
217 for (
int j = 0; j < 6; j++)
218 pic_param.wm[
i - 1].wmmat[j] =
s->cur_frame.gm_params[
i][j];
221 for (
int i = 0;
i < film_grain->num_y_points;
i++) {
222 pic_param.film_grain_info.point_y_value[
i] =
223 film_grain->point_y_value[
i];
224 pic_param.film_grain_info.point_y_scaling[
i] =
225 film_grain->point_y_scaling[
i];
227 for (
int i = 0;
i < film_grain->num_cb_points;
i++) {
228 pic_param.film_grain_info.point_cb_value[
i] =
229 film_grain->point_cb_value[
i];
230 pic_param.film_grain_info.point_cb_scaling[
i] =
231 film_grain->point_cb_scaling[
i];
233 for (
int i = 0;
i < film_grain->num_cr_points;
i++) {
234 pic_param.film_grain_info.point_cr_value[
i] =
235 film_grain->point_cr_value[
i];
236 pic_param.film_grain_info.point_cr_scaling[
i] =
237 film_grain->point_cr_scaling[
i];
239 for (
int i = 0;
i < 24;
i++) {
240 pic_param.film_grain_info.ar_coeffs_y[
i] =
241 film_grain->ar_coeffs_y_plus_128[
i] - 128;
243 for (
int i = 0;
i < 25;
i++) {
244 pic_param.film_grain_info.ar_coeffs_cb[
i] =
245 film_grain->ar_coeffs_cb_plus_128[
i] - 128;
246 pic_param.film_grain_info.ar_coeffs_cr[
i] =
247 film_grain->ar_coeffs_cr_plus_128[
i] - 128;
251 VAPictureParameterBufferType,
252 &pic_param,
sizeof(pic_param));
276 VASliceParameterBufferAV1 slice_param;
279 for (
int i = s->
tg_start; i <= s->tg_end;
i++) {
280 memset(&slice_param, 0,
sizeof(VASliceParameterBufferAV1));
282 slice_param = (VASliceParameterBufferAV1) {
285 .slice_data_flag = VA_SLICE_DATA_FLAG_ALL,
293 sizeof(VASliceParameterBufferAV1),
AV1RawFrameHeader * raw_frame_header
const AVHWAccel ff_av1_vaapi_hwaccel
static av_cold int init(AVCodecContext *avctx)
int ff_vaapi_decode_make_param_buffer(AVCodecContext *avctx, VAAPIDecodePicture *pic, int type, const void *data, size_t size)
static VASurfaceID ff_vaapi_get_surface_id(AVFrame *pic)
static int vaapi_av1_start_frame(AVCodecContext *avctx, av_unused const uint8_t *buffer, av_unused uint32_t size)
static int8_t vaapi_av1_get_bit_depth_idx(AVCodecContext *avctx)
int ff_vaapi_common_frame_params(AVCodecContext *avctx, AVBufferRef *hw_frames_ctx)
static int vaapi_av1_end_frame(AVCodecContext *avctx)
TileGroupInfo * tile_group_info
int ff_vaapi_decode_uninit(AVCodecContext *avctx)
int export_side_data
Bit set of AV_CODEC_EXPORT_DATA_* flags, which affects the kind of metadata exported in frame...
static av_cold int uninit(AVCodecContext *avctx)
int ff_vaapi_decode_issue(AVCodecContext *avctx, VAAPIDecodePicture *pic)
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
static VASurfaceID vaapi_av1_surface_id(AV1Frame *vf)
#define HWACCEL_CAP_ASYNC_SAFE
const char * name
Name of the hardware accelerated codec.
int ff_vaapi_decode_init(AVCodecContext *avctx)
static void bit_depth(AudioStatsContext *s, uint64_t mask, uint64_t imask, AVRational *depth)
AV1RawFilmGrainParams film_grain
#define AV_CODEC_EXPORT_DATA_FILM_GRAIN
Decoding only.
main external API structure.
static int vaapi_av1_decode_slice(AVCodecContext *avctx, const uint8_t *buffer, uint32_t size)
int ff_vaapi_decode_cancel(AVCodecContext *avctx, VAAPIDecodePicture *pic)
AV1RawSequenceHeader * raw_seq
void * hwaccel_picture_private
int ff_vaapi_decode_make_slice_buffer(AVCodecContext *avctx, VAAPIDecodePicture *pic, const void *params_data, size_t params_size, const void *slice_data, size_t slice_size)