Mutah
/
FloatImg4PythonBinding
forked from tTh/FloatImg
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

pretty formating the kluge

This commit is contained in:
Tonton Th 2019-05-27 17:11:43 +02:00
parent 2a40718410
commit 1fdd8901a1
1 changed files with 135 additions and 96 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

231
v4l2/v4l2grab.c
View File

@ -30,19 +30,34 @@
size_t length; size_t length;
}; };
static void xioctl(int fh, int request, void *arg) int verbosity;
{
int r;
do { /* --------------------------------------------------------------------- */
r = v4l2_ioctl(fh, request, arg);
} while (r == -1 && ((errno == EINTR) || (errno == EAGAIN)));
if (r == -1) { static void xioctl(int fh, int request, void *arg)
fprintf(stderr, "error %d, %s\n", errno, strerror(errno)); {
exit(EXIT_FAILURE); int r;
}
} do {
r = v4l2_ioctl(fh, request, arg);
} while (r == -1 && ((errno == EINTR) || (errno == EAGAIN)));
if (r == -1) {
fprintf(stderr, "error %d, %s\n", errno, strerror(errno));
exit(EXIT_FAILURE);
}
}
/* --------------------------------------------------------------------- */
void help(int v)
{
puts("options :");
puts("\t-d /dev/?\tselect video device");
puts("\t-n NNN\t\thow many frames ?");
puts("\t-p NNN\t\tperiod in seconds");
puts("\t-v\t\tincrease verbosity");
exit(0);
}
/* --------------------------------------------------------------------- */
int main(int argc, char **argv) int main(int argc, char **argv)
{ {
@ -59,106 +74,130 @@
FILE *fout; FILE *fout;
struct buffer *buffers; struct buffer *buffers;
fd = v4l2_open(dev_name, O_RDWR | O_NONBLOCK, 0); int period = 10; /* delai entre les captures */
if (fd < 0) { int nbre_capt = 1; /* nombre de captures */
perror("Cannot open device"); int opt;
exit(EXIT_FAILURE);
}
CLEAR(fmt); while ((opt = getopt(argc, argv, "d:hn:p:v")) != -1) {
fmt.type = V4L2_BUF_TYPE_VIDEO_CAPTURE; switch(opt) {
fmt.fmt.pix.width = 640; case 'd': dev_name = optarg; break;
fmt.fmt.pix.height = 480; case 'h': help(0); break;
fmt.fmt.pix.pixelformat = V4L2_PIX_FMT_RGB24; case 'n': nbre_capt = atoi(optarg); break;
fmt.fmt.pix.field = V4L2_FIELD_INTERLACED; case 'p': period = atoi(optarg); break;
xioctl(fd, VIDIOC_S_FMT, &fmt); case 'v': verbosity++; break;
if (fmt.fmt.pix.pixelformat != V4L2_PIX_FMT_RGB24) { }
printf("Libv4l didn't accept RGB24 format. Can't proceed.\n"); }
exit(EXIT_FAILURE);
}
if ((fmt.fmt.pix.width != 640) || (fmt.fmt.pix.height != 480))
printf("Warning: driver is sending image at %dx%d\n", fd = v4l2_open(dev_name, O_RDWR | O_NONBLOCK, 0);
if (fd < 0) {
perror("Cannot open device");
exit(EXIT_FAILURE);
}
CLEAR(fmt);
fmt.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
fmt.fmt.pix.width = 640;
fmt.fmt.pix.height = 480;
fmt.fmt.pix.pixelformat = V4L2_PIX_FMT_RGB24;
fmt.fmt.pix.field = V4L2_FIELD_INTERLACED;
xioctl(fd, VIDIOC_S_FMT, &fmt);
if (fmt.fmt.pix.pixelformat != V4L2_PIX_FMT_RGB24) {
printf("Libv4l didn't accept RGB24 format. Can't proceed.\n");
exit(EXIT_FAILURE);
}
if ((fmt.fmt.pix.width != 640) || (fmt.fmt.pix.height != 480)) {
printf("Warning: driver is sending image at %dx%d\n",
fmt.fmt.pix.width, fmt.fmt.pix.height); fmt.fmt.pix.width, fmt.fmt.pix.height);
}
CLEAR(req); CLEAR(req);
req.count = 2; req.count = 2;
req.type = V4L2_BUF_TYPE_VIDEO_CAPTURE; req.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
req.memory = V4L2_MEMORY_MMAP; req.memory = V4L2_MEMORY_MMAP;
xioctl(fd, VIDIOC_REQBUFS, &req); xioctl(fd, VIDIOC_REQBUFS, &req);
buffers = calloc(req.count, sizeof(*buffers)); buffers = calloc(req.count, sizeof(*buffers));
for (n_buffers = 0; n_buffers < req.count; ++n_buffers) { for (n_buffers = 0; n_buffers < req.count; ++n_buffers) {
CLEAR(buf); CLEAR(buf);
buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE; buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
buf.memory = V4L2_MEMORY_MMAP; buf.memory = V4L2_MEMORY_MMAP;
buf.index = n_buffers; buf.index = n_buffers;
xioctl(fd, VIDIOC_QUERYBUF, &buf); xioctl(fd, VIDIOC_QUERYBUF, &buf);
buffers[n_buffers].length = buf.length; buffers[n_buffers].length = buf.length;
buffers[n_buffers].start = v4l2_mmap(NULL, buf.length, buffers[n_buffers].start = v4l2_mmap(NULL, buf.length,
PROT_READ | PROT_WRITE, MAP_SHARED, PROT_READ | PROT_WRITE, MAP_SHARED,
fd, buf.m.offset); fd, buf.m.offset);
if (MAP_FAILED == buffers[n_buffers].start) { if (MAP_FAILED == buffers[n_buffers].start) {
perror("mmap"); perror("mmap");
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
} }
} }
for (i = 0; i < n_buffers; ++i) { for (i = 0; i < n_buffers; ++i) {
CLEAR(buf); CLEAR(buf);
buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE; buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
buf.memory = V4L2_MEMORY_MMAP; buf.memory = V4L2_MEMORY_MMAP;
buf.index = i; buf.index = i;
xioctl(fd, VIDIOC_QBUF, &buf); xioctl(fd, VIDIOC_QBUF, &buf);
} }
type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
xioctl(fd, VIDIOC_STREAMON, &type); type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
for (i = 0; i < 10; i++) {
do {
FD_ZERO(&fds);
FD_SET(fd, &fds);
/* Timeout. */ xioctl(fd, VIDIOC_STREAMON, &type);
tv.tv_sec = 2; for (i = 0; i < nbre_capt; i++) {
tv.tv_usec = 0; do {
FD_ZERO(&fds);
FD_SET(fd, &fds);
r = select(fd + 1, &fds, NULL, NULL, &tv); /* Timeout. */
} while ((r == -1 && (errno = EINTR))); tv.tv_sec = 2;
if (r == -1) { tv.tv_usec = 0;
perror("select");
return errno;
}
CLEAR(buf); r = select(fd + 1, &fds, NULL, NULL, &tv);
buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE; } while ((r == -1 && (errno = EINTR)));
buf.memory = V4L2_MEMORY_MMAP;
xioctl(fd, VIDIOC_DQBUF, &buf);
sprintf(out_name, "out%03d.ppm", i); if (r == -1) {
fprintf(stderr, "--> %s\n", out_name); perror("select");
fout = fopen(out_name, "w"); return errno;
if (!fout) { }
perror("Cannot open image");
exit(EXIT_FAILURE); CLEAR(buf);
} buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
fprintf(fout, "P6\n%d %d 255\n", buf.memory = V4L2_MEMORY_MMAP;
xioctl(fd, VIDIOC_DQBUF, &buf);
sprintf(out_name, "out%03d.ppm", i);
if (verbosity) fprintf(stderr, "--> %s\n", out_name);
fout = fopen(out_name, "w");
if (!fout) {
perror("Cannot open image");
exit(EXIT_FAILURE);
}
fprintf(fout, "P6\n%d %d 255\n",
fmt.fmt.pix.width, fmt.fmt.pix.height); fmt.fmt.pix.width, fmt.fmt.pix.height);
fwrite(buffers[buf.index].start, buf.bytesused, 1, fout); fwrite(buffers[buf.index].start, buf.bytesused, 1, fout);
fclose(fout); fclose(fout);
sleep(666); if (nbre_capt > 1 && period) {
sleep(period);
}
xioctl(fd, VIDIOC_QBUF, &buf); xioctl(fd, VIDIOC_QBUF, &buf);
} }
type = V4L2_BUF_TYPE_VIDEO_CAPTURE; type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
xioctl(fd, VIDIOC_STREAMOFF, &type); xioctl(fd, VIDIOC_STREAMOFF, &type);
for (i = 0; i < n_buffers; ++i) for (i = 0; i < n_buffers; ++i) {
v4l2_munmap(buffers[i].start, buffers[i].length); v4l2_munmap(buffers[i].start, buffers[i].length);
v4l2_close(fd); }
return 0; v4l2_close(fd);
}
return 0;
}