Added main loop for loading new frames

This commit is contained in:
Martijn Braam 2020-01-25 14:34:37 +01:00
parent 6190add3bc
commit 38ec7621fc
No known key found for this signature in database
GPG Key ID: C4280ACB000B060F
2 changed files with 165 additions and 153 deletions

315
main.c
View File

@ -19,6 +19,7 @@ struct buffer {
}; };
struct buffer *buffers; struct buffer *buffers;
static int *outbuffer;
static unsigned int n_buffers; static unsigned int n_buffers;
static char *rear_dev_name; static char *rear_dev_name;
static char *front_dev_name; static char *front_dev_name;
@ -53,54 +54,54 @@ start_capturing(int fd)
{ {
enum v4l2_buf_type type; enum v4l2_buf_type type;
switch (io) { switch (io) {
case IO_METHOD_READ: case IO_METHOD_READ:
/* Nothing to do. */ /* Nothing to do. */
break; break;
case IO_METHOD_MMAP: case IO_METHOD_MMAP:
for (int i = 0; i < n_buffers; ++i) { for (int i = 0; i < n_buffers; ++i) {
struct v4l2_buffer buf = { struct v4l2_buffer buf = {
.type = V4L2_BUF_TYPE_VIDEO_CAPTURE, .type = V4L2_BUF_TYPE_VIDEO_CAPTURE,
.memory = V4L2_MEMORY_MMAP, .memory = V4L2_MEMORY_MMAP,
.index = i, .index = i,
}; };
if (xioctl(fd, VIDIOC_QBUF, &buf) == -1) { if (xioctl(fd, VIDIOC_QBUF, &buf) == -1) {
errno_exit("VIDIOC_QBUF"); errno_exit("VIDIOC_QBUF");
}
} }
}
type = V4L2_BUF_TYPE_VIDEO_CAPTURE; type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
if (xioctl(fd, VIDIOC_STREAMON, &type) == -1) { if (xioctl(fd, VIDIOC_STREAMON, &type) == -1) {
errno_exit("VIDIOC_STREAMON"); errno_exit("VIDIOC_STREAMON");
}
break;
case IO_METHOD_USERPTR:
for (int i = 0; i < n_buffers; ++i) {
struct v4l2_buffer buf = {
.type = V4L2_BUF_TYPE_VIDEO_CAPTURE,
.memory = V4L2_MEMORY_USERPTR,
.index = i,
};
buf.m.userptr = (unsigned long)buffers[i].start;
buf.length = buffers[i].length;
if (xioctl(fd, VIDIOC_QBUF, &buf) == -1) {
errno_exit("VIDIOC_QBUF");
} }
} break;
case IO_METHOD_USERPTR:
for (int i = 0; i < n_buffers; ++i) {
struct v4l2_buffer buf = {
.type = V4L2_BUF_TYPE_VIDEO_CAPTURE,
.memory = V4L2_MEMORY_USERPTR,
.index = i,
};
buf.m.userptr = (unsigned long) buffers[i].start;
buf.length = buffers[i].length;
type = V4L2_BUF_TYPE_VIDEO_CAPTURE; if (xioctl(fd, VIDIOC_QBUF, &buf) == -1) {
if (xioctl(fd, VIDIOC_STREAMON, &type) == -1) { errno_exit("VIDIOC_QBUF");
errno_exit("VIDIOC_STREAMON"); }
} }
break;
type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
if (xioctl(fd, VIDIOC_STREAMON, &type) == -1) {
errno_exit("VIDIOC_STREAMON");
}
break;
} }
} }
static void static void
init_mmap(int fd) init_mmap(int fd)
{ {
struct v4l2_requestbuffers req = { 0 }; struct v4l2_requestbuffers req = {0};
req.count = 4; req.count = 4;
req.type = V4L2_BUF_TYPE_VIDEO_CAPTURE; req.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
req.memory = V4L2_MEMORY_MMAP; req.memory = V4L2_MEMORY_MMAP;
@ -108,7 +109,7 @@ init_mmap(int fd)
if (xioctl(fd, VIDIOC_REQBUFS, &req) == -1) { if (xioctl(fd, VIDIOC_REQBUFS, &req) == -1) {
if (errno == EINVAL) { if (errno == EINVAL) {
fprintf(stderr, "%s does not support memory mapping", fprintf(stderr, "%s does not support memory mapping",
dev_name); dev_name);
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
} else { } else {
errno_exit("VIDIOC_REQBUFS"); errno_exit("VIDIOC_REQBUFS");
@ -117,7 +118,7 @@ init_mmap(int fd)
if (req.count < 2) { if (req.count < 2) {
fprintf(stderr, "Insufficient buffer memory on %s\n", fprintf(stderr, "Insufficient buffer memory on %s\n",
dev_name); dev_name);
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
} }
@ -159,7 +160,7 @@ init_device(int fd)
if (xioctl(fd, VIDIOC_QUERYCAP, &cap) == -1) { if (xioctl(fd, VIDIOC_QUERYCAP, &cap) == -1) {
if (errno == EINVAL) { if (errno == EINVAL) {
fprintf(stderr, "%s is no V4L2 device\n", fprintf(stderr, "%s is no V4L2 device\n",
dev_name); dev_name);
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
} else { } else {
errno_exit("VIDIOC_QUERYCAP"); errno_exit("VIDIOC_QUERYCAP");
@ -168,26 +169,26 @@ init_device(int fd)
if (!(cap.capabilities & V4L2_CAP_VIDEO_CAPTURE)) { if (!(cap.capabilities & V4L2_CAP_VIDEO_CAPTURE)) {
fprintf(stderr, "%s is no video capture device\n", fprintf(stderr, "%s is no video capture device\n",
dev_name); dev_name);
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
} }
switch (io) { switch (io) {
case IO_METHOD_READ: case IO_METHOD_READ:
if (!(cap.capabilities & V4L2_CAP_READWRITE)) { if (!(cap.capabilities & V4L2_CAP_READWRITE)) {
fprintf(stderr, "%s does not support read i/o\n", fprintf(stderr, "%s does not support read i/o\n",
dev_name); dev_name);
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
} }
break; break;
case IO_METHOD_MMAP: case IO_METHOD_MMAP:
case IO_METHOD_USERPTR: case IO_METHOD_USERPTR:
if (!(cap.capabilities & V4L2_CAP_STREAMING)) { if (!(cap.capabilities & V4L2_CAP_STREAMING)) {
fprintf(stderr, "%s does not support streaming i/o\n", fprintf(stderr, "%s does not support streaming i/o\n",
dev_name); dev_name);
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
} }
break; break;
} }
/* Select video input, video standard and tune here. */ /* Select video input, video standard and tune here. */
@ -195,19 +196,19 @@ init_device(int fd)
.type = V4L2_BUF_TYPE_VIDEO_CAPTURE, .type = V4L2_BUF_TYPE_VIDEO_CAPTURE,
}; };
struct v4l2_crop crop = { 0 }; struct v4l2_crop crop = {0};
if (xioctl(fd, VIDIOC_CROPCAP, &cropcap) == 0) { if (xioctl(fd, VIDIOC_CROPCAP, &cropcap) == 0) {
crop.type = V4L2_BUF_TYPE_VIDEO_CAPTURE; crop.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
crop.c = cropcap.defrect; /* reset to default */ crop.c = cropcap.defrect; /* reset to default */
if (xioctl(fd, VIDIOC_S_CROP, &crop) == -1) { if (xioctl(fd, VIDIOC_S_CROP, &crop) == -1) {
switch (errno) { switch (errno) {
case EINVAL: case EINVAL:
/* Cropping not supported. */ /* Cropping not supported. */
break; break;
default: default:
/* Errors ignored. */ /* Errors ignored. */
break; break;
} }
} }
} else { } else {
@ -220,7 +221,7 @@ init_device(int fd)
}; };
if (preview_width > 0) { if (preview_width > 0) {
g_printerr("Setting camera to %dx%d fmt %d\n", g_printerr("Setting camera to %dx%d fmt %d\n",
preview_width, preview_height, preview_fmt); preview_width, preview_height, preview_fmt);
fmt.fmt.pix.width = preview_width; fmt.fmt.pix.width = preview_width;
fmt.fmt.pix.height = preview_height; fmt.fmt.pix.height = preview_height;
fmt.fmt.pix.pixelformat = preview_fmt; fmt.fmt.pix.pixelformat = preview_fmt;
@ -231,8 +232,8 @@ init_device(int fd)
} }
g_printerr("Driver returned %dx%d fmt %d\n", g_printerr("Driver returned %dx%d fmt %d\n",
fmt.fmt.pix.width, fmt.fmt.pix.height, fmt.fmt.pix.width, fmt.fmt.pix.height,
fmt.fmt.pix.pixelformat); fmt.fmt.pix.pixelformat);
/* Note VIDIOC_S_FMT may change width and height. */ /* Note VIDIOC_S_FMT may change width and height. */
@ -243,10 +244,12 @@ init_device(int fd)
errno_exit("VIDIOC_G_FMT"); errno_exit("VIDIOC_G_FMT");
} }
g_printerr("Driver returned %dx%d fmt %d\n", g_printerr("Driver returned %dx%d fmt %d\n",
fmt.fmt.pix.width, fmt.fmt.pix.height, fmt.fmt.pix.width, fmt.fmt.pix.height,
fmt.fmt.pix.pixelformat); fmt.fmt.pix.pixelformat);
preview_width = fmt.fmt.pix.width;
preview_height = fmt.fmt.pix.height;
} }
preview_fmt = fmt.fmt.pix.pixelformat;
/* Buggy driver paranoia. */ /* Buggy driver paranoia. */
unsigned int min = fmt.fmt.pix.width * 2; unsigned int min = fmt.fmt.pix.width * 2;
@ -259,109 +262,111 @@ init_device(int fd)
} }
switch (io) { switch (io) {
case IO_METHOD_READ: case IO_METHOD_READ:
//init_read(fmt.fmt.pix.sizeimage); //init_read(fmt.fmt.pix.sizeimage);
break; break;
case IO_METHOD_MMAP: case IO_METHOD_MMAP:
init_mmap(fd); init_mmap(fd);
break; break;
case IO_METHOD_USERPTR: case IO_METHOD_USERPTR:
//init_userp(fmt.fmt.pix.sizeimage); //init_userp(fmt.fmt.pix.sizeimage);
break; break;
} }
} }
static void static void
process_image(const void *p, int size) process_image(const int *p, int size)
{ {
GdkPixbuf *pixbuf = gdk_pixbuf_new_from_data(p, GDK_COLORSPACE_RGB, GdkPixbuf *pixbuf = gdk_pixbuf_new_from_data(p, GDK_COLORSPACE_RGB,
FALSE, 8, 640, 480, 2 * 640, NULL, NULL); FALSE, 8, 640, 480, 2 * 640,
gtk_image_set_from_pixbuf(preview_image, pixbuf); NULL, NULL);
gtk_image_set_from_pixbuf(preview_image, pixbuf
);
} }
static int static int
read_frame(int fd) read_frame(int fd)
{ {
struct v4l2_buffer buf = { 0 }; struct v4l2_buffer buf = {0};
switch (io) { switch (io) {
case IO_METHOD_READ: case IO_METHOD_READ:
if (read(fd, buffers[0].start, buffers[0].length) == -1) { if (read(fd, buffers[0].start, buffers[0].length) == -1) {
switch (errno) { switch (errno) {
case EAGAIN: case EAGAIN:
return 0; return 0;
case EIO: case EIO:
/* Could ignore EIO, see spec. */ /* Could ignore EIO, see spec. */
/* fallthrough */ /* fallthrough */
default: default:
errno_exit("read"); errno_exit("read");
break; break;
}
} }
} process_image(buffers[0].start, buffers[0].length);
process_image(buffers[0].start, buffers[0].length); break;
break; case IO_METHOD_MMAP:
case IO_METHOD_MMAP: buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE; buf.memory = V4L2_MEMORY_MMAP;
buf.memory = V4L2_MEMORY_MMAP; if (xioctl(fd, VIDIOC_DQBUF, &buf) == -1) {
if (xioctl(fd, VIDIOC_DQBUF, &buf) == -1) { switch (errno) {
switch (errno) { case EAGAIN:
case EAGAIN: return 0;
return 0; case EIO:
case EIO: /* Could ignore EIO, see spec. */
/* Could ignore EIO, see spec. */ /* fallthrough */
/* fallthrough */ default:
default: errno_exit("VIDIOC_DQBUF");
errno_exit("VIDIOC_DQBUF"); break;
break; }
} }
}
//assert(buf.index < n_buffers); //assert(buf.index < n_buffers);
process_image(buffers[buf.index].start, buf.bytesused); process_image(buffers[buf.index].start, buf.bytesused);
if (xioctl(fd, VIDIOC_QBUF, &buf) == -1) { if (xioctl(fd, VIDIOC_QBUF, &buf) == -1) {
errno_exit("VIDIOC_QBUF"); errno_exit("VIDIOC_QBUF");
}
break;
case IO_METHOD_USERPTR:
buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
buf.memory = V4L2_MEMORY_USERPTR;
if (xioctl(fd, VIDIOC_DQBUF, &buf) == -1) {
switch (errno) {
case EAGAIN:
return 0;
case EIO:
/* Could ignore EIO, see spec. */
/* fallthrough */
default:
errno_exit("VIDIOC_DQBUF");
break;
} }
} break;
unsigned int i; case IO_METHOD_USERPTR:
for (i = 0; i < n_buffers; ++i) { buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
if (buf.m.userptr == (unsigned long)buffers[i].start buf.memory = V4L2_MEMORY_USERPTR;
if (xioctl(fd, VIDIOC_DQBUF, &buf) == -1) {
switch (errno) {
case EAGAIN:
return 0;
case EIO:
/* Could ignore EIO, see spec. */
/* fallthrough */
default:
errno_exit("VIDIOC_DQBUF");
break;
}
}
unsigned int i;
for (i = 0; i < n_buffers; ++i) {
if (buf.m.userptr == (unsigned long) buffers[i].start
&& buf.length == buffers[i].length) { && buf.length == buffers[i].length) {
break; break;
}
} }
}
//assert(i < n_buffers); //assert(i < n_buffers);
process_image((void *)buf.m.userptr, buf.bytesused); process_image((void *) buf.m.userptr, buf.bytesused);
if (xioctl(fd, VIDIOC_QBUF, &buf) == -1) { if (xioctl(fd, VIDIOC_QBUF, &buf) == -1) {
errno_exit("VIDIOC_QBUF"); errno_exit("VIDIOC_QBUF");
} }
break; break;
} }
return 1; return 1;
} }
static void static gboolean
get_frame(int fd) get_frame(int fd)
{ {
while (1) { while (1) {
@ -393,31 +398,35 @@ get_frame(int fd)
} }
/* EAGAIN - continue select loop. */ /* EAGAIN - continue select loop. */
} }
return TRUE;
} }
static int static int
config_ini_handler(void *user, const char *section, const char *name, config_ini_handler(void *user, const char *section, const char *name,
const char *value) { const char *value)
{
if (strcmp(section, "preview") == 0) { if (strcmp(section, "preview") == 0) {
if (strcmp(name, "width") == 0) { if (strcmp(name, "width") == 0) {
preview_width = strtol(value, NULL, 10); preview_width = strtol(value, NULL, 10);
} else if (strcmp(name, "height") == 0) { } else if (strcmp(name, "height") == 0) {
preview_height = strtol(value, NULL, 10); preview_height = strtol(value, NULL, 10);
} else if (strcmp(name, "fmt") == 0) { } else if (strcmp(name, "fmt") == 0) {
if (strcmp(value, "RGB") == 0){ if (strcmp(value, "RGB") == 0) {
preview_fmt = V4L2_PIX_FMT_RGB24; preview_fmt = V4L2_PIX_FMT_RGB24;
} else if (strcmp(value, "UYVY8") == 0) { } else if (strcmp(value, "UYVY") == 0) {
preview_fmt = V4L2_PIX_FMT_UYVY; preview_fmt = V4L2_PIX_FMT_UYVY;
} else if (strcmp(value, "YUYV") == 0) {
preview_fmt = V4L2_PIX_FMT_YUYV;
} else if (strcmp(value, "JPEG") == 0) { } else if (strcmp(value, "JPEG") == 0) {
preview_fmt = V4L2_PIX_FMT_JPEG; preview_fmt = V4L2_PIX_FMT_JPEG;
} else if (strcmp(value, "NV12") == 0) { } else if (strcmp(value, "NV12") == 0) {
preview_fmt = V4L2_PIX_FMT_NV12; preview_fmt = V4L2_PIX_FMT_NV12;
} else if (strcmp(value, "YUV420") == 0 } else if (strcmp(value, "YUV420") == 0
|| strcmp(value, "I420") == 0 || strcmp(value, "I420") == 0
|| strcmp(value, "YU12") == 0) { || strcmp(value, "YU12") == 0) {
preview_fmt = V4L2_PIX_FMT_YUV420; preview_fmt = V4L2_PIX_FMT_YUV420;
} else if (strcmp(value, "YVU420") == 0 } else if (strcmp(value, "YVU420") == 0
|| strcmp(value, "YV12") == 0) { || strcmp(value, "YV12") == 0) {
preview_fmt = V4L2_PIX_FMT_YVU420; preview_fmt = V4L2_PIX_FMT_YVU420;
} else { } else {
g_printerr("Unsupported pixelformat %s\n", value); g_printerr("Unsupported pixelformat %s\n", value);
@ -454,8 +463,8 @@ main(int argc, char *argv[])
GError *error = NULL; GError *error = NULL;
gtk_init(&argc, &argv); gtk_init(&argc, &argv);
GtkBuilder *builder = gtk_builder_new(); GtkBuilder *builder = gtk_builder_new();
char* glade_file = "/usr/share/camera/ui/camera.glade"; char *glade_file = "/usr/share/camera/ui/camera.glade";
if (access( "camera.glade", F_OK) != -1) { if (access("camera.glade", F_OK) != -1) {
glade_file = "camera.glade"; glade_file = "camera.glade";
} }
if (gtk_builder_add_from_file(builder, glade_file, &error) == 0) { if (gtk_builder_add_from_file(builder, glade_file, &error) == 0) {
@ -470,15 +479,15 @@ main(int argc, char *argv[])
g_signal_connect(window, "destroy", G_CALLBACK(gtk_main_quit), NULL); g_signal_connect(window, "destroy", G_CALLBACK(gtk_main_quit), NULL);
GtkCssProvider *provider = gtk_css_provider_new(); GtkCssProvider *provider = gtk_css_provider_new();
if (access( "camera.css", F_OK) != -1) { if (access("camera.css", F_OK) != -1) {
gtk_css_provider_load_from_path(provider, "camera.css", NULL); gtk_css_provider_load_from_path(provider, "camera.css", NULL);
} else { } else {
gtk_css_provider_load_from_path(provider, "/usr/share/camera/ui/camera.css", NULL); gtk_css_provider_load_from_path(provider, "/usr/share/camera/ui/camera.css", NULL);
} }
GtkStyleContext *context = gtk_widget_get_style_context(preview_box); GtkStyleContext *context = gtk_widget_get_style_context(preview_box);
gtk_style_context_add_provider(context, gtk_style_context_add_provider(context,
GTK_STYLE_PROVIDER(provider), GTK_STYLE_PROVIDER(provider),
GTK_STYLE_PROVIDER_PRIORITY_USER); GTK_STYLE_PROVIDER_PRIORITY_USER);
int result = ini_parse(argv[1], config_ini_handler, NULL); int result = ini_parse(argv[1], config_ini_handler, NULL);
if (result == -1) { if (result == -1) {
@ -502,7 +511,9 @@ main(int argc, char *argv[])
init_device(fd); init_device(fd);
start_capturing(fd); start_capturing(fd);
get_frame(fd);
// Get a new frame every 34ms ~30fps
g_timeout_add(34, get_frame, fd);
gtk_widget_show(window); gtk_widget_show(window);
gtk_main(); gtk_main();

View File

@ -3,4 +3,5 @@ rear=/dev/video0
[preview] [preview]
width=640 width=640
height=480 height=480
fmt=YUYV