4431cbf5e3c0ef23317abcfcdcbb2938dd509026
[platform/upstream/gstreamer.git] / libs / gst / base / gstaggregator.c
1 /* GStreamer aggregator base class
2  * Copyright (C) 2014 Mathieu Duponchelle <mathieu.duponchelle@opencreed.com>
3  * Copyright (C) 2014 Thibault Saunier <tsaunier@gnome.org>
4  *
5  * gstaggregator.c:
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Library General Public
9  * License as published by the Free Software Foundation; either
10  * version 2 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Library General Public License for more details.
16  *
17  * You should have received a copy of the GNU Library General Public
18  * License along with this library; if not, write to the
19  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
20  * Boston, MA 02110-1301, USA.
21  */
22 /**
23  * SECTION: gstaggregator
24  * @title: GstAggregator
25  * @short_description: manages a set of pads with the purpose of
26  * aggregating their buffers.
27  * @see_also: gstcollectpads for historical reasons.
28  *
29  * Manages a set of pads with the purpose of aggregating their buffers.
30  * Control is given to the subclass when all pads have data.
31  *
32  *  * Base class for mixers and muxers. Subclasses should at least implement
33  *    the #GstAggregatorClass.aggregate() virtual method.
34  *
35  *  * When data is queued on all pads, tha aggregate vmethod is called.
36  *
37  *  * One can peek at the data on any given GstAggregatorPad with the
38  *    gst_aggregator_pad_get_buffer () method, and take ownership of it
39  *    with the gst_aggregator_pad_steal_buffer () method. When a buffer
40  *    has been taken with steal_buffer (), a new buffer can be queued
41  *    on that pad.
42  *
43  *  * If the subclass wishes to push a buffer downstream in its aggregate
44  *    implementation, it should do so through the
45  *    gst_aggregator_finish_buffer () method. This method will take care
46  *    of sending and ordering mandatory events such as stream start, caps
47  *    and segment.
48  *
49  *  * Same goes for EOS events, which should not be pushed directly by the
50  *    subclass, it should instead return GST_FLOW_EOS in its aggregate
51  *    implementation.
52  *
53  *  * Note that the aggregator logic regarding gap event handling is to turn
54  *    these into gap buffers with matching PTS and duration. It will also
55  *    flag these buffers with GST_BUFFER_FLAG_GAP and GST_BUFFER_FLAG_DROPPABLE
56  *    to ease their identification and subsequent processing.
57  *
58  */
59
60 #ifdef HAVE_CONFIG_H
61 #  include "config.h"
62 #endif
63
64 #include <string.h>             /* strlen */
65
66 #include "gstaggregator.h"
67
68 typedef enum
69 {
70   GST_AGGREGATOR_START_TIME_SELECTION_ZERO,
71   GST_AGGREGATOR_START_TIME_SELECTION_FIRST,
72   GST_AGGREGATOR_START_TIME_SELECTION_SET
73 } GstAggregatorStartTimeSelection;
74
75 static GType
76 gst_aggregator_start_time_selection_get_type (void)
77 {
78   static GType gtype = 0;
79
80   if (gtype == 0) {
81     static const GEnumValue values[] = {
82       {GST_AGGREGATOR_START_TIME_SELECTION_ZERO,
83           "Start at 0 running time (default)", "zero"},
84       {GST_AGGREGATOR_START_TIME_SELECTION_FIRST,
85           "Start at first observed input running time", "first"},
86       {GST_AGGREGATOR_START_TIME_SELECTION_SET,
87           "Set start time with start-time property", "set"},
88       {0, NULL, NULL}
89     };
90
91     gtype = g_enum_register_static ("GstAggregatorStartTimeSelection", values);
92   }
93   return gtype;
94 }
95
96 /*  Might become API */
97 static void gst_aggregator_merge_tags (GstAggregator * aggregator,
98     const GstTagList * tags, GstTagMergeMode mode);
99 static void gst_aggregator_set_latency_property (GstAggregator * agg,
100     gint64 latency);
101 static gint64 gst_aggregator_get_latency_property (GstAggregator * agg);
102
103
104 /* Locking order, locks in this element must always be taken in this order
105  *
106  * standard sink pad stream lock -> GST_PAD_STREAM_LOCK (aggpad)
107  * Aggregator pad flush lock -> PAD_FLUSH_LOCK(aggpad)
108  * standard src pad stream lock -> GST_PAD_STREAM_LOCK (srcpad)
109  * Aggregator src lock -> SRC_LOCK(agg) w/ SRC_WAIT/BROADCAST
110  * standard element object lock -> GST_OBJECT_LOCK(agg)
111  * Aggregator pad lock -> PAD_LOCK (aggpad) w/ PAD_WAIT/BROADCAST_EVENT(aggpad)
112  * standard src pad object lock -> GST_OBJECT_LOCK(srcpad)
113  * standard sink pad object lock -> GST_OBJECT_LOCK(aggpad)
114  */
115
116
117 static GstClockTime gst_aggregator_get_latency_unlocked (GstAggregator * self);
118
119 GST_DEBUG_CATEGORY_STATIC (aggregator_debug);
120 #define GST_CAT_DEFAULT aggregator_debug
121
122 /* GstAggregatorPad definitions */
123 #define PAD_LOCK(pad)   G_STMT_START {                                  \
124   GST_TRACE_OBJECT (pad, "Taking PAD lock from thread %p",              \
125         g_thread_self());                                               \
126   g_mutex_lock(&pad->priv->lock);                                       \
127   GST_TRACE_OBJECT (pad, "Took PAD lock from thread %p",                \
128         g_thread_self());                                               \
129   } G_STMT_END
130
131 #define PAD_UNLOCK(pad)  G_STMT_START {                                 \
132   GST_TRACE_OBJECT (pad, "Releasing PAD lock from thread %p",           \
133       g_thread_self());                                                 \
134   g_mutex_unlock(&pad->priv->lock);                                     \
135   GST_TRACE_OBJECT (pad, "Release PAD lock from thread %p",             \
136         g_thread_self());                                               \
137   } G_STMT_END
138
139
140 #define PAD_WAIT_EVENT(pad)   G_STMT_START {                            \
141   GST_LOG_OBJECT (pad, "Waiting for buffer to be consumed thread %p",   \
142         g_thread_self());                                               \
143   g_cond_wait(&(((GstAggregatorPad* )pad)->priv->event_cond),           \
144       (&((GstAggregatorPad*)pad)->priv->lock));                         \
145   GST_LOG_OBJECT (pad, "DONE Waiting for buffer to be consumed on thread %p", \
146         g_thread_self());                                               \
147   } G_STMT_END
148
149 #define PAD_BROADCAST_EVENT(pad) G_STMT_START {                        \
150   GST_LOG_OBJECT (pad, "Signaling buffer consumed from thread %p",     \
151         g_thread_self());                                              \
152   g_cond_broadcast(&(((GstAggregatorPad* )pad)->priv->event_cond));    \
153   } G_STMT_END
154
155
156 #define PAD_FLUSH_LOCK(pad)     G_STMT_START {                          \
157   GST_TRACE_OBJECT (pad, "Taking lock from thread %p",                  \
158         g_thread_self());                                               \
159   g_mutex_lock(&pad->priv->flush_lock);                                 \
160   GST_TRACE_OBJECT (pad, "Took lock from thread %p",                    \
161         g_thread_self());                                               \
162   } G_STMT_END
163
164 #define PAD_FLUSH_UNLOCK(pad)   G_STMT_START {                          \
165   GST_TRACE_OBJECT (pad, "Releasing lock from thread %p",               \
166         g_thread_self());                                               \
167   g_mutex_unlock(&pad->priv->flush_lock);                               \
168   GST_TRACE_OBJECT (pad, "Release lock from thread %p",                 \
169         g_thread_self());                                               \
170   } G_STMT_END
171
172 #define SRC_LOCK(self)   G_STMT_START {                             \
173   GST_TRACE_OBJECT (self, "Taking src lock from thread %p",         \
174       g_thread_self());                                             \
175   g_mutex_lock(&self->priv->src_lock);                              \
176   GST_TRACE_OBJECT (self, "Took src lock from thread %p",           \
177         g_thread_self());                                           \
178   } G_STMT_END
179
180 #define SRC_UNLOCK(self)  G_STMT_START {                            \
181   GST_TRACE_OBJECT (self, "Releasing src lock from thread %p",      \
182         g_thread_self());                                           \
183   g_mutex_unlock(&self->priv->src_lock);                            \
184   GST_TRACE_OBJECT (self, "Released src lock from thread %p",       \
185         g_thread_self());                                           \
186   } G_STMT_END
187
188 #define SRC_WAIT(self) G_STMT_START {                               \
189   GST_LOG_OBJECT (self, "Waiting for src on thread %p",             \
190         g_thread_self());                                           \
191   g_cond_wait(&(self->priv->src_cond), &(self->priv->src_lock));    \
192   GST_LOG_OBJECT (self, "DONE Waiting for src on thread %p",        \
193         g_thread_self());                                           \
194   } G_STMT_END
195
196 #define SRC_BROADCAST(self) G_STMT_START {                          \
197     GST_LOG_OBJECT (self, "Signaling src from thread %p",           \
198         g_thread_self());                                           \
199     if (self->priv->aggregate_id)                                   \
200       gst_clock_id_unschedule (self->priv->aggregate_id);           \
201     g_cond_broadcast(&(self->priv->src_cond));                      \
202   } G_STMT_END
203
204 struct _GstAggregatorPadPrivate
205 {
206   /* Following fields are protected by the PAD_LOCK */
207   GstFlowReturn flow_return;
208   gboolean pending_flush_start;
209   gboolean pending_flush_stop;
210   gboolean pending_eos;
211
212   gboolean first_buffer;
213
214   GQueue buffers;
215   GstBuffer *clipped_buffer;
216   guint num_buffers;
217   GstClockTime head_position;
218   GstClockTime tail_position;
219   GstClockTime head_time;
220   GstClockTime tail_time;
221   GstClockTime time_level;
222   GstSegment head_segment;      /* segment before the queue */
223
224   gboolean negotiated;
225
226   gboolean eos;
227
228   GMutex lock;
229   GCond event_cond;
230   /* This lock prevents a flush start processing happening while
231    * the chain function is also happening.
232    */
233   GMutex flush_lock;
234 };
235
236 /* Must be called with PAD_LOCK held */
237 static void
238 gst_aggregator_pad_reset_unlocked (GstAggregatorPad * aggpad)
239 {
240   aggpad->priv->pending_eos = FALSE;
241   aggpad->priv->eos = FALSE;
242   aggpad->priv->flow_return = GST_FLOW_OK;
243   GST_OBJECT_LOCK (aggpad);
244   gst_segment_init (&aggpad->segment, GST_FORMAT_UNDEFINED);
245   gst_segment_init (&aggpad->priv->head_segment, GST_FORMAT_UNDEFINED);
246   GST_OBJECT_UNLOCK (aggpad);
247   aggpad->priv->head_position = GST_CLOCK_TIME_NONE;
248   aggpad->priv->tail_position = GST_CLOCK_TIME_NONE;
249   aggpad->priv->head_time = GST_CLOCK_TIME_NONE;
250   aggpad->priv->tail_time = GST_CLOCK_TIME_NONE;
251   aggpad->priv->time_level = 0;
252   aggpad->priv->first_buffer = TRUE;
253 }
254
255 static gboolean
256 gst_aggregator_pad_flush (GstAggregatorPad * aggpad, GstAggregator * agg)
257 {
258   GstAggregatorPadClass *klass = GST_AGGREGATOR_PAD_GET_CLASS (aggpad);
259
260   PAD_LOCK (aggpad);
261   gst_aggregator_pad_reset_unlocked (aggpad);
262   PAD_UNLOCK (aggpad);
263
264   if (klass->flush)
265     return klass->flush (aggpad, agg);
266
267   return TRUE;
268 }
269
270 /*************************************
271  * GstAggregator implementation  *
272  *************************************/
273 static GstElementClass *aggregator_parent_class = NULL;
274
275 /* All members are protected by the object lock unless otherwise noted */
276
277 struct _GstAggregatorPrivate
278 {
279   gint max_padserial;
280
281   /* Our state is >= PAUSED */
282   gboolean running;             /* protected by src_lock */
283
284   gint seqnum;
285   gboolean send_stream_start;   /* protected by srcpad stream lock */
286   gboolean send_segment;
287   gboolean flush_seeking;
288   gboolean pending_flush_start;
289   gboolean send_eos;            /* protected by srcpad stream lock */
290
291   GstCaps *srccaps;             /* protected by the srcpad stream lock */
292
293   GstTagList *tags;
294   gboolean tags_changed;
295
296   gboolean peer_latency_live;   /* protected by src_lock */
297   GstClockTime peer_latency_min;        /* protected by src_lock */
298   GstClockTime peer_latency_max;        /* protected by src_lock */
299   gboolean has_peer_latency;    /* protected by src_lock */
300
301   GstClockTime sub_latency_min; /* protected by src_lock */
302   GstClockTime sub_latency_max; /* protected by src_lock */
303
304   /* aggregate */
305   GstClockID aggregate_id;      /* protected by src_lock */
306   GMutex src_lock;
307   GCond src_cond;
308
309   gboolean first_buffer;        /* protected by object lock */
310   GstAggregatorStartTimeSelection start_time_selection;
311   GstClockTime start_time;
312
313   /* protected by the object lock */
314   GstQuery *allocation_query;
315   GstAllocator *allocator;
316   GstBufferPool *pool;
317   GstAllocationParams allocation_params;
318
319   /* properties */
320   gint64 latency;               /* protected by both src_lock and all pad locks */
321 };
322
323 typedef struct
324 {
325   GstEvent *event;
326   gboolean result;
327   gboolean flush;
328   gboolean only_to_active_pads;
329
330   gboolean one_actually_seeked;
331 } EventData;
332
333 #define DEFAULT_LATENCY              0
334 #define DEFAULT_START_TIME_SELECTION GST_AGGREGATOR_START_TIME_SELECTION_ZERO
335 #define DEFAULT_START_TIME           (-1)
336
337 enum
338 {
339   PROP_0,
340   PROP_LATENCY,
341   PROP_START_TIME_SELECTION,
342   PROP_START_TIME,
343   PROP_LAST
344 };
345
346 static GstFlowReturn gst_aggregator_pad_chain_internal (GstAggregator * self,
347     GstAggregatorPad * aggpad, GstBuffer * buffer, gboolean head);
348
349 /**
350  * gst_aggregator_iterate_sinkpads:
351  * @self: The #GstAggregator
352  * @func: (scope call): The function to call.
353  * @user_data: (closure): The data to pass to @func.
354  *
355  * Iterate the sinkpads of aggregator to call a function on them.
356  *
357  * This method guarantees that @func will be called only once for each
358  * sink pad.
359  *
360  * Returns: %FALSE if there are no sinkpads or if @func returned %FALSE
361  */
362 gboolean
363 gst_aggregator_iterate_sinkpads (GstAggregator * self,
364     GstAggregatorPadForeachFunc func, gpointer user_data)
365 {
366   gboolean result = FALSE;
367   GstIterator *iter;
368   gboolean done = FALSE;
369   GValue item = { 0, };
370   GList *seen_pads = NULL;
371
372   iter = gst_element_iterate_sink_pads (GST_ELEMENT (self));
373
374   if (!iter)
375     goto no_iter;
376
377   while (!done) {
378     switch (gst_iterator_next (iter, &item)) {
379       case GST_ITERATOR_OK:
380       {
381         GstAggregatorPad *pad;
382
383         pad = g_value_get_object (&item);
384
385         /* if already pushed, skip. FIXME, find something faster to tag pads */
386         if (pad == NULL || g_list_find (seen_pads, pad)) {
387           g_value_reset (&item);
388           break;
389         }
390
391         GST_LOG_OBJECT (pad, "calling function %s on pad",
392             GST_DEBUG_FUNCPTR_NAME (func));
393
394         result = func (self, pad, user_data);
395
396         done = !result;
397
398         seen_pads = g_list_prepend (seen_pads, pad);
399
400         g_value_reset (&item);
401         break;
402       }
403       case GST_ITERATOR_RESYNC:
404         gst_iterator_resync (iter);
405         break;
406       case GST_ITERATOR_ERROR:
407         GST_ERROR_OBJECT (self,
408             "Could not iterate over internally linked pads");
409         done = TRUE;
410         break;
411       case GST_ITERATOR_DONE:
412         done = TRUE;
413         break;
414     }
415   }
416   g_value_unset (&item);
417   gst_iterator_free (iter);
418
419   if (seen_pads == NULL) {
420     GST_DEBUG_OBJECT (self, "No pad seen");
421     return FALSE;
422   }
423
424   g_list_free (seen_pads);
425
426 no_iter:
427   return result;
428 }
429
430 static gboolean
431 gst_aggregator_pad_queue_is_empty (GstAggregatorPad * pad)
432 {
433   return (g_queue_peek_tail (&pad->priv->buffers) == NULL &&
434       pad->priv->clipped_buffer == NULL);
435 }
436
437 static gboolean
438 gst_aggregator_check_pads_ready (GstAggregator * self)
439 {
440   GstAggregatorPad *pad;
441   GList *l, *sinkpads;
442   gboolean have_buffer = TRUE;
443   gboolean have_event = FALSE;
444
445   GST_LOG_OBJECT (self, "checking pads");
446
447   GST_OBJECT_LOCK (self);
448
449   sinkpads = GST_ELEMENT_CAST (self)->sinkpads;
450   if (sinkpads == NULL)
451     goto no_sinkpads;
452
453   for (l = sinkpads; l != NULL; l = l->next) {
454     pad = l->data;
455
456     PAD_LOCK (pad);
457
458     if (pad->priv->num_buffers == 0) {
459       if (!gst_aggregator_pad_queue_is_empty (pad))
460         have_event = TRUE;
461       if (!pad->priv->eos) {
462         have_buffer = FALSE;
463
464         /* If not live we need data on all pads, so leave the loop */
465         if (!self->priv->peer_latency_live) {
466           PAD_UNLOCK (pad);
467           goto pad_not_ready;
468         }
469       }
470     } else if (self->priv->peer_latency_live) {
471       /* In live mode, having a single pad with buffers is enough to
472        * generate a start time from it. In non-live mode all pads need
473        * to have a buffer
474        */
475       self->priv->first_buffer = FALSE;
476     }
477
478     PAD_UNLOCK (pad);
479   }
480
481   if (!have_buffer && !have_event)
482     goto pad_not_ready;
483
484   if (have_buffer)
485     self->priv->first_buffer = FALSE;
486
487   GST_OBJECT_UNLOCK (self);
488   GST_LOG_OBJECT (self, "pads are ready");
489   return TRUE;
490
491 no_sinkpads:
492   {
493     GST_LOG_OBJECT (self, "pads not ready: no sink pads");
494     GST_OBJECT_UNLOCK (self);
495     return FALSE;
496   }
497 pad_not_ready:
498   {
499     if (have_event)
500       GST_LOG_OBJECT (pad, "pad not ready to be aggregated yet,"
501           " but waking up for serialized event");
502     else
503       GST_LOG_OBJECT (pad, "pad not ready to be aggregated yet");
504     GST_OBJECT_UNLOCK (self);
505     return have_event;
506   }
507 }
508
509 static void
510 gst_aggregator_reset_flow_values (GstAggregator * self)
511 {
512   GST_OBJECT_LOCK (self);
513   self->priv->send_stream_start = TRUE;
514   self->priv->send_segment = TRUE;
515   gst_segment_init (&self->segment, GST_FORMAT_TIME);
516   self->priv->first_buffer = TRUE;
517   GST_OBJECT_UNLOCK (self);
518 }
519
520 static inline void
521 gst_aggregator_push_mandatory_events (GstAggregator * self)
522 {
523   GstAggregatorPrivate *priv = self->priv;
524   GstEvent *segment = NULL;
525   GstEvent *tags = NULL;
526
527   if (self->priv->send_stream_start) {
528     gchar s_id[32];
529
530     GST_INFO_OBJECT (self, "pushing stream start");
531     /* stream-start (FIXME: create id based on input ids) */
532     g_snprintf (s_id, sizeof (s_id), "agg-%08x", g_random_int ());
533     if (!gst_pad_push_event (self->srcpad, gst_event_new_stream_start (s_id))) {
534       GST_WARNING_OBJECT (self->srcpad, "Sending stream start event failed");
535     }
536     self->priv->send_stream_start = FALSE;
537   }
538
539   if (self->priv->srccaps) {
540
541     GST_INFO_OBJECT (self, "pushing caps: %" GST_PTR_FORMAT,
542         self->priv->srccaps);
543     if (!gst_pad_push_event (self->srcpad,
544             gst_event_new_caps (self->priv->srccaps))) {
545       GST_WARNING_OBJECT (self->srcpad, "Sending caps event failed");
546     }
547     gst_caps_unref (self->priv->srccaps);
548     self->priv->srccaps = NULL;
549   }
550
551   GST_OBJECT_LOCK (self);
552   if (self->priv->send_segment && !self->priv->flush_seeking) {
553     segment = gst_event_new_segment (&self->segment);
554
555     if (!self->priv->seqnum)
556       self->priv->seqnum = gst_event_get_seqnum (segment);
557     else
558       gst_event_set_seqnum (segment, self->priv->seqnum);
559     self->priv->send_segment = FALSE;
560
561     GST_DEBUG_OBJECT (self, "pushing segment %" GST_PTR_FORMAT, segment);
562   }
563
564   if (priv->tags && priv->tags_changed && !self->priv->flush_seeking) {
565     tags = gst_event_new_tag (gst_tag_list_ref (priv->tags));
566     priv->tags_changed = FALSE;
567   }
568   GST_OBJECT_UNLOCK (self);
569
570   if (segment)
571     gst_pad_push_event (self->srcpad, segment);
572   if (tags)
573     gst_pad_push_event (self->srcpad, tags);
574
575 }
576
577 /**
578  * gst_aggregator_set_src_caps:
579  * @self: The #GstAggregator
580  * @caps: The #GstCaps to set on the src pad.
581  *
582  * Sets the caps to be used on the src pad.
583  */
584 void
585 gst_aggregator_set_src_caps (GstAggregator * self, GstCaps * caps)
586 {
587   GST_PAD_STREAM_LOCK (self->srcpad);
588   gst_caps_replace (&self->priv->srccaps, caps);
589   gst_aggregator_push_mandatory_events (self);
590   GST_PAD_STREAM_UNLOCK (self->srcpad);
591 }
592
593 /**
594  * gst_aggregator_finish_buffer:
595  * @self: The #GstAggregator
596  * @buffer: (transfer full): the #GstBuffer to push.
597  *
598  * This method will push the provided output buffer downstream. If needed,
599  * mandatory events such as stream-start, caps, and segment events will be
600  * sent before pushing the buffer.
601  */
602 GstFlowReturn
603 gst_aggregator_finish_buffer (GstAggregator * self, GstBuffer * buffer)
604 {
605   gst_aggregator_push_mandatory_events (self);
606
607   GST_OBJECT_LOCK (self);
608   if (!self->priv->flush_seeking && gst_pad_is_active (self->srcpad)) {
609     GST_TRACE_OBJECT (self, "pushing buffer %" GST_PTR_FORMAT, buffer);
610     GST_OBJECT_UNLOCK (self);
611     return gst_pad_push (self->srcpad, buffer);
612   } else {
613     GST_INFO_OBJECT (self, "Not pushing (active: %i, flushing: %i)",
614         self->priv->flush_seeking, gst_pad_is_active (self->srcpad));
615     GST_OBJECT_UNLOCK (self);
616     gst_buffer_unref (buffer);
617     return GST_FLOW_OK;
618   }
619 }
620
621 static void
622 gst_aggregator_push_eos (GstAggregator * self)
623 {
624   GstEvent *event;
625   gst_aggregator_push_mandatory_events (self);
626
627   event = gst_event_new_eos ();
628
629   GST_OBJECT_LOCK (self);
630   self->priv->send_eos = FALSE;
631   gst_event_set_seqnum (event, self->priv->seqnum);
632   GST_OBJECT_UNLOCK (self);
633
634   gst_pad_push_event (self->srcpad, event);
635 }
636
637 static GstClockTime
638 gst_aggregator_get_next_time (GstAggregator * self)
639 {
640   GstAggregatorClass *klass = GST_AGGREGATOR_GET_CLASS (self);
641
642   if (klass->get_next_time)
643     return klass->get_next_time (self);
644
645   return GST_CLOCK_TIME_NONE;
646 }
647
648 static gboolean
649 gst_aggregator_wait_and_check (GstAggregator * self, gboolean * timeout)
650 {
651   GstClockTime latency;
652   GstClockTime start;
653   gboolean res;
654
655   *timeout = FALSE;
656
657   SRC_LOCK (self);
658
659   latency = gst_aggregator_get_latency_unlocked (self);
660
661   if (gst_aggregator_check_pads_ready (self)) {
662     GST_DEBUG_OBJECT (self, "all pads have data");
663     SRC_UNLOCK (self);
664
665     return TRUE;
666   }
667
668   /* Before waiting, check if we're actually still running */
669   if (!self->priv->running || !self->priv->send_eos) {
670     SRC_UNLOCK (self);
671
672     return FALSE;
673   }
674
675   start = gst_aggregator_get_next_time (self);
676
677   /* If we're not live, or if we use the running time
678    * of the first buffer as start time, we wait until
679    * all pads have buffers.
680    * Otherwise (i.e. if we are live!), we wait on the clock
681    * and if a pad does not have a buffer in time we ignore
682    * that pad.
683    */
684   GST_OBJECT_LOCK (self);
685   if (!GST_CLOCK_TIME_IS_VALID (latency) ||
686       !GST_IS_CLOCK (GST_ELEMENT_CLOCK (self)) ||
687       !GST_CLOCK_TIME_IS_VALID (start) ||
688       (self->priv->first_buffer
689           && self->priv->start_time_selection ==
690           GST_AGGREGATOR_START_TIME_SELECTION_FIRST)) {
691     /* We wake up here when something happened, and below
692      * then check if we're ready now. If we return FALSE,
693      * we will be directly called again.
694      */
695     GST_OBJECT_UNLOCK (self);
696     SRC_WAIT (self);
697   } else {
698     GstClockTime base_time, time;
699     GstClock *clock;
700     GstClockReturn status;
701     GstClockTimeDiff jitter;
702
703     GST_DEBUG_OBJECT (self, "got subclass start time: %" GST_TIME_FORMAT,
704         GST_TIME_ARGS (start));
705
706     base_time = GST_ELEMENT_CAST (self)->base_time;
707     clock = gst_object_ref (GST_ELEMENT_CLOCK (self));
708     GST_OBJECT_UNLOCK (self);
709
710     time = base_time + start;
711     time += latency;
712
713     GST_DEBUG_OBJECT (self, "possibly waiting for clock to reach %"
714         GST_TIME_FORMAT " (base %" GST_TIME_FORMAT " start %" GST_TIME_FORMAT
715         " latency %" GST_TIME_FORMAT " current %" GST_TIME_FORMAT ")",
716         GST_TIME_ARGS (time),
717         GST_TIME_ARGS (base_time),
718         GST_TIME_ARGS (start), GST_TIME_ARGS (latency),
719         GST_TIME_ARGS (gst_clock_get_time (clock)));
720
721     self->priv->aggregate_id = gst_clock_new_single_shot_id (clock, time);
722     gst_object_unref (clock);
723     SRC_UNLOCK (self);
724
725     jitter = 0;
726     status = gst_clock_id_wait (self->priv->aggregate_id, &jitter);
727
728     SRC_LOCK (self);
729     if (self->priv->aggregate_id) {
730       gst_clock_id_unref (self->priv->aggregate_id);
731       self->priv->aggregate_id = NULL;
732     }
733
734     GST_DEBUG_OBJECT (self,
735         "clock returned %d (jitter: %" GST_STIME_FORMAT ")",
736         status, GST_STIME_ARGS (jitter));
737
738     /* we timed out */
739     if (status == GST_CLOCK_OK || status == GST_CLOCK_EARLY) {
740       SRC_UNLOCK (self);
741       *timeout = TRUE;
742       return TRUE;
743     }
744   }
745
746   res = gst_aggregator_check_pads_ready (self);
747   SRC_UNLOCK (self);
748
749   return res;
750 }
751
752 static gboolean
753 check_events (GstAggregator * self, GstAggregatorPad * pad, gpointer user_data)
754 {
755   GstEvent *event = NULL;
756   GstQuery *query = NULL;
757   GstAggregatorClass *klass = NULL;
758   gboolean *processed_event = user_data;
759
760   do {
761     event = NULL;
762
763     PAD_LOCK (pad);
764     if (pad->priv->num_buffers == 0 && pad->priv->pending_eos) {
765       pad->priv->pending_eos = FALSE;
766       pad->priv->eos = TRUE;
767     }
768     if (pad->priv->clipped_buffer == NULL &&
769         !GST_IS_BUFFER (g_queue_peek_tail (&pad->priv->buffers))) {
770       if (GST_IS_EVENT (g_queue_peek_tail (&pad->priv->buffers)))
771         event = gst_event_ref (g_queue_peek_tail (&pad->priv->buffers));
772       if (GST_IS_QUERY (g_queue_peek_tail (&pad->priv->buffers)))
773         query = g_queue_peek_tail (&pad->priv->buffers);
774     }
775     PAD_UNLOCK (pad);
776     if (event || query) {
777       gboolean ret;
778
779       if (processed_event)
780         *processed_event = TRUE;
781       if (klass == NULL)
782         klass = GST_AGGREGATOR_GET_CLASS (self);
783
784       if (event) {
785         GST_LOG_OBJECT (pad, "Processing %" GST_PTR_FORMAT, event);
786         gst_event_ref (event);
787         ret = klass->sink_event (self, pad, event);
788
789         PAD_LOCK (pad);
790         if (GST_EVENT_TYPE (event) == GST_EVENT_CAPS)
791           pad->priv->negotiated = ret;
792         if (g_queue_peek_tail (&pad->priv->buffers) == event)
793           gst_event_unref (g_queue_pop_tail (&pad->priv->buffers));
794         gst_event_unref (event);
795       }
796
797       if (query) {
798         GST_LOG_OBJECT (pad, "Processing %" GST_PTR_FORMAT, query);
799         ret = klass->sink_query (self, pad, query);
800
801         PAD_LOCK (pad);
802         if (g_queue_peek_tail (&pad->priv->buffers) == query) {
803           GstStructure *s;
804
805           s = gst_query_writable_structure (query);
806           gst_structure_set (s, "gst-aggregator-retval", G_TYPE_BOOLEAN, ret,
807               NULL);
808           g_queue_pop_tail (&pad->priv->buffers);
809         }
810       }
811
812       PAD_BROADCAST_EVENT (pad);
813       PAD_UNLOCK (pad);
814     }
815   } while (event != NULL);
816
817   return TRUE;
818 }
819
820 static void
821 gst_aggregator_pad_set_flushing (GstAggregatorPad * aggpad,
822     GstFlowReturn flow_return, gboolean full)
823 {
824   GList *item;
825
826   PAD_LOCK (aggpad);
827   if (flow_return == GST_FLOW_NOT_LINKED)
828     aggpad->priv->flow_return = MIN (flow_return, aggpad->priv->flow_return);
829   else
830     aggpad->priv->flow_return = flow_return;
831
832   item = g_queue_peek_head_link (&aggpad->priv->buffers);
833   while (item) {
834     GList *next = item->next;
835
836     /* In partial flush, we do like the pad, we get rid of non-sticky events
837      * and EOS/SEGMENT.
838      */
839     if (full || GST_IS_BUFFER (item->data) ||
840         GST_EVENT_TYPE (item->data) == GST_EVENT_EOS ||
841         GST_EVENT_TYPE (item->data) == GST_EVENT_SEGMENT ||
842         !GST_EVENT_IS_STICKY (item->data)) {
843       if (!GST_IS_QUERY (item->data))
844         gst_mini_object_unref (item->data);
845       g_queue_delete_link (&aggpad->priv->buffers, item);
846     }
847     item = next;
848   }
849   aggpad->priv->num_buffers = 0;
850   gst_buffer_replace (&aggpad->priv->clipped_buffer, NULL);
851
852   PAD_BROADCAST_EVENT (aggpad);
853   PAD_UNLOCK (aggpad);
854 }
855
856 static GstFlowReturn
857 gst_aggregator_default_update_src_caps (GstAggregator * agg, GstCaps * caps,
858     GstCaps ** ret)
859 {
860   *ret = gst_caps_ref (caps);
861
862   return GST_FLOW_OK;
863 }
864
865 static GstCaps *
866 gst_aggregator_default_fixate_src_caps (GstAggregator * agg, GstCaps * caps)
867 {
868   caps = gst_caps_fixate (caps);
869
870   return caps;
871 }
872
873 static gboolean
874 gst_aggregator_default_negotiated_src_caps (GstAggregator * agg, GstCaps * caps)
875 {
876   return TRUE;
877 }
878
879
880 /* takes ownership of the pool, allocator and query */
881 static gboolean
882 gst_aggregator_set_allocation (GstAggregator * self,
883     GstBufferPool * pool, GstAllocator * allocator,
884     GstAllocationParams * params, GstQuery * query)
885 {
886   GstAllocator *oldalloc;
887   GstBufferPool *oldpool;
888   GstQuery *oldquery;
889
890   GST_DEBUG ("storing allocation query");
891
892   GST_OBJECT_LOCK (self);
893   oldpool = self->priv->pool;
894   self->priv->pool = pool;
895
896   oldalloc = self->priv->allocator;
897   self->priv->allocator = allocator;
898
899   oldquery = self->priv->allocation_query;
900   self->priv->allocation_query = query;
901
902   if (params)
903     self->priv->allocation_params = *params;
904   else
905     gst_allocation_params_init (&self->priv->allocation_params);
906   GST_OBJECT_UNLOCK (self);
907
908   if (oldpool) {
909     GST_DEBUG_OBJECT (self, "deactivating old pool %p", oldpool);
910     gst_buffer_pool_set_active (oldpool, FALSE);
911     gst_object_unref (oldpool);
912   }
913   if (oldalloc) {
914     gst_object_unref (oldalloc);
915   }
916   if (oldquery) {
917     gst_query_unref (oldquery);
918   }
919   return TRUE;
920 }
921
922
923 static gboolean
924 gst_aggregator_decide_allocation (GstAggregator * self, GstQuery * query)
925 {
926   GstAggregatorClass *aggclass = GST_AGGREGATOR_GET_CLASS (self);
927
928   if (aggclass->decide_allocation)
929     if (!aggclass->decide_allocation (self, query))
930       return FALSE;
931
932   return TRUE;
933 }
934
935 static gboolean
936 gst_aggregator_do_allocation (GstAggregator * self, GstCaps * caps)
937 {
938   GstQuery *query;
939   gboolean result = TRUE;
940   GstBufferPool *pool = NULL;
941   GstAllocator *allocator;
942   GstAllocationParams params;
943
944   /* find a pool for the negotiated caps now */
945   GST_DEBUG_OBJECT (self, "doing allocation query");
946   query = gst_query_new_allocation (caps, TRUE);
947   if (!gst_pad_peer_query (self->srcpad, query)) {
948     /* not a problem, just debug a little */
949     GST_DEBUG_OBJECT (self, "peer ALLOCATION query failed");
950   }
951
952   GST_DEBUG_OBJECT (self, "calling decide_allocation");
953   result = gst_aggregator_decide_allocation (self, query);
954
955   GST_DEBUG_OBJECT (self, "ALLOCATION (%d) params: %" GST_PTR_FORMAT, result,
956       query);
957
958   if (!result)
959     goto no_decide_allocation;
960
961   /* we got configuration from our peer or the decide_allocation method,
962    * parse them */
963   if (gst_query_get_n_allocation_params (query) > 0) {
964     gst_query_parse_nth_allocation_param (query, 0, &allocator, &params);
965   } else {
966     allocator = NULL;
967     gst_allocation_params_init (&params);
968   }
969
970   if (gst_query_get_n_allocation_pools (query) > 0)
971     gst_query_parse_nth_allocation_pool (query, 0, &pool, NULL, NULL, NULL);
972
973   /* now store */
974   result =
975       gst_aggregator_set_allocation (self, pool, allocator, &params, query);
976
977   return result;
978
979   /* Errors */
980 no_decide_allocation:
981   {
982     GST_WARNING_OBJECT (self, "Failed to decide allocation");
983     gst_query_unref (query);
984
985     return result;
986   }
987
988 }
989
990 /* WITH SRC_LOCK held */
991 static GstFlowReturn
992 gst_aggregator_update_src_caps (GstAggregator * self)
993 {
994   GstAggregatorClass *agg_klass = GST_AGGREGATOR_GET_CLASS (self);
995   GstCaps *downstream_caps, *template_caps, *caps = NULL;
996   GstFlowReturn ret = GST_FLOW_OK;
997
998   template_caps = gst_pad_get_pad_template_caps (self->srcpad);
999   downstream_caps = gst_pad_peer_query_caps (self->srcpad, template_caps);
1000
1001   if (gst_caps_is_empty (downstream_caps)) {
1002     GST_INFO_OBJECT (self, "Downstream caps (%"
1003         GST_PTR_FORMAT ") not compatible with pad template caps (%"
1004         GST_PTR_FORMAT ")", downstream_caps, template_caps);
1005     ret = GST_FLOW_NOT_NEGOTIATED;
1006     goto done;
1007   }
1008
1009   g_assert (agg_klass->update_src_caps);
1010   GST_DEBUG_OBJECT (self, "updating caps from %" GST_PTR_FORMAT,
1011       downstream_caps);
1012   ret = agg_klass->update_src_caps (self, downstream_caps, &caps);
1013   if (ret < GST_FLOW_OK) {
1014     GST_WARNING_OBJECT (self, "Subclass failed to update provided caps");
1015     goto done;
1016   }
1017   if ((caps == NULL || gst_caps_is_empty (caps)) && ret >= GST_FLOW_OK) {
1018     ret = GST_FLOW_NOT_NEGOTIATED;
1019     goto done;
1020   }
1021   GST_DEBUG_OBJECT (self, "               to %" GST_PTR_FORMAT, caps);
1022
1023 #ifdef GST_ENABLE_EXTRA_CHECKS
1024   if (!gst_caps_is_subset (caps, template_caps)) {
1025     GstCaps *intersection;
1026
1027     GST_ERROR_OBJECT (self,
1028         "update_src_caps returned caps %" GST_PTR_FORMAT
1029         " which are not a real subset of the template caps %"
1030         GST_PTR_FORMAT, caps, template_caps);
1031     g_warning ("%s: update_src_caps returned caps which are not a real "
1032         "subset of the filter caps", GST_ELEMENT_NAME (self));
1033
1034     intersection =
1035         gst_caps_intersect_full (template_caps, caps, GST_CAPS_INTERSECT_FIRST);
1036     gst_caps_unref (caps);
1037     caps = intersection;
1038   }
1039 #endif
1040
1041   if (gst_caps_is_any (caps)) {
1042     goto done;
1043   }
1044
1045   if (!gst_caps_is_fixed (caps)) {
1046     g_assert (agg_klass->fixate_src_caps);
1047
1048     GST_DEBUG_OBJECT (self, "fixate caps from %" GST_PTR_FORMAT, caps);
1049     if (!(caps = agg_klass->fixate_src_caps (self, caps))) {
1050       GST_WARNING_OBJECT (self, "Subclass failed to fixate provided caps");
1051       ret = GST_FLOW_NOT_NEGOTIATED;
1052       goto done;
1053     }
1054     GST_DEBUG_OBJECT (self, "             to %" GST_PTR_FORMAT, caps);
1055   }
1056
1057   if (agg_klass->negotiated_src_caps) {
1058     if (!agg_klass->negotiated_src_caps (self, caps)) {
1059       GST_WARNING_OBJECT (self, "Subclass failed to accept negotiated caps");
1060       ret = GST_FLOW_NOT_NEGOTIATED;
1061       goto done;
1062     }
1063   }
1064
1065   gst_aggregator_set_src_caps (self, caps);
1066
1067   if (!gst_aggregator_do_allocation (self, caps)) {
1068     GST_WARNING_OBJECT (self, "Allocation negotiation failed");
1069     ret = GST_FLOW_NOT_NEGOTIATED;
1070   }
1071
1072 done:
1073   gst_caps_unref (downstream_caps);
1074   gst_caps_unref (template_caps);
1075
1076   if (caps)
1077     gst_caps_unref (caps);
1078
1079   return ret;
1080 }
1081
1082 static void
1083 gst_aggregator_aggregate_func (GstAggregator * self)
1084 {
1085   GstAggregatorPrivate *priv = self->priv;
1086   GstAggregatorClass *klass = GST_AGGREGATOR_GET_CLASS (self);
1087   gboolean timeout = FALSE;
1088
1089   if (self->priv->running == FALSE) {
1090     GST_DEBUG_OBJECT (self, "Not running anymore");
1091     return;
1092   }
1093
1094   GST_LOG_OBJECT (self, "Checking aggregate");
1095   while (priv->send_eos && priv->running) {
1096     GstFlowReturn flow_return = GST_FLOW_OK;
1097     gboolean processed_event = FALSE;
1098
1099     gst_aggregator_iterate_sinkpads (self, check_events, NULL);
1100
1101     if (!gst_aggregator_wait_and_check (self, &timeout))
1102       continue;
1103
1104     gst_aggregator_iterate_sinkpads (self, check_events, &processed_event);
1105     if (processed_event)
1106       continue;
1107
1108     if (gst_pad_check_reconfigure (GST_AGGREGATOR_SRC_PAD (self))) {
1109       flow_return = gst_aggregator_update_src_caps (self);
1110       if (flow_return != GST_FLOW_OK)
1111         gst_pad_mark_reconfigure (GST_AGGREGATOR_SRC_PAD (self));
1112     }
1113
1114     if (timeout || flow_return >= GST_FLOW_OK) {
1115       GST_TRACE_OBJECT (self, "Actually aggregating!");
1116       flow_return = klass->aggregate (self, timeout);
1117     }
1118
1119     if (flow_return == GST_AGGREGATOR_FLOW_NEED_DATA)
1120       continue;
1121
1122     GST_OBJECT_LOCK (self);
1123     if (flow_return == GST_FLOW_FLUSHING && priv->flush_seeking) {
1124       /* We don't want to set the pads to flushing, but we want to
1125        * stop the thread, so just break here */
1126       GST_OBJECT_UNLOCK (self);
1127       break;
1128     }
1129     GST_OBJECT_UNLOCK (self);
1130
1131     if (flow_return == GST_FLOW_EOS || flow_return == GST_FLOW_ERROR) {
1132       gst_aggregator_push_eos (self);
1133     }
1134
1135     GST_LOG_OBJECT (self, "flow return is %s", gst_flow_get_name (flow_return));
1136
1137     if (flow_return != GST_FLOW_OK) {
1138       GList *item;
1139
1140       GST_OBJECT_LOCK (self);
1141       for (item = GST_ELEMENT (self)->sinkpads; item; item = item->next) {
1142         GstAggregatorPad *aggpad = GST_AGGREGATOR_PAD (item->data);
1143
1144         gst_aggregator_pad_set_flushing (aggpad, flow_return, TRUE);
1145       }
1146       GST_OBJECT_UNLOCK (self);
1147       break;
1148     }
1149   }
1150
1151   /* Pause the task here, the only ways to get here are:
1152    * 1) We're stopping, in which case the task is stopped anyway
1153    * 2) We got a flow error above, in which case it might take
1154    *    some time to forward the flow return upstream and we
1155    *    would otherwise call the task function over and over
1156    *    again without doing anything
1157    */
1158   gst_pad_pause_task (self->srcpad);
1159 }
1160
1161 static gboolean
1162 gst_aggregator_start (GstAggregator * self)
1163 {
1164   GstAggregatorClass *klass;
1165   gboolean result;
1166
1167   self->priv->send_stream_start = TRUE;
1168   self->priv->send_segment = TRUE;
1169   self->priv->send_eos = TRUE;
1170   self->priv->srccaps = NULL;
1171
1172   gst_aggregator_set_allocation (self, NULL, NULL, NULL, NULL);
1173
1174   klass = GST_AGGREGATOR_GET_CLASS (self);
1175
1176   if (klass->start)
1177     result = klass->start (self);
1178   else
1179     result = TRUE;
1180
1181   return result;
1182 }
1183
1184 static gboolean
1185 _check_pending_flush_stop (GstAggregatorPad * pad)
1186 {
1187   gboolean res;
1188
1189   PAD_LOCK (pad);
1190   res = (!pad->priv->pending_flush_stop && !pad->priv->pending_flush_start);
1191   PAD_UNLOCK (pad);
1192
1193   return res;
1194 }
1195
1196 static gboolean
1197 gst_aggregator_stop_srcpad_task (GstAggregator * self, GstEvent * flush_start)
1198 {
1199   gboolean res = TRUE;
1200
1201   GST_INFO_OBJECT (self, "%s srcpad task",
1202       flush_start ? "Pausing" : "Stopping");
1203
1204   SRC_LOCK (self);
1205   self->priv->running = FALSE;
1206   SRC_BROADCAST (self);
1207   SRC_UNLOCK (self);
1208
1209   if (flush_start) {
1210     res = gst_pad_push_event (self->srcpad, flush_start);
1211   }
1212
1213   gst_pad_stop_task (self->srcpad);
1214
1215   return res;
1216 }
1217
1218 static void
1219 gst_aggregator_start_srcpad_task (GstAggregator * self)
1220 {
1221   GST_INFO_OBJECT (self, "Starting srcpad task");
1222
1223   self->priv->running = TRUE;
1224   gst_pad_start_task (GST_PAD (self->srcpad),
1225       (GstTaskFunction) gst_aggregator_aggregate_func, self, NULL);
1226 }
1227
1228 static GstFlowReturn
1229 gst_aggregator_flush (GstAggregator * self)
1230 {
1231   GstFlowReturn ret = GST_FLOW_OK;
1232   GstAggregatorPrivate *priv = self->priv;
1233   GstAggregatorClass *klass = GST_AGGREGATOR_GET_CLASS (self);
1234
1235   GST_DEBUG_OBJECT (self, "Flushing everything");
1236   GST_OBJECT_LOCK (self);
1237   priv->send_segment = TRUE;
1238   priv->flush_seeking = FALSE;
1239   priv->tags_changed = FALSE;
1240   GST_OBJECT_UNLOCK (self);
1241   if (klass->flush)
1242     ret = klass->flush (self);
1243
1244   return ret;
1245 }
1246
1247
1248 /* Called with GstAggregator's object lock held */
1249
1250 static gboolean
1251 gst_aggregator_all_flush_stop_received_locked (GstAggregator * self)
1252 {
1253   GList *tmp;
1254   GstAggregatorPad *tmppad;
1255
1256   for (tmp = GST_ELEMENT (self)->sinkpads; tmp; tmp = tmp->next) {
1257     tmppad = (GstAggregatorPad *) tmp->data;
1258
1259     if (_check_pending_flush_stop (tmppad) == FALSE) {
1260       GST_DEBUG_OBJECT (tmppad, "Is not last %i -- %i",
1261           tmppad->priv->pending_flush_start, tmppad->priv->pending_flush_stop);
1262       return FALSE;
1263     }
1264   }
1265
1266   return TRUE;
1267 }
1268
1269 static void
1270 gst_aggregator_flush_start (GstAggregator * self, GstAggregatorPad * aggpad,
1271     GstEvent * event)
1272 {
1273   GstAggregatorPrivate *priv = self->priv;
1274   GstAggregatorPadPrivate *padpriv = aggpad->priv;
1275
1276   gst_aggregator_pad_set_flushing (aggpad, GST_FLOW_FLUSHING, FALSE);
1277
1278   PAD_FLUSH_LOCK (aggpad);
1279   PAD_LOCK (aggpad);
1280   if (padpriv->pending_flush_start) {
1281     GST_DEBUG_OBJECT (aggpad, "Expecting FLUSH_STOP now");
1282
1283     padpriv->pending_flush_start = FALSE;
1284     padpriv->pending_flush_stop = TRUE;
1285   }
1286   PAD_UNLOCK (aggpad);
1287
1288   GST_OBJECT_LOCK (self);
1289   if (priv->flush_seeking) {
1290     /* If flush_seeking we forward the first FLUSH_START */
1291     if (priv->pending_flush_start) {
1292       priv->pending_flush_start = FALSE;
1293       GST_OBJECT_UNLOCK (self);
1294
1295       GST_INFO_OBJECT (self, "Flushing, pausing srcpad task");
1296       gst_aggregator_stop_srcpad_task (self, event);
1297
1298       GST_INFO_OBJECT (self, "Getting STREAM_LOCK while seeking");
1299       GST_PAD_STREAM_LOCK (self->srcpad);
1300       GST_LOG_OBJECT (self, "GOT STREAM_LOCK");
1301       event = NULL;
1302     } else {
1303       GST_OBJECT_UNLOCK (self);
1304       gst_event_unref (event);
1305     }
1306   } else {
1307     GST_OBJECT_UNLOCK (self);
1308     gst_event_unref (event);
1309   }
1310   PAD_FLUSH_UNLOCK (aggpad);
1311 }
1312
1313 /* Must be called with the the PAD_LOCK held */
1314 static void
1315 update_time_level (GstAggregatorPad * aggpad, gboolean head)
1316 {
1317   if (head) {
1318     if (GST_CLOCK_TIME_IS_VALID (aggpad->priv->head_position) &&
1319         aggpad->priv->head_segment.format == GST_FORMAT_TIME)
1320       aggpad->priv->head_time =
1321           gst_segment_to_running_time (&aggpad->priv->head_segment,
1322           GST_FORMAT_TIME, aggpad->priv->head_position);
1323     else
1324       aggpad->priv->head_time = GST_CLOCK_TIME_NONE;
1325
1326     if (!GST_CLOCK_TIME_IS_VALID (aggpad->priv->tail_time))
1327       aggpad->priv->tail_time = aggpad->priv->head_time;
1328   } else {
1329     if (GST_CLOCK_TIME_IS_VALID (aggpad->priv->tail_position) &&
1330         aggpad->segment.format == GST_FORMAT_TIME)
1331       aggpad->priv->tail_time =
1332           gst_segment_to_running_time (&aggpad->segment,
1333           GST_FORMAT_TIME, aggpad->priv->tail_position);
1334     else
1335       aggpad->priv->tail_time = aggpad->priv->head_time;
1336   }
1337
1338   if (aggpad->priv->head_time == GST_CLOCK_TIME_NONE ||
1339       aggpad->priv->tail_time == GST_CLOCK_TIME_NONE) {
1340     aggpad->priv->time_level = 0;
1341     return;
1342   }
1343
1344   if (aggpad->priv->tail_time > aggpad->priv->head_time)
1345     aggpad->priv->time_level = 0;
1346   else
1347     aggpad->priv->time_level = aggpad->priv->head_time -
1348         aggpad->priv->tail_time;
1349 }
1350
1351
1352 /* GstAggregator vmethods default implementations */
1353 static gboolean
1354 gst_aggregator_default_sink_event (GstAggregator * self,
1355     GstAggregatorPad * aggpad, GstEvent * event)
1356 {
1357   gboolean res = TRUE;
1358   GstPad *pad = GST_PAD (aggpad);
1359   GstAggregatorPrivate *priv = self->priv;
1360
1361   switch (GST_EVENT_TYPE (event)) {
1362     case GST_EVENT_FLUSH_START:
1363     {
1364       gst_aggregator_flush_start (self, aggpad, event);
1365       /* We forward only in one case: right after flush_seeking */
1366       event = NULL;
1367       goto eat;
1368     }
1369     case GST_EVENT_FLUSH_STOP:
1370     {
1371       GST_DEBUG_OBJECT (aggpad, "Got FLUSH_STOP");
1372
1373       gst_aggregator_pad_flush (aggpad, self);
1374       GST_OBJECT_LOCK (self);
1375       if (priv->flush_seeking) {
1376         g_atomic_int_set (&aggpad->priv->pending_flush_stop, FALSE);
1377         if (gst_aggregator_all_flush_stop_received_locked (self)) {
1378           GST_OBJECT_UNLOCK (self);
1379           /* That means we received FLUSH_STOP/FLUSH_STOP on
1380            * all sinkpads -- Seeking is Done... sending FLUSH_STOP */
1381           gst_aggregator_flush (self);
1382           gst_pad_push_event (self->srcpad, event);
1383           event = NULL;
1384           SRC_LOCK (self);
1385           priv->send_eos = TRUE;
1386           SRC_BROADCAST (self);
1387           SRC_UNLOCK (self);
1388
1389           GST_INFO_OBJECT (self, "Releasing source pad STREAM_LOCK");
1390           GST_PAD_STREAM_UNLOCK (self->srcpad);
1391           gst_aggregator_start_srcpad_task (self);
1392         } else {
1393           GST_OBJECT_UNLOCK (self);
1394         }
1395       } else {
1396         GST_OBJECT_UNLOCK (self);
1397       }
1398
1399       /* We never forward the event */
1400       goto eat;
1401     }
1402     case GST_EVENT_EOS:
1403     {
1404       GST_DEBUG_OBJECT (aggpad, "EOS");
1405
1406       /* We still have a buffer, and we don't want the subclass to have to
1407        * check for it. Mark pending_eos, eos will be set when steal_buffer is
1408        * called
1409        */
1410       SRC_LOCK (self);
1411       PAD_LOCK (aggpad);
1412       if (aggpad->priv->num_buffers == 0) {
1413         aggpad->priv->eos = TRUE;
1414       } else {
1415         aggpad->priv->pending_eos = TRUE;
1416       }
1417       PAD_UNLOCK (aggpad);
1418
1419       SRC_BROADCAST (self);
1420       SRC_UNLOCK (self);
1421       goto eat;
1422     }
1423     case GST_EVENT_SEGMENT:
1424     {
1425       PAD_LOCK (aggpad);
1426       GST_OBJECT_LOCK (aggpad);
1427       gst_event_copy_segment (event, &aggpad->segment);
1428       /* We've got a new segment, tail_position is now meaningless
1429        * and may interfere with the time_level calculation
1430        */
1431       aggpad->priv->tail_position = GST_CLOCK_TIME_NONE;
1432       update_time_level (aggpad, FALSE);
1433       GST_OBJECT_UNLOCK (aggpad);
1434       PAD_UNLOCK (aggpad);
1435
1436       GST_OBJECT_LOCK (self);
1437       self->priv->seqnum = gst_event_get_seqnum (event);
1438       GST_OBJECT_UNLOCK (self);
1439       goto eat;
1440     }
1441     case GST_EVENT_STREAM_START:
1442     {
1443       goto eat;
1444     }
1445     case GST_EVENT_GAP:
1446     {
1447       GstClockTime pts, endpts;
1448       GstClockTime duration;
1449       GstBuffer *gapbuf;
1450
1451       gst_event_parse_gap (event, &pts, &duration);
1452       gapbuf = gst_buffer_new ();
1453
1454       if (GST_CLOCK_TIME_IS_VALID (duration))
1455         endpts = pts + duration;
1456       else
1457         endpts = GST_CLOCK_TIME_NONE;
1458
1459       GST_OBJECT_LOCK (aggpad);
1460       res = gst_segment_clip (&aggpad->segment, GST_FORMAT_TIME, pts, endpts,
1461           &pts, &endpts);
1462       GST_OBJECT_UNLOCK (aggpad);
1463
1464       if (!res) {
1465         GST_WARNING_OBJECT (self, "GAP event outside segment, dropping");
1466         goto eat;
1467       }
1468
1469       if (GST_CLOCK_TIME_IS_VALID (endpts) && GST_CLOCK_TIME_IS_VALID (pts))
1470         duration = endpts - pts;
1471       else
1472         duration = GST_CLOCK_TIME_NONE;
1473
1474       GST_BUFFER_PTS (gapbuf) = pts;
1475       GST_BUFFER_DURATION (gapbuf) = duration;
1476       GST_BUFFER_FLAG_SET (gapbuf, GST_BUFFER_FLAG_GAP);
1477       GST_BUFFER_FLAG_SET (gapbuf, GST_BUFFER_FLAG_DROPPABLE);
1478
1479       /* Remove GAP event so we can replace it with the buffer */
1480       if (g_queue_peek_tail (&aggpad->priv->buffers) == event)
1481         gst_event_unref (g_queue_pop_tail (&aggpad->priv->buffers));
1482
1483       if (gst_aggregator_pad_chain_internal (self, aggpad, gapbuf, FALSE) !=
1484           GST_FLOW_OK) {
1485         GST_WARNING_OBJECT (self, "Failed to chain gap buffer");
1486         res = FALSE;
1487       }
1488
1489       goto eat;
1490     }
1491     case GST_EVENT_TAG:
1492     {
1493       GstTagList *tags;
1494
1495       gst_event_parse_tag (event, &tags);
1496
1497       if (gst_tag_list_get_scope (tags) == GST_TAG_SCOPE_STREAM) {
1498         gst_aggregator_merge_tags (self, tags, GST_TAG_MERGE_REPLACE);
1499         gst_event_unref (event);
1500         event = NULL;
1501         goto eat;
1502       }
1503       break;
1504     }
1505     default:
1506     {
1507       break;
1508     }
1509   }
1510
1511   GST_DEBUG_OBJECT (pad, "Forwarding event: %" GST_PTR_FORMAT, event);
1512   return gst_pad_event_default (pad, GST_OBJECT (self), event);
1513
1514 eat:
1515   GST_DEBUG_OBJECT (pad, "Eating event: %" GST_PTR_FORMAT, event);
1516   if (event)
1517     gst_event_unref (event);
1518
1519   return res;
1520 }
1521
1522 static inline gboolean
1523 gst_aggregator_stop_pad (GstAggregator * self, GstAggregatorPad * pad,
1524     gpointer unused_udata)
1525 {
1526   gst_aggregator_pad_flush (pad, self);
1527
1528   PAD_LOCK (pad);
1529   pad->priv->flow_return = GST_FLOW_FLUSHING;
1530   pad->priv->negotiated = FALSE;
1531   PAD_BROADCAST_EVENT (pad);
1532   PAD_UNLOCK (pad);
1533
1534   return TRUE;
1535 }
1536
1537 static gboolean
1538 gst_aggregator_stop (GstAggregator * agg)
1539 {
1540   GstAggregatorClass *klass;
1541   gboolean result;
1542
1543   gst_aggregator_reset_flow_values (agg);
1544
1545   gst_aggregator_iterate_sinkpads (agg, gst_aggregator_stop_pad, NULL);
1546
1547   klass = GST_AGGREGATOR_GET_CLASS (agg);
1548
1549   if (klass->stop)
1550     result = klass->stop (agg);
1551   else
1552     result = TRUE;
1553
1554   agg->priv->has_peer_latency = FALSE;
1555   agg->priv->peer_latency_live = FALSE;
1556   agg->priv->peer_latency_min = agg->priv->peer_latency_max = FALSE;
1557
1558   if (agg->priv->tags)
1559     gst_tag_list_unref (agg->priv->tags);
1560   agg->priv->tags = NULL;
1561
1562   gst_aggregator_set_allocation (agg, NULL, NULL, NULL, NULL);
1563
1564   return result;
1565 }
1566
1567 /* GstElement vmethods implementations */
1568 static GstStateChangeReturn
1569 gst_aggregator_change_state (GstElement * element, GstStateChange transition)
1570 {
1571   GstStateChangeReturn ret;
1572   GstAggregator *self = GST_AGGREGATOR (element);
1573
1574   switch (transition) {
1575     case GST_STATE_CHANGE_READY_TO_PAUSED:
1576       if (!gst_aggregator_start (self))
1577         goto error_start;
1578       break;
1579     default:
1580       break;
1581   }
1582
1583   if ((ret =
1584           GST_ELEMENT_CLASS (aggregator_parent_class)->change_state (element,
1585               transition)) == GST_STATE_CHANGE_FAILURE)
1586     goto failure;
1587
1588
1589   switch (transition) {
1590     case GST_STATE_CHANGE_PAUSED_TO_READY:
1591       if (!gst_aggregator_stop (self)) {
1592         /* What to do in this case? Error out? */
1593         GST_ERROR_OBJECT (self, "Subclass failed to stop.");
1594       }
1595       break;
1596     default:
1597       break;
1598   }
1599
1600   return ret;
1601
1602 /* ERRORS */
1603 failure:
1604   {
1605     GST_ERROR_OBJECT (element, "parent failed state change");
1606     return ret;
1607   }
1608 error_start:
1609   {
1610     GST_ERROR_OBJECT (element, "Subclass failed to start");
1611     return GST_STATE_CHANGE_FAILURE;
1612   }
1613 }
1614
1615 static void
1616 gst_aggregator_release_pad (GstElement * element, GstPad * pad)
1617 {
1618   GstAggregator *self = GST_AGGREGATOR (element);
1619   GstAggregatorPad *aggpad = GST_AGGREGATOR_PAD (pad);
1620
1621   GST_INFO_OBJECT (pad, "Removing pad");
1622
1623   SRC_LOCK (self);
1624   gst_aggregator_pad_set_flushing (aggpad, GST_FLOW_FLUSHING, TRUE);
1625   gst_element_remove_pad (element, pad);
1626
1627   self->priv->has_peer_latency = FALSE;
1628   SRC_BROADCAST (self);
1629   SRC_UNLOCK (self);
1630 }
1631
1632 static GstAggregatorPad *
1633 gst_aggregator_default_create_new_pad (GstAggregator * self,
1634     GstPadTemplate * templ, const gchar * req_name, const GstCaps * caps)
1635 {
1636   GstAggregatorPad *agg_pad;
1637   GstAggregatorPrivate *priv = self->priv;
1638   gint serial = 0;
1639   gchar *name = NULL;
1640
1641   if (templ->direction != GST_PAD_SINK)
1642     goto not_sink;
1643
1644   if (templ->presence != GST_PAD_REQUEST)
1645     goto not_request;
1646
1647   GST_OBJECT_LOCK (self);
1648   if (req_name == NULL || strlen (req_name) < 6
1649       || !g_str_has_prefix (req_name, "sink_")) {
1650     /* no name given when requesting the pad, use next available int */
1651     serial = ++priv->max_padserial;
1652   } else {
1653     /* parse serial number from requested padname */
1654     serial = g_ascii_strtoull (&req_name[5], NULL, 10);
1655     if (serial > priv->max_padserial)
1656       priv->max_padserial = serial;
1657   }
1658
1659   name = g_strdup_printf ("sink_%u", serial);
1660   agg_pad = g_object_new (GST_AGGREGATOR_GET_CLASS (self)->sinkpads_type,
1661       "name", name, "direction", GST_PAD_SINK, "template", templ, NULL);
1662   g_free (name);
1663
1664   GST_OBJECT_UNLOCK (self);
1665
1666   return agg_pad;
1667
1668   /* errors */
1669 not_sink:
1670   {
1671     GST_WARNING_OBJECT (self, "request new pad that is not a SINK pad");
1672     return NULL;
1673   }
1674 not_request:
1675   {
1676     GST_WARNING_OBJECT (self, "request new pad that is not a REQUEST pad");
1677     return NULL;
1678   }
1679 }
1680
1681 static GstPad *
1682 gst_aggregator_request_new_pad (GstElement * element,
1683     GstPadTemplate * templ, const gchar * req_name, const GstCaps * caps)
1684 {
1685   GstAggregator *self;
1686   GstAggregatorPad *agg_pad;
1687   GstAggregatorClass *klass = GST_AGGREGATOR_GET_CLASS (element);
1688   GstAggregatorPrivate *priv = GST_AGGREGATOR (element)->priv;
1689
1690   self = GST_AGGREGATOR (element);
1691
1692   agg_pad = klass->create_new_pad (self, templ, req_name, caps);
1693   if (!agg_pad) {
1694     GST_ERROR_OBJECT (element, "Couldn't create new pad");
1695     return NULL;
1696   }
1697
1698   GST_DEBUG_OBJECT (element, "Adding pad %s", GST_PAD_NAME (agg_pad));
1699
1700   if (priv->running)
1701     gst_pad_set_active (GST_PAD (agg_pad), TRUE);
1702
1703   /* add the pad to the element */
1704   gst_element_add_pad (element, GST_PAD (agg_pad));
1705
1706   return GST_PAD (agg_pad);
1707 }
1708
1709 /* Must be called with SRC_LOCK held */
1710
1711 static gboolean
1712 gst_aggregator_query_latency_unlocked (GstAggregator * self, GstQuery * query)
1713 {
1714   gboolean query_ret, live;
1715   GstClockTime our_latency, min, max;
1716
1717   query_ret = gst_pad_query_default (self->srcpad, GST_OBJECT (self), query);
1718
1719   if (!query_ret) {
1720     GST_WARNING_OBJECT (self, "Latency query failed");
1721     return FALSE;
1722   }
1723
1724   gst_query_parse_latency (query, &live, &min, &max);
1725
1726   our_latency = self->priv->latency;
1727
1728   if (G_UNLIKELY (!GST_CLOCK_TIME_IS_VALID (min))) {
1729     GST_ERROR_OBJECT (self, "Invalid minimum latency %" GST_TIME_FORMAT
1730         ". Please file a bug at " PACKAGE_BUGREPORT ".", GST_TIME_ARGS (min));
1731     return FALSE;
1732   }
1733
1734   if (min > max && GST_CLOCK_TIME_IS_VALID (max)) {
1735     GST_ELEMENT_WARNING (self, CORE, CLOCK, (NULL),
1736         ("Impossible to configure latency: max %" GST_TIME_FORMAT " < min %"
1737             GST_TIME_FORMAT ". Add queues or other buffering elements.",
1738             GST_TIME_ARGS (max), GST_TIME_ARGS (min)));
1739     return FALSE;
1740   }
1741
1742   self->priv->peer_latency_live = live;
1743   self->priv->peer_latency_min = min;
1744   self->priv->peer_latency_max = max;
1745   self->priv->has_peer_latency = TRUE;
1746
1747   /* add our own */
1748   min += our_latency;
1749   min += self->priv->sub_latency_min;
1750   if (GST_CLOCK_TIME_IS_VALID (self->priv->sub_latency_max)
1751       && GST_CLOCK_TIME_IS_VALID (max))
1752     max += self->priv->sub_latency_max + our_latency;
1753   else
1754     max = GST_CLOCK_TIME_NONE;
1755
1756   SRC_BROADCAST (self);
1757
1758   GST_DEBUG_OBJECT (self, "configured latency live:%s min:%" G_GINT64_FORMAT
1759       " max:%" G_GINT64_FORMAT, live ? "true" : "false", min, max);
1760
1761   gst_query_set_latency (query, live, min, max);
1762
1763   return query_ret;
1764 }
1765
1766 /*
1767  * MUST be called with the src_lock held.
1768  *
1769  * See  gst_aggregator_get_latency() for doc
1770  */
1771 static GstClockTime
1772 gst_aggregator_get_latency_unlocked (GstAggregator * self)
1773 {
1774   GstClockTime latency;
1775
1776   g_return_val_if_fail (GST_IS_AGGREGATOR (self), 0);
1777
1778   if (!self->priv->has_peer_latency) {
1779     GstQuery *query = gst_query_new_latency ();
1780     gboolean ret;
1781
1782     ret = gst_aggregator_query_latency_unlocked (self, query);
1783     gst_query_unref (query);
1784     if (!ret)
1785       return GST_CLOCK_TIME_NONE;
1786   }
1787
1788   if (!self->priv->has_peer_latency || !self->priv->peer_latency_live)
1789     return GST_CLOCK_TIME_NONE;
1790
1791   /* latency_min is never GST_CLOCK_TIME_NONE by construction */
1792   latency = self->priv->peer_latency_min;
1793
1794   /* add our own */
1795   latency += self->priv->latency;
1796   latency += self->priv->sub_latency_min;
1797
1798   return latency;
1799 }
1800
1801 /**
1802  * gst_aggregator_get_latency:
1803  * @self: a #GstAggregator
1804  *
1805  * Retrieves the latency values reported by @self in response to the latency
1806  * query, or %GST_CLOCK_TIME_NONE if there is not live source connected and the element
1807  * will not wait for the clock.
1808  *
1809  * Typically only called by subclasses.
1810  *
1811  * Returns: The latency or %GST_CLOCK_TIME_NONE if the element does not sync
1812  */
1813 GstClockTime
1814 gst_aggregator_get_latency (GstAggregator * self)
1815 {
1816   GstClockTime ret;
1817
1818   SRC_LOCK (self);
1819   ret = gst_aggregator_get_latency_unlocked (self);
1820   SRC_UNLOCK (self);
1821
1822   return ret;
1823 }
1824
1825 static gboolean
1826 gst_aggregator_send_event (GstElement * element, GstEvent * event)
1827 {
1828   GstAggregator *self = GST_AGGREGATOR (element);
1829
1830   GST_STATE_LOCK (element);
1831   if (GST_EVENT_TYPE (event) == GST_EVENT_SEEK &&
1832       GST_STATE (element) < GST_STATE_PAUSED) {
1833     gdouble rate;
1834     GstFormat fmt;
1835     GstSeekFlags flags;
1836     GstSeekType start_type, stop_type;
1837     gint64 start, stop;
1838
1839     gst_event_parse_seek (event, &rate, &fmt, &flags, &start_type,
1840         &start, &stop_type, &stop);
1841
1842     GST_OBJECT_LOCK (self);
1843     gst_segment_do_seek (&self->segment, rate, fmt, flags, start_type, start,
1844         stop_type, stop, NULL);
1845     self->priv->seqnum = gst_event_get_seqnum (event);
1846     self->priv->first_buffer = FALSE;
1847     GST_OBJECT_UNLOCK (self);
1848
1849     GST_DEBUG_OBJECT (element, "Storing segment %" GST_PTR_FORMAT, event);
1850   }
1851   GST_STATE_UNLOCK (element);
1852
1853
1854   return GST_ELEMENT_CLASS (aggregator_parent_class)->send_event (element,
1855       event);
1856 }
1857
1858 static gboolean
1859 gst_aggregator_default_src_query (GstAggregator * self, GstQuery * query)
1860 {
1861   gboolean res = TRUE;
1862
1863   switch (GST_QUERY_TYPE (query)) {
1864     case GST_QUERY_SEEKING:
1865     {
1866       GstFormat format;
1867
1868       /* don't pass it along as some (file)sink might claim it does
1869        * whereas with a collectpads in between that will not likely work */
1870       gst_query_parse_seeking (query, &format, NULL, NULL, NULL);
1871       gst_query_set_seeking (query, format, FALSE, 0, -1);
1872       res = TRUE;
1873
1874       break;
1875     }
1876     case GST_QUERY_LATENCY:
1877       SRC_LOCK (self);
1878       res = gst_aggregator_query_latency_unlocked (self, query);
1879       SRC_UNLOCK (self);
1880       break;
1881     default:
1882       return gst_pad_query_default (self->srcpad, GST_OBJECT (self), query);
1883   }
1884
1885   return res;
1886 }
1887
1888 static gboolean
1889 gst_aggregator_event_forward_func (GstPad * pad, gpointer user_data)
1890 {
1891   EventData *evdata = user_data;
1892   gboolean ret = TRUE;
1893   GstPad *peer = gst_pad_get_peer (pad);
1894   GstAggregatorPad *aggpad = GST_AGGREGATOR_PAD (pad);
1895
1896   if (peer) {
1897     if (evdata->only_to_active_pads && aggpad->priv->first_buffer) {
1898       GST_DEBUG_OBJECT (pad, "not sending event to inactive pad");
1899       ret = TRUE;
1900     } else {
1901       ret = gst_pad_send_event (peer, gst_event_ref (evdata->event));
1902       GST_DEBUG_OBJECT (pad, "return of event push is %d", ret);
1903       gst_object_unref (peer);
1904     }
1905   }
1906
1907   if (ret == FALSE) {
1908     if (GST_EVENT_TYPE (evdata->event) == GST_EVENT_SEEK) {
1909       GstQuery *seeking = gst_query_new_seeking (GST_FORMAT_TIME);
1910
1911       GST_DEBUG_OBJECT (pad, "Event %" GST_PTR_FORMAT " failed", evdata->event);
1912
1913       if (gst_pad_query (peer, seeking)) {
1914         gboolean seekable;
1915
1916         gst_query_parse_seeking (seeking, NULL, &seekable, NULL, NULL);
1917
1918         if (seekable == FALSE) {
1919           GST_INFO_OBJECT (pad,
1920               "Source not seekable, We failed but it does not matter!");
1921
1922           ret = TRUE;
1923         }
1924       } else {
1925         GST_ERROR_OBJECT (pad, "Query seeking FAILED");
1926       }
1927
1928       gst_query_unref (seeking);
1929     }
1930
1931     if (evdata->flush) {
1932       PAD_LOCK (aggpad);
1933       aggpad->priv->pending_flush_start = FALSE;
1934       aggpad->priv->pending_flush_stop = FALSE;
1935       PAD_UNLOCK (aggpad);
1936     }
1937   } else {
1938     evdata->one_actually_seeked = TRUE;
1939   }
1940
1941   evdata->result &= ret;
1942
1943   /* Always send to all pads */
1944   return FALSE;
1945 }
1946
1947 static EventData
1948 gst_aggregator_forward_event_to_all_sinkpads (GstAggregator * self,
1949     GstEvent * event, gboolean flush, gboolean only_to_active_pads)
1950 {
1951   EventData evdata;
1952
1953   evdata.event = event;
1954   evdata.result = TRUE;
1955   evdata.flush = flush;
1956   evdata.one_actually_seeked = FALSE;
1957   evdata.only_to_active_pads = only_to_active_pads;
1958
1959   /* We first need to set all pads as flushing in a first pass
1960    * as flush_start flush_stop is sometimes sent synchronously
1961    * while we send the seek event */
1962   if (flush) {
1963     GList *l;
1964
1965     GST_OBJECT_LOCK (self);
1966     for (l = GST_ELEMENT_CAST (self)->sinkpads; l != NULL; l = l->next) {
1967       GstAggregatorPad *pad = l->data;
1968
1969       PAD_LOCK (pad);
1970       pad->priv->pending_flush_start = TRUE;
1971       pad->priv->pending_flush_stop = FALSE;
1972       PAD_UNLOCK (pad);
1973     }
1974     GST_OBJECT_UNLOCK (self);
1975   }
1976
1977   gst_pad_forward (self->srcpad, gst_aggregator_event_forward_func, &evdata);
1978
1979   gst_event_unref (event);
1980
1981   return evdata;
1982 }
1983
1984 static gboolean
1985 gst_aggregator_do_seek (GstAggregator * self, GstEvent * event)
1986 {
1987   gdouble rate;
1988   GstFormat fmt;
1989   GstSeekFlags flags;
1990   GstSeekType start_type, stop_type;
1991   gint64 start, stop;
1992   gboolean flush;
1993   EventData evdata;
1994   GstAggregatorPrivate *priv = self->priv;
1995
1996   gst_event_parse_seek (event, &rate, &fmt, &flags, &start_type,
1997       &start, &stop_type, &stop);
1998
1999   GST_INFO_OBJECT (self, "starting SEEK");
2000
2001   flush = flags & GST_SEEK_FLAG_FLUSH;
2002
2003   GST_OBJECT_LOCK (self);
2004   if (flush) {
2005     priv->pending_flush_start = TRUE;
2006     priv->flush_seeking = TRUE;
2007   }
2008
2009   gst_segment_do_seek (&self->segment, rate, fmt, flags, start_type, start,
2010       stop_type, stop, NULL);
2011
2012   /* Seeking sets a position */
2013   self->priv->first_buffer = FALSE;
2014   GST_OBJECT_UNLOCK (self);
2015
2016   /* forward the seek upstream */
2017   evdata =
2018       gst_aggregator_forward_event_to_all_sinkpads (self, event, flush, FALSE);
2019   event = NULL;
2020
2021   if (!evdata.result || !evdata.one_actually_seeked) {
2022     GST_OBJECT_LOCK (self);
2023     priv->flush_seeking = FALSE;
2024     priv->pending_flush_start = FALSE;
2025     GST_OBJECT_UNLOCK (self);
2026   }
2027
2028   GST_INFO_OBJECT (self, "seek done, result: %d", evdata.result);
2029
2030   return evdata.result;
2031 }
2032
2033 static gboolean
2034 gst_aggregator_default_src_event (GstAggregator * self, GstEvent * event)
2035 {
2036   EventData evdata;
2037   gboolean res = TRUE;
2038
2039   switch (GST_EVENT_TYPE (event)) {
2040     case GST_EVENT_SEEK:
2041     {
2042       gst_event_ref (event);
2043       res = gst_aggregator_do_seek (self, event);
2044       gst_event_unref (event);
2045       event = NULL;
2046       goto done;
2047     }
2048     case GST_EVENT_NAVIGATION:
2049     {
2050       /* navigation is rather pointless. */
2051       res = FALSE;
2052       gst_event_unref (event);
2053       goto done;
2054     }
2055     default:
2056     {
2057       break;
2058     }
2059   }
2060
2061   /* Don't forward QOS events to pads that had no active buffer yet. Otherwise
2062    * they will receive a QOS event that has earliest_time=0 (because we can't
2063    * have negative timestamps), and consider their buffer as too late */
2064   evdata =
2065       gst_aggregator_forward_event_to_all_sinkpads (self, event, FALSE,
2066       GST_EVENT_TYPE (event) == GST_EVENT_QOS);
2067   res = evdata.result;
2068
2069 done:
2070   return res;
2071 }
2072
2073 static gboolean
2074 gst_aggregator_src_pad_event_func (GstPad * pad, GstObject * parent,
2075     GstEvent * event)
2076 {
2077   GstAggregatorClass *klass = GST_AGGREGATOR_GET_CLASS (parent);
2078
2079   return klass->src_event (GST_AGGREGATOR (parent), event);
2080 }
2081
2082 static gboolean
2083 gst_aggregator_src_pad_query_func (GstPad * pad, GstObject * parent,
2084     GstQuery * query)
2085 {
2086   GstAggregatorClass *klass = GST_AGGREGATOR_GET_CLASS (parent);
2087
2088   return klass->src_query (GST_AGGREGATOR (parent), query);
2089 }
2090
2091 static gboolean
2092 gst_aggregator_src_pad_activate_mode_func (GstPad * pad,
2093     GstObject * parent, GstPadMode mode, gboolean active)
2094 {
2095   GstAggregator *self = GST_AGGREGATOR (parent);
2096   GstAggregatorClass *klass = GST_AGGREGATOR_GET_CLASS (parent);
2097
2098   if (klass->src_activate) {
2099     if (klass->src_activate (self, mode, active) == FALSE) {
2100       return FALSE;
2101     }
2102   }
2103
2104   if (active == TRUE) {
2105     switch (mode) {
2106       case GST_PAD_MODE_PUSH:
2107       {
2108         GST_INFO_OBJECT (pad, "Activating pad!");
2109         gst_aggregator_start_srcpad_task (self);
2110         return TRUE;
2111       }
2112       default:
2113       {
2114         GST_ERROR_OBJECT (pad, "Only supported mode is PUSH");
2115         return FALSE;
2116       }
2117     }
2118   }
2119
2120   /* deactivating */
2121   GST_INFO_OBJECT (self, "Deactivating srcpad");
2122   gst_aggregator_stop_srcpad_task (self, FALSE);
2123
2124   return TRUE;
2125 }
2126
2127 static gboolean
2128 gst_aggregator_default_sink_query (GstAggregator * self,
2129     GstAggregatorPad * aggpad, GstQuery * query)
2130 {
2131   GstPad *pad = GST_PAD (aggpad);
2132
2133   if (GST_QUERY_TYPE (query) == GST_QUERY_ALLOCATION) {
2134     GstQuery *decide_query = NULL;
2135     GstAggregatorClass *agg_class;
2136     gboolean ret;
2137
2138     GST_OBJECT_LOCK (self);
2139     PAD_LOCK (aggpad);
2140     if (G_UNLIKELY (!aggpad->priv->negotiated)) {
2141       GST_DEBUG_OBJECT (self,
2142           "not negotiated yet, can't answer ALLOCATION query");
2143       PAD_UNLOCK (aggpad);
2144       GST_OBJECT_UNLOCK (self);
2145
2146       return FALSE;
2147     }
2148
2149     if ((decide_query = self->priv->allocation_query))
2150       gst_query_ref (decide_query);
2151     PAD_UNLOCK (aggpad);
2152     GST_OBJECT_UNLOCK (self);
2153
2154     GST_DEBUG_OBJECT (self,
2155         "calling propose allocation with query %" GST_PTR_FORMAT, decide_query);
2156
2157     agg_class = GST_AGGREGATOR_GET_CLASS (self);
2158
2159     /* pass the query to the propose_allocation vmethod if any */
2160     if (agg_class->propose_allocation)
2161       ret = agg_class->propose_allocation (self, aggpad, decide_query, query);
2162     else
2163       ret = FALSE;
2164
2165     if (decide_query)
2166       gst_query_unref (decide_query);
2167
2168     GST_DEBUG_OBJECT (self, "ALLOCATION ret %d, %" GST_PTR_FORMAT, ret, query);
2169     return ret;
2170   }
2171
2172   return gst_pad_query_default (pad, GST_OBJECT (self), query);
2173 }
2174
2175 static void
2176 gst_aggregator_finalize (GObject * object)
2177 {
2178   GstAggregator *self = (GstAggregator *) object;
2179
2180   g_mutex_clear (&self->priv->src_lock);
2181   g_cond_clear (&self->priv->src_cond);
2182
2183   G_OBJECT_CLASS (aggregator_parent_class)->finalize (object);
2184 }
2185
2186 /*
2187  * gst_aggregator_set_latency_property:
2188  * @agg: a #GstAggregator
2189  * @latency: the new latency value (in nanoseconds).
2190  *
2191  * Sets the new latency value to @latency. This value is used to limit the
2192  * amount of time a pad waits for data to appear before considering the pad
2193  * as unresponsive.
2194  */
2195 static void
2196 gst_aggregator_set_latency_property (GstAggregator * self, gint64 latency)
2197 {
2198   gboolean changed;
2199
2200   g_return_if_fail (GST_IS_AGGREGATOR (self));
2201   g_return_if_fail (GST_CLOCK_TIME_IS_VALID (latency));
2202
2203   SRC_LOCK (self);
2204   changed = (self->priv->latency != latency);
2205
2206   if (changed) {
2207     GList *item;
2208
2209     GST_OBJECT_LOCK (self);
2210     /* First lock all the pads */
2211     for (item = GST_ELEMENT_CAST (self)->sinkpads; item; item = item->next) {
2212       GstAggregatorPad *aggpad = GST_AGGREGATOR_PAD (item->data);
2213       PAD_LOCK (aggpad);
2214     }
2215
2216     self->priv->latency = latency;
2217
2218     SRC_BROADCAST (self);
2219
2220     /* Now wake up the pads */
2221     for (item = GST_ELEMENT_CAST (self)->sinkpads; item; item = item->next) {
2222       GstAggregatorPad *aggpad = GST_AGGREGATOR_PAD (item->data);
2223       PAD_BROADCAST_EVENT (aggpad);
2224       PAD_UNLOCK (aggpad);
2225     }
2226     GST_OBJECT_UNLOCK (self);
2227   }
2228
2229   SRC_UNLOCK (self);
2230
2231   if (changed)
2232     gst_element_post_message (GST_ELEMENT_CAST (self),
2233         gst_message_new_latency (GST_OBJECT_CAST (self)));
2234 }
2235
2236 /*
2237  * gst_aggregator_get_latency_property:
2238  * @agg: a #GstAggregator
2239  *
2240  * Gets the latency value. See gst_aggregator_set_latency for
2241  * more details.
2242  *
2243  * Returns: The time in nanoseconds to wait for data to arrive on a sink pad
2244  * before a pad is deemed unresponsive. A value of -1 means an
2245  * unlimited time.
2246  */
2247 static gint64
2248 gst_aggregator_get_latency_property (GstAggregator * agg)
2249 {
2250   gint64 res;
2251
2252   g_return_val_if_fail (GST_IS_AGGREGATOR (agg), -1);
2253
2254   GST_OBJECT_LOCK (agg);
2255   res = agg->priv->latency;
2256   GST_OBJECT_UNLOCK (agg);
2257
2258   return res;
2259 }
2260
2261 static void
2262 gst_aggregator_set_property (GObject * object, guint prop_id,
2263     const GValue * value, GParamSpec * pspec)
2264 {
2265   GstAggregator *agg = GST_AGGREGATOR (object);
2266
2267   switch (prop_id) {
2268     case PROP_LATENCY:
2269       gst_aggregator_set_latency_property (agg, g_value_get_int64 (value));
2270       break;
2271     case PROP_START_TIME_SELECTION:
2272       agg->priv->start_time_selection = g_value_get_enum (value);
2273       break;
2274     case PROP_START_TIME:
2275       agg->priv->start_time = g_value_get_uint64 (value);
2276       break;
2277     default:
2278       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
2279       break;
2280   }
2281 }
2282
2283 static void
2284 gst_aggregator_get_property (GObject * object, guint prop_id,
2285     GValue * value, GParamSpec * pspec)
2286 {
2287   GstAggregator *agg = GST_AGGREGATOR (object);
2288
2289   switch (prop_id) {
2290     case PROP_LATENCY:
2291       g_value_set_int64 (value, gst_aggregator_get_latency_property (agg));
2292       break;
2293     case PROP_START_TIME_SELECTION:
2294       g_value_set_enum (value, agg->priv->start_time_selection);
2295       break;
2296     case PROP_START_TIME:
2297       g_value_set_uint64 (value, agg->priv->start_time);
2298       break;
2299     default:
2300       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
2301       break;
2302   }
2303 }
2304
2305 /* GObject vmethods implementations */
2306 static void
2307 gst_aggregator_class_init (GstAggregatorClass * klass)
2308 {
2309   GObjectClass *gobject_class = (GObjectClass *) klass;
2310   GstElementClass *gstelement_class = (GstElementClass *) klass;
2311
2312   aggregator_parent_class = g_type_class_peek_parent (klass);
2313   g_type_class_add_private (klass, sizeof (GstAggregatorPrivate));
2314
2315   GST_DEBUG_CATEGORY_INIT (aggregator_debug, "aggregator",
2316       GST_DEBUG_FG_MAGENTA, "GstAggregator");
2317
2318   klass->sinkpads_type = GST_TYPE_AGGREGATOR_PAD;
2319
2320   klass->sink_event = gst_aggregator_default_sink_event;
2321   klass->sink_query = gst_aggregator_default_sink_query;
2322
2323   klass->src_event = gst_aggregator_default_src_event;
2324   klass->src_query = gst_aggregator_default_src_query;
2325
2326   klass->create_new_pad = gst_aggregator_default_create_new_pad;
2327   klass->update_src_caps = gst_aggregator_default_update_src_caps;
2328   klass->fixate_src_caps = gst_aggregator_default_fixate_src_caps;
2329   klass->negotiated_src_caps = gst_aggregator_default_negotiated_src_caps;
2330
2331   gstelement_class->request_new_pad =
2332       GST_DEBUG_FUNCPTR (gst_aggregator_request_new_pad);
2333   gstelement_class->send_event = GST_DEBUG_FUNCPTR (gst_aggregator_send_event);
2334   gstelement_class->release_pad =
2335       GST_DEBUG_FUNCPTR (gst_aggregator_release_pad);
2336   gstelement_class->change_state =
2337       GST_DEBUG_FUNCPTR (gst_aggregator_change_state);
2338
2339   gobject_class->set_property = gst_aggregator_set_property;
2340   gobject_class->get_property = gst_aggregator_get_property;
2341   gobject_class->finalize = gst_aggregator_finalize;
2342
2343   g_object_class_install_property (gobject_class, PROP_LATENCY,
2344       g_param_spec_int64 ("latency", "Buffer latency",
2345           "Additional latency in live mode to allow upstream "
2346           "to take longer to produce buffers for the current "
2347           "position (in nanoseconds)", 0,
2348           (G_MAXLONG == G_MAXINT64) ? G_MAXINT64 : (G_MAXLONG * GST_SECOND - 1),
2349           DEFAULT_LATENCY, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
2350
2351   g_object_class_install_property (gobject_class, PROP_START_TIME_SELECTION,
2352       g_param_spec_enum ("start-time-selection", "Start Time Selection",
2353           "Decides which start time is output",
2354           gst_aggregator_start_time_selection_get_type (),
2355           DEFAULT_START_TIME_SELECTION,
2356           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
2357
2358   g_object_class_install_property (gobject_class, PROP_START_TIME,
2359       g_param_spec_uint64 ("start-time", "Start Time",
2360           "Start time to use if start-time-selection=set", 0,
2361           G_MAXUINT64,
2362           DEFAULT_START_TIME, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
2363
2364   GST_DEBUG_REGISTER_FUNCPTR (gst_aggregator_stop_pad);
2365 }
2366
2367 static void
2368 gst_aggregator_init (GstAggregator * self, GstAggregatorClass * klass)
2369 {
2370   GstPadTemplate *pad_template;
2371   GstAggregatorPrivate *priv;
2372
2373   g_return_if_fail (klass->aggregate != NULL);
2374
2375   self->priv =
2376       G_TYPE_INSTANCE_GET_PRIVATE (self, GST_TYPE_AGGREGATOR,
2377       GstAggregatorPrivate);
2378
2379   priv = self->priv;
2380
2381   pad_template =
2382       gst_element_class_get_pad_template (GST_ELEMENT_CLASS (klass), "src");
2383   g_return_if_fail (pad_template != NULL);
2384
2385   priv->max_padserial = -1;
2386   priv->tags_changed = FALSE;
2387
2388   self->priv->peer_latency_live = FALSE;
2389   self->priv->peer_latency_min = self->priv->sub_latency_min = 0;
2390   self->priv->peer_latency_max = self->priv->sub_latency_max = 0;
2391   self->priv->has_peer_latency = FALSE;
2392   gst_aggregator_reset_flow_values (self);
2393
2394   self->srcpad = gst_pad_new_from_template (pad_template, "src");
2395
2396   gst_pad_set_event_function (self->srcpad,
2397       GST_DEBUG_FUNCPTR (gst_aggregator_src_pad_event_func));
2398   gst_pad_set_query_function (self->srcpad,
2399       GST_DEBUG_FUNCPTR (gst_aggregator_src_pad_query_func));
2400   gst_pad_set_activatemode_function (self->srcpad,
2401       GST_DEBUG_FUNCPTR (gst_aggregator_src_pad_activate_mode_func));
2402
2403   gst_element_add_pad (GST_ELEMENT (self), self->srcpad);
2404
2405   self->priv->latency = DEFAULT_LATENCY;
2406   self->priv->start_time_selection = DEFAULT_START_TIME_SELECTION;
2407   self->priv->start_time = DEFAULT_START_TIME;
2408
2409   g_mutex_init (&self->priv->src_lock);
2410   g_cond_init (&self->priv->src_cond);
2411 }
2412
2413 /* we can't use G_DEFINE_ABSTRACT_TYPE because we need the klass in the _init
2414  * method to get to the padtemplates */
2415 GType
2416 gst_aggregator_get_type (void)
2417 {
2418   static volatile gsize type = 0;
2419
2420   if (g_once_init_enter (&type)) {
2421     GType _type;
2422     static const GTypeInfo info = {
2423       sizeof (GstAggregatorClass),
2424       NULL,
2425       NULL,
2426       (GClassInitFunc) gst_aggregator_class_init,
2427       NULL,
2428       NULL,
2429       sizeof (GstAggregator),
2430       0,
2431       (GInstanceInitFunc) gst_aggregator_init,
2432     };
2433
2434     _type = g_type_register_static (GST_TYPE_ELEMENT,
2435         "GstAggregator", &info, G_TYPE_FLAG_ABSTRACT);
2436     g_once_init_leave (&type, _type);
2437   }
2438   return type;
2439 }
2440
2441 /* Must be called with SRC lock and PAD lock held */
2442 static gboolean
2443 gst_aggregator_pad_has_space (GstAggregator * self, GstAggregatorPad * aggpad)
2444 {
2445   /* Empty queue always has space */
2446   if (aggpad->priv->num_buffers == 0 && aggpad->priv->clipped_buffer == NULL)
2447     return TRUE;
2448
2449   /* We also want at least two buffers, one is being processed and one is ready
2450    * for the next iteration when we operate in live mode. */
2451   if (self->priv->peer_latency_live && aggpad->priv->num_buffers < 2)
2452     return TRUE;
2453
2454   /* zero latency, if there is a buffer, it's full */
2455   if (self->priv->latency == 0)
2456     return FALSE;
2457
2458   /* Allow no more buffers than the latency */
2459   return (aggpad->priv->time_level <= self->priv->latency);
2460 }
2461
2462 /* Must be called with the PAD_LOCK held */
2463 static void
2464 apply_buffer (GstAggregatorPad * aggpad, GstBuffer * buffer, gboolean head)
2465 {
2466   GstClockTime timestamp;
2467
2468   if (GST_BUFFER_DTS_IS_VALID (buffer))
2469     timestamp = GST_BUFFER_DTS (buffer);
2470   else
2471     timestamp = GST_BUFFER_PTS (buffer);
2472
2473   if (timestamp == GST_CLOCK_TIME_NONE) {
2474     if (head)
2475       timestamp = aggpad->priv->head_position;
2476     else
2477       timestamp = aggpad->priv->tail_position;
2478   }
2479
2480   /* add duration */
2481   if (GST_BUFFER_DURATION_IS_VALID (buffer))
2482     timestamp += GST_BUFFER_DURATION (buffer);
2483
2484   if (head)
2485     aggpad->priv->head_position = timestamp;
2486   else
2487     aggpad->priv->tail_position = timestamp;
2488
2489   update_time_level (aggpad, head);
2490 }
2491
2492 static GstFlowReturn
2493 gst_aggregator_pad_chain_internal (GstAggregator * self,
2494     GstAggregatorPad * aggpad, GstBuffer * buffer, gboolean head)
2495 {
2496   GstFlowReturn flow_return;
2497   GstClockTime buf_pts;
2498
2499   GST_DEBUG_OBJECT (aggpad, "Start chaining a buffer %" GST_PTR_FORMAT, buffer);
2500
2501   PAD_FLUSH_LOCK (aggpad);
2502
2503   PAD_LOCK (aggpad);
2504   flow_return = aggpad->priv->flow_return;
2505   if (flow_return != GST_FLOW_OK)
2506     goto flushing;
2507
2508   if (aggpad->priv->pending_eos == TRUE)
2509     goto eos;
2510
2511   PAD_UNLOCK (aggpad);
2512
2513   buf_pts = GST_BUFFER_PTS (buffer);
2514
2515   for (;;) {
2516     SRC_LOCK (self);
2517     GST_OBJECT_LOCK (self);
2518     PAD_LOCK (aggpad);
2519
2520     if (aggpad->priv->first_buffer) {
2521       self->priv->has_peer_latency = FALSE;
2522       aggpad->priv->first_buffer = FALSE;
2523     }
2524
2525     if (gst_aggregator_pad_has_space (self, aggpad)
2526         && aggpad->priv->flow_return == GST_FLOW_OK) {
2527       if (head)
2528         g_queue_push_head (&aggpad->priv->buffers, buffer);
2529       else
2530         g_queue_push_tail (&aggpad->priv->buffers, buffer);
2531       apply_buffer (aggpad, buffer, head);
2532       aggpad->priv->num_buffers++;
2533       buffer = NULL;
2534       SRC_BROADCAST (self);
2535       break;
2536     }
2537
2538     flow_return = aggpad->priv->flow_return;
2539     if (flow_return != GST_FLOW_OK) {
2540       GST_OBJECT_UNLOCK (self);
2541       SRC_UNLOCK (self);
2542       goto flushing;
2543     }
2544     GST_DEBUG_OBJECT (aggpad, "Waiting for buffer to be consumed");
2545     GST_OBJECT_UNLOCK (self);
2546     SRC_UNLOCK (self);
2547     PAD_WAIT_EVENT (aggpad);
2548
2549     PAD_UNLOCK (aggpad);
2550   }
2551
2552   if (self->priv->first_buffer) {
2553     GstClockTime start_time;
2554
2555     switch (self->priv->start_time_selection) {
2556       case GST_AGGREGATOR_START_TIME_SELECTION_ZERO:
2557       default:
2558         start_time = 0;
2559         break;
2560       case GST_AGGREGATOR_START_TIME_SELECTION_FIRST:
2561         GST_OBJECT_LOCK (aggpad);
2562         if (aggpad->priv->head_segment.format == GST_FORMAT_TIME) {
2563           start_time = buf_pts;
2564           if (start_time != -1) {
2565             start_time = MAX (start_time, aggpad->priv->head_segment.start);
2566             start_time =
2567                 gst_segment_to_running_time (&aggpad->priv->head_segment,
2568                 GST_FORMAT_TIME, start_time);
2569           }
2570         } else {
2571           start_time = 0;
2572           GST_WARNING_OBJECT (aggpad,
2573               "Ignoring request of selecting the first start time "
2574               "as the segment is a %s segment instead of a time segment",
2575               gst_format_get_name (aggpad->segment.format));
2576         }
2577         GST_OBJECT_UNLOCK (aggpad);
2578         break;
2579       case GST_AGGREGATOR_START_TIME_SELECTION_SET:
2580         start_time = self->priv->start_time;
2581         if (start_time == -1)
2582           start_time = 0;
2583         break;
2584     }
2585
2586     if (start_time != -1) {
2587       if (self->segment.position == -1)
2588         self->segment.position = start_time;
2589       else
2590         self->segment.position = MIN (start_time, self->segment.position);
2591
2592       GST_DEBUG_OBJECT (self, "Selecting start time %" GST_TIME_FORMAT,
2593           GST_TIME_ARGS (start_time));
2594     }
2595   }
2596
2597   PAD_UNLOCK (aggpad);
2598   GST_OBJECT_UNLOCK (self);
2599   SRC_UNLOCK (self);
2600
2601   PAD_FLUSH_UNLOCK (aggpad);
2602
2603   GST_DEBUG_OBJECT (aggpad, "Done chaining");
2604
2605   return flow_return;
2606
2607 flushing:
2608   PAD_UNLOCK (aggpad);
2609   PAD_FLUSH_UNLOCK (aggpad);
2610
2611   GST_DEBUG_OBJECT (aggpad, "Pad is %s, dropping buffer",
2612       gst_flow_get_name (flow_return));
2613   if (buffer)
2614     gst_buffer_unref (buffer);
2615
2616   return flow_return;
2617
2618 eos:
2619   PAD_UNLOCK (aggpad);
2620   PAD_FLUSH_UNLOCK (aggpad);
2621
2622   gst_buffer_unref (buffer);
2623   GST_DEBUG_OBJECT (aggpad, "We are EOS already...");
2624
2625   return GST_FLOW_EOS;
2626 }
2627
2628 static GstFlowReturn
2629 gst_aggregator_pad_chain (GstPad * pad, GstObject * object, GstBuffer * buffer)
2630 {
2631   return gst_aggregator_pad_chain_internal (GST_AGGREGATOR_CAST (object),
2632       GST_AGGREGATOR_PAD_CAST (pad), buffer, TRUE);
2633 }
2634
2635 static gboolean
2636 gst_aggregator_pad_query_func (GstPad * pad, GstObject * parent,
2637     GstQuery * query)
2638 {
2639   GstAggregator *self = GST_AGGREGATOR (parent);
2640   GstAggregatorPad *aggpad = GST_AGGREGATOR_PAD (pad);
2641   GstAggregatorClass *klass = GST_AGGREGATOR_GET_CLASS (parent);
2642
2643   if (GST_QUERY_IS_SERIALIZED (query)) {
2644     GstStructure *s;
2645     gboolean ret = FALSE;
2646
2647     SRC_LOCK (self);
2648     PAD_LOCK (aggpad);
2649
2650     if (aggpad->priv->flow_return != GST_FLOW_OK) {
2651       SRC_UNLOCK (self);
2652       goto flushing;
2653     }
2654
2655     g_queue_push_head (&aggpad->priv->buffers, query);
2656     SRC_BROADCAST (self);
2657     SRC_UNLOCK (self);
2658
2659     while (!gst_aggregator_pad_queue_is_empty (aggpad)
2660         && aggpad->priv->flow_return == GST_FLOW_OK) {
2661       GST_DEBUG_OBJECT (aggpad, "Waiting for buffer to be consumed");
2662       PAD_WAIT_EVENT (aggpad);
2663     }
2664
2665     s = gst_query_writable_structure (query);
2666     if (gst_structure_get_boolean (s, "gst-aggregator-retval", &ret))
2667       gst_structure_remove_field (s, "gst-aggregator-retval");
2668     else
2669       g_queue_remove (&aggpad->priv->buffers, query);
2670
2671     if (aggpad->priv->flow_return != GST_FLOW_OK)
2672       goto flushing;
2673
2674     PAD_UNLOCK (aggpad);
2675
2676     return ret;
2677   }
2678
2679   return klass->sink_query (self, aggpad, query);
2680
2681 flushing:
2682   GST_DEBUG_OBJECT (aggpad, "Pad is %s, dropping query",
2683       gst_flow_get_name (aggpad->priv->flow_return));
2684   PAD_UNLOCK (aggpad);
2685
2686   return FALSE;
2687 }
2688
2689 static GstFlowReturn
2690 gst_aggregator_pad_event_func (GstPad * pad, GstObject * parent,
2691     GstEvent * event)
2692 {
2693   GstFlowReturn ret = GST_FLOW_OK;
2694   GstAggregator *self = GST_AGGREGATOR (parent);
2695   GstAggregatorPad *aggpad = GST_AGGREGATOR_PAD (pad);
2696   GstAggregatorClass *klass = GST_AGGREGATOR_GET_CLASS (parent);
2697
2698   if (GST_EVENT_IS_SERIALIZED (event) && GST_EVENT_TYPE (event) != GST_EVENT_EOS
2699       /* && GST_EVENT_TYPE (event) != GST_EVENT_SEGMENT_DONE */ ) {
2700     SRC_LOCK (self);
2701     PAD_LOCK (aggpad);
2702
2703     if (aggpad->priv->flow_return != GST_FLOW_OK
2704         && GST_EVENT_TYPE (event) != GST_EVENT_FLUSH_STOP) {
2705       ret = aggpad->priv->flow_return;
2706       goto flushing;
2707     }
2708
2709     if (GST_EVENT_TYPE (event) == GST_EVENT_SEGMENT) {
2710       GST_OBJECT_LOCK (aggpad);
2711       gst_event_copy_segment (event, &aggpad->priv->head_segment);
2712       aggpad->priv->head_position = aggpad->priv->head_segment.position;
2713       update_time_level (aggpad, TRUE);
2714       GST_OBJECT_UNLOCK (aggpad);
2715     }
2716
2717     if (GST_EVENT_TYPE (event) != GST_EVENT_FLUSH_STOP) {
2718       GST_DEBUG_OBJECT (aggpad, "Store event in queue: %" GST_PTR_FORMAT,
2719           event);
2720       g_queue_push_head (&aggpad->priv->buffers, event);
2721       event = NULL;
2722       SRC_BROADCAST (self);
2723     }
2724     PAD_UNLOCK (aggpad);
2725     SRC_UNLOCK (self);
2726   }
2727
2728   if (event) {
2729     if (!klass->sink_event (self, aggpad, event)) {
2730       /* Copied from GstPad to convert boolean to a GstFlowReturn in
2731        * the event handling func */
2732       ret = GST_FLOW_ERROR;
2733     }
2734   }
2735
2736   return ret;
2737
2738 flushing:
2739   GST_DEBUG_OBJECT (aggpad, "Pad is %s, dropping event",
2740       gst_flow_get_name (aggpad->priv->flow_return));
2741   PAD_UNLOCK (aggpad);
2742   SRC_UNLOCK (self);
2743   if (GST_EVENT_IS_STICKY (event))
2744     gst_pad_store_sticky_event (pad, event);
2745   gst_event_unref (event);
2746
2747   return ret;
2748 }
2749
2750 static gboolean
2751 gst_aggregator_pad_activate_mode_func (GstPad * pad,
2752     GstObject * parent, GstPadMode mode, gboolean active)
2753 {
2754   GstAggregator *self = GST_AGGREGATOR (parent);
2755   GstAggregatorPad *aggpad = GST_AGGREGATOR_PAD (pad);
2756
2757   if (active == FALSE) {
2758     SRC_LOCK (self);
2759     gst_aggregator_pad_set_flushing (aggpad, GST_FLOW_FLUSHING, TRUE);
2760     SRC_BROADCAST (self);
2761     SRC_UNLOCK (self);
2762   } else {
2763     PAD_LOCK (aggpad);
2764     aggpad->priv->flow_return = GST_FLOW_OK;
2765     PAD_BROADCAST_EVENT (aggpad);
2766     PAD_UNLOCK (aggpad);
2767   }
2768
2769   return TRUE;
2770 }
2771
2772 /***********************************
2773  * GstAggregatorPad implementation  *
2774  ************************************/
2775 G_DEFINE_TYPE (GstAggregatorPad, gst_aggregator_pad, GST_TYPE_PAD);
2776
2777 static void
2778 gst_aggregator_pad_constructed (GObject * object)
2779 {
2780   GstPad *pad = GST_PAD (object);
2781
2782   gst_pad_set_chain_function (pad,
2783       GST_DEBUG_FUNCPTR (gst_aggregator_pad_chain));
2784   gst_pad_set_event_full_function_full (pad,
2785       GST_DEBUG_FUNCPTR (gst_aggregator_pad_event_func), NULL, NULL);
2786   gst_pad_set_query_function (pad,
2787       GST_DEBUG_FUNCPTR (gst_aggregator_pad_query_func));
2788   gst_pad_set_activatemode_function (pad,
2789       GST_DEBUG_FUNCPTR (gst_aggregator_pad_activate_mode_func));
2790 }
2791
2792 static void
2793 gst_aggregator_pad_finalize (GObject * object)
2794 {
2795   GstAggregatorPad *pad = (GstAggregatorPad *) object;
2796
2797   g_cond_clear (&pad->priv->event_cond);
2798   g_mutex_clear (&pad->priv->flush_lock);
2799   g_mutex_clear (&pad->priv->lock);
2800
2801   G_OBJECT_CLASS (gst_aggregator_pad_parent_class)->finalize (object);
2802 }
2803
2804 static void
2805 gst_aggregator_pad_dispose (GObject * object)
2806 {
2807   GstAggregatorPad *pad = (GstAggregatorPad *) object;
2808
2809   gst_aggregator_pad_set_flushing (pad, GST_FLOW_FLUSHING, TRUE);
2810
2811   G_OBJECT_CLASS (gst_aggregator_pad_parent_class)->dispose (object);
2812 }
2813
2814 static void
2815 gst_aggregator_pad_class_init (GstAggregatorPadClass * klass)
2816 {
2817   GObjectClass *gobject_class = (GObjectClass *) klass;
2818
2819   g_type_class_add_private (klass, sizeof (GstAggregatorPadPrivate));
2820
2821   gobject_class->constructed = gst_aggregator_pad_constructed;
2822   gobject_class->finalize = gst_aggregator_pad_finalize;
2823   gobject_class->dispose = gst_aggregator_pad_dispose;
2824 }
2825
2826 static void
2827 gst_aggregator_pad_init (GstAggregatorPad * pad)
2828 {
2829   pad->priv =
2830       G_TYPE_INSTANCE_GET_PRIVATE (pad, GST_TYPE_AGGREGATOR_PAD,
2831       GstAggregatorPadPrivate);
2832
2833   g_queue_init (&pad->priv->buffers);
2834   g_cond_init (&pad->priv->event_cond);
2835
2836   g_mutex_init (&pad->priv->flush_lock);
2837   g_mutex_init (&pad->priv->lock);
2838
2839   gst_aggregator_pad_reset_unlocked (pad);
2840   pad->priv->negotiated = FALSE;
2841 }
2842
2843 /* Must be called with the PAD_LOCK held */
2844 static void
2845 gst_aggregator_pad_buffer_consumed (GstAggregatorPad * pad)
2846 {
2847   pad->priv->num_buffers--;
2848   GST_TRACE_OBJECT (pad, "Consuming buffer");
2849   if (gst_aggregator_pad_queue_is_empty (pad) && pad->priv->pending_eos) {
2850     pad->priv->pending_eos = FALSE;
2851     pad->priv->eos = TRUE;
2852   }
2853   PAD_BROADCAST_EVENT (pad);
2854 }
2855
2856 /* Must be called with the PAD_LOCK held */
2857 static void
2858 gst_aggregator_pad_clip_buffer_unlocked (GstAggregatorPad * pad)
2859 {
2860   GstAggregator *self = NULL;
2861   GstAggregatorClass *aggclass = NULL;
2862   GstBuffer *buffer = NULL;
2863
2864   while (pad->priv->clipped_buffer == NULL &&
2865       GST_IS_BUFFER (g_queue_peek_tail (&pad->priv->buffers))) {
2866     buffer = g_queue_pop_tail (&pad->priv->buffers);
2867
2868     apply_buffer (pad, buffer, FALSE);
2869
2870     /* We only take the parent here so that it's not taken if the buffer is
2871      * already clipped or if the queue is empty.
2872      */
2873     if (self == NULL) {
2874       self = GST_AGGREGATOR (gst_pad_get_parent_element (GST_PAD (pad)));
2875       if (self == NULL) {
2876         gst_buffer_unref (buffer);
2877         return;
2878       }
2879
2880       aggclass = GST_AGGREGATOR_GET_CLASS (self);
2881     }
2882
2883     if (aggclass->clip) {
2884       GST_TRACE_OBJECT (pad, "Clipping: %" GST_PTR_FORMAT, buffer);
2885
2886       buffer = aggclass->clip (self, pad, buffer);
2887
2888       if (buffer == NULL) {
2889         gst_aggregator_pad_buffer_consumed (pad);
2890         GST_TRACE_OBJECT (pad, "Clipping consumed the buffer");
2891       }
2892     }
2893
2894     pad->priv->clipped_buffer = buffer;
2895   }
2896
2897   if (self)
2898     gst_object_unref (self);
2899 }
2900
2901 /**
2902  * gst_aggregator_pad_steal_buffer:
2903  * @pad: the pad to get buffer from
2904  *
2905  * Steal the ref to the buffer currently queued in @pad.
2906  *
2907  * Returns: (transfer full): The buffer in @pad or NULL if no buffer was
2908  *   queued. You should unref the buffer after usage.
2909  */
2910 GstBuffer *
2911 gst_aggregator_pad_steal_buffer (GstAggregatorPad * pad)
2912 {
2913   GstBuffer *buffer;
2914
2915   PAD_LOCK (pad);
2916
2917   gst_aggregator_pad_clip_buffer_unlocked (pad);
2918
2919   buffer = pad->priv->clipped_buffer;
2920   pad->priv->clipped_buffer = NULL;
2921
2922   if (buffer) {
2923     gst_aggregator_pad_buffer_consumed (pad);
2924     GST_DEBUG_OBJECT (pad, "Consumed: %" GST_PTR_FORMAT, buffer);
2925   }
2926
2927   PAD_UNLOCK (pad);
2928
2929   return buffer;
2930 }
2931
2932 /**
2933  * gst_aggregator_pad_drop_buffer:
2934  * @pad: the pad where to drop any pending buffer
2935  *
2936  * Drop the buffer currently queued in @pad.
2937  *
2938  * Returns: TRUE if there was a buffer queued in @pad, or FALSE if not.
2939  */
2940 gboolean
2941 gst_aggregator_pad_drop_buffer (GstAggregatorPad * pad)
2942 {
2943   GstBuffer *buf;
2944
2945   buf = gst_aggregator_pad_steal_buffer (pad);
2946
2947   if (buf == NULL)
2948     return FALSE;
2949
2950   gst_buffer_unref (buf);
2951   return TRUE;
2952 }
2953
2954 /**
2955  * gst_aggregator_pad_get_buffer:
2956  * @pad: the pad to get buffer from
2957  *
2958  * Returns: (transfer full): A reference to the buffer in @pad or
2959  * NULL if no buffer was queued. You should unref the buffer after
2960  * usage.
2961  */
2962 GstBuffer *
2963 gst_aggregator_pad_get_buffer (GstAggregatorPad * pad)
2964 {
2965   GstBuffer *buffer;
2966
2967   PAD_LOCK (pad);
2968
2969   gst_aggregator_pad_clip_buffer_unlocked (pad);
2970
2971   if (pad->priv->clipped_buffer) {
2972     buffer = gst_buffer_ref (pad->priv->clipped_buffer);
2973   } else {
2974     buffer = NULL;
2975   }
2976   PAD_UNLOCK (pad);
2977
2978   return buffer;
2979 }
2980
2981 gboolean
2982 gst_aggregator_pad_is_eos (GstAggregatorPad * pad)
2983 {
2984   gboolean is_eos;
2985
2986   PAD_LOCK (pad);
2987   is_eos = pad->priv->eos;
2988   PAD_UNLOCK (pad);
2989
2990   return is_eos;
2991 }
2992
2993 /**
2994  * gst_aggregator_merge_tags:
2995  * @self: a #GstAggregator
2996  * @tags: a #GstTagList to merge
2997  * @mode: the #GstTagMergeMode to use
2998  *
2999  * Adds tags to so-called pending tags, which will be processed
3000  * before pushing out data downstream.
3001  *
3002  * Note that this is provided for convenience, and the subclass is
3003  * not required to use this and can still do tag handling on its own.
3004  *
3005  * MT safe.
3006  */
3007 void
3008 gst_aggregator_merge_tags (GstAggregator * self,
3009     const GstTagList * tags, GstTagMergeMode mode)
3010 {
3011   GstTagList *otags;
3012
3013   g_return_if_fail (GST_IS_AGGREGATOR (self));
3014   g_return_if_fail (tags == NULL || GST_IS_TAG_LIST (tags));
3015
3016   /* FIXME Check if we can use OBJECT lock here! */
3017   GST_OBJECT_LOCK (self);
3018   if (tags)
3019     GST_DEBUG_OBJECT (self, "merging tags %" GST_PTR_FORMAT, tags);
3020   otags = self->priv->tags;
3021   self->priv->tags = gst_tag_list_merge (self->priv->tags, tags, mode);
3022   if (otags)
3023     gst_tag_list_unref (otags);
3024   self->priv->tags_changed = TRUE;
3025   GST_OBJECT_UNLOCK (self);
3026 }
3027
3028 /**
3029  * gst_aggregator_set_latency:
3030  * @self: a #GstAggregator
3031  * @min_latency: minimum latency
3032  * @max_latency: maximum latency
3033  *
3034  * Lets #GstAggregator sub-classes tell the baseclass what their internal
3035  * latency is. Will also post a LATENCY message on the bus so the pipeline
3036  * can reconfigure its global latency.
3037  */
3038 void
3039 gst_aggregator_set_latency (GstAggregator * self,
3040     GstClockTime min_latency, GstClockTime max_latency)
3041 {
3042   gboolean changed = FALSE;
3043
3044   g_return_if_fail (GST_IS_AGGREGATOR (self));
3045   g_return_if_fail (GST_CLOCK_TIME_IS_VALID (min_latency));
3046   g_return_if_fail (max_latency >= min_latency);
3047
3048   SRC_LOCK (self);
3049   if (self->priv->sub_latency_min != min_latency) {
3050     self->priv->sub_latency_min = min_latency;
3051     changed = TRUE;
3052   }
3053   if (self->priv->sub_latency_max != max_latency) {
3054     self->priv->sub_latency_max = max_latency;
3055     changed = TRUE;
3056   }
3057
3058   if (changed)
3059     SRC_BROADCAST (self);
3060   SRC_UNLOCK (self);
3061
3062   if (changed) {
3063     gst_element_post_message (GST_ELEMENT_CAST (self),
3064         gst_message_new_latency (GST_OBJECT_CAST (self)));
3065   }
3066 }
3067
3068 /**
3069  * gst_aggregator_get_buffer_pool:
3070  * @self: a #GstAggregator
3071  *
3072  * Returns: (transfer full): the instance of the #GstBufferPool used
3073  * by @trans; free it after use it
3074  */
3075 GstBufferPool *
3076 gst_aggregator_get_buffer_pool (GstAggregator * self)
3077 {
3078   GstBufferPool *pool;
3079
3080   g_return_val_if_fail (GST_IS_AGGREGATOR (self), NULL);
3081
3082   GST_OBJECT_LOCK (self);
3083   pool = self->priv->pool;
3084   if (pool)
3085     gst_object_ref (pool);
3086   GST_OBJECT_UNLOCK (self);
3087
3088   return pool;
3089 }
3090
3091 /**
3092  * gst_aggregator_get_allocator:
3093  * @self: a #GstAggregator
3094  * @allocator: (out) (allow-none) (transfer full): the #GstAllocator
3095  * used
3096  * @params: (out) (allow-none) (transfer full): the
3097  * #GstAllocationParams of @allocator
3098  *
3099  * Lets #GstAggregator sub-classes get the memory @allocator
3100  * acquired by the base class and its @params.
3101  *
3102  * Unref the @allocator after use it.
3103  */
3104 void
3105 gst_aggregator_get_allocator (GstAggregator * self,
3106     GstAllocator ** allocator, GstAllocationParams * params)
3107 {
3108   g_return_if_fail (GST_IS_AGGREGATOR (self));
3109
3110   if (allocator)
3111     *allocator = self->priv->allocator ?
3112         gst_object_ref (self->priv->allocator) : NULL;
3113
3114   if (params)
3115     *params = self->priv->allocation_params;
3116 }