From 5f457fd9e3eb32b1beebc26921e4963e1935742c Mon Sep 17 00:00:00 2001
From: "T. van der Zwan" <zwantvd@gmail.com>
Date: Thu, 6 Mar 2014 21:01:14 +0100
Subject: [PATCH] Added qt screenshot test

Former-commit-id: c28e098ded780a6597952c83007fea1fe8da36fa
---
 test/TestQtScreenshot.cpp   | 56 +++++++++++++++++++++++++++++++++++--
 test/TestX11Performance.cpp |  4 +--
 2 files changed, 56 insertions(+), 4 deletions(-)

diff --git a/test/TestQtScreenshot.cpp b/test/TestQtScreenshot.cpp
index 27e03b0f..d2bed9e7 100644
--- a/test/TestQtScreenshot.cpp
+++ b/test/TestQtScreenshot.cpp
@@ -6,15 +6,67 @@
 #include <QApplication>
 #include <QDesktopWidget>
 #include <QPixmap>
+#include <Qfile>
+#include <QRgb>
+
+#include <QElapsedTimer>
+
+// Utils includes
+#include <utils/Image.h>
+#include <utils/ColorRgb.h>
+
+void createScreenshot(const int cropHorizontal, const int cropVertical, const int decimation, Image<ColorRgb> & image)
+{
+	// Create the full size screenshot
+	const QRect screenSize = QApplication::desktop()->screenGeometry();
+	const int croppedWidth  = screenSize.width()  - 2*cropVertical;
+	const int croppedHeight = screenSize.height() - 2*cropHorizontal;
+	const QPixmap fullSizeScreenshot = QPixmap::grabWindow(QApplication::desktop()->winId(), cropVertical, cropHorizontal, croppedWidth, croppedHeight);
+
+	// Scale the screenshot to the required size
+	const int width  = fullSizeScreenshot.width()/decimation;
+	const int height = fullSizeScreenshot.height()/decimation;
+	const QPixmap scaledScreenshot = fullSizeScreenshot.scaled(width, height, Qt::IgnoreAspectRatio, Qt::FastTransformation);
+
+	// Convert the QPixmap to QImage in order to get out RGB values
+	const QImage qImage = scaledScreenshot.toImage();
+
+	// Make sure that the output image has the right size
+	image.resize(width, height);
+
+	// Copy the data into the output image
+	for (int y=0; y<qImage.height(); ++y)
+	{
+		for (int x=0; x<qImage.width(); ++x)
+		{
+			// Get the pixel at [x;y] (format int32 #AARRGGBB)
+			const QRgb inPixel = qImage.pixel(x,y);
+
+			// Copy the color channels into the output pixel
+			ColorRgb & outPixel = image(x,y);
+			outPixel.red   = (inPixel & 0x00ff0000) >> 16;
+			outPixel.green = (inPixel & 0x0000ff00) >>  8;
+			outPixel.blue  = (inPixel & 0x000000ff);
+		}
+	}
+}
 
 int main(int argc, char** argv)
 {
+	int decimation = 10;
 
 	QApplication app(argc, argv);
+	QElapsedTimer timer;
 
-	QPixmap originalPixmap = QPixmap::grabWindow(QApplication::desktop()->winId());
+	Image<ColorRgb> screenshot(64,64);
 
-	std::cout << "Grabbed image: [" << originalPixmap.width() << "; " << originalPixmap.height() << "]" << std::endl;
+	int loopCnt = 100;
+	timer.start();
+	for (int i=0; i<loopCnt; ++i)
+	{
+		createScreenshot(0,0, decimation, screenshot);
+	}
+	std::cout << "Time required for single screenshot: " << timer.elapsed()/loopCnt << "ms" << std::endl;
 
 	return 0;
 }
diff --git a/test/TestX11Performance.cpp b/test/TestX11Performance.cpp
index 9b7fb402..b29e0587 100644
--- a/test/TestX11Performance.cpp
+++ b/test/TestX11Performance.cpp
@@ -67,7 +67,7 @@ void foo_1(int pixelDecimation)
 	// Cleanup allocated resources of the X11 grab
 	XDestroyImage(xImage);
 
-	std::cout << "Time required: " << timer.elapsed() << " us" << std::endl;
+	std::cout << "Time required: " << timer.elapsed() << " ms" << std::endl;
 
 	XCloseDisplay(x11Display);
 }
@@ -125,7 +125,7 @@ void foo_2(int pixelDecimation)
 			XDestroyImage(xImage);
 		}
 	}
-	std::cout << "Time required: " << timer.elapsed() << " us" << std::endl;
+	std::cout << "Time required: " << timer.elapsed() << " ms" << std::endl;
 
 
 	XCloseDisplay(x11Display);