summaryrefslogtreecommitdiffhomepage
path: root/tracker-pt/pt_video_widget.cpp
blob: 54bc9acd509c4bd3c61beedb27e4f6396672285c (plain)
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
/* 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 "pt_video_widget.h"
#include <opencv2/imgproc.hpp>

#include "opentrack/is-window-visible.hpp"

PTVideoWidget::PTVideoWidget(QWidget* parent) :
    QWidget(parent),
    freshp(false),
    visible(true)
{
    connect(&timer, SIGNAL(timeout()), this, SLOT(update_and_repaint()));
    timer.start(50);
}

void PTVideoWidget::update_image(const cv::Mat& frame)
{
    QMutexLocker foo(&mtx);

    if (!freshp)
    {
        if (_frame.cols != frame.cols || _frame.rows != frame.rows)
        {
            _frame = cv::Mat(frame.rows, frame.cols, CV_8U);
            _frame2 = cv::Mat(frame.rows, frame.cols, CV_8U);
        }
        frame.copyTo(_frame);
        freshp = true;
    }
}

void PTVideoWidget::paintEvent(QPaintEvent*)
{
    QMutexLocker foo(&mtx);
    QPainter painter(this);
    painter.drawImage(rect(), texture);
}

void PTVideoWidget::update_and_repaint()
{
    QMutexLocker l(&mtx);

    if (window_check_timer.elapsed_ms() > 250)
    {
        visible = is_window_visible(this);
        window_check_timer.start();
    }

    if (visible)
    {
        if (_frame.empty() || !freshp)
            return;
        cv::cvtColor(_frame, _frame2, cv::COLOR_RGB2BGR);

        if (_frame3.cols != width() || _frame3.rows != height())
            _frame3 = cv::Mat(height(), width(), CV_8U);

        cv::resize(_frame2, _frame3, cv::Size(width(), height()), 0, 0, cv::INTER_NEAREST);

        texture = QImage((const unsigned char*) _frame3.data, _frame3.cols, _frame3.rows, QImage::Format_RGB888);
        freshp = false;
        update();
    }
}