diff options
author | Stanislaw Halik <sthalik@misaki.pl> | 2019-02-11 20:07:41 +0100 |
---|---|---|
committer | Stanislaw Halik <sthalik@misaki.pl> | 2019-02-11 20:21:36 +0100 |
commit | 85fc6ac25d350950be56ba47b06c247216f2e22a (patch) | |
tree | 6a4b4a65e9084c9595c28c4045824376a8c577d3 /video/video-widget.cpp | |
parent | 5d6cccb406e4aa4fe3e8430690296e5f59474e48 (diff) |
video/widget: externalize opencv code
Diffstat (limited to 'video/video-widget.cpp')
-rw-r--r-- | video/video-widget.cpp | 96 |
1 files changed, 9 insertions, 87 deletions
diff --git a/video/video-widget.cpp b/video/video-widget.cpp index 90975f0c..3e6d9038 100644 --- a/video/video-widget.cpp +++ b/video/video-widget.cpp @@ -6,103 +6,24 @@ #include <cstddef> #include <cstring> -#ifdef OTR_VIDEO_HAS_OPENCV -# include <opencv2/imgproc.hpp> -#endif - #include <QPainter> -void cv_video_widget::init_image_nolock() +void video_widget::init_image_nolock() { texture = QImage(W, H, QImage::Format_ARGB32); texture.setDevicePixelRatio(devicePixelRatioF()); } -cv_video_widget::cv_video_widget(QWidget* parent) : QWidget(parent) +video_widget::video_widget(QWidget* parent) : QWidget(parent) { W = width(); H = height(); init_image_nolock(); texture.fill(Qt::gray); - connect(&timer, &QTimer::timeout, this, &cv_video_widget::update_and_repaint, Qt::DirectConnection); + connect(&timer, &QTimer::timeout, this, &video_widget::update_and_repaint, Qt::DirectConnection); timer.start(65); } -#ifdef OTR_VIDEO_HAS_OPENCV -void cv_video_widget::update_image(const cv::Mat& frame) -{ - QMutexLocker l(&mtx); - - if (freshp) - return; - - if (W < 1 || H < 1 || frame.rows < 1 || frame.cols < 1) - return; - - cv::Mat const* __restrict frame_scaled = nullptr; - - if (frame3.cols != W || frame3.rows != H) - { - frame3 = cv::Mat(H, W, frame.type()); - frame2 = cv::Mat(H, W, CV_8UC4); - - if (!frame2.isContinuous() || !frame3.isContinuous()) - std::abort(); - } - - if (frame.cols != W || frame.rows != H) - { - cv::resize(frame, frame3, { W, H }, 0, 0, cv::INTER_NEAREST); - frame_scaled = &frame3; - } - else if (!frame.isContinuous()) - { - frame.copyTo(frame3); - frame_scaled = &frame3; - } - else - frame_scaled = &frame; - - freshp = true; - - int color_cvt = 0; - constexpr int nchannels = 4; - - switch (frame_scaled->channels()) - { - case 1: - color_cvt = cv::COLOR_GRAY2BGRA; - break; - case 3: - color_cvt = cv::COLOR_BGR2BGRA; - break; - case nchannels: - break; - default: - unreachable(); - break; - } - - cv::Mat const* frame_color; - - if (color_cvt != cv::COLOR_COLORCVT_MAX) - { - cv::cvtColor(*frame_scaled, frame2, color_cvt); - frame_color = &frame2; - } - else - frame_color = frame_scaled; - - int stride = frame_color->step.p[0], rows = frame_color->rows; - int nbytes = rows * stride; - vec.resize(nbytes); vec.shrink_to_fit(); - std::memcpy(vec.data(), frame_color->data, nbytes); - - texture = QImage((const unsigned char*) vec.data(), W, H, stride, QImage::Format_ARGB32); - texture.setDevicePixelRatio(devicePixelRatioF()); -} -#endif - -void cv_video_widget::update_image(const QImage& img) +void video_widget::update_image(const QImage& img) { QMutexLocker l(&mtx); @@ -118,7 +39,7 @@ void cv_video_widget::update_image(const QImage& img) texture.setDevicePixelRatio(devicePixelRatioF()); } -void cv_video_widget::paintEvent(QPaintEvent*) +void video_widget::paintEvent(QPaintEvent*) { QMutexLocker foo(&mtx); @@ -132,7 +53,7 @@ void cv_video_widget::paintEvent(QPaintEvent*) painter.drawImage(rect(), texture); } -void cv_video_widget::update_and_repaint() +void video_widget::update_and_repaint() { if (!check_is_visible()) return; @@ -146,7 +67,7 @@ void cv_video_widget::update_and_repaint() } } -void cv_video_widget::resizeEvent(QResizeEvent*) +void video_widget::resizeEvent(QResizeEvent*) { QMutexLocker l(&mtx); double dpr = devicePixelRatioF(); @@ -155,8 +76,9 @@ void cv_video_widget::resizeEvent(QResizeEvent*) init_image_nolock(); } -void cv_video_widget::get_preview_size(int& w, int& h) +void video_widget::get_preview_size(int& w, int& h) { QMutexLocker l(&mtx); w = W; h = H; } + |