From f37eaa2bf1d10fda50a4e958ac63df84a58ce15c Mon Sep 17 00:00:00 2001
From: Tung Nguyen <tung.nguyen@hhi.fraunhofer.de>
Date: Thu, 31 Jan 2019 09:19:19 +0100
Subject: [PATCH] fix: do not apply TS to chroma when TS is applied to luma

---
 source/Lib/CommonLib/TrQuant.cpp | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/source/Lib/CommonLib/TrQuant.cpp b/source/Lib/CommonLib/TrQuant.cpp
index f2afd109db..6396fe5c3b 100644
--- a/source/Lib/CommonLib/TrQuant.cpp
+++ b/source/Lib/CommonLib/TrQuant.cpp
@@ -205,7 +205,7 @@ void TrQuant::invTransformNxN( TransformUnit &tu, const ComponentID &compID, Pel
     DTRACE_COEFF_BUF( D_TCOEFF, tempCoeff, tu, tu.cu->predMode, compID );
 
 #if JVET_M0464_UNI_MTS
-    if( tu.mtsIdx == 1 )
+    if( isLuma(compID) && tu.mtsIdx == 1 )
 #else
     if( tu.transformSkip[compID] )
 #endif
@@ -498,7 +498,7 @@ void TrQuant::transformNxN(TransformUnit &tu, const ComponentID &compID, const Q
     tu.mtsIdx = it->first;
     CoeffBuf tempCoeff( m_mtsCoeffs[tu.mtsIdx], rect );
 
-    if( tu.mtsIdx == 1 )
+    if( isLuma(compID) && tu.mtsIdx == 1 )
     {
       xTransformSkip( tu, compID, resiBuf, tempCoeff.buf );
     }
@@ -593,7 +593,7 @@ void TrQuant::transformNxN(TransformUnit &tu, const ComponentID &compID, const Q
 #if JVET_M0464_UNI_MTS
       if( !loadTr )
       {
-        if( tu.mtsIdx == 1 )
+        if( isLuma(compID) && tu.mtsIdx == 1 )
 #else
       if( tu.transformSkip[compID] )
 #endif
-- 
GitLab