mirrors/gstreamer
mirrors/gstreamer
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gstreamer.git synced 2024-06-18 05:50:41 +00:00
Code Issues 1 Projects Releases Wiki Activity

va: av1enc: Improve the LAST reference assignment

Browse source 
The last frame which has the smallest diff should be consider as
the first choice rather than the golden frame. Especially when only
one reference available, this way can improve the BD rate about 5
percentage.

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/6379>
This commit is contained in:
He Junyan 2024-03-14 23:17:32 +08:00 committed by GStreamer Marge Bot
parent 1745a82516
commit 40629b84ee
1 changed files with 34 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

50
subprojects/gst-plugins-bad/sys/va/gstvaav1enc.c
View file

@ -1419,6 +1419,7 @@ _av1_assign_ref_index (GstVaAV1Enc * self, GstVideoCodecFrame * frame)
gint forward_ref_num, backward_ref_num;; gint forward_ref_num, backward_ref_num;;
GstVaAV1EncFrame *va_frame = _enc_frame (frame); GstVaAV1EncFrame *va_frame = _enc_frame (frame);
gint i, num; gint i, num;
gboolean gf_assigned;
memset (va_frame->ref_frame_idx, -1, sizeof (va_frame->ref_frame_idx)); memset (va_frame->ref_frame_idx, -1, sizeof (va_frame->ref_frame_idx));
@ -1482,36 +1483,53 @@ _av1_assign_ref_index (GstVaAV1Enc * self, GstVideoCodecFrame * frame)
g_qsort_with_data (all_refs, ref_num, sizeof (GstVaAV1Ref), g_qsort_with_data (all_refs, ref_num, sizeof (GstVaAV1Ref),
_av1_sort_by_frame_num, NULL); _av1_sort_by_frame_num, NULL);
/* Setting the forward refs. GOLDEN is always set first. /* Assign the forward references in order of:
LAST is set to the nearest frame in the past if forward_ref_num * 1. The last frame which has the smallest diff.
is enough. LAST2 and LAST3 are set to next nearest frames in the * 2. The golden frame which can be a key frame with better quality.
past if forward_ref_num is enough. * 3. The other frames by inverse frame number order.
If forward_ref_num is not enough, they are just set to GOLDEN. */ */
va_frame->bidir_ref = FALSE; va_frame->bidir_ref = FALSE;
gf_assigned = FALSE;
num = forward_num - 1; num = forward_num - 1;
if (backward_num > 0) { g_assert (num >= 0);
forward_ref_num = self->gop.forward_ref_num - 1 /* already assign gf */ ; /* if forward only, should use forward_only_ref_num */
} else { forward_ref_num = backward_num > 0 ?
/* if forward only, should use forward_only_ref_num */ self->gop.forward_ref_num : self->gop.forward_only_ref_num;
forward_ref_num = g_assert (forward_ref_num > 0);
self->gop.forward_only_ref_num - 1 /* already assign gf */ ;
}
if (num >= 0 && all_refs[num].index_in_dpb == /* The golden frame happens to be the last frame. */
va_frame->ref_frame_idx[GST_AV1_REF_GOLDEN_FRAME]) if (all_refs[num].index_in_dpb ==
va_frame->ref_frame_idx[GST_AV1_REF_GOLDEN_FRAME]) {
num--; num--;
forward_ref_num--;
gf_assigned = TRUE;
}
if (num >= 0 && forward_ref_num > 0) { if (num >= 0 && forward_ref_num > 0) {
va_frame->ref_frame_idx[GST_AV1_REF_LAST_FRAME] = va_frame->ref_frame_idx[GST_AV1_REF_LAST_FRAME] =
all_refs[num].index_in_dpb; all_refs[num].index_in_dpb;
} else { } else {
/* At least one forward reference.
Just let the last frame be the same as the golden frame. */
g_assert (gf_assigned);
va_frame->ref_frame_idx[GST_AV1_REF_LAST_FRAME] = va_frame->ref_frame_idx[GST_AV1_REF_LAST_FRAME] =
va_frame->ref_frame_idx[GST_AV1_REF_GOLDEN_FRAME]; va_frame->ref_frame_idx[GST_AV1_REF_GOLDEN_FRAME];
} }
num--; num--;
forward_ref_num--; forward_ref_num--;
if (!gf_assigned) {
if (forward_ref_num <= 0) {
va_frame->ref_frame_idx[GST_AV1_REF_GOLDEN_FRAME] =
va_frame->ref_frame_idx[GST_AV1_REF_LAST_FRAME];
} else {
/* The golden frame index is already found. */
forward_ref_num--;
}
}
if (num >= 0 && all_refs[num].index_in_dpb == if (num >= 0 && all_refs[num].index_in_dpb ==
va_frame->ref_frame_idx[GST_AV1_REF_GOLDEN_FRAME]) va_frame->ref_frame_idx[GST_AV1_REF_GOLDEN_FRAME])
num--; num--;
@ -1521,7 +1539,7 @@ _av1_assign_ref_index (GstVaAV1Enc * self, GstVideoCodecFrame * frame)
all_refs[num].index_in_dpb; all_refs[num].index_in_dpb;
} else { } else {
va_frame->ref_frame_idx[GST_AV1_REF_LAST2_FRAME] = va_frame->ref_frame_idx[GST_AV1_REF_LAST2_FRAME] =
va_frame->ref_frame_idx[GST_AV1_REF_GOLDEN_FRAME]; va_frame->ref_frame_idx[GST_AV1_REF_LAST_FRAME];
} }
num--; num--;
@ -1535,7 +1553,7 @@ _av1_assign_ref_index (GstVaAV1Enc * self, GstVideoCodecFrame * frame)
all_refs[num].index_in_dpb; all_refs[num].index_in_dpb;
} else { } else {
va_frame->ref_frame_idx[GST_AV1_REF_LAST3_FRAME] = va_frame->ref_frame_idx[GST_AV1_REF_LAST3_FRAME] =
va_frame->ref_frame_idx[GST_AV1_REF_GOLDEN_FRAME]; va_frame->ref_frame_idx[GST_AV1_REF_LAST_FRAME];
} }
/* Setting the backward refs */ /* Setting the backward refs */