/* GStreamer * Copyright (C) <1999> Erik Walthinsen * * 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. */ #ifdef HAVE_CONFIG_H #include "config.h" #endif #include #include #include /*#define DEBUG_ENABLED */ #include "gstmpeg1systemencode.h" #include "main.h" /* GstMPEG1SystemEncode signals and args */ enum { /* FILL ME */ LAST_SIGNAL }; enum { ARG_0 /* FILL ME */ }; static GstStaticPadTemplate src_factory = GST_STATIC_PAD_TEMPLATE ("src", GST_PAD_SRC, GST_PAD_ALWAYS, GST_STATIC_CAPS ("video/mpeg, " "systemstream = (boolean) TRUE") ); static GstStaticPadTemplate video_sink_factory = GST_STATIC_PAD_TEMPLATE ("video_%d", GST_PAD_SINK, GST_PAD_REQUEST, GST_STATIC_CAPS ("video/mpeg, " "mpegversion = (int) 1, " "systemstream = (boolean) FALSE") ); static GstStaticPadTemplate audio_sink_factory = GST_STATIC_PAD_TEMPLATE ("audio_%d", GST_PAD_SINK, GST_PAD_REQUEST, GST_STATIC_CAPS ("audio/mpeg, " "mpegversion = (int) 1, " "layer = (int) [ 1, 2 ] ") ); static void gst_system_encode_class_init (GstMPEG1SystemEncodeClass * klass); static void gst_system_encode_base_init (GstMPEG1SystemEncodeClass * klass); static void gst_system_encode_init (GstMPEG1SystemEncode * system_encode); static GstPad *gst_system_encode_request_new_pad (GstElement * element, GstPadTemplate * templ, const gchar * unused); static void gst_system_encode_chain (GstPad * pad, GstData * _data); static void gst_system_encode_set_property (GObject * object, guint prop_id, const GValue * value, GParamSpec * pspec); static void gst_system_encode_get_property (GObject * object, guint prop_id, GValue * value, GParamSpec * pspec); static GstElementClass *parent_class = NULL; /*static guint gst_system_encode_signals[LAST_SIGNAL] = { 0 }; */ GType gst_mpeg1_system_encode_get_type (void) { static GType system_encode_type = 0; if (!system_encode_type) { static const GTypeInfo system_encode_info = { sizeof (GstMPEG1SystemEncodeClass), (GBaseInitFunc) gst_system_encode_base_init, NULL, (GClassInitFunc) gst_system_encode_class_init, NULL, NULL, sizeof (GstMPEG1SystemEncode), 0, (GInstanceInitFunc) gst_system_encode_init, NULL }; system_encode_type = g_type_register_static (GST_TYPE_ELEMENT, "GstMPEG1SystemEncode", &system_encode_info, 0); } return system_encode_type; } static void gst_system_encode_base_init (GstMPEG1SystemEncodeClass * klass) { GstElementClass *element_class = GST_ELEMENT_CLASS (klass); gst_element_class_add_static_pad_template (element_class, &src_factory); gst_element_class_add_static_pad_template (element_class, &audio_sink_factory); gst_element_class_add_static_pad_template (element_class, &video_sink_factory); gst_element_class_set_details_simple (element_class, "MPEG-1 muxer", "Codec/Muxer", "Multiplexes MPEG-1 Streams", "Wim Taymans "); } static void gst_system_encode_class_init (GstMPEG1SystemEncodeClass * klass) { GObjectClass *gobject_class; GstElementClass *gstelement_class; gobject_class = (GObjectClass *) klass; gstelement_class = (GstElementClass *) klass; parent_class = g_type_class_peek_parent (klass); gobject_class->set_property = gst_system_encode_set_property; gobject_class->get_property = gst_system_encode_get_property; gstelement_class->request_new_pad = gst_system_encode_request_new_pad; } static void gst_system_encode_init (GstMPEG1SystemEncode * system_encode) { system_encode->srcpad = gst_pad_new_from_static_template (&src_factory, "src"); gst_element_add_pad (GST_ELEMENT (system_encode), system_encode->srcpad); system_encode->video_buffer = mpeg1mux_buffer_new (BUFFER_TYPE_VIDEO, 0xE0); system_encode->audio_buffer = mpeg1mux_buffer_new (BUFFER_TYPE_AUDIO, 0xC0); system_encode->have_setup = FALSE; system_encode->mta = NULL; system_encode->packet_size = 2048; system_encode->lock = g_mutex_new (); system_encode->current_pack = system_encode->packets_per_pack = 3; system_encode->video_delay_ms = 0; system_encode->audio_delay_ms = 0; system_encode->sectors_delay = 0; system_encode->startup_delay = ~1; system_encode->which_streams = 0; system_encode->num_audio_pads = 0; system_encode->num_video_pads = 0; system_encode->pack = g_malloc (sizeof (Pack_struc)); system_encode->sys_header = g_malloc (sizeof (Sys_header_struc)); system_encode->sector = g_malloc (sizeof (Sector_struc)); } static GstPad * gst_system_encode_request_new_pad (GstElement * element, GstPadTemplate * templ, const gchar * unused) { GstMPEG1SystemEncode *system_encode; gchar *name = NULL; GstPad *newpad; g_return_val_if_fail (templ != NULL, NULL); if (templ->direction != GST_PAD_SINK) { g_warning ("system_encode: request pad that is not a SINK pad\n"); return NULL; } system_encode = GST_SYSTEM_ENCODE (element); if (templ == gst_static_pad_template_get (&audio_sink_factory)) { name = g_strdup_printf ("audio_%02d", system_encode->num_audio_pads); g_print ("%s\n", name); newpad = gst_pad_new_from_template (templ, name); gst_pad_set_element_private (newpad, GINT_TO_POINTER (system_encode->num_audio_pads)); system_encode->audio_pad[system_encode->num_audio_pads] = newpad; system_encode->num_audio_pads++; system_encode->which_streams |= STREAMS_AUDIO; } else if (templ == gst_static_pad_template_get (&video_sink_factory)) { name = g_strdup_printf ("video_%02d", system_encode->num_video_pads); g_print ("%s\n", name); newpad = gst_pad_new_from_template (templ, name); gst_pad_set_element_private (newpad, GINT_TO_POINTER (system_encode->num_video_pads)); system_encode->video_pad[system_encode->num_video_pads] = newpad; system_encode->num_video_pads++; system_encode->which_streams |= STREAMS_VIDEO; } else { g_warning ("system_encode: this is not our template!\n"); return NULL; } gst_pad_set_chain_function (newpad, gst_system_encode_chain); gst_element_add_pad (GST_ELEMENT (system_encode), newpad); return newpad; } /* return a list of all the highest prioripty streams */ static GList * gst_system_encode_pick_streams (GList * mta, GstMPEG1SystemEncode * system_encode) { guint64 lowest = ~1; GST_DEBUG ("pick_streams: %" G_GINT64_FORMAT ", %" G_GINT64_FORMAT, system_encode->video_buffer->next_frame_time, system_encode->audio_buffer->next_frame_time); if (system_encode->which_streams & STREAMS_VIDEO) { if (system_encode->video_buffer->next_frame_time < lowest - system_encode->video_delay) { lowest = system_encode->video_buffer->next_frame_time; } } if (system_encode->which_streams & STREAMS_AUDIO) { if (system_encode->audio_buffer->next_frame_time < lowest - system_encode->audio_delay) { lowest = system_encode->audio_buffer->next_frame_time; } } if (system_encode->which_streams & STREAMS_VIDEO) { if (system_encode->video_buffer->next_frame_time == lowest) { mta = g_list_append (mta, system_encode->video_buffer); } } if (system_encode->which_streams & STREAMS_AUDIO) { if (system_encode->audio_buffer->next_frame_time == lowest) { mta = g_list_append (mta, system_encode->audio_buffer); } } return mta; } static gboolean gst_system_encode_have_data (GstMPEG1SystemEncode * system_encode) { if (system_encode->which_streams == (STREAMS_VIDEO | STREAMS_AUDIO)) { if (MPEG1MUX_BUFFER_QUEUED (system_encode->audio_buffer) > 2 && MPEG1MUX_BUFFER_SPACE (system_encode->audio_buffer) > system_encode->packet_size * 2 && MPEG1MUX_BUFFER_QUEUED (system_encode->video_buffer) > 2 && MPEG1MUX_BUFFER_SPACE (system_encode->video_buffer) > system_encode->packet_size * 2) { return TRUE; } } if (system_encode->which_streams == STREAMS_VIDEO) { if (MPEG1MUX_BUFFER_QUEUED (system_encode->video_buffer) > 2 && MPEG1MUX_BUFFER_SPACE (system_encode->video_buffer) > system_encode->packet_size * 2) { return TRUE; } } if (system_encode->which_streams == STREAMS_VIDEO) { if (MPEG1MUX_BUFFER_QUEUED (system_encode->audio_buffer) > 2 && MPEG1MUX_BUFFER_SPACE (system_encode->audio_buffer) > system_encode->packet_size * 2) { return TRUE; } } return FALSE; } static GList * gst_system_encode_update_mta (GstMPEG1SystemEncode * system_encode, GList * mta, gulong size) { GList *streams = g_list_first (mta); Mpeg1MuxBuffer *mb = (Mpeg1MuxBuffer *) streams->data; GST_DEBUG ("system_encode::multiplex: update mta"); mpeg1mux_buffer_shrink (mb, size); mta = g_list_remove (mta, mb); return mta; } static void gst_system_setup_multiplex (GstMPEG1SystemEncode * system_encode) { Mpeg1MuxTimecode *video_tc, *audio_tc; system_encode->audio_buffer_size = 4 * 1024; system_encode->video_buffer_size = 46 * 1024; system_encode->bytes_output = 0; system_encode->min_packet_data = system_encode->packet_size - PACK_HEADER_SIZE - SYS_HEADER_SIZE - PACKET_HEADER_SIZE - AFTER_PACKET_LENGTH; system_encode->max_packet_data = system_encode->packet_size - PACKET_HEADER_SIZE - AFTER_PACKET_LENGTH; if (system_encode->which_streams & STREAMS_VIDEO) { system_encode->video_rate = system_encode->video_buffer->info.video.bit_rate * 50; } else system_encode->video_rate = 0; if (system_encode->which_streams & STREAMS_AUDIO) system_encode->audio_rate = system_encode->audio_buffer->info.audio.bit_rate * 128; else system_encode->audio_rate = 0; system_encode->data_rate = system_encode->video_rate + system_encode->audio_rate; system_encode->dmux_rate = ceil ((double) (system_encode->data_rate) * ((double) (system_encode->packet_size) / (double) (system_encode->min_packet_data) + ((double) (system_encode->packet_size) / (double) (system_encode->max_packet_data) * (double) (system_encode->packets_per_pack - 1.))) / (double) (system_encode->packets_per_pack)); system_encode->data_rate = ceil (system_encode->dmux_rate / 50.) * 50; GST_DEBUG ("system_encode::multiplex: data_rate %u, video_rate: %u, audio_rate: %u", system_encode->data_rate, system_encode->video_rate, system_encode->audio_rate); system_encode->video_delay = (double) system_encode->video_delay_ms * (double) (CLOCKS / 1000); system_encode->audio_delay = (double) system_encode->audio_delay_ms * (double) (CLOCKS / 1000); system_encode->mux_rate = ceil (system_encode->dmux_rate / 50.); system_encode->dmux_rate = system_encode->mux_rate * 50.; video_tc = MPEG1MUX_BUFFER_FIRST_TIMECODE (system_encode->video_buffer); audio_tc = MPEG1MUX_BUFFER_FIRST_TIMECODE (system_encode->audio_buffer); GST_DEBUG ("system_encode::video tc %" G_GINT64_FORMAT ", audio tc %" G_GINT64_FORMAT ":", video_tc->DTS, audio_tc->DTS); system_encode->delay = ((double) system_encode->sectors_delay + ceil ((double) video_tc->length / (double) system_encode->min_packet_data) + ceil ((double) video_tc->length / (double) system_encode->min_packet_data)) * (double) system_encode->packet_size / system_encode->dmux_rate * (double) CLOCKS; system_encode->audio_delay += system_encode->delay; system_encode->video_delay += system_encode->delay; system_encode->audio_delay = 0; system_encode->video_delay = 0; system_encode->delay = 0; GST_DEBUG ("system_encode::multiplex: delay %g, mux_rate: %lu", system_encode->delay, system_encode->mux_rate); } static void gst_system_encode_multiplex (GstMPEG1SystemEncode * system_encode) { GList *streams; Mpeg1MuxBuffer *mb = (Mpeg1MuxBuffer *) streams->data; guchar timestamps; guchar buffer_scale; GstBuffer *outbuf; Pack_struc *pack; Sys_header_struc *sys_header; Mpeg1MuxTimecode *tc; gulong buffer_size, non_scaled_buffer_size, total_queued; guint64 PTS, DTS; g_mutex_lock (system_encode->lock); while (gst_system_encode_have_data (system_encode)) { GST_DEBUG ("system_encode::multiplex: multiplexing"); if (!system_encode->have_setup) { gst_system_setup_multiplex (system_encode); system_encode->have_setup = TRUE; } if (system_encode->mta == NULL) { system_encode->mta = gst_system_encode_pick_streams (system_encode->mta, system_encode); } if (system_encode->mta == NULL) break; system_encode->SCR = (guint64) (system_encode->bytes_output + LAST_SCR_BYTE_IN_PACK) * CLOCKS / system_encode->dmux_rate; streams = g_list_first (system_encode->mta); mb = (Mpeg1MuxBuffer *) streams->data; if (system_encode->current_pack == system_encode->packets_per_pack) { create_pack (system_encode->pack, system_encode->SCR, system_encode->mux_rate); create_sys_header (system_encode->sys_header, system_encode->mux_rate, 1, 1, 1, 1, 1, 1, AUDIO_STR_0, 0, system_encode->audio_buffer_size / 128, VIDEO_STR_0, 1, system_encode->video_buffer_size / 1024, system_encode->which_streams); system_encode->current_pack = 0; pack = system_encode->pack; sys_header = system_encode->sys_header; } else { system_encode->current_pack++; pack = NULL; sys_header = NULL; } tc = MPEG1MUX_BUFFER_FIRST_TIMECODE (mb); if (mb->new_frame) { GST_DEBUG ("system_encode::multiplex: new frame"); if (tc->frame_type == FRAME_TYPE_AUDIO || tc->frame_type == FRAME_TYPE_IFRAME || tc->frame_type == FRAME_TYPE_PFRAME) { timestamps = TIMESTAMPS_PTS; } else { timestamps = TIMESTAMPS_PTS_DTS; } } else { timestamps = TIMESTAMPS_NO; } if (tc->frame_type != FRAME_TYPE_AUDIO) { if (tc->PTS < system_encode->startup_delay) system_encode->startup_delay = tc->PTS; } if (tc->frame_type == FRAME_TYPE_AUDIO) { buffer_scale = 0; non_scaled_buffer_size = system_encode->audio_buffer_size; buffer_size = system_encode->audio_buffer_size / 128; PTS = tc->PTS + system_encode->audio_delay + system_encode->startup_delay; DTS = tc->PTS + system_encode->audio_delay + system_encode->startup_delay; } else { buffer_scale = 1; non_scaled_buffer_size = system_encode->video_buffer_size; buffer_size = system_encode->video_buffer_size / 1024; PTS = tc->PTS + system_encode->video_delay; DTS = tc->DTS + system_encode->video_delay; } total_queued = mpeg1mux_buffer_update_queued (mb, system_encode->SCR); if (non_scaled_buffer_size - total_queued >= system_encode->packet_size) { /* write the pack/packet here */ create_sector (system_encode->sector, pack, sys_header, system_encode->packet_size, MPEG1MUX_BUFFER_DATA (mb), mb->stream_id, buffer_scale, buffer_size, TRUE, PTS, DTS, timestamps, system_encode->which_streams); /* update mta */ system_encode->mta = gst_system_encode_update_mta (system_encode, system_encode->mta, system_encode->sector->length_of_packet_data); } else { /* write a padding packet */ create_sector (system_encode->sector, pack, sys_header, system_encode->packet_size, NULL, PADDING_STR, 0, 0, FALSE, 0, 0, TIMESTAMPS_NO, system_encode->which_streams); } outbuf = gst_buffer_new (); GST_BUFFER_DATA (outbuf) = g_malloc (system_encode->sector->length_of_sector); GST_BUFFER_SIZE (outbuf) = system_encode->sector->length_of_sector; memcpy (GST_BUFFER_DATA (outbuf), system_encode->sector->buf, system_encode->sector->length_of_sector); system_encode->bytes_output += GST_BUFFER_SIZE (outbuf); gst_pad_push (system_encode->srcpad, GST_DATA (outbuf)); GST_DEBUG ("system_encode::multiplex: writing %02x", mb->stream_id); } GST_INFO ("system_encode::multiplex: data left in video buffer %lu\n", MPEG1MUX_BUFFER_SPACE (system_encode->video_buffer)); GST_INFO ("system_encode::multiplex: data left in audio buffer %lu\n", MPEG1MUX_BUFFER_SPACE (system_encode->audio_buffer)); g_mutex_unlock (system_encode->lock); } static void gst_system_encode_chain (GstPad * pad, GstData * _data) { GstBuffer *buf = GST_BUFFER (_data); GstMPEG1SystemEncode *system_encode; guchar *data; gulong size; const gchar *padname; gint channel; g_return_if_fail (pad != NULL); g_return_if_fail (GST_IS_PAD (pad)); g_return_if_fail (buf != NULL); system_encode = GST_SYSTEM_ENCODE (GST_OBJECT_PARENT (pad)); data = GST_BUFFER_DATA (buf); size = GST_BUFFER_SIZE (buf); GST_DEBUG ("system_encode::chain: system_encode: have buffer of size %lu", size); padname = GST_OBJECT_NAME (pad); if (strncmp (padname, "audio_", 6) == 0) { channel = atoi (&padname[6]); GST_DEBUG ("gst_system_encode_chain: got audio buffer in from audio channel %02d", channel); mpeg1mux_buffer_queue (system_encode->audio_buffer, buf); } else if (strncmp (padname, "video_", 6) == 0) { channel = atoi (&padname[6]); GST_DEBUG ("gst_system_encode_chain: got video buffer in from video channel %02d", channel); mpeg1mux_buffer_queue (system_encode->video_buffer, buf); } else { g_assert_not_reached (); } gst_system_encode_multiplex (system_encode); gst_buffer_unref (buf); } static void gst_system_encode_set_property (GObject * object, guint prop_id, const GValue * value, GParamSpec * pspec) { GstMPEG1SystemEncode *system_encode; g_return_if_fail (GST_IS_SYSTEM_ENCODE (object)); system_encode = GST_SYSTEM_ENCODE (object); switch (prop_id) { default: G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); break; } } static void gst_system_encode_get_property (GObject * object, guint prop_id, GValue * value, GParamSpec * pspec) { GstMPEG1SystemEncode *src; g_return_if_fail (GST_IS_SYSTEM_ENCODE (object)); src = GST_SYSTEM_ENCODE (object); switch (prop_id) { default: G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); break; } } static gboolean plugin_init (GstPlugin * plugin) { /* this filter needs the getbits functions */ if (!gst_library_load ("gstgetbits")) return FALSE; return gst_element_register (plugin, "mpeg1sysenc", GST_RANK_NONE, GST_TYPE_SYSTEM_ENCODE); } GST_PLUGIN_DEFINE (GST_VERSION_MAJOR, GST_VERSION_MINOR, "mpeg1sysenc", "MPEG-1 system stream encoder", plugin_init, VERSION, "GPL", GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)