Qt跨平台截图工具原创

Qt跨平台截图工具

文章目录

Qt跨平台截图工具

更多精彩内容
👉个人内容分类汇总 👈
👉Qt自定义模块、工具👈

1、概述

Qt版本：V5.12.5
兼容系统：
- Windows：这里测试了Windows10，其它的版本没有测试；
- Linux：这里测试了ubuntu18.04、20.04，其它的没有测试(ubuntu自带的截图软件没有这个功能)；
- Mac：等啥时候我有了Mac电脑再说。
实现功能：

全屏截图功能，截取当前屏幕；
矩形截图功能，可通过鼠标选取矩形区域进行截图；
窗口截图功能，通过获取鼠标坐标位置的窗口，截取所在窗口的图像（由于ubuntu下自带的截图工具没有这个功能，就很烦）；
保存截图功能，将截取的图像保存到本地；
实时显示截取图像功能，由于我不喜欢Windows自带截图软件会随着图片大小自动变大，所以我的这个窗口不会自动变大。
功能比Windows、ubuntu自带的截图工具更强大，目前还没开发完成，快捷键之类的后续再加上。

2、实现效果

Windows下实现效果
Linux下实现效果

3、软件构成

截图软件主要由屏幕截取功能、矩形选区类、窗口选取类、自定义全局鼠标监听、截图显示类几部分组成。

4、关键代码

由于使用到了系统API，所以pro文件中需要链接系统库

win32 {
LIBS+= -luser32    # 使用WindowsAPI需要链接库
}
unix:!macx{
LIBS += -lX11      # linux获取窗口信息需要用到xlib
}

图像截图功能

/**
 * @brief       开始截图
 * @param rect  截取的范围
 */
void MainWindow::grabPixmap(QRect rect)
{
#if defined(Q_OS_WIN)
    setWindowOpacity(0);       // 最好的方法是将当前窗口设置成完全透明
    QDesktopWidget *desk = QApplication::desktop();         // 获取桌面根窗口
    QScreen * screen = QGuiApplication::primaryScreen();    // 获取默认屏幕
    m_pixmap = screen->grabWindow(desk->winId(), rect.x(), rect.y(), rect.width(), rect.height());          // 抓取屏幕图像
    ui->centralwidget->updatePixmap(m_pixmap);                     // 显示捕获的图像
    setWindowOpacity(1);
#elif defined(Q_OS_LINUX)
    // linux下setWindowOpacity设置透明后截图还可以看到一个透明的边框，效果不是很好，所以使用hide
    this->hide();            // 截图之前将当前窗口隐藏，避免截取的图像中包含当前窗口，这种方法很慢，需要延时等待几百毫秒，否则还是会有当前窗口
    QThread::msleep(300);
    QDesktopWidget *desk = QApplication::desktop();         // 获取桌面根窗口
    QScreen * screen = QGuiApplication::primaryScreen();    // 获取默认屏幕
    m_pixmap = screen->grabWindow(desk->winId(), rect.x(), rect.y(), rect.width(), rect.height());          // 抓取屏幕图像
    ui->centralwidget->updatePixmap(m_pixmap);                     // 显示捕获的图像
    this->show();            // 截图完成后显示窗口
#endif
}

截图保存功能

/**
 * @brief         保存截图
 * @param checked
 */
void MainWindow::on_saveImage(bool checked)
{
    Q_UNUSED(checked)
    if(m_pixmap.isNull()) return;

    // linux下getSaveFileName不会返回默认文件后缀，所以需要在文件名中添加后缀，否则QImage::save无法通过后缀推测出文件类型，就会保存失败
    QString name = QString("%1.png").arg(QDateTime::currentDateTime().toString("yyyy-MM-dd hh-mm-ss"));
    QString strName = QFileDialog::getSaveFileName(this, "保存到", name, "便携式网络图形(*.png);;JPEG文件(*.jpg)");
    if(strName.isEmpty()) return;

    QImage image = m_pixmap.toImage();
    if(image.save(strName))
    {
        qDebug() << "保存成功！";
    }
    else
    {
        QMessageBox::warning(this, "注意！", "文件保存失败，请检查有没有输入文件后缀名。");
    }
}

矩形选区功能

#include "screenrect.h"

#include <qapplication.h>
#include <qpainter.h>
#include <QDesktopWidget>
#include <qscreen.h>
#include <QDebug>
#include <QMouseEvent>

ScreenRect::ScreenRect(QWidget *parent) : QWidget(parent)
{
    this->setWindowFlag(Qt::FramelessWindowHint);           // 去除标题栏
    this->setAttribute(Qt::WA_TranslucentBackground);       // 背景透明
    QScreen * screen = QGuiApplication::primaryScreen();    // 获取默认屏幕
    m_path.addRect(screen->geometry());
    this->setGeometry(screen->geometry());
}

/**
 * @brief        鼠标选择的启点
 * @param event
 */
void ScreenRect::mousePressEvent(QMouseEvent *event)
{
    if(event->button() == Qt::LeftButton)
    {
        m_rect.setTopLeft(event->pos());
    }
    QWidget::mousePressEvent(event);
}

/**
 * @brief       鼠标释放时如果选择了区域则将选择的矩形发送出去
 * @param event
 */
void ScreenRect::mouseReleaseEvent(QMouseEvent *event)
{
    if((event->button() == Qt::LeftButton) && (m_rect.x() > 0 || m_rect.y() > 0))  // 这里使用x >0,y>0判断是否选择区域，而不是使用!m_rect.isEmpty()判断，避免反着选
    {
        this->hide();
        emit this->selectRect(m_rect.normalized());   // 使用normalized防止出现反着选，例如从下往上选，从右往左选
        m_rect = QRect();      // 置为空
    }

    QWidget::mouseReleaseEvent(event);
}

/**
 * @brief       鼠标选择的终点
 * @param event
 */
void ScreenRect::mouseMoveEvent(QMouseEvent *event)
{
    if(m_rect.x() > 0 || m_rect.y() > 0)
    {
        m_rect.setBottomRight(event->pos());
        this->update();
    }
    QWidget::mouseMoveEvent(event);
}

/**
 * @brief           绘制全屏遮罩和选择的矩形框
 * @param event
 */
void ScreenRect::paintEvent(QPaintEvent *event)
{
    QWidget::paintEvent(event);

    QPainter painter(this);
    painter.setPen(QPen(QColor(85, 170, 255, 200), 1));
    painter.setBrush(QColor(85, 170, 255, 50));
    QPainterPath path;
    path.addRect(m_rect);
    painter.drawPath(m_path - path);       // 绘制选择的矩形框
}

全局鼠标事件单例

/**
 *  全局鼠标事件单例信号类
 */
class MouseEvent : public QObject
{
    Q_OBJECT
public:
    static MouseEvent* getInstance()
    {
        static MouseEvent mouseEvent;
        return &mouseEvent;
    }

signals:
    void mouseSignal(QEvent* event);

private:
    MouseEvent(){}
};

窗口选区功能

#include "windowrect.h"
#include <QDebug>
#include <QGridLayout>
#include <QEvent>
#include <QMouseEvent>

#if defined(Q_OS_WIN)
#include <Windows.h>
#include <windef.h>
#elif defined(Q_OS_LINUX)
#include <X11/Xlib.h>
#include <X11/Xatom.h>
#elif defined(Q_OS_MAC)
#endif

#if defined(Q_OS_WIN)
static HHOOK g_hook = nullptr;
/**
 * @brief           处理鼠标事件的回调函数
 * @param nCode
 * @param wParam
 * @param lParam
 * @return
 */
LRESULT CALLBACK CallBackProc(int nCode, WPARAM wParam, LPARAM lParam)
{
    switch (wParam)
    {
    case WM_LBUTTONDOWN:   // 鼠标左键按下
    {
        emit MouseEvent::getInstance()->mouseSignal(new QMouseEvent(QEvent::MouseButtonPress, QCursor::pos(), Qt::LeftButton, Qt::LeftButton, Qt::NoModifier));
        break;
    }
    default:
        break;
    }
    return CallNextHookEx(nullptr, nCode, wParam, lParam);   // 注意这一行一定不能少，否则会出大问题
}
#endif

WindowRect::WindowRect(QWidget *parent) : QWidget(parent)
{
#if defined(Q_OS_WIN)
    // linux下鼠标穿透要放在后面两行代码的全前面，否则无效(但是鼠标穿透了会导致一些奇怪的问题，如窗口显示不全，所以这里不使用)
    // windows下如果不设置鼠标穿透则只能捕获到当前窗口
    this->setAttribute(Qt::WA_TransparentForMouseEvents, true);
#endif
    this->setWindowFlags(Qt::FramelessWindowHint);                            // 去掉边框、标题栏
    this->setAttribute(Qt::WA_TranslucentBackground);                         // 背景透明
    this->setWindowFlags(this->windowFlags() | Qt::WindowStaysOnTopHint);     // 设置顶级窗口，防止遮挡

    // 在当前窗口上增加一层QWidget，否则不会显示边框
    QGridLayout* gridLayout = new QGridLayout(this);
    gridLayout->setSpacing(0);
    gridLayout->setContentsMargins(0, 0, 0, 0);
    gridLayout->addWidget(new QWidget(), 0, 0, 1, 1);

    this->setStyleSheet(" background-color: rgba(58, 196, 255, 40); border: 2px solid rgba(58, 196, 255, 200);"); // 设置窗口边框样式 dashed虚线，solid 实线

    connect(MouseEvent::getInstance(), &MouseEvent::mouseSignal, this, &WindowRect::on_mouseSignal);
    // 使用定时器定时获取当前鼠标位置的窗口位置信息
    connect(&m_timer, &QTimer::timeout, this, &WindowRect::on_timeout);
    m_timer.start(200);
}

WindowRect::~WindowRect()
{
#if defined(Q_OS_WIN)
    if(g_hook)
    {
        bool ret = UnhookWindowsHookEx(g_hook);
        if(ret)
        {
            qDebug() << "卸载鼠标钩子。";
        }
    }
#endif
}

/**
 * @brief         通过截图全局鼠标事件将当前窗口大小发生出去
 * @param event
 */
void WindowRect::on_mouseSignal(QEvent *event)
{
    delete event;
    this->hide();
    emit this->selectRect(QRect(this->pos(), this->size()));
}

/**
 * @brief        Windows使用全局鼠标钩子，显示窗口时挂载鼠标钩子
 * @param event
 */
void WindowRect::showEvent(QShowEvent *event)
{
#if defined(Q_OS_WIN)
    // 由于windows不透明的窗体如果不设置设置鼠标穿透WindowFromPoint只能捕捉到当前窗体，而设置鼠标穿透后想要获取鼠标事件只能通过鼠标钩子
    g_hook = SetWindowsHookExW(WH_MOUSE_LL, CallBackProc, GetModuleHandleW(nullptr), 0);  // 挂载全局鼠标钩子
    if (g_hook)
    {
        qDebug() << "鼠标钩子挂接成功,线程ID:" << GetCurrentThreadId();

    }
    else
    {
        qDebug() << "鼠标钩子挂接失败:" << GetLastError();
    }
#endif
    QWidget::showEvent(event);
}

/**
 * @brief       隐藏窗口时卸载鼠标钩子
 * @param event
 */
void WindowRect::hideEvent(QHideEvent *event)
{
#if defined(Q_OS_WIN)
    if(g_hook)
    {
        bool ret = UnhookWindowsHookEx(g_hook);
        if(ret)
        {
            qDebug() << "卸载鼠标钩子。";
            g_hook = nullptr;
        }
    }
#endif
    QWidget::hideEvent(event);
}

/**
 * @brief        linux下使用自带的鼠标点击事件就可以
 * @param event
 */
void WindowRect::mousePressEvent(QMouseEvent *event)
{
#if defined(Q_OS_LINUX)
    this->hide();
    emit this->selectRect(QRect(this->pos(), this->size()));
#endif
    QWidget::mousePressEvent(event);
}


void WindowRect::on_timeout()
{
    QPoint point = QCursor::pos();  // 获取鼠标当前位置
#if defined(Q_OS_WIN)
    POINT pos;
    pos.x = point.x();
    pos.y = point.y();

    HWND hwnd = nullptr;
    hwnd = WindowFromPoint(pos);   // 通过鼠标位置获取窗口句柄
    if(!hwnd) return;

    RECT lrect;
    bool ret = GetWindowRect(hwnd, &lrect);     //获取窗口位置
    if(!ret) return;

    QRect rect;
    rect.setX(lrect.left);
    rect.setY(lrect.top);
    rect.setWidth(lrect.right - lrect.left);
    rect.setHeight(lrect.bottom - lrect.top);
    this->setGeometry(rect);         // 设置窗口边框
#elif defined(Q_OS_LINUX)  // linux下使用x11获取的窗口大小有可能不太准确，例如浏览器的大小会偏小
    // 获取根窗口
    Display* display = XOpenDisplay(nullptr);
    Window rootWindow = DefaultRootWindow(display);

    Window root_return, parent_return;
    Window * children = nullptr;
    unsigned int nchildren = 0;
    // 函数详细说明见xlib文档：https://tronche.com/gui/x/xlib/window-information/XQueryTree.html
    // 该函数会返回父窗口的子窗口列表children，因为这里用的是当前桌面的根窗口作为父窗口，所以会返回所有子窗口
    // 注意：窗口顺序(z-order)为自底向上
    XQueryTree(display, rootWindow, &root_return, &parent_return, &children, &nchildren);

    QRect recte;                      // 保存鼠标当前所在窗口的范围
    for(unsigned int i = 0; i < nchildren; ++i)
    {
        if(children[i] == this->winId()) continue;           // 由于当前窗口一直在最顶层，所以这里要过滤掉当前窗口，否则一直获取到的就是当前窗口大小
        XWindowAttributes attrs;
        XGetWindowAttributes(display, children[i], &attrs);  // 获取窗口参数
        if (attrs.map_state == IsViewable)                   // 只处理可见的窗口， 三个状态：IsUnmapped, IsUnviewable, IsViewable
        {
#if 0
            QRect rect(attrs.x + 1, attrs.y, attrs.width, attrs.height);  // 这里x+1防止全屏显示，如果不+1，设置的大小等于屏幕大小是会自动切换成全屏显示状态，后面就无法缩小了
#else
            QRect rect(attrs.x, attrs.y, attrs.width, attrs.height);
#endif
            if(rect.contains(point))                        // 判断鼠标坐标是否在窗口范围内
            {
                recte = rect;                               // 记录最后一个窗口的范围
            }
        }
    }
#if 0  // 在linux下使用setGeometry设置窗口会有一些问题
    this->showNormal();               // 第一次显示是如果是屏幕大小，则后面无法缩小，这里需要设置还原
    this->setGeometry(recte);         // 设置窗口边框
#else  // 使用setFixedSize+move可以避免这些问题
    this->move(recte.x(), recte.y());
    this->setFixedSize(recte.width(), recte.height());
#endif
//    qDebug() << this->rect() <<recte<< this->windowState();

    // 注意释放资源
    XFree(children);
    XCloseDisplay(display);

#elif defined(Q_OS_MAC)
#endif
}

5、源代码

💡💡💡💡💡💡💡💡💡💡💡💡💡💡

posted @ 2023-11-07 23:02 mahuifa 阅读(0) 评论(0) 编辑收藏举报来源

刷新页面返回顶部

mahuifa

Qt跨平台截图工具 原创

Qt跨平台截图工具

文章目录

1、概述

2、实现效果

3、软件构成

4、关键代码

5、源代码

公告

Qt跨平台截图工具原创