From c83e0f8cff64ed7bfb8f8ca90fab1a110878d9b9 Mon Sep 17 00:00:00 2001
From: Clansty <i@gao4.pw>
Date: Fri, 12 Apr 2024 00:40:29 +0800
Subject: [PATCH] [F] Fix Chime scanning and (maybe) DX Pass crash

---
 AquaMai/Performance/ImproveLoadSpeed.cs | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/AquaMai/Performance/ImproveLoadSpeed.cs b/AquaMai/Performance/ImproveLoadSpeed.cs
index 1e659adc..006ac44a 100644
--- a/AquaMai/Performance/ImproveLoadSpeed.cs
+++ b/AquaMai/Performance/ImproveLoadSpeed.cs
@@ -1,6 +1,7 @@
 using System.Diagnostics;
 using HarmonyLib;
-using MelonLoader;
+using MAI2.Util;
+using Manager;
 using Process;
 
 namespace AquaMai.Performance
@@ -19,6 +20,8 @@ namespace AquaMai.Performance
                     traverse.Field("_state").SetValue((byte)4);
                     break;
                 case 5:
+                case 6:
+                case 7:
                     traverse.Field("_state").SetValue((byte)8);
                     break;
                 case 9:
@@ -38,9 +41,12 @@ namespace AquaMai.Performance
             switch (state)
             {
                 case 0:
-                case 1:
+                    traverse.Field("_state").SetValue((byte)1);
+                    break;
                 case 2:
-                    traverse.Field("_state").SetValue((byte)3);
+                    // AimeReader maybe typeof AimeReaderManager or ChimeReaderManager, must build with correct Assembly-CSharp.dll in Libs folder
+                    if(SingletonStateMachine<AmManager, AmManager.EState>.Instance.AimeReader.GetType().FullName == "Manager.AimeReaderManager")
+                        traverse.Field("_state").SetValue((byte)3);
                     break;
                 case 4:
                     traverse.Field("_state").SetValue((byte)5);