gstoggmux.c 50.5 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
#include <ogg/ogg.h>
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
29
/* memcpy - if someone knows a way to get rid of it, please speak up
Wim Taymans's avatar
Wim Taymans committed
30
31
32
 * note: the ogg docs even say you need this... */
#include <string.h>
#include <time.h>
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
33
#include <stdlib.h>             /* rand, srand, atoi */
Wim Taymans's avatar
Wim Taymans committed
34
35
36
37
38
39
40
41

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))
42
#define GST_IS_OGG_MUX_CLASS(klass) (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_OGG_MUX))
Wim Taymans's avatar
Wim Taymans committed
43

44
45
46
47
48
49
50
51
/* 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))

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

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

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

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

69
70
71
72
  /* These two buffers make a very simple queue - they enter as 'next_buffer'
   * and (usually) leave as 'buffer', except at EOS, when buffer will be NULL */
  GstBuffer *buffer;            /* the first waiting buffer for the pad */
  GstBuffer *next_buffer;       /* the second waiting buffer for the pad */
Wim Taymans's avatar
Wim Taymans committed
73
74
75
76
77

  gint serial;
  ogg_stream_state stream;
  gint64 packetno;              /* number of next packet */
  gint64 pageno;                /* number of next page */
78
  guint64 duration;             /* duration of current page */
Wim Taymans's avatar
Wim Taymans committed
79
  gboolean eos;
80
  gint64 offset;
81
82
83
84
85
86
87
  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
88

Wim Taymans's avatar
Wim Taymans committed
89
  GstOggPadState state;         /* state of the pad */
90
91

  GList *headers;
92

93
94
  GQueue *pagebuffers;          /* List of pages in buffers ready for pushing */

95
96
97
  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
98
99
100
101
102
103
104
}
GstOggPad;

struct _GstOggMux
{
  GstElement element;

Wim Taymans's avatar
Wim Taymans committed
105
  /* source pad */
Wim Taymans's avatar
Wim Taymans committed
106
107
  GstPad *srcpad;

Wim Taymans's avatar
Wim Taymans committed
108
109
  /* sinkpads */
  GstCollectPads *collect;
Wim Taymans's avatar
Wim Taymans committed
110

111
112
113
  /* number of pads which have not received EOS */
  gint active_pads;

Wim Taymans's avatar
Wim Taymans committed
114
  /* the pad we are currently using to fill a page */
Wim Taymans's avatar
Wim Taymans committed
115
116
117
118
119
  GstOggPad *pulling;

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

120
121
122
  /* Last timestamp actually output on src pad */
  GstClockTime last_ts;

Wim Taymans's avatar
Wim Taymans committed
123
124
  /* offset in stream */
  guint64 offset;
125
126
127

  /* need_headers */
  gboolean need_headers;
128
129

  guint64 max_delay;
130
  guint64 max_page_delay;
131

132
  GstOggPad *delta_pad;         /* when a delta frame is detected on a stream, we mark
133
134
                                   pages as delta frames up to the page that has the
                                   keyframe */
135

Wim Taymans's avatar
Wim Taymans committed
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
};

typedef enum
{
  GST_OGG_FLAG_BOS = GST_ELEMENT_FLAG_LAST,
  GST_OGG_FLAG_EOS
}
GstOggFlag;

struct _GstOggMuxClass
{
  GstElementClass parent_class;
};

/* elementfactory information */
Stefan Kost's avatar
Stefan Kost committed
151
152
static const GstElementDetails gst_ogg_mux_details =
GST_ELEMENT_DETAILS ("Ogg muxer",
Wim Taymans's avatar
Wim Taymans committed
153
154
155
156
157
158
159
160
161
162
163
    "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
164
/* set to 0.5 seconds by default */
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
165
166
#define DEFAULT_MAX_DELAY       G_GINT64_CONSTANT(500000000)
#define DEFAULT_MAX_PAGE_DELAY  G_GINT64_CONSTANT(500000000)
Wim Taymans's avatar
Wim Taymans committed
167
168
enum
{
169
170
  ARG_0,
  ARG_MAX_DELAY,
171
  ARG_MAX_PAGE_DELAY,
Wim Taymans's avatar
Wim Taymans committed
172
173
174
175
176
177
178
179
180
181
182
};

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,
183
184
    GST_STATIC_CAPS ("video/x-theora; "
        "audio/x-vorbis; audio/x-flac; audio/x-speex; "
185
        "application/x-ogm-video; application/x-ogm-audio; video/x-dirac; "
186
        "video/x-smoke; text/x-cmml, encoded = (boolean) TRUE")
Wim Taymans's avatar
Wim Taymans committed
187
188
189
190
191
    );

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

Wim Taymans's avatar
Wim Taymans committed
194
195
static GstFlowReturn
gst_ogg_mux_collected (GstCollectPads * pads, GstOggMux * ogg_mux);
Wim Taymans's avatar
Wim Taymans committed
196
197
198
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);
199
200
static void gst_ogg_mux_release_pad (GstElement * element, GstPad * pad);

Wim Taymans's avatar
Wim Taymans committed
201
202
203
204
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);
205
206
static GstStateChangeReturn gst_ogg_mux_change_state (GstElement * element,
    GstStateChange transition);
Wim Taymans's avatar
Wim Taymans committed
207
208
209
210
211
212
213
214
215
216

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;

217
  if (G_UNLIKELY (ogg_mux_type == 0)) {
Wim Taymans's avatar
Wim Taymans committed
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
249
250
251
252
253
254
255
256
257
258
    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;

259
  parent_class = g_type_class_peek_parent (klass);
Wim Taymans's avatar
Wim Taymans committed
260

261
  gobject_class->finalize = gst_ogg_mux_finalize;
262
263
264
  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
265
  gstelement_class->request_new_pad = gst_ogg_mux_request_new_pad;
266
  gstelement_class->release_pad = gst_ogg_mux_release_pad;
Wim Taymans's avatar
Wim Taymans committed
267

268
269
270
271
  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));
272
273
274
275
  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));
276

Wim Taymans's avatar
Wim Taymans committed
277
278
279
280
  gstelement_class->change_state = gst_ogg_mux_change_state;

}

Wim Taymans's avatar
Wim Taymans committed
281
#if 0
Wim Taymans's avatar
Wim Taymans committed
282
283
284
285
286
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
287
    {GST_EVENT_DISCONTINUOUS, 0},
Wim Taymans's avatar
Wim Taymans committed
288
289
290
291
292
    {0,}
  };

  return gst_ogg_mux_sink_event_masks;
}
Wim Taymans's avatar
Wim Taymans committed
293
#endif
Wim Taymans's avatar
Wim Taymans committed
294

295
296
297
298
299
300
301
302
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;
303
304
  ogg_mux->offset = 0;
  ogg_mux->next_ts = 0;
305
  ogg_mux->last_ts = GST_CLOCK_TIME_NONE;
306
307
}

Wim Taymans's avatar
Wim Taymans committed
308
309
310
311
312
313
314
315
316
317
318
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);

319
  GST_OBJECT_FLAG_SET (GST_ELEMENT (ogg_mux), GST_OGG_FLAG_BOS);
Wim Taymans's avatar
Wim Taymans committed
320
321
322
323

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

324
325
  ogg_mux->collect = gst_collect_pads_new ();
  gst_collect_pads_set_function (ogg_mux->collect,
326
327
      (GstCollectPadsFunction) GST_DEBUG_FUNCPTR (gst_ogg_mux_collected),
      ogg_mux);
328

329
  gst_ogg_mux_clear (ogg_mux);
Wim Taymans's avatar
Wim Taymans committed
330
331
}

332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
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
347
static GstPadLinkReturn
348
gst_ogg_mux_sinkconnect (GstPad * pad, GstPad * peer)
Wim Taymans's avatar
Wim Taymans committed
349
350
351
352
353
{
  GstOggMux *ogg_mux;

  ogg_mux = GST_OGG_MUX (gst_pad_get_parent (pad));

354
  GST_DEBUG_OBJECT (ogg_mux, "sinkconnect triggered on %s", GST_PAD_NAME (pad));
355

356
  gst_object_unref (ogg_mux);
Wim Taymans's avatar
Wim Taymans committed
357
358
359
360
361
362
363
364
365
366

  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
367
  GstElementClass *klass;
Wim Taymans's avatar
Wim Taymans committed
368
369
370

  g_return_val_if_fail (templ != NULL, NULL);

Wim Taymans's avatar
Wim Taymans committed
371
372
  if (templ->direction != GST_PAD_SINK)
    goto wrong_direction;
Wim Taymans's avatar
Wim Taymans committed
373
374
375
376

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

Wim Taymans's avatar
Wim Taymans committed
377
378
379
380
381
382
  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
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
    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
401
402
403
      GstOggPad *oggpad;

      oggpad = (GstOggPad *)
404
          gst_collect_pads_add_pad (ogg_mux->collect, newpad,
Wim Taymans's avatar
Wim Taymans committed
405
          sizeof (GstOggPad));
406
      ogg_mux->active_pads++;
Wim Taymans's avatar
Wim Taymans committed
407
408
409
410
411
412
413
414

      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;
415
416
417
      oggpad->new_page = TRUE;
      oggpad->first_delta = FALSE;
      oggpad->prev_delta = FALSE;
418
      oggpad->pagebuffers = g_queue_new ();
Wim Taymans's avatar
Wim Taymans committed
419
420
421
422
423
424
425
426
427
    }
  }

  /* 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
428
429
430
431
432
433
434
435
436
437
438
439

  /* 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
440
441
}

442
443
444
445
446
447
448
449
450
451
452
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));

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

  GST_OBJECT_LOCK (ogg_mux->collect);
453
  for (walk = ogg_mux->collect->data; walk; walk = g_slist_next (walk)) {
454
455
456
457
458
    GstOggPad *oggpad = (GstOggPad *) walk->data;
    GstCollectData *cdata = (GstCollectData *) walk->data;
    GstBuffer *buf;

    if (cdata->pad == pad) {
459
      ogg_stream_clear (&oggpad->stream);
460
461
462
463
464
465

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

      g_queue_free (oggpad->pagebuffers);
466
      oggpad->pagebuffers = NULL;
467
468
469
470
471
    }
  }
  GST_OBJECT_UNLOCK (ogg_mux->collect);

  gst_collect_pads_remove_pad (ogg_mux->collect, pad);
472
  gst_element_remove_pad (element, pad);
473
474
}

Wim Taymans's avatar
Wim Taymans committed
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
/* 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);
}

494
static GstBuffer *
495
gst_ogg_mux_buffer_from_page (GstOggMux * mux, ogg_page * page, gboolean delta)
Wim Taymans's avatar
Wim Taymans committed
496
497
498
499
{
  GstBuffer *buffer;

  /* allocate space for header and body */
Wim Taymans's avatar
Wim Taymans committed
500
  buffer = gst_buffer_new_and_alloc (page->header_len + page->body_len);
Wim Taymans's avatar
Wim Taymans committed
501
502
503
504
  memcpy (GST_BUFFER_DATA (buffer), page->header, page->header_len);
  memcpy (GST_BUFFER_DATA (buffer) + page->header_len,
      page->body, page->body_len);

505
506
507
508
  /* 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);
509
  if (delta)
510
    GST_BUFFER_FLAG_SET (buffer, GST_BUFFER_FLAG_DELTA_UNIT);
Wim Taymans's avatar
Wim Taymans committed
511

512
  GST_LOG_OBJECT (mux, GST_GP_FORMAT
513
      " created buffer %p from ogg page", ogg_page_granulepos (page), buffer);
514

515
516
517
  return buffer;
}

Wim Taymans's avatar
Wim Taymans committed
518
static GstFlowReturn
519
gst_ogg_mux_push_buffer (GstOggMux * mux, GstBuffer * buffer)
520
{
521
522
  GstCaps *caps;

523
524
525
526
  /* 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;
527

528
529
  /* Ensure we have monotonically increasing timestamps in the output. */
  if (GST_BUFFER_TIMESTAMP_IS_VALID (buffer)) {
530
531
    if (mux->last_ts != GST_CLOCK_TIME_NONE &&
        GST_BUFFER_TIMESTAMP (buffer) < mux->last_ts)
532
533
534
535
536
      GST_BUFFER_TIMESTAMP (buffer) = mux->last_ts;
    else
      mux->last_ts = GST_BUFFER_TIMESTAMP (buffer);
  }

537
538
539
540
  caps = gst_pad_get_negotiated_caps (mux->srcpad);
  gst_buffer_set_caps (buffer, caps);
  gst_caps_unref (caps);

541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
  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) {
567
568
        GST_LOG_OBJECT (pad->collect.pad,
            "pad is EOS, skipping for dequeue decision");
569
      } else {
570
571
        GST_LOG_OBJECT (pad->collect.pad,
            "no pages in this queue, can't dequeue");
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
        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) {
589
590
        GST_LOG_OBJECT (pad->collect.pad,
            "No page timestamps in queue, can't dequeue");
591
592
593
594
595
596
597
598
599
600
601
        return FALSE;
      }
    }

    walk = g_slist_next (walk);
  }

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

602
    /* any page with a granulepos of -1 can be pushed immediately.
603
604
605
     * 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
606
      GST_LOG_OBJECT (pad->collect.pad, "[gp        -1] pushing page");
607
608
609
610
611
612
613
614
615
      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) {
616
617
618
619
        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);
620
621
622
        opad = pad;
      } else {
        /* if we have an oldest, compare with this one */
623
624
625
626
627
        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);
628
629
630
631
632
633
634
635
636
637
          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
638
    GST_LOG_OBJECT (opad->collect.pad,
639
640
641
        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)));
642
643
644
645
646
647
648
    *flowret = gst_ogg_mux_push_buffer (mux, buf);
    ret = TRUE;
  }

  return ret;
}

649
650
651
652
653
654
655
656
657
658
659
660
661
/* 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.
 */
662
663
664
665
static GstFlowReturn
gst_ogg_mux_pad_queue_page (GstOggMux * mux, GstOggPad * pad, ogg_page * page,
    gboolean delta)
{
Wim Taymans's avatar
Wim Taymans committed
666
  GstFlowReturn ret;
667
  GstBuffer *buffer = gst_ogg_mux_buffer_from_page (mux, page, delta);
668

669
  /* take the timestamp of the first packet on this page */
670
  GST_BUFFER_TIMESTAMP (buffer) = pad->timestamp;
671
  GST_BUFFER_DURATION (buffer) = pad->timestamp_end - pad->timestamp;
672
673
  /* take the gp time of the last completed packet on this page */
  GST_BUFFER_OFFSET (buffer) = pad->gp_time;
674

675
  /* the next page will start where the current page's end time leaves off */
676
677
  pad->timestamp = pad->timestamp_end;

678
  g_queue_push_tail (pad->pagebuffers, buffer);
679
680
681
682
683
  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)),
684
685
      GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (buffer)),
      g_queue_get_length (pad->pagebuffers));
Wim Taymans's avatar
Wim Taymans committed
686

687
688
689
690
  while (gst_ogg_mux_dequeue_page (mux, &ret)) {
    if (ret != GST_FLOW_OK)
      break;
  }
Wim Taymans's avatar
Wim Taymans committed
691
692

  return ret;
Wim Taymans's avatar
Wim Taymans committed
693
694
695
}

/*
696
697
698
699
700
701
702
 * 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
703
704
 */
static gint
705
706
gst_ogg_mux_compare_pads (GstOggMux * ogg_mux, GstOggPad * first,
    GstOggPad * second)
Wim Taymans's avatar
Wim Taymans committed
707
{
708
  guint64 firsttime, secondtime;
Wim Taymans's avatar
Wim Taymans committed
709

710
711
  /* if the first pad doesn't contain anything or is even NULL, return
   * the second pad as best candidate and vice versa */
712
  if (first == NULL || (first->buffer == NULL && first->next_buffer == NULL))
Wim Taymans's avatar
Wim Taymans committed
713
    return 1;
714
  if (second == NULL || (second->buffer == NULL && second->next_buffer == NULL))
Wim Taymans's avatar
Wim Taymans committed
715
716
    return -1;

717
  /* no timestamp on first buffer, it must go first */
718
719
720
721
  if (first->buffer)
    firsttime = GST_BUFFER_TIMESTAMP (first->buffer);
  else
    firsttime = GST_BUFFER_TIMESTAMP (first->next_buffer);
722
  if (firsttime == GST_CLOCK_TIME_NONE)
Wim Taymans's avatar
Wim Taymans committed
723
724
    return -1;

725
  /* no timestamp on second buffer, it must go first */
726
727
728
729
  if (second->buffer)
    secondtime = GST_BUFFER_TIMESTAMP (second->buffer);
  else
    secondtime = GST_BUFFER_TIMESTAMP (second->next_buffer);
730
  if (secondtime == GST_CLOCK_TIME_NONE)
Wim Taymans's avatar
Wim Taymans committed
731
732
    return 1;

733
734
  /* first buffer has higher timestamp, second one should go first */
  if (secondtime < firsttime)
Wim Taymans's avatar
Wim Taymans committed
735
    return 1;
736
737
  /* second buffer has higher timestamp, first one should go first */
  else if (secondtime > firsttime)
Wim Taymans's avatar
Wim Taymans committed
738
739
740
741
    return -1;
  else {
    /* buffers with equal timestamps, prefer the pad that has the
     * least number of pages muxed */
742
    if (second->pageno < first->pageno)
Wim Taymans's avatar
Wim Taymans committed
743
      return 1;
744
    else if (second->pageno > first->pageno)
Wim Taymans's avatar
Wim Taymans committed
745
746
747
748
749
750
751
      return -1;
  }

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

752
753
754
755
756
757
758
759
760
761
762
763
764
/* 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
765
static GstOggPad *
Wim Taymans's avatar
Wim Taymans committed
766
gst_ogg_mux_queue_pads (GstOggMux * ogg_mux)
Wim Taymans's avatar
Wim Taymans committed
767
{
768
  GstOggPad *bestpad = NULL, *still_hungry = NULL;
Wim Taymans's avatar
Wim Taymans committed
769
770
771
  GSList *walk;

  /* try to make sure we have a buffer from each usable pad first */
Wim Taymans's avatar
Wim Taymans committed
772
  walk = ogg_mux->collect->data;
Wim Taymans's avatar
Wim Taymans committed
773
  while (walk) {
Wim Taymans's avatar
Wim Taymans committed
774
775
    GstOggPad *pad;
    GstCollectData *data;
Wim Taymans's avatar
Wim Taymans committed
776

Wim Taymans's avatar
Wim Taymans committed
777
778
    data = (GstCollectData *) walk->data;
    pad = (GstOggPad *) data;
Wim Taymans's avatar
Wim Taymans committed
779

Wim Taymans's avatar
Wim Taymans committed
780
781
    walk = g_slist_next (walk);

782
    GST_LOG_OBJECT (data->pad, "looking at pad for buffer");
783

Wim Taymans's avatar
Wim Taymans committed
784
    /* try to get a new buffer for this pad if needed and possible */
Wim Taymans's avatar
Wim Taymans committed
785
786
787
    if (pad->buffer == NULL) {
      GstBuffer *buf;
      gboolean incaps;
Wim Taymans's avatar
Wim Taymans committed
788

789
790
791
792
793
794
795
796
      /* 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;
      }

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

800
801
802
803
804
805
806
      /* 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) {
807
808
            GST_DEBUG_OBJECT (ogg_mux,
                "got incaps buffer in control state, ignoring");
809
810
811
812
            /* just ignore */
            gst_buffer_unref (buf);
            buf = NULL;
          } else {
813
814
            GST_DEBUG_OBJECT (ogg_mux,
                "got data buffer in control state, switching " "to data mode");
815
816
817
            /* this is a data buffer so switch to data state */
            pad->state = GST_OGG_PAD_STATE_DATA;
          }
Wim Taymans's avatar
Wim Taymans committed
818
        }
819
      } else {
820
        GST_DEBUG_OBJECT (data->pad, "EOS on pad");
821
822
823
824
        if (!pad->eos) {
          ogg_page page;
          GstFlowReturn ret;

825
826
827
          /* it's no longer active */
          ogg_mux->active_pads--;

828
829
830
831
832
833
834
835
836
837
838
839
840
          /* 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
841
      }
842

843
      pad->next_buffer = buf;
Wim Taymans's avatar
Wim Taymans committed
844
    }
Wim Taymans's avatar
Wim Taymans committed
845
846
847

    /* we should have a buffer now, see if it is the best pad to
     * pull on */
848
    if (pad->buffer || pad->next_buffer) {
849
      if (gst_ogg_mux_compare_pads (ogg_mux, bestpad, pad) > 0) {
850
851
852
        GST_LOG_OBJECT (data->pad,
            "new best pad, with buffers %" GST_PTR_FORMAT
            " and %" GST_PTR_FORMAT, pad->buffer, pad->next_buffer);
853

Wim Taymans's avatar
Wim Taymans committed
854
        bestpad = pad;
855
      }
856
    } else if (!pad->eos) {
857
      GST_LOG_OBJECT (data->pad, "hungry pad");
858
      still_hungry = pad;
Wim Taymans's avatar
Wim Taymans committed
859
860
    }
  }
861
862
863
864
865
866

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

869
870
871
872
873
874
static GList *
gst_ogg_mux_get_headers (GstOggPad * pad)
{
  GList *res = NULL;
  GstOggMux *ogg_mux;
  GstStructure *structure;
875
  GstCaps *caps;
Wim Taymans's avatar
Wim Taymans committed
876
  GstPad *thepad;
877

Wim Taymans's avatar
Wim Taymans committed
878
  thepad = pad->collect.pad;
879

Wim Taymans's avatar
Wim Taymans committed
880
  ogg_mux = GST_OGG_MUX (GST_PAD_PARENT (thepad));
881

882
  GST_LOG_OBJECT (thepad, "getting headers");
Wim Taymans's avatar
Wim Taymans committed
883
884

  caps = gst_pad_get_negotiated_caps (thepad);
885
886
887
888
889
890
  if (caps != NULL) {
    const GValue *streamheader;

    structure = gst_caps_get_structure (caps, 0);
    streamheader = gst_structure_get_value (structure, "streamheader");
    if (streamheader != NULL) {
891
      GST_LOG_OBJECT (thepad, "got header");
892
      if (G_VALUE_TYPE (streamheader) == GST_TYPE_ARRAY) {
893
894
895
        GArray *bufarr = g_value_peek_pointer (streamheader);
        gint i;

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

898
899
900
        for (i = 0; i < bufarr->len; i++) {
          GValue *bufval = &g_array_index (bufarr, GValue, i);

901
          GST_LOG_OBJECT (thepad, "item %d", i);
902
903
904
          if (G_VALUE_TYPE (bufval) == GST_TYPE_BUFFER) {
            GstBuffer *buf = g_value_peek_pointer (bufval);

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

Wim Taymans's avatar
Wim Taymans committed
907
            gst_buffer_ref (buf);
908
909
910
            res = g_list_append (res, buf);
          }
        }
Wim Taymans's avatar
Wim Taymans committed
911
      } else {
912
        GST_LOG_OBJECT (thepad, "streamheader is not fixed list");
913
      }
Wim Taymans's avatar
Wim Taymans committed
914
    } else {
915
      GST_LOG_OBJECT (thepad, "caps done have streamheader");
916
    }
917
    gst_caps_unref (caps);
Wim Taymans's avatar
Wim Taymans committed
918
  } else {
919
    GST_LOG_OBJECT (thepad, "got empty caps as negotiated format");
920
921
922
923
  }
  return res;
}

Wim Taymans's avatar
Wim Taymans committed
924
static GstCaps *
925
926
gst_ogg_mux_set_header_on_caps (GstCaps * caps, GList * buffers)
{
Wim Taymans's avatar
Wim Taymans committed
927
  GstStructure *structure;
928
  GValue array = { 0 };
929
930
  GList *walk = buffers;

Wim Taymans's avatar
Wim Taymans committed
931
932
933
934
  caps = gst_caps_make_writable (caps);

  structure = gst_caps_get_structure (caps, 0);

935
  /* put buffers in a fixed list */
936
  g_value_init (&array, GST_TYPE_ARRAY);
937
938
939

  while (walk) {
    GstBuffer *buf = GST_BUFFER (walk->data);
940
    GstBuffer *copy;
941
942
943
944
945
    GValue value = { 0 };

    walk = walk->next;

    /* mark buffer */
946
    GST_LOG ("Setting IN_CAPS on buffer of length %d", GST_BUFFER_SIZE (buf));
947
    GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_IN_CAPS);
948
949

    g_value_init (&value, GST_TYPE_BUFFER);
950
951
952
    copy = gst_buffer_copy (buf);
    gst_value_set_buffer (&value, copy);
    gst_buffer_unref (copy);
953
    gst_value_array_append_value (&array, &value);
954
955
    g_value_unset (&value);
  }
956
957
  gst_structure_set_value (structure, "streamheader", &array);
  g_value_unset (&array);
Wim Taymans's avatar
Wim Taymans committed
958
959

  return caps;
960
961
}

962
/*
963
964
965
966
967
968
969
 * 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
970
static GstFlowReturn
971
972
973
974
975
gst_ogg_mux_send_headers (GstOggMux * mux)
{
  GSList *walk;
  GList *hbufs, *hwalk;
  GstCaps *caps;
Wim Taymans's avatar
Wim Taymans committed
976
  GstFlowReturn ret;
977
978

  hbufs = NULL;
Wim Taymans's avatar
Wim Taymans committed
979
  ret = GST_FLOW_OK;
980

981
  GST_LOG_OBJECT (mux, "collecting headers");
982

Wim Taymans's avatar
Wim Taymans committed
983
  walk = mux->collect->data;
984
  while (walk) {
Wim Taymans's avatar
Wim Taymans committed
985
986
    GstOggPad *pad;
    GstPad *thepad;
987

Wim Taymans's avatar
Wim Taymans committed
988
989
    pad = (GstOggPad *) walk->data;
    thepad = pad->collect.pad;
990

Wim Taymans's avatar
Wim Taymans committed
991
992
    walk = g_slist_next (walk);

993
    GST_LOG_OBJECT (mux, "looking at pad %s:%s", GST_DEBUG_PAD_NAME (thepad));
994
995

    /* if the pad has no buffer, we don't care */
996
    if (pad->buffer == NULL && pad->next_buffer == NULL)
997
998
999
1000
1001
1002
      continue;

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

1003
  GST_LOG_OBJECT (mux, "creating BOS pages");
Wim Taymans's avatar
Wim Taymans committed
1004
  walk = mux->collect->data;
1005
  while (walk) {
Wim Taymans's avatar
Wim Taymans committed
1006
    GstOggPad *pad;
1007
1008
1009
    GstBuffer *buf;
    ogg_packet packet;
    ogg_page page;
Wim Taymans's avatar
Wim Taymans committed
1010
    GstPad *thepad;
1011
1012
1013
    GstCaps *caps;
    GstStructure *structure;
    GstBuffer *hbuf;
Wim Taymans's avatar
Wim Taymans committed
1014
1015
1016

    pad = (GstOggPad *) walk->data;
    thepad = pad->collect.pad;
1017
1018
    caps = gst_pad_get_negotiated_caps (thepad);
    structure = gst_caps_get_structure (caps, 0);
1019
1020
1021
1022
1023

    walk = walk->next;

    pad->packetno = 0;

1024
    GST_LOG_OBJECT (thepad, "looping over headers");
1025
1026
1027
1028

    if (pad->headers) {
      buf = GST_BUFFER (pad->headers->data);
      pad->headers = g_list_remove (pad->headers, buf);
1029
    } else if (pad->buffer) {
1030
1031
      buf = pad->buffer;
      gst_buffer_ref (buf);
1032
1033
1034
    } else if (pad->next_buffer) {
      buf = pad->next_buffer;
      gst_buffer_ref (buf);
1035
    } else {
1036
      /* fixme -- should be caught in the previous list traversal. */
1037
      GST_OBJECT_LOCK (pad);
1038
1039
      g_critical ("No headers or buffers on pad %s:%s",
          GST_DEBUG_PAD_NAME (pad));
1040
      GST_OBJECT_UNLOCK (pad);
1041
      continue;
1042
1043
1044
1045
1046
1047
    }

    /* 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);
1048
1049
    if (packet.granulepos == -1)
      packet.granulepos = 0;
1050
1051
1052
1053
1054
1055
1056
1057
1058
1059
    /* 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);

1060
1061
1062
1063
1064
    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);
1065

1066
1067
1068
1069
1070
1071
1072
1073
1074
1075
    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 {
1076
1077
      hbufs = g_list_append (hbufs, hbuf);
    }
1078
    gst_caps_unref (caps);
1079
1080
  }

1081
  GST_LOG_OBJECT (mux, "creating next headers");
Wim Taymans's avatar
Wim Taymans committed
1082
  walk = mux->collect->data;
1083
  while (walk) {
Wim Taymans's avatar
Wim Taymans committed
1084
1085
1086
1087
1088
    GstOggPad *pad;
    GstPad *thepad;

    pad = (GstOggPad *) walk->data;
    thepad = pad->collect.pad;
1089
1090
1091

    walk = walk->next;

1092
1093
    GST_LOG_OBJECT (mux, "looping over headers for pad %s:%s",
        GST_DEBUG_PAD_NAME (thepad));
1094
1095
1096
1097
1098
1099
1100
1101
1102
1103
1104
1105
1106

    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);
1107
1108
      if (packet.granulepos == -1)
        packet.granulepos = 0;
1109
1110
1111
1112
1113
1114
1115
1116
1117
1118
1119
1120
      /* 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) {
1121
        GST_LOG_OBJECT (mux,
Tim-Philipp Müller's avatar
Tim-Philipp Müller committed
1122
1123
            "flushing page as packet %" G_GUINT64_FORMAT " is first or "
            "last packet", pad->packetno);
1124
        while (ogg_stream_flush (&pad->stream, &page)) {
1125
          GstBuffer *hbuf = gst_ogg_mux_buffer_from_page (mux, &page, FALSE);
1126

1127
          GST_LOG_OBJECT (mux, "swapped out page");
1128
1129
1130
          hbufs = g_list_append (hbufs, hbuf);
        }
      } else {
1131
        GST_LOG_OBJECT (mux, "try to swap out page");
1132
1133
        /* just try to swap out a page then */
        while (ogg_stream_pageout (&pad->stream, &page) > 0) {
1134
          GstBuffer *hbuf = gst_ogg_mux_buffer_from_page (mux, &page, FALSE);
1135

1136
          GST_LOG_OBJECT (mux, "swapped out page");
1137
1138
1139
1140
          hbufs = g_list_append (hbufs, hbuf);
        }
      }
    }
Wim Taymans's avatar
Wim Taymans committed
1141
1142
    g_list_free (pad->headers);
    pad->headers = NULL;
1143
1144
1145
1146
1147
1148
  }
  /* 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
1149
1150
    caps = gst_ogg_mux_set_header_on_caps (caps, hbufs);
    gst_pad_set_caps (mux->srcpad, caps);
1151
    gst_caps_unref (caps);
1152
1153
1154
1155
1156
1157
1158
1159
  }
  /* and send the buffers */
  hwalk = hbufs;
  while (hwalk) {
    GstBuffer *buf = GST_BUFFER (hwalk->data);

    hwalk = hwalk->next;

1160
    if ((ret = gst_ogg_mux_push_buffer (mux, buf)) != GST_FLOW_OK)
Wim Taymans's avatar
Wim Taymans committed
1161
      break;
1162
  }
Wim Taymans's avatar
Wim Taymans committed
1163
  g_list_free (hbufs);
Wim Taymans's avatar
Wim Taymans committed
1164
1165

  return ret;
1166
1167
}

1168
/* this function is called to process data on the best pending pad.
Wim Taymans's avatar
Wim Taymans committed
1169
1170
 *
 * basic idea:
Wim Taymans's avatar
Wim Taymans committed
1171
 *
1172
 * 1) store the selected pad and keep on pulling until we fill a
1173
 *    complete ogg page or the ogg page is filled above the max-delay
1174
 *    threshold. This is needed because the ogg spec says that
Wim Taymans's avatar
Wim Taymans committed
1175
1176
 *    you should fill a complete page with data from the same logical
 *    stream. When the page is filled, go back to 1).
1177
 * 2) before filling a page, read ahead one more buffer to see if this
1178
1179
 *    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
James Doc Livingston's avatar