From 4d3688cf1768996a77415f7869fc07f3eeb902a9 Mon Sep 17 00:00:00 2001
From: siivonek <kari.siivonen@tuni.fi>
Date: Wed, 1 Jun 2022 14:24:27 +0300
Subject: [PATCH] [lfnst] Fix error in luma tr_skip check. All cbf was checked
 when only luma cbf needed to be checked.

---
 src/encode_coding_tree.c | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/encode_coding_tree.c b/src/encode_coding_tree.c
index 5cd69bed..a5d5262a 100644
--- a/src/encode_coding_tree.c
+++ b/src/encode_coding_tree.c
@@ -219,7 +219,7 @@ static bool encode_lfnst_idx(encoder_state_t * const state, cabac_data_t * const
         const int pu_y = y + ((i / tu_row_length) * tu_height);
         const cu_info_t* cur_tu = kvz_cu_array_at_const(frame->cu_array, pu_x, pu_y);
         assert(cur_tu != NULL && "NULL transform unit.");
-        bool cbf_set = cbf_is_set_any(cur_tu->cbf, tr_depth);
+        bool cbf_set = cbf_is_set(cur_tu->cbf, tr_depth, COLOR_Y);
 
         if (cur_tu != NULL && cbf_set && cur_tu->tr_idx == MTS_SKIP) {
           is_tr_skip = true;