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
133
134
|
/*! \example tutorial-video-recorder.cpp */
#include <visp3/core/vpTime.h>
#include <visp3/gui/vpDisplayGDI.h>
#include <visp3/gui/vpDisplayGTK.h>
#include <visp3/gui/vpDisplayOpenCV.h>
#include <visp3/gui/vpDisplayX.h>
#include <visp3/io/vpVideoWriter.h>
#include <visp3/sensor/vpV4l2Grabber.h>
#if defined(HAVE_OPENCV_VIDEOIO)
#include <opencv2/videoio.hpp>
#endif
/*!
This example allows to record a video from a camera.
It only requires that ViSP is build with OpenCV.
Example to save an mpeg video:
./tutorial-video-recorder --device 0 --name myvideo.mp4
Example to save a sequence of png images:
./tutorial-video-recorder --device 0 --name image%04d.png
Example to save one imags:
./tutorial-video-recorder --device 0 --name image.jpeg
*/
int main(int argc, const char *argv [])
{
#if ((defined(VISP_HAVE_V4L2) || defined(HAVE_OPENCV_VIDEOIO)) && \
(defined(VISP_HAVE_X11) || defined(VISP_HAVE_GDI) || defined(HAVE_OPENCV_HIGHGUI) || defined(VISP_HAVE_GTK)))
std::string opt_videoname = "video-recorded.mpg";
int opt_device = 0;
for (int i = 0; i < argc; i++) {
if (std::string(argv[i]) == "--device")
opt_device = atoi(argv[i + 1]);
else if (std::string(argv[i]) == "--name")
opt_videoname = std::string(argv[i + 1]);
else if (std::string(argv[i]) == "--help" || std::string(argv[i]) == "-h") {
std::cout << "\nUsage: " << argv[0] << " [--device <device number>] [--name <video name>] [--help][-h]\n"
<< std::endl;
return EXIT_SUCCESS;
}
}
std::cout << "Use device: " << opt_device << std::endl;
std::cout << "Record video in: " << opt_videoname << std::endl;
try {
// vpImage<vpRGBa> I; // for color images
vpImage<unsigned char> I; // for gray images
#if defined(VISP_HAVE_V4L2)
vpV4l2Grabber g;
std::ostringstream device;
device << "/dev/video" << opt_device;
g.setDevice(device.str());
g.setScale(1); // Acquire full resolution images
g.open(I);
g.acquire(I);
#elif defined(HAVE_OPENCV_VIDEOIO)
cv::VideoCapture g(opt_device);
if (!g.isOpened()) { // check if we succeeded
std::cout << "Failed to open the camera" << std::endl;
return EXIT_FAILURE;
}
cv::Mat frame;
g >> frame; // get a new frame from camera
vpImageConvert::convert(frame, I);
#endif
std::cout << "Image size: " << I.getWidth() << " " << I.getHeight() << std::endl;
#if defined(VISP_HAVE_X11)
vpDisplayX d;
#elif defined(VISP_HAVE_GDI)
vpDisplayGDI d;
#elif defined(HAVE_OPENCV_HIGHGUI)
vpDisplayOpenCV d;
#elif defined(VISP_HAVE_GTK)
vpDisplayGTK d;
#endif
d.init(I, 0, 0, "Camera view");
vpVideoWriter writer;
#if defined(HAVE_OPENCV_VIDEOIO)
#if (VISP_HAVE_OPENCV_VERSION >= 0x030000)
writer.setCodec(cv::VideoWriter::fourcc('P', 'I', 'M', '1')); // MPEG-1 codec
#else
writer.setCodec(CV_FOURCC('P', 'I', 'M', '1'));
#endif
#endif
writer.setFileName(opt_videoname);
writer.open(I);
bool recording = false;
for (;;) {
#if defined(VISP_HAVE_V4L2)
g.acquire(I);
#elif defined(HAVE_OPENCV_VIDEOIO)
g >> frame;
vpImageConvert::convert(frame, I);
#endif
vpDisplay::display(I);
if (recording == false) {
vpDisplay::displayText(I, 10, 10, "A click to start recording", vpColor::green);
if (vpDisplay::getClick(I, false))
recording = true;
}
else {
writer.saveFrame(I);
vpDisplay::displayText(I, 10, 10, "Recording: A click to stop and exit", vpColor::red);
if (vpDisplay::getClick(I, false))
break;
}
vpDisplay::flush(I);
}
std::cout << "The video was recorded in \"" << opt_videoname << "\"" << std::endl;
}
catch (const vpException &e) {
std::cout << "Catch an exception: " << e << std::endl;
}
#else
(void)argc;
(void)argv;
std::cout << "Install OpenCV and rebuild ViSP to use this example." << std::endl;
#endif
return EXIT_SUCCESS;
}
|
