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
123
124
125
126
127
128
129
|
/* Copyright (c) 2012 Patrick Ruoff
* Copyright (c) 2014-2016 Stanislaw Halik <sthalik@misaki.pl>
*
* Permission to use, copy, modify, and/or distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*/
#include "video-widget.hpp"
#include "compat/check-visible.hpp"
#include "compat/math.hpp"
#include <cstring>
#include <opencv2/imgproc.hpp>
cv_video_widget::cv_video_widget(QWidget* parent) : QWidget(parent)
{
connect(&timer, SIGNAL(timeout()), this, SLOT(update_and_repaint()), Qt::DirectConnection);
timer.start(65);
}
void cv_video_widget::update_image(const cv::Mat& frame)
{
QMutexLocker l(&mtx);
if (!freshp)
{
if (width < 1 || height < 1)
return;
if (frame1.cols != frame.cols || frame1.rows != frame.rows)
frame1 = cv::Mat(frame.rows, frame.cols, CV_8UC3);
frame.copyTo(frame1);
freshp = true;
if (frame2.cols != frame1.cols || frame2.rows != frame1.rows)
frame2 = cv::Mat(frame1.rows, frame1.cols, CV_8UC4);
if (frame3.cols != width || frame3.rows != height)
frame3 = cv::Mat(height, width, CV_8UC4);
cv::cvtColor(frame1, frame2, cv::COLOR_BGR2BGRA);
const cv::Mat* img;
if (frame1.cols != width || frame1.rows != height)
{
cv::resize(frame2, frame3, cv::Size(width, height), 0, 0, cv::INTER_NEAREST);
img = &frame3;
}
else
img = &frame2;
const unsigned nbytes = unsigned(4 * img->rows * img->cols);
vec.resize(nbytes);
std::memcpy(vec.data(), img->data, nbytes);
texture = QImage((const unsigned char*) vec.data(), width, height, QImage::Format_ARGB32);
}
}
void cv_video_widget::update_image(const QImage& img)
{
QMutexLocker l(&mtx);
if (freshp)
return;
const unsigned nbytes = unsigned(img.bytesPerLine() * img.height());
vec.resize(nbytes);
std::memcpy(vec.data(), img.constBits(), nbytes);
texture = QImage((const unsigned char*) vec.data(), img.width(), img.height(), img.format());
freshp = true;
}
void cv_video_widget::paintEvent(QPaintEvent*)
{
QMutexLocker foo(&mtx);
QPainter painter(this);
double dpr = devicePixelRatioF();
int W = iround(QWidget::width() * dpr);
int H = iround(QWidget::height() * dpr);
painter.drawImage(rect(), texture);
if (texture.width() != W || texture.height() != H)
{
texture = QImage(W, H, QImage::Format_ARGB32);
texture.setDevicePixelRatio(dpr);
width = W, height = H;
frame1 = cv::Mat();
frame2 = cv::Mat();
frame3 = cv::Mat();
}
}
void cv_video_widget::update_and_repaint()
{
if (!check_is_visible())
return;
QMutexLocker l(&mtx);
if (freshp)
{
freshp = false;
repaint();
}
}
void cv_video_widget::get_preview_size(int& w, int& h)
{
QMutexLocker l(&mtx);
w = width, h = height;
}
|
