diff --git a/gst/vaapi/Makefile.am b/gst/vaapi/Makefile.am index 3e54a41d18..016b3d92b8 100644 --- a/gst/vaapi/Makefile.am +++ b/gst/vaapi/Makefile.am @@ -105,7 +105,6 @@ libgstvaapi_source_c += $(libgstvaapi_vp8enc_source_c) libgstvaapi_source_h += $(libgstvaapi_vp8enc_source_h) endif - libgstvaapi_h265enc_source_c = gstvaapiencode_h265.c libgstvaapi_h265enc_source_h = gstvaapiencode_h265.h if USE_H265_ENCODER @@ -113,6 +112,13 @@ libgstvaapi_source_c += $(libgstvaapi_h265enc_source_c) libgstvaapi_source_h += $(libgstvaapi_h265enc_source_h) endif +libgstvaapi_vp9enc_source_c = gstvaapiencode_vp9.c +libgstvaapi_vp9enc_source_h = gstvaapiencode_vp9.h +if USE_VP9_ENCODER +libgstvaapi_source_c += $(libgstvaapi_vp9enc_source_c) +libgstvaapi_source_h += $(libgstvaapi_vp9enc_source_h) +endif + libgstvaapi_la_SOURCES = $(libgstvaapi_source_c) noinst_HEADERS = $(libgstvaapi_source_h) @@ -150,6 +156,8 @@ EXTRA_DIST = \ $(libgstvaapi_vp8enc_source_h) \ $(libgstvaapi_h265enc_source_c) \ $(libgstvaapi_h265enc_source_h) \ + $(libgstvaapi_vp9enc_source_c) \ + $(libgstvaapi_vp9enc_source_h) \ $(libgstvaapi_egl_source_c) \ $(libgstvaapi_egl_source_h) \ $(libgstvaapi_1_2p_source_c) \ diff --git a/gst/vaapi/gstvaapi.c b/gst/vaapi/gstvaapi.c index aabddc3590..396c7d0142 100644 --- a/gst/vaapi/gstvaapi.c +++ b/gst/vaapi/gstvaapi.c @@ -46,6 +46,10 @@ #include "gstvaapiencode_h265.h" #endif +#if USE_VP9_ENCODER +#include "gstvaapiencode_vp9.h" +#endif + #define PLUGIN_NAME "vaapi" #define PLUGIN_DESC "VA-API based elements" #define PLUGIN_LICENSE "LGPL" @@ -73,11 +77,14 @@ plugin_init (GstPlugin * plugin) gst_element_register (plugin, "vaapivp8enc", GST_RANK_PRIMARY, GST_TYPE_VAAPIENCODE_VP8); #endif - #if USE_H265_ENCODER gst_element_register (plugin, "vaapih265enc", GST_RANK_PRIMARY, GST_TYPE_VAAPIENCODE_H265); #endif +#if USE_VP9_ENCODER + gst_element_register (plugin, "vaapivp9enc", + GST_RANK_PRIMARY, GST_TYPE_VAAPIENCODE_VP9); +#endif gst_element_register (plugin, "vaapidecodebin", GST_RANK_PRIMARY + 2, GST_TYPE_VAAPI_DECODE_BIN); diff --git a/gst/vaapi/gstvaapiencode_vp9.c b/gst/vaapi/gstvaapiencode_vp9.c new file mode 100644 index 0000000000..74767575a2 --- /dev/null +++ b/gst/vaapi/gstvaapiencode_vp9.c @@ -0,0 +1,177 @@ +/* + * gstvaapiencode_vp9.c - VA-API VP9 encoder + * + * Copyright (C) 2016 Intel Corporation + * Author: Sreerenj Balachandran + * + * 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 + */ + +/** + * SECTION:element-vaapivp9enc + * @short_description: A VA-API based VP9 video encoder + * + * Encodes raw video streams into VP9 bitstreams. + * + * + * Example launch line + * |[ + * gst-launch-1.0 -ev videotestsrc num-buffers=60 ! timeoverlay ! vaapivp9enc ! matroskamux ! filesink location=test.mkv + * ]| + * + */ + +#include "gstcompat.h" +#include +#include +#include "gstvaapiencode_vp9.h" +#include "gstvaapipluginutil.h" +#include "gstvaapivideomemory.h" + +#define GST_PLUGIN_NAME "vaapivp9enc" +#define GST_PLUGIN_DESC "A VA-API based VP9 video encoder" + +GST_DEBUG_CATEGORY_STATIC (gst_vaapi_vp9_encode_debug); +#define GST_CAT_DEFAULT gst_vaapi_vp9_encode_debug + +#define GST_CODEC_CAPS \ + "video/x-vp9" + +/* *INDENT-OFF* */ +static const char gst_vaapiencode_vp9_sink_caps_str[] = + GST_VAAPI_MAKE_SURFACE_CAPS ", " + GST_CAPS_INTERLACED_FALSE "; " + GST_VIDEO_CAPS_MAKE (GST_VIDEO_FORMATS_ALL) ", " + GST_CAPS_INTERLACED_FALSE; +/* *INDENT-ON* */ + +/* *INDENT-OFF* */ +static const char gst_vaapiencode_vp9_src_caps_str[] = + GST_CODEC_CAPS; +/* *INDENT-ON* */ + +/* *INDENT-OFF* */ +static GstStaticPadTemplate gst_vaapiencode_vp9_sink_factory = + GST_STATIC_PAD_TEMPLATE ("sink", + GST_PAD_SINK, + GST_PAD_ALWAYS, + GST_STATIC_CAPS (gst_vaapiencode_vp9_sink_caps_str)); +/* *INDENT-ON* */ + +/* *INDENT-OFF* */ +static GstStaticPadTemplate gst_vaapiencode_vp9_src_factory = + GST_STATIC_PAD_TEMPLATE ("src", + GST_PAD_SRC, + GST_PAD_ALWAYS, + GST_STATIC_CAPS (gst_vaapiencode_vp9_src_caps_str)); +/* *INDENT-ON* */ + +/* vp9 encode */ +G_DEFINE_TYPE (GstVaapiEncodeVP9, gst_vaapiencode_vp9, GST_TYPE_VAAPIENCODE); + +static void +gst_vaapiencode_vp9_init (GstVaapiEncodeVP9 * encode) +{ + gst_vaapiencode_init_properties (GST_VAAPIENCODE_CAST (encode)); +} + +static void +gst_vaapiencode_vp9_finalize (GObject * object) +{ + G_OBJECT_CLASS (gst_vaapiencode_vp9_parent_class)->finalize (object); +} + +static void +gst_vaapiencode_vp9_set_property (GObject * object, + guint prop_id, const GValue * value, GParamSpec * pspec) +{ + GstVaapiEncodeClass *const encode_class = GST_VAAPIENCODE_GET_CLASS (object); + GstVaapiEncode *const base_encode = GST_VAAPIENCODE_CAST (object); + + switch (prop_id) { + default: + if (!encode_class->set_property (base_encode, prop_id, value)) + G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); + break; + } +} + +static void +gst_vaapiencode_vp9_get_property (GObject * object, + guint prop_id, GValue * value, GParamSpec * pspec) +{ + GstVaapiEncodeClass *const encode_class = GST_VAAPIENCODE_GET_CLASS (object); + GstVaapiEncode *const base_encode = GST_VAAPIENCODE_CAST (object); + + switch (prop_id) { + default: + if (!encode_class->get_property (base_encode, prop_id, value)) + G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); + break; + } +} + +static GstCaps * +gst_vaapiencode_vp9_get_caps (GstVaapiEncode * base_encode) +{ + GstCaps *caps; + + caps = gst_caps_from_string (GST_CODEC_CAPS); + + return caps; +} + +static GstVaapiEncoder * +gst_vaapiencode_vp9_alloc_encoder (GstVaapiEncode * base, + GstVaapiDisplay * display) +{ + return gst_vaapi_encoder_vp9_new (display); +} + +static void +gst_vaapiencode_vp9_class_init (GstVaapiEncodeVP9Class * klass) +{ + GObjectClass *const object_class = G_OBJECT_CLASS (klass); + GstElementClass *const element_class = GST_ELEMENT_CLASS (klass); + GstVaapiEncodeClass *const encode_class = GST_VAAPIENCODE_CLASS (klass); + + GST_DEBUG_CATEGORY_INIT (gst_vaapi_vp9_encode_debug, + GST_PLUGIN_NAME, 0, GST_PLUGIN_DESC); + + object_class->finalize = gst_vaapiencode_vp9_finalize; + object_class->set_property = gst_vaapiencode_vp9_set_property; + object_class->get_property = gst_vaapiencode_vp9_get_property; + + encode_class->get_properties = gst_vaapi_encoder_vp9_get_default_properties; + encode_class->get_caps = gst_vaapiencode_vp9_get_caps; + encode_class->alloc_encoder = gst_vaapiencode_vp9_alloc_encoder; + + gst_element_class_set_static_metadata (element_class, + "VA-API VP9 encoder", + "Codec/Encoder/Video", + GST_PLUGIN_DESC, + "Sreerenj Balachandran "); + + /* sink pad */ + gst_element_class_add_pad_template (element_class, + gst_static_pad_template_get (&gst_vaapiencode_vp9_sink_factory)); + + /* src pad */ + gst_element_class_add_pad_template (element_class, + gst_static_pad_template_get (&gst_vaapiencode_vp9_src_factory)); + + gst_vaapiencode_class_init_properties (encode_class); +} diff --git a/gst/vaapi/gstvaapiencode_vp9.h b/gst/vaapi/gstvaapiencode_vp9.h new file mode 100644 index 0000000000..09c1fe2a67 --- /dev/null +++ b/gst/vaapi/gstvaapiencode_vp9.h @@ -0,0 +1,69 @@ +/* + * gstvaapiencode_vp9.h - VA-API VP9 encoder + * + * Copyright (C) 2016 Intel Corporation + * Author: Sreerenj Balachandran + * + * 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_VP9_H +#define GST_VAAPIENCODE_VP9_H + +#include +#include "gstvaapiencode.h" + +G_BEGIN_DECLS + +#define GST_TYPE_VAAPIENCODE_VP9 \ + (gst_vaapiencode_vp9_get_type ()) +#define GST_VAAPIENCODE_VP9_CAST(obj) \ + ((GstVaapiEncodeVP9 *)(obj)) +#define GST_VAAPIENCODE_VP9(obj) \ + (G_TYPE_CHECK_INSTANCE_CAST ((obj), GST_TYPE_VAAPIENCODE_VP9, \ + GstVaapiEncodeVP9)) +#define GST_VAAPIENCODE_VP9_CLASS(klass) \ + (G_TYPE_CHECK_CLASS_CAST ((klass), GST_TYPE_VAAPIENCODE_VP9, \ + GstVaapiEncodeVP9Class)) +#define GST_VAAPIENCODE_VP9_GET_CLASS(obj) \ + (G_TYPE_INSTANCE_GET_CLASS ((obj), GST_TYPE_VAAPIENCODE_VP9, \ + GstVaapiEncodeVP9Class)) +#define GST_IS_VAAPIENCODE_VP9(obj) \ + (G_TYPE_CHECK_INSTANCE_TYPE ((obj), GST_TYPE_VAAPIENCODE_VP9)) +#define GST_IS_VAAPIENCODE_VP9_CLASS(klass) \ + (G_TYPE_CHECK_CLASS_TYPE ((klass), GST_TYPE_VAAPIENCODE_VP9)) + +typedef struct _GstVaapiEncodeVP9 GstVaapiEncodeVP9; +typedef struct _GstVaapiEncodeVP9Class GstVaapiEncodeVP9Class; + +struct _GstVaapiEncodeVP9 +{ + /*< private >*/ + GstVaapiEncode parent_instance; +}; + +struct _GstVaapiEncodeVP9Class +{ + /*< private >*/ + GstVaapiEncodeClass parent_class; +}; + +GType +gst_vaapiencode_vp9_get_type (void) G_GNUC_CONST; + +G_END_DECLS + +#endif /* GST_VAAPIENCODE_VP9_H */