diff options
Diffstat (limited to 'src/utils.c')
-rw-r--r-- | src/utils.c | 237 |
1 files changed, 0 insertions, 237 deletions
diff --git a/src/utils.c b/src/utils.c index 325f750..943fb2d 100644 --- a/src/utils.c +++ b/src/utils.c @@ -14,10 +14,8 @@ #include <xf86drmMode.h> #include <xf86drm.h> -#include <libdrm/drm_fourcc.h> #include <X11/Xatom.h> #include <X11/extensions/Xrandr.h> -#include <va/va_drmcommon.h> #include <libavcodec/avcodec.h> #include <libavutil/hwcontext_vaapi.h> @@ -663,241 +661,6 @@ bool video_codec_context_is_vaapi(AVCodecContext *video_codec_context) { return device_context->type == AV_HWDEVICE_TYPE_VAAPI; } -static uint32_t drm_fourcc_to_va_fourcc(uint32_t drm_fourcc) { - switch(drm_fourcc) { - case DRM_FORMAT_XRGB8888: return VA_FOURCC_BGRX; - case DRM_FORMAT_XBGR8888: return VA_FOURCC_RGBX; - case DRM_FORMAT_RGBX8888: return VA_FOURCC_XBGR; - case DRM_FORMAT_BGRX8888: return VA_FOURCC_XRGB; - case DRM_FORMAT_ARGB8888: return VA_FOURCC_BGRA; - case DRM_FORMAT_ABGR8888: return VA_FOURCC_RGBA; - case DRM_FORMAT_RGBA8888: return VA_FOURCC_ABGR; - case DRM_FORMAT_BGRA8888: return VA_FOURCC_ARGB; - default: return drm_fourcc; - } -} - -bool vaapi_copy_drm_planes_to_video_surface(AVCodecContext *video_codec_context, AVFrame *video_frame, vec2i source_pos, vec2i source_size, vec2i dest_pos, vec2i dest_size, uint32_t format, vec2i size, const int *fds, const uint32_t *offsets, const uint32_t *pitches, const uint64_t *modifiers, int num_planes) { - VAConfigID config_id = 0; - VAContextID context_id = 0; - VASurfaceID input_surface_id = 0; - VABufferID buffer_id = 0; - bool success = true; - - VADisplay va_dpy = video_codec_context_get_vaapi_display(video_codec_context); - if(!va_dpy) { - success = false; - goto done; - } - - VAStatus va_status = vaCreateConfig(va_dpy, VAProfileNone, VAEntrypointVideoProc, NULL, 0, &config_id); - if(va_status != VA_STATUS_SUCCESS) { - fprintf(stderr, "gsr error: vaapi_copy_drm_planes_to_video_surface: vaCreateConfig failed, error: %s\n", vaErrorStr(va_status)); - success = false; - goto done; - } - - VASurfaceID output_surface_id = (uintptr_t)video_frame->data[3]; - va_status = vaCreateContext(va_dpy, config_id, size.x, size.y, VA_PROGRESSIVE, &output_surface_id, 1, &context_id); - if(va_status != VA_STATUS_SUCCESS) { - fprintf(stderr, "gsr error: vaapi_copy_drm_planes_to_video_surface: vaCreateContext failed, error: %s\n", vaErrorStr(va_status)); - success = false; - goto done; - } - - VADRMPRIMESurfaceDescriptor buf = {0}; - buf.fourcc = drm_fourcc_to_va_fourcc(format);//VA_FOURCC_BGRX; // TODO: VA_FOURCC_BGRA, VA_FOURCC_X2R10G10B10 - buf.width = size.x; - buf.height = size.y; - buf.num_objects = num_planes; - buf.num_layers = 1; - buf.layers[0].drm_format = format; - buf.layers[0].num_planes = buf.num_objects; - for(int i = 0; i < num_planes; ++i) { - buf.objects[i].fd = fds[i]; - buf.objects[i].size = size.y * pitches[i]; // TODO: - buf.objects[i].drm_format_modifier = modifiers[i]; - - buf.layers[0].object_index[i] = i; - buf.layers[0].offset[i] = offsets[i]; - buf.layers[0].pitch[i] = pitches[i]; - } - - VASurfaceAttrib attribs[2] = {0}; - attribs[0].type = VASurfaceAttribMemoryType; - attribs[0].flags = VA_SURFACE_ATTRIB_SETTABLE; - attribs[0].value.type = VAGenericValueTypeInteger; - attribs[0].value.value.i = VA_SURFACE_ATTRIB_MEM_TYPE_DRM_PRIME_2; - attribs[1].type = VASurfaceAttribExternalBufferDescriptor; - attribs[1].flags = VA_SURFACE_ATTRIB_SETTABLE; - attribs[1].value.type = VAGenericValueTypePointer; - attribs[1].value.value.p = &buf; - - // TODO: RT_FORMAT with 10 bit/hdr, VA_RT_FORMAT_RGB32_10 - // TODO: Max size same as source_size - va_status = vaCreateSurfaces(va_dpy, VA_RT_FORMAT_RGB32, size.x, size.y, &input_surface_id, 1, attribs, 2); - if(va_status != VA_STATUS_SUCCESS) { - fprintf(stderr, "gsr error: vaapi_copy_drm_planes_to_video_surface: vaCreateSurfaces failed, error: %s\n", vaErrorStr(va_status)); - success = false; - goto done; - } - - const VARectangle source_region = { - .x = source_pos.x, - .y = source_pos.y, - .width = source_size.x, - .height = source_size.y - }; - - const VARectangle output_region = { - .x = dest_pos.x, - .y = dest_pos.y, - .width = dest_size.x, - .height = dest_size.y - }; - - const bool scaled = dest_size.x != source_size.x || dest_size.y != source_size.y; - - // Copying a surface to another surface will automatically perform the color conversion. Thanks vaapi! - VAProcPipelineParameterBuffer params = {0}; - params.surface = input_surface_id; - params.surface_region = NULL; - params.surface_region = &source_region; - params.output_region = &output_region; - params.output_background_color = 0; - params.filter_flags = scaled ? (VA_FILTER_SCALING_HQ | VA_FILTER_INTERPOLATION_BILINEAR) : 0; - params.pipeline_flags = VA_PROC_PIPELINE_FAST; - - params.input_color_properties.colour_primaries = 1; - params.input_color_properties.transfer_characteristics = 1; - params.input_color_properties.matrix_coefficients = 1; - params.surface_color_standard = VAProcColorStandardBT709; // TODO: - params.input_color_properties.color_range = video_frame->color_range == AVCOL_RANGE_JPEG ? VA_SOURCE_RANGE_FULL : VA_SOURCE_RANGE_REDUCED; - - params.output_color_properties.colour_primaries = 1; - params.output_color_properties.transfer_characteristics = 1; - params.output_color_properties.matrix_coefficients = 1; - params.output_color_standard = VAProcColorStandardBT709; // TODO: - params.output_color_properties.color_range = video_frame->color_range == AVCOL_RANGE_JPEG ? VA_SOURCE_RANGE_FULL : VA_SOURCE_RANGE_REDUCED; - - params.processing_mode = VAProcPerformanceMode; - - // VAProcPipelineCaps pipeline_caps = {0}; - // va_status = vaQueryVideoProcPipelineCaps(self->va_dpy, - // self->context_id, - // NULL, 0, - // &pipeline_caps); - // if(va_status == VA_STATUS_SUCCESS) { - // fprintf(stderr, "pipeline_caps: %u, %u\n", (unsigned int)pipeline_caps.rotation_flags, pipeline_caps.blend_flags); - // } - - // TODO: params.output_hdr_metadata - - // TODO: - // if (first surface to render) - // pipeline_param->output_background_color = 0xff000000; // black - - va_status = vaCreateBuffer(va_dpy, context_id, VAProcPipelineParameterBufferType, sizeof(params), 1, ¶ms, &buffer_id); - if(va_status != VA_STATUS_SUCCESS) { - fprintf(stderr, "gsr error: vaapi_copy_drm_planes_to_video_surface: vaCreateBuffer failed, error: %d\n", va_status); - success = false; - goto done; - } - - va_status = vaBeginPicture(va_dpy, context_id, output_surface_id); - if(va_status != VA_STATUS_SUCCESS) { - fprintf(stderr, "gsr error: vaapi_copy_drm_planes_to_video_surface: vaBeginPicture failed, error: %d\n", va_status); - success = false; - goto done; - } - - va_status = vaRenderPicture(va_dpy, context_id, &buffer_id, 1); - if(va_status != VA_STATUS_SUCCESS) { - vaEndPicture(va_dpy, context_id); - fprintf(stderr, "gsr error: vaapi_copy_drm_planes_to_video_surface: vaRenderPicture failed, error: %d\n", va_status); - success = false; - goto done; - } - - va_status = vaEndPicture(va_dpy, context_id); - if(va_status != VA_STATUS_SUCCESS) { - fprintf(stderr, "gsr error: vaapi_copy_drm_planes_to_video_surface: vaEndPicture failed, error: %d\n", va_status); - success = false; - goto done; - } - - // vaSyncBuffer(va_dpy, buffer_id, 1000 * 1000 * 1000); - // vaSyncSurface(va_dpy, input_surface_id); - // vaSyncSurface(va_dpy, output_surface_id); - - done: - if(buffer_id) - vaDestroyBuffer(va_dpy, buffer_id); - - if(input_surface_id) - vaDestroySurfaces(va_dpy, &input_surface_id, 1); - - if(context_id) - vaDestroyContext(va_dpy, context_id); - - if(config_id) - vaDestroyConfig(va_dpy, config_id); - - return success; -} - -bool vaapi_copy_egl_image_to_video_surface(gsr_egl *egl, EGLImage image, vec2i source_pos, vec2i source_size, vec2i dest_pos, vec2i dest_size, AVCodecContext *video_codec_context, AVFrame *video_frame) { - if(!image) - return false; - - int texture_fourcc = 0; - int texture_num_planes = 0; - uint64_t texture_modifiers = 0; - if(!egl->eglExportDMABUFImageQueryMESA(egl->egl_display, image, &texture_fourcc, &texture_num_planes, &texture_modifiers)) { - fprintf(stderr, "gsr error: gsr_capture_xcomposite_vaapi_tick: eglExportDMABUFImageQueryMESA failed\n"); - return false; - } - - if(texture_num_planes <= 0 || texture_num_planes > 8) { - fprintf(stderr, "gsr error: gsr_capture_xcomposite_vaapi_tick: expected planes size to be 0<planes<=8 for drm buf, got %d planes\n", texture_num_planes); - return false; - } - - int texture_fds[8]; - int32_t texture_strides[8]; - int32_t texture_offsets[8]; - - while(egl->eglGetError() != EGL_SUCCESS){} - if(!egl->eglExportDMABUFImageMESA(egl->egl_display, image, texture_fds, texture_strides, texture_offsets)) { - fprintf(stderr, "gsr error: gsr_capture_xcomposite_vaapi_tick: eglExportDMABUFImageMESA failed, error: %d\n", egl->eglGetError()); - return false; - } - - int fds[8]; - uint32_t offsets[8]; - uint32_t pitches[8]; - uint64_t modifiers[8]; - for(int i = 0; i < texture_num_planes; ++i) { - fds[i] = texture_fds[i]; - offsets[i] = texture_offsets[i]; - pitches[i] = texture_strides[i]; - modifiers[i] = texture_modifiers; - - if(fds[i] == -1) - texture_num_planes = i; - } - const bool success = texture_num_planes > 0 && vaapi_copy_drm_planes_to_video_surface(video_codec_context, video_frame, source_pos, source_size, dest_pos, dest_size, texture_fourcc, source_size, fds, offsets, pitches, modifiers, texture_num_planes); - - for(int i = 0; i < texture_num_planes; ++i) { - if(texture_fds[i] > 0) { - close(texture_fds[i]); - texture_fds[i] = -1; - } - } - - return success; -} - vec2i scale_keep_aspect_ratio(vec2i from, vec2i to) { if(from.x == 0 || from.y == 0) return (vec2i){0, 0}; |