mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-07-02 20:55:55 +00:00
218 lines
11 KiB
C
218 lines
11 KiB
C
/*
|
|
* gstvaapiencode.h - VA-API video encoder
|
|
*
|
|
* 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);
|
|
/* 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 */
|