gst/wavparse/gstwavparse.c

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