diff options
Diffstat (limited to 'examples')
| -rw-r--r-- | examples/dmabuf-capture.c | 909 | ||||
| -rw-r--r-- | examples/relative-pointer-unstable-v1.c | 489 |
2 files changed, 0 insertions, 1398 deletions
diff --git a/examples/dmabuf-capture.c b/examples/dmabuf-capture.c deleted file mode 100644 index 5a7debd2..00000000 --- a/examples/dmabuf-capture.c +++ /dev/null @@ -1,909 +0,0 @@ -#define _POSIX_C_SOURCE 199309L -#include <libavcodec/avcodec.h> -#include <libavformat/avformat.h> -#include <libavutil/display.h> -#include <libavutil/hwcontext_drm.h> -#include <libavutil/pixdesc.h> -#include <poll.h> -#include <signal.h> -#include <stdio.h> -#include <stdlib.h> -#include <string.h> -#include <unistd.h> -#include <pthread.h> -#include <stdbool.h> -#include <drm_fourcc.h> -#include "wlr-export-dmabuf-unstable-v1-client-protocol.h" - -struct wayland_output { - struct wl_list link; - uint32_t id; - struct wl_output *output; - char *make; - char *model; - int width; - int height; - AVRational framerate; -}; - -struct fifo_buffer { - AVFrame **queued_frames; - int num_queued_frames; - int max_queued_frames; - pthread_mutex_t lock; - pthread_cond_t cond; - pthread_mutex_t cond_lock; -}; - -struct capture_context { - AVClass *class; /* For pretty logging */ - struct wl_display *display; - struct wl_registry *registry; - struct zwlr_export_dmabuf_manager_v1 *export_manager; - - struct wl_list output_list; - - /* Target */ - struct wl_output *target_output; - bool with_cursor; - - /* Main frame callback */ - struct zwlr_export_dmabuf_frame_v1 *frame_callback; - - /* If something happens during capture */ - int err; - bool quit; - - /* FFmpeg specific parts */ - pthread_t vid_thread; - AVFrame *current_frame; - AVFormatContext *avf; - AVCodecContext *avctx; - AVBufferRef *drm_device_ref; - AVBufferRef *drm_frames_ref; - AVBufferRef *mapped_device_ref; - AVBufferRef *mapped_frames_ref; - - /* Sync stuff */ - struct fifo_buffer vid_frames; - - int64_t start_pts; - - /* Config */ - enum AVPixelFormat software_format; - enum AVHWDeviceType hw_device_type; - AVDictionary *encoder_opts; - int is_software_encoder; - char *hardware_device; - char *out_filename; - char *encoder_name; - float out_bitrate; -}; - -static int init_fifo(struct fifo_buffer *buf, int max_queued_frames) { - pthread_mutex_init(&buf->lock, NULL); - pthread_cond_init(&buf->cond, NULL); - pthread_mutex_init(&buf->cond_lock, NULL); - buf->num_queued_frames = 0; - buf->max_queued_frames = max_queued_frames; - buf->queued_frames = av_mallocz(buf->max_queued_frames * sizeof(AVFrame)); - return !buf->queued_frames ? AVERROR(ENOMEM) : 0; -} - -static int get_fifo_size(struct fifo_buffer *buf) { - pthread_mutex_lock(&buf->lock); - int ret = buf->num_queued_frames; - pthread_mutex_unlock(&buf->lock); - return ret; -} - -static int push_to_fifo(struct fifo_buffer *buf, AVFrame *f) { - int ret; - pthread_mutex_lock(&buf->lock); - if ((buf->num_queued_frames + 1) > buf->max_queued_frames) { - av_frame_free(&f); - ret = 1; - } else { - buf->queued_frames[buf->num_queued_frames++] = f; - ret = 0; - } - pthread_mutex_unlock(&buf->lock); - pthread_cond_signal(&buf->cond); - return ret; -} - -static AVFrame *pop_from_fifo(struct fifo_buffer *buf) { - pthread_mutex_lock(&buf->lock); - - if (!buf->num_queued_frames) { - pthread_mutex_unlock(&buf->lock); - pthread_cond_wait(&buf->cond, &buf->cond_lock); - pthread_mutex_lock(&buf->lock); - } - - AVFrame *rf = buf->queued_frames[0]; - for (int i = 1; i < buf->num_queued_frames; i++) { - buf->queued_frames[i - 1] = buf->queued_frames[i]; - } - buf->num_queued_frames--; - buf->queued_frames[buf->num_queued_frames] = NULL; - - pthread_mutex_unlock(&buf->lock); - return rf; -} - -static void free_fifo(struct fifo_buffer *buf) { - pthread_mutex_lock(&buf->lock); - if (buf->num_queued_frames) { - for (int i = 0; i < buf->num_queued_frames; i++) { - av_frame_free(&buf->queued_frames[i]); - } - } - av_freep(&buf->queued_frames); - pthread_mutex_unlock(&buf->lock); -} - -static void output_handle_geometry(void *data, struct wl_output *wl_output, - int32_t x, int32_t y, int32_t phys_width, int32_t phys_height, - int32_t subpixel, const char *make, const char *model, - int32_t transform) { - struct wayland_output *output = data; - output->make = av_strdup(make); - output->model = av_strdup(model); -} - -static void output_handle_mode(void *data, struct wl_output *wl_output, - uint32_t flags, int32_t width, int32_t height, int32_t refresh) { - if (flags & WL_OUTPUT_MODE_CURRENT) { - struct wayland_output *output = data; - output->width = width; - output->height = height; - output->framerate = (AVRational){ refresh, 1000 }; - } -} - -static void output_handle_done(void* data, struct wl_output *wl_output) { - /* Nothing to do */ -} - -static void output_handle_scale(void* data, struct wl_output *wl_output, - int32_t factor) { - /* Nothing to do */ -} - -static const struct wl_output_listener output_listener = { - .geometry = output_handle_geometry, - .mode = output_handle_mode, - .done = output_handle_done, - .scale = output_handle_scale, -}; - -static void registry_handle_add(void *data, struct wl_registry *reg, - uint32_t id, const char *interface, uint32_t ver) { - struct capture_context *ctx = data; - - if (!strcmp(interface, wl_output_interface.name)) { - struct wayland_output *output = av_mallocz(sizeof(*output)); - - output->id = id; - output->output = wl_registry_bind(reg, id, &wl_output_interface, 1); - - wl_output_add_listener(output->output, &output_listener, output); - wl_list_insert(&ctx->output_list, &output->link); - } - - if (!strcmp(interface, zwlr_export_dmabuf_manager_v1_interface.name)) { - ctx->export_manager = wl_registry_bind(reg, id, - &zwlr_export_dmabuf_manager_v1_interface, 1); - } -} - -static void remove_output(struct wayland_output *out) { - wl_list_remove(&out->link); - av_free(out->make); - av_free(out->model); - av_free(out); -} - -static struct wayland_output *find_output(struct capture_context *ctx, - struct wl_output *out, int id) { - struct wayland_output *output, *tmp; - wl_list_for_each_safe(output, tmp, &ctx->output_list, link) { - if (output->output == out || (id >= 0 && output->id == (uint32_t)id) - || id == -1) { - return output; - } - } - return NULL; -} - -static void registry_handle_remove(void *data, struct wl_registry *reg, - uint32_t id) { - remove_output(find_output((struct capture_context *)data, NULL, id)); -} - -static const struct wl_registry_listener registry_listener = { - .global = registry_handle_add, - .global_remove = registry_handle_remove, -}; - -static void frame_free(void *opaque, uint8_t *data) { - AVDRMFrameDescriptor *desc = (AVDRMFrameDescriptor *)data; - - if (desc) { - for (int i = 0; i < desc->nb_objects; ++i) { - close(desc->objects[i].fd); - } - av_free(data); - } - - zwlr_export_dmabuf_frame_v1_destroy(opaque); -} - -static void frame_start(void *data, struct zwlr_export_dmabuf_frame_v1 *frame, - uint32_t width, uint32_t height, uint32_t offset_x, uint32_t offset_y, - uint32_t buffer_flags, uint32_t flags, uint32_t format, - uint32_t mod_high, uint32_t mod_low, uint32_t num_objects) { - struct capture_context *ctx = data; - int err = 0; - - /* Allocate DRM specific struct */ - AVDRMFrameDescriptor *desc = av_mallocz(sizeof(*desc)); - if (!desc) { - err = AVERROR(ENOMEM); - goto fail; - } - - desc->nb_objects = num_objects; - desc->objects[0].format_modifier = ((uint64_t)mod_high << 32) | mod_low; - - desc->nb_layers = 1; - desc->layers[0].format = format; - - /* Allocate a frame */ - AVFrame *f = av_frame_alloc(); - if (!f) { - err = AVERROR(ENOMEM); - goto fail; - } - - /* Set base frame properties */ - ctx->current_frame = f; - f->width = width; - f->height = height; - f->format = AV_PIX_FMT_DRM_PRIME; - - /* Set the frame data to the DRM specific struct */ - f->buf[0] = av_buffer_create((uint8_t*)desc, sizeof(*desc), - &frame_free, frame, 0); - if (!f->buf[0]) { - err = AVERROR(ENOMEM); - goto fail; - } - - f->data[0] = (uint8_t*)desc; - - return; - -fail: - ctx->err = err; - frame_free(frame, (uint8_t *)desc); -} - -static void frame_object(void *data, struct zwlr_export_dmabuf_frame_v1 *frame, - uint32_t index, int32_t fd, uint32_t size, uint32_t offset, - uint32_t stride, uint32_t plane_index) { - struct capture_context *ctx = data; - AVFrame *f = ctx->current_frame; - AVDRMFrameDescriptor *desc = (AVDRMFrameDescriptor *)f->data[0]; - - desc->objects[index].fd = fd; - desc->objects[index].size = size; - - desc->layers[0].planes[plane_index].object_index = index; - desc->layers[0].planes[plane_index].offset = offset; - desc->layers[0].planes[plane_index].pitch = stride; -} - -static enum AVPixelFormat drm_fmt_to_pixfmt(uint32_t fmt) { - switch (fmt) { - case DRM_FORMAT_NV12: return AV_PIX_FMT_NV12; - case DRM_FORMAT_ARGB8888: return AV_PIX_FMT_BGRA; - case DRM_FORMAT_XRGB8888: return AV_PIX_FMT_BGR0; - case DRM_FORMAT_ABGR8888: return AV_PIX_FMT_RGBA; - case DRM_FORMAT_XBGR8888: return AV_PIX_FMT_RGB0; - case DRM_FORMAT_RGBA8888: return AV_PIX_FMT_ABGR; - case DRM_FORMAT_RGBX8888: return AV_PIX_FMT_0BGR; - case DRM_FORMAT_BGRA8888: return AV_PIX_FMT_ARGB; - case DRM_FORMAT_BGRX8888: return AV_PIX_FMT_0RGB; - default: return AV_PIX_FMT_NONE; - }; -} - -static int attach_drm_frames_ref(struct capture_context *ctx, AVFrame *f, - enum AVPixelFormat sw_format) { - int err = 0; - AVHWFramesContext *hwfc; - - if (ctx->drm_frames_ref) { - hwfc = (AVHWFramesContext*)ctx->drm_frames_ref->data; - if (hwfc->width == f->width && hwfc->height == f->height && - hwfc->sw_format == sw_format) { - goto attach; - } - av_buffer_unref(&ctx->drm_frames_ref); - } - - ctx->drm_frames_ref = av_hwframe_ctx_alloc(ctx->drm_device_ref); - if (!ctx->drm_frames_ref) { - err = AVERROR(ENOMEM); - goto fail; - } - - hwfc = (AVHWFramesContext*)ctx->drm_frames_ref->data; - - hwfc->format = f->format; - hwfc->sw_format = sw_format; - hwfc->width = f->width; - hwfc->height = f->height; - - err = av_hwframe_ctx_init(ctx->drm_frames_ref); - if (err) { - av_log(ctx, AV_LOG_ERROR, "AVHWFramesContext init failed: %s!\n", - av_err2str(err)); - goto fail; - } - -attach: - /* Set frame hardware context referencce */ - f->hw_frames_ctx = av_buffer_ref(ctx->drm_frames_ref); - if (!f->hw_frames_ctx) { - err = AVERROR(ENOMEM); - goto fail; - } - - return 0; - -fail: - av_buffer_unref(&ctx->drm_frames_ref); - return err; -} - -static void register_cb(struct capture_context *ctx); - -static void frame_ready(void *data, struct zwlr_export_dmabuf_frame_v1 *frame, - uint32_t tv_sec_hi, uint32_t tv_sec_lo, uint32_t tv_nsec) { - struct capture_context *ctx = data; - AVFrame *f = ctx->current_frame; - AVDRMFrameDescriptor *desc = (AVDRMFrameDescriptor *)f->data[0]; - enum AVPixelFormat pix_fmt = drm_fmt_to_pixfmt(desc->layers[0].format); - int err = 0; - - /* Timestamp, nanoseconds timebase */ - f->pts = ((((uint64_t)tv_sec_hi) << 32) | tv_sec_lo) * 1000000000 + tv_nsec; - - if (!ctx->start_pts) { - ctx->start_pts = f->pts; - } - - f->pts = av_rescale_q(f->pts - ctx->start_pts, (AVRational){ 1, 1000000000 }, - ctx->avctx->time_base); - - /* Attach the hardware frame context to the frame */ - err = attach_drm_frames_ref(ctx, f, pix_fmt); - if (err) { - goto end; - } - - /* TODO: support multiplane stuff */ - desc->layers[0].nb_planes = av_pix_fmt_count_planes(pix_fmt); - - AVFrame *mapped_frame = av_frame_alloc(); - if (!mapped_frame) { - err = AVERROR(ENOMEM); - goto end; - } - - AVHWFramesContext *mapped_hwfc; - mapped_hwfc = (AVHWFramesContext *)ctx->mapped_frames_ref->data; - mapped_frame->format = mapped_hwfc->format; - mapped_frame->pts = f->pts; - - /* Set frame hardware context referencce */ - mapped_frame->hw_frames_ctx = av_buffer_ref(ctx->mapped_frames_ref); - if (!mapped_frame->hw_frames_ctx) { - err = AVERROR(ENOMEM); - goto end; - } - - err = av_hwframe_map(mapped_frame, f, 0); - if (err) { - av_log(ctx, AV_LOG_ERROR, "Error mapping: %s!\n", av_err2str(err)); - goto end; - } - - if (push_to_fifo(&ctx->vid_frames, mapped_frame)) { - av_log(ctx, AV_LOG_WARNING, "Dropped frame!\n"); - } - - if (!ctx->quit && !ctx->err) { - register_cb(ctx); - } - -end: - ctx->err = err; - av_frame_free(&ctx->current_frame); -} - -static void frame_cancel(void *data, struct zwlr_export_dmabuf_frame_v1 *frame, - uint32_t reason) { - struct capture_context *ctx = data; - av_log(ctx, AV_LOG_WARNING, "Frame cancelled!\n"); - av_frame_free(&ctx->current_frame); - if (reason == ZWLR_EXPORT_DMABUF_FRAME_V1_CANCEL_REASON_PERMANENT) { - av_log(ctx, AV_LOG_ERROR, "Permanent failure, exiting\n"); - ctx->err = true; - } else { - register_cb(ctx); - } -} - -static const struct zwlr_export_dmabuf_frame_v1_listener frame_listener = { - .frame = frame_start, - .object = frame_object, - .ready = frame_ready, - .cancel = frame_cancel, -}; - -static void register_cb(struct capture_context *ctx) { - ctx->frame_callback = zwlr_export_dmabuf_manager_v1_capture_output( - ctx->export_manager, ctx->with_cursor, ctx->target_output); - - zwlr_export_dmabuf_frame_v1_add_listener(ctx->frame_callback, - &frame_listener, ctx); -} - -static void *vid_encode_thread(void *arg) { - int err = 0; - struct capture_context *ctx = arg; - - do { - AVFrame *f = NULL; - if (get_fifo_size(&ctx->vid_frames) || !ctx->quit) { - f = pop_from_fifo(&ctx->vid_frames); - } - - if (ctx->is_software_encoder && f) { - AVFrame *soft_frame = av_frame_alloc(); - av_hwframe_transfer_data(soft_frame, f, 0); - soft_frame->pts = f->pts; - av_frame_free(&f); - f = soft_frame; - } - - err = avcodec_send_frame(ctx->avctx, f); - - av_frame_free(&f); - - if (err) { - av_log(ctx, AV_LOG_ERROR, "Error encoding: %s!\n", av_err2str(err)); - goto end; - } - - while (1) { - AVPacket *pkt = av_packet_alloc(); - int ret = avcodec_receive_packet(ctx->avctx, pkt); - if (ret == AVERROR(EAGAIN)) { - av_packet_free(&pkt); - break; - } else if (ret == AVERROR_EOF) { - av_log(ctx, AV_LOG_INFO, "Encoder flushed!\n"); - av_packet_free(&pkt); - goto end; - } else if (ret) { - av_log(ctx, AV_LOG_ERROR, "Error encoding: %s!\n", - av_err2str(ret)); - av_packet_free(&pkt); - err = ret; - goto end; - } - - pkt->stream_index = 0; - err = av_interleaved_write_frame(ctx->avf, pkt); - - av_packet_free(&pkt); - - if (err) { - av_log(ctx, AV_LOG_ERROR, "Writing packet fail: %s!\n", - av_err2str(err)); - goto end; - } - }; - - int64_t frame_num; -#if LIBAVUTIL_VERSION_MAJOR >= 58 - frame_num = ctx->avctx->frame_num; -#else - frame_num = ctx->avctx->frame_number; -#endif - av_log(ctx, AV_LOG_INFO, "Encoded frame %"PRIi64" (%i in queue)\n", - frame_num, get_fifo_size(&ctx->vid_frames)); - - } while (!ctx->err); - -end: - if (!ctx->err) { - ctx->err = err; - } - return NULL; -} - -static int init_lavu_hwcontext(struct capture_context *ctx) { - /* DRM hwcontext */ - ctx->drm_device_ref = av_hwdevice_ctx_alloc(AV_HWDEVICE_TYPE_DRM); - if (!ctx->drm_device_ref) - return AVERROR(ENOMEM); - - AVHWDeviceContext *ref_data = (AVHWDeviceContext*)ctx->drm_device_ref->data; - AVDRMDeviceContext *hwctx = ref_data->hwctx; - - /* We don't need a device (we don't even know it and can't open it) */ - hwctx->fd = -1; - - av_hwdevice_ctx_init(ctx->drm_device_ref); - - /* Mapped hwcontext */ - int err = av_hwdevice_ctx_create(&ctx->mapped_device_ref, - ctx->hw_device_type, ctx->hardware_device, NULL, 0); - if (err < 0) { - av_log(ctx, AV_LOG_ERROR, "Failed to create a hardware device: %s\n", - av_err2str(err)); - return err; - } - - return 0; -} - -static int set_hwframe_ctx(struct capture_context *ctx, - AVBufferRef *hw_device_ctx) { - AVHWFramesContext *frames_ctx = NULL; - int err = 0; - - if (!(ctx->mapped_frames_ref = av_hwframe_ctx_alloc(hw_device_ctx))) { - return AVERROR(ENOMEM); - } - - AVHWFramesConstraints *cst = - av_hwdevice_get_hwframe_constraints(ctx->mapped_device_ref, NULL); - if (!cst) { - av_log(ctx, AV_LOG_ERROR, "Failed to get hw device constraints!\n"); - av_buffer_unref(&ctx->mapped_frames_ref); - return AVERROR(ENOMEM); - } - - frames_ctx = (AVHWFramesContext *)(ctx->mapped_frames_ref->data); - frames_ctx->format = cst->valid_hw_formats[0]; - frames_ctx->sw_format = ctx->avctx->pix_fmt; - frames_ctx->width = ctx->avctx->width; - frames_ctx->height = ctx->avctx->height; - - av_hwframe_constraints_free(&cst); - - if ((err = av_hwframe_ctx_init(ctx->mapped_frames_ref))) { - av_log(ctx, AV_LOG_ERROR, "Failed to initialize hw frame context: %s!\n", - av_err2str(err)); - av_buffer_unref(&ctx->mapped_frames_ref); - return err; - } - - if (!ctx->is_software_encoder) { - ctx->avctx->pix_fmt = frames_ctx->format; - ctx->avctx->hw_frames_ctx = av_buffer_ref(ctx->mapped_frames_ref); - if (!ctx->avctx->hw_frames_ctx) { - av_buffer_unref(&ctx->mapped_frames_ref); - err = AVERROR(ENOMEM); - } - } - - return err; -} - -static int init_encoding(struct capture_context *ctx) { - int err; - - /* lavf init */ - err = avformat_alloc_output_context2(&ctx->avf, NULL, - NULL, ctx->out_filename); - if (err) { - av_log(ctx, AV_LOG_ERROR, "Unable to init lavf context!\n"); - return err; - } - - AVStream *st = avformat_new_stream(ctx->avf, NULL); - if (!st) { - av_log(ctx, AV_LOG_ERROR, "Unable to alloc stream!\n"); - return 1; - } - - /* Find encoder */ - const AVCodec *out_codec = avcodec_find_encoder_by_name(ctx->encoder_name); - if (!out_codec) { - av_log(ctx, AV_LOG_ERROR, "Codec not found (not compiled in lavc?)!\n"); - return AVERROR(EINVAL); - } - ctx->avf->oformat = av_guess_format(ctx->encoder_name, NULL, NULL); - ctx->is_software_encoder = !(out_codec->capabilities & AV_CODEC_CAP_HARDWARE); - - ctx->avctx = avcodec_alloc_context3(out_codec); - if (!ctx->avctx) - return 1; - - ctx->avctx->opaque = ctx; - ctx->avctx->bit_rate = (int)ctx->out_bitrate*1000000.0f; - ctx->avctx->pix_fmt = ctx->software_format; - ctx->avctx->time_base = (AVRational){ 1, 1000 }; - ctx->avctx->compression_level = 7; - ctx->avctx->width = find_output(ctx, ctx->target_output, 0)->width; - ctx->avctx->height = find_output(ctx, ctx->target_output, 0)->height; - - if (ctx->avf->oformat->flags & AVFMT_GLOBALHEADER) { - ctx->avctx->flags |= AV_CODEC_FLAG_GLOBAL_HEADER; - } - - st->id = 0; - st->time_base = ctx->avctx->time_base; - st->avg_frame_rate = find_output(ctx, ctx->target_output, 0)->framerate; - - /* Init hw frames context */ - err = set_hwframe_ctx(ctx, ctx->mapped_device_ref); - if (err) { - return err; - } - - err = avcodec_open2(ctx->avctx, out_codec, &ctx->encoder_opts); - if (err) { - av_log(ctx, AV_LOG_ERROR, "Cannot open encoder: %s!\n", - av_err2str(err)); - return err; - } - - if (avcodec_parameters_from_context(st->codecpar, ctx->avctx) < 0) { - av_log(ctx, AV_LOG_ERROR, "Couldn't copy codec params: %s!\n", - av_err2str(err)); - return err; - } - - /* Debug print */ - av_dump_format(ctx->avf, 0, ctx->out_filename, 1); - - /* Open for writing */ - err = avio_open(&ctx->avf->pb, ctx->out_filename, AVIO_FLAG_WRITE); - if (err) { - av_log(ctx, AV_LOG_ERROR, "Couldn't open %s: %s!\n", ctx->out_filename, - av_err2str(err)); - return err; - } - - err = avformat_write_header(ctx->avf, NULL); - if (err) { - av_log(ctx, AV_LOG_ERROR, "Couldn't write header: %s!\n", av_err2str(err)); - return err; - } - - return err; -} - -struct capture_context *q_ctx = NULL; - -static void on_quit_signal(int signo) { - printf("\r"); - av_log(q_ctx, AV_LOG_WARNING, "Quitting!\n"); - q_ctx->quit = true; -} - -static int main_loop(struct capture_context *ctx) { - int err; - - q_ctx = ctx; - - if (signal(SIGINT, on_quit_signal) == SIG_ERR) { - av_log(ctx, AV_LOG_ERROR, "Unable to install signal handler!\n"); - return AVERROR(EINVAL); - } - - err = init_lavu_hwcontext(ctx); - if (err) { - return err; - } - - err = init_encoding(ctx); - if (err) { - return err; - } - - /* Start video encoding thread */ - err = init_fifo(&ctx->vid_frames, 16); - if (err) { - return err; - } - pthread_create(&ctx->vid_thread, NULL, vid_encode_thread, ctx); - - /* Start the frame callback */ - register_cb(ctx); - - /* Run capture */ - while (wl_display_dispatch(ctx->display) != -1 && !ctx->err && !ctx->quit); - - /* Join with encoder thread */ - pthread_join(ctx->vid_thread, NULL); - - err = av_write_trailer(ctx->avf); - if (err) { - av_log(ctx, AV_LOG_ERROR, "Error writing trailer: %s!\n", - av_err2str(err)); - return err; - } - - av_log(ctx, AV_LOG_INFO, "Wrote trailer!\n"); - - return ctx->err; -} - -static int init(struct capture_context *ctx) { - ctx->display = wl_display_connect(NULL); - if (!ctx->display) { - av_log(ctx, AV_LOG_ERROR, "Failed to connect to display!\n"); - return AVERROR(EINVAL); - } - - wl_list_init(&ctx->output_list); - - ctx->registry = wl_display_get_registry(ctx->display); - wl_registry_add_listener(ctx->registry, ®istry_listener, ctx); - - // First roundtrip to fetch globals - wl_display_roundtrip(ctx->display); - - // Second roundtrip to fetch wl_output information - wl_display_roundtrip(ctx->display); - - if (!ctx->export_manager) { - av_log(ctx, AV_LOG_ERROR, "Compositor doesn't support %s!\n", - zwlr_export_dmabuf_manager_v1_interface.name); - return -1; - } - - return 0; -} - -static void uninit(struct capture_context *ctx); - -static const char usage[] = "usage: dmabuf-capture [options...] <destination file path>\n" - " -o <output ID>\n" - " -t <hardware device type>\n" - " -d <device path>\n" - " -e <encoder>\n" - " -f <pixel format>\n" - " -r <bitrate in Mbps>\n" - "\n" - "Example:\n" - " dmabuf-capture -o 32 -t vaapi -d /dev/dri/renderD129 \\\n" - " -e libx264 -f nv12 -r 12 recording.mkv\n"; - -int main(int argc, char *argv[]) { - struct capture_context ctx = { - .hardware_device = "/dev/dri/renderD128", - .encoder_name = "libx264", - .out_bitrate = 12, - }; - int output_id = -1; - const char *hw_device_type = "vaapi"; - const char *software_format = "nv12"; - int opt; - while ((opt = getopt(argc, argv, "ho:t:d:e:f:r:")) != -1) { - char *end; - switch (opt) { - case 'o': - output_id = strtol(optarg, &end, 10); - if (optarg[0] == '\0' || end[0] != '\0') { - fprintf(stderr, "Output ID is not an integer\n"); - return 1; - } - break; - case 't': - hw_device_type = optarg; - break; - case 'd': - ctx.hardware_device = optarg; - break; - case 'e': - ctx.encoder_name = optarg; - break; - case 'f': - software_format = optarg; - break; - case 'r': - ctx.out_bitrate = strtof(optarg, &end); - if (optarg[0] == '\0' || end[0] != '\0') { - fprintf(stderr, "Bitrate is not a floating-pointer number\n"); - return 1; - } - break; - default: - fprintf(stderr, "%s", usage); - return 1; - } - } - - if (optind >= argc) { - fprintf(stderr, "Missing destination file argument\n"); - fprintf(stderr, "%s", usage); - return 1; - } - ctx.out_filename = argv[optind]; - - ctx.class = &((AVClass){ - .class_name = "dmabuf-capture", - .item_name = av_default_item_name, - .version = LIBAVUTIL_VERSION_INT, - }); - - int err = init(&ctx); - if (err) { - goto end; - } - - struct wayland_output *o, *tmp_o; - wl_list_for_each_reverse_safe(o, tmp_o, &ctx.output_list, link) { - printf("Capturable output: %s Model: %s: ID: %i\n", - o->make, o->model, o->id); - } - - o = find_output(&ctx, NULL, output_id); - if (!o) { - printf("Unable to find output with ID %d\n", output_id); - return 1; - } - - ctx.target_output = o->output; - ctx.with_cursor = true; - ctx.hw_device_type = av_hwdevice_find_type_by_name(hw_device_type); - ctx.software_format = av_get_pix_fmt(software_format); - - av_dict_set(&ctx.encoder_opts, "preset", "veryfast", 0); - - err = main_loop(&ctx); - if (err) { - goto end; - } - -end: - uninit(&ctx); - return err; -} - -static void uninit(struct capture_context *ctx) { - struct wayland_output *output, *tmp_o; - wl_list_for_each_safe(output, tmp_o, &ctx->output_list, link) { - remove_output(output); - } - - if (ctx->export_manager) { - zwlr_export_dmabuf_manager_v1_destroy(ctx->export_manager); - } - - free_fifo(&ctx->vid_frames); - - av_buffer_unref(&ctx->drm_frames_ref); - av_buffer_unref(&ctx->drm_device_ref); - av_buffer_unref(&ctx->mapped_frames_ref); - av_buffer_unref(&ctx->mapped_device_ref); - - av_dict_free(&ctx->encoder_opts); - - avcodec_close(ctx->avctx); - if (ctx->avf) { - avio_closep(&ctx->avf->pb); - } - avformat_free_context(ctx->avf); -} diff --git a/examples/relative-pointer-unstable-v1.c b/examples/relative-pointer-unstable-v1.c deleted file mode 100644 index 6722a5b9..00000000 --- a/examples/relative-pointer-unstable-v1.c +++ /dev/null @@ -1,489 +0,0 @@ -#include <stdio.h> -#include <stdlib.h> -#include <string.h> -#include <GLES2/gl2.h> -#include <linux/input-event-codes.h> -#include <wayland-egl.h> -#include "egl_common.h" -#include "pointer-constraints-unstable-v1-client-protocol.h" -#include "relative-pointer-unstable-v1-client-protocol.h" -#include "xdg-shell-client-protocol.h" - - -/** - * structs storing global state - * - * These are passed around as user data in the wayland globals. - */ - -struct egl_info { - struct wl_egl_window *egl_window; - struct wlr_egl_surface *egl_surface; - uint32_t width; - uint32_t height; - struct wl_surface *surface; - struct wl_callback *frame_callback; -}; - -struct window { - struct egl_info *egl_info; - struct wl_pointer *pointer; - struct zwp_locked_pointer_v1 *locked_pointer; - struct zwp_relative_pointer_v1 *relative_pointer; - int32_t pointer_x, pointer_y; - uint32_t last_draw; -}; - - -/** - * surface handling and helpers - * - * draw_cursor and draw_relative_cursor draw a small 5px by 5px box around the - * cursor and relative motion coordinates respectively. draw_background colors - * the background black. - * - * The functions are somewhat duplicated, but it doesn't really matter. - */ - -static void surface_callback_handle_done(void *data, - struct wl_callback *callback, uint32_t time) { - wl_callback_destroy(callback); - struct egl_info *e = data; - e->frame_callback = NULL; -} - -static struct wl_callback_listener surface_callback_listener = { - .done = surface_callback_handle_done, -}; - -static void draw_init(struct egl_info *e) { - eglMakeCurrent(egl_display, e->egl_surface, - e->egl_surface, egl_context); - glViewport(0, 0, e->width, e->height); -} - -static void draw_cursor(struct egl_info *e, int32_t x, int32_t y) { - glEnable(GL_SCISSOR_TEST); - glScissor(x, e->height - y, 5, 5); - glClearColor(1.0f, 1.0f, 1.0f, 1.0f); /* white */ - glClear(GL_COLOR_BUFFER_BIT); - glDisable(GL_SCISSOR_TEST); -} - -static void draw_relative_cursor(struct egl_info *e, int32_t x, int32_t y) { - glEnable(GL_SCISSOR_TEST); - glScissor(x, e->height - y, 5, 5); - glClearColor(1.0f, 0.0f, 0.0f, 1.0f); /* red */ - glClear(GL_COLOR_BUFFER_BIT); - glDisable(GL_SCISSOR_TEST); -} - -static void draw_background(struct egl_info *e) { - glClearColor(0.0f, 0.0f, 0.0f, 1.0f); /* black */ - glClear(GL_COLOR_BUFFER_BIT); -} - -static void draw_end(struct egl_info *e) { - e->frame_callback = wl_surface_frame(e->surface); - wl_callback_add_listener(e->frame_callback, &surface_callback_listener, e); - - eglSwapBuffers(egl_display, e->egl_surface); -} - - -/** - * registry and globals handling - */ - -static struct wl_compositor *compositor = NULL; -static struct wl_seat *seat = NULL; -static struct xdg_wm_base *wm_base = NULL; -static struct zwp_pointer_constraints_v1 *pointer_constraints = NULL; -static struct zwp_relative_pointer_manager_v1 *relative_pointer_manager = NULL; - -static void registry_handle_global(void *data, struct wl_registry *registry, - uint32_t name, const char *interface, uint32_t version) { - if (strcmp(interface, wl_compositor_interface.name) == 0) { - compositor = wl_registry_bind(registry, name, - &wl_compositor_interface, version); - } else if (strcmp(interface, xdg_wm_base_interface.name) == 0) { - wm_base = wl_registry_bind(registry, name, - &xdg_wm_base_interface, version); - } else if (strcmp(interface, wl_seat_interface.name) == 0) { - seat = wl_registry_bind(registry, name, - &wl_seat_interface, version); - } else if (strcmp(interface, zwp_pointer_constraints_v1_interface.name) == 0) { - pointer_constraints = wl_registry_bind(registry, name, - &zwp_pointer_constraints_v1_interface, version); - } else if (strcmp(interface, zwp_relative_pointer_manager_v1_interface.name) == 0) { - relative_pointer_manager = wl_registry_bind(registry, name, - &zwp_relative_pointer_manager_v1_interface, version); - } -} - -static void registry_handle_global_remove(void *data, struct wl_registry *registry, - uint32_t time) { - /* This space intentionally left blank */ -} - -static const struct wl_registry_listener registry_listener = { - .global = registry_handle_global, - .global_remove = registry_handle_global_remove, -}; - - -/** - * xdg_surface handling - */ - -static void xdg_surface_handle_configure(void *data, - struct xdg_surface *xdg_surface, uint32_t serial) { - struct egl_info *e = data; - xdg_surface_ack_configure(xdg_surface, serial); - wl_egl_window_resize(e->egl_window, e->width, e->height, 0, 0); - draw_init(e); - draw_background(e); - draw_end(e); -} - -static const struct xdg_surface_listener xdg_surface_listener = { - .configure = xdg_surface_handle_configure, -}; - - -/** - * xdg_toplevel handling - */ - -static void xdg_toplevel_handle_configure(void *data, - struct xdg_toplevel *xdg_toplevel, int32_t width, int32_t height, - struct wl_array *states) { - struct egl_info *e = data; - // TODO: Why do we get 0,0 on initialization here? (in rootston) - if (width == 0 && height == 0) { - return; - } - e->width = width; - e->height = height; -} - -static void xdg_toplevel_handle_close(void *data, - struct xdg_toplevel *xdg_toplevel) { - egl_finish(); - exit(EXIT_SUCCESS); -} - -static const struct xdg_toplevel_listener xdg_toplevel_listener = { - .configure = xdg_toplevel_handle_configure, - .close = xdg_toplevel_handle_close, -}; - - -/** - * zwp_locked_pointer handling - * - * Pointer unlocks need to be handled properly since the relative pointer needs - * to be released as well. Unlocks happen when the focus is changed, for - * example. - */ - -static void locked_pointer_handle_locked(void *data, - struct zwp_locked_pointer_v1 *zwp_locked_pointer_v1) { - /* This space intentionally left blank */ -} - -static void locked_pointer_handle_unlocked(void *data, - struct zwp_locked_pointer_v1 *zwp_locked_pointer_v1) { - struct window *w = data; - /* The locked pointer doesn't need to be destroyed since it was oneshot */ - w->locked_pointer = NULL; - if (w->relative_pointer) { - /* Destroy the relative pointer */ - zwp_relative_pointer_v1_destroy(w->relative_pointer); - w->relative_pointer = NULL; - } - draw_init(w->egl_info); - draw_background(w->egl_info); - draw_end(w->egl_info); -} - -static const struct zwp_locked_pointer_v1_listener locked_pointer_listener = { - .locked = locked_pointer_handle_locked, - .unlocked = locked_pointer_handle_unlocked, -}; - - -/** - * zwp_relative_pointer handling - * - * Handling relative_motion events is the meat of the code. The handler simply - * tries to indicate what the deltas look like. - */ - -static void relative_pointer_handle_relative_motion(void *data, - struct zwp_relative_pointer_v1 *zwp_relative_pointer_v1, - uint32_t utime_hi, uint32_t utime_lo, - wl_fixed_t dx, wl_fixed_t dy, - wl_fixed_t dx_unaccel, wl_fixed_t dy_unaccel) { - struct window *w = data; - - /** - * This renders the last location of the pointer (before it as locked), as - * well as what the position would have been after the given relative - * motion. Note that, if the device sends absolute motion events, the - * cursor location after relative motion is always identical to the actual - * cursor position. - */ - uint64_t utime = (((uint64_t) utime_hi << 32) + utime_lo) / 1000; - if (utime - w->last_draw > 30 && w->egl_info->frame_callback == NULL) { - w->last_draw = utime; - - struct egl_info *e = w->egl_info; - draw_init(e); - draw_background(e); - draw_cursor(e, w->pointer_x, w->pointer_y); - draw_relative_cursor(e, w->pointer_x + wl_fixed_to_int(dx), - w->pointer_y + wl_fixed_to_int(dy)); - draw_end(e); - } -} - -static const struct zwp_relative_pointer_v1_listener relative_pointer_listener = { - .relative_motion = relative_pointer_handle_relative_motion, -}; - - -/** - * wl_pointer handling - * - * The client toggles between locking the pointer and receiving relative motion - * events, and releasing the locked pointer and falling back to normal motion - * events, on a mouse button one (left mouse button) click. - * - * It additionally removes the cursor image, and indicates the pointer location - * via a small white box. - */ - -static void pointer_handle_button(void *data, struct wl_pointer *pointer, - uint32_t serial, uint32_t time, uint32_t button, uint32_t state_w) { - - struct window *w = data; - struct egl_info *e = w->egl_info; - - if (button == BTN_LEFT && state_w == WL_POINTER_BUTTON_STATE_PRESSED) { - if (w->locked_pointer == NULL && w->relative_pointer == NULL) { - /* Get a locked pointer and add listener */ - w->locked_pointer = zwp_pointer_constraints_v1_lock_pointer( - pointer_constraints, w->egl_info->surface, w->pointer, NULL, - ZWP_POINTER_CONSTRAINTS_V1_LIFETIME_ONESHOT); - zwp_locked_pointer_v1_add_listener(w->locked_pointer, - &locked_pointer_listener, w); - - /* Get relative pointer and add listener */ - w->relative_pointer = zwp_relative_pointer_manager_v1_get_relative_pointer( - relative_pointer_manager, w->pointer); - zwp_relative_pointer_v1_add_listener(w->relative_pointer, - &relative_pointer_listener, w); - - /* Commit the surface and render */ - wl_surface_commit(e->surface); - - draw_init(e); - draw_background(e); - draw_cursor(e, w->pointer_x, w->pointer_y); - draw_end(e); - } else if (w->locked_pointer && w->relative_pointer) { - /* Destroy the locked pointer */ - zwp_locked_pointer_v1_destroy(w->locked_pointer); - w->locked_pointer = NULL; - - /* Destroy the relative pointer */ - zwp_relative_pointer_v1_destroy(w->relative_pointer); - w->relative_pointer = NULL; - - /* Render */ - draw_init(e); - draw_background(e); - draw_cursor(e, w->pointer_x, w->pointer_y); - draw_end(e); - } else { - fprintf(stderr, "Unknown state!\n"); - exit(EXIT_FAILURE); - } - } -} - -static void pointer_handle_enter(void *data, struct wl_pointer *wl_pointer, - uint32_t serial, struct wl_surface *surface, - wl_fixed_t surface_x, wl_fixed_t surface_y) { - struct window *w = data; - wl_pointer_set_cursor(w->pointer, serial, NULL, 0, 0); -} - -static void pointer_handle_leave(void *data, struct wl_pointer *wl_pointer, - uint32_t serial, struct wl_surface *surface) { - /* This space intentionally left blank */ -} - -static void pointer_handle_motion(void *data, struct wl_pointer *wl_pointer, - uint32_t time, wl_fixed_t surface_x, wl_fixed_t surface_y) { - struct window *w = data; - struct egl_info *e = w->egl_info; - w->pointer_x = wl_fixed_to_int(surface_x); - w->pointer_y = wl_fixed_to_int(surface_y); - if (time - w->last_draw > 30 && e->frame_callback == NULL) { - w->last_draw = time; - draw_init(e); - draw_background(e); - draw_cursor(e, w->pointer_x, w->pointer_y); - draw_end(e); - } -} - -static void pointer_handle_axis(void *data, struct wl_pointer *wl_pointer, - uint32_t time, uint32_t axis, wl_fixed_t value) { - /* This space intentionally left blank */ -} - -static void pointer_handle_frame(void *data, struct wl_pointer *wl_pointer) { - /* This space intentionally left blank */ -} - -static void pointer_handle_axis_source(void *data, - struct wl_pointer *wl_pointer, uint32_t axis_source) { - /* This space intentionally left blank */ -} - -static void pointer_handle_axis_stop(void *data, - struct wl_pointer *wl_pointer, uint32_t time, uint32_t axis) { - /* This space intentionally left blank */ -} - -static void pointer_handle_axis_discrete(void *data, - struct wl_pointer *wl_pointer, uint32_t axis, int32_t discrete) { - /* This space intentionally left blank */ -} - -static const struct wl_pointer_listener pointer_listener = { - .enter = pointer_handle_enter, - .leave = pointer_handle_leave, - .motion = pointer_handle_motion, - .button = pointer_handle_button, - .axis = pointer_handle_axis, - .frame = pointer_handle_frame, - .axis_source = pointer_handle_axis_source, - .axis_stop = pointer_handle_axis_stop, - .axis_discrete = pointer_handle_axis_discrete, -}; - - -/** - * relative-pointer: - * - * This client servers as an example for the relative-pointer-v1 protocol, and - * to some extent the pointer-constraints protocol as well (the locked_pointer - * interface). - * - * The intended behavior is the following. In the default state, the client - * shows a black background, and renders the pointer location as a small white - * box. No cursor is shown. In the locked state, the pointer is locked and the - * client only listens for relative motion events, which are rendered relative - * to the last unlocked pointer location by a small red box. Clicking with the - * left mouse button toggles the state. - * - * Most of the code is standard. The interesting stuff happens in "wl_pointer - * handling" (toggling states), and "zwp_relative_pointer handling" (rendering - * relative motion events). - */ - -int main(int argc, char **argv) { - - /* Connect to the display */ - - struct wl_display *display = wl_display_connect(NULL); - if (display == NULL) { - fprintf(stderr, "Could not connect to a Wayland display\n"); - return EXIT_FAILURE; - } - - /* Get the registry and set listeners */ - - struct wl_registry *registry = wl_display_get_registry(display); - wl_registry_add_listener(registry, ®istry_listener, NULL); - wl_display_roundtrip(display); - - /* Check that all the global interfaces were captured */ - - if (compositor == NULL) { - fprintf(stderr, "wl_compositor not available\n"); - return EXIT_FAILURE; - } - if (seat == NULL) { - fprintf(stderr, "wl_seat not available\n"); - return EXIT_FAILURE; - } - if (wm_base == NULL) { - fprintf(stderr, "xdg_wm_base not available\n"); - return EXIT_FAILURE; - } - if (pointer_constraints == NULL) { - fprintf(stderr, "zwp_pointer_constraints_v1 not available\n"); - return EXIT_FAILURE; - } - if (relative_pointer_manager == NULL) { - fprintf(stderr, "zwp_relative_pointer_manager_v1 not available\n"); - return EXIT_FAILURE; - } - - /* Initialize EGL context */ - - struct egl_info *e = calloc(1, sizeof(*e)); - e->width = e->height = 512; - - egl_init(display); - - /* Create the surface and xdg_toplevels, and set listeners */ - - struct wl_surface *surface = wl_compositor_create_surface(compositor); - struct xdg_surface *xdg_surface = - xdg_wm_base_get_xdg_surface(wm_base, surface); - struct xdg_toplevel *xdg_toplevel = xdg_surface_get_toplevel(xdg_surface); - - xdg_surface_add_listener(xdg_surface, &xdg_surface_listener, e); - xdg_toplevel_add_listener(xdg_toplevel, &xdg_toplevel_listener, e); - - /* Create the egl window and surface */ - - wl_surface_commit(surface); - - e->egl_window = wl_egl_window_create(surface, e->width, e->height); - e->egl_surface = eglCreatePlatformWindowSurfaceEXT( - egl_display, egl_config, e->egl_window, NULL); - e->surface = surface; - - wl_display_roundtrip(display); - - /* Setup global state and render */ - - struct window *w = calloc(1, sizeof(*w)); - w->egl_info = e; - - draw_init(e); - draw_background(e); - draw_end(e); - - /* Setup the pointer */ - - struct wl_pointer *pointer = wl_seat_get_pointer(seat); - wl_pointer_add_listener(pointer, &pointer_listener, w); - - w->pointer = pointer; - - /* Run display */ - - while (wl_display_dispatch(display) != -1) { - /* This space intentionally left blank */ - } - - return EXIT_SUCCESS; -} |