From 741fccb0fd2dbe596fc83bd7c0ba736290d49993 Mon Sep 17 00:00:00 2001
From: Sjoerd van der Berg <harekiet@users.sourceforge.net>
Date: Sat, 19 Oct 2002 17:33:46 +0000
Subject: [PATCH] Added screenshot support.

Imported-from: https://svn.code.sf.net/p/dosbox/code-0/dosbox/trunk@369
---
 src/gui/render.cpp | 122 ++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 116 insertions(+), 6 deletions(-)

diff --git a/src/gui/render.cpp b/src/gui/render.cpp
index 04c3b475..0edcb9a3 100644
--- a/src/gui/render.cpp
+++ b/src/gui/render.cpp
@@ -16,14 +16,20 @@
  *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  */
 
+#include <png.h>
+#include <dirent.h>
+
 #include "dosbox.h"
 #include "video.h"
 #include "render.h"
-
-
+#include "setup.h"
+#include "keyboard.h"
+#include "cross.h"
 
 #define MAX_RES 2048
 
+
+
 struct PalData {
 	struct { 
 		Bit8u red;
@@ -51,8 +57,104 @@ static struct {
 	PalData pal;
 	bool remake;
 	bool enlarge;
+	bool screenshot;
 } render;
 
+static const char * snapshots_dir;
+
+/* Take a screenshot of the data that should be rendered */
+static void TakeScreenShot(Bit8u * bitmap) {
+	Bitu last=0;char file_name[CROSS_LEN];
+	DIR * dir;struct dirent * dir_ent;
+	png_structp png_ptr;
+	png_infop info_ptr;
+	png_bytep * row_pointers;
+	png_color palette[256];
+	Bitu i;
+
+/* First try to alloacte the png structures */
+	png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL,NULL, NULL);
+	if (!png_ptr) return;
+	info_ptr = png_create_info_struct(png_ptr);
+    if (!info_ptr) {
+		png_destroy_write_struct(&png_ptr,(png_infopp)NULL);
+		return;
+    }	
+/* Find a filename to open */
+	dir=opendir(snapshots_dir);
+	if (!dir) {
+		LOG_WARN("Can't open snapshot dir %s",snapshots_dir);
+		return;
+	}
+	while (dir_ent=readdir(dir)) {
+		char tempname[CROSS_LEN];
+		strcpy(tempname,dir_ent->d_name);
+		char * test=strstr(tempname,".png");
+		if (!test) continue;
+		*test=0;
+		if (strlen(tempname)<5) continue;
+		if (strncmp(tempname,"snap",4)!=0) continue;
+		Bitu num=atoi(&tempname[4]);
+		if (num>=last) last=num+1;
+	}
+	closedir(dir);
+	sprintf(file_name,"%s%csnap%05d.png",snapshots_dir,CROSS_FILESPLIT,last);
+/* Open the actual file */
+	FILE * fp=fopen(file_name,"wb");
+	if (!fp) {
+		LOG_WARN("Can't open snapshot file %s",file_name);
+		return;
+	}
+/* Finalize the initing of png library */
+	png_init_io(png_ptr, fp);
+	png_set_compression_level(png_ptr,Z_BEST_COMPRESSION);
+	
+	/* set other zlib parameters */
+	png_set_compression_mem_level(png_ptr, 8);
+	png_set_compression_strategy(png_ptr,Z_DEFAULT_STRATEGY);
+	png_set_compression_window_bits(png_ptr, 15);
+	png_set_compression_method(png_ptr, 8);
+	png_set_compression_buffer_size(png_ptr, 8192);
+
+	if (render.src.bpp==8) {
+		png_set_IHDR(png_ptr, info_ptr, render.src.width, render.src.height,
+			8, PNG_COLOR_TYPE_PALETTE, PNG_INTERLACE_NONE,
+			PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
+		for (i=0;i<256;i++) {
+			palette[i].red=render.pal.rgb[i].red;
+			palette[i].green=render.pal.rgb[i].green;
+			palette[i].blue=render.pal.rgb[i].blue;
+		}
+		png_set_PLTE(png_ptr, info_ptr, palette,256);
+	} else {
+		png_set_IHDR(png_ptr, info_ptr, render.src.width, render.src.height,
+			8, PNG_COLOR_TYPE_RGB, PNG_INTERLACE_NONE,
+			PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
+	}
+	/*Allocate an array of scanline pointers*/
+	row_pointers=(png_bytep*)malloc(render.src.height*sizeof(png_bytep));
+	for (i=0;i<render.src.height;i++) {
+		row_pointers[i]=(bitmap+i*render.src.pitch);
+	}
+	/*tell the png library what to encode.*/
+	png_set_rows(png_ptr, info_ptr, row_pointers);
+	
+	/*Write image to file*/
+	png_write_png(png_ptr, info_ptr, 0, NULL);
+
+	/*close file*/
+	fclose(fp);
+	
+	/*Destroy PNG structs*/
+	png_destroy_write_struct(&png_ptr, &info_ptr);
+	
+	/*clean up dynamically allocated RAM.*/
+	free(row_pointers);
+
+}
+
+
+
 /* This could go kinda bad with multiple threads */
 static void Check_Palette(void) {
 	if (render.pal.first>render.pal.last) return;
@@ -98,8 +200,11 @@ void RENDER_Draw(Bit8u * bitdata) {
 		render.remake=false;
 	}
 	render.handler(&src_data);
+	if (render.screenshot) {
+		TakeScreenShot(src_data);
+		render.screenshot=false;
+	}
 	Draw_8_Normal(src_data,bitdata);
-
 }
 
 
@@ -151,10 +256,15 @@ void RENDER_SetSize(Bitu width,Bitu height,Bitu bpp,Bitu pitch,float ratio,Bitu
 
 }
 
-
-void RENDER_Init(void) {
+static void EnableScreenShot(void) {
+	render.screenshot=true;
+}
+void RENDER_Init(Section * sec) {
+	Section_prop * section=static_cast<Section_prop *>(sec);
+	snapshots_dir=section->Get_string("snapshots");
 	render.pal.first=256;
 	render.pal.last=0;
-	render.enlarge=false;	
+	render.enlarge=false;
+	KEYBOARD_AddEvent(KBD_f5,CTRL_PRESSED,EnableScreenShot);
 }