mirror of
https://github.com/hyprwm/wlroots-hyprland.git
synced 2024-11-05 05:05:57 +01:00
examples: drop dead client code
Client examples have been moved to another repo, but it seems I
forgot to delete some files.
Fixes: 0bb445eeff
("examples: split clients in separate repository")
This commit is contained in:
parent
47bf87ade2
commit
ffa8e3ec81
2 changed files with 0 additions and 1398 deletions
|
@ -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);
|
||||
}
|
|
@ -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;
|
||||
}
|
Loading…
Reference in a new issue