1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
#include "UBWindowCaptureDelegate_win.h"
#include <windows.h>
UBWindowCaptureDelegate::UBWindowCaptureDelegate(QObject *parent)
:QObject(parent)
, mIsCapturing(false)
, mCancel(false)
, mLastPoint(0, 0)
{
// NOOP
}
UBWindowCaptureDelegate::~UBWindowCaptureDelegate(void)
{
// NOOP
}
const QPixmap UBWindowCaptureDelegate::getCapturedWindow()
{
return mCapturedPixmap;
}
int UBWindowCaptureDelegate::execute()
{
mIsCapturing = true;
mCancel = false;
qApp->installEventFilter(this);
while (mIsCapturing && !mCancel)
{
QPoint curPos = QCursor::pos();
if (mLastPoint != curPos)
{
processPos(curPos);
mLastPoint = curPos;
}
drawSelectionRect();
qApp->processEvents();
}
if (!mCancel)
{
BringWindowToTop(mCurrentWindow);
RedrawWindow(mCurrentWindow, NULL, NULL, RDW_INVALIDATE | RDW_UPDATENOW | RDW_ALLCHILDREN);
QWaitCondition sleep;
QMutex mutex;
mutex.lock();
sleep.wait(&mutex, 200);
mutex.unlock();
mCapturedPixmap = QPixmap::grabWindow(mCurrentWindow);
return QDialog::Accepted;
}
else
{
RedrawWindow(NULL, NULL, NULL, RDW_INVALIDATE | RDW_UPDATENOW | RDW_ALLCHILDREN);
return QDialog::Rejected;
}
}
void UBWindowCaptureDelegate::processPos(QPoint pPoint)
{
POINT wPoint;
wPoint.x = pPoint.x();
wPoint.y = pPoint.y();
HWND desktop = GetDesktopWindow();
HWND newWindow = ChildWindowFromPointEx(desktop, wPoint, CWP_SKIPINVISIBLE | CWP_SKIPTRANSPARENT);
if (mCurrentWindow != newWindow && newWindow != desktop)
{
RedrawWindow(NULL, NULL, NULL, RDW_INVALIDATE | RDW_UPDATENOW | RDW_ALLCHILDREN);
mCurrentWindow = newWindow;
}
}
void UBWindowCaptureDelegate::drawSelectionRect()
{
RECT windowRect;
if (GetWindowRect(mCurrentWindow, &windowRect))
{
HDC hDC;
HPEN hOldPen;
HBRUSH hOldBrush;
hDC = GetDC(NULL);
hOldPen = (HPEN)SelectObject(hDC, CreatePen(PS_SOLID, 5, RGB(255, 0, 0)));
hOldBrush = (HBRUSH)SelectObject(hDC, GetStockObject(NULL_BRUSH));
Rectangle(hDC, windowRect.left + 1, windowRect.top + 1, windowRect.right - 1, windowRect.bottom - 1);
SelectObject(hDC, hOldBrush);
DeleteObject(SelectObject(hDC, hOldPen));
ReleaseDC(NULL, hDC);
}
}
bool UBWindowCaptureDelegate::eventFilter(QObject *target, QEvent *event)
{
Q_UNUSED(target);
// once the application is deactivated, it means that the user clicked somewhere else and we shall
// take the capture
if (event->type() == QEvent::MouseButtonPress || event->type() == QEvent::WindowDeactivate)
{
mIsCapturing = false;
}
else if (event->type() == QEvent::KeyPress)
{
mCancel = true;
}
return true;
}