/* -*- mOde: C; tab-width: 2; indent-tabs-mode: t; c-basic-offset: 2 -*- */
-/* GStreamer
+/* GStreamer .wav encoder
* Copyright (C) <2002> Iain Holmes <iain@prettypeople.org>
+ * Copyright (C) <2006> Tim-Philipp Müller <tim centricular net>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* Boston, MA 02111-1307, USA.
*
*/
-
+/**
+ * SECTION:element-wavenc
+ *
+ * Format a audio stream into the wav format.
+ *
+ */
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include <string.h>
#include "gstwavenc.h"
-#include "riff.h"
+
+#include <gst/riff/riff-media.h>
GST_DEBUG_CATEGORY_STATIC (wavenc_debug);
#define GST_CAT_DEFAULT wavenc_debug
-static void gst_wavenc_base_init (gpointer g_class);
-static void gst_wavenc_class_init (GstWavEncClass * klass);
-static void gst_wavenc_init (GstWavEnc * wavenc);
-static GstFlowReturn gst_wavenc_chain (GstPad * pad, GstBuffer * buf);
-static gboolean gst_wavenc_event (GstPad * pad, GstEvent * event);
-
-#define WAVE_FORMAT_PCM 0x0001
-
struct riff_struct
{
guint8 id[4]; /* RIFF */
struct chunk_struct data;
};
-static GstElementDetails gst_wavenc_details =
-GST_ELEMENT_DETAILS ("WAV encoder",
- "Codec/Muxer/Audio",
- "Encode raw audio into WAV",
- "Iain Holmes <iain@prettypeople.org>");
+/* FIXME: mono doesn't produce correct files it seems, at least mplayer xruns */
+/* Max. of two channels, more channels need WAVFORMATEX with
+ * channel layout, which we do not support yet */
+#define SINK_CAPS \
+ "audio/x-raw-int, " \
+ "rate = (int) [ 1, MAX ], " \
+ "channels = (int) [ 1, 2 ], " \
+ "endianness = (int) LITTLE_ENDIAN, " \
+ "width = (int) 32, " \
+ "depth = (int) 32, " \
+ "signed = (boolean) true" \
+ "; " \
+ "audio/x-raw-int, " \
+ "rate = (int) [ 1, MAX ], " \
+ "channels = (int) [ 1, 2 ], " \
+ "endianness = (int) LITTLE_ENDIAN, " \
+ "width = (int) 24, " \
+ "depth = (int) 24, " \
+ "signed = (boolean) true" \
+ "; " \
+ "audio/x-raw-int, " \
+ "rate = (int) [ 1, MAX ], " \
+ "channels = (int) [ 1, 2 ], " \
+ "endianness = (int) LITTLE_ENDIAN, " \
+ "width = (int) 16, " \
+ "depth = (int) 16, " \
+ "signed = (boolean) true" \
+ "; " \
+ "audio/x-raw-int, " \
+ "rate = (int) [ 1, MAX ], " \
+ "channels = (int) [ 1, 2 ], " \
+ "width = (int) 8, " \
+ "depth = (int) 8, " \
+ "signed = (boolean) false" \
+ "; " \
+ "audio/x-raw-float, " \
+ "rate = (int) [ 1, MAX ], " \
+ "channels = (int) [ 1, 2 ], " \
+ "endianness = (int) LITTLE_ENDIAN, " \
+ "width = (int) { 32, 64 }; " \
+ "audio/x-alaw, " \
+ "rate = (int) [ 8000, 192000 ], " \
+ "channels = (int) [ 1, 2 ], " \
+ "width = (int) 8, " \
+ "depth = (int) 8, " \
+ "signed = (boolean) false; " \
+ "audio/x-mulaw, " \
+ "rate = (int) [ 8000, 192000 ], " \
+ "channels = (int) [ 1, 2 ], " \
+ "width = (int) 8, " \
+ "depth = (int) 8, " \
+ "signed = (boolean) false"
+
static GstStaticPadTemplate sink_factory = GST_STATIC_PAD_TEMPLATE ("sink",
GST_PAD_SINK,
GST_PAD_ALWAYS,
- GST_STATIC_CAPS ("audio/x-raw-int, "
- "rate = (int) [ 1, MAX ], "
- "channels = (int) [ 1, MAX ], "
- "endianness = (int) LITTLE_ENDIAN, "
- "width = (int) { 8, 16 }, "
- "depth = (int) { 8, 16 }, " "signed = (boolean) true")
+ GST_STATIC_CAPS (SINK_CAPS)
);
static GstStaticPadTemplate src_factory = GST_STATIC_PAD_TEMPLATE ("src",
GST_STATIC_CAPS ("audio/x-wav")
);
-enum
-{
- PROP_0
-};
-
-static GstElementClass *parent_class = NULL;
-
-static GType
-gst_wavenc_get_type (void)
-{
- static GType type = 0;
-
- if (type == 0) {
- static const GTypeInfo info = {
- sizeof (GstWavEncClass),
- gst_wavenc_base_init,
- NULL,
- (GClassInitFunc) gst_wavenc_class_init,
- NULL,
- NULL,
- sizeof (GstWavEnc),
- 0,
- (GInstanceInitFunc) gst_wavenc_init
- };
-
- type = g_type_register_static (GST_TYPE_ELEMENT, "GstWavEnc", &info, 0);
- }
-
- return type;
-}
-
-static GstStateChangeReturn
-gst_wavenc_change_state (GstElement * element, GstStateChange transition)
-{
- GstWavEnc *wavenc = GST_WAVENC (element);
-
- switch (transition) {
- case GST_STATE_CHANGE_PAUSED_TO_READY:
- wavenc->setup = FALSE;
- wavenc->flush_header = TRUE;
- break;
- default:
- break;
- }
+#define gst_wavenc_parent_class parent_class
+G_DEFINE_TYPE (GstWavEnc, gst_wavenc, GST_TYPE_ELEMENT);
- if (parent_class->change_state) {
- return parent_class->change_state (element, transition);
- }
-
- return GST_STATE_CHANGE_SUCCESS;
-}
+static GstFlowReturn gst_wavenc_chain (GstPad * pad, GstBuffer * buf);
+static gboolean gst_wavenc_event (GstPad * pad, GstEvent * event);
+static GstStateChangeReturn gst_wavenc_change_state (GstElement * element,
+ GstStateChange transition);
+static gboolean gst_wavenc_sink_setcaps (GstPad * pad, GstCaps * caps);
static void
-set_property (GObject * object,
- guint prop_id, const GValue * value, GParamSpec * pspec)
+gst_wavenc_class_init (GstWavEncClass * klass)
{
- GstWavEnc *enc;
-
- enc = GST_WAVENC (object);
+ GstElementClass *element_class;
- switch (prop_id) {
- default:
- break;
- }
-}
+ element_class = (GstElementClass *) klass;
-static void
-gst_wavenc_base_init (gpointer g_class)
-{
- GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
+ element_class->change_state = GST_DEBUG_FUNCPTR (gst_wavenc_change_state);
- gst_element_class_set_details (element_class, &gst_wavenc_details);
+ gst_element_class_set_details_simple (element_class, "WAV audio muxer",
+ "Codec/Muxer/Audio",
+ "Encode raw audio into WAV", "Iain Holmes <iain@prettypeople.org>");
gst_element_class_add_pad_template (element_class,
gst_static_pad_template_get (&src_factory));
gst_element_class_add_pad_template (element_class,
gst_static_pad_template_get (&sink_factory));
+
+ GST_DEBUG_CATEGORY_INIT (wavenc_debug, "wavenc", 0, "WAV encoder element");
}
+
static void
-gst_wavenc_class_init (GstWavEncClass * klass)
+gst_wavenc_init (GstWavEnc * wavenc)
{
- GstElementClass *element_class;
- GObjectClass *object_class;
-
- element_class = (GstElementClass *) klass;
- object_class = (GObjectClass *) klass;
- object_class->set_property = set_property;
-
- element_class->change_state = gst_wavenc_change_state;
-
- parent_class = g_type_class_ref (GST_TYPE_ELEMENT);
+ wavenc->sinkpad = gst_pad_new_from_static_template (&sink_factory, "sink");
+ gst_pad_set_chain_function (wavenc->sinkpad,
+ GST_DEBUG_FUNCPTR (gst_wavenc_chain));
+ gst_pad_set_event_function (wavenc->sinkpad,
+ GST_DEBUG_FUNCPTR (gst_wavenc_event));
+ gst_pad_use_fixed_caps (wavenc->sinkpad);
+ gst_element_add_pad (GST_ELEMENT (wavenc), wavenc->sinkpad);
- GST_DEBUG_CATEGORY_INIT (wavenc_debug, "wavenc", 0, "WAV encoder element");
+ wavenc->srcpad = gst_pad_new_from_static_template (&src_factory, "src");
+ gst_pad_use_fixed_caps (wavenc->srcpad);
+ gst_pad_set_caps (wavenc->srcpad,
+ gst_static_pad_template_get_caps (&src_factory));
+ gst_element_add_pad (GST_ELEMENT (wavenc), wavenc->srcpad);
}
-static gboolean
-gst_wavenc_setup (GstWavEnc * wavenc)
+#define WAV_HEADER_LEN 44
+
+static GstBuffer *
+gst_wavenc_create_header_buf (GstWavEnc * wavenc, guint audio_data_size)
{
struct wave_header wave;
- gint size = 0x7fffffff; /* Use a bogus size initially */
+ GstBuffer *buf;
+ guint8 *header;
+
+ buf = gst_buffer_new_and_alloc (WAV_HEADER_LEN);
+ header = gst_buffer_map (buf, NULL, NULL, GST_MAP_WRITE);
+ memset (header, 0, WAV_HEADER_LEN);
wave.common.wChannels = wavenc->channels;
- wave.common.wBitsPerSample = wavenc->bits;
+ wave.common.wBitsPerSample = wavenc->width;
wave.common.dwSamplesPerSec = wavenc->rate;
- memset (wavenc->header, 0, WAV_HEADER_LEN);
-
/* Fill out our wav-header with some information */
- strncpy ((char *) wave.riff.id, "RIFF", 4);
- wave.riff.len = size - 8;
- strncpy ((char *) wave.riff.wav_id, "WAVE", 4);
+ memcpy (wave.riff.id, "RIFF", 4);
+ wave.riff.len = audio_data_size + WAV_HEADER_LEN - 8;
+ memcpy (wave.riff.wav_id, "WAVE", 4);
- strncpy ((char *) wave.format.id, "fmt ", 4);
+ memcpy (wave.format.id, "fmt ", 4);
wave.format.len = 16;
- wave.common.wFormatTag = WAVE_FORMAT_PCM;
+ wave.common.wFormatTag = wavenc->format;
+ wave.common.wBlockAlign = (wavenc->width / 8) * wave.common.wChannels;
wave.common.dwAvgBytesPerSec =
- wave.common.wChannels * wave.common.dwSamplesPerSec *
- (wave.common.wBitsPerSample >> 3);
- wave.common.wBlockAlign =
- wave.common.wChannels * (wave.common.wBitsPerSample >> 3);
-
- strncpy ((char *) wave.data.id, "data", 4);
- wave.data.len = size - 44;
-
- strncpy ((char *) wavenc->header, (char *) wave.riff.id, 4);
- GST_WRITE_UINT32_LE (wavenc->header + 4, wave.riff.len);
- strncpy ((char *) wavenc->header + 8, (char *) wave.riff.wav_id, 4);
- strncpy ((char *) wavenc->header + 12, (char *) wave.format.id, 4);
- GST_WRITE_UINT32_LE (wavenc->header + 16, wave.format.len);
- GST_WRITE_UINT16_LE (wavenc->header + 20, wave.common.wFormatTag);
- GST_WRITE_UINT16_LE (wavenc->header + 22, wave.common.wChannels);
- GST_WRITE_UINT32_LE (wavenc->header + 24, wave.common.dwSamplesPerSec);
- GST_WRITE_UINT32_LE (wavenc->header + 28, wave.common.dwAvgBytesPerSec);
- GST_WRITE_UINT16_LE (wavenc->header + 32, wave.common.wBlockAlign);
- GST_WRITE_UINT16_LE (wavenc->header + 34, wave.common.wBitsPerSample);
- strncpy ((char *) wavenc->header + 36, (char *) wave.data.id, 4);
- GST_WRITE_UINT32_LE (wavenc->header + 40, wave.data.len);
-
- wavenc->length = 0;
- wavenc->setup = TRUE;
- return TRUE;
+ wave.common.wBlockAlign * wave.common.dwSamplesPerSec;
+
+ memcpy (wave.data.id, "data", 4);
+ wave.data.len = audio_data_size;
+
+ memcpy (header, (char *) wave.riff.id, 4);
+ GST_WRITE_UINT32_LE (header + 4, wave.riff.len);
+ memcpy (header + 8, (char *) wave.riff.wav_id, 4);
+ memcpy (header + 12, (char *) wave.format.id, 4);
+ GST_WRITE_UINT32_LE (header + 16, wave.format.len);
+ GST_WRITE_UINT16_LE (header + 20, wave.common.wFormatTag);
+ GST_WRITE_UINT16_LE (header + 22, wave.common.wChannels);
+ GST_WRITE_UINT32_LE (header + 24, wave.common.dwSamplesPerSec);
+ GST_WRITE_UINT32_LE (header + 28, wave.common.dwAvgBytesPerSec);
+ GST_WRITE_UINT16_LE (header + 32, wave.common.wBlockAlign);
+ GST_WRITE_UINT16_LE (header + 34, wave.common.wBitsPerSample);
+ memcpy (header + 36, (char *) wave.data.id, 4);
+ GST_WRITE_UINT32_LE (header + 40, wave.data.len);
+
+ gst_buffer_unmap (buf, header, -1);
+
+ return buf;
}
-static gboolean
-gst_wavenc_sink_setcaps (GstPad * pad, GstCaps * caps)
+static GstFlowReturn
+gst_wavenc_push_header (GstWavEnc * wavenc, guint audio_data_size)
{
- GstWavEnc *wavenc;
- GstStructure *structure;
+ GstFlowReturn ret;
+ GstBuffer *outbuf;
+ GstSegment segment;
- wavenc = GST_WAVENC (gst_pad_get_parent (pad));
- wavenc->setup = FALSE;
+ /* seek to beginning of file */
+ gst_segment_init (&segment, GST_FORMAT_BYTES);
+ gst_pad_push_event (wavenc->srcpad, gst_event_new_segment (&segment));
- structure = gst_caps_get_structure (caps, 0);
- gst_structure_get_int (structure, "channels", (gint *) & wavenc->channels);
- gst_structure_get_int (structure, "rate", (gint *) & wavenc->rate);
- gst_structure_get_int (structure, "depth", (gint *) & wavenc->bits);
+ GST_DEBUG_OBJECT (wavenc, "writing header with datasize=%u", audio_data_size);
- gst_wavenc_setup (wavenc);
+ outbuf = gst_wavenc_create_header_buf (wavenc, audio_data_size);
+ GST_BUFFER_OFFSET (outbuf) = 0;
- if (wavenc->setup) {
- return TRUE;
+ ret = gst_pad_push (wavenc->srcpad, outbuf);
+
+ if (ret != GST_FLOW_OK) {
+ GST_WARNING_OBJECT (wavenc, "push header failed: flow = %s",
+ gst_flow_get_name (ret));
}
- return FALSE;
+ return ret;
}
-static void
-gst_wavenc_stop_file (GstWavEnc * wavenc)
+static gboolean
+gst_wavenc_sink_setcaps (GstPad * pad, GstCaps * caps)
{
- GstEvent *event;
- GstBuffer *outbuf;
+ GstWavEnc *wavenc;
+ GstStructure *structure;
+ const gchar *name;
+ gint chans, rate, width;
+ GstCaps *ccaps;
- event = gst_event_new_newsegment (FALSE, 1.0, GST_FORMAT_BYTES,
- 0, GST_CLOCK_TIME_NONE, 0);
+ wavenc = GST_WAVENC (gst_pad_get_parent (pad));
- gst_pad_push_event (wavenc->srcpad, event);
+ ccaps = gst_pad_get_current_caps (pad);
+ if (wavenc->sent_header && ccaps && !gst_caps_can_intersect (caps, ccaps)) {
+ gst_caps_unref (ccaps);
+ GST_WARNING_OBJECT (wavenc, "cannot change format in middle of stream");
+ goto fail;
+ }
+ if (ccaps)
+ gst_caps_unref (ccaps);
- outbuf = gst_buffer_new_and_alloc (WAV_HEADER_LEN);
- GST_WRITE_UINT32_LE (wavenc->header + 4,
- wavenc->length + (WAV_HEADER_LEN - 8));
- GST_WRITE_UINT32_LE (wavenc->header + 40, wavenc->length);
- memcpy (GST_BUFFER_DATA (outbuf), wavenc->header, WAV_HEADER_LEN);
- GST_BUFFER_OFFSET (outbuf) = 0;
- GST_BUFFER_OFFSET_END (outbuf) = WAV_HEADER_LEN;
- GST_BUFFER_TIMESTAMP (outbuf) = 0;
+ GST_DEBUG_OBJECT (wavenc, "got caps: %" GST_PTR_FORMAT, caps);
- if ((gst_pad_push (wavenc->srcpad, outbuf)) != GST_FLOW_OK)
- GST_WARNING_OBJECT (wavenc, "couldn't push header....");
-}
+ structure = gst_caps_get_structure (caps, 0);
+ name = gst_structure_get_name (structure);
-static void
-gst_wavenc_init (GstWavEnc * wavenc)
-{
- GstElementClass *klass = GST_ELEMENT_GET_CLASS (wavenc);
+ if (!gst_structure_get_int (structure, "channels", &chans) ||
+ !gst_structure_get_int (structure, "rate", &rate)) {
+ GST_WARNING_OBJECT (wavenc, "caps incomplete");
+ goto fail;
+ }
- wavenc->sinkpad =
- gst_pad_new_from_template (gst_element_class_get_pad_template (klass,
- "sink"), "sink");
- gst_element_add_pad (GST_ELEMENT (wavenc), wavenc->sinkpad);
- gst_pad_set_chain_function (wavenc->sinkpad, gst_wavenc_chain);
- gst_pad_set_event_function (wavenc->sinkpad, gst_wavenc_event);
- gst_pad_set_setcaps_function (wavenc->sinkpad, gst_wavenc_sink_setcaps);
+ if (strcmp (name, "audio/x-raw-int") == 0) {
+ if (!gst_structure_get_int (structure, "width", &width)) {
+ GST_WARNING_OBJECT (wavenc, "caps incomplete");
+ goto fail;
+ }
+ wavenc->format = GST_RIFF_WAVE_FORMAT_PCM;
+ wavenc->width = width;
+ } else if (strcmp (name, "audio/x-raw-float") == 0) {
+ if (!gst_structure_get_int (structure, "width", &width)) {
+ GST_WARNING_OBJECT (wavenc, "caps incomplete");
+ goto fail;
+ }
+ wavenc->format = GST_RIFF_WAVE_FORMAT_IEEE_FLOAT;
+ wavenc->width = width;
+ } else if (strcmp (name, "audio/x-alaw") == 0) {
+ wavenc->format = GST_RIFF_WAVE_FORMAT_ALAW;
+ wavenc->width = 8;
+ } else if (strcmp (name, "audio/x-mulaw") == 0) {
+ wavenc->format = GST_RIFF_WAVE_FORMAT_MULAW;
+ wavenc->width = 8;
+ } else {
+ GST_WARNING_OBJECT (wavenc, "Unsupported format %s", name);
+ goto fail;
+ }
- wavenc->srcpad =
- gst_pad_new_from_template (gst_element_class_get_pad_template (klass,
- "src"), "src");
- gst_element_add_pad (GST_ELEMENT (wavenc), wavenc->srcpad);
+ wavenc->channels = chans;
+ wavenc->rate = rate;
+
+ GST_LOG_OBJECT (wavenc,
+ "accepted caps: format=0x%04x chans=%u width=%u rate=%u",
+ wavenc->format, wavenc->channels, wavenc->width, wavenc->rate);
+
+ gst_object_unref (wavenc);
+ return TRUE;
- wavenc->setup = FALSE;
- wavenc->flush_header = TRUE;
- wavenc->newmediacount = 0;
+fail:
+ gst_object_unref (wavenc);
+ return FALSE;
}
-struct _maps
+#if 0
+static struct _maps
{
- guint32 id;
- char *name;
-}
-maps[] = {
- {
- GST_RIFF_INFO_IARL, "Location"}
- , {
- GST_RIFF_INFO_IART, "Artist"}
- , {
- GST_RIFF_INFO_ICMS, "Commissioner"}
- , {
- GST_RIFF_INFO_ICMT, "Comment"}
- , {
- GST_RIFF_INFO_ICOP, "Copyright"}
- , {
- GST_RIFF_INFO_ICRD, "Creation Date"}
- , {
- GST_RIFF_INFO_IENG, "Engineer"}
- , {
- GST_RIFF_INFO_IGNR, "Genre"}
- , {
- GST_RIFF_INFO_IKEY, "Keywords"}
- , {
- GST_RIFF_INFO_INAM, "Title"}
- , /* Name */
+ const guint32 id;
+ const gchar *name;
+} maps[] = {
{
- GST_RIFF_INFO_IPRD, "Product"}
- , {
- GST_RIFF_INFO_ISBJ, "Subject"}
- , {
- GST_RIFF_INFO_ISFT, "Software"}
- , {
+ GST_RIFF_INFO_IARL, "Location"}, {
+ GST_RIFF_INFO_IART, "Artist"}, {
+ GST_RIFF_INFO_ICMS, "Commissioner"}, {
+ GST_RIFF_INFO_ICMT, "Comment"}, {
+ GST_RIFF_INFO_ICOP, "Copyright"}, {
+ GST_RIFF_INFO_ICRD, "Creation Date"}, {
+ GST_RIFF_INFO_IENG, "Engineer"}, {
+ GST_RIFF_INFO_IGNR, "Genre"}, {
+ GST_RIFF_INFO_IKEY, "Keywords"}, {
+ GST_RIFF_INFO_INAM, "Title"}, {
+ GST_RIFF_INFO_IPRD, "Product"}, {
+ GST_RIFF_INFO_ISBJ, "Subject"}, {
+ GST_RIFF_INFO_ISFT, "Software"}, {
GST_RIFF_INFO_ITCH, "Technician"}
- , {
- 0, NULL}
};
-#if 0
static guint32
get_id_from_name (const char *name)
{
int i;
- for (i = 0; maps[i].name; i++) {
+ for (i = 0; i < G_N_ELEMENTS (maps); i++) {
if (strcasecmp (maps[i].name, name) == 0) {
return maps[i].id;
}
gst_props_entry_get_string (entry, &text);
len = strlen (text) + 1; /* The length in the file includes the \0 */
- tmp = g_strdup_printf (GST_FOURCC_FORMAT "%d%s", GST_FOURCC_ARGS (id),
+ tmp = g_strdup_printf ("%" GST_FOURCC_FORMAT "%d%s", GST_FOURCC_ARGS (id),
GUINT32_TO_LE (len), text);
g_string_append (info_str, tmp);
g_free (tmp);
wavenc = GST_WAVENC (gst_pad_get_parent (pad));
switch (GST_EVENT_TYPE (event)) {
- case GST_EVENT_EOS:
- wavenc->pad_eos = TRUE;
+ case GST_EVENT_CAPS:
+ {
+ GstCaps *caps;
+
+ gst_event_parse_caps (event, &caps);
+ gst_wavenc_sink_setcaps (pad, caps);
+ /* have our own src caps */
+ gst_event_unref (event);
+ break;
+ }
+ case GST_EVENT_EOS:{
+ GST_DEBUG_OBJECT (wavenc, "got EOS");
#if 0
/* Write our metadata if we have any */
if (wavenc->metadata) {
write_labels (wavenc);
}
#endif
+ /* write header with correct length values */
+ gst_wavenc_push_header (wavenc, wavenc->length);
- gst_wavenc_stop_file (wavenc);
+ /* we're done with this file */
+ wavenc->finished_properly = TRUE;
+ /* and forward the EOS event */
+ res = gst_pad_event_default (pad, event);
break;
- case GST_EVENT_NEWSEGMENT:
- if (wavenc->newmediacount++ > 0) {
- gst_wavenc_stop_file (wavenc);
- wavenc->setup = FALSE;
- wavenc->flush_header = TRUE;
- gst_wavenc_setup (wavenc);
- }
+ }
+ case GST_EVENT_SEGMENT:
+ /* Just drop it, it's probably in TIME format
+ * anyway. We'll send our own newsegment event */
+ gst_event_unref (event);
break;
default:
+ res = gst_pad_event_default (pad, event);
break;
}
- /* and forward it */
- res = gst_pad_event_default (pad, event);
-
gst_object_unref (wavenc);
return res;
}
static GstFlowReturn
gst_wavenc_chain (GstPad * pad, GstBuffer * buf)
{
- GstWavEnc *wavenc;
- GstFlowReturn result = GST_FLOW_OK;
+ GstWavEnc *wavenc = GST_WAVENC (GST_PAD_PARENT (pad));
+ GstFlowReturn flow = GST_FLOW_OK;
- wavenc = GST_WAVENC (gst_pad_get_parent (pad));
+ g_return_val_if_fail (wavenc->channels > 0, GST_FLOW_WRONG_STATE);
+
+ if (!wavenc->sent_header) {
+ /* use bogus size initially, we'll write the real
+ * header when we get EOS and know the exact length */
+ flow = gst_wavenc_push_header (wavenc, 0x7FFF0000);
+
+ /* starting a file, means we have to finish it properly */
+ wavenc->finished_properly = FALSE;
- if (!wavenc->setup) {
- gst_buffer_unref (buf);
- GST_ELEMENT_ERROR (wavenc, CORE, NEGOTIATION, (NULL),
- ("encoder not initialised (input is not audio?)"));
- result = GST_FLOW_ERROR;
- goto beach;
+ if (flow != GST_FLOW_OK)
+ return flow;
+
+ GST_DEBUG_OBJECT (wavenc, "wrote dummy header");
+ wavenc->sent_header = TRUE;
}
- if (wavenc->flush_header) {
- GstBuffer *outbuf;
+ GST_LOG_OBJECT (wavenc, "pushing %u bytes raw audio, ts=%" GST_TIME_FORMAT,
+ gst_buffer_get_size (buf), GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (buf)));
- result = gst_pad_alloc_buffer (wavenc->srcpad, GST_BUFFER_OFFSET_NONE,
- WAV_HEADER_LEN, GST_PAD_CAPS (wavenc->srcpad), &outbuf);
- if (result != GST_FLOW_OK) {
- GST_WARNING_OBJECT (wavenc, "failed when allocating a %d bytes buffer "
- "for headers", WAV_HEADER_LEN);
- goto beach;
- }
- memcpy (GST_BUFFER_DATA (outbuf), wavenc->header, WAV_HEADER_LEN);
- GST_BUFFER_TIMESTAMP (outbuf) = GST_BUFFER_TIMESTAMP (buf);
+ buf = gst_buffer_make_writable (buf);
- result = gst_pad_push (wavenc->srcpad, outbuf);
- if (result != GST_FLOW_OK) {
- GST_WARNING_OBJECT (wavenc, "failed when pushing header buffer");
- goto beach;
- }
- wavenc->flush_header = FALSE;
+ GST_BUFFER_OFFSET (buf) = WAV_HEADER_LEN + wavenc->length;
+ GST_BUFFER_OFFSET_END (buf) = GST_BUFFER_OFFSET_NONE;
+
+ wavenc->length += gst_buffer_get_size (buf);
+
+ flow = gst_pad_push (wavenc->srcpad, buf);
+
+ return flow;
+}
+
+static GstStateChangeReturn
+gst_wavenc_change_state (GstElement * element, GstStateChange transition)
+{
+ GstStateChangeReturn ret = GST_STATE_CHANGE_SUCCESS;
+ GstWavEnc *wavenc = GST_WAVENC (element);
+
+ switch (transition) {
+ case GST_STATE_CHANGE_NULL_TO_READY:
+ wavenc->format = 0;
+ wavenc->channels = 0;
+ wavenc->width = 0;
+ wavenc->rate = 0;
+ wavenc->length = 0;
+ wavenc->sent_header = FALSE;
+ /* its true because we haven't writen anything */
+ wavenc->finished_properly = TRUE;
+ break;
+ default:
+ break;
}
- wavenc->length += GST_BUFFER_SIZE (buf);
- result = gst_pad_push (wavenc->srcpad, buf);
+ ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
+ if (ret != GST_STATE_CHANGE_SUCCESS)
+ return ret;
-beach:
- gst_object_unref (wavenc);
- return result;
+ switch (transition) {
+ case GST_STATE_CHANGE_PAUSED_TO_READY:
+ if (!wavenc->finished_properly) {
+ GST_ELEMENT_WARNING (wavenc, STREAM, MUX,
+ ("Wav stream not finished properly"),
+ ("Wav stream not finished properly, no EOS received "
+ "before shutdown"));
+ }
+ break;
+ default:
+ break;
+ }
+
+ return ret;
}
static gboolean
plugin_init (GstPlugin * plugin)
{
- return gst_element_register (plugin, "wavenc", GST_RANK_NONE,
+ return gst_element_register (plugin, "wavenc", GST_RANK_PRIMARY,
GST_TYPE_WAVENC);
}
projects / platform / upstream / gst-plugins-good.git / blobdiff