From 634cf72c5d79313f070096829ec1c4d35efcebcf Mon Sep 17 00:00:00 2001 From: Joose Sainio Date: Mon, 1 Jul 2024 12:23:38 +0300 Subject: [PATCH] [mts] Fix erroneous bitcost estimation --- src/encode_coding_tree.c | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/encode_coding_tree.c b/src/encode_coding_tree.c index 858d89f4..801e9943 100644 --- a/src/encode_coding_tree.c +++ b/src/encode_coding_tree.c @@ -52,8 +52,8 @@ bool uvg_is_mts_allowed(const encoder_state_t * const state, cu_info_t *const pr { uint32_t ts_max_size = 1 << state->encoder_control->cfg.trskip_max_size; const uint32_t max_size = 32; // CU::isIntra(cu) ? MTS_INTRA_MAX_CU_SIZE : MTS_INTER_MAX_CU_SIZE; - const uint32_t cu_width = cu_loc->width; - const uint32_t cu_height = cu_loc->height; + const uint32_t cu_width = 1 << pred_cu->log2_width; + const uint32_t cu_height = 1 << pred_cu->log2_height; //bool mts_allowed = cu.chType == CHANNEL_TYPE_LUMA && compID == COMPONENT_Y; uint8_t mts_type = state->encoder_control->cfg.mts;