FloatImg/v4l2/grabvidseq.c

452 lines
11 KiB
C
Raw Normal View History

2019-05-25 17:02:30 +02:00
/* V4L2 video picture grabber
2019-05-28 11:56:02 +02:00
Origin :V4L2GRAB.C - patched by tTh
2019-05-25 17:02:30 +02:00
Copyright (C) 2009 Mauro Carvalho Chehab <mchehab@infradead.org>
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation version 2 of the License.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
*/
2019-05-28 11:56:02 +02:00
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>
#include <errno.h>
#include <sys/ioctl.h>
#include <sys/types.h>
#include <sys/time.h>
#include <sys/mman.h>
#include <linux/videodev2.h>
#include <libv4l2.h>
2019-05-28 12:10:55 +02:00
#include "../floatimg.h"
2019-08-10 18:37:52 +02:00
#include "funcs.h"
/* --------------------------------------------------------------------- */
/* compilation control */
2019-08-10 21:26:39 +02:00
#define SAVE_AS_CUMUL 1
#define SAVE_AS_FIMG 0
2019-08-10 18:37:52 +02:00
2020-12-27 02:50:41 +01:00
#define NBR_BUFFERS 4
2019-08-10 18:37:52 +02:00
/* --------------------------------------------------------------------- */
2019-05-28 11:56:02 +02:00
#define CLEAR(x) memset(&(x), 0, sizeof(x))
struct buffer {
void *start;
size_t length;
};
2019-05-25 17:02:30 +02:00
2021-02-28 18:18:26 +01:00
int verbosity;
static int systrace;
2019-05-27 17:11:43 +02:00
/* --------------------------------------------------------------------- */
static void xioctl(int fh, int request, void *arg)
{
2020-07-24 14:04:54 +02:00
int r;
2019-05-27 17:11:43 +02:00
2021-02-28 18:18:26 +01:00
/* may be imagine a system for displaying all call to this function ? */
if (systrace) {
fprintf(stderr, "xioctl fd=%d req=%d arg=%p\n", fh, request, arg);
}
2020-07-24 14:04:54 +02:00
do {
2019-05-27 17:11:43 +02:00
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));
2019-07-20 04:09:39 +02:00
sleep(1);
2019-05-27 17:11:43 +02:00
exit(EXIT_FAILURE);
}
}
/* --------------------------------------------------------------------- */
void help(int v)
{
2019-08-12 22:20:25 +02:00
if (verbosity) {
printf("compiled %s at %s\n", __DATE__, __TIME__);
2020-01-05 12:30:48 +01:00
fimg_print_version(1);
2019-08-12 22:20:25 +02:00
}
2019-05-27 17:11:43 +02:00
puts("options :");
puts("\t-d /dev/?\tselect video device");
2019-09-02 11:47:51 +02:00
puts("\t-g\t\tconvert to gray");
2022-05-14 20:49:59 +02:00
puts("\t-m\t\ttry to add metadata");
2019-05-27 17:11:43 +02:00
puts("\t-n NNN\t\thow many frames ?");
2019-05-28 14:55:40 +02:00
puts("\t-O ./\t\tset Output dir");
puts("\t-o bla.xxx\tset output filename");
puts("\t-p NN.N\t\tperiod in seconds");
2020-07-04 19:00:04 +02:00
puts("\t-r NNN\t\trotate picture");
2019-09-02 04:31:27 +02:00
puts("\t-s WxH\t\tsize of capture");
puts("\t-c mode\t\tcontrast enhancement");
puts("\t-u\t\ttry upscaling...");
2019-05-27 17:11:43 +02:00
puts("\t-v\t\tincrease verbosity");
2021-02-28 18:18:26 +01:00
puts("\t-Z\t\tenable systrace");
2022-05-14 20:49:59 +02:00
if (verbosity > 1) {
2019-11-29 13:24:12 +01:00
puts("\n\t\tXXX list all the contrast modes, please\n");
}
2019-05-27 17:11:43 +02:00
exit(0);
}
/* --------------------------------------------------------------------- */
2019-05-25 17:02:30 +02:00
2019-05-28 11:56:02 +02:00
int main(int argc, char **argv)
{
struct v4l2_format fmt;
struct v4l2_buffer buf;
struct v4l2_requestbuffers req;
enum v4l2_buf_type type;
fd_set fds;
struct timeval tv;
int r, fd = -1;
unsigned int i, n_buffers;
char *dev_name = "/dev/video0";
2019-08-14 19:29:11 +02:00
2020-04-02 14:49:10 +02:00
// XXX FILE *fout;
2019-05-28 11:56:02 +02:00
struct buffer *buffers;
2020-04-02 14:49:10 +02:00
int foo;
2021-01-15 13:44:54 +01:00
double period = 10.0; /* delai entre les captures
en secondes */
int nbre_capt = 1; /* nombre de captures */
2019-05-28 11:56:02 +02:00
int opt;
2019-09-02 04:31:27 +02:00
int width = 640;
int height = 480;
double t_final, maxvalue;
2019-09-02 11:47:51 +02:00
int to_gray = 0;
int upscaling = 0;
2020-04-02 14:49:10 +02:00
int contrast = CONTRAST_NONE;
int rotfactor = 0; /* only 0 or 90 here */
char *dest_dir = "."; /* no trailing slash */
2022-05-14 20:49:59 +02:00
char *outfile = "out.fimg";
int add_metadata = 0;
FimgMetaData metadata;
2019-05-27 17:11:43 +02:00
2019-08-10 21:26:39 +02:00
#if SAVE_AS_CUMUL
2020-04-02 14:49:10 +02:00
FloatImg cumul, tmpfimg, *to_save;
2019-08-10 21:26:39 +02:00
#endif
2022-05-14 20:49:59 +02:00
while ((opt = getopt(argc, argv, "c:d:ghmn:o:O:p:r:s:uvZ")) != -1) {
2019-05-27 17:11:43 +02:00
switch(opt) {
2019-11-18 10:18:30 +01:00
case 'c': contrast = fimg_id_contraste(optarg);
if (contrast < 0) {
fputs("unknow contrast\n", stderr);
exit(1);
}
2019-11-18 10:18:30 +01:00
break;
2019-05-27 17:11:43 +02:00
case 'd': dev_name = optarg; break;
2019-09-02 11:47:51 +02:00
case 'g': to_gray = 1; break;
2019-05-27 17:11:43 +02:00
case 'h': help(0); break;
2022-05-14 20:49:59 +02:00
case 'm': add_metadata++; break;
2019-05-27 17:11:43 +02:00
case 'n': nbre_capt = atoi(optarg); break;
2019-05-28 14:55:40 +02:00
case 'O': dest_dir = optarg; break;
2019-08-10 21:26:39 +02:00
case 'o': outfile = optarg; break;
2019-09-16 12:28:47 +02:00
case 'p': foo = parse_double(optarg, &period);
if (foo<0) {
fprintf(stderr,
"error parsing -p arg '%s'\n",
optarg);
exit(1);
}
break;
2020-04-02 14:49:10 +02:00
case 'r': rotfactor = atoi(optarg); break;
case 's': parse_WxH(optarg, &width, &height);
break;
case 'u': upscaling = 1; break;
2019-05-27 17:11:43 +02:00
case 'v': verbosity++; break;
2021-02-28 18:18:26 +01:00
case 'Z': systrace = 1; break;
2020-04-02 14:49:10 +02:00
default:
fprintf(stderr, "option '%c' is wtf\n", opt);
exit(1);
2019-05-27 17:11:43 +02:00
}
}
2019-12-05 18:07:56 +01:00
if (verbosity > 1) {
2020-08-13 08:03:48 +02:00
fprintf(stderr, "*** GrabVidSeq (%s, %s) libv %d, pid=%d\n",
__DATE__, __TIME__, FIMG_VERSION, getpid());
fprintf(stderr, "grabing %d picz from %s, ", nbre_capt, dev_name);
2021-01-15 13:44:54 +01:00
fprintf(stderr, "period is %.3f seconds\n", period);
2019-09-02 04:31:27 +02:00
fprintf(stderr, "framesize is %dx%d\n", width, height);
2020-07-24 19:11:16 +02:00
// fprintf(stderr, "destdir is '%s'\n", dest_dir);
2019-11-14 22:21:18 +01:00
if (upscaling) fprintf(stderr, "upscaling is on\n");
2019-07-02 07:07:38 +02:00
}
2019-05-27 17:11:43 +02:00
2021-02-21 16:42:47 +01:00
if (upscaling && (nbre_capt%4)) {
fprintf(stderr, "WARN upscaling: %d bad nbre_capt\n",
nbre_capt);
}
2019-05-27 17:11:43 +02:00
fd = v4l2_open(dev_name, O_RDWR | O_NONBLOCK, 0);
if (fd < 0) {
2019-07-20 04:09:39 +02:00
perror(dev_name);
2019-05-27 17:11:43 +02:00
exit(EXIT_FAILURE);
}
CLEAR(fmt);
fmt.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
2019-09-02 04:31:27 +02:00
fmt.fmt.pix.width = width;
fmt.fmt.pix.height = height;
2019-05-27 17:11:43 +02:00
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) {
2019-09-02 11:47:51 +02:00
/* are others formats usable ? */
2019-11-14 22:21:18 +01:00
fprintf(stderr, "Libv4l didn't accept RGB24 format. Can't proceed.\n");
2019-05-27 17:11:43 +02:00
exit(EXIT_FAILURE);
}
2019-09-02 04:31:27 +02:00
if ((fmt.fmt.pix.width != width) || (fmt.fmt.pix.height != height)) {
2019-11-14 22:21:18 +01:00
fprintf(stderr, "Warning: driver is sending image at %dx%d\n",
2019-05-25 17:02:30 +02:00
fmt.fmt.pix.width, fmt.fmt.pix.height);
2019-05-27 17:11:43 +02:00
}
2020-12-29 00:54:15 +01:00
// fprintf(stderr,"--- Ok 1\n");
2019-05-27 17:11:43 +02:00
CLEAR(req);
2020-12-27 02:50:41 +01:00
req.count = NBR_BUFFERS;
2019-05-27 17:11:43 +02:00
req.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
req.memory = V4L2_MEMORY_MMAP;
xioctl(fd, VIDIOC_REQBUFS, &req);
buffers = calloc(req.count, sizeof(*buffers));
for (n_buffers = 0; n_buffers < req.count; ++n_buffers) {
CLEAR(buf);
buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
buf.memory = V4L2_MEMORY_MMAP;
buf.index = n_buffers;
xioctl(fd, VIDIOC_QUERYBUF, &buf);
buffers[n_buffers].length = buf.length;
buffers[n_buffers].start = v4l2_mmap(NULL, buf.length,
PROT_READ | PROT_WRITE, MAP_SHARED,
fd, buf.m.offset);
if (MAP_FAILED == buffers[n_buffers].start) {
2019-07-02 19:36:55 +02:00
perror("v4l2_mmap");
2019-05-27 17:11:43 +02:00
exit(EXIT_FAILURE);
}
}
2019-08-14 19:29:11 +02:00
(void)fimg_timer_set(0);
2019-07-02 07:07:38 +02:00
2020-12-27 02:50:41 +01:00
for (i = 0; i < NBR_BUFFERS; ++i) {
2019-05-27 17:11:43 +02:00
CLEAR(buf);
buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
buf.memory = V4L2_MEMORY_MMAP;
buf.index = i;
xioctl(fd, VIDIOC_QBUF, &buf);
}
2019-08-10 21:26:39 +02:00
#if SAVE_AS_CUMUL
if (upscaling) {
foo = fimg_create(&cumul,
fmt.fmt.pix.width*2, fmt.fmt.pix.height*2,
FIMG_TYPE_RGB);
}
else {
foo = fimg_create(&cumul,
fmt.fmt.pix.width, fmt.fmt.pix.height,
FIMG_TYPE_RGB);
}
2019-08-10 21:26:39 +02:00
fimg_clear(&cumul);
2020-08-31 17:51:12 +02:00
cumul.fval = 255.0; /* must be read from camera XXX */
2019-09-02 11:47:51 +02:00
cumul.count = 0;
2020-04-02 14:49:10 +02:00
to_save = &cumul;
2019-08-10 21:26:39 +02:00
#endif
2019-05-27 17:11:43 +02:00
type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
xioctl(fd, VIDIOC_STREAMON, &type);
2019-07-02 07:17:11 +02:00
#if 1
2023-02-13 14:27:01 +01:00
if (verbosity) fprintf(stderr,"pid %d is going to grab %d frames...\n",
2019-07-02 19:36:55 +02:00
getpid(), nbre_capt);
#endif
2019-07-02 07:17:11 +02:00
2020-12-26 19:04:22 +01:00
/*
* START ON THE GRABBING LOOP
*/
2019-05-27 17:11:43 +02:00
for (i = 0; i < nbre_capt; i++) {
do {
FD_ZERO(&fds);
FD_SET(fd, &fds);
/* Timeout. */
tv.tv_sec = 2;
tv.tv_usec = 0;
r = select(fd + 1, &fds, NULL, NULL, &tv);
} while ((r == -1 && (errno = EINTR)));
if (r == -1) {
perror("select");
2020-12-26 19:04:22 +01:00
return errno; /* WTF ? a rogue return
from the main() ? */
2019-05-27 17:11:43 +02:00
}
2019-08-14 19:29:11 +02:00
if(verbosity > 1) {
2021-01-15 13:44:54 +01:00
fprintf(stderr, "%6d / %6d %9.3f\r", i, nbre_capt,
fimg_timer_get(0));
fflush(stderr);
2019-08-14 19:29:11 +02:00
}
2019-05-27 17:11:43 +02:00
CLEAR(buf);
buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
buf.memory = V4L2_MEMORY_MMAP;
xioctl(fd, VIDIOC_DQBUF, &buf);
2020-12-27 02:50:41 +01:00
if(verbosity > 2) {
fprintf(stderr, "xioctl VIDIOC_DQBUF done\n");
fflush(stderr);
}
2019-08-10 21:26:39 +02:00
#if SAVE_AS_CUMUL
if (upscaling) {
x_upscaler_0(buffers[buf.index].start,
fmt.fmt.pix.width, fmt.fmt.pix.height, &cumul);
}
else {
x_add_rgb2fimg(buffers[buf.index].start,
fmt.fmt.pix.width, fmt.fmt.pix.height, &cumul);
}
2019-08-10 18:37:52 +02:00
#endif
#if SAVE_AS_FIMG
sprintf(out_name, "%s/%05d.fimg", dest_dir, i);
if (verbosity > 1) fprintf(stderr, "--> %s\n", out_name);
foo = x_rgb2file(buffers[buf.index].start,
fmt.fmt.pix.width, fmt.fmt.pix.height,
out_name);
#endif
2022-07-06 10:27:55 +02:00
if (nbre_capt > 1 && period > 0.0001) {
2021-01-15 13:44:54 +01:00
/* suspend execution for
microsecond intervals */
usleep((int)(period*1E6));
2019-05-27 17:11:43 +02:00
}
xioctl(fd, VIDIOC_QBUF, &buf);
}
2019-05-25 17:02:30 +02:00
2019-10-30 20:09:32 +01:00
if (verbosity) {
t_final = fimg_timer_get(0);
2019-12-12 12:50:08 +01:00
fprintf(stderr, "pid %d : elapsed %.3g s -> %.2f fps\n", getpid(),
t_final, (double)nbre_capt / t_final);
2019-10-30 20:09:32 +01:00
}
2019-08-10 21:26:39 +02:00
2019-09-02 11:47:51 +02:00
if (to_gray) {
if (verbosity) fputs("converting to gray\n", stderr);
foo = fimg_to_gray(&cumul);
}
2020-04-02 14:49:10 +02:00
2019-10-30 20:09:32 +01:00
#if SAVE_AS_CUMUL
2019-08-10 21:26:39 +02:00
// save cumul to file
2019-10-30 20:09:32 +01:00
if (verbosity) fprintf(stderr, "saving cumul to '%s'\n", outfile);
/* ----- nouveau 15 nov 2019 */
maxvalue = cumul.fval * cumul.count;
if (verbosity) {
2020-08-03 18:29:03 +02:00
fprintf(stderr, "theorical maxvalue = %g\n", maxvalue);
2020-08-31 17:51:12 +02:00
fprintf(stderr, "computed max value = %g\n",
fimg_get_maxvalue(&cumul));
}
switch (contrast) {
case CONTRAST_NONE:
// if (verbosity) fprintf(stderr, "contrast: none\n");
break;
case CONTRAST_SQRT:
fimg_square_root(&cumul, NULL, maxvalue);
break;
case CONTRAST_POW2:
fimg_power_2(&cumul, NULL, maxvalue);
break;
case CONTRAST_COS01:
fimg_cos_01(&cumul, NULL, maxvalue);
break;
case CONTRAST_COS010:
fimg_cos_010(&cumul, NULL, maxvalue);
break;
default:
fprintf(stderr, "bad contrast method\n");
break;
}
2022-07-06 10:27:55 +02:00
/* XXX for the debug of metadata thing */
// fprintf(stderr, "after grab & contrast: fval %f count %d\n",
// cumul.fval, cumul.count);
2020-04-02 14:49:10 +02:00
/* XXX warning, new from coronahome 26 mars 2020 */
2020-07-04 19:32:33 +02:00
to_save = &cumul;
2020-04-02 14:49:10 +02:00
if (90 == rotfactor) {
memset(&tmpfimg, 0, sizeof(FloatImg));
foo = fimg_rotate_90(&cumul, &tmpfimg, 0);
2020-07-04 19:32:33 +02:00
if (verbosity > 2) {
2020-04-02 14:49:10 +02:00
fprintf(stderr, "dump rot90 %p\n", &tmpfimg);
foo = fimg_save_as_png(&tmpfimg, "rot90.png", 0);
}
to_save = &tmpfimg;
2022-07-06 10:27:55 +02:00
/*
* CRITICAL : we have to restore those values !
*/
to_save->fval = cumul.fval;
to_save->count = cumul.count;
2020-04-02 14:49:10 +02:00
}
2019-10-30 20:09:32 +01:00
foo = format_from_extension(outfile);
switch (foo) {
case FILE_TYPE_FIMG:
2022-05-14 20:49:59 +02:00
if (add_metadata) {
2024-05-01 12:30:53 +02:00
fimg_default_metadata(&metadata, 0x51);
2022-07-06 10:27:55 +02:00
/* copy some values to mdstruct */
metadata.fval = to_save->fval;
metadata.count = to_save->count;
2024-05-01 12:30:53 +02:00
fprintf(stderr, "-> save '%s' with metadata\n",
outfile);
2022-05-14 20:49:59 +02:00
foo = fimg_dumpmd_to_file(to_save, outfile, \
&metadata, 0);
}
else {
foo = fimg_dump_to_file(to_save, outfile, 0);
2024-05-01 12:30:53 +02:00
fprintf(stderr, "%s no dump of metadata ?\n", outfile);
2022-05-14 20:49:59 +02:00
}
2019-10-30 20:09:32 +01:00
break;
case FILE_TYPE_PNM:
2020-07-04 19:32:33 +02:00
foo = fimg_save_as_pnm(to_save, outfile, 1);
2019-10-30 20:09:32 +01:00
break;
2020-01-25 17:22:31 +01:00
case FILE_TYPE_PNG:
2020-07-04 19:32:33 +02:00
foo = fimg_save_as_png(to_save, outfile, 0);
2020-01-25 17:22:31 +01:00
break;
2020-07-24 14:04:54 +02:00
case FILE_TYPE_FITS:
foo = fimg_save_R_as_fits(to_save, outfile, 0);
break;
2020-08-23 17:21:10 +02:00
case FILE_TYPE_TIFF:
foo = fimg_write_as_tiff(to_save, outfile, 0);
2020-08-24 11:40:49 +02:00
break;
2020-01-25 17:22:31 +01:00
default:
fprintf(stderr, "can't save as %s\n", outfile);
break;
2019-10-30 20:09:32 +01:00
}
2019-08-10 21:26:39 +02:00
// free buffers
2020-04-02 14:49:10 +02:00
2019-08-10 21:26:39 +02:00
fimg_destroy(&cumul);
#endif
2019-05-28 12:10:55 +02:00
2019-05-27 17:11:43 +02:00
type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
xioctl(fd, VIDIOC_STREAMOFF, &type);
2020-12-27 02:50:41 +01:00
for (i = 0; i < NBR_BUFFERS; ++i) {
2019-05-27 17:11:43 +02:00
v4l2_munmap(buffers[i].start, buffers[i].length);
}
2019-05-25 17:02:30 +02:00
2019-09-03 22:23:06 +02:00
// free(buffers); /* atomic bombing */
2019-05-27 17:11:43 +02:00
v4l2_close(fd);
2019-05-25 17:02:30 +02:00
2019-05-27 17:11:43 +02:00
return 0;
}