mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-11-03 16:09:39 +00:00
335c9f28d6
Majorly change the way that splitmuxsink collects incoming data and sends it to the output, so that it makes all decisions about when / where to split files on the input side. Use separate queues for each stream, so they can be grown individually and kept as small as possible. This removes raciness I observed where sometimes some data would end up put in a different output file over multiple runs with the same input. Also fixes hangs with input queues getting full and causing muxing to stall out.
172 lines
4.8 KiB
C
172 lines
4.8 KiB
C
/* GStreamer split muxer bin
|
|
* Copyright (C) 2014 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.
|
|
*/
|
|
|
|
#ifndef __GST_SPLITMUXSINK_H__
|
|
#define __GST_SPLITMUXSINK_H__
|
|
|
|
#include <gst/gst.h>
|
|
#include <gst/pbutils/pbutils.h>
|
|
|
|
G_BEGIN_DECLS
|
|
#define GST_TYPE_SPLITMUX_SINK (gst_splitmux_sink_get_type())
|
|
#define GST_SPLITMUX_SINK(obj) (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_SPLITMUX_SINK,GstSplitMuxSink))
|
|
#define GST_SPLITMUX_SINK_CLASS(klass) (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_SPLITMUX_SINK,GstSplitMuxSinkClass))
|
|
#define GST_IS_SPLITMUX_SINK(obj) (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_SPLITMUX_SINK))
|
|
#define GST_IS_SPLITMUX_SINK_CLASS(klass) (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_SPLITMUX_SINK))
|
|
typedef struct _GstSplitMuxSink GstSplitMuxSink;
|
|
typedef struct _GstSplitMuxSinkClass GstSplitMuxSinkClass;
|
|
|
|
GType gst_splitmux_sink_get_type (void);
|
|
gboolean register_splitmuxsink (GstPlugin * plugin);
|
|
|
|
typedef enum _SplitMuxInputState
|
|
{
|
|
SPLITMUX_INPUT_STATE_STOPPED,
|
|
SPLITMUX_INPUT_STATE_COLLECTING_GOP_START, /* Waiting for the next ref ctx keyframe */
|
|
SPLITMUX_INPUT_STATE_WAITING_GOP_COLLECT, /* Waiting for all streams to collect GOP */
|
|
SPLITMUX_INPUT_STATE_FINISHING_UP /* Got EOS from reference ctx, send everything */
|
|
} SplitMuxInputState;
|
|
|
|
typedef enum _SplitMuxOutputState
|
|
{
|
|
SPLITMUX_OUTPUT_STATE_STOPPED,
|
|
SPLITMUX_OUTPUT_STATE_AWAITING_COMMAND, /* Waiting first command packet from input */
|
|
SPLITMUX_OUTPUT_STATE_OUTPUT_GOP, /* Outputting a collected GOP */
|
|
SPLITMUX_OUTPUT_STATE_ENDING_FILE, /* Finishing the current fragment */
|
|
SPLITMUX_OUTPUT_STATE_START_NEXT_FILE /* Restarting after ENDING_FILE */
|
|
} SplitMuxOutputState;
|
|
|
|
typedef struct _SplitMuxOutputCommand
|
|
{
|
|
gboolean start_new_fragment; /* Whether to start a new fragment before advancing output ts */
|
|
GstClockTimeDiff max_output_ts; /* Set the limit to stop GOP output */
|
|
} SplitMuxOutputCommand;
|
|
|
|
typedef struct _MqStreamBuf
|
|
{
|
|
gboolean keyframe;
|
|
GstClockTimeDiff run_ts;
|
|
guint64 buf_size;
|
|
GstClockTime duration;
|
|
} MqStreamBuf;
|
|
|
|
typedef struct _MqStreamCtx
|
|
{
|
|
gint refcount;
|
|
|
|
GstSplitMuxSink *splitmux;
|
|
|
|
guint q_overrun_id;
|
|
guint sink_pad_block_id;
|
|
guint src_pad_block_id;
|
|
|
|
gboolean is_reference;
|
|
|
|
gboolean flushing;
|
|
gboolean in_eos;
|
|
gboolean out_eos;
|
|
gboolean need_unblock;
|
|
|
|
GstSegment in_segment;
|
|
GstSegment out_segment;
|
|
|
|
GstClockTimeDiff in_running_time;
|
|
GstClockTimeDiff out_running_time;
|
|
|
|
GstElement *q;
|
|
GQueue queued_bufs;
|
|
|
|
GstPad *sinkpad;
|
|
GstPad *srcpad;
|
|
|
|
GstBuffer *cur_buffer;
|
|
GstEvent *pending_gap;
|
|
} MqStreamCtx;
|
|
|
|
struct _GstSplitMuxSink
|
|
{
|
|
GstBin parent;
|
|
|
|
GMutex lock;
|
|
GCond input_cond;
|
|
GCond output_cond;
|
|
|
|
gdouble mux_overhead;
|
|
|
|
GstClockTime threshold_time;
|
|
guint64 threshold_bytes;
|
|
guint max_files;
|
|
gboolean send_keyframe_requests;
|
|
|
|
GstElement *muxer;
|
|
GstElement *sink;
|
|
|
|
GstElement *provided_muxer;
|
|
|
|
GstElement *provided_sink;
|
|
GstElement *active_sink;
|
|
|
|
gboolean ready_for_output;
|
|
|
|
gchar *location;
|
|
guint fragment_id;
|
|
|
|
GList *contexts;
|
|
|
|
SplitMuxInputState input_state;
|
|
GstClockTimeDiff max_in_running_time;
|
|
/* Number of bytes sent to the
|
|
* current fragment */
|
|
guint64 fragment_total_bytes;
|
|
/* Number of bytes we've collected into
|
|
* the GOP that's being collected */
|
|
guint64 gop_total_bytes;
|
|
/* Start time of the current fragment */
|
|
GstClockTimeDiff fragment_start_time;
|
|
/* Start time of the current GOP */
|
|
GstClockTimeDiff gop_start_time;
|
|
|
|
GQueue out_cmd_q; /* Queue of commands for output thread */
|
|
|
|
SplitMuxOutputState output_state;
|
|
GstClockTimeDiff max_out_running_time;
|
|
GstClockTimeDiff next_max_out_running_time;
|
|
|
|
GstClockTimeDiff muxed_out_time;
|
|
guint64 muxed_out_bytes;
|
|
|
|
MqStreamCtx *reference_ctx;
|
|
/* Count of queued keyframes in the reference ctx */
|
|
guint queued_keyframes;
|
|
|
|
gboolean switching_fragment;
|
|
|
|
gboolean have_video;
|
|
|
|
gboolean need_async_start;
|
|
gboolean async_pending;
|
|
};
|
|
|
|
struct _GstSplitMuxSinkClass
|
|
{
|
|
GstBinClass parent_class;
|
|
};
|
|
|
|
G_END_DECLS
|
|
#endif /* __GST_SPLITMUXSINK_H__ */
|