aboutsummaryrefslogtreecommitdiff
path: root/src/pipewire.c
diff options
context:
space:
mode:
authordec05eba <dec05eba@protonmail.com>2024-11-08 18:23:50 +0100
committerdec05eba <dec05eba@protonmail.com>2024-11-08 18:40:33 +0100
commit0686b924def291323c94f8d26d96fd0ef9d854c4 (patch)
tree5902cf270e7177559db240daae31f60577abf6d9 /src/pipewire.c
parentab662f1ed5c6165e54bb6cfa2ff4c06de12c1a99 (diff)
Prepare for pipewire audio routing
Diffstat (limited to 'src/pipewire.c')
-rw-r--r--src/pipewire.c788
1 files changed, 0 insertions, 788 deletions
diff --git a/src/pipewire.c b/src/pipewire.c
deleted file mode 100644
index 3bf54db..0000000
--- a/src/pipewire.c
+++ /dev/null
@@ -1,788 +0,0 @@
-#include "../include/pipewire.h"
-#include "../include/egl.h"
-#include "../include/utils.h"
-
-#include <pipewire/pipewire.h>
-#include <spa/param/video/format-utils.h>
-#include <spa/debug/types.h>
-
-#include <libdrm/drm_fourcc.h>
-
-#include <fcntl.h>
-#include <unistd.h>
-
-/* This code is partially based on xr-video-player pipewire implementation which is based on obs-studio's pipewire implementation */
-
-/* TODO: Make gsr_pipewire_init asynchronous */
-/* TODO: Support 10-bit capture (hdr) when pipewire supports it */
-/* TODO: Test all of the image formats */
-
-#ifndef SPA_POD_PROP_FLAG_DONT_FIXATE
-#define SPA_POD_PROP_FLAG_DONT_FIXATE (1 << 4)
-#endif
-
-#define CURSOR_META_SIZE(width, height) \
- (sizeof(struct spa_meta_cursor) + sizeof(struct spa_meta_bitmap) + \
- width * height * 4)
-
-static bool parse_pw_version(gsr_pipewire_data_version *dst, const char *version) {
- const int n_matches = sscanf(version, "%d.%d.%d", &dst->major, &dst->minor, &dst->micro);
- return n_matches == 3;
-}
-
-static bool check_pw_version(const gsr_pipewire_data_version *pw_version, int major, int minor, int micro) {
- if (pw_version->major != major)
- return pw_version->major > major;
- if (pw_version->minor != minor)
- return pw_version->minor > minor;
- return pw_version->micro >= micro;
-}
-
-static void update_pw_versions(gsr_pipewire *self, const char *version) {
- fprintf(stderr, "gsr info: pipewire: server version: %s\n", version);
- fprintf(stderr, "gsr info: pipewire: library version: %s\n", pw_get_library_version());
- fprintf(stderr, "gsr info: pipewire: header version: %s\n", pw_get_headers_version());
- if(!parse_pw_version(&self->server_version, version))
- fprintf(stderr, "gsr error: pipewire: failed to parse server version\n");
-}
-
-static void on_core_info_cb(void *user_data, const struct pw_core_info *info) {
- gsr_pipewire *self = user_data;
- update_pw_versions(self, info->version);
-}
-
-static void on_core_error_cb(void *user_data, uint32_t id, int seq, int res, const char *message) {
- gsr_pipewire *self = user_data;
- fprintf(stderr, "gsr error: pipewire: error id:%u seq:%d res:%d: %s\n", id, seq, res, message);
- pw_thread_loop_signal(self->thread_loop, false);
-}
-
-static void on_core_done_cb(void *user_data, uint32_t id, int seq) {
- gsr_pipewire *self = user_data;
- if (id == PW_ID_CORE && self->server_version_sync == seq)
- pw_thread_loop_signal(self->thread_loop, false);
-}
-
-static bool is_cursor_format_supported(const enum spa_video_format format) {
- switch(format) {
- case SPA_VIDEO_FORMAT_RGBx: return true;
- case SPA_VIDEO_FORMAT_BGRx: return true;
- case SPA_VIDEO_FORMAT_xRGB: return true;
- case SPA_VIDEO_FORMAT_xBGR: return true;
- case SPA_VIDEO_FORMAT_RGBA: return true;
- case SPA_VIDEO_FORMAT_BGRA: return true;
- case SPA_VIDEO_FORMAT_ARGB: return true;
- case SPA_VIDEO_FORMAT_ABGR: return true;
- default: break;
- }
- return false;
-}
-
-static const struct pw_core_events core_events = {
- PW_VERSION_CORE_EVENTS,
- .info = on_core_info_cb,
- .done = on_core_done_cb,
- .error = on_core_error_cb,
-};
-
-static void on_process_cb(void *user_data) {
- gsr_pipewire *self = user_data;
- struct spa_meta_cursor *cursor = NULL;
- //struct spa_meta *video_damage = NULL;
-
- /* Find the most recent buffer */
- struct pw_buffer *pw_buf = NULL;
- for(;;) {
- struct pw_buffer *aux = pw_stream_dequeue_buffer(self->stream);
- if(!aux)
- break;
- if(pw_buf)
- pw_stream_queue_buffer(self->stream, pw_buf);
- pw_buf = aux;
- }
-
- if(!pw_buf) {
- fprintf(stderr, "gsr info: pipewire: out of buffers!\n");
- return;
- }
-
- struct spa_buffer *buffer = pw_buf->buffer;
- const bool has_buffer = buffer->datas[0].chunk->size != 0;
- if(!has_buffer)
- goto read_metadata;
-
- pthread_mutex_lock(&self->mutex);
-
- if(buffer->datas[0].type == SPA_DATA_DmaBuf) {
- for(size_t i = 0; i < self->dmabuf_num_planes; ++i) {
- if(self->dmabuf_data[i].fd > 0) {
- close(self->dmabuf_data[i].fd);
- self->dmabuf_data[i].fd = -1;
- }
- }
-
- self->dmabuf_num_planes = buffer->n_datas;
- if(self->dmabuf_num_planes > GSR_PIPEWIRE_DMABUF_MAX_PLANES)
- self->dmabuf_num_planes = GSR_PIPEWIRE_DMABUF_MAX_PLANES;
-
- for(size_t i = 0; i < self->dmabuf_num_planes; ++i) {
- self->dmabuf_data[i].fd = dup(buffer->datas[i].fd);
- self->dmabuf_data[i].offset = buffer->datas[i].chunk->offset;
- self->dmabuf_data[i].stride = buffer->datas[i].chunk->stride;
- }
-
- self->damaged = true;
- } else {
- // TODO:
- }
-
- // TODO: Move down to read_metadata
- struct spa_meta_region *region = spa_buffer_find_meta_data(buffer, SPA_META_VideoCrop, sizeof(*region));
- if(region && spa_meta_region_is_valid(region)) {
- // fprintf(stderr, "gsr info: pipewire: crop Region available (%dx%d+%d+%d)\n",
- // region->region.position.x, region->region.position.y,
- // region->region.size.width, region->region.size.height);
- self->crop.x = region->region.position.x;
- self->crop.y = region->region.position.y;
- self->crop.width = region->region.size.width;
- self->crop.height = region->region.size.height;
- self->crop.valid = true;
- } else {
- self->crop.valid = false;
- }
-
- pthread_mutex_unlock(&self->mutex);
-
-read_metadata:
-
- // video_damage = spa_buffer_find_meta(buffer, SPA_META_VideoDamage);
- // if(video_damage) {
- // struct spa_meta_region *r = spa_meta_first(video_damage);
- // if(spa_meta_check(r, video_damage)) {
- // //fprintf(stderr, "damage: %d,%d %ux%u\n", r->region.position.x, r->region.position.y, r->region.size.width, r->region.size.height);
- // pthread_mutex_lock(&self->mutex);
- // self->damaged = true;
- // pthread_mutex_unlock(&self->mutex);
- // }
- // }
-
- cursor = spa_buffer_find_meta_data(buffer, SPA_META_Cursor, sizeof(*cursor));
- self->cursor.valid = cursor && spa_meta_cursor_is_valid(cursor);
-
- if (self->cursor.visible && self->cursor.valid) {
- pthread_mutex_lock(&self->mutex);
-
- struct spa_meta_bitmap *bitmap = NULL;
- if (cursor->bitmap_offset)
- bitmap = SPA_MEMBER(cursor, cursor->bitmap_offset, struct spa_meta_bitmap);
-
- if (bitmap && bitmap->size.width > 0 && bitmap->size.height && is_cursor_format_supported(bitmap->format)) {
- const uint8_t *bitmap_data = SPA_MEMBER(bitmap, bitmap->offset, uint8_t);
- fprintf(stderr, "gsr info: pipewire: cursor bitmap update, size: %dx%d, format: %s\n",
- (int)bitmap->size.width, (int)bitmap->size.height, spa_debug_type_find_name(spa_type_video_format, bitmap->format));
-
- const size_t bitmap_size = bitmap->size.width * bitmap->size.height * 4;
- uint8_t *new_bitmap_data = realloc(self->cursor.data, bitmap_size);
- if(new_bitmap_data) {
- self->cursor.data = new_bitmap_data;
- /* TODO: Convert bgr and other image formats to rgb here */
- memcpy(self->cursor.data, bitmap_data, bitmap_size);
- }
-
- self->cursor.hotspot_x = cursor->hotspot.x;
- self->cursor.hotspot_y = cursor->hotspot.y;
- self->cursor.width = bitmap->size.width;
- self->cursor.height = bitmap->size.height;
- }
-
- self->cursor.x = cursor->position.x;
- self->cursor.y = cursor->position.y;
- pthread_mutex_unlock(&self->mutex);
-
- //fprintf(stderr, "gsr info: pipewire: cursor: %d %d %d %d\n", cursor->hotspot.x, cursor->hotspot.y, cursor->position.x, cursor->position.y);
- }
-
- pw_stream_queue_buffer(self->stream, pw_buf);
-}
-
-static void on_param_changed_cb(void *user_data, uint32_t id, const struct spa_pod *param) {
- gsr_pipewire *self = user_data;
-
- if (!param || id != SPA_PARAM_Format)
- return;
-
- int result = spa_format_parse(param, &self->format.media_type, &self->format.media_subtype);
- if (result < 0)
- return;
-
- if (self->format.media_type != SPA_MEDIA_TYPE_video || self->format.media_subtype != SPA_MEDIA_SUBTYPE_raw)
- return;
-
- pthread_mutex_lock(&self->mutex);
- spa_format_video_raw_parse(param, &self->format.info.raw);
- pthread_mutex_unlock(&self->mutex);
-
- uint32_t buffer_types = 0;
- const bool has_modifier = spa_pod_find_prop(param, NULL, SPA_FORMAT_VIDEO_modifier) != NULL;
- if(has_modifier || check_pw_version(&self->server_version, 0, 3, 24))
- buffer_types |= 1 << SPA_DATA_DmaBuf;
-
- fprintf(stderr, "gsr info: pipewire: negotiated format:\n");
-
- fprintf(stderr, "gsr info: pipewire: Format: %d (%s)\n",
- self->format.info.raw.format,
- spa_debug_type_find_name(spa_type_video_format, self->format.info.raw.format));
-
- if(has_modifier) {
- fprintf(stderr, "gsr info: pipewire: Modifier: 0x%" PRIx64 "\n", self->format.info.raw.modifier);
- }
-
- fprintf(stderr, "gsr info: pipewire: Size: %dx%d\n", self->format.info.raw.size.width, self->format.info.raw.size.height);
- fprintf(stderr, "gsr info: pipewire: Framerate: %d/%d\n", self->format.info.raw.framerate.num, self->format.info.raw.framerate.denom);
-
- uint8_t params_buffer[1024];
- struct spa_pod_builder pod_builder = SPA_POD_BUILDER_INIT(params_buffer, sizeof(params_buffer));
- const struct spa_pod *params[4];
-
- params[0] = spa_pod_builder_add_object(
- &pod_builder, SPA_TYPE_OBJECT_ParamMeta, SPA_PARAM_Meta,
- SPA_PARAM_META_type, SPA_POD_Id(SPA_META_VideoCrop),
- SPA_PARAM_META_size,
- SPA_POD_Int(sizeof(struct spa_meta_region)));
-
- params[1] = spa_pod_builder_add_object(
- &pod_builder, SPA_TYPE_OBJECT_ParamMeta, SPA_PARAM_Meta,
- SPA_PARAM_META_type, SPA_POD_Id(SPA_META_VideoDamage),
- SPA_PARAM_META_size, SPA_POD_CHOICE_RANGE_Int(
- sizeof(struct spa_meta_region) * 16,
- sizeof(struct spa_meta_region) * 1,
- sizeof(struct spa_meta_region) * 16));
-
- params[2] = spa_pod_builder_add_object(
- &pod_builder, SPA_TYPE_OBJECT_ParamMeta, SPA_PARAM_Meta,
- SPA_PARAM_META_type, SPA_POD_Id(SPA_META_Cursor),
- SPA_PARAM_META_size,
- SPA_POD_CHOICE_RANGE_Int(CURSOR_META_SIZE(64, 64),
- CURSOR_META_SIZE(1, 1),
- CURSOR_META_SIZE(1024, 1024)));
-
- params[3] = spa_pod_builder_add_object(
- &pod_builder, SPA_TYPE_OBJECT_ParamBuffers, SPA_PARAM_Buffers,
- SPA_PARAM_BUFFERS_dataType, SPA_POD_Int(buffer_types));
-
- pw_stream_update_params(self->stream, params, 4);
- self->negotiated = true;
-}
-
-static void on_state_changed_cb(void *user_data, enum pw_stream_state old, enum pw_stream_state state, const char *error) {
- (void)old;
- gsr_pipewire *self = user_data;
-
- fprintf(stderr, "gsr info: pipewire: stream %p state: \"%s\" (error: %s)\n",
- (void*)self->stream, pw_stream_state_as_string(state),
- error ? error : "none");
-}
-
-static const struct pw_stream_events stream_events = {
- PW_VERSION_STREAM_EVENTS,
- .state_changed = on_state_changed_cb,
- .param_changed = on_param_changed_cb,
- .process = on_process_cb,
-};
-
-static inline struct spa_pod *build_format(struct spa_pod_builder *b,
- const gsr_pipewire_video_info *ovi,
- uint32_t format, const uint64_t *modifiers,
- size_t modifier_count)
-{
- struct spa_pod_frame format_frame;
-
- spa_pod_builder_push_object(b, &format_frame, SPA_TYPE_OBJECT_Format, SPA_PARAM_EnumFormat);
- spa_pod_builder_add(b, SPA_FORMAT_mediaType, SPA_POD_Id(SPA_MEDIA_TYPE_video), 0);
- spa_pod_builder_add(b, SPA_FORMAT_mediaSubtype, SPA_POD_Id(SPA_MEDIA_SUBTYPE_raw), 0);
-
- spa_pod_builder_add(b, SPA_FORMAT_VIDEO_format, SPA_POD_Id(format), 0);
-
- if (modifier_count > 0) {
- struct spa_pod_frame modifier_frame;
-
- spa_pod_builder_prop(b, SPA_FORMAT_VIDEO_modifier, SPA_POD_PROP_FLAG_MANDATORY | SPA_POD_PROP_FLAG_DONT_FIXATE);
- spa_pod_builder_push_choice(b, &modifier_frame, SPA_CHOICE_Enum, 0);
-
- /* The first element of choice pods is the preferred value. Here
- * we arbitrarily pick the first modifier as the preferred one.
- */
- // TODO:
- spa_pod_builder_long(b, modifiers[0]);
-
- for(uint32_t i = 0; i < modifier_count; i++)
- spa_pod_builder_long(b, modifiers[i]);
-
- spa_pod_builder_pop(b, &modifier_frame);
- }
-
- spa_pod_builder_add(b, SPA_FORMAT_VIDEO_size,
- SPA_POD_CHOICE_RANGE_Rectangle(
- &SPA_RECTANGLE(32, 32),
- &SPA_RECTANGLE(1, 1),
- &SPA_RECTANGLE(16384, 16384)),
- SPA_FORMAT_VIDEO_framerate,
- SPA_POD_CHOICE_RANGE_Fraction(
- &SPA_FRACTION(ovi->fps_num, ovi->fps_den),
- &SPA_FRACTION(0, 1), &SPA_FRACTION(500, 1)),
- 0);
- return spa_pod_builder_pop(b, &format_frame);
-}
-
-/* https://gstreamer.freedesktop.org/documentation/additional/design/mediatype-video-raw.html?gi-language=c#formats */
-/* For some reason gstreamer formats are in opposite order to drm formats */
-static int64_t spa_video_format_to_drm_format(const enum spa_video_format format) {
- switch(format) {
- case SPA_VIDEO_FORMAT_RGBx: return DRM_FORMAT_XBGR8888;
- case SPA_VIDEO_FORMAT_BGRx: return DRM_FORMAT_XRGB8888;
- case SPA_VIDEO_FORMAT_RGBA: return DRM_FORMAT_ABGR8888;
- case SPA_VIDEO_FORMAT_BGRA: return DRM_FORMAT_ARGB8888;
- case SPA_VIDEO_FORMAT_RGB: return DRM_FORMAT_XBGR8888;
- case SPA_VIDEO_FORMAT_BGR: return DRM_FORMAT_XRGB8888;
- default: break;
- }
- return DRM_FORMAT_INVALID;
-}
-
-static const enum spa_video_format video_formats[] = {
- SPA_VIDEO_FORMAT_BGRA,
- SPA_VIDEO_FORMAT_BGRx,
- SPA_VIDEO_FORMAT_BGR,
- SPA_VIDEO_FORMAT_RGBx,
- SPA_VIDEO_FORMAT_RGBA,
- SPA_VIDEO_FORMAT_RGB,
-};
-
-static bool gsr_pipewire_build_format_params(gsr_pipewire *self, struct spa_pod_builder *pod_builder, struct spa_pod **params, uint32_t *num_params) {
- *num_params = 0;
-
- if(!check_pw_version(&self->server_version, 0, 3, 33))
- return false;
-
- for(size_t i = 0; i < GSR_PIPEWIRE_NUM_VIDEO_FORMATS; i++) {
- if(self->supported_video_formats[i].modifiers_size == 0)
- continue;
- params[i] = build_format(pod_builder, &self->video_info, self->supported_video_formats[i].format, self->modifiers + self->supported_video_formats[i].modifiers_index, self->supported_video_formats[i].modifiers_size);
- ++(*num_params);
- }
-
- return true;
-}
-
-static void renegotiate_format(void *data, uint64_t expirations) {
- (void)expirations;
- gsr_pipewire *self = (gsr_pipewire*)data;
-
- pw_thread_loop_lock(self->thread_loop);
-
- struct spa_pod *params[GSR_PIPEWIRE_NUM_VIDEO_FORMATS];
- uint32_t num_video_formats = 0;
- uint8_t params_buffer[2048];
- struct spa_pod_builder pod_builder = SPA_POD_BUILDER_INIT(params_buffer, sizeof(params_buffer));
- if (!gsr_pipewire_build_format_params(self, &pod_builder, params, &num_video_formats)) {
- pw_thread_loop_unlock(self->thread_loop);
- return;
- }
-
- pw_stream_update_params(self->stream, (const struct spa_pod**)params, num_video_formats);
- pw_thread_loop_unlock(self->thread_loop);
-}
-
-static bool spa_video_format_get_modifiers(gsr_pipewire *self, const enum spa_video_format format, uint64_t *modifiers, int32_t max_modifiers, int32_t *num_modifiers) {
- *num_modifiers = 0;
-
- if(max_modifiers == 0) {
- fprintf(stderr, "gsr error: spa_video_format_get_modifiers: no space for modifiers left\n");
- //modifiers[0] = DRM_FORMAT_MOD_LINEAR;
- //modifiers[1] = DRM_FORMAT_MOD_INVALID;
- //*num_modifiers = 2;
- return false;
- }
-
- if(!self->egl->eglQueryDmaBufModifiersEXT) {
- fprintf(stderr, "gsr error: spa_video_format_get_modifiers: failed to initialize modifiers because eglQueryDmaBufModifiersEXT is not available\n");
- //modifiers[0] = DRM_FORMAT_MOD_LINEAR;
- //modifiers[1] = DRM_FORMAT_MOD_INVALID;
- //*num_modifiers = 2;
- return false;
- }
-
- const int64_t drm_format = spa_video_format_to_drm_format(format);
- if(!self->egl->eglQueryDmaBufModifiersEXT(self->egl->egl_display, drm_format, max_modifiers, modifiers, NULL, num_modifiers)) {
- fprintf(stderr, "gsr error: spa_video_format_get_modifiers: eglQueryDmaBufModifiersEXT failed with drm format %d, %" PRIi64 "\n", (int)format, drm_format);
- //modifiers[0] = DRM_FORMAT_MOD_LINEAR;
- //modifiers[1] = DRM_FORMAT_MOD_INVALID;
- //*num_modifiers = 2;
- *num_modifiers = 0;
- return false;
- }
-
- // if(*num_modifiers + 2 <= max_modifiers) {
- // modifiers[*num_modifiers + 0] = DRM_FORMAT_MOD_LINEAR;
- // modifiers[*num_modifiers + 1] = DRM_FORMAT_MOD_INVALID;
- // *num_modifiers += 2;
- // }
- return true;
-}
-
-static void gsr_pipewire_init_modifiers(gsr_pipewire *self) {
- for(size_t i = 0; i < GSR_PIPEWIRE_NUM_VIDEO_FORMATS; i++) {
- self->supported_video_formats[i].format = video_formats[i];
- int32_t num_modifiers = 0;
- spa_video_format_get_modifiers(self, self->supported_video_formats[i].format, self->modifiers + self->num_modifiers, GSR_PIPEWIRE_MAX_MODIFIERS - self->num_modifiers, &num_modifiers);
- self->supported_video_formats[i].modifiers_index = self->num_modifiers;
- self->supported_video_formats[i].modifiers_size = num_modifiers;
- }
-}
-
-static bool gsr_pipewire_setup_stream(gsr_pipewire *self) {
- struct spa_pod *params[GSR_PIPEWIRE_NUM_VIDEO_FORMATS];
- uint32_t num_video_formats = 0;
- uint8_t params_buffer[2048];
- struct spa_pod_builder pod_builder = SPA_POD_BUILDER_INIT(params_buffer, sizeof(params_buffer));
-
- self->thread_loop = pw_thread_loop_new("PipeWire thread loop", NULL);
- if(!self->thread_loop) {
- fprintf(stderr, "gsr error: gsr_pipewire_setup_stream: failed to create pipewire thread\n");
- goto error;
- }
-
- self->context = pw_context_new(pw_thread_loop_get_loop(self->thread_loop), NULL, 0);
- if(!self->context) {
- fprintf(stderr, "gsr error: gsr_pipewire_setup_stream: failed to create pipewire context\n");
- goto error;
- }
-
- if(pw_thread_loop_start(self->thread_loop) < 0) {
- fprintf(stderr, "gsr error: gsr_pipewire_setup_stream: failed to start thread\n");
- goto error;
- }
-
- pw_thread_loop_lock(self->thread_loop);
-
- // TODO: Why pass 5 to fcntl?
- self->core = pw_context_connect_fd(self->context, fcntl(self->fd, F_DUPFD_CLOEXEC, 5), NULL, 0);
- if(!self->core) {
- pw_thread_loop_unlock(self->thread_loop);
- fprintf(stderr, "gsr error: gsr_pipewire_setup_stream: failed to connect to fd %d\n", self->fd);
- goto error;
- }
-
- // TODO: Error check
- pw_core_add_listener(self->core, &self->core_listener, &core_events, self);
-
- gsr_pipewire_init_modifiers(self);
-
- // TODO: Cleanup?
- self->reneg = pw_loop_add_event(pw_thread_loop_get_loop(self->thread_loop), renegotiate_format, self);
- if(!self->reneg) {
- pw_thread_loop_unlock(self->thread_loop);
- fprintf(stderr, "gsr error: gsr_pipewire_setup_stream: pw_loop_add_event failed\n");
- goto error;
- }
-
- self->server_version_sync = pw_core_sync(self->core, PW_ID_CORE, 0);
- pw_thread_loop_wait(self->thread_loop);
-
- self->stream = pw_stream_new(self->core, "com.dec05eba.gpu_screen_recorder",
- pw_properties_new(PW_KEY_MEDIA_TYPE, "Video",
- PW_KEY_MEDIA_CATEGORY, "Capture",
- PW_KEY_MEDIA_ROLE, "Screen", NULL));
- if(!self->stream) {
- pw_thread_loop_unlock(self->thread_loop);
- fprintf(stderr, "gsr error: gsr_pipewire_setup_stream: failed to create stream\n");
- goto error;
- }
- pw_stream_add_listener(self->stream, &self->stream_listener, &stream_events, self);
-
- if(!gsr_pipewire_build_format_params(self, &pod_builder, params, &num_video_formats)) {
- pw_thread_loop_unlock(self->thread_loop);
- fprintf(stderr, "gsr error: gsr_pipewire_setup_stream: failed to build format params\n");
- goto error;
- }
-
- if(pw_stream_connect(
- self->stream, PW_DIRECTION_INPUT, self->node,
- PW_STREAM_FLAG_AUTOCONNECT | PW_STREAM_FLAG_MAP_BUFFERS, (const struct spa_pod**)params,
- num_video_formats) < 0)
- {
- pw_thread_loop_unlock(self->thread_loop);
- fprintf(stderr, "gsr error: gsr_pipewire_setup_stream: failed to connect stream\n");
- goto error;
- }
-
- pw_thread_loop_unlock(self->thread_loop);
- return true;
-
- error:
- if(self->thread_loop) {
- //pw_thread_loop_wait(self->thread_loop);
- pw_thread_loop_stop(self->thread_loop);
- }
-
- if(self->stream) {
- pw_stream_disconnect(self->stream);
- pw_stream_destroy(self->stream);
- self->stream = NULL;
- }
-
- if(self->core) {
- pw_core_disconnect(self->core);
- self->core = NULL;
- }
-
- if(self->context) {
- pw_context_destroy(self->context);
- self->context = NULL;
- }
-
- if(self->thread_loop) {
- pw_thread_loop_destroy(self->thread_loop);
- self->thread_loop = NULL;
- }
- return false;
-}
-
-static int pw_init_counter = 0;
-bool gsr_pipewire_init(gsr_pipewire *self, int pipewire_fd, uint32_t pipewire_node, int fps, bool capture_cursor, gsr_egl *egl) {
- if(pw_init_counter == 0)
- pw_init(NULL, NULL);
- ++pw_init_counter;
-
- memset(self, 0, sizeof(*self));
- self->egl = egl;
- self->fd = pipewire_fd;
- self->node = pipewire_node;
- if(pthread_mutex_init(&self->mutex, NULL) != 0) {
- fprintf(stderr, "gsr error: gsr_pipewire_init: failed to initialize mutex\n");
- gsr_pipewire_deinit(self);
- return false;
- }
- self->mutex_initialized = true;
- self->video_info.fps_num = fps;
- self->video_info.fps_den = 1;
- self->cursor.visible = capture_cursor;
-
- if(!gsr_pipewire_setup_stream(self)) {
- gsr_pipewire_deinit(self);
- return false;
- }
-
- return true;
-}
-
-void gsr_pipewire_deinit(gsr_pipewire *self) {
- if(self->thread_loop) {
- //pw_thread_loop_wait(self->thread_loop);
- pw_thread_loop_stop(self->thread_loop);
- }
-
- if(self->stream) {
- pw_stream_disconnect(self->stream);
- pw_stream_destroy(self->stream);
- self->stream = NULL;
- }
-
- if(self->core) {
- pw_core_disconnect(self->core);
- self->core = NULL;
- }
-
- if(self->context) {
- pw_context_destroy(self->context);
- self->context = NULL;
- }
-
- if(self->thread_loop) {
- pw_thread_loop_destroy(self->thread_loop);
- self->thread_loop = NULL;
- }
-
- if(self->fd > 0) {
- close(self->fd);
- self->fd = -1;
- }
-
- for(size_t i = 0; i < self->dmabuf_num_planes; ++i) {
- if(self->dmabuf_data[i].fd > 0) {
- close(self->dmabuf_data[i].fd);
- self->dmabuf_data[i].fd = -1;
- }
- }
- self->dmabuf_num_planes = 0;
-
- self->negotiated = false;
-
- if(self->mutex_initialized) {
- pthread_mutex_destroy(&self->mutex);
- self->mutex_initialized = false;
- }
-
- if(self->cursor.data) {
- free(self->cursor.data);
- self->cursor.data = NULL;
- }
-
- --pw_init_counter;
- if(pw_init_counter == 0) {
-#if PW_CHECK_VERSION(0, 3, 49)
- pw_deinit();
-#endif
- }
-}
-
-static EGLImage gsr_pipewire_create_egl_image(gsr_pipewire *self, const int *fds, const uint32_t *offsets, const uint32_t *pitches, const uint64_t *modifiers, bool use_modifiers) {
- intptr_t img_attr[44];
- setup_dma_buf_attrs(img_attr, spa_video_format_to_drm_format(self->format.info.raw.format), self->format.info.raw.size.width, self->format.info.raw.size.height,
- fds, offsets, pitches, modifiers, self->dmabuf_num_planes, use_modifiers);
- while(self->egl->eglGetError() != EGL_SUCCESS){}
- EGLImage image = self->egl->eglCreateImage(self->egl->egl_display, 0, EGL_LINUX_DMA_BUF_EXT, NULL, img_attr);
- if(!image || self->egl->eglGetError() != EGL_SUCCESS) {
- if(image)
- self->egl->eglDestroyImage(self->egl->egl_display, image);
- return NULL;
- }
- return image;
-}
-
-static EGLImage gsr_pipewire_create_egl_image_with_fallback(gsr_pipewire *self) {
- int fds[GSR_PIPEWIRE_DMABUF_MAX_PLANES];
- uint32_t offsets[GSR_PIPEWIRE_DMABUF_MAX_PLANES];
- uint32_t pitches[GSR_PIPEWIRE_DMABUF_MAX_PLANES];
- uint64_t modifiers[GSR_PIPEWIRE_DMABUF_MAX_PLANES];
- for(size_t i = 0; i < self->dmabuf_num_planes; ++i) {
- fds[i] = self->dmabuf_data[i].fd;
- offsets[i] = self->dmabuf_data[i].offset;
- pitches[i] = self->dmabuf_data[i].stride;
- modifiers[i] = self->format.info.raw.modifier;
- }
-
- EGLImage image = NULL;
- if(self->no_modifiers_fallback) {
- image = gsr_pipewire_create_egl_image(self, fds, offsets, pitches, modifiers, false);
- } else {
- image = gsr_pipewire_create_egl_image(self, fds, offsets, pitches, modifiers, true);
- if(!image) {
- fprintf(stderr, "gsr error: gsr_pipewire_create_egl_image_with_fallback: failed to create egl image with modifiers, trying without modifiers\n");
- self->no_modifiers_fallback = true;
- image = gsr_pipewire_create_egl_image(self, fds, offsets, pitches, modifiers, false);
- }
- }
- return image;
-}
-
-static bool gsr_pipewire_bind_image_to_texture(gsr_pipewire *self, EGLImage image, unsigned int texture_id, bool external_texture) {
- const int texture_target = external_texture ? GL_TEXTURE_EXTERNAL_OES : GL_TEXTURE_2D;
- while(self->egl->glGetError() != 0){}
- self->egl->glBindTexture(texture_target, texture_id);
- self->egl->glEGLImageTargetTexture2DOES(texture_target, image);
- const bool success = self->egl->glGetError() == 0;
- self->egl->glBindTexture(texture_target, 0);
- return success;
-}
-
-static void gsr_pipewire_bind_image_to_texture_with_fallback(gsr_pipewire *self, gsr_texture_map texture_map, EGLImage image) {
- if(self->external_texture_fallback) {
- gsr_pipewire_bind_image_to_texture(self, image, texture_map.external_texture_id, true);
- } else {
- if(!gsr_pipewire_bind_image_to_texture(self, image, texture_map.texture_id, false)) {
- fprintf(stderr, "gsr error: gsr_pipewire_map_texture: failed to bind image to texture, trying with external texture\n");
- self->external_texture_fallback = true;
- gsr_pipewire_bind_image_to_texture(self, image, texture_map.external_texture_id, true);
- }
- }
-}
-
-static void gsr_pipewire_update_cursor_texture(gsr_pipewire *self, gsr_texture_map texture_map) {
- if(!self->cursor.data)
- return;
-
- self->egl->glBindTexture(GL_TEXTURE_2D, texture_map.cursor_texture_id);
- // TODO: glTextureSubImage2D if same size
- self->egl->glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA8, self->cursor.width, self->cursor.height, 0, GL_RGBA, GL_UNSIGNED_BYTE, self->cursor.data);
- self->egl->glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
- self->egl->glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
- self->egl->glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
- self->egl->glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
- self->egl->glBindTexture(GL_TEXTURE_2D, 0);
-
- free(self->cursor.data);
- self->cursor.data = NULL;
-}
-
-bool gsr_pipewire_map_texture(gsr_pipewire *self, gsr_texture_map texture_map, gsr_pipewire_region *region, gsr_pipewire_region *cursor_region, gsr_pipewire_dmabuf_data *dmabuf_data, int *num_dmabuf_data, uint32_t *fourcc, uint64_t *modifiers, bool *using_external_image) {
- for(int i = 0; i < GSR_PIPEWIRE_DMABUF_MAX_PLANES; ++i) {
- memset(&dmabuf_data[i], 0, sizeof(gsr_pipewire_dmabuf_data));
- }
- *num_dmabuf_data = 0;
- *using_external_image = self->external_texture_fallback;
- *fourcc = 0;
- *modifiers = 0;
- pthread_mutex_lock(&self->mutex);
-
- if(!self->negotiated || self->dmabuf_data[0].fd <= 0) {
- pthread_mutex_unlock(&self->mutex);
- return false;
- }
-
- EGLImage image = gsr_pipewire_create_egl_image_with_fallback(self);
- if(image) {
- gsr_pipewire_bind_image_to_texture_with_fallback(self, texture_map, image);
- *using_external_image = self->external_texture_fallback;
- self->egl->eglDestroyImage(self->egl->egl_display, image);
- }
-
- gsr_pipewire_update_cursor_texture(self, texture_map);
-
- region->x = 0;
- region->y = 0;
-
- region->width = self->format.info.raw.size.width;
- region->height = self->format.info.raw.size.height;
-
- if(self->crop.valid) {
- region->x = self->crop.x;
- region->y = self->crop.y;
-
- region->width = self->crop.width;
- region->height = self->crop.height;
- }
-
- /* TODO: Test if cursor hotspot is correct */
- cursor_region->x = self->cursor.x - self->cursor.hotspot_x;
- cursor_region->y = self->cursor.y - self->cursor.hotspot_y;
-
- cursor_region->width = self->cursor.width;
- cursor_region->height = self->cursor.height;
-
- for(size_t i = 0; i < self->dmabuf_num_planes; ++i) {
- dmabuf_data[i] = self->dmabuf_data[i];
- self->dmabuf_data[i].fd = -1;
- }
- *num_dmabuf_data = self->dmabuf_num_planes;
- *fourcc = spa_video_format_to_drm_format(self->format.info.raw.format);
- *modifiers = self->format.info.raw.modifier;
- self->dmabuf_num_planes = 0;
-
- pthread_mutex_unlock(&self->mutex);
- return true;
-}
-
-bool gsr_pipewire_is_damaged(gsr_pipewire *self) {
- bool damaged = false;
- pthread_mutex_lock(&self->mutex);
- damaged = self->damaged;
- pthread_mutex_unlock(&self->mutex);
- return damaged;
-}
-
-void gsr_pipewire_clear_damage(gsr_pipewire *self) {
- pthread_mutex_lock(&self->mutex);
- self->damaged = false;
- pthread_mutex_unlock(&self->mutex);
-}