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