1 /* AVI muxer plugin for GStreamer
2 * Copyright (C) 2002 Ronald Bultje <rbultje@ronald.bitfreak.net>
3 * (C) 2006 Mark Nauwelaerts <manauw@skynet.be>
5 * This library is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Library General Public
7 * License as published by the Free Software Foundation; either
8 * version 2 of the License, or (at your option) any later version.
10 * This library is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Library General Public License for more details.
15 * You should have received a copy of the GNU Library General Public
16 * License along with this library; if not, write to the
17 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18 * Boston, MA 02111-1307, USA.
22 * - the old avimuxer (by Wim Taymans)
23 * - xawtv's aviwriter (by Gerd Knorr)
24 * - mjpegtools' avilib (by Rainer Johanni)
25 * - openDML large-AVI docs
29 * SECTION:element-avimux
31 * Muxes raw or compressed audio and/or video streams into an AVI file.
34 * <title>Example launch lines</title>
35 * <para>(write everything in one line, without the backslash characters)</para>
37 * gst-launch videotestsrc num-buffers=250 \
38 * ! 'video/x-raw-yuv,format=(fourcc)I420,width=320,height=240,framerate=(fraction)25/1' \
40 * audiotestsrc num-buffers=440 ! audioconvert \
41 * ! 'audio/x-raw-int,rate=44100,channels=2' ! queue ! mux. \
42 * avimux name=mux ! filesink location=test.avi
43 * ]| This will create an .AVI file containing an uncompressed video stream
44 * with a test picture and an uncompressed audio stream containing a
47 * gst-launch videotestsrc num-buffers=250 \
48 * ! 'video/x-raw-yuv,format=(fourcc)I420,width=320,height=240,framerate=(fraction)25/1' \
49 * ! xvidenc ! queue ! mux. \
50 * audiotestsrc num-buffers=440 ! audioconvert ! 'audio/x-raw-int,rate=44100,channels=2' \
51 * ! lame ! queue ! mux. \
52 * avimux name=mux ! filesink location=test.avi
53 * ]| This will create an .AVI file containing the same test video and sound
54 * as above, only that both streams will be compressed this time. This will
55 * only work if you have the necessary encoder elements installed of course.
63 #include "gst/gst-i18n-plugin.h"
68 #include <gst/video/video.h>
69 #include <gst/base/gstbytewriter.h>
71 #include "gstavimux.h"
73 GST_DEBUG_CATEGORY_STATIC (avimux_debug);
74 #define GST_CAT_DEFAULT avimux_debug
82 #define DEFAULT_BIGFILE TRUE
84 static GstStaticPadTemplate src_factory = GST_STATIC_PAD_TEMPLATE ("src",
87 GST_STATIC_CAPS ("video/x-msvideo")
90 static GstStaticPadTemplate video_sink_factory =
91 GST_STATIC_PAD_TEMPLATE ("video_%d",
94 GST_STATIC_CAPS ("video/x-raw-yuv, "
95 "format = (fourcc) { YUY2, I420 }, "
96 "width = (int) [ 16, 4096 ], "
97 "height = (int) [ 16, 4096 ], "
98 "framerate = (fraction) [ 0, MAX ]; "
100 "width = (int) [ 16, 4096 ], "
101 "height = (int) [ 16, 4096 ], "
102 "framerate = (fraction) [ 0, MAX ]; "
104 "width = (int) [ 16, 4096 ], "
105 "height = (int) [ 16, 4096 ], "
106 "framerate = (fraction) [ 0, MAX ], "
107 "divxversion = (int) [ 3, 5 ]; "
109 "width = (int) [ 16, 4096 ], "
110 "height = (int) [ 16, 4096 ], "
111 "framerate = (fraction) [ 0, MAX ]; "
113 "width = (int) [ 16, 4096 ], "
114 "height = (int) [ 16, 4096 ], "
115 "framerate = (fraction) [ 0, MAX ]; "
117 "width = (int) [ 16, 4096 ], "
118 "height = (int) [ 16, 4096 ], "
119 "framerate = (fraction) [ 0, MAX ], "
120 "msmpegversion = (int) [ 41, 43 ]; "
122 "width = (int) [ 16, 4096 ], "
123 "height = (int) [ 16, 4096 ], "
124 "framerate = (fraction) [ 0, MAX ], "
125 "mpegversion = (int) { 1, 2, 4}, "
126 "systemstream = (boolean) FALSE; "
128 "width = (int) [ 16, 4096 ], "
129 "height = (int) [ 16, 4096 ], "
130 "framerate = (fraction) [ 0, MAX ]; "
132 "width = (int) [ 16, 4096 ], "
133 "height = (int) [ 16, 4096 ], "
134 "framerate = (fraction) [ 0, MAX ]; "
136 "width = (int) 720, "
137 "height = (int) { 576, 480 }, "
138 "framerate = (fraction) [ 0, MAX ], "
139 "systemstream = (boolean) FALSE; "
141 "width = (int) [ 16, 4096 ], "
142 "height = (int) [ 16, 4096 ], " "framerate = (fraction) [ 0, MAX ];"
144 "width = (int) [ 16, 4096 ], "
145 "height = (int) [ 16, 4096 ], " "framerate = (fraction) [ 0, MAX ];"
147 "width = (int) [ 16, 4096 ], "
148 "height = (int) [ 16, 4096 ], " "framerate = (fraction) [ 0, MAX ], "
149 "wmvversion = (int) [ 1, 3];"
151 "width = (int) [ 1, 2147483647 ], "
152 "height = (int) [ 1, 2147483647 ], "
153 "framerate = (fraction) [ 0, MAX ];"
155 "width = (int) [ 1, 2147483647 ], "
156 "height = (int) [ 1, 2147483647 ], "
157 "framerate = (fraction) [ 0, MAX ]")
160 static GstStaticPadTemplate audio_sink_factory =
161 GST_STATIC_PAD_TEMPLATE ("audio_%d",
164 GST_STATIC_CAPS ("audio/x-raw-int, "
165 "endianness = (int) LITTLE_ENDIAN, "
166 "signed = (boolean) { TRUE, FALSE }, "
167 "width = (int) { 8, 16 }, "
168 "depth = (int) { 8, 16 }, "
169 "rate = (int) [ 1000, 96000 ], "
170 "channels = (int) [ 1, 2 ]; "
172 "mpegversion = (int) 1, "
173 "layer = (int) [ 1, 3 ], "
174 "rate = (int) [ 1000, 96000 ], " "channels = (int) [ 1, 2 ]; "
176 "mpegversion = (int) 4, "
177 "stream-format = (string) raw, "
178 "rate = (int) [ 1000, 96000 ], " "channels = (int) [ 1, 2 ]; "
179 /*#if 0 VC6 doesn't support #if here ...
181 "rate = (int) [ 1000, 96000 ], " "channels = (int) [ 1, 2 ]; "
184 "rate = (int) [ 1000, 96000 ], " "channels = (int) [ 1, 2 ]; "
186 "rate = (int) [ 1000, 48000 ], " "channels = (int) [ 1, 2 ]; "
188 "rate = (int) [ 1000, 48000 ], " "channels = (int) [ 1, 2 ]; "
190 "rate = (int) [ 1000, 96000 ], " "channels = (int) [ 1, 2 ], "
191 "wmaversion = (int) [ 1, 2 ] ")
194 static void gst_avi_mux_base_init (gpointer g_class);
195 static void gst_avi_mux_class_init (GstAviMuxClass * klass);
196 static void gst_avi_mux_init (GstAviMux * avimux);
197 static void gst_avi_mux_pad_reset (GstAviPad * avipad, gboolean free);
199 static GstFlowReturn gst_avi_mux_collect_pads (GstCollectPads * pads,
201 static gboolean gst_avi_mux_handle_event (GstPad * pad, GstEvent * event);
202 static GstPad *gst_avi_mux_request_new_pad (GstElement * element,
203 GstPadTemplate * templ, const gchar * name);
204 static void gst_avi_mux_release_pad (GstElement * element, GstPad * pad);
205 static void gst_avi_mux_set_property (GObject * object,
206 guint prop_id, const GValue * value, GParamSpec * pspec);
207 static void gst_avi_mux_get_property (GObject * object,
208 guint prop_id, GValue * value, GParamSpec * pspec);
209 static GstStateChangeReturn gst_avi_mux_change_state (GstElement * element,
210 GstStateChange transition);
212 static GstElementClass *parent_class = NULL;
215 gst_avi_mux_get_type (void)
217 static GType avimux_type = 0;
220 static const GTypeInfo avimux_info = {
221 sizeof (GstAviMuxClass),
222 gst_avi_mux_base_init,
224 (GClassInitFunc) gst_avi_mux_class_init,
229 (GInstanceInitFunc) gst_avi_mux_init,
231 static const GInterfaceInfo tag_setter_info = {
238 g_type_register_static (GST_TYPE_ELEMENT, "GstAviMux", &avimux_info, 0);
239 g_type_add_interface_static (avimux_type, GST_TYPE_TAG_SETTER,
246 gst_avi_mux_base_init (gpointer g_class)
248 GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
250 gst_element_class_add_pad_template (element_class,
251 gst_static_pad_template_get (&src_factory));
252 gst_element_class_add_pad_template (element_class,
253 gst_static_pad_template_get (&audio_sink_factory));
254 gst_element_class_add_pad_template (element_class,
255 gst_static_pad_template_get (&video_sink_factory));
257 gst_element_class_set_details_simple (element_class, "Avi muxer",
259 "Muxes audio and video into an avi stream",
260 "GStreamer maintainers <gstreamer-devel@lists.sourceforge.net>");
262 GST_DEBUG_CATEGORY_INIT (avimux_debug, "avimux", 0, "Muxer for AVI streams");
266 gst_avi_mux_finalize (GObject * object)
268 GstAviMux *mux = GST_AVI_MUX (object);
271 /* completely free each sinkpad */
272 node = mux->sinkpads;
274 GstAviPad *avipad = (GstAviPad *) node->data;
278 gst_avi_mux_pad_reset (avipad, TRUE);
281 g_slist_free (mux->sinkpads);
282 mux->sinkpads = NULL;
287 gst_object_unref (mux->collect);
289 G_OBJECT_CLASS (parent_class)->finalize (object);
293 gst_avi_mux_class_init (GstAviMuxClass * klass)
295 GObjectClass *gobject_class;
296 GstElementClass *gstelement_class;
298 gobject_class = (GObjectClass *) klass;
299 gstelement_class = (GstElementClass *) klass;
301 parent_class = g_type_class_peek_parent (klass);
303 gobject_class->get_property = gst_avi_mux_get_property;
304 gobject_class->set_property = gst_avi_mux_set_property;
305 gobject_class->finalize = gst_avi_mux_finalize;
307 g_object_class_install_property (gobject_class, ARG_BIGFILE,
308 g_param_spec_boolean ("bigfile", "Bigfile Support (>2GB)",
309 "Support for openDML-2.0 (big) AVI files", DEFAULT_BIGFILE,
310 G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
312 gstelement_class->request_new_pad =
313 GST_DEBUG_FUNCPTR (gst_avi_mux_request_new_pad);
314 gstelement_class->release_pad = GST_DEBUG_FUNCPTR (gst_avi_mux_release_pad);
315 gstelement_class->change_state = GST_DEBUG_FUNCPTR (gst_avi_mux_change_state);
318 /* reset pad to initial state
319 * free - if true, release all, not only stream related, data */
321 gst_avi_mux_pad_reset (GstAviPad * avipad, gboolean free)
324 memset (&(avipad->hdr), 0, sizeof (gst_riff_strh));
326 memset (&(avipad->idx[0]), 0, sizeof (avipad->idx));
329 g_free (avipad->tag);
331 g_free (avipad->idx_tag);
332 avipad->idx_tag = NULL;
335 if (avipad->is_video) {
336 GstAviVideoPad *vidpad = (GstAviVideoPad *) avipad;
338 avipad->hdr.type = GST_MAKE_FOURCC ('v', 'i', 'd', 's');
339 if (vidpad->vids_codec_data) {
340 gst_buffer_unref (vidpad->vids_codec_data);
341 vidpad->vids_codec_data = NULL;
344 if (vidpad->prepend_buffer) {
345 gst_buffer_unref (vidpad->prepend_buffer);
346 vidpad->prepend_buffer = NULL;
349 memset (&(vidpad->vids), 0, sizeof (gst_riff_strf_vids));
350 memset (&(vidpad->vprp), 0, sizeof (gst_riff_vprp));
352 GstAviAudioPad *audpad = (GstAviAudioPad *) avipad;
356 avipad->hdr.type = GST_MAKE_FOURCC ('a', 'u', 'd', 's');
357 if (audpad->auds_codec_data) {
358 gst_buffer_unref (audpad->auds_codec_data);
359 audpad->auds_codec_data = NULL;
362 memset (&(audpad->auds), 0, sizeof (gst_riff_strf_auds));
367 gst_avi_mux_reset (GstAviMux * avimux)
369 GSList *node, *newlist = NULL;
371 /* free and reset each sinkpad */
372 node = avimux->sinkpads;
374 GstAviPad *avipad = (GstAviPad *) node->data;
378 gst_avi_mux_pad_reset (avipad, FALSE);
379 /* if this pad has collectdata, keep it, otherwise dump it completely */
381 newlist = g_slist_append (newlist, avipad);
383 gst_avi_mux_pad_reset (avipad, TRUE);
388 /* free the old list of sinkpads, only keep the real collecting ones */
389 g_slist_free (avimux->sinkpads);
390 avimux->sinkpads = newlist;
393 avimux->num_frames = 0;
394 memset (&(avimux->avi_hdr), 0, sizeof (gst_riff_avih));
395 avimux->avi_hdr.max_bps = 10000000;
396 avimux->codec_data_size = 0;
398 if (avimux->tags_snap) {
399 gst_tag_list_free (avimux->tags_snap);
400 avimux->tags_snap = NULL;
403 g_free (avimux->idx);
407 avimux->write_header = TRUE;
410 gst_tag_setter_reset_tags (GST_TAG_SETTER (avimux));
414 gst_avi_mux_init (GstAviMux * avimux)
416 avimux->srcpad = gst_pad_new_from_static_template (&src_factory, "src");
417 gst_pad_use_fixed_caps (avimux->srcpad);
418 gst_element_add_pad (GST_ELEMENT (avimux), avimux->srcpad);
421 avimux->enable_large_avi = DEFAULT_BIGFILE;
423 avimux->collect = gst_collect_pads_new ();
424 gst_collect_pads_set_function (avimux->collect,
425 (GstCollectPadsFunction) (GST_DEBUG_FUNCPTR (gst_avi_mux_collect_pads)),
428 /* set to clean state */
429 gst_avi_mux_reset (avimux);
433 gst_avi_mux_vidsink_set_caps (GstPad * pad, GstCaps * vscaps)
436 GstAviVideoPad *avipad;
437 GstAviCollectData *collect_pad;
438 GstStructure *structure;
439 const gchar *mimetype;
440 const GValue *fps, *par;
441 const GValue *codec_data;
444 gboolean codec_data_in_headers = TRUE;
446 avimux = GST_AVI_MUX (gst_pad_get_parent (pad));
448 /* find stream data */
449 collect_pad = (GstAviCollectData *) gst_pad_get_element_private (pad);
450 g_assert (collect_pad);
451 avipad = (GstAviVideoPad *) collect_pad->avipad;
453 g_assert (avipad->parent.is_video);
454 g_assert (avipad->parent.hdr.type == GST_MAKE_FOURCC ('v', 'i', 'd', 's'));
456 GST_DEBUG_OBJECT (avimux, "%s:%s, caps=%" GST_PTR_FORMAT,
457 GST_DEBUG_PAD_NAME (pad), vscaps);
459 structure = gst_caps_get_structure (vscaps, 0);
460 mimetype = gst_structure_get_name (structure);
463 avipad->vids.size = sizeof (gst_riff_strf_vids);
464 avipad->vids.planes = 1;
465 if (!gst_structure_get_int (structure, "width", &width) ||
466 !gst_structure_get_int (structure, "height", &height)) {
470 avipad->vids.width = width;
471 avipad->vids.height = height;
473 fps = gst_structure_get_value (structure, "framerate");
474 if (fps == NULL || !GST_VALUE_HOLDS_FRACTION (fps))
477 avipad->parent.hdr.rate = gst_value_get_fraction_numerator (fps);
478 avipad->parent.hdr.scale = gst_value_get_fraction_denominator (fps);
480 /* (pixel) aspect ratio data, if any */
481 par = gst_structure_get_value (structure, "pixel-aspect-ratio");
482 /* only use video properties header if there is non-trivial aspect info */
483 if (par && GST_VALUE_HOLDS_FRACTION (par) &&
484 ((par_n = gst_value_get_fraction_numerator (par)) !=
485 (par_d = gst_value_get_fraction_denominator (par)))) {
486 GValue to_ratio = { 0, };
487 guint ratio_n, ratio_d;
489 /* some fraction voodoo to obtain simplest possible ratio */
490 g_value_init (&to_ratio, GST_TYPE_FRACTION);
491 gst_value_set_fraction (&to_ratio, width * par_n, height * par_d);
492 ratio_n = gst_value_get_fraction_numerator (&to_ratio);
493 ratio_d = gst_value_get_fraction_denominator (&to_ratio);
494 GST_DEBUG_OBJECT (avimux, "generating vprp data with aspect ratio %d/%d",
497 avipad->vprp.vert_rate = avipad->parent.hdr.rate / avipad->parent.hdr.scale;
498 avipad->vprp.hor_t_total = width;
499 avipad->vprp.vert_lines = height;
500 avipad->vprp.aspect = (ratio_n) << 16 | (ratio_d & 0xffff);
501 avipad->vprp.width = width;
502 avipad->vprp.height = height;
503 avipad->vprp.fields = 1;
504 avipad->vprp.field_info[0].compressed_bm_height = height;
505 avipad->vprp.field_info[0].compressed_bm_width = width;
506 avipad->vprp.field_info[0].valid_bm_height = height;
507 avipad->vprp.field_info[0].valid_bm_width = width;
510 if (!strcmp (mimetype, "video/x-raw-yuv")) {
513 gst_structure_get_fourcc (structure, "format", &format);
514 avipad->vids.compression = format;
516 case GST_MAKE_FOURCC ('Y', 'U', 'Y', '2'):
517 avipad->vids.bit_cnt = 16;
519 case GST_MAKE_FOURCC ('I', '4', '2', '0'):
520 avipad->vids.bit_cnt = 12;
524 avipad->vids.bit_cnt = 24;
525 avipad->vids.compression = 0;
528 if (!strcmp (mimetype, "video/x-huffyuv")) {
529 avipad->vids.compression = GST_MAKE_FOURCC ('H', 'F', 'Y', 'U');
530 } else if (!strcmp (mimetype, "image/jpeg")) {
531 avipad->vids.compression = GST_MAKE_FOURCC ('M', 'J', 'P', 'G');
532 } else if (!strcmp (mimetype, "video/x-divx")) {
535 gst_structure_get_int (structure, "divxversion", &divxversion);
536 switch (divxversion) {
538 avipad->vids.compression = GST_MAKE_FOURCC ('D', 'I', 'V', '3');
541 avipad->vids.compression = GST_MAKE_FOURCC ('D', 'I', 'V', 'X');
544 avipad->vids.compression = GST_MAKE_FOURCC ('D', 'X', '5', '0');
547 } else if (!strcmp (mimetype, "video/x-xvid")) {
548 avipad->vids.compression = GST_MAKE_FOURCC ('X', 'V', 'I', 'D');
549 } else if (!strcmp (mimetype, "video/x-3ivx")) {
550 avipad->vids.compression = GST_MAKE_FOURCC ('3', 'I', 'V', '2');
551 } else if (gst_structure_has_name (structure, "video/x-msmpeg")) {
554 gst_structure_get_int (structure, "msmpegversion", &msmpegversion);
555 switch (msmpegversion) {
557 avipad->vids.compression = GST_MAKE_FOURCC ('M', 'P', 'G', '4');
560 avipad->vids.compression = GST_MAKE_FOURCC ('M', 'P', '4', '2');
563 avipad->vids.compression = GST_MAKE_FOURCC ('M', 'P', '4', '3');
566 GST_INFO ("unhandled msmpegversion : %d, fall back to fourcc=MPEG",
568 avipad->vids.compression = GST_MAKE_FOURCC ('M', 'P', 'E', 'G');
571 } else if (!strcmp (mimetype, "video/x-dv")) {
572 avipad->vids.compression = GST_MAKE_FOURCC ('D', 'V', 'S', 'D');
573 } else if (!strcmp (mimetype, "video/x-h263")) {
574 avipad->vids.compression = GST_MAKE_FOURCC ('H', '2', '6', '3');
575 } else if (!strcmp (mimetype, "video/x-h264")) {
576 avipad->vids.compression = GST_MAKE_FOURCC ('H', '2', '6', '4');
577 } else if (!strcmp (mimetype, "video/mpeg")) {
580 gst_structure_get_int (structure, "mpegversion", &mpegversion);
582 switch (mpegversion) {
584 avipad->vids.compression = GST_MAKE_FOURCC ('M', 'P', 'G', '2');
587 /* mplayer/ffmpeg might not work with DIVX, but with FMP4 */
588 avipad->vids.compression = GST_MAKE_FOURCC ('D', 'I', 'V', 'X');
590 /* DIVX/XVID in AVI store the codec_data chunk as part of the
591 first data buffer. So for this case, we prepend the codec_data
592 blob (if any) to that first buffer */
593 codec_data_in_headers = FALSE;
596 GST_INFO ("unhandled mpegversion : %d, fall back to fourcc=MPEG",
598 avipad->vids.compression = GST_MAKE_FOURCC ('M', 'P', 'E', 'G');
601 } else if (!strcmp (mimetype, "video/x-dirac")) {
602 avipad->vids.compression = GST_MAKE_FOURCC ('d', 'r', 'a', 'c');
603 } else if (!strcmp (mimetype, "video/x-wmv")) {
606 if (gst_structure_get_int (structure, "wmvversion", &wmvversion)) {
607 switch (wmvversion) {
609 avipad->vids.compression = GST_MAKE_FOURCC ('W', 'M', 'V', '1');
612 avipad->vids.compression = GST_MAKE_FOURCC ('W', 'M', 'V', '2');
615 avipad->vids.compression = GST_MAKE_FOURCC ('W', 'M', 'V', '3');
620 } else if (!strcmp (mimetype, "image/x-jpc")) {
621 avipad->vids.compression = GST_MAKE_FOURCC ('M', 'J', '2', 'C');
622 } else if (!strcmp (mimetype, "video/x-vp8")) {
623 avipad->vids.compression = GST_MAKE_FOURCC ('V', 'P', '8', '0');
626 if (!avipad->vids.compression)
630 /* codec initialization data, if any */
631 codec_data = gst_structure_get_value (structure, "codec_data");
633 if (codec_data_in_headers) {
634 avipad->vids_codec_data = gst_value_get_buffer (codec_data);
635 gst_buffer_ref (avipad->vids_codec_data);
636 /* keep global track of size */
637 avimux->codec_data_size += gst_buffer_get_size (avipad->vids_codec_data);
639 avipad->prepend_buffer =
640 gst_buffer_ref (gst_value_get_buffer (codec_data));
644 avipad->parent.hdr.fcc_handler = avipad->vids.compression;
645 avipad->vids.image_size = avipad->vids.height * avipad->vids.width;
646 /* hm, maybe why avi only handles one stream well ... */
647 avimux->avi_hdr.width = avipad->vids.width;
648 avimux->avi_hdr.height = avipad->vids.height;
649 avimux->avi_hdr.us_frame = 1000000. * avipad->parent.hdr.scale /
650 avipad->parent.hdr.rate;
652 gst_object_unref (avimux);
657 GST_WARNING_OBJECT (avimux, "refused caps %" GST_PTR_FORMAT, vscaps);
658 gst_object_unref (avimux);
664 gst_avi_mux_audsink_scan_mpeg_audio (GstAviMux * avimux, GstAviPad * avipad,
675 data = gst_buffer_map (buffer, &size, NULL, GST_MAP_READ);
679 header = GST_READ_UINT32_BE (data);
681 if ((header & 0xffe00000) != 0xffe00000)
684 /* thanks go to mp3parse */
685 if (header & (1 << 20)) {
686 lsf = (header & (1 << 19)) ? 0 : 1;
693 version = 1 + lsf + mpg25;
694 layer = 4 - ((header >> 17) & 0x3);
696 /* see http://www.codeproject.com/audio/MPEGAudioInfo.asp */
701 else if (version == 1) {
704 /* MPEG-2 or "2.5" */
708 if (G_UNLIKELY (avipad->hdr.scale <= 1))
709 avipad->hdr.scale = spf;
710 else if (G_UNLIKELY (avipad->hdr.scale != spf)) {
711 GST_WARNING_OBJECT (avimux, "input mpeg audio has varying frame size");
715 gst_buffer_unmap (buffer, data, size);
722 GST_WARNING_OBJECT (avimux, "input mpeg audio is not parsed");
727 GST_WARNING_OBJECT (avimux, "falling back to CBR muxing");
728 avipad->hdr.scale = 1;
729 /* no need to check further */
736 gst_avi_mux_audsink_set_fields (GstAviMux * avimux, GstAviAudioPad * avipad)
738 if (avipad->parent.hdr.scale > 1) {
739 /* vbr case: fixed duration per frame/chunk */
740 avipad->parent.hdr.rate = avipad->auds.rate;
741 avipad->parent.hdr.samplesize = 0;
742 /* FIXME ?? some rumours say this should be largest audio chunk size */
743 avipad->auds.blockalign = avipad->parent.hdr.scale;
745 /* by spec, hdr.rate is av_bps related, is calculated that way in stop_file,
746 * and reduces to sample rate in PCM like cases */
747 avipad->parent.hdr.rate = avipad->auds.av_bps / avipad->auds.blockalign;
748 avipad->parent.hdr.samplesize = avipad->auds.blockalign;
749 avipad->parent.hdr.scale = 1;
754 gst_avi_mux_audsink_set_caps (GstPad * pad, GstCaps * vscaps)
757 GstAviAudioPad *avipad;
758 GstAviCollectData *collect_pad;
759 GstStructure *structure;
760 const gchar *mimetype;
761 const GValue *codec_data;
764 avimux = GST_AVI_MUX (gst_pad_get_parent (pad));
766 /* find stream data */
767 collect_pad = (GstAviCollectData *) gst_pad_get_element_private (pad);
768 g_assert (collect_pad);
769 avipad = (GstAviAudioPad *) collect_pad->avipad;
771 g_assert (!avipad->parent.is_video);
772 g_assert (avipad->parent.hdr.type == GST_MAKE_FOURCC ('a', 'u', 'd', 's'));
774 GST_DEBUG_OBJECT (avimux, "%s:%s, caps=%" GST_PTR_FORMAT,
775 GST_DEBUG_PAD_NAME (pad), vscaps);
777 structure = gst_caps_get_structure (vscaps, 0);
778 mimetype = gst_structure_get_name (structure);
780 /* we want these for all */
781 if (!gst_structure_get_int (structure, "channels", &channels) ||
782 !gst_structure_get_int (structure, "rate", &rate)) {
786 avipad->auds.channels = channels;
787 avipad->auds.rate = rate;
789 /* codec initialization data, if any */
790 codec_data = gst_structure_get_value (structure, "codec_data");
792 avipad->auds_codec_data = gst_value_get_buffer (codec_data);
793 gst_buffer_ref (avipad->auds_codec_data);
794 /* keep global track of size */
795 avimux->codec_data_size += gst_buffer_get_size (avipad->auds_codec_data);
798 if (!strcmp (mimetype, "audio/x-raw-int")) {
802 avipad->auds.format = GST_RIFF_WAVE_FORMAT_PCM;
804 if (!gst_structure_get_int (structure, "width", &width) ||
805 !gst_structure_get_int (structure, "depth", &depth) ||
806 !gst_structure_get_boolean (structure, "signed", &signedness)) {
807 GST_DEBUG_OBJECT (avimux,
808 "broken caps, width/depth/signed field missing");
812 /* no clear place to put different values for these while keeping to spec */
813 if (width != depth) {
814 GST_DEBUG_OBJECT (avimux, "width must be same as depth!");
818 /* because that's the way the caps will be recreated from riff data */
819 if ((width == 8 && signedness) || (width == 16 && !signedness)) {
820 GST_DEBUG_OBJECT (avimux,
821 "8-bit PCM must be unsigned, 16-bit PCM signed");
825 avipad->auds.blockalign = width;
826 avipad->auds.size = (width == 8) ? 8 : depth;
828 /* set some more info straight */
829 avipad->auds.blockalign /= 8;
830 avipad->auds.blockalign *= avipad->auds.channels;
831 avipad->auds.av_bps = avipad->auds.blockalign * avipad->auds.rate;
833 avipad->auds.format = 0;
834 /* set some defaults */
835 avipad->auds.blockalign = 1;
836 avipad->auds.av_bps = 0;
837 avipad->auds.size = 16;
839 if (!strcmp (mimetype, "audio/mpeg")) {
842 gst_structure_get_int (structure, "mpegversion", &mpegversion);
843 switch (mpegversion) {
846 gboolean parsed = FALSE;
848 gst_structure_get_int (structure, "layer", &layer);
849 gst_structure_get_boolean (structure, "parsed", &parsed);
852 avipad->auds.format = GST_RIFF_WAVE_FORMAT_MPEGL3;
856 avipad->auds.format = GST_RIFF_WAVE_FORMAT_MPEGL12;
860 /* treat as VBR, should also cover CBR case;
861 * setup hook to parse frame header and determine spf */
862 avipad->parent.hook = gst_avi_mux_audsink_scan_mpeg_audio;
864 GST_WARNING_OBJECT (avimux, "unparsed MPEG audio input (?), "
871 GstBuffer *codec_data_buf = avipad->auds_codec_data;
872 const gchar *stream_format;
876 stream_format = gst_structure_get_string (structure, "stream-format");
878 if (strcmp (stream_format, "raw") != 0) {
879 GST_WARNING_OBJECT (avimux, "AAC's stream format '%s' is not "
880 "supported, please use 'raw'", stream_format);
884 GST_WARNING_OBJECT (avimux, "AAC's stream-format not specified, "
888 /* vbr case needs some special handling */
889 if (!codec_data_buf || gst_buffer_get_size (codec_data_buf) < 2) {
890 GST_WARNING_OBJECT (avimux, "no (valid) codec_data for AAC audio");
893 avipad->auds.format = GST_RIFF_WAVE_FORMAT_AAC;
894 /* need to determine frame length */
895 gst_buffer_extract (codec_data_buf, 0, data, 2);
896 codec = GST_READ_UINT16_BE (data);
897 avipad->parent.hdr.scale = (codec & 0x4) ? 960 : 1024;
901 } else if (!strcmp (mimetype, "audio/x-vorbis")) {
902 avipad->auds.format = GST_RIFF_WAVE_FORMAT_VORBIS3;
903 } else if (!strcmp (mimetype, "audio/x-ac3")) {
904 avipad->auds.format = GST_RIFF_WAVE_FORMAT_A52;
905 } else if (!strcmp (mimetype, "audio/x-alaw")) {
906 avipad->auds.format = GST_RIFF_WAVE_FORMAT_ALAW;
907 avipad->auds.size = 8;
908 avipad->auds.blockalign = avipad->auds.channels;
909 avipad->auds.av_bps = avipad->auds.blockalign * avipad->auds.rate;
910 } else if (!strcmp (mimetype, "audio/x-mulaw")) {
911 avipad->auds.format = GST_RIFF_WAVE_FORMAT_MULAW;
912 avipad->auds.size = 8;
913 avipad->auds.blockalign = avipad->auds.channels;
914 avipad->auds.av_bps = avipad->auds.blockalign * avipad->auds.rate;
915 } else if (!strcmp (mimetype, "audio/x-wma")) {
920 if (gst_structure_get_int (structure, "wmaversion", &version)) {
923 avipad->auds.format = GST_RIFF_WAVE_FORMAT_WMAV1;
926 avipad->auds.format = GST_RIFF_WAVE_FORMAT_WMAV2;
933 if (avipad->auds.format != 0) {
934 if (gst_structure_get_int (structure, "block_align", &block_align)) {
935 avipad->auds.blockalign = block_align;
937 if (gst_structure_get_int (structure, "bitrate", &bitrate)) {
938 avipad->auds.av_bps = bitrate / 8;
944 if (!avipad->auds.format)
947 avipad->parent.hdr.fcc_handler = avipad->auds.format;
948 gst_avi_mux_audsink_set_fields (avimux, avipad);
950 gst_object_unref (avimux);
955 GST_WARNING_OBJECT (avimux, "refused caps %" GST_PTR_FORMAT, vscaps);
956 gst_object_unref (avimux);
963 gst_avi_mux_request_new_pad (GstElement * element,
964 GstPadTemplate * templ, const gchar * req_name)
969 GstElementClass *klass;
971 const gchar *pad_name = NULL;
972 GstPadSetCapsFunction setcapsfunc = NULL;
975 g_return_val_if_fail (templ != NULL, NULL);
977 if (templ->direction != GST_PAD_SINK)
978 goto wrong_direction;
980 g_return_val_if_fail (GST_IS_AVI_MUX (element), NULL);
981 avimux = GST_AVI_MUX (element);
983 if (!avimux->write_header)
986 klass = GST_ELEMENT_GET_CLASS (element);
988 /* FIXME-0.11: use %d instead of %02d for pad_names */
990 if (templ == gst_element_class_get_pad_template (klass, "audio_%d")) {
991 /* don't mix named and unnamed pads, if the pad already exists we fail when
992 * trying to add it */
993 if (req_name != NULL && sscanf (req_name, "audio_%02d", &pad_id) == 1) {
996 name = g_strdup_printf ("audio_%02d", avimux->audio_pads++);
999 setcapsfunc = GST_DEBUG_FUNCPTR (gst_avi_mux_audsink_set_caps);
1001 /* init pad specific data */
1002 avipad = g_malloc0 (sizeof (GstAviAudioPad));
1003 avipad->is_video = FALSE;
1004 avipad->hdr.type = GST_MAKE_FOURCC ('a', 'u', 'd', 's');
1005 /* audio goes last */
1006 avimux->sinkpads = g_slist_append (avimux->sinkpads, avipad);
1007 } else if (templ == gst_element_class_get_pad_template (klass, "video_%d")) {
1008 /* though streams are pretty generic and relatively self-contained,
1009 * some video info goes in a single avi header -and therefore mux struct-
1010 * so video restricted to one stream */
1011 if (avimux->video_pads > 0)
1012 goto too_many_video_pads;
1015 pad_name = "video_00";
1016 avimux->video_pads++;
1017 setcapsfunc = GST_DEBUG_FUNCPTR (gst_avi_mux_vidsink_set_caps);
1019 /* init pad specific data */
1020 avipad = g_malloc0 (sizeof (GstAviVideoPad));
1021 avipad->is_video = TRUE;
1022 avipad->hdr.type = GST_MAKE_FOURCC ('v', 'i', 'd', 's');
1023 /* video goes first */
1024 avimux->sinkpads = g_slist_prepend (avimux->sinkpads, avipad);
1026 goto wrong_template;
1028 newpad = gst_pad_new_from_template (templ, pad_name);
1029 gst_pad_set_setcaps_function (newpad, setcapsfunc);
1033 avipad->collect = gst_collect_pads_add_pad (avimux->collect,
1034 newpad, sizeof (GstAviCollectData));
1035 ((GstAviCollectData *) (avipad->collect))->avipad = avipad;
1036 /* FIXME: hacked way to override/extend the event function of
1037 * GstCollectPads; because it sets its own event function giving the
1038 * element no access to events */
1039 avimux->collect_event = (GstPadEventFunction) GST_PAD_EVENTFUNC (newpad);
1040 gst_pad_set_event_function (newpad,
1041 GST_DEBUG_FUNCPTR (gst_avi_mux_handle_event));
1043 if (!gst_element_add_pad (element, newpad))
1044 goto pad_add_failed;
1046 GST_DEBUG_OBJECT (newpad, "Added new request pad");
1053 g_warning ("avimux: request pad that is not a SINK pad\n");
1058 g_warning ("avimux: request pad cannot be added after streaming started\n");
1063 g_warning ("avimux: this is not our template!\n");
1066 too_many_video_pads:
1068 GST_WARNING_OBJECT (avimux, "Can only have one video stream");
1073 GST_WARNING_OBJECT (avimux, "Adding the new pad '%s' failed", pad_name);
1074 gst_object_unref (newpad);
1080 gst_avi_mux_release_pad (GstElement * element, GstPad * pad)
1082 GstAviMux *avimux = GST_AVI_MUX (element);
1085 node = avimux->sinkpads;
1087 GstAviPad *avipad = (GstAviPad *) node->data;
1089 if (avipad->collect->pad == pad) {
1090 /* pad count should not be adjusted,
1091 * as it also represent number of streams present */
1092 avipad->collect = NULL;
1093 GST_DEBUG_OBJECT (avimux, "removed pad '%s'", GST_PAD_NAME (pad));
1094 gst_collect_pads_remove_pad (avimux->collect, pad);
1095 gst_element_remove_pad (element, pad);
1096 /* if not started yet, we can remove any sign this pad ever existed */
1097 /* in this case _start will take care of the real pad count */
1098 if (avimux->write_header) {
1099 avimux->sinkpads = g_slist_remove (avimux->sinkpads, avipad);
1100 gst_avi_mux_pad_reset (avipad, TRUE);
1109 g_warning ("Unknown pad %s", GST_PAD_NAME (pad));
1113 gst_avi_mux_start_chunk (GstByteWriter * bw, const gchar * tag, guint32 fourcc)
1118 gst_byte_writer_put_data (bw, (const guint8 *) tag, 4);
1120 gst_byte_writer_put_uint32_le (bw, fourcc);
1122 chunk_offset = gst_byte_writer_get_pos (bw);
1123 /* real chunk size comes later */
1124 gst_byte_writer_put_uint32_le (bw, 0);
1126 return chunk_offset;
1130 gst_avi_mux_end_chunk (GstByteWriter * bw, guint chunk_offset)
1134 size = gst_byte_writer_get_pos (bw);
1136 gst_byte_writer_set_pos (bw, chunk_offset);
1137 gst_byte_writer_put_uint32_le (bw, size - chunk_offset - 4);
1138 gst_byte_writer_set_pos (bw, size);
1140 /* arrange for even padding */
1142 gst_byte_writer_put_uint8 (bw, 0);
1145 /* maybe some of these functions should be moved to riff.h? */
1148 gst_avi_mux_write_tag (const GstTagList * list, const gchar * tag,
1157 GST_RIFF_INFO_IARL, GST_TAG_LOCATION}, {
1158 GST_RIFF_INFO_IART, GST_TAG_ARTIST}, {
1159 GST_RIFF_INFO_ICMT, GST_TAG_COMMENT}, {
1160 GST_RIFF_INFO_ICOP, GST_TAG_COPYRIGHT}, {
1161 GST_RIFF_INFO_ICRD, GST_TAG_DATE}, {
1162 GST_RIFF_INFO_IGNR, GST_TAG_GENRE}, {
1163 GST_RIFF_INFO_IKEY, GST_TAG_KEYWORDS}, {
1164 GST_RIFF_INFO_INAM, GST_TAG_TITLE}, {
1165 GST_RIFF_INFO_ISFT, GST_TAG_ENCODER}, {
1166 GST_RIFF_INFO_ISRC, GST_TAG_ISRC}, {
1171 GstByteWriter *bw = data;
1174 for (n = 0; rifftags[n].fcc != 0; n++) {
1175 if (!strcmp (rifftags[n].tag, tag) &&
1176 gst_tag_list_get_string (list, tag, &str) && str) {
1177 chunk = gst_avi_mux_start_chunk (bw, NULL, rifftags[n].fcc);
1178 gst_byte_writer_put_string (bw, str);
1179 gst_avi_mux_end_chunk (bw, chunk);
1187 gst_avi_mux_riff_get_avi_header (GstAviMux * avimux)
1189 const GstTagList *tags;
1194 guint avih, riff, hdrl;
1198 GST_DEBUG_OBJECT (avimux, "creating avi header, data_size %u, idx_size %u",
1199 avimux->data_size, avimux->idx_size);
1201 if (avimux->tags_snap)
1202 tags = avimux->tags_snap;
1204 /* need to make snapshot of current state of tags to ensure the same set
1205 * is used next time around during header rewrite at the end */
1206 tags = gst_tag_setter_get_tag_list (GST_TAG_SETTER (avimux));
1208 tags = avimux->tags_snap = gst_tag_list_copy (tags);
1211 gst_byte_writer_init_with_size (&bw, 1024, FALSE);
1213 /* avi header metadata */
1214 riff = gst_avi_mux_start_chunk (&bw, "RIFF", 0);
1215 gst_byte_writer_put_data (&bw, (guint8 *) "AVI ", 4);
1216 hdrl = gst_avi_mux_start_chunk (&bw, "LIST", 0);
1217 gst_byte_writer_put_data (&bw, (guint8 *) "hdrl", 4);
1219 avih = gst_avi_mux_start_chunk (&bw, "avih", 0);
1220 /* the AVI header itself */
1221 gst_byte_writer_put_uint32_le (&bw, avimux->avi_hdr.us_frame);
1222 gst_byte_writer_put_uint32_le (&bw, avimux->avi_hdr.max_bps);
1223 gst_byte_writer_put_uint32_le (&bw, avimux->avi_hdr.pad_gran);
1224 gst_byte_writer_put_uint32_le (&bw, avimux->avi_hdr.flags);
1225 gst_byte_writer_put_uint32_le (&bw, avimux->avi_hdr.tot_frames);
1226 gst_byte_writer_put_uint32_le (&bw, avimux->avi_hdr.init_frames);
1227 gst_byte_writer_put_uint32_le (&bw, avimux->avi_hdr.streams);
1228 gst_byte_writer_put_uint32_le (&bw, avimux->avi_hdr.bufsize);
1229 gst_byte_writer_put_uint32_le (&bw, avimux->avi_hdr.width);
1230 gst_byte_writer_put_uint32_le (&bw, avimux->avi_hdr.height);
1231 gst_byte_writer_put_uint32_le (&bw, avimux->avi_hdr.scale);
1232 gst_byte_writer_put_uint32_le (&bw, avimux->avi_hdr.rate);
1233 gst_byte_writer_put_uint32_le (&bw, avimux->avi_hdr.start);
1234 gst_byte_writer_put_uint32_le (&bw, avimux->avi_hdr.length);
1235 gst_avi_mux_end_chunk (&bw, avih);
1238 node = avimux->sinkpads;
1240 GstAviPad *avipad = (GstAviPad *) node->data;
1241 GstAviVideoPad *vidpad = (GstAviVideoPad *) avipad;
1242 GstAviAudioPad *audpad = (GstAviAudioPad *) avipad;
1243 gint codec_size = 0;
1244 guint strh, strl, strf, indx;
1246 /* stream list metadata */
1247 strl = gst_avi_mux_start_chunk (&bw, "LIST", 0);
1248 gst_byte_writer_put_data (&bw, (guint8 *) "strl", 4);
1250 /* generic header */
1251 strh = gst_avi_mux_start_chunk (&bw, "strh", 0);
1252 gst_byte_writer_put_uint32_le (&bw, avipad->hdr.type);
1253 gst_byte_writer_put_uint32_le (&bw, avipad->hdr.fcc_handler);
1254 gst_byte_writer_put_uint32_le (&bw, avipad->hdr.flags);
1255 gst_byte_writer_put_uint32_le (&bw, avipad->hdr.priority);
1256 gst_byte_writer_put_uint32_le (&bw, avipad->hdr.init_frames);
1257 gst_byte_writer_put_uint32_le (&bw, avipad->hdr.scale);
1258 gst_byte_writer_put_uint32_le (&bw, avipad->hdr.rate);
1259 gst_byte_writer_put_uint32_le (&bw, avipad->hdr.start);
1260 gst_byte_writer_put_uint32_le (&bw, avipad->hdr.length);
1261 gst_byte_writer_put_uint32_le (&bw, avipad->hdr.bufsize);
1262 gst_byte_writer_put_uint32_le (&bw, avipad->hdr.quality);
1263 gst_byte_writer_put_uint32_le (&bw, avipad->hdr.samplesize);
1264 gst_byte_writer_put_uint16_le (&bw, 0);
1265 gst_byte_writer_put_uint16_le (&bw, 0);
1266 gst_byte_writer_put_uint16_le (&bw, 0);
1267 gst_byte_writer_put_uint16_le (&bw, 0);
1268 gst_avi_mux_end_chunk (&bw, strh);
1270 if (avipad->is_video) {
1271 codec_size = vidpad->vids_codec_data ?
1272 gst_buffer_get_size (vidpad->vids_codec_data) : 0;
1273 /* the video header */
1274 strf = gst_avi_mux_start_chunk (&bw, "strf", 0);
1275 /* the actual header */
1276 gst_byte_writer_put_uint32_le (&bw, vidpad->vids.size + codec_size);
1277 gst_byte_writer_put_uint32_le (&bw, vidpad->vids.width);
1278 gst_byte_writer_put_uint32_le (&bw, vidpad->vids.height);
1279 gst_byte_writer_put_uint16_le (&bw, vidpad->vids.planes);
1280 gst_byte_writer_put_uint16_le (&bw, vidpad->vids.bit_cnt);
1281 gst_byte_writer_put_uint32_le (&bw, vidpad->vids.compression);
1282 gst_byte_writer_put_uint32_le (&bw, vidpad->vids.image_size);
1283 gst_byte_writer_put_uint32_le (&bw, vidpad->vids.xpels_meter);
1284 gst_byte_writer_put_uint32_le (&bw, vidpad->vids.ypels_meter);
1285 gst_byte_writer_put_uint32_le (&bw, vidpad->vids.num_colors);
1286 gst_byte_writer_put_uint32_le (&bw, vidpad->vids.imp_colors);
1287 if (vidpad->vids_codec_data) {
1289 gst_buffer_map (vidpad->vids_codec_data, &bsize, NULL,
1291 gst_byte_writer_put_data (&bw, bdata, bsize);
1292 gst_buffer_unmap (vidpad->vids_codec_data, bdata, bsize);
1294 gst_avi_mux_end_chunk (&bw, strf);
1296 /* add video property data, mainly for aspect ratio, if any */
1297 if (vidpad->vprp.aspect) {
1301 /* let's be on the safe side */
1302 vidpad->vprp.fields = MIN (vidpad->vprp.fields,
1303 GST_RIFF_VPRP_VIDEO_FIELDS);
1304 /* the vprp header */
1305 vprp = gst_avi_mux_start_chunk (&bw, "vprp", 0);
1306 /* the actual data */
1307 gst_byte_writer_put_uint32_le (&bw, vidpad->vprp.format_token);
1308 gst_byte_writer_put_uint32_le (&bw, vidpad->vprp.standard);
1309 gst_byte_writer_put_uint32_le (&bw, vidpad->vprp.vert_rate);
1310 gst_byte_writer_put_uint32_le (&bw, vidpad->vprp.hor_t_total);
1311 gst_byte_writer_put_uint32_le (&bw, vidpad->vprp.vert_lines);
1312 gst_byte_writer_put_uint32_le (&bw, vidpad->vprp.aspect);
1313 gst_byte_writer_put_uint32_le (&bw, vidpad->vprp.width);
1314 gst_byte_writer_put_uint32_le (&bw, vidpad->vprp.height);
1315 gst_byte_writer_put_uint32_le (&bw, vidpad->vprp.fields);
1317 for (f = 0; f < vidpad->vprp.fields; ++f) {
1318 gst_riff_vprp_video_field_desc *fd;
1320 fd = &(vidpad->vprp.field_info[f]);
1321 gst_byte_writer_put_uint32_le (&bw, fd->compressed_bm_height);
1322 gst_byte_writer_put_uint32_le (&bw, fd->compressed_bm_width);
1323 gst_byte_writer_put_uint32_le (&bw, fd->valid_bm_height);
1324 gst_byte_writer_put_uint32_le (&bw, fd->valid_bm_width);
1325 gst_byte_writer_put_uint32_le (&bw, fd->valid_bm_x_offset);
1326 gst_byte_writer_put_uint32_le (&bw, fd->valid_bm_y_offset);
1327 gst_byte_writer_put_uint32_le (&bw, fd->video_x_t_offset);
1328 gst_byte_writer_put_uint32_le (&bw, fd->video_y_start);
1330 gst_avi_mux_end_chunk (&bw, vprp);
1333 codec_size = audpad->auds_codec_data ?
1334 gst_buffer_get_size (audpad->auds_codec_data) : 0;
1335 /* the audio header */
1336 strf = gst_avi_mux_start_chunk (&bw, "strf", 0);
1337 /* the actual header */
1338 gst_byte_writer_put_uint16_le (&bw, audpad->auds.format);
1339 gst_byte_writer_put_uint16_le (&bw, audpad->auds.channels);
1340 gst_byte_writer_put_uint32_le (&bw, audpad->auds.rate);
1341 gst_byte_writer_put_uint32_le (&bw, audpad->auds.av_bps);
1342 gst_byte_writer_put_uint16_le (&bw, audpad->auds.blockalign);
1343 gst_byte_writer_put_uint16_le (&bw, audpad->auds.size);
1344 gst_byte_writer_put_uint16_le (&bw, codec_size);
1345 if (audpad->auds_codec_data) {
1347 gst_buffer_map (audpad->auds_codec_data, &bsize, NULL,
1349 gst_byte_writer_put_data (&bw, bdata, bsize);
1350 gst_buffer_unmap (vidpad->vids_codec_data, bdata, bsize);
1352 gst_avi_mux_end_chunk (&bw, strf);
1355 /* odml superindex chunk */
1356 if (avipad->idx_index > 0)
1357 indx = gst_avi_mux_start_chunk (&bw, "indx", 0);
1359 indx = gst_avi_mux_start_chunk (&bw, "JUNK", 0);
1360 gst_byte_writer_put_uint16_le (&bw, 4); /* bytes per entry */
1361 gst_byte_writer_put_uint8 (&bw, 0); /* index subtype */
1362 gst_byte_writer_put_uint8 (&bw, GST_AVI_INDEX_OF_INDEXES); /* index type */
1363 gst_byte_writer_put_uint32_le (&bw, avipad->idx_index); /* entries in use */
1364 gst_byte_writer_put_data (&bw, (guint8 *) avipad->tag, 4); /* stream id */
1365 gst_byte_writer_put_uint32_le (&bw, 0); /* reserved */
1366 gst_byte_writer_put_uint32_le (&bw, 0); /* reserved */
1367 gst_byte_writer_put_uint32_le (&bw, 0); /* reserved */
1368 gst_byte_writer_put_data (&bw, (guint8 *) avipad->idx,
1369 GST_AVI_SUPERINDEX_COUNT * sizeof (gst_avi_superindex_entry));
1370 gst_avi_mux_end_chunk (&bw, indx);
1372 /* end strl for this stream */
1373 gst_avi_mux_end_chunk (&bw, strl);
1378 if (avimux->video_pads > 0) {
1381 odml = gst_avi_mux_start_chunk (&bw, "LIST", 0);
1382 gst_byte_writer_put_data (&bw, (guint8 *) "odml", 4);
1383 dmlh = gst_avi_mux_start_chunk (&bw, "dmlh", 0);
1384 gst_byte_writer_put_uint32_le (&bw, avimux->total_frames);
1385 gst_avi_mux_end_chunk (&bw, dmlh);
1386 gst_avi_mux_end_chunk (&bw, odml);
1390 gst_avi_mux_end_chunk (&bw, hdrl);
1396 info = gst_avi_mux_start_chunk (&bw, "LIST", 0);
1397 gst_byte_writer_put_data (&bw, (guint8 *) "INFO", 4);
1399 gst_tag_list_foreach (tags, gst_avi_mux_write_tag, &bw);
1400 if (info + 8 == gst_byte_writer_get_pos (&bw)) {
1401 /* no tags writen, remove the empty INFO LIST as it is useless
1402 * and prevents playback in vlc */
1403 gst_byte_writer_set_pos (&bw, info - 4);
1405 gst_avi_mux_end_chunk (&bw, info);
1410 gst_avi_mux_end_chunk (&bw, riff);
1412 /* avi data header */
1413 gst_byte_writer_put_data (&bw, (guint8 *) "LIST", 4);
1414 gst_byte_writer_put_uint32_le (&bw, avimux->data_size);
1415 gst_byte_writer_put_data (&bw, (guint8 *) "movi", 4);
1417 /* now get the data */
1418 buffer = gst_byte_writer_reset_and_get_buffer (&bw);
1420 /* ... but RIFF includes more than just header */
1421 bdata = gst_buffer_map (buffer, &bsize, NULL, GST_MAP_READWRITE);
1422 size = GST_READ_UINT32_LE (bdata + 4);
1423 size += 8 + avimux->data_size + avimux->idx_size;
1424 GST_WRITE_UINT32_LE (bdata + 4, size);
1426 GST_MEMDUMP_OBJECT (avimux, "avi header", bdata, bsize);
1427 gst_buffer_unmap (buffer, bdata, bsize);
1433 gst_avi_mux_riff_get_avix_header (guint32 datax_size)
1439 buffer = gst_buffer_new_and_alloc (24);
1441 bdata = gst_buffer_map (buffer, &bsize, NULL, GST_MAP_WRITE);
1442 memcpy (bdata + 0, "RIFF", 4);
1443 GST_WRITE_UINT32_LE (bdata + 4, datax_size + 3 * 4);
1444 memcpy (bdata + 8, "AVIX", 4);
1445 memcpy (bdata + 12, "LIST", 4);
1446 GST_WRITE_UINT32_LE (bdata + 16, datax_size);
1447 memcpy (bdata + 20, "movi", 4);
1448 gst_buffer_unmap (buffer, bdata, bsize);
1453 static inline GstBuffer *
1454 gst_avi_mux_riff_get_header (GstAviPad * avipad, guint32 video_frame_size)
1460 buffer = gst_buffer_new_and_alloc (8);
1462 bdata = gst_buffer_map (buffer, &bsize, NULL, GST_MAP_WRITE);
1463 memcpy (bdata + 0, avipad->tag, 4);
1464 GST_WRITE_UINT32_LE (bdata + 4, video_frame_size);
1465 gst_buffer_unmap (buffer, bdata, bsize);
1470 /* write an odml index chunk in the movi list */
1471 static GstFlowReturn
1472 gst_avi_mux_write_avix_index (GstAviMux * avimux, GstAviPad * avipad,
1473 gchar * code, gchar * chunk, gst_avi_superindex_entry * super_index,
1474 gint * super_index_count)
1479 gst_riff_index_entry *entry;
1481 guint32 size, entry_count;
1482 gboolean is_pcm = FALSE;
1483 guint32 pcm_samples = 0;
1487 /* check if it is pcm */
1488 if (avipad && !avipad->is_video) {
1489 GstAviAudioPad *audiopad = (GstAviAudioPad *) avipad;
1490 if (audiopad->auds.format == GST_RIFF_WAVE_FORMAT_PCM) {
1491 pcm_samples = audiopad->samples;
1496 /* allocate the maximum possible */
1497 buffer = gst_buffer_new_and_alloc (32 + 8 * avimux->idx_index);
1499 bdata = gst_buffer_map (buffer, &bsize, NULL, GST_MAP_WRITE);
1502 /* general index chunk info */
1503 memcpy (bdata + 0, chunk, 4); /* chunk id */
1504 GST_WRITE_UINT32_LE (bdata + 4, 0); /* chunk size; fill later */
1505 GST_WRITE_UINT16_LE (bdata + 8, 2); /* index entry is 2 words */
1506 bdata[10] = 0; /* index subtype */
1507 bdata[11] = GST_AVI_INDEX_OF_CHUNKS; /* index type: AVI_INDEX_OF_CHUNKS */
1508 GST_WRITE_UINT32_LE (bdata + 12, 0); /* entries in use; fill later */
1509 memcpy (bdata + 16, code, 4); /* stream to which index refers */
1510 GST_WRITE_UINT64_LE (bdata + 20, avimux->avix_start); /* base offset */
1511 GST_WRITE_UINT32_LE (bdata + 28, 0); /* reserved */
1514 /* now the actual index entries */
1515 i = avimux->idx_index;
1516 entry = avimux->idx;
1518 if (memcmp (&entry->id, code, 4) == 0) {
1519 /* enter relative offset to the data (!) */
1520 GST_WRITE_UINT32_LE (bdata, GUINT32_FROM_LE (entry->offset) + 8);
1521 /* msb is set if not (!) keyframe */
1522 GST_WRITE_UINT32_LE (bdata + 4, GUINT32_FROM_LE (entry->size)
1523 | (GUINT32_FROM_LE (entry->flags)
1524 & GST_RIFF_IF_KEYFRAME ? 0 : 1U << 31));
1531 /* ok, now we know the size and no of entries, fill in where needed */
1532 size = bdata - data;
1533 GST_WRITE_UINT32_LE (data + 4, size - 8);
1534 entry_count = (size - 32) / 8;
1535 GST_WRITE_UINT32_LE (data + 12, entry_count);
1536 gst_buffer_unmap (buffer, data, size);
1538 /* decorate and send */
1539 gst_buffer_set_caps (buffer, GST_PAD_CAPS (avimux->srcpad));
1540 if ((res = gst_pad_push (avimux->srcpad, buffer)) != GST_FLOW_OK)
1543 /* keep track of this in superindex (if room) ... */
1544 if (*super_index_count < GST_AVI_SUPERINDEX_COUNT) {
1545 i = *super_index_count;
1546 super_index[i].offset = GUINT64_TO_LE (avimux->total_data);
1547 super_index[i].size = GUINT32_TO_LE (size);
1549 super_index[i].duration = GUINT32_TO_LE (pcm_samples);
1551 super_index[i].duration = GUINT32_TO_LE (entry_count);
1553 (*super_index_count)++;
1555 GST_WARNING_OBJECT (avimux, "No more room in superindex of stream %s",
1558 /* ... and in size */
1559 avimux->total_data += size;
1560 if (avimux->is_bigfile)
1561 avimux->datax_size += size;
1563 avimux->data_size += size;
1568 /* some other usable functions (thankyou xawtv ;-) ) */
1571 gst_avi_mux_add_index (GstAviMux * avimux, GstAviPad * avipad, guint32 flags,
1574 gchar *code = avipad->tag;
1575 if (avimux->idx_index == avimux->idx_count) {
1576 avimux->idx_count += 256;
1578 g_realloc (avimux->idx,
1579 avimux->idx_count * sizeof (gst_riff_index_entry));
1582 /* in case of pcm audio, we need to count the number of samples for
1583 * putting in the indx entries */
1584 if (!avipad->is_video) {
1585 GstAviAudioPad *audiopad = (GstAviAudioPad *) avipad;
1586 if (audiopad->auds.format == GST_RIFF_WAVE_FORMAT_PCM) {
1587 audiopad->samples += size / audiopad->auds.blockalign;
1591 memcpy (&(avimux->idx[avimux->idx_index].id), code, 4);
1592 avimux->idx[avimux->idx_index].flags = GUINT32_TO_LE (flags);
1593 avimux->idx[avimux->idx_index].offset = GUINT32_TO_LE (avimux->idx_offset);
1594 avimux->idx[avimux->idx_index].size = GUINT32_TO_LE (size);
1595 avimux->idx_index++;
1598 static GstFlowReturn
1599 gst_avi_mux_write_index (GstAviMux * avimux)
1606 buffer = gst_buffer_new_and_alloc (8);
1608 buffdata = gst_buffer_map (buffer, &buffsize, NULL, GST_MAP_WRITE);
1609 memcpy (buffdata + 0, "idx1", 4);
1610 GST_WRITE_UINT32_LE (buffdata + 4,
1611 avimux->idx_index * sizeof (gst_riff_index_entry));
1612 gst_buffer_unmap (buffer, buffdata, buffsize);
1614 gst_buffer_set_caps (buffer, GST_PAD_CAPS (avimux->srcpad));
1615 res = gst_pad_push (avimux->srcpad, buffer);
1616 if (res != GST_FLOW_OK)
1619 buffer = gst_buffer_new ();
1621 buffsize = avimux->idx_index * sizeof (gst_riff_index_entry);
1622 buffdata = (guint8 *) avimux->idx;
1623 avimux->idx = NULL; /* will be free()'ed by gst_buffer_unref() */
1625 gst_buffer_take_memory (buffer,
1626 gst_memory_new_wrapped (0, buffdata, g_free, buffsize, 0, buffsize));
1628 avimux->total_data += buffsize + 8;
1630 gst_buffer_set_caps (buffer, GST_PAD_CAPS (avimux->srcpad));
1631 res = gst_pad_push (avimux->srcpad, buffer);
1632 if (res != GST_FLOW_OK)
1635 avimux->idx_size += avimux->idx_index * sizeof (gst_riff_index_entry) + 8;
1638 avimux->avi_hdr.flags |= GST_RIFF_AVIH_HASINDEX;
1642 static GstFlowReturn
1643 gst_avi_mux_bigfile (GstAviMux * avimux, gboolean last)
1645 GstFlowReturn res = GST_FLOW_OK;
1650 /* first some odml standard index chunks in the movi list */
1651 node = avimux->sinkpads;
1653 GstAviPad *avipad = (GstAviPad *) node->data;
1657 res = gst_avi_mux_write_avix_index (avimux, avipad, avipad->tag,
1658 avipad->idx_tag, avipad->idx, &avipad->idx_index);
1659 if (res != GST_FLOW_OK)
1663 if (avimux->is_bigfile) {
1665 event = gst_event_new_new_segment (FALSE, 1.0, GST_FORMAT_BYTES,
1666 avimux->avix_start, GST_CLOCK_TIME_NONE, avimux->avix_start);
1667 /* if the event succeeds */
1668 gst_pad_push_event (avimux->srcpad, event);
1670 /* rewrite AVIX header */
1671 header = gst_avi_mux_riff_get_avix_header (avimux->datax_size);
1672 gst_buffer_set_caps (header, GST_PAD_CAPS (avimux->srcpad));
1673 res = gst_pad_push (avimux->srcpad, header);
1675 /* go back to current location, at least try */
1676 event = gst_event_new_new_segment (FALSE, 1.0, GST_FORMAT_BYTES,
1677 avimux->total_data, GST_CLOCK_TIME_NONE, avimux->total_data);
1678 gst_pad_push_event (avimux->srcpad, event);
1680 if (res != GST_FLOW_OK)
1682 } else { /* write a standard index in the first riff chunk */
1683 res = gst_avi_mux_write_index (avimux);
1684 /* the index data/buffer is freed by pushing it */
1685 avimux->idx_count = 0;
1686 if (res != GST_FLOW_OK)
1690 avimux->avix_start = avimux->total_data;
1695 avimux->is_bigfile = TRUE;
1696 avimux->numx_frames = 0;
1697 avimux->datax_size = 4; /* movi tag */
1698 avimux->idx_index = 0;
1699 node = avimux->sinkpads;
1701 GstAviPad *avipad = (GstAviPad *) node->data;
1703 if (!avipad->is_video) {
1704 GstAviAudioPad *audiopad = (GstAviAudioPad *) avipad;
1705 audiopad->samples = 0;
1709 header = gst_avi_mux_riff_get_avix_header (0);
1710 avimux->total_data += gst_buffer_get_size (header);
1711 /* avix_start is used as base offset for the odml index chunk */
1712 avimux->idx_offset = avimux->total_data - avimux->avix_start;
1713 gst_buffer_set_caps (header, GST_PAD_CAPS (avimux->srcpad));
1714 return gst_pad_push (avimux->srcpad, header);
1717 /* enough header blabla now, let's go on to actually writing the headers */
1719 static GstFlowReturn
1720 gst_avi_mux_start_file (GstAviMux * avimux)
1727 avimux->total_data = 0;
1728 avimux->total_frames = 0;
1729 avimux->data_size = 4; /* movi tag */
1730 avimux->datax_size = 0;
1731 avimux->num_frames = 0;
1732 avimux->numx_frames = 0;
1733 avimux->avix_start = 0;
1735 avimux->idx_index = 0;
1736 avimux->idx_offset = 0; /* see 10 lines below */
1737 avimux->idx_size = 0;
1738 avimux->idx_count = 0;
1742 avimux->write_header = FALSE;
1743 avimux->restart = FALSE;
1745 /* init streams, see what we've got */
1746 node = avimux->sinkpads;
1747 avimux->audio_pads = avimux->video_pads = 0;
1749 GstAviPad *avipad = (GstAviPad *) node->data;
1753 if (!avipad->is_video) {
1754 /* audio stream numbers must start at 1 iff there is a video stream 0;
1755 * request_pad inserts video pad at head of list, so this test suffices */
1756 if (avimux->video_pads)
1757 avimux->audio_pads++;
1758 avipad->tag = g_strdup_printf ("%02uwb", avimux->audio_pads);
1759 avipad->idx_tag = g_strdup_printf ("ix%02u", avimux->audio_pads);
1760 if (!avimux->video_pads)
1761 avimux->audio_pads++;
1763 avipad->tag = g_strdup_printf ("%02udb", avimux->video_pads);
1764 avipad->idx_tag = g_strdup_printf ("ix%02u", avimux->video_pads++);
1768 caps = gst_caps_copy (gst_pad_get_pad_template_caps (avimux->srcpad));
1769 gst_pad_set_caps (avimux->srcpad, caps);
1770 gst_caps_unref (caps);
1772 /* let downstream know we think in BYTES and expect to do seeking later on */
1773 gst_pad_push_event (avimux->srcpad,
1774 gst_event_new_new_segment (FALSE, 1.0, GST_FORMAT_BYTES, 0, -1, 0));
1777 avimux->avi_hdr.streams = g_slist_length (avimux->sinkpads);
1778 avimux->is_bigfile = FALSE;
1780 header = gst_avi_mux_riff_get_avi_header (avimux);
1781 avimux->total_data += gst_buffer_get_size (header);
1783 gst_buffer_set_caps (header, GST_PAD_CAPS (avimux->srcpad));
1784 res = gst_pad_push (avimux->srcpad, header);
1786 avimux->idx_offset = avimux->total_data;
1791 static GstFlowReturn
1792 gst_avi_mux_stop_file (GstAviMux * avimux)
1794 GstFlowReturn res = GST_FLOW_OK;
1799 /* if bigfile, rewrite header, else write indexes */
1800 /* don't bail out at once if error, still try to re-write header */
1801 if (avimux->video_pads > 0) {
1802 if (avimux->is_bigfile) {
1803 res = gst_avi_mux_bigfile (avimux, TRUE);
1805 res = gst_avi_mux_write_index (avimux);
1809 /* we do our best to make it interleaved at least ... */
1810 if (avimux->audio_pads > 0 && avimux->video_pads > 0)
1811 avimux->avi_hdr.flags |= GST_RIFF_AVIH_ISINTERLEAVED;
1813 /* set rate and everything having to do with that */
1814 avimux->avi_hdr.max_bps = 0;
1815 node = avimux->sinkpads;
1817 GstAviPad *avipad = (GstAviPad *) node->data;
1821 if (!avipad->is_video) {
1822 GstAviAudioPad *audpad = (GstAviAudioPad *) avipad;
1824 /* calculate bps if needed */
1825 if (!audpad->auds.av_bps) {
1826 if (audpad->audio_time) {
1827 audpad->auds.av_bps =
1828 (GST_SECOND * audpad->audio_size) / audpad->audio_time;
1829 /* round bps to nearest multiple of 8;
1830 * which is much more likely to be the (cbr) bitrate in use;
1831 * which in turn results in better timestamp calculation on playback */
1832 audpad->auds.av_bps = GST_ROUND_UP_8 (audpad->auds.av_bps - 4);
1834 GST_ELEMENT_WARNING (avimux, STREAM, MUX,
1835 (_("No or invalid input audio, AVI stream will be corrupt.")),
1837 audpad->auds.av_bps = 0;
1840 gst_avi_mux_audsink_set_fields (avimux, audpad);
1841 avimux->avi_hdr.max_bps += audpad->auds.av_bps;
1842 avipad->hdr.length = gst_util_uint64_scale (audpad->audio_time,
1843 avipad->hdr.rate, avipad->hdr.scale * GST_SECOND);
1845 GstAviVideoPad *vidpad = (GstAviVideoPad *) avipad;
1847 avimux->avi_hdr.max_bps += ((vidpad->vids.bit_cnt + 7) / 8) *
1848 (1000000. / avimux->avi_hdr.us_frame) * vidpad->vids.image_size;
1849 avipad->hdr.length = avimux->total_frames;
1853 /* statistics/total_frames/... */
1854 avimux->avi_hdr.tot_frames = avimux->num_frames;
1856 /* seek and rewrite the header */
1857 header = gst_avi_mux_riff_get_avi_header (avimux);
1858 event = gst_event_new_new_segment (FALSE, 1.0, GST_FORMAT_BYTES,
1859 0, GST_CLOCK_TIME_NONE, 0);
1860 gst_pad_push_event (avimux->srcpad, event);
1862 gst_buffer_set_caps (header, GST_PAD_CAPS (avimux->srcpad));
1863 /* the first error survives */
1864 if (res == GST_FLOW_OK)
1865 res = gst_pad_push (avimux->srcpad, header);
1867 gst_pad_push (avimux->srcpad, header);
1869 event = gst_event_new_new_segment (FALSE, 1.0, GST_FORMAT_BYTES,
1870 avimux->total_data, GST_CLOCK_TIME_NONE, avimux->total_data);
1871 gst_pad_push_event (avimux->srcpad, event);
1873 avimux->write_header = TRUE;
1878 static GstFlowReturn
1879 gst_avi_mux_restart_file (GstAviMux * avimux)
1883 if ((res = gst_avi_mux_stop_file (avimux)) != GST_FLOW_OK)
1886 gst_pad_push_event (avimux->srcpad, gst_event_new_eos ());
1888 return gst_avi_mux_start_file (avimux);
1891 /* handle events (search) */
1893 gst_avi_mux_handle_event (GstPad * pad, GstEvent * event)
1898 avimux = GST_AVI_MUX (gst_pad_get_parent (pad));
1900 switch (GST_EVENT_TYPE (event)) {
1901 case GST_EVENT_TAG:{
1903 GstTagSetter *setter = GST_TAG_SETTER (avimux);
1904 const GstTagMergeMode mode = gst_tag_setter_get_tag_merge_mode (setter);
1906 gst_event_parse_tag (event, &list);
1907 gst_tag_setter_merge_tags (setter, list, mode);
1914 /* now GstCollectPads can take care of the rest, e.g. EOS */
1915 ret = avimux->collect_event (pad, event);
1917 gst_object_unref (avimux);
1922 /* send extra 'padding' data */
1923 static GstFlowReturn
1924 gst_avi_mux_send_pad_data (GstAviMux * avimux, gulong num_bytes)
1930 buffer = gst_buffer_new_and_alloc (num_bytes);
1932 bdata = gst_buffer_map (buffer, &bsize, NULL, GST_MAP_WRITE);
1933 memset (bdata, 0, num_bytes);
1934 gst_buffer_unmap (buffer, bdata, bsize);
1935 gst_buffer_set_caps (buffer, GST_PAD_CAPS (avimux->srcpad));
1936 return gst_pad_push (avimux->srcpad, buffer);
1940 static GstFlowReturn
1941 gst_avi_mux_do_buffer (GstAviMux * avimux, GstAviPad * avipad)
1944 GstBuffer *data, *header;
1945 gulong total_size, pad_bytes = 0;
1949 data = gst_collect_pads_pop (avimux->collect, avipad->collect);
1950 /* arrange downstream running time */
1951 data = gst_buffer_make_writable (data);
1952 GST_BUFFER_TIMESTAMP (data) =
1953 gst_segment_to_running_time (&avipad->collect->segment,
1954 GST_FORMAT_TIME, GST_BUFFER_TIMESTAMP (data));
1956 /* Prepend a special buffer to the first one for some formats */
1957 if (avipad->is_video) {
1958 GstAviVideoPad *vidpad = (GstAviVideoPad *) avipad;
1960 if (vidpad->prepend_buffer) {
1961 GstBuffer *newdata = gst_buffer_merge (vidpad->prepend_buffer, data);
1962 gst_buffer_copy_into (newdata, data, GST_BUFFER_COPY_TIMESTAMPS, 0, -1);
1963 gst_buffer_unref (data);
1964 gst_buffer_unref (vidpad->prepend_buffer);
1967 vidpad->prepend_buffer = NULL;
1971 if (avimux->restart) {
1972 if ((res = gst_avi_mux_restart_file (avimux)) != GST_FLOW_OK)
1976 datasize = gst_buffer_get_size (data);
1978 /* need to restart or start a next avix chunk ? */
1979 if ((avimux->is_bigfile ? avimux->datax_size : avimux->data_size) +
1980 datasize > GST_AVI_MAX_SIZE) {
1981 if (avimux->enable_large_avi) {
1982 if ((res = gst_avi_mux_bigfile (avimux, FALSE)) != GST_FLOW_OK)
1985 if ((res = gst_avi_mux_restart_file (avimux)) != GST_FLOW_OK)
1990 /* get header and record some stats */
1992 pad_bytes = 2 - (datasize & 1);
1994 header = gst_avi_mux_riff_get_header (avipad, datasize);
1995 total_size = gst_buffer_get_size (header) + datasize + pad_bytes;
1997 if (avimux->is_bigfile) {
1998 avimux->datax_size += total_size;
2000 avimux->data_size += total_size;
2003 if (G_UNLIKELY (avipad->hook))
2004 avipad->hook (avimux, avipad, data);
2006 /* the suggested buffer size is the max frame size */
2007 if (avipad->hdr.bufsize < datasize)
2008 avipad->hdr.bufsize = datasize;
2010 if (avipad->is_video) {
2011 avimux->total_frames++;
2013 if (avimux->is_bigfile) {
2014 avimux->numx_frames++;
2016 avimux->num_frames++;
2020 if (!GST_BUFFER_FLAG_IS_SET (data, GST_BUFFER_FLAG_DELTA_UNIT))
2023 GstAviAudioPad *audpad = (GstAviAudioPad *) avipad;
2026 audpad->audio_size += datasize;
2027 audpad->audio_time += GST_BUFFER_DURATION (data);
2030 gst_avi_mux_add_index (avimux, avipad, flags, datasize);
2032 /* prepare buffers for sending */
2033 gst_buffer_set_caps (header, GST_PAD_CAPS (avimux->srcpad));
2034 data = gst_buffer_make_writable (data);
2035 gst_buffer_set_caps (data, GST_PAD_CAPS (avimux->srcpad));
2037 GST_LOG_OBJECT (avimux, "pushing buffers: head, data");
2039 if ((res = gst_pad_push (avimux->srcpad, header)) != GST_FLOW_OK)
2041 if ((res = gst_pad_push (avimux->srcpad, data)) != GST_FLOW_OK)
2045 if ((res = gst_avi_mux_send_pad_data (avimux, pad_bytes)) != GST_FLOW_OK)
2049 /* if any push above fails, we're in trouble with file consistency anyway */
2050 avimux->total_data += total_size;
2051 avimux->idx_offset += total_size;
2056 /* pick the oldest buffer from the pads and push it */
2057 static GstFlowReturn
2058 gst_avi_mux_do_one_buffer (GstAviMux * avimux)
2060 GstAviPad *avipad, *best_pad;
2063 GstClockTime time, best_time, delay;
2065 node = avimux->sinkpads;
2067 best_time = GST_CLOCK_TIME_NONE;
2068 for (; node; node = node->next) {
2069 avipad = (GstAviPad *) node->data;
2071 if (!avipad->collect)
2074 if (!avipad->hdr.fcc_handler)
2075 goto not_negotiated;
2077 buffer = gst_collect_pads_peek (avimux->collect, avipad->collect);
2080 time = GST_BUFFER_TIMESTAMP (buffer);
2081 gst_buffer_unref (buffer);
2083 /* invalid should pass */
2084 if (G_LIKELY (GST_CLOCK_TIME_IS_VALID (time))) {
2085 time = gst_segment_to_running_time (&avipad->collect->segment,
2086 GST_FORMAT_TIME, time);
2087 if (G_UNLIKELY (!GST_CLOCK_TIME_IS_VALID (time))) {
2088 GST_DEBUG_OBJECT (avimux, "clipping buffer on pad %s outside segment",
2089 GST_PAD_NAME (avipad->collect->pad));
2090 buffer = gst_collect_pads_pop (avimux->collect, avipad->collect);
2091 gst_buffer_unref (buffer);
2096 delay = avipad->is_video ? GST_SECOND / 2 : 0;
2098 /* invalid timestamp buffers pass first,
2099 * these are probably initialization buffers */
2100 if (best_pad == NULL || !GST_CLOCK_TIME_IS_VALID (time)
2101 || (GST_CLOCK_TIME_IS_VALID (best_time) && time + delay < best_time)) {
2103 best_time = time + delay;
2108 GST_LOG_OBJECT (avimux, "selected pad %s with time %" GST_TIME_FORMAT,
2109 GST_PAD_NAME (best_pad->collect->pad), GST_TIME_ARGS (best_time));
2111 return gst_avi_mux_do_buffer (avimux, best_pad);
2113 /* simply finish off the file and send EOS */
2114 gst_avi_mux_stop_file (avimux);
2115 gst_pad_push_event (avimux->srcpad, gst_event_new_eos ());
2116 return GST_FLOW_UNEXPECTED;
2122 GST_ELEMENT_ERROR (avimux, CORE, NEGOTIATION, (NULL),
2123 ("pad %s not negotiated", GST_PAD_NAME (avipad->collect->pad)));
2124 return GST_FLOW_NOT_NEGOTIATED;
2128 static GstFlowReturn
2129 gst_avi_mux_collect_pads (GstCollectPads * pads, GstAviMux * avimux)
2133 if (G_UNLIKELY (avimux->write_header)) {
2134 if ((res = gst_avi_mux_start_file (avimux)) != GST_FLOW_OK)
2138 return gst_avi_mux_do_one_buffer (avimux);
2143 gst_avi_mux_get_property (GObject * object,
2144 guint prop_id, GValue * value, GParamSpec * pspec)
2148 avimux = GST_AVI_MUX (object);
2152 g_value_set_boolean (value, avimux->enable_large_avi);
2155 G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
2161 gst_avi_mux_set_property (GObject * object,
2162 guint prop_id, const GValue * value, GParamSpec * pspec)
2166 avimux = GST_AVI_MUX (object);
2170 avimux->enable_large_avi = g_value_get_boolean (value);
2173 G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
2178 static GstStateChangeReturn
2179 gst_avi_mux_change_state (GstElement * element, GstStateChange transition)
2182 GstStateChangeReturn ret;
2184 avimux = GST_AVI_MUX (element);
2186 switch (transition) {
2187 case GST_STATE_CHANGE_READY_TO_PAUSED:
2188 gst_collect_pads_start (avimux->collect);
2190 case GST_STATE_CHANGE_PAUSED_TO_PLAYING:
2192 case GST_STATE_CHANGE_PAUSED_TO_READY:
2193 gst_collect_pads_stop (avimux->collect);
2199 ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
2200 if (ret == GST_STATE_CHANGE_FAILURE)
2203 switch (transition) {
2204 case GST_STATE_CHANGE_PLAYING_TO_PAUSED:
2206 case GST_STATE_CHANGE_PAUSED_TO_READY:
2207 gst_avi_mux_reset (avimux);
2209 case GST_STATE_CHANGE_READY_TO_NULL: