gstreamer/subprojects/gst-plugins-good/gst/multifile/gstsplitmuxsink.h
Sebastian Dröge 4316be6571 splitmuxsink: Keep track of the pending input GOPs in a queue
This cleans up input GOP handling and makes it possible to handle more
than 2 pending GOPs, which could happen before if keyframes are arriving
with too short of a distance between them.

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/1005>
2021-10-05 12:35:19 +00:00

241 lines
6.8 KiB
C

/* GStreamer split muxer bin
* Copyright (C) 2014-2019 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>
#include <gst/base/base.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);
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_ENDING_STREAM, /* Finishing up the entire stream due to input EOS */
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 {
/* For the very first GOP if it was created from a GAP event */
gboolean from_gap;
/* Minimum start time (PTS or DTS) of the GOP */
GstClockTimeDiff start_time;
/* Start time (PTS) of the GOP */
GstClockTimeDiff start_time_pts;
/* Minimum start timecode of the GOP */
GstVideoTimeCode *start_tc;
/* Number of bytes we've collected into the GOP */
guint64 total_bytes;
/* Number of bytes from the reference context
* that we've collected into the GOP */
guint64 reference_bytes;
gboolean sent_fku;
} InputGop;
typedef struct _MqStreamCtx
{
GstSplitMuxSink *splitmux;
guint q_overrun_id;
guint sink_pad_block_id;
guint src_pad_block_id;
gulong fragment_block_id;
gboolean is_reference;
gboolean flushing;
gboolean in_eos;
gboolean out_eos;
gboolean out_eos_async_done;
gboolean need_unblock;
gboolean caps_change;
gboolean is_releasing;
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_out_buffer;
GstEvent *pending_gap;
} MqStreamCtx;
struct _GstSplitMuxSink
{
GstBin parent;
GMutex state_lock;
gboolean shutdown;
GMutex lock;
GCond input_cond;
GCond output_cond;
gdouble mux_overhead;
GstClockTime threshold_time;
guint64 threshold_bytes;
guint max_files;
gboolean send_keyframe_requests;
gchar *threshold_timecode_str;
/* created from threshold_timecode_str */
GstVideoTimeCodeInterval *tc_interval;
GstClockTime alignment_threshold;
/* expected running time of next force keyframe unit event */
GstClockTime next_fku_time;
gboolean reset_muxer;
GstElement *muxer;
GstElement *sink;
GstElement *provided_muxer;
GstElement *provided_sink;
GstElement *active_sink;
gboolean ready_for_output;
gchar *location;
guint fragment_id;
guint start_index;
GList *contexts;
SplitMuxInputState input_state;
GstClockTimeDiff max_in_running_time;
GstClockTimeDiff max_in_running_time_dts;
/* Number of bytes sent to the
* current fragment */
guint64 fragment_total_bytes;
/* Number of bytes for the reference
* stream in this fragment */
guint64 fragment_reference_bytes;
/* Minimum start time (PTS or DTS) of the current fragment */
GstClockTimeDiff fragment_start_time;
/* Start time (PTS) of the current fragment */
GstClockTimeDiff fragment_start_time_pts;
/* Minimum start timecode of the current fragment */
GstVideoTimeCode *fragment_start_tc;
/* Oldest GOP at head, newest GOP at tail */
GQueue pending_input_gops;
/* expected running time of next fragment in timecode mode */
GstClockTime next_fragment_start_tc_time;
GQueue out_cmd_q; /* Queue of commands for output thread */
SplitMuxOutputState output_state;
GstClockTimeDiff max_out_running_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;
gboolean use_robust_muxing;
gboolean muxer_has_reserved_props;
gboolean split_requested;
gboolean do_split_next_gop;
GstQueueArray *times_to_split;
/* Async finalize options */
gboolean async_finalize;
gchar *muxer_factory;
gchar *muxer_preset;
GstStructure *muxer_properties;
gchar *sink_factory;
gchar *sink_preset;
GstStructure *sink_properties;
GstStructure *muxerpad_map;
};
struct _GstSplitMuxSinkClass
{
GstBinClass parent_class;
/* actions */
void (*split_now) (GstSplitMuxSink * splitmux);
void (*split_after) (GstSplitMuxSink * splitmux);
void (*split_at_running_time) (GstSplitMuxSink * splitmux, GstClockTime split_time);
};
GST_ELEMENT_REGISTER_DECLARE (splitmuxsink);
G_END_DECLS
#endif /* __GST_SPLITMUXSINK_H__ */