yubioath-flutter/screenshot.h

#ifndef SCREENSHOT_H
#define SCREENSHOT_H
#include <QObject>
#include <QtWidgets>
#include <QVariant>
class ScreenShot: public QObject
{
    Q_OBJECT
public:
    explicit ScreenShot () : QObject() {}
    // Take a screenshot, convert it to a bitarray and return it with some metadata
    Q_INVOKABLE QVariantMap capture() {
        QScreen *screen = QGuiApplication::primaryScreen();
        // Virtual geometry grabs all pixels on the desktop, even with multiple monitors.
        QRect g = screen->virtualGeometry();
        QPixmap screenShot = screen->grabWindow(0, g.x(), g.y(), g.width(), g.height());
        // Monochrome, no dither
        QImage image = screenShot.toImage();
        image = image.convertToFormat(QImage::Format_Mono, Qt::ThresholdDither);
        // Get all pixels as 1 (black) or 0 (white)
        QByteArray imageArray;
        for (int row = 0; row < image.height(); ++row) {
            for (int col = 0; col < image.width(); ++col) {
                QRgb px = image.pixel(col, row);
                if (px == 0xFF000000) {
                    imageArray.append((char) 1);
                } else {
                    imageArray.append((char) 0);
                }
            }
        }
        QVariantMap map;
        map.insert("width", image.width());
        map.insert("height", image.height());
        map.insert("data", QString(imageArray.toBase64()));
        return map;
    }
};

#endif // SCREENSHOT_H