docs: get rid of 'Since: 0.10.x' markers
[platform/upstream/gst-plugins-good.git] / gst / wavparse / gstwavparse.c
1 /* -*- Mode: C; tab-width: 2; indent-tabs-mode: t; c-basic-offset: 2 -*- */
2 /* GStreamer
3  * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
4  * Copyright (C) <2006> Nokia Corporation, Stefan Kost <stefan.kost@nokia.com>.
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Library General Public
8  * License as published by the Free Software Foundation; either
9  * version 2 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Library General Public License for more details.
15  *
16  * You should have received a copy of the GNU Library General Public
17  * License along with this library; if not, write to the
18  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
19  * Boston, MA 02110-1301, USA.
20  */
21
22 /**
23  * SECTION:element-wavparse
24  *
25  * Parse a .wav file into raw or compressed audio.
26  *
27  * Wavparse supports both push and pull mode operations, making it possible to
28  * stream from a network source.
29  *
30  * <refsect2>
31  * <title>Example launch line</title>
32  * |[
33  * gst-launch-1.0 filesrc location=sine.wav ! wavparse ! audioconvert ! alsasink
34  * ]| Read a wav file and output to the soundcard using the ALSA element. The
35  * wav file is assumed to contain raw uncompressed samples.
36  * |[
37  * gst-launch-1.0 gnomevfssrc location=http://www.example.org/sine.wav ! queue ! wavparse ! audioconvert ! alsasink
38  * ]| Stream data from a network url.
39  * </refsect2>
40  *
41  * Last reviewed on 2007-02-14 (0.10.6)
42  */
43
44 /*
45  * TODO:
46  * http://replaygain.hydrogenaudio.org/file_format_wav.html
47  */
48
49 #ifdef HAVE_CONFIG_H
50 #include "config.h"
51 #endif
52
53 #include <string.h>
54 #include <math.h>
55
56 #include "gstwavparse.h"
57 #include "gst/riff/riff-media.h"
58 #include <gst/base/gsttypefindhelper.h>
59 #include <gst/gst-i18n-plugin.h>
60
61 GST_DEBUG_CATEGORY_STATIC (wavparse_debug);
62 #define GST_CAT_DEFAULT (wavparse_debug)
63
64 static void gst_wavparse_dispose (GObject * object);
65
66 static gboolean gst_wavparse_sink_activate (GstPad * sinkpad,
67     GstObject * parent);
68 static gboolean gst_wavparse_sink_activate_mode (GstPad * sinkpad,
69     GstObject * parent, GstPadMode mode, gboolean active);
70 static gboolean gst_wavparse_send_event (GstElement * element,
71     GstEvent * event);
72 static GstStateChangeReturn gst_wavparse_change_state (GstElement * element,
73     GstStateChange transition);
74
75 static gboolean gst_wavparse_pad_query (GstPad * pad, GstObject * parent,
76     GstQuery * query);
77 static gboolean gst_wavparse_pad_convert (GstPad * pad, GstFormat src_format,
78     gint64 src_value, GstFormat * dest_format, gint64 * dest_value);
79
80 static GstFlowReturn gst_wavparse_chain (GstPad * pad, GstObject * parent,
81     GstBuffer * buf);
82 static gboolean gst_wavparse_sink_event (GstPad * pad, GstObject * parent,
83     GstEvent * event);
84 static void gst_wavparse_loop (GstPad * pad);
85 static gboolean gst_wavparse_srcpad_event (GstPad * pad, GstObject * parent,
86     GstEvent * event);
87
88 static void gst_wavparse_set_property (GObject * object, guint prop_id,
89     const GValue * value, GParamSpec * pspec);
90 static void gst_wavparse_get_property (GObject * object, guint prop_id,
91     GValue * value, GParamSpec * pspec);
92
93 #define DEFAULT_IGNORE_LENGTH FALSE
94
95 enum
96 {
97   PROP_0,
98   PROP_IGNORE_LENGTH,
99 };
100
101 static GstStaticPadTemplate sink_template_factory =
102 GST_STATIC_PAD_TEMPLATE ("sink",
103     GST_PAD_SINK,
104     GST_PAD_ALWAYS,
105     GST_STATIC_CAPS ("audio/x-wav")
106     );
107
108 #define DEBUG_INIT \
109   GST_DEBUG_CATEGORY_INIT (wavparse_debug, "wavparse", 0, "WAV parser");
110
111 #define gst_wavparse_parent_class parent_class
112 G_DEFINE_TYPE_WITH_CODE (GstWavParse, gst_wavparse, GST_TYPE_ELEMENT,
113     DEBUG_INIT);
114
115 typedef struct
116 {
117   /* Offset Size    Description   Value
118    * 0x00   4       ID            unique identification value
119    * 0x04   4       Position      play order position
120    * 0x08   4       Data Chunk ID RIFF ID of corresponding data chunk
121    * 0x0c   4       Chunk Start   Byte Offset of Data Chunk *
122    * 0x10   4       Block Start   Byte Offset to sample of First Channel
123    * 0x14   4       Sample Offset Byte Offset to sample byte of First Channel
124    */
125   guint32 id;
126   guint32 position;
127   guint32 data_chunk_id;
128   guint32 chunk_start;
129   guint32 block_start;
130   guint32 sample_offset;
131 } GstWavParseCue;
132
133 typedef struct
134 {
135   /* Offset Size    Description     Value
136    * 0x08   4       Cue Point ID    0 - 0xFFFFFFFF
137    * 0x0c           Text
138    */
139   guint32 cue_point_id;
140   gchar *text;
141 } GstWavParseLabl, GstWavParseNote;
142
143 static void
144 gst_wavparse_class_init (GstWavParseClass * klass)
145 {
146   GstElementClass *gstelement_class;
147   GObjectClass *object_class;
148   GstPadTemplate *src_template;
149
150   gstelement_class = (GstElementClass *) klass;
151   object_class = (GObjectClass *) klass;
152
153   parent_class = g_type_class_peek_parent (klass);
154
155   object_class->dispose = gst_wavparse_dispose;
156
157   object_class->set_property = gst_wavparse_set_property;
158   object_class->get_property = gst_wavparse_get_property;
159
160   /**
161    * GstWavParse:ignore-length:
162    *
163    * This selects whether the length found in a data chunk
164    * should be ignored. This may be useful for streamed audio
165    * where the length is unknown until the end of streaming,
166    * and various software/hardware just puts some random value
167    * in there and hopes it doesn't break too much.
168    */
169   g_object_class_install_property (object_class, PROP_IGNORE_LENGTH,
170       g_param_spec_boolean ("ignore-length",
171           "Ignore length",
172           "Ignore length from the Wave header",
173           DEFAULT_IGNORE_LENGTH, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)
174       );
175
176   gstelement_class->change_state = gst_wavparse_change_state;
177   gstelement_class->send_event = gst_wavparse_send_event;
178
179   /* register pads */
180   gst_element_class_add_pad_template (gstelement_class,
181       gst_static_pad_template_get (&sink_template_factory));
182
183   src_template = gst_pad_template_new ("src", GST_PAD_SRC,
184       GST_PAD_ALWAYS, gst_riff_create_audio_template_caps ());
185   gst_element_class_add_pad_template (gstelement_class, src_template);
186
187   gst_element_class_set_static_metadata (gstelement_class, "WAV audio demuxer",
188       "Codec/Demuxer/Audio",
189       "Parse a .wav file into raw audio",
190       "Erik Walthinsen <omega@cse.ogi.edu>");
191 }
192
193 static void
194 gst_wavparse_reset (GstWavParse * wav)
195 {
196   wav->state = GST_WAVPARSE_START;
197
198   /* These will all be set correctly in the fmt chunk */
199   wav->depth = 0;
200   wav->rate = 0;
201   wav->width = 0;
202   wav->channels = 0;
203   wav->blockalign = 0;
204   wav->bps = 0;
205   wav->fact = 0;
206   wav->offset = 0;
207   wav->end_offset = 0;
208   wav->dataleft = 0;
209   wav->datasize = 0;
210   wav->datastart = 0;
211   wav->duration = 0;
212   wav->got_fmt = FALSE;
213   wav->first = TRUE;
214
215   if (wav->seek_event)
216     gst_event_unref (wav->seek_event);
217   wav->seek_event = NULL;
218   if (wav->adapter) {
219     gst_adapter_clear (wav->adapter);
220     g_object_unref (wav->adapter);
221     wav->adapter = NULL;
222   }
223   if (wav->tags)
224     gst_tag_list_unref (wav->tags);
225   wav->tags = NULL;
226   if (wav->toc)
227     gst_toc_unref (wav->toc);
228   wav->toc = NULL;
229   if (wav->cues)
230     g_list_free_full (wav->cues, g_free);
231   wav->cues = NULL;
232   if (wav->labls)
233     g_list_free_full (wav->labls, g_free);
234   wav->labls = NULL;
235   if (wav->caps)
236     gst_caps_unref (wav->caps);
237   wav->caps = NULL;
238   if (wav->start_segment)
239     gst_event_unref (wav->start_segment);
240   wav->start_segment = NULL;
241 }
242
243 static void
244 gst_wavparse_dispose (GObject * object)
245 {
246   GstWavParse *wav = GST_WAVPARSE (object);
247
248   GST_DEBUG_OBJECT (wav, "WAV: Dispose");
249   gst_wavparse_reset (wav);
250
251   G_OBJECT_CLASS (parent_class)->dispose (object);
252 }
253
254 static void
255 gst_wavparse_init (GstWavParse * wavparse)
256 {
257   gst_wavparse_reset (wavparse);
258
259   /* sink */
260   wavparse->sinkpad =
261       gst_pad_new_from_static_template (&sink_template_factory, "sink");
262   gst_pad_set_activate_function (wavparse->sinkpad,
263       GST_DEBUG_FUNCPTR (gst_wavparse_sink_activate));
264   gst_pad_set_activatemode_function (wavparse->sinkpad,
265       GST_DEBUG_FUNCPTR (gst_wavparse_sink_activate_mode));
266   gst_pad_set_chain_function (wavparse->sinkpad,
267       GST_DEBUG_FUNCPTR (gst_wavparse_chain));
268   gst_pad_set_event_function (wavparse->sinkpad,
269       GST_DEBUG_FUNCPTR (gst_wavparse_sink_event));
270   gst_element_add_pad (GST_ELEMENT_CAST (wavparse), wavparse->sinkpad);
271
272   /* src */
273   wavparse->srcpad =
274       gst_pad_new_from_template (gst_element_class_get_pad_template
275       (GST_ELEMENT_GET_CLASS (wavparse), "src"), "src");
276   gst_pad_use_fixed_caps (wavparse->srcpad);
277   gst_pad_set_query_function (wavparse->srcpad,
278       GST_DEBUG_FUNCPTR (gst_wavparse_pad_query));
279   gst_pad_set_event_function (wavparse->srcpad,
280       GST_DEBUG_FUNCPTR (gst_wavparse_srcpad_event));
281   gst_element_add_pad (GST_ELEMENT_CAST (wavparse), wavparse->srcpad);
282 }
283
284 /* FIXME: why is that not in use? */
285 #if 0
286 static void
287 gst_wavparse_parse_adtl (GstWavParse * wavparse, int len)
288 {
289   guint32 got_bytes;
290   GstByteStream *bs = wavparse->bs;
291   gst_riff_chunk *temp_chunk, chunk;
292   guint8 *tempdata;
293   struct _gst_riff_labl labl, *temp_labl;
294   struct _gst_riff_ltxt ltxt, *temp_ltxt;
295   struct _gst_riff_note note, *temp_note;
296   char *label_name;
297   GstProps *props;
298   GstPropsEntry *entry;
299   GstCaps *new_caps;
300   GList *caps = NULL;
301
302   props = wavparse->metadata->properties;
303
304   while (len > 0) {
305     got_bytes =
306         gst_bytestream_peek_bytes (bs, &tempdata, sizeof (gst_riff_chunk));
307     if (got_bytes != sizeof (gst_riff_chunk)) {
308       return;
309     }
310     temp_chunk = (gst_riff_chunk *) tempdata;
311
312     chunk.id = GUINT32_FROM_LE (temp_chunk->id);
313     chunk.size = GUINT32_FROM_LE (temp_chunk->size);
314
315     if (chunk.size == 0) {
316       gst_bytestream_flush (bs, sizeof (gst_riff_chunk));
317       len -= sizeof (gst_riff_chunk);
318       continue;
319     }
320
321     switch (chunk.id) {
322       case GST_RIFF_adtl_labl:
323         got_bytes =
324             gst_bytestream_peek_bytes (bs, &tempdata,
325             sizeof (struct _gst_riff_labl));
326         if (got_bytes != sizeof (struct _gst_riff_labl)) {
327           return;
328         }
329
330         temp_labl = (struct _gst_riff_labl *) tempdata;
331         labl.id = GUINT32_FROM_LE (temp_labl->id);
332         labl.size = GUINT32_FROM_LE (temp_labl->size);
333         labl.identifier = GUINT32_FROM_LE (temp_labl->identifier);
334
335         gst_bytestream_flush (bs, sizeof (struct _gst_riff_labl));
336         len -= sizeof (struct _gst_riff_labl);
337
338         got_bytes = gst_bytestream_peek_bytes (bs, &tempdata, labl.size - 4);
339         if (got_bytes != labl.size - 4) {
340           return;
341         }
342
343         label_name = (char *) tempdata;
344
345         gst_bytestream_flush (bs, ((labl.size - 4) + 1) & ~1);
346         len -= (((labl.size - 4) + 1) & ~1);
347
348         new_caps = gst_caps_new ("label",
349             "application/x-gst-metadata",
350             gst_props_new ("identifier", G_TYPE_INT (labl.identifier),
351                 "name", G_TYPE_STRING (label_name), NULL));
352
353         if (gst_props_get (props, "labels", &caps, NULL)) {
354           caps = g_list_append (caps, new_caps);
355         } else {
356           caps = g_list_append (NULL, new_caps);
357
358           entry = gst_props_entry_new ("labels", GST_PROPS_GLIST (caps));
359           gst_props_add_entry (props, entry);
360         }
361
362         break;
363
364       case GST_RIFF_adtl_ltxt:
365         got_bytes =
366             gst_bytestream_peek_bytes (bs, &tempdata,
367             sizeof (struct _gst_riff_ltxt));
368         if (got_bytes != sizeof (struct _gst_riff_ltxt)) {
369           return;
370         }
371
372         temp_ltxt = (struct _gst_riff_ltxt *) tempdata;
373         ltxt.id = GUINT32_FROM_LE (temp_ltxt->id);
374         ltxt.size = GUINT32_FROM_LE (temp_ltxt->size);
375         ltxt.identifier = GUINT32_FROM_LE (temp_ltxt->identifier);
376         ltxt.length = GUINT32_FROM_LE (temp_ltxt->length);
377         ltxt.purpose = GUINT32_FROM_LE (temp_ltxt->purpose);
378         ltxt.country = GUINT16_FROM_LE (temp_ltxt->country);
379         ltxt.language = GUINT16_FROM_LE (temp_ltxt->language);
380         ltxt.dialect = GUINT16_FROM_LE (temp_ltxt->dialect);
381         ltxt.codepage = GUINT16_FROM_LE (temp_ltxt->codepage);
382
383         gst_bytestream_flush (bs, sizeof (struct _gst_riff_ltxt));
384         len -= sizeof (struct _gst_riff_ltxt);
385
386         if (ltxt.size - 20 > 0) {
387           got_bytes = gst_bytestream_peek_bytes (bs, &tempdata, ltxt.size - 20);
388           if (got_bytes != ltxt.size - 20) {
389             return;
390           }
391
392           gst_bytestream_flush (bs, ((ltxt.size - 20) + 1) & ~1);
393           len -= (((ltxt.size - 20) + 1) & ~1);
394
395           label_name = (char *) tempdata;
396         } else {
397           label_name = "";
398         }
399
400         new_caps = gst_caps_new ("ltxt",
401             "application/x-gst-metadata",
402             gst_props_new ("identifier", G_TYPE_INT (ltxt.identifier),
403                 "name", G_TYPE_STRING (label_name),
404                 "length", G_TYPE_INT (ltxt.length), NULL));
405
406         if (gst_props_get (props, "ltxts", &caps, NULL)) {
407           caps = g_list_append (caps, new_caps);
408         } else {
409           caps = g_list_append (NULL, new_caps);
410
411           entry = gst_props_entry_new ("ltxts", GST_PROPS_GLIST (caps));
412           gst_props_add_entry (props, entry);
413         }
414
415         break;
416
417       case GST_RIFF_adtl_note:
418         got_bytes =
419             gst_bytestream_peek_bytes (bs, &tempdata,
420             sizeof (struct _gst_riff_note));
421         if (got_bytes != sizeof (struct _gst_riff_note)) {
422           return;
423         }
424
425         temp_note = (struct _gst_riff_note *) tempdata;
426         note.id = GUINT32_FROM_LE (temp_note->id);
427         note.size = GUINT32_FROM_LE (temp_note->size);
428         note.identifier = GUINT32_FROM_LE (temp_note->identifier);
429
430         gst_bytestream_flush (bs, sizeof (struct _gst_riff_note));
431         len -= sizeof (struct _gst_riff_note);
432
433         got_bytes = gst_bytestream_peek_bytes (bs, &tempdata, note.size - 4);
434         if (got_bytes != note.size - 4) {
435           return;
436         }
437
438         gst_bytestream_flush (bs, ((note.size - 4) + 1) & ~1);
439         len -= (((note.size - 4) + 1) & ~1);
440
441         label_name = (char *) tempdata;
442
443         new_caps = gst_caps_new ("note",
444             "application/x-gst-metadata",
445             gst_props_new ("identifier", G_TYPE_INT (note.identifier),
446                 "name", G_TYPE_STRING (label_name), NULL));
447
448         if (gst_props_get (props, "notes", &caps, NULL)) {
449           caps = g_list_append (caps, new_caps);
450         } else {
451           caps = g_list_append (NULL, new_caps);
452
453           entry = gst_props_entry_new ("notes", GST_PROPS_GLIST (caps));
454           gst_props_add_entry (props, entry);
455         }
456
457         break;
458
459       default:
460         g_print ("Unknown chunk: %" GST_FOURCC_FORMAT "\n",
461             GST_FOURCC_ARGS (chunk.id));
462         return;
463     }
464   }
465
466   g_object_notify (G_OBJECT (wavparse), "metadata");
467 }
468
469 static void
470 gst_wavparse_parse_cues (GstWavParse * wavparse, int len)
471 {
472   guint32 got_bytes;
473   GstByteStream *bs = wavparse->bs;
474   struct _gst_riff_cue *temp_cue, cue;
475   struct _gst_riff_cuepoints *points;
476   guint8 *tempdata;
477   int i;
478   GList *cues = NULL;
479   GstPropsEntry *entry;
480
481   while (len > 0) {
482     int required;
483
484     got_bytes =
485         gst_bytestream_peek_bytes (bs, &tempdata,
486         sizeof (struct _gst_riff_cue));
487     temp_cue = (struct _gst_riff_cue *) tempdata;
488
489     /* fixup for our big endian friends */
490     cue.id = GUINT32_FROM_LE (temp_cue->id);
491     cue.size = GUINT32_FROM_LE (temp_cue->size);
492     cue.cuepoints = GUINT32_FROM_LE (temp_cue->cuepoints);
493
494     gst_bytestream_flush (bs, sizeof (struct _gst_riff_cue));
495     if (got_bytes != sizeof (struct _gst_riff_cue)) {
496       return;
497     }
498
499     len -= sizeof (struct _gst_riff_cue);
500
501     /* -4 because cue.size contains the cuepoints size
502        and we've already flushed that out of the system */
503     required = cue.size - 4;
504     got_bytes = gst_bytestream_peek_bytes (bs, &tempdata, required);
505     gst_bytestream_flush (bs, ((required) + 1) & ~1);
506     if (got_bytes != required) {
507       return;
508     }
509
510     len -= (((cue.size - 4) + 1) & ~1);
511
512     /* now we have an array of struct _gst_riff_cuepoints in tempdata */
513     points = (struct _gst_riff_cuepoints *) tempdata;
514
515     for (i = 0; i < cue.cuepoints; i++) {
516       GstCaps *caps;
517
518       caps = gst_caps_new ("cues",
519           "application/x-gst-metadata",
520           gst_props_new ("identifier", G_TYPE_INT (points[i].identifier),
521               "position", G_TYPE_INT (points[i].offset), NULL));
522       cues = g_list_append (cues, caps);
523     }
524
525     entry = gst_props_entry_new ("cues", GST_PROPS_GLIST (cues));
526     gst_props_add_entry (wavparse->metadata->properties, entry);
527   }
528
529   g_object_notify (G_OBJECT (wavparse), "metadata");
530 }
531
532 /* Read 'fmt ' header */
533 static gboolean
534 gst_wavparse_fmt (GstWavParse * wav)
535 {
536   gst_riff_strf_auds *header = NULL;
537   GstCaps *caps;
538
539   if (!gst_riff_read_strf_auds (wav, &header))
540     goto no_fmt;
541
542   wav->format = header->format;
543   wav->rate = header->rate;
544   wav->channels = header->channels;
545   if (wav->channels == 0)
546     goto no_channels;
547
548   wav->blockalign = header->blockalign;
549   wav->width = (header->blockalign * 8) / header->channels;
550   wav->depth = header->size;
551   wav->bps = header->av_bps;
552   if (wav->bps <= 0)
553     goto no_bps;
554
555   /* Note: gst_riff_create_audio_caps might need to fix values in
556    * the header header depending on the format, so call it first */
557   /* FIXME: Need to handle the channel reorder map */
558   caps = gst_riff_create_audio_caps (header->format, NULL, header, NULL, NULL);
559   g_free (header);
560
561   if (caps == NULL)
562     goto no_caps;
563
564   gst_wavparse_create_sourcepad (wav);
565   gst_pad_use_fixed_caps (wav->srcpad);
566   gst_pad_set_active (wav->srcpad, TRUE);
567   gst_pad_set_caps (wav->srcpad, caps);
568   gst_caps_free (caps);
569   gst_element_add_pad (GST_ELEMENT_CAST (wav), wav->srcpad);
570   gst_element_no_more_pads (GST_ELEMENT_CAST (wav));
571
572   GST_DEBUG ("frequency %u, channels %u", wav->rate, wav->channels);
573
574   return TRUE;
575
576   /* ERRORS */
577 no_fmt:
578   {
579     GST_ELEMENT_ERROR (wav, STREAM, TYPE_NOT_FOUND, (NULL),
580         ("No FMT tag found"));
581     return FALSE;
582   }
583 no_channels:
584   {
585     GST_ELEMENT_ERROR (wav, STREAM, FAILED, (NULL),
586         ("Stream claims to contain zero channels - invalid data"));
587     g_free (header);
588     return FALSE;
589   }
590 no_bps:
591   {
592     GST_ELEMENT_ERROR (wav, STREAM, FAILED, (NULL),
593         ("Stream claims to bitrate of <= zero - invalid data"));
594     g_free (header);
595     return FALSE;
596   }
597 no_caps:
598   {
599     GST_ELEMENT_ERROR (wav, STREAM, TYPE_NOT_FOUND, (NULL), (NULL));
600     return FALSE;
601   }
602 }
603
604 static gboolean
605 gst_wavparse_other (GstWavParse * wav)
606 {
607   guint32 tag, length;
608
609   if (!gst_riff_peek_head (wav, &tag, &length, NULL)) {
610     GST_WARNING_OBJECT (wav, "could not peek head");
611     return FALSE;
612   }
613   GST_DEBUG_OBJECT (wav, "got tag (%08x) %4.4s, length %u", tag,
614       (const gchar *) &tag, length);
615
616   switch (tag) {
617     case GST_RIFF_TAG_LIST:
618       if (!(tag = gst_riff_peek_list (wav))) {
619         GST_WARNING_OBJECT (wav, "could not peek list");
620         return FALSE;
621       }
622
623       switch (tag) {
624         case GST_RIFF_LIST_INFO:
625           if (!gst_riff_read_list (wav, &tag) || !gst_riff_read_info (wav)) {
626             GST_WARNING_OBJECT (wav, "could not read list");
627             return FALSE;
628           }
629           break;
630
631         case GST_RIFF_LIST_adtl:
632           if (!gst_riff_read_skip (wav)) {
633             GST_WARNING_OBJECT (wav, "could not read skip");
634             return FALSE;
635           }
636           break;
637
638         default:
639           GST_DEBUG_OBJECT (wav, "skipping tag (%08x) %4.4s", tag,
640               (gchar *) & tag);
641           if (!gst_riff_read_skip (wav)) {
642             GST_WARNING_OBJECT (wav, "could not read skip");
643             return FALSE;
644           }
645           break;
646       }
647
648       break;
649
650     case GST_RIFF_TAG_data:
651       if (!gst_bytestream_flush (wav->bs, 8)) {
652         GST_WARNING_OBJECT (wav, "could not flush 8 bytes");
653         return FALSE;
654       }
655
656       GST_DEBUG_OBJECT (wav, "switching to data mode");
657       wav->state = GST_WAVPARSE_DATA;
658       wav->datastart = gst_bytestream_tell (wav->bs);
659       if (length == 0) {
660         guint64 file_length;
661
662         /* length is 0, data probably stretches to the end
663          * of file */
664         GST_DEBUG_OBJECT (wav, "length is 0 trying to find length");
665         /* get length of file */
666         file_length = gst_bytestream_length (wav->bs);
667         if (file_length == -1) {
668           GST_DEBUG_OBJECT (wav,
669               "could not get file length, assuming data to eof");
670           /* could not get length, assuming till eof */
671           length = G_MAXUINT32;
672         }
673         if (file_length > G_MAXUINT32) {
674           GST_DEBUG_OBJECT (wav, "file length %" G_GUINT64_FORMAT
675               ", clipping to 32 bits", file_length);
676           /* could not get length, assuming till eof */
677           length = G_MAXUINT32;
678         } else {
679           GST_DEBUG_OBJECT (wav, "file length %" G_GUINT64_FORMAT
680               ", datalength %u", file_length, length);
681           /* substract offset of datastart from length */
682           length = file_length - wav->datastart;
683           GST_DEBUG_OBJECT (wav, "datalength %u", length);
684         }
685       }
686       wav->datasize = (guint64) length;
687       GST_DEBUG_OBJECT (wav, "datasize = %ld", length)
688           break;
689
690     case GST_RIFF_TAG_cue:
691       if (!gst_riff_read_skip (wav)) {
692         GST_WARNING_OBJECT (wav, "could not read skip");
693         return FALSE;
694       }
695       break;
696
697     default:
698       GST_DEBUG_OBJECT (wav, "skipping tag (%08x) %4.4s", tag, (gchar *) & tag);
699       if (!gst_riff_read_skip (wav))
700         return FALSE;
701       break;
702   }
703
704   return TRUE;
705 }
706 #endif
707
708
709 static gboolean
710 gst_wavparse_parse_file_header (GstElement * element, GstBuffer * buf)
711 {
712   guint32 doctype;
713
714   if (!gst_riff_parse_file_header (element, buf, &doctype))
715     return FALSE;
716
717   if (doctype != GST_RIFF_RIFF_WAVE)
718     goto not_wav;
719
720   return TRUE;
721
722   /* ERRORS */
723 not_wav:
724   {
725     GST_ELEMENT_ERROR (element, STREAM, WRONG_TYPE, (NULL),
726         ("File is not a WAVE file: %" GST_FOURCC_FORMAT,
727             GST_FOURCC_ARGS (doctype)));
728     return FALSE;
729   }
730 }
731
732 static GstFlowReturn
733 gst_wavparse_stream_init (GstWavParse * wav)
734 {
735   GstFlowReturn res;
736   GstBuffer *buf = NULL;
737
738   if ((res = gst_pad_pull_range (wav->sinkpad,
739               wav->offset, 12, &buf)) != GST_FLOW_OK)
740     return res;
741   else if (!gst_wavparse_parse_file_header (GST_ELEMENT_CAST (wav), buf))
742     return GST_FLOW_ERROR;
743
744   wav->offset += 12;
745
746   return GST_FLOW_OK;
747 }
748
749 static gboolean
750 gst_wavparse_time_to_bytepos (GstWavParse * wav, gint64 ts, gint64 * bytepos)
751 {
752   /* -1 always maps to -1 */
753   if (ts == -1) {
754     *bytepos = -1;
755     return TRUE;
756   }
757
758   /* 0 always maps to 0 */
759   if (ts == 0) {
760     *bytepos = 0;
761     return TRUE;
762   }
763
764   if (wav->bps > 0) {
765     *bytepos = gst_util_uint64_scale_ceil (ts, (guint64) wav->bps, GST_SECOND);
766     return TRUE;
767   } else if (wav->fact) {
768     guint64 bps =
769         gst_util_uint64_scale_int (wav->datasize, wav->rate, wav->fact);
770     *bytepos = gst_util_uint64_scale_ceil (ts, bps, GST_SECOND);
771     return TRUE;
772   }
773
774   return FALSE;
775 }
776
777 /* This function is used to perform seeks on the element.
778  *
779  * It also works when event is NULL, in which case it will just
780  * start from the last configured segment. This technique is
781  * used when activating the element and to perform the seek in
782  * READY.
783  */
784 static gboolean
785 gst_wavparse_perform_seek (GstWavParse * wav, GstEvent * event)
786 {
787   gboolean res;
788   gdouble rate;
789   GstFormat format, bformat;
790   GstSeekFlags flags;
791   GstSeekType cur_type = GST_SEEK_TYPE_NONE, stop_type;
792   gint64 cur, stop, upstream_size;
793   gboolean flush;
794   gboolean update;
795   GstSegment seeksegment = { 0, };
796   gint64 last_stop;
797
798   if (event) {
799     GST_DEBUG_OBJECT (wav, "doing seek with event");
800
801     gst_event_parse_seek (event, &rate, &format, &flags,
802         &cur_type, &cur, &stop_type, &stop);
803
804     /* no negative rates yet */
805     if (rate < 0.0)
806       goto negative_rate;
807
808     if (format != wav->segment.format) {
809       GST_INFO_OBJECT (wav, "converting seek-event from %s to %s",
810           gst_format_get_name (format),
811           gst_format_get_name (wav->segment.format));
812       res = TRUE;
813       if (cur_type != GST_SEEK_TYPE_NONE)
814         res =
815             gst_pad_query_convert (wav->srcpad, format, cur,
816             wav->segment.format, &cur);
817       if (res && stop_type != GST_SEEK_TYPE_NONE)
818         res =
819             gst_pad_query_convert (wav->srcpad, format, stop,
820             wav->segment.format, &stop);
821       if (!res)
822         goto no_format;
823
824       format = wav->segment.format;
825     }
826   } else {
827     GST_DEBUG_OBJECT (wav, "doing seek without event");
828     flags = 0;
829     rate = 1.0;
830     cur_type = GST_SEEK_TYPE_SET;
831     stop_type = GST_SEEK_TYPE_SET;
832   }
833
834   /* in push mode, we must delegate to upstream */
835   if (wav->streaming) {
836     gboolean res = FALSE;
837
838     /* if streaming not yet started; only prepare initial newsegment */
839     if (!event || wav->state != GST_WAVPARSE_DATA) {
840       if (wav->start_segment)
841         gst_event_unref (wav->start_segment);
842       wav->start_segment = gst_event_new_segment (&wav->segment);
843       res = TRUE;
844     } else {
845       /* convert seek positions to byte positions in data sections */
846       if (format == GST_FORMAT_TIME) {
847         /* should not fail */
848         if (!gst_wavparse_time_to_bytepos (wav, cur, &cur))
849           goto no_position;
850         if (!gst_wavparse_time_to_bytepos (wav, stop, &stop))
851           goto no_position;
852       }
853       /* mind sample boundary and header */
854       if (cur >= 0) {
855         cur -= (cur % wav->bytes_per_sample);
856         cur += wav->datastart;
857       }
858       if (stop >= 0) {
859         stop -= (stop % wav->bytes_per_sample);
860         stop += wav->datastart;
861       }
862       GST_DEBUG_OBJECT (wav, "Pushing BYTE seek rate %g, "
863           "start %" G_GINT64_FORMAT ", stop %" G_GINT64_FORMAT, rate, cur,
864           stop);
865       /* BYTE seek event */
866       event = gst_event_new_seek (rate, GST_FORMAT_BYTES, flags, cur_type, cur,
867           stop_type, stop);
868       res = gst_pad_push_event (wav->sinkpad, event);
869     }
870     return res;
871   }
872
873   /* get flush flag */
874   flush = flags & GST_SEEK_FLAG_FLUSH;
875
876   /* now we need to make sure the streaming thread is stopped. We do this by
877    * either sending a FLUSH_START event downstream which will cause the
878    * streaming thread to stop with a WRONG_STATE.
879    * For a non-flushing seek we simply pause the task, which will happen as soon
880    * as it completes one iteration (and thus might block when the sink is
881    * blocking in preroll). */
882   if (flush) {
883     GST_DEBUG_OBJECT (wav, "sending flush start");
884     gst_pad_push_event (wav->srcpad, gst_event_new_flush_start ());
885   } else {
886     gst_pad_pause_task (wav->sinkpad);
887   }
888
889   /* we should now be able to grab the streaming thread because we stopped it
890    * with the above flush/pause code */
891   GST_PAD_STREAM_LOCK (wav->sinkpad);
892
893   /* save current position */
894   last_stop = wav->segment.position;
895
896   GST_DEBUG_OBJECT (wav, "stopped streaming at %" G_GINT64_FORMAT, last_stop);
897
898   /* copy segment, we need this because we still need the old
899    * segment when we close the current segment. */
900   memcpy (&seeksegment, &wav->segment, sizeof (GstSegment));
901
902   /* configure the seek parameters in the seeksegment. We will then have the
903    * right values in the segment to perform the seek */
904   if (event) {
905     GST_DEBUG_OBJECT (wav, "configuring seek");
906     gst_segment_do_seek (&seeksegment, rate, format, flags,
907         cur_type, cur, stop_type, stop, &update);
908   }
909
910   /* figure out the last position we need to play. If it's configured (stop !=
911    * -1), use that, else we play until the total duration of the file */
912   if ((stop = seeksegment.stop) == -1)
913     stop = seeksegment.duration;
914
915   GST_DEBUG_OBJECT (wav, "cur_type =%d", cur_type);
916   if ((cur_type != GST_SEEK_TYPE_NONE)) {
917     /* bring offset to bytes, if the bps is 0, we have the segment in BYTES and
918      * we can just copy the last_stop. If not, we use the bps to convert TIME to
919      * bytes. */
920     if (!gst_wavparse_time_to_bytepos (wav, seeksegment.position,
921             (gint64 *) & wav->offset))
922       wav->offset = seeksegment.position;
923     GST_LOG_OBJECT (wav, "offset=%" G_GUINT64_FORMAT, wav->offset);
924     wav->offset -= (wav->offset % wav->bytes_per_sample);
925     GST_LOG_OBJECT (wav, "offset=%" G_GUINT64_FORMAT, wav->offset);
926     wav->offset += wav->datastart;
927     GST_LOG_OBJECT (wav, "offset=%" G_GUINT64_FORMAT, wav->offset);
928   } else {
929     GST_LOG_OBJECT (wav, "continue from offset=%" G_GUINT64_FORMAT,
930         wav->offset);
931   }
932
933   if (stop_type != GST_SEEK_TYPE_NONE) {
934     if (!gst_wavparse_time_to_bytepos (wav, stop, (gint64 *) & wav->end_offset))
935       wav->end_offset = stop;
936     GST_LOG_OBJECT (wav, "end_offset=%" G_GUINT64_FORMAT, wav->end_offset);
937     wav->end_offset -= (wav->end_offset % wav->bytes_per_sample);
938     GST_LOG_OBJECT (wav, "end_offset=%" G_GUINT64_FORMAT, wav->end_offset);
939     wav->end_offset += wav->datastart;
940     GST_LOG_OBJECT (wav, "end_offset=%" G_GUINT64_FORMAT, wav->end_offset);
941   } else {
942     GST_LOG_OBJECT (wav, "continue to end_offset=%" G_GUINT64_FORMAT,
943         wav->end_offset);
944   }
945
946   /* make sure filesize is not exceeded due to rounding errors or so,
947    * same precaution as in _stream_headers */
948   bformat = GST_FORMAT_BYTES;
949   if (gst_pad_peer_query_duration (wav->sinkpad, bformat, &upstream_size))
950     wav->end_offset = MIN (wav->end_offset, upstream_size);
951
952   /* this is the range of bytes we will use for playback */
953   wav->offset = MIN (wav->offset, wav->end_offset);
954   wav->dataleft = wav->end_offset - wav->offset;
955
956   GST_DEBUG_OBJECT (wav,
957       "seek: rate %lf, offset %" G_GUINT64_FORMAT ", end %" G_GUINT64_FORMAT
958       ", segment %" GST_TIME_FORMAT " -- %" GST_TIME_FORMAT, rate, wav->offset,
959       wav->end_offset, GST_TIME_ARGS (seeksegment.start), GST_TIME_ARGS (stop));
960
961   /* prepare for streaming again */
962   if (flush) {
963     /* if we sent a FLUSH_START, we now send a FLUSH_STOP */
964     GST_DEBUG_OBJECT (wav, "sending flush stop");
965     gst_pad_push_event (wav->srcpad, gst_event_new_flush_stop (TRUE));
966   }
967
968   /* now we did the seek and can activate the new segment values */
969   memcpy (&wav->segment, &seeksegment, sizeof (GstSegment));
970
971   /* if we're doing a segment seek, post a SEGMENT_START message */
972   if (wav->segment.flags & GST_SEEK_FLAG_SEGMENT) {
973     gst_element_post_message (GST_ELEMENT_CAST (wav),
974         gst_message_new_segment_start (GST_OBJECT_CAST (wav),
975             wav->segment.format, wav->segment.position));
976   }
977
978   /* now create the newsegment */
979   GST_DEBUG_OBJECT (wav, "Creating newsegment from %" G_GINT64_FORMAT
980       " to %" G_GINT64_FORMAT, wav->segment.position, stop);
981
982   /* store the newsegment event so it can be sent from the streaming thread. */
983   if (wav->start_segment)
984     gst_event_unref (wav->start_segment);
985   wav->start_segment = gst_event_new_segment (&wav->segment);
986
987   /* mark discont if we are going to stream from another position. */
988   if (last_stop != wav->segment.position) {
989     GST_DEBUG_OBJECT (wav, "mark DISCONT, we did a seek to another position");
990     wav->discont = TRUE;
991   }
992
993   /* and start the streaming task again */
994   if (!wav->streaming) {
995     gst_pad_start_task (wav->sinkpad, (GstTaskFunction) gst_wavparse_loop,
996         wav->sinkpad, NULL);
997   }
998
999   GST_PAD_STREAM_UNLOCK (wav->sinkpad);
1000
1001   return TRUE;
1002
1003   /* ERRORS */
1004 negative_rate:
1005   {
1006     GST_DEBUG_OBJECT (wav, "negative playback rates are not supported yet.");
1007     return FALSE;
1008   }
1009 no_format:
1010   {
1011     GST_DEBUG_OBJECT (wav, "unsupported format given, seek aborted.");
1012     return FALSE;
1013   }
1014 no_position:
1015   {
1016     GST_DEBUG_OBJECT (wav,
1017         "Could not determine byte position for desired time");
1018     return FALSE;
1019   }
1020 }
1021
1022 /*
1023  * gst_wavparse_peek_chunk_info:
1024  * @wav Wavparse object
1025  * @tag holder for tag
1026  * @size holder for tag size
1027  *
1028  * Peek next chunk info (tag and size)
1029  *
1030  * Returns: %TRUE when the chunk info (header) is available
1031  */
1032 static gboolean
1033 gst_wavparse_peek_chunk_info (GstWavParse * wav, guint32 * tag, guint32 * size)
1034 {
1035   const guint8 *data = NULL;
1036
1037   if (gst_adapter_available (wav->adapter) < 8)
1038     return FALSE;
1039
1040   data = gst_adapter_map (wav->adapter, 8);
1041   *tag = GST_READ_UINT32_LE (data);
1042   *size = GST_READ_UINT32_LE (data + 4);
1043   gst_adapter_unmap (wav->adapter);
1044
1045   GST_DEBUG ("Next chunk size is %u bytes, type %" GST_FOURCC_FORMAT, *size,
1046       GST_FOURCC_ARGS (*tag));
1047
1048   return TRUE;
1049 }
1050
1051 /*
1052  * gst_wavparse_peek_chunk:
1053  * @wav Wavparse object
1054  * @tag holder for tag
1055  * @size holder for tag size
1056  *
1057  * Peek enough data for one full chunk
1058  *
1059  * Returns: %TRUE when the full chunk is available
1060  */
1061 static gboolean
1062 gst_wavparse_peek_chunk (GstWavParse * wav, guint32 * tag, guint32 * size)
1063 {
1064   guint32 peek_size = 0;
1065   guint available;
1066
1067   if (!gst_wavparse_peek_chunk_info (wav, tag, size))
1068     return FALSE;
1069
1070   /* size 0 -> empty data buffer would surprise most callers,
1071    * large size -> do not bother trying to squeeze that into adapter,
1072    * so we throw poor man's exception, which can be caught if caller really
1073    * wants to handle 0 size chunk */
1074   if (!(*size) || (*size) >= (1 << 30)) {
1075     GST_INFO ("Invalid/unexpected chunk size %u for tag %" GST_FOURCC_FORMAT,
1076         *size, GST_FOURCC_ARGS (*tag));
1077     /* chain should give up */
1078     wav->abort_buffering = TRUE;
1079     return FALSE;
1080   }
1081   peek_size = (*size + 1) & ~1;
1082   available = gst_adapter_available (wav->adapter);
1083
1084   if (available >= (8 + peek_size)) {
1085     return TRUE;
1086   } else {
1087     GST_LOG ("but only %u bytes available now", available);
1088     return FALSE;
1089   }
1090 }
1091
1092 /*
1093  * gst_wavparse_calculate_duration:
1094  * @wav: wavparse object
1095  *
1096  * Calculate duration on demand and store in @wav. Prefer bps, but use fact as a
1097  * fallback.
1098  *
1099  * Returns: %TRUE if duration is available.
1100  */
1101 static gboolean
1102 gst_wavparse_calculate_duration (GstWavParse * wav)
1103 {
1104   if (wav->duration > 0)
1105     return TRUE;
1106
1107   if (wav->bps > 0) {
1108     GST_INFO_OBJECT (wav, "Got datasize %" G_GUINT64_FORMAT, wav->datasize);
1109     wav->duration =
1110         gst_util_uint64_scale_ceil (wav->datasize, GST_SECOND,
1111         (guint64) wav->bps);
1112     GST_INFO_OBJECT (wav, "Got duration (bps) %" GST_TIME_FORMAT,
1113         GST_TIME_ARGS (wav->duration));
1114     return TRUE;
1115   } else if (wav->fact) {
1116     wav->duration =
1117         gst_util_uint64_scale_int_ceil (GST_SECOND, wav->fact, wav->rate);
1118     GST_INFO_OBJECT (wav, "Got duration (fact) %" GST_TIME_FORMAT,
1119         GST_TIME_ARGS (wav->duration));
1120     return TRUE;
1121   }
1122   return FALSE;
1123 }
1124
1125 static gboolean
1126 gst_waveparse_ignore_chunk (GstWavParse * wav, GstBuffer * buf, guint32 tag,
1127     guint32 size)
1128 {
1129   guint flush;
1130
1131   if (wav->streaming) {
1132     if (!gst_wavparse_peek_chunk (wav, &tag, &size))
1133       return FALSE;
1134   }
1135   GST_DEBUG_OBJECT (wav, "Ignoring tag %" GST_FOURCC_FORMAT,
1136       GST_FOURCC_ARGS (tag));
1137   flush = 8 + ((size + 1) & ~1);
1138   wav->offset += flush;
1139   if (wav->streaming) {
1140     gst_adapter_flush (wav->adapter, flush);
1141   } else {
1142     gst_buffer_unref (buf);
1143   }
1144
1145   return TRUE;
1146 }
1147
1148 /*
1149  * gst_wavparse_cue_chunk:
1150  * @wav GstWavParse object
1151  * @data holder for data
1152  * @size holder for data size
1153  *
1154  * Parse cue chunk from @data to wav->cues.
1155  *
1156  * Returns: %TRUE when cue chunk is available
1157  */
1158 static gboolean
1159 gst_wavparse_cue_chunk (GstWavParse * wav, const guint8 * data, guint32 size)
1160 {
1161   guint32 i, ncues;
1162   GList *cues = NULL;
1163   GstWavParseCue *cue;
1164
1165   if (wav->cues) {
1166     GST_WARNING_OBJECT (wav, "found another cue's");
1167     return TRUE;
1168   }
1169
1170   ncues = GST_READ_UINT32_LE (data);
1171
1172   if (size < 4 + ncues * 24) {
1173     GST_WARNING_OBJECT (wav, "broken file %d %d", size, ncues);
1174     return FALSE;
1175   }
1176
1177   /* parse data */
1178   data += 4;
1179   for (i = 0; i < ncues; i++) {
1180     cue = g_new0 (GstWavParseCue, 1);
1181     cue->id = GST_READ_UINT32_LE (data);
1182     cue->position = GST_READ_UINT32_LE (data + 4);
1183     cue->data_chunk_id = GST_READ_UINT32_LE (data + 8);
1184     cue->chunk_start = GST_READ_UINT32_LE (data + 12);
1185     cue->block_start = GST_READ_UINT32_LE (data + 16);
1186     cue->sample_offset = GST_READ_UINT32_LE (data + 20);
1187     cues = g_list_append (cues, cue);
1188     data += 24;
1189   }
1190
1191   wav->cues = cues;
1192
1193   return TRUE;
1194 }
1195
1196 /*
1197  * gst_wavparse_labl_chunk:
1198  * @wav GstWavParse object
1199  * @data holder for data
1200  * @size holder for data size
1201  *
1202  * Parse labl from @data to wav->labls.
1203  *
1204  * Returns: %TRUE when labl chunk is available
1205  */
1206 static gboolean
1207 gst_wavparse_labl_chunk (GstWavParse * wav, const guint8 * data, guint32 size)
1208 {
1209   GstWavParseLabl *labl;
1210
1211   if (size < 5)
1212     return FALSE;
1213
1214   labl = g_new0 (GstWavParseLabl, 1);
1215
1216   /* parse data */
1217   data += 8;
1218   labl->cue_point_id = GST_READ_UINT32_LE (data);
1219   labl->text = g_memdup (data + 4, size - 4);
1220
1221   wav->labls = g_list_append (wav->labls, labl);
1222
1223   return TRUE;
1224 }
1225
1226 /*
1227  * gst_wavparse_note_chunk:
1228  * @wav GstWavParse object
1229  * @data holder for data
1230  * @size holder for data size
1231  *
1232  * Parse note from @data to wav->notes.
1233  *
1234  * Returns: %TRUE when note chunk is available
1235  */
1236 static gboolean
1237 gst_wavparse_note_chunk (GstWavParse * wav, const guint8 * data, guint32 size)
1238 {
1239   GstWavParseNote *note;
1240
1241   if (size < 5)
1242     return FALSE;
1243
1244   note = g_new0 (GstWavParseNote, 1);
1245
1246   /* parse data */
1247   data += 8;
1248   note->cue_point_id = GST_READ_UINT32_LE (data);
1249   note->text = g_memdup (data + 4, size - 4);
1250
1251   wav->notes = g_list_append (wav->notes, note);
1252
1253   return TRUE;
1254 }
1255
1256 /*
1257  * gst_wavparse_adtl_chunk:
1258  * @wav GstWavParse object
1259  * @data holder for data
1260  * @size holder for data size
1261  *
1262  * Parse adtl from @data.
1263  *
1264  * Returns: %TRUE when adtl chunk is available
1265  */
1266 static gboolean
1267 gst_wavparse_adtl_chunk (GstWavParse * wav, const guint8 * data, guint32 size)
1268 {
1269   guint32 ltag, lsize, offset = 0;
1270
1271   while (size >= 8) {
1272     ltag = GST_READ_UINT32_LE (data + offset);
1273     lsize = GST_READ_UINT32_LE (data + offset + 4);
1274     switch (ltag) {
1275       case GST_RIFF_TAG_labl:
1276         gst_wavparse_labl_chunk (wav, data + offset, size);
1277         break;
1278       case GST_RIFF_TAG_note:
1279         gst_wavparse_note_chunk (wav, data + offset, size);
1280         break;
1281       default:
1282         break;
1283     }
1284     offset += 8 + GST_ROUND_UP_2 (lsize);
1285     size -= 8 + GST_ROUND_UP_2 (lsize);
1286   }
1287
1288   return TRUE;
1289 }
1290
1291 static GstTagList *
1292 gst_wavparse_get_tags_toc_entry (GstToc * toc, gchar * id)
1293 {
1294   GstTagList *tags = NULL;
1295   GstTocEntry *entry = NULL;
1296
1297   entry = gst_toc_find_entry (toc, id);
1298   if (entry != NULL) {
1299     tags = gst_toc_entry_get_tags (entry);
1300     if (tags == NULL) {
1301       tags = gst_tag_list_new_empty ();
1302       gst_toc_entry_set_tags (entry, tags);
1303     }
1304   }
1305
1306   return tags;
1307 }
1308
1309 /*
1310  * gst_wavparse_create_toc:
1311  * @wav GstWavParse object
1312  *
1313  * Create TOC from wav->cues and wav->labls.
1314  */
1315 static gboolean
1316 gst_wavparse_create_toc (GstWavParse * wav)
1317 {
1318   gint64 start, stop;
1319   gchar *id;
1320   GList *list;
1321   GstWavParseCue *cue;
1322   GstWavParseLabl *labl;
1323   GstWavParseNote *note;
1324   GstTagList *tags;
1325   GstToc *toc;
1326   GstTocEntry *entry = NULL, *cur_subentry = NULL, *prev_subentry = NULL;
1327
1328   GST_OBJECT_LOCK (wav);
1329   if (wav->toc) {
1330     GST_OBJECT_UNLOCK (wav);
1331     GST_WARNING_OBJECT (wav, "found another TOC");
1332     return FALSE;
1333   }
1334
1335   if (!wav->cues) {
1336     GST_OBJECT_UNLOCK (wav);
1337     return TRUE;
1338   }
1339
1340   /* FIXME: send CURRENT scope toc too */
1341   toc = gst_toc_new (GST_TOC_SCOPE_GLOBAL);
1342
1343   /* add cue edition */
1344   entry = gst_toc_entry_new (GST_TOC_ENTRY_TYPE_EDITION, "cue");
1345   gst_toc_entry_set_start_stop_times (entry, 0, wav->duration);
1346   gst_toc_append_entry (toc, entry);
1347
1348   /* add tracks in cue edition */
1349   list = wav->cues;
1350   while (list) {
1351     cue = list->data;
1352     prev_subentry = cur_subentry;
1353     /* previous track stop time = current track start time */
1354     if (prev_subentry != NULL) {
1355       gst_toc_entry_get_start_stop_times (prev_subentry, &start, NULL);
1356       stop = gst_util_uint64_scale_round (cue->position, GST_SECOND, wav->rate);
1357       gst_toc_entry_set_start_stop_times (prev_subentry, start, stop);
1358     }
1359     id = g_strdup_printf ("%08x", cue->id);
1360     cur_subentry = gst_toc_entry_new (GST_TOC_ENTRY_TYPE_TRACK, id);
1361     g_free (id);
1362     start = gst_util_uint64_scale_round (cue->position, GST_SECOND, wav->rate);
1363     stop = wav->duration;
1364     gst_toc_entry_set_start_stop_times (cur_subentry, start, stop);
1365     gst_toc_entry_append_sub_entry (entry, cur_subentry);
1366     list = g_list_next (list);
1367   }
1368
1369   /* add tags in tracks */
1370   list = wav->labls;
1371   while (list) {
1372     labl = list->data;
1373     id = g_strdup_printf ("%08x", labl->cue_point_id);
1374     tags = gst_wavparse_get_tags_toc_entry (toc, id);
1375     g_free (id);
1376     if (tags != NULL) {
1377       gst_tag_list_add (tags, GST_TAG_MERGE_APPEND, GST_TAG_TITLE, labl->text,
1378           NULL);
1379     }
1380     list = g_list_next (list);
1381   }
1382   list = wav->notes;
1383   while (list) {
1384     note = list->data;
1385     id = g_strdup_printf ("%08x", note->cue_point_id);
1386     tags = gst_wavparse_get_tags_toc_entry (toc, id);
1387     g_free (id);
1388     if (tags != NULL) {
1389       gst_tag_list_add (tags, GST_TAG_MERGE_PREPEND, GST_TAG_COMMENT,
1390           note->text, NULL);
1391     }
1392     list = g_list_next (list);
1393   }
1394
1395   /* send data as TOC */
1396   wav->toc = toc;
1397
1398   /* send TOC event */
1399   if (wav->toc) {
1400     GST_OBJECT_UNLOCK (wav);
1401     gst_pad_push_event (wav->srcpad, gst_event_new_toc (wav->toc, FALSE));
1402   }
1403
1404   return TRUE;
1405 }
1406
1407 #define MAX_BUFFER_SIZE 4096
1408
1409 static GstFlowReturn
1410 gst_wavparse_stream_headers (GstWavParse * wav)
1411 {
1412   GstFlowReturn res = GST_FLOW_OK;
1413   GstBuffer *buf = NULL;
1414   gst_riff_strf_auds *header = NULL;
1415   guint32 tag, size;
1416   gboolean gotdata = FALSE;
1417   GstCaps *caps = NULL;
1418   gchar *codec_name = NULL;
1419   GstEvent **event_p;
1420   gint64 upstream_size = 0;
1421
1422   /* search for "_fmt" chunk, which should be first */
1423   while (!wav->got_fmt) {
1424     GstBuffer *extra;
1425
1426     /* The header starts with a 'fmt ' tag */
1427     if (wav->streaming) {
1428       if (!gst_wavparse_peek_chunk (wav, &tag, &size))
1429         return res;
1430
1431       gst_adapter_flush (wav->adapter, 8);
1432       wav->offset += 8;
1433
1434       if (size) {
1435         buf = gst_adapter_take_buffer (wav->adapter, size);
1436         if (size & 1)
1437           gst_adapter_flush (wav->adapter, 1);
1438         wav->offset += GST_ROUND_UP_2 (size);
1439       } else {
1440         buf = gst_buffer_new ();
1441       }
1442     } else {
1443       if ((res = gst_riff_read_chunk (GST_ELEMENT_CAST (wav), wav->sinkpad,
1444                   &wav->offset, &tag, &buf)) != GST_FLOW_OK)
1445         return res;
1446     }
1447
1448     if (tag == GST_RIFF_TAG_JUNK || tag == GST_RIFF_TAG_JUNQ ||
1449         tag == GST_RIFF_TAG_bext || tag == GST_RIFF_TAG_BEXT ||
1450         tag == GST_RIFF_TAG_LIST || tag == GST_RIFF_TAG_ID32 ||
1451         tag == GST_RIFF_TAG_IDVX) {
1452       GST_DEBUG_OBJECT (wav, "skipping %" GST_FOURCC_FORMAT " chunk",
1453           GST_FOURCC_ARGS (tag));
1454       gst_buffer_unref (buf);
1455       buf = NULL;
1456       continue;
1457     }
1458
1459     if (tag != GST_RIFF_TAG_fmt)
1460       goto invalid_wav;
1461
1462     if (!(gst_riff_parse_strf_auds (GST_ELEMENT_CAST (wav), buf, &header,
1463                 &extra)))
1464       goto parse_header_error;
1465
1466     buf = NULL;                 /* parse_strf_auds() took ownership of buffer */
1467
1468     /* do sanity checks of header fields */
1469     if (header->channels == 0)
1470       goto no_channels;
1471     if (header->rate == 0)
1472       goto no_rate;
1473
1474     GST_DEBUG_OBJECT (wav, "creating the caps");
1475
1476     /* Note: gst_riff_create_audio_caps might need to fix values in
1477      * the header header depending on the format, so call it first */
1478     /* FIXME: Need to handle the channel reorder map */
1479     caps = gst_riff_create_audio_caps (header->format, NULL, header, extra,
1480         NULL, &codec_name, NULL);
1481
1482     if (extra)
1483       gst_buffer_unref (extra);
1484
1485     if (!caps)
1486       goto unknown_format;
1487
1488     /* do more sanity checks of header fields
1489      * (these can be sanitized by gst_riff_create_audio_caps()
1490      */
1491     wav->format = header->format;
1492     wav->rate = header->rate;
1493     wav->channels = header->channels;
1494     wav->blockalign = header->blockalign;
1495     wav->depth = header->bits_per_sample;
1496     wav->av_bps = header->av_bps;
1497     wav->vbr = FALSE;
1498
1499     g_free (header);
1500     header = NULL;
1501
1502     /* do format specific handling */
1503     switch (wav->format) {
1504       case GST_RIFF_WAVE_FORMAT_MPEGL12:
1505       case GST_RIFF_WAVE_FORMAT_MPEGL3:
1506       {
1507         /* Note: workaround for mp2/mp3 embedded in wav, that relies on the
1508          * bitrate inside the mpeg stream */
1509         GST_INFO ("resetting bps from %u to 0 for mp2/3", wav->av_bps);
1510         wav->bps = 0;
1511         break;
1512       }
1513       case GST_RIFF_WAVE_FORMAT_PCM:
1514         if (wav->blockalign > wav->channels * ((wav->depth + 7) / 8))
1515           goto invalid_blockalign;
1516         /* fall through */
1517       default:
1518         if (wav->av_bps > wav->blockalign * wav->rate)
1519           goto invalid_bps;
1520         /* use the configured bps */
1521         wav->bps = wav->av_bps;
1522         break;
1523     }
1524
1525     wav->width = (wav->blockalign * 8) / wav->channels;
1526     wav->bytes_per_sample = wav->channels * wav->width / 8;
1527
1528     if (wav->bytes_per_sample <= 0)
1529       goto no_bytes_per_sample;
1530
1531     GST_DEBUG_OBJECT (wav, "blockalign = %u", (guint) wav->blockalign);
1532     GST_DEBUG_OBJECT (wav, "width      = %u", (guint) wav->width);
1533     GST_DEBUG_OBJECT (wav, "depth      = %u", (guint) wav->depth);
1534     GST_DEBUG_OBJECT (wav, "av_bps     = %u", (guint) wav->av_bps);
1535     GST_DEBUG_OBJECT (wav, "frequency  = %u", (guint) wav->rate);
1536     GST_DEBUG_OBJECT (wav, "channels   = %u", (guint) wav->channels);
1537     GST_DEBUG_OBJECT (wav, "bytes_per_sample = %u", wav->bytes_per_sample);
1538
1539     /* bps can be 0 when we don't have a valid bitrate (mostly for compressed
1540      * formats). This will make the element output a BYTE format segment and
1541      * will not timestamp the outgoing buffers.
1542      */
1543     GST_DEBUG_OBJECT (wav, "bps        = %u", (guint) wav->bps);
1544
1545     GST_DEBUG_OBJECT (wav, "caps = %" GST_PTR_FORMAT, caps);
1546
1547     /* create pad later so we can sniff the first few bytes
1548      * of the real data and correct our caps if necessary */
1549     gst_caps_replace (&wav->caps, caps);
1550     gst_caps_replace (&caps, NULL);
1551
1552     wav->got_fmt = TRUE;
1553
1554     if (codec_name) {
1555       wav->tags = gst_tag_list_new_empty ();
1556
1557       gst_tag_list_add (wav->tags, GST_TAG_MERGE_REPLACE,
1558           GST_TAG_AUDIO_CODEC, codec_name, NULL);
1559
1560       g_free (codec_name);
1561       codec_name = NULL;
1562     }
1563
1564   }
1565
1566   gst_pad_peer_query_duration (wav->sinkpad, GST_FORMAT_BYTES, &upstream_size);
1567   GST_DEBUG_OBJECT (wav, "upstream size %" G_GUINT64_FORMAT, upstream_size);
1568
1569   /* loop headers until we get data */
1570   while (!gotdata) {
1571     if (wav->streaming) {
1572       if (!gst_wavparse_peek_chunk_info (wav, &tag, &size))
1573         goto exit;
1574     } else {
1575       GstMapInfo map;
1576
1577       buf = NULL;
1578       if ((res =
1579               gst_pad_pull_range (wav->sinkpad, wav->offset, 8,
1580                   &buf)) != GST_FLOW_OK)
1581         goto header_read_error;
1582       gst_buffer_map (buf, &map, GST_MAP_READ);
1583       tag = GST_READ_UINT32_LE (map.data);
1584       size = GST_READ_UINT32_LE (map.data + 4);
1585       gst_buffer_unmap (buf, &map);
1586     }
1587
1588     GST_INFO_OBJECT (wav,
1589         "Got TAG: %" GST_FOURCC_FORMAT ", offset %" G_GUINT64_FORMAT,
1590         GST_FOURCC_ARGS (tag), wav->offset);
1591
1592     /* wav is a st00pid format, we don't know for sure where data starts.
1593      * So we have to go bit by bit until we find the 'data' header
1594      */
1595     switch (tag) {
1596       case GST_RIFF_TAG_data:{
1597         GST_DEBUG_OBJECT (wav, "Got 'data' TAG, size : %u", size);
1598         if (wav->ignore_length) {
1599           GST_DEBUG_OBJECT (wav, "Ignoring length");
1600           size = 0;
1601         }
1602         if (wav->streaming) {
1603           gst_adapter_flush (wav->adapter, 8);
1604           gotdata = TRUE;
1605         } else {
1606           gst_buffer_unref (buf);
1607         }
1608         wav->offset += 8;
1609         wav->datastart = wav->offset;
1610         /* If size is zero, then the data chunk probably actually extends to
1611            the end of the file */
1612         if (size == 0 && upstream_size) {
1613           size = upstream_size - wav->datastart;
1614         }
1615         /* Or the file might be truncated */
1616         else if (upstream_size) {
1617           size = MIN (size, (upstream_size - wav->datastart));
1618         }
1619         wav->datasize = (guint64) size;
1620         wav->dataleft = (guint64) size;
1621         wav->end_offset = size + wav->datastart;
1622         if (!wav->streaming) {
1623           /* We will continue parsing tags 'till end */
1624           wav->offset += size;
1625         }
1626         GST_DEBUG_OBJECT (wav, "datasize = %u", size);
1627         break;
1628       }
1629       case GST_RIFF_TAG_fact:{
1630         if (wav->format != GST_RIFF_WAVE_FORMAT_MPEGL12 &&
1631             wav->format != GST_RIFF_WAVE_FORMAT_MPEGL3) {
1632           const guint data_size = 4;
1633
1634           GST_INFO_OBJECT (wav, "Have fact chunk");
1635           if (size < data_size) {
1636             if (!gst_waveparse_ignore_chunk (wav, buf, tag, size)) {
1637               /* need more data */
1638               goto exit;
1639             }
1640             GST_DEBUG_OBJECT (wav, "need %u, available %u; ignoring chunk",
1641                 data_size, size);
1642             break;
1643           }
1644           /* number of samples (for compressed formats) */
1645           if (wav->streaming) {
1646             const guint8 *data = NULL;
1647
1648             if (!gst_wavparse_peek_chunk (wav, &tag, &size)) {
1649               goto exit;
1650             }
1651             gst_adapter_flush (wav->adapter, 8);
1652             data = gst_adapter_map (wav->adapter, data_size);
1653             wav->fact = GST_READ_UINT32_LE (data);
1654             gst_adapter_unmap (wav->adapter);
1655             gst_adapter_flush (wav->adapter, GST_ROUND_UP_2 (size));
1656           } else {
1657             gst_buffer_unref (buf);
1658             buf = NULL;
1659             if ((res =
1660                     gst_pad_pull_range (wav->sinkpad, wav->offset + 8,
1661                         data_size, &buf)) != GST_FLOW_OK)
1662               goto header_read_error;
1663             gst_buffer_extract (buf, 0, &wav->fact, 4);
1664             wav->fact = GUINT32_FROM_LE (wav->fact);
1665             gst_buffer_unref (buf);
1666           }
1667           GST_DEBUG_OBJECT (wav, "have fact %u", wav->fact);
1668           wav->offset += 8 + GST_ROUND_UP_2 (size);
1669           break;
1670         } else {
1671           if (!gst_waveparse_ignore_chunk (wav, buf, tag, size)) {
1672             /* need more data */
1673             goto exit;
1674           }
1675         }
1676         break;
1677       }
1678       case GST_RIFF_TAG_acid:{
1679         const gst_riff_acid *acid = NULL;
1680         const guint data_size = sizeof (gst_riff_acid);
1681         gfloat tempo;
1682
1683         GST_INFO_OBJECT (wav, "Have acid chunk");
1684         if (size < data_size) {
1685           if (!gst_waveparse_ignore_chunk (wav, buf, tag, size)) {
1686             /* need more data */
1687             goto exit;
1688           }
1689           GST_DEBUG_OBJECT (wav, "need %u, available %u; ignoring chunk",
1690               data_size, size);
1691           break;
1692         }
1693         if (wav->streaming) {
1694           if (!gst_wavparse_peek_chunk (wav, &tag, &size)) {
1695             goto exit;
1696           }
1697           gst_adapter_flush (wav->adapter, 8);
1698           acid = (const gst_riff_acid *) gst_adapter_map (wav->adapter,
1699               data_size);
1700           tempo = acid->tempo;
1701           gst_adapter_unmap (wav->adapter);
1702         } else {
1703           GstMapInfo map;
1704           gst_buffer_unref (buf);
1705           buf = NULL;
1706           if ((res =
1707                   gst_pad_pull_range (wav->sinkpad, wav->offset + 8,
1708                       size, &buf)) != GST_FLOW_OK)
1709             goto header_read_error;
1710           gst_buffer_map (buf, &map, GST_MAP_READ);
1711           acid = (const gst_riff_acid *) map.data;
1712           tempo = acid->tempo;
1713           gst_buffer_unmap (buf, &map);
1714         }
1715         /* send data as tags */
1716         if (!wav->tags)
1717           wav->tags = gst_tag_list_new_empty ();
1718         gst_tag_list_add (wav->tags, GST_TAG_MERGE_REPLACE,
1719             GST_TAG_BEATS_PER_MINUTE, tempo, NULL);
1720
1721         size = GST_ROUND_UP_2 (size);
1722         if (wav->streaming) {
1723           gst_adapter_flush (wav->adapter, size);
1724         } else {
1725           gst_buffer_unref (buf);
1726         }
1727         wav->offset += 8 + size;
1728         break;
1729       }
1730         /* FIXME: all list tags after data are ignored in streaming mode */
1731       case GST_RIFF_TAG_LIST:{
1732         guint32 ltag;
1733
1734         if (wav->streaming) {
1735           const guint8 *data = NULL;
1736
1737           if (gst_adapter_available (wav->adapter) < 12) {
1738             goto exit;
1739           }
1740           data = gst_adapter_map (wav->adapter, 12);
1741           ltag = GST_READ_UINT32_LE (data + 8);
1742           gst_adapter_unmap (wav->adapter);
1743         } else {
1744           gst_buffer_unref (buf);
1745           buf = NULL;
1746           if ((res =
1747                   gst_pad_pull_range (wav->sinkpad, wav->offset, 12,
1748                       &buf)) != GST_FLOW_OK)
1749             goto header_read_error;
1750           gst_buffer_extract (buf, 8, &ltag, 4);
1751           ltag = GUINT32_FROM_LE (ltag);
1752         }
1753         switch (ltag) {
1754           case GST_RIFF_LIST_INFO:{
1755             const gint data_size = size - 4;
1756             GstTagList *new;
1757
1758             GST_INFO_OBJECT (wav, "Have LIST chunk INFO size %u", data_size);
1759             if (wav->streaming) {
1760               if (!gst_wavparse_peek_chunk (wav, &tag, &size)) {
1761                 goto exit;
1762               }
1763               gst_adapter_flush (wav->adapter, 12);
1764               wav->offset += 12;
1765               if (data_size > 0) {
1766                 buf = gst_adapter_take_buffer (wav->adapter, data_size);
1767                 if (data_size & 1)
1768                   gst_adapter_flush (wav->adapter, 1);
1769               }
1770             } else {
1771               wav->offset += 12;
1772               gst_buffer_unref (buf);
1773               buf = NULL;
1774               if (data_size > 0) {
1775                 if ((res =
1776                         gst_pad_pull_range (wav->sinkpad, wav->offset,
1777                             data_size, &buf)) != GST_FLOW_OK)
1778                   goto header_read_error;
1779               }
1780             }
1781             if (data_size > 0) {
1782               /* parse tags */
1783               gst_riff_parse_info (GST_ELEMENT (wav), buf, &new);
1784               if (new) {
1785                 GstTagList *old = wav->tags;
1786                 wav->tags =
1787                     gst_tag_list_merge (old, new, GST_TAG_MERGE_REPLACE);
1788                 if (old)
1789                   gst_tag_list_unref (old);
1790                 gst_tag_list_unref (new);
1791               }
1792               gst_buffer_unref (buf);
1793               wav->offset += GST_ROUND_UP_2 (data_size);
1794             }
1795             break;
1796           }
1797           case GST_RIFF_LIST_adtl:{
1798             const gint data_size = size;
1799
1800             GST_INFO_OBJECT (wav, "Have 'adtl' LIST, size %u", data_size);
1801             if (wav->streaming) {
1802               const guint8 *data = NULL;
1803
1804               gst_adapter_flush (wav->adapter, 12);
1805               data = gst_adapter_map (wav->adapter, data_size);
1806               gst_wavparse_adtl_chunk (wav, data, data_size);
1807               gst_adapter_unmap (wav->adapter);
1808             } else {
1809               GstMapInfo map;
1810
1811               gst_buffer_unref (buf);
1812               buf = NULL;
1813               if ((res =
1814                       gst_pad_pull_range (wav->sinkpad, wav->offset + 12,
1815                           data_size, &buf)) != GST_FLOW_OK)
1816                 goto header_read_error;
1817               gst_buffer_map (buf, &map, GST_MAP_READ);
1818               gst_wavparse_adtl_chunk (wav, (const guint8 *) map.data,
1819                   data_size);
1820               gst_buffer_unmap (buf, &map);
1821             }
1822           }
1823           default:
1824             GST_INFO_OBJECT (wav, "Ignoring LIST chunk %" GST_FOURCC_FORMAT,
1825                 GST_FOURCC_ARGS (ltag));
1826             if (!gst_waveparse_ignore_chunk (wav, buf, tag, size))
1827               /* need more data */
1828               goto exit;
1829             break;
1830         }
1831         break;
1832       }
1833       case GST_RIFF_TAG_cue:{
1834         const guint data_size = size;
1835
1836         GST_DEBUG_OBJECT (wav, "Have 'cue' TAG, size : %u", data_size);
1837         if (wav->streaming) {
1838           const guint8 *data = NULL;
1839
1840           if (!gst_wavparse_peek_chunk (wav, &tag, &size)) {
1841             goto exit;
1842           }
1843           gst_adapter_flush (wav->adapter, 8);
1844           wav->offset += 8;
1845           data = gst_adapter_map (wav->adapter, data_size);
1846           if (!gst_wavparse_cue_chunk (wav, data, data_size)) {
1847             goto header_read_error;
1848           }
1849           gst_adapter_unmap (wav->adapter);
1850         } else {
1851           GstMapInfo map;
1852
1853           wav->offset += 8;
1854           gst_buffer_unref (buf);
1855           buf = NULL;
1856           if ((res =
1857                   gst_pad_pull_range (wav->sinkpad, wav->offset,
1858                       data_size, &buf)) != GST_FLOW_OK)
1859             goto header_read_error;
1860           gst_buffer_map (buf, &map, GST_MAP_READ);
1861           if (!gst_wavparse_cue_chunk (wav, (const guint8 *) map.data,
1862                   data_size)) {
1863             goto header_read_error;
1864           }
1865           gst_buffer_unmap (buf, &map);
1866         }
1867         size = GST_ROUND_UP_2 (size);
1868         if (wav->streaming) {
1869           gst_adapter_flush (wav->adapter, size);
1870         } else {
1871           gst_buffer_unref (buf);
1872         }
1873         size = GST_ROUND_UP_2 (size);
1874         wav->offset += size;
1875         break;
1876       }
1877       default:
1878         if (!gst_waveparse_ignore_chunk (wav, buf, tag, size))
1879           /* need more data */
1880           goto exit;
1881         break;
1882     }
1883
1884     if (upstream_size && (wav->offset >= upstream_size)) {
1885       /* Now we are gone through the whole file */
1886       gotdata = TRUE;
1887     }
1888   }
1889
1890   GST_DEBUG_OBJECT (wav, "Finished parsing headers");
1891
1892   if (wav->bps <= 0 && wav->fact) {
1893 #if 0
1894     /* not a good idea, as for embedded mp2/mp3 we set bps to 0 earlier */
1895     wav->bps =
1896         (guint32) gst_util_uint64_scale ((guint64) wav->rate, wav->datasize,
1897         (guint64) wav->fact);
1898     GST_INFO_OBJECT (wav, "calculated bps : %u, enabling VBR", wav->bps);
1899 #endif
1900     wav->vbr = TRUE;
1901   }
1902
1903   if (gst_wavparse_calculate_duration (wav)) {
1904     gst_segment_init (&wav->segment, GST_FORMAT_TIME);
1905     if (!wav->ignore_length)
1906       wav->segment.duration = wav->duration;
1907     if (!wav->toc)
1908       gst_wavparse_create_toc (wav);
1909   } else {
1910     /* no bitrate, let downstream peer do the math, we'll feed it bytes. */
1911     gst_segment_init (&wav->segment, GST_FORMAT_BYTES);
1912     if (!wav->ignore_length)
1913       wav->segment.duration = wav->datasize;
1914   }
1915
1916   /* now we have all the info to perform a pending seek if any, if no
1917    * event, this will still do the right thing and it will also send
1918    * the right newsegment event downstream. */
1919   gst_wavparse_perform_seek (wav, wav->seek_event);
1920   /* remove pending event */
1921   event_p = &wav->seek_event;
1922   gst_event_replace (event_p, NULL);
1923
1924   /* we just started, we are discont */
1925   wav->discont = TRUE;
1926
1927   wav->state = GST_WAVPARSE_DATA;
1928
1929   /* determine reasonable max buffer size,
1930    * that is, buffers not too small either size or time wise
1931    * so we do not end up with too many of them */
1932   /* var abuse */
1933   upstream_size = 0;
1934   gst_wavparse_time_to_bytepos (wav, 40 * GST_MSECOND, &upstream_size);
1935   wav->max_buf_size = upstream_size;
1936   wav->max_buf_size = MAX (wav->max_buf_size, MAX_BUFFER_SIZE);
1937   if (wav->blockalign > 0)
1938     wav->max_buf_size -= (wav->max_buf_size % wav->blockalign);
1939
1940   GST_DEBUG_OBJECT (wav, "max buffer size %u", wav->max_buf_size);
1941
1942   return GST_FLOW_OK;
1943
1944   /* ERROR */
1945 exit:
1946   {
1947     if (codec_name)
1948       g_free (codec_name);
1949     if (header)
1950       g_free (header);
1951     if (caps)
1952       gst_caps_unref (caps);
1953     return res;
1954   }
1955 fail:
1956   {
1957     res = GST_FLOW_ERROR;
1958     goto exit;
1959   }
1960 invalid_wav:
1961   {
1962     GST_ELEMENT_ERROR (wav, STREAM, TYPE_NOT_FOUND, (NULL),
1963         ("Invalid WAV header (no fmt at start): %"
1964             GST_FOURCC_FORMAT, GST_FOURCC_ARGS (tag)));
1965     goto fail;
1966   }
1967 parse_header_error:
1968   {
1969     GST_ELEMENT_ERROR (wav, STREAM, DEMUX, (NULL),
1970         ("Couldn't parse audio header"));
1971     goto fail;
1972   }
1973 no_channels:
1974   {
1975     GST_ELEMENT_ERROR (wav, STREAM, FAILED, (NULL),
1976         ("Stream claims to contain no channels - invalid data"));
1977     goto fail;
1978   }
1979 no_rate:
1980   {
1981     GST_ELEMENT_ERROR (wav, STREAM, FAILED, (NULL),
1982         ("Stream with sample_rate == 0 - invalid data"));
1983     goto fail;
1984   }
1985 invalid_blockalign:
1986   {
1987     GST_ELEMENT_ERROR (wav, STREAM, FAILED, (NULL),
1988         ("Stream claims blockalign = %u, which is more than %u - invalid data",
1989             wav->blockalign, wav->channels * ((wav->depth + 7) / 8)));
1990     goto fail;
1991   }
1992 invalid_bps:
1993   {
1994     GST_ELEMENT_ERROR (wav, STREAM, FAILED, (NULL),
1995         ("Stream claims av_bsp = %u, which is more than %u - invalid data",
1996             wav->av_bps, wav->blockalign * wav->rate));
1997     goto fail;
1998   }
1999 no_bytes_per_sample:
2000   {
2001     GST_ELEMENT_ERROR (wav, STREAM, FAILED, (NULL),
2002         ("Could not caluclate bytes per sample - invalid data"));
2003     goto fail;
2004   }
2005 unknown_format:
2006   {
2007     GST_ELEMENT_ERROR (wav, STREAM, TYPE_NOT_FOUND, (NULL),
2008         ("No caps found for format 0x%x, %u channels, %u Hz",
2009             wav->format, wav->channels, wav->rate));
2010     goto fail;
2011   }
2012 header_read_error:
2013   {
2014     GST_ELEMENT_ERROR (wav, STREAM, DEMUX, (NULL),
2015         ("Couldn't read in header %d (%s)", res, gst_flow_get_name (res)));
2016     goto fail;
2017   }
2018 }
2019
2020 /*
2021  * Read WAV file tag when streaming
2022  */
2023 static GstFlowReturn
2024 gst_wavparse_parse_stream_init (GstWavParse * wav)
2025 {
2026   if (gst_adapter_available (wav->adapter) >= 12) {
2027     GstBuffer *tmp;
2028
2029     /* _take flushes the data */
2030     tmp = gst_adapter_take_buffer (wav->adapter, 12);
2031
2032     GST_DEBUG ("Parsing wav header");
2033     if (!gst_wavparse_parse_file_header (GST_ELEMENT_CAST (wav), tmp))
2034       return GST_FLOW_ERROR;
2035
2036     wav->offset += 12;
2037     /* Go to next state */
2038     wav->state = GST_WAVPARSE_HEADER;
2039   }
2040   return GST_FLOW_OK;
2041 }
2042
2043 /* handle an event sent directly to the element.
2044  *
2045  * This event can be sent either in the READY state or the
2046  * >READY state. The only event of interest really is the seek
2047  * event.
2048  *
2049  * In the READY state we can only store the event and try to
2050  * respect it when going to PAUSED. We assume we are in the
2051  * READY state when our parsing state != GST_WAVPARSE_DATA.
2052  *
2053  * When we are steaming, we can simply perform the seek right
2054  * away.
2055  */
2056 static gboolean
2057 gst_wavparse_send_event (GstElement * element, GstEvent * event)
2058 {
2059   GstWavParse *wav = GST_WAVPARSE (element);
2060   gboolean res = FALSE;
2061   GstEvent **event_p;
2062
2063   GST_DEBUG_OBJECT (wav, "received event %s", GST_EVENT_TYPE_NAME (event));
2064
2065   switch (GST_EVENT_TYPE (event)) {
2066     case GST_EVENT_SEEK:
2067       if (wav->state == GST_WAVPARSE_DATA) {
2068         /* we can handle the seek directly when streaming data */
2069         res = gst_wavparse_perform_seek (wav, event);
2070       } else {
2071         GST_DEBUG_OBJECT (wav, "queuing seek for later");
2072
2073         event_p = &wav->seek_event;
2074         gst_event_replace (event_p, event);
2075
2076         /* we always return true */
2077         res = TRUE;
2078       }
2079       break;
2080     default:
2081       break;
2082   }
2083   gst_event_unref (event);
2084   return res;
2085 }
2086
2087 static gboolean
2088 gst_wavparse_have_dts_caps (const GstCaps * caps, GstTypeFindProbability prob)
2089 {
2090   GstStructure *s;
2091
2092   s = gst_caps_get_structure (caps, 0);
2093   if (!gst_structure_has_name (s, "audio/x-dts"))
2094     return FALSE;
2095   if (prob >= GST_TYPE_FIND_LIKELY)
2096     return TRUE;
2097   /* DTS at non-0 offsets and without second sync may yield POSSIBLE .. */
2098   if (prob < GST_TYPE_FIND_POSSIBLE)
2099     return FALSE;
2100   /* .. in which case we want at least a valid-looking rate and channels */
2101   if (!gst_structure_has_field (s, "channels"))
2102     return FALSE;
2103   /* and for extra assurance we could also check the rate from the DTS frame
2104    * against the one in the wav header, but for now let's not do that */
2105   return gst_structure_has_field (s, "rate");
2106 }
2107
2108 static void
2109 gst_wavparse_add_src_pad (GstWavParse * wav, GstBuffer * buf)
2110 {
2111   GstStructure *s;
2112
2113   GST_DEBUG_OBJECT (wav, "adding src pad");
2114
2115   g_assert (wav->caps != NULL);
2116
2117   s = gst_caps_get_structure (wav->caps, 0);
2118   if (s && gst_structure_has_name (s, "audio/x-raw") && buf != NULL) {
2119     GstTypeFindProbability prob;
2120     GstCaps *tf_caps;
2121
2122     tf_caps = gst_type_find_helper_for_buffer (GST_OBJECT (wav), buf, &prob);
2123     if (tf_caps != NULL) {
2124       GST_LOG ("typefind caps = %" GST_PTR_FORMAT ", P=%d", tf_caps, prob);
2125       if (gst_wavparse_have_dts_caps (tf_caps, prob)) {
2126         GST_INFO_OBJECT (wav, "Found DTS marker in file marked as raw PCM");
2127         gst_caps_unref (wav->caps);
2128         wav->caps = tf_caps;
2129
2130         gst_tag_list_add (wav->tags, GST_TAG_MERGE_REPLACE,
2131             GST_TAG_AUDIO_CODEC, "dts", NULL);
2132       } else {
2133         GST_DEBUG_OBJECT (wav, "found caps %" GST_PTR_FORMAT " for stream "
2134             "marked as raw PCM audio, but ignoring for now", tf_caps);
2135         gst_caps_unref (tf_caps);
2136       }
2137     }
2138   }
2139
2140   gst_pad_set_caps (wav->srcpad, wav->caps);
2141   gst_caps_replace (&wav->caps, NULL);
2142
2143   if (wav->start_segment) {
2144     GST_DEBUG_OBJECT (wav, "Send start segment event on newpad");
2145     gst_pad_push_event (wav->srcpad, wav->start_segment);
2146     wav->start_segment = NULL;
2147   }
2148
2149   if (wav->tags) {
2150     gst_pad_push_event (wav->srcpad, gst_event_new_tag (wav->tags));
2151     wav->tags = NULL;
2152   }
2153 }
2154
2155 static GstFlowReturn
2156 gst_wavparse_stream_data (GstWavParse * wav)
2157 {
2158   GstBuffer *buf = NULL;
2159   GstFlowReturn res = GST_FLOW_OK;
2160   guint64 desired, obtained;
2161   GstClockTime timestamp, next_timestamp, duration;
2162   guint64 pos, nextpos;
2163
2164 iterate_adapter:
2165   GST_LOG_OBJECT (wav,
2166       "offset: %" G_GINT64_FORMAT " , end: %" G_GINT64_FORMAT " , dataleft: %"
2167       G_GINT64_FORMAT, wav->offset, wav->end_offset, wav->dataleft);
2168
2169   /* Get the next n bytes and output them */
2170   if (wav->dataleft == 0 || wav->dataleft < wav->blockalign)
2171     goto found_eos;
2172
2173   /* scale the amount of data by the segment rate so we get equal
2174    * amounts of data regardless of the playback rate */
2175   desired =
2176       MIN (gst_guint64_to_gdouble (wav->dataleft),
2177       wav->max_buf_size * ABS (wav->segment.rate));
2178
2179   if (desired >= wav->blockalign && wav->blockalign > 0)
2180     desired -= (desired % wav->blockalign);
2181
2182   GST_LOG_OBJECT (wav, "Fetching %" G_GINT64_FORMAT " bytes of data "
2183       "from the sinkpad", desired);
2184
2185   if (wav->streaming) {
2186     guint avail = gst_adapter_available (wav->adapter);
2187     guint extra;
2188
2189     /* flush some bytes if evil upstream sends segment that starts
2190      * before data or does is not send sample aligned segment */
2191     if (G_LIKELY (wav->offset >= wav->datastart)) {
2192       extra = (wav->offset - wav->datastart) % wav->bytes_per_sample;
2193     } else {
2194       extra = wav->datastart - wav->offset;
2195     }
2196
2197     if (G_UNLIKELY (extra)) {
2198       extra = wav->bytes_per_sample - extra;
2199       if (extra <= avail) {
2200         GST_DEBUG_OBJECT (wav, "flushing %u bytes to sample boundary", extra);
2201         gst_adapter_flush (wav->adapter, extra);
2202         wav->offset += extra;
2203         wav->dataleft -= extra;
2204         goto iterate_adapter;
2205       } else {
2206         GST_DEBUG_OBJECT (wav, "flushing %u bytes", avail);
2207         gst_adapter_clear (wav->adapter);
2208         wav->offset += avail;
2209         wav->dataleft -= avail;
2210         return GST_FLOW_OK;
2211       }
2212     }
2213
2214     if (avail < desired) {
2215       GST_LOG_OBJECT (wav, "Got only %u bytes of data from the sinkpad", avail);
2216       return GST_FLOW_OK;
2217     }
2218
2219     buf = gst_adapter_take_buffer (wav->adapter, desired);
2220   } else {
2221     if ((res = gst_pad_pull_range (wav->sinkpad, wav->offset,
2222                 desired, &buf)) != GST_FLOW_OK)
2223       goto pull_error;
2224
2225     /* we may get a short buffer at the end of the file */
2226     if (gst_buffer_get_size (buf) < desired) {
2227       gsize size = gst_buffer_get_size (buf);
2228
2229       GST_LOG_OBJECT (wav, "Got only %" G_GSIZE_FORMAT " bytes of data", size);
2230       if (size >= wav->blockalign) {
2231         buf = gst_buffer_make_writable (buf);
2232         gst_buffer_resize (buf, 0, size - (size % wav->blockalign));
2233       } else {
2234         gst_buffer_unref (buf);
2235         goto found_eos;
2236       }
2237     }
2238   }
2239
2240   obtained = gst_buffer_get_size (buf);
2241
2242   /* our positions in bytes */
2243   pos = wav->offset - wav->datastart;
2244   nextpos = pos + obtained;
2245
2246   /* update offsets, does not overflow. */
2247   buf = gst_buffer_make_writable (buf);
2248   GST_BUFFER_OFFSET (buf) = pos / wav->bytes_per_sample;
2249   GST_BUFFER_OFFSET_END (buf) = nextpos / wav->bytes_per_sample;
2250
2251   /* first chunk of data? create the source pad. We do this only here so
2252    * we can detect broken .wav files with dts disguised as raw PCM (sigh) */
2253   if (G_UNLIKELY (wav->first)) {
2254     wav->first = FALSE;
2255     /* this will also push the segment events */
2256     gst_wavparse_add_src_pad (wav, buf);
2257   } else {
2258     /* If we have a pending start segment, send it now. */
2259     if (G_UNLIKELY (wav->start_segment != NULL)) {
2260       gst_pad_push_event (wav->srcpad, wav->start_segment);
2261       wav->start_segment = NULL;
2262     }
2263   }
2264
2265   if (wav->bps > 0) {
2266     /* and timestamps if we have a bitrate, be careful for overflows */
2267     timestamp =
2268         gst_util_uint64_scale_ceil (pos, GST_SECOND, (guint64) wav->bps);
2269     next_timestamp =
2270         gst_util_uint64_scale_ceil (nextpos, GST_SECOND, (guint64) wav->bps);
2271     duration = next_timestamp - timestamp;
2272
2273     /* update current running segment position */
2274     if (G_LIKELY (next_timestamp >= wav->segment.start))
2275       wav->segment.position = next_timestamp;
2276   } else if (wav->fact) {
2277     guint64 bps =
2278         gst_util_uint64_scale_int (wav->datasize, wav->rate, wav->fact);
2279     /* and timestamps if we have a bitrate, be careful for overflows */
2280     timestamp = gst_util_uint64_scale_ceil (pos, GST_SECOND, bps);
2281     next_timestamp = gst_util_uint64_scale_ceil (nextpos, GST_SECOND, bps);
2282     duration = next_timestamp - timestamp;
2283   } else {
2284     /* no bitrate, all we know is that the first sample has timestamp 0, all
2285      * other positions and durations have unknown timestamp. */
2286     if (pos == 0)
2287       timestamp = 0;
2288     else
2289       timestamp = GST_CLOCK_TIME_NONE;
2290     duration = GST_CLOCK_TIME_NONE;
2291     /* update current running segment position with byte offset */
2292     if (G_LIKELY (nextpos >= wav->segment.start))
2293       wav->segment.position = nextpos;
2294   }
2295   if ((pos > 0) && wav->vbr) {
2296     /* don't set timestamps for VBR files if it's not the first buffer */
2297     timestamp = GST_CLOCK_TIME_NONE;
2298     duration = GST_CLOCK_TIME_NONE;
2299   }
2300   if (wav->discont) {
2301     GST_DEBUG_OBJECT (wav, "marking DISCONT");
2302     GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_DISCONT);
2303     wav->discont = FALSE;
2304   }
2305
2306   GST_BUFFER_TIMESTAMP (buf) = timestamp;
2307   GST_BUFFER_DURATION (buf) = duration;
2308
2309   GST_LOG_OBJECT (wav,
2310       "Got buffer. timestamp:%" GST_TIME_FORMAT " , duration:%" GST_TIME_FORMAT
2311       ", size:%" G_GSIZE_FORMAT, GST_TIME_ARGS (timestamp),
2312       GST_TIME_ARGS (duration), gst_buffer_get_size (buf));
2313
2314   if ((res = gst_pad_push (wav->srcpad, buf)) != GST_FLOW_OK)
2315     goto push_error;
2316
2317   if (obtained < wav->dataleft) {
2318     wav->offset += obtained;
2319     wav->dataleft -= obtained;
2320   } else {
2321     wav->offset += wav->dataleft;
2322     wav->dataleft = 0;
2323   }
2324
2325   /* Iterate until need more data, so adapter size won't grow */
2326   if (wav->streaming) {
2327     GST_LOG_OBJECT (wav,
2328         "offset: %" G_GINT64_FORMAT " , end: %" G_GINT64_FORMAT, wav->offset,
2329         wav->end_offset);
2330     goto iterate_adapter;
2331   }
2332   return res;
2333
2334   /* ERROR */
2335 found_eos:
2336   {
2337     GST_DEBUG_OBJECT (wav, "found EOS");
2338     return GST_FLOW_EOS;
2339   }
2340 pull_error:
2341   {
2342     /* check if we got EOS */
2343     if (res == GST_FLOW_EOS)
2344       goto found_eos;
2345
2346     GST_WARNING_OBJECT (wav,
2347         "Error getting %" G_GINT64_FORMAT " bytes from the "
2348         "sinkpad (dataleft = %" G_GINT64_FORMAT ")", desired, wav->dataleft);
2349     return res;
2350   }
2351 push_error:
2352   {
2353     GST_INFO_OBJECT (wav,
2354         "Error pushing on srcpad %s:%s, reason %s, is linked? = %d",
2355         GST_DEBUG_PAD_NAME (wav->srcpad), gst_flow_get_name (res),
2356         gst_pad_is_linked (wav->srcpad));
2357     return res;
2358   }
2359 }
2360
2361 static void
2362 gst_wavparse_loop (GstPad * pad)
2363 {
2364   GstFlowReturn ret;
2365   GstWavParse *wav = GST_WAVPARSE (GST_PAD_PARENT (pad));
2366   GstEvent *event;
2367   gchar *stream_id;
2368
2369   GST_LOG_OBJECT (wav, "process data");
2370
2371   switch (wav->state) {
2372     case GST_WAVPARSE_START:
2373       GST_INFO_OBJECT (wav, "GST_WAVPARSE_START");
2374       if ((ret = gst_wavparse_stream_init (wav)) != GST_FLOW_OK)
2375         goto pause;
2376
2377       stream_id =
2378           gst_pad_create_stream_id (wav->srcpad, GST_ELEMENT_CAST (wav), NULL);
2379       event = gst_event_new_stream_start (stream_id);
2380       gst_event_set_group_id (event, gst_util_group_id_next ());
2381       gst_pad_push_event (wav->srcpad, event);
2382       g_free (stream_id);
2383
2384       wav->state = GST_WAVPARSE_HEADER;
2385       /* fall-through */
2386
2387     case GST_WAVPARSE_HEADER:
2388       GST_INFO_OBJECT (wav, "GST_WAVPARSE_HEADER");
2389       if ((ret = gst_wavparse_stream_headers (wav)) != GST_FLOW_OK)
2390         goto pause;
2391
2392       wav->state = GST_WAVPARSE_DATA;
2393       GST_INFO_OBJECT (wav, "GST_WAVPARSE_DATA");
2394       /* fall-through */
2395
2396     case GST_WAVPARSE_DATA:
2397       if ((ret = gst_wavparse_stream_data (wav)) != GST_FLOW_OK)
2398         goto pause;
2399       break;
2400     default:
2401       g_assert_not_reached ();
2402   }
2403   return;
2404
2405   /* ERRORS */
2406 pause:
2407   {
2408     const gchar *reason = gst_flow_get_name (ret);
2409
2410     GST_DEBUG_OBJECT (wav, "pausing task, reason %s", reason);
2411     gst_pad_pause_task (pad);
2412
2413     if (ret == GST_FLOW_EOS) {
2414       /* handle end-of-stream/segment */
2415       /* so align our position with the end of it, if there is one
2416        * this ensures a subsequent will arrive at correct base/acc time */
2417       if (wav->segment.format == GST_FORMAT_TIME) {
2418         if (wav->segment.rate > 0.0 &&
2419             GST_CLOCK_TIME_IS_VALID (wav->segment.stop))
2420           wav->segment.position = wav->segment.stop;
2421         else if (wav->segment.rate < 0.0)
2422           wav->segment.position = wav->segment.start;
2423       }
2424       if (wav->state == GST_WAVPARSE_START) {
2425         GST_ELEMENT_ERROR (wav, STREAM, WRONG_TYPE, (NULL),
2426             ("No valid input found before end of stream"));
2427         gst_pad_push_event (wav->srcpad, gst_event_new_eos ());
2428       } else {
2429         /* add pad before we perform EOS */
2430         if (G_UNLIKELY (wav->first)) {
2431           wav->first = FALSE;
2432           gst_wavparse_add_src_pad (wav, NULL);
2433         }
2434
2435         /* perform EOS logic */
2436         if (wav->segment.flags & GST_SEEK_FLAG_SEGMENT) {
2437           GstClockTime stop;
2438
2439           if ((stop = wav->segment.stop) == -1)
2440             stop = wav->segment.duration;
2441
2442           gst_element_post_message (GST_ELEMENT_CAST (wav),
2443               gst_message_new_segment_done (GST_OBJECT_CAST (wav),
2444                   wav->segment.format, stop));
2445           gst_pad_push_event (wav->srcpad,
2446               gst_event_new_segment_done (wav->segment.format, stop));
2447         } else {
2448           gst_pad_push_event (wav->srcpad, gst_event_new_eos ());
2449         }
2450       }
2451     } else if (ret == GST_FLOW_NOT_LINKED || ret < GST_FLOW_EOS) {
2452       /* for fatal errors we post an error message, post the error
2453        * first so the app knows about the error first. */
2454       GST_ELEMENT_ERROR (wav, STREAM, FAILED,
2455           (_("Internal data flow error.")),
2456           ("streaming task paused, reason %s (%d)", reason, ret));
2457       gst_pad_push_event (wav->srcpad, gst_event_new_eos ());
2458     }
2459     return;
2460   }
2461 }
2462
2463 static GstFlowReturn
2464 gst_wavparse_chain (GstPad * pad, GstObject * parent, GstBuffer * buf)
2465 {
2466   GstFlowReturn ret;
2467   GstWavParse *wav = GST_WAVPARSE (parent);
2468
2469   GST_LOG_OBJECT (wav, "adapter_push %" G_GSIZE_FORMAT " bytes",
2470       gst_buffer_get_size (buf));
2471
2472   gst_adapter_push (wav->adapter, buf);
2473
2474   switch (wav->state) {
2475     case GST_WAVPARSE_START:
2476       GST_INFO_OBJECT (wav, "GST_WAVPARSE_START");
2477       if ((ret = gst_wavparse_parse_stream_init (wav)) != GST_FLOW_OK)
2478         goto done;
2479
2480       if (wav->state != GST_WAVPARSE_HEADER)
2481         break;
2482
2483       /* otherwise fall-through */
2484     case GST_WAVPARSE_HEADER:
2485       GST_INFO_OBJECT (wav, "GST_WAVPARSE_HEADER");
2486       if ((ret = gst_wavparse_stream_headers (wav)) != GST_FLOW_OK)
2487         goto done;
2488
2489       if (!wav->got_fmt || wav->datastart == 0)
2490         break;
2491
2492       wav->state = GST_WAVPARSE_DATA;
2493       GST_INFO_OBJECT (wav, "GST_WAVPARSE_DATA");
2494
2495       /* fall-through */
2496     case GST_WAVPARSE_DATA:
2497       if (buf && GST_BUFFER_FLAG_IS_SET (buf, GST_BUFFER_FLAG_DISCONT))
2498         wav->discont = TRUE;
2499       if ((ret = gst_wavparse_stream_data (wav)) != GST_FLOW_OK)
2500         goto done;
2501       break;
2502     default:
2503       g_return_val_if_reached (GST_FLOW_ERROR);
2504   }
2505 done:
2506   if (G_UNLIKELY (wav->abort_buffering)) {
2507     wav->abort_buffering = FALSE;
2508     ret = GST_FLOW_ERROR;
2509     /* sort of demux/parse error */
2510     GST_ELEMENT_ERROR (wav, STREAM, DEMUX, (NULL), ("unhandled buffer size"));
2511   }
2512
2513   return ret;
2514 }
2515
2516 static GstFlowReturn
2517 gst_wavparse_flush_data (GstWavParse * wav)
2518 {
2519   GstFlowReturn ret = GST_FLOW_OK;
2520   guint av;
2521
2522   if ((av = gst_adapter_available (wav->adapter)) > 0) {
2523     wav->dataleft = av;
2524     wav->end_offset = wav->offset + av;
2525     ret = gst_wavparse_stream_data (wav);
2526   }
2527
2528   return ret;
2529 }
2530
2531 static gboolean
2532 gst_wavparse_sink_event (GstPad * pad, GstObject * parent, GstEvent * event)
2533 {
2534   GstWavParse *wav = GST_WAVPARSE (parent);
2535   gboolean ret = TRUE;
2536
2537   GST_LOG_OBJECT (wav, "handling %s event", GST_EVENT_TYPE_NAME (event));
2538
2539   switch (GST_EVENT_TYPE (event)) {
2540     case GST_EVENT_CAPS:
2541     {
2542       /* discard, we'll come up with proper src caps */
2543       gst_event_unref (event);
2544       break;
2545     }
2546     case GST_EVENT_SEGMENT:
2547     {
2548       gint64 start, stop, offset = 0, end_offset = -1;
2549       GstSegment segment;
2550
2551       /* some debug output */
2552       gst_event_copy_segment (event, &segment);
2553       GST_DEBUG_OBJECT (wav, "received newsegment %" GST_SEGMENT_FORMAT,
2554           &segment);
2555
2556       if (wav->state != GST_WAVPARSE_DATA) {
2557         GST_DEBUG_OBJECT (wav, "still starting, eating event");
2558         goto exit;
2559       }
2560
2561       /* now we are either committed to TIME or BYTE format,
2562        * and we only expect a BYTE segment, e.g. following a seek */
2563       if (segment.format == GST_FORMAT_BYTES) {
2564         /* handle (un)signed issues */
2565         start = segment.start;
2566         stop = segment.stop;
2567         if (start > 0) {
2568           offset = start;
2569           start -= wav->datastart;
2570           start = MAX (start, 0);
2571         }
2572         if (stop > 0) {
2573           end_offset = stop;
2574           segment.stop -= wav->datastart;
2575           segment.stop = MAX (stop, 0);
2576         }
2577         if (wav->segment.format == GST_FORMAT_TIME) {
2578           guint64 bps = wav->bps;
2579
2580           /* operating in format TIME, so we can convert */
2581           if (!bps && wav->fact)
2582             bps =
2583                 gst_util_uint64_scale_int (wav->datasize, wav->rate, wav->fact);
2584           if (bps) {
2585             if (start >= 0)
2586               start =
2587                   gst_util_uint64_scale_ceil (start, GST_SECOND,
2588                   (guint64) wav->bps);
2589             if (stop >= 0)
2590               stop =
2591                   gst_util_uint64_scale_ceil (stop, GST_SECOND,
2592                   (guint64) wav->bps);
2593           }
2594         }
2595       } else {
2596         GST_DEBUG_OBJECT (wav, "unsupported segment format, ignoring");
2597         goto exit;
2598       }
2599
2600       segment.start = start;
2601       segment.stop = stop;
2602
2603       /* accept upstream's notion of segment and distribute along */
2604       segment.format = wav->segment.format;
2605       segment.time = segment.position = segment.start;
2606       segment.duration = wav->segment.duration;
2607       segment.base = gst_segment_to_running_time (&wav->segment,
2608           GST_FORMAT_TIME, wav->segment.position);
2609
2610       gst_segment_copy_into (&segment, &wav->segment);
2611
2612       /* also store the newsegment event for the streaming thread */
2613       if (wav->start_segment)
2614         gst_event_unref (wav->start_segment);
2615       GST_DEBUG_OBJECT (wav, "Storing newseg %" GST_SEGMENT_FORMAT, &segment);
2616       wav->start_segment = gst_event_new_segment (&segment);
2617
2618       /* stream leftover data in current segment */
2619       gst_wavparse_flush_data (wav);
2620       /* and set up streaming thread for next one */
2621       wav->offset = offset;
2622       wav->end_offset = end_offset;
2623       if (wav->end_offset > 0) {
2624         wav->dataleft = wav->end_offset - wav->offset;
2625       } else {
2626         /* infinity; upstream will EOS when done */
2627         wav->dataleft = G_MAXUINT64;
2628       }
2629     exit:
2630       gst_event_unref (event);
2631       break;
2632     }
2633     case GST_EVENT_EOS:
2634       if (wav->state == GST_WAVPARSE_START) {
2635         GST_ELEMENT_ERROR (wav, STREAM, WRONG_TYPE, (NULL),
2636             ("No valid input found before end of stream"));
2637       } else {
2638         /* add pad if needed so EOS is seen downstream */
2639         if (G_UNLIKELY (wav->first)) {
2640           wav->first = FALSE;
2641           gst_wavparse_add_src_pad (wav, NULL);
2642         } else {
2643           /* stream leftover data in current segment */
2644           gst_wavparse_flush_data (wav);
2645         }
2646       }
2647
2648       /* fall-through */
2649     case GST_EVENT_FLUSH_STOP:
2650     {
2651       GstClockTime dur;
2652
2653       gst_adapter_clear (wav->adapter);
2654       wav->discont = TRUE;
2655       dur = wav->segment.duration;
2656       gst_segment_init (&wav->segment, wav->segment.format);
2657       wav->segment.duration = dur;
2658       /* fall-through */
2659     }
2660     default:
2661       ret = gst_pad_event_default (wav->sinkpad, parent, event);
2662       break;
2663   }
2664
2665   return ret;
2666 }
2667
2668 #if 0
2669 /* convert and query stuff */
2670 static const GstFormat *
2671 gst_wavparse_get_formats (GstPad * pad)
2672 {
2673   static GstFormat formats[] = {
2674     GST_FORMAT_TIME,
2675     GST_FORMAT_BYTES,
2676     GST_FORMAT_DEFAULT,         /* a "frame", ie a set of samples per Hz */
2677     0
2678   };
2679
2680   return formats;
2681 }
2682 #endif
2683
2684 static gboolean
2685 gst_wavparse_pad_convert (GstPad * pad,
2686     GstFormat src_format, gint64 src_value,
2687     GstFormat * dest_format, gint64 * dest_value)
2688 {
2689   GstWavParse *wavparse;
2690   gboolean res = TRUE;
2691
2692   wavparse = GST_WAVPARSE (GST_PAD_PARENT (pad));
2693
2694   if (*dest_format == src_format) {
2695     *dest_value = src_value;
2696     return TRUE;
2697   }
2698
2699   if ((wavparse->bps == 0) && !wavparse->fact)
2700     goto no_bps_fact;
2701
2702   GST_INFO_OBJECT (wavparse, "converting value from %s to %s",
2703       gst_format_get_name (src_format), gst_format_get_name (*dest_format));
2704
2705   switch (src_format) {
2706     case GST_FORMAT_BYTES:
2707       switch (*dest_format) {
2708         case GST_FORMAT_DEFAULT:
2709           *dest_value = src_value / wavparse->bytes_per_sample;
2710           /* make sure we end up on a sample boundary */
2711           *dest_value -= *dest_value % wavparse->bytes_per_sample;
2712           break;
2713         case GST_FORMAT_TIME:
2714           /* src_value + datastart = offset */
2715           GST_INFO_OBJECT (wavparse,
2716               "src=%" G_GINT64_FORMAT ", offset=%" G_GINT64_FORMAT, src_value,
2717               wavparse->offset);
2718           if (wavparse->bps > 0)
2719             *dest_value = gst_util_uint64_scale_ceil (src_value, GST_SECOND,
2720                 (guint64) wavparse->bps);
2721           else if (wavparse->fact) {
2722             guint64 bps = gst_util_uint64_scale_int_ceil (wavparse->datasize,
2723                 wavparse->rate, wavparse->fact);
2724
2725             *dest_value =
2726                 gst_util_uint64_scale_int_ceil (src_value, GST_SECOND, bps);
2727           } else {
2728             res = FALSE;
2729           }
2730           break;
2731         default:
2732           res = FALSE;
2733           goto done;
2734       }
2735       break;
2736
2737     case GST_FORMAT_DEFAULT:
2738       switch (*dest_format) {
2739         case GST_FORMAT_BYTES:
2740           *dest_value = src_value * wavparse->bytes_per_sample;
2741           break;
2742         case GST_FORMAT_TIME:
2743           *dest_value = gst_util_uint64_scale (src_value, GST_SECOND,
2744               (guint64) wavparse->rate);
2745           break;
2746         default:
2747           res = FALSE;
2748           goto done;
2749       }
2750       break;
2751
2752     case GST_FORMAT_TIME:
2753       switch (*dest_format) {
2754         case GST_FORMAT_BYTES:
2755           if (wavparse->bps > 0)
2756             *dest_value = gst_util_uint64_scale (src_value,
2757                 (guint64) wavparse->bps, GST_SECOND);
2758           else {
2759             guint64 bps = gst_util_uint64_scale_int (wavparse->datasize,
2760                 wavparse->rate, wavparse->fact);
2761
2762             *dest_value = gst_util_uint64_scale (src_value, bps, GST_SECOND);
2763           }
2764           /* make sure we end up on a sample boundary */
2765           *dest_value -= *dest_value % wavparse->blockalign;
2766           break;
2767         case GST_FORMAT_DEFAULT:
2768           *dest_value = gst_util_uint64_scale (src_value,
2769               (guint64) wavparse->rate, GST_SECOND);
2770           break;
2771         default:
2772           res = FALSE;
2773           goto done;
2774       }
2775       break;
2776
2777     default:
2778       res = FALSE;
2779       goto done;
2780   }
2781
2782 done:
2783   return res;
2784
2785   /* ERRORS */
2786 no_bps_fact:
2787   {
2788     GST_DEBUG_OBJECT (wavparse, "bps 0 or no fact chunk, cannot convert");
2789     res = FALSE;
2790     goto done;
2791   }
2792 }
2793
2794 /* handle queries for location and length in requested format */
2795 static gboolean
2796 gst_wavparse_pad_query (GstPad * pad, GstObject * parent, GstQuery * query)
2797 {
2798   gboolean res = TRUE;
2799   GstWavParse *wav = GST_WAVPARSE (parent);
2800
2801   /* only if we know */
2802   if (wav->state != GST_WAVPARSE_DATA) {
2803     return FALSE;
2804   }
2805
2806   GST_LOG_OBJECT (pad, "%s query", GST_QUERY_TYPE_NAME (query));
2807
2808   switch (GST_QUERY_TYPE (query)) {
2809     case GST_QUERY_POSITION:
2810     {
2811       gint64 curb;
2812       gint64 cur;
2813       GstFormat format;
2814
2815       /* this is not very precise, as we have pushed severla buffer upstream for prerolling */
2816       curb = wav->offset - wav->datastart;
2817       gst_query_parse_position (query, &format, NULL);
2818       GST_INFO_OBJECT (wav, "pos query at %" G_GINT64_FORMAT, curb);
2819
2820       switch (format) {
2821         case GST_FORMAT_BYTES:
2822           format = GST_FORMAT_BYTES;
2823           cur = curb;
2824           break;
2825         default:
2826           res = gst_wavparse_pad_convert (pad, GST_FORMAT_BYTES, curb,
2827               &format, &cur);
2828           break;
2829       }
2830       if (res)
2831         gst_query_set_position (query, format, cur);
2832       break;
2833     }
2834     case GST_QUERY_DURATION:
2835     {
2836       gint64 duration = 0;
2837       GstFormat format;
2838
2839       if (wav->ignore_length) {
2840         res = FALSE;
2841         break;
2842       }
2843
2844       gst_query_parse_duration (query, &format, NULL);
2845
2846       switch (format) {
2847         case GST_FORMAT_BYTES:{
2848           format = GST_FORMAT_BYTES;
2849           duration = wav->datasize;
2850           break;
2851         }
2852         case GST_FORMAT_TIME:
2853           if ((res = gst_wavparse_calculate_duration (wav))) {
2854             duration = wav->duration;
2855           }
2856           break;
2857         default:
2858           res = FALSE;
2859           break;
2860       }
2861       if (res)
2862         gst_query_set_duration (query, format, duration);
2863       break;
2864     }
2865     case GST_QUERY_CONVERT:
2866     {
2867       gint64 srcvalue, dstvalue;
2868       GstFormat srcformat, dstformat;
2869
2870       gst_query_parse_convert (query, &srcformat, &srcvalue,
2871           &dstformat, &dstvalue);
2872       res = gst_wavparse_pad_convert (pad, srcformat, srcvalue,
2873           &dstformat, &dstvalue);
2874       if (res)
2875         gst_query_set_convert (query, srcformat, srcvalue, dstformat, dstvalue);
2876       break;
2877     }
2878     case GST_QUERY_SEEKING:{
2879       GstFormat fmt;
2880       gboolean seekable = FALSE;
2881
2882       gst_query_parse_seeking (query, &fmt, NULL, NULL, NULL);
2883       if (fmt == wav->segment.format) {
2884         if (wav->streaming) {
2885           GstQuery *q;
2886
2887           q = gst_query_new_seeking (GST_FORMAT_BYTES);
2888           if ((res = gst_pad_peer_query (wav->sinkpad, q))) {
2889             gst_query_parse_seeking (q, &fmt, &seekable, NULL, NULL);
2890             GST_LOG_OBJECT (wav, "upstream BYTE seekable %d", seekable);
2891           }
2892           gst_query_unref (q);
2893         } else {
2894           GST_LOG_OBJECT (wav, "looping => seekable");
2895           seekable = TRUE;
2896           res = TRUE;
2897         }
2898       } else if (fmt == GST_FORMAT_TIME) {
2899         res = TRUE;
2900       }
2901       if (res) {
2902         gst_query_set_seeking (query, fmt, seekable, 0, wav->segment.duration);
2903       }
2904       break;
2905     }
2906     default:
2907       res = gst_pad_query_default (pad, parent, query);
2908       break;
2909   }
2910   return res;
2911 }
2912
2913 static gboolean
2914 gst_wavparse_srcpad_event (GstPad * pad, GstObject * parent, GstEvent * event)
2915 {
2916   GstWavParse *wavparse = GST_WAVPARSE (parent);
2917   gboolean res = FALSE;
2918
2919   GST_DEBUG_OBJECT (wavparse, "%s event", GST_EVENT_TYPE_NAME (event));
2920
2921   switch (GST_EVENT_TYPE (event)) {
2922     case GST_EVENT_SEEK:
2923       /* can only handle events when we are in the data state */
2924       if (wavparse->state == GST_WAVPARSE_DATA) {
2925         res = gst_wavparse_perform_seek (wavparse, event);
2926       }
2927       gst_event_unref (event);
2928       break;
2929
2930     case GST_EVENT_TOC_SELECT:
2931     {
2932       char *uid = NULL;
2933       GstTocEntry *entry = NULL;
2934       GstEvent *seek_event;
2935       gint64 start_pos;
2936
2937       if (!wavparse->toc) {
2938         GST_DEBUG_OBJECT (wavparse, "no TOC to select");
2939         return FALSE;
2940       } else {
2941         gst_event_parse_toc_select (event, &uid);
2942         if (uid != NULL) {
2943           GST_OBJECT_LOCK (wavparse);
2944           entry = gst_toc_find_entry (wavparse->toc, uid);
2945           if (entry == NULL) {
2946             GST_OBJECT_UNLOCK (wavparse);
2947             GST_WARNING_OBJECT (wavparse, "no TOC entry with given UID: %s",
2948                 uid);
2949             res = FALSE;
2950           } else {
2951             gst_toc_entry_get_start_stop_times (entry, &start_pos, NULL);
2952             GST_OBJECT_UNLOCK (wavparse);
2953             seek_event = gst_event_new_seek (1.0,
2954                 GST_FORMAT_TIME,
2955                 GST_SEEK_FLAG_FLUSH,
2956                 GST_SEEK_TYPE_SET, start_pos, GST_SEEK_TYPE_SET, -1);
2957             res = gst_wavparse_perform_seek (wavparse, seek_event);
2958             gst_event_unref (seek_event);
2959           }
2960           g_free (uid);
2961         } else {
2962           GST_WARNING_OBJECT (wavparse, "received empty TOC select event");
2963           res = FALSE;
2964         }
2965       }
2966       gst_event_unref (event);
2967       break;
2968     }
2969
2970     default:
2971       res = gst_pad_push_event (wavparse->sinkpad, event);
2972       break;
2973   }
2974   return res;
2975 }
2976
2977 static gboolean
2978 gst_wavparse_sink_activate (GstPad * sinkpad, GstObject * parent)
2979 {
2980   GstWavParse *wav = GST_WAVPARSE (parent);
2981   GstQuery *query;
2982   gboolean pull_mode;
2983
2984   if (wav->adapter) {
2985     gst_adapter_clear (wav->adapter);
2986     g_object_unref (wav->adapter);
2987     wav->adapter = NULL;
2988   }
2989
2990   query = gst_query_new_scheduling ();
2991
2992   if (!gst_pad_peer_query (sinkpad, query)) {
2993     gst_query_unref (query);
2994     goto activate_push;
2995   }
2996
2997   pull_mode = gst_query_has_scheduling_mode_with_flags (query,
2998       GST_PAD_MODE_PULL, GST_SCHEDULING_FLAG_SEEKABLE);
2999   gst_query_unref (query);
3000
3001   if (!pull_mode)
3002     goto activate_push;
3003
3004   GST_DEBUG_OBJECT (sinkpad, "activating pull");
3005   wav->streaming = FALSE;
3006   return gst_pad_activate_mode (sinkpad, GST_PAD_MODE_PULL, TRUE);
3007
3008 activate_push:
3009   {
3010     GST_DEBUG_OBJECT (sinkpad, "activating push");
3011     wav->streaming = TRUE;
3012     wav->adapter = gst_adapter_new ();
3013     return gst_pad_activate_mode (sinkpad, GST_PAD_MODE_PUSH, TRUE);
3014   }
3015 }
3016
3017
3018 static gboolean
3019 gst_wavparse_sink_activate_mode (GstPad * sinkpad, GstObject * parent,
3020     GstPadMode mode, gboolean active)
3021 {
3022   gboolean res;
3023
3024   switch (mode) {
3025     case GST_PAD_MODE_PUSH:
3026       res = TRUE;
3027       break;
3028     case GST_PAD_MODE_PULL:
3029       if (active) {
3030         /* if we have a scheduler we can start the task */
3031         res = gst_pad_start_task (sinkpad, (GstTaskFunction) gst_wavparse_loop,
3032             sinkpad, NULL);
3033       } else {
3034         res = gst_pad_stop_task (sinkpad);
3035       }
3036       break;
3037     default:
3038       res = FALSE;
3039       break;
3040   }
3041   return res;
3042 }
3043
3044 static GstStateChangeReturn
3045 gst_wavparse_change_state (GstElement * element, GstStateChange transition)
3046 {
3047   GstStateChangeReturn ret;
3048   GstWavParse *wav = GST_WAVPARSE (element);
3049
3050   switch (transition) {
3051     case GST_STATE_CHANGE_NULL_TO_READY:
3052       break;
3053     case GST_STATE_CHANGE_READY_TO_PAUSED:
3054       gst_wavparse_reset (wav);
3055       break;
3056     case GST_STATE_CHANGE_PAUSED_TO_PLAYING:
3057       break;
3058     default:
3059       break;
3060   }
3061
3062   ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
3063
3064   switch (transition) {
3065     case GST_STATE_CHANGE_PLAYING_TO_PAUSED:
3066       break;
3067     case GST_STATE_CHANGE_PAUSED_TO_READY:
3068       gst_wavparse_reset (wav);
3069       break;
3070     case GST_STATE_CHANGE_READY_TO_NULL:
3071       break;
3072     default:
3073       break;
3074   }
3075   return ret;
3076 }
3077
3078 static void
3079 gst_wavparse_set_property (GObject * object, guint prop_id,
3080     const GValue * value, GParamSpec * pspec)
3081 {
3082   GstWavParse *self;
3083
3084   g_return_if_fail (GST_IS_WAVPARSE (object));
3085   self = GST_WAVPARSE (object);
3086
3087   switch (prop_id) {
3088     case PROP_IGNORE_LENGTH:
3089       self->ignore_length = g_value_get_boolean (value);
3090       break;
3091     default:
3092       G_OBJECT_WARN_INVALID_PROPERTY_ID (self, prop_id, pspec);
3093   }
3094
3095 }
3096
3097 static void
3098 gst_wavparse_get_property (GObject * object, guint prop_id,
3099     GValue * value, GParamSpec * pspec)
3100 {
3101   GstWavParse *self;
3102
3103   g_return_if_fail (GST_IS_WAVPARSE (object));
3104   self = GST_WAVPARSE (object);
3105
3106   switch (prop_id) {
3107     case PROP_IGNORE_LENGTH:
3108       g_value_set_boolean (value, self->ignore_length);
3109       break;
3110     default:
3111       G_OBJECT_WARN_INVALID_PROPERTY_ID (self, prop_id, pspec);
3112   }
3113 }
3114
3115 static gboolean
3116 plugin_init (GstPlugin * plugin)
3117 {
3118   gst_riff_init ();
3119
3120   return gst_element_register (plugin, "wavparse", GST_RANK_PRIMARY,
3121       GST_TYPE_WAVPARSE);
3122 }
3123
3124 GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
3125     GST_VERSION_MINOR,
3126     wavparse,
3127     "Parse a .wav file into raw audio",
3128     plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)