This client opens a V4L2 device, usually exposed as /dev/videoN, and retrieves its frames as dmabuf for later import into the compositor. It supports both single- and multi-planar devices, and any format exposed by the V4L2 device the Wayland compositor accepts. This client never changes the v4l2 settings, use `v4l2-ctl -c` if you want to change those. Signed-off-by: Emmanuel Gil Peyrot <emmanuel.peyrot@collabora.com> Acked-by: Derek Foreman <derekf@osg.samsung.com> Tested-by: Derek Foreman <derekf@osg.samsung.com> Maniphest Tasks: T90 Differential Revision: https://phabricator.freedesktop.org/D339dev
parent
8ef2957873
commit
5d43af3e88
@ -0,0 +1,945 @@ |
||||
/*
|
||||
* Copyright © 2015 Collabora Ltd. |
||||
* |
||||
* Permission to use, copy, modify, distribute, and sell this software and its |
||||
* documentation for any purpose is hereby granted without fee, provided that |
||||
* the above copyright notice appear in all copies and that both that copyright |
||||
* notice and this permission notice appear in supporting documentation, and |
||||
* that the name of the copyright holders not be used in advertising or |
||||
* publicity pertaining to distribution of the software without specific, |
||||
* written prior permission. The copyright holders make no representations |
||||
* about the suitability of this software for any purpose. It is provided "as |
||||
* is" without express or implied warranty. |
||||
* |
||||
* THE COPYRIGHT HOLDERS DISCLAIM ALL WARRANTIES WITH REGARD TO THIS SOFTWARE, |
||||
* INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS, IN NO |
||||
* EVENT SHALL THE COPYRIGHT HOLDERS BE LIABLE FOR ANY SPECIAL, INDIRECT OR |
||||
* CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, |
||||
* DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER |
||||
* TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE |
||||
* OF THIS SOFTWARE. |
||||
*/ |
||||
|
||||
#include <config.h> |
||||
|
||||
#include <stdio.h> |
||||
#include <stdlib.h> |
||||
#include <string.h> |
||||
#include <stdbool.h> |
||||
#include <assert.h> |
||||
#include <unistd.h> |
||||
#include <sys/mman.h> |
||||
#include <signal.h> |
||||
#include <fcntl.h> |
||||
|
||||
#include <drm_fourcc.h> |
||||
|
||||
#include <stropts.h> |
||||
#include <sys/mman.h> |
||||
#include <sys/stat.h> |
||||
#include <fcntl.h> |
||||
#include <errno.h> |
||||
#include <linux/videodev2.h> |
||||
#include <linux/input.h> |
||||
|
||||
#include <wayland-client.h> |
||||
#include "xdg-shell-unstable-v5-client-protocol.h" |
||||
#include "fullscreen-shell-unstable-v1-client-protocol.h" |
||||
#include "linux-dmabuf-unstable-v1-client-protocol.h" |
||||
|
||||
#define CLEAR(x) memset(&(x), 0, sizeof(x)) |
||||
|
||||
static int |
||||
xioctl(int fh, int request, void *arg) |
||||
{ |
||||
int r; |
||||
|
||||
do { |
||||
r = ioctl(fh, request, arg); |
||||
} while (r == -1 && errno == EINTR); |
||||
|
||||
return r; |
||||
} |
||||
|
||||
static uint32_t |
||||
parse_format(const char fmt[4]) |
||||
{ |
||||
return fourcc_code(fmt[0], fmt[1], fmt[2], fmt[3]); |
||||
} |
||||
|
||||
static inline const char * |
||||
dump_format(uint32_t format, char out[4]) |
||||
{ |
||||
#if BYTE_ORDER == BIG_ENDIAN |
||||
format = __builtin_bswap32(format); |
||||
#endif |
||||
memcpy(out, &format, 4); |
||||
return out; |
||||
} |
||||
|
||||
struct buffer_format { |
||||
int width; |
||||
int height; |
||||
enum v4l2_buf_type type; |
||||
uint32_t format; |
||||
|
||||
unsigned num_planes; |
||||
unsigned strides[VIDEO_MAX_PLANES]; |
||||
}; |
||||
|
||||
struct display { |
||||
struct wl_display *display; |
||||
struct wl_registry *registry; |
||||
struct wl_compositor *compositor; |
||||
struct wl_seat *seat; |
||||
struct wl_keyboard *keyboard; |
||||
struct xdg_shell *shell; |
||||
struct zwp_fullscreen_shell_v1 *fshell; |
||||
struct zwp_linux_dmabuf_v1 *dmabuf; |
||||
bool requested_format_found; |
||||
|
||||
int v4l_fd; |
||||
struct buffer_format format; |
||||
uint32_t drm_format; |
||||
}; |
||||
|
||||
struct buffer { |
||||
struct wl_buffer *buffer; |
||||
struct display *display; |
||||
int busy; |
||||
int index; |
||||
|
||||
int dmabuf_fds[VIDEO_MAX_PLANES]; |
||||
int data_offsets[VIDEO_MAX_PLANES]; |
||||
}; |
||||
|
||||
#define NUM_BUFFERS 3 |
||||
|
||||
struct window { |
||||
struct display *display; |
||||
struct wl_surface *surface; |
||||
struct xdg_surface *xdg_surface; |
||||
struct buffer buffers[NUM_BUFFERS]; |
||||
struct wl_callback *callback; |
||||
}; |
||||
|
||||
static bool running = true; |
||||
|
||||
static int |
||||
queue(struct display *display, struct buffer *buffer) |
||||
{ |
||||
struct v4l2_buffer buf; |
||||
struct v4l2_plane planes[VIDEO_MAX_PLANES]; |
||||
unsigned i; |
||||
|
||||
CLEAR(buf); |
||||
buf.type = display->format.type; |
||||
buf.memory = V4L2_MEMORY_MMAP; |
||||
buf.index = buffer->index; |
||||
|
||||
if (display->format.type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE) { |
||||
CLEAR(planes); |
||||
buf.length = VIDEO_MAX_PLANES; |
||||
buf.m.planes = planes; |
||||
} |
||||
|
||||
if (xioctl(display->v4l_fd, VIDIOC_QUERYBUF, &buf) == -1) { |
||||
perror("VIDIOC_QUERYBUF"); |
||||
return 0; |
||||
} |
||||
|
||||
if (xioctl(display->v4l_fd, VIDIOC_QBUF, &buf) == -1) { |
||||
perror("VIDIOC_QBUF"); |
||||
return 0; |
||||
} |
||||
|
||||
if (display->format.type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE) { |
||||
if (display->format.num_planes != buf.length) { |
||||
fprintf(stderr, "Wrong number of planes returned by " |
||||
"QUERYBUF\n"); |
||||
return 0; |
||||
} |
||||
|
||||
for (i = 0; i < buf.length; ++i) |
||||
buffer->data_offsets[i] = buf.m.planes[i].data_offset; |
||||
} |
||||
|
||||
return 1; |
||||
} |
||||
|
||||
static void |
||||
buffer_release(void *data, struct wl_buffer *buffer) |
||||
{ |
||||
struct buffer *mybuf = data; |
||||
|
||||
mybuf->busy = 0; |
||||
|
||||
if (!queue(mybuf->display, mybuf)) |
||||
running = false; |
||||
} |
||||
|
||||
static const struct wl_buffer_listener buffer_listener = { |
||||
buffer_release |
||||
}; |
||||
|
||||
static unsigned int |
||||
set_format(struct display *display, uint32_t format) |
||||
{ |
||||
struct v4l2_format fmt; |
||||
char buf[4]; |
||||
|
||||
CLEAR(fmt); |
||||
|
||||
fmt.type = display->format.type; |
||||
|
||||
if (xioctl(display->v4l_fd, VIDIOC_G_FMT, &fmt) == -1) { |
||||
perror("VIDIOC_G_FMT"); |
||||
return 0; |
||||
} |
||||
|
||||
/* No need to set the format if it already is the one we want */ |
||||
if (display->format.type == V4L2_BUF_TYPE_VIDEO_CAPTURE && |
||||
fmt.fmt.pix.pixelformat == format) |
||||
return 1; |
||||
if (display->format.type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE && |
||||
fmt.fmt.pix_mp.pixelformat == format) |
||||
return fmt.fmt.pix_mp.num_planes; |
||||
|
||||
if (display->format.type == V4L2_BUF_TYPE_VIDEO_CAPTURE) |
||||
fmt.fmt.pix.pixelformat = format; |
||||
else |
||||
fmt.fmt.pix_mp.pixelformat = format; |
||||
|
||||
if (xioctl(display->v4l_fd, VIDIOC_S_FMT, &fmt) == -1) { |
||||
perror("VIDIOC_S_FMT"); |
||||
return 0; |
||||
} |
||||
|
||||
if ((display->format.type == V4L2_BUF_TYPE_VIDEO_CAPTURE && |
||||
fmt.fmt.pix.pixelformat != format) || |
||||
(display->format.type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE && |
||||
fmt.fmt.pix_mp.pixelformat != format)) { |
||||
fprintf(stderr, "Failed to set format to %.4s\n", |
||||
dump_format(format, buf)); |
||||
return 0; |
||||
} |
||||
|
||||
if (display->format.type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE) |
||||
return fmt.fmt.pix_mp.num_planes; |
||||
|
||||
return 1; |
||||
} |
||||
|
||||
static int |
||||
v4l_connect(struct display *display, const char *dev_name) |
||||
{ |
||||
struct v4l2_capability cap; |
||||
struct v4l2_requestbuffers req; |
||||
unsigned int num_planes; |
||||
|
||||
display->v4l_fd = open(dev_name, O_RDWR); |
||||
if (display->v4l_fd < 0) { |
||||
perror(dev_name); |
||||
return 0; |
||||
} |
||||
|
||||
if (xioctl(display->v4l_fd, VIDIOC_QUERYCAP, &cap) == -1) { |
||||
if (errno == EINVAL) { |
||||
fprintf(stderr, "%s is no V4L2 device\n", dev_name); |
||||
} else { |
||||
perror("VIDIOC_QUERYCAP"); |
||||
} |
||||
return 0; |
||||
} |
||||
|
||||
if (cap.capabilities & V4L2_CAP_VIDEO_CAPTURE) |
||||
display->format.type = V4L2_BUF_TYPE_VIDEO_CAPTURE; |
||||
else if (cap.capabilities & V4L2_CAP_VIDEO_CAPTURE_MPLANE) |
||||
display->format.type = V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE; |
||||
else { |
||||
fprintf(stderr, "%s is no video capture device\n", dev_name); |
||||
return 0; |
||||
} |
||||
|
||||
if (!(cap.capabilities & V4L2_CAP_STREAMING)) { |
||||
fprintf(stderr, "%s does not support dmabuf i/o\n", dev_name); |
||||
return 0; |
||||
} |
||||
|
||||
/* Select video input, video standard and tune here */ |
||||
|
||||
num_planes = set_format(display, display->format.format); |
||||
if (num_planes < 1) |
||||
return 0; |
||||
|
||||
CLEAR(req); |
||||
|
||||
req.type = display->format.type; |
||||
req.memory = V4L2_MEMORY_MMAP; |
||||
req.count = NUM_BUFFERS * num_planes; |
||||
|
||||
if (xioctl(display->v4l_fd, VIDIOC_REQBUFS, &req) == -1) { |
||||
if (errno == EINVAL) { |
||||
fprintf(stderr, "%s does not support dmabuf\n", |
||||
dev_name); |
||||
} else { |
||||
perror("VIDIOC_REQBUFS"); |
||||
} |
||||
return 0; |
||||
} |
||||
|
||||
if (req.count < NUM_BUFFERS * num_planes) { |
||||
fprintf(stderr, "Insufficient buffer memory on %s\n", dev_name); |
||||
return 0; |
||||
} |
||||
|
||||
printf("Created %d buffers\n", req.count); |
||||
|
||||
return 1; |
||||
} |
||||
|
||||
static void |
||||
v4l_shutdown(struct display *display) |
||||
{ |
||||
close(display->v4l_fd); |
||||
} |
||||
|
||||
static void |
||||
create_succeeded(void *data, |
||||
struct zwp_linux_buffer_params_v1 *params, |
||||
struct wl_buffer *new_buffer) |
||||
{ |
||||
struct buffer *buffer = data; |
||||
unsigned i; |
||||
|
||||
buffer->buffer = new_buffer; |
||||
wl_buffer_add_listener(buffer->buffer, &buffer_listener, buffer); |
||||
|
||||
zwp_linux_buffer_params_v1_destroy(params); |
||||
|
||||
for (i = 0; i < buffer->display->format.num_planes; ++i) |
||||
close(buffer->dmabuf_fds[i]); |
||||
} |
||||
|
||||
static void |
||||
create_failed(void *data, struct zwp_linux_buffer_params_v1 *params) |
||||
{ |
||||
struct buffer *buffer = data; |
||||
unsigned i; |
||||
|
||||
buffer->buffer = NULL; |
||||
|
||||
zwp_linux_buffer_params_v1_destroy(params); |
||||
|
||||
for (i = 0; i < buffer->display->format.num_planes; ++i) |
||||
close(buffer->dmabuf_fds[i]); |
||||
|
||||
running = false; |
||||
|
||||
fprintf(stderr, "Error: zwp_linux_buffer_params.create failed.\n"); |
||||
} |
||||
|
||||
static const struct zwp_linux_buffer_params_v1_listener params_listener = { |
||||
create_succeeded, |
||||
create_failed |
||||
}; |
||||
|
||||
static void |
||||
create_dmabuf_buffer(struct display *display, struct buffer *buffer) |
||||
{ |
||||
struct zwp_linux_buffer_params_v1 *params; |
||||
uint64_t modifier; |
||||
uint32_t flags; |
||||
unsigned i; |
||||
|
||||
modifier = 0; |
||||
flags = ZWP_LINUX_BUFFER_PARAMS_V1_FLAGS_Y_INVERT; |
||||
|
||||
params = zwp_linux_dmabuf_v1_create_params(display->dmabuf); |
||||
for (i = 0; i < display->format.num_planes; ++i) |
||||
zwp_linux_buffer_params_v1_add(params, |
||||
buffer->dmabuf_fds[i], |
||||
i, /* plane_idx */ |
||||
buffer->data_offsets[i], /* offset */ |
||||
display->format.strides[i], |
||||
modifier >> 32, |
||||
modifier & 0xffffffff); |
||||
zwp_linux_buffer_params_v1_add_listener(params, ¶ms_listener, |
||||
buffer); |
||||
zwp_linux_buffer_params_v1_create(params, |
||||
display->format.width, |
||||
display->format.height, |
||||
display->drm_format, |
||||
flags); |
||||
} |
||||
|
||||
static int |
||||
buffer_export(struct display *display, int index, int dmafd[]) |
||||
{ |
||||
struct v4l2_exportbuffer expbuf; |
||||
unsigned i; |
||||
|
||||
CLEAR(expbuf); |
||||
|
||||
for (i = 0; i < display->format.num_planes; ++i) { |
||||
expbuf.type = display->format.type; |
||||
expbuf.index = index; |
||||
expbuf.plane = i; |
||||
if (xioctl(display->v4l_fd, VIDIOC_EXPBUF, &expbuf) == -1) { |
||||
perror("VIDIOC_EXPBUF"); |
||||
while (i) |
||||
close(dmafd[--i]); |
||||
return 0; |
||||
} |
||||
dmafd[i] = expbuf.fd; |
||||
} |
||||
|
||||
return 1; |
||||
} |
||||
|
||||
static int |
||||
queue_initial_buffers(struct display *display, |
||||
struct buffer buffers[NUM_BUFFERS]) |
||||
{ |
||||
struct buffer *buffer; |
||||
int index; |
||||
|
||||
for (index = 0; index < NUM_BUFFERS; ++index) { |
||||
buffer = &buffers[index]; |
||||
buffer->display = display; |
||||
buffer->index = index; |
||||
|
||||
if (!queue(display, buffer)) { |
||||
fprintf(stderr, "Failed to queue buffer\n"); |
||||
return 0; |
||||
} |
||||
|
||||
assert(!buffer->buffer); |
||||
if (!buffer_export(display, index, buffer->dmabuf_fds)) |
||||
return 0; |
||||
|
||||
create_dmabuf_buffer(display, buffer); |
||||
} |
||||
|
||||
return 1; |
||||
} |
||||
|
||||
static int |
||||
dequeue(struct display *display) |
||||
{ |
||||
struct v4l2_buffer buf; |
||||
struct v4l2_plane planes[VIDEO_MAX_PLANES]; |
||||
|
||||
CLEAR(buf); |
||||
buf.type = display->format.type; |
||||
buf.memory = V4L2_MEMORY_MMAP; |
||||
buf.length = VIDEO_MAX_PLANES; |
||||
buf.m.planes = planes; |
||||
|
||||
/* This ioctl is blocking until a buffer is ready to be displayed */ |
||||
if (xioctl(display->v4l_fd, VIDIOC_DQBUF, &buf) == -1) { |
||||
perror("VIDIOC_DQBUF"); |
||||
return -1; |
||||
} |
||||
|
||||
assert(buf.flags & V4L2_BUF_FLAG_DONE); |
||||
|
||||
return buf.index; |
||||
} |
||||
|
||||
static int |
||||
fill_buffer_format(struct display *display) |
||||
{ |
||||
struct v4l2_format fmt; |
||||
struct v4l2_pix_format *pix; |
||||
struct v4l2_pix_format_mplane *pix_mp; |
||||
int i; |
||||
char buf[4]; |
||||
|
||||
CLEAR(fmt); |
||||
fmt.type = display->format.type; |
||||
|
||||
/* Preserve original settings as set by v4l2-ctl for example */ |
||||
if (xioctl(display->v4l_fd, VIDIOC_G_FMT, &fmt) == -1) { |
||||
perror("VIDIOC_G_FMT"); |
||||
return 0; |
||||
} |
||||
|
||||
if (display->format.type == V4L2_BUF_TYPE_VIDEO_CAPTURE) { |
||||
pix = &fmt.fmt.pix; |
||||
|
||||
printf("%d×%d, %.4s\n", pix->width, pix->height, |
||||
dump_format(pix->pixelformat, buf)); |
||||
|
||||
display->format.num_planes = 1; |
||||
display->format.width = pix->width; |
||||
display->format.height = pix->height; |
||||
display->format.strides[0] = pix->bytesperline; |
||||
} else { |
||||
pix_mp = &fmt.fmt.pix_mp; |
||||
|
||||
display->format.num_planes = pix_mp->num_planes; |
||||
display->format.width = pix_mp->width; |
||||
display->format.height = pix_mp->height; |
||||
|
||||
for (i = 0; i < pix_mp->num_planes; ++i) |
||||
display->format.strides[i] = pix_mp->plane_fmt[i].bytesperline; |
||||
|
||||
printf("%d×%d, %.4s, %d planes\n", |
||||
pix_mp->width, pix_mp->height, |
||||
dump_format(pix_mp->pixelformat, buf), |
||||
pix_mp->num_planes); |
||||
} |
||||
|
||||
return 1; |
||||
} |
||||
|
||||
static int |
||||
v4l_init(struct display *display, struct buffer buffers[NUM_BUFFERS]) { |
||||
if (!fill_buffer_format(display)) { |
||||
fprintf(stderr, "Failed to fill buffer format\n"); |
||||
return 0; |
||||
} |
||||
|
||||
if (!queue_initial_buffers(display, buffers)) { |
||||
fprintf(stderr, "Failed to queue initial buffers\n"); |
||||
return 0; |
||||
} |
||||
|
||||
return 1; |
||||
} |
||||
|
||||
static int |
||||
start_capture(struct display *display) |
||||
{ |
||||
int type = display->format.type; |
||||
|
||||
if (xioctl(display->v4l_fd, VIDIOC_STREAMON, &type) == -1) { |
||||
perror("VIDIOC_STREAMON"); |
||||
return 0; |
||||
} |
||||
|
||||
return 1; |
||||
} |
||||
|
||||
static void |
||||
handle_configure(void *data, struct xdg_surface *surface, |
||||
int32_t width, int32_t height, |
||||
struct wl_array *states, uint32_t serial) |
||||
{ |
||||
} |
||||
|
||||
static void |
||||
handle_delete(void *data, struct xdg_surface *xdg_surface) |
||||
{ |
||||
running = false; |
||||
} |
||||
|
||||
static const struct xdg_surface_listener xdg_surface_listener = { |
||||
handle_configure, |
||||
handle_delete, |
||||
}; |
||||
|
||||
static struct window * |
||||
create_window(struct display *display) |
||||
{ |
||||
struct window *window; |
||||
|
||||
window = calloc(1, sizeof *window); |
||||
if (!window) |
||||
return NULL; |
||||
|
||||
window->callback = NULL; |
||||
window->display = display; |
||||
window->surface = wl_compositor_create_surface(display->compositor); |
||||
|
||||
if (display->shell) { |
||||
window->xdg_surface = |
||||
xdg_shell_get_xdg_surface(display->shell, |
||||
window->surface); |
||||
|
||||
assert(window->xdg_surface); |
||||
|
||||
xdg_surface_add_listener(window->xdg_surface, |
||||
&xdg_surface_listener, window); |
||||
|
||||
xdg_surface_set_title(window->xdg_surface, "simple-dmabuf-v4l"); |
||||
} else if (display->fshell) { |
||||
zwp_fullscreen_shell_v1_present_surface(display->fshell, |
||||
window->surface, |
||||
ZWP_FULLSCREEN_SHELL_V1_PRESENT_METHOD_DEFAULT, |
||||
NULL); |
||||
} else { |
||||
assert(0); |
||||
} |
||||
|
||||
return window; |
||||
} |
||||
|
||||
static void |
||||
destroy_window(struct window *window) |
||||
{ |
||||
int i; |
||||
unsigned j; |
||||
|
||||
if (window->callback) |
||||
wl_callback_destroy(window->callback); |
||||
|
||||
if (window->xdg_surface) |
||||
xdg_surface_destroy(window->xdg_surface); |
||||
wl_surface_destroy(window->surface); |
||||
|
||||
for (i = 0; i < NUM_BUFFERS; i++) { |
||||
if (!window->buffers[i].buffer) |
||||
continue; |
||||
|
||||
wl_buffer_destroy(window->buffers[i].buffer); |
||||
for (j = 0; j < window->display->format.num_planes; ++j) |
||||
close(window->buffers[i].dmabuf_fds[j]); |
||||
} |
||||
|
||||
v4l_shutdown(window->display); |
||||
|
||||
free(window); |
||||
} |
||||
|
||||
static const struct wl_callback_listener frame_listener; |
||||
|
||||
static void |
||||
redraw(void *data, struct wl_callback *callback, uint32_t time) |
||||
{ |
||||
struct window *window = data; |
||||
struct buffer *buffer; |
||||
int index, num_busy = 0; |
||||
|
||||
/* Check for a deadlock situation where we would block forever trying
|
||||
* to dequeue a buffer while all of them are locked by the compositor. |
||||
*/ |
||||
for (index = 0; index < NUM_BUFFERS; ++index) |
||||
if (window->buffers[index].busy) |
||||
++num_busy; |
||||
|
||||
/* A robust application would just postpone redraw until it has queued
|
||||
* a buffer. |
||||
*/ |
||||
assert(num_busy < NUM_BUFFERS); |
||||
|
||||
index = dequeue(window->display); |
||||
if (index < 0) { |
||||
/* We couldn’t get any buffer out of the camera, exiting. */ |
||||
running = false; |
||||
return; |
||||
} |
||||
|
||||
buffer = &window->buffers[index]; |
||||
assert(!buffer->busy); |
||||
|
||||
wl_surface_attach(window->surface, buffer->buffer, 0, 0); |
||||
wl_surface_damage(window->surface, 0, 0, |
||||
window->display->format.width, |
||||
window->display->format.height); |
||||
|
||||
if (callback) |
||||
wl_callback_destroy(callback); |
||||
|
||||
window->callback = wl_surface_frame(window->surface); |
||||
wl_callback_add_listener(window->callback, &frame_listener, window); |
||||
wl_surface_commit(window->surface); |
||||
buffer->busy = 1; |
||||
} |
||||
|
||||
static const struct wl_callback_listener frame_listener = { |
||||
redraw |
||||
}; |
||||
|
||||
static void |
||||
dmabuf_format(void *data, struct zwp_linux_dmabuf_v1 *zwp_linux_dmabuf, |
||||
uint32_t format) |
||||
{ |
||||
struct display *d = data; |
||||
|
||||
if (format == d->drm_format) |
||||
d->requested_format_found = true; |
||||
} |
||||
|
||||
static const struct zwp_linux_dmabuf_v1_listener dmabuf_listener = { |
||||
dmabuf_format |
||||
}; |
||||
|
||||
static void |
||||
keyboard_handle_keymap(void *data, struct wl_keyboard *keyboard, |
||||
uint32_t format, int fd, uint32_t size) |
||||
{ |
||||
/* Just so we don’t leak the keymap fd */ |
||||
close(fd); |
||||
} |
||||
|
||||
static void |
||||
keyboard_handle_enter(void *data, struct wl_keyboard *keyboard, |
||||
uint32_t serial, struct wl_surface *surface, |
||||
struct wl_array *keys) |
||||
{ |
||||
} |
||||
|
||||
static void |
||||
keyboard_handle_leave(void *data, struct wl_keyboard *keyboard, |
||||
uint32_t serial, struct wl_surface *surface) |
||||
{ |
||||
} |
||||
|
||||
static void |
||||
keyboard_handle_key(void *data, struct wl_keyboard *keyboard, |
||||
uint32_t serial, uint32_t time, uint32_t key, |
||||
uint32_t state) |
||||
{ |
||||
struct display *d = data; |
||||
|
||||
if (!d->shell) |
||||
return; |
||||
|
||||
if (key == KEY_ESC && state) |
||||
running = false; |
||||
} |
||||
|
||||
static void |
||||
keyboard_handle_modifiers(void *data, struct wl_keyboard *keyboard, |
||||
uint32_t serial, uint32_t mods_depressed, |
||||
uint32_t mods_latched, uint32_t mods_locked, |
||||
uint32_t group) |
||||
{ |
||||
} |
||||
|
||||
static const struct wl_keyboard_listener keyboard_listener = { |
||||
keyboard_handle_keymap, |
||||
keyboard_handle_enter, |
||||
keyboard_handle_leave, |
||||
keyboard_handle_key, |
||||
keyboard_handle_modifiers, |
||||
}; |
||||
|
||||
static void |
||||
seat_handle_capabilities(void *data, struct wl_seat *seat, |
||||
enum wl_seat_capability caps) |
||||
{ |
||||
struct display *d = data; |
||||
|
||||
if ((caps & WL_SEAT_CAPABILITY_KEYBOARD) && !d->keyboard) { |
||||
d->keyboard = wl_seat_get_keyboard(seat); |
||||
wl_keyboard_add_listener(d->keyboard, &keyboard_listener, d); |
||||
} else if (!(caps & WL_SEAT_CAPABILITY_KEYBOARD) && d->keyboard) { |
||||
wl_keyboard_destroy(d->keyboard); |
||||
d->keyboard = NULL; |
||||
} |
||||
} |
||||
|
||||
static const struct wl_seat_listener seat_listener = { |
||||
seat_handle_capabilities, |
||||
}; |
||||
|
||||
static void |
||||
xdg_shell_ping(void *data, struct xdg_shell *shell, uint32_t serial) |
||||
{ |
||||
xdg_shell_pong(shell, serial); |
||||
} |
||||
|
||||
static const struct xdg_shell_listener xdg_shell_listener = { |
||||
xdg_shell_ping, |
||||
}; |
||||
|
||||
#define XDG_VERSION 5 /* The version of xdg-shell that we implement */ |
||||
#ifdef static_assert |
||||
static_assert(XDG_VERSION == XDG_SHELL_VERSION_CURRENT, |
||||
"Interface version doesn't match implementation version"); |
||||
#endif |
||||
|
||||
static void |
||||
registry_handle_global(void *data, struct wl_registry *registry, |
||||
uint32_t id, const char *interface, uint32_t version) |
||||
{ |
||||
struct display *d = data; |
||||
|
||||
if (strcmp(interface, "wl_compositor") == 0) { |
||||
d->compositor = |
||||
wl_registry_bind(registry, |
||||
id, &wl_compositor_interface, 1); |
||||
} else if (strcmp(interface, "wl_seat") == 0) { |
||||
d->seat = wl_registry_bind(registry, |
||||
id, &wl_seat_interface, 1); |
||||
wl_seat_add_listener(d->seat, &seat_listener, d); |
||||
} else if (strcmp(interface, "xdg_shell") == 0) { |
||||
d->shell = wl_registry_bind(registry, |
||||
id, &xdg_shell_interface, 1); |
||||
xdg_shell_use_unstable_version(d->shell, XDG_VERSION); |
||||
xdg_shell_add_listener(d->shell, &xdg_shell_listener, d); |
||||
} else if (strcmp(interface, "zwp_fullscreen_shell_v1") == 0) { |
||||
d->fshell = wl_registry_bind(registry, |
||||
id, &zwp_fullscreen_shell_v1_interface, |
||||
1); |
||||
} else if (strcmp(interface, "zwp_linux_dmabuf_v1") == 0) { |
||||
d->dmabuf = wl_registry_bind(registry, |
||||
id, &zwp_linux_dmabuf_v1_interface, |
||||
1); |
||||
zwp_linux_dmabuf_v1_add_listener(d->dmabuf, &dmabuf_listener, |
||||
d); |
||||
} |
||||
} |
||||
|
||||
static void |
||||
registry_handle_global_remove(void *data, struct wl_registry *registry, |
||||
uint32_t name) |
||||
{ |
||||
} |
||||
|
||||
static const struct wl_registry_listener registry_listener = { |
||||
registry_handle_global, |
||||
registry_handle_global_remove |
||||
}; |
||||
|
||||
static struct display * |
||||
create_display(uint32_t requested_format) |
||||
{ |
||||
struct display *display; |
||||
|
||||
display = malloc(sizeof *display); |
||||
if (display == NULL) { |
||||
fprintf(stderr, "out of memory\n"); |
||||
exit(1); |
||||
} |
||||
display->display = wl_display_connect(NULL); |
||||
assert(display->display); |
||||
|
||||
display->drm_format = requested_format; |
||||
|
||||
display->registry = wl_display_get_registry(display->display); |
||||
wl_registry_add_listener(display->registry, |
||||
®istry_listener, display); |
||||
wl_display_roundtrip(display->display); |
||||
if (display->dmabuf == NULL) { |
||||
fprintf(stderr, "No zwp_linux_dmabuf global\n"); |
||||
exit(1); |
||||
} |
||||
|
||||
wl_display_roundtrip(display->display); |
||||
|
||||
/* XXX: fake, because the compositor does not yet advertise anything */ |
||||
display->requested_format_found = true; |
||||
|
||||
if (!display->requested_format_found) { |
||||
fprintf(stderr, "DRM_FORMAT_YUYV not available\n"); |
||||
exit(1); |
||||
} |
||||
|
||||
return display; |
||||
} |
||||
|
||||
static void |
||||
destroy_display(struct display *display) |
||||
{ |
||||
if (display->dmabuf) |
||||
zwp_linux_dmabuf_v1_destroy(display->dmabuf); |
||||
|
||||
if (display->shell) |
||||
xdg_shell_destroy(display->shell); |
||||
|
||||
if (display->fshell) |
||||
zwp_fullscreen_shell_v1_release(display->fshell); |
||||
|
||||
if (display->compositor) |
||||
wl_compositor_destroy(display->compositor); |
||||
|
||||
wl_registry_destroy(display->registry); |
||||
wl_display_flush(display->display); |
||||
wl_display_disconnect(display->display); |
||||
free(display); |
||||
} |
||||
|
||||
static void |
||||
usage(const char *argv0) |
||||
{ |
||||
printf("Usage: %s [V4L2 device] [V4L2 format] [DRM format]\n" |
||||
"\n" |
||||
"The default V4L2 device is /dev/video0\n" |
||||
"\n" |
||||
"Both formats are FOURCC values (see http://fourcc.org/)\n" |
||||
"V4L2 formats are defined in <linux/videodev2.h>\n" |
||||
"DRM formats are defined in <libdrm/drm_fourcc.h>\n" |
||||
"The default for both formats is YUYV.\n" |
||||
"If the V4L2 and DRM formats differ, the data is simply " |
||||
"reinterpreted rather than converted.\n", argv0); |
||||
exit(0); |
||||
} |
||||
|
||||
static void |
||||
signal_int(int signum) |
||||
{ |
||||
running = false; |
||||
} |
||||
|
||||
int |
||||
main(int argc, char **argv) |
||||
{ |
||||
struct sigaction sigint; |
||||
struct display *display; |
||||
struct window *window; |
||||
const char *v4l_device; |
||||
uint32_t v4l_format, drm_format; |
||||
int ret = 0; |
||||
|
||||
if (argc < 2) { |
||||
v4l_device = "/dev/video0"; |
||||
} else if (!strcmp(argv[1], "--help")) { |
||||
usage(argv[0]); |
||||
} else { |
||||
v4l_device = argv[1]; |
||||
} |
||||
|
||||
if (argc < 3) |
||||
v4l_format = parse_format("YUYV"); |
||||
else |
||||
v4l_format = parse_format(argv[2]); |
||||
|
||||
if (argc < 4) |
||||
drm_format = v4l_format; |
||||
else |
||||
drm_format = parse_format(argv[3]); |
||||
|
||||
display = create_display(drm_format); |
||||
display->format.format = v4l_format; |
||||
|
||||
window = create_window(display); |
||||
if (!window) |
||||
return 1; |
||||
|
||||
if (!v4l_connect(display, v4l_device)) |
||||
return 1; |
||||
|
||||
if (!v4l_init(display, window->buffers)) |
||||
return 1; |
||||
|
||||
sigint.sa_handler = signal_int; |
||||
sigemptyset(&sigint.sa_mask); |
||||
sigint.sa_flags = SA_RESETHAND; |
||||
sigaction(SIGINT, &sigint, NULL); |
||||
|
||||
/* Here we retrieve the linux-dmabuf objects, or error */ |
||||
wl_display_roundtrip(display->display); |
||||
|
||||
/* In case of error, running will be 0 */ |
||||
if (!running) |
||||
return 1; |
||||
|
||||
/* We got all of our buffers, we can start the capture! */ |
||||
if (!start_capture(display)) |
||||
return 1; |
||||
|
||||
redraw(window, NULL, 0); |
||||
|
||||
while (running && ret != -1) |
||||
ret = wl_display_dispatch(display->display); |
||||
|
||||
fprintf(stderr, "simple-dmabuf-v4l exiting\n"); |
||||
destroy_window(window); |
||||
destroy_display(display); |
||||
|
||||
return 0; |
||||
} |
Loading…
Reference in new issue