gstreamer-encoder: Include dmabuf encoding conditionally for Linux

add dmabuf encoding if `drm/drm_fourcc.h` is present and
gstreamer is at least 1.24 du to
`gst_video_dma_drm_fourcc_to_format()`.

Signed-off-by: Michael Scherle <michael.scherle@rz.uni-freiburg.de>
This commit is contained in:
Michael Scherle 2024-11-26 11:24:04 +01:00
parent 31643ca0ea
commit 1af9ad8035
2 changed files with 36 additions and 8 deletions

View File

@ -131,15 +131,23 @@ endforeach
spice_server_has_gstreamer = false
spice_server_gst_version = get_option('gstreamer')
if spice_server_gst_version != 'no'
gstreamer_version = ''
gst_deps = ['gstreamer', 'gstreamer-base', 'gstreamer-app', 'gstreamer-video', 'gstreamer-allocators']
foreach dep : gst_deps
dep = '@0@-@1@'.format(dep, spice_server_gst_version)
spice_server_deps += dependency(dep)
dep = dependency(dep)
if gstreamer_version == '' or dep.version().version_compare('< ' + gstreamer_version)
gstreamer_version = dep.version()
endif
spice_server_deps += dep
endforeach
spice_server_deps += dependency('orc-0.4')
gst_def = 'HAVE_GSTREAMER_1_0'
spice_server_config_data.set(gst_def, '1')
if compiler.has_header('drm/drm_fourcc.h') and gstreamer_version.version_compare('>= 1.24')
spice_server_config_data.set('HAVE_GSTREAMER_DMABUF_ENCODING', '1')
endif
spice_server_has_gstreamer = true
endif

View File

@ -27,11 +27,13 @@
# pragma GCC diagnostic ignored "-Wunused-const-variable"
#endif
#include <gst/gst.h>
#include <gst/allocators/gstdmabuf.h>
#include <gst/app/gstappsrc.h>
#include <gst/app/gstappsink.h>
#include <gst/video/video.h>
#ifdef HAVE_GSTREAMER_DMABUF_ENCODING
#include <gst/video/video-info-dma.h>
#include <gst/allocators/gstdmabuf.h>
#endif
#include <orc/orcprogram.h>
#if defined(__GNUC__) && (__GNUC__ >= 6)
# pragma GCC diagnostic pop
@ -41,7 +43,12 @@
#include "video-encoder.h"
#include "utils.h"
#include "common/udev.h"
#ifdef HAVE_GSTREAMER_DMABUF_ENCODING
#include "drm/drm_fourcc.h"
#endif
#ifndef DRM_FORMAT_INVALID
#define DRM_FORMAT_INVALID 0
#endif
#define SPICE_GST_DEFAULT_FPS 30
@ -793,6 +800,7 @@ static const SpiceFormatForGStreamer *map_format(SpiceBitmapFmt format)
return GSTREAMER_FORMAT_INVALID;
}
#ifdef HAVE_GSTREAMER_DMABUF_ENCODING
static SpiceFormatForGStreamer drm_format_map[] = {
DRM_FMT_DESC(DRM_FORMAT_INVALID, 0),
DRM_FMT_DESC(DRM_FORMAT_XRGB8888, 32),
@ -822,7 +830,7 @@ static const SpiceFormatForGStreamer *map_drm_format(uint32_t fourcc)
}
drm_format_map[i].gst_format = gst_format;
strncpy(drm_format_map[i].format,
g_strlcpy(drm_format_map[i].format,
gst_video_format_to_string(gst_format),
format_size - 1);
drm_format_map[i].format[format_size - 1] = '\0';
@ -833,6 +841,7 @@ static const SpiceFormatForGStreamer *map_drm_format(uint32_t fourcc)
return GSTREAMER_DRM_FORMAT_INVALID;
}
#endif
static void set_appsrc_caps(SpiceGstEncoder *encoder)
{
@ -1579,7 +1588,9 @@ static void spice_gst_encoder_destroy(VideoEncoder *video_encoder)
{
SpiceGstEncoder *encoder = (SpiceGstEncoder*)video_encoder;
#ifdef HAVE_GSTREAMER_DMABUF_ENCODING
gst_object_unref(encoder->allocator);
#endif
free_pipeline(encoder);
pthread_mutex_destroy(&encoder->outbuf_mutex);
pthread_cond_destroy(&encoder->outbuf_cond);
@ -1619,12 +1630,17 @@ spice_gst_encoder_configure_pipeline(SpiceGstEncoder *encoder,
if (spice_format != SPICE_BITMAP_FMT_INVALID) {
format = map_format(spice_format);
} else if (drm_format != DRM_FORMAT_INVALID) {
}
#ifdef HAVE_GSTREAMER_DMABUF_ENCODING
else if (drm_format != DRM_FORMAT_INVALID) {
format = map_drm_format(drm_format);
}
if (format == GSTREAMER_FORMAT_INVALID ||
format == GSTREAMER_DRM_FORMAT_INVALID) {
#endif
if (format == GSTREAMER_FORMAT_INVALID
#ifdef HAVE_GSTREAMER_DMABUF_ENCODING
|| format == GSTREAMER_DRM_FORMAT_INVALID
#endif
) {
spice_warning("unable to map format type %d or %u",
spice_format, drm_format);
encoder->errors = 4;
@ -1738,6 +1754,7 @@ spice_gst_encoder_encode_frame(VideoEncoder *video_encoder,
return rc;
}
#ifdef HAVE_GSTREAMER_DMABUF_ENCODING
static void
spice_gst_mem_free_cb(VideoEncoderDmabufData *dmabuf_data, GstMiniObject *obj)
{
@ -1806,6 +1823,7 @@ spice_gst_encoder_encode_dmabuf(VideoEncoder *video_encoder,
spice_gst_encoder_add_frame(encoder, outbuf, start, frame_mm_time);
return rc;
}
#endif
static void spice_gst_encoder_client_stream_report(VideoEncoder *video_encoder,
uint32_t num_frames,
@ -1997,7 +2015,10 @@ VideoEncoder *gstreamer_encoder_new(SpiceVideoCodecType codec_type,
SpiceGstEncoder *encoder = g_new0(SpiceGstEncoder, 1);
encoder->base.destroy = spice_gst_encoder_destroy;
encoder->base.encode_frame = spice_gst_encoder_encode_frame;
#ifdef HAVE_GSTREAMER_DMABUF_ENCODING
encoder->base.encode_dmabuf = spice_gst_encoder_encode_dmabuf;
encoder->allocator = gst_dmabuf_allocator_new();
#endif
encoder->base.client_stream_report = spice_gst_encoder_client_stream_report;
encoder->base.notify_server_frame_drop = spice_gst_encoder_notify_server_frame_drop;
encoder->base.get_bit_rate = spice_gst_encoder_get_bit_rate;
@ -2010,7 +2031,6 @@ VideoEncoder *gstreamer_encoder_new(SpiceVideoCodecType codec_type,
encoder->bitmap_ref = bitmap_ref;
encoder->bitmap_unref = bitmap_unref;
encoder->format = GSTREAMER_FORMAT_INVALID;
encoder->allocator = gst_dmabuf_allocator_new();
pthread_mutex_init(&encoder->outbuf_mutex, NULL);
pthread_cond_init(&encoder->outbuf_cond, NULL);