| /* GStreamer |
| * Copyright (C) 1999,2000 Erik Walthinsen <omega@cse.ogi.edu> |
| * 2003 Colin Walters <cwalters@gnome.org> |
| * 2000,2005,2007 Wim Taymans <wim.taymans@gmail.com> |
| * 2007 Thiago Sousa Santos <thiagoss@lcc.ufcg.edu.br> |
| * SA 2010 ST-Ericsson <benjamin.gaignard@stericsson.com> |
| * |
| * gstqueue2.c: |
| * |
| * 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. |
| */ |
| |
| /** |
| * SECTION:element-queue2 |
| * |
| * Data is queued until one of the limits specified by the |
| * #GstQueue2:max-size-buffers, #GstQueue2:max-size-bytes and/or |
| * #GstQueue2:max-size-time properties has been reached. Any attempt to push |
| * more buffers into the queue will block the pushing thread until more space |
| * becomes available. |
| * |
| * The queue will create a new thread on the source pad to decouple the |
| * processing on sink and source pad. |
| * |
| * You can query how many buffers are queued by reading the |
| * #GstQueue2:current-level-buffers property. |
| * |
| * The default queue size limits are 100 buffers, 2MB of data, or |
| * two seconds worth of data, whichever is reached first. |
| * |
| * If you set temp-template to a value such as /tmp/gstreamer-XXXXXX, the element |
| * will allocate a random free filename and buffer data in the file. |
| * By using this, it will buffer the entire stream data on the file independently |
| * of the queue size limits, they will only be used for buffering statistics. |
| * |
| * The temp-location property will be used to notify the application of the |
| * allocated filename. |
| */ |
| |
| #ifdef HAVE_CONFIG_H |
| #include "config.h" |
| #endif |
| |
| #include "gstqueue2.h" |
| |
| #include <glib/gstdio.h> |
| |
| #include "gst/gst-i18n-lib.h" |
| #include "gst/glib-compat-private.h" |
| |
| #include <string.h> |
| |
| #ifdef G_OS_WIN32 |
| #include <io.h> /* lseek, open, close, read */ |
| #undef lseek |
| #define lseek _lseeki64 |
| #undef off_t |
| #define off_t guint64 |
| #else |
| #include <unistd.h> |
| #endif |
| |
| #ifdef __BIONIC__ /* Android */ |
| #undef lseek |
| #define lseek lseek64 |
| #undef off_t |
| #define off_t guint64 |
| #include <fcntl.h> |
| #endif |
| |
| static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink", |
| GST_PAD_SINK, |
| GST_PAD_ALWAYS, |
| GST_STATIC_CAPS_ANY); |
| |
| static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src", |
| GST_PAD_SRC, |
| GST_PAD_ALWAYS, |
| GST_STATIC_CAPS_ANY); |
| |
| GST_DEBUG_CATEGORY_STATIC (queue_debug); |
| #define GST_CAT_DEFAULT (queue_debug) |
| GST_DEBUG_CATEGORY_STATIC (queue_dataflow); |
| |
| enum |
| { |
| LAST_SIGNAL |
| }; |
| |
| /* other defines */ |
| #define DEFAULT_BUFFER_SIZE 4096 |
| #define QUEUE_IS_USING_TEMP_FILE(queue) ((queue)->temp_template != NULL) |
| #define QUEUE_IS_USING_RING_BUFFER(queue) ((queue)->ring_buffer_max_size != 0) /* for consistency with the above macro */ |
| #define QUEUE_IS_USING_QUEUE(queue) (!QUEUE_IS_USING_TEMP_FILE(queue) && !QUEUE_IS_USING_RING_BUFFER (queue)) |
| |
| #define QUEUE_MAX_BYTES(queue) MIN((queue)->max_level.bytes, (queue)->ring_buffer_max_size) |
| |
| /* default property values */ |
| #define DEFAULT_MAX_SIZE_BUFFERS 100 /* 100 buffers */ |
| #define DEFAULT_MAX_SIZE_BYTES (2 * 1024 * 1024) /* 2 MB */ |
| #define DEFAULT_MAX_SIZE_TIME 2 * GST_SECOND /* 2 seconds */ |
| #define DEFAULT_USE_BUFFERING FALSE |
| #define DEFAULT_USE_TAGS_BITRATE FALSE |
| #define DEFAULT_USE_RATE_ESTIMATE TRUE |
| #define DEFAULT_LOW_PERCENT 10 |
| #define DEFAULT_HIGH_PERCENT 99 |
| #define DEFAULT_TEMP_REMOVE TRUE |
| #define DEFAULT_RING_BUFFER_MAX_SIZE 0 |
| |
| enum |
| { |
| PROP_0, |
| PROP_CUR_LEVEL_BUFFERS, |
| PROP_CUR_LEVEL_BYTES, |
| PROP_CUR_LEVEL_TIME, |
| PROP_MAX_SIZE_BUFFERS, |
| PROP_MAX_SIZE_BYTES, |
| PROP_MAX_SIZE_TIME, |
| PROP_USE_BUFFERING, |
| PROP_USE_TAGS_BITRATE, |
| PROP_USE_RATE_ESTIMATE, |
| PROP_LOW_PERCENT, |
| PROP_HIGH_PERCENT, |
| PROP_TEMP_TEMPLATE, |
| PROP_TEMP_LOCATION, |
| PROP_TEMP_REMOVE, |
| PROP_RING_BUFFER_MAX_SIZE, |
| PROP_AVG_IN_RATE, |
| PROP_LAST |
| }; |
| |
| #define GST_QUEUE2_CLEAR_LEVEL(l) G_STMT_START { \ |
| l.buffers = 0; \ |
| l.bytes = 0; \ |
| l.time = 0; \ |
| l.rate_time = 0; \ |
| } G_STMT_END |
| |
| #define STATUS(queue, pad, msg) \ |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, \ |
| "(%s:%s) " msg ": %u of %u buffers, %u of %u " \ |
| "bytes, %" G_GUINT64_FORMAT " of %" G_GUINT64_FORMAT \ |
| " ns, %"G_GUINT64_FORMAT" items", \ |
| GST_DEBUG_PAD_NAME (pad), \ |
| queue->cur_level.buffers, \ |
| queue->max_level.buffers, \ |
| queue->cur_level.bytes, \ |
| queue->max_level.bytes, \ |
| queue->cur_level.time, \ |
| queue->max_level.time, \ |
| (guint64) (!QUEUE_IS_USING_QUEUE(queue) ? \ |
| queue->current->writing_pos - queue->current->max_reading_pos : \ |
| queue->queue.length)) |
| |
| #define GST_QUEUE2_MUTEX_LOCK(q) G_STMT_START { \ |
| g_mutex_lock (&q->qlock); \ |
| } G_STMT_END |
| |
| #define GST_QUEUE2_MUTEX_LOCK_CHECK(q,res,label) G_STMT_START { \ |
| GST_QUEUE2_MUTEX_LOCK (q); \ |
| if (res != GST_FLOW_OK) \ |
| goto label; \ |
| } G_STMT_END |
| |
| #define GST_QUEUE2_MUTEX_UNLOCK(q) G_STMT_START { \ |
| g_mutex_unlock (&q->qlock); \ |
| } G_STMT_END |
| |
| #define GST_QUEUE2_WAIT_DEL_CHECK(q, res, label) G_STMT_START { \ |
| STATUS (queue, q->sinkpad, "wait for DEL"); \ |
| q->waiting_del = TRUE; \ |
| g_cond_wait (&q->item_del, &queue->qlock); \ |
| q->waiting_del = FALSE; \ |
| if (res != GST_FLOW_OK) { \ |
| STATUS (queue, q->srcpad, "received DEL wakeup"); \ |
| goto label; \ |
| } \ |
| STATUS (queue, q->sinkpad, "received DEL"); \ |
| } G_STMT_END |
| |
| #define GST_QUEUE2_WAIT_ADD_CHECK(q, res, label) G_STMT_START { \ |
| STATUS (queue, q->srcpad, "wait for ADD"); \ |
| q->waiting_add = TRUE; \ |
| g_cond_wait (&q->item_add, &q->qlock); \ |
| q->waiting_add = FALSE; \ |
| if (res != GST_FLOW_OK) { \ |
| STATUS (queue, q->srcpad, "received ADD wakeup"); \ |
| goto label; \ |
| } \ |
| STATUS (queue, q->srcpad, "received ADD"); \ |
| } G_STMT_END |
| |
| #define GST_QUEUE2_SIGNAL_DEL(q) G_STMT_START { \ |
| if (q->waiting_del) { \ |
| STATUS (q, q->srcpad, "signal DEL"); \ |
| g_cond_signal (&q->item_del); \ |
| } \ |
| } G_STMT_END |
| |
| #define GST_QUEUE2_SIGNAL_ADD(q) G_STMT_START { \ |
| if (q->waiting_add) { \ |
| STATUS (q, q->sinkpad, "signal ADD"); \ |
| g_cond_signal (&q->item_add); \ |
| } \ |
| } G_STMT_END |
| |
| #define SET_PERCENT(q, perc) G_STMT_START { \ |
| if (perc != q->buffering_percent) { \ |
| q->buffering_percent = perc; \ |
| q->percent_changed = TRUE; \ |
| GST_DEBUG_OBJECT (q, "buffering %d percent", perc); \ |
| get_buffering_stats (q, perc, &q->mode, &q->avg_in, &q->avg_out, \ |
| &q->buffering_left); \ |
| } \ |
| } G_STMT_END |
| |
| #define _do_init \ |
| GST_DEBUG_CATEGORY_INIT (queue_debug, "queue2", 0, "queue element"); \ |
| GST_DEBUG_CATEGORY_INIT (queue_dataflow, "queue2_dataflow", 0, \ |
| "dataflow inside the queue element"); |
| #define gst_queue2_parent_class parent_class |
| G_DEFINE_TYPE_WITH_CODE (GstQueue2, gst_queue2, GST_TYPE_ELEMENT, _do_init); |
| |
| static void gst_queue2_finalize (GObject * object); |
| |
| static void gst_queue2_set_property (GObject * object, |
| guint prop_id, const GValue * value, GParamSpec * pspec); |
| static void gst_queue2_get_property (GObject * object, |
| guint prop_id, GValue * value, GParamSpec * pspec); |
| |
| static GstFlowReturn gst_queue2_chain (GstPad * pad, GstObject * parent, |
| GstBuffer * buffer); |
| static GstFlowReturn gst_queue2_chain_list (GstPad * pad, GstObject * parent, |
| GstBufferList * buffer_list); |
| static GstFlowReturn gst_queue2_push_one (GstQueue2 * queue); |
| static void gst_queue2_loop (GstPad * pad); |
| |
| static gboolean gst_queue2_handle_sink_event (GstPad * pad, GstObject * parent, |
| GstEvent * event); |
| static gboolean gst_queue2_handle_sink_query (GstPad * pad, GstObject * parent, |
| GstQuery * query); |
| |
| static gboolean gst_queue2_handle_src_event (GstPad * pad, GstObject * parent, |
| GstEvent * event); |
| static gboolean gst_queue2_handle_src_query (GstPad * pad, GstObject * parent, |
| GstQuery * query); |
| static gboolean gst_queue2_handle_query (GstElement * element, |
| GstQuery * query); |
| |
| static GstFlowReturn gst_queue2_get_range (GstPad * pad, GstObject * parent, |
| guint64 offset, guint length, GstBuffer ** buffer); |
| |
| static gboolean gst_queue2_src_activate_mode (GstPad * pad, GstObject * parent, |
| GstPadMode mode, gboolean active); |
| static gboolean gst_queue2_sink_activate_mode (GstPad * pad, GstObject * parent, |
| GstPadMode mode, gboolean active); |
| static GstStateChangeReturn gst_queue2_change_state (GstElement * element, |
| GstStateChange transition); |
| |
| static gboolean gst_queue2_is_empty (GstQueue2 * queue); |
| static gboolean gst_queue2_is_filled (GstQueue2 * queue); |
| |
| static void update_cur_level (GstQueue2 * queue, GstQueue2Range * range); |
| static void update_in_rates (GstQueue2 * queue); |
| static void gst_queue2_post_buffering (GstQueue2 * queue); |
| |
| typedef enum |
| { |
| GST_QUEUE2_ITEM_TYPE_UNKNOWN = 0, |
| GST_QUEUE2_ITEM_TYPE_BUFFER, |
| GST_QUEUE2_ITEM_TYPE_BUFFER_LIST, |
| GST_QUEUE2_ITEM_TYPE_EVENT, |
| GST_QUEUE2_ITEM_TYPE_QUERY |
| } GstQueue2ItemType; |
| |
| typedef struct |
| { |
| GstQueue2ItemType type; |
| GstMiniObject *item; |
| } GstQueue2Item; |
| |
| /* static guint gst_queue2_signals[LAST_SIGNAL] = { 0 }; */ |
| |
| static void |
| gst_queue2_class_init (GstQueue2Class * klass) |
| { |
| GObjectClass *gobject_class = G_OBJECT_CLASS (klass); |
| GstElementClass *gstelement_class = GST_ELEMENT_CLASS (klass); |
| |
| gobject_class->set_property = gst_queue2_set_property; |
| gobject_class->get_property = gst_queue2_get_property; |
| |
| /* properties */ |
| g_object_class_install_property (gobject_class, PROP_CUR_LEVEL_BYTES, |
| g_param_spec_uint ("current-level-bytes", "Current level (kB)", |
| "Current amount of data in the queue (bytes)", |
| 0, G_MAXUINT, 0, G_PARAM_READABLE | G_PARAM_STATIC_STRINGS)); |
| g_object_class_install_property (gobject_class, PROP_CUR_LEVEL_BUFFERS, |
| g_param_spec_uint ("current-level-buffers", "Current level (buffers)", |
| "Current number of buffers in the queue", |
| 0, G_MAXUINT, 0, G_PARAM_READABLE | G_PARAM_STATIC_STRINGS)); |
| g_object_class_install_property (gobject_class, PROP_CUR_LEVEL_TIME, |
| g_param_spec_uint64 ("current-level-time", "Current level (ns)", |
| "Current amount of data in the queue (in ns)", |
| 0, G_MAXUINT64, 0, G_PARAM_READABLE | G_PARAM_STATIC_STRINGS)); |
| |
| g_object_class_install_property (gobject_class, PROP_MAX_SIZE_BYTES, |
| g_param_spec_uint ("max-size-bytes", "Max. size (kB)", |
| "Max. amount of data in the queue (bytes, 0=disable)", |
| 0, G_MAXUINT, DEFAULT_MAX_SIZE_BYTES, |
| G_PARAM_READWRITE | GST_PARAM_MUTABLE_PLAYING | |
| G_PARAM_STATIC_STRINGS)); |
| g_object_class_install_property (gobject_class, PROP_MAX_SIZE_BUFFERS, |
| g_param_spec_uint ("max-size-buffers", "Max. size (buffers)", |
| "Max. number of buffers in the queue (0=disable)", 0, G_MAXUINT, |
| DEFAULT_MAX_SIZE_BUFFERS, |
| G_PARAM_READWRITE | GST_PARAM_MUTABLE_PLAYING | |
| G_PARAM_STATIC_STRINGS)); |
| g_object_class_install_property (gobject_class, PROP_MAX_SIZE_TIME, |
| g_param_spec_uint64 ("max-size-time", "Max. size (ns)", |
| "Max. amount of data in the queue (in ns, 0=disable)", 0, G_MAXUINT64, |
| DEFAULT_MAX_SIZE_TIME, G_PARAM_READWRITE | GST_PARAM_MUTABLE_PLAYING | |
| G_PARAM_STATIC_STRINGS)); |
| |
| g_object_class_install_property (gobject_class, PROP_USE_BUFFERING, |
| g_param_spec_boolean ("use-buffering", "Use buffering", |
| "Emit GST_MESSAGE_BUFFERING based on low-/high-percent thresholds", |
| DEFAULT_USE_BUFFERING, G_PARAM_READWRITE | GST_PARAM_MUTABLE_PLAYING | |
| G_PARAM_STATIC_STRINGS)); |
| g_object_class_install_property (gobject_class, PROP_USE_TAGS_BITRATE, |
| g_param_spec_boolean ("use-tags-bitrate", "Use bitrate from tags", |
| "Use a bitrate from upstream tags to estimate buffer duration if not provided", |
| DEFAULT_USE_TAGS_BITRATE, |
| G_PARAM_READWRITE | GST_PARAM_MUTABLE_PLAYING | |
| G_PARAM_STATIC_STRINGS)); |
| g_object_class_install_property (gobject_class, PROP_USE_RATE_ESTIMATE, |
| g_param_spec_boolean ("use-rate-estimate", "Use Rate Estimate", |
| "Estimate the bitrate of the stream to calculate time level", |
| DEFAULT_USE_RATE_ESTIMATE, |
| G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)); |
| g_object_class_install_property (gobject_class, PROP_LOW_PERCENT, |
| g_param_spec_int ("low-percent", "Low percent", |
| "Low threshold for buffering to start. Only used if use-buffering is True", |
| 0, 100, DEFAULT_LOW_PERCENT, |
| G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)); |
| g_object_class_install_property (gobject_class, PROP_HIGH_PERCENT, |
| g_param_spec_int ("high-percent", "High percent", |
| "High threshold for buffering to finish. Only used if use-buffering is True", |
| 0, 100, DEFAULT_HIGH_PERCENT, |
| G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)); |
| |
| g_object_class_install_property (gobject_class, PROP_TEMP_TEMPLATE, |
| g_param_spec_string ("temp-template", "Temporary File Template", |
| "File template to store temporary files in, should contain directory " |
| "and XXXXXX. (NULL == disabled)", |
| NULL, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)); |
| |
| g_object_class_install_property (gobject_class, PROP_TEMP_LOCATION, |
| g_param_spec_string ("temp-location", "Temporary File Location", |
| "Location to store temporary files in (Only read this property, " |
| "use temp-template to configure the name template)", |
| NULL, G_PARAM_READABLE | G_PARAM_STATIC_STRINGS)); |
| |
| /** |
| * GstQueue2:temp-remove |
| * |
| * When temp-template is set, remove the temporary file when going to READY. |
| */ |
| g_object_class_install_property (gobject_class, PROP_TEMP_REMOVE, |
| g_param_spec_boolean ("temp-remove", "Remove the Temporary File", |
| "Remove the temp-location after use", |
| DEFAULT_TEMP_REMOVE, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)); |
| |
| /** |
| * GstQueue2:ring-buffer-max-size |
| * |
| * The maximum size of the ring buffer in bytes. If set to 0, the ring |
| * buffer is disabled. Default 0. |
| */ |
| g_object_class_install_property (gobject_class, PROP_RING_BUFFER_MAX_SIZE, |
| g_param_spec_uint64 ("ring-buffer-max-size", |
| "Max. ring buffer size (bytes)", |
| "Max. amount of data in the ring buffer (bytes, 0 = disabled)", |
| 0, G_MAXUINT64, DEFAULT_RING_BUFFER_MAX_SIZE, |
| G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)); |
| |
| /** |
| * GstQueue2:avg-in-rate |
| * |
| * The average input data rate. |
| */ |
| g_object_class_install_property (gobject_class, PROP_AVG_IN_RATE, |
| g_param_spec_int64 ("avg-in-rate", "Input data rate (bytes/s)", |
| "Average input data rate (bytes/s)", |
| 0, G_MAXINT64, 0, G_PARAM_READABLE | G_PARAM_STATIC_STRINGS)); |
| |
| /* set several parent class virtual functions */ |
| gobject_class->finalize = gst_queue2_finalize; |
| |
| gst_element_class_add_static_pad_template (gstelement_class, &srctemplate); |
| gst_element_class_add_static_pad_template (gstelement_class, &sinktemplate); |
| |
| gst_element_class_set_static_metadata (gstelement_class, "Queue 2", |
| "Generic", |
| "Simple data queue", |
| "Erik Walthinsen <omega@cse.ogi.edu>, " |
| "Wim Taymans <wim.taymans@gmail.com>"); |
| |
| gstelement_class->change_state = GST_DEBUG_FUNCPTR (gst_queue2_change_state); |
| gstelement_class->query = GST_DEBUG_FUNCPTR (gst_queue2_handle_query); |
| } |
| |
| static void |
| gst_queue2_init (GstQueue2 * queue) |
| { |
| queue->sinkpad = gst_pad_new_from_static_template (&sinktemplate, "sink"); |
| |
| gst_pad_set_chain_function (queue->sinkpad, |
| GST_DEBUG_FUNCPTR (gst_queue2_chain)); |
| gst_pad_set_chain_list_function (queue->sinkpad, |
| GST_DEBUG_FUNCPTR (gst_queue2_chain_list)); |
| gst_pad_set_activatemode_function (queue->sinkpad, |
| GST_DEBUG_FUNCPTR (gst_queue2_sink_activate_mode)); |
| gst_pad_set_event_function (queue->sinkpad, |
| GST_DEBUG_FUNCPTR (gst_queue2_handle_sink_event)); |
| gst_pad_set_query_function (queue->sinkpad, |
| GST_DEBUG_FUNCPTR (gst_queue2_handle_sink_query)); |
| GST_PAD_SET_PROXY_CAPS (queue->sinkpad); |
| gst_element_add_pad (GST_ELEMENT (queue), queue->sinkpad); |
| |
| queue->srcpad = gst_pad_new_from_static_template (&srctemplate, "src"); |
| |
| gst_pad_set_activatemode_function (queue->srcpad, |
| GST_DEBUG_FUNCPTR (gst_queue2_src_activate_mode)); |
| gst_pad_set_getrange_function (queue->srcpad, |
| GST_DEBUG_FUNCPTR (gst_queue2_get_range)); |
| gst_pad_set_event_function (queue->srcpad, |
| GST_DEBUG_FUNCPTR (gst_queue2_handle_src_event)); |
| gst_pad_set_query_function (queue->srcpad, |
| GST_DEBUG_FUNCPTR (gst_queue2_handle_src_query)); |
| GST_PAD_SET_PROXY_CAPS (queue->srcpad); |
| gst_element_add_pad (GST_ELEMENT (queue), queue->srcpad); |
| |
| /* levels */ |
| GST_QUEUE2_CLEAR_LEVEL (queue->cur_level); |
| queue->max_level.buffers = DEFAULT_MAX_SIZE_BUFFERS; |
| queue->max_level.bytes = DEFAULT_MAX_SIZE_BYTES; |
| queue->max_level.time = DEFAULT_MAX_SIZE_TIME; |
| queue->max_level.rate_time = DEFAULT_MAX_SIZE_TIME; |
| queue->use_buffering = DEFAULT_USE_BUFFERING; |
| queue->use_rate_estimate = DEFAULT_USE_RATE_ESTIMATE; |
| queue->low_percent = DEFAULT_LOW_PERCENT; |
| queue->high_percent = DEFAULT_HIGH_PERCENT; |
| |
| gst_segment_init (&queue->sink_segment, GST_FORMAT_TIME); |
| gst_segment_init (&queue->src_segment, GST_FORMAT_TIME); |
| |
| queue->sinktime = GST_CLOCK_TIME_NONE; |
| queue->srctime = GST_CLOCK_TIME_NONE; |
| queue->sink_tainted = TRUE; |
| queue->src_tainted = TRUE; |
| |
| queue->srcresult = GST_FLOW_FLUSHING; |
| queue->sinkresult = GST_FLOW_FLUSHING; |
| queue->is_eos = FALSE; |
| queue->in_timer = g_timer_new (); |
| queue->out_timer = g_timer_new (); |
| |
| g_mutex_init (&queue->qlock); |
| queue->waiting_add = FALSE; |
| g_cond_init (&queue->item_add); |
| queue->waiting_del = FALSE; |
| g_cond_init (&queue->item_del); |
| g_queue_init (&queue->queue); |
| |
| g_cond_init (&queue->query_handled); |
| queue->last_query = FALSE; |
| |
| g_mutex_init (&queue->buffering_post_lock); |
| queue->buffering_percent = 100; |
| |
| /* tempfile related */ |
| queue->temp_template = NULL; |
| queue->temp_location = NULL; |
| queue->temp_remove = DEFAULT_TEMP_REMOVE; |
| |
| queue->ring_buffer = NULL; |
| queue->ring_buffer_max_size = DEFAULT_RING_BUFFER_MAX_SIZE; |
| |
| GST_DEBUG_OBJECT (queue, |
| "initialized queue's not_empty & not_full conditions"); |
| } |
| |
| /* called only once, as opposed to dispose */ |
| static void |
| gst_queue2_finalize (GObject * object) |
| { |
| GstQueue2 *queue = GST_QUEUE2 (object); |
| |
| GST_DEBUG_OBJECT (queue, "finalizing queue"); |
| |
| while (!g_queue_is_empty (&queue->queue)) { |
| GstQueue2Item *qitem = g_queue_pop_head (&queue->queue); |
| |
| if (qitem->type != GST_QUEUE2_ITEM_TYPE_QUERY) |
| gst_mini_object_unref (qitem->item); |
| g_slice_free (GstQueue2Item, qitem); |
| } |
| |
| queue->last_query = FALSE; |
| g_queue_clear (&queue->queue); |
| g_mutex_clear (&queue->qlock); |
| g_mutex_clear (&queue->buffering_post_lock); |
| g_cond_clear (&queue->item_add); |
| g_cond_clear (&queue->item_del); |
| g_cond_clear (&queue->query_handled); |
| g_timer_destroy (queue->in_timer); |
| g_timer_destroy (queue->out_timer); |
| |
| /* temp_file path cleanup */ |
| g_free (queue->temp_template); |
| g_free (queue->temp_location); |
| |
| G_OBJECT_CLASS (parent_class)->finalize (object); |
| } |
| |
| static void |
| debug_ranges (GstQueue2 * queue) |
| { |
| GstQueue2Range *walk; |
| |
| for (walk = queue->ranges; walk; walk = walk->next) { |
| GST_DEBUG_OBJECT (queue, |
| "range [%" G_GUINT64_FORMAT "-%" G_GUINT64_FORMAT "] (rb [%" |
| G_GUINT64_FORMAT "-%" G_GUINT64_FORMAT "]), reading %" G_GUINT64_FORMAT |
| " current range? %s", walk->offset, walk->writing_pos, walk->rb_offset, |
| walk->rb_writing_pos, walk->reading_pos, |
| walk == queue->current ? "**y**" : " n "); |
| } |
| } |
| |
| /* clear all the downloaded ranges */ |
| static void |
| clean_ranges (GstQueue2 * queue) |
| { |
| GST_DEBUG_OBJECT (queue, "clean queue ranges"); |
| |
| g_slice_free_chain (GstQueue2Range, queue->ranges, next); |
| queue->ranges = NULL; |
| queue->current = NULL; |
| } |
| |
| /* find a range that contains @offset or NULL when nothing does */ |
| static GstQueue2Range * |
| find_range (GstQueue2 * queue, guint64 offset) |
| { |
| GstQueue2Range *range = NULL; |
| GstQueue2Range *walk; |
| |
| /* first do a quick check for the current range */ |
| for (walk = queue->ranges; walk; walk = walk->next) { |
| if (offset >= walk->offset && offset <= walk->writing_pos) { |
| /* we can reuse an existing range */ |
| range = walk; |
| break; |
| } |
| } |
| if (range) { |
| GST_DEBUG_OBJECT (queue, |
| "found range for %" G_GUINT64_FORMAT ": [%" G_GUINT64_FORMAT "-%" |
| G_GUINT64_FORMAT "]", offset, range->offset, range->writing_pos); |
| } else { |
| GST_DEBUG_OBJECT (queue, "no range for %" G_GUINT64_FORMAT, offset); |
| } |
| return range; |
| } |
| |
| static void |
| update_cur_level (GstQueue2 * queue, GstQueue2Range * range) |
| { |
| guint64 max_reading_pos, writing_pos; |
| |
| writing_pos = range->writing_pos; |
| max_reading_pos = range->max_reading_pos; |
| |
| if (writing_pos > max_reading_pos) |
| queue->cur_level.bytes = writing_pos - max_reading_pos; |
| else |
| queue->cur_level.bytes = 0; |
| } |
| |
| /* make a new range for @offset or reuse an existing range */ |
| static GstQueue2Range * |
| add_range (GstQueue2 * queue, guint64 offset, gboolean update_existing) |
| { |
| GstQueue2Range *range, *prev, *next; |
| |
| GST_DEBUG_OBJECT (queue, "find range for %" G_GUINT64_FORMAT, offset); |
| |
| if ((range = find_range (queue, offset))) { |
| GST_DEBUG_OBJECT (queue, |
| "reusing range %" G_GUINT64_FORMAT "-%" G_GUINT64_FORMAT, range->offset, |
| range->writing_pos); |
| if (update_existing && range->writing_pos != offset) { |
| GST_DEBUG_OBJECT (queue, "updating range writing position to " |
| "%" G_GUINT64_FORMAT, offset); |
| range->writing_pos = offset; |
| } |
| } else { |
| GST_DEBUG_OBJECT (queue, |
| "new range %" G_GUINT64_FORMAT "-%" G_GUINT64_FORMAT, offset, offset); |
| |
| range = g_slice_new0 (GstQueue2Range); |
| range->offset = offset; |
| /* we want to write to the next location in the ring buffer */ |
| range->rb_offset = queue->current ? queue->current->rb_writing_pos : 0; |
| range->writing_pos = offset; |
| range->rb_writing_pos = range->rb_offset; |
| range->reading_pos = offset; |
| range->max_reading_pos = offset; |
| |
| /* insert sorted */ |
| prev = NULL; |
| next = queue->ranges; |
| while (next) { |
| if (next->offset > offset) { |
| /* insert before next */ |
| GST_DEBUG_OBJECT (queue, |
| "insert before range %p, offset %" G_GUINT64_FORMAT, next, |
| next->offset); |
| break; |
| } |
| /* try next */ |
| prev = next; |
| next = next->next; |
| } |
| range->next = next; |
| if (prev) |
| prev->next = range; |
| else |
| queue->ranges = range; |
| } |
| debug_ranges (queue); |
| |
| /* update the stats for this range */ |
| update_cur_level (queue, range); |
| |
| return range; |
| } |
| |
| |
| /* clear and init the download ranges for offset 0 */ |
| static void |
| init_ranges (GstQueue2 * queue) |
| { |
| GST_DEBUG_OBJECT (queue, "init queue ranges"); |
| |
| /* get rid of all the current ranges */ |
| clean_ranges (queue); |
| /* make a range for offset 0 */ |
| queue->current = add_range (queue, 0, TRUE); |
| } |
| |
| /* calculate the diff between running time on the sink and src of the queue. |
| * This is the total amount of time in the queue. */ |
| static void |
| update_time_level (GstQueue2 * queue) |
| { |
| if (queue->sink_tainted) { |
| queue->sinktime = |
| gst_segment_to_running_time (&queue->sink_segment, GST_FORMAT_TIME, |
| queue->sink_segment.position); |
| queue->sink_tainted = FALSE; |
| } |
| |
| if (queue->src_tainted) { |
| queue->srctime = |
| gst_segment_to_running_time (&queue->src_segment, GST_FORMAT_TIME, |
| queue->src_segment.position); |
| queue->src_tainted = FALSE; |
| } |
| |
| GST_DEBUG_OBJECT (queue, "sink %" GST_TIME_FORMAT ", src %" GST_TIME_FORMAT, |
| GST_TIME_ARGS (queue->sinktime), GST_TIME_ARGS (queue->srctime)); |
| |
| if (queue->sinktime != GST_CLOCK_TIME_NONE |
| && queue->srctime != GST_CLOCK_TIME_NONE |
| && queue->sinktime >= queue->srctime) |
| queue->cur_level.time = queue->sinktime - queue->srctime; |
| else |
| queue->cur_level.time = 0; |
| } |
| |
| /* take a SEGMENT event and apply the values to segment, updating the time |
| * level of queue. */ |
| static void |
| apply_segment (GstQueue2 * queue, GstEvent * event, GstSegment * segment, |
| gboolean is_sink) |
| { |
| gst_event_copy_segment (event, segment); |
| |
| if (segment->format == GST_FORMAT_BYTES) { |
| if (!QUEUE_IS_USING_QUEUE (queue) && is_sink) { |
| /* start is where we'll be getting from and as such writing next */ |
| queue->current = add_range (queue, segment->start, TRUE); |
| } |
| } |
| |
| /* now configure the values, we use these to track timestamps on the |
| * sinkpad. */ |
| if (segment->format != GST_FORMAT_TIME) { |
| /* non-time format, pretend the current time segment is closed with a |
| * 0 start and unknown stop time. */ |
| segment->format = GST_FORMAT_TIME; |
| segment->start = 0; |
| segment->stop = -1; |
| segment->time = 0; |
| } |
| |
| GST_DEBUG_OBJECT (queue, "configured SEGMENT %" GST_SEGMENT_FORMAT, segment); |
| |
| if (is_sink) |
| queue->sink_tainted = TRUE; |
| else |
| queue->src_tainted = TRUE; |
| |
| /* segment can update the time level of the queue */ |
| update_time_level (queue); |
| } |
| |
| static void |
| apply_gap (GstQueue2 * queue, GstEvent * event, |
| GstSegment * segment, gboolean is_sink) |
| { |
| GstClockTime timestamp; |
| GstClockTime duration; |
| |
| gst_event_parse_gap (event, ×tamp, &duration); |
| |
| if (GST_CLOCK_TIME_IS_VALID (timestamp)) { |
| |
| if (GST_CLOCK_TIME_IS_VALID (duration)) { |
| timestamp += duration; |
| } |
| |
| segment->position = timestamp; |
| |
| if (is_sink) |
| queue->sink_tainted = TRUE; |
| else |
| queue->src_tainted = TRUE; |
| |
| /* calc diff with other end */ |
| update_time_level (queue); |
| } |
| } |
| |
| /* take a buffer and update segment, updating the time level of the queue. */ |
| static void |
| apply_buffer (GstQueue2 * queue, GstBuffer * buffer, GstSegment * segment, |
| guint64 size, gboolean is_sink) |
| { |
| GstClockTime duration, timestamp; |
| |
| timestamp = GST_BUFFER_DTS_OR_PTS (buffer); |
| duration = GST_BUFFER_DURATION (buffer); |
| |
| /* If we have no duration, pick one from the bitrate if we can */ |
| if (duration == GST_CLOCK_TIME_NONE && queue->use_tags_bitrate) { |
| guint bitrate = |
| is_sink ? queue->sink_tags_bitrate : queue->src_tags_bitrate; |
| if (bitrate) |
| duration = gst_util_uint64_scale (size, 8 * GST_SECOND, bitrate); |
| } |
| |
| /* if no timestamp is set, assume it's continuous with the previous |
| * time */ |
| if (timestamp == GST_CLOCK_TIME_NONE) |
| timestamp = segment->position; |
| |
| /* add duration */ |
| if (duration != GST_CLOCK_TIME_NONE) |
| timestamp += duration; |
| |
| GST_DEBUG_OBJECT (queue, "position updated to %" GST_TIME_FORMAT, |
| GST_TIME_ARGS (timestamp)); |
| |
| segment->position = timestamp; |
| |
| if (is_sink) |
| queue->sink_tainted = TRUE; |
| else |
| queue->src_tainted = TRUE; |
| |
| /* calc diff with other end */ |
| update_time_level (queue); |
| } |
| |
| struct BufListData |
| { |
| GstClockTime timestamp; |
| guint bitrate; |
| }; |
| |
| static gboolean |
| buffer_list_apply_time (GstBuffer ** buf, guint idx, gpointer data) |
| { |
| struct BufListData *bld = data; |
| GstClockTime *timestamp = &bld->timestamp; |
| GstClockTime btime; |
| |
| GST_TRACE ("buffer %u has pts %" GST_TIME_FORMAT " dts %" GST_TIME_FORMAT |
| " duration %" GST_TIME_FORMAT, idx, |
| GST_TIME_ARGS (GST_BUFFER_PTS (*buf)), |
| GST_TIME_ARGS (GST_BUFFER_DTS (*buf)), |
| GST_TIME_ARGS (GST_BUFFER_DURATION (*buf))); |
| |
| btime = GST_BUFFER_DTS_OR_PTS (*buf); |
| if (GST_CLOCK_TIME_IS_VALID (btime)) |
| *timestamp = btime; |
| |
| if (GST_BUFFER_DURATION_IS_VALID (*buf)) |
| *timestamp += GST_BUFFER_DURATION (*buf); |
| else if (bld->bitrate != 0) { |
| guint64 size = gst_buffer_get_size (*buf); |
| |
| /* If we have no duration, pick one from the bitrate if we can */ |
| *timestamp += gst_util_uint64_scale (bld->bitrate, 8 * GST_SECOND, size); |
| } |
| |
| |
| GST_TRACE ("ts now %" GST_TIME_FORMAT, GST_TIME_ARGS (*timestamp)); |
| return TRUE; |
| } |
| |
| /* take a buffer list and update segment, updating the time level of the queue */ |
| static void |
| apply_buffer_list (GstQueue2 * queue, GstBufferList * buffer_list, |
| GstSegment * segment, gboolean is_sink) |
| { |
| struct BufListData bld; |
| |
| /* if no timestamp is set, assume it's continuous with the previous time */ |
| bld.timestamp = segment->position; |
| |
| if (queue->use_tags_bitrate) { |
| if (is_sink) |
| bld.bitrate = queue->sink_tags_bitrate; |
| else |
| bld.bitrate = queue->src_tags_bitrate; |
| } else |
| bld.bitrate = 0; |
| |
| gst_buffer_list_foreach (buffer_list, buffer_list_apply_time, &bld); |
| |
| GST_DEBUG_OBJECT (queue, "last_stop updated to %" GST_TIME_FORMAT, |
| GST_TIME_ARGS (bld.timestamp)); |
| |
| segment->position = bld.timestamp; |
| |
| if (is_sink) |
| queue->sink_tainted = TRUE; |
| else |
| queue->src_tainted = TRUE; |
| |
| /* calc diff with other end */ |
| update_time_level (queue); |
| } |
| |
| static inline gint |
| get_percent (guint64 cur_level, guint64 max_level, guint64 alt_max) |
| { |
| guint64 p; |
| |
| if (max_level == 0) |
| return 0; |
| |
| if (alt_max > 0) |
| p = gst_util_uint64_scale (cur_level, 100, MIN (max_level, alt_max)); |
| else |
| p = gst_util_uint64_scale (cur_level, 100, max_level); |
| |
| return MIN (p, 100); |
| } |
| |
| static gboolean |
| get_buffering_percent (GstQueue2 * queue, gboolean * is_buffering, |
| gint * percent) |
| { |
| gint perc, perc2; |
| |
| if (queue->high_percent <= 0) { |
| if (percent) |
| *percent = 100; |
| if (is_buffering) |
| *is_buffering = FALSE; |
| return FALSE; |
| } |
| #define GET_PERCENT(format,alt_max) \ |
| get_percent(queue->cur_level.format,queue->max_level.format,(alt_max)) |
| |
| if (queue->is_eos) { |
| /* on EOS we are always 100% full, we set the var here so that it we can |
| * reuse the logic below to stop buffering */ |
| perc = 100; |
| GST_LOG_OBJECT (queue, "we are EOS"); |
| } else { |
| GST_LOG_OBJECT (queue, |
| "Cur level bytes/time/buffers %u/%" GST_TIME_FORMAT "/%u", |
| queue->cur_level.bytes, GST_TIME_ARGS (queue->cur_level.time), |
| queue->cur_level.buffers); |
| |
| /* figure out the percent we are filled, we take the max of all formats. */ |
| if (!QUEUE_IS_USING_RING_BUFFER (queue)) { |
| perc = GET_PERCENT (bytes, 0); |
| } else { |
| guint64 rb_size = queue->ring_buffer_max_size; |
| perc = GET_PERCENT (bytes, rb_size); |
| } |
| |
| perc2 = GET_PERCENT (time, 0); |
| perc = MAX (perc, perc2); |
| |
| perc2 = GET_PERCENT (buffers, 0); |
| perc = MAX (perc, perc2); |
| |
| /* also apply the rate estimate when we need to */ |
| if (queue->use_rate_estimate) { |
| perc2 = GET_PERCENT (rate_time, 0); |
| perc = MAX (perc, perc2); |
| } |
| |
| /* Don't get to 0% unless we're really empty */ |
| if (queue->cur_level.bytes > 0) |
| perc = MAX (1, perc); |
| } |
| #undef GET_PERCENT |
| |
| if (is_buffering) |
| *is_buffering = queue->is_buffering; |
| |
| /* scale to high percent so that it becomes the 100% mark */ |
| perc = perc * 100 / queue->high_percent; |
| /* clip */ |
| if (perc > 100) |
| perc = 100; |
| |
| if (percent) |
| *percent = perc; |
| |
| GST_DEBUG_OBJECT (queue, "buffering %d, percent %d", queue->is_buffering, |
| perc); |
| |
| return TRUE; |
| } |
| |
| static void |
| get_buffering_stats (GstQueue2 * queue, gint percent, GstBufferingMode * mode, |
| gint * avg_in, gint * avg_out, gint64 * buffering_left) |
| { |
| if (mode) { |
| if (!QUEUE_IS_USING_QUEUE (queue)) { |
| if (QUEUE_IS_USING_RING_BUFFER (queue)) |
| *mode = GST_BUFFERING_TIMESHIFT; |
| else |
| *mode = GST_BUFFERING_DOWNLOAD; |
| } else { |
| *mode = GST_BUFFERING_STREAM; |
| } |
| } |
| |
| if (avg_in) |
| *avg_in = queue->byte_in_rate; |
| if (avg_out) |
| *avg_out = queue->byte_out_rate; |
| |
| if (buffering_left) { |
| *buffering_left = (percent == 100 ? 0 : -1); |
| |
| if (queue->use_rate_estimate) { |
| guint64 max, cur; |
| |
| max = queue->max_level.rate_time; |
| cur = queue->cur_level.rate_time; |
| |
| if (percent != 100 && max > cur) |
| *buffering_left = (max - cur) / 1000000; |
| } |
| } |
| } |
| |
| static void |
| gst_queue2_post_buffering (GstQueue2 * queue) |
| { |
| GstMessage *msg = NULL; |
| |
| g_mutex_lock (&queue->buffering_post_lock); |
| GST_QUEUE2_MUTEX_LOCK (queue); |
| if (queue->percent_changed) { |
| gint percent = queue->buffering_percent; |
| |
| queue->percent_changed = FALSE; |
| |
| GST_DEBUG_OBJECT (queue, "Going to post buffering: %d%%", percent); |
| msg = gst_message_new_buffering (GST_OBJECT_CAST (queue), percent); |
| |
| gst_message_set_buffering_stats (msg, queue->mode, queue->avg_in, |
| queue->avg_out, queue->buffering_left); |
| } |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| |
| if (msg != NULL) |
| gst_element_post_message (GST_ELEMENT_CAST (queue), msg); |
| |
| g_mutex_unlock (&queue->buffering_post_lock); |
| } |
| |
| static void |
| update_buffering (GstQueue2 * queue) |
| { |
| gint percent; |
| |
| /* Ensure the variables used to calculate buffering state are up-to-date. */ |
| if (queue->current) |
| update_cur_level (queue, queue->current); |
| update_in_rates (queue); |
| |
| if (!get_buffering_percent (queue, NULL, &percent)) |
| return; |
| |
| if (queue->is_buffering) { |
| /* if we were buffering see if we reached the high watermark */ |
| if (percent >= 100) |
| queue->is_buffering = FALSE; |
| |
| SET_PERCENT (queue, percent); |
| } else { |
| /* we were not buffering, check if we need to start buffering if we drop |
| * below the low threshold */ |
| if (percent < queue->low_percent) { |
| queue->is_buffering = TRUE; |
| SET_PERCENT (queue, percent); |
| } |
| } |
| } |
| |
| static void |
| reset_rate_timer (GstQueue2 * queue) |
| { |
| queue->bytes_in = 0; |
| queue->bytes_out = 0; |
| queue->byte_in_rate = 0.0; |
| queue->byte_in_period = 0; |
| queue->byte_out_rate = 0.0; |
| queue->last_update_in_rates_elapsed = 0.0; |
| queue->last_in_elapsed = 0.0; |
| queue->last_out_elapsed = 0.0; |
| queue->in_timer_started = FALSE; |
| queue->out_timer_started = FALSE; |
| } |
| |
| /* the interval in seconds to recalculate the rate */ |
| #define RATE_INTERVAL 0.2 |
| /* Tuning for rate estimation. We use a large window for the input rate because |
| * it should be stable when connected to a network. The output rate is less |
| * stable (the elements preroll, queues behind a demuxer fill, ...) and should |
| * therefore adapt more quickly. |
| * However, initial input rate may be subject to a burst, and should therefore |
| * initially also adapt more quickly to changes, and only later on give higher |
| * weight to previous values. */ |
| #define AVG_IN(avg,val,w1,w2) ((avg) * (w1) + (val) * (w2)) / ((w1) + (w2)) |
| #define AVG_OUT(avg,val) ((avg) * 3.0 + (val)) / 4.0 |
| |
| static void |
| update_in_rates (GstQueue2 * queue) |
| { |
| gdouble elapsed, period; |
| gdouble byte_in_rate; |
| |
| if (!queue->in_timer_started) { |
| queue->in_timer_started = TRUE; |
| g_timer_start (queue->in_timer); |
| return; |
| } |
| |
| queue->last_update_in_rates_elapsed = elapsed = |
| g_timer_elapsed (queue->in_timer, NULL); |
| |
| /* recalc after each interval. */ |
| if (queue->last_in_elapsed + RATE_INTERVAL < elapsed) { |
| period = elapsed - queue->last_in_elapsed; |
| |
| GST_DEBUG_OBJECT (queue, |
| "rates: period %f, in %" G_GUINT64_FORMAT ", global period %f", |
| period, queue->bytes_in, queue->byte_in_period); |
| |
| byte_in_rate = queue->bytes_in / period; |
| |
| if (queue->byte_in_rate == 0.0) |
| queue->byte_in_rate = byte_in_rate; |
| else |
| queue->byte_in_rate = AVG_IN (queue->byte_in_rate, byte_in_rate, |
| (double) queue->byte_in_period, period); |
| |
| /* another data point, cap at 16 for long time running average */ |
| if (queue->byte_in_period < 16 * RATE_INTERVAL) |
| queue->byte_in_period += period; |
| |
| /* reset the values to calculate rate over the next interval */ |
| queue->last_in_elapsed = elapsed; |
| queue->bytes_in = 0; |
| } |
| |
| if (queue->byte_in_rate > 0.0) { |
| queue->cur_level.rate_time = |
| queue->cur_level.bytes / queue->byte_in_rate * GST_SECOND; |
| } |
| GST_DEBUG_OBJECT (queue, "rates: in %f, time %" GST_TIME_FORMAT, |
| queue->byte_in_rate, GST_TIME_ARGS (queue->cur_level.rate_time)); |
| } |
| |
| static void |
| update_out_rates (GstQueue2 * queue) |
| { |
| gdouble elapsed, period; |
| gdouble byte_out_rate; |
| |
| if (!queue->out_timer_started) { |
| queue->out_timer_started = TRUE; |
| g_timer_start (queue->out_timer); |
| return; |
| } |
| |
| elapsed = g_timer_elapsed (queue->out_timer, NULL); |
| |
| /* recalc after each interval. */ |
| if (queue->last_out_elapsed + RATE_INTERVAL < elapsed) { |
| period = elapsed - queue->last_out_elapsed; |
| |
| GST_DEBUG_OBJECT (queue, |
| "rates: period %f, out %" G_GUINT64_FORMAT, period, queue->bytes_out); |
| |
| byte_out_rate = queue->bytes_out / period; |
| |
| if (queue->byte_out_rate == 0.0) |
| queue->byte_out_rate = byte_out_rate; |
| else |
| queue->byte_out_rate = AVG_OUT (queue->byte_out_rate, byte_out_rate); |
| |
| /* reset the values to calculate rate over the next interval */ |
| queue->last_out_elapsed = elapsed; |
| queue->bytes_out = 0; |
| } |
| if (queue->byte_in_rate > 0.0) { |
| queue->cur_level.rate_time = |
| queue->cur_level.bytes / queue->byte_in_rate * GST_SECOND; |
| } |
| GST_DEBUG_OBJECT (queue, "rates: out %f, time %" GST_TIME_FORMAT, |
| queue->byte_out_rate, GST_TIME_ARGS (queue->cur_level.rate_time)); |
| } |
| |
| static void |
| update_cur_pos (GstQueue2 * queue, GstQueue2Range * range, guint64 pos) |
| { |
| guint64 reading_pos, max_reading_pos; |
| |
| reading_pos = pos; |
| max_reading_pos = range->max_reading_pos; |
| |
| max_reading_pos = MAX (max_reading_pos, reading_pos); |
| |
| GST_DEBUG_OBJECT (queue, |
| "updating max_reading_pos from %" G_GUINT64_FORMAT " to %" |
| G_GUINT64_FORMAT, range->max_reading_pos, max_reading_pos); |
| range->max_reading_pos = max_reading_pos; |
| |
| update_cur_level (queue, range); |
| } |
| |
| static gboolean |
| perform_seek_to_offset (GstQueue2 * queue, guint64 offset) |
| { |
| GstEvent *event; |
| gboolean res; |
| |
| /* until we receive the FLUSH_STOP from this seek, we skip data */ |
| queue->seeking = TRUE; |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| |
| debug_ranges (queue); |
| |
| GST_DEBUG_OBJECT (queue, "Seeking to %" G_GUINT64_FORMAT, offset); |
| |
| event = |
| gst_event_new_seek (1.0, GST_FORMAT_BYTES, |
| GST_SEEK_FLAG_FLUSH | GST_SEEK_FLAG_ACCURATE, GST_SEEK_TYPE_SET, offset, |
| GST_SEEK_TYPE_NONE, -1); |
| |
| res = gst_pad_push_event (queue->sinkpad, event); |
| GST_QUEUE2_MUTEX_LOCK (queue); |
| |
| if (res) { |
| /* Between us sending the seek event and re-acquiring the lock, the source |
| * thread might already have pushed data and moved along the range's |
| * writing_pos beyond the seek offset. In that case we don't want to set |
| * the writing position back to the requested seek position, as it would |
| * cause data to be written to the wrong offset in the file or ring buffer. |
| * We still do the add_range call to switch the current range to the |
| * requested range, or create one if one doesn't exist yet. */ |
| queue->current = add_range (queue, offset, FALSE); |
| } |
| |
| return res; |
| } |
| |
| /* get the threshold for when we decide to seek rather than wait */ |
| static guint64 |
| get_seek_threshold (GstQueue2 * queue) |
| { |
| guint64 threshold; |
| |
| /* FIXME, find a good threshold based on the incoming rate. */ |
| threshold = 1024 * 512; |
| |
| if (QUEUE_IS_USING_RING_BUFFER (queue)) { |
| threshold = MIN (threshold, |
| QUEUE_MAX_BYTES (queue) - queue->cur_level.bytes); |
| } |
| return threshold; |
| } |
| |
| /* see if there is enough data in the file to read a full buffer */ |
| static gboolean |
| gst_queue2_have_data (GstQueue2 * queue, guint64 offset, guint length) |
| { |
| GstQueue2Range *range; |
| |
| GST_DEBUG_OBJECT (queue, "looking for offset %" G_GUINT64_FORMAT ", len %u", |
| offset, length); |
| |
| if ((range = find_range (queue, offset))) { |
| if (queue->current != range) { |
| GST_DEBUG_OBJECT (queue, "switching ranges, do seek to range position"); |
| perform_seek_to_offset (queue, range->writing_pos); |
| } |
| |
| GST_INFO_OBJECT (queue, "cur_level.bytes %u (max %" G_GUINT64_FORMAT ")", |
| queue->cur_level.bytes, QUEUE_MAX_BYTES (queue)); |
| |
| /* we have a range for offset */ |
| GST_DEBUG_OBJECT (queue, |
| "we have a range %p, offset %" G_GUINT64_FORMAT ", writing_pos %" |
| G_GUINT64_FORMAT, range, range->offset, range->writing_pos); |
| |
| if (!QUEUE_IS_USING_RING_BUFFER (queue) && queue->is_eos) |
| return TRUE; |
| |
| if (offset + length <= range->writing_pos) |
| return TRUE; |
| else |
| GST_DEBUG_OBJECT (queue, |
| "Need more data (%" G_GUINT64_FORMAT " bytes more)", |
| (offset + length) - range->writing_pos); |
| |
| } else { |
| GST_INFO_OBJECT (queue, "not found in any range off %" G_GUINT64_FORMAT |
| " len %u", offset, length); |
| /* we don't have the range, see how far away we are */ |
| if (!queue->is_eos && queue->current) { |
| guint64 threshold = get_seek_threshold (queue); |
| |
| if (offset >= queue->current->offset && offset <= |
| queue->current->writing_pos + threshold) { |
| GST_INFO_OBJECT (queue, |
| "requested data is within range, wait for data"); |
| return FALSE; |
| } |
| } |
| |
| /* too far away, do a seek */ |
| perform_seek_to_offset (queue, offset); |
| } |
| |
| return FALSE; |
| } |
| |
| #ifdef HAVE_FSEEKO |
| #define FSEEK_FILE(file,offset) (fseeko (file, (off_t) offset, SEEK_SET) != 0) |
| #elif defined (G_OS_UNIX) || defined (G_OS_WIN32) |
| #define FSEEK_FILE(file,offset) (lseek (fileno (file), (off_t) offset, SEEK_SET) == (off_t) -1) |
| #else |
| #define FSEEK_FILE(file,offset) (fseek (file, offset, SEEK_SET) != 0) |
| #endif |
| |
| static GstFlowReturn |
| gst_queue2_read_data_at_offset (GstQueue2 * queue, guint64 offset, guint length, |
| guint8 * dst, gint64 * read_return) |
| { |
| guint8 *ring_buffer; |
| size_t res; |
| |
| ring_buffer = queue->ring_buffer; |
| |
| if (QUEUE_IS_USING_TEMP_FILE (queue) && FSEEK_FILE (queue->temp_file, offset)) |
| goto seek_failed; |
| |
| /* this should not block */ |
| GST_LOG_OBJECT (queue, "Reading %d bytes from offset %" G_GUINT64_FORMAT, |
| length, offset); |
| if (QUEUE_IS_USING_TEMP_FILE (queue)) { |
| res = fread (dst, 1, length, queue->temp_file); |
| } else { |
| memcpy (dst, ring_buffer + offset, length); |
| res = length; |
| } |
| |
| GST_LOG_OBJECT (queue, "read %" G_GSIZE_FORMAT " bytes", res); |
| |
| if (G_UNLIKELY (res < length)) { |
| if (!QUEUE_IS_USING_TEMP_FILE (queue)) |
| goto could_not_read; |
| /* check for errors or EOF */ |
| if (ferror (queue->temp_file)) |
| goto could_not_read; |
| if (feof (queue->temp_file) && length > 0) |
| goto eos; |
| } |
| |
| *read_return = res; |
| |
| return GST_FLOW_OK; |
| |
| seek_failed: |
| { |
| GST_ELEMENT_ERROR (queue, RESOURCE, SEEK, (NULL), GST_ERROR_SYSTEM); |
| return GST_FLOW_ERROR; |
| } |
| could_not_read: |
| { |
| GST_ELEMENT_ERROR (queue, RESOURCE, READ, (NULL), GST_ERROR_SYSTEM); |
| return GST_FLOW_ERROR; |
| } |
| eos: |
| { |
| GST_DEBUG ("non-regular file hits EOS"); |
| return GST_FLOW_EOS; |
| } |
| } |
| |
| static GstFlowReturn |
| gst_queue2_create_read (GstQueue2 * queue, guint64 offset, guint length, |
| GstBuffer ** buffer) |
| { |
| GstBuffer *buf; |
| GstMapInfo info; |
| guint8 *data; |
| guint64 file_offset; |
| guint block_length, remaining, read_length; |
| guint64 rb_size; |
| guint64 max_size; |
| guint64 rpos; |
| GstFlowReturn ret = GST_FLOW_OK; |
| |
| /* allocate the output buffer of the requested size */ |
| if (*buffer == NULL) |
| buf = gst_buffer_new_allocate (NULL, length, NULL); |
| else |
| buf = *buffer; |
| |
| if (!gst_buffer_map (buf, &info, GST_MAP_WRITE)) |
| goto buffer_write_fail; |
| data = info.data; |
| |
| GST_DEBUG_OBJECT (queue, "Reading %u bytes from %" G_GUINT64_FORMAT, length, |
| offset); |
| |
| rpos = offset; |
| rb_size = queue->ring_buffer_max_size; |
| max_size = QUEUE_MAX_BYTES (queue); |
| |
| remaining = length; |
| while (remaining > 0) { |
| /* configure how much/whether to read */ |
| if (!gst_queue2_have_data (queue, rpos, remaining)) { |
| read_length = 0; |
| |
| if (QUEUE_IS_USING_RING_BUFFER (queue)) { |
| guint64 level; |
| |
| /* calculate how far away the offset is */ |
| if (queue->current->writing_pos > rpos) |
| level = queue->current->writing_pos - rpos; |
| else |
| level = 0; |
| |
| GST_DEBUG_OBJECT (queue, |
| "reading %" G_GUINT64_FORMAT ", writing %" G_GUINT64_FORMAT |
| ", level %" G_GUINT64_FORMAT ", max %" G_GUINT64_FORMAT, |
| rpos, queue->current->writing_pos, level, max_size); |
| |
| if (level >= max_size) { |
| /* we don't have the data but if we have a ring buffer that is full, we |
| * need to read */ |
| GST_DEBUG_OBJECT (queue, |
| "ring buffer full, reading QUEUE_MAX_BYTES %" |
| G_GUINT64_FORMAT " bytes", max_size); |
| read_length = max_size; |
| } else if (queue->is_eos) { |
| /* won't get any more data so read any data we have */ |
| if (level) { |
| GST_DEBUG_OBJECT (queue, |
| "EOS hit but read %" G_GUINT64_FORMAT " bytes that we have", |
| level); |
| read_length = level; |
| remaining = level; |
| length = level; |
| } else |
| goto hit_eos; |
| } |
| } |
| |
| if (read_length == 0) { |
| if (QUEUE_IS_USING_RING_BUFFER (queue)) { |
| GST_DEBUG_OBJECT (queue, |
| "update current position [%" G_GUINT64_FORMAT "-%" |
| G_GUINT64_FORMAT "]", rpos, queue->current->max_reading_pos); |
| update_cur_pos (queue, queue->current, rpos); |
| GST_QUEUE2_SIGNAL_DEL (queue); |
| } |
| |
| if (queue->use_buffering) |
| update_buffering (queue); |
| |
| GST_DEBUG_OBJECT (queue, "waiting for add"); |
| GST_QUEUE2_WAIT_ADD_CHECK (queue, queue->srcresult, out_flushing); |
| continue; |
| } |
| } else { |
| /* we have the requested data so read it */ |
| read_length = remaining; |
| } |
| |
| /* set range reading_pos to actual reading position for this read */ |
| queue->current->reading_pos = rpos; |
| |
| /* configure how much and from where to read */ |
| if (QUEUE_IS_USING_RING_BUFFER (queue)) { |
| file_offset = |
| (queue->current->rb_offset + (rpos - |
| queue->current->offset)) % rb_size; |
| if (file_offset + read_length > rb_size) { |
| block_length = rb_size - file_offset; |
| } else { |
| block_length = read_length; |
| } |
| } else { |
| file_offset = rpos; |
| block_length = read_length; |
| } |
| |
| /* while we still have data to read, we loop */ |
| while (read_length > 0) { |
| gint64 read_return; |
| |
| ret = |
| gst_queue2_read_data_at_offset (queue, file_offset, block_length, |
| data, &read_return); |
| if (ret != GST_FLOW_OK) |
| goto read_error; |
| |
| file_offset += read_return; |
| if (QUEUE_IS_USING_RING_BUFFER (queue)) |
| file_offset %= rb_size; |
| |
| data += read_return; |
| read_length -= read_return; |
| block_length = read_length; |
| remaining -= read_return; |
| |
| rpos = (queue->current->reading_pos += read_return); |
| update_cur_pos (queue, queue->current, queue->current->reading_pos); |
| } |
| GST_QUEUE2_SIGNAL_DEL (queue); |
| GST_DEBUG_OBJECT (queue, "%u bytes left to read", remaining); |
| } |
| |
| gst_buffer_unmap (buf, &info); |
| gst_buffer_resize (buf, 0, length); |
| |
| GST_BUFFER_OFFSET (buf) = offset; |
| GST_BUFFER_OFFSET_END (buf) = offset + length; |
| |
| *buffer = buf; |
| |
| return ret; |
| |
| /* ERRORS */ |
| hit_eos: |
| { |
| GST_DEBUG_OBJECT (queue, "EOS hit and we don't have any requested data"); |
| gst_buffer_unmap (buf, &info); |
| if (*buffer == NULL) |
| gst_buffer_unref (buf); |
| return GST_FLOW_EOS; |
| } |
| out_flushing: |
| { |
| GST_DEBUG_OBJECT (queue, "we are flushing"); |
| gst_buffer_unmap (buf, &info); |
| if (*buffer == NULL) |
| gst_buffer_unref (buf); |
| return GST_FLOW_FLUSHING; |
| } |
| read_error: |
| { |
| GST_DEBUG_OBJECT (queue, "we have a read error"); |
| gst_buffer_unmap (buf, &info); |
| if (*buffer == NULL) |
| gst_buffer_unref (buf); |
| return ret; |
| } |
| buffer_write_fail: |
| { |
| GST_ELEMENT_ERROR (queue, RESOURCE, WRITE, (NULL), |
| ("Can't write to buffer")); |
| if (*buffer == NULL) |
| gst_buffer_unref (buf); |
| return GST_FLOW_ERROR; |
| } |
| } |
| |
| /* should be called with QUEUE_LOCK */ |
| static GstMiniObject * |
| gst_queue2_read_item_from_file (GstQueue2 * queue) |
| { |
| GstMiniObject *item; |
| |
| if (queue->stream_start_event != NULL) { |
| item = GST_MINI_OBJECT_CAST (queue->stream_start_event); |
| queue->stream_start_event = NULL; |
| } else if (queue->starting_segment != NULL) { |
| item = GST_MINI_OBJECT_CAST (queue->starting_segment); |
| queue->starting_segment = NULL; |
| } else { |
| GstFlowReturn ret; |
| GstBuffer *buffer = NULL; |
| guint64 reading_pos; |
| |
| reading_pos = queue->current->reading_pos; |
| |
| ret = |
| gst_queue2_create_read (queue, reading_pos, DEFAULT_BUFFER_SIZE, |
| &buffer); |
| |
| switch (ret) { |
| case GST_FLOW_OK: |
| item = GST_MINI_OBJECT_CAST (buffer); |
| break; |
| case GST_FLOW_EOS: |
| item = GST_MINI_OBJECT_CAST (gst_event_new_eos ()); |
| break; |
| default: |
| item = NULL; |
| break; |
| } |
| } |
| return item; |
| } |
| |
| /* must be called with MUTEX_LOCK. Will briefly release the lock when notifying |
| * the temp filename. */ |
| static gboolean |
| gst_queue2_open_temp_location_file (GstQueue2 * queue) |
| { |
| gint fd = -1; |
| gchar *name = NULL; |
| |
| if (queue->temp_file) |
| goto already_opened; |
| |
| GST_DEBUG_OBJECT (queue, "opening temp file %s", queue->temp_template); |
| |
| /* If temp_template was set, allocate a filename and open that file */ |
| |
| /* nothing to do */ |
| if (queue->temp_template == NULL) |
| goto no_directory; |
| |
| /* make copy of the template, we don't want to change this */ |
| name = g_strdup (queue->temp_template); |
| |
| #ifdef __BIONIC__ |
| fd = g_mkstemp_full (name, O_RDWR | O_LARGEFILE, S_IRUSR | S_IWUSR); |
| #else |
| fd = g_mkstemp (name); |
| #endif |
| |
| if (fd == -1) |
| goto mkstemp_failed; |
| |
| /* open the file for update/writing */ |
| queue->temp_file = fdopen (fd, "wb+"); |
| /* error creating file */ |
| if (queue->temp_file == NULL) |
| goto open_failed; |
| |
| g_free (queue->temp_location); |
| queue->temp_location = name; |
| |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| |
| /* we can't emit the notify with the lock */ |
| g_object_notify (G_OBJECT (queue), "temp-location"); |
| |
| GST_QUEUE2_MUTEX_LOCK (queue); |
| |
| GST_DEBUG_OBJECT (queue, "opened temp file %s", queue->temp_template); |
| |
| return TRUE; |
| |
| /* ERRORS */ |
| already_opened: |
| { |
| GST_DEBUG_OBJECT (queue, "temp file was already open"); |
| return TRUE; |
| } |
| no_directory: |
| { |
| GST_ELEMENT_ERROR (queue, RESOURCE, NOT_FOUND, |
| (_("No Temp directory specified.")), (NULL)); |
| return FALSE; |
| } |
| mkstemp_failed: |
| { |
| GST_ELEMENT_ERROR (queue, RESOURCE, OPEN_READ, |
| (_("Could not create temp file \"%s\"."), queue->temp_template), |
| GST_ERROR_SYSTEM); |
| g_free (name); |
| return FALSE; |
| } |
| open_failed: |
| { |
| GST_ELEMENT_ERROR (queue, RESOURCE, OPEN_READ, |
| (_("Could not open file \"%s\" for reading."), name), GST_ERROR_SYSTEM); |
| g_free (name); |
| if (fd != -1) |
| close (fd); |
| return FALSE; |
| } |
| } |
| |
| static void |
| gst_queue2_close_temp_location_file (GstQueue2 * queue) |
| { |
| /* nothing to do */ |
| if (queue->temp_file == NULL) |
| return; |
| |
| GST_DEBUG_OBJECT (queue, "closing temp file"); |
| |
| fflush (queue->temp_file); |
| fclose (queue->temp_file); |
| |
| if (queue->temp_remove) { |
| if (remove (queue->temp_location) < 0) { |
| GST_WARNING_OBJECT (queue, "Failed to remove temporary file %s: %s", |
| queue->temp_location, g_strerror (errno)); |
| } |
| } |
| |
| queue->temp_file = NULL; |
| clean_ranges (queue); |
| } |
| |
| static void |
| gst_queue2_flush_temp_file (GstQueue2 * queue) |
| { |
| if (queue->temp_file == NULL) |
| return; |
| |
| GST_DEBUG_OBJECT (queue, "flushing temp file"); |
| |
| queue->temp_file = g_freopen (queue->temp_location, "wb+", queue->temp_file); |
| } |
| |
| static void |
| gst_queue2_locked_flush (GstQueue2 * queue, gboolean full, gboolean clear_temp) |
| { |
| if (!QUEUE_IS_USING_QUEUE (queue)) { |
| if (QUEUE_IS_USING_TEMP_FILE (queue) && clear_temp) |
| gst_queue2_flush_temp_file (queue); |
| init_ranges (queue); |
| } else { |
| while (!g_queue_is_empty (&queue->queue)) { |
| GstQueue2Item *qitem = g_queue_pop_head (&queue->queue); |
| |
| if (!full && qitem->type == GST_QUEUE2_ITEM_TYPE_EVENT |
| && GST_EVENT_IS_STICKY (qitem->item) |
| && GST_EVENT_TYPE (qitem->item) != GST_EVENT_SEGMENT |
| && GST_EVENT_TYPE (qitem->item) != GST_EVENT_EOS) { |
| gst_pad_store_sticky_event (queue->srcpad, |
| GST_EVENT_CAST (qitem->item)); |
| } |
| |
| /* Then lose another reference because we are supposed to destroy that |
| data when flushing */ |
| if (qitem->type != GST_QUEUE2_ITEM_TYPE_QUERY) |
| gst_mini_object_unref (qitem->item); |
| g_slice_free (GstQueue2Item, qitem); |
| } |
| } |
| queue->last_query = FALSE; |
| g_cond_signal (&queue->query_handled); |
| GST_QUEUE2_CLEAR_LEVEL (queue->cur_level); |
| gst_segment_init (&queue->sink_segment, GST_FORMAT_TIME); |
| gst_segment_init (&queue->src_segment, GST_FORMAT_TIME); |
| queue->sinktime = queue->srctime = GST_CLOCK_TIME_NONE; |
| queue->sink_tainted = queue->src_tainted = TRUE; |
| if (queue->starting_segment != NULL) |
| gst_event_unref (queue->starting_segment); |
| queue->starting_segment = NULL; |
| queue->segment_event_received = FALSE; |
| gst_event_replace (&queue->stream_start_event, NULL); |
| |
| /* we deleted a lot of something */ |
| GST_QUEUE2_SIGNAL_DEL (queue); |
| } |
| |
| static gboolean |
| gst_queue2_wait_free_space (GstQueue2 * queue) |
| { |
| /* We make space available if we're "full" according to whatever |
| * the user defined as "full". */ |
| if (gst_queue2_is_filled (queue)) { |
| gboolean started; |
| |
| /* pause the timer while we wait. The fact that we are waiting does not mean |
| * the byterate on the input pad is lower */ |
| if ((started = queue->in_timer_started)) |
| g_timer_stop (queue->in_timer); |
| |
| GST_CAT_DEBUG_OBJECT (queue_dataflow, queue, |
| "queue is full, waiting for free space"); |
| do { |
| /* Wait for space to be available, we could be unlocked because of a flush. */ |
| GST_QUEUE2_WAIT_DEL_CHECK (queue, queue->sinkresult, out_flushing); |
| } |
| while (gst_queue2_is_filled (queue)); |
| |
| /* and continue if we were running before */ |
| if (started) |
| g_timer_continue (queue->in_timer); |
| } |
| return TRUE; |
| |
| /* ERRORS */ |
| out_flushing: |
| { |
| GST_CAT_DEBUG_OBJECT (queue_dataflow, queue, "queue is flushing"); |
| return FALSE; |
| } |
| } |
| |
| static gboolean |
| gst_queue2_create_write (GstQueue2 * queue, GstBuffer * buffer) |
| { |
| GstMapInfo info; |
| guint8 *data, *ring_buffer; |
| guint size, rb_size; |
| guint64 writing_pos, new_writing_pos; |
| GstQueue2Range *range, *prev, *next; |
| gboolean do_seek = FALSE; |
| |
| if (QUEUE_IS_USING_RING_BUFFER (queue)) |
| writing_pos = queue->current->rb_writing_pos; |
| else |
| writing_pos = queue->current->writing_pos; |
| ring_buffer = queue->ring_buffer; |
| rb_size = queue->ring_buffer_max_size; |
| |
| if (!gst_buffer_map (buffer, &info, GST_MAP_READ)) |
| goto buffer_read_error; |
| |
| size = info.size; |
| data = info.data; |
| |
| GST_DEBUG_OBJECT (queue, "Writing %u bytes to %" G_GUINT64_FORMAT, size, |
| writing_pos); |
| |
| /* sanity check */ |
| if (GST_BUFFER_OFFSET_IS_VALID (buffer) && |
| GST_BUFFER_OFFSET (buffer) != queue->current->writing_pos) { |
| GST_WARNING_OBJECT (queue, "buffer offset does not match current writing " |
| "position! %" G_GINT64_FORMAT " != %" G_GINT64_FORMAT, |
| GST_BUFFER_OFFSET (buffer), queue->current->writing_pos); |
| } |
| |
| while (size > 0) { |
| guint to_write; |
| |
| if (QUEUE_IS_USING_RING_BUFFER (queue)) { |
| gint64 space; |
| |
| /* calculate the space in the ring buffer not used by data from |
| * the current range */ |
| while (QUEUE_MAX_BYTES (queue) <= queue->cur_level.bytes) { |
| /* wait until there is some free space */ |
| GST_QUEUE2_WAIT_DEL_CHECK (queue, queue->sinkresult, out_flushing); |
| } |
| /* get the amount of space we have */ |
| space = QUEUE_MAX_BYTES (queue) - queue->cur_level.bytes; |
| |
| /* calculate if we need to split or if we can write the entire |
| * buffer now */ |
| to_write = MIN (size, space); |
| |
| /* the writing position in the ring buffer after writing (part |
| * or all of) the buffer */ |
| new_writing_pos = (writing_pos + to_write) % rb_size; |
| |
| prev = NULL; |
| range = queue->ranges; |
| |
| /* if we need to overwrite data in the ring buffer, we need to |
| * update the ranges |
| * |
| * warning: this code is complicated and includes some |
| * simplifications - pen, paper and diagrams for the cases |
| * recommended! */ |
| while (range) { |
| guint64 range_data_start, range_data_end; |
| GstQueue2Range *range_to_destroy = NULL; |
| |
| range_data_start = range->rb_offset; |
| range_data_end = range->rb_writing_pos; |
| |
| /* handle the special case where the range has no data in it */ |
| if (range->writing_pos == range->offset) { |
| if (range != queue->current) { |
| GST_DEBUG_OBJECT (queue, |
| "Removing range: offset %" G_GUINT64_FORMAT ", wpos %" |
| G_GUINT64_FORMAT, range->offset, range->writing_pos); |
| /* remove range */ |
| range_to_destroy = range; |
| if (prev) |
| prev->next = range->next; |
| } |
| goto next_range; |
| } |
| |
| if (range_data_end > range_data_start) { |
| if (writing_pos >= range_data_end && new_writing_pos >= writing_pos) |
| goto next_range; |
| |
| if (new_writing_pos > range_data_start) { |
| if (new_writing_pos >= range_data_end) { |
| GST_DEBUG_OBJECT (queue, |
| "Removing range: offset %" G_GUINT64_FORMAT ", wpos %" |
| G_GUINT64_FORMAT, range->offset, range->writing_pos); |
| /* remove range */ |
| range_to_destroy = range; |
| if (prev) |
| prev->next = range->next; |
| } else { |
| GST_DEBUG_OBJECT (queue, |
| "advancing offsets from %" G_GUINT64_FORMAT " (%" |
| G_GUINT64_FORMAT ") to %" G_GUINT64_FORMAT " (%" |
| G_GUINT64_FORMAT ")", range->offset, range->rb_offset, |
| range->offset + new_writing_pos - range_data_start, |
| new_writing_pos); |
| range->offset += (new_writing_pos - range_data_start); |
| range->rb_offset = new_writing_pos; |
| } |
| } |
| } else { |
| guint64 new_wpos_virt = writing_pos + to_write; |
| |
| if (new_wpos_virt <= range_data_start) |
| goto next_range; |
| |
| if (new_wpos_virt > rb_size && new_writing_pos >= range_data_end) { |
| GST_DEBUG_OBJECT (queue, |
| "Removing range: offset %" G_GUINT64_FORMAT ", wpos %" |
| G_GUINT64_FORMAT, range->offset, range->writing_pos); |
| /* remove range */ |
| range_to_destroy = range; |
| if (prev) |
| prev->next = range->next; |
| } else { |
| GST_DEBUG_OBJECT (queue, |
| "advancing offsets from %" G_GUINT64_FORMAT " (%" |
| G_GUINT64_FORMAT ") to %" G_GUINT64_FORMAT " (%" |
| G_GUINT64_FORMAT ")", range->offset, range->rb_offset, |
| range->offset + new_writing_pos - range_data_start, |
| new_writing_pos); |
| range->offset += (new_wpos_virt - range_data_start); |
| range->rb_offset = new_writing_pos; |
| } |
| } |
| |
| next_range: |
| if (!range_to_destroy) |
| prev = range; |
| |
| range = range->next; |
| if (range_to_destroy) { |
| if (range_to_destroy == queue->ranges) |
| queue->ranges = range; |
| g_slice_free (GstQueue2Range, range_to_destroy); |
| range_to_destroy = NULL; |
| } |
| } |
| } else { |
| to_write = size; |
| new_writing_pos = writing_pos + to_write; |
| } |
| |
| if (QUEUE_IS_USING_TEMP_FILE (queue) |
| && FSEEK_FILE (queue->temp_file, writing_pos)) |
| goto seek_failed; |
| |
| if (new_writing_pos > writing_pos) { |
| GST_INFO_OBJECT (queue, |
| "writing %u bytes to range [%" G_GUINT64_FORMAT "-%" G_GUINT64_FORMAT |
| "] (rb wpos %" G_GUINT64_FORMAT ")", to_write, queue->current->offset, |
| queue->current->writing_pos, queue->current->rb_writing_pos); |
| /* either not using ring buffer or no wrapping, just write */ |
| if (QUEUE_IS_USING_TEMP_FILE (queue)) { |
| if (fwrite (data, to_write, 1, queue->temp_file) != 1) |
| goto handle_error; |
| } else { |
| memcpy (ring_buffer + writing_pos, data, to_write); |
| } |
| |
| if (!QUEUE_IS_USING_RING_BUFFER (queue)) { |
| /* try to merge with next range */ |
| while ((next = queue->current->next)) { |
| GST_INFO_OBJECT (queue, |
| "checking merge with next range %" G_GUINT64_FORMAT " < %" |
| G_GUINT64_FORMAT, new_writing_pos, next->offset); |
| if (new_writing_pos < next->offset) |
| break; |
| |
| GST_DEBUG_OBJECT (queue, "merging ranges %" G_GUINT64_FORMAT, |
| next->writing_pos); |
| |
| /* remove the group */ |
| queue->current->next = next->next; |
| |
| /* We use the threshold to decide if we want to do a seek or simply |
| * read the data again. If there is not so much data in the range we |
| * prefer to avoid to seek and read it again. */ |
| if (next->writing_pos > new_writing_pos + get_seek_threshold (queue)) { |
| /* the new range had more data than the threshold, it's worth keeping |
| * it and doing a seek. */ |
| new_writing_pos = next->writing_pos; |
| do_seek = TRUE; |
| } |
| g_slice_free (GstQueue2Range, next); |
| } |
| goto update_and_signal; |
| } |
| } else { |
| /* wrapping */ |
| guint block_one, block_two; |
| |
| block_one = rb_size - writing_pos; |
| block_two = to_write - block_one; |
| |
| if (block_one > 0) { |
| GST_INFO_OBJECT (queue, "writing %u bytes", block_one); |
| /* write data to end of ring buffer */ |
| if (QUEUE_IS_USING_TEMP_FILE (queue)) { |
| if (fwrite (data, block_one, 1, queue->temp_file) != 1) |
| goto handle_error; |
| } else { |
| memcpy (ring_buffer + writing_pos, data, block_one); |
| } |
| } |
| |
| if (QUEUE_IS_USING_TEMP_FILE (queue) && FSEEK_FILE (queue->temp_file, 0)) |
| goto seek_failed; |
| |
| if (block_two > 0) { |
| GST_INFO_OBJECT (queue, "writing %u bytes", block_two); |
| if (QUEUE_IS_USING_TEMP_FILE (queue)) { |
| if (fwrite (data + block_one, block_two, 1, queue->temp_file) != 1) |
| goto handle_error; |
| } else { |
| memcpy (ring_buffer, data + block_one, block_two); |
| } |
| } |
| } |
| |
| update_and_signal: |
| /* update the writing positions */ |
| size -= to_write; |
| GST_INFO_OBJECT (queue, |
| "wrote %u bytes to %" G_GUINT64_FORMAT " (%u bytes remaining to write)", |
| to_write, writing_pos, size); |
| |
| if (QUEUE_IS_USING_RING_BUFFER (queue)) { |
| data += to_write; |
| queue->current->writing_pos += to_write; |
| queue->current->rb_writing_pos = writing_pos = new_writing_pos; |
| } else { |
| queue->current->writing_pos = writing_pos = new_writing_pos; |
| } |
| if (do_seek) |
| perform_seek_to_offset (queue, new_writing_pos); |
| |
| update_cur_level (queue, queue->current); |
| |
| /* update the buffering status */ |
| if (queue->use_buffering) |
| update_buffering (queue); |
| |
| GST_INFO_OBJECT (queue, "cur_level.bytes %u (max %" G_GUINT64_FORMAT ")", |
| queue->cur_level.bytes, QUEUE_MAX_BYTES (queue)); |
| |
| GST_QUEUE2_SIGNAL_ADD (queue); |
| } |
| |
| gst_buffer_unmap (buffer, &info); |
| |
| return TRUE; |
| |
| /* ERRORS */ |
| out_flushing: |
| { |
| GST_DEBUG_OBJECT (queue, "we are flushing"); |
| gst_buffer_unmap (buffer, &info); |
| /* FIXME - GST_FLOW_EOS ? */ |
| return FALSE; |
| } |
| seek_failed: |
| { |
| GST_ELEMENT_ERROR (queue, RESOURCE, SEEK, (NULL), GST_ERROR_SYSTEM); |
| gst_buffer_unmap (buffer, &info); |
| return FALSE; |
| } |
| handle_error: |
| { |
| switch (errno) { |
| case ENOSPC:{ |
| GST_ELEMENT_ERROR (queue, RESOURCE, NO_SPACE_LEFT, (NULL), (NULL)); |
| break; |
| } |
| default:{ |
| GST_ELEMENT_ERROR (queue, RESOURCE, WRITE, |
| (_("Error while writing to download file.")), |
| ("%s", g_strerror (errno))); |
| } |
| } |
| gst_buffer_unmap (buffer, &info); |
| return FALSE; |
| } |
| buffer_read_error: |
| { |
| GST_ELEMENT_ERROR (queue, RESOURCE, READ, (NULL), |
| ("Can't read from buffer")); |
| return FALSE; |
| } |
| } |
| |
| static gboolean |
| buffer_list_create_write (GstBuffer ** buf, guint idx, gpointer q) |
| { |
| GstQueue2 *queue = q; |
| |
| GST_TRACE_OBJECT (queue, |
| "writing buffer %u of size %" G_GSIZE_FORMAT " bytes", idx, |
| gst_buffer_get_size (*buf)); |
| |
| if (!gst_queue2_create_write (queue, *buf)) { |
| GST_INFO_OBJECT (queue, "create_write() returned FALSE, bailing out"); |
| return FALSE; |
| } |
| return TRUE; |
| } |
| |
| static gboolean |
| buffer_list_calc_size (GstBuffer ** buf, guint idx, gpointer data) |
| { |
| guint *p_size = data; |
| gsize buf_size; |
| |
| buf_size = gst_buffer_get_size (*buf); |
| GST_TRACE ("buffer %u in has size %" G_GSIZE_FORMAT, idx, buf_size); |
| *p_size += buf_size; |
| return TRUE; |
| } |
| |
| /* enqueue an item an update the level stats */ |
| static void |
| gst_queue2_locked_enqueue (GstQueue2 * queue, gpointer item, |
| GstQueue2ItemType item_type) |
| { |
| if (item_type == GST_QUEUE2_ITEM_TYPE_BUFFER) { |
| GstBuffer *buffer; |
| guint size; |
| |
| buffer = GST_BUFFER_CAST (item); |
| size = gst_buffer_get_size (buffer); |
| |
| /* add buffer to the statistics */ |
| if (QUEUE_IS_USING_QUEUE (queue)) { |
| queue->cur_level.buffers++; |
| queue->cur_level.bytes += size; |
| } |
| queue->bytes_in += size; |
| |
| /* apply new buffer to segment stats */ |
| apply_buffer (queue, buffer, &queue->sink_segment, size, TRUE); |
| /* update the byterate stats */ |
| update_in_rates (queue); |
| |
| if (!QUEUE_IS_USING_QUEUE (queue)) { |
| /* FIXME - check return value? */ |
| gst_queue2_create_write (queue, buffer); |
| } |
| } else if (item_type == GST_QUEUE2_ITEM_TYPE_BUFFER_LIST) { |
| GstBufferList *buffer_list; |
| guint size = 0; |
| |
| buffer_list = GST_BUFFER_LIST_CAST (item); |
| |
| gst_buffer_list_foreach (buffer_list, buffer_list_calc_size, &size); |
| GST_LOG_OBJECT (queue, "total size of buffer list: %u bytes", size); |
| |
| /* add buffer to the statistics */ |
| if (QUEUE_IS_USING_QUEUE (queue)) { |
| queue->cur_level.buffers += gst_buffer_list_length (buffer_list); |
| queue->cur_level.bytes += size; |
| } |
| queue->bytes_in += size; |
| |
| /* apply new buffer to segment stats */ |
| apply_buffer_list (queue, buffer_list, &queue->sink_segment, TRUE); |
| |
| /* update the byterate stats */ |
| update_in_rates (queue); |
| |
| if (!QUEUE_IS_USING_QUEUE (queue)) { |
| gst_buffer_list_foreach (buffer_list, buffer_list_create_write, queue); |
| } |
| } else if (item_type == GST_QUEUE2_ITEM_TYPE_EVENT) { |
| GstEvent *event; |
| |
| event = GST_EVENT_CAST (item); |
| |
| switch (GST_EVENT_TYPE (event)) { |
| case GST_EVENT_EOS: |
| /* Zero the thresholds, this makes sure the queue is completely |
| * filled and we can read all data from the queue. */ |
| GST_DEBUG_OBJECT (queue, "we have EOS"); |
| queue->is_eos = TRUE; |
| break; |
| case GST_EVENT_SEGMENT: |
| apply_segment (queue, event, &queue->sink_segment, TRUE); |
| /* This is our first new segment, we hold it |
| * as we can't save it on the temp file */ |
| if (!QUEUE_IS_USING_QUEUE (queue)) { |
| if (queue->segment_event_received) |
| goto unexpected_event; |
| |
| queue->segment_event_received = TRUE; |
| if (queue->starting_segment != NULL) |
| gst_event_unref (queue->starting_segment); |
| queue->starting_segment = event; |
| item = NULL; |
| } |
| /* a new segment allows us to accept more buffers if we got EOS |
| * from downstream */ |
| queue->unexpected = FALSE; |
| break; |
| case GST_EVENT_GAP: |
| apply_gap (queue, event, &queue->sink_segment, TRUE); |
| break; |
| case GST_EVENT_STREAM_START: |
| if (!QUEUE_IS_USING_QUEUE (queue)) { |
| gst_event_replace (&queue->stream_start_event, event); |
| gst_event_unref (event); |
| item = NULL; |
| } |
| break; |
| case GST_EVENT_CAPS:{ |
| GstCaps *caps; |
| |
| gst_event_parse_caps (event, &caps); |
| GST_INFO ("got caps: %" GST_PTR_FORMAT, caps); |
| |
| if (!QUEUE_IS_USING_QUEUE (queue)) { |
| GST_LOG ("Dropping caps event, not using queue"); |
| gst_event_unref (event); |
| item = NULL; |
| } |
| break; |
| } |
| default: |
| if (!QUEUE_IS_USING_QUEUE (queue)) |
| goto unexpected_event; |
| break; |
| } |
| } else if (GST_IS_QUERY (item)) { |
| /* Can't happen as we check that in the caller */ |
| if (!QUEUE_IS_USING_QUEUE (queue)) |
| g_assert_not_reached (); |
| } else { |
| g_warning ("Unexpected item %p added in queue %s (refcounting problem?)", |
| item, GST_OBJECT_NAME (queue)); |
| /* we can't really unref since we don't know what it is */ |
| item = NULL; |
| } |
| |
| if (item) { |
| /* update the buffering status */ |
| if (queue->use_buffering) |
| update_buffering (queue); |
| |
| if (QUEUE_IS_USING_QUEUE (queue)) { |
| GstQueue2Item *qitem = g_slice_new (GstQueue2Item); |
| qitem->type = item_type; |
| qitem->item = item; |
| g_queue_push_tail (&queue->queue, qitem); |
| } else { |
| gst_mini_object_unref (GST_MINI_OBJECT_CAST (item)); |
| } |
| |
| GST_QUEUE2_SIGNAL_ADD (queue); |
| } |
| |
| return; |
| |
| /* ERRORS */ |
| unexpected_event: |
| { |
| gboolean is_custom = GST_EVENT_TYPE (item) < GST_EVENT_CUSTOM_UPSTREAM; |
| |
| GST_WARNING_OBJECT (queue, "%s%s event can't be added to temp file: " |
| "%" GST_PTR_FORMAT, is_custom ? "Unexpected " : "", |
| GST_EVENT_TYPE_NAME (item), GST_EVENT_CAST (item)); |
| gst_event_unref (GST_EVENT_CAST (item)); |
| return; |
| } |
| } |
| |
| /* dequeue an item from the queue and update level stats */ |
| static GstMiniObject * |
| gst_queue2_locked_dequeue (GstQueue2 * queue, GstQueue2ItemType * item_type) |
| { |
| GstMiniObject *item; |
| |
| if (!QUEUE_IS_USING_QUEUE (queue)) { |
| item = gst_queue2_read_item_from_file (queue); |
| } else { |
| GstQueue2Item *qitem = g_queue_pop_head (&queue->queue); |
| |
| if (qitem == NULL) |
| goto no_item; |
| |
| item = qitem->item; |
| g_slice_free (GstQueue2Item, qitem); |
| } |
| |
| if (item == NULL) |
| goto no_item; |
| |
| if (GST_IS_BUFFER (item)) { |
| GstBuffer *buffer; |
| guint size; |
| |
| buffer = GST_BUFFER_CAST (item); |
| size = gst_buffer_get_size (buffer); |
| *item_type = GST_QUEUE2_ITEM_TYPE_BUFFER; |
| |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, |
| "retrieved buffer %p from queue", buffer); |
| |
| if (QUEUE_IS_USING_QUEUE (queue)) { |
| queue->cur_level.buffers--; |
| queue->cur_level.bytes -= size; |
| } |
| queue->bytes_out += size; |
| |
| apply_buffer (queue, buffer, &queue->src_segment, size, FALSE); |
| /* update the byterate stats */ |
| update_out_rates (queue); |
| /* update the buffering */ |
| if (queue->use_buffering) |
| update_buffering (queue); |
| |
| } else if (GST_IS_EVENT (item)) { |
| GstEvent *event = GST_EVENT_CAST (item); |
| |
| *item_type = GST_QUEUE2_ITEM_TYPE_EVENT; |
| |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, |
| "retrieved event %p from queue", event); |
| |
| switch (GST_EVENT_TYPE (event)) { |
| case GST_EVENT_EOS: |
| /* queue is empty now that we dequeued the EOS */ |
| GST_QUEUE2_CLEAR_LEVEL (queue->cur_level); |
| break; |
| case GST_EVENT_SEGMENT: |
| apply_segment (queue, event, &queue->src_segment, FALSE); |
| break; |
| case GST_EVENT_GAP: |
| apply_gap (queue, event, &queue->src_segment, FALSE); |
| break; |
| default: |
| break; |
| } |
| } else if (GST_IS_BUFFER_LIST (item)) { |
| GstBufferList *buffer_list; |
| guint size = 0; |
| |
| buffer_list = GST_BUFFER_LIST_CAST (item); |
| gst_buffer_list_foreach (buffer_list, buffer_list_calc_size, &size); |
| *item_type = GST_QUEUE2_ITEM_TYPE_BUFFER_LIST; |
| |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, |
| "retrieved buffer list %p from queue", buffer_list); |
| |
| if (QUEUE_IS_USING_QUEUE (queue)) { |
| queue->cur_level.buffers -= gst_buffer_list_length (buffer_list); |
| queue->cur_level.bytes -= size; |
| } |
| queue->bytes_out += size; |
| |
| apply_buffer_list (queue, buffer_list, &queue->src_segment, FALSE); |
| /* update the byterate stats */ |
| update_out_rates (queue); |
| /* update the buffering */ |
| if (queue->use_buffering) |
| update_buffering (queue); |
| } else if (GST_IS_QUERY (item)) { |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, |
| "retrieved query %p from queue", item); |
| *item_type = GST_QUEUE2_ITEM_TYPE_QUERY; |
| } else { |
| g_warning |
| ("Unexpected item %p dequeued from queue %s (refcounting problem?)", |
| item, GST_OBJECT_NAME (queue)); |
| item = NULL; |
| *item_type = GST_QUEUE2_ITEM_TYPE_UNKNOWN; |
| } |
| GST_QUEUE2_SIGNAL_DEL (queue); |
| |
| return item; |
| |
| /* ERRORS */ |
| no_item: |
| { |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, "the queue is empty"); |
| return NULL; |
| } |
| } |
| |
| static gboolean |
| gst_queue2_handle_sink_event (GstPad * pad, GstObject * parent, |
| GstEvent * event) |
| { |
| gboolean ret = TRUE; |
| GstQueue2 *queue; |
| |
| queue = GST_QUEUE2 (parent); |
| |
| switch (GST_EVENT_TYPE (event)) { |
| case GST_EVENT_FLUSH_START: |
| { |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, "received flush start event"); |
| if (GST_PAD_MODE (queue->srcpad) == GST_PAD_MODE_PUSH) { |
| /* forward event */ |
| ret = gst_pad_push_event (queue->srcpad, event); |
| |
| /* now unblock the chain function */ |
| GST_QUEUE2_MUTEX_LOCK (queue); |
| queue->srcresult = GST_FLOW_FLUSHING; |
| queue->sinkresult = GST_FLOW_FLUSHING; |
| /* unblock the loop and chain functions */ |
| GST_QUEUE2_SIGNAL_ADD (queue); |
| GST_QUEUE2_SIGNAL_DEL (queue); |
| queue->last_query = FALSE; |
| g_cond_signal (&queue->query_handled); |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| |
| /* make sure it pauses, this should happen since we sent |
| * flush_start downstream. */ |
| gst_pad_pause_task (queue->srcpad); |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, "loop stopped"); |
| } else { |
| GST_QUEUE2_MUTEX_LOCK (queue); |
| /* flush the sink pad */ |
| queue->sinkresult = GST_FLOW_FLUSHING; |
| GST_QUEUE2_SIGNAL_DEL (queue); |
| queue->last_query = FALSE; |
| g_cond_signal (&queue->query_handled); |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| |
| gst_event_unref (event); |
| } |
| break; |
| } |
| case GST_EVENT_FLUSH_STOP: |
| { |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, "received flush stop event"); |
| |
| if (GST_PAD_MODE (queue->srcpad) == GST_PAD_MODE_PUSH) { |
| /* forward event */ |
| ret = gst_pad_push_event (queue->srcpad, event); |
| |
| GST_QUEUE2_MUTEX_LOCK (queue); |
| gst_queue2_locked_flush (queue, FALSE, TRUE); |
| queue->srcresult = GST_FLOW_OK; |
| queue->sinkresult = GST_FLOW_OK; |
| queue->is_eos = FALSE; |
| queue->unexpected = FALSE; |
| queue->seeking = FALSE; |
| queue->src_tags_bitrate = queue->sink_tags_bitrate = 0; |
| /* reset rate counters */ |
| reset_rate_timer (queue); |
| gst_pad_start_task (queue->srcpad, (GstTaskFunction) gst_queue2_loop, |
| queue->srcpad, NULL); |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| } else { |
| GST_QUEUE2_MUTEX_LOCK (queue); |
| queue->segment_event_received = FALSE; |
| queue->is_eos = FALSE; |
| queue->unexpected = FALSE; |
| queue->sinkresult = GST_FLOW_OK; |
| queue->seeking = FALSE; |
| queue->src_tags_bitrate = queue->sink_tags_bitrate = 0; |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| |
| gst_event_unref (event); |
| } |
| break; |
| } |
| case GST_EVENT_TAG:{ |
| if (queue->use_tags_bitrate) { |
| GstTagList *tags; |
| guint bitrate; |
| |
| gst_event_parse_tag (event, &tags); |
| if (gst_tag_list_get_uint (tags, GST_TAG_BITRATE, &bitrate) || |
| gst_tag_list_get_uint (tags, GST_TAG_NOMINAL_BITRATE, &bitrate)) { |
| GST_QUEUE2_MUTEX_LOCK (queue); |
| queue->sink_tags_bitrate = bitrate; |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| GST_LOG_OBJECT (queue, "Sink pad bitrate from tags now %u", bitrate); |
| } |
| } |
| /* Fall-through */ |
| } |
| default: |
| if (GST_EVENT_IS_SERIALIZED (event)) { |
| /* serialized events go in the queue */ |
| GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->sinkresult, out_flushing); |
| if (queue->srcresult != GST_FLOW_OK) { |
| /* Errors in sticky event pushing are no problem and ignored here |
| * as they will cause more meaningful errors during data flow. |
| * For EOS events, that are not followed by data flow, we still |
| * return FALSE here though and report an error. |
| */ |
| if (!GST_EVENT_IS_STICKY (event)) { |
| goto out_flow_error; |
| } else if (GST_EVENT_TYPE (event) == GST_EVENT_EOS) { |
| if (queue->srcresult == GST_FLOW_NOT_LINKED |
| || queue->srcresult < GST_FLOW_EOS) { |
| GST_ELEMENT_ERROR (queue, STREAM, FAILED, |
| (_("Internal data flow error.")), |
| ("streaming task paused, reason %s (%d)", |
| gst_flow_get_name (queue->srcresult), queue->srcresult)); |
| } |
| goto out_flow_error; |
| } |
| } |
| /* refuse more events on EOS */ |
| if (queue->is_eos) |
| goto out_eos; |
| gst_queue2_locked_enqueue (queue, event, GST_QUEUE2_ITEM_TYPE_EVENT); |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| gst_queue2_post_buffering (queue); |
| } else { |
| /* non-serialized events are passed upstream. */ |
| ret = gst_pad_push_event (queue->srcpad, event); |
| } |
| break; |
| } |
| return ret; |
| |
| /* ERRORS */ |
| out_flushing: |
| { |
| GST_DEBUG_OBJECT (queue, "refusing event, we are flushing"); |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| gst_event_unref (event); |
| return FALSE; |
| } |
| out_eos: |
| { |
| GST_DEBUG_OBJECT (queue, "refusing event, we are EOS"); |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| gst_event_unref (event); |
| return FALSE; |
| } |
| out_flow_error: |
| { |
| GST_LOG_OBJECT (queue, |
| "refusing event, we have a downstream flow error: %s", |
| gst_flow_get_name (queue->srcresult)); |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| gst_event_unref (event); |
| return FALSE; |
| } |
| } |
| |
| static gboolean |
| gst_queue2_handle_sink_query (GstPad * pad, GstObject * parent, |
| GstQuery * query) |
| { |
| GstQueue2 *queue; |
| gboolean res; |
| |
| queue = GST_QUEUE2 (parent); |
| |
| switch (GST_QUERY_TYPE (query)) { |
| default: |
| if (GST_QUERY_IS_SERIALIZED (query)) { |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, "received query %p", query); |
| /* serialized events go in the queue. We need to be certain that we |
| * don't cause deadlocks waiting for the query return value. We check if |
| * the queue is empty (nothing is blocking downstream and the query can |
| * be pushed for sure) or we are not buffering. If we are buffering, |
| * the pipeline waits to unblock downstream until our queue fills up |
| * completely, which can not happen if we block on the query.. |
| * Therefore we only potentially block when we are not buffering. */ |
| GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->sinkresult, out_flushing); |
| if (QUEUE_IS_USING_QUEUE (queue) && (gst_queue2_is_empty (queue) |
| || !queue->use_buffering)) { |
| if (!g_atomic_int_get (&queue->downstream_may_block)) { |
| gst_queue2_locked_enqueue (queue, query, |
| GST_QUEUE2_ITEM_TYPE_QUERY); |
| |
| STATUS (queue, queue->sinkpad, "wait for QUERY"); |
| g_cond_wait (&queue->query_handled, &queue->qlock); |
| if (queue->sinkresult != GST_FLOW_OK) |
| goto out_flushing; |
| res = queue->last_query; |
| } else { |
| GST_DEBUG_OBJECT (queue, "refusing query, downstream might block"); |
| res = FALSE; |
| } |
| } else { |
| GST_DEBUG_OBJECT (queue, |
| "refusing query, we are not using the queue"); |
| res = FALSE; |
| } |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| gst_queue2_post_buffering (queue); |
| } else { |
| res = gst_pad_query_default (pad, parent, query); |
| } |
| break; |
| } |
| return res; |
| |
| /* ERRORS */ |
| out_flushing: |
| { |
| GST_DEBUG_OBJECT (queue, "refusing query, we are flushing"); |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| return FALSE; |
| } |
| } |
| |
| static gboolean |
| gst_queue2_is_empty (GstQueue2 * queue) |
| { |
| /* never empty on EOS */ |
| if (queue->is_eos) |
| return FALSE; |
| |
| if (!QUEUE_IS_USING_QUEUE (queue) && queue->current) { |
| return queue->current->writing_pos <= queue->current->max_reading_pos; |
| } else { |
| if (queue->queue.length == 0) |
| return TRUE; |
| } |
| |
| return FALSE; |
| } |
| |
| static gboolean |
| gst_queue2_is_filled (GstQueue2 * queue) |
| { |
| gboolean res; |
| |
| /* always filled on EOS */ |
| if (queue->is_eos) |
| return TRUE; |
| |
| #define CHECK_FILLED(format,alt_max) ((queue->max_level.format) > 0 && \ |
| (queue->cur_level.format) >= ((alt_max) ? \ |
| MIN ((queue->max_level.format), (alt_max)) : (queue->max_level.format))) |
| |
| /* if using a ring buffer we're filled if all ring buffer space is used |
| * _by the current range_ */ |
| if (QUEUE_IS_USING_RING_BUFFER (queue)) { |
| guint64 rb_size = queue->ring_buffer_max_size; |
| GST_DEBUG_OBJECT (queue, |
| "max bytes %u, rb size %" G_GUINT64_FORMAT ", cur bytes %u", |
| queue->max_level.bytes, rb_size, queue->cur_level.bytes); |
| return CHECK_FILLED (bytes, rb_size); |
| } |
| |
| /* if using file, we're never filled if we don't have EOS */ |
| if (QUEUE_IS_USING_TEMP_FILE (queue)) |
| return FALSE; |
| |
| /* we are never filled when we have no buffers at all */ |
| if (queue->cur_level.buffers == 0) |
| return FALSE; |
| |
| /* we are filled if one of the current levels exceeds the max */ |
| res = CHECK_FILLED (buffers, 0) || CHECK_FILLED (bytes, 0) |
| || CHECK_FILLED (time, 0); |
| |
| /* if we need to, use the rate estimate to check against the max time we are |
| * allowed to queue */ |
| if (queue->use_rate_estimate) |
| res |= CHECK_FILLED (rate_time, 0); |
| |
| #undef CHECK_FILLED |
| return res; |
| } |
| |
| static GstFlowReturn |
| gst_queue2_chain_buffer_or_buffer_list (GstQueue2 * queue, |
| GstMiniObject * item, GstQueue2ItemType item_type) |
| { |
| /* we have to lock the queue since we span threads */ |
| GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->sinkresult, out_flushing); |
| /* when we received EOS, we refuse more data */ |
| if (queue->is_eos) |
| goto out_eos; |
| /* when we received unexpected from downstream, refuse more buffers */ |
| if (queue->unexpected) |
| goto out_unexpected; |
| |
| /* while we didn't receive the newsegment, we're seeking and we skip data */ |
| if (queue->seeking) |
| goto out_seeking; |
| |
| if (!gst_queue2_wait_free_space (queue)) |
| goto out_flushing; |
| |
| /* put buffer in queue now */ |
| gst_queue2_locked_enqueue (queue, item, item_type); |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| gst_queue2_post_buffering (queue); |
| |
| return GST_FLOW_OK; |
| |
| /* special conditions */ |
| out_flushing: |
| { |
| GstFlowReturn ret = queue->sinkresult; |
| |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, |
| "exit because task paused, reason: %s", gst_flow_get_name (ret)); |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| gst_mini_object_unref (item); |
| |
| return ret; |
| } |
| out_eos: |
| { |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, "exit because we received EOS"); |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| gst_mini_object_unref (item); |
| |
| return GST_FLOW_EOS; |
| } |
| out_seeking: |
| { |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, "exit because we are seeking"); |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| gst_mini_object_unref (item); |
| |
| return GST_FLOW_OK; |
| } |
| out_unexpected: |
| { |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, "exit because we received EOS"); |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| gst_mini_object_unref (item); |
| |
| return GST_FLOW_EOS; |
| } |
| } |
| |
| static GstFlowReturn |
| gst_queue2_chain (GstPad * pad, GstObject * parent, GstBuffer * buffer) |
| { |
| GstQueue2 *queue; |
| |
| queue = GST_QUEUE2 (parent); |
| |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, "received buffer %p of " |
| "size %" G_GSIZE_FORMAT ", time %" GST_TIME_FORMAT ", duration %" |
| GST_TIME_FORMAT, buffer, gst_buffer_get_size (buffer), |
| GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (buffer)), |
| GST_TIME_ARGS (GST_BUFFER_DURATION (buffer))); |
| |
| return gst_queue2_chain_buffer_or_buffer_list (queue, |
| GST_MINI_OBJECT_CAST (buffer), GST_QUEUE2_ITEM_TYPE_BUFFER); |
| } |
| |
| static GstFlowReturn |
| gst_queue2_chain_list (GstPad * pad, GstObject * parent, |
| GstBufferList * buffer_list) |
| { |
| GstQueue2 *queue; |
| |
| queue = GST_QUEUE2 (parent); |
| |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, |
| "received buffer list %p", buffer_list); |
| |
| return gst_queue2_chain_buffer_or_buffer_list (queue, |
| GST_MINI_OBJECT_CAST (buffer_list), GST_QUEUE2_ITEM_TYPE_BUFFER_LIST); |
| } |
| |
| static GstMiniObject * |
| gst_queue2_dequeue_on_eos (GstQueue2 * queue, GstQueue2ItemType * item_type) |
| { |
| GstMiniObject *data; |
| |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, "got EOS from downstream"); |
| |
| /* stop pushing buffers, we dequeue all items until we see an item that we |
| * can push again, which is EOS or SEGMENT. If there is nothing in the |
| * queue we can push, we set a flag to make the sinkpad refuse more |
| * buffers with an EOS return value until we receive something |
| * pushable again or we get flushed. */ |
| while ((data = gst_queue2_locked_dequeue (queue, item_type))) { |
| if (*item_type == GST_QUEUE2_ITEM_TYPE_BUFFER) { |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, |
| "dropping EOS buffer %p", data); |
| gst_buffer_unref (GST_BUFFER_CAST (data)); |
| } else if (*item_type == GST_QUEUE2_ITEM_TYPE_EVENT) { |
| GstEvent *event = GST_EVENT_CAST (data); |
| GstEventType type = GST_EVENT_TYPE (event); |
| |
| if (type == GST_EVENT_EOS || type == GST_EVENT_SEGMENT) { |
| /* we found a pushable item in the queue, push it out */ |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, |
| "pushing pushable event %s after EOS", GST_EVENT_TYPE_NAME (event)); |
| return data; |
| } |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, |
| "dropping EOS event %p", event); |
| gst_event_unref (event); |
| } else if (*item_type == GST_QUEUE2_ITEM_TYPE_BUFFER_LIST) { |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, |
| "dropping EOS buffer list %p", data); |
| gst_buffer_list_unref (GST_BUFFER_LIST_CAST (data)); |
| } else if (*item_type == GST_QUEUE2_ITEM_TYPE_QUERY) { |
| queue->last_query = FALSE; |
| g_cond_signal (&queue->query_handled); |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, "dropping EOS query %p", data); |
| } |
| } |
| /* no more items in the queue. Set the unexpected flag so that upstream |
| * make us refuse any more buffers on the sinkpad. Since we will still |
| * accept EOS and SEGMENT we return _FLOW_OK to the caller so that the |
| * task function does not shut down. */ |
| queue->unexpected = TRUE; |
| return NULL; |
| } |
| |
| /* dequeue an item from the queue an push it downstream. This functions returns |
| * the result of the push. */ |
| static GstFlowReturn |
| gst_queue2_push_one (GstQueue2 * queue) |
| { |
| GstFlowReturn result = queue->srcresult; |
| GstMiniObject *data; |
| GstQueue2ItemType item_type; |
| |
| data = gst_queue2_locked_dequeue (queue, &item_type); |
| if (data == NULL) |
| goto no_item; |
| |
| next: |
| STATUS (queue, queue->srcpad, "We have something dequeud"); |
| g_atomic_int_set (&queue->downstream_may_block, |
| item_type == GST_QUEUE2_ITEM_TYPE_BUFFER || |
| item_type == GST_QUEUE2_ITEM_TYPE_BUFFER_LIST); |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| gst_queue2_post_buffering (queue); |
| |
| if (item_type == GST_QUEUE2_ITEM_TYPE_BUFFER) { |
| GstBuffer *buffer; |
| |
| buffer = GST_BUFFER_CAST (data); |
| |
| result = gst_pad_push (queue->srcpad, buffer); |
| g_atomic_int_set (&queue->downstream_may_block, 0); |
| |
| /* need to check for srcresult here as well */ |
| GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->srcresult, out_flushing); |
| if (result == GST_FLOW_EOS) { |
| data = gst_queue2_dequeue_on_eos (queue, &item_type); |
| if (data != NULL) |
| goto next; |
| /* Since we will still accept EOS and SEGMENT we return _FLOW_OK |
| * to the caller so that the task function does not shut down */ |
| result = GST_FLOW_OK; |
| } |
| } else if (item_type == GST_QUEUE2_ITEM_TYPE_EVENT) { |
| GstEvent *event = GST_EVENT_CAST (data); |
| GstEventType type = GST_EVENT_TYPE (event); |
| |
| if (type == GST_EVENT_TAG) { |
| if (queue->use_tags_bitrate) { |
| GstTagList *tags; |
| guint bitrate; |
| |
| gst_event_parse_tag (event, &tags); |
| if (gst_tag_list_get_uint (tags, GST_TAG_BITRATE, &bitrate) || |
| gst_tag_list_get_uint (tags, GST_TAG_NOMINAL_BITRATE, &bitrate)) { |
| GST_QUEUE2_MUTEX_LOCK (queue); |
| queue->src_tags_bitrate = bitrate; |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| GST_LOG_OBJECT (queue, "src pad bitrate from tags now %u", bitrate); |
| } |
| } |
| } |
| |
| gst_pad_push_event (queue->srcpad, event); |
| |
| /* if we're EOS, return EOS so that the task pauses. */ |
| if (type == GST_EVENT_EOS) { |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, |
| "pushed EOS event %p, return EOS", event); |
| result = GST_FLOW_EOS; |
| } |
| |
| GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->srcresult, out_flushing); |
| } else if (item_type == GST_QUEUE2_ITEM_TYPE_BUFFER_LIST) { |
| GstBufferList *buffer_list; |
| |
| buffer_list = GST_BUFFER_LIST_CAST (data); |
| |
| result = gst_pad_push_list (queue->srcpad, buffer_list); |
| g_atomic_int_set (&queue->downstream_may_block, 0); |
| |
| /* need to check for srcresult here as well */ |
| GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->srcresult, out_flushing); |
| if (result == GST_FLOW_EOS) { |
| data = gst_queue2_dequeue_on_eos (queue, &item_type); |
| if (data != NULL) |
| goto next; |
| /* Since we will still accept EOS and SEGMENT we return _FLOW_OK |
| * to the caller so that the task function does not shut down */ |
| result = GST_FLOW_OK; |
| } |
| } else if (item_type == GST_QUEUE2_ITEM_TYPE_QUERY) { |
| GstQuery *query = GST_QUERY_CAST (data); |
| |
| GST_LOG_OBJECT (queue->srcpad, "Peering query %p", query); |
| queue->last_query = gst_pad_peer_query (queue->srcpad, query); |
| GST_LOG_OBJECT (queue->srcpad, "Peered query"); |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, |
| "did query %p, return %d", query, queue->last_query); |
| g_cond_signal (&queue->query_handled); |
| GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->srcresult, out_flushing); |
| result = GST_FLOW_OK; |
| } |
| return result; |
| |
| /* ERRORS */ |
| no_item: |
| { |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, |
| "exit because we have no item in the queue"); |
| return GST_FLOW_ERROR; |
| } |
| out_flushing: |
| { |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, "exit because we are flushing"); |
| return GST_FLOW_FLUSHING; |
| } |
| } |
| |
| /* called repeatedly with @pad as the source pad. This function should push out |
| * data to the peer element. */ |
| static void |
| gst_queue2_loop (GstPad * pad) |
| { |
| GstQueue2 *queue; |
| GstFlowReturn ret; |
| |
| queue = GST_QUEUE2 (GST_PAD_PARENT (pad)); |
| |
| /* have to lock for thread-safety */ |
| GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->srcresult, out_flushing); |
| |
| if (gst_queue2_is_empty (queue)) { |
| gboolean started; |
| |
| /* pause the timer while we wait. The fact that we are waiting does not mean |
| * the byterate on the output pad is lower */ |
| if ((started = queue->out_timer_started)) |
| g_timer_stop (queue->out_timer); |
| |
| GST_CAT_DEBUG_OBJECT (queue_dataflow, queue, |
| "queue is empty, waiting for new data"); |
| do { |
| /* Wait for data to be available, we could be unlocked because of a flush. */ |
| GST_QUEUE2_WAIT_ADD_CHECK (queue, queue->srcresult, out_flushing); |
| } |
| while (gst_queue2_is_empty (queue)); |
| |
| /* and continue if we were running before */ |
| if (started) |
| g_timer_continue (queue->out_timer); |
| } |
| ret = gst_queue2_push_one (queue); |
| queue->srcresult = ret; |
| queue->sinkresult = ret; |
| if (ret != GST_FLOW_OK) |
| goto out_flushing; |
| |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| gst_queue2_post_buffering (queue); |
| |
| return; |
| |
| /* ERRORS */ |
| out_flushing: |
| { |
| gboolean eos = queue->is_eos; |
| GstFlowReturn ret = queue->srcresult; |
| |
| gst_pad_pause_task (queue->srcpad); |
| if (ret == GST_FLOW_FLUSHING) { |
| gst_queue2_locked_flush (queue, FALSE, FALSE); |
| } else { |
| GST_QUEUE2_SIGNAL_DEL (queue); |
| queue->last_query = FALSE; |
| g_cond_signal (&queue->query_handled); |
| } |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| GST_CAT_LOG_OBJECT (queue_dataflow, queue, |
| "pause task, reason: %s", gst_flow_get_name (queue->srcresult)); |
| /* let app know about us giving up if upstream is not expected to do so */ |
| /* EOS is already taken care of elsewhere */ |
| if (eos && (ret == GST_FLOW_NOT_LINKED || ret < GST_FLOW_EOS)) { |
| GST_ELEMENT_ERROR (queue, STREAM, FAILED, |
| (_("Internal data flow error.")), |
| ("streaming task paused, reason %s (%d)", |
| gst_flow_get_name (ret), ret)); |
| gst_pad_push_event (queue->srcpad, gst_event_new_eos ()); |
| } |
| return; |
| } |
| } |
| |
| static gboolean |
| gst_queue2_handle_src_event (GstPad * pad, GstObject * parent, GstEvent * event) |
| { |
| gboolean res = TRUE; |
| GstQueue2 *queue = GST_QUEUE2 (parent); |
| |
| #ifndef GST_DISABLE_GST_DEBUG |
| GST_CAT_DEBUG_OBJECT (queue_dataflow, queue, "got event %p (%s)", |
| event, GST_EVENT_TYPE_NAME (event)); |
| #endif |
| |
| switch (GST_EVENT_TYPE (event)) { |
| case GST_EVENT_FLUSH_START: |
| if (QUEUE_IS_USING_QUEUE (queue)) { |
| /* just forward upstream */ |
| res = gst_pad_push_event (queue->sinkpad, event); |
| } else { |
| /* now unblock the getrange function */ |
| GST_QUEUE2_MUTEX_LOCK (queue); |
| GST_DEBUG_OBJECT (queue, "flushing"); |
| queue->srcresult = GST_FLOW_FLUSHING; |
| GST_QUEUE2_SIGNAL_ADD (queue); |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| |
| /* when using a temp file, we eat the event */ |
| res = TRUE; |
| gst_event_unref (event); |
| } |
| break; |
| case GST_EVENT_FLUSH_STOP: |
| if (QUEUE_IS_USING_QUEUE (queue)) { |
| /* just forward upstream */ |
| res = gst_pad_push_event (queue->sinkpad, event); |
| } else { |
| /* now unblock the getrange function */ |
| GST_QUEUE2_MUTEX_LOCK (queue); |
| queue->srcresult = GST_FLOW_OK; |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| |
| /* when using a temp file, we eat the event */ |
| res = TRUE; |
| gst_event_unref (event); |
| } |
| break; |
| case GST_EVENT_RECONFIGURE: |
| GST_QUEUE2_MUTEX_LOCK (queue); |
| /* assume downstream is linked now and try to push again */ |
| if (queue->srcresult == GST_FLOW_NOT_LINKED) { |
| queue->srcresult = GST_FLOW_OK; |
| queue->sinkresult = GST_FLOW_OK; |
| if (GST_PAD_MODE (pad) == GST_PAD_MODE_PUSH) { |
| gst_pad_start_task (pad, (GstTaskFunction) gst_queue2_loop, pad, |
| NULL); |
| } |
| } |
| GST_QUEUE2_MUTEX_UNLOCK (queue); |
| |
| res = gst_pad_push_event (queue->sinkpad, event); |
| break; |
| default: |
| res = gst_pad_push_event (queue->sinkpad, event); |
| break; |
| } |
| |
| return res; |
| } |
| |
| static gboolean |
| gst_queue2_handle_src_query (GstPad * pad, GstObject * parent, GstQuery * query) |
| { |
| GstQueue2 *queue; |
| |
| queue = GST_QUEUE2 (parent); |
| |
| switch (GST_QUERY_TYPE (query)) { |
| case GST_QUERY_POSITION: |
| { |
| gint64 p
|