gstoggmux.c 46.4 KB
Newer Older
Wim Taymans's avatar
Wim Taymans committed
1
2
/* OGG muxer plugin for GStreamer
 * Copyright (C) 2004 Wim Taymans <wim@fluendo.com>
3
 * Copyright (C) 2006 Thomas Vander Stichele <thomas at apestaart dot org>
Wim Taymans's avatar
Wim Taymans committed
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Library General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public
 * License along with this library; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <gst/gst.h>
Wim Taymans's avatar
Wim Taymans committed
26
27
#include <gst/base/gstcollectpads.h>

Wim Taymans's avatar
Wim Taymans committed
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
#include <ogg/ogg.h>
/* memcpy - if someone knows a way to get rid of it, please speak up 
 * note: the ogg docs even say you need this... */
#include <string.h>
#include <time.h>

GST_DEBUG_CATEGORY_STATIC (gst_ogg_mux_debug);
#define GST_CAT_DEFAULT gst_ogg_mux_debug

#define GST_TYPE_OGG_MUX (gst_ogg_mux_get_type())
#define GST_OGG_MUX(obj) (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_OGG_MUX, GstOggMux))
#define GST_OGG_MUX_CLASS(klass) (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_OGG_MUX, GstOggMux))
#define GST_IS_OGG_MUX(obj) (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_OGG_MUX))
#define GST_IS_OGG_MUX_CLASS(obj) (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_OGG_MUX))

43
44
45
46
47
48
49
50
/* This isn't generally what you'd want with an end-time macro, because
   technically the end time of a buffer with invalid duration is invalid. But
   for sorting ogg pages this is what we want. */
#define GST_BUFFER_END_TIME(buf) \
    (GST_BUFFER_DURATION_IS_VALID (buf) \
    ? GST_BUFFER_TIMESTAMP (buf) + GST_BUFFER_DURATION (buf) \
    : GST_BUFFER_TIMESTAMP (buf))

51
52
#define GST_GP_FORMAT "[gp %8" G_GINT64_FORMAT "]"

Wim Taymans's avatar
Wim Taymans committed
53
54
55
typedef struct _GstOggMux GstOggMux;
typedef struct _GstOggMuxClass GstOggMuxClass;

Wim Taymans's avatar
Wim Taymans committed
56
57
58
59
60
61
62
typedef enum
{
  GST_OGG_PAD_STATE_CONTROL = 0,
  GST_OGG_PAD_STATE_DATA = 1
}
GstOggPadState;

Wim Taymans's avatar
Wim Taymans committed
63
64
65
/* all information needed for one ogg stream */
typedef struct
{
Wim Taymans's avatar
Wim Taymans committed
66
  GstCollectData collect;       /* we extend the CollectData */
Wim Taymans's avatar
Wim Taymans committed
67
68
69
70
71
72
73

  GstBuffer *buffer;            /* the queued buffer for this pad */

  gint serial;
  ogg_stream_state stream;
  gint64 packetno;              /* number of next packet */
  gint64 pageno;                /* number of next page */
74
  guint64 duration;             /* duration of current page */
Wim Taymans's avatar
Wim Taymans committed
75
  gboolean eos;
76
  gint64 offset;
77
78
79
80
81
82
83
  GstClockTime timestamp;       /* timestamp of the first packet on the next
                                 * page to be dequeued */
  GstClockTime timestamp_end;   /* end timestamp of last complete packet on
                                   the next page to be dequeued */
  GstClockTime gp_time;         /* time corresponding to the gp value of the
                                   last complete packet on the next page to be
                                   dequeued */
Wim Taymans's avatar
Wim Taymans committed
84

Wim Taymans's avatar
Wim Taymans committed
85
  GstOggPadState state;         /* state of the pad */
86
87

  GList *headers;
88

89
90
  GQueue *pagebuffers;          /* List of pages in buffers ready for pushing */

91
92
93
  gboolean new_page;            /* starting a new page */
  gboolean first_delta;         /* was the first packet in the page a delta */
  gboolean prev_delta;          /* was the previous buffer a delta frame */
Wim Taymans's avatar
Wim Taymans committed
94
95
96
97
98
99
100
}
GstOggPad;

struct _GstOggMux
{
  GstElement element;

Wim Taymans's avatar
Wim Taymans committed
101
  /* source pad */
Wim Taymans's avatar
Wim Taymans committed
102
103
  GstPad *srcpad;

Wim Taymans's avatar
Wim Taymans committed
104
105
  /* sinkpads */
  GstCollectPads *collect;
Wim Taymans's avatar
Wim Taymans committed
106

Wim Taymans's avatar
Wim Taymans committed
107
  /* the pad we are currently using to fill a page */
Wim Taymans's avatar
Wim Taymans committed
108
109
110
111
112
113
114
  GstOggPad *pulling;

  /* next timestamp for the page */
  GstClockTime next_ts;

  /* offset in stream */
  guint64 offset;
115
116
117

  /* need_headers */
  gboolean need_headers;
118
119

  guint64 max_delay;
120
  guint64 max_page_delay;
121

122
  GstOggPad *delta_pad;         /* when a delta frame is detected on a stream, we mark
123
124
                                   pages as delta frames up to the page that has the
                                   keyframe */
125

Wim Taymans's avatar
Wim Taymans committed
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
};

typedef enum
{
  GST_OGG_FLAG_BOS = GST_ELEMENT_FLAG_LAST,
  GST_OGG_FLAG_EOS
}
GstOggFlag;

struct _GstOggMuxClass
{
  GstElementClass parent_class;
};

/* elementfactory information */
static GstElementDetails gst_ogg_mux_details = GST_ELEMENT_DETAILS ("ogg muxer",
    "Codec/Muxer",
    "mux ogg streams (info about ogg: http://xiph.org)",
    "Wim Taymans <wim@fluendo.com>");

/* OggMux signals and args */
enum
{
  /* FILL ME */
  LAST_SIGNAL
};

Wim Taymans's avatar
Wim Taymans committed
153
/* set to 0.5 seconds by default */
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
154
155
#define DEFAULT_MAX_DELAY       G_GINT64_CONSTANT(500000000)
#define DEFAULT_MAX_PAGE_DELAY  G_GINT64_CONSTANT(500000000)
Wim Taymans's avatar
Wim Taymans committed
156
157
enum
{
158
159
  ARG_0,
  ARG_MAX_DELAY,
160
  ARG_MAX_PAGE_DELAY,
Wim Taymans's avatar
Wim Taymans committed
161
162
163
164
165
166
167
168
169
170
171
};

static GstStaticPadTemplate src_factory = GST_STATIC_PAD_TEMPLATE ("src",
    GST_PAD_SRC,
    GST_PAD_ALWAYS,
    GST_STATIC_CAPS ("application/ogg")
    );

static GstStaticPadTemplate sink_factory = GST_STATIC_PAD_TEMPLATE ("sink_%d",
    GST_PAD_SINK,
    GST_PAD_REQUEST,
172
173
    GST_STATIC_CAPS ("video/x-theora; "
        "audio/x-vorbis; audio/x-flac; audio/x-speex; "
174
        "application/x-ogm-video; application/x-ogm-audio; video/x-dirac")
Wim Taymans's avatar
Wim Taymans committed
175
176
177
178
179
    );

static void gst_ogg_mux_base_init (gpointer g_class);
static void gst_ogg_mux_class_init (GstOggMuxClass * klass);
static void gst_ogg_mux_init (GstOggMux * ogg_mux);
180
static void gst_ogg_mux_finalize (GObject * object);
Wim Taymans's avatar
Wim Taymans committed
181

Wim Taymans's avatar
Wim Taymans committed
182
183
static GstFlowReturn
gst_ogg_mux_collected (GstCollectPads * pads, GstOggMux * ogg_mux);
Wim Taymans's avatar
Wim Taymans committed
184
185
186
static gboolean gst_ogg_mux_handle_src_event (GstPad * pad, GstEvent * event);
static GstPad *gst_ogg_mux_request_new_pad (GstElement * element,
    GstPadTemplate * templ, const gchar * name);
187
188
static void gst_ogg_mux_release_pad (GstElement * element, GstPad * pad);

Wim Taymans's avatar
Wim Taymans committed
189
190
191
192
static void gst_ogg_mux_set_property (GObject * object,
    guint prop_id, const GValue * value, GParamSpec * pspec);
static void gst_ogg_mux_get_property (GObject * object,
    guint prop_id, GValue * value, GParamSpec * pspec);
193
194
static GstStateChangeReturn gst_ogg_mux_change_state (GstElement * element,
    GstStateChange transition);
Wim Taymans's avatar
Wim Taymans committed
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248

static GstElementClass *parent_class = NULL;

/*static guint gst_ogg_mux_signals[LAST_SIGNAL] = { 0 }; */

GType
gst_ogg_mux_get_type (void)
{
  static GType ogg_mux_type = 0;

  if (!ogg_mux_type) {
    static const GTypeInfo ogg_mux_info = {
      sizeof (GstOggMuxClass),
      gst_ogg_mux_base_init,
      NULL,
      (GClassInitFunc) gst_ogg_mux_class_init,
      NULL,
      NULL,
      sizeof (GstOggMux),
      0,
      (GInstanceInitFunc) gst_ogg_mux_init,
    };

    ogg_mux_type =
        g_type_register_static (GST_TYPE_ELEMENT, "GstOggMux", &ogg_mux_info,
        0);
  }
  return ogg_mux_type;
}

static void
gst_ogg_mux_base_init (gpointer g_class)
{
  GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);

  gst_element_class_add_pad_template (element_class,
      gst_static_pad_template_get (&src_factory));
  gst_element_class_add_pad_template (element_class,
      gst_static_pad_template_get (&sink_factory));

  gst_element_class_set_details (element_class, &gst_ogg_mux_details);
}

static void
gst_ogg_mux_class_init (GstOggMuxClass * klass)
{
  GObjectClass *gobject_class;
  GstElementClass *gstelement_class;

  gobject_class = (GObjectClass *) klass;
  gstelement_class = (GstElementClass *) klass;

  parent_class = g_type_class_ref (GST_TYPE_ELEMENT);

249
  gobject_class->finalize = gst_ogg_mux_finalize;
250
251
252
  gobject_class->get_property = gst_ogg_mux_get_property;
  gobject_class->set_property = gst_ogg_mux_set_property;

Wim Taymans's avatar
Wim Taymans committed
253
  gstelement_class->request_new_pad = gst_ogg_mux_request_new_pad;
254
  gstelement_class->release_pad = gst_ogg_mux_release_pad;
Wim Taymans's avatar
Wim Taymans committed
255

256
257
258
259
  g_object_class_install_property (gobject_class, ARG_MAX_DELAY,
      g_param_spec_uint64 ("max-delay", "Max delay",
          "Maximum delay in multiplexing streams", 0, G_MAXUINT64,
          DEFAULT_MAX_DELAY, (GParamFlags) G_PARAM_READWRITE));
260
261
262
263
  g_object_class_install_property (gobject_class, ARG_MAX_PAGE_DELAY,
      g_param_spec_uint64 ("max-page-delay", "Max page delay",
          "Maximum delay for sending out a page", 0, G_MAXUINT64,
          DEFAULT_MAX_PAGE_DELAY, (GParamFlags) G_PARAM_READWRITE));
264

Wim Taymans's avatar
Wim Taymans committed
265
266
267
268
  gstelement_class->change_state = gst_ogg_mux_change_state;

}

Wim Taymans's avatar
Wim Taymans committed
269
#if 0
Wim Taymans's avatar
Wim Taymans committed
270
271
272
273
274
static const GstEventMask *
gst_ogg_mux_get_sink_event_masks (GstPad * pad)
{
  static const GstEventMask gst_ogg_mux_sink_event_masks[] = {
    {GST_EVENT_EOS, 0},
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
275
    {GST_EVENT_DISCONTINUOUS, 0},
Wim Taymans's avatar
Wim Taymans committed
276
277
278
279
280
    {0,}
  };

  return gst_ogg_mux_sink_event_masks;
}
Wim Taymans's avatar
Wim Taymans committed
281
#endif
Wim Taymans's avatar
Wim Taymans committed
282

283
284
285
286
287
288
289
290
291
292
static void
gst_ogg_mux_clear (GstOggMux * ogg_mux)
{
  ogg_mux->pulling = NULL;
  ogg_mux->need_headers = TRUE;
  ogg_mux->max_delay = DEFAULT_MAX_DELAY;
  ogg_mux->max_page_delay = DEFAULT_MAX_PAGE_DELAY;
  ogg_mux->delta_pad = NULL;
}

Wim Taymans's avatar
Wim Taymans committed
293
294
295
296
297
298
299
300
301
302
303
static void
gst_ogg_mux_init (GstOggMux * ogg_mux)
{
  GstElementClass *klass = GST_ELEMENT_GET_CLASS (ogg_mux);

  ogg_mux->srcpad =
      gst_pad_new_from_template (gst_element_class_get_pad_template (klass,
          "src"), "src");
  gst_pad_set_event_function (ogg_mux->srcpad, gst_ogg_mux_handle_src_event);
  gst_element_add_pad (GST_ELEMENT (ogg_mux), ogg_mux->srcpad);

304
  GST_OBJECT_FLAG_SET (GST_ELEMENT (ogg_mux), GST_OGG_FLAG_BOS);
Wim Taymans's avatar
Wim Taymans committed
305
306
307
308

  /* seed random number generator for creation of serial numbers */
  srand (time (NULL));

309
310
  ogg_mux->collect = gst_collect_pads_new ();
  gst_collect_pads_set_function (ogg_mux->collect,
311
      (GstCollectPadsFunction) gst_ogg_mux_collected, ogg_mux);
312

313
  gst_ogg_mux_clear (ogg_mux);
Wim Taymans's avatar
Wim Taymans committed
314
315
}

316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
static void
gst_ogg_mux_finalize (GObject * object)
{
  GstOggMux *ogg_mux;

  ogg_mux = GST_OGG_MUX (object);

  if (ogg_mux->collect) {
    gst_object_unref (ogg_mux->collect);
    ogg_mux->collect = NULL;
  }

  G_OBJECT_CLASS (parent_class)->finalize (object);
}

Wim Taymans's avatar
Wim Taymans committed
331
static GstPadLinkReturn
332
gst_ogg_mux_sinkconnect (GstPad * pad, GstPad * peer)
Wim Taymans's avatar
Wim Taymans committed
333
334
335
336
337
{
  GstOggMux *ogg_mux;

  ogg_mux = GST_OGG_MUX (gst_pad_get_parent (pad));

338
  GST_DEBUG_OBJECT (ogg_mux, "sinkconnect triggered on %s", GST_PAD_NAME (pad));
339

340
  gst_object_unref (ogg_mux);
Wim Taymans's avatar
Wim Taymans committed
341
342
343
344
345
346
347
348
349
350

  return GST_PAD_LINK_OK;
}

static GstPad *
gst_ogg_mux_request_new_pad (GstElement * element,
    GstPadTemplate * templ, const gchar * req_name)
{
  GstOggMux *ogg_mux;
  GstPad *newpad;
Wim Taymans's avatar
Wim Taymans committed
351
  GstElementClass *klass;
Wim Taymans's avatar
Wim Taymans committed
352
353
354

  g_return_val_if_fail (templ != NULL, NULL);

Wim Taymans's avatar
Wim Taymans committed
355
356
  if (templ->direction != GST_PAD_SINK)
    goto wrong_direction;
Wim Taymans's avatar
Wim Taymans committed
357
358
359
360

  g_return_val_if_fail (GST_IS_OGG_MUX (element), NULL);
  ogg_mux = GST_OGG_MUX (element);

Wim Taymans's avatar
Wim Taymans committed
361
362
363
364
365
366
  klass = GST_ELEMENT_GET_CLASS (element);

  if (templ != gst_element_class_get_pad_template (klass, "sink_%d"))
    goto wrong_template;

  {
Wim Taymans's avatar
Wim Taymans committed
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
    gint serial;
    gchar *name;

    if (req_name == NULL || strlen (req_name) < 6) {
      /* no name given when requesting the pad, use random serial number */
      serial = rand ();
    } else {
      /* parse serial number from requested padname */
      serial = atoi (&req_name[5]);
    }
    /* create new pad with the name */
    name = g_strdup_printf ("sink_%d", serial);
    newpad = gst_pad_new_from_template (templ, name);
    g_free (name);

    /* construct our own wrapper data structure for the pad to
     * keep track of its status */
    {
Wim Taymans's avatar
Wim Taymans committed
385
386
387
      GstOggPad *oggpad;

      oggpad = (GstOggPad *)
388
          gst_collect_pads_add_pad (ogg_mux->collect, newpad,
Wim Taymans's avatar
Wim Taymans committed
389
          sizeof (GstOggPad));
Wim Taymans's avatar
Wim Taymans committed
390
391
392
393
394
395
396
397

      oggpad->serial = serial;
      ogg_stream_init (&oggpad->stream, serial);
      oggpad->packetno = 0;
      oggpad->pageno = 0;
      oggpad->eos = FALSE;
      /* we assume there will be some control data first for this pad */
      oggpad->state = GST_OGG_PAD_STATE_CONTROL;
398
399
400
      oggpad->new_page = TRUE;
      oggpad->first_delta = FALSE;
      oggpad->prev_delta = FALSE;
401
      oggpad->pagebuffers = g_queue_new ();
Wim Taymans's avatar
Wim Taymans committed
402
403
404
405
406
407
408
409
410
    }
  }

  /* setup some pad functions */
  gst_pad_set_link_function (newpad, gst_ogg_mux_sinkconnect);
  /* dd the pad to the element */
  gst_element_add_pad (element, newpad);

  return newpad;
Wim Taymans's avatar
Wim Taymans committed
411
412
413
414
415
416
417
418
419
420
421
422

  /* ERRORS */
wrong_direction:
  {
    g_warning ("ogg_mux: request pad that is not a SINK pad\n");
    return NULL;
  }
wrong_template:
  {
    g_warning ("ogg_mux: this is not our template!\n");
    return NULL;
  }
Wim Taymans's avatar
Wim Taymans committed
423
424
}

425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
static void
gst_ogg_mux_release_pad (GstElement * element, GstPad * pad)
{
  GstOggMux *ogg_mux;
  GSList *walk;

  ogg_mux = GST_OGG_MUX (gst_pad_get_parent (pad));

  /* FIXME: When a request pad is released while paused or playing, 
   * we probably need to do something to finalise its stream in the
   * ogg data we're producing, but I'm not sure what */

  /* Find out GstOggPad in the collect pads info and clean it up */

  GST_OBJECT_LOCK (ogg_mux->collect);
  walk = ogg_mux->collect->data;
  while (walk) {
    GstOggPad *oggpad = (GstOggPad *) walk->data;
    GstCollectData *cdata = (GstCollectData *) walk->data;
    GstBuffer *buf;

    if (cdata->pad == pad) {
      /* FIXME: clear the ogg stream stuff? - 
       *    ogg_stream_clear (&oggpad->stream); */

      while ((buf = g_queue_pop_head (oggpad->pagebuffers)) != NULL) {
        gst_buffer_unref (buf);
      }

      g_queue_free (oggpad->pagebuffers);
    }
    walk = g_slist_next (walk);
  }
  GST_OBJECT_UNLOCK (ogg_mux->collect);

  gst_collect_pads_remove_pad (ogg_mux->collect, pad);
}

Wim Taymans's avatar
Wim Taymans committed
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
/* handle events */
static gboolean
gst_ogg_mux_handle_src_event (GstPad * pad, GstEvent * event)
{
  GstEventType type;

  type = event ? GST_EVENT_TYPE (event) : GST_EVENT_UNKNOWN;

  switch (type) {
    case GST_EVENT_SEEK:
      /* disable seeking for now */
      return FALSE;
    default:
      break;
  }

  return gst_pad_event_default (pad, event);
}

482
static GstBuffer *
483
gst_ogg_mux_buffer_from_page (GstOggMux * mux, ogg_page * page, gboolean delta)
Wim Taymans's avatar
Wim Taymans committed
484
485
486
487
{
  GstBuffer *buffer;

  /* allocate space for header and body */
Wim Taymans's avatar
Wim Taymans committed
488
  buffer = gst_buffer_new_and_alloc (page->header_len + page->body_len);
Wim Taymans's avatar
Wim Taymans committed
489
490
491
492
  memcpy (GST_BUFFER_DATA (buffer), page->header, page->header_len);
  memcpy (GST_BUFFER_DATA (buffer) + page->header_len,
      page->body, page->body_len);

493
494
495
496
  /* Here we set granulepos as our OFFSET_END to give easy direct access to
   * this value later. Before we push it, we reset this to OFFSET + SIZE
   * (see gst_ogg_mux_push_buffer). */
  GST_BUFFER_OFFSET_END (buffer) = ogg_page_granulepos (page);
497
  if (delta)
498
    GST_BUFFER_FLAG_SET (buffer, GST_BUFFER_FLAG_DELTA_UNIT);
Wim Taymans's avatar
Wim Taymans committed
499

500
501
502
  GST_LOG_OBJECT (mux, GST_GP_FORMAT
      " created buffer %p from ogg page", ogg_page_granulepos (page));

503
504
505
  return buffer;
}

Wim Taymans's avatar
Wim Taymans committed
506
static GstFlowReturn
507
gst_ogg_mux_push_buffer (GstOggMux * mux, GstBuffer * buffer)
508
{
509
510
511
512
  /* fix up OFFSET and OFFSET_END again */
  GST_BUFFER_OFFSET (buffer) = mux->offset;
  mux->offset += GST_BUFFER_SIZE (buffer);
  GST_BUFFER_OFFSET_END (buffer) = mux->offset;
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539

  return gst_pad_push (mux->srcpad, buffer);
}

/* if all queues have at least one page, dequeue the page with the lowest
 * timestamp */
static gboolean
gst_ogg_mux_dequeue_page (GstOggMux * mux, GstFlowReturn * flowret)
{
  GSList *walk;
  GstOggPad *opad = NULL;       /* "oldest" pad */
  GstClockTime oldest = GST_CLOCK_TIME_NONE;
  GstBuffer *buf = NULL;
  gboolean ret = FALSE;

  *flowret = GST_FLOW_OK;

  walk = mux->collect->data;
  while (walk) {
    GstOggPad *pad = (GstOggPad *) walk->data;

    /* We need each queue to either be at EOS, or have one or more pages
     * available with a set granulepos (i.e. not -1), otherwise we don't have
     * enough data yet to determine which stream needs to go next for correct
     * time ordering. */
    if (pad->pagebuffers->length == 0) {
      if (pad->eos) {
540
541
        GST_LOG_OBJECT (pad->collect.pad,
            "pad is EOS, skipping for dequeue decision");
542
      } else {
543
544
        GST_LOG_OBJECT (pad->collect.pad,
            "no pages in this queue, can't dequeue");
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
        return FALSE;
      }
    } else {
      /* We then need to check for a non-negative granulepos */
      int i;
      gboolean valid = FALSE;

      for (i = 0; i < pad->pagebuffers->length; i++) {
        buf = g_queue_peek_nth (pad->pagebuffers, i);
        /* Here we check the OFFSET_END, which is actually temporarily the
         * granulepos value for this buffer */
        if (GST_BUFFER_OFFSET_END (buf) != -1) {
          valid = TRUE;
          break;
        }
      }
      if (!valid) {
562
563
        GST_LOG_OBJECT (pad->collect.pad,
            "No page timestamps in queue, can't dequeue");
564
565
566
567
568
569
570
571
572
573
574
        return FALSE;
      }
    }

    walk = g_slist_next (walk);
  }

  walk = mux->collect->data;
  while (walk) {
    GstOggPad *pad = (GstOggPad *) walk->data;

575
    /* any page with a granulepos of -1 can be pushed immediately.
576
577
578
     * TODO: it CAN be, but it seems silly to do so? */
    buf = g_queue_peek_head (pad->pagebuffers);
    while (buf && GST_BUFFER_OFFSET_END (buf) == -1) {
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
579
      GST_LOG_OBJECT (pad->collect.pad, "[gp        -1] pushing page");
580
581
582
583
584
585
586
587
588
      g_queue_pop_head (pad->pagebuffers);
      *flowret = gst_ogg_mux_push_buffer (mux, buf);
      buf = g_queue_peek_head (pad->pagebuffers);
      ret = TRUE;
    }

    if (buf) {
      /* if no oldest buffer yet, take this one */
      if (oldest == GST_CLOCK_TIME_NONE) {
589
590
591
592
        GST_LOG_OBJECT (mux, "no oldest yet, taking buffer %p from pad %"
            GST_PTR_FORMAT " with gp time %" GST_TIME_FORMAT,
            buf, pad->collect.pad, GST_TIME_ARGS (GST_BUFFER_OFFSET (buf)));
        oldest = GST_BUFFER_OFFSET (buf);
593
594
595
        opad = pad;
      } else {
        /* if we have an oldest, compare with this one */
596
597
598
599
600
        if (GST_BUFFER_OFFSET (buf) < oldest) {
          GST_LOG_OBJECT (mux, "older buffer %p, taking from pad %"
              GST_PTR_FORMAT " with gp time %" GST_TIME_FORMAT,
              buf, pad->collect.pad, GST_TIME_ARGS (GST_BUFFER_OFFSET (buf)));
          oldest = GST_BUFFER_OFFSET (buf);
601
602
603
604
605
606
607
608
609
610
          opad = pad;
        }
      }
    }
    walk = g_slist_next (walk);
  }

  if (oldest != GST_CLOCK_TIME_NONE) {
    g_assert (opad);
    buf = g_queue_pop_head (opad->pagebuffers);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
611
    GST_LOG_OBJECT (opad->collect.pad,
612
613
614
        GST_GP_FORMAT " pushing oldest page buffer %p (granulepos time %"
        GST_TIME_FORMAT ")", GST_BUFFER_OFFSET_END (buf), buf,
        GST_TIME_ARGS (GST_BUFFER_OFFSET (buf)));
615
616
617
618
619
620
621
    *flowret = gst_ogg_mux_push_buffer (mux, buf);
    ret = TRUE;
  }

  return ret;
}

622
623
624
625
626
627
628
629
630
631
632
633
634
/* put the given ogg page on a per-pad queue, timestamping it correctly.
 * after that, dequeue and push as many pages as possible.
 * Caller should make sure:
 * pad->timestamp     was set with the timestamp of the first packet put
 *                    on the page
 * pad->timestamp_end was set with the timestamp + duration of the last packet
 *                    put on the page
 * pad->gp_time       was set with the time matching the gp of the last
 *                    packet put on the page
 *
 * will also reset timestamp and timestamp_end, so caller func can restart
 * counting.
 */
635
636
637
638
static GstFlowReturn
gst_ogg_mux_pad_queue_page (GstOggMux * mux, GstOggPad * pad, ogg_page * page,
    gboolean delta)
{
Wim Taymans's avatar
Wim Taymans committed
639
  GstFlowReturn ret;
640
  GstBuffer *buffer = gst_ogg_mux_buffer_from_page (mux, page, delta);
641

642
  /* take the timestamp of the first packet on this page */
643
  GST_BUFFER_TIMESTAMP (buffer) = pad->timestamp;
644
  GST_BUFFER_DURATION (buffer) = pad->timestamp_end - pad->timestamp;
645
646
  /* take the gp time of the last completed packet on this page */
  GST_BUFFER_OFFSET (buffer) = pad->gp_time;
647

648
  /* the next page will start where the current page's end time leaves off */
649
650
  pad->timestamp = pad->timestamp_end;

651
  g_queue_push_tail (pad->pagebuffers, buffer);
652
653
654
655
656
  GST_LOG_OBJECT (pad->collect.pad, GST_GP_FORMAT
      " queued buffer page %p (gp time %"
      GST_TIME_FORMAT ", timestamp %" GST_TIME_FORMAT
      "), %d page buffers queued", ogg_page_granulepos (page),
      buffer, GST_TIME_ARGS (GST_BUFFER_OFFSET (buffer)),
657
658
      GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (buffer)),
      g_queue_get_length (pad->pagebuffers));
Wim Taymans's avatar
Wim Taymans committed
659

660
661
662
663
  while (gst_ogg_mux_dequeue_page (mux, &ret)) {
    if (ret != GST_FLOW_OK)
      break;
  }
Wim Taymans's avatar
Wim Taymans committed
664
665

  return ret;
Wim Taymans's avatar
Wim Taymans committed
666
667
668
}

/*
669
670
671
672
673
674
675
 * Given two pads, compare the buffers queued on it.
 * Returns:
 *  0 if they have an equal priority
 * -1 if the first is better
 *  1 if the second is better
 * Priority decided by: a) validity, b) older timestamp, c) smaller number
 * of muxed pages
Wim Taymans's avatar
Wim Taymans committed
676
677
 */
static gint
678
679
gst_ogg_mux_compare_pads (GstOggMux * ogg_mux, GstOggPad * first,
    GstOggPad * second)
Wim Taymans's avatar
Wim Taymans committed
680
{
681
  guint64 firsttime, secondtime;
Wim Taymans's avatar
Wim Taymans committed
682

683
684
685
  /* if the first pad doesn't contain anything or is even NULL, return
   * the second pad as best candidate and vice versa */
  if (first == NULL || first->buffer == NULL)
Wim Taymans's avatar
Wim Taymans committed
686
    return 1;
687
  if (second == NULL || second->buffer == NULL)
Wim Taymans's avatar
Wim Taymans committed
688
689
    return -1;

690
691
692
  /* no timestamp on first buffer, it must go first */
  firsttime = GST_BUFFER_TIMESTAMP (first->buffer);
  if (firsttime == GST_CLOCK_TIME_NONE)
Wim Taymans's avatar
Wim Taymans committed
693
694
    return -1;

695
696
697
  /* no timestamp on second buffer, it must go first */
  secondtime = GST_BUFFER_TIMESTAMP (second->buffer);
  if (secondtime == GST_CLOCK_TIME_NONE)
Wim Taymans's avatar
Wim Taymans committed
698
699
    return 1;

700
701
  /* first buffer has higher timestamp, second one should go first */
  if (secondtime < firsttime)
Wim Taymans's avatar
Wim Taymans committed
702
    return 1;
703
704
  /* second buffer has higher timestamp, first one should go first */
  else if (secondtime > firsttime)
Wim Taymans's avatar
Wim Taymans committed
705
706
707
708
    return -1;
  else {
    /* buffers with equal timestamps, prefer the pad that has the
     * least number of pages muxed */
709
    if (second->pageno < first->pageno)
Wim Taymans's avatar
Wim Taymans committed
710
      return 1;
711
    else if (second->pageno > first->pageno)
Wim Taymans's avatar
Wim Taymans committed
712
713
714
715
716
717
718
719
      return -1;
  }

  /* same priority if all of the above failed */
  return 0;
}

/* make sure a buffer is queued on all pads, returns a pointer to an oggpad
720
721
 * that holds the best buffer or NULL when no pad was usable.
 * "best" means the buffer marked with the lowest timestamp */
Wim Taymans's avatar
Wim Taymans committed
722
static GstOggPad *
Wim Taymans's avatar
Wim Taymans committed
723
gst_ogg_mux_queue_pads (GstOggMux * ogg_mux)
Wim Taymans's avatar
Wim Taymans committed
724
{
725
  GstOggPad *bestpad = NULL, *still_hungry = NULL;
Wim Taymans's avatar
Wim Taymans committed
726
727
728
  GSList *walk;

  /* try to make sure we have a buffer from each usable pad first */
Wim Taymans's avatar
Wim Taymans committed
729
  walk = ogg_mux->collect->data;
Wim Taymans's avatar
Wim Taymans committed
730
  while (walk) {
Wim Taymans's avatar
Wim Taymans committed
731
732
    GstOggPad *pad;
    GstCollectData *data;
Wim Taymans's avatar
Wim Taymans committed
733

Wim Taymans's avatar
Wim Taymans committed
734
735
    data = (GstCollectData *) walk->data;
    pad = (GstOggPad *) data;
Wim Taymans's avatar
Wim Taymans committed
736

Wim Taymans's avatar
Wim Taymans committed
737
738
    walk = g_slist_next (walk);

739
    GST_LOG_OBJECT (data->pad, "looking at pad for buffer");
740

Wim Taymans's avatar
Wim Taymans committed
741
    /* try to get a new buffer for this pad if needed and possible */
Wim Taymans's avatar
Wim Taymans committed
742
743
744
    if (pad->buffer == NULL) {
      GstBuffer *buf;
      gboolean incaps;
Wim Taymans's avatar
Wim Taymans committed
745

746
      buf = gst_collect_pads_pop (ogg_mux->collect, data);
747
      GST_LOG_OBJECT (data->pad, "popped buffer %" GST_PTR_FORMAT, buf);
Wim Taymans's avatar
Wim Taymans committed
748

749
750
751
752
753
754
755
      /* On EOS we get a NULL buffer */
      if (buf != NULL) {
        incaps = GST_BUFFER_FLAG_IS_SET (buf, GST_BUFFER_FLAG_IN_CAPS);
        /* if we need headers */
        if (pad->state == GST_OGG_PAD_STATE_CONTROL) {
          /* and we have one */
          if (incaps) {
756
757
            GST_DEBUG_OBJECT (ogg_mux,
                "got incaps buffer in control state, ignoring");
758
759
760
761
            /* just ignore */
            gst_buffer_unref (buf);
            buf = NULL;
          } else {
762
763
            GST_DEBUG_OBJECT (ogg_mux,
                "got data buffer in control state, switching " "to data mode");
764
765
766
            /* this is a data buffer so switch to data state */
            pad->state = GST_OGG_PAD_STATE_DATA;
          }
Wim Taymans's avatar
Wim Taymans committed
767
        }
768
      } else {
769
        GST_DEBUG_OBJECT (data->pad, "EOS on pad");
770
        pad->eos = TRUE;
Wim Taymans's avatar
Wim Taymans committed
771
      }
772

Wim Taymans's avatar
Wim Taymans committed
773
774
      pad->buffer = buf;
    }
Wim Taymans's avatar
Wim Taymans committed
775
776
777

    /* we should have a buffer now, see if it is the best pad to
     * pull on */
778
    if (pad->buffer) {
779
      if (gst_ogg_mux_compare_pads (ogg_mux, bestpad, pad) > 0) {
780
        GST_LOG_OBJECT (data->pad, "new best pad");
781

Wim Taymans's avatar
Wim Taymans committed
782
        bestpad = pad;
783
      }
784
    } else if (!pad->eos) {
785
      GST_LOG_OBJECT (data->pad, "hungry pad");
786
      still_hungry = pad;
Wim Taymans's avatar
Wim Taymans committed
787
788
    }
  }
789
790
791
792
793
794

  if (still_hungry)
    /* drop back into collectpads... */
    return still_hungry;
  else
    return bestpad;
Wim Taymans's avatar
Wim Taymans committed
795
796
}

797
798
799
800
801
802
803
static GList *
gst_ogg_mux_get_headers (GstOggPad * pad)
{
  GList *res = NULL;
  GstOggMux *ogg_mux;
  GstStructure *structure;
  const GstCaps *caps;
Wim Taymans's avatar
Wim Taymans committed
804
  GstPad *thepad;
805

Wim Taymans's avatar
Wim Taymans committed
806
  thepad = pad->collect.pad;
807

Wim Taymans's avatar
Wim Taymans committed
808
  ogg_mux = GST_OGG_MUX (GST_PAD_PARENT (thepad));
809

810
  GST_LOG_OBJECT (thepad, "getting headers");
Wim Taymans's avatar
Wim Taymans committed
811
812

  caps = gst_pad_get_negotiated_caps (thepad);
813
814
815
816
817
818
  if (caps != NULL) {
    const GValue *streamheader;

    structure = gst_caps_get_structure (caps, 0);
    streamheader = gst_structure_get_value (structure, "streamheader");
    if (streamheader != NULL) {
819
      GST_LOG_OBJECT (thepad, "got header");
820
      if (G_VALUE_TYPE (streamheader) == GST_TYPE_ARRAY) {
821
822
823
        GArray *bufarr = g_value_peek_pointer (streamheader);
        gint i;

824
        GST_LOG_OBJECT (thepad, "got fixed list");
Wim Taymans's avatar
Wim Taymans committed
825

826
827
828
        for (i = 0; i < bufarr->len; i++) {
          GValue *bufval = &g_array_index (bufarr, GValue, i);

829
          GST_LOG_OBJECT (thepad, "item %d", i);
830
831
832
          if (G_VALUE_TYPE (bufval) == GST_TYPE_BUFFER) {
            GstBuffer *buf = g_value_peek_pointer (bufval);

833
            GST_LOG_OBJECT (thepad, "adding item %d to header list", i);
Wim Taymans's avatar
Wim Taymans committed
834

Wim Taymans's avatar
Wim Taymans committed
835
            gst_buffer_ref (buf);
836
837
838
            res = g_list_append (res, buf);
          }
        }
Wim Taymans's avatar
Wim Taymans committed
839
      } else {
840
        GST_LOG_OBJECT (thepad, "streamheader is not fixed list");
841
      }
Wim Taymans's avatar
Wim Taymans committed
842
    } else {
843
      GST_LOG_OBJECT (thepad, "caps done have streamheader");
844
    }
Wim Taymans's avatar
Wim Taymans committed
845
  } else {
846
    GST_LOG_OBJECT (thepad, "got empty caps as negotiated format");
847
848
849
850
  }
  return res;
}

Wim Taymans's avatar
Wim Taymans committed
851
static GstCaps *
852
853
gst_ogg_mux_set_header_on_caps (GstCaps * caps, GList * buffers)
{
Wim Taymans's avatar
Wim Taymans committed
854
  GstStructure *structure;
855
  GValue array = { 0 };
856
857
  GList *walk = buffers;

Wim Taymans's avatar
Wim Taymans committed
858
859
860
861
  caps = gst_caps_make_writable (caps);

  structure = gst_caps_get_structure (caps, 0);

862
  /* put buffers in a fixed list */
863
  g_value_init (&array, GST_TYPE_ARRAY);
864
865
866
867
868
869
870
871

  while (walk) {
    GstBuffer *buf = GST_BUFFER (walk->data);
    GValue value = { 0 };

    walk = walk->next;

    /* mark buffer */
872
    GST_LOG ("Setting IN_CAPS on buffer of length %d", GST_BUFFER_SIZE (buf));
873
    GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_IN_CAPS);
874
875

    g_value_init (&value, GST_TYPE_BUFFER);
876
    gst_value_set_buffer (&value, buf);
877
    gst_value_array_append_value (&array, &value);
878
879
    g_value_unset (&value);
  }
880
881
  gst_structure_set_value (structure, "streamheader", &array);
  g_value_unset (&array);
Wim Taymans's avatar
Wim Taymans committed
882
883

  return caps;
884
885
}

886
/*
887
888
889
890
891
892
893
 * For each pad we need to write out one (small) header in one
 * page that allows decoders to identify the type of the stream.
 * After that we need to write out all extra info for the decoders.
 * In the case of a codec that also needs data as configuration, we can
 * find that info in the streamcaps. 
 * After writing the headers we must start a new page for the data.
 */
Wim Taymans's avatar
Wim Taymans committed
894
static GstFlowReturn
895
896
897
898
899
gst_ogg_mux_send_headers (GstOggMux * mux)
{
  GSList *walk;
  GList *hbufs, *hwalk;
  GstCaps *caps;
Wim Taymans's avatar
Wim Taymans committed
900
  GstFlowReturn ret;
901
902

  hbufs = NULL;
Wim Taymans's avatar
Wim Taymans committed
903
  ret = GST_FLOW_OK;
904

905
  GST_LOG_OBJECT (mux, "collecting headers");
906

Wim Taymans's avatar
Wim Taymans committed
907
  walk = mux->collect->data;
908
  while (walk) {
Wim Taymans's avatar
Wim Taymans committed
909
910
    GstOggPad *pad;
    GstPad *thepad;
911

Wim Taymans's avatar
Wim Taymans committed
912
913
    pad = (GstOggPad *) walk->data;
    thepad = pad->collect.pad;
914

Wim Taymans's avatar
Wim Taymans committed
915
916
    walk = g_slist_next (walk);

917
    GST_LOG_OBJECT (mux, "looking at pad %s:%s", GST_DEBUG_PAD_NAME (thepad));
918
919
920
921
922
923
924
925
926

    /* if the pad has no buffer, we don't care */
    if (pad->buffer == NULL)
      continue;

    /* now figure out the headers */
    pad->headers = gst_ogg_mux_get_headers (pad);
  }

927
  GST_LOG_OBJECT (mux, "creating BOS pages");
Wim Taymans's avatar
Wim Taymans committed
928
  walk = mux->collect->data;
929
  while (walk) {
Wim Taymans's avatar
Wim Taymans committed
930
    GstOggPad *pad;
931
932
933
    GstBuffer *buf;
    ogg_packet packet;
    ogg_page page;
Wim Taymans's avatar
Wim Taymans committed
934
    GstPad *thepad;
935
936
937
    GstCaps *caps;
    GstStructure *structure;
    GstBuffer *hbuf;
Wim Taymans's avatar
Wim Taymans committed
938
939
940

    pad = (GstOggPad *) walk->data;
    thepad = pad->collect.pad;
941
942
    caps = gst_pad_get_negotiated_caps (thepad);
    structure = gst_caps_get_structure (caps, 0);
943
944
945
946
947

    walk = walk->next;

    pad->packetno = 0;

948
    GST_LOG_OBJECT (thepad, "looping over headers");
949
950
951
952

    if (pad->headers) {
      buf = GST_BUFFER (pad->headers->data);
      pad->headers = g_list_remove (pad->headers, buf);
953
    } else if (pad->buffer) {
954
955
      buf = pad->buffer;
      gst_buffer_ref (buf);
956
    } else {
957
      /* fixme -- should be caught in the previous list traversal. */
958
      GST_OBJECT_LOCK (pad);
959
960
      g_critical ("No headers or buffers on pad %s:%s",
          GST_DEBUG_PAD_NAME (pad));
961
      GST_OBJECT_UNLOCK (pad);
962
      continue;
963
964
965
966
967
968
    }

    /* create a packet from the buffer */
    packet.packet = GST_BUFFER_DATA (buf);
    packet.bytes = GST_BUFFER_SIZE (buf);
    packet.granulepos = GST_BUFFER_OFFSET_END (buf);
969
970
    if (packet.granulepos == -1)
      packet.granulepos = 0;
971
972
973
974
975
976
977
978
979
980
    /* mark BOS and packet number */
    packet.b_o_s = (pad->packetno == 0);
    packet.packetno = pad->packetno++;
    /* mark EOS */
    packet.e_o_s = 0;

    /* swap the packet in */
    ogg_stream_packetin (&pad->stream, &packet);
    gst_buffer_unref (buf);

981
982
983
984
985
    GST_LOG_OBJECT (thepad, "flushing out BOS page");
    if (!ogg_stream_flush (&pad->stream, &page))
      g_critical ("Could not flush BOS page");

    hbuf = gst_ogg_mux_buffer_from_page (mux, &page, FALSE);
986

987
988
989
990
991
992
993
994
995
996
    GST_LOG_OBJECT (mux, "swapped out page with mime type %s",
        gst_structure_get_name (structure));

    /* quick hack: put theora pages at the front.
     * Ideally, we would have a settable enum for which Ogg
     * profile we work with, and order based on that */
    if (strcmp (gst_structure_get_name (structure), "video/x-theora") == 0) {
      GST_DEBUG_OBJECT (thepad, "putting Theora page at the front");
      hbufs = g_list_prepend (hbufs, hbuf);
    } else {
997
998
      hbufs = g_list_append (hbufs, hbuf);
    }
999
    gst_caps_unref (caps);
1000
1001
  }

1002
  GST_LOG_OBJECT (mux, "creating next headers");
Wim Taymans's avatar
Wim Taymans committed
1003
  walk = mux->collect->data;
1004
  while (walk) {
Wim Taymans's avatar
Wim Taymans committed
1005
1006
1007
1008
1009
    GstOggPad *pad;
    GstPad *thepad;

    pad = (GstOggPad *) walk->data;
    thepad = pad->collect.pad;
1010
1011
1012

    walk = walk->next;

1013
1014
    GST_LOG_OBJECT (mux, "looping over headers for pad %s:%s",
        GST_DEBUG_PAD_NAME (thepad));
1015
1016
1017
1018
1019
1020
1021
1022
1023
1024
1025
1026
1027

    hwalk = pad->headers;
    while (hwalk) {
      GstBuffer *buf = GST_BUFFER (hwalk->data);
      ogg_packet packet;
      ogg_page page;

      hwalk = hwalk->next;

      /* create a packet from the buffer */
      packet.packet = GST_BUFFER_DATA (buf);
      packet.bytes = GST_BUFFER_SIZE (buf);
      packet.granulepos = GST_BUFFER_OFFSET_END (buf);
1028
1029
      if (packet.granulepos == -1)
        packet.granulepos = 0;
1030
1031
1032
1033
1034
1035
1036
1037
1038
1039
1040
1041
      /* mark BOS and packet number */
      packet.b_o_s = (pad->packetno == 0);
      packet.packetno = pad->packetno++;
      /* mark EOS */
      packet.e_o_s = 0;

      /* swap the packet in */
      ogg_stream_packetin (&pad->stream, &packet);
      gst_buffer_unref (buf);

      /* if last header, flush page */
      if (hwalk == NULL) {
1042
1043
        GST_LOG_OBJECT (mux,
            "flushing page as packet %d is first or last packet",
1044
1045
            pad->packetno);
        while (ogg_stream_flush (&pad->stream, &page)) {
1046
          GstBuffer *hbuf = gst_ogg_mux_buffer_from_page (mux, &page, FALSE);
1047

1048
          GST_LOG_OBJECT (mux, "swapped out page");
1049
1050
1051
          hbufs = g_list_append (hbufs, hbuf);
        }
      } else {
1052
        GST_LOG_OBJECT (mux, "try to swap out page");
1053
1054
        /* just try to swap out a page then */
        while (ogg_stream_pageout (&pad->stream, &page) > 0) {
1055
          GstBuffer *hbuf = gst_ogg_mux_buffer_from_page (mux, &page, FALSE);
1056

1057
          GST_LOG_OBJECT (mux, "swapped out page");
1058
1059
1060
1061
          hbufs = g_list_append (hbufs, hbuf);
        }
      }
    }
Wim Taymans's avatar
Wim Taymans committed
1062
1063
    g_list_free (pad->headers);
    pad->headers = NULL;
1064
1065
1066
1067
1068
1069
  }
  /* hbufs holds all buffers for the headers now */

  /* create caps with the buffers */
  caps = gst_pad_get_caps (mux->srcpad);
  if (caps) {
Wim Taymans's avatar
Wim Taymans committed
1070
1071
    caps = gst_ogg_mux_set_header_on_caps (caps, hbufs);
    gst_pad_set_caps (mux->srcpad, caps);
1072
    gst_caps_unref (caps);
1073
1074
1075
1076
1077
1078
1079
1080
  }
  /* and send the buffers */
  hwalk = hbufs;
  while (hwalk) {
    GstBuffer *buf = GST_BUFFER (hwalk->data);

    hwalk = hwalk->next;

1081
    if ((ret = gst_ogg_mux_push_buffer (mux, buf)) != GST_FLOW_OK)
Wim Taymans's avatar
Wim Taymans committed
1082
      break;
1083
  }
Wim Taymans's avatar
Wim Taymans committed
1084
  g_list_free (hbufs);
Wim Taymans's avatar
Wim Taymans committed
1085
1086

  return ret;
1087
1088
}

Wim Taymans's avatar
Wim Taymans committed
1089
1090
1091
/* this function is called when there is data on all pads.
 *
 * basic idea:
Wim Taymans's avatar
Wim Taymans committed
1092
 *
1093
 * 1) find a pad to pull on, this is done by looking at the buffers
1094
 *    to decide which one to use, we use the 'oldest' one first.
1095
1096
 * 2) store the selected pad and keep on pulling until we fill a
 *    complete ogg page or the ogg page is filled above the max-delay
1097
 *    threshold. This is needed because the ogg spec says that
Wim Taymans's avatar
Wim Taymans committed
1098
1099
 *    you should fill a complete page with data from the same logical
 *    stream. When the page is filled, go back to 1).
1100
 * 3) before filling a page, read ahead one more buffer to see if this
1101
1102
 *    packet is the last of the stream. We need to do this because the ogg
 *    spec mandates that the last packet should have the EOS flag set before
1103
 *    sending it to ogg. FIXME: Apparently we're allowed to send empty 'nil'
1104
1105
1106
 *    pages with the EOS flag set for EOS, so we could do this. Not sure how
 *    that works, though. TODO: 'read ahead one more buffer' is a bit funky
 *    with collectpads. Rethink this.
1107
1108
 * 4) pages get queued on a per-pad queue. Every time a page is queued, a
 *    dequeue is called, which will dequeue the oldest page on any pad, provided
1109
1110
 *    that ALL pads have at least one marked page in the queue (or remaining
 *    pad are at EOS)
Wim Taymans's avatar
Wim Taymans committed
1111
 */
Wim Taymans's avatar
Wim Taymans committed
1112
1113
static GstFlowReturn
gst_ogg_mux_collected (GstCollectPads * pads, GstOggMux * ogg_mux)
Wim Taymans's avatar
Wim Taymans committed
1114
{
1115
  GstOggPad *best;
1116
  gboolean delta_unit;
Wim Taymans's avatar
Wim Taymans committed
1117
  GstFlowReturn ret;
1118
  gint64 granulepos = 0;
1119
  GstClockTime timestamp, gp_time;
Wim Taymans's avatar
Wim Taymans committed
1120

1121
  GST_LOG_OBJECT (ogg_mux, "collected");
Wim Taymans's avatar
Wim Taymans committed
1122

1123
  /* queue buffers on all pads; find a buffer with the lowest timestamp */
Wim Taymans's avatar
Wim Taymans committed
1124
  best = gst_ogg_mux_queue_pads (ogg_mux);
1125
1126
  if (best && !best->buffer) {
    GST_DEBUG_OBJECT (ogg_mux, "No buffer available on best pad");
Wim Taymans's avatar
Wim Taymans committed
1127
    return GST_FLOW_OK;
1128
  }
Wim Taymans's avatar
Wim Taymans committed
1129

1130
1131
1132
  if (!best) {
    /* EOS : FIXME !! We need to handle EOS correctly, and set EOS
       flags on the ogg pages. */
1133
    GST_DEBUG_OBJECT (ogg_mux, "Pushing EOS");
1134
1135
1136
1137
    gst_pad_push_event (ogg_mux->srcpad, gst_event_new_eos ());
    return GST_FLOW_WRONG_STATE;
  }

1138
1139
1140
  GST_LOG_OBJECT (ogg_mux, "best pad %" GST_PTR_FORMAT
      ", currently pulling from %" GST_PTR_FORMAT, best->collect.pad,
      ogg_mux->pulling);
1141

1142
1143
1144
  /* if we were already pulling from one pad, but the new "best" buffer is
   * from another pad, we need to check if we have reason to flush a page
   * for the pad we were pulling from before */
1145
1146
  if (ogg_mux->pulling && best &&
      ogg_mux->pulling != best && ogg_mux->pulling->buffer) {
1147
1148
    GstOggPad *pad = ogg_mux->pulling;

1149
    GstClockTime last_ts = GST_BUFFER_END_TIME (pad->buffer);
Wim Taymans's avatar
Wim Taymans committed
1150

1151
    /* if the next packet in the current page is going to make the page
1152
1153
1154
1155
     * too long, we need to flush */
    if (last_ts > ogg_mux->next_ts + ogg_mux->max_delay) {
      ogg_page page;

1156
      GST_LOG_OBJECT (pad->collect.pad,
1157
1158
1159
1160
          GST_GP_FORMAT " stored packet %" G_GINT64_FORMAT
          " will make page too long, flushing",
          GST_BUFFER_OFFSET_END (pad->buffer), pad->stream.packetno);

1161
      while (ogg_stream_flush (&pad->stream, &page)) {
1162
1163
        /* end time of this page is the timestamp of the next buffer */
        ogg_mux->pulling->timestamp_end = GST_BUFFER_TIMESTAMP (pad->buffer);
1164
1165
1166
        /* Place page into the per-pad queue */
        ret = gst_ogg_mux_pad_queue_page (ogg_mux, pad, &page,
            pad->first_delta);
1167
        /* increment the page number counter */
1168
1169
1170
        pad->pageno++;
        /* mark other pages as delta */
        pad->first_delta = TRUE;
1171
      }
1172
      pad->new_page = TRUE;
1173
1174
1175
1176
1177
      ogg_mux->pulling = NULL;
    }
  }

  /* if we don't know which pad to pull on, use the best one */
Wim Taymans's avatar
Wim Taymans committed
1178
  if (ogg_mux->pulling == NULL) {
1179
    ogg_mux->pulling = best;
1180
    GST_LOG_OBJECT (ogg_mux->pulling->collect.pad, "pulling from best pad");
1181

1182
    /* remember timestamp and gp time of first buffer for this new pad */
Wim Taymans's avatar
Wim Taymans committed
1183
1184
    if (ogg_mux->pulling != NULL) {
      ogg_mux->next_ts = GST_BUFFER_TIMESTAMP (ogg_mux->pulling->buffer);
1185
1186
      GST_LOG_OBJECT (ogg_mux->pulling->collect.pad, "updated times, next ts %"
          GST_TIME_FORMAT, GST_TIME_ARGS (ogg_mux->next_ts));
Wim Taymans's avatar
Wim Taymans committed
1187
1188
    } else {
      /* no pad to pull on, send EOS */