浅谈基于QT的截图工具的设计与实现

2023-10-18 18:45:58 浏览数 (2)

前言:QT绘图基础

在介绍截图工具设计与实现前,让我们先通过介绍QT的绘图基础知识,让读者有一个比较感性的认识。

本文理论上并非是完整的QT框架使用介绍,但是我们总是需要用一款支持绘图的GUI框架来介绍关于截图的知识,于是笔者就拿较为熟悉的QT框架来说明。但只要读者理解到了截图工具的本质,举一反三,其它的GUI框架也能完成截图的目的。

对于绘图来说,我们通常遵循“数据驱动渲染的模型。具体一点,我们会围绕数据展开绘图,图像的绘制总是来源于数据的定义。那么如何实现动态图形呢?只需要通过某些操作改变数据即可。

这样的模型,数据的修改和数据的渲染是解耦的,我们编写处理绘图部分的时候,只需要根据已有的数据进行绘制,可以完全不用关心数据是怎么变化的;而当操作数据的时候,完全可以不用关心渲染部分。基于该模型,可以让我们在开发类似于截图软件的时候,极大降低心智负担。

回到实际的部分,我们先使用QT编写一个窗体widget,然后重写窗体的paintEvent方法:

代码语言:javascript复制
class DemoWidget: public QWidget {
public:
  void paintEvent(QPaintEvent *event) override {
    QPainter painter(this);
    painter.setPen(QPen(Qt::red));
    painter.drawRect(10, 10, 100, 60);
  }
};

paintEvent函数体代码就三行:

  1. 使用当前窗体指针构造一个QPainter(QPainter painter(this););
  2. 设置画笔的颜色;
  3. 在坐标(10, 10)处绘制一个宽100像素,高60像素的矩形。

然后,我们编写main方法,创建这个DemoWidget类的实例,将它show出来:

代码语言:javascript复制
int main(int argc, char *argv[]) {
  QApplication a(argc, argv);
  DemoWidget w;
  w.resize(200, 100);
  w.show();
  return QApplication::exec();
}

整体代码和运行效果如下:

没错,QT中在一个窗体中进行绘图就是这么简单。接下来让我们更进一步,将矩形数据(x,y,w,h)提升到到类成员变量层级,并让painter绘制矩形的时候读取类成员变量:

代码语言:javascript复制
class DemoWidget: public QWidget {
public:
  void paintEvent(QPaintEvent *event) override {
    QPainter painter(this);
    painter.setPen(QPen(Qt::red));
-   painter.drawRect(10, 10, 100, 60);
    painter.drawRect(x_, y_, w_, h_); // 读取类成员变量
  }
  private:
   int x_ = 10, y_ = 10, w_ = 100, h_ = 60;
};

然后,我们重写QWidget的onKeyPress事件,代码如下:

代码语言:javascript复制
void keyPressEvent(QKeyEvent *event) override {
  auto key = event->key();
  switch (key) {
    case Qt::Key_Up: y_ -= 5;
      break;
    case Qt::Key_Down: y_  = 5;
      break;
    case Qt::Key_Left: x_ -= 5;
      break;
    case Qt::Key_Right: x_  = 5;
      break;
    default:break;
  }
}

这段代码的作用是当我们按下方向键后,就能够修改x_y_变量的值,于是矩形的xy坐标会按照对应方向移动5像素。理论上讲,如果此时触发绘图事件,而我们使用painter又在读取类成员变量x_y_等数据进行矩形绘制,那么就会看到矩形跟随方向键在上下左右移动。

然而,当我们操作时候却发现无论怎么按方向键界面似乎没有任何反应:

为什么呢?让我们引入qdebug向控制台输出一些信息一探究竟:

应用运行以后,通过QDebug,我们可以在调试模式下看到控制台的输出内容:

通过控制台可以看到,一开始触发了几次绘图事件(paintEvent)。之后,当我们按下方向键时,触发了按键事件(keyPressEvent),此时x_y_的值的确已经发生了改变,但是控件上的矩形没有任何的变化。实际上,造成这种问题的根本原因在于我们重写的绘图事件没有触发,于是导致最新的效果并没有绘制到界面上,所以看不出效果。

那么,QT的绘图事件什么时候触发呢?大致会有一下几种情况:

  1. 当控件第一次显示时,系统会自动产生一个绘图事件。比如上面的动图中第一次的paintEvent
  2. 窗体失去焦点,获得焦点等,之后几次paintEvent出发就是因此产生的。
  3. 当窗口控件被其他部件遮挡,然后又显示出来时,会对隐藏的区域产生一个重绘事件。比如最小化再出现。
  4. 重新调整窗口大小时。
  5. repaint()update()函数被调用时。

上面的例子中,在按下方向键以后界面没有效果,如果此时我们最小化它再恢复它,就会看到绘图事件被触发,同时界面也有所改变:

当然,我们不可能为了触发绘图事件而手动操作窗体。为了达到触发绘图事件的目的,我们一般会调用控件的update方法系列方法或repaint的系列方法,来主动告诉QT需要进行控件的重新绘制,进而让QT触发paintEvent,绘制界面:

再次运行程序,并按下方向键,我们可以清楚的看到paintEvent在每次按下方向键以后都被调用,同时,矩形也表现出移动的效果:

这里我们调用的是update方法,同时,我们还提到QT还提供一个repaint方法,二者区别在于:repaint一旦调用,QT内部就会立刻调用触发paintEvent,而update只是将触发绘图事件的任务放到事件队列,等统一事件调用。所以,绝对不能在paintEvent中调用repaint,这样会死循环。

此外使用update还有一个优点在于,QT会将多个update的请求通过算法机制尽可能的合并为一个paintEvent,从而提高运行的效率。比如,我们可以在调用update的地方多赋值几次调用:

在实际调用中,只会触发一次paintEvent

如果换成调用5次repaint就会发现每调用一次就会触发一次paintEvent,读者可以自行测试。

正文:截图思路

在介绍了QT绘图基础以后,我们终于可以开始讨论正题了:截图工具的设计与实现。实际上,截图工具实现起来并不复杂。可以想象一下,我们首先通过某种API获取到桌面屏幕的图片,然后把这个图片放到一个窗体里面,最后再把这个窗体最大化的方式展现在屏幕上。此时就达到了我们截取了屏幕并让整个屏幕“冻结”,等待我们操作的效果。

此时窗体全屏幕覆盖,接下来我们就需要在上面进行某个区域的获取。

PS:这个动图使用了跨平台视频剪辑工具Kdenlive制作,并转为gif,有空写一个教程,哈哈。

区域截取状态

一般来说,截图过程就是按下鼠标,然后移动鼠标,此时界面上会显示整个鼠标拖拽产生的一个区域,直到松开鼠标,这个区域就被“截取”下来了:

想要实现这样的效果并不复杂,代码如下何解释如下:

在上图代码中我分别标注了两个部分:

  1. 捕获指定区域所需要的数据;
  2. 将指定数据转化为图形进行绘制。

首先讲解第一部分:捕获指定区域所需要的数据。这里我使用了三组数据,分别是:鼠标按下的起始位置、鼠标当前的位置、是否处于捕获中状态。不难看出,只需要这三组数据,我们就可以描述这样一个画面:如果没有在捕获状态,那么界面上不会出现矩形;如果处于捕获状态,那么我们使用起始位置和当前位置得到一个矩形:

paintEvent中的代码实现也正是如此:

代码语言:javascript复制
void paintEvent(QPaintEvent *event) override {
  if (!isCapturing) {
    return;
  }
  QPainter painter(this);
  painter.setPen(QPen(Qt::red));
  int w = abs(currX - startX);
  int h = abs(currY - startY);
  painter.drawRect(startX, startY, w, h);
}

也就是说,按照数据驱动渲染的模型,我们完成了由数据到渲染的部分:

接下来,我们完全只需要关注如何修改数据即可。在本例中,我们的操作行为是按下鼠标开始截取区域,移动过程中界面绘制开始点和当前鼠标构成的矩形,松开鼠标完成区域截取。很明显,我们会利用到鼠标事件。在QT中提供了三个鼠标事件供我们使用:

  1. mousePresssEvent,鼠标按下事件;
  2. mouseReleaseEvent,鼠标松开事件;
  3. mouseMoveEvent,鼠标移动事件。

当我们按下鼠标的时候,就进入了“捕获状态”(isCapturing置为true),并且记录鼠标此时按下的位置(startXstartY);在鼠标移动过程中,不断的更新当前鼠标位置(设置currXcurrY);松开鼠标时就退出“捕获状态”(isCapturing置为false)。代码如下:

代码语言:javascript复制
void mousePressEvent(QMouseEvent *event) override {
  isCapturing = true;
  startX = event->pos().x();
  startY = event->pos().y();
  this->update();
}
void mouseReleaseEvent(QMouseEvent *event) override {
  isCapturing = false;
  this->update();
}
void mouseMoveEvent(QMouseEvent *event) override {
  auto pos = event->pos();
  currX = pos.x();
  currY = pos.y();
  this->update();
}

注意事项1:这里每一个操作都要调用update告诉QT需要触发绘图事件,否则你会发现界面上没有任何的动静。另外,怎么知道什么时候应该调用update方法呢?很简单,只要在某处的代码修改了paintEvent中所依赖的数据,就应该在之后调用update

注意事项2:在QT中,mouseMoveEvent并不是随时都在触发,该事件默认只有在鼠标按下以后的移动过程才会触发,QT这样设计考虑的点是因为鼠标的移动是很频繁的,随时触发会降低性能。如果你在某些场景下就是需要随时出发移动事件,需要在控件的构造函数中调用"setMouseTracking(true);"(可以看代码清单图中11行)。

区域捕获到这里就结束了吗?非也。让我们来演示上面代码的问题:

很明显可以看到,当我们将鼠标向右下拖动的时候,矩形很正常的在动态显示,而向左上角拖动的时候,就出现了问题。原因在于,QT的drawRect等API绘制矩形的时候,位置参数总是矩形的左上角位置,而我们总是将鼠标按下的位置作为左上角位置。然而,鼠标按下的位置就应该是矩形的左上角吗?不总是。当我们拖动鼠标向右下角移动的时候,左上角的start位置确实是可以作为矩形的xy坐标。但一旦我们将鼠标移动到左上角,位于起始位置的左边和上边的时候,就应该用当前鼠标的位置作为矩形的左上角了:

于是,我们需要适当修改以下paintEvent中的代码:

代码语言:javascript复制
  void paintEvent(QPaintEvent *event) override {
    if (!isCapturing) {
      return;
    }
    QPainter painter(this);
    painter.setPen(QPen(Qt::red));
    int w = abs(currX - startX);
    int h = abs(currY - startY);
    int left = startX < currX ? startX : currX;
    int top = startY < currY ? startY : currY;
-   painter.drawRect(startX, startY, w, h);
    painter.drawRect(left, top, w, h);
  }

就能看到合适的效果了:

捕获完成状态与整体流转

一般截图工具都会在我们松开鼠标的时候,将被截取的区域固定下来,然后我们可以在上面写写画画(譬如添加额外的标记、文字等)。为了达到这个目的,我们首先要考虑如何将一个区域“固定”下来。在前面,我们引入了一个状态:“是否正在捕获中”(使用isCapturing作为标记)。在这里,为了描述“区域截取完成之后”的情形,我们需要引入一个新的状态:截取完成。于是,在整个截图操作的过程中,我们的状态流转如下:

为了后续代码更好的设计,我们使用枚举来表达状态:

代码语言:javascript复制
enum Status {
  Explore = 0,
  Capturing,
  Captured
};

这里的Status::CapturingStatus::Captured不必多说,要单独解释一下Explore单词的含义。实际上,Explore就是指上面的“默认”,只是在笔者看来,当我们还没有进行截图的时候,鼠标就是在整个窗口上移动“探索”,所以笔者将这个状态取名为Explore

然后,我们需要对现有的代码进行适当的修改。首先是成员变量,由于我们引入了枚举来表达截图的状态,所以原先isCapturing字段就可以舍弃,取而代之的是使用枚举并默认为Status::Explore。同时,我们还需要引入一个矩形数据变量,来存储当我们松开鼠标的时候,截取到的区域的矩形信息。于是变动如下:

代码语言:javascript复制
private:
  int startX = 0, startY = 0;
  int currX = 0, currY = 0;
- bool isCapturing;
  QRect capturedRect; // 存储截取的区域信息,这里使用QT的QRect类
  Status status = Explore; // 替换原有的bool,并默认为Explore状态

对于数据的定义发生了变化,我们优先考虑渲染部分的变化,也就是paintEvent需要做出适配。正对不同的状态,paintEvent会绘制不同的效果:

  1. Explore态,我们认为界面上什么操作也没有,所以什么都不需要做;
  2. Capturing态,其实就是我们上面isCapturingtrue的处理;
  3. Captured态,截取完成后,我们把截取到的区域用蓝色矩形框住,而矩形数据就是上面新增的成员变量capturedRect

于是,整个代码如下:

代码语言:javascript复制
void paintEvent(QPaintEvent *event) override {
  if (status == Explore) {
    return;
  }
  if (status == Capturing) {
    QPainter painter(this);
    painter.setPen(QPen(Qt::red));
    int w = abs(currX - startX);
    int h = abs(currY - startY);
    int left = startX < currX ? startX : currX;
    int top = startY < currY ? startY : currY;
    painter.drawRect(left, top, w, h);
    return;
  }
  if (status == Captured) {
    QPainter painter(this);
    painter.setPen(QPen(Qt::blue));
    painter.drawRect(capturedRect);
    return;
  }
}

同样的,考虑完了数据以及如何绘制以后,我们需要回到模型的“数据操作”部分,考虑这些数据是如何变化的。按照上面的"默认" -> "截图中" -> "截图后"状态流转图,我们就可以很轻易写出数据修改的代码。

首先是鼠标按下事件。当鼠标按下的时候,如果我们处于Explore,那么就进入Capturing并记录鼠标起始位置;如果处于Captured,那么就什么也不干(理论上是不会有Capturing情况下的鼠标按下事件的),代码如下:

代码语言:javascript复制
void mousePressEvent(QMouseEvent *event) override {
  switch (status) {
    case Explore: {
      status = Capturing; // 进入Capturing
      startX = event->pos().x();
      startY = event->pos().y();
      break;
    }
    default:break;  // 其余状态都不关心
  }
  this->update();
}

接着是鼠标松开事件。当鼠标松开的时候,如果是Explore(理论上是不会出现的)或Captured,就什么也不做;如果是Capturing,则进行Captured状态,同时要存储下此时截取的区域,代码如下:

代码语言:javascript复制
void mouseReleaseEvent(QMouseEvent *event) override {
  switch (status) {
    case Capturing: {
      // 进入Captured态
      status = Captured;
      // 保存区域
      int w = abs(currX - startX);
      int h = abs(currY - startY);
      int left = startX < currX ? startX : currX;
      int top = startY < currY ? startY : currY;
      capturedRect = QRect(left, top, w, h);
      break;
    }
    default: break;
  }
  this->update();
}

然后是鼠标移动过程的状态处理。如果是Explore或是Captured,那么什么也不做;如果是Capturing,那么不断更新当前鼠标位置,代码如下:

代码语言:javascript复制
void mouseMoveEvent(QMouseEvent *event) override {
  switch (status) {
    case Capturing: {
      auto pos = event->pos();
      currX = pos.x();
      currY = pos.y();
      break;
    }
    default:break;
  }
  this->update();
}

此时,我们还差一个将状态从Caputred切回到Explore的处理,我们重写keyPressEvent事件的,如果在Captured状态按下了ECS,就进入Explore态:

代码语言:javascript复制
void keyPressEvent(QKeyEvent *event) override {
  if (event->key() == Qt::Key_Escape) {
    status = Explore;
  }
  this->update();
}

在所有代码准备好以后,让我们启用应用看一下效果:

细心的读者如果实践到此处,会发现一个小问题:每一次按下ESC键以后,下一次进入Capturing状态,在鼠标拖动开始的一瞬间,会有一个矩形框闪现,原因是currXcurrY还是上一次的数据,没有即时清理。解决办法也比较简单,就是在按下的一瞬间,同时更新startcurr的坐标数据为同一位置即可:

代码语言:javascript复制
void mousePressEvent(QMouseEvent *event) override {
  switch (status) {
    case Explore: {
      status = Capturing; // 进入Capturing
      startX = event->pos().x();
      startY = event->pos().y();
      currX = startX; // 同时更新start和curr
      currY = startY;
      break;
    }
    default:break;  // 其余状态都不关心
  }
  this->update();
}

完成图像截取

终于,我们还剩最后一步了,就是截取这个区域的图像。在之前的介绍中,我们一直在一个空白的窗体上进行绘图。在本节,我们将通过QT的API,来获取当前鼠标所在的屏幕图像,并把图像作为这个窗体的背景图。然后,我们照旧在上面进行区域的截取,来达到所谓的屏幕截图的效果。

首先,我们需要做一些准备工作:

准备工作以下几步:

  1. DemoWidget类中定义一个QImage的指针类成员变量;
  2. 修改构造函数,让外部传入这个QImage实例指针并进行存储;
  3. 调用如下QT提供的相关API来获取屏幕图像:
代码语言:javascript复制
// 获取鼠标所在屏幕
QScreen *screen = QApplication::screenAt(QCursor().pos());
// 获取屏幕的图像数据
QImage screenImg = screen->grabWindow(0).toImage();
  1. 我们将screenImg的地址作为指针变量作为DemoWidget的构造函数入参传入。

图像的获取与存储完成以后,我们将会在paintEvent中,优先绘制屏幕图像,然后才根据状态来绘制对应的矩形:

于是,界面运行以后,我们就能看屏幕截图填充在窗口里面的效果:

接下来,我们增加一种操作:当处于屏幕截取完成的状态(Captured)的时候,只要按下回车键,就能将截取的屏幕保存到粘贴板中,并回到Explore状态。很自然的,我们需要在keyPressEvent新增关于该操作的代码:

代码语言:javascript复制
void keyPressEvent(QKeyEvent *event) override {
  if (event->key() == Qt::Key_Return && status == Captured) {
    // 1. 获取捕获的图像区域
    // 2. 从保存的屏幕图像中获取指定区域的图像数据
    // 3. 将图像数据写入到操作系统粘贴板
    // 4. 回到Explore
    return;
  }
  if (event->key() == Qt::Key_Escape) {
    status = Explore;
  }
  this->update();
}

注意,QT中回车键的枚举值是Key_Return,不是Key_Enter。

对于步骤1,我们在前文已经使用capturedRect类成员变量保存了当区域截取完成以后的区域数据;

对于步骤2,QImage有一个名为copy的方法:

代码语言:javascript复制
[[nodiscard]] QImage copy(int x, int y, int w, int h) const;

它可以从已有的图像中复制指定区域的图像,得到一个新的图像数据;

对于步骤3,我们可以使用QT提供的QClipboard类来操作系统粘贴板。于是,你可以这样调用来将图像数据保存到粘贴板中:

代码语言:javascript复制
QClipboard *clipboard = QGuiApplication::clipboard();
clipboard->setImage(/* QImage对象 */);

对于步骤4就比较简单了,切换status的状态为Explore即可。

按照上面的过程描述,我们编写如下的代码:

代码语言:javascript复制
void keyPressEvent(QKeyEvent *event) override {
  if (event->key() == Qt::Key_Enter && status == Captured) {
    // 1. 获取捕获的图像区域
    auto imgRect = this->capturedRect;
    // 2. 从保存的屏幕图像中获取指定区域的图像数据
    auto copiedImg = this->screenImg->copy(imgRect);
    // 3. 将图像数据写入到操作系统粘贴板
    QClipboard *clipboard = QGuiApplication::clipboard();
    clipboard->setImage(copiedImg);
    // 4. 回到Explore
    status = Explore;
    return;
  }
  // 其余代码 ... ...
}

当我们兴致勃勃的运行应用并进行截图操作的时候,会发现在粘贴板中的图像,和我们截取的区域不太一致!

注意,我们截取了右下角有紫蓝色的区域,但是实际获取的图像却不是。这个问题的核心原因是,我们截取的capturedRect是这个窗体界面上的区域,但并不是图像真正的区域capturedRect需要进行比例转换,才能得到实际在图片上的区域。

也就是说,我们需要将capturedRect转化为实际imgRect:

代码语言:javascript复制
void keyPressEvent(QKeyEvent *event) override {
  if (event->key() == Qt::Key_Return && status == Captured) {
    // 1. 获取捕获的图像区域
    auto picRealSize = screenImg->size();
    auto winSize = this->size();
    // 比例计算
    int realRectX = capturedRect.x() * picRealSize.width() / winSize.width();
    int realRectY = capturedRect.y() * picRealSize.height() / winSize.height();
    int realRectW =
        capturedRect.width() * picRealSize.width() / winSize.width();
    int realRectH =
        capturedRect.height() * picRealSize.height() / winSize.height();
    // 得到实际Rect
    QRect imgRect(realRectX, realRectY, realRectW, realRectH);
    // 2. 从保存的屏幕图像中获取指定区域的图像数据
    auto copiedImg = this->screenImg->copy(imgRect);
    // 3. 将图像数据写入到操作系统粘贴板
    QClipboard *clipboard = QGuiApplication::clipboard();
    clipboard->setImage(copiedImg);
    // 4. 回到Explore
    status = Explore;
    return;
  }
  if (event->key() == Qt::Key_Escape) {
    status = Explore;
  }
  this->update();
}

按照比例换算以后的代码如上,此时我们再看效果,会发现没有问题了:

最后

这篇文章算不上是比较深入的讲解截图工具的实现,只是通过demo来大体上讲解了截图的机制,让读者有一个入门的认识,像是截图区域确定以后我们还可以在上面添加方框、圆形、文字等操作都没有在这篇文章中体现。这篇文章只是一个入门,读者可以在掌握了基本的开发模式以后,实现更有意思的功能。

另外,笔者自己编写的截图软件capi(仓库地址:w4ngzhen/capi)已经有了基本的雏形,后续还会持续的往里面增加功能的,这里厚着脸皮希望有小伙伴能给个start。值得提到的是,笔者的截图软件capi目前是基于QT编写的,但是笔者正在做的是将截图的模块和QT的模块进行完全的解耦(其实已经差不多了),使用C 17的标准实现了截图功能核心模块的概念抽象,其目的在于笔者准备将QT换成另一个跨平台GUI框架wxWidgets来实现,为了实现这个目的,截图模块与具体的GUI框架解耦是十分必要的。

0 人点赞