gstoggmux.c 51.1 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
 *
 * 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.
 */

21
22
23
24
25
/**
 * SECTION:element-oggmux
 * @see_also: <link linkend="gst-plugins-base-plugins-deoggmux">oggdemux</link>
 *
 * This element merges streams (audio and video) into ogg files.
26
27
 *
 * <refsect2>
28
 * <title>Example pipelines</title>
29
 * |[
30
 * gst-launch v4l2src num-buffers=500 ! video/x-raw-yuv,width=320,height=240 ! ffmpegcolorspace ! theoraenc ! oggmux ! filesink location=video.ogg
31
 * ]| Encodes a video stream captured from a v4l2-compatible camera to Ogg/Theora
32
 * (the encoding will stop automatically after 500 frames)
33
34
35
36
37
 * </refsect2>
 *
 * Last reviewed on 2008-02-06 (0.10.17)
 */

Wim Taymans's avatar
Wim Taymans committed
38
39
40
41
42
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

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

45
46
#include "gstoggmux.h"

Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
47
/* memcpy - if someone knows a way to get rid of it, please speak up
Wim Taymans's avatar
Wim Taymans committed
48
49
50
 * note: the ogg docs even say you need this... */
#include <string.h>
#include <time.h>
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
51
#include <stdlib.h>             /* rand, srand, atoi */
Wim Taymans's avatar
Wim Taymans committed
52
53
54
55

GST_DEBUG_CATEGORY_STATIC (gst_ogg_mux_debug);
#define GST_CAT_DEFAULT gst_ogg_mux_debug

56
57
58
59
60
61
62
63
/* 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))

64
65
#define GST_GP_FORMAT "[gp %8" G_GINT64_FORMAT "]"

Wim Taymans's avatar
Wim Taymans committed
66
67
68
69
70
71
72
73
typedef enum
{
  GST_OGG_FLAG_BOS = GST_ELEMENT_FLAG_LAST,
  GST_OGG_FLAG_EOS
}
GstOggFlag;

/* elementfactory information */
Stefan Kost's avatar
Stefan Kost committed
74
75
static const GstElementDetails gst_ogg_mux_details =
GST_ELEMENT_DETAILS ("Ogg muxer",
Wim Taymans's avatar
Wim Taymans committed
76
77
78
79
80
81
82
83
84
85
86
    "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
87
/* set to 0.5 seconds by default */
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
88
89
#define DEFAULT_MAX_DELAY       G_GINT64_CONSTANT(500000000)
#define DEFAULT_MAX_PAGE_DELAY  G_GINT64_CONSTANT(500000000)
Wim Taymans's avatar
Wim Taymans committed
90
91
enum
{
92
93
  ARG_0,
  ARG_MAX_DELAY,
94
  ARG_MAX_PAGE_DELAY,
Wim Taymans's avatar
Wim Taymans committed
95
96
97
98
99
100
101
102
103
104
105
};

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,
106
    GST_STATIC_CAPS ("video/x-theora; "
107
        "audio/x-vorbis; audio/x-flac; audio/x-speex; audio/x-celt; "
108
        "application/x-ogm-video; application/x-ogm-audio; video/x-dirac; "
109
        "video/x-smoke; text/x-cmml, encoded = (boolean) TRUE")
Wim Taymans's avatar
Wim Taymans committed
110
111
112
113
114
    );

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);
115
static void gst_ogg_mux_finalize (GObject * object);
Wim Taymans's avatar
Wim Taymans committed
116

Wim Taymans's avatar
Wim Taymans committed
117
118
static GstFlowReturn
gst_ogg_mux_collected (GstCollectPads * pads, GstOggMux * ogg_mux);
Wim Taymans's avatar
Wim Taymans committed
119
120
121
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);
122
123
static void gst_ogg_mux_release_pad (GstElement * element, GstPad * pad);

Wim Taymans's avatar
Wim Taymans committed
124
125
126
127
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);
128
129
static GstStateChangeReturn gst_ogg_mux_change_state (GstElement * element,
    GstStateChange transition);
Wim Taymans's avatar
Wim Taymans committed
130
131
132
133
134
135
136
137
138
139

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;

140
  if (G_UNLIKELY (ogg_mux_type == 0)) {
Wim Taymans's avatar
Wim Taymans committed
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
    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;

182
  parent_class = g_type_class_peek_parent (klass);
Wim Taymans's avatar
Wim Taymans committed
183

184
  gobject_class->finalize = gst_ogg_mux_finalize;
185
186
187
  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
188
  gstelement_class->request_new_pad = gst_ogg_mux_request_new_pad;
189
  gstelement_class->release_pad = gst_ogg_mux_release_pad;
Wim Taymans's avatar
Wim Taymans committed
190

191
192
193
  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,
194
195
          DEFAULT_MAX_DELAY,
          (GParamFlags) G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
196
197
198
  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,
199
200
          DEFAULT_MAX_PAGE_DELAY,
          (GParamFlags) G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
201

Wim Taymans's avatar
Wim Taymans committed
202
203
204
205
  gstelement_class->change_state = gst_ogg_mux_change_state;

}

Wim Taymans's avatar
Wim Taymans committed
206
#if 0
Wim Taymans's avatar
Wim Taymans committed
207
208
209
210
211
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
212
    {GST_EVENT_DISCONTINUOUS, 0},
Wim Taymans's avatar
Wim Taymans committed
213
214
215
216
217
    {0,}
  };

  return gst_ogg_mux_sink_event_masks;
}
Wim Taymans's avatar
Wim Taymans committed
218
#endif
Wim Taymans's avatar
Wim Taymans committed
219

220
221
222
223
224
225
226
227
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;
228
229
  ogg_mux->offset = 0;
  ogg_mux->next_ts = 0;
230
  ogg_mux->last_ts = GST_CLOCK_TIME_NONE;
231
232
}

Wim Taymans's avatar
Wim Taymans committed
233
234
235
236
237
238
239
240
241
242
243
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);

244
  GST_OBJECT_FLAG_SET (GST_ELEMENT (ogg_mux), GST_OGG_FLAG_BOS);
Wim Taymans's avatar
Wim Taymans committed
245
246
247
248

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

249
250
  ogg_mux->collect = gst_collect_pads_new ();
  gst_collect_pads_set_function (ogg_mux->collect,
251
252
      (GstCollectPadsFunction) GST_DEBUG_FUNCPTR (gst_ogg_mux_collected),
      ogg_mux);
253

254
  gst_ogg_mux_clear (ogg_mux);
Wim Taymans's avatar
Wim Taymans committed
255
256
}

257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
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);
}

272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
static void
gst_ogg_mux_ogg_pad_destroy_notify (GstCollectData * data)
{
  GstOggPad *oggpad = (GstOggPad *) data;
  GstBuffer *buf;

  ogg_stream_clear (&oggpad->stream);

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

Wim Taymans's avatar
Wim Taymans committed
289
static GstPadLinkReturn
290
gst_ogg_mux_sinkconnect (GstPad * pad, GstPad * peer)
Wim Taymans's avatar
Wim Taymans committed
291
292
293
294
295
{
  GstOggMux *ogg_mux;

  ogg_mux = GST_OGG_MUX (gst_pad_get_parent (pad));

296
  GST_DEBUG_OBJECT (ogg_mux, "sinkconnect triggered on %s", GST_PAD_NAME (pad));
297

298
  gst_object_unref (ogg_mux);
Wim Taymans's avatar
Wim Taymans committed
299
300
301
302

  return GST_PAD_LINK_OK;
}

303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
static gboolean
gst_ogg_mux_sink_event (GstPad * pad, GstEvent * event)
{
  GstOggMux *ogg_mux = GST_OGG_MUX (gst_pad_get_parent (pad));
  GstOggPad *ogg_pad = (GstOggPad *) gst_pad_get_element_private (pad);
  gboolean ret;

  GST_DEBUG ("Got %s event on pad %s:%s", GST_EVENT_TYPE_NAME (event),
      GST_DEBUG_PAD_NAME (pad));

  switch (GST_EVENT_TYPE (event)) {
    case GST_EVENT_NEWSEGMENT:
      /* We don't support NEWSEGMENT events */
      gst_event_unref (event);
      ret = FALSE;
      break;
    default:
      ret = TRUE;
      break;
  }

  /* now GstCollectPads can take care of the rest, e.g. EOS */
  if (ret)
    ret = ogg_pad->collect_event (pad, event);

  gst_object_unref (ogg_mux);
  return ret;
}

Wim Taymans's avatar
Wim Taymans committed
332
333
334
335
336
337
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
338
  GstElementClass *klass;
Wim Taymans's avatar
Wim Taymans committed
339
340
341

  g_return_val_if_fail (templ != NULL, NULL);

Wim Taymans's avatar
Wim Taymans committed
342
343
  if (templ->direction != GST_PAD_SINK)
    goto wrong_direction;
Wim Taymans's avatar
Wim Taymans committed
344
345
346
347

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

Wim Taymans's avatar
Wim Taymans committed
348
349
350
351
352
353
  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
354
355
356
357
358
359
360
361
362
363
364
    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 */
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
365
    GST_DEBUG_OBJECT (ogg_mux, "Creating new pad for serial %d", serial);
Wim Taymans's avatar
Wim Taymans committed
366
367
368
369
370
371
372
    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
373
374
375
      GstOggPad *oggpad;

      oggpad = (GstOggPad *)
376
377
          gst_collect_pads_add_pad_full (ogg_mux->collect, newpad,
          sizeof (GstOggPad), gst_ogg_mux_ogg_pad_destroy_notify);
378
      ogg_mux->active_pads++;
Wim Taymans's avatar
Wim Taymans committed
379
380
381
382
383
384
385
386

      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;
387
388
389
      oggpad->new_page = TRUE;
      oggpad->first_delta = FALSE;
      oggpad->prev_delta = FALSE;
390
      oggpad->pagebuffers = g_queue_new ();
391
392
393
394

      oggpad->collect_event = (GstPadEventFunction) GST_PAD_EVENTFUNC (newpad);
      gst_pad_set_event_function (newpad,
          GST_DEBUG_FUNCPTR (gst_ogg_mux_sink_event));
Wim Taymans's avatar
Wim Taymans committed
395
396
397
398
399
    }
  }

  /* setup some pad functions */
  gst_pad_set_link_function (newpad, gst_ogg_mux_sinkconnect);
400

Wim Taymans's avatar
Wim Taymans committed
401
402
403
404
  /* dd the pad to the element */
  gst_element_add_pad (element, newpad);

  return newpad;
Wim Taymans's avatar
Wim Taymans committed
405
406
407
408
409
410
411
412
413
414
415
416

  /* 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
417
418
}

419
420
421
422
423
424
425
426
static void
gst_ogg_mux_release_pad (GstElement * element, GstPad * pad)
{
  GstOggMux *ogg_mux;

  ogg_mux = GST_OGG_MUX (gst_pad_get_parent (pad));

  gst_collect_pads_remove_pad (ogg_mux->collect, pad);
427
  gst_element_remove_pad (element, pad);
428
429

  gst_object_unref (ogg_mux);
430
431
}

Wim Taymans's avatar
Wim Taymans committed
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
/* 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);
}

451
static GstBuffer *
452
gst_ogg_mux_buffer_from_page (GstOggMux * mux, ogg_page * page, gboolean delta)
Wim Taymans's avatar
Wim Taymans committed
453
454
455
456
{
  GstBuffer *buffer;

  /* allocate space for header and body */
Wim Taymans's avatar
Wim Taymans committed
457
  buffer = gst_buffer_new_and_alloc (page->header_len + page->body_len);
Wim Taymans's avatar
Wim Taymans committed
458
459
460
461
  memcpy (GST_BUFFER_DATA (buffer), page->header, page->header_len);
  memcpy (GST_BUFFER_DATA (buffer) + page->header_len,
      page->body, page->body_len);

462
463
464
465
  /* 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);
466
  if (delta)
467
    GST_BUFFER_FLAG_SET (buffer, GST_BUFFER_FLAG_DELTA_UNIT);
Wim Taymans's avatar
Wim Taymans committed
468

469
  GST_LOG_OBJECT (mux, GST_GP_FORMAT
470
      " created buffer %p from ogg page", ogg_page_granulepos (page), buffer);
471

472
473
474
  return buffer;
}

Wim Taymans's avatar
Wim Taymans committed
475
static GstFlowReturn
476
gst_ogg_mux_push_buffer (GstOggMux * mux, GstBuffer * buffer)
477
{
478
479
  GstCaps *caps;

480
481
482
483
  /* 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;
484

485
486
  /* Ensure we have monotonically increasing timestamps in the output. */
  if (GST_BUFFER_TIMESTAMP_IS_VALID (buffer)) {
487
488
    if (mux->last_ts != GST_CLOCK_TIME_NONE &&
        GST_BUFFER_TIMESTAMP (buffer) < mux->last_ts)
489
490
491
492
493
      GST_BUFFER_TIMESTAMP (buffer) = mux->last_ts;
    else
      mux->last_ts = GST_BUFFER_TIMESTAMP (buffer);
  }

494
495
496
497
  caps = gst_pad_get_negotiated_caps (mux->srcpad);
  gst_buffer_set_caps (buffer, caps);
  gst_caps_unref (caps);

498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
  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) {
524
525
        GST_LOG_OBJECT (pad->collect.pad,
            "pad is EOS, skipping for dequeue decision");
526
      } else {
527
528
        GST_LOG_OBJECT (pad->collect.pad,
            "no pages in this queue, can't dequeue");
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
        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) {
546
547
        GST_LOG_OBJECT (pad->collect.pad,
            "No page timestamps in queue, can't dequeue");
548
549
550
551
552
553
554
555
556
557
558
        return FALSE;
      }
    }

    walk = g_slist_next (walk);
  }

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

559
    /* any page with a granulepos of -1 can be pushed immediately.
560
561
562
     * 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
563
      GST_LOG_OBJECT (pad->collect.pad, "[gp        -1] pushing page");
564
565
566
567
568
569
570
571
572
      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) {
573
574
575
576
        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);
577
578
579
        opad = pad;
      } else {
        /* if we have an oldest, compare with this one */
580
581
582
583
584
        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);
585
586
587
588
589
590
591
592
593
594
          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
595
    GST_LOG_OBJECT (opad->collect.pad,
596
597
598
        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)));
599
600
601
602
603
604
605
    *flowret = gst_ogg_mux_push_buffer (mux, buf);
    ret = TRUE;
  }

  return ret;
}

606
607
608
609
610
611
612
613
614
615
616
617
618
/* 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.
 */
619
620
621
622
static GstFlowReturn
gst_ogg_mux_pad_queue_page (GstOggMux * mux, GstOggPad * pad, ogg_page * page,
    gboolean delta)
{
Wim Taymans's avatar
Wim Taymans committed
623
  GstFlowReturn ret;
624
  GstBuffer *buffer = gst_ogg_mux_buffer_from_page (mux, page, delta);
625

626
  /* take the timestamp of the first packet on this page */
627
  GST_BUFFER_TIMESTAMP (buffer) = pad->timestamp;
628
  GST_BUFFER_DURATION (buffer) = pad->timestamp_end - pad->timestamp;
629
630
  /* take the gp time of the last completed packet on this page */
  GST_BUFFER_OFFSET (buffer) = pad->gp_time;
631

632
  /* the next page will start where the current page's end time leaves off */
633
634
  pad->timestamp = pad->timestamp_end;

635
  g_queue_push_tail (pad->pagebuffers, buffer);
636
637
638
639
640
  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)),
641
642
      GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (buffer)),
      g_queue_get_length (pad->pagebuffers));
Wim Taymans's avatar
Wim Taymans committed
643

644
645
646
647
  while (gst_ogg_mux_dequeue_page (mux, &ret)) {
    if (ret != GST_FLOW_OK)
      break;
  }
Wim Taymans's avatar
Wim Taymans committed
648
649

  return ret;
Wim Taymans's avatar
Wim Taymans committed
650
651
652
}

/*
653
654
655
656
657
658
659
 * 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
660
661
 */
static gint
662
663
gst_ogg_mux_compare_pads (GstOggMux * ogg_mux, GstOggPad * first,
    GstOggPad * second)
Wim Taymans's avatar
Wim Taymans committed
664
{
665
  guint64 firsttime, secondtime;
Wim Taymans's avatar
Wim Taymans committed
666

667
668
  /* if the first pad doesn't contain anything or is even NULL, return
   * the second pad as best candidate and vice versa */
669
  if (first == NULL || (first->buffer == NULL && first->next_buffer == NULL))
Wim Taymans's avatar
Wim Taymans committed
670
    return 1;
671
  if (second == NULL || (second->buffer == NULL && second->next_buffer == NULL))
Wim Taymans's avatar
Wim Taymans committed
672
673
    return -1;

674
  /* no timestamp on first buffer, it must go first */
675
676
677
678
  if (first->buffer)
    firsttime = GST_BUFFER_TIMESTAMP (first->buffer);
  else
    firsttime = GST_BUFFER_TIMESTAMP (first->next_buffer);
679
  if (firsttime == GST_CLOCK_TIME_NONE)
Wim Taymans's avatar
Wim Taymans committed
680
681
    return -1;

682
  /* no timestamp on second buffer, it must go first */
683
684
685
686
  if (second->buffer)
    secondtime = GST_BUFFER_TIMESTAMP (second->buffer);
  else
    secondtime = GST_BUFFER_TIMESTAMP (second->next_buffer);
687
  if (secondtime == GST_CLOCK_TIME_NONE)
Wim Taymans's avatar
Wim Taymans committed
688
689
    return 1;

690
691
  /* first buffer has higher timestamp, second one should go first */
  if (secondtime < firsttime)
Wim Taymans's avatar
Wim Taymans committed
692
    return 1;
693
694
  /* second buffer has higher timestamp, first one should go first */
  else if (secondtime > firsttime)
Wim Taymans's avatar
Wim Taymans committed
695
696
697
698
    return -1;
  else {
    /* buffers with equal timestamps, prefer the pad that has the
     * least number of pages muxed */
699
    if (second->pageno < first->pageno)
Wim Taymans's avatar
Wim Taymans committed
700
      return 1;
701
    else if (second->pageno > first->pageno)
Wim Taymans's avatar
Wim Taymans committed
702
703
704
705
706
707
708
      return -1;
  }

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

709
710
711
712
713
714
715
716
717
718
719
720
721
/* make sure at least one buffer is queued on all pads, two if possible
 * 
 * if pad->buffer == NULL, pad->next_buffer !=  NULL, then
 *   we do not know if the buffer is the last or not
 * if pad->buffer != NULL, pad->next_buffer != NULL, then
 *   pad->buffer is not the last buffer for the pad
 * if pad->buffer != NULL, pad->next_buffer == NULL, then
 *   pad->buffer if the last buffer for the pad
 * 
 * returns a pointer to an oggpad that holds the best buffer, or
 * NULL when no pad was usable. "best" means the buffer marked
 * with the lowest timestamp. If best->buffer == NULL then nothing
 * should be done until more data arrives */
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
747
748
749
750
751
752
753
      /* shift the buffer along if needed (it's okay if next_buffer is NULL) */
      if (pad->buffer == NULL) {
        GST_LOG_OBJECT (data->pad, "shifting buffer %" GST_PTR_FORMAT,
            pad->next_buffer);
        pad->buffer = pad->next_buffer;
        pad->next_buffer = NULL;
      }

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

757
758
      /* On EOS we get a NULL buffer */
      if (buf != NULL) {
759
760
761
762
        if (ogg_mux->delta_pad == NULL &&
            GST_BUFFER_FLAG_IS_SET (buf, GST_BUFFER_FLAG_DELTA_UNIT))
          ogg_mux->delta_pad = pad;

763
764
765
766
767
        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) {
768
769
            GST_DEBUG_OBJECT (ogg_mux,
                "got incaps buffer in control state, ignoring");
770
771
772
773
            /* just ignore */
            gst_buffer_unref (buf);
            buf = NULL;
          } else {
774
775
            GST_DEBUG_OBJECT (ogg_mux,
                "got data buffer in control state, switching " "to data mode");
776
777
778
            /* this is a data buffer so switch to data state */
            pad->state = GST_OGG_PAD_STATE_DATA;
          }
Wim Taymans's avatar
Wim Taymans committed
779
        }
780
      } else {
781
        GST_DEBUG_OBJECT (data->pad, "EOS on pad");
782
783
784
785
        if (!pad->eos) {
          ogg_page page;
          GstFlowReturn ret;

786
787
788
          /* it's no longer active */
          ogg_mux->active_pads--;

789
790
791
792
793
794
795
796
797
798
799
800
801
          /* Just gone to EOS. Flush existing page(s) */
          pad->eos = TRUE;

          while (ogg_stream_flush (&pad->stream, &page)) {
            /* Place page into the per-pad queue */
            ret = gst_ogg_mux_pad_queue_page (ogg_mux, pad, &page,
                pad->first_delta);
            /* increment the page number counter */
            pad->pageno++;
            /* mark other pages as delta */
            pad->first_delta = TRUE;
          }
        }
Wim Taymans's avatar
Wim Taymans committed
802
      }
803

804
      pad->next_buffer = buf;
Wim Taymans's avatar
Wim Taymans committed
805
    }
Wim Taymans's avatar
Wim Taymans committed
806
807
808

    /* we should have a buffer now, see if it is the best pad to
     * pull on */
809
    if (pad->buffer || pad->next_buffer) {
810
      if (gst_ogg_mux_compare_pads (ogg_mux, bestpad, pad) > 0) {
811
812
813
        GST_LOG_OBJECT (data->pad,
            "new best pad, with buffers %" GST_PTR_FORMAT
            " and %" GST_PTR_FORMAT, pad->buffer, pad->next_buffer);
814

Wim Taymans's avatar
Wim Taymans committed
815
        bestpad = pad;
816
      }
817
    } else if (!pad->eos) {
818
      GST_LOG_OBJECT (data->pad, "hungry pad");
819
      still_hungry = pad;
Wim Taymans's avatar
Wim Taymans committed
820
821
    }
  }
822
823
824
825
826
827

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

830
831
832
833
834
835
static GList *
gst_ogg_mux_get_headers (GstOggPad * pad)
{
  GList *res = NULL;
  GstOggMux *ogg_mux;
  GstStructure *structure;
836
  GstCaps *caps;
Wim Taymans's avatar
Wim Taymans committed
837
  GstPad *thepad;
838

Wim Taymans's avatar
Wim Taymans committed
839
  thepad = pad->collect.pad;
840

Wim Taymans's avatar
Wim Taymans committed
841
  ogg_mux = GST_OGG_MUX (GST_PAD_PARENT (thepad));
842

843
  GST_LOG_OBJECT (thepad, "getting headers");
Wim Taymans's avatar
Wim Taymans committed
844
845

  caps = gst_pad_get_negotiated_caps (thepad);
846
847
848
849
  if (caps != NULL) {
    const GValue *streamheader;

    structure = gst_caps_get_structure (caps, 0);
850
851
852
853
854
855
    streamheader = gst_structure_get_value (structure, "streamheader");
    if (streamheader != NULL) {
      GST_LOG_OBJECT (thepad, "got header");
      if (G_VALUE_TYPE (streamheader) == GST_TYPE_ARRAY) {
        GArray *bufarr = g_value_peek_pointer (streamheader);
        gint i;
856

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

859
860
        for (i = 0; i < bufarr->len; i++) {
          GValue *bufval = &g_array_index (bufarr, GValue, i);
861

862
863
864
          GST_LOG_OBJECT (thepad, "item %d", i);
          if (G_VALUE_TYPE (bufval) == GST_TYPE_BUFFER) {
            GstBuffer *buf = g_value_peek_pointer (bufval);
865

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

868
869
            gst_buffer_ref (buf);
            res = g_list_append (res, buf);
870
871
          }
        }
Wim Taymans's avatar
Wim Taymans committed
872
      } else {
873
        GST_LOG_OBJECT (thepad, "streamheader is not fixed list");
874
      }
875
876
877
878
    } else if (gst_structure_has_name (structure, "video/x-dirac")) {
      res = g_list_append (res, pad->buffer);
      pad->buffer = pad->next_buffer;
      pad->next_buffer = NULL;
879
      pad->always_flush_page = TRUE;
880
881
    } else {
      GST_LOG_OBJECT (thepad, "caps don't have streamheader");
882
    }
883
    gst_caps_unref (caps);
Wim Taymans's avatar
Wim Taymans committed
884
  } else {
885
    GST_LOG_OBJECT (thepad, "got empty caps as negotiated format");
886
887
888
889
  }
  return res;
}

Wim Taymans's avatar
Wim Taymans committed
890
static GstCaps *
891
892
gst_ogg_mux_set_header_on_caps (GstCaps * caps, GList * buffers)
{
Wim Taymans's avatar
Wim Taymans committed
893
  GstStructure *structure;
894
  GValue array = { 0 };
895
896
  GList *walk = buffers;

Wim Taymans's avatar
Wim Taymans committed
897
898
899
900
  caps = gst_caps_make_writable (caps);

  structure = gst_caps_get_structure (caps, 0);

901
  /* put buffers in a fixed list */
902
  g_value_init (&array, GST_TYPE_ARRAY);
903
904
905

  while (walk) {
    GstBuffer *buf = GST_BUFFER (walk->data);
906
    GstBuffer *copy;
907
908
909
910
911
    GValue value = { 0 };

    walk = walk->next;

    /* mark buffer */
912
    GST_LOG ("Setting IN_CAPS on buffer of length %d", GST_BUFFER_SIZE (buf));
913
    GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_IN_CAPS);
914
915

    g_value_init (&value, GST_TYPE_BUFFER);
916
917
918
    copy = gst_buffer_copy (buf);
    gst_value_set_buffer (&value, copy);
    gst_buffer_unref (copy);
919
    gst_value_array_append_value (&array, &value);
920
921
    g_value_unset (&value);
  }
922
923
  gst_structure_set_value (structure, "streamheader", &array);
  g_value_unset (&array);
Wim Taymans's avatar
Wim Taymans committed
924
925

  return caps;
926
927
}

928
/*
929
930
931
932
933
934
935
 * 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
936
static GstFlowReturn
937
938
939
940
941
gst_ogg_mux_send_headers (GstOggMux * mux)
{
  GSList *walk;
  GList *hbufs, *hwalk;
  GstCaps *caps;
Wim Taymans's avatar
Wim Taymans committed
942
  GstFlowReturn ret;
943
944

  hbufs = NULL;
Wim Taymans's avatar
Wim Taymans committed
945
  ret = GST_FLOW_OK;
946

947
  GST_LOG_OBJECT (mux, "collecting headers");
948

Wim Taymans's avatar
Wim Taymans committed
949
  walk = mux->collect->data;
950
  while (walk) {
Wim Taymans's avatar
Wim Taymans committed
951
952
    GstOggPad *pad;
    GstPad *thepad;
953

Wim Taymans's avatar
Wim Taymans committed
954
955
    pad = (GstOggPad *) walk->data;
    thepad = pad->collect.pad;
956

Wim Taymans's avatar
Wim Taymans committed
957
958
    walk = g_slist_next (walk);

959
    GST_LOG_OBJECT (mux, "looking at pad %s:%s", GST_DEBUG_PAD_NAME (thepad));
960
961

    /* if the pad has no buffer, we don't care */
962
    if (pad->buffer == NULL && pad->next_buffer == NULL)
963
964
965
966
967
968
      continue;

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

969
  GST_LOG_OBJECT (mux, "creating BOS pages");
Wim Taymans's avatar
Wim Taymans committed
970
  walk = mux->collect->data;
971
  while (walk) {
Wim Taymans's avatar
Wim Taymans committed
972
    GstOggPad *pad;
973
974
975
    GstBuffer *buf;
    ogg_packet packet;
    ogg_page page;
Wim Taymans's avatar
Wim Taymans committed
976
    GstPad *thepad;
977
978
979
    GstCaps *caps;
    GstStructure *structure;
    GstBuffer *hbuf;
Wim Taymans's avatar
Wim Taymans committed
980
981
982

    pad = (GstOggPad *) walk->data;
    thepad = pad->collect.pad;
983
984
    caps = gst_pad_get_negotiated_caps (thepad);
    structure = gst_caps_get_structure (caps, 0);
985
986
987
988
989

    walk = walk->next;

    pad->packetno = 0;

990
    GST_LOG_OBJECT (thepad, "looping over headers");
991
992
993
994

    if (pad->headers) {
      buf = GST_BUFFER (pad->headers->data);
      pad->headers = g_list_remove (pad->headers, buf);
995
    } else if (pad->buffer) {
996
997
      buf = pad->buffer;
      gst_buffer_ref (buf);
998
999
1000
    } else if (pad->next_buffer) {
      buf = pad->next_buffer;
      gst_buffer_ref (buf);
1001
    } else {
1002
      /* fixme -- should be caught in the previous list traversal. */
1003
      GST_OBJECT_LOCK (pad);
1004
1005
      g_critical ("No headers or buffers on pad %s:%s",
          GST_DEBUG_PAD_NAME (pad));
1006
      GST_OBJECT_UNLOCK (pad);
1007
      continue;
1008
1009
1010
1011
1012
1013
    }

    /* 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);
1014
1015
    if (packet.granulepos == -1)
      packet.granulepos = 0;
1016
1017
1018
1019
1020
1021
1022
1023
1024
1025
    /* 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);

1026
1027
1028
1029
1030
    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);
1031

1032
1033
1034
    GST_LOG_OBJECT (mux, "swapped out page with mime type %s",
        gst_structure_get_name (structure));

1035
    /* quick hack: put Theora and Dirac video pages at the front.
1036
     * Ideally, we would have a settable enum for which Ogg
1037
1038
1039
1040
1041
1042
1043
     * profile we work with, and order based on that.
     * (FIXME: if there is more than one video stream, shouldn't we only put
     * one's BOS into the first page, followed by an audio stream's BOS, and
     * only then followed by the remaining video and audio streams?) */
    if (gst_structure_has_name (structure, "video/x-theora")) {
      GST_DEBUG_OBJECT (thepad, "putting %s page at the front", "Theora");
      hbufs = g_list_prepend (hbufs, hbuf);
1044
      pad->always_flush_page = TRUE;
1045
1046
    } else if (gst_structure_has_name (structure, "video/x-dirac")) {
      GST_DEBUG_OBJECT (thepad, "putting %s page at the front", "Dirac");
1047
      hbufs = g_list_prepend (hbufs, hbuf);
1048
      pad->always_flush_page = TRUE;
1049
    } else {
1050
1051
      hbufs = g_list_append (hbufs, hbuf);
    }
1052
    gst_caps_unref (caps);
1053
1054
  }

1055
  GST_LOG_OBJECT (mux, "creating next headers");
Wim Taymans's avatar
Wim Taymans committed
1056
  walk = mux->collect->data;
1057
  while (walk) {
Wim Taymans's avatar
Wim Taymans committed
1058
1059
1060
1061
1062
    GstOggPad *pad;
    GstPad *thepad;

    pad = (GstOggPad *) walk->data;
    thepad = pad->collect.pad;
1063
1064
1065

    walk = walk->next;

1066
1067
    GST_LOG_OBJECT (mux, "looping over headers for pad %s:%s",
        GST_DEBUG_PAD_NAME (thepad));
1068
1069
1070
1071
1072
1073
1074
1075
1076
1077
1078
1079
1080

    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);
1081
1082
      if (packet.granulepos == -1)
        packet.granulepos = 0;
1083
1084
1085
1086
1087
1088
1089
1090
1091
1092
1093
1094
      /* 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) {
1095
        GST_LOG_OBJECT (mux,
Tim-Philipp Müller's avatar
Tim-Philipp Müller committed
1096
1097
            "flushing page as packet %" G_GUINT64_FORMAT " is first or "
            "last packet", pad->packetno);
1098
        while (ogg_stream_flush (&pad->stream, &page)) {
1099
          GstBuffer *hbuf = gst_ogg_mux_buffer_from_page (mux, &page, FALSE);
1100

1101
          GST_LOG_OBJECT (mux, "swapped out page");
1102
1103
1104
          hbufs = g_list_append (hbufs, hbuf);
        }
      } else {
1105
        GST_LOG_OBJECT (mux, "try to swap out page");
1106
1107
        /* just try to swap out a page then */
        while (ogg_stream_pageout (&pad->stream, &page) > 0) {
1108
          GstBuffer *hbuf = gst_ogg_mux_buffer_from_page (mux, &page, FALSE);
1109

1110
          GST_LOG_OBJECT (mux, "swapped out page");
1111
1112
1113
1114
          hbufs = g_list_append (hbufs, hbuf);
        }
      }
    }
Wim Taymans's avatar
Wim Taymans committed
1115
1116
    g_list_free (pad->headers);
    pad->headers = NULL;
1117
1118
1119
1120
1121
1122
  }
  /* 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
1123
1124
    caps = gst_ogg_mux_set_header_on_caps (caps, hbufs);
    gst_pad_set_caps (mux->srcpad, caps);
1125
    gst_caps_unref (caps);
1126
1127
1128
1129
1130
1131
1132
1133
  }
  /* and send the buffers */
  hwalk = hbufs;
  while (hwalk) {
    GstBuffer *buf = GST_BUFFER (hwalk->data);

    hwalk = hwalk->next;

1134
    if ((ret = gst_ogg_mux_push_buffer (mux, buf)) != GST_FLOW_OK)
Wim Taymans's avatar
Wim Taymans committed
1135
      break;
1136
  }
Wim Taymans's avatar
Wim Taymans committed
1137
  g_list_free (hbufs);
Wim Taymans's avatar
Wim Taymans committed
1138
1139

  return ret;
1140
1141
}

1142
/* this function is called to process data on the best pending pad.
Wim Taymans's avatar
Wim Taymans committed
1143
1144
 *
 * basic idea:
Wim Taymans's avatar
Wim Taymans committed
1145
 *
1146
 * 1) store the selected pad and keep on pulling until we fill a
1147
 *    complete ogg page or the ogg page is filled above the max-delay
1148
 *    threshold. This is needed because the ogg spec says that
Wim Taymans's avatar
Wim Taymans committed
1149
1150
 *    you should fill a complete page with data from the same logical
 *    stream. When the page is filled, go back to 1).
1151
 * 2) before filling a page, read ahead one more buffer to see if this
1152
1153
 *    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
1154
1155
1156
 *    sending it to ogg. if pad->buffer is NULL we need to wait to find out
 *    whether there are any more buffers.
 * 3) pages get queued on a per-pad queue. Every time a page is queued, a
1157
 *    dequeue is called, which will dequeue the oldest page on any pad, provided
1158
 *    that ALL pads have at least one marked page in the queue (or remaining
1159
 *    pads are at EOS)