landmark-annotator-desktop/viewtab.cpp at main · welvin21/landmark-annotator-desktop · 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
#include "viewtab.h"

ViewTab::ViewTab(DesktopApp* parent)
{
    this->parent = parent;

    this->timer = new QTimer;
    QObject::connect(timer, &QTimer::timeout, [this]() {
        if (this->parent->deviceCount > 0) {
            switch (k4a_device_get_capture(this->parent->device, &this->parent->capture, K4A_WAIT_INFINITE)) {
            case K4A_WAIT_RESULT_SUCCEEDED:
                break;
            }

            if (this->parent->capture) {
                k4a_image_t k4aColorImage = k4a_capture_get_color_image(this->parent->capture);

                if (k4aColorImage != NULL) {
                    this->parent->colorImageQueue.push(k4aColorImage);

                    int width = this->parent->ui.graphicsViewVideo->width(), height = this->parent->ui.graphicsViewVideo->height();
                    QImage qColorImage = (this->parent->getQColorImage()).scaled(width, height, Qt::KeepAspectRatio);

                    // Deallocate heap memory used by previous GGraphicsScene object
                    if (this->parent->ui.graphicsViewVideo->scene()) {
                        delete this->parent->ui.graphicsViewVideo->scene();
                    }

                    QGraphicsPixmapItem* item = new QGraphicsPixmapItem(QPixmap::fromImage(qColorImage));
                    QGraphicsScene* scene = new QGraphicsScene;
                    scene->addItem(item);

                    this->parent->ui.graphicsViewVideo->setScene(scene);
                    this->parent->ui.graphicsViewVideo->show();

                    while (this->parent->colorImageQueue.size() > MAX_IMAGE_QUEUE_SIZE) {
                        k4a_image_release(this->parent->colorImageQueue.front());
                        this->parent->colorImageQueue.pop();
                    }
                }

                k4a_image_t k4aDepthImage = k4a_capture_get_depth_image(this->parent->capture);

                if (k4aDepthImage != NULL) {
                    this->parent->depthImageQueue.push(k4aDepthImage);

                    int width = this->parent->ui.graphicsViewVideo2->width(), height = this->parent->ui.graphicsViewVideo2->height();
                    QImage qDepthImage = (this->parent->getQDepthImage()).scaled(width, height, Qt::KeepAspectRatio);

                    // Deallocate heap memory used by previous GGraphicsScene object
                    if (this->parent->ui.graphicsViewVideo->scene()) {
                        delete this->parent->ui.graphicsViewVideo2->scene();
                    }

                    QGraphicsPixmapItem* item = new QGraphicsPixmapItem(QPixmap::fromImage(qDepthImage));
                    QGraphicsScene* scene = new QGraphicsScene;
                    scene->addItem(item);

                    this->parent->ui.graphicsViewVideo2->setScene(scene);
                    this->parent->ui.graphicsViewVideo2->show();

                    while (this->parent->depthImageQueue.size() > MAX_IMAGE_QUEUE_SIZE) {
                        k4a_image_release(this->parent->depthImageQueue.front());
                        this->parent->depthImageQueue.pop();
                    }
                }

                k4a_image_t k4aIRImage = k4a_capture_get_ir_image(this->parent->capture);

                if (k4aIRImage != NULL) {
                    this->parent->irImageQueue.push(k4aIRImage);

                    int width = this->parent->ui.graphicsViewVideo3->width(), height = this->parent->ui.graphicsViewVideo3->height();
                    QImage qIRImage = (this->parent->getQIRImage()).scaled(width, height, Qt::KeepAspectRatio);

                    // Deallocate heap memory used by previous GGraphicsScene object
                    if (this->parent->ui.graphicsViewVideo->scene()) {
                        delete this->parent->ui.graphicsViewVideo3->scene();
                    }

                    QGraphicsPixmapItem* item = new QGraphicsPixmapItem(QPixmap::fromImage(qIRImage));
                    QGraphicsScene* scene = new QGraphicsScene;
                    scene->addItem(item);

                    this->parent->ui.graphicsViewVideo3->setScene(scene);
                    this->parent->ui.graphicsViewVideo3->show();

                    while (this->parent->irImageQueue.size() > MAX_IMAGE_QUEUE_SIZE) {
                        k4a_image_release(this->parent->irImageQueue.front());
                        this->parent->irImageQueue.pop();
                    }
                }

                k4a_capture_release(this->parent->capture);
            }
            else {
                qDebug() << "No capture found\n";
            }
        }
    });
}