apps/ffmpeg-rockchip
1
0
Fork 0
mirror of https://github.com/nyanmisaka/ffmpeg-rockchip.git synced 2025-10-31 20:42:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

avcodec/av1dec: honor the requested skip_frame level

Browse Source 
This supports dropping non-intra, non-key, or all frames.

Tested-by: nevcairiel
Signed-off-by: James Almer <jamrial@gmail.com>
This commit is contained in:
James Almer
2022-01-07 16:11:12 -03:00
parent 3c2b674468
commit 4b053b8db1
1 changed files with 40 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
libavcodec/av1dec.c
View File

@@ -585,16 +585,19 @@ static int av1_frame_ref(AVCodecContext *avctx, AV1Frame *dst, const AV1Frame *s
{ {
int ret; int ret;
ret = ff_thread_ref_frame(&dst->tf, &src->tf); ret = av_buffer_replace(&dst->header_ref, src->header_ref);
if (ret < 0) if (ret < 0)
return ret; return ret;
dst->header_ref = av_buffer_ref(src->header_ref);
if (!dst->header_ref)
goto fail;
dst->raw_frame_header = src->raw_frame_header; dst->raw_frame_header = src->raw_frame_header;
if (!src->tf.f->buf[0])
return 0;
ret = ff_thread_ref_frame(&dst->tf, &src->tf);
if (ret < 0)
goto fail;
if (src->hwaccel_picture_private) { if (src->hwaccel_picture_private) {
dst->hwaccel_priv_buf = av_buffer_ref(src->hwaccel_priv_buf); dst->hwaccel_priv_buf = av_buffer_ref(src->hwaccel_priv_buf);
if (!dst->hwaccel_priv_buf) if (!dst->hwaccel_priv_buf)
@@ -800,12 +803,6 @@ static int av1_frame_alloc(AVCodecContext *avctx, AV1Frame *f)
return ret; return ret;
} }
f->header_ref = av_buffer_ref(s->header_ref);
if (!f->header_ref)
return AVERROR(ENOMEM);
f->raw_frame_header = s->raw_frame_header;
if ((ret = ff_thread_get_buffer(avctx, &f->tf, AV_GET_BUFFER_FLAG_REF)) < 0) if ((ret = ff_thread_get_buffer(avctx, &f->tf, AV_GET_BUFFER_FLAG_REF)) < 0)
goto fail; goto fail;
@@ -945,8 +942,7 @@ static int update_reference_list(AVCodecContext *avctx)
for (int i = 0; i < AV1_NUM_REF_FRAMES; i++) { for (int i = 0; i < AV1_NUM_REF_FRAMES; i++) {
if (header->refresh_frame_flags & (1 << i)) { if (header->refresh_frame_flags & (1 << i)) {
if (s->ref[i].tf.f->buf[0]) av1_frame_unref(avctx, &s->ref[i]);
av1_frame_unref(avctx, &s->ref[i]);
if ((ret = av1_frame_ref(avctx, &s->ref[i], &s->cur_frame)) < 0) { if ((ret = av1_frame_ref(avctx, &s->ref[i], &s->cur_frame)) < 0) {
av_log(avctx, AV_LOG_ERROR, av_log(avctx, AV_LOG_ERROR,
"Failed to update frame %d in reference list\n", i); "Failed to update frame %d in reference list\n", i);
@@ -962,8 +958,27 @@ static int get_current_frame(AVCodecContext *avctx)
AV1DecContext *s = avctx->priv_data; AV1DecContext *s = avctx->priv_data;
int ret; int ret;
if (s->cur_frame.tf.f->buf[0]) av1_frame_unref(avctx, &s->cur_frame);
av1_frame_unref(avctx, &s->cur_frame);
s->cur_frame.header_ref = av_buffer_ref(s->header_ref);
if (!s->cur_frame.header_ref)
return AVERROR(ENOMEM);
s->cur_frame.raw_frame_header = s->raw_frame_header;
ret = init_tile_data(s);
if (ret < 0) {
av_log(avctx, AV_LOG_ERROR, "Failed to init tile data.\n");
return ret;
}
if ((avctx->skip_frame >= AVDISCARD_NONINTRA &&
(s->raw_frame_header->frame_type != AV1_FRAME_KEY &&
s->raw_frame_header->frame_type != AV1_FRAME_INTRA_ONLY)) ||
(avctx->skip_frame >= AVDISCARD_NONKEY &&
s->raw_frame_header->frame_type != AV1_FRAME_KEY) ||
avctx->skip_frame >= AVDISCARD_ALL)
return 0;
ret = av1_frame_alloc(avctx, &s->cur_frame); ret = av1_frame_alloc(avctx, &s->cur_frame);
if (ret < 0) { if (ret < 0) {
@@ -972,12 +987,6 @@ static int get_current_frame(AVCodecContext *avctx)
return ret; return ret;
} }
ret = init_tile_data(s);
if (ret < 0) {
av_log(avctx, AV_LOG_ERROR, "Failed to init tile data.\n");
return ret;
}
global_motion_params(s); global_motion_params(s);
skip_mode_params(s); skip_mode_params(s);
coded_lossless_param(s); coded_lossless_param(s);
@@ -1077,8 +1086,7 @@ static int av1_decode_frame(AVCodecContext *avctx, void *frame,
s->raw_frame_header = &obu->obu.frame_header; s->raw_frame_header = &obu->obu.frame_header;
if (s->raw_frame_header->show_existing_frame) { if (s->raw_frame_header->show_existing_frame) {
if (s->cur_frame.tf.f->buf[0]) av1_frame_unref(avctx, &s->cur_frame);
av1_frame_unref(avctx, &s->cur_frame);
ret = av1_frame_ref(avctx, &s->cur_frame, ret = av1_frame_ref(avctx, &s->cur_frame,
&s->ref[s->raw_frame_header->frame_to_show_map_idx]); &s->ref[s->raw_frame_header->frame_to_show_map_idx]);
@@ -1093,9 +1101,11 @@ static int av1_decode_frame(AVCodecContext *avctx, void *frame,
goto end; goto end;
} }
ret = set_output_frame(avctx, frame, pkt, got_frame); if (s->cur_frame.tf.f->buf[0]) {
if (ret < 0) ret = set_output_frame(avctx, frame, pkt, got_frame);
av_log(avctx, AV_LOG_ERROR, "Set output frame error.\n"); if (ret < 0)
av_log(avctx, AV_LOG_ERROR, "Set output frame error.\n");
}
s->raw_frame_header = NULL; s->raw_frame_header = NULL;
@@ -1111,7 +1121,7 @@ static int av1_decode_frame(AVCodecContext *avctx, void *frame,
s->cur_frame.spatial_id = header->spatial_id; s->cur_frame.spatial_id = header->spatial_id;
s->cur_frame.temporal_id = header->temporal_id; s->cur_frame.temporal_id = header->temporal_id;
if (avctx->hwaccel) { if (avctx->hwaccel && s->cur_frame.tf.f->buf[0]) {
ret = avctx->hwaccel->start_frame(avctx, unit->data, ret = avctx->hwaccel->start_frame(avctx, unit->data,
unit->data_size); unit->data_size);
if (ret < 0) { if (ret < 0) {
@@ -1138,7 +1148,7 @@ static int av1_decode_frame(AVCodecContext *avctx, void *frame,
if (ret < 0) if (ret < 0)
goto end; goto end;
if (avctx->hwaccel) { if (avctx->hwaccel && s->cur_frame.tf.f->buf[0]) {
ret = avctx->hwaccel->decode_slice(avctx, ret = avctx->hwaccel->decode_slice(avctx,
raw_tile_group->tile_data.data, raw_tile_group->tile_data.data,
raw_tile_group->tile_data.data_size); raw_tile_group->tile_data.data_size);
@@ -1161,7 +1171,7 @@ static int av1_decode_frame(AVCodecContext *avctx, void *frame,
} }
if (raw_tile_group && (s->tile_num == raw_tile_group->tg_end + 1)) { if (raw_tile_group && (s->tile_num == raw_tile_group->tg_end + 1)) {
if (avctx->hwaccel) { if (avctx->hwaccel && s->cur_frame.tf.f->buf[0]) {
ret = avctx->hwaccel->end_frame(avctx); ret = avctx->hwaccel->end_frame(avctx);
if (ret < 0) { if (ret < 0) {
av_log(avctx, AV_LOG_ERROR, "HW accel end frame fail.\n"); av_log(avctx, AV_LOG_ERROR, "HW accel end frame fail.\n");
@@ -1175,7 +1185,7 @@ static int av1_decode_frame(AVCodecContext *avctx, void *frame,
goto end; goto end;
} }
if (s->raw_frame_header->show_frame) { if (s->raw_frame_header->show_frame && s->cur_frame.tf.f->buf[0]) {
ret = set_output_frame(avctx, frame, pkt, got_frame); ret = set_output_frame(avctx, frame, pkt, got_frame);
if (ret < 0) { if (ret < 0) {
av_log(avctx, AV_LOG_ERROR, "Set output frame error\n"); av_log(avctx, AV_LOG_ERROR, "Set output frame error\n");