plugins/elements/gstqueue2.c

   1 /* GStreamer
   2  * Copyright (C) 1999,2000 Erik Walthinsen <omega@cse.ogi.edu>
   3  *                    2003 Colin Walters <cwalters@gnome.org>
   4  *                    2000,2005,2007 Wim Taymans <wim.taymans@gmail.com>
   5  *                    2007 Thiago Sousa Santos <thiagoss@lcc.ufcg.edu.br>
   6  *                 SA 2010 ST-Ericsson <benjamin.gaignard@stericsson.com>
   7  *
   8  * gstqueue2.c:
   9  *
  10  * This library is free software; you can redistribute it and/or
  11  * modify it under the terms of the GNU Library General Public
  12  * License as published by the Free Software Foundation; either
  13  * version 2 of the License, or (at your option) any later version.
  14  *
  15  * This library is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  18  * Library General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU Library General Public
  21  * License along with this library; if not, write to the
  22  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
  23  * Boston, MA 02110-1301, USA.
  24  */
  25
  26 /**
  27  * SECTION:element-queue2
  28  *
  29  * Data is queued until one of the limits specified by the
  30  * #GstQueue2:max-size-buffers, #GstQueue2:max-size-bytes and/or
  31  * #GstQueue2:max-size-time properties has been reached. Any attempt to push
  32  * more buffers into the queue will block the pushing thread until more space
  33  * becomes available.
  34  *
  35  * The queue will create a new thread on the source pad to decouple the
  36  * processing on sink and source pad.
  37  *
  38  * You can query how many buffers are queued by reading the
  39  * #GstQueue2:current-level-buffers property.
  40  *
  41  * The default queue size limits are 100 buffers, 2MB of data, or
  42  * two seconds worth of data, whichever is reached first.
  43  *
  44  * If you set temp-template to a value such as /tmp/gstreamer-XXXXXX, the element
  45  * will allocate a random free filename and buffer data in the file.
  46  * By using this, it will buffer the entire stream data on the file independently
  47  * of the queue size limits, they will only be used for buffering statistics.
  48  *
  49  * The temp-location property will be used to notify the application of the
  50  * allocated filename.
  51  */
  52
  53 #ifdef HAVE_CONFIG_H
  54 #include "config.h"
  55 #endif
  56
  57 #include "gstqueue2.h"
  58
  59 #include <glib/gstdio.h>
  60
  61 #include "gst/gst-i18n-lib.h"
  62 #include "gst/glib-compat-private.h"
  63
  64 #include <string.h>
  65
  66 #ifdef G_OS_WIN32
  67 #include <io.h>                 /* lseek, open, close, read */
  68 #undef lseek
  69 #define lseek _lseeki64
  70 #undef off_t
  71 #define off_t guint64
  72 #else
  73 #include <unistd.h>
  74 #endif
  75
  76 #ifdef __BIONIC__               /* Android */
  77 #undef lseek
  78 #define lseek lseek64
  79 #undef off_t
  80 #define off_t guint64
  81 #include <fcntl.h>
  82 #endif
  83
  84 static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
  85     GST_PAD_SINK,
  86     GST_PAD_ALWAYS,
  87     GST_STATIC_CAPS_ANY);
  88
  89 static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",
  90     GST_PAD_SRC,
  91     GST_PAD_ALWAYS,
  92     GST_STATIC_CAPS_ANY);
  93
  94 GST_DEBUG_CATEGORY_STATIC (queue_debug);
  95 #define GST_CAT_DEFAULT (queue_debug)
  96 GST_DEBUG_CATEGORY_STATIC (queue_dataflow);
  97
  98 enum
  99 {
 100   LAST_SIGNAL
 101 };
 102
 103 /* other defines */
 104 #define DEFAULT_BUFFER_SIZE 4096
 105 #define QUEUE_IS_USING_TEMP_FILE(queue) ((queue)->temp_template != NULL)
 106 #define QUEUE_IS_USING_RING_BUFFER(queue) ((queue)->ring_buffer_max_size != 0)  /* for consistency with the above macro */
 107 #define QUEUE_IS_USING_QUEUE(queue) (!QUEUE_IS_USING_TEMP_FILE(queue) && !QUEUE_IS_USING_RING_BUFFER (queue))
 108
 109 #define QUEUE_MAX_BYTES(queue) MIN((queue)->max_level.bytes, (queue)->ring_buffer_max_size)
 110
 111 /* default property values */
 112 #define DEFAULT_MAX_SIZE_BUFFERS   100  /* 100 buffers */
 113 #define DEFAULT_MAX_SIZE_BYTES     (2 * 1024 * 1024)    /* 2 MB */
 114 #define DEFAULT_MAX_SIZE_TIME      2 * GST_SECOND       /* 2 seconds */
 115 #define DEFAULT_USE_BUFFERING      FALSE
 116 #define DEFAULT_USE_TAGS_BITRATE   FALSE
 117 #define DEFAULT_USE_RATE_ESTIMATE  TRUE
 118 #define DEFAULT_LOW_PERCENT        10
 119 #define DEFAULT_HIGH_PERCENT       99
 120 #define DEFAULT_TEMP_REMOVE        TRUE
 121 #define DEFAULT_RING_BUFFER_MAX_SIZE 0
 122
 123 enum
 124 {
 125   PROP_0,
 126   PROP_CUR_LEVEL_BUFFERS,
 127   PROP_CUR_LEVEL_BYTES,
 128   PROP_CUR_LEVEL_TIME,
 129   PROP_MAX_SIZE_BUFFERS,
 130   PROP_MAX_SIZE_BYTES,
 131   PROP_MAX_SIZE_TIME,
 132   PROP_USE_BUFFERING,
 133   PROP_USE_TAGS_BITRATE,
 134   PROP_USE_RATE_ESTIMATE,
 135   PROP_LOW_PERCENT,
 136   PROP_HIGH_PERCENT,
 137   PROP_TEMP_TEMPLATE,
 138   PROP_TEMP_LOCATION,
 139   PROP_TEMP_REMOVE,
 140   PROP_RING_BUFFER_MAX_SIZE,
 141   PROP_AVG_IN_RATE,
 142   PROP_LAST
 143 };
 144
 145 #define GST_QUEUE2_CLEAR_LEVEL(l) G_STMT_START {         \
 146   l.buffers = 0;                                        \
 147   l.bytes = 0;                                          \
 148   l.time = 0;                                           \
 149   l.rate_time = 0;                                      \
 150 } G_STMT_END
 151
 152 #define STATUS(queue, pad, msg) \
 153   GST_CAT_LOG_OBJECT (queue_dataflow, queue, \
 154                       "(%s:%s) " msg ": %u of %u buffers, %u of %u " \
 155                       "bytes, %" G_GUINT64_FORMAT " of %" G_GUINT64_FORMAT \
 156                       " ns, %"G_GUINT64_FORMAT" items", \
 157                       GST_DEBUG_PAD_NAME (pad), \
 158                       queue->cur_level.buffers, \
 159                       queue->max_level.buffers, \
 160                       queue->cur_level.bytes, \
 161                       queue->max_level.bytes, \
 162                       queue->cur_level.time, \
 163                       queue->max_level.time, \
 164                       (guint64) (!QUEUE_IS_USING_QUEUE(queue) ? \
 165                         queue->current->writing_pos - queue->current->max_reading_pos : \
 166                         queue->queue.length))
 167
 168 #define GST_QUEUE2_MUTEX_LOCK(q) G_STMT_START {                          \
 169   g_mutex_lock (&q->qlock);                                              \
 170 } G_STMT_END
 171
 172 #define GST_QUEUE2_MUTEX_LOCK_CHECK(q,res,label) G_STMT_START {         \
 173   GST_QUEUE2_MUTEX_LOCK (q);                                            \
 174   if (res != GST_FLOW_OK)                                               \
 175     goto label;                                                         \
 176 } G_STMT_END
 177
 178 #define GST_QUEUE2_MUTEX_UNLOCK(q) G_STMT_START {                        \
 179   g_mutex_unlock (&q->qlock);                                            \
 180 } G_STMT_END
 181
 182 #define GST_QUEUE2_WAIT_DEL_CHECK(q, res, label) G_STMT_START {         \
 183   STATUS (queue, q->sinkpad, "wait for DEL");                           \
 184   q->waiting_del = TRUE;                                                \
 185   g_cond_wait (&q->item_del, &queue->qlock);                              \
 186   q->waiting_del = FALSE;                                               \
 187   if (res != GST_FLOW_OK) {                                             \
 188     STATUS (queue, q->srcpad, "received DEL wakeup");                   \
 189     goto label;                                                         \
 190   }                                                                     \
 191   STATUS (queue, q->sinkpad, "received DEL");                           \
 192 } G_STMT_END
 193
 194 #define GST_QUEUE2_WAIT_ADD_CHECK(q, res, label) G_STMT_START {         \
 195   STATUS (queue, q->srcpad, "wait for ADD");                            \
 196   q->waiting_add = TRUE;                                                \
 197   g_cond_wait (&q->item_add, &q->qlock);                                  \
 198   q->waiting_add = FALSE;                                               \
 199   if (res != GST_FLOW_OK) {                                             \
 200     STATUS (queue, q->srcpad, "received ADD wakeup");                   \
 201     goto label;                                                         \
 202   }                                                                     \
 203   STATUS (queue, q->srcpad, "received ADD");                            \
 204 } G_STMT_END
 205
 206 #define GST_QUEUE2_SIGNAL_DEL(q) G_STMT_START {                          \
 207   if (q->waiting_del) {                                                 \
 208     STATUS (q, q->srcpad, "signal DEL");                                \
 209     g_cond_signal (&q->item_del);                                        \
 210   }                                                                     \
 211 } G_STMT_END
 212
 213 #define GST_QUEUE2_SIGNAL_ADD(q) G_STMT_START {                          \
 214   if (q->waiting_add) {                                                 \
 215     STATUS (q, q->sinkpad, "signal ADD");                               \
 216     g_cond_signal (&q->item_add);                                        \
 217   }                                                                     \
 218 } G_STMT_END
 219
 220 #define SET_PERCENT(q, perc) G_STMT_START {                              \
 221   if (perc != q->buffering_percent) {                                    \
 222     q->buffering_percent = perc;                                         \
 223     q->percent_changed = TRUE;                                           \
 224     GST_DEBUG_OBJECT (q, "buffering %d percent", perc);                  \
 225     get_buffering_stats (q, perc, &q->mode, &q->avg_in, &q->avg_out,     \
 226         &q->buffering_left);                                             \
 227   }                                                                      \
 228 } G_STMT_END
 229
 230 #define _do_init \
 231     GST_DEBUG_CATEGORY_INIT (queue_debug, "queue2", 0, "queue element"); \
 232     GST_DEBUG_CATEGORY_INIT (queue_dataflow, "queue2_dataflow", 0, \
 233         "dataflow inside the queue element");
 234 #define gst_queue2_parent_class parent_class
 235 G_DEFINE_TYPE_WITH_CODE (GstQueue2, gst_queue2, GST_TYPE_ELEMENT, _do_init);
 236
 237 static void gst_queue2_finalize (GObject * object);
 238
 239 static void gst_queue2_set_property (GObject * object,
 240     guint prop_id, const GValue * value, GParamSpec * pspec);
 241 static void gst_queue2_get_property (GObject * object,
 242     guint prop_id, GValue * value, GParamSpec * pspec);
 243
 244 static GstFlowReturn gst_queue2_chain (GstPad * pad, GstObject * parent,
 245     GstBuffer * buffer);
 246 static GstFlowReturn gst_queue2_chain_list (GstPad * pad, GstObject * parent,
 247     GstBufferList * buffer_list);
 248 static GstFlowReturn gst_queue2_push_one (GstQueue2 * queue);
 249 static void gst_queue2_loop (GstPad * pad);
 250
 251 static gboolean gst_queue2_handle_sink_event (GstPad * pad, GstObject * parent,
 252     GstEvent * event);
 253 static gboolean gst_queue2_handle_sink_query (GstPad * pad, GstObject * parent,
 254     GstQuery * query);
 255
 256 static gboolean gst_queue2_handle_src_event (GstPad * pad, GstObject * parent,
 257     GstEvent * event);
 258 static gboolean gst_queue2_handle_src_query (GstPad * pad, GstObject * parent,
 259     GstQuery * query);
 260 static gboolean gst_queue2_handle_query (GstElement * element,
 261     GstQuery * query);
 262
 263 static GstFlowReturn gst_queue2_get_range (GstPad * pad, GstObject * parent,
 264     guint64 offset, guint length, GstBuffer ** buffer);
 265
 266 static gboolean gst_queue2_src_activate_mode (GstPad * pad, GstObject * parent,
 267     GstPadMode mode, gboolean active);
 268 static gboolean gst_queue2_sink_activate_mode (GstPad * pad, GstObject * parent,
 269     GstPadMode mode, gboolean active);
 270 static GstStateChangeReturn gst_queue2_change_state (GstElement * element,
 271     GstStateChange transition);
 272
 273 static gboolean gst_queue2_is_empty (GstQueue2 * queue);
 274 static gboolean gst_queue2_is_filled (GstQueue2 * queue);
 275
 276 static void update_cur_level (GstQueue2 * queue, GstQueue2Range * range);
 277 static void update_in_rates (GstQueue2 * queue);
 278 static void gst_queue2_post_buffering (GstQueue2 * queue);
 279
 280 typedef enum
 281 {
 282   GST_QUEUE2_ITEM_TYPE_UNKNOWN = 0,
 283   GST_QUEUE2_ITEM_TYPE_BUFFER,
 284   GST_QUEUE2_ITEM_TYPE_BUFFER_LIST,
 285   GST_QUEUE2_ITEM_TYPE_EVENT,
 286   GST_QUEUE2_ITEM_TYPE_QUERY
 287 } GstQueue2ItemType;
 288
 289 typedef struct
 290 {
 291   GstQueue2ItemType type;
 292   GstMiniObject *item;
 293 } GstQueue2Item;
 294
 295 /* static guint gst_queue2_signals[LAST_SIGNAL] = { 0 }; */
 296
 297 static void
 298 gst_queue2_class_init (GstQueue2Class * klass)
 299 {
 300   GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
 301   GstElementClass *gstelement_class = GST_ELEMENT_CLASS (klass);
 302
 303   gobject_class->set_property = gst_queue2_set_property;
 304   gobject_class->get_property = gst_queue2_get_property;
 305
 306   /* properties */
 307   g_object_class_install_property (gobject_class, PROP_CUR_LEVEL_BYTES,
 308       g_param_spec_uint ("current-level-bytes", "Current level (kB)",
 309           "Current amount of data in the queue (bytes)",
 310           0, G_MAXUINT, 0, G_PARAM_READABLE | G_PARAM_STATIC_STRINGS));
 311   g_object_class_install_property (gobject_class, PROP_CUR_LEVEL_BUFFERS,
 312       g_param_spec_uint ("current-level-buffers", "Current level (buffers)",
 313           "Current number of buffers in the queue",
 314           0, G_MAXUINT, 0, G_PARAM_READABLE | G_PARAM_STATIC_STRINGS));
 315   g_object_class_install_property (gobject_class, PROP_CUR_LEVEL_TIME,
 316       g_param_spec_uint64 ("current-level-time", "Current level (ns)",
 317           "Current amount of data in the queue (in ns)",
 318           0, G_MAXUINT64, 0, G_PARAM_READABLE | G_PARAM_STATIC_STRINGS));
 319
 320   g_object_class_install_property (gobject_class, PROP_MAX_SIZE_BYTES,
 321       g_param_spec_uint ("max-size-bytes", "Max. size (kB)",
 322           "Max. amount of data in the queue (bytes, 0=disable)",
 323           0, G_MAXUINT, DEFAULT_MAX_SIZE_BYTES,
 324           G_PARAM_READWRITE | GST_PARAM_MUTABLE_PLAYING |
 325           G_PARAM_STATIC_STRINGS));
 326   g_object_class_install_property (gobject_class, PROP_MAX_SIZE_BUFFERS,
 327       g_param_spec_uint ("max-size-buffers", "Max. size (buffers)",
 328           "Max. number of buffers in the queue (0=disable)", 0, G_MAXUINT,
 329           DEFAULT_MAX_SIZE_BUFFERS,
 330           G_PARAM_READWRITE | GST_PARAM_MUTABLE_PLAYING |
 331           G_PARAM_STATIC_STRINGS));
 332   g_object_class_install_property (gobject_class, PROP_MAX_SIZE_TIME,
 333       g_param_spec_uint64 ("max-size-time", "Max. size (ns)",
 334           "Max. amount of data in the queue (in ns, 0=disable)", 0, G_MAXUINT64,
 335           DEFAULT_MAX_SIZE_TIME, G_PARAM_READWRITE | GST_PARAM_MUTABLE_PLAYING |
 336           G_PARAM_STATIC_STRINGS));
 337
 338   g_object_class_install_property (gobject_class, PROP_USE_BUFFERING,
 339       g_param_spec_boolean ("use-buffering", "Use buffering",
 340           "Emit GST_MESSAGE_BUFFERING based on low-/high-percent thresholds",
 341           DEFAULT_USE_BUFFERING, G_PARAM_READWRITE | GST_PARAM_MUTABLE_PLAYING |
 342           G_PARAM_STATIC_STRINGS));
 343   g_object_class_install_property (gobject_class, PROP_USE_TAGS_BITRATE,
 344       g_param_spec_boolean ("use-tags-bitrate", "Use bitrate from tags",
 345           "Use a bitrate from upstream tags to estimate buffer duration if not provided",
 346           DEFAULT_USE_TAGS_BITRATE,
 347           G_PARAM_READWRITE | GST_PARAM_MUTABLE_PLAYING |
 348           G_PARAM_STATIC_STRINGS));
 349   g_object_class_install_property (gobject_class, PROP_USE_RATE_ESTIMATE,
 350       g_param_spec_boolean ("use-rate-estimate", "Use Rate Estimate",
 351           "Estimate the bitrate of the stream to calculate time level",
 352           DEFAULT_USE_RATE_ESTIMATE,
 353           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 354   g_object_class_install_property (gobject_class, PROP_LOW_PERCENT,
 355       g_param_spec_int ("low-percent", "Low percent",
 356           "Low threshold for buffering to start. Only used if use-buffering is True",
 357           0, 100, DEFAULT_LOW_PERCENT,
 358           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 359   g_object_class_install_property (gobject_class, PROP_HIGH_PERCENT,
 360       g_param_spec_int ("high-percent", "High percent",
 361           "High threshold for buffering to finish. Only used if use-buffering is True",
 362           0, 100, DEFAULT_HIGH_PERCENT,
 363           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 364
 365   g_object_class_install_property (gobject_class, PROP_TEMP_TEMPLATE,
 366       g_param_spec_string ("temp-template", "Temporary File Template",
 367           "File template to store temporary files in, should contain directory "
 368           "and XXXXXX. (NULL == disabled)",
 369           NULL, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 370
 371   g_object_class_install_property (gobject_class, PROP_TEMP_LOCATION,
 372       g_param_spec_string ("temp-location", "Temporary File Location",
 373           "Location to store temporary files in (Only read this property, "
 374           "use temp-template to configure the name template)",
 375           NULL, G_PARAM_READABLE | G_PARAM_STATIC_STRINGS));
 376
 377   /**
 378    * GstQueue2:temp-remove
 379    *
 380    * When temp-template is set, remove the temporary file when going to READY.
 381    */
 382   g_object_class_install_property (gobject_class, PROP_TEMP_REMOVE,
 383       g_param_spec_boolean ("temp-remove", "Remove the Temporary File",
 384           "Remove the temp-location after use",
 385           DEFAULT_TEMP_REMOVE, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 386
 387   /**
 388    * GstQueue2:ring-buffer-max-size
 389    *
 390    * The maximum size of the ring buffer in bytes. If set to 0, the ring
 391    * buffer is disabled. Default 0.
 392    */
 393   g_object_class_install_property (gobject_class, PROP_RING_BUFFER_MAX_SIZE,
 394       g_param_spec_uint64 ("ring-buffer-max-size",
 395           "Max. ring buffer size (bytes)",
 396           "Max. amount of data in the ring buffer (bytes, 0 = disabled)",
 397           0, G_MAXUINT64, DEFAULT_RING_BUFFER_MAX_SIZE,
 398           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 399
 400   /**
 401    * GstQueue2:avg-in-rate
 402    *
 403    * The average input data rate.
 404    */
 405   g_object_class_install_property (gobject_class, PROP_AVG_IN_RATE,
 406       g_param_spec_int64 ("avg-in-rate", "Input data rate (bytes/s)",
 407           "Average input data rate (bytes/s)",
 408           0, G_MAXINT64, 0, G_PARAM_READABLE | G_PARAM_STATIC_STRINGS));
 409
 410   /* set several parent class virtual functions */
 411   gobject_class->finalize = gst_queue2_finalize;
 412
 413   gst_element_class_add_pad_template (gstelement_class,
 414       gst_static_pad_template_get (&srctemplate));
 415   gst_element_class_add_pad_template (gstelement_class,
 416       gst_static_pad_template_get (&sinktemplate));
 417
 418   gst_element_class_set_static_metadata (gstelement_class, "Queue 2",
 419       "Generic",
 420       "Simple data queue",
 421       "Erik Walthinsen <omega@cse.ogi.edu>, "
 422       "Wim Taymans <wim.taymans@gmail.com>");
 423
 424   gstelement_class->change_state = GST_DEBUG_FUNCPTR (gst_queue2_change_state);
 425   gstelement_class->query = GST_DEBUG_FUNCPTR (gst_queue2_handle_query);
 426 }
 427
 428 static void
 429 gst_queue2_init (GstQueue2 * queue)
 430 {
 431   queue->sinkpad = gst_pad_new_from_static_template (&sinktemplate, "sink");
 432
 433   gst_pad_set_chain_function (queue->sinkpad,
 434       GST_DEBUG_FUNCPTR (gst_queue2_chain));
 435   gst_pad_set_chain_list_function (queue->sinkpad,
 436       GST_DEBUG_FUNCPTR (gst_queue2_chain_list));
 437   gst_pad_set_activatemode_function (queue->sinkpad,
 438       GST_DEBUG_FUNCPTR (gst_queue2_sink_activate_mode));
 439   gst_pad_set_event_function (queue->sinkpad,
 440       GST_DEBUG_FUNCPTR (gst_queue2_handle_sink_event));
 441   gst_pad_set_query_function (queue->sinkpad,
 442       GST_DEBUG_FUNCPTR (gst_queue2_handle_sink_query));
 443   GST_PAD_SET_PROXY_CAPS (queue->sinkpad);
 444   gst_element_add_pad (GST_ELEMENT (queue), queue->sinkpad);
 445
 446   queue->srcpad = gst_pad_new_from_static_template (&srctemplate, "src");
 447
 448   gst_pad_set_activatemode_function (queue->srcpad,
 449       GST_DEBUG_FUNCPTR (gst_queue2_src_activate_mode));
 450   gst_pad_set_getrange_function (queue->srcpad,
 451       GST_DEBUG_FUNCPTR (gst_queue2_get_range));
 452   gst_pad_set_event_function (queue->srcpad,
 453       GST_DEBUG_FUNCPTR (gst_queue2_handle_src_event));
 454   gst_pad_set_query_function (queue->srcpad,
 455       GST_DEBUG_FUNCPTR (gst_queue2_handle_src_query));
 456   GST_PAD_SET_PROXY_CAPS (queue->srcpad);
 457   gst_element_add_pad (GST_ELEMENT (queue), queue->srcpad);
 458
 459   /* levels */
 460   GST_QUEUE2_CLEAR_LEVEL (queue->cur_level);
 461   queue->max_level.buffers = DEFAULT_MAX_SIZE_BUFFERS;
 462   queue->max_level.bytes = DEFAULT_MAX_SIZE_BYTES;
 463   queue->max_level.time = DEFAULT_MAX_SIZE_TIME;
 464   queue->max_level.rate_time = DEFAULT_MAX_SIZE_TIME;
 465   queue->use_buffering = DEFAULT_USE_BUFFERING;
 466   queue->use_rate_estimate = DEFAULT_USE_RATE_ESTIMATE;
 467   queue->low_percent = DEFAULT_LOW_PERCENT;
 468   queue->high_percent = DEFAULT_HIGH_PERCENT;
 469
 470   gst_segment_init (&queue->sink_segment, GST_FORMAT_TIME);
 471   gst_segment_init (&queue->src_segment, GST_FORMAT_TIME);
 472
 473   queue->sinktime = GST_CLOCK_TIME_NONE;
 474   queue->srctime = GST_CLOCK_TIME_NONE;
 475   queue->sink_tainted = TRUE;
 476   queue->src_tainted = TRUE;
 477
 478   queue->srcresult = GST_FLOW_FLUSHING;
 479   queue->sinkresult = GST_FLOW_FLUSHING;
 480   queue->is_eos = FALSE;
 481   queue->in_timer = g_timer_new ();
 482   queue->out_timer = g_timer_new ();
 483
 484   g_mutex_init (&queue->qlock);
 485   queue->waiting_add = FALSE;
 486   g_cond_init (&queue->item_add);
 487   queue->waiting_del = FALSE;
 488   g_cond_init (&queue->item_del);
 489   g_queue_init (&queue->queue);
 490
 491   g_cond_init (&queue->query_handled);
 492   queue->last_query = FALSE;
 493
 494   g_mutex_init (&queue->buffering_post_lock);
 495   queue->buffering_percent = 100;
 496
 497   /* tempfile related */
 498   queue->temp_template = NULL;
 499   queue->temp_location = NULL;
 500   queue->temp_remove = DEFAULT_TEMP_REMOVE;
 501
 502   queue->ring_buffer = NULL;
 503   queue->ring_buffer_max_size = DEFAULT_RING_BUFFER_MAX_SIZE;
 504
 505   GST_DEBUG_OBJECT (queue,
 506       "initialized queue's not_empty & not_full conditions");
 507 }
 508
 509 /* called only once, as opposed to dispose */
 510 static void
 511 gst_queue2_finalize (GObject * object)
 512 {
 513   GstQueue2 *queue = GST_QUEUE2 (object);
 514
 515   GST_DEBUG_OBJECT (queue, "finalizing queue");
 516
 517   while (!g_queue_is_empty (&queue->queue)) {
 518     GstQueue2Item *qitem = g_queue_pop_head (&queue->queue);
 519
 520     if (qitem->type != GST_QUEUE2_ITEM_TYPE_QUERY)
 521       gst_mini_object_unref (qitem->item);
 522     g_slice_free (GstQueue2Item, qitem);
 523   }
 524
 525   queue->last_query = FALSE;
 526   g_queue_clear (&queue->queue);
 527   g_mutex_clear (&queue->qlock);
 528   g_mutex_clear (&queue->buffering_post_lock);
 529   g_cond_clear (&queue->item_add);
 530   g_cond_clear (&queue->item_del);
 531   g_cond_clear (&queue->query_handled);
 532   g_timer_destroy (queue->in_timer);
 533   g_timer_destroy (queue->out_timer);
 534
 535   /* temp_file path cleanup  */
 536   g_free (queue->temp_template);
 537   g_free (queue->temp_location);
 538
 539   G_OBJECT_CLASS (parent_class)->finalize (object);
 540 }
 541
 542 static void
 543 debug_ranges (GstQueue2 * queue)
 544 {
 545   GstQueue2Range *walk;
 546
 547   for (walk = queue->ranges; walk; walk = walk->next) {
 548     GST_DEBUG_OBJECT (queue,
 549         "range [%" G_GUINT64_FORMAT "-%" G_GUINT64_FORMAT "] (rb [%"
 550         G_GUINT64_FORMAT "-%" G_GUINT64_FORMAT "]), reading %" G_GUINT64_FORMAT
 551         " current range? %s", walk->offset, walk->writing_pos, walk->rb_offset,
 552         walk->rb_writing_pos, walk->reading_pos,
 553         walk == queue->current ? "**y**" : "  n  ");
 554   }
 555 }
 556
 557 /* clear all the downloaded ranges */
 558 static void
 559 clean_ranges (GstQueue2 * queue)
 560 {
 561   GST_DEBUG_OBJECT (queue, "clean queue ranges");
 562
 563   g_slice_free_chain (GstQueue2Range, queue->ranges, next);
 564   queue->ranges = NULL;
 565   queue->current = NULL;
 566 }
 567
 568 /* find a range that contains @offset or NULL when nothing does */
 569 static GstQueue2Range *
 570 find_range (GstQueue2 * queue, guint64 offset)
 571 {
 572   GstQueue2Range *range = NULL;
 573   GstQueue2Range *walk;
 574
 575   /* first do a quick check for the current range */
 576   for (walk = queue->ranges; walk; walk = walk->next) {
 577     if (offset >= walk->offset && offset <= walk->writing_pos) {
 578       /* we can reuse an existing range */
 579       range = walk;
 580       break;
 581     }
 582   }
 583   if (range) {
 584     GST_DEBUG_OBJECT (queue,
 585         "found range for %" G_GUINT64_FORMAT ": [%" G_GUINT64_FORMAT "-%"
 586         G_GUINT64_FORMAT "]", offset, range->offset, range->writing_pos);
 587   } else {
 588     GST_DEBUG_OBJECT (queue, "no range for %" G_GUINT64_FORMAT, offset);
 589   }
 590   return range;
 591 }
 592
 593 static void
 594 update_cur_level (GstQueue2 * queue, GstQueue2Range * range)
 595 {
 596   guint64 max_reading_pos, writing_pos;
 597
 598   writing_pos = range->writing_pos;
 599   max_reading_pos = range->max_reading_pos;
 600
 601   if (writing_pos > max_reading_pos)
 602     queue->cur_level.bytes = writing_pos - max_reading_pos;
 603   else
 604     queue->cur_level.bytes = 0;
 605 }
 606
 607 /* make a new range for @offset or reuse an existing range */
 608 static GstQueue2Range *
 609 add_range (GstQueue2 * queue, guint64 offset, gboolean update_existing)
 610 {
 611   GstQueue2Range *range, *prev, *next;
 612
 613   GST_DEBUG_OBJECT (queue, "find range for %" G_GUINT64_FORMAT, offset);
 614
 615   if ((range = find_range (queue, offset))) {
 616     GST_DEBUG_OBJECT (queue,
 617         "reusing range %" G_GUINT64_FORMAT "-%" G_GUINT64_FORMAT, range->offset,
 618         range->writing_pos);
 619     if (update_existing && range->writing_pos != offset) {
 620       GST_DEBUG_OBJECT (queue, "updating range writing position to "
 621           "%" G_GUINT64_FORMAT, offset);
 622       range->writing_pos = offset;
 623     }
 624   } else {
 625     GST_DEBUG_OBJECT (queue,
 626         "new range %" G_GUINT64_FORMAT "-%" G_GUINT64_FORMAT, offset, offset);
 627
 628     range = g_slice_new0 (GstQueue2Range);
 629     range->offset = offset;
 630     /* we want to write to the next location in the ring buffer */
 631     range->rb_offset = queue->current ? queue->current->rb_writing_pos : 0;
 632     range->writing_pos = offset;
 633     range->rb_writing_pos = range->rb_offset;
 634     range->reading_pos = offset;
 635     range->max_reading_pos = offset;
 636
 637     /* insert sorted */
 638     prev = NULL;
 639     next = queue->ranges;
 640     while (next) {
 641       if (next->offset > offset) {
 642         /* insert before next */
 643         GST_DEBUG_OBJECT (queue,
 644             "insert before range %p, offset %" G_GUINT64_FORMAT, next,
 645             next->offset);
 646         break;
 647       }
 648       /* try next */
 649       prev = next;
 650       next = next->next;
 651     }
 652     range->next = next;
 653     if (prev)
 654       prev->next = range;
 655     else
 656       queue->ranges = range;
 657   }
 658   debug_ranges (queue);
 659
 660   /* update the stats for this range */
 661   update_cur_level (queue, range);
 662
 663   return range;
 664 }
 665
 666
 667 /* clear and init the download ranges for offset 0 */
 668 static void
 669 init_ranges (GstQueue2 * queue)
 670 {
 671   GST_DEBUG_OBJECT (queue, "init queue ranges");
 672
 673   /* get rid of all the current ranges */
 674   clean_ranges (queue);
 675   /* make a range for offset 0 */
 676   queue->current = add_range (queue, 0, TRUE);
 677 }
 678
 679 /* calculate the diff between running time on the sink and src of the queue.
 680  * This is the total amount of time in the queue. */
 681 static void
 682 update_time_level (GstQueue2 * queue)
 683 {
 684   if (queue->sink_tainted) {
 685     queue->sinktime =
 686         gst_segment_to_running_time (&queue->sink_segment, GST_FORMAT_TIME,
 687         queue->sink_segment.position);
 688     queue->sink_tainted = FALSE;
 689   }
 690
 691   if (queue->src_tainted) {
 692     queue->srctime =
 693         gst_segment_to_running_time (&queue->src_segment, GST_FORMAT_TIME,
 694         queue->src_segment.position);
 695     queue->src_tainted = FALSE;
 696   }
 697
 698   GST_DEBUG_OBJECT (queue, "sink %" GST_TIME_FORMAT ", src %" GST_TIME_FORMAT,
 699       GST_TIME_ARGS (queue->sinktime), GST_TIME_ARGS (queue->srctime));
 700
 701   if (queue->sinktime != GST_CLOCK_TIME_NONE
 702       && queue->srctime != GST_CLOCK_TIME_NONE
 703       && queue->sinktime >= queue->srctime)
 704     queue->cur_level.time = queue->sinktime - queue->srctime;
 705   else
 706     queue->cur_level.time = 0;
 707 }
 708
 709 /* take a SEGMENT event and apply the values to segment, updating the time
 710  * level of queue. */
 711 static void
 712 apply_segment (GstQueue2 * queue, GstEvent * event, GstSegment * segment,
 713     gboolean is_sink)
 714 {
 715   gst_event_copy_segment (event, segment);
 716
 717   if (segment->format == GST_FORMAT_BYTES) {
 718     if (!QUEUE_IS_USING_QUEUE (queue) && is_sink) {
 719       /* start is where we'll be getting from and as such writing next */
 720       queue->current = add_range (queue, segment->start, TRUE);
 721     }
 722   }
 723
 724   /* now configure the values, we use these to track timestamps on the
 725    * sinkpad. */
 726   if (segment->format != GST_FORMAT_TIME) {
 727     /* non-time format, pretend the current time segment is closed with a
 728      * 0 start and unknown stop time. */
 729     segment->format = GST_FORMAT_TIME;
 730     segment->start = 0;
 731     segment->stop = -1;
 732     segment->time = 0;
 733   }
 734
 735   GST_DEBUG_OBJECT (queue, "configured SEGMENT %" GST_SEGMENT_FORMAT, segment);
 736
 737   if (is_sink)
 738     queue->sink_tainted = TRUE;
 739   else
 740     queue->src_tainted = TRUE;
 741
 742   /* segment can update the time level of the queue */
 743   update_time_level (queue);
 744 }
 745
 746 static void
 747 apply_gap (GstQueue2 * queue, GstEvent * event,
 748     GstSegment * segment, gboolean is_sink)
 749 {
 750   GstClockTime timestamp;
 751   GstClockTime duration;
 752
 753   gst_event_parse_gap (event, &timestamp, &duration);
 754
 755   if (GST_CLOCK_TIME_IS_VALID (timestamp)) {
 756
 757     if (GST_CLOCK_TIME_IS_VALID (duration)) {
 758       timestamp += duration;
 759     }
 760
 761     segment->position = timestamp;
 762
 763     if (is_sink)
 764       queue->sink_tainted = TRUE;
 765     else
 766       queue->src_tainted = TRUE;
 767
 768     /* calc diff with other end */
 769     update_time_level (queue);
 770   }
 771 }
 772
 773 /* take a buffer and update segment, updating the time level of the queue. */
 774 static void
 775 apply_buffer (GstQueue2 * queue, GstBuffer * buffer, GstSegment * segment,
 776     guint64 size, gboolean is_sink)
 777 {
 778   GstClockTime duration, timestamp;
 779
 780   timestamp = GST_BUFFER_DTS_OR_PTS (buffer);
 781   duration = GST_BUFFER_DURATION (buffer);
 782
 783   /* If we have no duration, pick one from the bitrate if we can */
 784   if (duration == GST_CLOCK_TIME_NONE && queue->use_tags_bitrate) {
 785     guint bitrate =
 786         is_sink ? queue->sink_tags_bitrate : queue->src_tags_bitrate;
 787     if (bitrate)
 788       duration = gst_util_uint64_scale (size, 8 * GST_SECOND, bitrate);
 789   }
 790
 791   /* if no timestamp is set, assume it's continuous with the previous
 792    * time */
 793   if (timestamp == GST_CLOCK_TIME_NONE)
 794     timestamp = segment->position;
 795
 796   /* add duration */
 797   if (duration != GST_CLOCK_TIME_NONE)
 798     timestamp += duration;
 799
 800   GST_DEBUG_OBJECT (queue, "position updated to %" GST_TIME_FORMAT,
 801       GST_TIME_ARGS (timestamp));
 802
 803   segment->position = timestamp;
 804
 805   if (is_sink)
 806     queue->sink_tainted = TRUE;
 807   else
 808     queue->src_tainted = TRUE;
 809
 810   /* calc diff with other end */
 811   update_time_level (queue);
 812 }
 813
 814 struct BufListData
 815 {
 816   GstClockTime timestamp;
 817   guint bitrate;
 818 };
 819
 820 static gboolean
 821 buffer_list_apply_time (GstBuffer ** buf, guint idx, gpointer data)
 822 {
 823   struct BufListData *bld = data;
 824   GstClockTime *timestamp = &bld->timestamp;
 825   GstClockTime btime;
 826
 827   GST_TRACE ("buffer %u has pts %" GST_TIME_FORMAT " dts %" GST_TIME_FORMAT
 828       " duration %" GST_TIME_FORMAT, idx,
 829       GST_TIME_ARGS (GST_BUFFER_PTS (*buf)),
 830       GST_TIME_ARGS (GST_BUFFER_DTS (*buf)),
 831       GST_TIME_ARGS (GST_BUFFER_DURATION (*buf)));
 832
 833   btime = GST_BUFFER_DTS_OR_PTS (*buf);
 834   if (GST_CLOCK_TIME_IS_VALID (btime))
 835     *timestamp = btime;
 836
 837   if (GST_BUFFER_DURATION_IS_VALID (*buf))
 838     *timestamp += GST_BUFFER_DURATION (*buf);
 839   else if (bld->bitrate != 0) {
 840     guint64 size = gst_buffer_get_size (*buf);
 841
 842     /* If we have no duration, pick one from the bitrate if we can */
 843     *timestamp += gst_util_uint64_scale (bld->bitrate, 8 * GST_SECOND, size);
 844   }
 845
 846
 847   GST_TRACE ("ts now %" GST_TIME_FORMAT, GST_TIME_ARGS (*timestamp));
 848   return TRUE;
 849 }
 850
 851 /* take a buffer list and update segment, updating the time level of the queue */
 852 static void
 853 apply_buffer_list (GstQueue2 * queue, GstBufferList * buffer_list,
 854     GstSegment * segment, gboolean is_sink)
 855 {
 856   struct BufListData bld;
 857
 858   /* if no timestamp is set, assume it's continuous with the previous time */
 859   bld.timestamp = segment->position;
 860
 861   if (queue->use_tags_bitrate) {
 862     if (is_sink)
 863       bld.bitrate = queue->sink_tags_bitrate;
 864     else
 865       bld.bitrate = queue->src_tags_bitrate;
 866   } else
 867     bld.bitrate = 0;
 868
 869   gst_buffer_list_foreach (buffer_list, buffer_list_apply_time, &bld);
 870
 871   GST_DEBUG_OBJECT (queue, "last_stop updated to %" GST_TIME_FORMAT,
 872       GST_TIME_ARGS (bld.timestamp));
 873
 874   segment->position = bld.timestamp;
 875
 876   if (is_sink)
 877     queue->sink_tainted = TRUE;
 878   else
 879     queue->src_tainted = TRUE;
 880
 881   /* calc diff with other end */
 882   update_time_level (queue);
 883 }
 884
 885 static inline gint
 886 get_percent (guint64 cur_level, guint64 max_level, guint64 alt_max)
 887 {
 888   guint64 p;
 889
 890   if (max_level == 0)
 891     return 0;
 892
 893   if (alt_max > 0)
 894     p = gst_util_uint64_scale (cur_level, 100, MIN (max_level, alt_max));
 895   else
 896     p = gst_util_uint64_scale (cur_level, 100, max_level);
 897
 898   return MIN (p, 100);
 899 }
 900
 901 static gboolean
 902 get_buffering_percent (GstQueue2 * queue, gboolean * is_buffering,
 903     gint * percent)
 904 {
 905   gint perc, perc2;
 906
 907   if (queue->high_percent <= 0) {
 908     if (percent)
 909       *percent = 100;
 910     if (is_buffering)
 911       *is_buffering = FALSE;
 912     return FALSE;
 913   }
 914 #define GET_PERCENT(format,alt_max) \
 915     get_percent(queue->cur_level.format,queue->max_level.format,(alt_max))
 916
 917   if (queue->is_eos) {
 918     /* on EOS we are always 100% full, we set the var here so that it we can
 919      * reuse the logic below to stop buffering */
 920     perc = 100;
 921     GST_LOG_OBJECT (queue, "we are EOS");
 922   } else {
 923     GST_LOG_OBJECT (queue,
 924         "Cur level bytes/time/buffers %u/%" GST_TIME_FORMAT "/%u",
 925         queue->cur_level.bytes, GST_TIME_ARGS (queue->cur_level.time),
 926         queue->cur_level.buffers);
 927
 928     /* figure out the percent we are filled, we take the max of all formats. */
 929     if (!QUEUE_IS_USING_RING_BUFFER (queue)) {
 930       perc = GET_PERCENT (bytes, 0);
 931     } else {
 932       guint64 rb_size = queue->ring_buffer_max_size;
 933       perc = GET_PERCENT (bytes, rb_size);
 934     }
 935
 936     perc2 = GET_PERCENT (time, 0);
 937     perc = MAX (perc, perc2);
 938
 939     perc2 = GET_PERCENT (buffers, 0);
 940     perc = MAX (perc, perc2);
 941
 942     /* also apply the rate estimate when we need to */
 943     if (queue->use_rate_estimate) {
 944       perc2 = GET_PERCENT (rate_time, 0);
 945       perc = MAX (perc, perc2);
 946     }
 947
 948     /* Don't get to 0% unless we're really empty */
 949     if (queue->cur_level.bytes > 0)
 950       perc = MAX (1, perc);
 951   }
 952 #undef GET_PERCENT
 953
 954   if (is_buffering)
 955     *is_buffering = queue->is_buffering;
 956
 957   /* scale to high percent so that it becomes the 100% mark */
 958   perc = perc * 100 / queue->high_percent;
 959   /* clip */
 960   if (perc > 100)
 961     perc = 100;
 962
 963   if (percent)
 964     *percent = perc;
 965
 966   GST_DEBUG_OBJECT (queue, "buffering %d, percent %d", queue->is_buffering,
 967       perc);
 968
 969   return TRUE;
 970 }
 971
 972 static void
 973 get_buffering_stats (GstQueue2 * queue, gint percent, GstBufferingMode * mode,
 974     gint * avg_in, gint * avg_out, gint64 * buffering_left)
 975 {
 976   if (mode) {
 977     if (!QUEUE_IS_USING_QUEUE (queue)) {
 978       if (QUEUE_IS_USING_RING_BUFFER (queue))
 979         *mode = GST_BUFFERING_TIMESHIFT;
 980       else
 981         *mode = GST_BUFFERING_DOWNLOAD;
 982     } else {
 983       *mode = GST_BUFFERING_STREAM;
 984     }
 985   }
 986
 987   if (avg_in)
 988     *avg_in = queue->byte_in_rate;
 989   if (avg_out)
 990     *avg_out = queue->byte_out_rate;
 991
 992   if (buffering_left) {
 993     *buffering_left = (percent == 100 ? 0 : -1);
 994
 995     if (queue->use_rate_estimate) {
 996       guint64 max, cur;
 997
 998       max = queue->max_level.rate_time;
 999       cur = queue->cur_level.rate_time;
1000
1001       if (percent != 100 && max > cur)
1002         *buffering_left = (max - cur) / 1000000;
1003     }
1004   }
1005 }
1006
1007 static void
1008 gst_queue2_post_buffering (GstQueue2 * queue)
1009 {
1010   GstMessage *msg = NULL;
1011
1012   g_mutex_lock (&queue->buffering_post_lock);
1013   GST_QUEUE2_MUTEX_LOCK (queue);
1014   if (queue->percent_changed) {
1015     gint percent = queue->buffering_percent;
1016
1017     queue->percent_changed = FALSE;
1018
1019     GST_DEBUG_OBJECT (queue, "Going to post buffering: %d%%", percent);
1020     msg = gst_message_new_buffering (GST_OBJECT_CAST (queue), percent);
1021
1022     gst_message_set_buffering_stats (msg, queue->mode, queue->avg_in,
1023         queue->avg_out, queue->buffering_left);
1024   }
1025   GST_QUEUE2_MUTEX_UNLOCK (queue);
1026
1027   if (msg != NULL)
1028     gst_element_post_message (GST_ELEMENT_CAST (queue), msg);
1029
1030   g_mutex_unlock (&queue->buffering_post_lock);
1031 }
1032
1033 static void
1034 update_buffering (GstQueue2 * queue)
1035 {
1036   gint percent;
1037
1038   /* Ensure the variables used to calculate buffering state are up-to-date. */
1039   if (queue->current)
1040     update_cur_level (queue, queue->current);
1041   update_in_rates (queue);
1042
1043   if (!get_buffering_percent (queue, NULL, &percent))
1044     return;
1045
1046   if (queue->is_buffering) {
1047     /* if we were buffering see if we reached the high watermark */
1048     if (percent >= 100)
1049       queue->is_buffering = FALSE;
1050
1051     SET_PERCENT (queue, percent);
1052   } else {
1053     /* we were not buffering, check if we need to start buffering if we drop
1054      * below the low threshold */
1055     if (percent < queue->low_percent) {
1056       queue->is_buffering = TRUE;
1057       SET_PERCENT (queue, percent);
1058     }
1059   }
1060 }
1061
1062 static void
1063 reset_rate_timer (GstQueue2 * queue)
1064 {
1065   queue->bytes_in = 0;
1066   queue->bytes_out = 0;
1067   queue->byte_in_rate = 0.0;
1068   queue->byte_in_period = 0;
1069   queue->byte_out_rate = 0.0;
1070   queue->last_update_in_rates_elapsed = 0.0;
1071   queue->last_in_elapsed = 0.0;
1072   queue->last_out_elapsed = 0.0;
1073   queue->in_timer_started = FALSE;
1074   queue->out_timer_started = FALSE;
1075 }
1076
1077 /* the interval in seconds to recalculate the rate */
1078 #define RATE_INTERVAL    0.2
1079 /* Tuning for rate estimation. We use a large window for the input rate because
1080  * it should be stable when connected to a network. The output rate is less
1081  * stable (the elements preroll, queues behind a demuxer fill, ...) and should
1082  * therefore adapt more quickly.
1083  * However, initial input rate may be subject to a burst, and should therefore
1084  * initially also adapt more quickly to changes, and only later on give higher
1085  * weight to previous values. */
1086 #define AVG_IN(avg,val,w1,w2)  ((avg) * (w1) + (val) * (w2)) / ((w1) + (w2))
1087 #define AVG_OUT(avg,val) ((avg) * 3.0 + (val)) / 4.0
1088
1089 static void
1090 update_in_rates (GstQueue2 * queue)
1091 {
1092   gdouble elapsed, period;
1093   gdouble byte_in_rate;
1094
1095   if (!queue->in_timer_started) {
1096     queue->in_timer_started = TRUE;
1097     g_timer_start (queue->in_timer);
1098     return;
1099   }
1100
1101   queue->last_update_in_rates_elapsed = elapsed =
1102       g_timer_elapsed (queue->in_timer, NULL);
1103
1104   /* recalc after each interval. */
1105   if (queue->last_in_elapsed + RATE_INTERVAL < elapsed) {
1106     period = elapsed - queue->last_in_elapsed;
1107
1108     GST_DEBUG_OBJECT (queue,
1109         "rates: period %f, in %" G_GUINT64_FORMAT ", global period %f",
1110         period, queue->bytes_in, queue->byte_in_period);
1111
1112     byte_in_rate = queue->bytes_in / period;
1113
1114     if (queue->byte_in_rate == 0.0)
1115       queue->byte_in_rate = byte_in_rate;
1116     else
1117       queue->byte_in_rate = AVG_IN (queue->byte_in_rate, byte_in_rate,
1118           (double) queue->byte_in_period, period);
1119
1120     /* another data point, cap at 16 for long time running average */
1121     if (queue->byte_in_period < 16 * RATE_INTERVAL)
1122       queue->byte_in_period += period;
1123
1124     /* reset the values to calculate rate over the next interval */
1125     queue->last_in_elapsed = elapsed;
1126     queue->bytes_in = 0;
1127   }
1128
1129   if (queue->byte_in_rate > 0.0) {
1130     queue->cur_level.rate_time =
1131         queue->cur_level.bytes / queue->byte_in_rate * GST_SECOND;
1132   }
1133   GST_DEBUG_OBJECT (queue, "rates: in %f, time %" GST_TIME_FORMAT,
1134       queue->byte_in_rate, GST_TIME_ARGS (queue->cur_level.rate_time));
1135 }
1136
1137 static void
1138 update_out_rates (GstQueue2 * queue)
1139 {
1140   gdouble elapsed, period;
1141   gdouble byte_out_rate;
1142
1143   if (!queue->out_timer_started) {
1144     queue->out_timer_started = TRUE;
1145     g_timer_start (queue->out_timer);
1146     return;
1147   }
1148
1149   elapsed = g_timer_elapsed (queue->out_timer, NULL);
1150
1151   /* recalc after each interval. */
1152   if (queue->last_out_elapsed + RATE_INTERVAL < elapsed) {
1153     period = elapsed - queue->last_out_elapsed;
1154
1155     GST_DEBUG_OBJECT (queue,
1156         "rates: period %f, out %" G_GUINT64_FORMAT, period, queue->bytes_out);
1157
1158     byte_out_rate = queue->bytes_out / period;
1159
1160     if (queue->byte_out_rate == 0.0)
1161       queue->byte_out_rate = byte_out_rate;
1162     else
1163       queue->byte_out_rate = AVG_OUT (queue->byte_out_rate, byte_out_rate);
1164
1165     /* reset the values to calculate rate over the next interval */
1166     queue->last_out_elapsed = elapsed;
1167     queue->bytes_out = 0;
1168   }
1169   if (queue->byte_in_rate > 0.0) {
1170     queue->cur_level.rate_time =
1171         queue->cur_level.bytes / queue->byte_in_rate * GST_SECOND;
1172   }
1173   GST_DEBUG_OBJECT (queue, "rates: out %f, time %" GST_TIME_FORMAT,
1174       queue->byte_out_rate, GST_TIME_ARGS (queue->cur_level.rate_time));
1175 }
1176
1177 static void
1178 update_cur_pos (GstQueue2 * queue, GstQueue2Range * range, guint64 pos)
1179 {
1180   guint64 reading_pos, max_reading_pos;
1181
1182   reading_pos = pos;
1183   max_reading_pos = range->max_reading_pos;
1184
1185   max_reading_pos = MAX (max_reading_pos, reading_pos);
1186
1187   GST_DEBUG_OBJECT (queue,
1188       "updating max_reading_pos from %" G_GUINT64_FORMAT " to %"
1189       G_GUINT64_FORMAT, range->max_reading_pos, max_reading_pos);
1190   range->max_reading_pos = max_reading_pos;
1191
1192   update_cur_level (queue, range);
1193 }
1194
1195 static gboolean
1196 perform_seek_to_offset (GstQueue2 * queue, guint64 offset)
1197 {
1198   GstEvent *event;
1199   gboolean res;
1200
1201   /* until we receive the FLUSH_STOP from this seek, we skip data */
1202   queue->seeking = TRUE;
1203   GST_QUEUE2_MUTEX_UNLOCK (queue);
1204
1205   debug_ranges (queue);
1206
1207   GST_DEBUG_OBJECT (queue, "Seeking to %" G_GUINT64_FORMAT, offset);
1208
1209   event =
1210       gst_event_new_seek (1.0, GST_FORMAT_BYTES,
1211       GST_SEEK_FLAG_FLUSH | GST_SEEK_FLAG_ACCURATE, GST_SEEK_TYPE_SET, offset,
1212       GST_SEEK_TYPE_NONE, -1);
1213
1214   res = gst_pad_push_event (queue->sinkpad, event);
1215   GST_QUEUE2_MUTEX_LOCK (queue);
1216
1217   if (res) {
1218     /* Between us sending the seek event and re-acquiring the lock, the source
1219      * thread might already have pushed data and moved along the range's
1220      * writing_pos beyond the seek offset. In that case we don't want to set
1221      * the writing position back to the requested seek position, as it would
1222      * cause data to be written to the wrong offset in the file or ring buffer.
1223      * We still do the add_range call to switch the current range to the
1224      * requested range, or create one if one doesn't exist yet. */
1225     queue->current = add_range (queue, offset, FALSE);
1226   }
1227
1228   return res;
1229 }
1230
1231 /* get the threshold for when we decide to seek rather than wait */
1232 static guint64
1233 get_seek_threshold (GstQueue2 * queue)
1234 {
1235   guint64 threshold;
1236
1237   /* FIXME, find a good threshold based on the incoming rate. */
1238   threshold = 1024 * 512;
1239
1240   if (QUEUE_IS_USING_RING_BUFFER (queue)) {
1241     threshold = MIN (threshold,
1242         QUEUE_MAX_BYTES (queue) - queue->cur_level.bytes);
1243   }
1244   return threshold;
1245 }
1246
1247 /* see if there is enough data in the file to read a full buffer */
1248 static gboolean
1249 gst_queue2_have_data (GstQueue2 * queue, guint64 offset, guint length)
1250 {
1251   GstQueue2Range *range;
1252
1253   GST_DEBUG_OBJECT (queue, "looking for offset %" G_GUINT64_FORMAT ", len %u",
1254       offset, length);
1255
1256   if ((range = find_range (queue, offset))) {
1257     if (queue->current != range) {
1258       GST_DEBUG_OBJECT (queue, "switching ranges, do seek to range position");
1259       perform_seek_to_offset (queue, range->writing_pos);
1260     }
1261
1262     GST_INFO_OBJECT (queue, "cur_level.bytes %u (max %" G_GUINT64_FORMAT ")",
1263         queue->cur_level.bytes, QUEUE_MAX_BYTES (queue));
1264
1265     /* we have a range for offset */
1266     GST_DEBUG_OBJECT (queue,
1267         "we have a range %p, offset %" G_GUINT64_FORMAT ", writing_pos %"
1268         G_GUINT64_FORMAT, range, range->offset, range->writing_pos);
1269
1270     if (!QUEUE_IS_USING_RING_BUFFER (queue) && queue->is_eos)
1271       return TRUE;
1272
1273     if (offset + length <= range->writing_pos)
1274       return TRUE;
1275     else
1276       GST_DEBUG_OBJECT (queue,
1277           "Need more data (%" G_GUINT64_FORMAT " bytes more)",
1278           (offset + length) - range->writing_pos);
1279
1280   } else {
1281     GST_INFO_OBJECT (queue, "not found in any range off %" G_GUINT64_FORMAT
1282         " len %u", offset, length);
1283     /* we don't have the range, see how far away we are */
1284     if (!queue->is_eos && queue->current) {
1285       guint64 threshold = get_seek_threshold (queue);
1286
1287       if (offset >= queue->current->offset && offset <=
1288           queue->current->writing_pos + threshold) {
1289         GST_INFO_OBJECT (queue,
1290             "requested data is within range, wait for data");
1291         return FALSE;
1292       }
1293     }
1294
1295     /* too far away, do a seek */
1296     perform_seek_to_offset (queue, offset);
1297   }
1298
1299   return FALSE;
1300 }
1301
1302 #ifdef HAVE_FSEEKO
1303 #define FSEEK_FILE(file,offset)  (fseeko (file, (off_t) offset, SEEK_SET) != 0)
1304 #elif defined (G_OS_UNIX) || defined (G_OS_WIN32)
1305 #define FSEEK_FILE(file,offset)  (lseek (fileno (file), (off_t) offset, SEEK_SET) == (off_t) -1)
1306 #else
1307 #define FSEEK_FILE(file,offset)  (fseek (file, offset, SEEK_SET) != 0)
1308 #endif
1309
1310 static GstFlowReturn
1311 gst_queue2_read_data_at_offset (GstQueue2 * queue, guint64 offset, guint length,
1312     guint8 * dst, gint64 * read_return)
1313 {
1314   guint8 *ring_buffer;
1315   size_t res;
1316
1317   ring_buffer = queue->ring_buffer;
1318
1319   if (QUEUE_IS_USING_TEMP_FILE (queue) && FSEEK_FILE (queue->temp_file, offset))
1320     goto seek_failed;
1321
1322   /* this should not block */
1323   GST_LOG_OBJECT (queue, "Reading %d bytes from offset %" G_GUINT64_FORMAT,
1324       length, offset);
1325   if (QUEUE_IS_USING_TEMP_FILE (queue)) {
1326     res = fread (dst, 1, length, queue->temp_file);
1327   } else {
1328     memcpy (dst, ring_buffer + offset, length);
1329     res = length;
1330   }
1331
1332   GST_LOG_OBJECT (queue, "read %" G_GSIZE_FORMAT " bytes", res);
1333
1334   if (G_UNLIKELY (res < length)) {
1335     if (!QUEUE_IS_USING_TEMP_FILE (queue))
1336       goto could_not_read;
1337     /* check for errors or EOF */
1338     if (ferror (queue->temp_file))
1339       goto could_not_read;
1340     if (feof (queue->temp_file) && length > 0)
1341       goto eos;
1342   }
1343
1344   *read_return = res;
1345
1346   return GST_FLOW_OK;
1347
1348 seek_failed:
1349   {
1350     GST_ELEMENT_ERROR (queue, RESOURCE, SEEK, (NULL), GST_ERROR_SYSTEM);
1351     return GST_FLOW_ERROR;
1352   }
1353 could_not_read:
1354   {
1355     GST_ELEMENT_ERROR (queue, RESOURCE, READ, (NULL), GST_ERROR_SYSTEM);
1356     return GST_FLOW_ERROR;
1357   }
1358 eos:
1359   {
1360     GST_DEBUG ("non-regular file hits EOS");
1361     return GST_FLOW_EOS;
1362   }
1363 }
1364
1365 static GstFlowReturn
1366 gst_queue2_create_read (GstQueue2 * queue, guint64 offset, guint length,
1367     GstBuffer ** buffer)
1368 {
1369   GstBuffer *buf;
1370   GstMapInfo info;
1371   guint8 *data;
1372   guint64 file_offset;
1373   guint block_length, remaining, read_length;
1374   guint64 rb_size;
1375   guint64 max_size;
1376   guint64 rpos;
1377   GstFlowReturn ret = GST_FLOW_OK;
1378
1379   /* allocate the output buffer of the requested size */
1380   if (*buffer == NULL)
1381     buf = gst_buffer_new_allocate (NULL, length, NULL);
1382   else
1383     buf = *buffer;
1384
1385   gst_buffer_map (buf, &info, GST_MAP_WRITE);
1386   data = info.data;
1387
1388   GST_DEBUG_OBJECT (queue, "Reading %u bytes from %" G_GUINT64_FORMAT, length,
1389       offset);
1390
1391   rpos = offset;
1392   rb_size = queue->ring_buffer_max_size;
1393   max_size = QUEUE_MAX_BYTES (queue);
1394
1395   remaining = length;
1396   while (remaining > 0) {
1397     /* configure how much/whether to read */
1398     if (!gst_queue2_have_data (queue, rpos, remaining)) {
1399       read_length = 0;
1400
1401       if (QUEUE_IS_USING_RING_BUFFER (queue)) {
1402         guint64 level;
1403
1404         /* calculate how far away the offset is */
1405         if (queue->current->writing_pos > rpos)
1406           level = queue->current->writing_pos - rpos;
1407         else
1408           level = 0;
1409
1410         GST_DEBUG_OBJECT (queue,
1411             "reading %" G_GUINT64_FORMAT ", writing %" G_GUINT64_FORMAT
1412             ", level %" G_GUINT64_FORMAT ", max %" G_GUINT64_FORMAT,
1413             rpos, queue->current->writing_pos, level, max_size);
1414
1415         if (level >= max_size) {
1416           /* we don't have the data but if we have a ring buffer that is full, we
1417            * need to read */
1418           GST_DEBUG_OBJECT (queue,
1419               "ring buffer full, reading QUEUE_MAX_BYTES %"
1420               G_GUINT64_FORMAT " bytes", max_size);
1421           read_length = max_size;
1422         } else if (queue->is_eos) {
1423           /* won't get any more data so read any data we have */
1424           if (level) {
1425             GST_DEBUG_OBJECT (queue,
1426                 "EOS hit but read %" G_GUINT64_FORMAT " bytes that we have",
1427                 level);
1428             read_length = level;
1429             remaining = level;
1430             length = level;
1431           } else
1432             goto hit_eos;
1433         }
1434       }
1435
1436       if (read_length == 0) {
1437         if (QUEUE_IS_USING_RING_BUFFER (queue)) {
1438           GST_DEBUG_OBJECT (queue,
1439               "update current position [%" G_GUINT64_FORMAT "-%"
1440               G_GUINT64_FORMAT "]", rpos, queue->current->max_reading_pos);
1441           update_cur_pos (queue, queue->current, rpos);
1442           GST_QUEUE2_SIGNAL_DEL (queue);
1443         }
1444
1445         if (queue->use_buffering)
1446           update_buffering (queue);
1447
1448         GST_DEBUG_OBJECT (queue, "waiting for add");
1449         GST_QUEUE2_WAIT_ADD_CHECK (queue, queue->srcresult, out_flushing);
1450         continue;
1451       }
1452     } else {
1453       /* we have the requested data so read it */
1454       read_length = remaining;
1455     }
1456
1457     /* set range reading_pos to actual reading position for this read */
1458     queue->current->reading_pos = rpos;
1459
1460     /* configure how much and from where to read */
1461     if (QUEUE_IS_USING_RING_BUFFER (queue)) {
1462       file_offset =
1463           (queue->current->rb_offset + (rpos -
1464               queue->current->offset)) % rb_size;
1465       if (file_offset + read_length > rb_size) {
1466         block_length = rb_size - file_offset;
1467       } else {
1468         block_length = read_length;
1469       }
1470     } else {
1471       file_offset = rpos;
1472       block_length = read_length;
1473     }
1474
1475     /* while we still have data to read, we loop */
1476     while (read_length > 0) {
1477       gint64 read_return;
1478
1479       ret =
1480           gst_queue2_read_data_at_offset (queue, file_offset, block_length,
1481           data, &read_return);
1482       if (ret != GST_FLOW_OK)
1483         goto read_error;
1484
1485       file_offset += read_return;
1486       if (QUEUE_IS_USING_RING_BUFFER (queue))
1487         file_offset %= rb_size;
1488
1489       data += read_return;
1490       read_length -= read_return;
1491       block_length = read_length;
1492       remaining -= read_return;
1493
1494       rpos = (queue->current->reading_pos += read_return);
1495       update_cur_pos (queue, queue->current, queue->current->reading_pos);
1496     }
1497     GST_QUEUE2_SIGNAL_DEL (queue);
1498     GST_DEBUG_OBJECT (queue, "%u bytes left to read", remaining);
1499   }
1500
1501   gst_buffer_unmap (buf, &info);
1502   gst_buffer_resize (buf, 0, length);
1503
1504   GST_BUFFER_OFFSET (buf) = offset;
1505   GST_BUFFER_OFFSET_END (buf) = offset + length;
1506
1507   *buffer = buf;
1508
1509   return ret;
1510
1511   /* ERRORS */
1512 hit_eos:
1513   {
1514     GST_DEBUG_OBJECT (queue, "EOS hit and we don't have any requested data");
1515     gst_buffer_unmap (buf, &info);
1516     if (*buffer == NULL)
1517       gst_buffer_unref (buf);
1518     return GST_FLOW_EOS;
1519   }
1520 out_flushing:
1521   {
1522     GST_DEBUG_OBJECT (queue, "we are flushing");
1523     gst_buffer_unmap (buf, &info);
1524     if (*buffer == NULL)
1525       gst_buffer_unref (buf);
1526     return GST_FLOW_FLUSHING;
1527   }
1528 read_error:
1529   {
1530     GST_DEBUG_OBJECT (queue, "we have a read error");
1531     gst_buffer_unmap (buf, &info);
1532     if (*buffer == NULL)
1533       gst_buffer_unref (buf);
1534     return ret;
1535   }
1536 }
1537
1538 /* should be called with QUEUE_LOCK */
1539 static GstMiniObject *
1540 gst_queue2_read_item_from_file (GstQueue2 * queue)
1541 {
1542   GstMiniObject *item;
1543
1544   if (queue->stream_start_event != NULL) {
1545     item = GST_MINI_OBJECT_CAST (queue->stream_start_event);
1546     queue->stream_start_event = NULL;
1547   } else if (queue->starting_segment != NULL) {
1548     item = GST_MINI_OBJECT_CAST (queue->starting_segment);
1549     queue->starting_segment = NULL;
1550   } else {
1551     GstFlowReturn ret;
1552     GstBuffer *buffer = NULL;
1553     guint64 reading_pos;
1554
1555     reading_pos = queue->current->reading_pos;
1556
1557     ret =
1558         gst_queue2_create_read (queue, reading_pos, DEFAULT_BUFFER_SIZE,
1559         &buffer);
1560
1561     switch (ret) {
1562       case GST_FLOW_OK:
1563         item = GST_MINI_OBJECT_CAST (buffer);
1564         break;
1565       case GST_FLOW_EOS:
1566         item = GST_MINI_OBJECT_CAST (gst_event_new_eos ());
1567         break;
1568       default:
1569         item = NULL;
1570         break;
1571     }
1572   }
1573   return item;
1574 }
1575
1576 /* must be called with MUTEX_LOCK. Will briefly release the lock when notifying
1577  * the temp filename. */
1578 static gboolean
1579 gst_queue2_open_temp_location_file (GstQueue2 * queue)
1580 {
1581   gint fd = -1;
1582   gchar *name = NULL;
1583
1584   if (queue->temp_file)
1585     goto already_opened;
1586
1587   GST_DEBUG_OBJECT (queue, "opening temp file %s", queue->temp_template);
1588
1589   /* If temp_template was set, allocate a filename and open that file */
1590
1591   /* nothing to do */
1592   if (queue->temp_template == NULL)
1593     goto no_directory;
1594
1595   /* make copy of the template, we don't want to change this */
1596   name = g_strdup (queue->temp_template);
1597
1598 #ifdef __BIONIC__
1599   fd = g_mkstemp_full (name, O_RDWR | O_LARGEFILE, S_IRUSR | S_IWUSR);
1600 #else
1601   fd = g_mkstemp (name);
1602 #endif
1603
1604   if (fd == -1)
1605     goto mkstemp_failed;
1606
1607   /* open the file for update/writing */
1608   queue->temp_file = fdopen (fd, "wb+");
1609   /* error creating file */
1610   if (queue->temp_file == NULL)
1611     goto open_failed;
1612
1613   g_free (queue->temp_location);
1614   queue->temp_location = name;
1615
1616   GST_QUEUE2_MUTEX_UNLOCK (queue);
1617
1618   /* we can't emit the notify with the lock */
1619   g_object_notify (G_OBJECT (queue), "temp-location");
1620
1621   GST_QUEUE2_MUTEX_LOCK (queue);
1622
1623   GST_DEBUG_OBJECT (queue, "opened temp file %s", queue->temp_template);
1624
1625   return TRUE;
1626
1627   /* ERRORS */
1628 already_opened:
1629   {
1630     GST_DEBUG_OBJECT (queue, "temp file was already open");
1631     return TRUE;
1632   }
1633 no_directory:
1634   {
1635     GST_ELEMENT_ERROR (queue, RESOURCE, NOT_FOUND,
1636         (_("No Temp directory specified.")), (NULL));
1637     return FALSE;
1638   }
1639 mkstemp_failed:
1640   {
1641     GST_ELEMENT_ERROR (queue, RESOURCE, OPEN_READ,
1642         (_("Could not create temp file \"%s\"."), queue->temp_template),
1643         GST_ERROR_SYSTEM);
1644     g_free (name);
1645     return FALSE;
1646   }
1647 open_failed:
1648   {
1649     GST_ELEMENT_ERROR (queue, RESOURCE, OPEN_READ,
1650         (_("Could not open file \"%s\" for reading."), name), GST_ERROR_SYSTEM);
1651     g_free (name);
1652     if (fd != -1)
1653       close (fd);
1654     return FALSE;
1655   }
1656 }
1657
1658 static void
1659 gst_queue2_close_temp_location_file (GstQueue2 * queue)
1660 {
1661   /* nothing to do */
1662   if (queue->temp_file == NULL)
1663     return;
1664
1665   GST_DEBUG_OBJECT (queue, "closing temp file");
1666
1667   fflush (queue->temp_file);
1668   fclose (queue->temp_file);
1669
1670   if (queue->temp_remove) {
1671     if (remove (queue->temp_location) < 0) {
1672       GST_WARNING_OBJECT (queue, "Failed to remove temporary file %s: %s",
1673           queue->temp_location, g_strerror (errno));
1674     }
1675   }
1676
1677   queue->temp_file = NULL;
1678   clean_ranges (queue);
1679 }
1680
1681 static void
1682 gst_queue2_flush_temp_file (GstQueue2 * queue)
1683 {
1684   if (queue->temp_file == NULL)
1685     return;
1686
1687   GST_DEBUG_OBJECT (queue, "flushing temp file");
1688
1689   queue->temp_file = g_freopen (queue->temp_location, "wb+", queue->temp_file);
1690 }
1691
1692 static void
1693 gst_queue2_locked_flush (GstQueue2 * queue, gboolean full, gboolean clear_temp)
1694 {
1695   if (!QUEUE_IS_USING_QUEUE (queue)) {
1696     if (QUEUE_IS_USING_TEMP_FILE (queue) && clear_temp)
1697       gst_queue2_flush_temp_file (queue);
1698     init_ranges (queue);
1699   } else {
1700     while (!g_queue_is_empty (&queue->queue)) {
1701       GstQueue2Item *qitem = g_queue_pop_head (&queue->queue);
1702
1703       if (!full && qitem->type == GST_QUEUE2_ITEM_TYPE_EVENT
1704           && GST_EVENT_IS_STICKY (qitem->item)
1705           && GST_EVENT_TYPE (qitem->item) != GST_EVENT_SEGMENT
1706           && GST_EVENT_TYPE (qitem->item) != GST_EVENT_EOS) {
1707         gst_pad_store_sticky_event (queue->srcpad,
1708             GST_EVENT_CAST (qitem->item));
1709       }
1710
1711       /* Then lose another reference because we are supposed to destroy that
1712          data when flushing */
1713       if (qitem->type != GST_QUEUE2_ITEM_TYPE_QUERY)
1714         gst_mini_object_unref (qitem->item);
1715       g_slice_free (GstQueue2Item, qitem);
1716     }
1717   }
1718   queue->last_query = FALSE;
1719   g_cond_signal (&queue->query_handled);
1720   GST_QUEUE2_CLEAR_LEVEL (queue->cur_level);
1721   gst_segment_init (&queue->sink_segment, GST_FORMAT_TIME);
1722   gst_segment_init (&queue->src_segment, GST_FORMAT_TIME);
1723   queue->sinktime = queue->srctime = GST_CLOCK_TIME_NONE;
1724   queue->sink_tainted = queue->src_tainted = TRUE;
1725   if (queue->starting_segment != NULL)
1726     gst_event_unref (queue->starting_segment);
1727   queue->starting_segment = NULL;
1728   queue->segment_event_received = FALSE;
1729   gst_event_replace (&queue->stream_start_event, NULL);
1730
1731   /* we deleted a lot of something */
1732   GST_QUEUE2_SIGNAL_DEL (queue);
1733 }
1734
1735 static gboolean
1736 gst_queue2_wait_free_space (GstQueue2 * queue)
1737 {
1738   /* We make space available if we're "full" according to whatever
1739    * the user defined as "full". */
1740   if (gst_queue2_is_filled (queue)) {
1741     gboolean started;
1742
1743     /* pause the timer while we wait. The fact that we are waiting does not mean
1744      * the byterate on the input pad is lower */
1745     if ((started = queue->in_timer_started))
1746       g_timer_stop (queue->in_timer);
1747
1748     GST_CAT_DEBUG_OBJECT (queue_dataflow, queue,
1749         "queue is full, waiting for free space");
1750     do {
1751       /* Wait for space to be available, we could be unlocked because of a flush. */
1752       GST_QUEUE2_WAIT_DEL_CHECK (queue, queue->sinkresult, out_flushing);
1753     }
1754     while (gst_queue2_is_filled (queue));
1755
1756     /* and continue if we were running before */
1757     if (started)
1758       g_timer_continue (queue->in_timer);
1759   }
1760   return TRUE;
1761
1762   /* ERRORS */
1763 out_flushing:
1764   {
1765     GST_CAT_DEBUG_OBJECT (queue_dataflow, queue, "queue is flushing");
1766     return FALSE;
1767   }
1768 }
1769
1770 static gboolean
1771 gst_queue2_create_write (GstQueue2 * queue, GstBuffer * buffer)
1772 {
1773   GstMapInfo info;
1774   guint8 *data, *ring_buffer;
1775   guint size, rb_size;
1776   guint64 writing_pos, new_writing_pos;
1777   GstQueue2Range *range, *prev, *next;
1778   gboolean do_seek = FALSE;
1779
1780   if (QUEUE_IS_USING_RING_BUFFER (queue))
1781     writing_pos = queue->current->rb_writing_pos;
1782   else
1783     writing_pos = queue->current->writing_pos;
1784   ring_buffer = queue->ring_buffer;
1785   rb_size = queue->ring_buffer_max_size;
1786
1787   gst_buffer_map (buffer, &info, GST_MAP_READ);
1788
1789   size = info.size;
1790   data = info.data;
1791
1792   GST_DEBUG_OBJECT (queue, "Writing %u bytes to %" G_GUINT64_FORMAT, size,
1793       writing_pos);
1794
1795   /* sanity check */
1796   if (GST_BUFFER_OFFSET_IS_VALID (buffer) &&
1797       GST_BUFFER_OFFSET (buffer) != queue->current->writing_pos) {
1798     GST_WARNING_OBJECT (queue, "buffer offset does not match current writing "
1799         "position! %" G_GINT64_FORMAT " != %" G_GINT64_FORMAT,
1800         GST_BUFFER_OFFSET (buffer), queue->current->writing_pos);
1801   }
1802
1803   while (size > 0) {
1804     guint to_write;
1805
1806     if (QUEUE_IS_USING_RING_BUFFER (queue)) {
1807       gint64 space;
1808
1809       /* calculate the space in the ring buffer not used by data from
1810        * the current range */
1811       while (QUEUE_MAX_BYTES (queue) <= queue->cur_level.bytes) {
1812         /* wait until there is some free space */
1813         GST_QUEUE2_WAIT_DEL_CHECK (queue, queue->sinkresult, out_flushing);
1814       }
1815       /* get the amount of space we have */
1816       space = QUEUE_MAX_BYTES (queue) - queue->cur_level.bytes;
1817
1818       /* calculate if we need to split or if we can write the entire
1819        * buffer now */
1820       to_write = MIN (size, space);
1821
1822       /* the writing position in the ring buffer after writing (part
1823        * or all of) the buffer */
1824       new_writing_pos = (writing_pos + to_write) % rb_size;
1825
1826       prev = NULL;
1827       range = queue->ranges;
1828
1829       /* if we need to overwrite data in the ring buffer, we need to
1830        * update the ranges
1831        *
1832        * warning: this code is complicated and includes some
1833        * simplifications - pen, paper and diagrams for the cases
1834        * recommended! */
1835       while (range) {
1836         guint64 range_data_start, range_data_end;
1837         GstQueue2Range *range_to_destroy = NULL;
1838
1839         range_data_start = range->rb_offset;
1840         range_data_end = range->rb_writing_pos;
1841
1842         /* handle the special case where the range has no data in it */
1843         if (range->writing_pos == range->offset) {
1844           if (range != queue->current) {
1845             GST_DEBUG_OBJECT (queue,
1846                 "Removing range: offset %" G_GUINT64_FORMAT ", wpos %"
1847                 G_GUINT64_FORMAT, range->offset, range->writing_pos);
1848             /* remove range */
1849             range_to_destroy = range;
1850             if (prev)
1851               prev->next = range->next;
1852           }
1853           goto next_range;
1854         }
1855
1856         if (range_data_end > range_data_start) {
1857           if (writing_pos >= range_data_end && new_writing_pos >= writing_pos)
1858             goto next_range;
1859
1860           if (new_writing_pos > range_data_start) {
1861             if (new_writing_pos >= range_data_end) {
1862               GST_DEBUG_OBJECT (queue,
1863                   "Removing range: offset %" G_GUINT64_FORMAT ", wpos %"
1864                   G_GUINT64_FORMAT, range->offset, range->writing_pos);
1865               /* remove range */
1866               range_to_destroy = range;
1867               if (prev)
1868                 prev->next = range->next;
1869             } else {
1870               GST_DEBUG_OBJECT (queue,
1871                   "advancing offsets from %" G_GUINT64_FORMAT " (%"
1872                   G_GUINT64_FORMAT ") to %" G_GUINT64_FORMAT " (%"
1873                   G_GUINT64_FORMAT ")", range->offset, range->rb_offset,
1874                   range->offset + new_writing_pos - range_data_start,
1875                   new_writing_pos);
1876               range->offset += (new_writing_pos - range_data_start);
1877               range->rb_offset = new_writing_pos;
1878             }
1879           }
1880         } else {
1881           guint64 new_wpos_virt = writing_pos + to_write;
1882
1883           if (new_wpos_virt <= range_data_start)
1884             goto next_range;
1885
1886           if (new_wpos_virt > rb_size && new_writing_pos >= range_data_end) {
1887             GST_DEBUG_OBJECT (queue,
1888                 "Removing range: offset %" G_GUINT64_FORMAT ", wpos %"
1889                 G_GUINT64_FORMAT, range->offset, range->writing_pos);
1890             /* remove range */
1891             range_to_destroy = range;
1892             if (prev)
1893               prev->next = range->next;
1894           } else {
1895             GST_DEBUG_OBJECT (queue,
1896                 "advancing offsets from %" G_GUINT64_FORMAT " (%"
1897                 G_GUINT64_FORMAT ") to %" G_GUINT64_FORMAT " (%"
1898                 G_GUINT64_FORMAT ")", range->offset, range->rb_offset,
1899                 range->offset + new_writing_pos - range_data_start,
1900                 new_writing_pos);
1901             range->offset += (new_wpos_virt - range_data_start);
1902             range->rb_offset = new_writing_pos;
1903           }
1904         }
1905
1906       next_range:
1907         if (!range_to_destroy)
1908           prev = range;
1909
1910         range = range->next;
1911         if (range_to_destroy) {
1912           if (range_to_destroy == queue->ranges)
1913             queue->ranges = range;
1914           g_slice_free (GstQueue2Range, range_to_destroy);
1915           range_to_destroy = NULL;
1916         }
1917       }
1918     } else {
1919       to_write = size;
1920       new_writing_pos = writing_pos + to_write;
1921     }
1922
1923     if (QUEUE_IS_USING_TEMP_FILE (queue)
1924         && FSEEK_FILE (queue->temp_file, writing_pos))
1925       goto seek_failed;
1926
1927     if (new_writing_pos > writing_pos) {
1928       GST_INFO_OBJECT (queue,
1929           "writing %u bytes to range [%" G_GUINT64_FORMAT "-%" G_GUINT64_FORMAT
1930           "] (rb wpos %" G_GUINT64_FORMAT ")", to_write, queue->current->offset,
1931           queue->current->writing_pos, queue->current->rb_writing_pos);
1932       /* either not using ring buffer or no wrapping, just write */
1933       if (QUEUE_IS_USING_TEMP_FILE (queue)) {
1934         if (fwrite (data, to_write, 1, queue->temp_file) != 1)
1935           goto handle_error;
1936       } else {
1937         memcpy (ring_buffer + writing_pos, data, to_write);
1938       }
1939
1940       if (!QUEUE_IS_USING_RING_BUFFER (queue)) {
1941         /* try to merge with next range */
1942         while ((next = queue->current->next)) {
1943           GST_INFO_OBJECT (queue,
1944               "checking merge with next range %" G_GUINT64_FORMAT " < %"
1945               G_GUINT64_FORMAT, new_writing_pos, next->offset);
1946           if (new_writing_pos < next->offset)
1947             break;
1948
1949           GST_DEBUG_OBJECT (queue, "merging ranges %" G_GUINT64_FORMAT,
1950               next->writing_pos);
1951
1952           /* remove the group */
1953           queue->current->next = next->next;
1954
1955           /* We use the threshold to decide if we want to do a seek or simply
1956            * read the data again. If there is not so much data in the range we
1957            * prefer to avoid to seek and read it again. */
1958           if (next->writing_pos > new_writing_pos + get_seek_threshold (queue)) {
1959             /* the new range had more data than the threshold, it's worth keeping
1960              * it and doing a seek. */
1961             new_writing_pos = next->writing_pos;
1962             do_seek = TRUE;
1963           }
1964           g_slice_free (GstQueue2Range, next);
1965         }
1966         goto update_and_signal;
1967       }
1968     } else {
1969       /* wrapping */
1970       guint block_one, block_two;
1971
1972       block_one = rb_size - writing_pos;
1973       block_two = to_write - block_one;
1974
1975       if (block_one > 0) {
1976         GST_INFO_OBJECT (queue, "writing %u bytes", block_one);
1977         /* write data to end of ring buffer */
1978         if (QUEUE_IS_USING_TEMP_FILE (queue)) {
1979           if (fwrite (data, block_one, 1, queue->temp_file) != 1)
1980             goto handle_error;
1981         } else {
1982           memcpy (ring_buffer + writing_pos, data, block_one);
1983         }
1984       }
1985
1986       if (QUEUE_IS_USING_TEMP_FILE (queue) && FSEEK_FILE (queue->temp_file, 0))
1987         goto seek_failed;
1988
1989       if (block_two > 0) {
1990         GST_INFO_OBJECT (queue, "writing %u bytes", block_two);
1991         if (QUEUE_IS_USING_TEMP_FILE (queue)) {
1992           if (fwrite (data + block_one, block_two, 1, queue->temp_file) != 1)
1993             goto handle_error;
1994         } else {
1995           memcpy (ring_buffer, data + block_one, block_two);
1996         }
1997       }
1998     }
1999
2000   update_and_signal:
2001     /* update the writing positions */
2002     size -= to_write;
2003     GST_INFO_OBJECT (queue,
2004         "wrote %u bytes to %" G_GUINT64_FORMAT " (%u bytes remaining to write)",
2005         to_write, writing_pos, size);
2006
2007     if (QUEUE_IS_USING_RING_BUFFER (queue)) {
2008       data += to_write;
2009       queue->current->writing_pos += to_write;
2010       queue->current->rb_writing_pos = writing_pos = new_writing_pos;
2011     } else {
2012       queue->current->writing_pos = writing_pos = new_writing_pos;
2013     }
2014     if (do_seek)
2015       perform_seek_to_offset (queue, new_writing_pos);
2016
2017     update_cur_level (queue, queue->current);
2018
2019     /* update the buffering status */
2020     if (queue->use_buffering)
2021       update_buffering (queue);
2022
2023     GST_INFO_OBJECT (queue, "cur_level.bytes %u (max %" G_GUINT64_FORMAT ")",
2024         queue->cur_level.bytes, QUEUE_MAX_BYTES (queue));
2025
2026     GST_QUEUE2_SIGNAL_ADD (queue);
2027   }
2028
2029   gst_buffer_unmap (buffer, &info);
2030
2031   return TRUE;
2032
2033   /* ERRORS */
2034 out_flushing:
2035   {
2036     GST_DEBUG_OBJECT (queue, "we are flushing");
2037     gst_buffer_unmap (buffer, &info);
2038     /* FIXME - GST_FLOW_EOS ? */
2039     return FALSE;
2040   }
2041 seek_failed:
2042   {
2043     GST_ELEMENT_ERROR (queue, RESOURCE, SEEK, (NULL), GST_ERROR_SYSTEM);
2044     gst_buffer_unmap (buffer, &info);
2045     return FALSE;
2046   }
2047 handle_error:
2048   {
2049     switch (errno) {
2050       case ENOSPC:{
2051         GST_ELEMENT_ERROR (queue, RESOURCE, NO_SPACE_LEFT, (NULL), (NULL));
2052         break;
2053       }
2054       default:{
2055         GST_ELEMENT_ERROR (queue, RESOURCE, WRITE,
2056             (_("Error while writing to download file.")),
2057             ("%s", g_strerror (errno)));
2058       }
2059     }
2060     gst_buffer_unmap (buffer, &info);
2061     return FALSE;
2062   }
2063 }
2064
2065 static gboolean
2066 buffer_list_create_write (GstBuffer ** buf, guint idx, gpointer q)
2067 {
2068   GstQueue2 *queue = q;
2069
2070   GST_TRACE_OBJECT (queue,
2071       "writing buffer %u of size %" G_GSIZE_FORMAT " bytes", idx,
2072       gst_buffer_get_size (*buf));
2073
2074   if (!gst_queue2_create_write (queue, *buf)) {
2075     GST_INFO_OBJECT (queue, "create_write() returned FALSE, bailing out");
2076     return FALSE;
2077   }
2078   return TRUE;
2079 }
2080
2081 static gboolean
2082 buffer_list_calc_size (GstBuffer ** buf, guint idx, gpointer data)
2083 {
2084   guint *p_size = data;
2085   gsize buf_size;
2086
2087   buf_size = gst_buffer_get_size (*buf);
2088   GST_TRACE ("buffer %u in has size %" G_GSIZE_FORMAT, idx, buf_size);
2089   *p_size += buf_size;
2090   return TRUE;
2091 }
2092
2093 /* enqueue an item an update the level stats */
2094 static void
2095 gst_queue2_locked_enqueue (GstQueue2 * queue, gpointer item,
2096     GstQueue2ItemType item_type)
2097 {
2098   if (item_type == GST_QUEUE2_ITEM_TYPE_BUFFER) {
2099     GstBuffer *buffer;
2100     guint size;
2101
2102     buffer = GST_BUFFER_CAST (item);
2103     size = gst_buffer_get_size (buffer);
2104
2105     /* add buffer to the statistics */
2106     if (QUEUE_IS_USING_QUEUE (queue)) {
2107       queue->cur_level.buffers++;
2108       queue->cur_level.bytes += size;
2109     }
2110     queue->bytes_in += size;
2111
2112     /* apply new buffer to segment stats */
2113     apply_buffer (queue, buffer, &queue->sink_segment, size, TRUE);
2114     /* update the byterate stats */
2115     update_in_rates (queue);
2116
2117     if (!QUEUE_IS_USING_QUEUE (queue)) {
2118       /* FIXME - check return value? */
2119       gst_queue2_create_write (queue, buffer);
2120     }
2121   } else if (item_type == GST_QUEUE2_ITEM_TYPE_BUFFER_LIST) {
2122     GstBufferList *buffer_list;
2123     guint size = 0;
2124
2125     buffer_list = GST_BUFFER_LIST_CAST (item);
2126
2127     gst_buffer_list_foreach (buffer_list, buffer_list_calc_size, &size);
2128     GST_LOG_OBJECT (queue, "total size of buffer list: %u bytes", size);
2129
2130     /* add buffer to the statistics */
2131     if (QUEUE_IS_USING_QUEUE (queue)) {
2132       queue->cur_level.buffers += gst_buffer_list_length (buffer_list);
2133       queue->cur_level.bytes += size;
2134     }
2135     queue->bytes_in += size;
2136
2137     /* apply new buffer to segment stats */
2138     apply_buffer_list (queue, buffer_list, &queue->sink_segment, TRUE);
2139
2140     /* update the byterate stats */
2141     update_in_rates (queue);
2142
2143     if (!QUEUE_IS_USING_QUEUE (queue)) {
2144       gst_buffer_list_foreach (buffer_list, buffer_list_create_write, queue);
2145     }
2146   } else if (item_type == GST_QUEUE2_ITEM_TYPE_EVENT) {
2147     GstEvent *event;
2148
2149     event = GST_EVENT_CAST (item);
2150
2151     switch (GST_EVENT_TYPE (event)) {
2152       case GST_EVENT_EOS:
2153         /* Zero the thresholds, this makes sure the queue is completely
2154          * filled and we can read all data from the queue. */
2155         GST_DEBUG_OBJECT (queue, "we have EOS");
2156         queue->is_eos = TRUE;
2157         break;
2158       case GST_EVENT_SEGMENT:
2159         apply_segment (queue, event, &queue->sink_segment, TRUE);
2160         /* This is our first new segment, we hold it
2161          * as we can't save it on the temp file */
2162         if (!QUEUE_IS_USING_QUEUE (queue)) {
2163           if (queue->segment_event_received)
2164             goto unexpected_event;
2165
2166           queue->segment_event_received = TRUE;
2167           if (queue->starting_segment != NULL)
2168             gst_event_unref (queue->starting_segment);
2169           queue->starting_segment = event;
2170           item = NULL;
2171         }
2172         /* a new segment allows us to accept more buffers if we got EOS
2173          * from downstream */
2174         queue->unexpected = FALSE;
2175         break;
2176       case GST_EVENT_GAP:
2177         apply_gap (queue, event, &queue->sink_segment, TRUE);
2178         break;
2179       case GST_EVENT_STREAM_START:
2180         if (!QUEUE_IS_USING_QUEUE (queue)) {
2181           gst_event_replace (&queue->stream_start_event, event);
2182           gst_event_unref (event);
2183           item = NULL;
2184         }
2185         break;
2186       case GST_EVENT_CAPS:{
2187         GstCaps *caps;
2188
2189         gst_event_parse_caps (event, &caps);
2190         GST_INFO ("got caps: %" GST_PTR_FORMAT, caps);
2191
2192         if (!QUEUE_IS_USING_QUEUE (queue)) {
2193           GST_LOG ("Dropping caps event, not using queue");
2194           gst_event_unref (event);
2195           item = NULL;
2196         }
2197         break;
2198       }
2199       default:
2200         if (!QUEUE_IS_USING_QUEUE (queue))
2201           goto unexpected_event;
2202         break;
2203     }
2204   } else if (GST_IS_QUERY (item)) {
2205     /* Can't happen as we check that in the caller */
2206     if (!QUEUE_IS_USING_QUEUE (queue))
2207       g_assert_not_reached ();
2208   } else {
2209     g_warning ("Unexpected item %p added in queue %s (refcounting problem?)",
2210         item, GST_OBJECT_NAME (queue));
2211     /* we can't really unref since we don't know what it is */
2212     item = NULL;
2213   }
2214
2215   if (item) {
2216     /* update the buffering status */
2217     if (queue->use_buffering)
2218       update_buffering (queue);
2219
2220     if (QUEUE_IS_USING_QUEUE (queue)) {
2221       GstQueue2Item *qitem = g_slice_new (GstQueue2Item);
2222       qitem->type = item_type;
2223       qitem->item = item;
2224       g_queue_push_tail (&queue->queue, qitem);
2225     } else {
2226       gst_mini_object_unref (GST_MINI_OBJECT_CAST (item));
2227     }
2228
2229     GST_QUEUE2_SIGNAL_ADD (queue);
2230   }
2231
2232   return;
2233
2234   /* ERRORS */
2235 unexpected_event:
2236   {
2237     gboolean is_custom = GST_EVENT_TYPE (item) < GST_EVENT_CUSTOM_UPSTREAM;
2238
2239     GST_WARNING_OBJECT (queue, "%s%s event can't be added to temp file: "
2240         "%" GST_PTR_FORMAT, is_custom ? "Unexpected " : "",
2241         GST_EVENT_TYPE_NAME (item), GST_EVENT_CAST (item));
2242     gst_event_unref (GST_EVENT_CAST (item));
2243     return;
2244   }
2245 }
2246
2247 /* dequeue an item from the queue and update level stats */
2248 static GstMiniObject *
2249 gst_queue2_locked_dequeue (GstQueue2 * queue, GstQueue2ItemType * item_type)
2250 {
2251   GstMiniObject *item;
2252
2253   if (!QUEUE_IS_USING_QUEUE (queue)) {
2254     item = gst_queue2_read_item_from_file (queue);
2255   } else {
2256     GstQueue2Item *qitem = g_queue_pop_head (&queue->queue);
2257
2258     if (qitem == NULL)
2259       goto no_item;
2260
2261     item = qitem->item;
2262     g_slice_free (GstQueue2Item, qitem);
2263   }
2264
2265   if (item == NULL)
2266     goto no_item;
2267
2268   if (GST_IS_BUFFER (item)) {
2269     GstBuffer *buffer;
2270     guint size;
2271
2272     buffer = GST_BUFFER_CAST (item);
2273     size = gst_buffer_get_size (buffer);
2274     *item_type = GST_QUEUE2_ITEM_TYPE_BUFFER;
2275
2276     GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2277         "retrieved buffer %p from queue", buffer);
2278
2279     if (QUEUE_IS_USING_QUEUE (queue)) {
2280       queue->cur_level.buffers--;
2281       queue->cur_level.bytes -= size;
2282     }
2283     queue->bytes_out += size;
2284
2285     apply_buffer (queue, buffer, &queue->src_segment, size, FALSE);
2286     /* update the byterate stats */
2287     update_out_rates (queue);
2288     /* update the buffering */
2289     if (queue->use_buffering)
2290       update_buffering (queue);
2291
2292   } else if (GST_IS_EVENT (item)) {
2293     GstEvent *event = GST_EVENT_CAST (item);
2294
2295     *item_type = GST_QUEUE2_ITEM_TYPE_EVENT;
2296
2297     GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2298         "retrieved event %p from queue", event);
2299
2300     switch (GST_EVENT_TYPE (event)) {
2301       case GST_EVENT_EOS:
2302         /* queue is empty now that we dequeued the EOS */
2303         GST_QUEUE2_CLEAR_LEVEL (queue->cur_level);
2304         break;
2305       case GST_EVENT_SEGMENT:
2306         apply_segment (queue, event, &queue->src_segment, FALSE);
2307         break;
2308       case GST_EVENT_GAP:
2309         apply_gap (queue, event, &queue->src_segment, FALSE);
2310         break;
2311       default:
2312         break;
2313     }
2314   } else if (GST_IS_BUFFER_LIST (item)) {
2315     GstBufferList *buffer_list;
2316     guint size = 0;
2317
2318     buffer_list = GST_BUFFER_LIST_CAST (item);
2319     gst_buffer_list_foreach (buffer_list, buffer_list_calc_size, &size);
2320     *item_type = GST_QUEUE2_ITEM_TYPE_BUFFER_LIST;
2321
2322     GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2323         "retrieved buffer list %p from queue", buffer_list);
2324
2325     if (QUEUE_IS_USING_QUEUE (queue)) {
2326       queue->cur_level.buffers -= gst_buffer_list_length (buffer_list);
2327       queue->cur_level.bytes -= size;
2328     }
2329     queue->bytes_out += size;
2330
2331     apply_buffer_list (queue, buffer_list, &queue->src_segment, FALSE);
2332     /* update the byterate stats */
2333     update_out_rates (queue);
2334     /* update the buffering */
2335     if (queue->use_buffering)
2336       update_buffering (queue);
2337   } else if (GST_IS_QUERY (item)) {
2338     GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2339         "retrieved query %p from queue", item);
2340     *item_type = GST_QUEUE2_ITEM_TYPE_QUERY;
2341   } else {
2342     g_warning
2343         ("Unexpected item %p dequeued from queue %s (refcounting problem?)",
2344         item, GST_OBJECT_NAME (queue));
2345     item = NULL;
2346     *item_type = GST_QUEUE2_ITEM_TYPE_UNKNOWN;
2347   }
2348   GST_QUEUE2_SIGNAL_DEL (queue);
2349
2350   return item;
2351
2352   /* ERRORS */
2353 no_item:
2354   {
2355     GST_CAT_LOG_OBJECT (queue_dataflow, queue, "the queue is empty");
2356     return NULL;
2357   }
2358 }
2359
2360 static gboolean
2361 gst_queue2_handle_sink_event (GstPad * pad, GstObject * parent,
2362     GstEvent * event)
2363 {
2364   gboolean ret = TRUE;
2365   GstQueue2 *queue;
2366
2367   queue = GST_QUEUE2 (parent);
2368
2369   switch (GST_EVENT_TYPE (event)) {
2370     case GST_EVENT_FLUSH_START:
2371     {
2372       GST_CAT_LOG_OBJECT (queue_dataflow, queue, "received flush start event");
2373       if (GST_PAD_MODE (queue->srcpad) == GST_PAD_MODE_PUSH) {
2374         /* forward event */
2375         ret = gst_pad_push_event (queue->srcpad, event);
2376
2377         /* now unblock the chain function */
2378         GST_QUEUE2_MUTEX_LOCK (queue);
2379         queue->srcresult = GST_FLOW_FLUSHING;
2380         queue->sinkresult = GST_FLOW_FLUSHING;
2381         /* unblock the loop and chain functions */
2382         GST_QUEUE2_SIGNAL_ADD (queue);
2383         GST_QUEUE2_SIGNAL_DEL (queue);
2384         queue->last_query = FALSE;
2385         g_cond_signal (&queue->query_handled);
2386         GST_QUEUE2_MUTEX_UNLOCK (queue);
2387
2388         /* make sure it pauses, this should happen since we sent
2389          * flush_start downstream. */
2390         gst_pad_pause_task (queue->srcpad);
2391         GST_CAT_LOG_OBJECT (queue_dataflow, queue, "loop stopped");
2392       } else {
2393         GST_QUEUE2_MUTEX_LOCK (queue);
2394         /* flush the sink pad */
2395         queue->sinkresult = GST_FLOW_FLUSHING;
2396         GST_QUEUE2_SIGNAL_DEL (queue);
2397         queue->last_query = FALSE;
2398         g_cond_signal (&queue->query_handled);
2399         GST_QUEUE2_MUTEX_UNLOCK (queue);
2400
2401         gst_event_unref (event);
2402       }
2403       break;
2404     }
2405     case GST_EVENT_FLUSH_STOP:
2406     {
2407       GST_CAT_LOG_OBJECT (queue_dataflow, queue, "received flush stop event");
2408
2409       if (GST_PAD_MODE (queue->srcpad) == GST_PAD_MODE_PUSH) {
2410         /* forward event */
2411         ret = gst_pad_push_event (queue->srcpad, event);
2412
2413         GST_QUEUE2_MUTEX_LOCK (queue);
2414         gst_queue2_locked_flush (queue, FALSE, TRUE);
2415         queue->srcresult = GST_FLOW_OK;
2416         queue->sinkresult = GST_FLOW_OK;
2417         queue->is_eos = FALSE;
2418         queue->unexpected = FALSE;
2419         queue->seeking = FALSE;
2420         queue->src_tags_bitrate = queue->sink_tags_bitrate = 0;
2421         /* reset rate counters */
2422         reset_rate_timer (queue);
2423         gst_pad_start_task (queue->srcpad, (GstTaskFunction) gst_queue2_loop,
2424             queue->srcpad, NULL);
2425         GST_QUEUE2_MUTEX_UNLOCK (queue);
2426       } else {
2427         GST_QUEUE2_MUTEX_LOCK (queue);
2428         queue->segment_event_received = FALSE;
2429         queue->is_eos = FALSE;
2430         queue->unexpected = FALSE;
2431         queue->sinkresult = GST_FLOW_OK;
2432         queue->seeking = FALSE;
2433         queue->src_tags_bitrate = queue->sink_tags_bitrate = 0;
2434         GST_QUEUE2_MUTEX_UNLOCK (queue);
2435
2436         gst_event_unref (event);
2437       }
2438       break;
2439     }
2440     case GST_EVENT_TAG:{
2441       if (queue->use_tags_bitrate) {
2442         GstTagList *tags;
2443         guint bitrate;
2444
2445         gst_event_parse_tag (event, &tags);
2446         if (gst_tag_list_get_uint (tags, GST_TAG_BITRATE, &bitrate) ||
2447             gst_tag_list_get_uint (tags, GST_TAG_NOMINAL_BITRATE, &bitrate)) {
2448           GST_QUEUE2_MUTEX_LOCK (queue);
2449           queue->sink_tags_bitrate = bitrate;
2450           GST_QUEUE2_MUTEX_UNLOCK (queue);
2451           GST_LOG_OBJECT (queue, "Sink pad bitrate from tags now %u", bitrate);
2452         }
2453       }
2454       /* Fall-through */
2455     }
2456     default:
2457       if (GST_EVENT_IS_SERIALIZED (event)) {
2458         /* serialized events go in the queue */
2459         GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->sinkresult, out_flushing);
2460         if (queue->srcresult != GST_FLOW_OK) {
2461           /* Errors in sticky event pushing are no problem and ignored here
2462            * as they will cause more meaningful errors during data flow.
2463            * For EOS events, that are not followed by data flow, we still
2464            * return FALSE here though and report an error.
2465            */
2466           if (!GST_EVENT_IS_STICKY (event)) {
2467             goto out_flow_error;
2468           } else if (GST_EVENT_TYPE (event) == GST_EVENT_EOS) {
2469             if (queue->srcresult == GST_FLOW_NOT_LINKED
2470                 || queue->srcresult < GST_FLOW_EOS) {
2471               GST_ELEMENT_ERROR (queue, STREAM, FAILED,
2472                   (_("Internal data flow error.")),
2473                   ("streaming task paused, reason %s (%d)",
2474                       gst_flow_get_name (queue->srcresult), queue->srcresult));
2475             }
2476             goto out_flow_error;
2477           }
2478         }
2479         /* refuse more events on EOS */
2480         if (queue->is_eos)
2481           goto out_eos;
2482         gst_queue2_locked_enqueue (queue, event, GST_QUEUE2_ITEM_TYPE_EVENT);
2483         GST_QUEUE2_MUTEX_UNLOCK (queue);
2484         gst_queue2_post_buffering (queue);
2485       } else {
2486         /* non-serialized events are passed upstream. */
2487         ret = gst_pad_push_event (queue->srcpad, event);
2488       }
2489       break;
2490   }
2491   return ret;
2492
2493   /* ERRORS */
2494 out_flushing:
2495   {
2496     GST_DEBUG_OBJECT (queue, "refusing event, we are flushing");
2497     GST_QUEUE2_MUTEX_UNLOCK (queue);
2498     gst_event_unref (event);
2499     return FALSE;
2500   }
2501 out_eos:
2502   {
2503     GST_DEBUG_OBJECT (queue, "refusing event, we are EOS");
2504     GST_QUEUE2_MUTEX_UNLOCK (queue);
2505     gst_event_unref (event);
2506     return FALSE;
2507   }
2508 out_flow_error:
2509   {
2510     GST_LOG_OBJECT (queue,
2511         "refusing event, we have a downstream flow error: %s",
2512         gst_flow_get_name (queue->srcresult));
2513     GST_QUEUE2_MUTEX_UNLOCK (queue);
2514     gst_event_unref (event);
2515     return FALSE;
2516   }
2517 }
2518
2519 static gboolean
2520 gst_queue2_handle_sink_query (GstPad * pad, GstObject * parent,
2521     GstQuery * query)
2522 {
2523   GstQueue2 *queue;
2524   gboolean res;
2525
2526   queue = GST_QUEUE2 (parent);
2527
2528   switch (GST_QUERY_TYPE (query)) {
2529     default:
2530       if (GST_QUERY_IS_SERIALIZED (query)) {
2531         GST_CAT_LOG_OBJECT (queue_dataflow, queue, "received query %p", query);
2532         /* serialized events go in the queue. We need to be certain that we
2533          * don't cause deadlocks waiting for the query return value. We check if
2534          * the queue is empty (nothing is blocking downstream and the query can
2535          * be pushed for sure) or we are not buffering. If we are buffering,
2536          * the pipeline waits to unblock downstream until our queue fills up
2537          * completely, which can not happen if we block on the query..
2538          * Therefore we only potentially block when we are not buffering. */
2539         GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->sinkresult, out_flushing);
2540         if (QUEUE_IS_USING_QUEUE (queue) && (gst_queue2_is_empty (queue)
2541                 || !queue->use_buffering)) {
2542           if (!g_atomic_int_get (&queue->downstream_may_block)) {
2543             gst_queue2_locked_enqueue (queue, query,
2544                 GST_QUEUE2_ITEM_TYPE_QUERY);
2545
2546             STATUS (queue, queue->sinkpad, "wait for QUERY");
2547             g_cond_wait (&queue->query_handled, &queue->qlock);
2548             if (queue->sinkresult != GST_FLOW_OK)
2549               goto out_flushing;
2550             res = queue->last_query;
2551           } else {
2552             GST_DEBUG_OBJECT (queue, "refusing query, downstream might block");
2553             res = FALSE;
2554           }
2555         } else {
2556           GST_DEBUG_OBJECT (queue,
2557               "refusing query, we are not using the queue");
2558           res = FALSE;
2559         }
2560         GST_QUEUE2_MUTEX_UNLOCK (queue);
2561         gst_queue2_post_buffering (queue);
2562       } else {
2563         res = gst_pad_query_default (pad, parent, query);
2564       }
2565       break;
2566   }
2567   return res;
2568
2569   /* ERRORS */
2570 out_flushing:
2571   {
2572     GST_DEBUG_OBJECT (queue, "refusing query, we are flushing");
2573     GST_QUEUE2_MUTEX_UNLOCK (queue);
2574     return FALSE;
2575   }
2576 }
2577
2578 static gboolean
2579 gst_queue2_is_empty (GstQueue2 * queue)
2580 {
2581   /* never empty on EOS */
2582   if (queue->is_eos)
2583     return FALSE;
2584
2585   if (!QUEUE_IS_USING_QUEUE (queue) && queue->current) {
2586     return queue->current->writing_pos <= queue->current->max_reading_pos;
2587   } else {
2588     if (queue->queue.length == 0)
2589       return TRUE;
2590   }
2591
2592   return FALSE;
2593 }
2594
2595 static gboolean
2596 gst_queue2_is_filled (GstQueue2 * queue)
2597 {
2598   gboolean res;
2599
2600   /* always filled on EOS */
2601   if (queue->is_eos)
2602     return TRUE;
2603
2604 #define CHECK_FILLED(format,alt_max) ((queue->max_level.format) > 0 && \
2605     (queue->cur_level.format) >= ((alt_max) ? \
2606       MIN ((queue->max_level.format), (alt_max)) : (queue->max_level.format)))
2607
2608   /* if using a ring buffer we're filled if all ring buffer space is used
2609    * _by the current range_ */
2610   if (QUEUE_IS_USING_RING_BUFFER (queue)) {
2611     guint64 rb_size = queue->ring_buffer_max_size;
2612     GST_DEBUG_OBJECT (queue,
2613         "max bytes %u, rb size %" G_GUINT64_FORMAT ", cur bytes %u",
2614         queue->max_level.bytes, rb_size, queue->cur_level.bytes);
2615     return CHECK_FILLED (bytes, rb_size);
2616   }
2617
2618   /* if using file, we're never filled if we don't have EOS */
2619   if (QUEUE_IS_USING_TEMP_FILE (queue))
2620     return FALSE;
2621
2622   /* we are never filled when we have no buffers at all */
2623   if (queue->cur_level.buffers == 0)
2624     return FALSE;
2625
2626   /* we are filled if one of the current levels exceeds the max */
2627   res = CHECK_FILLED (buffers, 0) || CHECK_FILLED (bytes, 0)
2628       || CHECK_FILLED (time, 0);
2629
2630   /* if we need to, use the rate estimate to check against the max time we are
2631    * allowed to queue */
2632   if (queue->use_rate_estimate)
2633     res |= CHECK_FILLED (rate_time, 0);
2634
2635 #undef CHECK_FILLED
2636   return res;
2637 }
2638
2639 static GstFlowReturn
2640 gst_queue2_chain_buffer_or_buffer_list (GstQueue2 * queue,
2641     GstMiniObject * item, GstQueue2ItemType item_type)
2642 {
2643   /* we have to lock the queue since we span threads */
2644   GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->sinkresult, out_flushing);
2645   /* when we received EOS, we refuse more data */
2646   if (queue->is_eos)
2647     goto out_eos;
2648   /* when we received unexpected from downstream, refuse more buffers */
2649   if (queue->unexpected)
2650     goto out_unexpected;
2651
2652   /* while we didn't receive the newsegment, we're seeking and we skip data */
2653   if (queue->seeking)
2654     goto out_seeking;
2655
2656   if (!gst_queue2_wait_free_space (queue))
2657     goto out_flushing;
2658
2659   /* put buffer in queue now */
2660   gst_queue2_locked_enqueue (queue, item, item_type);
2661   GST_QUEUE2_MUTEX_UNLOCK (queue);
2662   gst_queue2_post_buffering (queue);
2663
2664   return GST_FLOW_OK;
2665
2666   /* special conditions */
2667 out_flushing:
2668   {
2669     GstFlowReturn ret = queue->sinkresult;
2670
2671     GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2672         "exit because task paused, reason: %s", gst_flow_get_name (ret));
2673     GST_QUEUE2_MUTEX_UNLOCK (queue);
2674     gst_mini_object_unref (item);
2675
2676     return ret;
2677   }
2678 out_eos:
2679   {
2680     GST_CAT_LOG_OBJECT (queue_dataflow, queue, "exit because we received EOS");
2681     GST_QUEUE2_MUTEX_UNLOCK (queue);
2682     gst_mini_object_unref (item);
2683
2684     return GST_FLOW_EOS;
2685   }
2686 out_seeking:
2687   {
2688     GST_CAT_LOG_OBJECT (queue_dataflow, queue, "exit because we are seeking");
2689     GST_QUEUE2_MUTEX_UNLOCK (queue);
2690     gst_mini_object_unref (item);
2691
2692     return GST_FLOW_OK;
2693   }
2694 out_unexpected:
2695   {
2696     GST_CAT_LOG_OBJECT (queue_dataflow, queue, "exit because we received EOS");
2697     GST_QUEUE2_MUTEX_UNLOCK (queue);
2698     gst_mini_object_unref (item);
2699
2700     return GST_FLOW_EOS;
2701   }
2702 }
2703
2704 static GstFlowReturn
2705 gst_queue2_chain (GstPad * pad, GstObject * parent, GstBuffer * buffer)
2706 {
2707   GstQueue2 *queue;
2708
2709   queue = GST_QUEUE2 (parent);
2710
2711   GST_CAT_LOG_OBJECT (queue_dataflow, queue, "received buffer %p of "
2712       "size %" G_GSIZE_FORMAT ", time %" GST_TIME_FORMAT ", duration %"
2713       GST_TIME_FORMAT, buffer, gst_buffer_get_size (buffer),
2714       GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (buffer)),
2715       GST_TIME_ARGS (GST_BUFFER_DURATION (buffer)));
2716
2717   return gst_queue2_chain_buffer_or_buffer_list (queue,
2718       GST_MINI_OBJECT_CAST (buffer), GST_QUEUE2_ITEM_TYPE_BUFFER);
2719 }
2720
2721 static GstFlowReturn
2722 gst_queue2_chain_list (GstPad * pad, GstObject * parent,
2723     GstBufferList * buffer_list)
2724 {
2725   GstQueue2 *queue;
2726
2727   queue = GST_QUEUE2 (parent);
2728
2729   GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2730       "received buffer list %p", buffer_list);
2731
2732   return gst_queue2_chain_buffer_or_buffer_list (queue,
2733       GST_MINI_OBJECT_CAST (buffer_list), GST_QUEUE2_ITEM_TYPE_BUFFER_LIST);
2734 }
2735
2736 static GstMiniObject *
2737 gst_queue2_dequeue_on_eos (GstQueue2 * queue, GstQueue2ItemType * item_type)
2738 {
2739   GstMiniObject *data;
2740
2741   GST_CAT_LOG_OBJECT (queue_dataflow, queue, "got EOS from downstream");
2742
2743   /* stop pushing buffers, we dequeue all items until we see an item that we
2744    * can push again, which is EOS or SEGMENT. If there is nothing in the
2745    * queue we can push, we set a flag to make the sinkpad refuse more
2746    * buffers with an EOS return value until we receive something
2747    * pushable again or we get flushed. */
2748   while ((data = gst_queue2_locked_dequeue (queue, item_type))) {
2749     if (*item_type == GST_QUEUE2_ITEM_TYPE_BUFFER) {
2750       GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2751           "dropping EOS buffer %p", data);
2752       gst_buffer_unref (GST_BUFFER_CAST (data));
2753     } else if (*item_type == GST_QUEUE2_ITEM_TYPE_EVENT) {
2754       GstEvent *event = GST_EVENT_CAST (data);
2755       GstEventType type = GST_EVENT_TYPE (event);
2756
2757       if (type == GST_EVENT_EOS || type == GST_EVENT_SEGMENT) {
2758         /* we found a pushable item in the queue, push it out */
2759         GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2760             "pushing pushable event %s after EOS", GST_EVENT_TYPE_NAME (event));
2761         return data;
2762       }
2763       GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2764           "dropping EOS event %p", event);
2765       gst_event_unref (event);
2766     } else if (*item_type == GST_QUEUE2_ITEM_TYPE_BUFFER_LIST) {
2767       GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2768           "dropping EOS buffer list %p", data);
2769       gst_buffer_list_unref (GST_BUFFER_LIST_CAST (data));
2770     } else if (*item_type == GST_QUEUE2_ITEM_TYPE_QUERY) {
2771       queue->last_query = FALSE;
2772       g_cond_signal (&queue->query_handled);
2773       GST_CAT_LOG_OBJECT (queue_dataflow, queue, "dropping EOS query %p", data);
2774     }
2775   }
2776   /* no more items in the queue. Set the unexpected flag so that upstream
2777    * make us refuse any more buffers on the sinkpad. Since we will still
2778    * accept EOS and SEGMENT we return _FLOW_OK to the caller so that the
2779    * task function does not shut down. */
2780   queue->unexpected = TRUE;
2781   return NULL;
2782 }
2783
2784 /* dequeue an item from the queue an push it downstream. This functions returns
2785  * the result of the push. */
2786 static GstFlowReturn
2787 gst_queue2_push_one (GstQueue2 * queue)
2788 {
2789   GstFlowReturn result = queue->srcresult;
2790   GstMiniObject *data;
2791   GstQueue2ItemType item_type;
2792
2793   data = gst_queue2_locked_dequeue (queue, &item_type);
2794   if (data == NULL)
2795     goto no_item;
2796
2797 next:
2798   STATUS (queue, queue->srcpad, "We have something dequeud");
2799   g_atomic_int_set (&queue->downstream_may_block,
2800       item_type == GST_QUEUE2_ITEM_TYPE_BUFFER ||
2801       item_type == GST_QUEUE2_ITEM_TYPE_BUFFER_LIST);
2802   GST_QUEUE2_MUTEX_UNLOCK (queue);
2803   gst_queue2_post_buffering (queue);
2804
2805   if (item_type == GST_QUEUE2_ITEM_TYPE_BUFFER) {
2806     GstBuffer *buffer;
2807
2808     buffer = GST_BUFFER_CAST (data);
2809
2810     result = gst_pad_push (queue->srcpad, buffer);
2811     g_atomic_int_set (&queue->downstream_may_block, 0);
2812
2813     /* need to check for srcresult here as well */
2814     GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->srcresult, out_flushing);
2815     if (result == GST_FLOW_EOS) {
2816       data = gst_queue2_dequeue_on_eos (queue, &item_type);
2817       if (data != NULL)
2818         goto next;
2819       /* Since we will still accept EOS and SEGMENT we return _FLOW_OK
2820        * to the caller so that the task function does not shut down */
2821       result = GST_FLOW_OK;
2822     }
2823   } else if (item_type == GST_QUEUE2_ITEM_TYPE_EVENT) {
2824     GstEvent *event = GST_EVENT_CAST (data);
2825     GstEventType type = GST_EVENT_TYPE (event);
2826
2827     if (type == GST_EVENT_TAG) {
2828       if (queue->use_tags_bitrate) {
2829         GstTagList *tags;
2830         guint bitrate;
2831
2832         gst_event_parse_tag (event, &tags);
2833         if (gst_tag_list_get_uint (tags, GST_TAG_BITRATE, &bitrate) ||
2834             gst_tag_list_get_uint (tags, GST_TAG_NOMINAL_BITRATE, &bitrate)) {
2835           GST_QUEUE2_MUTEX_LOCK (queue);
2836           queue->src_tags_bitrate = bitrate;
2837           GST_QUEUE2_MUTEX_UNLOCK (queue);
2838           GST_LOG_OBJECT (queue, "src pad bitrate from tags now %u", bitrate);
2839         }
2840       }
2841     }
2842
2843     gst_pad_push_event (queue->srcpad, event);
2844
2845     /* if we're EOS, return EOS so that the task pauses. */
2846     if (type == GST_EVENT_EOS) {
2847       GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2848           "pushed EOS event %p, return EOS", event);
2849       result = GST_FLOW_EOS;
2850     }
2851
2852     GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->srcresult, out_flushing);
2853   } else if (item_type == GST_QUEUE2_ITEM_TYPE_BUFFER_LIST) {
2854     GstBufferList *buffer_list;
2855
2856     buffer_list = GST_BUFFER_LIST_CAST (data);
2857
2858     result = gst_pad_push_list (queue->srcpad, buffer_list);
2859     g_atomic_int_set (&queue->downstream_may_block, 0);
2860
2861     /* need to check for srcresult here as well */
2862     GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->srcresult, out_flushing);
2863     if (result == GST_FLOW_EOS) {
2864       data = gst_queue2_dequeue_on_eos (queue, &item_type);
2865       if (data != NULL)
2866         goto next;
2867       /* Since we will still accept EOS and SEGMENT we return _FLOW_OK
2868        * to the caller so that the task function does not shut down */
2869       result = GST_FLOW_OK;
2870     }
2871   } else if (item_type == GST_QUEUE2_ITEM_TYPE_QUERY) {
2872     GstQuery *query = GST_QUERY_CAST (data);
2873
2874     GST_LOG_OBJECT (queue->srcpad, "Peering query %p", query);
2875     queue->last_query = gst_pad_peer_query (queue->srcpad, query);
2876     GST_LOG_OBJECT (queue->srcpad, "Peered query");
2877     GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2878         "did query %p, return %d", query, queue->last_query);
2879     g_cond_signal (&queue->query_handled);
2880     GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->srcresult, out_flushing);
2881     result = GST_FLOW_OK;
2882   }
2883   return result;
2884
2885   /* ERRORS */
2886 no_item:
2887   {
2888     GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2889         "exit because we have no item in the queue");
2890     return GST_FLOW_ERROR;
2891   }
2892 out_flushing:
2893   {
2894     GST_CAT_LOG_OBJECT (queue_dataflow, queue, "exit because we are flushing");
2895     return GST_FLOW_FLUSHING;
2896   }
2897 }
2898
2899 /* called repeatedly with @pad as the source pad. This function should push out
2900  * data to the peer element. */
2901 static void
2902 gst_queue2_loop (GstPad * pad)
2903 {
2904   GstQueue2 *queue;
2905   GstFlowReturn ret;
2906
2907   queue = GST_QUEUE2 (GST_PAD_PARENT (pad));
2908
2909   /* have to lock for thread-safety */
2910   GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->srcresult, out_flushing);
2911
2912   if (gst_queue2_is_empty (queue)) {
2913     gboolean started;
2914
2915     /* pause the timer while we wait. The fact that we are waiting does not mean
2916      * the byterate on the output pad is lower */
2917     if ((started = queue->out_timer_started))
2918       g_timer_stop (queue->out_timer);
2919
2920     GST_CAT_DEBUG_OBJECT (queue_dataflow, queue,
2921         "queue is empty, waiting for new data");
2922     do {
2923       /* Wait for data to be available, we could be unlocked because of a flush. */
2924       GST_QUEUE2_WAIT_ADD_CHECK (queue, queue->srcresult, out_flushing);
2925     }
2926     while (gst_queue2_is_empty (queue));
2927
2928     /* and continue if we were running before */
2929     if (started)
2930       g_timer_continue (queue->out_timer);
2931   }
2932   ret = gst_queue2_push_one (queue);
2933   queue->srcresult = ret;
2934   queue->sinkresult = ret;
2935   if (ret != GST_FLOW_OK)
2936     goto out_flushing;
2937
2938   GST_QUEUE2_MUTEX_UNLOCK (queue);
2939   gst_queue2_post_buffering (queue);
2940
2941   return;
2942
2943   /* ERRORS */
2944 out_flushing:
2945   {
2946     gboolean eos = queue->is_eos;
2947     GstFlowReturn ret = queue->srcresult;
2948
2949     gst_pad_pause_task (queue->srcpad);
2950     if (ret == GST_FLOW_FLUSHING) {
2951       gst_queue2_locked_flush (queue, FALSE, FALSE);
2952     } else {
2953       GST_QUEUE2_SIGNAL_DEL (queue);
2954       queue->last_query = FALSE;
2955       g_cond_signal (&queue->query_handled);
2956     }
2957     GST_QUEUE2_MUTEX_UNLOCK (queue);
2958     GST_CAT_LOG_OBJECT (queue_dataflow, queue,
2959         "pause task, reason:  %s", gst_flow_get_name (queue->srcresult));
2960     /* let app know about us giving up if upstream is not expected to do so */
2961     /* EOS is already taken care of elsewhere */
2962     if (eos && (ret == GST_FLOW_NOT_LINKED || ret < GST_FLOW_EOS)) {
2963       GST_ELEMENT_ERROR (queue, STREAM, FAILED,
2964           (_("Internal data flow error.")),
2965           ("streaming task paused, reason %s (%d)",
2966               gst_flow_get_name (ret), ret));
2967       gst_pad_push_event (queue->srcpad, gst_event_new_eos ());
2968     }
2969     return;
2970   }
2971 }
2972
2973 static gboolean
2974 gst_queue2_handle_src_event (GstPad * pad, GstObject * parent, GstEvent * event)
2975 {
2976   gboolean res = TRUE;
2977   GstQueue2 *queue = GST_QUEUE2 (parent);
2978
2979 #ifndef GST_DISABLE_GST_DEBUG
2980   GST_CAT_DEBUG_OBJECT (queue_dataflow, queue, "got event %p (%s)",
2981       event, GST_EVENT_TYPE_NAME (event));
2982 #endif
2983
2984   switch (GST_EVENT_TYPE (event)) {
2985     case GST_EVENT_FLUSH_START:
2986       if (QUEUE_IS_USING_QUEUE (queue)) {
2987         /* just forward upstream */
2988         res = gst_pad_push_event (queue->sinkpad, event);
2989       } else {
2990         /* now unblock the getrange function */
2991         GST_QUEUE2_MUTEX_LOCK (queue);
2992         GST_DEBUG_OBJECT (queue, "flushing");
2993         queue->srcresult = GST_FLOW_FLUSHING;
2994         GST_QUEUE2_SIGNAL_ADD (queue);
2995         GST_QUEUE2_MUTEX_UNLOCK (queue);
2996
2997         /* when using a temp file, we eat the event */
2998         res = TRUE;
2999         gst_event_unref (event);
3000       }
3001       break;
3002     case GST_EVENT_FLUSH_STOP:
3003       if (QUEUE_IS_USING_QUEUE (queue)) {
3004         /* just forward upstream */
3005         res = gst_pad_push_event (queue->sinkpad, event);
3006       } else {
3007         /* now unblock the getrange function */
3008         GST_QUEUE2_MUTEX_LOCK (queue);
3009         queue->srcresult = GST_FLOW_OK;
3010         GST_QUEUE2_MUTEX_UNLOCK (queue);
3011
3012         /* when using a temp file, we eat the event */
3013         res = TRUE;
3014         gst_event_unref (event);
3015       }
3016       break;
3017     case GST_EVENT_RECONFIGURE:
3018       GST_QUEUE2_MUTEX_LOCK (queue);
3019       /* assume downstream is linked now and try to push again */
3020       if (queue->srcresult == GST_FLOW_NOT_LINKED) {
3021         queue->srcresult = GST_FLOW_OK;
3022         queue->sinkresult = GST_FLOW_OK;
3023         if (GST_PAD_MODE (pad) == GST_PAD_MODE_PUSH) {
3024           gst_pad_start_task (pad, (GstTaskFunction) gst_queue2_loop, pad,
3025               NULL);
3026         }
3027       }
3028       GST_QUEUE2_MUTEX_UNLOCK (queue);
3029
3030       res = gst_pad_push_event (queue->sinkpad, event);
3031       break;
3032     default:
3033       res = gst_pad_push_event (queue->sinkpad, event);
3034       break;
3035   }
3036
3037   return res;
3038 }
3039
3040 static gboolean
3041 gst_queue2_handle_src_query (GstPad * pad, GstObject * parent, GstQuery * query)
3042 {
3043   GstQueue2 *queue;
3044
3045   queue = GST_QUEUE2 (parent);
3046
3047   switch (GST_QUERY_TYPE (query)) {
3048     case GST_QUERY_POSITION:
3049     {
3050       gint64 peer_pos;
3051       GstFormat format;
3052
3053       if (!gst_pad_peer_query (queue->sinkpad, query))
3054         goto peer_failed;
3055
3056       /* get peer position */
3057       gst_query_parse_position (query, &format, &peer_pos);
3058
3059       /* FIXME: this code assumes that there's no discont in the queue */
3060       switch (format) {
3061         case GST_FORMAT_BYTES:
3062           peer_pos -= queue->cur_level.bytes;
3063           break;
3064         case GST_FORMAT_TIME:
3065           peer_pos -= queue->cur_level.time;
3066           break;
3067         default:
3068           GST_WARNING_OBJECT (queue, "dropping query in %s format, don't "
3069               "know how to adjust value", gst_format_get_name (format));
3070           return FALSE;
3071       }
3072       /* set updated position */
3073       gst_query_set_position (query, format, peer_pos);
3074       break;
3075     }
3076     case GST_QUERY_DURATION:
3077     {
3078       GST_DEBUG_OBJECT (queue, "doing peer query");
3079
3080       if (!gst_pad_peer_query (queue->sinkpad, query))
3081         goto peer_failed;
3082
3083       GST_DEBUG_OBJECT (queue, "peer query success");
3084       break;
3085     }
3086     case GST_QUERY_BUFFERING:
3087     {
3088       gint percent;
3089       gboolean is_buffering;
3090       GstBufferingMode mode;
3091       gint avg_in, avg_out;
3092       gint64 buffering_left;
3093
3094       GST_DEBUG_OBJECT (queue, "query buffering");
3095
3096       get_buffering_percent (queue, &is_buffering, &percent);
3097       gst_query_set_buffering_percent (query, is_buffering, percent);
3098
3099       get_buffering_stats (queue, percent, &mode, &avg_in, &avg_out,
3100           &buffering_left);
3101       gst_query_set_buffering_stats (query, mode, avg_in, avg_out,
3102           buffering_left);
3103
3104       if (!QUEUE_IS_USING_QUEUE (queue)) {
3105         /* add ranges for download and ringbuffer buffering */
3106         GstFormat format;
3107         gint64 start, stop, range_start, range_stop;
3108         guint64 writing_pos;
3109         gint64 estimated_total;
3110         gint64 duration;
3111         gboolean peer_res, is_eos;
3112         GstQueue2Range *queued_ranges;
3113
3114         /* we need a current download region */
3115         if (queue->current == NULL)
3116           return FALSE;
3117
3118         writing_pos = queue->current->writing_pos;
3119         is_eos = queue->is_eos;
3120
3121         if (is_eos) {
3122           /* we're EOS, we know the duration in bytes now */
3123           peer_res = TRUE;
3124           duration = writing_pos;
3125         } else {
3126           /* get duration of upstream in bytes */
3127           peer_res = gst_pad_peer_query_duration (queue->sinkpad,
3128               GST_FORMAT_BYTES, &duration);
3129         }
3130
3131         GST_DEBUG_OBJECT (queue, "percent %d, duration %" G_GINT64_FORMAT
3132             ", writing %" G_GINT64_FORMAT, percent, duration, writing_pos);
3133
3134         /* calculate remaining and total download time */
3135         if (peer_res && avg_in > 0.0)
3136           estimated_total = ((duration - writing_pos) * 1000) / avg_in;
3137         else
3138           estimated_total = -1;
3139
3140         GST_DEBUG_OBJECT (queue, "estimated-total %" G_GINT64_FORMAT,
3141             estimated_total);
3142
3143         gst_query_parse_buffering_range (query, &format, NULL, NULL, NULL);
3144
3145         switch (format) {
3146           case GST_FORMAT_PERCENT:
3147             /* we need duration */
3148             if (!peer_res)
3149               goto peer_failed;
3150
3151             start = 0;
3152             /* get our available data relative to the duration */
3153             if (duration != -1)
3154               stop =
3155                   gst_util_uint64_scale (GST_FORMAT_PERCENT_MAX, writing_pos,
3156                   duration);
3157             else
3158               stop = -1;
3159             break;
3160           case GST_FORMAT_BYTES:
3161             start = 0;
3162             stop = writing_pos;
3163             break;
3164           default:
3165             start = -1;
3166             stop = -1;
3167             break;
3168         }
3169
3170         /* fill out the buffered ranges */
3171         for (queued_ranges = queue->ranges; queued_ranges;
3172             queued_ranges = queued_ranges->next) {
3173           switch (format) {
3174             case GST_FORMAT_PERCENT:
3175               if (duration == -1) {
3176                 range_start = 0;
3177                 range_stop = 0;
3178                 break;
3179               }
3180               range_start =
3181                   gst_util_uint64_scale (GST_FORMAT_PERCENT_MAX,
3182                   queued_ranges->offset, duration);
3183               range_stop =
3184                   gst_util_uint64_scale (GST_FORMAT_PERCENT_MAX,
3185                   queued_ranges->writing_pos, duration);
3186               break;
3187             case GST_FORMAT_BYTES:
3188               range_start = queued_ranges->offset;
3189               range_stop = queued_ranges->writing_pos;
3190               break;
3191             default:
3192               range_start = -1;
3193               range_stop = -1;
3194               break;
3195           }
3196           if (range_start == range_stop)
3197             continue;
3198           GST_DEBUG_OBJECT (queue,
3199               "range starting at %" G_GINT64_FORMAT " and finishing at %"
3200               G_GINT64_FORMAT, range_start, range_stop);
3201           gst_query_add_buffering_range (query, range_start, range_stop);
3202         }
3203
3204         gst_query_set_buffering_range (query, format, start, stop,
3205             estimated_total);
3206       }
3207       break;
3208     }
3209     case GST_QUERY_SCHEDULING:
3210     {
3211       gboolean pull_mode;
3212       GstSchedulingFlags flags = 0;
3213
3214       if (!gst_pad_peer_query (queue->sinkpad, query))
3215         goto peer_failed;
3216
3217       gst_query_parse_scheduling (query, &flags, NULL, NULL, NULL);
3218
3219       /* we can operate in pull mode when we are using a tempfile */
3220       pull_mode = !QUEUE_IS_USING_QUEUE (queue);
3221
3222       if (pull_mode)
3223         flags |= GST_SCHEDULING_FLAG_SEEKABLE;
3224       gst_query_set_scheduling (query, flags, 0, -1, 0);
3225       if (pull_mode)
3226         gst_query_add_scheduling_mode (query, GST_PAD_MODE_PULL);
3227       gst_query_add_scheduling_mode (query, GST_PAD_MODE_PUSH);
3228       break;
3229     }
3230     default:
3231       /* peer handled other queries */
3232       if (!gst_pad_query_default (pad, parent, query))
3233         goto peer_failed;
3234       break;
3235   }
3236
3237   return TRUE;
3238
3239   /* ERRORS */
3240 peer_failed:
3241   {
3242     GST_DEBUG_OBJECT (queue, "failed peer query");
3243     return FALSE;
3244   }
3245 }
3246
3247 static gboolean
3248 gst_queue2_handle_query (GstElement * element, GstQuery * query)
3249 {
3250   GstQueue2 *queue = GST_QUEUE2 (element);
3251
3252   /* simply forward to the srcpad query function */
3253   return gst_queue2_handle_src_query (queue->srcpad, GST_OBJECT_CAST (element),
3254       query);
3255 }
3256
3257 static void
3258 gst_queue2_update_upstream_size (GstQueue2 * queue)
3259 {
3260   gint64 upstream_size = -1;
3261
3262   if (gst_pad_peer_query_duration (queue->sinkpad, GST_FORMAT_BYTES,
3263           &upstream_size)) {
3264     GST_INFO_OBJECT (queue, "upstream size: %" G_GINT64_FORMAT, upstream_size);
3265
3266     /* upstream_size can be negative but queue->upstream_size is unsigned.
3267      * Prevent setting negative values to it (the query can return -1) */
3268     if (upstream_size >= 0)
3269       queue->upstream_size = upstream_size;
3270     else
3271       queue->upstream_size = 0;
3272   }
3273 }
3274
3275 static GstFlowReturn
3276 gst_queue2_get_range (GstPad * pad, GstObject * parent, guint64 offset,
3277     guint length, GstBuffer ** buffer)
3278 {
3279   GstQueue2 *queue;
3280   GstFlowReturn ret;
3281
3282   queue = GST_QUEUE2_CAST (parent);
3283
3284   length = (length == -1) ? DEFAULT_BUFFER_SIZE : length;
3285   GST_QUEUE2_MUTEX_LOCK_CHECK (queue, queue->srcresult, out_flushing);
3286   offset = (offset == -1) ? queue->current->reading_pos : offset;
3287
3288   GST_DEBUG_OBJECT (queue,
3289       "Getting range: offset %" G_GUINT64_FORMAT ", length %u", offset, length);
3290
3291   /* catch any reads beyond the size of the file here to make sure queue2
3292    * doesn't send seek events beyond the size of the file upstream, since
3293    * that would confuse elements such as souphttpsrc and/or http servers.
3294    * Demuxers often just loop until EOS at the end of the file to figure out
3295    * when they've read all the end-headers or index chunks. */
3296   if (G_UNLIKELY (offset >= queue->upstream_size)) {
3297     gst_queue2_update_upstream_size (queue);
3298     if (queue->upstream_size > 0 && offset >= queue->upstream_size)
3299       goto out_unexpected;
3300   }
3301
3302   if (G_UNLIKELY (offset + length > queue->upstream_size)) {
3303     gst_queue2_update_upstream_size (queue);
3304     if (queue->upstream_size > 0 && offset + length >= queue->upstream_size) {
3305       length = queue->upstream_size - offset;
3306       GST_DEBUG_OBJECT (queue, "adjusting length downto %d", length);
3307     }
3308   }
3309
3310   /* FIXME - function will block when the range is not yet available */
3311   ret = gst_queue2_create_read (queue, offset, length, buffer);
3312   GST_QUEUE2_MUTEX_UNLOCK (queue);
3313   gst_queue2_post_buffering (queue);
3314
3315   return ret;
3316
3317   /* ERRORS */
3318 out_flushing:
3319   {
3320     ret = queue->srcresult;
3321
3322     GST_DEBUG_OBJECT (queue, "we are flushing");
3323     GST_QUEUE2_MUTEX_UNLOCK (queue);
3324     return ret;
3325   }
3326 out_unexpected:
3327   {
3328     GST_DEBUG_OBJECT (queue, "read beyond end of file");
3329     GST_QUEUE2_MUTEX_UNLOCK (queue);
3330     return GST_FLOW_EOS;
3331   }
3332 }
3333
3334 /* sink currently only operates in push mode */
3335 static gboolean
3336 gst_queue2_sink_activate_mode (GstPad * pad, GstObject * parent,
3337     GstPadMode mode, gboolean active)
3338 {
3339   gboolean result;
3340   GstQueue2 *queue;
3341
3342   queue = GST_QUEUE2 (parent);
3343
3344   switch (mode) {
3345     case GST_PAD_MODE_PUSH:
3346       if (active) {
3347         GST_QUEUE2_MUTEX_LOCK (queue);
3348         GST_DEBUG_OBJECT (queue, "activating push mode");
3349         queue->srcresult = GST_FLOW_OK;
3350         queue->sinkresult = GST_FLOW_OK;
3351         queue->is_eos = FALSE;
3352         queue->unexpected = FALSE;
3353         reset_rate_timer (queue);
3354         GST_QUEUE2_MUTEX_UNLOCK (queue);
3355       } else {
3356         /* unblock chain function */
3357         GST_QUEUE2_MUTEX_LOCK (queue);
3358         GST_DEBUG_OBJECT (queue, "deactivating push mode");
3359         queue->srcresult = GST_FLOW_FLUSHING;
3360         queue->sinkresult = GST_FLOW_FLUSHING;
3361         GST_QUEUE2_SIGNAL_DEL (queue);
3362         /* Unblock query handler */
3363         queue->last_query = FALSE;
3364         g_cond_signal (&queue->query_handled);
3365         GST_QUEUE2_MUTEX_UNLOCK (queue);
3366
3367         /* wait until it is unblocked and clean up */
3368         GST_PAD_STREAM_LOCK (pad);
3369         GST_QUEUE2_MUTEX_LOCK (queue);
3370         gst_queue2_locked_flush (queue, TRUE, FALSE);
3371         GST_QUEUE2_MUTEX_UNLOCK (queue);
3372         GST_PAD_STREAM_UNLOCK (pad);
3373       }
3374       result = TRUE;
3375       break;
3376     default:
3377       result = FALSE;
3378       break;
3379   }
3380   return result;
3381 }
3382
3383 /* src operating in push mode, we start a task on the source pad that pushes out
3384  * buffers from the queue */
3385 static gboolean
3386 gst_queue2_src_activate_push (GstPad * pad, GstObject * parent, gboolean active)
3387 {
3388   gboolean result = FALSE;
3389   GstQueue2 *queue;
3390
3391   queue = GST_QUEUE2 (parent);
3392
3393   if (active) {
3394     GST_QUEUE2_MUTEX_LOCK (queue);
3395     GST_DEBUG_OBJECT (queue, "activating push mode");
3396     queue->srcresult = GST_FLOW_OK;
3397     queue->sinkresult = GST_FLOW_OK;
3398     queue->is_eos = FALSE;
3399     queue->unexpected = FALSE;
3400     result =
3401         gst_pad_start_task (pad, (GstTaskFunction) gst_queue2_loop, pad, NULL);
3402     GST_QUEUE2_MUTEX_UNLOCK (queue);
3403   } else {
3404     /* unblock loop function */
3405     GST_QUEUE2_MUTEX_LOCK (queue);
3406     GST_DEBUG_OBJECT (queue, "deactivating push mode");
3407     queue->srcresult = GST_FLOW_FLUSHING;
3408     queue->sinkresult = GST_FLOW_FLUSHING;
3409     /* the item add signal will unblock */
3410     GST_QUEUE2_SIGNAL_ADD (queue);
3411     GST_QUEUE2_MUTEX_UNLOCK (queue);
3412
3413     /* step 2, make sure streaming finishes */
3414     result = gst_pad_stop_task (pad);
3415   }
3416
3417   return result;
3418 }
3419
3420 /* pull mode, downstream will call our getrange function */
3421 static gboolean
3422 gst_queue2_src_activate_pull (GstPad * pad, GstObject * parent, gboolean active)
3423 {
3424   gboolean result;
3425   GstQueue2 *queue;
3426
3427   queue = GST_QUEUE2 (parent);
3428
3429   if (active) {
3430     GST_QUEUE2_MUTEX_LOCK (queue);
3431     if (!QUEUE_IS_USING_QUEUE (queue)) {
3432       if (QUEUE_IS_USING_TEMP_FILE (queue)) {
3433         /* open the temp file now */
3434         result = gst_queue2_open_temp_location_file (queue);
3435       } else if (!queue->ring_buffer) {
3436         queue->ring_buffer = g_malloc (queue->ring_buffer_max_size);
3437         result = ! !queue->ring_buffer;
3438       } else {
3439         result = TRUE;
3440       }
3441
3442       GST_DEBUG_OBJECT (queue, "activating pull mode");
3443       init_ranges (queue);
3444       queue->srcresult = GST_FLOW_OK;
3445       queue->sinkresult = GST_FLOW_OK;
3446       queue->is_eos = FALSE;
3447       queue->unexpected = FALSE;
3448       queue->upstream_size = 0;
3449     } else {
3450       GST_DEBUG_OBJECT (queue, "no temp file, cannot activate pull mode");
3451       /* this is not allowed, we cannot operate in pull mode without a temp
3452        * file. */
3453       queue->srcresult = GST_FLOW_FLUSHING;
3454       queue->sinkresult = GST_FLOW_FLUSHING;
3455       result = FALSE;
3456     }
3457     GST_QUEUE2_MUTEX_UNLOCK (queue);
3458   } else {
3459     GST_QUEUE2_MUTEX_LOCK (queue);
3460     GST_DEBUG_OBJECT (queue, "deactivating pull mode");
3461     queue->srcresult = GST_FLOW_FLUSHING;
3462     queue->sinkresult = GST_FLOW_FLUSHING;
3463     /* this will unlock getrange */
3464     GST_QUEUE2_SIGNAL_ADD (queue);
3465     result = TRUE;
3466     GST_QUEUE2_MUTEX_UNLOCK (queue);
3467   }
3468
3469   return result;
3470 }
3471
3472 static gboolean
3473 gst_queue2_src_activate_mode (GstPad * pad, GstObject * parent, GstPadMode mode,
3474     gboolean active)
3475 {
3476   gboolean res;
3477
3478   switch (mode) {
3479     case GST_PAD_MODE_PULL:
3480       res = gst_queue2_src_activate_pull (pad, parent, active);
3481       break;
3482     case GST_PAD_MODE_PUSH:
3483       res = gst_queue2_src_activate_push (pad, parent, active);
3484       break;
3485     default:
3486       GST_LOG_OBJECT (pad, "unknown activation mode %d", mode);
3487       res = FALSE;
3488       break;
3489   }
3490   return res;
3491 }
3492
3493 static GstStateChangeReturn
3494 gst_queue2_change_state (GstElement * element, GstStateChange transition)
3495 {
3496   GstQueue2 *queue;
3497   GstStateChangeReturn ret = GST_STATE_CHANGE_SUCCESS;
3498
3499   queue = GST_QUEUE2 (element);
3500
3501   switch (transition) {
3502     case GST_STATE_CHANGE_NULL_TO_READY:
3503       break;
3504     case GST_STATE_CHANGE_READY_TO_PAUSED:
3505       GST_QUEUE2_MUTEX_LOCK (queue);
3506       if (!QUEUE_IS_USING_QUEUE (queue)) {
3507         if (QUEUE_IS_USING_TEMP_FILE (queue)) {
3508           if (!gst_queue2_open_temp_location_file (queue))
3509             ret = GST_STATE_CHANGE_FAILURE;
3510         } else {
3511           if (queue->ring_buffer) {
3512             g_free (queue->ring_buffer);
3513             queue->ring_buffer = NULL;
3514           }
3515           if (!(queue->ring_buffer = g_malloc (queue->ring_buffer_max_size)))
3516             ret = GST_STATE_CHANGE_FAILURE;
3517         }
3518         init_ranges (queue);
3519       }
3520       queue->segment_event_received = FALSE;
3521       queue->starting_segment = NULL;
3522       gst_event_replace (&queue->stream_start_event, NULL);
3523       GST_QUEUE2_MUTEX_UNLOCK (queue);
3524       break;
3525     case GST_STATE_CHANGE_PAUSED_TO_PLAYING:
3526       break;
3527     default:
3528       break;
3529   }
3530
3531   if (ret == GST_STATE_CHANGE_FAILURE)
3532     return ret;
3533
3534   ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
3535
3536   if (ret == GST_STATE_CHANGE_FAILURE)
3537     return ret;
3538
3539   switch (transition) {
3540     case GST_STATE_CHANGE_PLAYING_TO_PAUSED:
3541       break;
3542     case GST_STATE_CHANGE_PAUSED_TO_READY:
3543       GST_QUEUE2_MUTEX_LOCK (queue);
3544       if (!QUEUE_IS_USING_QUEUE (queue)) {
3545         if (QUEUE_IS_USING_TEMP_FILE (queue)) {
3546           gst_queue2_close_temp_location_file (queue);
3547         } else if (queue->ring_buffer) {
3548           g_free (queue->ring_buffer);
3549           queue->ring_buffer = NULL;
3550         }
3551         clean_ranges (queue);
3552       }
3553       if (queue->starting_segment != NULL) {
3554         gst_event_unref (queue->starting_segment);
3555         queue->starting_segment = NULL;
3556       }
3557       gst_event_replace (&queue->stream_start_event, NULL);
3558       GST_QUEUE2_MUTEX_UNLOCK (queue);
3559       break;
3560     case GST_STATE_CHANGE_READY_TO_NULL:
3561       break;
3562     default:
3563       break;
3564   }
3565
3566   return ret;
3567 }
3568
3569 /* changing the capacity of the queue must wake up
3570  * the _chain function, it might have more room now
3571  * to store the buffer/event in the queue */
3572 #define QUEUE_CAPACITY_CHANGE(q) \
3573   GST_QUEUE2_SIGNAL_DEL (queue); \
3574   if (queue->use_buffering)      \
3575     update_buffering (queue);
3576
3577 /* Changing the minimum required fill level must
3578  * wake up the _loop function as it might now
3579  * be able to preceed.
3580  */
3581 #define QUEUE_THRESHOLD_CHANGE(q)\
3582   GST_QUEUE2_SIGNAL_ADD (queue);
3583
3584 static void
3585 gst_queue2_set_temp_template (GstQueue2 * queue, const gchar * template)
3586 {
3587   GstState state;
3588
3589   /* the element must be stopped in order to do this */
3590   GST_OBJECT_LOCK (queue);
3591   state = GST_STATE (queue);
3592   if (state != GST_STATE_READY && state != GST_STATE_NULL)
3593     goto wrong_state;
3594   GST_OBJECT_UNLOCK (queue);
3595
3596   /* set new location */
3597   g_free (queue->temp_template);
3598   queue->temp_template = g_strdup (template);
3599
3600   return;
3601
3602 /* ERROR */
3603 wrong_state:
3604   {
3605     GST_WARNING_OBJECT (queue, "setting temp-template property in wrong state");
3606     GST_OBJECT_UNLOCK (queue);
3607   }
3608 }
3609
3610 static void
3611 gst_queue2_set_property (GObject * object,
3612     guint prop_id, const GValue * value, GParamSpec * pspec)
3613 {
3614   GstQueue2 *queue = GST_QUEUE2 (object);
3615
3616   /* someone could change levels here, and since this
3617    * affects the get/put funcs, we need to lock for safety. */
3618   GST_QUEUE2_MUTEX_LOCK (queue);
3619
3620   switch (prop_id) {
3621     case PROP_MAX_SIZE_BYTES:
3622       queue->max_level.bytes = g_value_get_uint (value);
3623       QUEUE_CAPACITY_CHANGE (queue);
3624       break;
3625     case PROP_MAX_SIZE_BUFFERS:
3626       queue->max_level.buffers = g_value_get_uint (value);
3627       QUEUE_CAPACITY_CHANGE (queue);
3628       break;
3629     case PROP_MAX_SIZE_TIME:
3630       queue->max_level.time = g_value_get_uint64 (value);
3631       /* set rate_time to the same value. We use an extra field in the level
3632        * structure so that we can easily access and compare it */
3633       queue->max_level.rate_time = queue->max_level.time;
3634       QUEUE_CAPACITY_CHANGE (queue);
3635       break;
3636     case PROP_USE_BUFFERING:
3637       queue->use_buffering = g_value_get_boolean (value);
3638       if (!queue->use_buffering && queue->is_buffering) {
3639         GST_DEBUG_OBJECT (queue, "Disabled buffering while buffering, "
3640             "posting 100%% message");
3641         SET_PERCENT (queue, 100);
3642         queue->is_buffering = FALSE;
3643       }
3644
3645       if (queue->use_buffering) {
3646         queue->is_buffering = TRUE;
3647         update_buffering (queue);
3648       }
3649       break;
3650     case PROP_USE_TAGS_BITRATE:
3651       queue->use_tags_bitrate = g_value_get_boolean (value);
3652       break;
3653     case PROP_USE_RATE_ESTIMATE:
3654       queue->use_rate_estimate = g_value_get_boolean (value);
3655       break;
3656     case PROP_LOW_PERCENT:
3657       queue->low_percent = g_value_get_int (value);
3658       break;
3659     case PROP_HIGH_PERCENT:
3660       queue->high_percent = g_value_get_int (value);
3661       break;
3662     case PROP_TEMP_TEMPLATE:
3663       gst_queue2_set_temp_template (queue, g_value_get_string (value));
3664       break;
3665     case PROP_TEMP_REMOVE:
3666       queue->temp_remove = g_value_get_boolean (value);
3667       break;
3668     case PROP_RING_BUFFER_MAX_SIZE:
3669       queue->ring_buffer_max_size = g_value_get_uint64 (value);
3670       break;
3671     default:
3672       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
3673       break;
3674   }
3675
3676   GST_QUEUE2_MUTEX_UNLOCK (queue);
3677   gst_queue2_post_buffering (queue);
3678 }
3679
3680 static void
3681 gst_queue2_get_property (GObject * object,
3682     guint prop_id, GValue * value, GParamSpec * pspec)
3683 {
3684   GstQueue2 *queue = GST_QUEUE2 (object);
3685
3686   GST_QUEUE2_MUTEX_LOCK (queue);
3687
3688   switch (prop_id) {
3689     case PROP_CUR_LEVEL_BYTES:
3690       g_value_set_uint (value, queue->cur_level.bytes);
3691       break;
3692     case PROP_CUR_LEVEL_BUFFERS:
3693       g_value_set_uint (value, queue->cur_level.buffers);
3694       break;
3695     case PROP_CUR_LEVEL_TIME:
3696       g_value_set_uint64 (value, queue->cur_level.time);
3697       break;
3698     case PROP_MAX_SIZE_BYTES:
3699       g_value_set_uint (value, queue->max_level.bytes);
3700       break;
3701     case PROP_MAX_SIZE_BUFFERS:
3702       g_value_set_uint (value, queue->max_level.buffers);
3703       break;
3704     case PROP_MAX_SIZE_TIME:
3705       g_value_set_uint64 (value, queue->max_level.time);
3706       break;
3707     case PROP_USE_BUFFERING:
3708       g_value_set_boolean (value, queue->use_buffering);
3709       break;
3710     case PROP_USE_TAGS_BITRATE:
3711       g_value_set_boolean (value, queue->use_tags_bitrate);
3712       break;
3713     case PROP_USE_RATE_ESTIMATE:
3714       g_value_set_boolean (value, queue->use_rate_estimate);
3715       break;
3716     case PROP_LOW_PERCENT:
3717       g_value_set_int (value, queue->low_percent);
3718       break;
3719     case PROP_HIGH_PERCENT:
3720       g_value_set_int (value, queue->high_percent);
3721       break;
3722     case PROP_TEMP_TEMPLATE:
3723       g_value_set_string (value, queue->temp_template);
3724       break;
3725     case PROP_TEMP_LOCATION:
3726       g_value_set_string (value, queue->temp_location);
3727       break;
3728     case PROP_TEMP_REMOVE:
3729       g_value_set_boolean (value, queue->temp_remove);
3730       break;
3731     case PROP_RING_BUFFER_MAX_SIZE:
3732       g_value_set_uint64 (value, queue->ring_buffer_max_size);
3733       break;
3734     case PROP_AVG_IN_RATE:
3735     {
3736       gdouble in_rate = queue->byte_in_rate;
3737
3738       /* During the first RATE_INTERVAL, byte_in_rate will not have been
3739        * calculated, so calculate it here. */
3740       if (in_rate == 0.0 && queue->bytes_in
3741           && queue->last_update_in_rates_elapsed > 0.0)
3742         in_rate = queue->bytes_in / queue->last_update_in_rates_elapsed;
3743
3744       g_value_set_int64 (value, (gint64) in_rate);
3745       break;
3746     }
3747     default:
3748       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
3749       break;
3750   }
3751
3752   GST_QUEUE2_MUTEX_UNLOCK (queue);
3753 }