From 3afff7940c016d44d212aae342825f3acb4fda74 Mon Sep 17 00:00:00 2001
From: Joose Sainio <joose.sainio@tuni.fi>
Date: Mon, 6 Jun 2022 09:04:00 +0300
Subject: [PATCH] [jccr] update tr-skip context for jccr blocks

---
 src/search.c | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/search.c b/src/search.c
index b181d5fe..6ce6b9a4 100644
--- a/src/search.c
+++ b/src/search.c
@@ -592,6 +592,9 @@ static double cu_rd_cost_tr_split_accurate(const encoder_state_t* const state,
           chroma_width);
         chroma_ssd = ssd_u_joint + ssd_v_joint;
       }
+      if (chroma_can_use_tr_skip) {
+        CABAC_FBITS_UPDATE(cabac, &cabac->ctx.transform_skip_model_chroma, tr_cu->tr_skip & 2, tr_tree_bits, "transform_skip_flag");
+      }
       coeff_bits += uvg_get_coeff_cost(state, &lcu->coeff.joint_uv[index], NULL, chroma_width, COLOR_U, scan_order, 0);
     }
   }