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     if (codec_name)
1695       g_free (codec_name);
1696     if (header)
1697       g_free (header);
1698     if (caps)
1699       gst_caps_unref (caps);
1700     return res;
1701   }
1702 fail:
1703   {
1704     res = GST_FLOW_ERROR;
1705     goto exit;
1706   }
1707 invalid_wav:
1708   {
1709     GST_ELEMENT_ERROR (wav, STREAM, TYPE_NOT_FOUND, (NULL),
1710         ("Invalid WAV header (no fmt at start): %"
1711             GST_FOURCC_FORMAT, GST_FOURCC_ARGS (tag)));
1712     goto fail;
1713   }
1714 parse_header_error:
1715   {
1716     GST_ELEMENT_ERROR (wav, STREAM, DEMUX, (NULL),
1717         ("Couldn't parse audio header"));
1718     goto fail;
1719   }
1720 no_channels:
1721   {
1722     GST_ELEMENT_ERROR (wav, STREAM, FAILED, (NULL),
1723         ("Stream claims to contain no channels - invalid data"));
1724     goto fail;
1725   }
1726 no_rate:
1727   {
1728     GST_ELEMENT_ERROR (wav, STREAM, FAILED, (NULL),
1729         ("Stream with sample_rate == 0 - invalid data"));
1730     goto fail;
1731   }
1732 invalid_blockalign:
1733   {
1734     GST_ELEMENT_ERROR (wav, STREAM, FAILED, (NULL),
1735         ("Stream claims blockalign = %u, which is more than %u - invalid data",
1736             wav->blockalign, wav->channels * ((wav->depth + 7) / 8)));
1737     goto fail;
1738   }
1739 invalid_bps:
1740   {
1741     GST_ELEMENT_ERROR (wav, STREAM, FAILED, (NULL),
1742         ("Stream claims av_bsp = %u, which is more than %u - invalid data",
1743             wav->av_bps, wav->blockalign * wav->rate));
1744     goto fail;
1745   }
1746 no_bytes_per_sample:
1747   {
1748     GST_ELEMENT_ERROR (wav, STREAM, FAILED, (NULL),
1749         ("Could not caluclate bytes per sample - invalid data"));
1750     goto fail;
1751   }
1752 unknown_format:
1753   {
1754     GST_ELEMENT_ERROR (wav, STREAM, TYPE_NOT_FOUND, (NULL),
1755         ("No caps found for format 0x%x, %u channels, %u Hz",
1756             wav->format, wav->channels, wav->rate));
1757     goto fail;
1758   }
1759 header_read_error:
1760   {
1761     GST_ELEMENT_ERROR (wav, STREAM, DEMUX, (NULL),
1762         ("Couldn't read in header %d (%s)", res, gst_flow_get_name (res)));
1763     goto fail;
1764   }
1765 }
1766
1767 /*
1768  * Read WAV file tag when streaming
1769  */
1770 static GstFlowReturn
1771 gst_wavparse_parse_stream_init (GstWavParse * wav)
1772 {
1773   if (gst_adapter_available (wav->adapter) >= 12) {
1774     GstBuffer *tmp;
1775
1776     /* _take flushes the data */
1777     tmp = gst_adapter_take_buffer (wav->adapter, 12);
1778
1779     GST_DEBUG ("Parsing wav header");
1780     if (!gst_wavparse_parse_file_header (GST_ELEMENT_CAST (wav), tmp))
1781       return GST_FLOW_ERROR;
1782
1783     wav->offset += 12;
1784     /* Go to next state */
1785     wav->state = GST_WAVPARSE_HEADER;
1786   }
1787   return GST_FLOW_OK;
1788 }
1789
1790 /* handle an event sent directly to the element.
1791  *
1792  * This event can be sent either in the READY state or the
1793  * >READY state. The only event of interest really is the seek
1794  * event.
1795  *
1796  * In the READY state we can only store the event and try to
1797  * respect it when going to PAUSED. We assume we are in the
1798  * READY state when our parsing state != GST_WAVPARSE_DATA.
1799  *
1800  * When we are steaming, we can simply perform the seek right
1801  * away.
1802  */
1803 static gboolean
1804 gst_wavparse_send_event (GstElement * element, GstEvent * event)
1805 {
1806   GstWavParse *wav = GST_WAVPARSE (element);
1807   gboolean res = FALSE;
1808   GstEvent **event_p;
1809
1810   GST_DEBUG_OBJECT (wav, "received event %s", GST_EVENT_TYPE_NAME (event));
1811
1812   switch (GST_EVENT_TYPE (event)) {
1813     case GST_EVENT_SEEK:
1814       if (wav->state == GST_WAVPARSE_DATA) {
1815         /* we can handle the seek directly when streaming data */
1816         res = gst_wavparse_perform_seek (wav, event);
1817       } else {
1818         GST_DEBUG_OBJECT (wav, "queuing seek for later");
1819
1820         event_p = &wav->seek_event;
1821         gst_event_replace (event_p, event);
1822
1823         /* we always return true */
1824         res = TRUE;
1825       }
1826       break;
1827     default:
1828       break;
1829   }
1830   gst_event_unref (event);
1831   return res;
1832 }
1833
1834 static gboolean
1835 gst_wavparse_have_dts_caps (const GstCaps * caps, GstTypeFindProbability prob)
1836 {
1837   GstStructure *s;
1838
1839   s = gst_caps_get_structure (caps, 0);
1840   if (!gst_structure_has_name (s, "audio/x-dts"))
1841     return FALSE;
1842   /* typefind behavior for DTS:
1843    *  MAXIMUM: multiple frame syncs detected, certainly DTS
1844    *  LIKELY: single frame sync at offset 0.  Maybe DTS?
1845    *  POSSIBLE: single frame sync, not at offset 0.  Highly unlikely
1846    *    to be DTS.  */
1847   if (prob > GST_TYPE_FIND_LIKELY)
1848     return TRUE;
1849   if (prob <= GST_TYPE_FIND_POSSIBLE)
1850     return FALSE;
1851   /* for maybe, check for at least a valid-looking rate and channels */
1852   if (!gst_structure_has_field (s, "channels"))
1853     return FALSE;
1854   /* and for extra assurance we could also check the rate from the DTS frame
1855    * against the one in the wav header, but for now let's not do that */
1856   return gst_structure_has_field (s, "rate");
1857 }
1858
1859 static GstTagList *
1860 gst_wavparse_get_upstream_tags (GstWavParse * wav, GstTagScope scope)
1861 {
1862   GstTagList *tags = NULL;
1863   GstEvent *ev;
1864   gint i;
1865
1866   i = 0;
1867   while ((ev = gst_pad_get_sticky_event (wav->sinkpad, GST_EVENT_TAG, i++))) {
1868     gst_event_parse_tag (ev, &tags);
1869     if (tags != NULL && gst_tag_list_get_scope (tags) == scope) {
1870       tags = gst_tag_list_copy (tags);
1871       gst_tag_list_remove_tag (tags, GST_TAG_CONTAINER_FORMAT);
1872       gst_event_unref (ev);
1873       break;
1874     }
1875     tags = NULL;
1876     gst_event_unref (ev);
1877   }
1878   return tags;
1879 }
1880
1881 static void
1882 gst_wavparse_add_src_pad (GstWavParse * wav, GstBuffer * buf)
1883 {
1884   GstStructure *s;
1885   GstTagList *tags, *utags;
1886
1887   GST_DEBUG_OBJECT (wav, "adding src pad");
1888
1889   g_assert (wav->caps != NULL);
1890
1891   s = gst_caps_get_structure (wav->caps, 0);
1892   if (s && gst_structure_has_name (s, "audio/x-raw") && buf != NULL) {
1893     GstTypeFindProbability prob;
1894     GstCaps *tf_caps;
1895
1896     tf_caps = gst_type_find_helper_for_buffer (GST_OBJECT (wav), buf, &prob);
1897     if (tf_caps != NULL) {
1898       GST_LOG ("typefind caps = %" GST_PTR_FORMAT ", P=%d", tf_caps, prob);
1899       if (gst_wavparse_have_dts_caps (tf_caps, prob)) {
1900         GST_INFO_OBJECT (wav, "Found DTS marker in file marked as raw PCM");
1901         gst_caps_unref (wav->caps);
1902         wav->caps = tf_caps;
1903
1904         gst_tag_list_add (wav->tags, GST_TAG_MERGE_REPLACE,
1905             GST_TAG_AUDIO_CODEC, "dts", NULL);
1906       } else {
1907         GST_DEBUG_OBJECT (wav, "found caps %" GST_PTR_FORMAT " for stream "
1908             "marked as raw PCM audio, but ignoring for now", tf_caps);
1909         gst_caps_unref (tf_caps);
1910       }
1911     }
1912   }
1913
1914   gst_pad_set_caps (wav->srcpad, wav->caps);
1915   gst_caps_replace (&wav->caps, NULL);
1916
1917   if (wav->start_segment) {
1918     GST_DEBUG_OBJECT (wav, "Send start segment event on newpad");
1919     gst_pad_push_event (wav->srcpad, wav->start_segment);
1920     wav->start_segment = NULL;
1921   }
1922
1923   /* upstream tags, e.g. from id3/ape tag before the wav file; assume for now
1924    * that there'll be only one scope/type of tag list from upstream, if any */
1925   utags = gst_wavparse_get_upstream_tags (wav, GST_TAG_SCOPE_GLOBAL);
1926   if (utags == NULL)
1927     utags = gst_wavparse_get_upstream_tags (wav, GST_TAG_SCOPE_STREAM);
1928
1929   /* if there's a tag upstream it's probably been added to override the
1930    * tags from inside the wav header, so keep upstream tags if in doubt */
1931   tags = gst_tag_list_merge (utags, wav->tags, GST_TAG_MERGE_KEEP);
1932
1933   if (wav->tags != NULL) {
1934     gst_tag_list_unref (wav->tags);
1935     wav->tags = NULL;
1936   }
1937
1938   if (utags != NULL)
1939     gst_tag_list_unref (utags);
1940
1941   /* send tags downstream, if any */
1942   if (tags != NULL)
1943     gst_pad_push_event (wav->srcpad, gst_event_new_tag (tags));
1944 }
1945
1946 static GstFlowReturn
1947 gst_wavparse_stream_data (GstWavParse * wav)
1948 {
1949   GstBuffer *buf = NULL;
1950   GstFlowReturn res = GST_FLOW_OK;
1951   guint64 desired, obtained;
1952   GstClockTime timestamp, next_timestamp, duration;
1953   guint64 pos, nextpos;
1954
1955 iterate_adapter:
1956   GST_LOG_OBJECT (wav,
1957       "offset: %" G_GINT64_FORMAT " , end: %" G_GINT64_FORMAT " , dataleft: %"
1958       G_GINT64_FORMAT, wav->offset, wav->end_offset, wav->dataleft);
1959
1960   /* Get the next n bytes and output them */
1961   if (wav->dataleft == 0 || wav->dataleft < wav->blockalign)
1962     goto found_eos;
1963
1964   /* scale the amount of data by the segment rate so we get equal
1965    * amounts of data regardless of the playback rate */
1966   desired =
1967       MIN (gst_guint64_to_gdouble (wav->dataleft),
1968       wav->max_buf_size * ABS (wav->segment.rate));
1969
1970   if (desired >= wav->blockalign && wav->blockalign > 0)
1971     desired -= (desired % wav->blockalign);
1972
1973   GST_LOG_OBJECT (wav, "Fetching %" G_GINT64_FORMAT " bytes of data "
1974       "from the sinkpad", desired);
1975
1976   if (wav->streaming) {
1977     guint avail = gst_adapter_available (wav->adapter);
1978     guint extra;
1979
1980     /* flush some bytes if evil upstream sends segment that starts
1981      * before data or does is not send sample aligned segment */
1982     if (G_LIKELY (wav->offset >= wav->datastart)) {
1983       extra = (wav->offset - wav->datastart) % wav->bytes_per_sample;
1984     } else {
1985       extra = wav->datastart - wav->offset;
1986     }
1987
1988     if (G_UNLIKELY (extra)) {
1989       extra = wav->bytes_per_sample - extra;
1990       if (extra <= avail) {
1991         GST_DEBUG_OBJECT (wav, "flushing %u bytes to sample boundary", extra);
1992         gst_adapter_flush (wav->adapter, extra);
1993         wav->offset += extra;
1994         wav->dataleft -= extra;
1995         goto iterate_adapter;
1996       } else {
1997         GST_DEBUG_OBJECT (wav, "flushing %u bytes", avail);
1998         gst_adapter_clear (wav->adapter);
1999         wav->offset += avail;
2000         wav->dataleft -= avail;
2001         return GST_FLOW_OK;
2002       }
2003     }
2004
2005     if (avail < desired) {
2006       GST_LOG_OBJECT (wav, "Got only %u bytes of data from the sinkpad", avail);
2007       return GST_FLOW_OK;
2008     }
2009
2010     buf = gst_adapter_take_buffer (wav->adapter, desired);
2011   } else {
2012     if ((res = gst_pad_pull_range (wav->sinkpad, wav->offset,
2013                 desired, &buf)) != GST_FLOW_OK)
2014       goto pull_error;
2015
2016     /* we may get a short buffer at the end of the file */
2017     if (gst_buffer_get_size (buf) < desired) {
2018       gsize size = gst_buffer_get_size (buf);
2019
2020       GST_LOG_OBJECT (wav, "Got only %" G_GSIZE_FORMAT " bytes of data", size);
2021       if (size >= wav->blockalign) {
2022         if (wav->blockalign > 0) {
2023           buf = gst_buffer_make_writable (buf);
2024           gst_buffer_resize (buf, 0, size - (size % wav->blockalign));
2025         }
2026       } else {
2027         gst_buffer_unref (buf);
2028         goto found_eos;
2029       }
2030     }
2031   }
2032
2033   obtained = gst_buffer_get_size (buf);
2034
2035   /* our positions in bytes */
2036   pos = wav->offset - wav->datastart;
2037   nextpos = pos + obtained;
2038
2039   /* update offsets, does not overflow. */
2040   buf = gst_buffer_make_writable (buf);
2041   GST_BUFFER_OFFSET (buf) = pos / wav->bytes_per_sample;
2042   GST_BUFFER_OFFSET_END (buf) = nextpos / wav->bytes_per_sample;
2043
2044   /* first chunk of data? create the source pad. We do this only here so
2045    * we can detect broken .wav files with dts disguised as raw PCM (sigh) */
2046   if (G_UNLIKELY (wav->first)) {
2047     wav->first = FALSE;
2048     /* this will also push the segment events */
2049     gst_wavparse_add_src_pad (wav, buf);
2050   } else {
2051     /* If we have a pending start segment, send it now. */
2052     if (G_UNLIKELY (wav->start_segment != NULL)) {
2053       gst_pad_push_event (wav->srcpad, wav->start_segment);
2054       wav->start_segment = NULL;
2055     }
2056   }
2057
2058   if (wav->bps > 0) {
2059     /* and timestamps if we have a bitrate, be careful for overflows */
2060     timestamp =
2061         gst_util_uint64_scale_ceil (pos, GST_SECOND, (guint64) wav->bps);
2062     next_timestamp =
2063         gst_util_uint64_scale_ceil (nextpos, GST_SECOND, (guint64) wav->bps);
2064     duration = next_timestamp - timestamp;
2065
2066     /* update current running segment position */
2067     if (G_LIKELY (next_timestamp >= wav->segment.start))
2068       wav->segment.position = next_timestamp;
2069   } else if (wav->fact) {
2070     guint64 bps =
2071         gst_util_uint64_scale_int (wav->datasize, wav->rate, wav->fact);
2072     /* and timestamps if we have a bitrate, be careful for overflows */
2073     timestamp = gst_util_uint64_scale_ceil (pos, GST_SECOND, bps);
2074     next_timestamp = gst_util_uint64_scale_ceil (nextpos, GST_SECOND, bps);
2075     duration = next_timestamp - timestamp;
2076   } else {
2077     /* no bitrate, all we know is that the first sample has timestamp 0, all
2078      * other positions and durations have unknown timestamp. */
2079     if (pos == 0)
2080       timestamp = 0;
2081     else
2082       timestamp = GST_CLOCK_TIME_NONE;
2083     duration = GST_CLOCK_TIME_NONE;
2084     /* update current running segment position with byte offset */
2085     if (G_LIKELY (nextpos >= wav->segment.start))
2086       wav->segment.position = nextpos;
2087   }
2088   if ((pos > 0) && wav->vbr) {
2089     /* don't set timestamps for VBR files if it's not the first buffer */
2090     timestamp = GST_CLOCK_TIME_NONE;
2091     duration = GST_CLOCK_TIME_NONE;
2092   }
2093   if (wav->discont) {
2094     GST_DEBUG_OBJECT (wav, "marking DISCONT");
2095     GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_DISCONT);
2096     wav->discont = FALSE;
2097   }
2098
2099   GST_BUFFER_TIMESTAMP (buf) = timestamp;
2100   GST_BUFFER_DURATION (buf) = duration;
2101
2102   GST_LOG_OBJECT (wav,
2103       "Got buffer. timestamp:%" GST_TIME_FORMAT " , duration:%" GST_TIME_FORMAT
2104       ", size:%" G_GSIZE_FORMAT, GST_TIME_ARGS (timestamp),
2105       GST_TIME_ARGS (duration), gst_buffer_get_size (buf));
2106
2107   if ((res = gst_pad_push (wav->srcpad, buf)) != GST_FLOW_OK)
2108     goto push_error;
2109
2110   if (obtained < wav->dataleft) {
2111     wav->offset += obtained;
2112     wav->dataleft -= obtained;
2113   } else {
2114     wav->offset += wav->dataleft;
2115     wav->dataleft = 0;
2116   }
2117
2118   /* Iterate until need more data, so adapter size won't grow */
2119   if (wav->streaming) {
2120     GST_LOG_OBJECT (wav,
2121         "offset: %" G_GINT64_FORMAT " , end: %" G_GINT64_FORMAT, wav->offset,
2122         wav->end_offset);
2123     goto iterate_adapter;
2124   }
2125   return res;
2126
2127   /* ERROR */
2128 found_eos:
2129   {
2130     GST_DEBUG_OBJECT (wav, "found EOS");
2131     return GST_FLOW_EOS;
2132   }
2133 pull_error:
2134   {
2135     /* check if we got EOS */
2136     if (res == GST_FLOW_EOS)
2137       goto found_eos;
2138
2139     GST_WARNING_OBJECT (wav,
2140         "Error getting %" G_GINT64_FORMAT " bytes from the "
2141         "sinkpad (dataleft = %" G_GINT64_FORMAT ")", desired, wav->dataleft);
2142     return res;
2143   }
2144 push_error:
2145   {
2146     GST_INFO_OBJECT (wav,
2147         "Error pushing on srcpad %s:%s, reason %s, is linked? = %d",
2148         GST_DEBUG_PAD_NAME (wav->srcpad), gst_flow_get_name (res),
2149         gst_pad_is_linked (wav->srcpad));
2150     return res;
2151   }
2152 }
2153
2154 static void
2155 gst_wavparse_loop (GstPad * pad)
2156 {
2157   GstFlowReturn ret;
2158   GstWavParse *wav = GST_WAVPARSE (GST_PAD_PARENT (pad));
2159   GstEvent *event;
2160   gchar *stream_id;
2161
2162   GST_LOG_OBJECT (wav, "process data");
2163
2164   switch (wav->state) {
2165     case GST_WAVPARSE_START:
2166       GST_INFO_OBJECT (wav, "GST_WAVPARSE_START");
2167       if ((ret = gst_wavparse_stream_init (wav)) != GST_FLOW_OK)
2168         goto pause;
2169
2170       stream_id =
2171           gst_pad_create_stream_id (wav->srcpad, GST_ELEMENT_CAST (wav), NULL);
2172       event = gst_event_new_stream_start (stream_id);
2173       gst_event_set_group_id (event, gst_util_group_id_next ());
2174       gst_pad_push_event (wav->srcpad, event);
2175       g_free (stream_id);
2176
2177       wav->state = GST_WAVPARSE_HEADER;
2178       /* fall-through */
2179
2180     case GST_WAVPARSE_HEADER:
2181       GST_INFO_OBJECT (wav, "GST_WAVPARSE_HEADER");
2182       if ((ret = gst_wavparse_stream_headers (wav)) != GST_FLOW_OK)
2183         goto pause;
2184
2185       wav->state = GST_WAVPARSE_DATA;
2186       GST_INFO_OBJECT (wav, "GST_WAVPARSE_DATA");
2187       /* fall-through */
2188
2189     case GST_WAVPARSE_DATA:
2190       if ((ret = gst_wavparse_stream_data (wav)) != GST_FLOW_OK)
2191         goto pause;
2192       break;
2193     default:
2194       g_assert_not_reached ();
2195   }
2196   return;
2197
2198   /* ERRORS */
2199 pause:
2200   {
2201     const gchar *reason = gst_flow_get_name (ret);
2202
2203     GST_DEBUG_OBJECT (wav, "pausing task, reason %s", reason);
2204     gst_pad_pause_task (pad);
2205
2206     if (ret == GST_FLOW_EOS) {
2207       /* handle end-of-stream/segment */
2208       /* so align our position with the end of it, if there is one
2209        * this ensures a subsequent will arrive at correct base/acc time */
2210       if (wav->segment.format == GST_FORMAT_TIME) {
2211         if (wav->segment.rate > 0.0 &&
2212             GST_CLOCK_TIME_IS_VALID (wav->segment.stop))
2213           wav->segment.position = wav->segment.stop;
2214         else if (wav->segment.rate < 0.0)
2215           wav->segment.position = wav->segment.start;
2216       }
2217       if (wav->state == GST_WAVPARSE_START) {
2218         GST_ELEMENT_ERROR (wav, STREAM, WRONG_TYPE, (NULL),
2219             ("No valid input found before end of stream"));
2220         gst_pad_push_event (wav->srcpad, gst_event_new_eos ());
2221       } else {
2222         /* add pad before we perform EOS */
2223         if (G_UNLIKELY (wav->first)) {
2224           wav->first = FALSE;
2225           gst_wavparse_add_src_pad (wav, NULL);
2226         }
2227
2228         /* perform EOS logic */
2229         if (wav->segment.flags & GST_SEEK_FLAG_SEGMENT) {
2230           GstClockTime stop;
2231
2232           if ((stop = wav->segment.stop) == -1)
2233             stop = wav->segment.duration;
2234
2235           gst_element_post_message (GST_ELEMENT_CAST (wav),
2236               gst_message_new_segment_done (GST_OBJECT_CAST (wav),
2237                   wav->segment.format, stop));
2238           gst_pad_push_event (wav->srcpad,
2239               gst_event_new_segment_done (wav->segment.format, stop));
2240         } else {
2241           gst_pad_push_event (wav->srcpad, gst_event_new_eos ());
2242         }
2243       }
2244     } else if (ret == GST_FLOW_NOT_LINKED || ret < GST_FLOW_EOS) {
2245       /* for fatal errors we post an error message, post the error
2246        * first so the app knows about the error first. */
2247       GST_ELEMENT_ERROR (wav, STREAM, FAILED,
2248           (_("Internal data flow error.")),
2249           ("streaming task paused, reason %s (%d)", reason, ret));
2250       gst_pad_push_event (wav->srcpad, gst_event_new_eos ());
2251     }
2252     return;
2253   }
2254 }
2255
2256 static GstFlowReturn
2257 gst_wavparse_chain (GstPad * pad, GstObject * parent, GstBuffer * buf)
2258 {
2259   GstFlowReturn ret;
2260   GstWavParse *wav = GST_WAVPARSE (parent);
2261
2262   GST_LOG_OBJECT (wav, "adapter_push %" G_GSIZE_FORMAT " bytes",
2263       gst_buffer_get_size (buf));
2264
2265   gst_adapter_push (wav->adapter, buf);
2266
2267   switch (wav->state) {
2268     case GST_WAVPARSE_START:
2269       GST_INFO_OBJECT (wav, "GST_WAVPARSE_START");
2270       if ((ret = gst_wavparse_parse_stream_init (wav)) != GST_FLOW_OK)
2271         goto done;
2272
2273       if (wav->state != GST_WAVPARSE_HEADER)
2274         break;
2275
2276       /* otherwise fall-through */
2277     case GST_WAVPARSE_HEADER:
2278       GST_INFO_OBJECT (wav, "GST_WAVPARSE_HEADER");
2279       if ((ret = gst_wavparse_stream_headers (wav)) != GST_FLOW_OK)
2280         goto done;
2281
2282       if (!wav->got_fmt || wav->datastart == 0)
2283         break;
2284
2285       wav->state = GST_WAVPARSE_DATA;
2286       GST_INFO_OBJECT (wav, "GST_WAVPARSE_DATA");
2287
2288       /* fall-through */
2289     case GST_WAVPARSE_DATA:
2290       if (buf && GST_BUFFER_FLAG_IS_SET (buf, GST_BUFFER_FLAG_DISCONT))
2291         wav->discont = TRUE;
2292       if ((ret = gst_wavparse_stream_data (wav)) != GST_FLOW_OK)
2293         goto done;
2294       break;
2295     default:
2296       g_return_val_if_reached (GST_FLOW_ERROR);
2297   }
2298 done:
2299   if (G_UNLIKELY (wav->abort_buffering)) {
2300     wav->abort_buffering = FALSE;
2301     ret = GST_FLOW_ERROR;
2302     /* sort of demux/parse error */
2303     GST_ELEMENT_ERROR (wav, STREAM, DEMUX, (NULL), ("unhandled buffer size"));
2304   }
2305
2306   return ret;
2307 }
2308
2309 static GstFlowReturn
2310 gst_wavparse_flush_data (GstWavParse * wav)
2311 {
2312   GstFlowReturn ret = GST_FLOW_OK;
2313   guint av;
2314
2315   if ((av = gst_adapter_available (wav->adapter)) > 0) {
2316     wav->dataleft = av;
2317     wav->end_offset = wav->offset + av;
2318     ret = gst_wavparse_stream_data (wav);
2319   }
2320
2321   return ret;
2322 }
2323
2324 static gboolean
2325 gst_wavparse_sink_event (GstPad * pad, GstObject * parent, GstEvent * event)
2326 {
2327   GstWavParse *wav = GST_WAVPARSE (parent);
2328   gboolean ret = TRUE;
2329
2330   GST_LOG_OBJECT (wav, "handling %s event", GST_EVENT_TYPE_NAME (event));
2331
2332   switch (GST_EVENT_TYPE (event)) {
2333     case GST_EVENT_CAPS:
2334     {
2335       /* discard, we'll come up with proper src caps */
2336       gst_event_unref (event);
2337       break;
2338     }
2339     case GST_EVENT_SEGMENT:
2340     {
2341       gint64 start, stop, offset = 0, end_offset = -1;
2342       GstSegment segment;
2343
2344       /* some debug output */
2345       gst_event_copy_segment (event, &segment);
2346       GST_DEBUG_OBJECT (wav, "received newsegment %" GST_SEGMENT_FORMAT,
2347           &segment);
2348
2349       if (wav->state != GST_WAVPARSE_DATA) {
2350         GST_DEBUG_OBJECT (wav, "still starting, eating event");
2351         goto exit;
2352       }
2353
2354       /* now we are either committed to TIME or BYTE format,
2355        * and we only expect a BYTE segment, e.g. following a seek */
2356       if (segment.format == GST_FORMAT_BYTES) {
2357         /* handle (un)signed issues */
2358         start = segment.start;
2359         stop = segment.stop;
2360         if (start > 0) {
2361           offset = start;
2362           start -= wav->datastart;
2363           start = MAX (start, 0);
2364         }
2365         if (stop > 0) {
2366           end_offset = stop;
2367           stop -= wav->datastart;
2368           stop = MAX (stop, 0);
2369         }
2370         if (wav->segment.format == GST_FORMAT_TIME) {
2371           guint64 bps = wav->bps;
2372
2373           /* operating in format TIME, so we can convert */
2374           if (!bps && wav->fact)
2375             bps =
2376                 gst_util_uint64_scale_int (wav->datasize, wav->rate, wav->fact);
2377           if (bps) {
2378             if (start >= 0)
2379               start =
2380                   gst_util_uint64_scale_ceil (start, GST_SECOND,
2381                   (guint64) wav->bps);
2382             if (stop >= 0)
2383               stop =
2384                   gst_util_uint64_scale_ceil (stop, GST_SECOND,
2385                   (guint64) wav->bps);
2386           }
2387         }
2388       } else {
2389         GST_DEBUG_OBJECT (wav, "unsupported segment format, ignoring");
2390         goto exit;
2391       }
2392
2393       segment.start = start;
2394       segment.stop = stop;
2395
2396       /* accept upstream's notion of segment and distribute along */
2397       segment.format = wav->segment.format;
2398       segment.time = segment.position = segment.start;
2399       segment.duration = wav->segment.duration;
2400       segment.base = gst_segment_to_running_time (&wav->segment,
2401           GST_FORMAT_TIME, wav->segment.position);
2402
2403       gst_segment_copy_into (&segment, &wav->segment);
2404
2405       /* also store the newsegment event for the streaming thread */
2406       if (wav->start_segment)
2407         gst_event_unref (wav->start_segment);
2408       GST_DEBUG_OBJECT (wav, "Storing newseg %" GST_SEGMENT_FORMAT, &segment);
2409       wav->start_segment = gst_event_new_segment (&segment);
2410
2411       /* stream leftover data in current segment */
2412       gst_wavparse_flush_data (wav);
2413       /* and set up streaming thread for next one */
2414       wav->offset = offset;
2415       wav->end_offset = end_offset;
2416       if (wav->end_offset > 0) {
2417         wav->dataleft = wav->end_offset - wav->offset;
2418       } else {
2419         /* infinity; upstream will EOS when done */
2420         wav->dataleft = G_MAXUINT64;
2421       }
2422     exit:
2423       gst_event_unref (event);
2424       break;
2425     }
2426     case GST_EVENT_EOS:
2427       if (wav->state == GST_WAVPARSE_START) {
2428         GST_ELEMENT_ERROR (wav, STREAM, WRONG_TYPE, (NULL),
2429             ("No valid input found before end of stream"));
2430       } else {
2431         /* add pad if needed so EOS is seen downstream */
2432         if (G_UNLIKELY (wav->first)) {
2433           wav->first = FALSE;
2434           gst_wavparse_add_src_pad (wav, NULL);
2435         } else {
2436           /* stream leftover data in current segment */
2437           gst_wavparse_flush_data (wav);
2438         }
2439       }
2440
2441       /* fall-through */
2442     case GST_EVENT_FLUSH_STOP:
2443     {
2444       GstClockTime dur;
2445
2446       gst_adapter_clear (wav->adapter);
2447       wav->discont = TRUE;
2448       dur = wav->segment.duration;
2449       gst_segment_init (&wav->segment, wav->segment.format);
2450       wav->segment.duration = dur;
2451       /* fall-through */
2452     }
2453     default:
2454       ret = gst_pad_event_default (wav->sinkpad, parent, event);
2455       break;
2456   }
2457
2458   return ret;
2459 }
2460
2461 #if 0
2462 /* convert and query stuff */
2463 static const GstFormat *
2464 gst_wavparse_get_formats (GstPad * pad)
2465 {
2466   static const GstFormat formats[] = {
2467     GST_FORMAT_TIME,
2468     GST_FORMAT_BYTES,
2469     GST_FORMAT_DEFAULT,         /* a "frame", ie a set of samples per Hz */
2470     0
2471   };
2472
2473   return formats;
2474 }
2475 #endif
2476
2477 static gboolean
2478 gst_wavparse_pad_convert (GstPad * pad,
2479     GstFormat src_format, gint64 src_value,
2480     GstFormat * dest_format, gint64 * dest_value)
2481 {
2482   GstWavParse *wavparse;
2483   gboolean res = TRUE;
2484
2485   wavparse = GST_WAVPARSE (GST_PAD_PARENT (pad));
2486
2487   if (*dest_format == src_format) {
2488     *dest_value = src_value;
2489     return TRUE;
2490   }
2491
2492   if ((wavparse->bps == 0) && !wavparse->fact)
2493     goto no_bps_fact;
2494
2495   GST_INFO_OBJECT (wavparse, "converting value from %s to %s",
2496       gst_format_get_name (src_format), gst_format_get_name (*dest_format));
2497
2498   switch (src_format) {
2499     case GST_FORMAT_BYTES:
2500       switch (*dest_format) {
2501         case GST_FORMAT_DEFAULT:
2502           *dest_value = src_value / wavparse->bytes_per_sample;
2503           /* make sure we end up on a sample boundary */
2504           *dest_value -= *dest_value % wavparse->bytes_per_sample;
2505           break;
2506         case GST_FORMAT_TIME:
2507           /* src_value + datastart = offset */
2508           GST_INFO_OBJECT (wavparse,
2509               "src=%" G_GINT64_FORMAT ", offset=%" G_GINT64_FORMAT, src_value,
2510               wavparse->offset);
2511           if (wavparse->bps > 0)
2512             *dest_value = gst_util_uint64_scale_ceil (src_value, GST_SECOND,
2513                 (guint64) wavparse->bps);
2514           else if (wavparse->fact) {
2515             guint64 bps = gst_util_uint64_scale_int_ceil (wavparse->datasize,
2516                 wavparse->rate, wavparse->fact);
2517
2518             *dest_value =
2519                 gst_util_uint64_scale_int_ceil (src_value, GST_SECOND, bps);
2520           } else {
2521             res = FALSE;
2522           }
2523           break;
2524         default:
2525           res = FALSE;
2526           goto done;
2527       }
2528       break;
2529
2530     case GST_FORMAT_DEFAULT:
2531       switch (*dest_format) {
2532         case GST_FORMAT_BYTES:
2533           *dest_value = src_value * wavparse->bytes_per_sample;
2534           break;
2535         case GST_FORMAT_TIME:
2536           *dest_value = gst_util_uint64_scale (src_value, GST_SECOND,
2537               (guint64) wavparse->rate);
2538           break;
2539         default:
2540           res = FALSE;
2541           goto done;
2542       }
2543       break;
2544
2545     case GST_FORMAT_TIME:
2546       switch (*dest_format) {
2547         case GST_FORMAT_BYTES:
2548           if (wavparse->bps > 0)
2549             *dest_value = gst_util_uint64_scale (src_value,
2550                 (guint64) wavparse->bps, GST_SECOND);
2551           else {
2552             guint64 bps = gst_util_uint64_scale_int (wavparse->datasize,
2553                 wavparse->rate, wavparse->fact);
2554
2555             *dest_value = gst_util_uint64_scale (src_value, bps, GST_SECOND);
2556           }
2557           /* make sure we end up on a sample boundary */
2558           *dest_value -= *dest_value % wavparse->blockalign;
2559           break;
2560         case GST_FORMAT_DEFAULT:
2561           *dest_value = gst_util_uint64_scale (src_value,
2562               (guint64) wavparse->rate, GST_SECOND);
2563           break;
2564         default:
2565           res = FALSE;
2566           goto done;
2567       }
2568       break;
2569
2570     default:
2571       res = FALSE;
2572       goto done;
2573   }
2574
2575 done:
2576   return res;
2577
2578   /* ERRORS */
2579 no_bps_fact:
2580   {
2581     GST_DEBUG_OBJECT (wavparse, "bps 0 or no fact chunk, cannot convert");
2582     res = FALSE;
2583     goto done;
2584   }
2585 }
2586
2587 /* handle queries for location and length in requested format */
2588 static gboolean
2589 gst_wavparse_pad_query (GstPad * pad, GstObject * parent, GstQuery * query)
2590 {
2591   gboolean res = TRUE;
2592   GstWavParse *wav = GST_WAVPARSE (parent);
2593
2594   /* only if we know */
2595   if (wav->state != GST_WAVPARSE_DATA) {
2596     return FALSE;
2597   }
2598
2599   GST_LOG_OBJECT (pad, "%s query", GST_QUERY_TYPE_NAME (query));
2600
2601   switch (GST_QUERY_TYPE (query)) {
2602     case GST_QUERY_POSITION:
2603     {
2604       gint64 curb;
2605       gint64 cur;
2606       GstFormat format;
2607
2608       /* this is not very precise, as we have pushed severla buffer upstream for prerolling */
2609       curb = wav->offset - wav->datastart;
2610       gst_query_parse_position (query, &format, NULL);
2611       GST_INFO_OBJECT (wav, "pos query at %" G_GINT64_FORMAT, curb);
2612
2613       switch (format) {
2614         case GST_FORMAT_BYTES:
2615           format = GST_FORMAT_BYTES;
2616           cur = curb;
2617           break;
2618         default:
2619           res = gst_wavparse_pad_convert (pad, GST_FORMAT_BYTES, curb,
2620               &format, &cur);
2621           break;
2622       }
2623       if (res)
2624         gst_query_set_position (query, format, cur);
2625       break;
2626     }
2627     case GST_QUERY_DURATION:
2628     {
2629       gint64 duration = 0;
2630       GstFormat format;
2631
2632       if (wav->ignore_length) {
2633         res = FALSE;
2634         break;
2635       }
2636
2637       gst_query_parse_duration (query, &format, NULL);
2638
2639       switch (format) {
2640         case GST_FORMAT_BYTES:{
2641           format = GST_FORMAT_BYTES;
2642           duration = wav->datasize;
2643           break;
2644         }
2645         case GST_FORMAT_TIME:
2646           if ((res = gst_wavparse_calculate_duration (wav))) {
2647             duration = wav->duration;
2648           }
2649           break;
2650         default:
2651           res = FALSE;
2652           break;
2653       }
2654       if (res)
2655         gst_query_set_duration (query, format, duration);
2656       break;
2657     }
2658     case GST_QUERY_CONVERT:
2659     {
2660       gint64 srcvalue, dstvalue;
2661       GstFormat srcformat, dstformat;
2662
2663       gst_query_parse_convert (query, &srcformat, &srcvalue,
2664           &dstformat, &dstvalue);
2665       res = gst_wavparse_pad_convert (pad, srcformat, srcvalue,
2666           &dstformat, &dstvalue);
2667       if (res)
2668         gst_query_set_convert (query, srcformat, srcvalue, dstformat, dstvalue);
2669       break;
2670     }
2671     case GST_QUERY_SEEKING:{
2672       GstFormat fmt;
2673       gboolean seekable = FALSE;
2674
2675       gst_query_parse_seeking (query, &fmt, NULL, NULL, NULL);
2676       if (fmt == wav->segment.format) {
2677         if (wav->streaming) {
2678           GstQuery *q;
2679
2680           q = gst_query_new_seeking (GST_FORMAT_BYTES);
2681           if ((res = gst_pad_peer_query (wav->sinkpad, q))) {
2682             gst_query_parse_seeking (q, &fmt, &seekable, NULL, NULL);
2683             GST_LOG_OBJECT (wav, "upstream BYTE seekable %d", seekable);
2684           }
2685           gst_query_unref (q);
2686         } else {
2687           GST_LOG_OBJECT (wav, "looping => seekable");
2688           seekable = TRUE;
2689           res = TRUE;
2690         }
2691       } else if (fmt == GST_FORMAT_TIME) {
2692         res = TRUE;
2693       }
2694       if (res) {
2695         gst_query_set_seeking (query, fmt, seekable, 0, wav->segment.duration);
2696       }
2697       break;
2698     }
2699     default:
2700       res = gst_pad_query_default (pad, parent, query);
2701       break;
2702   }
2703   return res;
2704 }
2705
2706 static gboolean
2707 gst_wavparse_srcpad_event (GstPad * pad, GstObject * parent, GstEvent * event)
2708 {
2709   GstWavParse *wavparse = GST_WAVPARSE (parent);
2710   gboolean res = FALSE;
2711
2712   GST_DEBUG_OBJECT (wavparse, "%s event", GST_EVENT_TYPE_NAME (event));
2713
2714   switch (GST_EVENT_TYPE (event)) {
2715     case GST_EVENT_SEEK:
2716       /* can only handle events when we are in the data state */
2717       if (wavparse->state == GST_WAVPARSE_DATA) {
2718         res = gst_wavparse_perform_seek (wavparse, event);
2719       }
2720       gst_event_unref (event);
2721       break;
2722
2723     case GST_EVENT_TOC_SELECT:
2724     {
2725       char *uid = NULL;
2726       GstTocEntry *entry = NULL;
2727       GstEvent *seek_event;
2728       gint64 start_pos;
2729
2730       if (!wavparse->toc) {
2731         GST_DEBUG_OBJECT (wavparse, "no TOC to select");
2732         return FALSE;
2733       } else {
2734         gst_event_parse_toc_select (event, &uid);
2735         if (uid != NULL) {
2736           GST_OBJECT_LOCK (wavparse);
2737           entry = gst_toc_find_entry (wavparse->toc, uid);
2738           if (entry == NULL) {
2739             GST_OBJECT_UNLOCK (wavparse);
2740             GST_WARNING_OBJECT (wavparse, "no TOC entry with given UID: %s",
2741                 uid);
2742             res = FALSE;
2743           } else {
2744             gst_toc_entry_get_start_stop_times (entry, &start_pos, NULL);
2745             GST_OBJECT_UNLOCK (wavparse);
2746             seek_event = gst_event_new_seek (1.0,
2747                 GST_FORMAT_TIME,
2748                 GST_SEEK_FLAG_FLUSH,
2749                 GST_SEEK_TYPE_SET, start_pos, GST_SEEK_TYPE_SET, -1);
2750             res = gst_wavparse_perform_seek (wavparse, seek_event);
2751             gst_event_unref (seek_event);
2752           }
2753           g_free (uid);
2754         } else {
2755           GST_WARNING_OBJECT (wavparse, "received empty TOC select event");
2756           res = FALSE;
2757         }
2758       }
2759       gst_event_unref (event);
2760       break;
2761     }
2762
2763     default:
2764       res = gst_pad_push_event (wavparse->sinkpad, event);
2765       break;
2766   }
2767   return res;
2768 }
2769
2770 static gboolean
2771 gst_wavparse_sink_activate (GstPad * sinkpad, GstObject * parent)
2772 {
2773   GstWavParse *wav = GST_WAVPARSE (parent);
2774   GstQuery *query;
2775   gboolean pull_mode;
2776
2777   if (wav->adapter) {
2778     gst_adapter_clear (wav->adapter);
2779     g_object_unref (wav->adapter);
2780     wav->adapter = NULL;
2781   }
2782
2783   query = gst_query_new_scheduling ();
2784
2785   if (!gst_pad_peer_query (sinkpad, query)) {
2786     gst_query_unref (query);
2787     goto activate_push;
2788   }
2789
2790   pull_mode = gst_query_has_scheduling_mode_with_flags (query,
2791       GST_PAD_MODE_PULL, GST_SCHEDULING_FLAG_SEEKABLE);
2792   gst_query_unref (query);
2793
2794   if (!pull_mode)
2795     goto activate_push;
2796
2797   GST_DEBUG_OBJECT (sinkpad, "activating pull");
2798   wav->streaming = FALSE;
2799   return gst_pad_activate_mode (sinkpad, GST_PAD_MODE_PULL, TRUE);
2800
2801 activate_push:
2802   {
2803     GST_DEBUG_OBJECT (sinkpad, "activating push");
2804     wav->streaming = TRUE;
2805     wav->adapter = gst_adapter_new ();
2806     return gst_pad_activate_mode (sinkpad, GST_PAD_MODE_PUSH, TRUE);
2807   }
2808 }
2809
2810
2811 static gboolean
2812 gst_wavparse_sink_activate_mode (GstPad * sinkpad, GstObject * parent,
2813     GstPadMode mode, gboolean active)
2814 {
2815   gboolean res;
2816
2817   switch (mode) {
2818     case GST_PAD_MODE_PUSH:
2819       res = TRUE;
2820       break;
2821     case GST_PAD_MODE_PULL:
2822       if (active) {
2823         /* if we have a scheduler we can start the task */
2824         res = gst_pad_start_task (sinkpad, (GstTaskFunction) gst_wavparse_loop,
2825             sinkpad, NULL);
2826       } else {
2827         res = gst_pad_stop_task (sinkpad);
2828       }
2829       break;
2830     default:
2831       res = FALSE;
2832       break;
2833   }
2834   return res;
2835 }
2836
2837 static GstStateChangeReturn
2838 gst_wavparse_change_state (GstElement * element, GstStateChange transition)
2839 {
2840   GstStateChangeReturn ret;
2841   GstWavParse *wav = GST_WAVPARSE (element);
2842
2843   switch (transition) {
2844     case GST_STATE_CHANGE_NULL_TO_READY:
2845       break;
2846     case GST_STATE_CHANGE_READY_TO_PAUSED:
2847       gst_wavparse_reset (wav);
2848       break;
2849     case GST_STATE_CHANGE_PAUSED_TO_PLAYING:
2850       break;
2851     default:
2852       break;
2853   }
2854
2855   ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
2856
2857   switch (transition) {
2858     case GST_STATE_CHANGE_PLAYING_TO_PAUSED:
2859       break;
2860     case GST_STATE_CHANGE_PAUSED_TO_READY:
2861       gst_wavparse_reset (wav);
2862       break;
2863     case GST_STATE_CHANGE_READY_TO_NULL:
2864       break;
2865     default:
2866       break;
2867   }
2868   return ret;
2869 }
2870
2871 static void
2872 gst_wavparse_set_property (GObject * object, guint prop_id,
2873     const GValue * value, GParamSpec * pspec)
2874 {
2875   GstWavParse *self;
2876
2877   g_return_if_fail (GST_IS_WAVPARSE (object));
2878   self = GST_WAVPARSE (object);
2879
2880   switch (prop_id) {
2881     case PROP_IGNORE_LENGTH:
2882       self->ignore_length = g_value_get_boolean (value);
2883       break;
2884     default:
2885       G_OBJECT_WARN_INVALID_PROPERTY_ID (self, prop_id, pspec);
2886   }
2887
2888 }
2889
2890 static void
2891 gst_wavparse_get_property (GObject * object, guint prop_id,
2892     GValue * value, GParamSpec * pspec)
2893 {
2894   GstWavParse *self;
2895
2896   g_return_if_fail (GST_IS_WAVPARSE (object));
2897   self = GST_WAVPARSE (object);
2898
2899   switch (prop_id) {
2900     case PROP_IGNORE_LENGTH:
2901       g_value_set_boolean (value, self->ignore_length);
2902       break;
2903     default:
2904       G_OBJECT_WARN_INVALID_PROPERTY_ID (self, prop_id, pspec);
2905   }
2906 }
2907
2908 static gboolean
2909 plugin_init (GstPlugin * plugin)
2910 {
2911   gst_riff_init ();
2912
2913   return gst_element_register (plugin, "wavparse", GST_RANK_PRIMARY,
2914       GST_TYPE_WAVPARSE);
2915 }
2916
2917 GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
2918     GST_VERSION_MINOR,
2919     wavparse,
2920     "Parse a .wav file into raw audio",
2921     plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)