mirror of
https://github.com/zhuzichu520/FluentUI.git
synced 2024-11-23 11:17:15 +08:00
71 lines
2.7 KiB
C++
71 lines
2.7 KiB
C++
#include "Screenshot.h"
|
|
#include <QGuiApplication>
|
|
#include <QScreen>
|
|
#include <QQuickWindow>
|
|
#include <QDir>
|
|
#include <QtMath>
|
|
#include <QDateTime>
|
|
#include <QThreadPool>
|
|
#include "Def.h"
|
|
|
|
Screenshot::Screenshot(QQuickItem* parent):QQuickPaintedItem(parent){
|
|
_desktopGeometry = qApp->primaryScreen()->virtualGeometry();
|
|
maskColor(QColor(0,0,0,150));
|
|
start(QPoint(0,0));
|
|
end(QPoint(0,0));
|
|
connect(this,&Screenshot::startChanged,this,[=]{update();});
|
|
connect(this,&Screenshot::endChanged,this,[=]{update();});
|
|
}
|
|
|
|
void Screenshot::paint(QPainter* painter){
|
|
painter->save();
|
|
painter->fillRect(_desktopGeometry,_maskColor);
|
|
painter->setCompositionMode(QPainter::CompositionMode_Clear);
|
|
painter->fillRect(QRect(_start.x(),_start.y(),_end.x()-_start.x(),_end.y()-_start.y()), Qt::black);
|
|
painter->restore();
|
|
}
|
|
|
|
ScreenshotBackground::ScreenshotBackground(QQuickItem* parent) : QQuickPaintedItem(parent){
|
|
_devicePixelRatio = qApp->primaryScreen()->devicePixelRatio();
|
|
_desktopGeometry = qApp->primaryScreen()->virtualGeometry();
|
|
_desktopPixmap = qApp->primaryScreen()->grabWindow(0, _desktopGeometry.x(), _desktopGeometry.y(), _desktopGeometry.width(), _desktopGeometry.height());
|
|
int w = qApp->primaryScreen()->geometry().width();
|
|
int h = qApp->primaryScreen()->geometry().height();
|
|
foreach (auto item, qApp->screens()) {
|
|
if(item != qApp->primaryScreen()){
|
|
w = w + item->geometry().width()/qApp->primaryScreen()->devicePixelRatio();
|
|
}
|
|
}
|
|
setWidth(w);
|
|
setHeight(h);
|
|
}
|
|
|
|
void ScreenshotBackground::paint(QPainter* painter){
|
|
painter->save();
|
|
_sourcePixmap = _desktopPixmap.copy();
|
|
painter->drawPixmap(_desktopGeometry,_sourcePixmap);
|
|
painter->restore();
|
|
}
|
|
|
|
void ScreenshotBackground::capture(const QPoint& start,const QPoint& end){
|
|
update();
|
|
auto pixelRatio = qApp->primaryScreen()->devicePixelRatio();
|
|
auto x = qMin(start.x(),end.x()) * pixelRatio;
|
|
auto y = qMin(start.y(),end.y()) * pixelRatio;
|
|
auto w = qAbs(end.x()-start.x()) * pixelRatio;
|
|
auto h = qAbs(end.y()-start.y()) * pixelRatio;
|
|
_captureRect = QRect(x,y,w,h);
|
|
if(_captureMode == FluScreenshotType::CaptrueMode::Pixmap){
|
|
Q_EMIT captrueToPixmapCompleted(_sourcePixmap.copy(_captureRect));
|
|
}
|
|
if(_captureMode == FluScreenshotType::CaptrueMode::File){
|
|
QDir dir = _saveFolder;
|
|
if (!dir.exists(_saveFolder)){
|
|
dir.mkpath(_saveFolder);
|
|
}
|
|
auto filePath = _saveFolder.append("/").append(QString::number(QDateTime::currentDateTime().toMSecsSinceEpoch())).append(".png");
|
|
_sourcePixmap.copy(_captureRect).save(filePath);
|
|
Q_EMIT captrueToFileCompleted(QUrl::fromLocalFile(filePath));
|
|
}
|
|
}
|