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