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