gstreamer/subprojects/gst-plugins-good/tests/examples/splitmux/splitmuxsrc-extract.c
Jan Schmidt 682db96a41 splitmuxsrc: Add add-fragment signal and examples
Add a signal that allows adding fragments with a specific offset
and duration directly to splitmuxsrc's list. By providing the
fragment's offset on the playback timeline and duration directly,
splitmuxsrc doesn't need to measure the fragment making for faster
startup times.

Add a bus message that's published when fragments are measured,
reporting the offset and duration, so they can be cached by an
application and used on future invocations.

Add examples for handling the bus message and using the 'add-fragment'
signal.

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/7053>
2024-08-02 15:30:04 +10:00

135 lines
3.9 KiB
C

/* GStreamer
* Copyright (C) 2024 Jan Schmidt <jan@centricular.com>
*
* 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., 51 Franklin St, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
/*
* This example uses splitmuxsrc to play a set of splitmuxed-files,
* listening for the fragment-info messages from splitmuxsrc
* and writing a CSV file with the fragment offsets and durations
*/
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <gst/gst.h>
GMainLoop *loop;
FILE *out_csv = NULL;
gsize num_fragments = 0;
static gboolean
message_handler (GstBus * bus, GstMessage * message, gpointer data)
{
GstElement *pipe = data;
if (message->type == GST_MESSAGE_ELEMENT) {
const GstStructure *s = gst_message_get_structure (message);
const gchar *name = gst_structure_get_name (s);
if (strcmp (name, "splitmuxsrc-fragment-info") == 0) {
const gchar *fname = gst_structure_get_string (s, "location");
GstClockTime start_offset, duration;
if (!gst_structure_get_uint64 (s, "fragment-offset", &start_offset) ||
!gst_structure_get_uint64 (s, "fragment-duration", &duration)) {
g_assert_not_reached ();
}
fprintf (out_csv, "\"%s\",%" G_GUINT64_FORMAT ",%" G_GUINT64_FORMAT "\n",
fname, start_offset, duration);
num_fragments++;
}
} else if (message->type == GST_MESSAGE_EOS) {
g_main_loop_quit (loop);
} else if (message->type == GST_MESSAGE_STATE_CHANGED) {
GstState old_state, new_state, pending_state;
gst_message_parse_state_changed (message, &old_state, &new_state,
&pending_state);
if (GST_MESSAGE_SRC (message) == GST_OBJECT (pipe)
&& new_state == GST_STATE_PLAYING) {
g_print ("splitmuxsrc scanned %" G_GSIZE_FORMAT " files. Exiting\n",
num_fragments);
g_main_loop_quit (loop);
}
}
return TRUE;
}
static void
on_pad_added (GstElement * src, GstPad * pad, GstBin * pipe)
{
GstElement *sink = gst_element_factory_make ("fakesink", NULL);
gst_bin_add (pipe, sink);
GstPad *sinkpad = gst_element_get_static_pad (sink, "sink");
gst_pad_link (pad, sinkpad);
gst_object_unref (sinkpad);
gst_element_sync_state_with_parent (sink);
}
int
main (int argc, char *argv[])
{
GstElement *pipe;
GstElement *src;
GstBus *bus;
gst_init (&argc, &argv);
if (argc < 3) {
g_printerr
("Usage: %s *.mp4 out.csv\n Pass splitmux file glob and fragment info will be dumped to out.csv\n",
argv[0]);
return 1;
}
out_csv = fopen (argv[2], "w");
if (out_csv == NULL) {
g_printerr ("Failed to open output file %s", argv[2]);
return 2;
}
pipe = gst_pipeline_new (NULL);
src = gst_element_factory_make ("splitmuxsrc", "src");
g_assert (src != NULL);
/* Set the files glob on src */
g_object_set (src, "location", argv[1], NULL);
/* Connect to pad-added to attach fakesink elements */
g_signal_connect (src, "pad-added", G_CALLBACK (on_pad_added), pipe);
gst_bin_add (GST_BIN (pipe), src);
bus = gst_element_get_bus (pipe);
gst_bus_add_watch (bus, message_handler, pipe);
gst_object_unref (bus);
gst_element_set_state (pipe, GST_STATE_PLAYING);
loop = g_main_loop_new (NULL, FALSE);
g_main_loop_run (loop);
fclose (out_csv);
gst_element_set_state (pipe, GST_STATE_NULL);
gst_object_unref (pipe);
return 0;
}