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