-
Notifications
You must be signed in to change notification settings - Fork 0
/
cv_camera.cpp
133 lines (113 loc) · 4.22 KB
/
cv_camera.cpp
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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
//
// Created by tekatod on 4/2/18.
//
#include <opencv2/imgproc.hpp>
#include <cv.hpp>
#include <QtWidgets/QGridLayout>
#include <fcntl.h>
#include <linux/videodev2.h>
#include <sys/ioctl.h>
#include <iostream>
#include "cv_camera.h"
cv_camera::cv_camera(QWidget* parent) : QWidget(parent) {
m_cameras = QCameraInfo::availableCameras();
m_timer = new QTimer(this);
emit set_up_camera("/dev/video0");
// m_cap = cv::VideoCapture(0);
// m_cap.set(cv::CAP_PROP_FRAME_WIDTH, 1920);
// m_cap.set(cv::CAP_PROP_FRAME_HEIGHT, 1080);
// m_v4l_device = open("/dev/video0", O_RDWR, 0);
m_zoom_slider = new QSlider(Qt::Horizontal, this);
m_combo_box = new QComboBox();
m_zoom_slider->setMinimum(100);
m_zoom_slider->setMaximum(500);
m_cap_button = new QPushButton("capture");
m_record_button = new QPushButton("record");
m_label = new QLabel(this);
m_label->setSizePolicy(QSizePolicy::Expanding,
QSizePolicy::Expanding);
m_label->setAlignment(Qt::AlignCenter);
for(auto&& info : m_cameras) {
m_combo_box->addItem(info.deviceName());
}
this->setMinimumSize(640, 480);
// m_label->setMinimumSize(1280, 720);
auto* layout1 = new QGridLayout;
layout1->addWidget(m_label);
layout1->addWidget(m_zoom_slider);
layout1->addWidget(m_cap_button);
layout1->addWidget(m_record_button);
layout1->addWidget(m_combo_box);
this->setLayout(layout1);
QObject::connect(m_timer, SIGNAL(timeout()), this, SLOT(update_picture()));
QObject::connect(m_zoom_slider, SIGNAL(valueChanged(int)), this, SLOT(change_zoom(int)));
QObject::connect(m_cap_button, SIGNAL(clicked()), this, SLOT(capture()));
QObject::connect(m_record_button, SIGNAL(clicked()), this, SLOT(record()));
QObject::connect(m_combo_box, SIGNAL(currentIndexChanged(const QString&)),
this, SLOT(set_up_camera(const QString&)));
}
void cv_camera::update_picture() {
m_captured = false;
m_cap >> m_image;
m_captured = true;
cv::Mat dst;
auto t_size = this->size();
dst = this->resize_keep_ar(m_image, cv::Size(t_size.width() - 50, t_size.height() - 50));
cv::cvtColor(dst, dst, CV_BGR2RGB);
QImage image1 = QImage((uchar*) dst.data, dst.cols, dst.rows, dst.step, QImage::Format_RGB888);
//show Qimage using QLabel
m_label->setPixmap(QPixmap::fromImage(image1));
m_label->adjustSize();
if(m_record) {
cv::imwrite("image_" + std::to_string(m_index) + ".jpg", m_image);
m_index++;
}
// m_label->show();
}
cv::Mat cv_camera::resize_keep_ar(const cv::Mat& input, const cv::Size& dstSize)
{
cv::Mat output;
double h1 = dstSize.width * (input.rows/(double)input.cols);
double w2 = dstSize.height * (input.cols/(double)input.rows);
if( h1 <= dstSize.height) {
cv::resize( input, output, cv::Size(dstSize.width, h1));
} else {
cv::resize( input, output, cv::Size(w2, dstSize.height));
}
int top = (dstSize.height-output.rows) / 2;
int down = (dstSize.height-output.rows+1) / 2;
int left = (dstSize.width - output.cols) / 2;
int right = (dstSize.width - output.cols+1) / 2;
cv::copyMakeBorder(output, output, top, down, left, right, cv::BORDER_CONSTANT, cv::Scalar(0, 0, 0));
return output;
}
void cv_camera::change_zoom(int zoom) {
struct v4l2_control v4l2_ctrl = {V4L2_CID_ZOOM_ABSOLUTE, zoom};
ioctl(m_v4l_device, VIDIOC_S_CTRL, &v4l2_ctrl);
}
void cv_camera::capture() {
if(m_captured) {
cv::imwrite("image_" + std::to_string(m_index) + ".jpg", m_image);
m_index++;
}
}
void cv_camera::record() {
if(!m_record) {
m_record = true;
m_record_button->setText("stop");
m_cap_button->setDisabled(true);
} else {
m_record = false;
m_record_button->setText("record");
m_cap_button->setDisabled(false);
}
}
void cv_camera::set_up_camera(const QString& camera_name) {
m_timer->stop();
m_cap.release();
m_cap = cv::VideoCapture(camera_name.toStdString().back() - '0');
m_cap.set(cv::CAP_PROP_FRAME_WIDTH, 1920);
m_cap.set(cv::CAP_PROP_FRAME_HEIGHT, 1080);
m_v4l_device = open(camera_name.toStdString().c_str(), O_RDWR, 0);
m_timer->start(5);
}