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