-/* Gnome-Streamer
- * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
+/* AVI muxer plugin for GStreamer
+ * Copyright (C) 2002 Ronald Bultje <rbultje@ronald.bitfreak.net>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* Boston, MA 02111-1307, USA.
*/
+/* based on:
+ * - the old avimuxer (by Wim Taymans)
+ * - xawtv's aviwriter (by Gerd Knorr)
+ * - mjpegtools' avilib (by Rainer Johanni)
+ * - openDML large-AVI docs
+ */
+
+#include <config.h>
#include <stdlib.h>
#include <string.h>
-#include "gstavimux.h"
+#include <gst/video/video.h>
+#include "gstavimux.h"
+#ifndef LE_FROM_GUINT16
+#define LE_FROM_GUINT16 GUINT16_FROM_LE
+#endif
+#ifndef LE_FROM_GUINT32
+#define LE_FROM_GUINT32 GUINT32_FROM_LE
+#endif
/* elementfactory information */
static GstElementDetails
gst_avimux_details =
{
- ".avi mux",
- "Mux/Video",
- "Encodes audio and video into an avi stream",
+ "Avi multiplexer",
+ "Codec/Muxer",
+ "LGPL",
+ "Muxes audio and video into an avi stream",
VERSION,
- "Wim Taymans <wim.taymans@chello.be>",
- "(C) 2000",
+ "Ronald Bultje <rbultje@ronald.bitfreak.net>",
+ "(C) 2002",
};
/* AviMux signals and args */
enum {
ARG_0,
- /* FILL ME */
+ ARG_BIGFILE,
};
-GST_PADTEMPLATE_FACTORY (src_factory,
+GST_PAD_TEMPLATE_FACTORY (src_factory,
"src",
GST_PAD_SRC,
GST_PAD_ALWAYS,
GST_CAPS_NEW (
- "sink_video",
+ "avimux_src_video",
"video/avi",
NULL
)
)
-GST_PADTEMPLATE_FACTORY (video_sink_factory,
- "video_%02d",
+GST_PAD_TEMPLATE_FACTORY (video_sink_factory,
+ "video_%d",
GST_PAD_SINK,
GST_PAD_REQUEST,
GST_CAPS_NEW (
- "sink_video",
+ "avimux_sink_video",
"video/avi",
"format", GST_PROPS_STRING ("strf_vids")
+ ),
+ GST_CAPS_NEW (
+ "avimux_sink_video",
+ "video/raw",
+ "format", GST_PROPS_LIST (
+ GST_PROPS_FOURCC (GST_MAKE_FOURCC('Y','U','Y','2')),
+ GST_PROPS_FOURCC (GST_MAKE_FOURCC('I','4','2','0')),
+ GST_PROPS_FOURCC (GST_MAKE_FOURCC('Y','4','1','P'))
+ ),
+ "width", GST_PROPS_INT_RANGE (16, 4096),
+ "height", GST_PROPS_INT_RANGE (16, 4096)
+ ),
+ GST_CAPS_NEW (
+ "avimux_sink_video",
+ "video/raw",
+ "format", GST_PROPS_FOURCC (GST_MAKE_FOURCC('R','G','B',' ')),
+ "width", GST_PROPS_INT_RANGE (16, 4096),
+ "height", GST_PROPS_INT_RANGE (16, 4096),
+ "depth", GST_PROPS_LIST(
+ GST_PROPS_INT(16),
+ GST_PROPS_INT(16),
+ GST_PROPS_INT(24),
+ GST_PROPS_INT(32)
+ ),
+ "bpp", GST_PROPS_LIST(
+ GST_PROPS_INT(15),
+ GST_PROPS_INT(16),
+ GST_PROPS_INT(24),
+ GST_PROPS_INT(32)
+ )
+ ),
+ GST_CAPS_NEW (
+ "avimux_sink_video",
+ "video/jpeg",
+ "width", GST_PROPS_INT_RANGE (16, 4096),
+ "height", GST_PROPS_INT_RANGE (16, 4096)
)
)
-GST_PADTEMPLATE_FACTORY (audio_sink_factory,
- "audio_%02d",
+GST_PAD_TEMPLATE_FACTORY (audio_sink_factory,
+ "audio_%d",
GST_PAD_SINK,
GST_PAD_REQUEST,
GST_CAPS_NEW (
- "sink_audio",
+ "avimux_sink_audio",
"video/avi",
"format", GST_PROPS_STRING ("strf_auds")
+ ),
+ GST_CAPS_NEW (
+ "avimux_sink_audio",
+ "audio/raw",
+ "format", GST_PROPS_STRING ("int"),
+ "law", GST_PROPS_INT (0),
+ "endianness", GST_PROPS_INT (G_BYTE_ORDER),
+ "signed", GST_PROPS_LIST (
+ GST_PROPS_BOOLEAN (TRUE),
+ GST_PROPS_BOOLEAN (FALSE)
+ ),
+ "width", GST_PROPS_LIST (
+ GST_PROPS_INT (8),
+ GST_PROPS_INT (16)
+ ),
+ "depth", GST_PROPS_LIST (
+ GST_PROPS_INT (8),
+ GST_PROPS_INT (16)
+ ),
+ "rate", GST_PROPS_INT_RANGE (1000, 48000),
+ "channels", GST_PROPS_INT_RANGE (1, 2)
+ ),
+ GST_CAPS_NEW (
+ "avimux_sink_audio",
+ "audio/x-mp3",
+ NULL
+ ),
+ GST_CAPS_NEW (
+ "avimux_sink_audio",
+ "application/x-ogg",
+ NULL
)
)
-static void gst_avimux_class_init (GstAviMuxClass *klass);
-static void gst_avimux_init (GstAviMux *avimux);
-
-static void gst_avimux_chain (GstPad *pad, GstBuffer *buf);
-static GstPad* gst_avimux_request_new_pad (GstElement *element, GstPadTemplate *templ, const gchar *name);
-
-static void gst_avimux_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec);
-static void gst_avimux_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec);
-
+static void gst_avimux_class_init (GstAviMuxClass *klass);
+static void gst_avimux_init (GstAviMux *avimux);
+
+static void gst_avimux_loop (GstElement *element);
+static gboolean gst_avimux_handle_event (GstPad *pad,
+ GstEvent *event);
+static GstPad* gst_avimux_request_new_pad (GstElement *element,
+ GstPadTemplate *templ,
+ const gchar *name);
+static void gst_avimux_set_property (GObject *object,
+ guint prop_id,
+ const GValue *value,
+ GParamSpec *pspec);
+static void gst_avimux_get_property (GObject *object,
+ guint prop_id,
+ GValue *value,
+ GParamSpec *pspec);
+static GstElementStateReturn gst_avimux_change_state (GstElement *element);
static GstElementClass *parent_class = NULL;
-//static guint gst_avimux_signals[LAST_SIGNAL] = { 0 };
+/*static guint gst_avimux_signals[LAST_SIGNAL] = { 0 }; */
GType
gst_avimux_get_type (void)
parent_class = g_type_class_ref (GST_TYPE_ELEMENT);
- gobject_class->set_property = gst_avimux_set_property;
- gobject_class->get_property = gst_avimux_get_property;
+ g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_BIGFILE,
+ g_param_spec_boolean("bigfile","Bigfile Support","Whether to capture large or small AVI files",
+ 0,G_PARAM_READWRITE));
gstelement_class->request_new_pad = gst_avimux_request_new_pad;
+
+ gstelement_class->change_state = gst_avimux_change_state;
+
+ gstelement_class->get_property = gst_avimux_get_property;
+ gstelement_class->set_property = gst_avimux_set_property;
+}
+
+static const GstEventMask *
+gst_avimux_get_event_masks (GstPad *pad)
+{
+ static const GstEventMask gst_avimux_src_event_masks[] = {
+ { GST_EVENT_NEW_MEDIA, 0 },
+ { 0, }
+ };
+ static const GstEventMask gst_avimux_sink_event_masks[] = {
+ { GST_EVENT_EOS, 0 },
+ { 0, }
+ };
+
+ if (GST_PAD_IS_SRC(pad))
+ return gst_avimux_src_event_masks;
+ else
+ return gst_avimux_sink_event_masks;
}
static void
gst_avimux_init (GstAviMux *avimux)
{
avimux->srcpad = gst_pad_new_from_template (
- GST_PADTEMPLATE_GET (src_factory), "src");
+ GST_PAD_TEMPLATE_GET (src_factory), "src");
gst_element_add_pad (GST_ELEMENT (avimux), avimux->srcpad);
- avimux->state = GST_AVIMUX_INITIAL;
- avimux->riff = NULL;
- avimux->num_audio_pads = 0;
- avimux->num_video_pads = 0;
- avimux->next_time = 0;
+ GST_FLAG_SET (GST_ELEMENT(avimux), GST_ELEMENT_EVENT_AWARE);
+ gst_pad_set_event_function(avimux->srcpad, gst_avimux_handle_event);
+ gst_pad_set_event_mask_function(avimux->srcpad, gst_avimux_get_event_masks);
+
+ avimux->audiosinkpad = NULL;
+ avimux->audio_pad_connected = FALSE;
+ avimux->videosinkpad = NULL;
+ avimux->video_pad_connected = FALSE;
+
+ avimux->audio_buffer_queue = NULL;
+ avimux->video_buffer_queue = NULL;
+
+ avimux->num_frames = 0;
+
+ /* audio/video/AVI header initialisation */
+ memset(&(avimux->avi_hdr),0,sizeof(gst_riff_avih));
+ memset(&(avimux->vids_hdr),0,sizeof(gst_riff_strh));
+ memset(&(avimux->vids),0,sizeof(gst_riff_strf_vids));
+ memset(&(avimux->auds_hdr),0,sizeof(gst_riff_strh));
+ memset(&(avimux->auds),0,sizeof(gst_riff_strf_auds));
+ avimux->vids_hdr.type = GST_MAKE_FOURCC('v','i','d','s');
+ avimux->vids_hdr.rate = 1000000;
+ avimux->auds_hdr.type = GST_MAKE_FOURCC('a','u','d','s');
+
+ avimux->idx = NULL;
+
+ avimux->write_header = TRUE;
+
+ avimux->enable_large_avi = TRUE;
- avimux->riff = gst_riff_encoder_new (GST_RIFF_RIFF_AVI);
- avimux->aviheader = g_malloc0 (sizeof (gst_riff_avih));
+ avimux->framerate = 0;
+
+ gst_element_set_loop_function(GST_ELEMENT(avimux), gst_avimux_loop);
}
static GstPadConnectReturn
-gst_avimux_sinkconnect (GstPad *pad, GstCaps *caps)
+gst_avimux_sinkconnect (GstPad *pad, GstCaps *vscaps)
{
GstAviMux *avimux;
- const gchar* format = gst_caps_get_string (caps, "format");
- gint padnum = GPOINTER_TO_INT (gst_pad_get_element_private (pad));
+ GstCaps *caps;
avimux = GST_AVIMUX (gst_pad_get_parent (pad));
- GST_DEBUG (0, "avimux: sinkconnect triggered on %s (%d), %s\n", gst_pad_get_name (pad),
- padnum, format);
+ /* we are not going to act on variable caps */
+ if (!GST_CAPS_IS_FIXED (vscaps))
+ return GST_PAD_CONNECT_DELAYED;
+
+ GST_DEBUG (0, "avimux: sinkconnect triggered on %s", gst_pad_get_name (pad));
+
+ for (caps = vscaps; caps != NULL; caps = vscaps = vscaps->next)
+ {
+ const gchar* mimetype = gst_caps_get_mime(caps);
+
+ if (!strcmp (mimetype, "video/avi"))
+ {
+ const gchar* format;
+
+ gst_caps_get_string (caps, "format", &format);
+
+ if (!strncmp (format, "strf_vids", 9)) {
+ avimux->vids.size = sizeof(gst_riff_strf_vids);
+ gst_caps_get (caps,
+ "width", &avimux->vids.width,
+ "height", &avimux->vids.height,
+ "planes", &avimux->vids.planes,
+ "bit_cnt", &avimux->vids.bit_cnt,
+ "compression", &avimux->vids.compression,
+ "image_size", &avimux->vids.image_size,
+ "xpels_meter", &avimux->vids.xpels_meter,
+ "ypels_meter", &avimux->vids.ypels_meter,
+ "num_colors", &avimux->vids.num_colors,
+ "imp_colors", &avimux->vids.imp_colors,
+ NULL);
+ avimux->vids_hdr.fcc_handler = avimux->vids.compression;
+ avimux->avi_hdr.width = avimux->vids.width;
+ avimux->avi_hdr.height = avimux->vids.height;
+ goto done;
+ }
+ else if (!strncmp (format, "strf_auds", 9)) {
+ gst_caps_get (caps,
+ "format", &avimux->auds.format,
+ "channels", &avimux->auds.channels,
+ "rate", &avimux->auds.rate,
+ "av_bps", &avimux->auds.av_bps,
+ "blockalign", &avimux->auds.blockalign,
+ "size", &avimux->auds.size,
+ NULL);
+ avimux->auds_hdr.samplesize = avimux->auds_hdr.scale = avimux->auds.blockalign;
+ avimux->auds_hdr.rate = avimux->auds.av_bps;
+ goto done;
+ }
+ }
+ else if (!strcmp (mimetype, "video/raw"))
+ {
+ guint32 format;
+ gint temp;
+
+ gst_caps_get_fourcc_int (caps, "format", &format);
+ switch (format)
+ {
+ case GST_MAKE_FOURCC('Y','U','Y','2'):
+ case GST_MAKE_FOURCC('I','4','2','0'):
+ case GST_MAKE_FOURCC('Y','4','1','P'):
+ case GST_MAKE_FOURCC('R','G','B',' '):
+ avimux->vids.size = sizeof(gst_riff_strf_vids);
+ gst_caps_get (caps, "width", &avimux->vids.width,
+ "height", &avimux->vids.height, NULL);
+ avimux->vids.planes = 1;
+ switch (format)
+ {
+ case GST_MAKE_FOURCC('Y','U','Y','2'):
+ avimux->vids.bit_cnt = 16; /* YUY2 */
+ break;
+ case GST_MAKE_FOURCC('R','G','B',' '):
+ gst_caps_get_int (caps, "bpp", &temp); /* RGB */
+ avimux->vids.bit_cnt = temp;
+ break;
+ case GST_MAKE_FOURCC('Y','4','1','P'):
+ case GST_MAKE_FOURCC('I','4','2','0'):
+ avimux->vids.bit_cnt = 12; /* Y41P or I420 */
+ break;
+ }
+ gst_caps_get_fourcc_int(caps, "format", &avimux->vids.compression);
+ avimux->vids_hdr.fcc_handler = avimux->vids.compression;
+ avimux->vids.image_size = avimux->vids.height * avimux->vids.width;
+ avimux->avi_hdr.width = avimux->vids.width;
+ avimux->avi_hdr.height = avimux->vids.height;
+ goto done;
+ default:
+ break;
+ }
+ }
+ else if (!strcmp (mimetype, "video/jpeg"))
+ {
+ avimux->vids.size = sizeof(gst_riff_strf_vids);
+ gst_caps_get (caps, "width", &avimux->vids.width,
+ "height", &avimux->vids.height, NULL);
+ avimux->vids.planes = 1;
+ avimux->vids.bit_cnt = 24;
+ avimux->vids_hdr.fcc_handler = avimux->vids.compression = GST_MAKE_FOURCC('M','J','P','G');
+ avimux->avi_hdr.width = avimux->vids.width;
+ avimux->avi_hdr.height = avimux->vids.height;
+ avimux->vids.image_size = avimux->vids.height * avimux->vids.width;
+ goto done;
+ }
+ else if (!strcmp (mimetype, "audio/raw"))
+ {
+ gint width;
+
+ avimux->auds.format = GST_RIFF_WAVE_FORMAT_PCM;
+ gst_caps_get (caps, "channels", &avimux->auds.channels,
+ "rate", &avimux->auds.rate,
+ "width", &width,
+ "depth", &avimux->auds.size,
+ NULL);
+ avimux->auds_hdr.rate = avimux->auds.av_bps = width * avimux->auds.rate * avimux->auds.channels / 8;
+ avimux->auds_hdr.samplesize = avimux->auds_hdr.scale = avimux->auds.blockalign = width * avimux->auds.channels/8;
+ goto done;
+ }
+ else if (!strcmp (mimetype, "audio/x-mp3"))
+ {
+ gint layer;
+
+ gst_caps_get_int(caps, "layer", &layer);
+
+ /* we don't need to do anything here, compressed mp3 contains it all */
+ avimux->auds.format = (layer == 3?
+ GST_RIFF_WAVE_FORMAT_MPEGL3 :
+ GST_RIFF_WAVE_FORMAT_MPEGL12);
+ goto done;
+ }
+ else if (!strcmp (mimetype, "application/x-ogg"))
+ {
+ avimux->auds.format = GST_RIFF_WAVE_FORMAT_VORBIS1;
+ goto done;
+ }
+ }
+ return GST_PAD_CONNECT_REFUSED;
- if (!strncmp (format, "strf_vids", 9)) {
- gst_riff_strf_vids *strf_vids = g_malloc(sizeof(gst_riff_strf_vids));
+done:
+ return GST_PAD_CONNECT_OK;
+}
- strf_vids->size = sizeof(gst_riff_strf_vids);
- strf_vids->width = gst_caps_get_int (caps, "width");
- strf_vids->height = gst_caps_get_int (caps, "height");;
- strf_vids->planes = gst_caps_get_int (caps, "planes");;
- strf_vids->bit_cnt = gst_caps_get_int (caps, "bit_cnt");;
- strf_vids->compression = gst_caps_get_fourcc_int (caps, "compression");;
- strf_vids->image_size = gst_caps_get_int (caps, "image_size");;
- strf_vids->xpels_meter = gst_caps_get_int (caps, "xpels_meter");;
- strf_vids->ypels_meter = gst_caps_get_int (caps, "ypels_meter");;
- strf_vids->num_colors = gst_caps_get_int (caps, "num_colors");;
- strf_vids->imp_colors = gst_caps_get_int (caps, "imp_colors");;
+static void
+gst_avimux_pad_connect (GstPad *pad,
+ GstPad *peer,
+ gpointer data)
+{
+ GstAviMux *avimux = GST_AVIMUX(data);
+ const gchar *padname = gst_pad_get_name (pad);
- avimux->video_header[padnum] = strf_vids;
+ if (pad == avimux->audiosinkpad)
+ {
+ avimux->audio_pad_connected = TRUE;
}
- else if (!strncmp (format, "strf_auds", 9)) {
+ else if (pad == avimux->videosinkpad)
+ {
+ avimux->video_pad_connected = TRUE;
+ }
+ else
+ {
+ g_warning("Unknown padname '%s'", padname);
+ return;
+ }
+
+ GST_DEBUG(GST_CAT_PLUGIN_INFO, "pad '%s' connected", padname);
+}
+
+static void
+gst_avimux_pad_disconnect (GstPad *pad,
+ GstPad *peer,
+ gpointer data)
+{
+ GstAviMux *avimux = GST_AVIMUX(data);
+ const gchar *padname = gst_pad_get_name (pad);
+ if (pad == avimux->audiosinkpad)
+ {
+ avimux->audio_pad_connected = FALSE;
+ avimux->audiosinkpad = NULL;
}
- return GST_PAD_CONNECT_OK;
+ else if (pad == avimux->videosinkpad)
+ {
+ avimux->video_pad_connected = FALSE;
+ avimux->videosinkpad = NULL;
+ }
+ else
+ {
+ g_warning("Unknown padname '%s'", padname);
+ return;
+ }
+
+ GST_DEBUG(GST_CAT_PLUGIN_INFO, "pad '%s' disconnected", padname);
+
+ /* is this allowed? */
+ gst_pad_destroy(pad);
}
static GstPad*
const gchar *req_name)
{
GstAviMux *avimux;
- gchar *name = NULL;
GstPad *newpad;
-
+
g_return_val_if_fail (templ != NULL, NULL);
if (templ->direction != GST_PAD_SINK) {
avimux = GST_AVIMUX (element);
- if (templ == GST_PADTEMPLATE_GET (audio_sink_factory)) {
- name = g_strdup_printf ("audio_%02d", avimux->num_audio_pads);
- newpad = gst_pad_new_from_template (templ, name);
- gst_pad_set_element_private (newpad, GINT_TO_POINTER (avimux->num_audio_pads));
-
- avimux->audio_pad[avimux->num_audio_pads] = newpad;
- avimux->num_audio_pads++;
+ if (templ == GST_PAD_TEMPLATE_GET (audio_sink_factory)) {
+ g_return_val_if_fail(avimux->audiosinkpad == NULL, NULL);
+ newpad = gst_pad_new_from_template (templ, "audio_00");
+ avimux->audiosinkpad = newpad;
}
- else if (templ == GST_PADTEMPLATE_GET (video_sink_factory)) {
- name = g_strdup_printf ("video_%02d", avimux->num_video_pads);
- newpad = gst_pad_new_from_template (templ, name);
- gst_pad_set_element_private (newpad, GINT_TO_POINTER (avimux->num_video_pads));
-
- avimux->video_pad[avimux->num_video_pads] = newpad;
- avimux->num_video_pads++;
+ else if (templ == GST_PAD_TEMPLATE_GET (video_sink_factory)) {
+ g_return_val_if_fail(avimux->videosinkpad == NULL, NULL);
+ newpad = gst_pad_new_from_template (templ, "video_00");
+ avimux->videosinkpad = newpad;
}
else {
g_warning ("avimux: this is not our template!\n");
return NULL;
}
- gst_pad_set_chain_function (newpad, gst_avimux_chain);
+ g_signal_connect(newpad, "connected",
+ G_CALLBACK(gst_avimux_pad_connect), (gpointer)avimux);
+ g_signal_connect(newpad, "disconnected",
+ G_CALLBACK(gst_avimux_pad_disconnect), (gpointer)avimux);
gst_pad_set_connect_function (newpad, gst_avimux_sinkconnect);
gst_element_add_pad (element, newpad);
+ gst_pad_set_event_function(newpad, gst_avimux_handle_event);
+ gst_pad_set_event_mask_function(newpad, gst_avimux_get_event_masks);
return newpad;
}
+/* maybe some of these functions should be moved to riff.h? */
+
+/* DISCLAIMER: this function is ugly. So be it (i.e. it makes the rest easier) */
+
+static GstBuffer *
+gst_avimux_riff_get_avi_header (GstAviMux *avimux)
+{
+ GstBuffer *buffer;
+ guint8 *buffdata;
+ guint16 temp16;
+ guint32 temp32;
+
+ buffer = gst_buffer_new();
+
+ /* first, let's see what actually needs to be in the buffer */
+ GST_BUFFER_SIZE(buffer) = 0;
+ GST_BUFFER_SIZE(buffer) += 32 + sizeof(gst_riff_avih); /* avi header */
+ if (avimux->video_pad_connected)
+ { /* we have video */
+ GST_BUFFER_SIZE(buffer) += 28 + sizeof(gst_riff_strh) + sizeof(gst_riff_strf_vids); /* vid hdr */
+ GST_BUFFER_SIZE(buffer) += 24; /* odml header */
+ }
+ if (avimux->audio_pad_connected)
+ { /* we have audio */
+ GST_BUFFER_SIZE(buffer) += 28 + sizeof(gst_riff_strh) + sizeof(gst_riff_strf_auds); /* aud hdr */
+ }
+ /* this is the "riff size" */
+ avimux->header_size = GST_BUFFER_SIZE(buffer);
+ GST_BUFFER_SIZE(buffer) += 12; /* avi data header */
+
+ /* allocate the buffer */
+ buffdata = GST_BUFFER_DATA(buffer) = g_malloc(GST_BUFFER_SIZE(buffer));
+
+ /* avi header metadata */
+ memcpy(buffdata, "RIFF", 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->header_size + avimux->idx_size + avimux->data_size);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ memcpy(buffdata, "AVI ", 4); buffdata += 4;
+ memcpy(buffdata, "LIST", 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->header_size - 4*5);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ memcpy(buffdata, "hdrl", 4); buffdata += 4;
+ memcpy(buffdata, "avih", 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(sizeof(gst_riff_avih));
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ /* the AVI header itself */
+ temp32 = LE_FROM_GUINT32(avimux->avi_hdr.us_frame);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->avi_hdr.max_bps);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->avi_hdr.pad_gran);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->avi_hdr.flags);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->avi_hdr.tot_frames);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->avi_hdr.init_frames);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->avi_hdr.streams);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->avi_hdr.bufsize);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->avi_hdr.width);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->avi_hdr.height);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->avi_hdr.scale);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->avi_hdr.rate);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->avi_hdr.start);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->avi_hdr.length);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+
+ if (avimux->video_pad_connected)
+ {
+ /* video header metadata */
+ memcpy(buffdata, "LIST", 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(sizeof(gst_riff_strh) + sizeof(gst_riff_strf_vids) + 4*5);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ memcpy(buffdata, "strl", 4); buffdata += 4;
+ /* generic header */
+ memcpy(buffdata, "strh", 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(sizeof(gst_riff_strh));
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ /* the actual header */
+ temp32 = LE_FROM_GUINT32(avimux->vids_hdr.type);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->vids_hdr.fcc_handler);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->vids_hdr.flags);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->vids_hdr.priority);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->vids_hdr.init_frames);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->vids_hdr.scale);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->vids_hdr.rate);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->vids_hdr.start);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->vids_hdr.length);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->vids_hdr.bufsize);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->vids_hdr.quality);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->vids_hdr.samplesize);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ /* the video header */
+ memcpy(buffdata, "strf", 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(sizeof(gst_riff_strf_vids));
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ /* the actual header */
+ temp32 = LE_FROM_GUINT32(avimux->vids.size);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->vids.width);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->vids.height);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp16 = LE_FROM_GUINT16(avimux->vids.planes);
+ memcpy(buffdata, &temp16, 2); buffdata += 2;
+ temp16 = LE_FROM_GUINT16(avimux->vids.bit_cnt);
+ memcpy(buffdata, &temp16, 2); buffdata += 2;
+ temp32 = LE_FROM_GUINT32(avimux->vids.compression);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->vids.image_size);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->vids.xpels_meter);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->vids.ypels_meter);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->vids.num_colors);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->vids.imp_colors);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ }
+
+ if (avimux->audio_pad_connected)
+ {
+ /* audio header */
+ memcpy(buffdata, "LIST", 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(sizeof(gst_riff_strh) + sizeof(gst_riff_strf_auds) + 4*5);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ memcpy(buffdata, "strl", 4); buffdata += 4;
+ /* generic header */
+ memcpy(buffdata, "strh", 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(sizeof(gst_riff_strh));
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ /* the actual header */
+ temp32 = LE_FROM_GUINT32(avimux->auds_hdr.type);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->auds_hdr.fcc_handler);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->auds_hdr.flags);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->auds_hdr.priority);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->auds_hdr.init_frames);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->auds_hdr.scale);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->auds_hdr.rate);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->auds_hdr.start);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->auds_hdr.length);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->auds_hdr.bufsize);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->auds_hdr.quality);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->auds_hdr.samplesize);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ /* the audio header */
+ memcpy(buffdata, "strf", 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(sizeof(gst_riff_strf_auds));
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ /* the actual header */
+ temp16 = LE_FROM_GUINT16(avimux->auds.format);
+ memcpy(buffdata, &temp16, 2); buffdata += 2;
+ temp16 = LE_FROM_GUINT16(avimux->auds.channels);
+ memcpy(buffdata, &temp16, 2); buffdata += 2;
+ temp32 = LE_FROM_GUINT32(avimux->auds.rate);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->auds.av_bps);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp16 = LE_FROM_GUINT16(avimux->auds.blockalign);
+ memcpy(buffdata, &temp16, 2); buffdata += 2;
+ temp16 = LE_FROM_GUINT16(avimux->auds.size);
+ memcpy(buffdata, &temp16, 2); buffdata += 2;
+ }
+
+ if (avimux->video_pad_connected)
+ {
+ /* odml header */
+ memcpy(buffdata, "LIST", 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(sizeof(guint32)+4*3);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ memcpy(buffdata, "odml", 4); buffdata += 4;
+ memcpy(buffdata, "dmlh", 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(sizeof(guint32));
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->total_frames);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ }
+
+ /* avi data header */
+ memcpy(buffdata, "LIST", 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(avimux->data_size);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ memcpy(buffdata, "movi", 4);
+
+ return buffer;
+}
+
+static GstBuffer *
+gst_avimux_riff_get_avix_header (guint32 datax_size)
+{
+ GstBuffer *buffer;
+ guint8 *buffdata;
+ guint32 temp32;
+
+ buffer = gst_buffer_new();
+ GST_BUFFER_SIZE(buffer) = 24;
+ buffdata = GST_BUFFER_DATA(buffer) = g_malloc(GST_BUFFER_SIZE(buffer));
+
+ memcpy(buffdata, "LIST", 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(datax_size+4*4);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ memcpy(buffdata, "AVIX", 4); buffdata += 4;
+ memcpy(buffdata, "LIST", 4); buffdata += 4;
+ temp32 = LE_FROM_GUINT32(datax_size);
+ memcpy(buffdata, &temp32, 4); buffdata += 4;
+ memcpy(buffdata, "movi", 4);
+
+ return buffer;
+}
+
+static GstBuffer *
+gst_avimux_riff_get_video_header (guint32 video_frame_size)
+{
+ GstBuffer *buffer;
+ guint32 temp32;
+
+ buffer = gst_buffer_new();
+ GST_BUFFER_DATA(buffer) = g_malloc(8);
+ GST_BUFFER_SIZE(buffer) = 8;
+ memcpy(GST_BUFFER_DATA(buffer), "00db", 4);
+ temp32 = LE_FROM_GUINT32(video_frame_size);
+ memcpy(GST_BUFFER_DATA(buffer)+4, &temp32, 4);
+
+ return buffer;
+}
+
+static GstBuffer *
+gst_avimux_riff_get_audio_header (guint32 audio_sample_size)
+{
+ GstBuffer *buffer;
+ guint32 temp32;
+
+ buffer = gst_buffer_new();
+ GST_BUFFER_DATA(buffer) = g_malloc(8);
+ GST_BUFFER_SIZE(buffer) = 8;
+ memcpy(GST_BUFFER_DATA(buffer), "01wb", 4);
+ temp32 = LE_FROM_GUINT32(audio_sample_size);
+ memcpy(GST_BUFFER_DATA(buffer)+4, &temp32, 4);
+
+ return buffer;
+}
+
+/* some other usable functions (thankyou xawtv ;-) ) */
+
+static void
+gst_avimux_add_index (GstAviMux *avimux, guchar *code, guint32 flags, guint32 size)
+{
+ if (avimux->idx_index == avimux->idx_count)
+ {
+ avimux->idx_count += 256;
+ avimux->idx = realloc(avimux->idx, avimux->idx_count*sizeof(gst_riff_index_entry));
+ }
+ memcpy(&(avimux->idx[avimux->idx_index].id), code, 4);
+ avimux->idx[avimux->idx_index].flags = LE_FROM_GUINT32(flags);
+ avimux->idx[avimux->idx_index].offset = LE_FROM_GUINT32(avimux->idx_offset);
+ avimux->idx[avimux->idx_index].size = LE_FROM_GUINT32(size);
+ avimux->idx_index++;
+ avimux->idx_offset += size;
+}
+
+static void
+gst_avimux_write_index (GstAviMux *avimux)
+{
+ GstBuffer *buffer;
+ guint32 temp32;
+
+ buffer = gst_buffer_new();
+ GST_BUFFER_SIZE(buffer) = 8;
+ GST_BUFFER_DATA(buffer) = g_malloc(8);
+ memcpy(GST_BUFFER_DATA(buffer), "idx1", 4);
+ temp32 = LE_FROM_GUINT32(avimux->idx_index * sizeof(gst_riff_index_entry));
+ memcpy(GST_BUFFER_DATA(buffer)+4, &temp32, 4);
+ gst_pad_push(avimux->srcpad, buffer);
+
+ buffer = gst_buffer_new();
+ GST_BUFFER_SIZE(buffer) = avimux->idx_index * sizeof(gst_riff_index_entry);
+ GST_BUFFER_DATA(buffer) = (unsigned char*) avimux->idx;
+ avimux->idx = NULL; /* will be free()'ed by gst_buffer_unref() */
+ avimux->total_data += GST_BUFFER_SIZE(buffer);
+ gst_pad_push(avimux->srcpad, buffer);
+
+ avimux->idx_size += avimux->idx_index * sizeof(gst_riff_index_entry) + 8;
+
+ /* update header */
+ avimux->avi_hdr.flags |= GST_RIFF_AVIH_HASINDEX;
+}
+
static void
-gst_avimux_make_header (GstAviMux *avimux)
+gst_avimux_bigfile(GstAviMux *avimux, gboolean last)
{
- gint i;
+ GstBuffer *header;
+ GstEvent *event;
+
+ if (avimux->is_bigfile)
+ {
+ /* sarch back */
+ event = gst_event_new_seek (GST_FORMAT_BYTES |
+ GST_SEEK_METHOD_SET |
+ GST_SEEK_FLAG_FLUSH,
+ avimux->avix_start);
+ /* if the event succeeds */
+ if (gst_pad_send_event(GST_PAD_PEER(avimux->srcpad), event)) {
+
+ /* rewrite AVIX header */
+ header = gst_avimux_riff_get_avix_header(avimux->datax_size);
+ gst_pad_push(avimux->srcpad, header);
+
+ /* go back to current location */
+ event = gst_event_new_seek (GST_FORMAT_BYTES |
+ GST_SEEK_METHOD_SET |
+ GST_SEEK_FLAG_FLUSH,
+ avimux->total_data);
+ gst_pad_send_event(GST_PAD_PEER(avimux->srcpad), event);
+ }
+ }
+ avimux->avix_start = avimux->total_data;
- gst_riff_strh strh;
+ if (last)
+ return;
+
+ avimux->is_bigfile = TRUE;
+ avimux->numx_frames = 0;
+ avimux->datax_size = 0;
+
+ header = gst_avimux_riff_get_avix_header(0);
+ avimux->total_data += GST_BUFFER_SIZE(header);
+ gst_pad_push(avimux->srcpad, header);
+}
- avimux->aviheader->us_frame = 40000;
- avimux->aviheader->streams = avimux->num_video_pads + avimux->num_audio_pads;
- avimux->aviheader->width = -1;
- avimux->aviheader->height = -1;
- gst_riff_encoder_avih(avimux->riff, avimux->aviheader, sizeof(gst_riff_avih));
+/* enough header blabla now, let's go on to actually writing the headers */
- memset(&strh, 0, sizeof(gst_riff_strh));
- strh.scale = 40000;
+static void
+gst_avimux_start_file (GstAviMux *avimux)
+{
+ GstBuffer *header;
+
+ avimux->total_data = 0;
+ avimux->total_frames = 0;
+ avimux->data_size = 4; /* ? */
+ avimux->datax_size = 0;
+ avimux->num_frames = 0;
+ avimux->numx_frames = 0;
+ avimux->audio_size = 0;
+ avimux->avix_start = 0;
+
+ avimux->idx_index = 0;
+ avimux->idx_offset = 0; /* see 10 lines below */
+ avimux->idx_size = 0;
+ avimux->idx_count = 0;
+ avimux->idx = NULL;
+
+ /* header */
+ avimux->avi_hdr.streams = avimux->video_pad_connected?1:0 + avimux->audio_pad_connected?1:0;
+ avimux->is_bigfile = FALSE;
+
+ header = gst_avimux_riff_get_avi_header(avimux);
+ avimux->idx_offset = avimux->header_size + 12;
+ avimux->total_data += GST_BUFFER_SIZE(header);
+ gst_pad_push(avimux->srcpad, header);
+
+ avimux->write_header = FALSE;
+ avimux->restart = FALSE;
+}
- gst_riff_encoder_strh(avimux->riff, GST_RIFF_FCC_vids, &strh, sizeof(gst_riff_strh));
+static void
+gst_avimux_stop_file (GstAviMux *avimux)
+{
+ GstEvent *event;
+ GstBuffer *header;
+
+ /* if bigfile, rewrite header, else write indexes */
+ if (avimux->video_pad_connected)
+ {
+ if (avimux->is_bigfile)
+ {
+ gst_avimux_bigfile(avimux, TRUE);
+ avimux->idx_size = 0;
+ }
+ else
+ {
+ gst_avimux_write_index(avimux);
+ }
+ }
- for (i=0; i<avimux->num_video_pads; i++) {
- gst_riff_encoder_strf(avimux->riff, avimux->video_header[i], sizeof(gst_riff_strf_vids));
+ /* statistics/total_frames/... */
+ avimux->avi_hdr.tot_frames = avimux->num_frames;
+ if (avimux->video_pad_connected)
+ avimux->vids_hdr.length = avimux->num_frames;
+ if (avimux->audio_pad_connected)
+ {
+ if (avimux->auds_hdr.scale)
+ avimux->auds_hdr.length = avimux->audio_size/avimux->auds_hdr.scale;
+ else
+ avimux->auds_hdr.length = 0; /* urm...? FIXME! ;-) */
}
+
+ /* set rate and everything having to do with that */
+ avimux->avi_hdr.us_frame = avimux->vids_hdr.scale = 1000000/avimux->framerate;
+ avimux->avi_hdr.max_bps = 0;
+ if (avimux->audio_pad_connected)
+ avimux->avi_hdr.max_bps += avimux->auds.av_bps;
+ if (avimux->video_pad_connected)
+ avimux->avi_hdr.max_bps += ((avimux->vids.bit_cnt+7)/8) *
+ avimux->framerate *
+ avimux->vids.image_size;
+
+ /* seek and rewrite the header */
+ header = gst_avimux_riff_get_avi_header(avimux);
+ event = gst_event_new_seek (GST_FORMAT_BYTES |
+ GST_SEEK_METHOD_SET |
+ GST_SEEK_FLAG_FLUSH, 0);
+ gst_pad_send_event(GST_PAD_PEER(avimux->srcpad), event);
+ gst_pad_push(avimux->srcpad, header);
+
+ avimux->write_header = TRUE;
}
static void
-gst_avimux_chain (GstPad *pad, GstBuffer *buf)
+gst_avimux_restart_file (GstAviMux *avimux)
{
- GstAviMux *avimux;
- guchar *data;
- gulong size;
- const gchar *padname;
- gint channel;
- GstBuffer *newbuf;
+ GstEvent *event;
- g_return_if_fail (pad != NULL);
- g_return_if_fail (GST_IS_PAD (pad));
- g_return_if_fail (buf != NULL);
- g_return_if_fail (GST_BUFFER_DATA (buf) != NULL);
+ gst_avimux_stop_file(avimux);
+
+ event = gst_event_new(GST_EVENT_NEW_MEDIA);
+ gst_pad_send_event(avimux->srcpad, event);
+
+ gst_avimux_start_file(avimux);
+}
+
+/* handle events (search) */
+static gboolean
+gst_avimux_handle_event (GstPad *pad, GstEvent *event)
+{
+ GstAviMux *avimux;
+ GstEventType type;
avimux = GST_AVIMUX (gst_pad_get_parent (pad));
+
+ type = event ? GST_EVENT_TYPE (event) : GST_EVENT_UNKNOWN;
- data = (guchar *)GST_BUFFER_DATA(buf);
- size = GST_BUFFER_SIZE(buf);
-
- switch(avimux->state) {
- case GST_AVIMUX_INITIAL:
- GST_DEBUG (0,"gst_avimux_chain: writing header\n");
- gst_avimux_make_header(avimux);
- newbuf = gst_riff_encoder_get_and_reset_buffer(avimux->riff);
- gst_pad_push(avimux->srcpad, newbuf);
- avimux->state = GST_AVIMUX_MOVI;
- case GST_AVIMUX_MOVI:
- padname = gst_pad_get_name (pad);
- channel = GPOINTER_TO_INT (gst_pad_get_element_private (pad));
-
- if (strncmp(padname, "audio_", 6) == 0) {
- GST_DEBUG (0,"gst_avimux_chain: got audio buffer in from channel %02d %lu\n", channel, size);
- gst_riff_encoder_chunk(avimux->riff, GST_RIFF_01wb, NULL, size);
- newbuf = gst_riff_encoder_get_and_reset_buffer(avimux->riff);
- gst_pad_push(avimux->srcpad, newbuf);
- }
- else if (strncmp(padname, "video_", 6) == 0) {
- GST_DEBUG (0,"gst_avimux_chain: got video buffer in from channel %02d %lu\n", channel, size);
- gst_riff_encoder_chunk(avimux->riff, GST_RIFF_00db, NULL, size);
- newbuf = gst_riff_encoder_get_and_reset_buffer(avimux->riff);
- GST_DEBUG (0,"gst_avimux_chain: encoded %u\n", GST_BUFFER_SIZE(newbuf));
- gst_pad_push(avimux->srcpad, newbuf);
- }
- GST_BUFFER_SIZE(buf) = (GST_BUFFER_SIZE(buf)+1)&~1;
- gst_pad_push(avimux->srcpad, buf);
+ switch (type) {
+ case GST_EVENT_NEW_MEDIA:
+ avimux->restart = TRUE;
+ break;
+ case GST_EVENT_EOS:
+ /* is this allright? */
+ if (pad == avimux->videosinkpad)
+ avimux->video_pad_eos = TRUE;
+ else if (pad == avimux->audiosinkpad)
+ avimux->audio_pad_eos = TRUE;
+ else
+ g_warning("Unknown pad for EOS!");
break;
default:
break;
}
- //gst_buffer_unref(buf);
+ return TRUE;
}
-static void
-gst_avimux_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec)
+
+/* fill the internal queue for each available pad */
+static void
+gst_avimux_fill_queue (GstAviMux *avimux)
+{
+ GstBuffer *buffer;
+
+ if (!avimux->audio_buffer_queue &&
+ GST_PAD_IS_USABLE(avimux->audiosinkpad) &&
+ !avimux->audio_pad_eos)
+ {
+ while (1)
+ {
+ buffer = gst_pad_pull(avimux->audiosinkpad);
+ if (GST_IS_EVENT(buffer))
+ {
+ gst_avimux_handle_event(avimux->audiosinkpad, GST_EVENT(buffer));
+ }
+ else
+ {
+ avimux->audio_buffer_queue = buffer;
+ break;
+ }
+ }
+ }
+
+ if (!avimux->video_buffer_queue &&
+ GST_PAD_IS_USABLE(avimux->videosinkpad) &&
+ !avimux->video_pad_eos)
+ {
+ while (1)
+ {
+ buffer = gst_pad_pull(avimux->videosinkpad);
+ if (GST_IS_EVENT(buffer))
+ {
+ gst_avimux_handle_event(avimux->videosinkpad, GST_EVENT(buffer));
+ }
+ else
+ {
+ avimux->video_buffer_queue = buffer;
+ if (avimux->framerate < 0)
+ avimux->framerate = gst_video_frame_rate(GST_PAD_PEER(avimux->videosinkpad));
+ break;
+ }
+ }
+ }
+}
+
+
+/* send extra 'padding' data */
+static void
+gst_avimux_send_pad_data (GstAviMux *avimux,
+ gulong num_bytes)
+{
+ GstBuffer *buffer;
+
+ buffer = gst_buffer_new();
+ GST_BUFFER_SIZE(buffer) = num_bytes;
+ GST_BUFFER_DATA(buffer) = g_malloc(num_bytes);
+ memset(GST_BUFFER_DATA(buffer), 0, num_bytes);
+
+ gst_pad_push(avimux->srcpad, buffer);
+}
+
+/* do audio buffer */
+static void
+gst_avimux_do_audio_buffer (GstAviMux *avimux)
+{
+ GstBuffer *data = avimux->audio_buffer_queue, *header;
+ gulong total_size, pad_bytes;
+
+ /* write a audio header + index entry */
+ header = gst_avimux_riff_get_audio_header((GST_BUFFER_SIZE(data)+3)&~3);
+ total_size = GST_BUFFER_SIZE(header) + GST_BUFFER_SIZE(data);
+ avimux->total_data += total_size;
+
+ if (avimux->is_bigfile)
+ {
+ avimux->datax_size += total_size;
+ }
+ else
+ {
+ avimux->data_size += total_size;
+ avimux->audio_size += GST_BUFFER_SIZE(data);
+ gst_avimux_add_index(avimux, "01wb", 0x0, total_size);
+ }
+
+ gst_pad_push(avimux->srcpad, header);
+ pad_bytes = ((GST_BUFFER_SIZE(data)+3)&~3) - GST_BUFFER_SIZE(data);
+ if (pad_bytes)
+ if (GST_BUFFER_MAXSIZE(data) >= GST_BUFFER_SIZE(data) + pad_bytes)
+ {
+ GST_BUFFER_SIZE(data) += pad_bytes;
+ pad_bytes = 0;
+ }
+ gst_pad_push(avimux->srcpad, data);
+ if (pad_bytes)
+ gst_avimux_send_pad_data(avimux, pad_bytes);
+ avimux->audio_buffer_queue = NULL;
+}
+
+
+/* do video buffer */
+static void
+gst_avimux_do_video_buffer (GstAviMux *avimux)
+{
+ GstBuffer *data = avimux->video_buffer_queue, *header;
+ gulong total_size, pad_bytes;
+
+ if (avimux->restart)
+ gst_avimux_restart_file(avimux);
+
+ /* write a video header + index entry */
+ if ((avimux->is_bigfile?avimux->datax_size:avimux->data_size)+GST_BUFFER_SIZE(data)>1024*1024*2000)
+ {
+ if (avimux->enable_large_avi)
+ gst_avimux_bigfile(avimux, FALSE);
+ else
+ gst_avimux_restart_file(avimux);
+ }
+
+ header = gst_avimux_riff_get_video_header((GST_BUFFER_SIZE(data)+3)&~3);
+ total_size = GST_BUFFER_SIZE(header) + GST_BUFFER_SIZE(data);
+ avimux->total_data += total_size;
+ avimux->total_frames++;
+
+ if (avimux->is_bigfile)
+ {
+ avimux->datax_size += total_size;
+ avimux->numx_frames++;
+ }
+ else
+ {
+ avimux->data_size += total_size;
+ avimux->num_frames++;
+ gst_avimux_add_index(avimux, "00db", 0x12, total_size);
+ }
+
+ gst_pad_push(avimux->srcpad, header);
+ pad_bytes = ((GST_BUFFER_SIZE(data)+3)&~3) - GST_BUFFER_SIZE(data);
+ if (pad_bytes)
+ if (GST_BUFFER_MAXSIZE(data) >= GST_BUFFER_SIZE(data) + pad_bytes)
+ {
+ GST_BUFFER_SIZE(data) += pad_bytes;
+ pad_bytes = 0;
+ }
+ gst_pad_push(avimux->srcpad, data);
+ if (pad_bytes)
+ gst_avimux_send_pad_data(avimux, pad_bytes);
+ avimux->video_buffer_queue = NULL;
+}
+
+
+/* take the oldest buffer in our internal queue and push-it */
+static gboolean
+gst_avimux_do_one_buffer (GstAviMux *avimux)
{
- GstAviMux *src;
+ if (avimux->video_buffer_queue &&
+ avimux->audio_buffer_queue)
+ {
+ if (GST_BUFFER_TIMESTAMP(avimux->video_buffer_queue) <=
+ GST_BUFFER_TIMESTAMP(avimux->audio_buffer_queue))
+ gst_avimux_do_video_buffer(avimux);
+ else
+ gst_avimux_do_audio_buffer(avimux);
+ }
+ else if (avimux->video_buffer_queue ||
+ avimux->audio_buffer_queue)
+ {
+ if (avimux->video_buffer_queue)
+ gst_avimux_do_video_buffer(avimux);
+ else
+ gst_avimux_do_audio_buffer(avimux);
+ }
+ else
+ return FALSE;
+
+ return TRUE;
+}
+
+
+static void
+gst_avimux_loop (GstElement *element)
+{
+ GstAviMux *avimux;
+
+ avimux = GST_AVIMUX(element);
+
+ /* first fill queue (some elements only set caps when
+ * flowing data), then write header */
+ gst_avimux_fill_queue(avimux);
+
+ if (avimux->write_header)
+ gst_avimux_start_file(avimux);
+
+ gst_avimux_do_one_buffer(avimux);
+}
+
+static void
+gst_avimux_get_property (GObject *object,
+ guint prop_id,
+ GValue *value,
+ GParamSpec *pspec)
+{
+ GstAviMux *avimux;
/* it's not null if we got it, but it might not be ours */
g_return_if_fail(GST_IS_AVIMUX(object));
- src = GST_AVIMUX(object);
+ avimux = GST_AVIMUX(object);
- switch(prop_id) {
+ switch (prop_id)
+ {
+ case ARG_BIGFILE:
+ g_value_set_boolean(value, avimux->enable_large_avi);
+ break;
default:
+ G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;
}
}
-static void
-gst_avimux_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec)
+static void
+gst_avimux_set_property (GObject *object,
+ guint prop_id,
+ const GValue *value,
+ GParamSpec *pspec)
{
- GstAviMux *src;
+ GstAviMux *avimux;
/* it's not null if we got it, but it might not be ours */
g_return_if_fail(GST_IS_AVIMUX(object));
- src = GST_AVIMUX(object);
+ avimux = GST_AVIMUX(object);
- switch(prop_id) {
+ switch (prop_id)
+ {
+ case ARG_BIGFILE:
+ avimux->enable_large_avi = g_value_get_boolean(value);
+ break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;
}
}
+static GstElementStateReturn
+gst_avimux_change_state (GstElement *element)
+{
+ GstAviMux *avimux;
+ gint transition = GST_STATE_TRANSITION (element);
+
+ g_return_val_if_fail(GST_IS_AVIMUX(element), GST_STATE_FAILURE);
+
+ avimux = GST_AVIMUX(element);
+
+ switch (transition) {
+ case GST_STATE_READY_TO_PAUSED:
+ break;
+ case GST_STATE_PAUSED_TO_PLAYING:
+ avimux->framerate = -1; /* means that we fill it in later */
+ avimux->video_pad_eos = avimux->audio_pad_eos = FALSE;
+ break;
+ case GST_STATE_PLAYING_TO_PAUSED:
+ /* this function returns TRUE while it handles buffers */
+ while (gst_avimux_do_one_buffer(avimux));
+ gst_avimux_stop_file(avimux);
+ break;
+ case GST_STATE_PAUSED_TO_READY:
+ break;
+ }
+
+ if (GST_ELEMENT_CLASS (parent_class)->change_state)
+ return GST_ELEMENT_CLASS (parent_class)->change_state (element);
+
+ return GST_STATE_SUCCESS;
+}
+
static gboolean
plugin_init (GModule *module, GstPlugin *plugin)
{
GstElementFactory *factory;
- /* this filter needs the riff parser */
- if (!gst_library_load ("gstriff")) {
- gst_info ("avimux: could not load support library: 'gstriff'\n");
+ if (!gst_library_load("gstvideo"))
return FALSE;
- }
/* create an elementfactory for the avimux element */
- factory = gst_elementfactory_new ("avimux", GST_TYPE_AVIMUX,
- &gst_avimux_details);
+ factory = gst_element_factory_new ("avimux", GST_TYPE_AVIMUX,
+ &gst_avimux_details);
g_return_val_if_fail (factory != NULL, FALSE);
- gst_elementfactory_add_padtemplate (factory, GST_PADTEMPLATE_GET (src_factory));
- gst_elementfactory_add_padtemplate (factory, GST_PADTEMPLATE_GET (audio_sink_factory));
- gst_elementfactory_add_padtemplate (factory, GST_PADTEMPLATE_GET (video_sink_factory));
+ gst_element_factory_add_pad_template (factory, GST_PAD_TEMPLATE_GET (src_factory));
+ gst_element_factory_add_pad_template (factory, GST_PAD_TEMPLATE_GET (audio_sink_factory));
+ gst_element_factory_add_pad_template (factory, GST_PAD_TEMPLATE_GET (video_sink_factory));
gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (factory));
"avimux",
plugin_init
};
-