/*
 *  Copyright (c) 2020 The WebM project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */

#include "vp9/encoder/vp9_ext_ratectrl.h"
#include "vp9/common/vp9_common.h"

void vp9_extrc_init(EXT_RATECTRL *ext_ratectrl) { vp9_zero(*ext_ratectrl); }

void vp9_extrc_create(vpx_rc_funcs_t funcs, vpx_rc_config_t ratectrl_config,
                      EXT_RATECTRL *ext_ratectrl) {
  vpx_rc_firstpass_stats_t *rc_firstpass_stats;
  vp9_extrc_delete(ext_ratectrl);
  ext_ratectrl->funcs = funcs;
  ext_ratectrl->ratectrl_config = ratectrl_config;
  ext_ratectrl->funcs.create_model(ext_ratectrl->funcs.priv,
                                   &ext_ratectrl->ratectrl_config,
                                   ext_ratectrl->model);
  rc_firstpass_stats = &ext_ratectrl->rc_firstpass_stats;
  rc_firstpass_stats->num_frames = ratectrl_config.show_frame_count;
  rc_firstpass_stats->frame_stats =
      vpx_malloc(sizeof(*rc_firstpass_stats->frame_stats) *
                 rc_firstpass_stats->num_frames);
  ext_ratectrl->ready = 1;
}

void vp9_extrc_delete(EXT_RATECTRL *ext_ratectrl) {
  if (ext_ratectrl->ready) {
    ext_ratectrl->funcs.delete_model(ext_ratectrl->model);
  }
  vp9_extrc_init(ext_ratectrl);
}

static void gen_rc_firstpass_stats(const FIRSTPASS_STATS *stats,
                                   double *rc_frame_stats) {
  rc_frame_stats[0] = stats->frame;
  rc_frame_stats[1] = stats->weight;
  rc_frame_stats[2] = stats->intra_error;
  rc_frame_stats[3] = stats->coded_error;
  rc_frame_stats[4] = stats->sr_coded_error;
  rc_frame_stats[5] = stats->frame_noise_energy;
  rc_frame_stats[6] = stats->pcnt_inter;
  rc_frame_stats[7] = stats->pcnt_motion;
  rc_frame_stats[8] = stats->pcnt_second_ref;
  rc_frame_stats[9] = stats->pcnt_neutral;
  rc_frame_stats[10] = stats->pcnt_intra_low;
  rc_frame_stats[11] = stats->pcnt_intra_high;
  rc_frame_stats[12] = stats->intra_skip_pct;
  rc_frame_stats[13] = stats->intra_smooth_pct;
  rc_frame_stats[14] = stats->inactive_zone_rows;
  rc_frame_stats[15] = stats->inactive_zone_cols;
  rc_frame_stats[16] = stats->MVr;
  rc_frame_stats[17] = stats->mvr_abs;
  rc_frame_stats[18] = stats->MVc;
  rc_frame_stats[19] = stats->mvc_abs;
  rc_frame_stats[20] = stats->MVrv;
  rc_frame_stats[21] = stats->MVcv;
  rc_frame_stats[22] = stats->mv_in_out_count;
  rc_frame_stats[23] = stats->duration;
  rc_frame_stats[24] = stats->count;
}

void vp9_extrc_send_firstpass_stats(const FIRST_PASS_INFO *first_pass_info,
                                    EXT_RATECTRL *ext_ratectrl) {
  if (ext_ratectrl->ready) {
    vpx_rc_firstpass_stats_t *rc_firstpass_stats =
        &ext_ratectrl->rc_firstpass_stats;
    int i;
    assert(rc_firstpass_stats->num_frames == first_pass_info->num_frames);
    for (i = 0; i < rc_firstpass_stats->num_frames; ++i) {
      gen_rc_firstpass_stats(&first_pass_info->stats[i],
                             rc_firstpass_stats->frame_stats[i]);
    }
    ext_ratectrl->funcs.send_firstpass_stats(ext_ratectrl->model,
                                             rc_firstpass_stats);
  }
}