diff options
author | Stanislaw Halik <sthalik@misaki.pl> | 2018-01-18 23:45:42 +0100 |
---|---|---|
committer | Stanislaw Halik <sthalik@misaki.pl> | 2018-01-18 23:55:02 +0100 |
commit | 0657dc14e07ee705a8ab0bba67dfbe04c603db49 (patch) | |
tree | e554cf0c59ddf710bb8c2b3bae3c478f5b9ffb66 /tracker-pt/frame.cpp | |
parent | 65523fe1304a7e61889abb1e6854192a43c6dcf0 (diff) |
tracker/pt: move impl bits away from pt-base
Diffstat (limited to 'tracker-pt/frame.cpp')
-rw-r--r-- | tracker-pt/frame.cpp | 83 |
1 files changed, 0 insertions, 83 deletions
diff --git a/tracker-pt/frame.cpp b/tracker-pt/frame.cpp deleted file mode 100644 index e403af07..00000000 --- a/tracker-pt/frame.cpp +++ /dev/null @@ -1,83 +0,0 @@ -#include "frame.hpp" - -#include "compat/math.hpp" - -#include <cstring> -#include <tuple> - -#include <opencv2/imgproc.hpp> - -using namespace pt_module; - -Preview& Preview::operator=(const pt_frame& frame_) -{ - const cv::Mat& frame = frame_.as_const<const Frame>()->mat; - ensure_size(frame_copy, frame_out.cols, frame_out.rows, CV_8UC3); - - if (frame.channels() != 3) - { - once_only(qDebug() << "tracker/pt: camera frame depth: 3 !=" << frame.channels()); - return *this; - } - - const bool need_resize = frame.cols != frame_out.cols || frame.rows != frame_out.rows; - if (need_resize) - cv::resize(frame, frame_copy, cv::Size(frame_out.cols, frame_out.rows), 0, 0, cv::INTER_NEAREST); - else - frame.copyTo(frame_copy); - - return *this; -} - -Preview::Preview(int w, int h) -{ - ensure_size(frame_out, w, h, CV_8UC4); - - frame_out.setTo(cv::Scalar(0, 0, 0, 0)); -} - -QImage Preview::get_bitmap() -{ - int stride = frame_out.step.p[0]; - - if (stride < 64 || stride < frame_out.cols * 4) - { - once_only(qDebug() << "bad stride" << stride - << "for bitmap size" << frame_copy.cols << frame_copy.rows); - return QImage(); - } - - cv::cvtColor(frame_copy, frame_out, cv::COLOR_BGR2BGRA); - - return QImage((const unsigned char*) frame_out.data, - frame_out.cols, frame_out.rows, - stride, - QImage::Format_ARGB32); -} - -void Preview::draw_head_center(double x, double y) -{ - double px_, py_; - - std::tie(px_, py_) = to_pixel_pos(x, y, frame_copy.cols, frame_copy.rows); - - int px = iround(px_), py = iround(py_); - - constexpr int len = 9; - - static const cv::Scalar color(0, 255, 255); - cv::line(frame_copy, - cv::Point(px - len, py), - cv::Point(px + len, py), - color, 1); - cv::line(frame_copy, - cv::Point(px, py - len), - cv::Point(px, py + len), - color, 1); -} - -void Preview::ensure_size(cv::Mat& frame, int w, int h, int type) -{ - if (frame.cols != w || frame.rows != h) - frame = cv::Mat(h, w, type); -} |