mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-28 03:00:35 +00:00
flvmux: Write an index table to the end of the file
This commit is contained in:
Sebastian Dröge
parent
fac0992929
commit
22d95bfc09
2 changed files with 151 additions and 5 deletions
|
|
@ -1,6 +1,6 @@
|
|||
/* GStreamer
|
||||
*
|
||||
* Copyright (c) 2008 Sebastian Dröge <sebastian.droege@collabora.co.uk>
|
||||
* Copyright (c) 2008,2009 Sebastian Dröge <sebastian.droege@collabora.co.uk>
|
||||
*
|
||||
* This library is free software; you can redistribute it and/or
|
||||
* modify it under the terms of the GNU Library General Public
|
||||
|
|
@ -100,6 +100,18 @@ gst_flv_mux_change_state (GstElement * element, GstStateChange transition);
|
|||
|
||||
static void gst_flv_mux_reset (GstElement * element);
|
||||
|
||||
typedef struct
|
||||
{
|
||||
gdouble position;
|
||||
gdouble time;
|
||||
} GstFlvMuxIndexEntry;
|
||||
|
||||
static void
|
||||
gst_flv_mux_index_entry_free (GstFlvMuxIndexEntry * entry)
|
||||
{
|
||||
g_slice_free (GstFlvMuxIndexEntry, entry);
|
||||
}
|
||||
|
||||
static void
|
||||
gst_flv_mux_base_init (gpointer g_class)
|
||||
{
|
||||
|
|
@ -183,6 +195,11 @@ gst_flv_mux_reset (GstElement * element)
|
|||
gst_tag_list_free (mux->tags);
|
||||
mux->tags = NULL;
|
||||
|
||||
g_list_foreach (mux->index, (GFunc) gst_flv_mux_index_entry_free, NULL);
|
||||
g_list_free (mux->index);
|
||||
mux->index = NULL;
|
||||
mux->byte_count = 0;
|
||||
|
||||
mux->state = GST_FLV_MUX_STATE_HEADER;
|
||||
}
|
||||
|
||||
|
|
@ -527,6 +544,22 @@ gst_flv_mux_release_pad (GstElement * element, GstPad * pad)
|
|||
gst_element_remove_pad (element, pad);
|
||||
}
|
||||
|
||||
static GstFlowReturn
|
||||
gst_flv_mux_push (GstFlvMux * mux, GstBuffer * buffer)
|
||||
{
|
||||
mux->byte_count += GST_BUFFER_SIZE (buffer);
|
||||
|
||||
if (GST_BUFFER_TIMESTAMP_IS_VALID (buffer)) {
|
||||
GstFlvMuxIndexEntry *entry = g_slice_new (GstFlvMuxIndexEntry);
|
||||
entry->position = mux->byte_count;
|
||||
entry->time =
|
||||
gst_guint64_to_gdouble (GST_BUFFER_TIMESTAMP (buffer)) / GST_MSECOND;
|
||||
mux->index = g_list_prepend (mux->index, entry);
|
||||
}
|
||||
|
||||
return gst_pad_push (mux->srcpad, buffer);
|
||||
}
|
||||
|
||||
static GstFlowReturn
|
||||
gst_flv_mux_write_metadata (GstFlvMux * mux)
|
||||
{
|
||||
|
|
@ -749,7 +782,7 @@ gst_flv_mux_write_metadata (GstFlvMux * mux)
|
|||
GST_WRITE_UINT32_BE (data + 11 + 13 + 1, tags_written);
|
||||
|
||||
gst_buffer_set_caps (script_tag, GST_PAD_CAPS (mux->srcpad));
|
||||
ret = gst_pad_push (mux->srcpad, script_tag);
|
||||
ret = gst_flv_mux_push (mux, script_tag);
|
||||
|
||||
if (merged_tags)
|
||||
gst_tag_list_free (merged_tags);
|
||||
|
|
@ -782,7 +815,7 @@ gst_flv_mux_write_header (GstFlvMux * mux)
|
|||
|
||||
GST_WRITE_UINT32_BE (data + 9, 0); /* previous tag size */
|
||||
|
||||
ret = gst_pad_push (mux->srcpad, header);
|
||||
ret = gst_flv_mux_push (mux, header);
|
||||
if (ret != GST_FLOW_OK)
|
||||
return ret;
|
||||
|
||||
|
|
@ -825,6 +858,7 @@ next:
|
|||
size += 4;
|
||||
|
||||
tag = gst_buffer_new_and_alloc (size);
|
||||
GST_BUFFER_TIMESTAMP (tag) = timestamp * GST_MSECOND;
|
||||
data = GST_BUFFER_DATA (tag);
|
||||
memset (data, 0, size);
|
||||
|
||||
|
|
@ -899,7 +933,7 @@ next:
|
|||
second_run = FALSE;
|
||||
cpad->sent_codec_data = TRUE;
|
||||
|
||||
ret = gst_pad_push (mux->srcpad, tag);
|
||||
ret = gst_flv_mux_push (mux, tag);
|
||||
if (ret != GST_FLOW_OK) {
|
||||
gst_buffer_unref (buffer);
|
||||
return ret;
|
||||
|
|
@ -917,7 +951,7 @@ next:
|
|||
|
||||
gst_buffer_unref (buffer);
|
||||
|
||||
ret = gst_pad_push (mux->srcpad, tag);
|
||||
ret = gst_flv_mux_push (mux, tag);
|
||||
|
||||
if (ret == GST_FLOW_OK)
|
||||
cpad->last_timestamp = timestamp;
|
||||
|
|
@ -925,6 +959,115 @@ next:
|
|||
return ret;
|
||||
}
|
||||
|
||||
static GstFlowReturn
|
||||
gst_flv_mux_write_index (GstFlvMux * mux)
|
||||
{
|
||||
GstFlowReturn ret = GST_FLOW_OK;
|
||||
GstBuffer *script_tag, *tmp;
|
||||
guint8 *data;
|
||||
GList *l;
|
||||
guint32 index_len;
|
||||
guint32 i, index_skip;
|
||||
|
||||
if (!mux->index)
|
||||
return GST_FLOW_OK;
|
||||
|
||||
script_tag = gst_buffer_new_and_alloc (11);
|
||||
data = GST_BUFFER_DATA (script_tag);
|
||||
|
||||
data[0] = 18;
|
||||
|
||||
/* Data size, unknown for now */
|
||||
data[1] = 0;
|
||||
data[2] = 0;
|
||||
data[3] = 0;
|
||||
|
||||
/* Timestamp */
|
||||
data[4] = data[5] = data[6] = data[7] = 0;
|
||||
|
||||
/* Stream ID */
|
||||
data[8] = data[9] = data[10] = 0;
|
||||
|
||||
tmp = gst_buffer_new_and_alloc (13);
|
||||
data = GST_BUFFER_DATA (tmp);
|
||||
data[0] = 2; /* string */
|
||||
data[1] = 0;
|
||||
data[2] = 0x0a; /* length 10 */
|
||||
memcpy (&data[3], "onMetaData", sizeof ("onMetaData"));
|
||||
|
||||
script_tag = gst_buffer_join (script_tag, tmp);
|
||||
|
||||
tmp = gst_buffer_new_and_alloc (5);
|
||||
data = GST_BUFFER_DATA (tmp);
|
||||
data[0] = 8; /* ECMA array */
|
||||
GST_WRITE_UINT32_BE (data + 1, 2);
|
||||
script_tag = gst_buffer_join (script_tag, tmp);
|
||||
|
||||
mux->index = g_list_reverse (mux->index);
|
||||
index_len = g_list_length (mux->index);
|
||||
|
||||
/* We write at most 128 elements */
|
||||
index_skip = (index_len > 128) ? 1 + index_len / 128 : 1;
|
||||
index_len =
|
||||
(index_len <= 128) ? 1 : (index_len + index_skip - 1) / index_skip;
|
||||
|
||||
tmp = gst_buffer_new_and_alloc (2 + 5 + 1 + 4 + index_len * (1 + 8));
|
||||
data = GST_BUFFER_DATA (tmp);
|
||||
data[0] = 0; /* 5 bytes name */
|
||||
data[1] = 5;
|
||||
memcpy (&data[2], "times", 5);
|
||||
data[7] = 10; /* array */
|
||||
GST_WRITE_UINT32_BE (&data[8], index_len);
|
||||
data += 12;
|
||||
|
||||
for (i = 0, l = mux->index; l; l = l->next, i++) {
|
||||
GstFlvMuxIndexEntry *entry = l->data;
|
||||
|
||||
if (i % index_skip != 0)
|
||||
continue;
|
||||
|
||||
data[0] = 0;
|
||||
GST_WRITE_DOUBLE_BE (&data[1], entry->time);
|
||||
data += 9;
|
||||
}
|
||||
script_tag = gst_buffer_join (script_tag, tmp);
|
||||
|
||||
tmp = gst_buffer_new_and_alloc (2 + 13 + 1 + 4 + index_len * (1 + 8));
|
||||
data = GST_BUFFER_DATA (tmp);
|
||||
data[0] = 0; /* 13 bytes name */
|
||||
data[1] = 13;
|
||||
memcpy (&data[2], "filepositions", 13);
|
||||
data[15] = 10; /* array */
|
||||
GST_WRITE_UINT32_BE (&data[16], index_len);
|
||||
data += 20;
|
||||
|
||||
for (i = 0, l = mux->index; l; l = l->next, i++) {
|
||||
GstFlvMuxIndexEntry *entry = l->data;
|
||||
|
||||
if (i % index_skip != 0)
|
||||
continue;
|
||||
data[0] = 0;
|
||||
GST_WRITE_DOUBLE_BE (&data[1], entry->position);
|
||||
data += 9;
|
||||
}
|
||||
script_tag = gst_buffer_join (script_tag, tmp);
|
||||
|
||||
tmp = gst_buffer_new_and_alloc (4);
|
||||
data = GST_BUFFER_DATA (tmp);
|
||||
GST_WRITE_UINT32_BE (data, GST_BUFFER_SIZE (script_tag));
|
||||
script_tag = gst_buffer_join (script_tag, tmp);
|
||||
|
||||
data = GST_BUFFER_DATA (script_tag);
|
||||
data[1] = ((GST_BUFFER_SIZE (script_tag) - 11 - 4) >> 16) & 0xff;
|
||||
data[2] = ((GST_BUFFER_SIZE (script_tag) - 11 - 4) >> 8) & 0xff;
|
||||
data[3] = ((GST_BUFFER_SIZE (script_tag) - 11 - 4) >> 0) & 0xff;
|
||||
|
||||
gst_buffer_set_caps (script_tag, GST_PAD_CAPS (mux->srcpad));
|
||||
ret = gst_flv_mux_push (mux, script_tag);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
static GstFlowReturn
|
||||
gst_flv_mux_collected (GstCollectPads * pads, gpointer user_data)
|
||||
{
|
||||
|
|
@ -994,6 +1137,7 @@ gst_flv_mux_collected (GstCollectPads * pads, gpointer user_data)
|
|||
if (!eos && best) {
|
||||
return gst_flv_mux_write_buffer (mux, best);
|
||||
} else if (eos) {
|
||||
gst_flv_mux_write_index (mux);
|
||||
gst_pad_push_event (mux->srcpad, gst_event_new_eos ());
|
||||
return GST_FLOW_UNEXPECTED;
|
||||
} else {
|
||||
|
|
|
|||
|
|
@ -76,6 +76,8 @@ typedef struct _GstFlvMux {
|
|||
gboolean have_video;
|
||||
|
||||
GstTagList *tags;
|
||||
GList *index;
|
||||
guint64 byte_count;
|
||||
} GstFlvMux;
|
||||
|
||||
typedef struct _GstFlvMuxClass {
|
||||
|
|
|
|||
Loading…
Reference in a new issue