Fix "Akaya Akashiya Ayakashi" umd switch

Core/HLE/sceUmd.cpp

@@ -38,6 +38,9 @@
 #include "Core/FileSystems/MetaFileSystem.h"
 #include "Core/FileSystems/ISOFileSystem.h"
 #include "Core/FileSystems/VirtualDiscFileSystem.h"
+#ifdef USING_WIN_UI
+#include "Windows/MainWindowMenu.h"
+#endif
 
 const u64 MICRO_DELAY_ACTIVATE = 4000;
 
@@ -47,10 +50,12 @@ static u32 umdErrorStat = 0;
 static int driveCBId = 0;
 static int umdStatTimeoutEvent = -1;
 static int umdStatChangeEvent = -1;
+static int umdInsertChangeEvent = -1;
 static std::vector<SceUID> umdWaitingThreads;
 static std::map<SceUID, u64> umdPausedWaits;
 
 bool UMDReplacePermit = false;
+bool UMDInserted = true;
 
 struct PspUmdInfo {
 	u32_le size;
@@ -59,13 +64,15 @@ struct PspUmdInfo {
 
 void __UmdStatTimeout(u64 userdata, int cyclesLate);
 void __UmdStatChange(u64 userdata, int cyclesLate);
+void __UmdInsertChange(u64 userdata, int cyclesLate);
 void __UmdBeginCallback(SceUID threadID, SceUID prevCallbackId);
 void __UmdEndCallback(SceUID threadID, SceUID prevCallbackId);
 
 void __UmdInit()
 {
 	umdStatTimeoutEvent = CoreTiming::RegisterEvent("UmdTimeout", __UmdStatTimeout);
 	umdStatChangeEvent = CoreTiming::RegisterEvent("UmdChange", __UmdStatChange);
+	umdInsertChangeEvent = CoreTiming::RegisterEvent("UmdInsertChange", __UmdInsertChange);
 	umdActivated = 1;
 	umdStatus = 0;
 	umdErrorStat = 0;
@@ -108,6 +115,11 @@ static u8 __KernelUmdGetState()
 	return state;
 }
 
+void __UmdInsertChange(u64 userdata, int cyclesLate)
+{
+	UMDInserted = true;
+}
+
 void __UmdStatChange(u64 userdata, int cyclesLate)
 {
 	// TODO: Why not a bool anyway?
@@ -225,8 +237,12 @@ void __UmdEndCallback(SceUID threadID, SceUID prevCallbackId)
 
 static int sceUmdCheckMedium()
 {
-	DEBUG_LOG(SCEIO, "1=sceUmdCheckMedium()");
-	return 1; //non-zero: disc in drive
+	if (UMDInserted) {
+		DEBUG_LOG(SCEIO, "1=sceUmdCheckMedium()");
+		return 1; //non-zero: disc in drive
+	}
+	DEBUG_LOG(SCEIO, "0=sceUmdCheckMedium()");
+	return 0;
 }
 
 static u32 sceUmdGetDiscInfo(u32 infoAddr)
@@ -495,7 +511,8 @@ void __UmdReplace(std::string filepath) {
 		}
 	}
 	delete currentUMD;
-
+	UMDInserted = false;
+	CoreTiming::ScheduleEvent(usToCycles(200*1000), umdInsertChangeEvent, 0); // Wait sceUmdCheckMedium call
 	// TODO Is this always correct if UMD was not activated?
 	u32 notifyArg = PSP_UMD_PRESENT | PSP_UMD_READABLE | PSP_UMD_CHANGED;
 	if (driveCBId != -1)
@@ -510,13 +527,19 @@ static u32 sceUmdReplaceProhibit()
 {
 	UMDReplacePermit = false;
 	DEBUG_LOG(SCEIO,"sceUmdReplaceProhibit()");
+#ifdef USING_WIN_UI
+	MainWindow::ChangeMenu();
+#endif
 	return 0;
 }
 
 static u32 sceUmdReplacePermit()
 {
 	UMDReplacePermit = true;
 	DEBUG_LOG(SCEIO,"sceUmdReplacePermit()");
+#ifdef USING_WIN_UI
+	MainWindow::ChangeMenu();
+#endif
 	return 0;
 }
 

Windows/MainWindowMenu.cpp

@@ -1258,4 +1258,8 @@ namespace MainWindow {
 		}
 		return FALSE;
 	}
+
+	void ChangeMenu() {
+		SetIngameMenuItemStates(GetMenu(GetHWND()), UISTATE_INGAME);
+	}
 }

Windows/MainWindowMenu.h

@@ -10,4 +10,5 @@ namespace MainWindow {
 	void BrowseAndBoot(std::string defaultPath, bool browseDirectory = false);
 	void BrowseAndBootDone();
 	void setTexScalingMultiplier(int level);
+	void ChangeMenu();
 }