/* V4L2 video picture grabber Origin :V4L2GRAB.C - patched by tTh Copyright (C) 2009 Mauro Carvalho Chehab 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. */ #include #include #include #include #include #include #include #include #include #include #include #include "../floatimg.h" #include "funcs.h" /* --------------------------------------------------------------------- */ /* compilation control */ #define SAVE_AS_CUMUL 1 #define SAVE_AS_PNM 0 #define SAVE_AS_FIMG 0 /* --------------------------------------------------------------------- */ #define CLEAR(x) memset(&(x), 0, sizeof(x)) struct buffer { void *start; size_t length; }; int verbosity; /* --------------------------------------------------------------------- */ static void xioctl(int fh, int request, void *arg) { 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)); sleep(1); exit(EXIT_FAILURE); } } /* --------------------------------------------------------------------- */ void help(int v) { if (verbosity) { printf("compiled %s at %s\n", __DATE__, __TIME__); fimg_print_version(1); } puts("options :"); puts("\t-d /dev/?\tselect video device"); puts("\t-g\t\tconvert to gray"); puts("\t-n NNN\t\thow many frames ?"); puts("\t-O ./\t\tset Output dir"); puts("\t-o bla.xxx\tset output filename"); puts("\t-p NN.N\t\tperiod in seconds"); puts("\t-r NNN\t\trotate picture"); puts("\t-s WxH\t\tsize of capture"); puts("\t-c mode\t\tcontrast enhancement"); puts("\t-u\t\ttry upscaling..."); puts("\t-v\t\tincrease verbosity"); if (verbosity) { puts("\n\t\tXXX list all the contrast modes, please\n"); } exit(0); } /* --------------------------------------------------------------------- */ 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"; // XXX FILE *fout; struct buffer *buffers; int foo; double period = 10.0; /* delai entre les captures */ int nbre_capt = 1; /* nombre de captures */ int opt; int width = 640; int height = 480; double t_final, maxvalue; int to_gray = 0; int upscaling = 0; int contrast = CONTRAST_NONE; int rotfactor = 0; /* only 0 or 90 here */ char *dest_dir = "."; /* no trailing slash */ char *outfile = "out.pnm"; #if SAVE_AS_CUMUL FloatImg cumul, tmpfimg, *to_save; #endif while ((opt = getopt(argc, argv, "c:d:ghn:o:O:p:r:s:uv")) != -1) { switch(opt) { case 'c': contrast = fimg_id_contraste(optarg); if (contrast < 0) { fputs("unknow contrast\n", stderr); exit(1); } break; case 'd': dev_name = optarg; break; case 'g': to_gray = 1; break; case 'h': help(0); break; case 'n': nbre_capt = atoi(optarg); break; case 'O': dest_dir = optarg; break; case 'o': outfile = optarg; break; case 'p': foo = parse_double(optarg, &period); if (foo<0) { fprintf(stderr, "error parsing -p arg '%s'\n", optarg); exit(1); } period *= 1e6; break; case 'r': rotfactor = atoi(optarg); break; case 's': parse_WxH(optarg, &width, &height); break; case 'u': upscaling = 1; break; case 'v': verbosity++; break; default: fprintf(stderr, "option '%c' is wtf\n", opt); exit(1); } } if (verbosity > 1) { fprintf(stderr, "*** GrabVidSeq (%s, %s) libv %d, pid=%d\n", __DATE__, __TIME__, FIMG_VERSION, getpid()); fprintf(stderr, "grabing %d picz, ", nbre_capt); fprintf(stderr, "period is %.3f milliseconds\n", period/1e3); fprintf(stderr, "framesize is %dx%d\n", width, height); // fprintf(stderr, "destdir is '%s'\n", dest_dir); if (upscaling) fprintf(stderr, "upscaling is on\n"); } fd = v4l2_open(dev_name, O_RDWR | O_NONBLOCK, 0); if (fd < 0) { perror(dev_name); exit(EXIT_FAILURE); } CLEAR(fmt); fmt.type = V4L2_BUF_TYPE_VIDEO_CAPTURE; fmt.fmt.pix.width = width; fmt.fmt.pix.height = height; 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) { /* are others formats usable ? */ fprintf(stderr, "Libv4l didn't accept RGB24 format. Can't proceed.\n"); exit(EXIT_FAILURE); } if ((fmt.fmt.pix.width != width) || (fmt.fmt.pix.height != height)) { fprintf(stderr, "Warning: driver is sending image at %dx%d\n", fmt.fmt.pix.width, fmt.fmt.pix.height); } CLEAR(req); req.count = 2; 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) { perror("v4l2_mmap"); exit(EXIT_FAILURE); } } (void)fimg_timer_set(0); for (i = 0; i < n_buffers; ++i) { CLEAR(buf); buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE; buf.memory = V4L2_MEMORY_MMAP; buf.index = i; xioctl(fd, VIDIOC_QBUF, &buf); } #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); } fimg_clear(&cumul); cumul.fval = 255.0; cumul.count = 0; to_save = &cumul; #endif type = V4L2_BUF_TYPE_VIDEO_CAPTURE; xioctl(fd, VIDIOC_STREAMON, &type); #if 0 if (verbosity) fprintf(stderr,"pid %d grabbing %d picz...\n", getpid(), nbre_capt); #endif 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"); return errno; } if(verbosity > 1) { printf("%6d / %6d %9.3f\r", i, nbre_capt, fimg_timer_get(0)); } CLEAR(buf); buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE; buf.memory = V4L2_MEMORY_MMAP; xioctl(fd, VIDIOC_DQBUF, &buf); #if SAVE_AS_PNM /* at this time,we'v got a picture in-memory, so we can blast in on storage */ sprintf(out_name, "%s/%05d.ppm", dest_dir, i); if (verbosity > 1) 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); fwrite(buffers[buf.index].start, buf.bytesused, 1, fout); fclose(fout); #endif #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); } #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 if (nbre_capt > 1 && period > 1.0) { usleep(period); } xioctl(fd, VIDIOC_QBUF, &buf); } fflush(stdout); if (verbosity) { t_final = fimg_timer_get(0); fprintf(stderr, "pid %d : elapsed %.3g s -> %.2f fps\n", getpid(), t_final, (double)nbre_capt / t_final); } if (to_gray) { if (verbosity) fputs("converting to gray\n", stderr); foo = fimg_to_gray(&cumul); } #if SAVE_AS_CUMUL // save cumul to file if (verbosity) fprintf(stderr, "saving cumul to '%s'\n", outfile); /* ----- nouveau 15 nov 2019 */ maxvalue = cumul.fval * cumul.count; if (verbosity) { fprintf(stderr, "theorical maxvalue = %g\n", maxvalue); } 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; } /* XXX warning, new from coronahome 26 mars 2020 */ to_save = &cumul; if (90 == rotfactor) { memset(&tmpfimg, 0, sizeof(FloatImg)); foo = fimg_rotate_90(&cumul, &tmpfimg, 0); if (verbosity > 2) { fprintf(stderr, "dump rot90 %p\n", &tmpfimg); foo = fimg_save_as_png(&tmpfimg, "rot90.png", 0); } to_save = &tmpfimg; } foo = format_from_extension(outfile); switch (foo) { case FILE_TYPE_FIMG: foo = fimg_dump_to_file(to_save, outfile, 0); break; case FILE_TYPE_PNM: foo = fimg_save_as_pnm(to_save, outfile, 1); break; case FILE_TYPE_PNG: foo = fimg_save_as_png(to_save, outfile, 0); break; case FILE_TYPE_FITS: foo = fimg_save_R_as_fits(to_save, outfile, 0); break; case FILE_TYPE_TIFF: foo = fimg_write_as_tiff(to_save, outfile, 0); break; default: fprintf(stderr, "can't save as %s\n", outfile); break; } // free buffers fimg_destroy(&cumul); #endif type = V4L2_BUF_TYPE_VIDEO_CAPTURE; xioctl(fd, VIDIOC_STREAMOFF, &type); for (i = 0; i < n_buffers; ++i) { v4l2_munmap(buffers[i].start, buffers[i].length); } // free(buffers); /* atomic bombing */ v4l2_close(fd); return 0; }