From a664a479a854be26b78f2d475845695fbf56dd70 Mon Sep 17 00:00:00 2001 From: Sreerenj Balachandran Date: Fri, 13 Feb 2015 13:45:32 +0200 Subject: [PATCH] plugins: Add VP8 Encoder --- gst/vaapi/Makefile.am | 9 ++ gst/vaapi/gstvaapi.c | 9 ++ gst/vaapi/gstvaapiencode_vp8.c | 172 +++++++++++++++++++++++++++++++++ gst/vaapi/gstvaapiencode_vp8.h | 69 +++++++++++++ 4 files changed, 259 insertions(+) create mode 100644 gst/vaapi/gstvaapiencode_vp8.c create mode 100644 gst/vaapi/gstvaapiencode_vp8.h diff --git a/gst/vaapi/Makefile.am b/gst/vaapi/Makefile.am index 097ca6a1a5..6a49fdf0f8 100644 --- a/gst/vaapi/Makefile.am +++ b/gst/vaapi/Makefile.am @@ -84,6 +84,13 @@ libgstvaapi_source_c += $(libgstvaapi_jpegenc_source_c) libgstvaapi_source_h += $(libgstvaapi_jpegenc_source_h) endif +libgstvaapi_vp8enc_source_c = gstvaapiencode_vp8.c +libgstvaapi_vp8enc_source_h = gstvaapiencode_vp8.h +if USE_VP8_ENCODER +libgstvaapi_source_c += $(libgstvaapi_vp8enc_source_c) +libgstvaapi_source_h += $(libgstvaapi_vp8enc_source_h) +endif + libgstvaapi_x11_source_c = gstvaapivideoconverter_x11.c libgstvaapi_x11_source_h = gstvaapivideoconverter_x11.h @@ -246,6 +253,8 @@ EXTRA_DIST = \ $(libgstvaapi_enc_source_h) \ $(libgstvaapi_jpegenc_source_c) \ $(libgstvaapi_jpegenc_source_h) \ + $(libgstvaapi_vp8enc_source_c) \ + $(libgstvaapi_vp8enc_source_h) \ $(libgstvaapi_x11_source_c) \ $(libgstvaapi_x11_source_h) \ $(libgstvaapi_glx_source_c) \ diff --git a/gst/vaapi/gstvaapi.c b/gst/vaapi/gstvaapi.c index 8abd23b637..f15731b66f 100644 --- a/gst/vaapi/gstvaapi.c +++ b/gst/vaapi/gstvaapi.c @@ -39,6 +39,10 @@ #include "gstvaapiencode_jpeg.h" #endif +#if USE_VP8_ENCODER +#include "gstvaapiencode_vp8.h" +#endif + #define PLUGIN_NAME "vaapi" #define PLUGIN_DESC "VA-API based elements" #define PLUGIN_LICENSE "LGPL" @@ -76,6 +80,11 @@ plugin_init (GstPlugin *plugin) GST_RANK_PRIMARY, GST_TYPE_VAAPIENCODE_JPEG); #endif +#if USE_VP8_ENCODER + gst_element_register(plugin, "vaapiencode_vp8", + GST_RANK_PRIMARY, + GST_TYPE_VAAPIENCODE_VP8); +#endif return TRUE; } diff --git a/gst/vaapi/gstvaapiencode_vp8.c b/gst/vaapi/gstvaapiencode_vp8.c new file mode 100644 index 0000000000..e236206cb4 --- /dev/null +++ b/gst/vaapi/gstvaapiencode_vp8.c @@ -0,0 +1,172 @@ +/* + * gstvaapiencode_vp8.c - VA-API VP8 encoder + * + * Copyright (C) 2015 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 + */ + +#include "gst/vaapi/sysdeps.h" +#include +#include +#include "gstvaapiencode_vp8.h" +#include "gstvaapipluginutil.h" +#if GST_CHECK_VERSION(1,0,0) +#include "gstvaapivideomemory.h" +#endif + +#define GST_PLUGIN_NAME "vaapiencode_vp8" +#define GST_PLUGIN_DESC "A VA-API based VP8 video encoder" + +GST_DEBUG_CATEGORY_STATIC (gst_vaapi_vp8_encode_debug); +#define GST_CAT_DEFAULT gst_vaapi_vp8_encode_debug + +#define GST_CODEC_CAPS \ + "video/x-vp8" + +/* *INDENT-OFF* */ +static const char gst_vaapiencode_vp8_sink_caps_str[] = +#if GST_CHECK_VERSION(1,1,0) + GST_VIDEO_CAPS_MAKE_WITH_FEATURES (GST_CAPS_FEATURE_MEMORY_VAAPI_SURFACE, + "{ ENCODED, NV12, I420, YV12 }") ", " +#else + GST_VAAPI_SURFACE_CAPS ", " +#endif + GST_CAPS_INTERLACED_FALSE "; " +#if GST_CHECK_VERSION(1,0,0) + GST_VIDEO_CAPS_MAKE (GST_VIDEO_FORMATS_ALL) ", " +#endif + GST_CAPS_INTERLACED_FALSE; +/* *INDENT-ON* */ + +/* *INDENT-OFF* */ +static const char gst_vaapiencode_vp8_src_caps_str[] = + GST_CODEC_CAPS; +/* *INDENT-ON* */ + +/* *INDENT-OFF* */ +static GstStaticPadTemplate gst_vaapiencode_vp8_sink_factory = + GST_STATIC_PAD_TEMPLATE ("sink", + GST_PAD_SINK, + GST_PAD_ALWAYS, + GST_STATIC_CAPS (gst_vaapiencode_vp8_sink_caps_str)); +/* *INDENT-ON* */ + +/* *INDENT-OFF* */ +static GstStaticPadTemplate gst_vaapiencode_vp8_src_factory = + GST_STATIC_PAD_TEMPLATE ("src", + GST_PAD_SRC, + GST_PAD_ALWAYS, + GST_STATIC_CAPS (gst_vaapiencode_vp8_src_caps_str)); +/* *INDENT-ON* */ + +/* vp8 encode */ +G_DEFINE_TYPE (GstVaapiEncodeVP8, gst_vaapiencode_vp8, GST_TYPE_VAAPIENCODE); + +static void +gst_vaapiencode_vp8_init (GstVaapiEncodeVP8 * encode) +{ + gst_vaapiencode_init_properties (GST_VAAPIENCODE_CAST (encode)); +} + +static void +gst_vaapiencode_vp8_finalize (GObject * object) +{ + G_OBJECT_CLASS (gst_vaapiencode_vp8_parent_class)->finalize (object); +} + +static void +gst_vaapiencode_vp8_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_vp8_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_vp8_get_caps (GstVaapiEncode * base_encode) +{ + GstCaps *caps; + + caps = gst_caps_from_string (GST_CODEC_CAPS); + + return caps; +} + +static GstVaapiEncoder * +gst_vaapiencode_vp8_alloc_encoder (GstVaapiEncode * base, + GstVaapiDisplay * display) +{ + return gst_vaapi_encoder_vp8_new (display); +} + +static void +gst_vaapiencode_vp8_class_init (GstVaapiEncodeVP8Class * 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_vp8_encode_debug, + GST_PLUGIN_NAME, 0, GST_PLUGIN_DESC); + + object_class->finalize = gst_vaapiencode_vp8_finalize; + object_class->set_property = gst_vaapiencode_vp8_set_property; + object_class->get_property = gst_vaapiencode_vp8_get_property; + + encode_class->get_properties = gst_vaapi_encoder_vp8_get_default_properties; + encode_class->get_caps = gst_vaapiencode_vp8_get_caps; + encode_class->alloc_encoder = gst_vaapiencode_vp8_alloc_encoder; + + gst_element_class_set_static_metadata (element_class, + "VA-API VP8 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_vp8_sink_factory)); + + /* src pad */ + gst_element_class_add_pad_template (element_class, + gst_static_pad_template_get (&gst_vaapiencode_vp8_src_factory)); + + gst_vaapiencode_class_init_properties (encode_class); +} diff --git a/gst/vaapi/gstvaapiencode_vp8.h b/gst/vaapi/gstvaapiencode_vp8.h new file mode 100644 index 0000000000..e71c25505e --- /dev/null +++ b/gst/vaapi/gstvaapiencode_vp8.h @@ -0,0 +1,69 @@ +/* + * gstvaapiencode_vp8.h - VA-API VP8 encoder + * + * Copyright (C) 2015 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_VP8_H +#define GST_VAAPIENCODE_VP8_H + +#include +#include "gstvaapiencode.h" + +G_BEGIN_DECLS + +#define GST_TYPE_VAAPIENCODE_VP8 \ + (gst_vaapiencode_vp8_get_type ()) +#define GST_VAAPIENCODE_VP8_CAST(obj) \ + ((GstVaapiEncodeVP8 *)(obj)) +#define GST_VAAPIENCODE_VP8(obj) \ + (G_TYPE_CHECK_INSTANCE_CAST ((obj), GST_TYPE_VAAPIENCODE_VP8, \ + GstVaapiEncodeVP8)) +#define GST_VAAPIENCODE_VP8_CLASS(klass) \ + (G_TYPE_CHECK_CLASS_CAST ((klass), GST_TYPE_VAAPIENCODE_VP8, \ + GstVaapiEncodeVP8Class)) +#define GST_VAAPIENCODE_VP8_GET_CLASS(obj) \ + (G_TYPE_INSTANCE_GET_CLASS ((obj), GST_TYPE_VAAPIENCODE_VP8, \ + GstVaapiEncodeVP8Class)) +#define GST_IS_VAAPIENCODE_VP8(obj) \ + (G_TYPE_CHECK_INSTANCE_TYPE ((obj), GST_TYPE_VAAPIENCODE_VP8)) +#define GST_IS_VAAPIENCODE_VP8_CLASS(klass) \ + (G_TYPE_CHECK_CLASS_TYPE ((klass), GST_TYPE_VAAPIENCODE_VP8)) + +typedef struct _GstVaapiEncodeVP8 GstVaapiEncodeVP8; +typedef struct _GstVaapiEncodeVP8Class GstVaapiEncodeVP8Class; + +struct _GstVaapiEncodeVP8 +{ + /*< private >*/ + GstVaapiEncode parent_instance; +}; + +struct _GstVaapiEncodeVP8Class +{ + /*< private >*/ + GstVaapiEncodeClass parent_class; +}; + +GType +gst_vaapiencode_vp8_get_type (void) G_GNUC_CONST; + +G_END_DECLS + +#endif /* GST_VAAPIENCODE_VP8_H */