veye_viewer/Capture.cpp at main · veyeimaging/veye_viewer · GitHub

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
108
109
110
111
#include "Capture.h"
#include "CommStuct.h"
#include "QxHelp.h"

Capture::Capture(CommDevCfg devCfg, QObject *parent)
    : QThread{parent}
    , m_devCfg(devCfg)
{
    qRegisterMetaType<timeval>("timeval");
    qRegisterMetaType<StCamInfo>("StCamInfo");
    qRegisterMetaType<EuCamStatus>("EuCamStatus");
    m_timer = new QTimer(this);
}

Capture::~Capture()
{
    if (m_timer) {
        m_timer->stop();

        m_timer->deleteLater();
        m_timer = nullptr;
    }
}

void Capture::setMainWidget(QObject *mainWidget)
{
    m_mainWidget = mainWidget;
}

void Capture::startStream(int fd, StCamParam &param)
{
    m_bThread = true;
    std::thread tSave([this]() { saveThread(); });
    tSave.detach();

    m_fd = fd;
    m_stCamParam = param;
    if (!m_timer->isActive()) {
        connect(m_timer, &QTimer::timeout, this, [this]() {
            StCamInfo info;
            info.fps = m_nFps * 1.0;
            info.height = m_height;
            info.width = m_width;
            info.left = m_x;
            info.top = m_y;
            emit sndFrameInfo(info);
            m_nFps = 0;
        });
        m_timer->start(1000);
    }
}

void Capture::stopStream()
{
    m_bThread = false;
}

void Capture::rcvSaveImg(QString strPath, int count)
{
    QxHelp::makeFolder(strPath);
    QxHelp::makeFolderTail(strPath);
    m_strSavePah = strPath;
    m_nSaveCount = count;
    m_saveIndex = 0;
    m_bSave = true;
}

void Capture::rcvShow(bool bShow)
{
    m_bShow = bShow;
}

void Capture::saveThread()
{
    int nCount = 1000;
    while (m_bThread) {
        StSaveImg si;
        int size = 0;
        m_mtxSaveImg.lock();
        size = m_listSaveImg.size();
        if (size > 0) {
            si = m_listSaveImg.front();
            m_listSaveImg.pop_front();
        }
        m_mtxSaveImg.unlock();

        if (size > 0) {
            nCount = 0;
            if (QxHelp::writeFile((char *) si.buf, si.len, si.filename)) {
                qDebug() << "save.... " + si.filename + " OK";
            } else {
                qDebug() << "save.... " + si.filename + " error";
            }

            // si.img.save(si.filename + "_.png", "PNG");
            // QImage img = QImage(si.buf, si.width, si.height, QImage::Format_Grayscale8);
            // img.save(si.filename + ".png", "PNG");
            delete[] si.buf;
        } else {
            if (++nCount > 500 && !m_bSave) {
                if (501 == nCount) {
                    emit sndSaveFinish(true);
                } else {
                    nCount = 502;
                }
            }
        }

        QThread::msleep(1);
    }
}