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 /cv | |
parent | 5d6cccb406e4aa4fe3e8430690296e5f59474e48 (diff) |
video/widget: externalize opencv code
Diffstat (limited to 'cv')
-rw-r--r-- | cv/CMakeLists.txt | 2 | ||||
-rw-r--r-- | cv/video-widget.cpp | 78 | ||||
-rw-r--r-- | cv/video-widget.hpp | 20 |
3 files changed, 99 insertions, 1 deletions
diff --git a/cv/CMakeLists.txt b/cv/CMakeLists.txt index 60350a99..dfc983aa 100644 --- a/cv/CMakeLists.txt +++ b/cv/CMakeLists.txt @@ -1,6 +1,6 @@ find_package(OpenCV QUIET) if(OpenCV_FOUND) otr_module(cv STATIC) - target_link_libraries(${self} opencv_videoio opencv_core) + target_link_libraries(${self} opencv_videoio opencv_core opentrack-video) target_include_directories(${self} SYSTEM PRIVATE ${OpenCV_INCLUDE_DIRS}) endif() diff --git a/cv/video-widget.cpp b/cv/video-widget.cpp new file mode 100644 index 00000000..7cc6ca2d --- /dev/null +++ b/cv/video-widget.cpp @@ -0,0 +1,78 @@ +#include "video-widget.hpp" + +#include <opencv2/imgproc.hpp> + +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()); +} + +cv_video_widget::cv_video_widget(QWidget* parent) : video_widget(parent) {} diff --git a/cv/video-widget.hpp b/cv/video-widget.hpp new file mode 100644 index 00000000..9d62f19e --- /dev/null +++ b/cv/video-widget.hpp @@ -0,0 +1,20 @@ +/* Copyright (c) 2019 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. + */ + +#pragma once + +#include "video/video-widget.hpp" +#include <opencv2/core.hpp> + +struct cv_video_widget final : video_widget +{ + cv_video_widget(QWidget* parent = nullptr); + void update_image(const cv::Mat& frame); + +private: + cv::Mat frame2, frame3; +}; |