RC: Implement send_firstpass and tests

Bug: b:450252793
Change-Id: Ie9135bb4974b886f5b55ee1bb4369e6d903ef016

av1/encoder/av1_ext_ratectrl.c

diff --git a/av1/encoder/av1_ext_ratectrl.c b/av1/encoder/av1_ext_ratectrl.c
index bf24fb9..65e1092 100644
--- a/av1/encoder/av1_ext_ratectrl.c
+++ b/av1/encoder/av1_ext_ratectrl.c
@@ -50,6 +50,61 @@
   return AOM_CODEC_OK;
 }
 
+static void gen_rc_firstpass_stats(const FIRSTPASS_STATS *stats,
+                                   aom_rc_frame_stats_t *rc_frame_stats) {
+  rc_frame_stats->frame = stats->frame;
+  rc_frame_stats->weight = stats->weight;
+  rc_frame_stats->intra_error = stats->intra_error;
+  rc_frame_stats->coded_error = stats->coded_error;
+  rc_frame_stats->sr_coded_error = stats->sr_coded_error;
+  rc_frame_stats->frame_avg_wavelet_energy = stats->frame_avg_wavelet_energy;
+  rc_frame_stats->pcnt_inter = stats->pcnt_inter;
+  rc_frame_stats->pcnt_motion = stats->pcnt_motion;
+  rc_frame_stats->pcnt_second_ref = stats->pcnt_second_ref;
+  rc_frame_stats->pcnt_neutral = stats->pcnt_neutral;
+  rc_frame_stats->intra_skip_pct = stats->intra_skip_pct;
+  rc_frame_stats->inactive_zone_rows = stats->inactive_zone_rows;
+  rc_frame_stats->inactive_zone_cols = stats->inactive_zone_cols;
+  rc_frame_stats->MVr = stats->MVr;
+  rc_frame_stats->mvr_abs = stats->mvr_abs;
+  rc_frame_stats->MVc = stats->MVc;
+  rc_frame_stats->mvc_abs = stats->mvc_abs;
+  rc_frame_stats->MVrv = stats->MVrv;
+  rc_frame_stats->MVcv = stats->MVcv;
+  rc_frame_stats->mv_in_out_count = stats->mv_in_out_count;
+  rc_frame_stats->duration = stats->duration;
+  rc_frame_stats->count = stats->count;
+  rc_frame_stats->new_mv_count = stats->new_mv_count;
+  rc_frame_stats->raw_error_stdev = stats->raw_error_stdev;
+  rc_frame_stats->is_flash = stats->is_flash;
+  rc_frame_stats->noise_var = stats->noise_var;
+  rc_frame_stats->cor_coeff = stats->cor_coeff;
+  rc_frame_stats->log_intra_error = stats->log_intra_error;
+  rc_frame_stats->log_coded_error = stats->log_coded_error;
+}
+
+aom_codec_err_t av1_extrc_send_firstpass_stats(
+    AOM_EXT_RATECTRL *ext_ratectrl, const FIRSTPASS_INFO *first_pass_info) {
+  assert(first_pass_info != NULL);
+  assert(ext_ratectrl != NULL);
+  if (ext_ratectrl->ready) {
+    aom_rc_status_t rc_status;
+    aom_rc_firstpass_stats_t *rc_firstpass_stats =
+        &ext_ratectrl->rc_firstpass_stats;
+    assert(rc_firstpass_stats->num_frames == first_pass_info->stats_buf_size);
+    for (int i = 0; i < rc_firstpass_stats->num_frames; ++i) {
+      gen_rc_firstpass_stats(&first_pass_info->stats_buf[i],
+                             &rc_firstpass_stats->frame_stats[i]);
+    }
+    rc_status = ext_ratectrl->funcs.send_firstpass_stats(ext_ratectrl->model,
+                                                         rc_firstpass_stats);
+    if (rc_status == AOM_RC_ERROR) {
+      return AOM_CODEC_ERROR;
+    }
+  }
+  return AOM_CODEC_OK;
+}
+
 aom_codec_err_t av1_extrc_delete(AOM_EXT_RATECTRL *ext_ratectrl) {
   if (ext_ratectrl == NULL) {
     return AOM_CODEC_INVALID_PARAM;

av1/encoder/pass2_strategy.c

diff --git a/av1/encoder/pass2_strategy.c b/av1/encoder/pass2_strategy.c
index a1d6640..9cd5225 100644
--- a/av1/encoder/pass2_strategy.c
+++ b/av1/encoder/pass2_strategy.c
@@ -3757,6 +3757,16 @@
     return;
   }
 
+  if (cpi->common.current_frame.frame_number == 0 &&
+      cpi->ext_ratectrl.funcs.send_firstpass_stats != NULL) {
+    const aom_codec_err_t codec_status = av1_extrc_send_firstpass_stats(
+        &cpi->ext_ratectrl, &cpi->ppi->twopass.firstpass_info);
+    if (codec_status != AOM_CODEC_OK) {
+      aom_internal_error(cpi->common.error, codec_status,
+                         "av1_extrc_send_firstpass_stats() failed");
+    }
+  }
+
   const FIRSTPASS_STATS *const start_pos = cpi->twopass_frame.stats_in;
   int update_total_stats = 0;
 

test/ext_ratectrl_test.cc

diff --git a/test/ext_ratectrl_test.cc b/test/ext_ratectrl_test.cc
index ef7f92e..a84def7 100644
--- a/test/ext_ratectrl_test.cc
+++ b/test/ext_ratectrl_test.cc
@@ -33,6 +33,9 @@
 // A flag to indicate if delete_model() is called.
 bool is_delete_model_called = false;
 
+// A flag to indicate if send_firstpass_stats() is called.
+bool is_send_firstpass_stats_called = false;
+
 aom_rc_status_t mock_create_model(void *priv,
                                   const aom_rc_config_t *ratectrl_config,
                                   aom_rc_model_t *ratectrl_model) {
@@ -54,8 +57,11 @@
 aom_rc_status_t mock_send_firstpass_stats(
     aom_rc_model_t ratectrl_model,
     const aom_rc_firstpass_stats_t *firstpass_stats) {
-  (void)ratectrl_model;
-  (void)firstpass_stats;
+  EXPECT_NE(ratectrl_model, nullptr);
+  EXPECT_NE(firstpass_stats, nullptr);
+  EXPECT_EQ(firstpass_stats->num_frames, kFrameNum);
+  EXPECT_NE(firstpass_stats->frame_stats, nullptr);
+  is_send_firstpass_stats_called = true;
   return AOM_RC_OK;
 }
 
@@ -94,6 +100,9 @@
     InitializeConfig(static_cast<libaom_test::TestMode>(GET_PARAM(1)));
     cfg_.g_threads = 1;
     cfg_.g_limit = kFrameNum;
+    is_create_model_called = false;
+    is_delete_model_called = false;
+    is_send_firstpass_stats_called = false;
   }
 
   void PreEncodeFrameHook(::libaom_test::VideoSource *video,
@@ -112,6 +121,7 @@
   ::libaom_test::Y4mVideoSource video("screendata.y4m", 0, kFrameNum);
   ASSERT_NO_FATAL_FAILURE(RunLoop(&video));
   EXPECT_TRUE(is_create_model_called);
+  EXPECT_TRUE(is_send_firstpass_stats_called);
   EXPECT_TRUE(is_delete_model_called);
 }