gstreamer/gst/vaapi/gstvaapiencode.h

217 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 */