Add reference buffer manager into gop structure
Change-Id: Ib43bed68eaf6a29054eb446aaae4574352774ef2
diff --git a/av1/encoder/firstpass.h b/av1/encoder/firstpass.h
index e37926c..f5ea986 100644
--- a/av1/encoder/firstpass.h
+++ b/av1/encoder/firstpass.h
@@ -187,6 +187,8 @@
// This is currently only populated for AOM_Q mode
unsigned char q_val[MAX_STATIC_GF_GROUP_LENGTH];
int bit_allocation[MAX_STATIC_GF_GROUP_LENGTH];
+ FRAME_TYPE frame_type[MAX_STATIC_GF_GROUP_LENGTH];
+ REFBUF_STATE refbuf_state[MAX_STATIC_GF_GROUP_LENGTH];
int arf_index; // the index in the gf group of ARF, if no arf, then -1
int size;
/*!\endcond */
diff --git a/av1/encoder/gop_structure.c b/av1/encoder/gop_structure.c
index 4cd5bce..362261a 100644
--- a/av1/encoder/gop_structure.c
+++ b/av1/encoder/gop_structure.c
@@ -47,6 +47,8 @@
gf_group->layer_depth[*frame_ind] = MAX_ARF_LAYERS;
gf_group->arf_boost[*frame_ind] = av1_calc_arf_boost(
twopass, rc, frame_info, start, end - start, 0, NULL, NULL);
+ gf_group->frame_type[*frame_ind] = INTER_FRAME;
+ gf_group->refbuf_state[*frame_ind] = REFBUF_UPDATE;
gf_group->max_layer_depth =
AOMMAX(gf_group->max_layer_depth, layer_depth);
++(*frame_ind);
@@ -60,6 +62,8 @@
gf_group->arf_src_offset[*frame_ind] = m - start - 1;
gf_group->cur_frame_idx[*frame_ind] = *cur_frame_idx;
gf_group->layer_depth[*frame_ind] = layer_depth;
+ gf_group->frame_type[*frame_ind] = INTER_FRAME;
+ gf_group->refbuf_state[*frame_ind] = REFBUF_UPDATE;
// Get the boost factor for intermediate ARF frames.
gf_group->arf_boost[*frame_ind] = av1_calc_arf_boost(
@@ -76,6 +80,8 @@
gf_group->cur_frame_idx[*frame_ind] = *cur_frame_idx;
gf_group->arf_boost[*frame_ind] = 0;
gf_group->layer_depth[*frame_ind] = layer_depth;
+ gf_group->frame_type[*frame_ind] = INTER_FRAME;
+ gf_group->refbuf_state[*frame_ind] = REFBUF_UPDATE;
++(*frame_ind);
++(*cur_frame_idx);
@@ -104,6 +110,8 @@
gf_group->arf_src_offset[frame_index] = 0;
gf_group->cur_frame_idx[frame_index] = cur_frame_index;
gf_group->layer_depth[frame_index] = 0;
+ gf_group->frame_type[frame_index] = KEY_FRAME;
+ gf_group->refbuf_state[frame_index] = REFBUF_RESET;
gf_group->max_layer_depth = 0;
++frame_index;
@@ -111,6 +119,8 @@
gf_group->arf_src_offset[frame_index] = 0;
gf_group->cur_frame_idx[frame_index] = cur_frame_index;
gf_group->layer_depth[frame_index] = 0;
+ gf_group->frame_type[frame_index] = INTER_FRAME;
+ gf_group->refbuf_state[frame_index] = REFBUF_UPDATE;
gf_group->max_layer_depth = 0;
++frame_index;
cur_frame_index++;
@@ -120,6 +130,10 @@
gf_group->cur_frame_idx[frame_index] = cur_frame_index;
gf_group->layer_depth[frame_index] =
first_frame_update_type == OVERLAY_UPDATE ? MAX_ARF_LAYERS + 1 : 0;
+ gf_group->frame_type[frame_index] =
+ (first_frame_update_type == KF_UPDATE) ? KEY_FRAME : INTER_FRAME;
+ gf_group->refbuf_state[frame_index] =
+ (first_frame_update_type == KF_UPDATE) ? REFBUF_RESET : REFBUF_UPDATE;
gf_group->max_layer_depth = 0;
++frame_index;
++cur_frame_index;
@@ -127,12 +141,15 @@
// ALTREF.
const int use_altref = gf_group->max_layer_depth_allowed > 0;
+ int is_fwd_kf = (gf_interval == cpi->rc.frames_to_key);
if (use_altref) {
gf_group->update_type[frame_index] = ARF_UPDATE;
gf_group->arf_src_offset[frame_index] = gf_interval - 1;
gf_group->cur_frame_idx[frame_index] = cur_frame_index;
gf_group->layer_depth[frame_index] = 1;
gf_group->arf_boost[frame_index] = cpi->rc.gfu_boost;
+ gf_group->frame_type[frame_index] = is_fwd_kf ? KEY_FRAME : INTER_FRAME;
+ gf_group->refbuf_state[frame_index] = REFBUF_UPDATE;
gf_group->max_layer_depth = 1;
gf_group->arf_index = frame_index;
++frame_index;
@@ -150,6 +167,9 @@
gf_group->cur_frame_idx[frame_index] = cur_frame_index;
gf_group->layer_depth[frame_index] = MAX_ARF_LAYERS;
gf_group->arf_boost[frame_index] = NORMAL_BOOST;
+ gf_group->frame_type[frame_index] = INTER_FRAME;
+ gf_group->refbuf_state[frame_index] =
+ is_fwd_kf ? REFBUF_RESET : REFBUF_UPDATE;
}
return frame_index;
}
diff --git a/av1/encoder/ratectrl.c b/av1/encoder/ratectrl.c
index 4a6bb1a..3b65e0d 100644
--- a/av1/encoder/ratectrl.c
+++ b/av1/encoder/ratectrl.c
@@ -2602,6 +2602,8 @@
rc->kf_boost = DEFAULT_KF_BOOST_RT;
rc->source_alt_ref_active = 0;
gf_group->update_type[gf_group->index] = KF_UPDATE;
+ gf_group->frame_type[gf_group->index] = KEY_FRAME;
+ gf_group->refbuf_state[gf_group->index] = REFBUF_RESET;
if (cpi->use_svc) {
if (cm->current_frame.frame_number > 0)
av1_svc_reset_temporal_layers(cpi, 1);
@@ -2610,6 +2612,8 @@
} else {
frame_params->frame_type = INTER_FRAME;
gf_group->update_type[gf_group->index] = LF_UPDATE;
+ gf_group->frame_type[gf_group->index] = INTER_FRAME;
+ gf_group->refbuf_state[gf_group->index] = REFBUF_UPDATE;
if (cpi->use_svc) {
LAYER_CONTEXT *lc = &svc->layer_context[layer];
lc->is_key_frame =
diff --git a/av1/encoder/ratectrl.h b/av1/encoder/ratectrl.h
index 42b276c..37edf2b 100644
--- a/av1/encoder/ratectrl.h
+++ b/av1/encoder/ratectrl.h
@@ -76,7 +76,11 @@
FRAME_UPDATE_TYPES
} UENUM1BYTE(FRAME_UPDATE_TYPE);
-enum { REFBUF_RESET, REFBUF_UPDATE, REFBUF_STATES } UENUM1BYTE(REFBUF_STATE);
+enum {
+ REFBUF_RESET, // Clear reference frame buffer
+ REFBUF_UPDATE, // Refresh reference frame buffer
+ REFBUF_STATES
+} UENUM1BYTE(REFBUF_STATE);
typedef enum {
NO_RESIZE = 0,