summaryrefslogtreecommitdiffhomepage
path: root/opentrack/opencv-camera-dialog.hpp
blob: d6a2df198133477f5b4661f6a0a86dedcbe4cc93 (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
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
/* Copyright (c) 2015, 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

#if !defined(QT_MOC_RUN)

#include <QTimer>
#include <QMutex>
#include <QMutexLocker>
#include <opencv2/videoio.hpp>
#include "opentrack-compat/camera-names.hpp"

#ifdef __linux
#   include <QProcess>
#   include "opentrack-compat/sleep.hpp"
#endif

#ifdef _WIN32
#   include <objbase.h>
#   include <winerror.h>
#   include <windows.h>
#endif

#ifdef _WIN32
static void init_com_threading(void)
{
    HRESULT hr = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
    if (FAILED(hr))
        qDebug() << "failed CoInitializeEx" << hr << "code" << GetLastError();
}
#endif

static void maybe_grab_frame(cv::VideoCapture& cap)
{
    for (int i = 0; i < 60; i++)
    {
        if (cap.grab())
            break;
        portable::sleep(50);
    }
}

class camera_dialog
{
public:
    inline virtual ~camera_dialog() {}
#ifdef __linux
    void open_camera_settings(cv::VideoCapture *, const QString &camera_name, QMutex *)
    {
        int idx = camera_name_to_index(camera_name);
        QProcess::startDetached("qv4l2", QStringList() << "-d" << ("/dev/video" + QString::number(idx)));
    }
#elif defined(_WIN32)
    void open_camera_settings(cv::VideoCapture* cap, const QString& camera_name, QMutex* camera_mtx)
    {
        if (cap)
        {
            QMutexLocker l(camera_mtx);

            if (cap->isOpened())
            {
                init_com_threading();
                maybe_grab_frame(*cap);
                cap->set(cv::CAP_PROP_SETTINGS, 1);
                return;
            }
        }

        if (t.isActive())
            return;

        // don't hog the camera capture
        if (!t.isSingleShot())
            QObject::connect(&t, &QTimer::timeout, [&]() -> void { delete_capture(); });

        init_com_threading();
        fake_capture = cv::VideoCapture(camera_name_to_index(camera_name));
        maybe_grab_frame(fake_capture);
        fake_capture.set(cv::CAP_PROP_SETTINGS, 1);

        t.setSingleShot(true);
        t.setInterval(5000);

        // HACK: we're not notified when it's safe to close the capture
        t.start();
    }
#else
    void open_camera_settings(cv::VideoCapture*, const QString&, QMutex*) {}
#endif
private:
#if defined(_WIN32)
    cv::VideoCapture fake_capture;
    QTimer t;
    void delete_capture()
    {
        fake_capture.open("");
    }
#endif
};

#endif