fixed some graphics problems with loading savestates (for example, wrong colors on title screen of metroid prime 3)

1 files changed, 8 insertions, 1 deletions

diff --git a/Source/Core/VideoCommon/Src/VideoState.cpp b/Source/Core/VideoCommon/Src/VideoState.cpp
index b1c1ed0889..5a677a2460 100644
--- a/Source/Core/VideoCommon/Src/VideoState.cpp
+++ b/Source/Core/VideoCommon/Src/VideoState.cpp
@@ -24,6 +24,8 @@
 #include "Fifo.h"
 #include "CommandProcessor.h"
 #include "PixelEngine.h"
+#include "PixelShaderManager.h"
+#include "VertexShaderManager.h"
 
 static void DoState(PointerWrap &p)
 {
@@ -48,12 +50,17 @@ static void DoState(PointerWrap &p)
 
     CommandProcessor::DoState(p);
     PixelEngine::DoState(p);
+
+	// the old way of replaying current bpmem as writes to push side effects to pixel shader manager doesn't really work.
+	PixelShaderManager::DoState(p);
+	VertexShaderManager::DoState(p);
+
+	// TODO: search for more data that should be saved and add it here
 }
 
 void VideoCommon_DoState(PointerWrap &p)
 {
     DoState(p);
-	//TODO: search for more data that should be saved and add it here
 }
 
 void VideoCommon_RunLoop(bool enable)