Implement threaded JPEG screenshot for GeneralsMD

Author: bobtista

GeneralsMD/Code/GameEngineDevice/Include/W3DDevice/GameClient/W3DDisplay.h

@@ -121,6 +121,7 @@ class W3DDisplay : public Display
 	virtual VideoBuffer*	createVideoBuffer( void ) ;							///< Create a video buffer that can be used for this display
 
 	virtual void takeScreenShot(void);						//save screenshot to file
+	virtual void takeScreenShotCompressed(void);	//save compressed screenshot to file (JPG/PNG) without stalling
 	virtual void toggleMovieCapture(void);			//enable AVI or frame capture mode.
 
 	virtual void toggleLetterBox(void);	///<enabled letter-boxed display

GeneralsMD/Code/GameEngineDevice/Source/W3DDevice/GameClient/W3DDisplay.cpp

@@ -39,6 +39,11 @@ static void drawFramerateBar(void);
 #include <windows.h>
 #include <io.h>
 #include <time.h>
+#include <thread>
+#include <memory>
+
+#define STB_IMAGE_WRITE_IMPLEMENTATION
+#include <stb_image_write.h>
 
 // USER INCLUDES //////////////////////////////////////////////////////////////
 #include "Common/FramePacer.h"
@@ -3141,6 +3146,85 @@ void W3DDisplay::takeScreenShot(void)
 	TheInGameUI->message(TheGameText->fetch("GUI:ScreenCapture"), ufileName.str());
 }
 
+void W3DDisplay::takeScreenShotCompressed(void)
+{
+	char leafname[256];
+	char pathname[1024];
+	static int frame_number = 1;
+
+	Bool done = false;
+	while (!done) {
+		sprintf(leafname, "sshot%.3d.jpg", frame_number++);
+		strcpy(pathname, TheGlobalData->getPath_UserData().str());
+		strlcat(pathname, leafname, ARRAY_SIZE(pathname));
+		if (_access(pathname, 0) == -1)
+			done = true;
+	}
+
+	SurfaceClass* surface = DX8Wrapper::_Get_DX8_Back_Buffer();
+	SurfaceClass::SurfaceDescription surfaceDesc;
+	surface->Get_Description(surfaceDesc);
+
+	SurfaceClass* surfaceCopy = NEW_REF(SurfaceClass, (DX8Wrapper::_Create_DX8_Surface(surfaceDesc.Width, surfaceDesc.Height, surfaceDesc.Format)));
+	DX8Wrapper::_Copy_DX8_Rects(surface->Peek_D3D_Surface(), NULL, 0, surfaceCopy->Peek_D3D_Surface(), NULL);
+
+	surface->Release_Ref();
+	surface = NULL;
+
+	struct Rect
+	{
+		int Pitch;
+		void* pBits;
+	} lrect;
+
+	lrect.pBits = surfaceCopy->Lock(&lrect.Pitch);
+	if (lrect.pBits == NULL)
+	{
+		surfaceCopy->Release_Ref();
+		return;
+	}
+
+	unsigned int x, y, index, index2;
+	unsigned int width = surfaceDesc.Width;
+	unsigned int height = surfaceDesc.Height;
+
+	std::shared_ptr<unsigned char> imageData(new unsigned char[3 * width * height], 
+		std::default_delete<unsigned char[]>());
+	unsigned char* image = imageData.get();
+
+	for (y = 0; y < height; y++)
+	{
+		for (x = 0; x < width; x++)
+		{
+			index = 3 * (x + y * width);
+			index2 = y * lrect.Pitch + 4 * x;
+
+			image[index]     = *((unsigned char*)lrect.pBits + index2 + 2);
+			image[index + 1] = *((unsigned char*)lrect.pBits + index2 + 1);
+			image[index + 2] = *((unsigned char*)lrect.pBits + index2 + 0);
+		}
+	}
+
+	surfaceCopy->Unlock();
+	surfaceCopy->Release_Ref();
+	surfaceCopy = NULL;
+
+	std::string pathnameCopy(pathname);
+	std::string leafnameCopy(leafname);
+
+	std::thread([imageData, width, height, pathnameCopy, leafnameCopy]() {
+		int result = stbi_write_jpg(pathnameCopy.c_str(), width, height, 3, imageData.get(), 90);
+		
+		if (!result) {
+			OutputDebugStringA("Failed to write screenshot JPEG\n");
+		}
+	}).detach();
+
+	UnicodeString ufileName;
+	ufileName.translate(leafnameCopy.c_str());
+	TheInGameUI->message(TheGameText->fetch("GUI:ScreenCapture"), ufileName.str());
+}
+
 /** Start/Stop capturing an AVI movie*/
 void W3DDisplay::toggleMovieCapture(void)
 {