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., 59 Temple Place - Suite 330,
  19  * Boston, MA 02111-1307, 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 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 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_free (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   /* FIXME: send CURRENT scope toc too */
1287   toc = gst_toc_new (GST_TOC_SCOPE_GLOBAL);
1288
1289   /* add cue edition */
1290   entry = gst_toc_entry_new (GST_TOC_ENTRY_TYPE_EDITION, "cue");
1291   gst_toc_entry_set_start_stop_times (entry, 0, wav->duration);
1292   gst_toc_append_entry (toc, entry);
1293
1294   /* add chapters in cue edition */
1295   list = g_list_first (wav->cues);
1296   while (list != NULL) {
1297     cue = list->data;
1298     prev_subentry = cur_subentry;
1299     /* previous chapter stop time = current chapter start time */
1300     if (prev_subentry != NULL) {
1301       gst_toc_entry_get_start_stop_times (prev_subentry, &start, NULL);
1302       stop = gst_util_uint64_scale_round (cue->position, GST_SECOND, wav->rate);
1303       gst_toc_entry_set_start_stop_times (prev_subentry, start, stop);
1304     }
1305     id = g_strdup_printf ("%08x", cue->id);
1306     cur_subentry = gst_toc_entry_new (GST_TOC_ENTRY_TYPE_CHAPTER, id);
1307     g_free (id);
1308     start = gst_util_uint64_scale_round (cue->position, GST_SECOND, wav->rate);
1309     stop = wav->duration;
1310     gst_toc_entry_set_start_stop_times (cur_subentry, start, stop);
1311     gst_toc_entry_append_sub_entry (entry, cur_subentry);
1312     list = g_list_next (list);
1313   }
1314
1315   /* add tags in chapters */
1316   list = g_list_first (wav->labls);
1317   while (list != NULL) {
1318     labl = list->data;
1319     id = g_strdup_printf ("%08x", labl->cue_point_id);
1320     cur_subentry = gst_toc_find_entry (toc, id);
1321     g_free (id);
1322     if (cur_subentry != NULL) {
1323       tags = gst_tag_list_new_empty ();
1324       gst_tag_list_add (tags, GST_TAG_MERGE_APPEND, GST_TAG_TITLE, labl->text,
1325           NULL);
1326       gst_toc_entry_set_tags (cur_subentry, tags);
1327     }
1328     list = g_list_next (list);
1329   }
1330
1331   /* send data as TOC */
1332   wav->toc = toc;
1333
1334   /* send TOC event */
1335   if (wav->toc) {
1336     GST_OBJECT_UNLOCK (wav);
1337     gst_pad_push_event (wav->srcpad, gst_event_new_toc (wav->toc, FALSE));
1338   }
1339
1340   return TRUE;
1341 }
1342
1343 #define MAX_BUFFER_SIZE 4096
1344
1345 static GstFlowReturn
1346 gst_wavparse_stream_headers (GstWavParse * wav)
1347 {
1348   GstFlowReturn res = GST_FLOW_OK;
1349   GstBuffer *buf = NULL;
1350   gst_riff_strf_auds *header = NULL;
1351   guint32 tag, size;
1352   gboolean gotdata = FALSE;
1353   GstCaps *caps = NULL;
1354   gchar *codec_name = NULL;
1355   GstEvent **event_p;
1356   gint64 upstream_size = 0;
1357
1358   /* search for "_fmt" chunk, which should be first */
1359   while (!wav->got_fmt) {
1360     GstBuffer *extra;
1361
1362     /* The header starts with a 'fmt ' tag */
1363     if (wav->streaming) {
1364       if (!gst_wavparse_peek_chunk (wav, &tag, &size))
1365         return res;
1366
1367       gst_adapter_flush (wav->adapter, 8);
1368       wav->offset += 8;
1369
1370       if (size) {
1371         buf = gst_adapter_take_buffer (wav->adapter, size);
1372         if (size & 1)
1373           gst_adapter_flush (wav->adapter, 1);
1374         wav->offset += GST_ROUND_UP_2 (size);
1375       } else {
1376         buf = gst_buffer_new ();
1377       }
1378     } else {
1379       if ((res = gst_riff_read_chunk (GST_ELEMENT_CAST (wav), wav->sinkpad,
1380                   &wav->offset, &tag, &buf)) != GST_FLOW_OK)
1381         return res;
1382     }
1383
1384     if (tag == GST_RIFF_TAG_JUNK || tag == GST_RIFF_TAG_JUNQ ||
1385         tag == GST_RIFF_TAG_bext || tag == GST_RIFF_TAG_BEXT ||
1386         tag == GST_RIFF_TAG_LIST || tag == GST_RIFF_TAG_ID32 ||
1387         tag == GST_RIFF_TAG_IDVX) {
1388       GST_DEBUG_OBJECT (wav, "skipping %" GST_FOURCC_FORMAT " chunk",
1389           GST_FOURCC_ARGS (tag));
1390       gst_buffer_unref (buf);
1391       buf = NULL;
1392       continue;
1393     }
1394
1395     if (tag != GST_RIFF_TAG_fmt)
1396       goto invalid_wav;
1397
1398     if (!(gst_riff_parse_strf_auds (GST_ELEMENT_CAST (wav), buf, &header,
1399                 &extra)))
1400       goto parse_header_error;
1401
1402     buf = NULL;                 /* parse_strf_auds() took ownership of buffer */
1403
1404     /* do sanity checks of header fields */
1405     if (header->channels == 0)
1406       goto no_channels;
1407     if (header->rate == 0)
1408       goto no_rate;
1409
1410     GST_DEBUG_OBJECT (wav, "creating the caps");
1411
1412     /* Note: gst_riff_create_audio_caps might need to fix values in
1413      * the header header depending on the format, so call it first */
1414     /* FIXME: Need to handle the channel reorder map */
1415     caps = gst_riff_create_audio_caps (header->format, NULL, header, extra,
1416         NULL, &codec_name, NULL);
1417
1418     if (extra)
1419       gst_buffer_unref (extra);
1420
1421     if (!caps)
1422       goto unknown_format;
1423
1424     /* do more sanity checks of header fields
1425      * (these can be sanitized by gst_riff_create_audio_caps()
1426      */
1427     wav->format = header->format;
1428     wav->rate = header->rate;
1429     wav->channels = header->channels;
1430     wav->blockalign = header->blockalign;
1431     wav->depth = header->bits_per_sample;
1432     wav->av_bps = header->av_bps;
1433     wav->vbr = FALSE;
1434
1435     g_free (header);
1436     header = NULL;
1437
1438     /* do format specific handling */
1439     switch (wav->format) {
1440       case GST_RIFF_WAVE_FORMAT_MPEGL12:
1441       case GST_RIFF_WAVE_FORMAT_MPEGL3:
1442       {
1443         /* Note: workaround for mp2/mp3 embedded in wav, that relies on the
1444          * bitrate inside the mpeg stream */
1445         GST_INFO ("resetting bps from %u to 0 for mp2/3", wav->av_bps);
1446         wav->bps = 0;
1447         break;
1448       }
1449       case GST_RIFF_WAVE_FORMAT_PCM:
1450         if (wav->blockalign > wav->channels * ((wav->depth + 7) / 8))
1451           goto invalid_blockalign;
1452         /* fall through */
1453       default:
1454         if (wav->av_bps > wav->blockalign * wav->rate)
1455           goto invalid_bps;
1456         /* use the configured bps */
1457         wav->bps = wav->av_bps;
1458         break;
1459     }
1460
1461     wav->width = (wav->blockalign * 8) / wav->channels;
1462     wav->bytes_per_sample = wav->channels * wav->width / 8;
1463
1464     if (wav->bytes_per_sample <= 0)
1465       goto no_bytes_per_sample;
1466
1467     GST_DEBUG_OBJECT (wav, "blockalign = %u", (guint) wav->blockalign);
1468     GST_DEBUG_OBJECT (wav, "width      = %u", (guint) wav->width);
1469     GST_DEBUG_OBJECT (wav, "depth      = %u", (guint) wav->depth);
1470     GST_DEBUG_OBJECT (wav, "av_bps     = %u", (guint) wav->av_bps);
1471     GST_DEBUG_OBJECT (wav, "frequency  = %u", (guint) wav->rate);
1472     GST_DEBUG_OBJECT (wav, "channels   = %u", (guint) wav->channels);
1473     GST_DEBUG_OBJECT (wav, "bytes_per_sample = %u", wav->bytes_per_sample);
1474
1475     /* bps can be 0 when we don't have a valid bitrate (mostly for compressed
1476      * formats). This will make the element output a BYTE format segment and
1477      * will not timestamp the outgoing buffers.
1478      */
1479     GST_DEBUG_OBJECT (wav, "bps        = %u", (guint) wav->bps);
1480
1481     GST_DEBUG_OBJECT (wav, "caps = %" GST_PTR_FORMAT, caps);
1482
1483     /* create pad later so we can sniff the first few bytes
1484      * of the real data and correct our caps if necessary */
1485     gst_caps_replace (&wav->caps, caps);
1486     gst_caps_replace (&caps, NULL);
1487
1488     wav->got_fmt = TRUE;
1489
1490     if (codec_name) {
1491       wav->tags = gst_tag_list_new_empty ();
1492
1493       gst_tag_list_add (wav->tags, GST_TAG_MERGE_REPLACE,
1494           GST_TAG_AUDIO_CODEC, codec_name, NULL);
1495
1496       g_free (codec_name);
1497       codec_name = NULL;
1498     }
1499
1500   }
1501
1502   gst_pad_peer_query_duration (wav->sinkpad, GST_FORMAT_BYTES, &upstream_size);
1503   GST_DEBUG_OBJECT (wav, "upstream size %" G_GUINT64_FORMAT, upstream_size);
1504
1505   /* loop headers until we get data */
1506   while (!gotdata) {
1507     if (wav->streaming) {
1508       if (!gst_wavparse_peek_chunk_info (wav, &tag, &size))
1509         goto exit;
1510     } else {
1511       GstMapInfo map;
1512
1513       buf = NULL;
1514       if ((res =
1515               gst_pad_pull_range (wav->sinkpad, wav->offset, 8,
1516                   &buf)) != GST_FLOW_OK)
1517         goto header_read_error;
1518       gst_buffer_map (buf, &map, GST_MAP_READ);
1519       tag = GST_READ_UINT32_LE (map.data);
1520       size = GST_READ_UINT32_LE (map.data + 4);
1521       gst_buffer_unmap (buf, &map);
1522     }
1523
1524     GST_INFO_OBJECT (wav,
1525         "Got TAG: %" GST_FOURCC_FORMAT ", offset %" G_GUINT64_FORMAT,
1526         GST_FOURCC_ARGS (tag), wav->offset);
1527
1528     /* wav is a st00pid format, we don't know for sure where data starts.
1529      * So we have to go bit by bit until we find the 'data' header
1530      */
1531     switch (tag) {
1532       case GST_RIFF_TAG_data:{
1533         GST_DEBUG_OBJECT (wav, "Got 'data' TAG, size : %u", size);
1534         if (wav->ignore_length) {
1535           GST_DEBUG_OBJECT (wav, "Ignoring length");
1536           size = 0;
1537         }
1538         if (wav->streaming) {
1539           gst_adapter_flush (wav->adapter, 8);
1540           gotdata = TRUE;
1541         } else {
1542           gst_buffer_unref (buf);
1543         }
1544         wav->offset += 8;
1545         wav->datastart = wav->offset;
1546         /* If size is zero, then the data chunk probably actually extends to
1547            the end of the file */
1548         if (size == 0 && upstream_size) {
1549           size = upstream_size - wav->datastart;
1550         }
1551         /* Or the file might be truncated */
1552         else if (upstream_size) {
1553           size = MIN (size, (upstream_size - wav->datastart));
1554         }
1555         wav->datasize = (guint64) size;
1556         wav->dataleft = (guint64) size;
1557         wav->end_offset = size + wav->datastart;
1558         if (!wav->streaming) {
1559           /* We will continue parsing tags 'till end */
1560           wav->offset += size;
1561         }
1562         GST_DEBUG_OBJECT (wav, "datasize = %u", size);
1563         break;
1564       }
1565       case GST_RIFF_TAG_fact:{
1566         if (wav->format != GST_RIFF_WAVE_FORMAT_MPEGL12 &&
1567             wav->format != GST_RIFF_WAVE_FORMAT_MPEGL3) {
1568           const guint data_size = 4;
1569
1570           GST_INFO_OBJECT (wav, "Have fact chunk");
1571           if (size < data_size) {
1572             if (!gst_waveparse_ignore_chunk (wav, buf, tag, size)) {
1573               /* need more data */
1574               goto exit;
1575             }
1576             GST_DEBUG_OBJECT (wav, "need %u, available %u; ignoring chunk",
1577                 data_size, size);
1578             break;
1579           }
1580           /* number of samples (for compressed formats) */
1581           if (wav->streaming) {
1582             const guint8 *data = NULL;
1583
1584             if (!gst_wavparse_peek_chunk (wav, &tag, &size)) {
1585               goto exit;
1586             }
1587             gst_adapter_flush (wav->adapter, 8);
1588             data = gst_adapter_map (wav->adapter, data_size);
1589             wav->fact = GST_READ_UINT32_LE (data);
1590             gst_adapter_unmap (wav->adapter);
1591             gst_adapter_flush (wav->adapter, GST_ROUND_UP_2 (size));
1592           } else {
1593             gst_buffer_unref (buf);
1594             buf = NULL;
1595             if ((res =
1596                     gst_pad_pull_range (wav->sinkpad, wav->offset + 8,
1597                         data_size, &buf)) != GST_FLOW_OK)
1598               goto header_read_error;
1599             gst_buffer_extract (buf, 0, &wav->fact, 4);
1600             wav->fact = GUINT32_FROM_LE (wav->fact);
1601             gst_buffer_unref (buf);
1602           }
1603           GST_DEBUG_OBJECT (wav, "have fact %u", wav->fact);
1604           wav->offset += 8 + GST_ROUND_UP_2 (size);
1605           break;
1606         } else {
1607           if (!gst_waveparse_ignore_chunk (wav, buf, tag, size)) {
1608             /* need more data */
1609             goto exit;
1610           }
1611         }
1612         break;
1613       }
1614       case GST_RIFF_TAG_acid:{
1615         const gst_riff_acid *acid = NULL;
1616         const guint data_size = sizeof (gst_riff_acid);
1617         gfloat tempo;
1618
1619         GST_INFO_OBJECT (wav, "Have acid chunk");
1620         if (size < data_size) {
1621           if (!gst_waveparse_ignore_chunk (wav, buf, tag, size)) {
1622             /* need more data */
1623             goto exit;
1624           }
1625           GST_DEBUG_OBJECT (wav, "need %u, available %u; ignoring chunk",
1626               data_size, size);
1627           break;
1628         }
1629         if (wav->streaming) {
1630           if (!gst_wavparse_peek_chunk (wav, &tag, &size)) {
1631             goto exit;
1632           }
1633           gst_adapter_flush (wav->adapter, 8);
1634           acid = (const gst_riff_acid *) gst_adapter_map (wav->adapter,
1635               data_size);
1636           tempo = acid->tempo;
1637           gst_adapter_unmap (wav->adapter);
1638         } else {
1639           GstMapInfo map;
1640           gst_buffer_unref (buf);
1641           buf = NULL;
1642           if ((res =
1643                   gst_pad_pull_range (wav->sinkpad, wav->offset + 8,
1644                       size, &buf)) != GST_FLOW_OK)
1645             goto header_read_error;
1646           gst_buffer_map (buf, &map, GST_MAP_READ);
1647           acid = (const gst_riff_acid *) map.data;
1648           tempo = acid->tempo;
1649           gst_buffer_unmap (buf, &map);
1650         }
1651         /* send data as tags */
1652         if (!wav->tags)
1653           wav->tags = gst_tag_list_new_empty ();
1654         gst_tag_list_add (wav->tags, GST_TAG_MERGE_REPLACE,
1655             GST_TAG_BEATS_PER_MINUTE, tempo, NULL);
1656
1657         size = GST_ROUND_UP_2 (size);
1658         if (wav->streaming) {
1659           gst_adapter_flush (wav->adapter, size);
1660         } else {
1661           gst_buffer_unref (buf);
1662         }
1663         wav->offset += 8 + size;
1664         break;
1665       }
1666         /* FIXME: all list tags after data are ignored in streaming mode */
1667       case GST_RIFF_TAG_LIST:{
1668         guint32 ltag;
1669
1670         if (wav->streaming) {
1671           const guint8 *data = NULL;
1672
1673           if (gst_adapter_available (wav->adapter) < 12) {
1674             goto exit;
1675           }
1676           data = gst_adapter_map (wav->adapter, 12);
1677           ltag = GST_READ_UINT32_LE (data + 8);
1678           gst_adapter_unmap (wav->adapter);
1679         } else {
1680           gst_buffer_unref (buf);
1681           buf = NULL;
1682           if ((res =
1683                   gst_pad_pull_range (wav->sinkpad, wav->offset, 12,
1684                       &buf)) != GST_FLOW_OK)
1685             goto header_read_error;
1686           gst_buffer_extract (buf, 8, &ltag, 4);
1687           ltag = GUINT32_FROM_LE (ltag);
1688         }
1689         switch (ltag) {
1690           case GST_RIFF_LIST_INFO:{
1691             const gint data_size = size - 4;
1692             GstTagList *new;
1693
1694             GST_INFO_OBJECT (wav, "Have LIST chunk INFO size %u", data_size);
1695             if (wav->streaming) {
1696               if (!gst_wavparse_peek_chunk (wav, &tag, &size)) {
1697                 goto exit;
1698               }
1699               gst_adapter_flush (wav->adapter, 12);
1700               wav->offset += 12;
1701               if (data_size > 0) {
1702                 buf = gst_adapter_take_buffer (wav->adapter, data_size);
1703                 if (data_size & 1)
1704                   gst_adapter_flush (wav->adapter, 1);
1705               }
1706             } else {
1707               wav->offset += 12;
1708               gst_buffer_unref (buf);
1709               buf = NULL;
1710               if (data_size > 0) {
1711                 if ((res =
1712                         gst_pad_pull_range (wav->sinkpad, wav->offset,
1713                             data_size, &buf)) != GST_FLOW_OK)
1714                   goto header_read_error;
1715               }
1716             }
1717             if (data_size > 0) {
1718               /* parse tags */
1719               gst_riff_parse_info (GST_ELEMENT (wav), buf, &new);
1720               if (new) {
1721                 GstTagList *old = wav->tags;
1722                 wav->tags =
1723                     gst_tag_list_merge (old, new, GST_TAG_MERGE_REPLACE);
1724                 if (old)
1725                   gst_tag_list_free (old);
1726                 gst_tag_list_free (new);
1727               }
1728               gst_buffer_unref (buf);
1729               wav->offset += GST_ROUND_UP_2 (data_size);
1730             }
1731             break;
1732           }
1733           case GST_RIFF_LIST_adtl:{
1734             const gint data_size = size;
1735
1736             GST_INFO_OBJECT (wav, "Have 'adtl' LIST, size %u", data_size);
1737             if (wav->streaming) {
1738               const guint8 *data = NULL;
1739
1740               gst_adapter_flush (wav->adapter, 12);
1741               data = gst_adapter_map (wav->adapter, data_size);
1742               gst_wavparse_adtl_chunk (wav, data, data_size);
1743               gst_adapter_unmap (wav->adapter);
1744             } else {
1745               GstMapInfo map;
1746
1747               gst_buffer_unref (buf);
1748               buf = NULL;
1749               if ((res =
1750                       gst_pad_pull_range (wav->sinkpad, wav->offset + 12,
1751                           data_size, &buf)) != GST_FLOW_OK)
1752                 goto header_read_error;
1753               gst_buffer_map (buf, &map, GST_MAP_READ);
1754               gst_wavparse_adtl_chunk (wav, (const guint8 *) map.data,
1755                   data_size);
1756               gst_buffer_unmap (buf, &map);
1757             }
1758           }
1759           default:
1760             GST_INFO_OBJECT (wav, "Ignoring LIST chunk %" GST_FOURCC_FORMAT,
1761                 GST_FOURCC_ARGS (ltag));
1762             if (!gst_waveparse_ignore_chunk (wav, buf, tag, size))
1763               /* need more data */
1764               goto exit;
1765             break;
1766         }
1767         break;
1768       }
1769       case GST_RIFF_TAG_cue:{
1770         const guint data_size = size;
1771
1772         GST_DEBUG_OBJECT (wav, "Have 'cue' TAG, size : %u", data_size);
1773         if (wav->streaming) {
1774           const guint8 *data = NULL;
1775
1776           if (!gst_wavparse_peek_chunk (wav, &tag, &size)) {
1777             goto exit;
1778           }
1779           gst_adapter_flush (wav->adapter, 8);
1780           wav->offset += 8;
1781           data = gst_adapter_map (wav->adapter, data_size);
1782           if (!gst_wavparse_cue_chunk (wav, data, data_size)) {
1783             goto header_read_error;
1784           }
1785           gst_adapter_unmap (wav->adapter);
1786         } else {
1787           GstMapInfo map;
1788
1789           wav->offset += 8;
1790           gst_buffer_unref (buf);
1791           buf = NULL;
1792           if ((res =
1793                   gst_pad_pull_range (wav->sinkpad, wav->offset,
1794                       data_size, &buf)) != GST_FLOW_OK)
1795             goto header_read_error;
1796           gst_buffer_map (buf, &map, GST_MAP_READ);
1797           if (!gst_wavparse_cue_chunk (wav, (const guint8 *) map.data,
1798                   data_size)) {
1799             goto header_read_error;
1800           }
1801           gst_buffer_unmap (buf, &map);
1802         }
1803         size = GST_ROUND_UP_2 (size);
1804         if (wav->streaming) {
1805           gst_adapter_flush (wav->adapter, size);
1806         } else {
1807           gst_buffer_unref (buf);
1808         }
1809         size = GST_ROUND_UP_2 (size);
1810         wav->offset += size;
1811         break;
1812       }
1813       default:
1814         if (!gst_waveparse_ignore_chunk (wav, buf, tag, size))
1815           /* need more data */
1816           goto exit;
1817         break;
1818     }
1819
1820     if (upstream_size && (wav->offset >= upstream_size)) {
1821       /* Now we are gone through the whole file */
1822       gotdata = TRUE;
1823     }
1824   }
1825
1826   GST_DEBUG_OBJECT (wav, "Finished parsing headers");
1827
1828   if (wav->bps <= 0 && wav->fact) {
1829 #if 0
1830     /* not a good idea, as for embedded mp2/mp3 we set bps to 0 earlier */
1831     wav->bps =
1832         (guint32) gst_util_uint64_scale ((guint64) wav->rate, wav->datasize,
1833         (guint64) wav->fact);
1834     GST_INFO_OBJECT (wav, "calculated bps : %u, enabling VBR", wav->bps);
1835 #endif
1836     wav->vbr = TRUE;
1837   }
1838
1839   if (gst_wavparse_calculate_duration (wav)) {
1840     gst_segment_init (&wav->segment, GST_FORMAT_TIME);
1841     if (!wav->ignore_length)
1842       wav->segment.duration = wav->duration;
1843     if (!wav->toc)
1844       gst_wavparse_create_toc (wav);
1845   } else {
1846     /* no bitrate, let downstream peer do the math, we'll feed it bytes. */
1847     gst_segment_init (&wav->segment, GST_FORMAT_BYTES);
1848     if (!wav->ignore_length)
1849       wav->segment.duration = wav->datasize;
1850   }
1851
1852   /* now we have all the info to perform a pending seek if any, if no
1853    * event, this will still do the right thing and it will also send
1854    * the right newsegment event downstream. */
1855   gst_wavparse_perform_seek (wav, wav->seek_event);
1856   /* remove pending event */
1857   event_p = &wav->seek_event;
1858   gst_event_replace (event_p, NULL);
1859
1860   /* we just started, we are discont */
1861   wav->discont = TRUE;
1862
1863   wav->state = GST_WAVPARSE_DATA;
1864
1865   /* determine reasonable max buffer size,
1866    * that is, buffers not too small either size or time wise
1867    * so we do not end up with too many of them */
1868   /* var abuse */
1869   upstream_size = 0;
1870   gst_wavparse_time_to_bytepos (wav, 40 * GST_MSECOND, &upstream_size);
1871   wav->max_buf_size = upstream_size;
1872   wav->max_buf_size = MAX (wav->max_buf_size, MAX_BUFFER_SIZE);
1873   if (wav->blockalign > 0)
1874     wav->max_buf_size -= (wav->max_buf_size % wav->blockalign);
1875
1876   GST_DEBUG_OBJECT (wav, "max buffer size %u", wav->max_buf_size);
1877
1878   return GST_FLOW_OK;
1879
1880   /* ERROR */
1881 exit:
1882   {
1883     if (codec_name)
1884       g_free (codec_name);
1885     if (header)
1886       g_free (header);
1887     if (caps)
1888       gst_caps_unref (caps);
1889     return res;
1890   }
1891 fail:
1892   {
1893     res = GST_FLOW_ERROR;
1894     goto exit;
1895   }
1896 invalid_wav:
1897   {
1898     GST_ELEMENT_ERROR (wav, STREAM, TYPE_NOT_FOUND, (NULL),
1899         ("Invalid WAV header (no fmt at start): %"
1900             GST_FOURCC_FORMAT, GST_FOURCC_ARGS (tag)));
1901     goto fail;
1902   }
1903 parse_header_error:
1904   {
1905     GST_ELEMENT_ERROR (wav, STREAM, DEMUX, (NULL),
1906         ("Couldn't parse audio header"));
1907     goto fail;
1908   }
1909 no_channels:
1910   {
1911     GST_ELEMENT_ERROR (wav, STREAM, FAILED, (NULL),
1912         ("Stream claims to contain no channels - invalid data"));
1913     goto fail;
1914   }
1915 no_rate:
1916   {
1917     GST_ELEMENT_ERROR (wav, STREAM, FAILED, (NULL),
1918         ("Stream with sample_rate == 0 - invalid data"));
1919     goto fail;
1920   }
1921 invalid_blockalign:
1922   {
1923     GST_ELEMENT_ERROR (wav, STREAM, FAILED, (NULL),
1924         ("Stream claims blockalign = %u, which is more than %u - invalid data",
1925             wav->blockalign, wav->channels * ((wav->depth + 7) / 8)));
1926     goto fail;
1927   }
1928 invalid_bps:
1929   {
1930     GST_ELEMENT_ERROR (wav, STREAM, FAILED, (NULL),
1931         ("Stream claims av_bsp = %u, which is more than %u - invalid data",
1932             wav->av_bps, wav->blockalign * wav->rate));
1933     goto fail;
1934   }
1935 no_bytes_per_sample:
1936   {
1937     GST_ELEMENT_ERROR (wav, STREAM, FAILED, (NULL),
1938         ("Could not caluclate bytes per sample - invalid data"));
1939     goto fail;
1940   }
1941 unknown_format:
1942   {
1943     GST_ELEMENT_ERROR (wav, STREAM, TYPE_NOT_FOUND, (NULL),
1944         ("No caps found for format 0x%x, %u channels, %u Hz",
1945             wav->format, wav->channels, wav->rate));
1946     goto fail;
1947   }
1948 header_read_error:
1949   {
1950     GST_ELEMENT_ERROR (wav, STREAM, DEMUX, (NULL),
1951         ("Couldn't read in header %d (%s)", res, gst_flow_get_name (res)));
1952     goto fail;
1953   }
1954 }
1955
1956 /*
1957  * Read WAV file tag when streaming
1958  */
1959 static GstFlowReturn
1960 gst_wavparse_parse_stream_init (GstWavParse * wav)
1961 {
1962   if (gst_adapter_available (wav->adapter) >= 12) {
1963     GstBuffer *tmp;
1964
1965     /* _take flushes the data */
1966     tmp = gst_adapter_take_buffer (wav->adapter, 12);
1967
1968     GST_DEBUG ("Parsing wav header");
1969     if (!gst_wavparse_parse_file_header (GST_ELEMENT_CAST (wav), tmp))
1970       return GST_FLOW_ERROR;
1971
1972     wav->offset += 12;
1973     /* Go to next state */
1974     wav->state = GST_WAVPARSE_HEADER;
1975   }
1976   return GST_FLOW_OK;
1977 }
1978
1979 /* handle an event sent directly to the element.
1980  *
1981  * This event can be sent either in the READY state or the
1982  * >READY state. The only event of interest really is the seek
1983  * event.
1984  *
1985  * In the READY state we can only store the event and try to
1986  * respect it when going to PAUSED. We assume we are in the
1987  * READY state when our parsing state != GST_WAVPARSE_DATA.
1988  *
1989  * When we are steaming, we can simply perform the seek right
1990  * away.
1991  */
1992 static gboolean
1993 gst_wavparse_send_event (GstElement * element, GstEvent * event)
1994 {
1995   GstWavParse *wav = GST_WAVPARSE (element);
1996   gboolean res = FALSE;
1997   GstEvent **event_p;
1998
1999   GST_DEBUG_OBJECT (wav, "received event %s", GST_EVENT_TYPE_NAME (event));
2000
2001   switch (GST_EVENT_TYPE (event)) {
2002     case GST_EVENT_SEEK:
2003       if (wav->state == GST_WAVPARSE_DATA) {
2004         /* we can handle the seek directly when streaming data */
2005         res = gst_wavparse_perform_seek (wav, event);
2006       } else {
2007         GST_DEBUG_OBJECT (wav, "queuing seek for later");
2008
2009         event_p = &wav->seek_event;
2010         gst_event_replace (event_p, event);
2011
2012         /* we always return true */
2013         res = TRUE;
2014       }
2015       break;
2016     default:
2017       break;
2018   }
2019   gst_event_unref (event);
2020   return res;
2021 }
2022
2023 static gboolean
2024 gst_wavparse_have_dts_caps (const GstCaps * caps, GstTypeFindProbability prob)
2025 {
2026   GstStructure *s;
2027
2028   s = gst_caps_get_structure (caps, 0);
2029   if (!gst_structure_has_name (s, "audio/x-dts"))
2030     return FALSE;
2031   if (prob >= GST_TYPE_FIND_LIKELY)
2032     return TRUE;
2033   /* DTS at non-0 offsets and without second sync may yield POSSIBLE .. */
2034   if (prob < GST_TYPE_FIND_POSSIBLE)
2035     return FALSE;
2036   /* .. in which case we want at least a valid-looking rate and channels */
2037   if (!gst_structure_has_field (s, "channels"))
2038     return FALSE;
2039   /* and for extra assurance we could also check the rate from the DTS frame
2040    * against the one in the wav header, but for now let's not do that */
2041   return gst_structure_has_field (s, "rate");
2042 }
2043
2044 static void
2045 gst_wavparse_add_src_pad (GstWavParse * wav, GstBuffer * buf)
2046 {
2047   GstStructure *s;
2048
2049   GST_DEBUG_OBJECT (wav, "adding src pad");
2050
2051   if (wav->caps) {
2052     s = gst_caps_get_structure (wav->caps, 0);
2053     if (s && gst_structure_has_name (s, "audio/x-raw") && buf != NULL) {
2054       GstTypeFindProbability prob;
2055       GstCaps *tf_caps;
2056
2057       tf_caps = gst_type_find_helper_for_buffer (GST_OBJECT (wav), buf, &prob);
2058       if (tf_caps != NULL) {
2059         GST_LOG ("typefind caps = %" GST_PTR_FORMAT ", P=%d", tf_caps, prob);
2060         if (gst_wavparse_have_dts_caps (tf_caps, prob)) {
2061           GST_INFO_OBJECT (wav, "Found DTS marker in file marked as raw PCM");
2062           gst_caps_unref (wav->caps);
2063           wav->caps = tf_caps;
2064
2065           gst_tag_list_add (wav->tags, GST_TAG_MERGE_REPLACE,
2066               GST_TAG_AUDIO_CODEC, "dts", NULL);
2067         } else {
2068           GST_DEBUG_OBJECT (wav, "found caps %" GST_PTR_FORMAT " for stream "
2069               "marked as raw PCM audio, but ignoring for now", tf_caps);
2070           gst_caps_unref (tf_caps);
2071         }
2072       }
2073     }
2074   }
2075
2076   gst_pad_set_caps (wav->srcpad, wav->caps);
2077   gst_caps_replace (&wav->caps, NULL);
2078
2079   if (wav->start_segment) {
2080     GST_DEBUG_OBJECT (wav, "Send start segment event on newpad");
2081     gst_pad_push_event (wav->srcpad, wav->start_segment);
2082     wav->start_segment = NULL;
2083   }
2084
2085   if (wav->tags) {
2086     gst_pad_push_event (wav->srcpad, gst_event_new_tag (wav->tags));
2087     wav->tags = NULL;
2088   }
2089 }
2090
2091 static GstFlowReturn
2092 gst_wavparse_stream_data (GstWavParse * wav)
2093 {
2094   GstBuffer *buf = NULL;
2095   GstFlowReturn res = GST_FLOW_OK;
2096   guint64 desired, obtained;
2097   GstClockTime timestamp, next_timestamp, duration;
2098   guint64 pos, nextpos;
2099
2100 iterate_adapter:
2101   GST_LOG_OBJECT (wav,
2102       "offset: %" G_GINT64_FORMAT " , end: %" G_GINT64_FORMAT " , dataleft: %"
2103       G_GINT64_FORMAT, wav->offset, wav->end_offset, wav->dataleft);
2104
2105   /* Get the next n bytes and output them */
2106   if (wav->dataleft == 0 || wav->dataleft < wav->blockalign)
2107     goto found_eos;
2108
2109   /* scale the amount of data by the segment rate so we get equal
2110    * amounts of data regardless of the playback rate */
2111   desired =
2112       MIN (gst_guint64_to_gdouble (wav->dataleft),
2113       wav->max_buf_size * ABS (wav->segment.rate));
2114
2115   if (desired >= wav->blockalign && wav->blockalign > 0)
2116     desired -= (desired % wav->blockalign);
2117
2118   GST_LOG_OBJECT (wav, "Fetching %" G_GINT64_FORMAT " bytes of data "
2119       "from the sinkpad", desired);
2120
2121   if (wav->streaming) {
2122     guint avail = gst_adapter_available (wav->adapter);
2123     guint extra;
2124
2125     /* flush some bytes if evil upstream sends segment that starts
2126      * before data or does is not send sample aligned segment */
2127     if (G_LIKELY (wav->offset >= wav->datastart)) {
2128       extra = (wav->offset - wav->datastart) % wav->bytes_per_sample;
2129     } else {
2130       extra = wav->datastart - wav->offset;
2131     }
2132
2133     if (G_UNLIKELY (extra)) {
2134       extra = wav->bytes_per_sample - extra;
2135       if (extra <= avail) {
2136         GST_DEBUG_OBJECT (wav, "flushing %u bytes to sample boundary", extra);
2137         gst_adapter_flush (wav->adapter, extra);
2138         wav->offset += extra;
2139         wav->dataleft -= extra;
2140         goto iterate_adapter;
2141       } else {
2142         GST_DEBUG_OBJECT (wav, "flushing %u bytes", avail);
2143         gst_adapter_clear (wav->adapter);
2144         wav->offset += avail;
2145         wav->dataleft -= avail;
2146         return GST_FLOW_OK;
2147       }
2148     }
2149
2150     if (avail < desired) {
2151       GST_LOG_OBJECT (wav, "Got only %u bytes of data from the sinkpad", avail);
2152       return GST_FLOW_OK;
2153     }
2154
2155     buf = gst_adapter_take_buffer (wav->adapter, desired);
2156   } else {
2157     if ((res = gst_pad_pull_range (wav->sinkpad, wav->offset,
2158                 desired, &buf)) != GST_FLOW_OK)
2159       goto pull_error;
2160
2161     /* we may get a short buffer at the end of the file */
2162     if (gst_buffer_get_size (buf) < desired) {
2163       gsize size = gst_buffer_get_size (buf);
2164
2165       GST_LOG_OBJECT (wav, "Got only %" G_GSIZE_FORMAT " bytes of data", size);
2166       if (size >= wav->blockalign) {
2167         buf = gst_buffer_make_writable (buf);
2168         gst_buffer_resize (buf, 0, size - (size % wav->blockalign));
2169       } else {
2170         gst_buffer_unref (buf);
2171         goto found_eos;
2172       }
2173     }
2174   }
2175
2176   obtained = gst_buffer_get_size (buf);
2177
2178   /* our positions in bytes */
2179   pos = wav->offset - wav->datastart;
2180   nextpos = pos + obtained;
2181
2182   /* update offsets, does not overflow. */
2183   buf = gst_buffer_make_writable (buf);
2184   GST_BUFFER_OFFSET (buf) = pos / wav->bytes_per_sample;
2185   GST_BUFFER_OFFSET_END (buf) = nextpos / wav->bytes_per_sample;
2186
2187   /* first chunk of data? create the source pad. We do this only here so
2188    * we can detect broken .wav files with dts disguised as raw PCM (sigh) */
2189   if (G_UNLIKELY (wav->first)) {
2190     wav->first = FALSE;
2191     /* this will also push the segment events */
2192     gst_wavparse_add_src_pad (wav, buf);
2193   } else {
2194     /* If we have a pending start segment, send it now. */
2195     if (G_UNLIKELY (wav->start_segment != NULL)) {
2196       gst_pad_push_event (wav->srcpad, wav->start_segment);
2197       wav->start_segment = NULL;
2198     }
2199   }
2200
2201   if (wav->bps > 0) {
2202     /* and timestamps if we have a bitrate, be careful for overflows */
2203     timestamp =
2204         gst_util_uint64_scale_ceil (pos, GST_SECOND, (guint64) wav->bps);
2205     next_timestamp =
2206         gst_util_uint64_scale_ceil (nextpos, GST_SECOND, (guint64) wav->bps);
2207     duration = next_timestamp - timestamp;
2208
2209     /* update current running segment position */
2210     if (G_LIKELY (next_timestamp >= wav->segment.start))
2211       wav->segment.position = next_timestamp;
2212   } else if (wav->fact) {
2213     guint64 bps =
2214         gst_util_uint64_scale_int (wav->datasize, wav->rate, wav->fact);
2215     /* and timestamps if we have a bitrate, be careful for overflows */
2216     timestamp = gst_util_uint64_scale_ceil (pos, GST_SECOND, bps);
2217     next_timestamp = gst_util_uint64_scale_ceil (nextpos, GST_SECOND, bps);
2218     duration = next_timestamp - timestamp;
2219   } else {
2220     /* no bitrate, all we know is that the first sample has timestamp 0, all
2221      * other positions and durations have unknown timestamp. */
2222     if (pos == 0)
2223       timestamp = 0;
2224     else
2225       timestamp = GST_CLOCK_TIME_NONE;
2226     duration = GST_CLOCK_TIME_NONE;
2227     /* update current running segment position with byte offset */
2228     if (G_LIKELY (nextpos >= wav->segment.start))
2229       wav->segment.position = nextpos;
2230   }
2231   if ((pos > 0) && wav->vbr) {
2232     /* don't set timestamps for VBR files if it's not the first buffer */
2233     timestamp = GST_CLOCK_TIME_NONE;
2234     duration = GST_CLOCK_TIME_NONE;
2235   }
2236   if (wav->discont) {
2237     GST_DEBUG_OBJECT (wav, "marking DISCONT");
2238     GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_DISCONT);
2239     wav->discont = FALSE;
2240   }
2241
2242   GST_BUFFER_TIMESTAMP (buf) = timestamp;
2243   GST_BUFFER_DURATION (buf) = duration;
2244
2245   GST_LOG_OBJECT (wav,
2246       "Got buffer. timestamp:%" GST_TIME_FORMAT " , duration:%" GST_TIME_FORMAT
2247       ", size:%" G_GSIZE_FORMAT, GST_TIME_ARGS (timestamp),
2248       GST_TIME_ARGS (duration), gst_buffer_get_size (buf));
2249
2250   if ((res = gst_pad_push (wav->srcpad, buf)) != GST_FLOW_OK)
2251     goto push_error;
2252
2253   if (obtained < wav->dataleft) {
2254     wav->offset += obtained;
2255     wav->dataleft -= obtained;
2256   } else {
2257     wav->offset += wav->dataleft;
2258     wav->dataleft = 0;
2259   }
2260
2261   /* Iterate until need more data, so adapter size won't grow */
2262   if (wav->streaming) {
2263     GST_LOG_OBJECT (wav,
2264         "offset: %" G_GINT64_FORMAT " , end: %" G_GINT64_FORMAT, wav->offset,
2265         wav->end_offset);
2266     goto iterate_adapter;
2267   }
2268   return res;
2269
2270   /* ERROR */
2271 found_eos:
2272   {
2273     GST_DEBUG_OBJECT (wav, "found EOS");
2274     return GST_FLOW_EOS;
2275   }
2276 pull_error:
2277   {
2278     /* check if we got EOS */
2279     if (res == GST_FLOW_EOS)
2280       goto found_eos;
2281
2282     GST_WARNING_OBJECT (wav,
2283         "Error getting %" G_GINT64_FORMAT " bytes from the "
2284         "sinkpad (dataleft = %" G_GINT64_FORMAT ")", desired, wav->dataleft);
2285     return res;
2286   }
2287 push_error:
2288   {
2289     GST_INFO_OBJECT (wav,
2290         "Error pushing on srcpad %s:%s, reason %s, is linked? = %d",
2291         GST_DEBUG_PAD_NAME (wav->srcpad), gst_flow_get_name (res),
2292         gst_pad_is_linked (wav->srcpad));
2293     return res;
2294   }
2295 }
2296
2297 static void
2298 gst_wavparse_loop (GstPad * pad)
2299 {
2300   GstFlowReturn ret;
2301   GstWavParse *wav = GST_WAVPARSE (GST_PAD_PARENT (pad));
2302
2303   GST_LOG_OBJECT (wav, "process data");
2304
2305   switch (wav->state) {
2306     case GST_WAVPARSE_START:
2307       GST_INFO_OBJECT (wav, "GST_WAVPARSE_START");
2308       if ((ret = gst_wavparse_stream_init (wav)) != GST_FLOW_OK)
2309         goto pause;
2310
2311       wav->state = GST_WAVPARSE_HEADER;
2312       /* fall-through */
2313
2314     case GST_WAVPARSE_HEADER:
2315       GST_INFO_OBJECT (wav, "GST_WAVPARSE_HEADER");
2316       if ((ret = gst_wavparse_stream_headers (wav)) != GST_FLOW_OK)
2317         goto pause;
2318
2319       wav->state = GST_WAVPARSE_DATA;
2320       GST_INFO_OBJECT (wav, "GST_WAVPARSE_DATA");
2321       /* fall-through */
2322
2323     case GST_WAVPARSE_DATA:
2324       if ((ret = gst_wavparse_stream_data (wav)) != GST_FLOW_OK)
2325         goto pause;
2326       break;
2327     default:
2328       g_assert_not_reached ();
2329   }
2330   return;
2331
2332   /* ERRORS */
2333 pause:
2334   {
2335     const gchar *reason = gst_flow_get_name (ret);
2336
2337     GST_DEBUG_OBJECT (wav, "pausing task, reason %s", reason);
2338     gst_pad_pause_task (pad);
2339
2340     if (ret == GST_FLOW_EOS) {
2341       /* handle end-of-stream/segment */
2342       /* so align our position with the end of it, if there is one
2343        * this ensures a subsequent will arrive at correct base/acc time */
2344       if (wav->segment.format == GST_FORMAT_TIME) {
2345         if (wav->segment.rate > 0.0 &&
2346             GST_CLOCK_TIME_IS_VALID (wav->segment.stop))
2347           wav->segment.position = wav->segment.stop;
2348         else if (wav->segment.rate < 0.0)
2349           wav->segment.position = wav->segment.start;
2350       }
2351       /* add pad before we perform EOS */
2352       if (G_UNLIKELY (wav->first)) {
2353         wav->first = FALSE;
2354         gst_wavparse_add_src_pad (wav, NULL);
2355       }
2356
2357       if (wav->state == GST_WAVPARSE_START)
2358         GST_ELEMENT_ERROR (wav, STREAM, WRONG_TYPE,
2359             ("No valid input found before end of stream"), (NULL));
2360
2361       /* perform EOS logic */
2362       if (wav->segment.flags & GST_SEEK_FLAG_SEGMENT) {
2363         GstClockTime stop;
2364
2365         if ((stop = wav->segment.stop) == -1)
2366           stop = wav->segment.duration;
2367
2368         gst_element_post_message (GST_ELEMENT_CAST (wav),
2369             gst_message_new_segment_done (GST_OBJECT_CAST (wav),
2370                 wav->segment.format, stop));
2371         gst_pad_push_event (wav->srcpad,
2372             gst_event_new_segment_done (wav->segment.format, stop));
2373       } else {
2374         gst_pad_push_event (wav->srcpad, gst_event_new_eos ());
2375       }
2376     } else if (ret == GST_FLOW_NOT_LINKED || ret < GST_FLOW_EOS) {
2377       /* for fatal errors we post an error message, post the error
2378        * first so the app knows about the error first. */
2379       GST_ELEMENT_ERROR (wav, STREAM, FAILED,
2380           (_("Internal data flow error.")),
2381           ("streaming task paused, reason %s (%d)", reason, ret));
2382       gst_pad_push_event (wav->srcpad, gst_event_new_eos ());
2383     }
2384     return;
2385   }
2386 }
2387
2388 static GstFlowReturn
2389 gst_wavparse_chain (GstPad * pad, GstObject * parent, GstBuffer * buf)
2390 {
2391   GstFlowReturn ret;
2392   GstWavParse *wav = GST_WAVPARSE (parent);
2393
2394   GST_LOG_OBJECT (wav, "adapter_push %" G_GSIZE_FORMAT " bytes",
2395       gst_buffer_get_size (buf));
2396
2397   gst_adapter_push (wav->adapter, buf);
2398
2399   switch (wav->state) {
2400     case GST_WAVPARSE_START:
2401       GST_INFO_OBJECT (wav, "GST_WAVPARSE_START");
2402       if ((ret = gst_wavparse_parse_stream_init (wav)) != GST_FLOW_OK)
2403         goto done;
2404
2405       if (wav->state != GST_WAVPARSE_HEADER)
2406         break;
2407
2408       /* otherwise fall-through */
2409     case GST_WAVPARSE_HEADER:
2410       GST_INFO_OBJECT (wav, "GST_WAVPARSE_HEADER");
2411       if ((ret = gst_wavparse_stream_headers (wav)) != GST_FLOW_OK)
2412         goto done;
2413
2414       if (!wav->got_fmt || wav->datastart == 0)
2415         break;
2416
2417       wav->state = GST_WAVPARSE_DATA;
2418       GST_INFO_OBJECT (wav, "GST_WAVPARSE_DATA");
2419
2420       /* fall-through */
2421     case GST_WAVPARSE_DATA:
2422       if (buf && GST_BUFFER_FLAG_IS_SET (buf, GST_BUFFER_FLAG_DISCONT))
2423         wav->discont = TRUE;
2424       if ((ret = gst_wavparse_stream_data (wav)) != GST_FLOW_OK)
2425         goto done;
2426       break;
2427     default:
2428       g_return_val_if_reached (GST_FLOW_ERROR);
2429   }
2430 done:
2431   if (G_UNLIKELY (wav->abort_buffering)) {
2432     wav->abort_buffering = FALSE;
2433     ret = GST_FLOW_ERROR;
2434     /* sort of demux/parse error */
2435     GST_ELEMENT_ERROR (wav, STREAM, DEMUX, (NULL), ("unhandled buffer size"));
2436   }
2437
2438   return ret;
2439 }
2440
2441 static GstFlowReturn
2442 gst_wavparse_flush_data (GstWavParse * wav)
2443 {
2444   GstFlowReturn ret = GST_FLOW_OK;
2445   guint av;
2446
2447   if ((av = gst_adapter_available (wav->adapter)) > 0) {
2448     wav->dataleft = av;
2449     wav->end_offset = wav->offset + av;
2450     ret = gst_wavparse_stream_data (wav);
2451   }
2452
2453   return ret;
2454 }
2455
2456 static gboolean
2457 gst_wavparse_sink_event (GstPad * pad, GstObject * parent, GstEvent * event)
2458 {
2459   GstWavParse *wav = GST_WAVPARSE (parent);
2460   gboolean ret = TRUE;
2461
2462   GST_LOG_OBJECT (wav, "handling %s event", GST_EVENT_TYPE_NAME (event));
2463
2464   switch (GST_EVENT_TYPE (event)) {
2465     case GST_EVENT_CAPS:
2466     {
2467       /* discard, we'll come up with proper src caps */
2468       gst_event_unref (event);
2469       break;
2470     }
2471     case GST_EVENT_SEGMENT:
2472     {
2473       gint64 start, stop, offset = 0, end_offset = -1;
2474       GstSegment segment;
2475
2476       /* some debug output */
2477       gst_event_copy_segment (event, &segment);
2478       GST_DEBUG_OBJECT (wav, "received newsegment %" GST_SEGMENT_FORMAT,
2479           &segment);
2480
2481       if (wav->state != GST_WAVPARSE_DATA) {
2482         GST_DEBUG_OBJECT (wav, "still starting, eating event");
2483         goto exit;
2484       }
2485
2486       /* now we are either committed to TIME or BYTE format,
2487        * and we only expect a BYTE segment, e.g. following a seek */
2488       if (segment.format == GST_FORMAT_BYTES) {
2489         /* handle (un)signed issues */
2490         start = segment.start;
2491         stop = segment.stop;
2492         if (start > 0) {
2493           offset = start;
2494           start -= wav->datastart;
2495           start = MAX (start, 0);
2496         }
2497         if (stop > 0) {
2498           end_offset = stop;
2499           segment.stop -= wav->datastart;
2500           segment.stop = MAX (stop, 0);
2501         }
2502         if (wav->segment.format == GST_FORMAT_TIME) {
2503           guint64 bps = wav->bps;
2504
2505           /* operating in format TIME, so we can convert */
2506           if (!bps && wav->fact)
2507             bps =
2508                 gst_util_uint64_scale_int (wav->datasize, wav->rate, wav->fact);
2509           if (bps) {
2510             if (start >= 0)
2511               start =
2512                   gst_util_uint64_scale_ceil (start, GST_SECOND,
2513                   (guint64) wav->bps);
2514             if (stop >= 0)
2515               stop =
2516                   gst_util_uint64_scale_ceil (stop, GST_SECOND,
2517                   (guint64) wav->bps);
2518           }
2519         }
2520       } else {
2521         GST_DEBUG_OBJECT (wav, "unsupported segment format, ignoring");
2522         goto exit;
2523       }
2524
2525       segment.start = start;
2526       segment.stop = stop;
2527
2528       /* accept upstream's notion of segment and distribute along */
2529       segment.format = wav->segment.format;
2530       segment.time = segment.position = segment.start;
2531       segment.duration = wav->segment.duration;
2532       segment.base = gst_segment_to_running_time (&wav->segment,
2533           GST_FORMAT_TIME, wav->segment.position);
2534
2535       gst_segment_copy_into (&segment, &wav->segment);
2536
2537       /* also store the newsegment event for the streaming thread */
2538       if (wav->start_segment)
2539         gst_event_unref (wav->start_segment);
2540       GST_DEBUG_OBJECT (wav, "Storing newseg %" GST_SEGMENT_FORMAT, &segment);
2541       wav->start_segment = gst_event_new_segment (&segment);
2542
2543       /* stream leftover data in current segment */
2544       gst_wavparse_flush_data (wav);
2545       /* and set up streaming thread for next one */
2546       wav->offset = offset;
2547       wav->end_offset = end_offset;
2548       if (wav->end_offset > 0) {
2549         wav->dataleft = wav->end_offset - wav->offset;
2550       } else {
2551         /* infinity; upstream will EOS when done */
2552         wav->dataleft = G_MAXUINT64;
2553       }
2554     exit:
2555       gst_event_unref (event);
2556       break;
2557     }
2558     case GST_EVENT_EOS:
2559       /* add pad if needed so EOS is seen downstream */
2560       if (G_UNLIKELY (wav->first)) {
2561         wav->first = FALSE;
2562         gst_wavparse_add_src_pad (wav, NULL);
2563       } else {
2564         /* stream leftover data in current segment */
2565         gst_wavparse_flush_data (wav);
2566       }
2567
2568       if (wav->state == GST_WAVPARSE_START)
2569         GST_ELEMENT_ERROR (wav, STREAM, WRONG_TYPE,
2570             ("No valid input found before end of stream"), (NULL));
2571
2572       /* fall-through */
2573     case GST_EVENT_FLUSH_STOP:
2574     {
2575       GstClockTime dur;
2576
2577       gst_adapter_clear (wav->adapter);
2578       wav->discont = TRUE;
2579       dur = wav->segment.duration;
2580       gst_segment_init (&wav->segment, wav->segment.format);
2581       wav->segment.duration = dur;
2582       /* fall-through */
2583     }
2584     default:
2585       ret = gst_pad_event_default (wav->sinkpad, parent, event);
2586       break;
2587   }
2588
2589   return ret;
2590 }
2591
2592 #if 0
2593 /* convert and query stuff */
2594 static const GstFormat *
2595 gst_wavparse_get_formats (GstPad * pad)
2596 {
2597   static GstFormat formats[] = {
2598     GST_FORMAT_TIME,
2599     GST_FORMAT_BYTES,
2600     GST_FORMAT_DEFAULT,         /* a "frame", ie a set of samples per Hz */
2601     0
2602   };
2603
2604   return formats;
2605 }
2606 #endif
2607
2608 static gboolean
2609 gst_wavparse_pad_convert (GstPad * pad,
2610     GstFormat src_format, gint64 src_value,
2611     GstFormat * dest_format, gint64 * dest_value)
2612 {
2613   GstWavParse *wavparse;
2614   gboolean res = TRUE;
2615
2616   wavparse = GST_WAVPARSE (GST_PAD_PARENT (pad));
2617
2618   if (*dest_format == src_format) {
2619     *dest_value = src_value;
2620     return TRUE;
2621   }
2622
2623   if ((wavparse->bps == 0) && !wavparse->fact)
2624     goto no_bps_fact;
2625
2626   GST_INFO_OBJECT (wavparse, "converting value from %s to %s",
2627       gst_format_get_name (src_format), gst_format_get_name (*dest_format));
2628
2629   switch (src_format) {
2630     case GST_FORMAT_BYTES:
2631       switch (*dest_format) {
2632         case GST_FORMAT_DEFAULT:
2633           *dest_value = src_value / wavparse->bytes_per_sample;
2634           /* make sure we end up on a sample boundary */
2635           *dest_value -= *dest_value % wavparse->bytes_per_sample;
2636           break;
2637         case GST_FORMAT_TIME:
2638           /* src_value + datastart = offset */
2639           GST_INFO_OBJECT (wavparse,
2640               "src=%" G_GINT64_FORMAT ", offset=%" G_GINT64_FORMAT, src_value,
2641               wavparse->offset);
2642           if (wavparse->bps > 0)
2643             *dest_value = gst_util_uint64_scale_ceil (src_value, GST_SECOND,
2644                 (guint64) wavparse->bps);
2645           else if (wavparse->fact) {
2646             guint64 bps = gst_util_uint64_scale_int_ceil (wavparse->datasize,
2647                 wavparse->rate, wavparse->fact);
2648
2649             *dest_value =
2650                 gst_util_uint64_scale_int_ceil (src_value, GST_SECOND, bps);
2651           } else {
2652             res = FALSE;
2653           }
2654           break;
2655         default:
2656           res = FALSE;
2657           goto done;
2658       }
2659       break;
2660
2661     case GST_FORMAT_DEFAULT:
2662       switch (*dest_format) {
2663         case GST_FORMAT_BYTES:
2664           *dest_value = src_value * wavparse->bytes_per_sample;
2665           break;
2666         case GST_FORMAT_TIME:
2667           *dest_value = gst_util_uint64_scale (src_value, GST_SECOND,
2668               (guint64) wavparse->rate);
2669           break;
2670         default:
2671           res = FALSE;
2672           goto done;
2673       }
2674       break;
2675
2676     case GST_FORMAT_TIME:
2677       switch (*dest_format) {
2678         case GST_FORMAT_BYTES:
2679           if (wavparse->bps > 0)
2680             *dest_value = gst_util_uint64_scale (src_value,
2681                 (guint64) wavparse->bps, GST_SECOND);
2682           else {
2683             guint64 bps = gst_util_uint64_scale_int (wavparse->datasize,
2684                 wavparse->rate, wavparse->fact);
2685
2686             *dest_value = gst_util_uint64_scale (src_value, bps, GST_SECOND);
2687           }
2688           /* make sure we end up on a sample boundary */
2689           *dest_value -= *dest_value % wavparse->blockalign;
2690           break;
2691         case GST_FORMAT_DEFAULT:
2692           *dest_value = gst_util_uint64_scale (src_value,
2693               (guint64) wavparse->rate, GST_SECOND);
2694           break;
2695         default:
2696           res = FALSE;
2697           goto done;
2698       }
2699       break;
2700
2701     default:
2702       res = FALSE;
2703       goto done;
2704   }
2705
2706 done:
2707   return res;
2708
2709   /* ERRORS */
2710 no_bps_fact:
2711   {
2712     GST_DEBUG_OBJECT (wavparse, "bps 0 or no fact chunk, cannot convert");
2713     res = FALSE;
2714     goto done;
2715   }
2716 }
2717
2718 /* handle queries for location and length in requested format */
2719 static gboolean
2720 gst_wavparse_pad_query (GstPad * pad, GstObject * parent, GstQuery * query)
2721 {
2722   gboolean res = TRUE;
2723   GstWavParse *wav = GST_WAVPARSE (parent);
2724
2725   /* only if we know */
2726   if (wav->state != GST_WAVPARSE_DATA) {
2727     return FALSE;
2728   }
2729
2730   GST_LOG_OBJECT (pad, "%s query", GST_QUERY_TYPE_NAME (query));
2731
2732   switch (GST_QUERY_TYPE (query)) {
2733     case GST_QUERY_POSITION:
2734     {
2735       gint64 curb;
2736       gint64 cur;
2737       GstFormat format;
2738
2739       /* this is not very precise, as we have pushed severla buffer upstream for prerolling */
2740       curb = wav->offset - wav->datastart;
2741       gst_query_parse_position (query, &format, NULL);
2742       GST_INFO_OBJECT (wav, "pos query at %" G_GINT64_FORMAT, curb);
2743
2744       switch (format) {
2745         case GST_FORMAT_BYTES:
2746           format = GST_FORMAT_BYTES;
2747           cur = curb;
2748           break;
2749         default:
2750           res = gst_wavparse_pad_convert (pad, GST_FORMAT_BYTES, curb,
2751               &format, &cur);
2752           break;
2753       }
2754       if (res)
2755         gst_query_set_position (query, format, cur);
2756       break;
2757     }
2758     case GST_QUERY_DURATION:
2759     {
2760       gint64 duration = 0;
2761       GstFormat format;
2762
2763       if (wav->ignore_length) {
2764         res = FALSE;
2765         break;
2766       }
2767
2768       gst_query_parse_duration (query, &format, NULL);
2769
2770       switch (format) {
2771         case GST_FORMAT_BYTES:{
2772           format = GST_FORMAT_BYTES;
2773           duration = wav->datasize;
2774           break;
2775         }
2776         case GST_FORMAT_TIME:
2777           if ((res = gst_wavparse_calculate_duration (wav))) {
2778             duration = wav->duration;
2779           }
2780           break;
2781         default:
2782           res = FALSE;
2783           break;
2784       }
2785       if (res)
2786         gst_query_set_duration (query, format, duration);
2787       break;
2788     }
2789     case GST_QUERY_CONVERT:
2790     {
2791       gint64 srcvalue, dstvalue;
2792       GstFormat srcformat, dstformat;
2793
2794       gst_query_parse_convert (query, &srcformat, &srcvalue,
2795           &dstformat, &dstvalue);
2796       res = gst_wavparse_pad_convert (pad, srcformat, srcvalue,
2797           &dstformat, &dstvalue);
2798       if (res)
2799         gst_query_set_convert (query, srcformat, srcvalue, dstformat, dstvalue);
2800       break;
2801     }
2802     case GST_QUERY_SEEKING:{
2803       GstFormat fmt;
2804       gboolean seekable = FALSE;
2805
2806       gst_query_parse_seeking (query, &fmt, NULL, NULL, NULL);
2807       if (fmt == wav->segment.format) {
2808         if (wav->streaming) {
2809           GstQuery *q;
2810
2811           q = gst_query_new_seeking (GST_FORMAT_BYTES);
2812           if ((res = gst_pad_peer_query (wav->sinkpad, q))) {
2813             gst_query_parse_seeking (q, &fmt, &seekable, NULL, NULL);
2814             GST_LOG_OBJECT (wav, "upstream BYTE seekable %d", seekable);
2815           }
2816           gst_query_unref (q);
2817         } else {
2818           GST_LOG_OBJECT (wav, "looping => seekable");
2819           seekable = TRUE;
2820           res = TRUE;
2821         }
2822       } else if (fmt == GST_FORMAT_TIME) {
2823         res = TRUE;
2824       }
2825       if (res) {
2826         gst_query_set_seeking (query, fmt, seekable, 0, wav->segment.duration);
2827       }
2828       break;
2829     }
2830     default:
2831       res = gst_pad_query_default (pad, parent, query);
2832       break;
2833   }
2834   return res;
2835 }
2836
2837 static gboolean
2838 gst_wavparse_srcpad_event (GstPad * pad, GstObject * parent, GstEvent * event)
2839 {
2840   GstWavParse *wavparse = GST_WAVPARSE (parent);
2841   gboolean res = FALSE;
2842
2843   GST_DEBUG_OBJECT (wavparse, "%s event", GST_EVENT_TYPE_NAME (event));
2844
2845   switch (GST_EVENT_TYPE (event)) {
2846     case GST_EVENT_SEEK:
2847       /* can only handle events when we are in the data state */
2848       if (wavparse->state == GST_WAVPARSE_DATA) {
2849         res = gst_wavparse_perform_seek (wavparse, event);
2850       }
2851       gst_event_unref (event);
2852       break;
2853
2854     case GST_EVENT_TOC_SELECT:
2855     {
2856       char *uid = NULL;
2857       GstTocEntry *entry = NULL;
2858       GstEvent *seek_event;
2859       gint64 start_pos;
2860
2861       if (!wavparse->toc) {
2862         GST_DEBUG_OBJECT (wavparse, "no TOC to select");
2863         return FALSE;
2864       } else {
2865         gst_event_parse_toc_select (event, &uid);
2866         if (uid != NULL) {
2867           GST_OBJECT_LOCK (wavparse);
2868           entry = gst_toc_find_entry (wavparse->toc, uid);
2869           if (entry == NULL) {
2870             GST_OBJECT_UNLOCK (wavparse);
2871             GST_WARNING_OBJECT (wavparse, "no TOC entry with given UID: %s",
2872                 uid);
2873             res = FALSE;
2874           } else {
2875             gst_toc_entry_get_start_stop_times (entry, &start_pos, NULL);
2876             GST_OBJECT_UNLOCK (wavparse);
2877             seek_event = gst_event_new_seek (1.0,
2878                 GST_FORMAT_TIME,
2879                 GST_SEEK_FLAG_FLUSH,
2880                 GST_SEEK_TYPE_SET, start_pos, GST_SEEK_TYPE_SET, -1);
2881             res = gst_wavparse_perform_seek (wavparse, seek_event);
2882             gst_event_unref (seek_event);
2883           }
2884           g_free (uid);
2885         } else {
2886           GST_WARNING_OBJECT (wavparse, "received empty TOC select event");
2887           res = FALSE;
2888         }
2889       }
2890       gst_event_unref (event);
2891       break;
2892     }
2893
2894     default:
2895       res = gst_pad_push_event (wavparse->sinkpad, event);
2896       break;
2897   }
2898   return res;
2899 }
2900
2901 static gboolean
2902 gst_wavparse_sink_activate (GstPad * sinkpad, GstObject * parent)
2903 {
2904   GstWavParse *wav = GST_WAVPARSE (parent);
2905   GstQuery *query;
2906   gboolean pull_mode;
2907
2908   if (wav->adapter) {
2909     gst_adapter_clear (wav->adapter);
2910     g_object_unref (wav->adapter);
2911     wav->adapter = NULL;
2912   }
2913
2914   query = gst_query_new_scheduling ();
2915
2916   if (!gst_pad_peer_query (sinkpad, query)) {
2917     gst_query_unref (query);
2918     goto activate_push;
2919   }
2920
2921   pull_mode = gst_query_has_scheduling_mode (query, GST_PAD_MODE_PULL);
2922   gst_query_unref (query);
2923
2924   if (!pull_mode)
2925     goto activate_push;
2926
2927   GST_DEBUG_OBJECT (sinkpad, "activating pull");
2928   wav->streaming = FALSE;
2929   return gst_pad_activate_mode (sinkpad, GST_PAD_MODE_PULL, TRUE);
2930
2931 activate_push:
2932   {
2933     GST_DEBUG_OBJECT (sinkpad, "activating push");
2934     wav->streaming = TRUE;
2935     wav->adapter = gst_adapter_new ();
2936     return gst_pad_activate_mode (sinkpad, GST_PAD_MODE_PUSH, TRUE);
2937   }
2938 }
2939
2940
2941 static gboolean
2942 gst_wavparse_sink_activate_mode (GstPad * sinkpad, GstObject * parent,
2943     GstPadMode mode, gboolean active)
2944 {
2945   gboolean res;
2946
2947   switch (mode) {
2948     case GST_PAD_MODE_PUSH:
2949       res = TRUE;
2950       break;
2951     case GST_PAD_MODE_PULL:
2952       if (active) {
2953         /* if we have a scheduler we can start the task */
2954         res = gst_pad_start_task (sinkpad, (GstTaskFunction) gst_wavparse_loop,
2955             sinkpad, NULL);
2956       } else {
2957         res = gst_pad_stop_task (sinkpad);
2958       }
2959       break;
2960     default:
2961       res = FALSE;
2962       break;
2963   }
2964   return res;
2965 }
2966
2967 static GstStateChangeReturn
2968 gst_wavparse_change_state (GstElement * element, GstStateChange transition)
2969 {
2970   GstStateChangeReturn ret;
2971   GstWavParse *wav = GST_WAVPARSE (element);
2972
2973   switch (transition) {
2974     case GST_STATE_CHANGE_NULL_TO_READY:
2975       break;
2976     case GST_STATE_CHANGE_READY_TO_PAUSED:
2977       gst_wavparse_reset (wav);
2978       break;
2979     case GST_STATE_CHANGE_PAUSED_TO_PLAYING:
2980       break;
2981     default:
2982       break;
2983   }
2984
2985   ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
2986
2987   switch (transition) {
2988     case GST_STATE_CHANGE_PLAYING_TO_PAUSED:
2989       break;
2990     case GST_STATE_CHANGE_PAUSED_TO_READY:
2991       gst_wavparse_reset (wav);
2992       break;
2993     case GST_STATE_CHANGE_READY_TO_NULL:
2994       break;
2995     default:
2996       break;
2997   }
2998   return ret;
2999 }
3000
3001 static void
3002 gst_wavparse_set_property (GObject * object, guint prop_id,
3003     const GValue * value, GParamSpec * pspec)
3004 {
3005   GstWavParse *self;
3006
3007   g_return_if_fail (GST_IS_WAVPARSE (object));
3008   self = GST_WAVPARSE (object);
3009
3010   switch (prop_id) {
3011     case PROP_IGNORE_LENGTH:
3012       self->ignore_length = g_value_get_boolean (value);
3013       break;
3014     default:
3015       G_OBJECT_WARN_INVALID_PROPERTY_ID (self, prop_id, pspec);
3016   }
3017
3018 }
3019
3020 static void
3021 gst_wavparse_get_property (GObject * object, guint prop_id,
3022     GValue * value, GParamSpec * pspec)
3023 {
3024   GstWavParse *self;
3025
3026   g_return_if_fail (GST_IS_WAVPARSE (object));
3027   self = GST_WAVPARSE (object);
3028
3029   switch (prop_id) {
3030     case PROP_IGNORE_LENGTH:
3031       g_value_set_boolean (value, self->ignore_length);
3032       break;
3033     default:
3034       G_OBJECT_WARN_INVALID_PROPERTY_ID (self, prop_id, pspec);
3035   }
3036 }
3037
3038 static gboolean
3039 plugin_init (GstPlugin * plugin)
3040 {
3041   gst_riff_init ();
3042
3043   return gst_element_register (plugin, "wavparse", GST_RANK_PRIMARY,
3044       GST_TYPE_WAVPARSE);
3045 }
3046
3047 GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
3048     GST_VERSION_MINOR,
3049     wavparse,
3050     "Parse a .wav file into raw audio",
3051     plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)