From c0726d89f0918e0317c0516a445bead8c527372a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9C=B1=E5=AD=90=E6=A5=9A=5Czhuzi?=
 <zhuzichu520@outlook.com>
Date: Thu, 17 Aug 2023 19:09:56 +0800
Subject: [PATCH] update

---
 src/Screenshot.cpp | 12 ++++--------
 src/Screenshot.h   |  1 +
 2 files changed, 5 insertions(+), 8 deletions(-)

diff --git a/src/Screenshot.cpp b/src/Screenshot.cpp
index e5936780..c84f287b 100644
--- a/src/Screenshot.cpp
+++ b/src/Screenshot.cpp
@@ -42,28 +42,24 @@ ScreenshotBackground::ScreenshotBackground(QQuickItem* parent) : QQuickPaintedIt
 void ScreenshotBackground::paint(QPainter* painter)
 {
     painter->save();
-    QPixmap sourcePixmap;
-    QPainter p(&sourcePixmap);
+    _sourcePixmap = QPixmap(QSize(_desktopGeometry.width(),_desktopGeometry.height()));
+    QPainter p(&_sourcePixmap);
     p.drawPixmap(_desktopGeometry,_desktopPixmap);
-    painter->drawPixmap(_desktopGeometry,sourcePixmap);
+    painter->drawPixmap(_desktopGeometry,_sourcePixmap);
     painter->restore();
 }
 
 void ScreenshotBackground::capture(const QPoint& start,const QPoint& end){
-    qDebug()<<start;
-    qDebug()<<end;
     _grabResult = grabToImage();
     auto x = qMin(start.x(),end.x());
     auto y = qMin(start.y(),end.y());
     auto w = qAbs(end.x()-start.x());
     auto h = qAbs(end.y()-start.y());
     _captureRect = QRect(x,y,w,h);
-    qDebug()<<_captureRect;
     connect(_grabResult.data(), &QQuickItemGrabResult::ready, this, &ScreenshotBackground::handleGrabResult);
 }
 
 void ScreenshotBackground::handleGrabResult(){
-    _grabResult.data()->image().copy(_captureRect).save("aaa.png");
-
+    _sourcePixmap.copy(_captureRect).save("aaa.png");
 }
 
diff --git a/src/Screenshot.h b/src/Screenshot.h
index 17487d84..a2c0527d 100644
--- a/src/Screenshot.h
+++ b/src/Screenshot.h
@@ -19,6 +19,7 @@ public:
 private:
     QRect _desktopGeometry;
     QPixmap _desktopPixmap;
+    QPixmap _sourcePixmap;
     qreal _devicePixelRatio;
     QSharedPointer<QQuickItemGrabResult>  _grabResult;
     QRect _captureRect;