/* AVI muxer plugin for GStreamer * Copyright (C) 2002 Ronald Bultje * * 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 */ #ifdef HAVE_CONFIG_H #include "config.h" #endif #include "gst/gst-i18n-plugin.h" #include #include #include #include "gstavimux.h" #ifndef LE_FROM_GUINT16 #define LE_FROM_GUINT16 GUINT16_FROM_LE #endif #ifndef LE_FROM_GUINT32 #define LE_FROM_GUINT32 GUINT32_FROM_LE #endif /* AviMux signals and args */ enum { /* FILL ME */ LAST_SIGNAL }; enum { ARG_0, ARG_BIGFILE, }; 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 ]; " "video/x-jpeg, " "width = (int) [ 16, 4096 ], " "height = (int) [ 16, 4096 ]; " "video/x-divx, " "width = (int) [ 16, 4096 ], " "height = (int) [ 16, 4096 ], " "divxversion = (int) [ 3, 5 ]; " "video/x-xvid, " "width = (int) [ 16, 4096 ], " "height = (int) [ 16, 4096 ]; " "video/x-3ivx, " "width = (int) [ 16, 4096 ], " "height = (int) [ 16, 4096 ]; " "video/x-msmpeg, " "width = (int) [ 16, 4096 ], " "height = (int) [ 16, 4096 ], " "msmpegversion = (int) [ 41, 43 ]; " "video/mpeg, " "width = (int) [ 16, 4096 ], " "height = (int) [ 16, 4096 ], " "mpegversion = (int) 1, " "systemstream = (boolean) FALSE; " "video/x-h263, " "width = (int) [ 16, 4096 ], " "height = (int) [ 16, 4096 ]; " "video/x-dv, " "width = (int) 720, " "height = (int) { 576, 480 }, " "systemstream = (boolean) FALSE; " "video/x-huffyuv, " "width = (int) [ 16, 4096 ], " "height = (int) [ 16, 4096 ]" ) ); 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/x-vorbis, " "rate = (int) [ 1000, 96000 ], " "channels = (int) [ 1, 2 ]; " "audio/x-ac3, " "rate = (int) [ 1000, 96000 ], " "channels = (int) [ 1, 2 ]" ) ); static void gst_avimux_base_init (gpointer g_class); static void gst_avimux_class_init (GstAviMuxClass *klass); static void gst_avimux_init (GstAviMux *avimux); static void gst_avimux_loop (GstElement *element); static gboolean gst_avimux_handle_event (GstPad *pad, GstEvent *event); static GstPad* gst_avimux_request_new_pad (GstElement *element, GstPadTemplate *templ, const gchar *name); static void gst_avimux_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec); static void gst_avimux_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec); static GstElementStateReturn gst_avimux_change_state (GstElement *element); static GstElementClass *parent_class = NULL; /*static guint gst_avimux_signals[LAST_SIGNAL] = { 0 }; */ GType gst_avimux_get_type (void) { static GType avimux_type = 0; if (!avimux_type) { static const GTypeInfo avimux_info = { sizeof(GstAviMuxClass), gst_avimux_base_init, NULL, (GClassInitFunc)gst_avimux_class_init, NULL, NULL, sizeof(GstAviMux), 0, (GInstanceInitFunc)gst_avimux_init, }; avimux_type = g_type_register_static(GST_TYPE_ELEMENT, "GstAviMux", &avimux_info, 0); } return avimux_type; } static void gst_avimux_base_init (gpointer g_class) { GstElementClass *element_class = GST_ELEMENT_CLASS (g_class); static GstElementDetails gst_avimux_details = GST_ELEMENT_DETAILS ( "Avi multiplexer", "Codec/Muxer", "Muxes audio and video into an avi stream", "Ronald Bultje " ); 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_avimux_details); } static void gst_avimux_class_init (GstAviMuxClass *klass) { GObjectClass *gobject_class; GstElementClass *gstelement_class; gobject_class = (GObjectClass*)klass; gstelement_class = (GstElementClass*)klass; parent_class = g_type_class_ref (GST_TYPE_ELEMENT); g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_BIGFILE, g_param_spec_boolean("bigfile","Bigfile Support", "Support for openDML-2.0 (big) AVI files", 0,G_PARAM_READWRITE)); gstelement_class->request_new_pad = gst_avimux_request_new_pad; gstelement_class->change_state = gst_avimux_change_state; gstelement_class->get_property = gst_avimux_get_property; gstelement_class->set_property = gst_avimux_set_property; } static const GstEventMask * gst_avimux_get_event_masks (GstPad *pad) { static const GstEventMask gst_avimux_sink_event_masks[] = { { GST_EVENT_EOS, 0 }, { 0, } }; return gst_avimux_sink_event_masks; } static void gst_avimux_init (GstAviMux *avimux) { GstElementClass *klass = GST_ELEMENT_GET_CLASS (avimux); avimux->srcpad = gst_pad_new_from_template ( gst_element_class_get_pad_template (klass, "src"), "src"); gst_element_add_pad (GST_ELEMENT (avimux), avimux->srcpad); GST_FLAG_SET (GST_ELEMENT(avimux), GST_ELEMENT_EVENT_AWARE); avimux->audiosinkpad = NULL; avimux->audio_pad_connected = FALSE; avimux->videosinkpad = NULL; avimux->video_pad_connected = FALSE; avimux->audio_buffer_queue = NULL; avimux->video_buffer_queue = NULL; avimux->num_frames = 0; /* audio/video/AVI header initialisation */ memset(&(avimux->avi_hdr),0,sizeof(gst_riff_avih)); memset(&(avimux->vids_hdr),0,sizeof(gst_riff_strh)); memset(&(avimux->vids),0,sizeof(gst_riff_strf_vids)); memset(&(avimux->auds_hdr),0,sizeof(gst_riff_strh)); memset(&(avimux->auds),0,sizeof(gst_riff_strf_auds)); avimux->vids_hdr.type = GST_MAKE_FOURCC('v','i','d','s'); avimux->vids_hdr.rate = 1000000; avimux->avi_hdr.max_bps = 10000000; avimux->auds_hdr.type = GST_MAKE_FOURCC('a','u','d','s'); avimux->vids_hdr.quality = 0xFFFFFFFF; avimux->auds_hdr.quality = 0xFFFFFFFF; avimux->idx = NULL; avimux->write_header = TRUE; avimux->enable_large_avi = TRUE; gst_element_set_loop_function(GST_ELEMENT(avimux), gst_avimux_loop); } static GstPadLinkReturn gst_avimux_vidsinkconnect (GstPad *pad, const GstCaps *vscaps) { GstAviMux *avimux; GstStructure *structure; const gchar* mimetype; gdouble fps = 0.; gboolean ret; avimux = GST_AVIMUX (gst_pad_get_parent (pad)); GST_DEBUG ("avimux: video sinkconnect triggered on %s", gst_pad_get_name (pad)); structure = gst_caps_get_structure (vscaps, 0); mimetype = gst_structure_get_name (structure); /* global */ avimux->vids.size = sizeof(gst_riff_strf_vids); avimux->vids.planes = 1; ret = gst_structure_get_int (structure, "width", &avimux->vids.width); ret &= gst_structure_get_int (structure, "height", &avimux->vids.height); ret &= gst_structure_get_double (structure, "framerate", &fps); if (!ret) return GST_PAD_LINK_REFUSED; if (fps != 0.) avimux->vids_hdr.scale = avimux->vids_hdr.rate / fps; if (!strcmp (mimetype, "video/x-raw-yuv")) { guint32 format; gst_structure_get_fourcc (structure, "format", &format); avimux->vids.compression = format; switch (format) { case GST_MAKE_FOURCC('Y','U','Y','2'): avimux->vids.bit_cnt = 16; break; case GST_MAKE_FOURCC('I','4','2','0'): avimux->vids.bit_cnt = 12; break; } } else { avimux->vids.bit_cnt = 24; avimux->vids.compression = 0; /* find format */ if (!strcmp (mimetype, "video/x-huffyuv")) { avimux->vids.compression = GST_MAKE_FOURCC('H','F','Y','U'); } else if (!strcmp (mimetype, "video/x-jpeg")) { avimux->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: avimux->vids.compression = GST_MAKE_FOURCC('D','I','V','3'); break; case 4: avimux->vids.compression = GST_MAKE_FOURCC('D','I','V','X'); break; case 5: avimux->vids.compression = GST_MAKE_FOURCC('D','X','5','0'); break; } } else if (!strcmp (mimetype, "video/x-xvid")) { avimux->vids.compression = GST_MAKE_FOURCC('X','V','I','D'); } else if (!strcmp (mimetype, "video/x-3ivx")) { avimux->vids.compression = GST_MAKE_FOURCC('3','I','V','2'); } else if (!strcmp (mimetype, "video/x-msmpeg")) { gint msmpegversion; gst_structure_get_int (structure, "msmpegversion", &msmpegversion); switch (msmpegversion) { case 41: avimux->vids.compression = GST_MAKE_FOURCC('M','P','G','4'); break; case 42: avimux->vids.compression = GST_MAKE_FOURCC('M','P','4','2'); break; case 43: avimux->vids.compression = GST_MAKE_FOURCC('M','P','4','3'); break; } } else if (!strcmp (mimetype, "video/x-dv")) { avimux->vids.compression = GST_MAKE_FOURCC('D','V','S','D'); } else if (!strcmp (mimetype, "video/x-h263")) { avimux->vids.compression = GST_MAKE_FOURCC('H','2','6','3'); } else if (!strcmp (mimetype, "video/mpeg")) { avimux->vids.compression = GST_MAKE_FOURCC('M','P','E','G'); } if (!avimux->vids.compression) { return GST_PAD_LINK_DELAYED; } } avimux->vids_hdr.fcc_handler = avimux->vids.compression; avimux->vids.image_size = avimux->vids.height * avimux->vids.width; avimux->avi_hdr.width = avimux->vids.width; avimux->avi_hdr.height = avimux->vids.height; avimux->avi_hdr.us_frame = avimux->vids_hdr.scale; return GST_PAD_LINK_OK; } static GstPadLinkReturn gst_avimux_audsinkconnect (GstPad *pad, const GstCaps *vscaps) { GstAviMux *avimux; GstStructure *structure; const gchar* mimetype; int i; avimux = GST_AVIMUX (gst_pad_get_parent (pad)); GST_DEBUG ("avimux: audio sinkconnect triggered on %s", gst_pad_get_name (pad)); structure = gst_caps_get_structure (vscaps, 0); mimetype = gst_structure_get_name (structure); /* we want these for all */ gst_structure_get_int (structure, "channels", &i); avimux->auds.channels = i; gst_structure_get_int (structure, "rate", &i); avimux->auds.rate = i; if (!strcmp (mimetype, "audio/x-raw-int")) { avimux->auds.format = GST_RIFF_WAVE_FORMAT_PCM; gst_structure_get_int (structure, "width", &i); avimux->auds.blockalign = i; gst_structure_get_int (structure, "depth", &i); avimux->auds.size = i; /* set some more info straight */ avimux->auds.blockalign /= 8; avimux->auds.blockalign *= avimux->auds.channels; avimux->auds.av_bps = avimux->auds.blockalign * avimux->auds.rate; } else if (!strcmp (mimetype, "audio/mpeg") || !strcmp (mimetype, "audio/x-vorbis") || !strcmp (mimetype, "audio/x-ac3")) { avimux->auds.format = 0; if (!strcmp (mimetype, "audio/mpeg")) { gint layer = 3; gst_structure_get_int (structure, "layer", &layer); switch (layer) { case 3: avimux->auds.format = GST_RIFF_WAVE_FORMAT_MPEGL3; break; case 1: case 2: avimux->auds.format = GST_RIFF_WAVE_FORMAT_MPEGL12; break; } } else if (!strcmp (mimetype, "audio/x-vorbis")) { avimux->auds.format = GST_RIFF_WAVE_FORMAT_VORBIS3; } else if (!strcmp (mimetype, "audio/x-ac3")) { avimux->auds.format = GST_RIFF_WAVE_FORMAT_A52; } avimux->auds.blockalign = 1; avimux->auds.av_bps = 0; avimux->auds.size = 16; if (!avimux->auds.format) { return GST_PAD_LINK_REFUSED; } } avimux->auds_hdr.rate = avimux->auds.blockalign * avimux->auds.rate; avimux->auds_hdr.samplesize = avimux->auds.blockalign; avimux->auds_hdr.scale = avimux->auds.blockalign; return GST_PAD_LINK_OK; } static void gst_avimux_pad_link (GstPad *pad, GstPad *peer, gpointer data) { GstAviMux *avimux = GST_AVIMUX(data); const gchar *padname = gst_pad_get_name (pad); if (pad == avimux->audiosinkpad) { avimux->audio_pad_connected = TRUE; } else if (pad == avimux->videosinkpad) { avimux->video_pad_connected = TRUE; } else { g_warning("Unknown padname '%s'", padname); return; } GST_DEBUG ("pad '%s' connected", padname); } static void gst_avimux_pad_unlink (GstPad *pad, GstPad *peer, gpointer data) { GstAviMux *avimux = GST_AVIMUX(data); const gchar *padname = gst_pad_get_name (pad); if (pad == avimux->audiosinkpad) { avimux->audio_pad_connected = FALSE; avimux->audiosinkpad = NULL; } else if (pad == avimux->videosinkpad) { avimux->video_pad_connected = FALSE; avimux->videosinkpad = NULL; } else { g_warning("Unknown padname '%s'", padname); return; } GST_DEBUG ("pad '%s' unlinked", padname); } static GstPad* gst_avimux_request_new_pad (GstElement *element, GstPadTemplate *templ, const gchar *req_name) { GstAviMux *avimux; GstPad *newpad; GstElementClass *klass = GST_ELEMENT_GET_CLASS (element); g_return_val_if_fail (templ != NULL, NULL); if (templ->direction != GST_PAD_SINK) { g_warning ("avimux: request pad that is not a SINK pad\n"); return NULL; } g_return_val_if_fail (GST_IS_AVIMUX (element), NULL); avimux = GST_AVIMUX (element); if (templ == gst_element_class_get_pad_template (klass, "audio_%d")) { g_return_val_if_fail(avimux->audiosinkpad == NULL, NULL); newpad = gst_pad_new_from_template (templ, "audio_00"); gst_pad_set_link_function (newpad, gst_avimux_audsinkconnect); avimux->audiosinkpad = newpad; } else if (templ == gst_element_class_get_pad_template (klass, "video_%d")) { g_return_val_if_fail(avimux->videosinkpad == NULL, NULL); newpad = gst_pad_new_from_template (templ, "video_00"); gst_pad_set_link_function (newpad, gst_avimux_vidsinkconnect); avimux->videosinkpad = newpad; } else { g_warning ("avimux: this is not our template!\n"); return NULL; } g_signal_connect(newpad, "linked", G_CALLBACK(gst_avimux_pad_link), (gpointer)avimux); g_signal_connect(newpad, "unlinked", G_CALLBACK(gst_avimux_pad_unlink), (gpointer)avimux); gst_element_add_pad (element, newpad); gst_pad_set_event_mask_function(newpad, gst_avimux_get_event_masks); return newpad; } /* maybe some of these functions should be moved to riff.h? */ /* DISCLAIMER: this function is ugly. So be it (i.e. it makes the rest easier) */ static GstBuffer * gst_avimux_riff_get_avi_header (GstAviMux *avimux) { GstBuffer *buffer; guint8 *buffdata; guint16 temp16; guint32 temp32; buffer = gst_buffer_new(); /* first, let's see what actually needs to be in the buffer */ GST_BUFFER_SIZE(buffer) = 0; GST_BUFFER_SIZE(buffer) += 32 + sizeof(gst_riff_avih); /* avi header */ if (avimux->video_pad_connected) { /* we have video */ GST_BUFFER_SIZE(buffer) += 28 + sizeof(gst_riff_strh) + sizeof(gst_riff_strf_vids); /* vid hdr */ GST_BUFFER_SIZE(buffer) += 24; /* odml header */ } if (avimux->audio_pad_connected) { /* we have audio */ GST_BUFFER_SIZE(buffer) += 28 + sizeof(gst_riff_strh) + sizeof(gst_riff_strf_auds); /* aud hdr */ } /* this is the "riff size" */ avimux->header_size = GST_BUFFER_SIZE(buffer); GST_BUFFER_SIZE(buffer) += 12; /* avi data header */ /* allocate the buffer */ buffdata = GST_BUFFER_DATA(buffer) = g_malloc(GST_BUFFER_SIZE(buffer)); /* avi header metadata */ memcpy(buffdata, "RIFF", 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->header_size + avimux->idx_size + avimux->data_size); memcpy(buffdata, &temp32, 4); buffdata += 4; memcpy(buffdata, "AVI ", 4); buffdata += 4; memcpy(buffdata, "LIST", 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->header_size - 4*5); memcpy(buffdata, &temp32, 4); buffdata += 4; memcpy(buffdata, "hdrl", 4); buffdata += 4; memcpy(buffdata, "avih", 4); buffdata += 4; temp32 = LE_FROM_GUINT32(sizeof(gst_riff_avih)); memcpy(buffdata, &temp32, 4); buffdata += 4; /* the AVI header itself */ temp32 = LE_FROM_GUINT32(avimux->avi_hdr.us_frame); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->avi_hdr.max_bps); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->avi_hdr.pad_gran); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->avi_hdr.flags); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->avi_hdr.tot_frames); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->avi_hdr.init_frames); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->avi_hdr.streams); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->avi_hdr.bufsize); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->avi_hdr.width); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->avi_hdr.height); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->avi_hdr.scale); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->avi_hdr.rate); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->avi_hdr.start); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->avi_hdr.length); memcpy(buffdata, &temp32, 4); buffdata += 4; if (avimux->video_pad_connected) { /* video header metadata */ memcpy(buffdata, "LIST", 4); buffdata += 4; temp32 = LE_FROM_GUINT32(sizeof(gst_riff_strh) + sizeof(gst_riff_strf_vids) + 4*5); memcpy(buffdata, &temp32, 4); buffdata += 4; memcpy(buffdata, "strl", 4); buffdata += 4; /* generic header */ memcpy(buffdata, "strh", 4); buffdata += 4; temp32 = LE_FROM_GUINT32(sizeof(gst_riff_strh)); memcpy(buffdata, &temp32, 4); buffdata += 4; /* the actual header */ temp32 = LE_FROM_GUINT32(avimux->vids_hdr.type); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->vids_hdr.fcc_handler); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->vids_hdr.flags); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->vids_hdr.priority); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->vids_hdr.init_frames); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->vids_hdr.scale); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->vids_hdr.rate); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->vids_hdr.start); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->vids_hdr.length); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->vids_hdr.bufsize); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->vids_hdr.quality); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->vids_hdr.samplesize); memcpy(buffdata, &temp32, 4); buffdata += 4; /* the video header */ memcpy(buffdata, "strf", 4); buffdata += 4; temp32 = LE_FROM_GUINT32(sizeof(gst_riff_strf_vids)); memcpy(buffdata, &temp32, 4); buffdata += 4; /* the actual header */ temp32 = LE_FROM_GUINT32(avimux->vids.size); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->vids.width); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->vids.height); memcpy(buffdata, &temp32, 4); buffdata += 4; temp16 = LE_FROM_GUINT16(avimux->vids.planes); memcpy(buffdata, &temp16, 2); buffdata += 2; temp16 = LE_FROM_GUINT16(avimux->vids.bit_cnt); memcpy(buffdata, &temp16, 2); buffdata += 2; temp32 = LE_FROM_GUINT32(avimux->vids.compression); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->vids.image_size); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->vids.xpels_meter); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->vids.ypels_meter); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->vids.num_colors); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->vids.imp_colors); memcpy(buffdata, &temp32, 4); buffdata += 4; } if (avimux->audio_pad_connected) { /* audio header */ memcpy(buffdata, "LIST", 4); buffdata += 4; temp32 = LE_FROM_GUINT32(sizeof(gst_riff_strh) + sizeof(gst_riff_strf_auds) + 4*5); memcpy(buffdata, &temp32, 4); buffdata += 4; memcpy(buffdata, "strl", 4); buffdata += 4; /* generic header */ memcpy(buffdata, "strh", 4); buffdata += 4; temp32 = LE_FROM_GUINT32(sizeof(gst_riff_strh)); memcpy(buffdata, &temp32, 4); buffdata += 4; /* the actual header */ temp32 = LE_FROM_GUINT32(avimux->auds_hdr.type); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->auds_hdr.fcc_handler); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->auds_hdr.flags); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->auds_hdr.priority); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->auds_hdr.init_frames); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->auds_hdr.scale); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->auds_hdr.rate); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->auds_hdr.start); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->auds_hdr.length); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->auds_hdr.bufsize); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->auds_hdr.quality); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->auds_hdr.samplesize); memcpy(buffdata, &temp32, 4); buffdata += 4; /* the audio header */ memcpy(buffdata, "strf", 4); buffdata += 4; temp32 = LE_FROM_GUINT32(sizeof(gst_riff_strf_auds)); memcpy(buffdata, &temp32, 4); buffdata += 4; /* the actual header */ temp16 = LE_FROM_GUINT16(avimux->auds.format); memcpy(buffdata, &temp16, 2); buffdata += 2; temp16 = LE_FROM_GUINT16(avimux->auds.channels); memcpy(buffdata, &temp16, 2); buffdata += 2; temp32 = LE_FROM_GUINT32(avimux->auds.rate); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->auds.av_bps); memcpy(buffdata, &temp32, 4); buffdata += 4; temp16 = LE_FROM_GUINT16(avimux->auds.blockalign); memcpy(buffdata, &temp16, 2); buffdata += 2; temp16 = LE_FROM_GUINT16(avimux->auds.size); memcpy(buffdata, &temp16, 2); buffdata += 2; } if (avimux->video_pad_connected) { /* odml header */ memcpy(buffdata, "LIST", 4); buffdata += 4; temp32 = LE_FROM_GUINT32(sizeof(guint32)+4*3); memcpy(buffdata, &temp32, 4); buffdata += 4; memcpy(buffdata, "odml", 4); buffdata += 4; memcpy(buffdata, "dmlh", 4); buffdata += 4; temp32 = LE_FROM_GUINT32(sizeof(guint32)); memcpy(buffdata, &temp32, 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->total_frames); memcpy(buffdata, &temp32, 4); buffdata += 4; } /* avi data header */ memcpy(buffdata, "LIST", 4); buffdata += 4; temp32 = LE_FROM_GUINT32(avimux->data_size); memcpy(buffdata, &temp32, 4); buffdata += 4; memcpy(buffdata, "movi", 4); return buffer; } static GstBuffer * gst_avimux_riff_get_avix_header (guint32 datax_size) { GstBuffer *buffer; guint8 *buffdata; guint32 temp32; buffer = gst_buffer_new(); GST_BUFFER_SIZE(buffer) = 24; buffdata = GST_BUFFER_DATA(buffer) = g_malloc(GST_BUFFER_SIZE(buffer)); memcpy(buffdata, "LIST", 4); buffdata += 4; temp32 = LE_FROM_GUINT32(datax_size+4*4); memcpy(buffdata, &temp32, 4); buffdata += 4; memcpy(buffdata, "AVIX", 4); buffdata += 4; memcpy(buffdata, "LIST", 4); buffdata += 4; temp32 = LE_FROM_GUINT32(datax_size); memcpy(buffdata, &temp32, 4); buffdata += 4; memcpy(buffdata, "movi", 4); return buffer; } static GstBuffer * gst_avimux_riff_get_video_header (guint32 video_frame_size) { GstBuffer *buffer; guint32 temp32; buffer = gst_buffer_new(); GST_BUFFER_DATA(buffer) = g_malloc(8); GST_BUFFER_SIZE(buffer) = 8; memcpy(GST_BUFFER_DATA(buffer), "00db", 4); temp32 = LE_FROM_GUINT32(video_frame_size); memcpy(GST_BUFFER_DATA(buffer)+4, &temp32, 4); return buffer; } static GstBuffer * gst_avimux_riff_get_audio_header (guint32 audio_sample_size) { GstBuffer *buffer; guint32 temp32; buffer = gst_buffer_new(); GST_BUFFER_DATA(buffer) = g_malloc(8); GST_BUFFER_SIZE(buffer) = 8; memcpy(GST_BUFFER_DATA(buffer), "01wb", 4); temp32 = LE_FROM_GUINT32(audio_sample_size); memcpy(GST_BUFFER_DATA(buffer)+4, &temp32, 4); return buffer; } /* some other usable functions (thankyou xawtv ;-) ) */ static void gst_avimux_add_index (GstAviMux *avimux, guchar *code, guint32 flags, guint32 size) { if (avimux->idx_index == avimux->idx_count) { avimux->idx_count += 256; avimux->idx = realloc(avimux->idx, avimux->idx_count*sizeof(gst_riff_index_entry)); } memcpy(&(avimux->idx[avimux->idx_index].id), code, 4); avimux->idx[avimux->idx_index].flags = LE_FROM_GUINT32(flags); avimux->idx[avimux->idx_index].offset = LE_FROM_GUINT32(avimux->idx_offset); avimux->idx[avimux->idx_index].size = LE_FROM_GUINT32(size); avimux->idx_index++; } static void gst_avimux_write_index (GstAviMux *avimux) { GstBuffer *buffer; guint32 temp32; buffer = gst_buffer_new(); GST_BUFFER_SIZE(buffer) = 8; GST_BUFFER_DATA(buffer) = g_malloc(8); memcpy(GST_BUFFER_DATA(buffer), "idx1", 4); temp32 = LE_FROM_GUINT32(avimux->idx_index * sizeof(gst_riff_index_entry)); memcpy(GST_BUFFER_DATA(buffer)+4, &temp32, 4); gst_pad_push(avimux->srcpad, GST_DATA (buffer)); buffer = gst_buffer_new(); GST_BUFFER_SIZE(buffer) = avimux->idx_index * sizeof(gst_riff_index_entry); GST_BUFFER_DATA(buffer) = (unsigned char*) avimux->idx; avimux->idx = NULL; /* will be free()'ed by gst_buffer_unref() */ avimux->total_data += GST_BUFFER_SIZE(buffer); gst_pad_push(avimux->srcpad, GST_DATA (buffer)); avimux->idx_size += avimux->idx_index * sizeof(gst_riff_index_entry) + 8; /* update header */ avimux->avi_hdr.flags |= GST_RIFF_AVIH_HASINDEX; } static void gst_avimux_bigfile(GstAviMux *avimux, gboolean last) { GstBuffer *header; GstEvent *event; if (avimux->is_bigfile) { /* sarch back */ event = gst_event_new_seek (GST_FORMAT_BYTES | GST_SEEK_METHOD_SET | GST_SEEK_FLAG_FLUSH, avimux->avix_start); /* if the event succeeds */ gst_pad_push(avimux->srcpad, GST_DATA(event)); /* rewrite AVIX header */ header = gst_avimux_riff_get_avix_header(avimux->datax_size); gst_pad_push(avimux->srcpad, GST_DATA (header)); /* go back to current location */ event = gst_event_new_seek (GST_FORMAT_BYTES | GST_SEEK_METHOD_SET | GST_SEEK_FLAG_FLUSH, avimux->total_data); gst_pad_push(avimux->srcpad, GST_DATA(event)); } avimux->avix_start = avimux->total_data; if (last) return; avimux->is_bigfile = TRUE; avimux->numx_frames = 0; avimux->datax_size = 0; header = gst_avimux_riff_get_avix_header(0); avimux->total_data += GST_BUFFER_SIZE(header); gst_pad_push(avimux->srcpad, GST_DATA (header)); } /* enough header blabla now, let's go on to actually writing the headers */ static void gst_avimux_start_file (GstAviMux *avimux) { GstBuffer *header; avimux->total_data = 0; avimux->total_frames = 0; avimux->data_size = 4; /* ? */ avimux->datax_size = 0; avimux->num_frames = 0; avimux->numx_frames = 0; avimux->audio_size = 0; avimux->audio_time = 0; avimux->avix_start = 0; avimux->idx_index = 0; avimux->idx_offset = 0; /* see 10 lines below */ avimux->idx_size = 0; avimux->idx_count = 0; avimux->idx = NULL; /* header */ avimux->avi_hdr.streams = (avimux->video_pad_connected?1:0) + (avimux->audio_pad_connected?1:0); avimux->is_bigfile = FALSE; header = gst_avimux_riff_get_avi_header(avimux); avimux->total_data += GST_BUFFER_SIZE(header); avimux->idx_offset = avimux->total_data; gst_pad_push(avimux->srcpad, GST_DATA (header)); avimux->write_header = FALSE; avimux->restart = FALSE; } static void gst_avimux_stop_file (GstAviMux *avimux) { GstEvent *event; GstBuffer *header; /* if bigfile, rewrite header, else write indexes */ if (avimux->video_pad_connected) { if (avimux->is_bigfile) { gst_avimux_bigfile(avimux, TRUE); avimux->idx_size = 0; } else { gst_avimux_write_index(avimux); } } /* statistics/total_frames/... */ avimux->avi_hdr.tot_frames = avimux->num_frames; if (avimux->video_pad_connected) { avimux->vids_hdr.length = avimux->num_frames; } if (avimux->audio_pad_connected) { avimux->auds_hdr.length = (avimux->audio_time * avimux->auds.rate)/GST_SECOND; } /* set rate and everything having to do with that */ avimux->avi_hdr.max_bps = 0; if (avimux->audio_pad_connected) { /* calculate bps if needed */ if (!avimux->auds.av_bps) { if (avimux->audio_time) { avimux->auds_hdr.rate = (GST_SECOND * avimux->audio_size) / avimux->audio_time; } else { GST_ELEMENT_ERROR (avimux, STREAM, MUX, (_("No or invalid input audio, AVI stream will be corrupt.")), (NULL)); avimux->auds_hdr.rate = 0; } avimux->auds.av_bps = avimux->auds_hdr.rate * avimux->auds_hdr.scale; } avimux->avi_hdr.max_bps += avimux->auds.av_bps; } if (avimux->video_pad_connected) { avimux->avi_hdr.max_bps += ((avimux->vids.bit_cnt+7)/8) * (1000000. / avimux->avi_hdr.us_frame) * avimux->vids.image_size; } /* seek and rewrite the header */ header = gst_avimux_riff_get_avi_header(avimux); event = gst_event_new_seek (GST_FORMAT_BYTES | GST_SEEK_METHOD_SET, 0); gst_pad_push(avimux->srcpad, GST_DATA(event)); gst_pad_push(avimux->srcpad, GST_DATA (header)); event = gst_event_new_seek (GST_FORMAT_BYTES | GST_SEEK_METHOD_SET, avimux->total_data); gst_pad_push(avimux->srcpad, GST_DATA(event)); avimux->write_header = TRUE; } static void gst_avimux_restart_file (GstAviMux *avimux) { GstEvent *event; gst_avimux_stop_file(avimux); event = gst_event_new(GST_EVENT_EOS); gst_pad_push(avimux->srcpad, GST_DATA(event)); gst_avimux_start_file(avimux); } /* handle events (search) */ static gboolean gst_avimux_handle_event (GstPad *pad, GstEvent *event) { GstAviMux *avimux; GstEventType type; avimux = GST_AVIMUX (gst_pad_get_parent (pad)); type = event ? GST_EVENT_TYPE (event) : GST_EVENT_UNKNOWN; switch (type) { case GST_EVENT_EOS: /* is this allright? */ if (pad == avimux->videosinkpad) { avimux->video_pad_eos = TRUE; } else if (pad == avimux->audiosinkpad) { avimux->audio_pad_eos = TRUE; } else { g_warning("Unknown pad for EOS!"); } break; default: break; } return TRUE; } /* fill the internal queue for each available pad */ static void gst_avimux_fill_queue (GstAviMux *avimux) { GstBuffer *buffer; while (!avimux->audio_buffer_queue && avimux->audiosinkpad && avimux->audio_pad_connected && GST_PAD_IS_USABLE(avimux->audiosinkpad) && !avimux->audio_pad_eos) { buffer = GST_BUFFER (gst_pad_pull(avimux->audiosinkpad)); if (GST_IS_EVENT(buffer)) { gst_avimux_handle_event(avimux->audiosinkpad, GST_EVENT(buffer)); } else { avimux->audio_buffer_queue = buffer; break; } } while (!avimux->video_buffer_queue && avimux->videosinkpad && avimux->video_pad_connected && GST_PAD_IS_USABLE(avimux->videosinkpad) && !avimux->video_pad_eos) { buffer = GST_BUFFER (gst_pad_pull(avimux->videosinkpad)); if (GST_IS_EVENT(buffer)) { gst_avimux_handle_event(avimux->videosinkpad, GST_EVENT(buffer)); } else { avimux->video_buffer_queue = buffer; break; } } } /* send extra 'padding' data */ static void gst_avimux_send_pad_data (GstAviMux *avimux, gulong num_bytes) { GstBuffer *buffer; buffer = gst_buffer_new(); GST_BUFFER_SIZE(buffer) = num_bytes; GST_BUFFER_DATA(buffer) = g_malloc(num_bytes); memset(GST_BUFFER_DATA(buffer), 0, num_bytes); gst_pad_push(avimux->srcpad, GST_DATA (buffer)); } /* do audio buffer */ static void gst_avimux_do_audio_buffer (GstAviMux *avimux) { GstBuffer *data = avimux->audio_buffer_queue, *header; gulong total_size, pad_bytes = 0; /* write a audio header + index entry */ if (GST_BUFFER_SIZE(data) & 1) { pad_bytes = 2 - (GST_BUFFER_SIZE(data) & 1); } header = gst_avimux_riff_get_audio_header(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; avimux->audio_size += GST_BUFFER_SIZE(data); avimux->audio_time += GST_BUFFER_DURATION(data); gst_avimux_add_index(avimux, "01wb", 0x0, GST_BUFFER_SIZE(data)); } gst_pad_push(avimux->srcpad, GST_DATA (header)); gst_pad_push(avimux->srcpad, GST_DATA (data)); if (pad_bytes) { gst_avimux_send_pad_data(avimux, pad_bytes); } avimux->total_data += total_size; avimux->idx_offset += total_size; avimux->audio_buffer_queue = NULL; } /* do video buffer */ static void gst_avimux_do_video_buffer (GstAviMux *avimux) { GstBuffer *data = avimux->video_buffer_queue, *header; gulong total_size, pad_bytes = 0; if (avimux->restart) gst_avimux_restart_file(avimux); /* write a video header + index entry */ if ((avimux->is_bigfile?avimux->datax_size:avimux->data_size)+GST_BUFFER_SIZE(data)>1024*1024*2000) { if (avimux->enable_large_avi) gst_avimux_bigfile(avimux, FALSE); else gst_avimux_restart_file(avimux); } if (GST_BUFFER_SIZE(data) & 1) { pad_bytes = 2 - (GST_BUFFER_SIZE(data) & 1); } header = gst_avimux_riff_get_video_header(GST_BUFFER_SIZE(data)); total_size = GST_BUFFER_SIZE(header) + GST_BUFFER_SIZE(data) + pad_bytes; avimux->total_frames++; if (avimux->is_bigfile) { avimux->datax_size += total_size; avimux->numx_frames++; } else { guint flags = 0x2; if (GST_BUFFER_FLAG_IS_SET (data, GST_BUFFER_KEY_UNIT)) flags |= 0x10; avimux->data_size += total_size; avimux->num_frames++; gst_avimux_add_index(avimux, "00db", flags, GST_BUFFER_SIZE(data)); } gst_pad_push(avimux->srcpad, GST_DATA (header)); gst_pad_push(avimux->srcpad, GST_DATA (data)); if (pad_bytes) { gst_avimux_send_pad_data(avimux, pad_bytes); } avimux->total_data += total_size; avimux->idx_offset += total_size; avimux->video_buffer_queue = NULL; } /* take the oldest buffer in our internal queue and push-it */ static gboolean gst_avimux_do_one_buffer (GstAviMux *avimux) { if (avimux->video_buffer_queue && avimux->audio_buffer_queue) { if (GST_BUFFER_TIMESTAMP(avimux->video_buffer_queue) <= GST_BUFFER_TIMESTAMP(avimux->audio_buffer_queue)) gst_avimux_do_video_buffer(avimux); else gst_avimux_do_audio_buffer(avimux); } else if (avimux->video_buffer_queue || avimux->audio_buffer_queue) { if (avimux->video_buffer_queue) gst_avimux_do_video_buffer(avimux); else gst_avimux_do_audio_buffer(avimux); } else { /* simply finish off the file and send EOS */ gst_avimux_stop_file(avimux); gst_pad_push(avimux->srcpad, GST_DATA(gst_event_new(GST_EVENT_EOS))); gst_element_set_eos(GST_ELEMENT(avimux)); return FALSE; } return TRUE; } static void gst_avimux_loop (GstElement *element) { GstAviMux *avimux; avimux = GST_AVIMUX(element); /* first fill queue (some elements only set caps when * flowing data), then write header */ gst_avimux_fill_queue(avimux); if (avimux->write_header) gst_avimux_start_file(avimux); gst_avimux_do_one_buffer(avimux); } static void gst_avimux_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec) { GstAviMux *avimux; /* it's not null if we got it, but it might not be ours */ g_return_if_fail(GST_IS_AVIMUX(object)); avimux = GST_AVIMUX(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_avimux_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec) { GstAviMux *avimux; /* it's not null if we got it, but it might not be ours */ g_return_if_fail(GST_IS_AVIMUX(object)); avimux = GST_AVIMUX(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 GstElementStateReturn gst_avimux_change_state (GstElement *element) { GstAviMux *avimux; gint transition = GST_STATE_TRANSITION (element); g_return_val_if_fail(GST_IS_AVIMUX(element), GST_STATE_FAILURE); avimux = GST_AVIMUX(element); switch (transition) { case GST_STATE_PAUSED_TO_PLAYING: avimux->video_pad_eos = avimux->audio_pad_eos = FALSE; break; } if (GST_ELEMENT_CLASS (parent_class)->change_state) return GST_ELEMENT_CLASS (parent_class)->change_state (element); return GST_STATE_SUCCESS; }