/* -*- c-basic-offset: 2 -*- * GStreamer * Copyright (C) 1999-2001 Erik Walthinsen * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Library General Public * License as published by the Free Software Foundation; either * version 2 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Library General Public License for more details. * * You should have received a copy of the GNU Library General Public * License along with this library; if not, write to the * Free Software Foundation, Inc., 59 Temple Place - Suite 330, * Boston, MA 02111-1307, USA. */ #ifdef HAVE_CONFIG_H #include "config.h" #endif #include #include #include #include #include "gstvolume.h" static GstElementDetails volume_details = { "Volume", "Filter/Effect/Audio", "Set volume on audio/raw streams", "Andy Wingo ", }; /* Filter signals and args */ enum { /* FILL ME */ LAST_SIGNAL }; enum { ARG_0, ARG_SILENT, ARG_MUTE, ARG_VOLUME }; GST_PAD_TEMPLATE_FACTORY (volume_sink_factory, "sink", GST_PAD_SINK, GST_PAD_ALWAYS, gst_caps_new ( "volume_float_sink", "audio/x-raw-float", GST_AUDIO_FLOAT_STANDARD_PAD_TEMPLATE_PROPS ), GST_CAPS_NEW ( "volume_int_sink", "audio/x-raw-int", "channels", GST_PROPS_INT_RANGE (1, G_MAXINT), "rate", GST_PROPS_INT_RANGE (1, G_MAXINT), "endianness", GST_PROPS_INT (G_BYTE_ORDER), "width", GST_PROPS_INT (16), "depth", GST_PROPS_INT (16), "signed", GST_PROPS_BOOLEAN (TRUE) ) ); GST_PAD_TEMPLATE_FACTORY (volume_src_factory, "src", GST_PAD_SRC, GST_PAD_ALWAYS, gst_caps_new ( "volume_float_src", "audio/x-raw-float", GST_AUDIO_FLOAT_STANDARD_PAD_TEMPLATE_PROPS ), GST_CAPS_NEW ( "volume_int_src", "audio/x-raw-int", "channels", GST_PROPS_INT_RANGE (1, G_MAXINT), "rate", GST_PROPS_INT_RANGE (1, G_MAXINT), "endianness", GST_PROPS_INT (G_BYTE_ORDER), "width", GST_PROPS_INT (16), "depth", GST_PROPS_INT (16), "signed", GST_PROPS_BOOLEAN (TRUE) ) ); static void volume_base_init (gpointer g_class); static void volume_class_init (GstVolumeClass *klass); static void volume_init (GstVolume *filter); static void volume_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec); static void volume_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec); static void volume_update_volume (const GValue *value, gpointer data); static void volume_update_mute (const GValue *value, gpointer data); static gboolean volume_parse_caps (GstVolume *filter, GstCaps *caps); static void volume_chain_float (GstPad *pad, GstData *_data); static void volume_chain_int16 (GstPad *pad, GstData *_data); static GstElementClass *parent_class = NULL; /*static guint gst_filter_signals[LAST_SIGNAL] = { 0 }; */ static GstBufferPool* volume_get_bufferpool (GstPad *pad) { GstVolume *filter; filter = GST_VOLUME (gst_pad_get_parent (pad)); return gst_pad_get_bufferpool (filter->srcpad); } static GstPadLinkReturn volume_connect (GstPad *pad, GstCaps *caps) { GstVolume *filter; GstPad *otherpad; gint rate; filter = GST_VOLUME (gst_pad_get_parent (pad)); g_return_val_if_fail (filter != NULL, GST_PAD_LINK_REFUSED); g_return_val_if_fail (GST_IS_VOLUME (filter), GST_PAD_LINK_REFUSED); otherpad = (pad == filter->srcpad ? filter->sinkpad : filter->srcpad); if (GST_CAPS_IS_FIXED (caps)) { GstPadLinkReturn set_retval; if (!volume_parse_caps (filter, caps)) return GST_PAD_LINK_REFUSED; if ((set_retval = gst_pad_try_set_caps(otherpad, caps)) > 0) if (gst_caps_get_int (caps, "rate", &rate)){ gst_dpman_set_rate(filter->dpman, rate); } return set_retval; } return GST_PAD_LINK_DELAYED; } static gboolean volume_parse_caps (GstVolume *filter, GstCaps *caps) { const gchar *mimetype; g_return_val_if_fail(filter!=NULL,FALSE); g_return_val_if_fail(caps!=NULL,FALSE); mimetype = gst_caps_get_mime (caps); if (strcmp(mimetype, "audio/x-raw-int")==0) { gst_pad_set_chain_function(filter->sinkpad,volume_chain_int16); return TRUE; } if (strcmp(mimetype, "audio/x-raw-float")==0) { gst_pad_set_chain_function(filter->sinkpad,volume_chain_float); return TRUE; } return FALSE; } GType gst_volume_get_type(void) { static GType volume_type = 0; if (!volume_type) { static const GTypeInfo volume_info = { sizeof(GstVolumeClass), volume_base_init, NULL, (GClassInitFunc)volume_class_init, NULL, NULL, sizeof(GstVolume), 0, (GInstanceInitFunc)volume_init, }; volume_type = g_type_register_static(GST_TYPE_ELEMENT, "GstVolume", &volume_info, 0); } return volume_type; } static void volume_base_init (gpointer g_class) { GstElementClass *element_class = GST_ELEMENT_CLASS (g_class); gst_element_class_add_pad_template (element_class, volume_src_factory ()); gst_element_class_add_pad_template (element_class, volume_sink_factory ()); gst_element_class_set_details (element_class, &volume_details); } static void volume_class_init (GstVolumeClass *klass) { GObjectClass *gobject_class; GstElementClass *gstelement_class; gobject_class = (GObjectClass*)klass; gstelement_class = (GstElementClass*)klass; parent_class = g_type_class_ref(GST_TYPE_ELEMENT); g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_MUTE, g_param_spec_boolean("mute","mute","mute", FALSE,G_PARAM_READWRITE)); g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_VOLUME, g_param_spec_float("volume","volume","volume", 0.0,4.0,1.0,G_PARAM_READWRITE)); gobject_class->set_property = volume_set_property; gobject_class->get_property = volume_get_property; } static void volume_init (GstVolume *filter) { filter->sinkpad = gst_pad_new_from_template(volume_sink_factory (),"sink"); gst_pad_set_link_function(filter->sinkpad,volume_connect); gst_pad_set_bufferpool_function(filter->sinkpad,volume_get_bufferpool); filter->srcpad = gst_pad_new_from_template(volume_src_factory (),"src"); gst_pad_set_link_function(filter->srcpad,volume_connect); gst_element_add_pad(GST_ELEMENT(filter),filter->sinkpad); gst_element_add_pad(GST_ELEMENT(filter),filter->srcpad); gst_pad_set_chain_function(filter->sinkpad,volume_chain_int16); filter->mute = FALSE; filter->volume_i = 8192; filter->volume_f = 1.0; filter->real_vol_i = 8192; filter->real_vol_f = 1.0; filter->dpman = gst_dpman_new ("volume_dpman", GST_ELEMENT(filter)); gst_dpman_add_required_dparam_callback ( filter->dpman, g_param_spec_int("mute","Mute","Mute the audio", 0, 1, 0, G_PARAM_READWRITE), "int", volume_update_mute, filter ); gst_dpman_add_required_dparam_callback ( filter->dpman, g_param_spec_float("volume","Volume","Volume of the audio", 0.0, 4.0, 1.0, G_PARAM_READWRITE), "scalar", volume_update_volume, filter ); } static void volume_chain_float (GstPad *pad, GstData *_data) { GstBuffer *buf = GST_BUFFER (_data); GstVolume *filter; GstBuffer *out_buf; gfloat *data; gint i, num_samples; g_return_if_fail(GST_IS_PAD(pad)); g_return_if_fail(buf != NULL); filter = GST_VOLUME(GST_OBJECT_PARENT (pad)); g_return_if_fail(GST_IS_VOLUME(filter)); out_buf = gst_buffer_copy_on_write (buf); data = (gfloat *)GST_BUFFER_DATA(out_buf); num_samples = GST_BUFFER_SIZE(out_buf)/sizeof(gfloat); GST_DPMAN_PREPROCESS(filter->dpman, num_samples, GST_BUFFER_TIMESTAMP(out_buf)); i = 0; while(GST_DPMAN_PROCESS(filter->dpman, i)) { data[i++] *= filter->real_vol_f; } gst_pad_push(filter->srcpad,GST_DATA (out_buf)); } static void volume_chain_int16 (GstPad *pad, GstData *_data) { GstBuffer *buf = GST_BUFFER (_data); GstVolume *filter; GstBuffer *out_buf; gint16 *data; gint i, num_samples; g_return_if_fail(GST_IS_PAD(pad)); g_return_if_fail(buf != NULL); filter = GST_VOLUME(GST_OBJECT_PARENT (pad)); g_return_if_fail(GST_IS_VOLUME(filter)); out_buf = gst_buffer_copy_on_write (buf); data = (gint16 *) GST_BUFFER_DATA (out_buf); g_assert (data); num_samples = GST_BUFFER_SIZE(out_buf)/sizeof(gint16); GST_DPMAN_PREPROCESS(filter->dpman, num_samples, GST_BUFFER_TIMESTAMP(out_buf)); i = 0; while(GST_DPMAN_PROCESS(filter->dpman, i)) { /* only clamp if the gain is greater than 1.0 */ if (filter->real_vol_i > 8192){ while (i < GST_DPMAN_NEXT_UPDATE_FRAME(filter->dpman)){ data[i] = (gint16)CLAMP(filter->real_vol_i * (gint)data[i] / 8192, -32768, 32767); i++; } } else { while (i < GST_DPMAN_NEXT_UPDATE_FRAME(filter->dpman)){ data[i] = (gint16)(filter->real_vol_i * (gint)data[i] / 8192); i++; } } } gst_pad_push(filter->srcpad,GST_DATA (out_buf)); } static void volume_update_mute(const GValue *value, gpointer data) { GstVolume *filter = (GstVolume*)data; g_return_if_fail(GST_IS_VOLUME(filter)); if (G_VALUE_HOLDS_BOOLEAN(value)){ filter->mute = g_value_get_boolean(value); } else if (G_VALUE_HOLDS_INT(value)){ filter->mute = (g_value_get_int(value) == 1); } if (filter->mute){ filter->real_vol_f = 0.0; filter->real_vol_i = 0; } else { filter->real_vol_f = filter->volume_f; filter->real_vol_i = filter->volume_i; } } static void volume_update_volume(const GValue *value, gpointer data) { GstVolume *filter = (GstVolume*)data; g_return_if_fail(GST_IS_VOLUME(filter)); filter->volume_f = g_value_get_float (value); filter->volume_i = filter->volume_f*8192; if (filter->mute){ filter->real_vol_f = 0.0; filter->real_vol_i = 0; } else { filter->real_vol_f = filter->volume_f; filter->real_vol_i = filter->volume_i; } } static void volume_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec) { GstVolume *filter; /* it's not null if we got it, but it might not be ours */ g_return_if_fail(GST_IS_VOLUME(object)); filter = GST_VOLUME(object); switch (prop_id) { case ARG_MUTE: gst_dpman_bypass_dparam(filter->dpman, "mute"); volume_update_mute(value, filter); break; case ARG_VOLUME: gst_dpman_bypass_dparam(filter->dpman, "volume"); volume_update_volume(value, filter); break; default: G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); break; } } static void volume_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec) { GstVolume *filter; /* it's not null if we got it, but it might not be ours */ g_return_if_fail(GST_IS_VOLUME(object)); filter = GST_VOLUME(object); switch (prop_id) { case ARG_MUTE: g_value_set_boolean (value, filter->mute); break; case ARG_VOLUME: g_value_set_float (value, filter->volume_f); break; default: G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); break; } } static gboolean plugin_init (GstPlugin *plugin) { gst_control_init(NULL,NULL); return gst_element_register (plugin, "volume", GST_RANK_PRIMARY, GST_TYPE_VOLUME); } GST_PLUGIN_DEFINE ( GST_VERSION_MAJOR, GST_VERSION_MINOR, "volume", "element for controlling audio volume", plugin_init, VERSION, GST_LICENSE, GST_PACKAGE, GST_ORIGIN )