diff -r bc39b352897e -r 69c7080681bf gst_plugins_good/gst/avi/gstavimux.c --- a/gst_plugins_good/gst/avi/gstavimux.c Fri Jul 09 16:26:45 2010 -0500 +++ /dev/null Thu Jan 01 00:00:00 1970 +0000 @@ -1,1982 +0,0 @@ -/* AVI muxer plugin for GStreamer - * Copyright (C) 2002 Ronald Bultje - * (C) 2006 Mark Nauwelaerts - * - * This library is free software; you can redistribute it and/or - * modify it under the terms of the GNU Library General Public - * License as published by the Free Software Foundation; either - * version 2 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 - * Library General Public License for more details. - * - * You should have received a copy of the GNU Library General Public - * License along with this library; if not, write to the - * Free Software Foundation, Inc., 59 Temple Place - Suite 330, - * 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 - */ - -/** - * SECTION:element-avimux - * - * Muxes raw or compressed audio and/or video streams into an AVI file. - * - * - * Example launch lines - * (write everything in one line, without the backslash characters) - * |[ - * gst-launch videotestsrc num-buffers=250 \ - * ! 'video/x-raw-yuv,format=(fourcc)I420,width=320,height=240,framerate=(fraction)25/1' \ - * ! queue ! mux. \ - * audiotestsrc num-buffers=440 ! audioconvert \ - * ! 'audio/x-raw-int,rate=44100,channels=2' ! queue ! mux. \ - * avimux name=mux ! filesink location=test.avi - * ]| This will create an .AVI file containing an uncompressed video stream - * with a test picture and an uncompressed audio stream containing a - * test sound. - * |[ - * gst-launch videotestsrc num-buffers=250 \ - * ! 'video/x-raw-yuv,format=(fourcc)I420,width=320,height=240,framerate=(fraction)25/1' \ - * ! xvidenc ! queue ! mux. \ - * audiotestsrc num-buffers=440 ! audioconvert ! 'audio/x-raw-int,rate=44100,channels=2' \ - * ! lame ! queue ! mux. \ - * avimux name=mux ! filesink location=test.avi - * ]| This will create an .AVI file containing the same test video and sound - * as above, only that both streams will be compressed this time. This will - * only work if you have the necessary encoder elements installed of course. - * - */ - -#ifdef HAVE_CONFIG_H -#include "config.h" -#endif - -#include "gst/gst-i18n-plugin.h" -#include -#include - -#include - -#include "gstavimux.h" - -GST_DEBUG_CATEGORY_STATIC (avimux_debug); -#define GST_CAT_DEFAULT avimux_debug - -enum -{ - ARG_0, - ARG_BIGFILE -}; - -#define DEFAULT_BIGFILE TRUE - -static const GstElementDetails gst_avi_mux_details = -GST_ELEMENT_DETAILS ("Avi muxer", - "Codec/Muxer", - "Muxes audio and video into an avi stream", - "Ronald Bultje "); - -static GstStaticPadTemplate src_factory = GST_STATIC_PAD_TEMPLATE ("src", - GST_PAD_SRC, - GST_PAD_ALWAYS, - GST_STATIC_CAPS ("video/x-msvideo") - ); - -static GstStaticPadTemplate video_sink_factory = - GST_STATIC_PAD_TEMPLATE ("video_%d", - GST_PAD_SINK, - GST_PAD_REQUEST, - GST_STATIC_CAPS ("video/x-raw-yuv, " - "format = (fourcc) { YUY2, I420 }, " - "width = (int) [ 16, 4096 ], " - "height = (int) [ 16, 4096 ], " - "framerate = (fraction) [ 0, MAX ]; " - "image/jpeg, " - "width = (int) [ 16, 4096 ], " - "height = (int) [ 16, 4096 ], " - "framerate = (fraction) [ 0, MAX ]; " - "video/x-divx, " - "width = (int) [ 16, 4096 ], " - "height = (int) [ 16, 4096 ], " - "framerate = (fraction) [ 0, MAX ], " - "divxversion = (int) [ 3, 5 ]; " - "video/x-xvid, " - "width = (int) [ 16, 4096 ], " - "height = (int) [ 16, 4096 ], " - "framerate = (fraction) [ 0, MAX ]; " - "video/x-3ivx, " - "width = (int) [ 16, 4096 ], " - "height = (int) [ 16, 4096 ], " - "framerate = (fraction) [ 0, MAX ]; " - "video/x-msmpeg, " - "width = (int) [ 16, 4096 ], " - "height = (int) [ 16, 4096 ], " - "framerate = (fraction) [ 0, MAX ], " - "msmpegversion = (int) [ 41, 43 ]; " - "video/mpeg, " - "width = (int) [ 16, 4096 ], " - "height = (int) [ 16, 4096 ], " - "framerate = (fraction) [ 0, MAX ], " - "mpegversion = (int) { 1, 2, 4}, " - "systemstream = (boolean) FALSE; " - "video/x-h263, " - "width = (int) [ 16, 4096 ], " - "height = (int) [ 16, 4096 ], " - "framerate = (fraction) [ 0, MAX ]; " - "video/x-h264, " - "width = (int) [ 16, 4096 ], " - "height = (int) [ 16, 4096 ], " - "framerate = (fraction) [ 0, MAX ]; " - "video/x-dv, " - "width = (int) 720, " - "height = (int) { 576, 480 }, " - "framerate = (fraction) [ 0, MAX ], " - "systemstream = (boolean) FALSE; " - "video/x-huffyuv, " - "width = (int) [ 16, 4096 ], " - "height = (int) [ 16, 4096 ], " "framerate = (fraction) [ 0, MAX ];" - "video/x-dirac, " - "width = (int) [ 16, 4096 ], " - "height = (int) [ 16, 4096 ], " "framerate = (fraction) [ 0, MAX ];" - "video/x-wmv, " - "width = (int) [ 16, 4096 ], " - "height = (int) [ 16, 4096 ], " "framerate = (fraction) [ 0, MAX ], " - "wmvversion = (int) [ 1, 3]") - ); - -static GstStaticPadTemplate audio_sink_factory = - GST_STATIC_PAD_TEMPLATE ("audio_%d", - GST_PAD_SINK, - GST_PAD_REQUEST, - GST_STATIC_CAPS ("audio/x-raw-int, " - "endianness = (int) LITTLE_ENDIAN, " - "signed = (boolean) { TRUE, FALSE }, " - "width = (int) { 8, 16 }, " - "depth = (int) { 8, 16 }, " - "rate = (int) [ 1000, 96000 ], " - "channels = (int) [ 1, 2 ]; " - "audio/mpeg, " - "mpegversion = (int) 1, " - "layer = (int) [ 1, 3 ], " - "rate = (int) [ 1000, 96000 ], " "channels = (int) [ 1, 2 ]; " - "audio/mpeg, " - "mpegversion = (int) 4, " - "rate = (int) [ 1000, 96000 ], " "channels = (int) [ 1, 2 ]; " -/*#if 0 VC6 doesn't support #if here ... - "audio/x-vorbis, " - "rate = (int) [ 1000, 96000 ], " "channels = (int) [ 1, 2 ]; " -#endif*/ - "audio/x-ac3, " - "rate = (int) [ 1000, 96000 ], " "channels = (int) [ 1, 2 ]; " - "audio/x-alaw, " - "rate = (int) [ 1000, 48000 ], " "channels = (int) [ 1, 2 ]; " - "audio/x-mulaw, " - "rate = (int) [ 1000, 48000 ], " "channels = (int) [ 1, 2 ]; " - "audio/x-wma, " - "rate = (int) [ 1000, 96000 ], " "channels = (int) [ 1, 2 ], " - "wmaversion = (int) [ 1, 2 ] ") - ); - -static void gst_avi_mux_base_init (gpointer g_class); -static void gst_avi_mux_class_init (GstAviMuxClass * klass); -static void gst_avi_mux_init (GstAviMux * avimux); -static void gst_avi_mux_pad_reset (GstAviPad * avipad, gboolean free); - -static GstFlowReturn gst_avi_mux_collect_pads (GstCollectPads * pads, - GstAviMux * avimux); -static gboolean gst_avi_mux_handle_event (GstPad * pad, GstEvent * event); -static GstPad *gst_avi_mux_request_new_pad (GstElement * element, - GstPadTemplate * templ, const gchar * name); -static void gst_avi_mux_release_pad (GstElement * element, GstPad * pad); -static void gst_avi_mux_set_property (GObject * object, - guint prop_id, const GValue * value, GParamSpec * pspec); -static void gst_avi_mux_get_property (GObject * object, - guint prop_id, GValue * value, GParamSpec * pspec); -static GstStateChangeReturn gst_avi_mux_change_state (GstElement * element, - GstStateChange transition); - -static GstElementClass *parent_class = NULL; - -GType -gst_avi_mux_get_type (void) -{ - static GType avimux_type = 0; - - if (!avimux_type) { - static const GTypeInfo avimux_info = { - sizeof (GstAviMuxClass), - gst_avi_mux_base_init, - NULL, - (GClassInitFunc) gst_avi_mux_class_init, - NULL, - NULL, - sizeof (GstAviMux), - 0, - (GInstanceInitFunc) gst_avi_mux_init, - }; - static const GInterfaceInfo tag_setter_info = { - NULL, - NULL, - NULL - }; - - avimux_type = - g_type_register_static (GST_TYPE_ELEMENT, "GstAviMux", &avimux_info, 0); - g_type_add_interface_static (avimux_type, GST_TYPE_TAG_SETTER, - &tag_setter_info); - } - return avimux_type; -} - -static void -gst_avi_mux_base_init (gpointer g_class) -{ - GstElementClass *element_class = GST_ELEMENT_CLASS (g_class); - - gst_element_class_add_pad_template (element_class, - gst_static_pad_template_get (&src_factory)); - gst_element_class_add_pad_template (element_class, - gst_static_pad_template_get (&audio_sink_factory)); - gst_element_class_add_pad_template (element_class, - gst_static_pad_template_get (&video_sink_factory)); - - gst_element_class_set_details (element_class, &gst_avi_mux_details); - - GST_DEBUG_CATEGORY_INIT (avimux_debug, "avimux", 0, "Muxer for AVI streams"); -} - -static void -gst_avi_mux_finalize (GObject * object) -{ - GstAviMux *mux = GST_AVI_MUX (object); - GSList *node; - - /* completely free each sinkpad */ - node = mux->sinkpads; - while (node) { - GstAviPad *avipad = (GstAviPad *) node->data; - - node = node->next; - - gst_avi_mux_pad_reset (avipad, TRUE); - g_free (avipad); - } - g_slist_free (mux->sinkpads); - mux->sinkpads = NULL; - - g_free (mux->idx); - mux->idx = NULL; - - gst_object_unref (mux->collect); - - G_OBJECT_CLASS (parent_class)->finalize (object); -} - -static void -gst_avi_mux_class_init (GstAviMuxClass * klass) -{ - GObjectClass *gobject_class; - GstElementClass *gstelement_class; - - gobject_class = (GObjectClass *) klass; - gstelement_class = (GstElementClass *) klass; - - parent_class = g_type_class_peek_parent (klass); - - gobject_class->get_property = gst_avi_mux_get_property; - gobject_class->set_property = gst_avi_mux_set_property; - gobject_class->finalize = gst_avi_mux_finalize; - - g_object_class_install_property (gobject_class, ARG_BIGFILE, - g_param_spec_boolean ("bigfile", "Bigfile Support (>2GB)", - "Support for openDML-2.0 (big) AVI files", DEFAULT_BIGFILE, - G_PARAM_READWRITE)); - - gstelement_class->request_new_pad = - GST_DEBUG_FUNCPTR (gst_avi_mux_request_new_pad); - gstelement_class->release_pad = GST_DEBUG_FUNCPTR (gst_avi_mux_release_pad); - gstelement_class->change_state = GST_DEBUG_FUNCPTR (gst_avi_mux_change_state); -} - -/* reset pad to initial state - * free - if true, release all, not only stream related, data */ -static void -gst_avi_mux_pad_reset (GstAviPad * avipad, gboolean free) -{ - /* generic part */ - memset (&(avipad->hdr), 0, sizeof (gst_riff_strh)); - - memset (&(avipad->idx[0]), 0, sizeof (avipad->idx)); - - if (free) { - g_free (avipad->tag); - avipad->tag = NULL; - g_free (avipad->idx_tag); - avipad->idx_tag = NULL; - } - - if (avipad->is_video) { - GstAviVideoPad *vidpad = (GstAviVideoPad *) avipad; - - avipad->hdr.type = GST_MAKE_FOURCC ('v', 'i', 'd', 's'); - if (vidpad->vids_codec_data) { - gst_buffer_unref (vidpad->vids_codec_data); - vidpad->vids_codec_data = NULL; - } - - memset (&(vidpad->vids), 0, sizeof (gst_riff_strf_vids)); - memset (&(vidpad->vprp), 0, sizeof (gst_riff_vprp)); - } else { - GstAviAudioPad *audpad = (GstAviAudioPad *) avipad; - - avipad->hdr.type = GST_MAKE_FOURCC ('a', 'u', 'd', 's'); - if (audpad->auds_codec_data) { - gst_buffer_unref (audpad->auds_codec_data); - audpad->auds_codec_data = NULL; - } - - memset (&(audpad->auds), 0, sizeof (gst_riff_strf_auds)); - } -} - -static void -gst_avi_mux_reset (GstAviMux * avimux) -{ - GSList *node, *newlist = NULL; - - /* free and reset each sinkpad */ - node = avimux->sinkpads; - while (node) { - GstAviPad *avipad = (GstAviPad *) node->data; - - node = node->next; - - gst_avi_mux_pad_reset (avipad, FALSE); - /* if this pad has collectdata, keep it, otherwise dump it completely */ - if (avipad->collect) - newlist = g_slist_append (newlist, avipad); - else { - gst_avi_mux_pad_reset (avipad, TRUE); - g_free (avipad); - } - } - - /* free the old list of sinkpads, only keep the real collecting ones */ - g_slist_free (avimux->sinkpads); - avimux->sinkpads = newlist; - - /* avi data */ - avimux->num_frames = 0; - memset (&(avimux->avi_hdr), 0, sizeof (gst_riff_avih)); - avimux->avi_hdr.max_bps = 10000000; - avimux->codec_data_size = 0; - - if (avimux->tags_snap) { - gst_tag_list_free (avimux->tags_snap); - avimux->tags_snap = NULL; - } - - g_free (avimux->idx); - avimux->idx = NULL; - - /* state info */ - avimux->write_header = TRUE; - - /* tags */ - gst_tag_setter_reset_tags (GST_TAG_SETTER (avimux)); -} - -static void -gst_avi_mux_init (GstAviMux * avimux) -{ - avimux->srcpad = gst_pad_new_from_static_template (&src_factory, "src"); - gst_pad_use_fixed_caps (avimux->srcpad); - gst_element_add_pad (GST_ELEMENT (avimux), avimux->srcpad); - - /* property */ - avimux->enable_large_avi = DEFAULT_BIGFILE; - - avimux->collect = gst_collect_pads_new (); - gst_collect_pads_set_function (avimux->collect, - (GstCollectPadsFunction) (GST_DEBUG_FUNCPTR (gst_avi_mux_collect_pads)), - avimux); - - /* set to clean state */ - gst_avi_mux_reset (avimux); -} - -static gboolean -gst_avi_mux_vidsink_set_caps (GstPad * pad, GstCaps * vscaps) -{ - GstAviMux *avimux; - GstAviVideoPad *avipad; - GstAviCollectData *collect_pad; - GstStructure *structure; - const gchar *mimetype; - const GValue *fps, *par; - const GValue *codec_data; - gint width, height; - gint par_n, par_d; - - avimux = GST_AVI_MUX (gst_pad_get_parent (pad)); - - /* find stream data */ - collect_pad = (GstAviCollectData *) gst_pad_get_element_private (pad); - g_assert (collect_pad); - avipad = (GstAviVideoPad *) collect_pad->avipad; - g_assert (avipad); - g_assert (avipad->parent.is_video); - g_assert (avipad->parent.hdr.type == GST_MAKE_FOURCC ('v', 'i', 'd', 's')); - - GST_DEBUG_OBJECT (avimux, "%s:%s, caps=%" GST_PTR_FORMAT, - GST_DEBUG_PAD_NAME (pad), vscaps); - - structure = gst_caps_get_structure (vscaps, 0); - mimetype = gst_structure_get_name (structure); - - /* global */ - avipad->vids.size = sizeof (gst_riff_strf_vids); - avipad->vids.planes = 1; - if (!gst_structure_get_int (structure, "width", &width) || - !gst_structure_get_int (structure, "height", &height)) { - goto refuse_caps; - } - - avipad->vids.width = width; - avipad->vids.height = height; - - fps = gst_structure_get_value (structure, "framerate"); - if (fps == NULL || !GST_VALUE_HOLDS_FRACTION (fps)) - goto refuse_caps; - - avipad->parent.hdr.rate = gst_value_get_fraction_numerator (fps); - avipad->parent.hdr.scale = gst_value_get_fraction_denominator (fps); - - /* (pixel) aspect ratio data, if any */ - par = gst_structure_get_value (structure, "pixel-aspect-ratio"); - /* only use video properties header if there is non-trivial aspect info */ - if (par && GST_VALUE_HOLDS_FRACTION (par) && - ((par_n = gst_value_get_fraction_numerator (par)) != - (par_d = gst_value_get_fraction_denominator (par)))) { - GValue to_ratio = { 0, }; - guint ratio_n, ratio_d; - - /* some fraction voodoo to obtain simplest possible ratio */ - g_value_init (&to_ratio, GST_TYPE_FRACTION); - gst_value_set_fraction (&to_ratio, width * par_n, height * par_d); - ratio_n = gst_value_get_fraction_numerator (&to_ratio); - ratio_d = gst_value_get_fraction_denominator (&to_ratio); - GST_DEBUG_OBJECT (avimux, "generating vprp data with aspect ratio %d/%d", - ratio_n, ratio_d); - /* simply fill in */ - avipad->vprp.vert_rate = avipad->parent.hdr.rate / avipad->parent.hdr.scale; - avipad->vprp.hor_t_total = width; - avipad->vprp.vert_lines = height; - avipad->vprp.aspect = (ratio_n) << 16 | (ratio_d & 0xffff); - avipad->vprp.width = width; - avipad->vprp.height = height; - avipad->vprp.fields = 1; - avipad->vprp.field_info[0].compressed_bm_height = height; - avipad->vprp.field_info[0].compressed_bm_width = width; - avipad->vprp.field_info[0].valid_bm_height = height; - avipad->vprp.field_info[0].valid_bm_width = width; - } - - /* codec initialization data, if any */ - codec_data = gst_structure_get_value (structure, "codec_data"); - if (codec_data) { - avipad->vids_codec_data = gst_value_get_buffer (codec_data); - gst_buffer_ref (avipad->vids_codec_data); - /* keep global track of size */ - avimux->codec_data_size += GST_BUFFER_SIZE (avipad->vids_codec_data); - } - - if (!strcmp (mimetype, "video/x-raw-yuv")) { - guint32 format; - - gst_structure_get_fourcc (structure, "format", &format); - avipad->vids.compression = format; - switch (format) { - case GST_MAKE_FOURCC ('Y', 'U', 'Y', '2'): - avipad->vids.bit_cnt = 16; - break; - case GST_MAKE_FOURCC ('I', '4', '2', '0'): - avipad->vids.bit_cnt = 12; - break; - } - } else { - avipad->vids.bit_cnt = 24; - avipad->vids.compression = 0; - - /* find format */ - if (!strcmp (mimetype, "video/x-huffyuv")) { - avipad->vids.compression = GST_MAKE_FOURCC ('H', 'F', 'Y', 'U'); - } else if (!strcmp (mimetype, "image/jpeg")) { - avipad->vids.compression = GST_MAKE_FOURCC ('M', 'J', 'P', 'G'); - } else if (!strcmp (mimetype, "video/x-divx")) { - gint divxversion; - - gst_structure_get_int (structure, "divxversion", &divxversion); - switch (divxversion) { - case 3: - avipad->vids.compression = GST_MAKE_FOURCC ('D', 'I', 'V', '3'); - break; - case 4: - avipad->vids.compression = GST_MAKE_FOURCC ('D', 'I', 'V', 'X'); - break; - case 5: - avipad->vids.compression = GST_MAKE_FOURCC ('D', 'X', '5', '0'); - break; - } - } else if (!strcmp (mimetype, "video/x-xvid")) { - avipad->vids.compression = GST_MAKE_FOURCC ('X', 'V', 'I', 'D'); - } else if (!strcmp (mimetype, "video/x-3ivx")) { - avipad->vids.compression = GST_MAKE_FOURCC ('3', 'I', 'V', '2'); - } else if (gst_structure_has_name (structure, "video/x-msmpeg")) { - gint msmpegversion; - - gst_structure_get_int (structure, "msmpegversion", &msmpegversion); - switch (msmpegversion) { - case 41: - avipad->vids.compression = GST_MAKE_FOURCC ('M', 'P', 'G', '4'); - break; - case 42: - avipad->vids.compression = GST_MAKE_FOURCC ('M', 'P', '4', '2'); - break; - case 43: - avipad->vids.compression = GST_MAKE_FOURCC ('M', 'P', '4', '3'); - break; - default: - GST_INFO ("unhandled msmpegversion : %d, fall back to fourcc=MPEG", - msmpegversion); - avipad->vids.compression = GST_MAKE_FOURCC ('M', 'P', 'E', 'G'); - break; - } - } else if (!strcmp (mimetype, "video/x-dv")) { - avipad->vids.compression = GST_MAKE_FOURCC ('D', 'V', 'S', 'D'); - } else if (!strcmp (mimetype, "video/x-h263")) { - avipad->vids.compression = GST_MAKE_FOURCC ('H', '2', '6', '3'); - } else if (!strcmp (mimetype, "video/x-h264")) { - avipad->vids.compression = GST_MAKE_FOURCC ('H', '2', '6', '4'); - } else if (!strcmp (mimetype, "video/mpeg")) { - gint mpegversion; - - gst_structure_get_int (structure, "mpegversion", &mpegversion); - - switch (mpegversion) { - case 2: - avipad->vids.compression = GST_MAKE_FOURCC ('M', 'P', 'G', '2'); - break; - case 4: - /* mplayer/ffmpeg might not work with DIVX, but with FMP4 */ - avipad->vids.compression = GST_MAKE_FOURCC ('D', 'I', 'V', 'X'); - break; - default: - GST_INFO ("unhandled mpegversion : %d, fall back to fourcc=MPEG", - mpegversion); - avipad->vids.compression = GST_MAKE_FOURCC ('M', 'P', 'E', 'G'); - break; - } - } else if (!strcmp (mimetype, "video/x-dirac")) { - avipad->vids.compression = GST_MAKE_FOURCC ('d', 'r', 'a', 'c'); - } else if (!strcmp (mimetype, "video/x-wmv")) { - gint wmvversion; - - if (gst_structure_get_int (structure, "wmvversion", &wmvversion)) { - switch (wmvversion) { - case 1: - avipad->vids.compression = GST_MAKE_FOURCC ('W', 'M', 'V', '1'); - break; - case 2: - avipad->vids.compression = GST_MAKE_FOURCC ('W', 'M', 'V', '2'); - break; - case 3: - avipad->vids.compression = GST_MAKE_FOURCC ('W', 'M', 'V', '3'); - default: - break; - } - } - } - - if (!avipad->vids.compression) - goto refuse_caps; - } - - avipad->parent.hdr.fcc_handler = avipad->vids.compression; - avipad->vids.image_size = avipad->vids.height * avipad->vids.width; - /* hm, maybe why avi only handles one stream well ... */ - avimux->avi_hdr.width = avipad->vids.width; - avimux->avi_hdr.height = avipad->vids.height; - avimux->avi_hdr.us_frame = 1000000. * avipad->parent.hdr.scale / - avipad->parent.hdr.rate; - - gst_object_unref (avimux); - return TRUE; - -refuse_caps: - { - GST_WARNING_OBJECT (avimux, "refused caps %" GST_PTR_FORMAT, vscaps); - gst_object_unref (avimux); - return FALSE; - } -} - -static gboolean -gst_avi_mux_audsink_set_caps (GstPad * pad, GstCaps * vscaps) -{ - GstAviMux *avimux; - GstAviAudioPad *avipad; - GstAviCollectData *collect_pad; - GstStructure *structure; - const gchar *mimetype; - const GValue *codec_data; - gint channels, rate; - - avimux = GST_AVI_MUX (gst_pad_get_parent (pad)); - - /* find stream data */ - collect_pad = (GstAviCollectData *) gst_pad_get_element_private (pad); - g_assert (collect_pad); - avipad = (GstAviAudioPad *) collect_pad->avipad; - g_assert (avipad); - g_assert (!avipad->parent.is_video); - g_assert (avipad->parent.hdr.type == GST_MAKE_FOURCC ('a', 'u', 'd', 's')); - - GST_DEBUG_OBJECT (avimux, "%s:%s, caps=%" GST_PTR_FORMAT, - GST_DEBUG_PAD_NAME (pad), vscaps); - - structure = gst_caps_get_structure (vscaps, 0); - mimetype = gst_structure_get_name (structure); - - /* we want these for all */ - if (!gst_structure_get_int (structure, "channels", &channels) || - !gst_structure_get_int (structure, "rate", &rate)) { - goto refuse_caps; - } - - avipad->auds.channels = channels; - avipad->auds.rate = rate; - - /* codec initialization data, if any */ - codec_data = gst_structure_get_value (structure, "codec_data"); - if (codec_data) { - avipad->auds_codec_data = gst_value_get_buffer (codec_data); - gst_buffer_ref (avipad->auds_codec_data); - /* keep global track of size */ - avimux->codec_data_size += GST_BUFFER_SIZE (avipad->auds_codec_data); - } - - if (!strcmp (mimetype, "audio/x-raw-int")) { - gint width, depth; - gboolean signedness; - - avipad->auds.format = GST_RIFF_WAVE_FORMAT_PCM; - - if (!gst_structure_get_int (structure, "width", &width) || - !gst_structure_get_int (structure, "depth", &depth) || - !gst_structure_get_boolean (structure, "signed", &signedness)) { - GST_DEBUG_OBJECT (avimux, - "broken caps, width/depth/signed field missing"); - goto refuse_caps; - } - - /* no clear place to put different values for these while keeping to spec */ - if (width != depth) { - GST_DEBUG_OBJECT (avimux, "width must be same as depth!"); - goto refuse_caps; - } - - /* because that's the way the caps will be recreated from riff data */ - if ((width == 8 && signedness) || (width == 16 && !signedness)) { - GST_DEBUG_OBJECT (avimux, - "8-bit PCM must be unsigned, 16-bit PCM signed"); - goto refuse_caps; - } - - avipad->auds.blockalign = width; - avipad->auds.size = (width == 8) ? 8 : depth; - - /* set some more info straight */ - avipad->auds.blockalign /= 8; - avipad->auds.blockalign *= avipad->auds.channels; - avipad->auds.av_bps = avipad->auds.blockalign * avipad->auds.rate; - } else { - avipad->auds.format = 0; - /* set some defaults */ - avipad->auds.blockalign = 1; - avipad->auds.av_bps = 0; - avipad->auds.size = 16; - - if (!strcmp (mimetype, "audio/mpeg")) { - gint mpegversion; - - gst_structure_get_int (structure, "mpegversion", &mpegversion); - switch (mpegversion) { - case 1:{ - gint layer = 3; - - gst_structure_get_int (structure, "layer", &layer); - switch (layer) { - case 3: - avipad->auds.format = GST_RIFF_WAVE_FORMAT_MPEGL3; - break; - case 1: - case 2: - avipad->auds.format = GST_RIFF_WAVE_FORMAT_MPEGL12; - break; - } - break; - } - case 4: - GST_WARNING ("AAC"); - avipad->auds.format = GST_RIFF_WAVE_FORMAT_AAC; - break; - } - } else if (!strcmp (mimetype, "audio/x-vorbis")) { - avipad->auds.format = GST_RIFF_WAVE_FORMAT_VORBIS3; - } else if (!strcmp (mimetype, "audio/x-ac3")) { - avipad->auds.format = GST_RIFF_WAVE_FORMAT_A52; - } else if (!strcmp (mimetype, "audio/x-alaw")) { - avipad->auds.format = GST_RIFF_WAVE_FORMAT_ALAW; - avipad->auds.size = 8; - avipad->auds.blockalign = avipad->auds.channels; - avipad->auds.av_bps = avipad->auds.blockalign * avipad->auds.rate; - } else if (!strcmp (mimetype, "audio/x-mulaw")) { - avipad->auds.format = GST_RIFF_WAVE_FORMAT_MULAW; - avipad->auds.size = 8; - avipad->auds.blockalign = avipad->auds.channels; - avipad->auds.av_bps = avipad->auds.blockalign * avipad->auds.rate; - } else if (!strcmp (mimetype, "audio/x-wma")) { - gint version; - gint bitrate; - gint block_align; - - if (gst_structure_get_int (structure, "wmaversion", &version)) { - switch (version) { - case 1: - avipad->auds.format = GST_RIFF_WAVE_FORMAT_WMAV1; - break; - case 2: - avipad->auds.format = GST_RIFF_WAVE_FORMAT_WMAV2; - break; - default: - break; - } - } - - if (avipad->auds.format != 0) { - if (gst_structure_get_int (structure, "block_align", &block_align)) { - avipad->auds.blockalign = block_align; - } - if (gst_structure_get_int (structure, "bitrate", &bitrate)) { - avipad->auds.av_bps = bitrate / 8; - } - } - } - } - - if (!avipad->auds.format) - goto refuse_caps; - - /* by spec, hdr.rate is av_bps related, is calculated that way in stop_file, - * and reduces to sample rate in PCM like cases */ - avipad->parent.hdr.rate = avipad->auds.av_bps / avipad->auds.blockalign; - avipad->parent.hdr.samplesize = avipad->auds.blockalign; - avipad->parent.hdr.scale = 1; - - gst_object_unref (avimux); - return TRUE; - -refuse_caps: - { - GST_WARNING_OBJECT (avimux, "refused caps %" GST_PTR_FORMAT, vscaps); - gst_object_unref (avimux); - return FALSE; - } -} - - -static GstPad * -gst_avi_mux_request_new_pad (GstElement * element, - GstPadTemplate * templ, const gchar * req_name) -{ - GstAviMux *avimux; - GstPad *newpad; - GstAviPad *avipad; - GstElementClass *klass; - - g_return_val_if_fail (templ != NULL, NULL); - - if (templ->direction != GST_PAD_SINK) - goto wrong_direction; - - g_return_val_if_fail (GST_IS_AVI_MUX (element), NULL); - avimux = GST_AVI_MUX (element); - - if (!avimux->write_header) - goto too_late; - - klass = GST_ELEMENT_GET_CLASS (element); - - if (templ == gst_element_class_get_pad_template (klass, "audio_%d")) { - gchar *name; - - /* setup pad */ - name = g_strdup_printf ("audio_%02d", avimux->audio_pads); - GST_DEBUG_OBJECT (avimux, "adding new pad: %s", name); - newpad = gst_pad_new_from_template (templ, name); - g_free (name); - gst_pad_set_setcaps_function (newpad, - GST_DEBUG_FUNCPTR (gst_avi_mux_audsink_set_caps)); - - /* init pad specific data */ - avipad = g_malloc0 (sizeof (GstAviAudioPad)); - avipad->is_video = FALSE; - avipad->hdr.type = GST_MAKE_FOURCC ('a', 'u', 'd', 's'); - avimux->audio_pads++; - /* audio goes last */ - avimux->sinkpads = g_slist_append (avimux->sinkpads, avipad); - } else if (templ == gst_element_class_get_pad_template (klass, "video_%d")) { - /* though streams are pretty generic and relatively self-contained, - * some video info goes in a single avi header -and therefore mux struct- - * so video restricted to one stream */ - if (avimux->video_pads > 0) - return NULL; - /* setup pad */ - GST_DEBUG_OBJECT (avimux, "adding new pad: video_00"); - newpad = gst_pad_new_from_template (templ, "video_00"); - gst_pad_set_setcaps_function (newpad, - GST_DEBUG_FUNCPTR (gst_avi_mux_vidsink_set_caps)); - avipad = g_malloc0 (sizeof (GstAviVideoPad)); - - /* init pad specific data */ - avipad->is_video = TRUE; - avipad->hdr.type = GST_MAKE_FOURCC ('v', 'i', 'd', 's'); - avimux->video_pads++; - /* video goes first */ - avimux->sinkpads = g_slist_prepend (avimux->sinkpads, avipad); - } else - goto wrong_template; - - avipad->collect = gst_collect_pads_add_pad (avimux->collect, - newpad, sizeof (GstAviCollectData)); - ((GstAviCollectData *) (avipad->collect))->avipad = avipad; - /* FIXME: hacked way to override/extend the event function of - * GstCollectPads; because it sets its own event function giving the - * element no access to events */ - avimux->collect_event = (GstPadEventFunction) GST_PAD_EVENTFUNC (newpad); - gst_pad_set_event_function (newpad, - GST_DEBUG_FUNCPTR (gst_avi_mux_handle_event)); - - gst_element_add_pad (element, newpad); - - return newpad; - - /* ERRORS */ -wrong_direction: - { - g_warning ("avimux: request pad that is not a SINK pad\n"); - return NULL; - } -too_late: - { - g_warning ("avimux: request pad cannot be added after streaming started\n"); - return NULL; - } -wrong_template: - { - g_warning ("avimuxx: this is not our template!\n"); - return NULL; - } -} - -static void -gst_avi_mux_release_pad (GstElement * element, GstPad * pad) -{ - GstAviMux *avimux = GST_AVI_MUX (element); - GSList *node; - - node = avimux->sinkpads; - while (node) { - GstAviPad *avipad = (GstAviPad *) node->data; - - if (avipad->collect->pad == pad) { - /* pad count should not be adjusted, - * as it also represent number of streams present */ - avipad->collect = NULL; - GST_DEBUG_OBJECT (avimux, "removed pad '%s'", GST_PAD_NAME (pad)); - gst_collect_pads_remove_pad (avimux->collect, pad); - gst_element_remove_pad (element, pad); - /* if not started yet, we can remove any sign this pad ever existed */ - /* in this case _start will take care of the real pad count */ - if (avimux->write_header) { - avimux->sinkpads = g_slist_remove (avimux->sinkpads, avipad); - gst_avi_mux_pad_reset (avipad, TRUE); - g_free (avipad); - } - return; - } - - node = node->next; - } - - g_warning ("Unknown pad %s", GST_PAD_NAME (pad)); -} - -/* maybe some of these functions should be moved to riff.h? */ - -/* DISCLAIMER: this function is fairly ugly. So be it (i.e. it makes the rest easier) - * so is this struct */ - -typedef struct _GstMarkedBuffer -{ - guint *highmark; - GstBuffer *buffer; -} GstMarkedBuffer; - -static void -gst_avi_mux_write_tag (const GstTagList * list, const gchar * tag, - gpointer data) -{ - const struct - { - guint32 fcc; - gchar *tag; - } rifftags[] = { - { - GST_RIFF_INFO_IARL, GST_TAG_LOCATION}, { - GST_RIFF_INFO_IART, GST_TAG_ARTIST}, { - GST_RIFF_INFO_ICMT, GST_TAG_COMMENT}, { - GST_RIFF_INFO_ICOP, GST_TAG_COPYRIGHT}, { - GST_RIFF_INFO_ICRD, GST_TAG_DATE}, { - GST_RIFF_INFO_IGNR, GST_TAG_GENRE}, { - GST_RIFF_INFO_IKEY, GST_TAG_KEYWORDS}, { - GST_RIFF_INFO_INAM, GST_TAG_TITLE}, { - GST_RIFF_INFO_ISFT, GST_TAG_ENCODER}, { - GST_RIFF_INFO_ISRC, GST_TAG_ISRC}, { - 0, NULL} - }; - gint n, len, plen; - GstBuffer *buf = ((GstMarkedBuffer *) data)->buffer; - guint *highmark = ((GstMarkedBuffer *) data)->highmark; - guint8 *buffdata = GST_BUFFER_DATA (buf) + *highmark; - gchar *str; - - for (n = 0; rifftags[n].fcc != 0; n++) { - if (!strcmp (rifftags[n].tag, tag) && - gst_tag_list_get_string (list, tag, &str) && str) { - len = strlen (str); - plen = len + 1; - if (plen & 1) - plen++; - if (GST_BUFFER_SIZE (buf) >= *highmark + 8 + plen) { - GST_WRITE_UINT32_LE (buffdata, rifftags[n].fcc); - GST_WRITE_UINT32_LE (buffdata + 4, len + 1); - memcpy (buffdata + 8, str, len); - buffdata[8 + len] = 0; - *highmark += 8 + plen; - GST_DEBUG ("writing tag in buffer %p, highmark at %d", buf, *highmark); - } - g_free (str); - break; - } - } -} - -#define ODML_SUPERINDEX_SIZE \ - (32 + GST_AVI_SUPERINDEX_COUNT * sizeof (gst_avi_superindex_entry)) - -static GstBuffer * -gst_avi_mux_riff_get_avi_header (GstAviMux * avimux) -{ - const GstTagList *tags; - GstBuffer *buffer; - guint8 *buffdata; - guint size = 0; - guint highmark = 0; - - /* pointer to list size field */ - guint8 *riff_size, *hdrl_size; - GSList *node; - - GST_DEBUG_OBJECT (avimux, "creating avi header, data_size %u, idx_size %u", - avimux->data_size, avimux->idx_size); - - if (avimux->tags_snap) - tags = avimux->tags_snap; - else { - /* need to make snapshot of current state of tags to ensure the same set - * is used next time around during header rewrite at the end */ - tags = gst_tag_setter_get_tag_list (GST_TAG_SETTER (avimux)); - if (tags) - tags = avimux->tags_snap = gst_tag_list_copy (tags); - } - if (tags) { - /* that should be the strlen of all tags + header sizes - * not all of tags end up in a avi, still this is a good estimate - */ - gchar *str = gst_structure_to_string (tags); - size += strlen (str) + 8 * gst_structure_n_fields (tags); - g_free (str); - } - - /* allocate the buffer, starting with some wild/safe upper bound */ - size += avimux->codec_data_size + 100 + sizeof (gst_riff_avih) - + (g_slist_length (avimux->sinkpads) * (100 + sizeof (gst_riff_strh_full) - + sizeof (gst_riff_strf_vids) - + sizeof (gst_riff_vprp) - + sizeof (gst_riff_vprp_video_field_desc) * 2 - + sizeof (gst_riff_strf_auds) + 2 + ODML_SUPERINDEX_SIZE)); - buffer = gst_buffer_new_and_alloc (size); - buffdata = GST_BUFFER_DATA (buffer); - highmark = 0; - GST_DEBUG_OBJECT (avimux, "creating buffer %p, size %d, highmark at 0", - buffer, GST_BUFFER_SIZE (buffer)); - - /* avi header metadata */ - memcpy (buffdata + 0, "RIFF", 4); - /* fill in RIFF size later */ - riff_size = buffdata + 4; - memcpy (buffdata + 8, "AVI ", 4); - memcpy (buffdata + 12, "LIST", 4); - /* fill in header size later */ - hdrl_size = buffdata + 16; - memcpy (buffdata + 20, "hdrl", 4); - memcpy (buffdata + 24, "avih", 4); - GST_WRITE_UINT32_LE (buffdata + 28, sizeof (gst_riff_avih)); - buffdata += 32; - highmark += 32; - - /* the AVI header itself */ - GST_WRITE_UINT32_LE (buffdata + 0, avimux->avi_hdr.us_frame); - GST_WRITE_UINT32_LE (buffdata + 4, avimux->avi_hdr.max_bps); - GST_WRITE_UINT32_LE (buffdata + 8, avimux->avi_hdr.pad_gran); - GST_WRITE_UINT32_LE (buffdata + 12, avimux->avi_hdr.flags); - GST_WRITE_UINT32_LE (buffdata + 16, avimux->avi_hdr.tot_frames); - GST_WRITE_UINT32_LE (buffdata + 20, avimux->avi_hdr.init_frames); - GST_WRITE_UINT32_LE (buffdata + 24, avimux->avi_hdr.streams); - GST_WRITE_UINT32_LE (buffdata + 28, avimux->avi_hdr.bufsize); - GST_WRITE_UINT32_LE (buffdata + 32, avimux->avi_hdr.width); - GST_WRITE_UINT32_LE (buffdata + 36, avimux->avi_hdr.height); - GST_WRITE_UINT32_LE (buffdata + 40, avimux->avi_hdr.scale); - GST_WRITE_UINT32_LE (buffdata + 44, avimux->avi_hdr.rate); - GST_WRITE_UINT32_LE (buffdata + 48, avimux->avi_hdr.start); - GST_WRITE_UINT32_LE (buffdata + 52, avimux->avi_hdr.length); - buffdata += 56; - highmark += 56; - - /* stream data */ - node = avimux->sinkpads; - while (node) { - GstAviPad *avipad = (GstAviPad *) node->data; - GstAviVideoPad *vidpad = (GstAviVideoPad *) avipad; - GstAviAudioPad *audpad = (GstAviAudioPad *) avipad; - guint codec_size = 0, strl_size = 0, vprp_size = 0; - - if (avipad->is_video) { - if (vidpad->vids_codec_data) - codec_size = GST_BUFFER_SIZE (vidpad->vids_codec_data); - strl_size = sizeof (gst_riff_strh_full) + sizeof (gst_riff_strf_vids) - + GST_ROUND_UP_2 (codec_size) + 4 * 5 + ODML_SUPERINDEX_SIZE; - if (vidpad->vprp.aspect) { - /* let's be on the safe side */ - vidpad->vprp.fields = MIN (vidpad->vprp.fields, - GST_RIFF_VPRP_VIDEO_FIELDS); - vprp_size = G_STRUCT_OFFSET (gst_riff_vprp, field_info) - + (vidpad->vprp.fields * sizeof (gst_riff_vprp_video_field_desc)); - strl_size += 4 * 2 + vprp_size; - } - } else { - if (audpad->auds_codec_data) - codec_size = GST_BUFFER_SIZE (audpad->auds_codec_data); - /* +2 is codec_size field, not part of gst_riff_strf_auds */ - strl_size = sizeof (gst_riff_strh_full) + sizeof (gst_riff_strf_auds) + 2 - + GST_ROUND_UP_2 (codec_size) + 4 * 5 + ODML_SUPERINDEX_SIZE; - } - - /* stream list metadata */ - memcpy (buffdata + 0, "LIST", 4); - GST_WRITE_UINT32_LE (buffdata + 4, strl_size); - memcpy (buffdata + 8, "strl", 4); - /* generic header */ - memcpy (buffdata + 12, "strh", 4); - GST_WRITE_UINT32_LE (buffdata + 16, sizeof (gst_riff_strh_full)); - /* the actual header */ - GST_WRITE_UINT32_LE (buffdata + 20, avipad->hdr.type); - GST_WRITE_UINT32_LE (buffdata + 24, avipad->hdr.fcc_handler); - GST_WRITE_UINT32_LE (buffdata + 28, avipad->hdr.flags); - GST_WRITE_UINT32_LE (buffdata + 32, avipad->hdr.priority); - GST_WRITE_UINT32_LE (buffdata + 36, avipad->hdr.init_frames); - GST_WRITE_UINT32_LE (buffdata + 40, avipad->hdr.scale); - GST_WRITE_UINT32_LE (buffdata + 44, avipad->hdr.rate); - GST_WRITE_UINT32_LE (buffdata + 48, avipad->hdr.start); - GST_WRITE_UINT32_LE (buffdata + 52, avipad->hdr.length); - GST_WRITE_UINT32_LE (buffdata + 56, avipad->hdr.bufsize); - GST_WRITE_UINT32_LE (buffdata + 60, avipad->hdr.quality); - GST_WRITE_UINT32_LE (buffdata + 64, avipad->hdr.samplesize); - GST_WRITE_UINT16_LE (buffdata + 68, 0); - GST_WRITE_UINT16_LE (buffdata + 70, 0); - GST_WRITE_UINT16_LE (buffdata + 72, 0); - GST_WRITE_UINT16_LE (buffdata + 74, 0); - buffdata += 76; - highmark += 76; - - if (avipad->is_video) { - /* the video header */ - memcpy (buffdata + 0, "strf", 4); - GST_WRITE_UINT32_LE (buffdata + 4, - sizeof (gst_riff_strf_vids) + codec_size); - /* the actual header */ - GST_WRITE_UINT32_LE (buffdata + 8, vidpad->vids.size + codec_size); - GST_WRITE_UINT32_LE (buffdata + 12, vidpad->vids.width); - GST_WRITE_UINT32_LE (buffdata + 16, vidpad->vids.height); - GST_WRITE_UINT16_LE (buffdata + 20, vidpad->vids.planes); - GST_WRITE_UINT16_LE (buffdata + 22, vidpad->vids.bit_cnt); - GST_WRITE_UINT32_LE (buffdata + 24, vidpad->vids.compression); - GST_WRITE_UINT32_LE (buffdata + 28, vidpad->vids.image_size); - GST_WRITE_UINT32_LE (buffdata + 32, vidpad->vids.xpels_meter); - GST_WRITE_UINT32_LE (buffdata + 36, vidpad->vids.ypels_meter); - GST_WRITE_UINT32_LE (buffdata + 40, vidpad->vids.num_colors); - GST_WRITE_UINT32_LE (buffdata + 44, vidpad->vids.imp_colors); - buffdata += 48; - highmark += 48; - - /* include codec data, if any */ - if (codec_size) { - memcpy (buffdata, GST_BUFFER_DATA (vidpad->vids_codec_data), - codec_size); - - buffdata += codec_size; - highmark += codec_size; - } - /* padding */ - if (highmark & 0x1) { - highmark++; - buffdata++; - } - - /* add video property data, mainly for aspect ratio, if any */ - if (vprp_size) { - gint f; - - /* the vprp header */ - memcpy (buffdata + 0, "vprp", 4); - GST_WRITE_UINT32_LE (buffdata + 4, vprp_size); - /* the actual data */ - GST_WRITE_UINT32_LE (buffdata + 8, vidpad->vprp.format_token); - GST_WRITE_UINT32_LE (buffdata + 12, vidpad->vprp.standard); - GST_WRITE_UINT32_LE (buffdata + 16, vidpad->vprp.vert_rate); - GST_WRITE_UINT32_LE (buffdata + 20, vidpad->vprp.hor_t_total); - GST_WRITE_UINT32_LE (buffdata + 24, vidpad->vprp.vert_lines); - GST_WRITE_UINT32_LE (buffdata + 28, vidpad->vprp.aspect); - GST_WRITE_UINT32_LE (buffdata + 32, vidpad->vprp.width); - GST_WRITE_UINT32_LE (buffdata + 36, vidpad->vprp.height); - GST_WRITE_UINT32_LE (buffdata + 40, vidpad->vprp.fields); - buffdata += 44; - highmark += 44; - for (f = 0; f < vidpad->vprp.fields; ++f) { - gst_riff_vprp_video_field_desc *fd; - - fd = &(vidpad->vprp.field_info[f]); - GST_WRITE_UINT32_LE (buffdata + 0, fd->compressed_bm_height); - GST_WRITE_UINT32_LE (buffdata + 4, fd->compressed_bm_width); - GST_WRITE_UINT32_LE (buffdata + 8, fd->valid_bm_height); - GST_WRITE_UINT32_LE (buffdata + 12, fd->valid_bm_width); - GST_WRITE_UINT32_LE (buffdata + 16, fd->valid_bm_x_offset); - GST_WRITE_UINT32_LE (buffdata + 20, fd->valid_bm_y_offset); - GST_WRITE_UINT32_LE (buffdata + 24, fd->video_x_t_offset); - GST_WRITE_UINT32_LE (buffdata + 28, fd->video_y_start); - buffdata += 32; - highmark += 32; - } - } - } else { - /* the audio header */ - memcpy (buffdata + 0, "strf", 4); - GST_WRITE_UINT32_LE (buffdata + 4, - sizeof (gst_riff_strf_auds) + 2 + codec_size); - /* the actual header */ - GST_WRITE_UINT16_LE (buffdata + 8, audpad->auds.format); - GST_WRITE_UINT16_LE (buffdata + 10, audpad->auds.channels); - GST_WRITE_UINT32_LE (buffdata + 12, audpad->auds.rate); - GST_WRITE_UINT32_LE (buffdata + 16, audpad->auds.av_bps); - GST_WRITE_UINT16_LE (buffdata + 20, audpad->auds.blockalign); - GST_WRITE_UINT16_LE (buffdata + 22, audpad->auds.size); - GST_WRITE_UINT16_LE (buffdata + 24, codec_size); - buffdata += 26; - highmark += 26; - - /* include codec data, if any */ - if (codec_size) { - memcpy (buffdata, GST_BUFFER_DATA (audpad->auds_codec_data), - codec_size); - - buffdata += codec_size; - highmark += codec_size; - } - /* padding */ - if (highmark & 0x1) { - highmark++; - buffdata++; - } - } - - /* odml superindex chunk */ - if (avipad->idx_index > 0) - memcpy (buffdata, "indx", 4); - else - memcpy (buffdata, "JUNK", 4); - GST_WRITE_UINT32_LE (buffdata + 4, ODML_SUPERINDEX_SIZE - 8); /* chunk size */ - GST_WRITE_UINT16_LE (buffdata + 8, 4); /* bytes per entry */ - buffdata[10] = 0; /* index subtype */ - buffdata[11] = GST_AVI_INDEX_OF_INDEXES; /* index type */ - GST_WRITE_UINT32_LE (buffdata + 12, avipad->idx_index); /* entries in use */ - memcpy (buffdata + 16, avipad->tag, 4); /* stream id */ - GST_WRITE_UINT32_LE (buffdata + 20, 0); /* reserved */ - GST_WRITE_UINT32_LE (buffdata + 24, 0); /* reserved */ - GST_WRITE_UINT32_LE (buffdata + 28, 0); /* reserved */ - memcpy (buffdata + 32, avipad->idx, - GST_AVI_SUPERINDEX_COUNT * sizeof (gst_avi_superindex_entry)); - buffdata += ODML_SUPERINDEX_SIZE; - highmark += ODML_SUPERINDEX_SIZE; - - node = node->next; - } - - if (avimux->video_pads > 0) { - /* odml header */ - memcpy (buffdata + 0, "LIST", 4); - GST_WRITE_UINT32_LE (buffdata + 4, sizeof (guint32) + 4 * 3); - memcpy (buffdata + 8, "odml", 4); - memcpy (buffdata + 12, "dmlh", 4); - GST_WRITE_UINT32_LE (buffdata + 16, sizeof (guint32)); - GST_WRITE_UINT32_LE (buffdata + 20, avimux->total_frames); - buffdata += 24; - highmark += 24; - } - - GST_WRITE_UINT32_LE (hdrl_size, (guint32) (buffdata - hdrl_size) - 4); - - /* tags */ - if (tags) { - guint8 *ptr; - guint startsize; - GstMarkedBuffer data ; - data.highmark = &highmark; - data.buffer = buffer; - - memcpy (buffdata + 0, "LIST", 4); - ptr = buffdata + 4; /* fill in later */ - startsize = highmark + 4; - memcpy (buffdata + 8, "INFO", 4); - buffdata += 12; - highmark += 12; - - /* 12 bytes is needed for data header */ - GST_BUFFER_SIZE (buffer) -= 12; - gst_tag_list_foreach (tags, gst_avi_mux_write_tag, &data); - GST_BUFFER_SIZE (buffer) += 12; - buffdata = GST_BUFFER_DATA (buffer) + highmark; - - /* update list size */ - GST_WRITE_UINT32_LE (ptr, highmark - startsize - 4); - } - - /* avi data header */ - memcpy (buffdata + 0, "LIST", 4); - GST_WRITE_UINT32_LE (buffdata + 4, avimux->data_size); - memcpy (buffdata + 8, "movi", 4); - buffdata += 12; - highmark += 12; - - /* finally we can fill in the RIFF size */ - /* note that riff only counts the first avi chunk */ - GST_WRITE_UINT32_LE (riff_size, (guint32) (buffdata - riff_size - 4) /* header and movi tags */ - +avimux->idx_size + avimux->data_size - 4); /* movi data and index */ - - { /* only the part that is filled in actually makes up the header - * unref the parent as we only need this part from now on */ - GstBuffer *subbuffer = gst_buffer_create_sub (buffer, 0, highmark); - - gst_buffer_unref (buffer); - return subbuffer; - } -} - -static GstBuffer * -gst_avi_mux_riff_get_avix_header (guint32 datax_size) -{ - GstBuffer *buffer; - guint8 *buffdata; - - buffer = gst_buffer_new_and_alloc (24); - buffdata = GST_BUFFER_DATA (buffer); - - memcpy (buffdata + 0, "RIFF", 4); - GST_WRITE_UINT32_LE (buffdata + 4, datax_size + 3 * 4); - memcpy (buffdata + 8, "AVIX", 4); - memcpy (buffdata + 12, "LIST", 4); - GST_WRITE_UINT32_LE (buffdata + 16, datax_size); - memcpy (buffdata + 20, "movi", 4); - - return buffer; -} - -static inline GstBuffer * -gst_avi_mux_riff_get_header (GstAviPad * avipad, guint32 video_frame_size) -{ - GstBuffer *buffer; - guint8 *buffdata; - - buffer = gst_buffer_new_and_alloc (8); - buffdata = GST_BUFFER_DATA (buffer); - memcpy (buffdata + 0, avipad->tag, 4); - GST_WRITE_UINT32_LE (buffdata + 4, video_frame_size); - - return buffer; -} - -/* write an odml index chunk in the movi list */ -static GstFlowReturn -gst_avi_mux_write_avix_index (GstAviMux * avimux, gchar * code, - gchar * chunk, gst_avi_superindex_entry * super_index, - gint * super_index_count) -{ - GstFlowReturn res; - GstBuffer *buffer; - guint8 *buffdata, *data; - gst_riff_index_entry *entry; - gint i; - guint32 size, entry_count; - - /* allocate the maximum possible */ - buffer = gst_buffer_new_and_alloc (32 + 8 * avimux->idx_index); - buffdata = GST_BUFFER_DATA (buffer); - - /* general index chunk info */ - memcpy (buffdata + 0, chunk, 4); /* chunk id */ - GST_WRITE_UINT32_LE (buffdata + 4, 0); /* chunk size; fill later */ - GST_WRITE_UINT16_LE (buffdata + 8, 2); /* index entry is 2 words */ - buffdata[10] = 0; /* index subtype */ - buffdata[11] = GST_AVI_INDEX_OF_CHUNKS; /* index type: AVI_INDEX_OF_CHUNKS */ - GST_WRITE_UINT32_LE (buffdata + 12, 0); /* entries in use; fill later */ - memcpy (buffdata + 16, code, 4); /* stream to which index refers */ - GST_WRITE_UINT64_LE (buffdata + 20, avimux->avix_start); /* base offset */ - GST_WRITE_UINT32_LE (buffdata + 28, 0); /* reserved */ - buffdata += 32; - - /* now the actual index entries */ - i = avimux->idx_index; - entry = avimux->idx; - while (i > 0) { - if (memcmp (&entry->id, code, 4) == 0) { - /* enter relative offset to the data (!) */ - GST_WRITE_UINT32_LE (buffdata, GUINT32_FROM_LE (entry->offset) + 8); - /* msb is set if not (!) keyframe */ - GST_WRITE_UINT32_LE (buffdata + 4, GUINT32_FROM_LE (entry->size) - | (GUINT32_FROM_LE (entry->flags) - & GST_RIFF_IF_KEYFRAME ? 0 : 1U << 31)); - buffdata += 8; - } - i--; - entry++; - } - - /* ok, now we know the size and no of entries, fill in where needed */ - data = GST_BUFFER_DATA (buffer); - GST_BUFFER_SIZE (buffer) = size = buffdata - data; - GST_WRITE_UINT32_LE (data + 4, size - 8); - entry_count = (size - 32) / 8; - GST_WRITE_UINT32_LE (data + 12, entry_count); - - /* decorate and send */ - gst_buffer_set_caps (buffer, GST_PAD_CAPS (avimux->srcpad)); - if ((res = gst_pad_push (avimux->srcpad, buffer)) != GST_FLOW_OK) - return res; - - /* keep track of this in superindex (if room) ... */ - if (*super_index_count < GST_AVI_SUPERINDEX_COUNT) { - i = *super_index_count; - super_index[i].offset = GUINT64_TO_LE (avimux->total_data); - super_index[i].size = GUINT32_TO_LE (size); - super_index[i].duration = GUINT32_TO_LE (entry_count); - (*super_index_count)++; - } else - GST_WARNING_OBJECT (avimux, "No more room in superindex of stream %s", - code); - - /* ... and in size */ - avimux->total_data += size; - if (avimux->is_bigfile) - avimux->datax_size += size; - else - avimux->data_size += size; - - return GST_FLOW_OK; -} - -/* some other usable functions (thankyou xawtv ;-) ) */ - -static void -gst_avi_mux_add_index (GstAviMux * avimux, gchar * code, guint32 flags, - guint32 size) -{ - if (avimux->idx_index == avimux->idx_count) { - avimux->idx_count += 256; - avimux->idx = - g_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 = GUINT32_TO_LE (flags); - avimux->idx[avimux->idx_index].offset = GUINT32_TO_LE (avimux->idx_offset); - avimux->idx[avimux->idx_index].size = GUINT32_TO_LE (size); - avimux->idx_index++; -} - -static GstFlowReturn -gst_avi_mux_write_index (GstAviMux * avimux) -{ - GstFlowReturn res; - GstBuffer *buffer; - guint8 *buffdata; - - buffer = gst_buffer_new_and_alloc (8); - buffdata = GST_BUFFER_DATA (buffer); - memcpy (buffdata + 0, "idx1", 4); - GST_WRITE_UINT32_LE (buffdata + 4, - avimux->idx_index * sizeof (gst_riff_index_entry)); - - gst_buffer_set_caps (buffer, GST_PAD_CAPS (avimux->srcpad)); - res = gst_pad_push (avimux->srcpad, buffer); - if (res != GST_FLOW_OK) - return res; - - buffer = gst_buffer_new (); - GST_BUFFER_SIZE (buffer) = avimux->idx_index * sizeof (gst_riff_index_entry); - GST_BUFFER_DATA (buffer) = (guint8 *) avimux->idx; - GST_BUFFER_MALLOCDATA (buffer) = GST_BUFFER_DATA (buffer); - avimux->idx = NULL; /* will be free()'ed by gst_buffer_unref() */ - avimux->total_data += GST_BUFFER_SIZE (buffer) + 8; - - gst_buffer_set_caps (buffer, GST_PAD_CAPS (avimux->srcpad)); - res = gst_pad_push (avimux->srcpad, buffer); - if (res != GST_FLOW_OK) - return res; - - avimux->idx_size += avimux->idx_index * sizeof (gst_riff_index_entry) + 8; - - /* update header */ - avimux->avi_hdr.flags |= GST_RIFF_AVIH_HASINDEX; - return GST_FLOW_OK; -} - -static GstFlowReturn -gst_avi_mux_bigfile (GstAviMux * avimux, gboolean last) -{ - GstFlowReturn res = GST_FLOW_OK; - GstBuffer *header; - GstEvent *event; - GSList *node; - - /* first some odml standard index chunks in the movi list */ - node = avimux->sinkpads; - while (node) { - GstAviPad *avipad = (GstAviPad *) node->data; - - node = node->next; - - res = gst_avi_mux_write_avix_index (avimux, avipad->tag, - avipad->idx_tag, avipad->idx, &avipad->idx_index); - if (res != GST_FLOW_OK) - return res; - } - - if (avimux->is_bigfile) { - /* search back */ - event = gst_event_new_new_segment (FALSE, 1.0, GST_FORMAT_BYTES, - avimux->avix_start, GST_CLOCK_TIME_NONE, avimux->avix_start); - /* if the event succeeds */ - gst_pad_push_event (avimux->srcpad, event); - - /* rewrite AVIX header */ - header = gst_avi_mux_riff_get_avix_header (avimux->datax_size); - gst_buffer_set_caps (header, GST_PAD_CAPS (avimux->srcpad)); - res = gst_pad_push (avimux->srcpad, header); - - /* go back to current location, at least try */ - event = gst_event_new_new_segment (FALSE, 1.0, GST_FORMAT_BYTES, - avimux->total_data, GST_CLOCK_TIME_NONE, avimux->total_data); - gst_pad_push_event (avimux->srcpad, event); - - if (res != GST_FLOW_OK) - return res; - } else { /* write a standard index in the first riff chunk */ - res = gst_avi_mux_write_index (avimux); - /* the index data/buffer is freed by pushing it */ - avimux->idx_count = 0; - if (res != GST_FLOW_OK) - return res; - } - - avimux->avix_start = avimux->total_data; - - if (last) - return res; - - avimux->is_bigfile = TRUE; - avimux->numx_frames = 0; - avimux->datax_size = 4; /* movi tag */ - avimux->idx_index = 0; - - header = gst_avi_mux_riff_get_avix_header (0); - avimux->total_data += GST_BUFFER_SIZE (header); - /* avix_start is used as base offset for the odml index chunk */ - avimux->idx_offset = avimux->total_data - avimux->avix_start; - gst_buffer_set_caps (header, GST_PAD_CAPS (avimux->srcpad)); - return gst_pad_push (avimux->srcpad, header); -} - -/* enough header blabla now, let's go on to actually writing the headers */ - -static GstFlowReturn -gst_avi_mux_start_file (GstAviMux * avimux) -{ - GstFlowReturn res; - GstBuffer *header; - GSList *node; - - avimux->total_data = 0; - avimux->total_frames = 0; - avimux->data_size = 4; /* movi tag */ - avimux->datax_size = 0; - avimux->num_frames = 0; - avimux->numx_frames = 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; - - /* state */ - avimux->write_header = FALSE; - avimux->restart = FALSE; - - /* init streams, see what we've got */ - node = avimux->sinkpads; - avimux->audio_pads = avimux->video_pads = 0; - while (node) { - GstAviPad *avipad = (GstAviPad *) node->data; - - node = node->next; - - if (!avipad->is_video) { - /* audio stream numbers must start at 1 iff there is a video stream 0; - * request_pad inserts video pad at head of list, so this test suffices */ - if (avimux->video_pads) - avimux->audio_pads++; - avipad->tag = g_strdup_printf ("%02uwb", avimux->audio_pads); - avipad->idx_tag = g_strdup_printf ("ix%02u", avimux->audio_pads); - if (!avimux->video_pads) - avimux->audio_pads++; - } else { - avipad->tag = g_strdup_printf ("%02udb", avimux->video_pads); - avipad->idx_tag = g_strdup_printf ("ix%02u", avimux->video_pads++); - } - } - - /* let downstream know we think in BYTES and expect to do seeking later on */ - gst_pad_push_event (avimux->srcpad, - gst_event_new_new_segment (FALSE, 1.0, GST_FORMAT_BYTES, 0, -1, 0)); - - /* header */ - avimux->avi_hdr.streams = g_slist_length (avimux->sinkpads); - avimux->is_bigfile = FALSE; - - header = gst_avi_mux_riff_get_avi_header (avimux); - avimux->total_data += GST_BUFFER_SIZE (header); - - gst_buffer_set_caps (header, GST_PAD_CAPS (avimux->srcpad)); - res = gst_pad_push (avimux->srcpad, header); - - avimux->idx_offset = avimux->total_data; - - return res; -} - -static GstFlowReturn -gst_avi_mux_stop_file (GstAviMux * avimux) -{ - GstFlowReturn res = GST_FLOW_OK; - GstEvent *event; - GstBuffer *header; - GSList *node; - - /* if bigfile, rewrite header, else write indexes */ - /* don't bail out at once if error, still try to re-write header */ - if (avimux->video_pads > 0) { - if (avimux->is_bigfile) { - res = gst_avi_mux_bigfile (avimux, TRUE); - } else { - res = gst_avi_mux_write_index (avimux); - } - } - - /* we do our best to make it interleaved at least ... */ - if (avimux->audio_pads > 0 && avimux->video_pads > 0) - avimux->avi_hdr.flags |= GST_RIFF_AVIH_ISINTERLEAVED; - - /* set rate and everything having to do with that */ - avimux->avi_hdr.max_bps = 0; - node = avimux->sinkpads; - while (node) { - GstAviPad *avipad = (GstAviPad *) node->data; - - node = node->next; - - if (!avipad->is_video) { - GstAviAudioPad *audpad = (GstAviAudioPad *) avipad; - - /* calculate bps if needed */ - if (!audpad->auds.av_bps) { - if (audpad->audio_time) { - audpad->auds.av_bps = - (GST_SECOND * audpad->audio_size) / audpad->audio_time; - /* round bps to nearest multiple of 8; - * which is much more likely to be the (cbr) bitrate in use; - * which in turn results in better timestamp calculation on playback */ - audpad->auds.av_bps = GST_ROUND_UP_8 (audpad->auds.av_bps - 4); - } else { - GST_ELEMENT_WARNING (avimux, STREAM, MUX, - (_("No or invalid input audio, AVI stream will be corrupt.")), - (NULL)); - audpad->auds.av_bps = 0; - } - avipad->hdr.rate = audpad->auds.av_bps * avipad->hdr.scale; - } - avimux->avi_hdr.max_bps += audpad->auds.av_bps; - avipad->hdr.length = (audpad->audio_time * avipad->hdr.rate) / GST_SECOND; - } else { - GstAviVideoPad *vidpad = (GstAviVideoPad *) avipad; - - avimux->avi_hdr.max_bps += ((vidpad->vids.bit_cnt + 7) / 8) * - (1000000. / avimux->avi_hdr.us_frame) * vidpad->vids.image_size; - avipad->hdr.length = avimux->total_frames; - } - } - - /* statistics/total_frames/... */ - avimux->avi_hdr.tot_frames = avimux->num_frames; - - /* seek and rewrite the header */ - header = gst_avi_mux_riff_get_avi_header (avimux); - event = gst_event_new_new_segment (FALSE, 1.0, GST_FORMAT_BYTES, - 0, GST_CLOCK_TIME_NONE, 0); - gst_pad_push_event (avimux->srcpad, event); - - gst_buffer_set_caps (header, GST_PAD_CAPS (avimux->srcpad)); - /* the first error survives */ - if (res == GST_FLOW_OK) - res = gst_pad_push (avimux->srcpad, header); - else - gst_pad_push (avimux->srcpad, header); - - event = gst_event_new_new_segment (FALSE, 1.0, GST_FORMAT_BYTES, - avimux->total_data, GST_CLOCK_TIME_NONE, avimux->total_data); - gst_pad_push_event (avimux->srcpad, event); - - avimux->write_header = TRUE; - - return res; -} - -static GstFlowReturn -gst_avi_mux_restart_file (GstAviMux * avimux) -{ - GstFlowReturn res; - - if ((res = gst_avi_mux_stop_file (avimux)) != GST_FLOW_OK) - return res; - - gst_pad_push_event (avimux->srcpad, gst_event_new_eos ()); - - return gst_avi_mux_start_file (avimux); -} - -/* handle events (search) */ -static gboolean -gst_avi_mux_handle_event (GstPad * pad, GstEvent * event) -{ - GstAviMux *avimux; - gboolean ret; - - avimux = GST_AVI_MUX (gst_pad_get_parent (pad)); - - switch (GST_EVENT_TYPE (event)) { - case GST_EVENT_TAG:{ - GstTagList *list; - GstTagSetter *setter = GST_TAG_SETTER (avimux); - const GstTagMergeMode mode = gst_tag_setter_get_tag_merge_mode (setter); - - gst_event_parse_tag (event, &list); - gst_tag_setter_merge_tags (setter, list, mode); - break; - } - default: - break; - } - - /* now GstCollectPads can take care of the rest, e.g. EOS */ - ret = avimux->collect_event (pad, event); - - gst_object_unref (avimux); - - return ret; -} - -/* send extra 'padding' data */ -static GstFlowReturn -gst_avi_mux_send_pad_data (GstAviMux * avimux, gulong num_bytes) -{ - GstBuffer *buffer; - - buffer = gst_buffer_new_and_alloc (num_bytes); - memset (GST_BUFFER_DATA (buffer), 0, num_bytes); - gst_buffer_set_caps (buffer, GST_PAD_CAPS (avimux->srcpad)); - return gst_pad_push (avimux->srcpad, buffer); -} - -/* do buffer */ -static GstFlowReturn -gst_avi_mux_do_buffer (GstAviMux * avimux, GstAviPad * avipad) -{ - GstFlowReturn res; - GstBuffer *data, *header; - gulong total_size, pad_bytes = 0; - guint flags; - - data = gst_collect_pads_pop (avimux->collect, avipad->collect); - - if (avimux->restart) { - if ((res = gst_avi_mux_restart_file (avimux)) != GST_FLOW_OK) - return res; - } - - /* need to restart or start a next avix chunk ? */ - if ((avimux->is_bigfile ? avimux->datax_size : avimux->data_size) + - GST_BUFFER_SIZE (data) > 1024 * 1024 * 2000) { - if (avimux->enable_large_avi) { - if ((res = gst_avi_mux_bigfile (avimux, FALSE)) != GST_FLOW_OK) - return res; - } else { - if ((res = gst_avi_mux_restart_file (avimux)) != GST_FLOW_OK) - return res; - } - } - - /* get header and record some stats */ - if (GST_BUFFER_SIZE (data) & 1) { - pad_bytes = 2 - (GST_BUFFER_SIZE (data) & 1); - } - header = gst_avi_mux_riff_get_header (avipad, GST_BUFFER_SIZE (data)); - total_size = GST_BUFFER_SIZE (header) + GST_BUFFER_SIZE (data) + pad_bytes; - - if (avimux->is_bigfile) { - avimux->datax_size += total_size; - } else { - avimux->data_size += total_size; - } - - if (avipad->is_video) { - avimux->total_frames++; - - if (avimux->is_bigfile) { - avimux->numx_frames++; - } else { - avimux->num_frames++; - } - - flags = 0x02; - if (!GST_BUFFER_FLAG_IS_SET (data, GST_BUFFER_FLAG_DELTA_UNIT)) - flags |= 0x10; - } else { - GstAviAudioPad *audpad = (GstAviAudioPad *) avipad; - - flags = 0; - audpad->audio_size += GST_BUFFER_SIZE (data); - audpad->audio_time += GST_BUFFER_DURATION (data); - } - - gst_avi_mux_add_index (avimux, avipad->tag, flags, GST_BUFFER_SIZE (data)); - - /* prepare buffers for sending */ - gst_buffer_set_caps (header, GST_PAD_CAPS (avimux->srcpad)); - data = gst_buffer_make_metadata_writable (data); - gst_buffer_set_caps (data, GST_PAD_CAPS (avimux->srcpad)); - - GST_LOG_OBJECT (avimux, "pushing buffers: head, data"); - - if ((res = gst_pad_push (avimux->srcpad, header)) != GST_FLOW_OK) - return res; - if ((res = gst_pad_push (avimux->srcpad, data)) != GST_FLOW_OK) - return res; - - if (pad_bytes) { - if ((res = gst_avi_mux_send_pad_data (avimux, pad_bytes)) != GST_FLOW_OK) - return res; - } - - /* if any push above fails, we're in trouble with file consistency anyway */ - avimux->total_data += total_size; - avimux->idx_offset += total_size; - - return res; -} - -/* pick the oldest buffer from the pads and push it */ -static GstFlowReturn -gst_avi_mux_do_one_buffer (GstAviMux * avimux) -{ - GstAviPad *avipad, *best_pad; - GSList *node; - GstBuffer *buffer; - GstClockTime time, best_time; - - node = avimux->sinkpads; - best_pad = NULL; - best_time = GST_CLOCK_TIME_NONE; - for (; node; node = node->next) { - avipad = (GstAviPad *) node->data; - - if (!avipad->collect) - continue; - - buffer = gst_collect_pads_peek (avimux->collect, avipad->collect); - if (!buffer) - continue; - time = GST_BUFFER_TIMESTAMP (buffer); - gst_buffer_unref (buffer); - - /* invalid timestamp buffers pass first, - * these are probably initialization buffers */ - if (best_pad == NULL || !GST_CLOCK_TIME_IS_VALID (time) - || (GST_CLOCK_TIME_IS_VALID (best_time) && time < best_time)) { - best_pad = avipad; - best_time = time; - } - } - - if (best_pad) { - GST_LOG_OBJECT (avimux, "selected pad %s with time %" GST_TIME_FORMAT, - GST_PAD_NAME (best_pad->collect->pad), GST_TIME_ARGS (best_time)); - - return gst_avi_mux_do_buffer (avimux, best_pad); - } else { - /* simply finish off the file and send EOS */ - gst_avi_mux_stop_file (avimux); - gst_pad_push_event (avimux->srcpad, gst_event_new_eos ()); - return GST_FLOW_UNEXPECTED; - } -} - -static GstFlowReturn -gst_avi_mux_collect_pads (GstCollectPads * pads, GstAviMux * avimux) -{ - GstFlowReturn res; - - if (G_UNLIKELY (avimux->write_header)) { - if ((res = gst_avi_mux_start_file (avimux)) != GST_FLOW_OK) - return res; - } - - return gst_avi_mux_do_one_buffer (avimux); -} - - -static void -gst_avi_mux_get_property (GObject * object, - guint prop_id, GValue * value, GParamSpec * pspec) -{ - GstAviMux *avimux; - - avimux = GST_AVI_MUX (object); - - 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_avi_mux_set_property (GObject * object, - guint prop_id, const GValue * value, GParamSpec * pspec) -{ - GstAviMux *avimux; - - avimux = GST_AVI_MUX (object); - - 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 GstStateChangeReturn -gst_avi_mux_change_state (GstElement * element, GstStateChange transition) -{ - GstAviMux *avimux; - GstStateChangeReturn ret; - - avimux = GST_AVI_MUX (element); - - switch (transition) { - case GST_STATE_CHANGE_READY_TO_PAUSED: - gst_collect_pads_start (avimux->collect); - break; - case GST_STATE_CHANGE_PAUSED_TO_PLAYING: - break; - case GST_STATE_CHANGE_PAUSED_TO_READY: - gst_collect_pads_stop (avimux->collect); - break; - default: - break; - } - - ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition); - if (ret == GST_STATE_CHANGE_FAILURE) - goto done; - - switch (transition) { - case GST_STATE_CHANGE_PLAYING_TO_PAUSED: - break; - case GST_STATE_CHANGE_PAUSED_TO_READY: - gst_avi_mux_reset (avimux); - break; - case GST_STATE_CHANGE_READY_TO_NULL: - break; - default: - break; - } - -done: - return ret; -}