mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-11-06 09:29:42 +00:00
4316be6571
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>
241 lines
6.8 KiB
C
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__ */
|