gstreamer/gst/volume/gstvolume.c

451 lines
12 KiB
C
Raw Normal View History

/* -*- c-basic-offset: 2 -*-
* GStreamer
* Copyright (C) 1999-2001 Erik Walthinsen <omega@cse.ogi.edu>
*
* 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.
*/
#include <string.h>
#include <gst/gst.h>
#include <gst/audio/audio.h>
#include <gst/control/control.h>
#include "gstvolume.h"
static GstElementDetails volume_details = {
"Volume",
"Filter/Audio/Effect",
"Set volume on audio/raw streams",
VERSION,
"Andy Wingo <apwingo@eos.ncsu.edu>",
"(C) 2001"
};
/* 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/raw",
"rate", GST_PROPS_INT_RANGE (1, G_MAXINT),
"format", GST_PROPS_STRING ("float"),
"layout", GST_PROPS_STRING ("gfloat"),
"intercept", GST_PROPS_FLOAT(0.0),
"slope", GST_PROPS_FLOAT(1.0),
"channels", GST_PROPS_INT (1)
),
GST_CAPS_NEW (
"volume_int_sink",
"audio/raw",
"format", GST_PROPS_STRING ("int"),
"channels", GST_PROPS_INT_RANGE (1, G_MAXINT),
"rate", GST_PROPS_INT_RANGE (1, G_MAXINT),
"law", GST_PROPS_INT (0),
"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/raw",
"rate", GST_PROPS_INT_RANGE (1, G_MAXINT),
"format", GST_PROPS_STRING ("float"),
"layout", GST_PROPS_STRING ("gfloat"),
"intercept", GST_PROPS_FLOAT(0.0),
"slope", GST_PROPS_FLOAT(1.0),
"channels", GST_PROPS_INT (1)
),
GST_CAPS_NEW (
"volume_int_src",
"audio/raw",
"format", GST_PROPS_STRING ("int"),
"channels", GST_PROPS_INT_RANGE (1, G_MAXINT),
"rate", GST_PROPS_INT_RANGE (1, G_MAXINT),
"law", GST_PROPS_INT (0),
"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_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, GstBuffer *buf);
static void volume_chain_int16 (GstPad *pad, GstBuffer *buf);
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 GstPadConnectReturn
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_CONNECT_REFUSED);
g_return_val_if_fail (GST_IS_VOLUME (filter), GST_PAD_CONNECT_REFUSED);
otherpad = (pad == filter->srcpad ? filter->sinkpad : filter->srcpad);
if (GST_CAPS_IS_FIXED (caps)) {
if (!volume_parse_caps (filter, caps) || !gst_pad_try_set_caps (otherpad, caps))
return GST_PAD_CONNECT_REFUSED;
if (gst_caps_get_int (caps, "rate", &rate)){
gst_dpman_set_rate(filter->dpman, rate);
}
return GST_PAD_CONNECT_OK;
}
return GST_PAD_CONNECT_DELAYED;
}
static gboolean
volume_parse_caps (GstVolume *filter, GstCaps *caps)
{
const gchar *format;
g_return_val_if_fail(filter!=NULL,FALSE);
g_return_val_if_fail(caps!=NULL,FALSE);
gst_caps_get_string (caps, "format", &format);
if (strcmp(format, "int")==0) {
gst_pad_set_chain_function(filter->sinkpad,volume_chain_int16);
return TRUE;
}
if (strcmp(format, "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), NULL,
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_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_connect_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_connect_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, GstBuffer *buf)
{
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,out_buf);
}
static void
volume_chain_int16 (GstPad *pad, GstBuffer *buf)
{
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);
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,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 (GModule *module, GstPlugin *plugin)
{
GstElementFactory *factory;
factory = gst_element_factory_new("volume",GST_TYPE_VOLUME,
&volume_details);
g_return_val_if_fail(factory != NULL, FALSE);
gst_element_factory_add_pad_template (factory, volume_src_factory ());
gst_element_factory_add_pad_template (factory, volume_sink_factory ());
gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (factory));
/* load dparam support library */
if (!gst_library_load ("gstcontrol"))
{
gst_info ("volume: could not load support library: 'gstcontrol'\n");
return FALSE;
}
return TRUE;
}
GstPluginDesc plugin_desc = {
GST_VERSION_MAJOR,
GST_VERSION_MINOR,
"volume",
plugin_init
};