Add function to set the SSIM based rdmult
In tune=ssim mode, this CL scales rdmult by the precalculated scaling
factor in the block partition search stage.
Change-Id: I82c9c27fe6cc5eddeb8d01efb3dbb210125fbbe7
diff --git a/av1/encoder/encodeframe.c b/av1/encoder/encodeframe.c
index 0d1e621..8f68871 100644
--- a/av1/encoder/encodeframe.c
+++ b/av1/encoder/encodeframe.c
@@ -209,6 +209,41 @@
}
#endif // !CONFIG_REALTIME_ONLY
+static void set_ssim_rdmult(const AV1_COMP *const cpi, MACROBLOCK *const x,
+ const BLOCK_SIZE bsize, const int mi_row,
+ const int mi_col, int *const rdmult) {
+ const AV1_COMMON *const cm = &cpi->common;
+
+ const int bsize_base = BLOCK_16X16;
+ const int num_mi_w = mi_size_wide[bsize_base];
+ const int num_mi_h = mi_size_high[bsize_base];
+ const int num_cols = (cm->mi_cols + num_mi_w - 1) / num_mi_w;
+ const int num_rows = (cm->mi_rows + num_mi_h - 1) / num_mi_h;
+ const int num_bcols = (mi_size_wide[bsize] + num_mi_w - 1) / num_mi_w;
+ const int num_brows = (mi_size_high[bsize] + num_mi_h - 1) / num_mi_h;
+ int row, col;
+ double num_of_mi = 0.0;
+ double geom_mean_of_scale = 0.0;
+
+ assert(cpi->oxcf.tuning == AOM_TUNE_SSIM);
+
+ for (row = mi_row / num_mi_w;
+ row < num_rows && row < mi_row / num_mi_w + num_brows; ++row) {
+ for (col = mi_col / num_mi_h;
+ col < num_cols && col < mi_col / num_mi_h + num_bcols; ++col) {
+ const int index = row * num_cols + col;
+ geom_mean_of_scale += log(cpi->ssim_rdmult_scaling_factors[index]);
+ num_of_mi += 1.0;
+ }
+ }
+ geom_mean_of_scale = exp(geom_mean_of_scale / num_of_mi);
+
+ *rdmult = (int)((double)(*rdmult) * geom_mean_of_scale);
+ *rdmult = AOMMAX(*rdmult, 0);
+ set_error_per_bit(x, *rdmult);
+ aom_clear_system_state();
+}
+
static void set_offsets_without_segment_id(const AV1_COMP *const cpi,
const TileInfo *const tile,
MACROBLOCK *const x, int mi_row,
@@ -278,6 +313,9 @@
}
av1_init_plane_quantizers(cpi, x, mbmi->segment_id);
}
+ if (cpi->oxcf.tuning == AOM_TUNE_SSIM) {
+ set_ssim_rdmult(cpi, x, bsize, mi_row, mi_col, &x->rdmult);
+ }
}
static void update_filter_type_count(uint8_t allow_update_cdf,
@@ -638,6 +676,9 @@
// Set error per bit for current rdmult
set_error_per_bit(x, x->rdmult);
+ if (cpi->oxcf.tuning == AOM_TUNE_SSIM) {
+ set_ssim_rdmult(cpi, x, bsize, mi_row, mi_col, &x->rdmult);
+ }
av1_rd_cost_update(x->rdmult, &best_rd);
// Find best coding mode & reconstruct the MB so it is available
