mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-19 06:46:38 +00:00
some first trial code for ffmux_* disable ffmux/ffdemux by default since they cause havoc
Original commit message from CVS: * some first trial code for ffmux_* * disable ffmux/ffdemux by default since they cause havoc I'll enable them locally for testing purposes until I've got them working in a general setup, too.
This commit is contained in:
Ronald S. Bultje
parent
59d7f9b535
commit
e8d306adb4
2 changed files with 349 additions and 254 deletions
|
|
@ -35,7 +35,7 @@
|
||||||
extern gboolean gst_ffmpegdemux_register (GstPlugin *plugin);
|
extern gboolean gst_ffmpegdemux_register (GstPlugin *plugin);
|
||||||
extern gboolean gst_ffmpegdec_register (GstPlugin *plugin);
|
extern gboolean gst_ffmpegdec_register (GstPlugin *plugin);
|
||||||
extern gboolean gst_ffmpegenc_register (GstPlugin *plugin);
|
extern gboolean gst_ffmpegenc_register (GstPlugin *plugin);
|
||||||
extern gboolean gst_ffmpegtypes_register (GstPlugin *plugin);
|
extern gboolean gst_ffmpegmux_register (GstPlugin *plugin);
|
||||||
|
|
||||||
extern URLProtocol gstreamer_protocol;
|
extern URLProtocol gstreamer_protocol;
|
||||||
|
|
||||||
|
|
@ -51,7 +51,8 @@ plugin_init (GModule *module, GstPlugin *plugin)
|
||||||
|
|
||||||
gst_ffmpegenc_register (plugin);
|
gst_ffmpegenc_register (plugin);
|
||||||
gst_ffmpegdec_register (plugin);
|
gst_ffmpegdec_register (plugin);
|
||||||
gst_ffmpegdemux_register (plugin);
|
/*gst_ffmpegdemux_register (plugin);*/
|
||||||
|
//gst_ffmpegmux_register (plugin);
|
||||||
|
|
||||||
register_protocol (&gstreamer_protocol);
|
register_protocol (&gstreamer_protocol);
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -17,8 +17,11 @@
|
||||||
* Boston, MA 02111-1307, USA.
|
* Boston, MA 02111-1307, USA.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include <string.h>
|
#ifdef HAVE_CONFIG_H
|
||||||
#include "config.h"
|
#include "config.h"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include <string.h>
|
||||||
#ifdef HAVE_FFMPEG_UNINSTALLED
|
#ifdef HAVE_FFMPEG_UNINSTALLED
|
||||||
#include <avformat.h>
|
#include <avformat.h>
|
||||||
#else
|
#else
|
||||||
|
|
@ -27,6 +30,8 @@
|
||||||
|
|
||||||
#include <gst/gst.h>
|
#include <gst/gst.h>
|
||||||
|
|
||||||
|
#include "gstffmpegcodecmap.h"
|
||||||
|
|
||||||
typedef struct _GstFFMpegMux GstFFMpegMux;
|
typedef struct _GstFFMpegMux GstFFMpegMux;
|
||||||
|
|
||||||
struct _GstFFMpegMux {
|
struct _GstFFMpegMux {
|
||||||
|
|
@ -34,18 +39,32 @@ struct _GstFFMpegMux {
|
||||||
|
|
||||||
/* We need to keep track of our pads, so we do so here. */
|
/* We need to keep track of our pads, so we do so here. */
|
||||||
GstPad *srcpad;
|
GstPad *srcpad;
|
||||||
GstPad *sinkpad;
|
|
||||||
|
|
||||||
AVCodecContext *context;
|
AVFormatContext *context;
|
||||||
AVFrame *picture;
|
gboolean opened;
|
||||||
|
|
||||||
|
GstPad *sinkpads[MAX_STREAMS];
|
||||||
|
gint videopads, audiopads;
|
||||||
|
GstBuffer *bufferqueue[MAX_STREAMS];
|
||||||
|
gboolean eos[MAX_STREAMS];
|
||||||
};
|
};
|
||||||
|
|
||||||
|
typedef struct _GstFFMpegMuxClassParams {
|
||||||
|
AVOutputFormat *in_plugin;
|
||||||
|
GstPadTemplate *srctempl;
|
||||||
|
GstPadTemplate *videosinktempl;
|
||||||
|
GstPadTemplate *audiosinktempl;
|
||||||
|
} GstFFMpegMuxClassParams;
|
||||||
|
|
||||||
typedef struct _GstFFMpegMuxClass GstFFMpegMuxClass;
|
typedef struct _GstFFMpegMuxClass GstFFMpegMuxClass;
|
||||||
|
|
||||||
struct _GstFFMpegMuxClass {
|
struct _GstFFMpegMuxClass {
|
||||||
GstElementClass parent_class;
|
GstElementClass parent_class;
|
||||||
|
|
||||||
AVCodec *in_plugin;
|
AVOutputFormat *in_plugin;
|
||||||
|
GstPadTemplate *srctempl;
|
||||||
|
GstPadTemplate *videosinktempl;
|
||||||
|
GstPadTemplate *audiosinktempl;
|
||||||
};
|
};
|
||||||
|
|
||||||
#define GST_TYPE_FFMPEGMUX \
|
#define GST_TYPE_FFMPEGMUX \
|
||||||
|
|
@ -69,71 +88,23 @@ enum {
|
||||||
/* FILL ME */
|
/* FILL ME */
|
||||||
};
|
};
|
||||||
|
|
||||||
/* This factory is much simpler, and defines the source pad. */
|
|
||||||
GST_PAD_TEMPLATE_FACTORY (gst_ffmpegmux_sink_factory,
|
|
||||||
"sink",
|
|
||||||
GST_PAD_SINK,
|
|
||||||
GST_PAD_ALWAYS,
|
|
||||||
GST_CAPS_NEW (
|
|
||||||
"ffmpegmux_sink",
|
|
||||||
"video/avi",
|
|
||||||
"format", GST_PROPS_STRING ("strf_vids")
|
|
||||||
),
|
|
||||||
GST_CAPS_NEW (
|
|
||||||
"ffmpegmux_sink",
|
|
||||||
"video/mpeg",
|
|
||||||
NULL
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
/* This factory is much simpler, and defines the source pad. */
|
|
||||||
GST_PAD_TEMPLATE_FACTORY (gst_ffmpegmux_audio_src_factory,
|
|
||||||
"src",
|
|
||||||
GST_PAD_SRC,
|
|
||||||
GST_PAD_ALWAYS,
|
|
||||||
GST_CAPS_NEW (
|
|
||||||
"ffmpegmux_src",
|
|
||||||
"audio/raw",
|
|
||||||
"format", GST_PROPS_STRING ("int"),
|
|
||||||
"law", GST_PROPS_INT (0),
|
|
||||||
"endianness", GST_PROPS_INT (G_BYTE_ORDER),
|
|
||||||
"signed", GST_PROPS_BOOLEAN (TRUE),
|
|
||||||
"width", GST_PROPS_INT (16),
|
|
||||||
"depth", GST_PROPS_INT (16),
|
|
||||||
"rate", GST_PROPS_INT_RANGE (8000, 96000),
|
|
||||||
"channels", GST_PROPS_INT_RANGE (1, 2)
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
/* This factory is much simpler, and defines the source pad. */
|
|
||||||
GST_PAD_TEMPLATE_FACTORY (gst_ffmpegmux_video_src_factory,
|
|
||||||
"src",
|
|
||||||
GST_PAD_SRC,
|
|
||||||
GST_PAD_ALWAYS,
|
|
||||||
GST_CAPS_NEW (
|
|
||||||
"ffmpegmux_src",
|
|
||||||
"video/raw",
|
|
||||||
"format", GST_PROPS_LIST (
|
|
||||||
GST_PROPS_FOURCC (GST_STR_FOURCC ("I420"))
|
|
||||||
),
|
|
||||||
"width", GST_PROPS_INT_RANGE (16, 4096),
|
|
||||||
"height", GST_PROPS_INT_RANGE (16, 4096)
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
static GHashTable *global_plugins;
|
static GHashTable *global_plugins;
|
||||||
|
|
||||||
/* A number of functon prototypes are given so we can refer to them later. */
|
/* A number of functon prototypes are given so we can refer to them later. */
|
||||||
static void gst_ffmpegmux_class_init (GstFFMpegMuxClass *klass);
|
static void gst_ffmpegmux_class_init (GstFFMpegMuxClass *klass);
|
||||||
static void gst_ffmpegmux_init (GstFFMpegMux *ffmpegmux);
|
static void gst_ffmpegmux_init (GstFFMpegMux *ffmpegmux);
|
||||||
|
static void gst_ffmpegmux_dispose (GObject *object);
|
||||||
|
|
||||||
static void gst_ffmpegmux_chain_audio (GstPad *pad, GstBuffer *buffer);
|
static GstPadLinkReturn
|
||||||
static void gst_ffmpegmux_chain_video (GstPad *pad, GstBuffer *buffer);
|
gst_ffmpegmux_connect (GstPad *pad,
|
||||||
|
GstCaps *caps);
|
||||||
|
static GstPad * gst_ffmpegmux_request_new_pad (GstElement *element,
|
||||||
|
GstPadTemplate *templ,
|
||||||
|
const gchar *name);
|
||||||
|
static void gst_ffmpegmux_loop (GstElement *element);
|
||||||
|
|
||||||
static void gst_ffmpegmux_set_property (GObject *object, guint prop_id, const GValue *value,
|
static GstElementStateReturn
|
||||||
GParamSpec *pspec);
|
gst_ffmpegmux_change_state (GstElement *element);
|
||||||
static void gst_ffmpegmux_get_property (GObject *object, guint prop_id, GValue *value,
|
|
||||||
GParamSpec *pspec);
|
|
||||||
|
|
||||||
static GstElementClass *parent_class = NULL;
|
static GstElementClass *parent_class = NULL;
|
||||||
|
|
||||||
|
|
@ -144,49 +115,24 @@ gst_ffmpegmux_class_init (GstFFMpegMuxClass *klass)
|
||||||
{
|
{
|
||||||
GObjectClass *gobject_class;
|
GObjectClass *gobject_class;
|
||||||
GstElementClass *gstelement_class;
|
GstElementClass *gstelement_class;
|
||||||
|
GstFFMpegMuxClassParams *params;
|
||||||
|
|
||||||
gobject_class = (GObjectClass*)klass;
|
gobject_class = (GObjectClass*)klass;
|
||||||
gstelement_class = (GstElementClass*)klass;
|
gstelement_class = (GstElementClass*)klass;
|
||||||
|
|
||||||
parent_class = g_type_class_ref(GST_TYPE_ELEMENT);
|
parent_class = g_type_class_ref(GST_TYPE_ELEMENT);
|
||||||
|
|
||||||
klass->in_plugin = g_hash_table_lookup (global_plugins,
|
params = g_hash_table_lookup (global_plugins,
|
||||||
GINT_TO_POINTER (G_OBJECT_CLASS_TYPE (gobject_class)));
|
GINT_TO_POINTER (G_OBJECT_CLASS_TYPE (gobject_class)));
|
||||||
|
|
||||||
gobject_class->set_property = gst_ffmpegmux_set_property;
|
klass->in_plugin = params->in_plugin;
|
||||||
gobject_class->get_property = gst_ffmpegmux_get_property;
|
klass->videosinktempl = params->videosinktempl;
|
||||||
}
|
klass->audiosinktempl = params->audiosinktempl;
|
||||||
|
klass->srctempl = params->srctempl;
|
||||||
|
|
||||||
static GstPadLinkReturn
|
gstelement_class->request_new_pad = gst_ffmpegmux_request_new_pad;
|
||||||
gst_ffmpegmux_sinkconnect (GstPad *pad, GstCaps *caps)
|
gstelement_class->change_state = gst_ffmpegmux_change_state;
|
||||||
{
|
gobject_class->dispose = gst_ffmpegmux_dispose;
|
||||||
GstFFMpegMux *ffmpegmux = (GstFFMpegMux *)(gst_pad_get_parent (pad));
|
|
||||||
GstFFMpegMuxClass *oclass = (GstFFMpegMuxClass*)(G_OBJECT_GET_CLASS (ffmpegmux));
|
|
||||||
|
|
||||||
if (!GST_CAPS_IS_FIXED (caps))
|
|
||||||
return GST_PAD_LINK_DELAYED;
|
|
||||||
|
|
||||||
if (gst_caps_has_property_typed (caps, "width", GST_PROPS_INT_TYPE))
|
|
||||||
gst_caps_get_int (caps, "width", &ffmpegmux->context->width);
|
|
||||||
if (gst_caps_has_property_typed (caps, "height", GST_PROPS_INT_TYPE))
|
|
||||||
gst_caps_get_int (caps, "height", &ffmpegmux->context->height);
|
|
||||||
|
|
||||||
ffmpegmux->context->pix_fmt = PIX_FMT_YUV420P;
|
|
||||||
ffmpegmux->context->frame_rate = 23 * DEFAULT_FRAME_RATE_BASE;
|
|
||||||
ffmpegmux->context->frame_rate_base = DEFAULT_FRAME_RATE_BASE;
|
|
||||||
ffmpegmux->context->bit_rate = 0;
|
|
||||||
|
|
||||||
/* FIXME bug in ffmpeg */
|
|
||||||
if (avcodec_open (ffmpegmux->context, avcodec_find_encoder(CODEC_ID_MPEG1VIDEO)) <0 ) {
|
|
||||||
g_warning ("ffmpegmux: could not open codec");
|
|
||||||
return GST_PAD_LINK_REFUSED;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (avcodec_open (ffmpegmux->context, oclass->in_plugin) < 0) {
|
|
||||||
g_warning ("ffmpegmux: could not open codec");
|
|
||||||
return GST_PAD_LINK_REFUSED;
|
|
||||||
}
|
|
||||||
return GST_PAD_LINK_OK;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static void
|
static void
|
||||||
|
|
@ -194,158 +140,263 @@ gst_ffmpegmux_init(GstFFMpegMux *ffmpegmux)
|
||||||
{
|
{
|
||||||
GstFFMpegMuxClass *oclass = (GstFFMpegMuxClass*)(G_OBJECT_GET_CLASS (ffmpegmux));
|
GstFFMpegMuxClass *oclass = (GstFFMpegMuxClass*)(G_OBJECT_GET_CLASS (ffmpegmux));
|
||||||
|
|
||||||
ffmpegmux->context = avcodec_alloc_context();
|
ffmpegmux->srcpad = gst_pad_new_from_template (oclass->srctempl,
|
||||||
|
"src");
|
||||||
|
gst_element_add_pad (GST_ELEMENT (ffmpegmux),
|
||||||
|
ffmpegmux->srcpad);
|
||||||
|
gst_element_set_loop_function (GST_ELEMENT (ffmpegmux),
|
||||||
|
gst_ffmpegmux_loop);
|
||||||
|
|
||||||
ffmpegmux->sinkpad = gst_pad_new_from_template (
|
ffmpegmux->context = g_new0 (AVFormatContext, 1);
|
||||||
GST_PAD_TEMPLATE_GET (gst_ffmpegmux_sink_factory), "sink");
|
memset (ffmpegmux->context, 0, sizeof (AVFormatContext));
|
||||||
gst_pad_set_link_function (ffmpegmux->sinkpad, gst_ffmpegmux_sinkconnect);
|
ffmpegmux->context->oformat = oclass->in_plugin;
|
||||||
|
ffmpegmux->context->nb_streams = 0;
|
||||||
|
sprintf (ffmpegmux->context->filename, "gstreamer://%p",
|
||||||
|
ffmpegmux->srcpad);
|
||||||
|
ffmpegmux->opened = FALSE;
|
||||||
|
|
||||||
if (oclass->in_plugin->type == CODEC_TYPE_VIDEO) {
|
ffmpegmux->videopads = 0;
|
||||||
ffmpegmux->srcpad = gst_pad_new_from_template (
|
ffmpegmux->audiopads = 0;
|
||||||
GST_PAD_TEMPLATE_GET (gst_ffmpegmux_video_src_factory), "src");
|
|
||||||
gst_pad_set_chain_function (ffmpegmux->sinkpad, gst_ffmpegmux_chain_video);
|
|
||||||
}
|
|
||||||
else if (oclass->in_plugin->type == CODEC_TYPE_AUDIO) {
|
|
||||||
ffmpegmux->srcpad = gst_pad_new_from_template (
|
|
||||||
GST_PAD_TEMPLATE_GET (gst_ffmpegmux_audio_src_factory), "src");
|
|
||||||
gst_pad_set_chain_function (ffmpegmux->sinkpad, gst_ffmpegmux_chain_audio);
|
|
||||||
}
|
|
||||||
|
|
||||||
gst_element_add_pad (GST_ELEMENT (ffmpegmux), ffmpegmux->sinkpad);
|
|
||||||
gst_element_add_pad (GST_ELEMENT (ffmpegmux), ffmpegmux->srcpad);
|
|
||||||
|
|
||||||
ffmpegmux->picture = g_malloc0 (sizeof (AVFrame));
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static void
|
static void
|
||||||
gst_ffmpegmux_chain_audio (GstPad *pad, GstBuffer *inbuf)
|
gst_ffmpegmux_dispose (GObject *object)
|
||||||
{
|
{
|
||||||
/*GstFFMpegMux *ffmpegmux = (GstFFMpegMux *)(gst_pad_get_parent (pad)); */
|
GstFFMpegMux *ffmpegmux = (GstFFMpegMux *) object;
|
||||||
gpointer data;
|
|
||||||
gint size;
|
|
||||||
|
|
||||||
data = GST_BUFFER_DATA (inbuf);
|
if (ffmpegmux->opened) {
|
||||||
size = GST_BUFFER_SIZE (inbuf);
|
url_fclose (&ffmpegmux->context->pb);
|
||||||
|
ffmpegmux->opened = FALSE;
|
||||||
|
}
|
||||||
|
|
||||||
GST_DEBUG (0, "got buffer %p %d", data, size);
|
g_free (ffmpegmux->context);
|
||||||
|
|
||||||
gst_buffer_unref (inbuf);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static void
|
static GstPad *
|
||||||
gst_ffmpegmux_chain_video (GstPad *pad, GstBuffer *inbuf)
|
gst_ffmpegmux_request_new_pad (GstElement *element,
|
||||||
|
GstPadTemplate *templ,
|
||||||
|
const gchar *name)
|
||||||
|
{
|
||||||
|
GstFFMpegMux *ffmpegmux = (GstFFMpegMux *) element;
|
||||||
|
GstFFMpegMuxClass *oclass = (GstFFMpegMuxClass*)(G_OBJECT_GET_CLASS (ffmpegmux));
|
||||||
|
gchar *padname;
|
||||||
|
GstPad *pad;
|
||||||
|
AVStream *st;
|
||||||
|
enum CodecType type;
|
||||||
|
gint padnum;
|
||||||
|
|
||||||
|
g_return_val_if_fail (templ != NULL, NULL);
|
||||||
|
g_return_val_if_fail (templ->direction == GST_PAD_SINK, NULL);
|
||||||
|
g_return_val_if_fail (ffmpegmux->opened == FALSE, NULL);
|
||||||
|
|
||||||
|
/* figure out a name that *we* like */
|
||||||
|
if (templ == oclass->videosinktempl) {
|
||||||
|
padname = g_strdup_printf ("video_%02d",
|
||||||
|
ffmpegmux->videopads++);
|
||||||
|
type = CODEC_TYPE_VIDEO;
|
||||||
|
} else if (templ == oclass->audiosinktempl) {
|
||||||
|
padname = g_strdup_printf ("audio_%02d",
|
||||||
|
ffmpegmux->audiopads++);
|
||||||
|
type = CODEC_TYPE_AUDIO;
|
||||||
|
} else {
|
||||||
|
g_warning("ffmux: unknown pad template!");
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* create pad */
|
||||||
|
pad = gst_pad_new_from_template (templ, padname);
|
||||||
|
padnum = ffmpegmux->context->nb_streams++;
|
||||||
|
ffmpegmux->sinkpads[padnum] = pad;
|
||||||
|
gst_pad_set_link_function (pad, gst_ffmpegmux_connect);
|
||||||
|
gst_element_add_pad (element, pad);
|
||||||
|
|
||||||
|
/* AVStream needs to be created */
|
||||||
|
st = g_malloc (sizeof (AVStream));
|
||||||
|
memset(st, 0, sizeof (AVStream));
|
||||||
|
avcodec_get_context_defaults (&st->codec);
|
||||||
|
st->codec.codec_type = type;
|
||||||
|
st->codec.codec_id = CODEC_ID_NONE; /* this is a check afterwards */
|
||||||
|
st->index = padnum;
|
||||||
|
st->stream_copy = 1; /* we're not the actual encoder */
|
||||||
|
/* we fill in other codec-related info during capsnego */
|
||||||
|
ffmpegmux->context->streams[padnum] = st;
|
||||||
|
|
||||||
|
/* we love debug output (c) (tm) (r) */
|
||||||
|
GST_DEBUG (GST_CAT_PLUGIN_INFO,
|
||||||
|
"Created %s pad for ffmux_%s element",
|
||||||
|
padname, oclass->in_plugin->name);
|
||||||
|
g_free (padname);
|
||||||
|
|
||||||
|
return pad;
|
||||||
|
}
|
||||||
|
|
||||||
|
static GstPadLinkReturn
|
||||||
|
gst_ffmpegmux_connect (GstPad *pad,
|
||||||
|
GstCaps *caps)
|
||||||
{
|
{
|
||||||
GstBuffer *outbuf;
|
|
||||||
GstFFMpegMux *ffmpegmux = (GstFFMpegMux *)(gst_pad_get_parent (pad));
|
GstFFMpegMux *ffmpegmux = (GstFFMpegMux *)(gst_pad_get_parent (pad));
|
||||||
guchar *data;
|
gint i;
|
||||||
gint size, frame_size, len;
|
AVStream *st;
|
||||||
gint have_picture;
|
|
||||||
|
|
||||||
data = GST_BUFFER_DATA (inbuf);
|
g_return_val_if_fail (ffmpegmux->opened == FALSE,
|
||||||
size = GST_BUFFER_SIZE (inbuf);
|
GST_PAD_LINK_REFUSED);
|
||||||
|
|
||||||
do {
|
if (!GST_CAPS_IS_FIXED (caps))
|
||||||
ffmpegmux->context->frame_number++;
|
return GST_PAD_LINK_DELAYED;
|
||||||
|
|
||||||
len = avcodec_decode_video (ffmpegmux->context, ffmpegmux->picture,
|
for (i = 0; i < ffmpegmux->context->nb_streams; i++) {
|
||||||
&have_picture, data, size);
|
if (pad == ffmpegmux->sinkpads[i]) {
|
||||||
|
|
||||||
if (len < 0) {
|
|
||||||
g_warning ("ffmpegmux: decoding error");
|
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
if (i == ffmpegmux->context->nb_streams) {
|
||||||
|
g_warning ("Unknown pad given during capsnego: %p", pad);
|
||||||
|
return GST_PAD_LINK_REFUSED;
|
||||||
|
}
|
||||||
|
st = ffmpegmux->context->streams[i];
|
||||||
|
|
||||||
if (have_picture) {
|
/* for the format-specific guesses, we'll go to
|
||||||
guchar *picdata, *picdata2, *outdata, *outdata2;
|
* our famous codec mapper */
|
||||||
gint xsize, i, width, height;
|
for ( ; caps != NULL; caps = caps->next) {
|
||||||
|
if (gst_ffmpeg_caps_to_codecid (caps,
|
||||||
width = ffmpegmux->context->width;
|
&st->codec.codec_id) == CODEC_ID_NONE) {
|
||||||
height = ffmpegmux->context->height;
|
ffmpegmux->eos[i] = FALSE;
|
||||||
|
return GST_PAD_LINK_OK;
|
||||||
if (!GST_PAD_CAPS (ffmpegmux->srcpad)) {
|
}
|
||||||
gst_pad_try_set_caps (ffmpegmux->srcpad,
|
|
||||||
GST_CAPS_NEW (
|
|
||||||
"ffmpegmux_src",
|
|
||||||
"video/raw",
|
|
||||||
"format", GST_PROPS_FOURCC (GST_STR_FOURCC ("I420")),
|
|
||||||
"width", GST_PROPS_INT (width),
|
|
||||||
"height", GST_PROPS_INT (height)
|
|
||||||
));
|
|
||||||
}
|
}
|
||||||
|
|
||||||
frame_size = width * height;
|
return GST_PAD_LINK_REFUSED;
|
||||||
|
|
||||||
outbuf = gst_buffer_new ();
|
|
||||||
GST_BUFFER_SIZE (outbuf) = (frame_size*3)>>1;
|
|
||||||
outdata = GST_BUFFER_DATA (outbuf) = g_malloc (GST_BUFFER_SIZE (outbuf));
|
|
||||||
GST_BUFFER_TIMESTAMP (outbuf) = GST_BUFFER_TIMESTAMP (inbuf);
|
|
||||||
|
|
||||||
picdata = ffmpegmux->picture->data[0];
|
|
||||||
xsize = ffmpegmux->picture->linesize[0];
|
|
||||||
for (i=height; i; i--) {
|
|
||||||
memcpy (outdata, picdata, width);
|
|
||||||
outdata += width;
|
|
||||||
picdata += xsize;
|
|
||||||
}
|
|
||||||
|
|
||||||
frame_size >>= 2;
|
|
||||||
width >>= 1;
|
|
||||||
height >>= 1;
|
|
||||||
outdata2 = outdata + frame_size;
|
|
||||||
|
|
||||||
picdata = ffmpegmux->picture->data[1];
|
|
||||||
picdata2 = ffmpegmux->picture->data[2];
|
|
||||||
xsize = ffmpegmux->picture->linesize[1];
|
|
||||||
for (i=height; i; i--) {
|
|
||||||
memcpy (outdata, picdata, width);
|
|
||||||
memcpy (outdata2, picdata2, width);
|
|
||||||
outdata += width; outdata2 += width;
|
|
||||||
picdata += xsize; picdata2 += xsize;
|
|
||||||
}
|
|
||||||
|
|
||||||
gst_pad_push (ffmpegmux->srcpad, outbuf);
|
|
||||||
}
|
|
||||||
|
|
||||||
size -= len;
|
|
||||||
data += len;
|
|
||||||
}
|
|
||||||
while (size > 0);
|
|
||||||
|
|
||||||
gst_buffer_unref (inbuf);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static void
|
static void
|
||||||
gst_ffmpegmux_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec)
|
gst_ffmpegmux_loop (GstElement *element)
|
||||||
{
|
{
|
||||||
GstFFMpegMux *ffmpegmux;
|
GstFFMpegMux *ffmpegmux = (GstFFMpegMux *) element;
|
||||||
|
GstFFMpegMuxClass *oclass = (GstFFMpegMuxClass*)(G_OBJECT_GET_CLASS (ffmpegmux));
|
||||||
|
gint i, bufnum;
|
||||||
|
GstBuffer *buf;
|
||||||
|
|
||||||
/* Get a pointer of the right type. */
|
/* we do need all streams to have started capsnego,
|
||||||
ffmpegmux = (GstFFMpegMux *)(object);
|
* or things will go horribly wrong */
|
||||||
|
for (i = 0; i < ffmpegmux->context->nb_streams; i++) {
|
||||||
|
AVStream *st = ffmpegmux->context->streams[i];
|
||||||
|
|
||||||
/* Check the argument id to see which argument we're setting. */
|
/* check whether the pad has successfully completed capsnego */
|
||||||
switch (prop_id) {
|
if (st->codec.codec_id == CODEC_ID_NONE) {
|
||||||
default:
|
GST_DEBUG (GST_CAT_PLUGIN_INFO,
|
||||||
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
"ffmux loop function called without capsnego on pad %d",
|
||||||
break;
|
i);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/* open "file" (gstreamer protocol to next element) */
|
||||||
|
if (!ffmpegmux->opened) {
|
||||||
|
if (url_fopen (&ffmpegmux->context->pb,
|
||||||
|
ffmpegmux->context->filename,
|
||||||
|
URL_WRONLY) < 0) {
|
||||||
|
gst_element_error (element,
|
||||||
|
"Failed to open stream context in ffmux");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* we're now opened */
|
||||||
|
ffmpegmux->opened = TRUE;
|
||||||
|
|
||||||
|
/* now open the mux format */
|
||||||
|
oclass->in_plugin->write_header (ffmpegmux->context);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* FIXME: do we support more events? */
|
||||||
|
|
||||||
|
/* start by filling an internal queue of buffers */
|
||||||
|
for (i = 0; i < ffmpegmux->context->nb_streams; i++) {
|
||||||
|
GstPad *pad = ffmpegmux->sinkpads[i];
|
||||||
|
|
||||||
|
/* check for "pull'ability" */
|
||||||
|
while (pad != NULL &&
|
||||||
|
GST_PAD_IS_USABLE (pad) &&
|
||||||
|
/*GST_PAD_IS_ACTIVE (pad) &&*/
|
||||||
|
GST_PAD_IS_LINKED (pad) &&
|
||||||
|
ffmpegmux->eos[i] == FALSE &&
|
||||||
|
ffmpegmux->bufferqueue[i] == NULL) {
|
||||||
|
/* we can pull a buffer! */
|
||||||
|
buf = gst_pad_pull (pad);
|
||||||
|
if (GST_IS_EVENT (buf)) {
|
||||||
|
if (GST_EVENT_TYPE (GST_EVENT (buf)) == GST_EVENT_EOS) {
|
||||||
|
/* flag EOS on this stream */
|
||||||
|
ffmpegmux->eos[i] = TRUE;
|
||||||
|
gst_data_unref (GST_DATA (buf));
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
ffmpegmux->bufferqueue[i] = buf;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/* take the one with earliest timestamp,
|
||||||
|
* and push it forward */
|
||||||
|
bufnum = -1;
|
||||||
|
for (i = 0; i < ffmpegmux->context->nb_streams; i++) {
|
||||||
|
/* if there's no buffer, just continue */
|
||||||
|
if (ffmpegmux->bufferqueue[i] == NULL) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* if we have no buffer yet, just use the first one */
|
||||||
|
if (bufnum == -1) {
|
||||||
|
bufnum = i;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* if we do have one, only use this one if it's older */
|
||||||
|
if (GST_BUFFER_TIMESTAMP (ffmpegmux->bufferqueue[i]) <
|
||||||
|
GST_BUFFER_TIMESTAMP (ffmpegmux->bufferqueue[bufnum])) {
|
||||||
|
bufnum = i;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/* now handle the buffer, or signal EOS if we have
|
||||||
|
* no buffers left */
|
||||||
|
if (bufnum >= 0) {
|
||||||
|
/* push out current buffer */
|
||||||
|
buf = ffmpegmux->bufferqueue[bufnum];
|
||||||
|
ffmpegmux->bufferqueue[bufnum] = NULL;
|
||||||
|
oclass->in_plugin->write_packet (ffmpegmux->context,
|
||||||
|
bufnum,
|
||||||
|
GST_BUFFER_DATA (buf),
|
||||||
|
GST_BUFFER_SIZE (buf), 0);
|
||||||
|
gst_buffer_unref (buf);
|
||||||
|
} else {
|
||||||
|
/* close down */
|
||||||
|
oclass->in_plugin->write_trailer (ffmpegmux->context);
|
||||||
|
url_fclose (&ffmpegmux->context->pb);
|
||||||
|
ffmpegmux->opened = FALSE;
|
||||||
|
gst_element_set_eos (element);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/* The set function is simply the inverse of the get fuction. */
|
static GstElementStateReturn
|
||||||
static void
|
gst_ffmpegmux_change_state (GstElement *element)
|
||||||
gst_ffmpegmux_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec)
|
|
||||||
{
|
{
|
||||||
GstFFMpegMux *ffmpegmux;
|
GstFFMpegMux *ffmpegmux = (GstFFMpegMux *)(element);
|
||||||
|
gint transition = GST_STATE_TRANSITION (element);
|
||||||
|
|
||||||
/* It's not null if we got it, but it might not be ours */
|
switch (transition) {
|
||||||
ffmpegmux = (GstFFMpegMux *)(object);
|
case GST_STATE_PAUSED_TO_READY:
|
||||||
|
if (ffmpegmux->opened) {
|
||||||
switch (prop_id) {
|
url_fclose (&ffmpegmux->context->pb);
|
||||||
default:
|
ffmpegmux->opened = FALSE;
|
||||||
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
}
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (GST_ELEMENT_CLASS (parent_class)->change_state)
|
||||||
|
return GST_ELEMENT_CLASS (parent_class)->change_state (element);
|
||||||
|
|
||||||
|
return GST_STATE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
gboolean
|
gboolean
|
||||||
gst_ffmpegmux_register (GstPlugin *plugin)
|
gst_ffmpegmux_register (GstPlugin *plugin)
|
||||||
{
|
{
|
||||||
|
|
@ -363,24 +414,57 @@ gst_ffmpegmux_register (GstPlugin *plugin)
|
||||||
};
|
};
|
||||||
GType type;
|
GType type;
|
||||||
GstElementDetails *details;
|
GstElementDetails *details;
|
||||||
AVCodec *in_plugin;
|
AVOutputFormat *in_plugin;
|
||||||
|
GstFFMpegMuxClassParams *params;
|
||||||
|
AVCodec *in_codec;
|
||||||
|
|
||||||
in_plugin = first_avcodec;
|
in_plugin = first_oformat;
|
||||||
|
|
||||||
global_plugins = g_hash_table_new (NULL, NULL);
|
global_plugins = g_hash_table_new (NULL, NULL);
|
||||||
|
|
||||||
while (in_plugin) {
|
while (in_plugin) {
|
||||||
gchar *type_name;
|
gchar *type_name;
|
||||||
gchar *codec_type;
|
gchar *p;
|
||||||
|
GstCaps *srccaps, *audiosinkcaps, *videosinkcaps;
|
||||||
|
|
||||||
if (in_plugin->decode) {
|
/* Try to find the caps that belongs here */
|
||||||
codec_type = "dec";
|
srccaps = gst_ffmpeg_formatid_to_caps (in_plugin->name);
|
||||||
}
|
if (!srccaps) {
|
||||||
else {
|
|
||||||
goto next;
|
goto next;
|
||||||
}
|
}
|
||||||
|
/* This is a bit ugly, but we just take all formats
|
||||||
|
* for the pad template. We'll get an exact match
|
||||||
|
* when we open the stream */
|
||||||
|
audiosinkcaps = NULL;
|
||||||
|
videosinkcaps = NULL;
|
||||||
|
for (in_codec = first_avcodec; in_codec != NULL;
|
||||||
|
in_codec = in_codec->next) {
|
||||||
|
GstCaps *temp = gst_ffmpeg_codecid_to_caps (in_codec->id, NULL);
|
||||||
|
if (!temp) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
switch (in_codec->type) {
|
||||||
|
case CODEC_TYPE_VIDEO:
|
||||||
|
videosinkcaps = gst_caps_append (videosinkcaps, temp);
|
||||||
|
break;
|
||||||
|
case CODEC_TYPE_AUDIO:
|
||||||
|
audiosinkcaps = gst_caps_append (audiosinkcaps, temp);
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
gst_caps_unref (temp);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/* construct the type */
|
/* construct the type */
|
||||||
type_name = g_strdup_printf("ff%s_%s", codec_type, in_plugin->name);
|
type_name = g_strdup_printf("ffmux_%s", in_plugin->name);
|
||||||
|
|
||||||
|
p = type_name;
|
||||||
|
|
||||||
|
while (*p) {
|
||||||
|
if (*p == '.') *p = '_';
|
||||||
|
p++;
|
||||||
|
}
|
||||||
|
|
||||||
/* if it's already registered, drop it */
|
/* if it's already registered, drop it */
|
||||||
if (g_type_from_name(type_name)) {
|
if (g_type_from_name(type_name)) {
|
||||||
|
|
@ -388,40 +472,50 @@ gst_ffmpegmux_register (GstPlugin *plugin)
|
||||||
goto next;
|
goto next;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* create the gtk type now */
|
/* create the type now */
|
||||||
type = g_type_register_static(GST_TYPE_ELEMENT, type_name , &typeinfo, 0);
|
type = g_type_register_static(GST_TYPE_ELEMENT, type_name , &typeinfo, 0);
|
||||||
|
|
||||||
/* construct the element details struct */
|
/* construct the element details struct */
|
||||||
details = g_new0 (GstElementDetails,1);
|
details = g_new0 (GstElementDetails, 1);
|
||||||
details->longname = g_strdup (in_plugin->name);
|
details->longname = g_strdup (in_plugin->long_name);
|
||||||
details->klass = "Codec/FFMpeg";
|
details->klass = g_strdup ("Codec/Muxer");
|
||||||
details->license = "LGPL";
|
details->license = g_strdup ("LGPL");
|
||||||
details->description = g_strdup (in_plugin->name);
|
details->description = g_strdup_printf ("FFMPEG %s Muxer",
|
||||||
details->version = g_strdup("1.0.0");
|
in_plugin->name);
|
||||||
details->author = g_strdup("The FFMPEG crew, GStreamer plugin by Wim Taymans <wim.taymans@chello.be>");
|
details->version = g_strdup (VERSION);
|
||||||
details->copyright = g_strdup("(c) 2001");
|
details->author = g_strdup ("The FFMPEG crew\n"
|
||||||
|
"Wim Taymans <wim.taymans@chello.be>\n"
|
||||||
g_hash_table_insert (global_plugins,
|
"Ronald Bultje <rbultje@ronald.bitfreak.net>");
|
||||||
GINT_TO_POINTER (type),
|
details->copyright = g_strdup ("(c) 2002-2003");
|
||||||
(gpointer) in_plugin);
|
|
||||||
|
|
||||||
/* register the plugin with gstreamer */
|
/* register the plugin with gstreamer */
|
||||||
factory = gst_element_factory_new(type_name,type,details);
|
factory = gst_element_factory_new(type_name,type,details);
|
||||||
g_return_val_if_fail(factory != NULL, FALSE);
|
g_return_val_if_fail(factory != NULL, FALSE);
|
||||||
|
|
||||||
gst_element_factory_set_rank (factory, GST_ELEMENT_RANK_NONE);
|
/* create a cache for these properties */
|
||||||
|
params = g_new0 (GstFFMpegMuxClassParams, 1);
|
||||||
|
params->in_plugin = in_plugin;
|
||||||
|
params->srctempl = gst_pad_template_new ("sink", GST_PAD_SRC,
|
||||||
|
GST_PAD_ALWAYS,
|
||||||
|
srccaps, NULL);
|
||||||
|
gst_element_factory_add_pad_template (factory,
|
||||||
|
params->srctempl);
|
||||||
|
params->audiosinktempl = gst_pad_template_new ("audio_%02d",
|
||||||
|
GST_PAD_SINK,
|
||||||
|
GST_PAD_REQUEST,
|
||||||
|
audiosinkcaps, NULL);
|
||||||
|
gst_element_factory_add_pad_template (factory,
|
||||||
|
params->audiosinktempl);
|
||||||
|
params->videosinktempl = gst_pad_template_new ("video_%02d",
|
||||||
|
GST_PAD_SINK,
|
||||||
|
GST_PAD_REQUEST,
|
||||||
|
videosinkcaps, NULL);
|
||||||
|
gst_element_factory_add_pad_template (factory,
|
||||||
|
params->videosinktempl);
|
||||||
|
|
||||||
gst_element_factory_add_pad_template (factory,
|
g_hash_table_insert (global_plugins,
|
||||||
GST_PAD_TEMPLATE_GET (gst_ffmpegmux_sink_factory));
|
GINT_TO_POINTER (type),
|
||||||
|
(gpointer) params);
|
||||||
if (in_plugin->type == CODEC_TYPE_VIDEO) {
|
|
||||||
gst_element_factory_add_pad_template (factory,
|
|
||||||
GST_PAD_TEMPLATE_GET (gst_ffmpegmux_video_src_factory));
|
|
||||||
}
|
|
||||||
else if (in_plugin->type == CODEC_TYPE_AUDIO) {
|
|
||||||
gst_element_factory_add_pad_template (factory,
|
|
||||||
GST_PAD_TEMPLATE_GET (gst_ffmpegmux_audio_src_factory));
|
|
||||||
}
|
|
||||||
|
|
||||||
/* The very last thing is to register the elementfactory with the plugin. */
|
/* The very last thing is to register the elementfactory with the plugin. */
|
||||||
gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (factory));
|
gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (factory));
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue