From 8a86737ff091b12e4aa0599fab265e5b3007cf19 Mon Sep 17 00:00:00 2001
From: feos <feos-theos@yandex.ru>
Date: Wed, 21 Oct 2015 00:04:24 +0300
Subject: [PATCH] tastudio: color stated frames on WasLag too.

---
 .../tools/TAStudio/TAStudio.ListView.cs             | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/BizHawk.Client.EmuHawk/tools/TAStudio/TAStudio.ListView.cs b/BizHawk.Client.EmuHawk/tools/TAStudio/TAStudio.ListView.cs
index aef3c4b631..e2e9673c57 100644
--- a/BizHawk.Client.EmuHawk/tools/TAStudio/TAStudio.ListView.cs
+++ b/BizHawk.Client.EmuHawk/tools/TAStudio/TAStudio.ListView.cs
@@ -189,7 +189,7 @@ namespace BizHawk.Client.EmuHawk
 			}
 			else if (record.Lagged.HasValue)
 			{
-                if (!CurrentTasMovie.TasStateManager.HasState(index) && TasView.denoteStatesWithBGColor)
+				if (!record.HasState && TasView.denoteStatesWithBGColor)
                     color = record.Lagged.Value ?
                         LagZone_InputLog :
                         GreenZone_InputLog;
@@ -200,9 +200,14 @@ namespace BizHawk.Client.EmuHawk
 			}
 			else if (record.WasLagged.HasValue)
 			{
-				color = record.WasLagged.Value ?
-					LagZone_InputLog_Invalidated :
-					GreenZone_InputLog_Invalidated;
+				if (!record.HasState && TasView.denoteStatesWithBGColor)
+					color = record.WasLagged.Value ?
+						LagZone_InputLog_Invalidated :
+						GreenZone_InputLog_Invalidated;
+				else
+					color = record.WasLagged.Value ?
+						LagZone_InputLog_Stated :
+						GreenZone_InputLog_Stated;
 			}
 			else
 			{