mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-25 09:40:37 +00:00
220 lines
8.9 KiB
C
220 lines
8.9 KiB
C
/*
|
|
* GStreamer
|
|
* Copyright (C) 2023 Igalia, S.L.
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Library General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Library General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Library General Public
|
|
* License along with this library; if not, write to the
|
|
* Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
|
|
* Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <gst/vulkan/vulkan.h>
|
|
|
|
#define GST_TYPE_VULKAN_ENCODER (gst_vulkan_encoder_get_type())
|
|
#define GST_VULKAN_ENCODER(o) (G_TYPE_CHECK_INSTANCE_CAST((o), GST_TYPE_VULKAN_ENCODER, GstVulkanEncoder))
|
|
#define GST_VULKAN_ENCODER_CLASS(k) (G_TYPE_CHECK_CLASS_CAST((k), GST_TYPE_VULKAN_ENCODER, GstVulkanEncoderClass))
|
|
#define GST_IS_VULKAN_ENCODER(o) (G_TYPE_CHECK_INSTANCE_TYPE((o), GST_TYPE_VULKAN_ENCODER))
|
|
#define GST_IS_VULKAN_ENCODER_CLASS(k) (G_TYPE_CHECK_CLASS_TYPE((k), GST_TYPE_VULKAN_ENCODER))
|
|
#define GST_VULKAN_ENCODER_GET_CLASS(o) (G_TYPE_INSTANCE_GET_CLASS((o), GST_TYPE_VULKAN_ENCODER, GstVulkanEncoderClass))
|
|
GST_VULKAN_API
|
|
GType gst_vulkan_encoder_get_type (void);
|
|
|
|
#define GST_TYPE_VULKAN_ENCODER_RATE_CONTROL_MODE (gst_vulkan_encoder_rate_control_mode_get_type())
|
|
GST_VULKAN_API
|
|
GType gst_vulkan_encoder_rate_control_mode_get_type (void);
|
|
|
|
typedef struct _GstVulkanEncoder GstVulkanEncoder;
|
|
typedef struct _GstVulkanEncoderClass GstVulkanEncoderClass;
|
|
typedef struct _GstVulkanEncoderQualityPoperties GstVulkanEncoderQualityProperties;
|
|
typedef union _GstVulkanEncoderParameters GstVulkanEncoderParameters;
|
|
typedef union _GstVulkanEncoderParametersOverrides GstVulkanEncoderParametersOverrides;
|
|
typedef union _GstVulkanEncoderParametersFeedback GstVulkanEncoderParametersFeedback;
|
|
typedef struct _GstVulkanEncoderPicture GstVulkanEncoderPicture;
|
|
typedef struct _GstVulkaneEncoderCallbacks GstVulkanEncoderCallbacks;
|
|
|
|
/**
|
|
* GstVulkaneEncoderCallbacks:
|
|
* @setup_codec_pic: Called after VkVideoEncodeInfoKHR and
|
|
* VkVideoReferenceSlotInfoKHR are filled so they can be chained with the
|
|
* specific codec structures. Called in gst_vulkan_encoder_encode().
|
|
* @setup_rc_pic: Called after VkVideoEncodeRateControlInfoKHR and
|
|
* VkVideoEncodeRateControlLayerInfoKHR are filled so they can be chained
|
|
* with the specific codec structures. Called in
|
|
* gst_vulkan_encoder_encode().
|
|
*
|
|
* See gst_vulkan_encoder_set_callbacks()
|
|
*/
|
|
struct _GstVulkaneEncoderCallbacks
|
|
{
|
|
void (*setup_codec_pic) (GstVulkanEncoderPicture * pic,
|
|
VkVideoEncodeInfoKHR * info, gpointer data);
|
|
void (*setup_rc_pic) (GstVulkanEncoderPicture * pic,
|
|
VkVideoEncodeRateControlInfoKHR * info,
|
|
VkVideoEncodeRateControlLayerInfoKHR * layer, gpointer data);
|
|
};
|
|
|
|
/**
|
|
* GstVulkanEncoderPicture:
|
|
* @slotIndex: slot index
|
|
* @offset: headers offset
|
|
* @in_buffer: input buffer
|
|
* @out_buffer: output buffer
|
|
*
|
|
* It contains the whole state for encoding a single picture.
|
|
*
|
|
* Since: 1.24
|
|
*/
|
|
struct _GstVulkanEncoderPicture
|
|
{
|
|
guint64 offset;
|
|
|
|
GstBuffer *in_buffer;
|
|
GstBuffer *dpb_buffer;
|
|
GstBuffer *out_buffer;
|
|
|
|
/* Input frame */
|
|
GstVulkanImageView *img_view;
|
|
GstVulkanImageView *dpb_view;
|
|
|
|
VkVideoPictureResourceInfoKHR dpb;
|
|
VkVideoReferenceSlotInfoKHR dpb_slot;
|
|
|
|
gpointer codec_rc_info;
|
|
};
|
|
|
|
/**
|
|
* GstVulkanEncoder:
|
|
* @parent: the parent #GstObject
|
|
* @queue: the #GstVulkanQueue to command buffers will be allocated from
|
|
*
|
|
* Since: 1.24
|
|
**/
|
|
struct _GstVulkanEncoder
|
|
{
|
|
GstObject parent;
|
|
|
|
GstVulkanQueue *queue;
|
|
|
|
guint codec;
|
|
|
|
/* <private> */
|
|
gpointer _reserved [GST_PADDING];
|
|
};
|
|
|
|
/**
|
|
* GstVulkanEncoderClass:
|
|
* @parent_class: the parent #GstObjectClass
|
|
*
|
|
* Since: 1.24
|
|
*/
|
|
struct _GstVulkanEncoderClass
|
|
{
|
|
GstObjectClass parent;
|
|
/* <private> */
|
|
gpointer _reserved [GST_PADDING];
|
|
};
|
|
|
|
union _GstVulkanEncoderParameters
|
|
{
|
|
/*< private >*/
|
|
VkVideoEncodeH264SessionParametersCreateInfoKHR h264;
|
|
VkVideoEncodeH265SessionParametersCreateInfoKHR h265;
|
|
};
|
|
|
|
union _GstVulkanEncoderParametersOverrides
|
|
{
|
|
/*< private >*/
|
|
VkVideoEncodeH264SessionParametersGetInfoKHR h264;
|
|
VkVideoEncodeH265SessionParametersGetInfoKHR h265;
|
|
};
|
|
|
|
union _GstVulkanEncoderParametersFeedback
|
|
{
|
|
VkVideoEncodeH264SessionParametersFeedbackInfoKHR h264;
|
|
VkVideoEncodeH265SessionParametersFeedbackInfoKHR h265;
|
|
};
|
|
|
|
struct _GstVulkanEncoderQualityPoperties
|
|
{
|
|
gint32 quality_level;
|
|
union
|
|
{
|
|
VkVideoEncodeH264QualityLevelPropertiesKHR h264;
|
|
VkVideoEncodeH265QualityLevelPropertiesKHR h265;
|
|
} codec;
|
|
};
|
|
|
|
G_DEFINE_AUTOPTR_CLEANUP_FUNC (GstVulkanEncoder, gst_object_unref)
|
|
|
|
GST_VULKAN_API
|
|
GstVulkanEncoder * gst_vulkan_encoder_create_from_queue (GstVulkanQueue * queue,
|
|
guint codec);
|
|
|
|
GST_VULKAN_API
|
|
void gst_vulkan_encoder_set_callbacks (GstVulkanEncoder * self,
|
|
GstVulkanEncoderCallbacks * callbacks,
|
|
gpointer user_data,
|
|
GDestroyNotify notify);
|
|
GST_VULKAN_API
|
|
void gst_vulkan_encoder_set_rc_mode (GstVulkanEncoder * self,
|
|
VkVideoEncodeRateControlModeFlagBitsKHR rc_mode);
|
|
|
|
GST_VULKAN_API
|
|
gboolean gst_vulkan_encoder_start (GstVulkanEncoder * self,
|
|
GstVulkanVideoProfile * profile,
|
|
GstVulkanEncoderQualityProperties * codec_quality_props,
|
|
GError ** error);
|
|
GST_VULKAN_API
|
|
gboolean gst_vulkan_encoder_stop (GstVulkanEncoder * self);
|
|
GST_VULKAN_API
|
|
gboolean gst_vulkan_encoder_update_video_session_parameters
|
|
(GstVulkanEncoder * self,
|
|
GstVulkanEncoderParameters *enc_params,
|
|
GError ** error);
|
|
GST_VULKAN_API
|
|
gboolean gst_vulkan_encoder_video_session_parameters_overrides
|
|
(GstVulkanEncoder * self,
|
|
GstVulkanEncoderParametersOverrides * params,
|
|
GstVulkanEncoderParametersFeedback * feedback,
|
|
gsize * data_size,
|
|
gpointer * data,
|
|
GError ** error);
|
|
GST_VULKAN_API
|
|
gboolean gst_vulkan_encoder_create_dpb_pool (GstVulkanEncoder * self,
|
|
GstCaps * caps);
|
|
GST_VULKAN_API
|
|
gboolean gst_vulkan_encoder_encode (GstVulkanEncoder * self,
|
|
GstVideoInfo * info,
|
|
GstVulkanEncoderPicture * pic,
|
|
guint nb_refs,
|
|
GstVulkanEncoderPicture ** ref_pics);
|
|
GST_VULKAN_API
|
|
gboolean gst_vulkan_encoder_caps (GstVulkanEncoder * self,
|
|
GstVulkanVideoCapabilities * caps);
|
|
GST_VULKAN_API
|
|
gboolean gst_vulkan_encoder_is_started (GstVulkanEncoder * self);
|
|
GST_VULKAN_API
|
|
GstCaps * gst_vulkan_encoder_profile_caps (GstVulkanEncoder * self);
|
|
GST_VULKAN_API
|
|
gint32 gst_vulkan_encoder_quality_level (GstVulkanEncoder * self);
|
|
|
|
GST_VULKAN_API
|
|
gboolean gst_vulkan_encoder_picture_init (GstVulkanEncoderPicture * pic,
|
|
GstVulkanEncoder * self,
|
|
GstBuffer * in_buffer,
|
|
gsize size);
|
|
GST_VULKAN_API
|
|
void gst_vulkan_encoder_picture_clear (GstVulkanEncoderPicture * pic,
|
|
GstVulkanEncoder * self);
|