gst/wavparse/gstwavparse.c

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