gstreamer/subprojects/gstreamer-vaapi/gst/vaapi/gstvaapiencode.h

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

218 lines
11 KiB
C
Raw Normal View History

/*
* gstvaapiencode.h - VA-API video encoder
*
2014-01-22 17:54:14 +00:00
* Copyright (C) 2013-2014 Intel Corporation
* Author: Wind Yuan <feng.yuan@intel.com>
* Author: Gwenole Beauchesne <gwenole.beauchesne@intel.com>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public License
* as published by the Free Software Foundation; either version 2.1
* 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
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free
* Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
* Boston, MA 02110-1301 USA
*/
#ifndef GST_VAAPIENCODE_H
#define GST_VAAPIENCODE_H
#include "gstvaapipluginbase.h"
#include <gst/vaapi/gstvaapiencoder.h>
G_BEGIN_DECLS
#define GST_TYPE_VAAPIENCODE \
(gst_vaapiencode_get_type ())
#define GST_VAAPIENCODE_CAST(obj) \
((GstVaapiEncode *)(obj))
#define GST_VAAPIENCODE(obj) \
(G_TYPE_CHECK_INSTANCE_CAST ((obj), GST_TYPE_VAAPIENCODE, GstVaapiEncode))
#define GST_VAAPIENCODE_CLASS(klass) \
(G_TYPE_CHECK_CLASS_CAST ((klass), GST_TYPE_VAAPIENCODE, GstVaapiEncodeClass))
#define GST_VAAPIENCODE_GET_CLASS(obj) \
(G_TYPE_INSTANCE_GET_CLASS ((obj), GST_TYPE_VAAPIENCODE, GstVaapiEncodeClass))
#define GST_IS_VAAPIENCODE(obj) \
(G_TYPE_CHECK_INSTANCE_TYPE ((obj), GST_TYPE_VAAPIENCODE))
#define GST_IS_VAAPIENCODE_CLASS(klass) \
(G_TYPE_CHECK_CLASS_TYPE ((klass), GST_TYPE_VAAPIENCODE))
typedef struct _GstVaapiEncodeInitData GstVaapiEncodeInitData;
struct _GstVaapiEncodeInitData
{
GstCaps *sink_caps;
GstCaps *src_caps;
};
/* *INDENT-OFF* */
#define GST_VAAPI_ENCODE_STATIC_SINK_CAPS \
GST_VAAPI_MAKE_SURFACE_CAPS ", " \
GST_CAPS_INTERLACED_FALSE "; " \
GST_VIDEO_CAPS_MAKE (GST_VAAPI_FORMATS_ALL) ", " \
GST_CAPS_INTERLACED_FALSE ";" \
GST_VIDEO_CAPS_MAKE_WITH_FEATURES( \
GST_CAPS_FEATURE_MEMORY_DMABUF, \
GST_VAAPI_FORMATS_ALL) "," \
GST_CAPS_INTERLACED_FALSE
/* *INDENT-ON* */
#define GST_VAAPI_ENCODE_REGISTER_TYPE(NAME, CODEC, CLASS, _EXT_FMT_, FUN) \
static GstVaapiEncodeInitData encode_init_data = { NULL, NULL }; \
static GType encode_type = G_TYPE_INVALID; \
static gpointer gst_vaapiencode_##NAME##_parent_class = NULL; \
static void \
gst_vaapiencode_##NAME##_class_init ( \
GstVaapiEncode##CLASS##Class * klass, gpointer data); \
static void gst_vaapiencode_##NAME##_class_intern_init (gpointer klass, \
gpointer data) \
{ \
gst_vaapiencode_##NAME##_parent_class = \
g_type_class_peek_parent (klass); \
gst_vaapiencode_##NAME##_class_init (klass, data); \
} \
static void \
gst_vaapiencode_##NAME##_init (GstVaapiEncode##CLASS * encode); \
GType \
gst_vaapiencode_##NAME##_register_type (GstVaapiDisplay * display) \
{ \
GstCaps *caps; \
guint i, n; \
GTypeInfo type_info = { \
sizeof (GstVaapiEncodeClass), \
NULL, \
NULL, \
(GClassInitFunc) gst_vaapiencode_##NAME##_class_intern_init, \
NULL, \
NULL, \
sizeof (GstVaapiEncode##CLASS), \
0, \
(GInstanceInitFunc) gst_vaapiencode_##NAME##_init, \
}; \
GArray *extra_fmts = NULL; \
GstVideoFormat ext_video_fmts[] = _EXT_FMT_; \
\
GST_DEBUG_CATEGORY_INIT (gst_vaapi_##NAME##_encode_debug, \
GST_PLUGIN_NAME, 0, GST_PLUGIN_DESC); \
\
if ((n = G_N_ELEMENTS (ext_video_fmts))) { \
extra_fmts = \
g_array_sized_new (FALSE, FALSE, sizeof (GstVideoFormat), n); \
for (i = 0; i < n; i++) \
g_array_append_val (extra_fmts, ext_video_fmts[i]); \
} \
caps = gst_vaapi_build_template_raw_caps_by_codec (display, \
GST_VAAPI_CONTEXT_USAGE_ENCODE, \
GST_VAAPI_CODEC_##CODEC, extra_fmts); \
g_clear_pointer (&extra_fmts, g_array_unref); \
if (!caps) { \
GST_ERROR ("failed to get sink caps for " #CODEC \
" encode, can not register"); \
return G_TYPE_INVALID; \
} \
\
for (i = 0; i < gst_caps_get_size (caps); i++) { \
GstStructure *structure = gst_caps_get_structure (caps, i); \
if (!structure) \
continue; \
gst_structure_set (structure, "interlace-mode", G_TYPE_STRING, \
"progressive", NULL); \
} \
GST_DEBUG (#CODEC" encode's sink caps %" GST_PTR_FORMAT, caps); \
\
/* class data will be leaked if the element never gets instantiated */ \
GST_MINI_OBJECT_FLAG_SET (caps, GST_MINI_OBJECT_FLAG_MAY_BE_LEAKED); \
encode_init_data.sink_caps = caps; \
\
caps = gst_vaapi_build_template_coded_caps_by_codec (display, \
GST_VAAPI_CONTEXT_USAGE_ENCODE, \
GST_VAAPI_CODEC_##CODEC, GST_CODEC_CAPS, \
FUN); \
if (!caps) { \
GST_ERROR ("failed to get src caps for " #CODEC \
" encode, can not register"); \
gst_caps_unref (encode_init_data.sink_caps); \
return G_TYPE_INVALID; \
} \
GST_DEBUG (#CODEC" encode's src caps %" GST_PTR_FORMAT, caps); \
/* class data will be leaked if the element never gets instantiated */ \
GST_MINI_OBJECT_FLAG_SET (caps, GST_MINI_OBJECT_FLAG_MAY_BE_LEAKED); \
encode_init_data.src_caps = caps; \
type_info.class_data = &encode_init_data; \
encode_type = g_type_register_static (GST_TYPE_VAAPIENCODE, \
"GstVaapiEncode"#CLASS, &type_info, 0); \
\
return encode_type; \
} \
\
GType \
gst_vaapiencode_##NAME##_get_type (void) \
{ \
g_assert (encode_type != G_TYPE_INVALID); \
return encode_type; \
}
typedef struct _GstVaapiEncode GstVaapiEncode;
typedef struct _GstVaapiEncodeClass GstVaapiEncodeClass;
struct _GstVaapiEncode
{
/*< private >*/
GstVaapiPluginBase parent_instance;
GstVaapiEncoder *encoder;
GstVideoCodecState *input_state;
gboolean input_state_changed;
/* needs to be set by the subclass implementation */
gboolean need_codec_data;
GstVideoCodecState *output_state;
GPtrArray *prop_values;
GstCaps *allowed_sinkpad_caps;
};
struct _GstVaapiEncodeClass
{
/*< private >*/
GstVaapiPluginBaseClass parent_class;
guint prop_num;
gboolean (*set_config) (GstVaapiEncode * encode);
GstCaps * (*get_caps) (GstVaapiEncode * encode);
GstVaapiEncoder * (*alloc_encoder) (GstVaapiEncode * encode,
GstVaapiDisplay * display);
GstFlowReturn (*alloc_buffer) (GstVaapiEncode * encode,
GstVaapiCodedBuffer * coded_buf,
GstBuffer ** outbuf_ptr);
plugin: encode: List all possible profiles to detect input formats. The current get_profile just return one possible profile for the encode, which is not enough. For example, if we want to support HEVC 4:4:4 profile, the input of encode should be VYUA rather than NV12 in HEVC main profile. So the command line: gst-launch-1.0 videotestsrc num-buffers=200 ! capsfilter \ caps=video/x-raw,format=VUYA,width=800,height=600 ! vaapih265enc \ tune=low-power init-qp=30 ! fakesink can not work because vaapih265enc just report NV12 in sink caps, we need to specify the profile obviously like: gst-launch-1.0 videotestsrc num-buffers=200 ! capsfilter \ caps=video/x-raw,format=VUYA,width=800,height=600 ! vaapih265enc \ tune=low-power init-qp=30 ! capsfilter caps=video/x-h265, \ profile=main-444 ! fakesink The encode should have the ability to choose the profile based on input format automatically. If the input video format is VUYA, the main-444 profile should be auto choosed. We modify to let get_allowed_profiles of each encode sub class to return an array of all supported profiles based on downstream's allowed caps, or return NULL if no valid profiles specified by downstream. If no allowed profiles found, all profiles which belong to the current encoder's codec will be the candidates. The function gst_vaapi_encoder_get_surface_attributes collects the surface's attributes for that profile list we just get. So for this case, both NV12 and VUYA should be returned. TODO: some codec like VP9, need to implement the get_profile() function.
2019-12-23 06:29:08 +00:00
/* Get all possible profiles based on allowed caps */
GArray * (*get_allowed_profiles) (GstVaapiEncode * encode,
GstCaps * allowed);
};
GType
gst_vaapiencode_get_type (void) G_GNUC_CONST;
G_GNUC_INTERNAL
void
gst_vaapiencode_set_property_subclass (GObject * object,
guint prop_id, const GValue * value, GParamSpec * pspec);
G_GNUC_INTERNAL
void
gst_vaapiencode_get_property_subclass (GObject * object, guint prop_id,
GValue * value, GParamSpec * pspec);
G_GNUC_INTERNAL
gboolean
gst_vaapiencode_class_install_properties (GstVaapiEncodeClass * klass,
GObjectClass * encoder_class);
G_END_DECLS
#endif /* GST_VAAPIENCODE_H */