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