mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-10-06 02:32:23 +00:00
plugins: add mpeg2 encoder element.
Add GstVaapiEncodeMPEG2 element object. The actual plug-in element is called "vaapiencode_mpeg2". Valid properties: - rate-control: rate control mode (default: cqp - constant QP) - bitrate: desired bitrate in kbps (default: auto-calculated) - key-period: maximal distance between two key frames (default: 30) - max-bframes: number of B-frames between I and P (default: 2) - quantizer: constant quantizer (default: 8) Signed-off-by: Gwenole Beauchesne <gwenole.beauchesne@intel.com>
This commit is contained in:
parent
f1c33feef9
commit
2e356b0f7e
4 changed files with 330 additions and 0 deletions
|
@ -59,11 +59,13 @@ libgstvaapi_source_h = \
|
||||||
libgstvaapi_enc_source_c = \
|
libgstvaapi_enc_source_c = \
|
||||||
gstvaapiencode.c \
|
gstvaapiencode.c \
|
||||||
gstvaapiencode_h264.c \
|
gstvaapiencode_h264.c \
|
||||||
|
gstvaapiencode_mpeg2.c \
|
||||||
$(NULL)
|
$(NULL)
|
||||||
|
|
||||||
libgstvaapi_enc_source_h = \
|
libgstvaapi_enc_source_h = \
|
||||||
gstvaapiencode.h \
|
gstvaapiencode.h \
|
||||||
gstvaapiencode_h264.h \
|
gstvaapiencode_h264.h \
|
||||||
|
gstvaapiencode_mpeg2.h \
|
||||||
$(NULL)
|
$(NULL)
|
||||||
|
|
||||||
if USE_ENCODERS
|
if USE_ENCODERS
|
||||||
|
|
|
@ -32,6 +32,7 @@
|
||||||
|
|
||||||
#if USE_ENCODERS
|
#if USE_ENCODERS
|
||||||
#include "gstvaapiencode_h264.h"
|
#include "gstvaapiencode_h264.h"
|
||||||
|
#include "gstvaapiencode_mpeg2.h"
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#define PLUGIN_NAME "vaapi"
|
#define PLUGIN_NAME "vaapi"
|
||||||
|
@ -62,6 +63,9 @@ plugin_init (GstPlugin *plugin)
|
||||||
gst_element_register(plugin, "vaapiencode_h264",
|
gst_element_register(plugin, "vaapiencode_h264",
|
||||||
GST_RANK_PRIMARY,
|
GST_RANK_PRIMARY,
|
||||||
GST_TYPE_VAAPIENCODE_H264);
|
GST_TYPE_VAAPIENCODE_H264);
|
||||||
|
gst_element_register(plugin, "vaapiencode_mpeg2",
|
||||||
|
GST_RANK_PRIMARY,
|
||||||
|
GST_TYPE_VAAPIENCODE_MPEG2);
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
return TRUE;
|
return TRUE;
|
||||||
|
|
248
gst/vaapi/gstvaapiencode_mpeg2.c
Normal file
248
gst/vaapi/gstvaapiencode_mpeg2.c
Normal file
|
@ -0,0 +1,248 @@
|
||||||
|
/*
|
||||||
|
* gstvaapiencode_mpeg2.c - VA-API MPEG2 encoder
|
||||||
|
*
|
||||||
|
* Copyright (C) 2012-2013 Intel Corporation
|
||||||
|
*
|
||||||
|
* 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 "gst/vaapi/gstvaapicompat.h"
|
||||||
|
|
||||||
|
#include "gstvaapiencode_mpeg2.h"
|
||||||
|
#include "gstvaapipluginutil.h"
|
||||||
|
#include "gst/vaapi/gstvaapiencoder_mpeg2.h"
|
||||||
|
#include "gst/vaapi/gstvaapiencoder_mpeg2_priv.h"
|
||||||
|
#include "gst/vaapi/gstvaapidisplay.h"
|
||||||
|
#include "gst/vaapi/gstvaapivalue.h"
|
||||||
|
#include "gst/vaapi/gstvaapisurface.h"
|
||||||
|
|
||||||
|
#include <string.h>
|
||||||
|
|
||||||
|
GST_DEBUG_CATEGORY_STATIC (gst_vaapi_mpeg2_encode_debug);
|
||||||
|
#define GST_CAT_DEFAULT gst_vaapi_mpeg2_encode_debug
|
||||||
|
|
||||||
|
#define GST_CAPS_CODEC(CODEC) CODEC "; "
|
||||||
|
|
||||||
|
static const char gst_vaapiencode_mpeg2_sink_caps_str[] =
|
||||||
|
GST_VIDEO_CAPS_MAKE (GST_VIDEO_FORMATS_ALL) "; "
|
||||||
|
GST_VAAPI_SURFACE_CAPS;
|
||||||
|
|
||||||
|
static const char gst_vaapiencode_mpeg2_src_caps_str[] =
|
||||||
|
GST_CAPS_CODEC ("video/mpeg,"
|
||||||
|
"mpegversion = (int) 2, " "systemstream = (boolean) false");
|
||||||
|
|
||||||
|
static GstStaticPadTemplate gst_vaapiencode_mpeg2_sink_factory =
|
||||||
|
GST_STATIC_PAD_TEMPLATE ("sink",
|
||||||
|
GST_PAD_SINK,
|
||||||
|
GST_PAD_ALWAYS,
|
||||||
|
GST_STATIC_CAPS (gst_vaapiencode_mpeg2_sink_caps_str));
|
||||||
|
|
||||||
|
static GstStaticPadTemplate gst_vaapiencode_mpeg2_src_factory =
|
||||||
|
GST_STATIC_PAD_TEMPLATE ("src",
|
||||||
|
GST_PAD_SRC,
|
||||||
|
GST_PAD_ALWAYS,
|
||||||
|
GST_STATIC_CAPS (gst_vaapiencode_mpeg2_src_caps_str));
|
||||||
|
|
||||||
|
/* mpeg2 encode */
|
||||||
|
G_DEFINE_TYPE (GstVaapiEncodeMpeg2, gst_vaapiencode_mpeg2, GST_TYPE_VAAPIENCODE)
|
||||||
|
|
||||||
|
enum
|
||||||
|
{
|
||||||
|
PROP_0,
|
||||||
|
PROP_RATE_CONTROL,
|
||||||
|
PROP_BITRATE,
|
||||||
|
PROP_QUANTIZER,
|
||||||
|
PROP_KEY_PERIOD,
|
||||||
|
PROP_MAX_BFRAMES
|
||||||
|
};
|
||||||
|
|
||||||
|
static void
|
||||||
|
gst_vaapiencode_mpeg2_init (GstVaapiEncodeMpeg2 * mpeg2_encode)
|
||||||
|
{
|
||||||
|
mpeg2_encode->rate_control = GST_VAAPI_ENCODER_MPEG2_DEFAULT_RATE_CONTROL;
|
||||||
|
mpeg2_encode->bitrate = 0;
|
||||||
|
mpeg2_encode->quantizer = GST_VAAPI_ENCODER_MPEG2_DEFAULT_CQP;
|
||||||
|
mpeg2_encode->intra_period = GST_VAAPI_ENCODER_MPEG2_DEFAULT_GOP_SIZE;
|
||||||
|
mpeg2_encode->ip_period = GST_VAAPI_ENCODER_MPEG2_DEFAULT_MAX_BFRAMES;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
gst_vaapiencode_mpeg2_finalize (GObject * object)
|
||||||
|
{
|
||||||
|
G_OBJECT_CLASS (gst_vaapiencode_mpeg2_parent_class)->finalize (object);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
gst_vaapiencode_mpeg2_set_property (GObject * object,
|
||||||
|
guint prop_id, const GValue * value, GParamSpec * pspec)
|
||||||
|
{
|
||||||
|
GstVaapiEncodeMpeg2 *encode = GST_VAAPIENCODE_MPEG2 (object);
|
||||||
|
|
||||||
|
switch (prop_id) {
|
||||||
|
case PROP_RATE_CONTROL:
|
||||||
|
{
|
||||||
|
GstVaapiRateControl rate_control = g_value_get_enum (value);
|
||||||
|
if (rate_control == GST_VAAPI_RATECONTROL_CBR ||
|
||||||
|
rate_control == GST_VAAPI_RATECONTROL_CQP) {
|
||||||
|
encode->rate_control = g_value_get_enum (value);
|
||||||
|
} else {
|
||||||
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case PROP_BITRATE:
|
||||||
|
encode->bitrate = g_value_get_uint (value);
|
||||||
|
break;
|
||||||
|
case PROP_QUANTIZER:
|
||||||
|
encode->quantizer = g_value_get_uint (value);
|
||||||
|
break;
|
||||||
|
case PROP_KEY_PERIOD:
|
||||||
|
encode->intra_period = g_value_get_uint (value);
|
||||||
|
break;
|
||||||
|
case PROP_MAX_BFRAMES:
|
||||||
|
encode->ip_period = g_value_get_uint (value);
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
gst_vaapiencode_mpeg2_get_property (GObject * object,
|
||||||
|
guint prop_id, GValue * value, GParamSpec * pspec)
|
||||||
|
{
|
||||||
|
GstVaapiEncodeMpeg2 *encode = GST_VAAPIENCODE_MPEG2 (object);
|
||||||
|
|
||||||
|
switch (prop_id) {
|
||||||
|
case PROP_RATE_CONTROL:
|
||||||
|
g_value_set_enum (value, encode->rate_control);
|
||||||
|
break;
|
||||||
|
case PROP_BITRATE:
|
||||||
|
g_value_set_uint (value, encode->bitrate);
|
||||||
|
break;
|
||||||
|
case PROP_QUANTIZER:
|
||||||
|
g_value_set_uint (value, encode->quantizer);
|
||||||
|
break;
|
||||||
|
case PROP_KEY_PERIOD:
|
||||||
|
g_value_set_uint (value, encode->intra_period);
|
||||||
|
break;
|
||||||
|
case PROP_MAX_BFRAMES:
|
||||||
|
g_value_set_uint (value, encode->ip_period);
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static GstVaapiEncoder *
|
||||||
|
gst_vaapiencode_mpeg2_create_encoder (GstVaapiEncode * base,
|
||||||
|
GstVaapiDisplay * display)
|
||||||
|
{
|
||||||
|
GstVaapiEncodeMpeg2 *encode = GST_VAAPIENCODE_MPEG2 (base);
|
||||||
|
GstVaapiEncoder *ret;
|
||||||
|
GstVaapiEncoderMpeg2 *mpeg2encoder;
|
||||||
|
|
||||||
|
ret = gst_vaapi_encoder_mpeg2_new (display);
|
||||||
|
mpeg2encoder = GST_VAAPI_ENCODER_MPEG2 (ret);
|
||||||
|
|
||||||
|
mpeg2encoder->profile = GST_VAAPI_ENCODER_MPEG2_DEFAULT_PROFILE;
|
||||||
|
mpeg2encoder->level = GST_VAAPI_ENCODER_MPEG2_DEFAULT_LEVEL;
|
||||||
|
GST_VAAPI_ENCODER_RATE_CONTROL (mpeg2encoder) = encode->rate_control;
|
||||||
|
mpeg2encoder->bitrate = encode->bitrate;
|
||||||
|
mpeg2encoder->cqp = encode->quantizer;
|
||||||
|
mpeg2encoder->intra_period = encode->intra_period;
|
||||||
|
mpeg2encoder->ip_period = encode->ip_period;
|
||||||
|
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
gst_vaapiencode_mpeg2_class_init (GstVaapiEncodeMpeg2Class * 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_mpeg2_encode_debug,
|
||||||
|
"vaapimpeg2encode", 0, "vaapimpeg2encode element");
|
||||||
|
|
||||||
|
object_class->finalize = gst_vaapiencode_mpeg2_finalize;
|
||||||
|
object_class->set_property = gst_vaapiencode_mpeg2_set_property;
|
||||||
|
object_class->get_property = gst_vaapiencode_mpeg2_get_property;
|
||||||
|
|
||||||
|
encode_class->create_encoder = gst_vaapiencode_mpeg2_create_encoder;
|
||||||
|
|
||||||
|
gst_element_class_set_static_metadata (element_class,
|
||||||
|
"VA-API mpeg2 encoder",
|
||||||
|
"Codec/Encoder/Video",
|
||||||
|
"A VA-API based video encoder", "Guangxin Xu <guangxin.xu@intel.com>");
|
||||||
|
|
||||||
|
/* sink pad */
|
||||||
|
gst_element_class_add_pad_template (element_class,
|
||||||
|
gst_static_pad_template_get (&gst_vaapiencode_mpeg2_sink_factory)
|
||||||
|
);
|
||||||
|
|
||||||
|
/* src pad */
|
||||||
|
gst_element_class_add_pad_template (element_class,
|
||||||
|
gst_static_pad_template_get (&gst_vaapiencode_mpeg2_src_factory)
|
||||||
|
);
|
||||||
|
|
||||||
|
g_object_class_install_property (object_class,
|
||||||
|
PROP_RATE_CONTROL,
|
||||||
|
g_param_spec_enum ("rate-control",
|
||||||
|
"Rate Control",
|
||||||
|
"Rate control mode (CQP or CBR only)",
|
||||||
|
GST_VAAPI_TYPE_RATE_CONTROL,
|
||||||
|
GST_VAAPI_RATECONTROL_NONE,
|
||||||
|
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
||||||
|
|
||||||
|
g_object_class_install_property (object_class,
|
||||||
|
PROP_BITRATE,
|
||||||
|
g_param_spec_uint ("bitrate",
|
||||||
|
"Bitrate (kbps)",
|
||||||
|
"The desired bitrate expressed in kbps (0: auto-calculate)",
|
||||||
|
0, GST_VAAPI_ENCODER_MPEG2_MAX_BITRATE, 0, G_PARAM_READWRITE));
|
||||||
|
|
||||||
|
g_object_class_install_property (object_class,
|
||||||
|
PROP_QUANTIZER,
|
||||||
|
g_param_spec_uint ("quantizer",
|
||||||
|
"Constant Quantizer",
|
||||||
|
"Constant quantizer (if rate-control mode is CQP)",
|
||||||
|
GST_VAAPI_ENCODER_MPEG2_MIN_CQP,
|
||||||
|
GST_VAAPI_ENCODER_MPEG2_MAX_CQP,
|
||||||
|
GST_VAAPI_ENCODER_MPEG2_DEFAULT_CQP, G_PARAM_READWRITE));
|
||||||
|
|
||||||
|
g_object_class_install_property (object_class,
|
||||||
|
PROP_KEY_PERIOD,
|
||||||
|
g_param_spec_uint ("key-period",
|
||||||
|
"Key Period",
|
||||||
|
"Maximal distance between two key-frames",
|
||||||
|
1,
|
||||||
|
GST_VAAPI_ENCODER_MPEG2_MAX_GOP_SIZE,
|
||||||
|
GST_VAAPI_ENCODER_MPEG2_DEFAULT_GOP_SIZE, G_PARAM_READWRITE));
|
||||||
|
|
||||||
|
g_object_class_install_property (object_class,
|
||||||
|
PROP_MAX_BFRAMES,
|
||||||
|
g_param_spec_uint ("max-bframes",
|
||||||
|
"Max B-Frames",
|
||||||
|
"Number of B-frames between I and P",
|
||||||
|
0,
|
||||||
|
GST_VAAPI_ENCODER_MPEG2_MAX_MAX_BFRAMES,
|
||||||
|
GST_VAAPI_ENCODER_MPEG2_DEFAULT_MAX_BFRAMES, G_PARAM_READWRITE));
|
||||||
|
}
|
76
gst/vaapi/gstvaapiencode_mpeg2.h
Normal file
76
gst/vaapi/gstvaapiencode_mpeg2.h
Normal file
|
@ -0,0 +1,76 @@
|
||||||
|
/*
|
||||||
|
* gstvaapiencode_mpeg2.h - VA-API MPEG2 encoder
|
||||||
|
*
|
||||||
|
* Copyright (C) 2012-2013 Intel Corporation
|
||||||
|
*
|
||||||
|
* 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_MPEG2_H
|
||||||
|
#define GST_VAAPIENCODE_MPEG2_H
|
||||||
|
|
||||||
|
#include <gst/gst.h>
|
||||||
|
#include "gstvaapiencode.h"
|
||||||
|
|
||||||
|
G_BEGIN_DECLS
|
||||||
|
|
||||||
|
#define GST_TYPE_VAAPIENCODE_MPEG2 \
|
||||||
|
(gst_vaapiencode_mpeg2_get_type())
|
||||||
|
#define GST_IS_VAAPIENCODE_MPEG2(obj) \
|
||||||
|
(G_TYPE_CHECK_INSTANCE_TYPE ((obj), GST_TYPE_VAAPIENCODE_MPEG2))
|
||||||
|
#define GST_IS_VAAPIENCODE_MPEG2_CLASS(klass) \
|
||||||
|
(G_TYPE_CHECK_CLASS_TYPE ((klass), GST_TYPE_VAAPIENCODE_MPEG2))
|
||||||
|
|
||||||
|
#define GST_VAAPIENCODE_MPEG2_GET_CLASS(obj) \
|
||||||
|
(G_TYPE_INSTANCE_GET_CLASS ((obj), \
|
||||||
|
GST_TYPE_VAAPIENCODE_MPEG2, \
|
||||||
|
GstVaapiEncodeMpeg2Class))
|
||||||
|
|
||||||
|
#define GST_VAAPIENCODE_MPEG2(obj) \
|
||||||
|
(G_TYPE_CHECK_INSTANCE_CAST ((obj), \
|
||||||
|
GST_TYPE_VAAPIENCODE_MPEG2, \
|
||||||
|
GstVaapiEncodeMpeg2))
|
||||||
|
|
||||||
|
#define GST_VAAPIENCODE_MPEG2_CLASS(klass) \
|
||||||
|
(G_TYPE_CHECK_CLASS_CAST ((klass), \
|
||||||
|
GST_TYPE_VAAPIENCODE_MPEG2, \
|
||||||
|
GstVaapiEncodeMpeg2Class))
|
||||||
|
|
||||||
|
typedef struct _GstVaapiEncodeMpeg2 GstVaapiEncodeMpeg2;
|
||||||
|
typedef struct _GstVaapiEncodeMpeg2Class GstVaapiEncodeMpeg2Class;
|
||||||
|
|
||||||
|
struct _GstVaapiEncodeMpeg2
|
||||||
|
{
|
||||||
|
GstVaapiEncode parent;
|
||||||
|
|
||||||
|
GstVaapiRateControl rate_control;
|
||||||
|
guint32 bitrate; /* kbps */
|
||||||
|
guint32 quantizer;
|
||||||
|
guint32 intra_period;
|
||||||
|
guint32 ip_period;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct _GstVaapiEncodeMpeg2Class
|
||||||
|
{
|
||||||
|
GstVaapiEncodeClass parent_class;
|
||||||
|
};
|
||||||
|
|
||||||
|
GType
|
||||||
|
gst_vaapiencode_mpeg2_get_type (void) G_GNUC_CONST;
|
||||||
|
|
||||||
|
G_END_DECLS
|
||||||
|
|
||||||
|
#endif /* GST_VAAPIENCODE_MPEG2_H */
|
Loading…
Reference in a new issue