140 lines
4.1 KiB
C++
140 lines
4.1 KiB
C++
|
|
#include <QElapsedTimer>
|
|
|
|
#include <utils/Image.h>
|
|
#include <utils/ColorRgb.h>
|
|
|
|
// X11 includes
|
|
#include <X11/Xlib.h>
|
|
#include <X11/Xutil.h>
|
|
|
|
void foo_1(int pixelDecimation)
|
|
{
|
|
int cropWidth = 0;
|
|
int cropHeight = 0;
|
|
|
|
Image<ColorRgb> image(64, 64);
|
|
|
|
/// Reference to the X11 display (nullptr if not opened)
|
|
Display * x11Display;
|
|
|
|
const char * display_name = nullptr;
|
|
x11Display = XOpenDisplay(display_name);
|
|
|
|
std::cout << "Opened display: " << x11Display << std::endl;
|
|
|
|
XWindowAttributes window_attributes_return;
|
|
XGetWindowAttributes(x11Display, DefaultRootWindow(x11Display), &window_attributes_return);
|
|
|
|
int screenWidth = window_attributes_return.width;
|
|
int screenHeight = window_attributes_return.height;
|
|
std::cout << "[" << screenWidth << "x" << screenHeight <<"]" << std::endl;
|
|
|
|
// Update the size of the buffer used to transfer the screenshot
|
|
int width = (screenWidth - 2 * cropWidth + pixelDecimation/2) / pixelDecimation;
|
|
int height = (screenHeight - 2 * cropHeight + pixelDecimation/2) / pixelDecimation;
|
|
image.resize(width, height);
|
|
|
|
const int croppedWidth = screenWidth - 2*cropWidth;
|
|
const int croppedHeight = screenHeight - 2*cropHeight;
|
|
|
|
QElapsedTimer timer;
|
|
timer.start();
|
|
|
|
XImage * xImage = XGetImage(x11Display, DefaultRootWindow(x11Display), cropWidth, cropHeight, croppedWidth, croppedHeight, AllPlanes, ZPixmap);
|
|
|
|
std::cout << "Captured image: " << xImage << std::endl;
|
|
|
|
// Copy the capture XImage to the local image (and apply required decimation)
|
|
ColorRgb * outputPtr = image.memptr();
|
|
for (int iY=(pixelDecimation/2); iY<croppedHeight; iY+=pixelDecimation)
|
|
{
|
|
for (int iX=(pixelDecimation/2); iX<croppedWidth; iX+=pixelDecimation)
|
|
{
|
|
// Extract the pixel from the X11-image
|
|
const uint32_t pixel = uint32_t(XGetPixel(xImage, iX, iY));
|
|
|
|
// Assign the color value
|
|
outputPtr->red = uint8_t((pixel >> 16) & 0xff);
|
|
outputPtr->green = uint8_t((pixel >> 8) & 0xff);
|
|
outputPtr->blue = uint8_t((pixel >> 0) & 0xff);
|
|
|
|
// Move to the next output pixel
|
|
++outputPtr;
|
|
}
|
|
}
|
|
|
|
// Cleanup allocated resources of the X11 grab
|
|
XDestroyImage(xImage);
|
|
|
|
std::cout << "Time required: " << timer.elapsed() << " ms" << std::endl;
|
|
|
|
XCloseDisplay(x11Display);
|
|
}
|
|
|
|
void foo_2(int pixelDecimation)
|
|
{
|
|
int cropWidth = 0;
|
|
int cropHeight = 0;
|
|
|
|
Image<ColorRgb> image(64, 64);
|
|
|
|
/// Reference to the X11 display (nullptr if not opened)
|
|
Display * x11Display;
|
|
|
|
const char * display_name = nullptr;
|
|
x11Display = XOpenDisplay(display_name);
|
|
|
|
XWindowAttributes window_attributes_return;
|
|
XGetWindowAttributes(x11Display, DefaultRootWindow(x11Display), &window_attributes_return);
|
|
|
|
int screenWidth = window_attributes_return.width;
|
|
int screenHeight = window_attributes_return.height;
|
|
std::cout << "[" << screenWidth << "x" << screenHeight <<"]" << std::endl;
|
|
|
|
// Update the size of the buffer used to transfer the screenshot
|
|
int width = (screenWidth - 2 * cropWidth + pixelDecimation/2) / pixelDecimation;
|
|
int height = (screenHeight - 2 * cropHeight + pixelDecimation/2) / pixelDecimation;
|
|
image.resize(width, height);
|
|
|
|
const int croppedWidth = screenWidth - 2*cropWidth;
|
|
const int croppedHeight = screenHeight - 2*cropHeight;
|
|
|
|
QElapsedTimer timer;
|
|
timer.start();
|
|
|
|
// Copy the capture XImage to the local image (and apply required decimation)
|
|
ColorRgb * outputPtr = image.memptr();
|
|
for (int iY=(pixelDecimation/2); iY<croppedHeight; iY+=pixelDecimation)
|
|
{
|
|
for (int iX=(pixelDecimation/2); iX<croppedWidth; iX+=pixelDecimation)
|
|
{
|
|
XImage * xImage = XGetImage(x11Display, DefaultRootWindow(x11Display), iX, iY, 1, 1, AllPlanes, ZPixmap);
|
|
// Extract the pixel from the X11-image
|
|
const uint32_t pixel = uint32_t(XGetPixel(xImage, 0, 0));
|
|
|
|
// Assign the color value
|
|
outputPtr->red = uint8_t((pixel >> 16) & 0xff);
|
|
outputPtr->green = uint8_t((pixel >> 8) & 0xff);
|
|
outputPtr->blue = uint8_t((pixel >> 0) & 0xff);
|
|
|
|
// Move to the next output pixel
|
|
++outputPtr;
|
|
|
|
// Cleanup allocated resources of the X11 grab
|
|
XDestroyImage(xImage);
|
|
}
|
|
}
|
|
std::cout << "Time required: " << timer.elapsed() << " ms" << std::endl;
|
|
|
|
|
|
XCloseDisplay(x11Display);
|
|
}
|
|
|
|
int main()
|
|
{
|
|
foo_1(10);
|
|
foo_2(10);
|
|
return 0;
|
|
}
|