gstreamer/subprojects/gst-plugins-bad/sys/va/gstvabaseenc.h
Víctor Manuel Jáquez Leal 9f5b2c4e25 va: use GstVideoInfoDmaDrm when importing buffers
In the case of encoders and filters when importing a DMABuf, use
GstVideoInfoDmaDrm to get the drm fourcc and modifier.

In both cases, instead of keeping the original GstVideoInfoDmaDrm from caps, the
GstVideoInfo part of the structure is converted as canonical one, given the
format from the fourcc. It's kept in the way to handle V4L2 linear DMABufs and
to avoid too many changes in the current code.

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/5264>
2023-11-09 17:36:06 +00:00

152 lines
6.2 KiB
C

/* GStreamer
* Copyright (C) 2022 Intel Corporation
* Author: He Junyan <junyan.he@intel.com>
*
* 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 the0
* Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
#pragma once
#include "gstvadevice.h"
#include "gstvaencoder.h"
#include "gstvaprofile.h"
G_BEGIN_DECLS
#define GST_TYPE_VA_BASE_ENC (gst_va_base_enc_get_type())
#define GST_VA_BASE_ENC(obj) (G_TYPE_CHECK_INSTANCE_CAST((obj), GST_TYPE_VA_BASE_ENC, GstVaBaseEnc))
#define GST_IS_VA_BASE_ENC(obj) (G_TYPE_CHECK_INSTANCE_TYPE((obj), GST_TYPE_VA_BASE_ENC))
#define GST_VA_BASE_ENC_CLASS(klass) (G_TYPE_CHECK_CLASS_CAST((klass), GST_TYPE_VA_BASE_ENC, GstVaBaseEncClass))
#define GST_IS_VA_BASE_ENC_CLASS(klass) (G_TYPE_CHECK_CLASS_TYPE((klass), GST_TYPE_VA_BASE_ENC))
#define GST_VA_BASE_ENC_GET_CLASS(obj) (G_TYPE_INSTANCE_GET_CLASS((obj), GST_TYPE_VA_BASE_ENC, GstVaBaseEncClass))
#define GST_VA_BASE_ENC_ENTRYPOINT(obj) (GST_VA_BASE_ENC_GET_CLASS(obj)->entrypoint)
typedef struct _GstVaBaseEnc GstVaBaseEnc;
typedef struct _GstVaBaseEncClass GstVaBaseEncClass;
typedef struct _GstVaBaseEncPrivate GstVaBaseEncPrivate;
struct _GstVaBaseEnc
{
GstVideoEncoder parent_instance;
GstVaDisplay *display;
GstVaEncoder *encoder;
gboolean reconf;
VAProfile profile;
gint width;
gint height;
guint rt_format;
guint codedbuf_size;
GstClockTime start_pts;
GstClockTime frame_duration;
/* Total frames we handled since reconfig. */
guint input_frame_count;
guint output_frame_count;
GQueue reorder_list;
GQueue ref_list;
GQueue output_list;
GstVideoCodecState *input_state;
union {
GstVideoInfo in_info;
GstVideoInfoDmaDrm in_drm_info;
};
/*< private >*/
GstVaBaseEncPrivate *priv;
gpointer _padding[GST_PADDING];
};
struct _GstVaBaseEncClass
{
GstVideoEncoderClass parent_class;
void (*reset_state) (GstVaBaseEnc * encoder);
gboolean (*reconfig) (GstVaBaseEnc * encoder);
gboolean (*new_frame) (GstVaBaseEnc * encoder,
GstVideoCodecFrame * frame);
gboolean (*reorder_frame) (GstVaBaseEnc * base,
GstVideoCodecFrame * frame,
gboolean bump_all,
GstVideoCodecFrame ** out_frame);
GstFlowReturn (*encode_frame) (GstVaBaseEnc * encoder,
GstVideoCodecFrame * frame,
gboolean is_last);
void (*prepare_output) (GstVaBaseEnc * encoder,
GstVideoCodecFrame * frame);
GstVaCodecs codec;
VAEntrypoint entrypoint;
gchar *render_device_path;
gpointer _padding[GST_PADDING];
};
struct CData
{
VAEntrypoint entrypoint;
gchar *render_device_path;
gchar *description;
GstCaps *sink_caps;
GstCaps *src_caps;
};
GType gst_va_base_enc_get_type (void);
gboolean gst_va_base_enc_add_rate_control_parameter (GstVaBaseEnc * base,
GstVaEncodePicture * picture,
guint32 rc_mode,
guint max_bitrate_bits,
guint target_percentage,
guint32 qp_i,
guint32 min_qp,
guint32 max_qp,
guint32 mbbrc);
gboolean gst_va_base_enc_add_quality_level_parameter (GstVaBaseEnc * base,
GstVaEncodePicture * picture,
guint target_usage);
gboolean gst_va_base_enc_add_frame_rate_parameter (GstVaBaseEnc * base,
GstVaEncodePicture * picture);
gboolean gst_va_base_enc_add_hrd_parameter (GstVaBaseEnc * base,
GstVaEncodePicture * picture,
guint32 rc_mode,
guint cpb_length_bits);
gboolean gst_va_base_enc_add_trellis_parameter (GstVaBaseEnc * base,
GstVaEncodePicture * picture,
gboolean use_trellis);
void gst_va_base_enc_add_codec_tag (GstVaBaseEnc * base,
const gchar * codec_name);
void gst_va_base_enc_reset_state (GstVaBaseEnc * base);
void gst_va_base_enc_update_property_uint (GstVaBaseEnc * base,
guint32 * old_val,
guint32 new_val,
GParamSpec * pspec);
void gst_va_base_enc_update_property_bool (GstVaBaseEnc * base,
gboolean * old_val,
gboolean new_val,
GParamSpec * pspec);
G_DEFINE_AUTOPTR_CLEANUP_FUNC(GstVaBaseEnc, gst_object_unref)
G_END_DECLS