MYNT-EYE-S-SDK/samples/uvc/camera.cc

131 lines
3.4 KiB
C++
Raw Permalink Normal View History

2018-03-25 18:03:04 +03:00
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <chrono>
#include <condition_variable>
#include <iomanip>
#include <iostream>
#include <mutex>
2018-04-09 04:28:58 +03:00
#include "mynteye/glog_init.h"
2018-03-22 07:03:08 +02:00
#include "mynteye/mynteye.h"
2018-04-06 04:12:09 +03:00
#include "mynteye/types.h"
2018-03-22 07:03:08 +02:00
#include "uvc/uvc.h"
2018-03-25 18:03:04 +03:00
struct frame {
const void *data = nullptr;
~frame() {
data = nullptr;
}
};
2018-03-22 07:03:08 +02:00
MYNTEYE_USE_NAMESPACE
int main(int argc, char *argv[]) {
glog_init _(argc, argv);
2018-03-25 18:03:04 +03:00
std::vector<std::shared_ptr<uvc::device>> mynteye_devices;
2018-03-22 07:03:08 +02:00
auto context = uvc::create_context();
auto devices = uvc::query_devices(context);
LOG_IF(FATAL, devices.size() <= 0) << "No devices :(";
for (auto &&device : devices) {
2018-03-25 18:03:04 +03:00
auto vid = uvc::get_vendor_id(*device);
// auto pid = uvc::get_product_id(*device);
// LOG(INFO) << "vid: " << vid << ", pid: " << pid;
if (vid == MYNTEYE_VID) {
mynteye_devices.push_back(device);
}
}
// std::string dashes(80, '-');
size_t n = mynteye_devices.size();
LOG_IF(FATAL, n <= 0) << "No MYNT EYE devices :(";
2018-03-31 10:54:37 +03:00
LOG(INFO) << "MYNT EYE devices: ";
2018-03-25 18:03:04 +03:00
for (size_t i = 0; i < n; i++) {
auto device = mynteye_devices[i];
2018-03-30 11:39:35 +03:00
auto name = uvc::get_video_name(*device);
2018-03-22 07:03:08 +02:00
auto vid = uvc::get_vendor_id(*device);
auto pid = uvc::get_product_id(*device);
2018-03-31 10:54:37 +03:00
LOG(INFO) << " index: " << i << ", name: " << name << ", vid: 0x"
<< std::hex << vid << ", pid: 0x" << std::hex << pid;
2018-03-25 18:03:04 +03:00
}
std::shared_ptr<uvc::device> device = nullptr;
if (n <= 1) {
device = mynteye_devices[0];
LOG(INFO) << "Only one MYNT EYE device, select index: 0";
} else {
while (true) {
size_t i;
LOG(INFO) << "There are " << n << " MYNT EYE devices, select index: ";
std::cin >> i;
if (i >= n) {
LOG(WARNING) << "Index out of range :(";
continue;
}
device = mynteye_devices[i];
break;
}
}
std::mutex mtx;
std::condition_variable cv;
std::vector<frame> frames;
const auto frame_ready = [&frames]() { return !frames.empty(); };
const auto frame_empty = [&frames]() { return frames.empty(); };
uvc::set_device_mode(
2018-03-31 10:54:37 +03:00
*device, 752, 480, static_cast<int>(Format::YUYV), 25,
2018-03-25 18:03:04 +03:00
[&mtx, &cv, &frames, &frame_ready](const void *data) {
// reinterpret_cast<const std::uint8_t *>(data);
std::unique_lock<std::mutex> lock(mtx);
frame frame;
frame.data = data; // not copy
frames.push_back(frame);
if (frame_ready())
cv.notify_one();
});
2018-03-30 11:24:43 +03:00
LOG(INFO) << "Press ESC/Q on windows to terminate";
2018-03-25 18:03:04 +03:00
cv::namedWindow("frame");
uvc::start_streaming(*device, 0);
double t, fps = 0;
while (true) {
t = static_cast<double>(cv::getTickCount());
std::unique_lock<std::mutex> lock(mtx);
if (frame_empty()) {
if (!cv.wait_for(lock, std::chrono::seconds(2), frame_ready))
throw std::runtime_error("Timeout waiting for frame.");
}
auto frame = frames.back(); // only last one is valid
cv::Mat img(480, 752, CV_8UC2, const_cast<void *>(frame.data));
cv::cvtColor(img, img, cv::COLOR_YUV2BGR_YUY2);
cv::imshow("frame", img);
frames.clear();
char key = static_cast<char>(cv::waitKey(1));
if (key == 27 || key == 'q' || key == 'Q') { // ESC/Q
break;
}
t = static_cast<double>(cv::getTickCount() - t);
fps = cv::getTickFrequency() / t;
2018-03-22 07:03:08 +02:00
}
2018-03-25 18:03:04 +03:00
UNUSED(fps)
2018-03-22 07:03:08 +02:00
2018-03-25 18:03:04 +03:00
uvc::stop_streaming(*device);
// cv::destroyAllWindows();
2018-03-22 07:03:08 +02:00
return 0;
}