mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-09-09 13:48:55 +00:00
478 lines
21 KiB
C
478 lines
21 KiB
C
/*
|
|
* GStreamer
|
|
* Copyright (C) 2007 Sebastian Dröge <slomo@circular-chaos.org>
|
|
* Copyright (C) 2006 Stefan Kost <ensonic@users.sf.net>
|
|
*
|
|
* 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., 51 Franklin St, Fifth Floor,
|
|
* Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
/**
|
|
* SECTION:element-audioamplify
|
|
* @title: audioamplify
|
|
*
|
|
* Amplifies an audio stream by a given factor and allows the selection of different clipping modes.
|
|
* The difference between the clipping modes is best evaluated by testing.
|
|
*
|
|
* ## Example launch line
|
|
* |[
|
|
* gst-launch-1.0 audiotestsrc wave=saw ! audioamplify amplification=1.5 ! alsasink
|
|
* gst-launch-1.0 filesrc location="melo1.ogg" ! oggdemux ! vorbisdec ! audioconvert ! audioamplify amplification=1.5 clipping-method=wrap-negative ! alsasink
|
|
* gst-launch-1.0 audiotestsrc wave=saw ! audioconvert ! audioamplify amplification=1.5 clipping-method=wrap-positive ! audioconvert ! alsasink
|
|
* ]|
|
|
*
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include <gst/gst.h>
|
|
#include <gst/base/gstbasetransform.h>
|
|
#include <gst/audio/audio.h>
|
|
#include <gst/audio/gstaudiofilter.h>
|
|
|
|
#include "audioamplify.h"
|
|
|
|
#define GST_CAT_DEFAULT gst_audio_amplify_debug
|
|
GST_DEBUG_CATEGORY_STATIC (GST_CAT_DEFAULT);
|
|
|
|
/* Filter signals and args */
|
|
enum
|
|
{
|
|
/* FILL ME */
|
|
LAST_SIGNAL
|
|
};
|
|
|
|
enum
|
|
{
|
|
PROP_0,
|
|
PROP_AMPLIFICATION,
|
|
PROP_CLIPPING_METHOD
|
|
};
|
|
|
|
enum
|
|
{
|
|
METHOD_CLIP = 0,
|
|
METHOD_WRAP_NEGATIVE,
|
|
METHOD_WRAP_POSITIVE,
|
|
METHOD_NOCLIP,
|
|
NUM_METHODS
|
|
};
|
|
|
|
#define GST_TYPE_AUDIO_AMPLIFY_CLIPPING_METHOD (gst_audio_amplify_clipping_method_get_type ())
|
|
static GType
|
|
gst_audio_amplify_clipping_method_get_type (void)
|
|
{
|
|
static GType gtype = 0;
|
|
|
|
if (gtype == 0) {
|
|
static const GEnumValue values[] = {
|
|
{METHOD_CLIP, "Normal clipping (default)", "clip"},
|
|
{METHOD_WRAP_NEGATIVE,
|
|
"Push overdriven values back from the opposite side",
|
|
"wrap-negative"},
|
|
{METHOD_WRAP_POSITIVE, "Push overdriven values back from the same side",
|
|
"wrap-positive"},
|
|
{METHOD_NOCLIP, "No clipping", "none"},
|
|
{0, NULL, NULL}
|
|
};
|
|
gtype = g_enum_register_static ("GstAudioAmplifyClippingMethod", values);
|
|
}
|
|
return gtype;
|
|
}
|
|
|
|
#define ALLOWED_CAPS \
|
|
"audio/x-raw," \
|
|
" format=(string) {S8,"GST_AUDIO_NE(S16)","GST_AUDIO_NE(S32)"," \
|
|
GST_AUDIO_NE(F32)","GST_AUDIO_NE(F64)"}," \
|
|
" rate=(int)[1,MAX]," \
|
|
" channels=(int)[1,MAX], " \
|
|
" layout=(string) {interleaved, non-interleaved}"
|
|
|
|
G_DEFINE_TYPE (GstAudioAmplify, gst_audio_amplify, GST_TYPE_AUDIO_FILTER);
|
|
|
|
static gboolean gst_audio_amplify_set_process_function (GstAudioAmplify *
|
|
filter, gint clipping, GstAudioFormat format);
|
|
static void gst_audio_amplify_set_property (GObject * object, guint prop_id,
|
|
const GValue * value, GParamSpec * pspec);
|
|
static void gst_audio_amplify_get_property (GObject * object, guint prop_id,
|
|
GValue * value, GParamSpec * pspec);
|
|
|
|
static gboolean gst_audio_amplify_setup (GstAudioFilter * filter,
|
|
const GstAudioInfo * info);
|
|
static GstFlowReturn gst_audio_amplify_transform_ip (GstBaseTransform * base,
|
|
GstBuffer * buf);
|
|
|
|
#define MIN_gint8 G_MININT8
|
|
#define MAX_gint8 G_MAXINT8
|
|
#define MIN_gint16 G_MININT16
|
|
#define MAX_gint16 G_MAXINT16
|
|
#define MIN_gint32 G_MININT32
|
|
#define MAX_gint32 G_MAXINT32
|
|
|
|
#define MAKE_INT_FUNCS(type,largetype) \
|
|
static void \
|
|
gst_audio_amplify_transform_##type##_clip (GstAudioAmplify * filter, \
|
|
void * data, guint num_samples) \
|
|
{ \
|
|
type *d = data; \
|
|
\
|
|
while (num_samples--) { \
|
|
largetype val = *d * filter->amplification; \
|
|
*d++ = CLAMP (val, MIN_##type, MAX_##type); \
|
|
} \
|
|
} \
|
|
static void \
|
|
gst_audio_amplify_transform_##type##_wrap_negative (GstAudioAmplify * filter, \
|
|
void * data, guint num_samples) \
|
|
{ \
|
|
type *d = data; \
|
|
\
|
|
while (num_samples--) { \
|
|
largetype val = *d * filter->amplification; \
|
|
if (val > MAX_##type) \
|
|
val = MIN_##type + (val - MIN_##type) % ((largetype) MAX_##type + 1 - \
|
|
MIN_##type); \
|
|
else if (val < MIN_##type) \
|
|
val = MAX_##type - (MAX_##type - val) % ((largetype) MAX_##type + 1 - \
|
|
MIN_##type); \
|
|
*d++ = val; \
|
|
} \
|
|
} \
|
|
static void \
|
|
gst_audio_amplify_transform_##type##_wrap_positive (GstAudioAmplify * filter, \
|
|
void * data, guint num_samples) \
|
|
{ \
|
|
type *d = data; \
|
|
\
|
|
while (num_samples--) { \
|
|
largetype val = *d * filter->amplification; \
|
|
do { \
|
|
if (val > MAX_##type) \
|
|
val = MAX_##type - (val - MAX_##type); \
|
|
else if (val < MIN_##type) \
|
|
val = MIN_##type + (MIN_##type - val); \
|
|
else \
|
|
break; \
|
|
} while (1); \
|
|
*d++ = val; \
|
|
} \
|
|
} \
|
|
static void \
|
|
gst_audio_amplify_transform_##type##_noclip (GstAudioAmplify * filter, \
|
|
void * data, guint num_samples) \
|
|
{ \
|
|
type *d = data; \
|
|
\
|
|
while (num_samples--) \
|
|
*d++ *= filter->amplification; \
|
|
}
|
|
|
|
#define MAKE_FLOAT_FUNCS(type) \
|
|
static void \
|
|
gst_audio_amplify_transform_##type##_clip (GstAudioAmplify * filter, \
|
|
void * data, guint num_samples) \
|
|
{ \
|
|
type *d = data; \
|
|
\
|
|
while (num_samples--) { \
|
|
type val = *d* filter->amplification; \
|
|
*d++ = CLAMP (val, -1.0, +1.0); \
|
|
} \
|
|
} \
|
|
static void \
|
|
gst_audio_amplify_transform_##type##_wrap_negative (GstAudioAmplify * \
|
|
filter, void * data, guint num_samples) \
|
|
{ \
|
|
type *d = data; \
|
|
\
|
|
while (num_samples--) { \
|
|
type val = *d * filter->amplification; \
|
|
do { \
|
|
if (val > 1.0) \
|
|
val = -1.0 + (val - 1.0); \
|
|
else if (val < -1.0) \
|
|
val = 1.0 - (1.0 - val); \
|
|
else \
|
|
break; \
|
|
} while (1); \
|
|
*d++ = val; \
|
|
} \
|
|
} \
|
|
static void \
|
|
gst_audio_amplify_transform_##type##_wrap_positive (GstAudioAmplify * filter, \
|
|
void * data, guint num_samples) \
|
|
{ \
|
|
type *d = data; \
|
|
\
|
|
while (num_samples--) { \
|
|
type val = *d* filter->amplification; \
|
|
do { \
|
|
if (val > 1.0) \
|
|
val = 1.0 - (val - 1.0); \
|
|
else if (val < -1.0) \
|
|
val = -1.0 + (-1.0 - val); \
|
|
else \
|
|
break; \
|
|
} while (1); \
|
|
*d++ = val; \
|
|
} \
|
|
} \
|
|
static void \
|
|
gst_audio_amplify_transform_##type##_noclip (GstAudioAmplify * filter, \
|
|
void * data, guint num_samples) \
|
|
{ \
|
|
type *d = data; \
|
|
\
|
|
while (num_samples--) \
|
|
*d++ *= filter->amplification; \
|
|
}
|
|
|
|
/* *INDENT-OFF* */
|
|
MAKE_INT_FUNCS (gint8,gint)
|
|
MAKE_INT_FUNCS (gint16,gint)
|
|
MAKE_INT_FUNCS (gint32,gint64)
|
|
MAKE_FLOAT_FUNCS (gfloat)
|
|
MAKE_FLOAT_FUNCS (gdouble)
|
|
/* *INDENT-ON* */
|
|
|
|
/* GObject vmethod implementations */
|
|
|
|
static void
|
|
gst_audio_amplify_class_init (GstAudioAmplifyClass * klass)
|
|
{
|
|
GObjectClass *gobject_class;
|
|
GstElementClass *gstelement_class;
|
|
GstCaps *caps;
|
|
|
|
GST_DEBUG_CATEGORY_INIT (gst_audio_amplify_debug, "audioamplify", 0,
|
|
"audioamplify element");
|
|
|
|
gobject_class = (GObjectClass *) klass;
|
|
gstelement_class = (GstElementClass *) klass;
|
|
|
|
gobject_class->set_property = gst_audio_amplify_set_property;
|
|
gobject_class->get_property = gst_audio_amplify_get_property;
|
|
|
|
g_object_class_install_property (gobject_class, PROP_AMPLIFICATION,
|
|
g_param_spec_float ("amplification", "Amplification",
|
|
"Factor of amplification", -G_MAXFLOAT, G_MAXFLOAT,
|
|
1.0,
|
|
G_PARAM_READWRITE | GST_PARAM_CONTROLLABLE | G_PARAM_STATIC_STRINGS));
|
|
|
|
/**
|
|
* GstAudioAmplify:clipping-method
|
|
*
|
|
* Clipping method: clip mode set values higher than the maximum to the
|
|
* maximum. The wrap-negative mode pushes those values back from the
|
|
* opposite side, wrap-positive pushes them back from the same side.
|
|
*
|
|
**/
|
|
g_object_class_install_property (gobject_class, PROP_CLIPPING_METHOD,
|
|
g_param_spec_enum ("clipping-method", "Clipping method",
|
|
"Selects how to handle values higher than the maximum",
|
|
GST_TYPE_AUDIO_AMPLIFY_CLIPPING_METHOD, METHOD_CLIP,
|
|
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
|
|
|
gst_element_class_set_static_metadata (gstelement_class, "Audio amplifier",
|
|
"Filter/Effect/Audio",
|
|
"Amplifies an audio stream by a given factor",
|
|
"Sebastian Dröge <slomo@circular-chaos.org>");
|
|
|
|
caps = gst_caps_from_string (ALLOWED_CAPS);
|
|
gst_audio_filter_class_add_pad_templates (GST_AUDIO_FILTER_CLASS (klass),
|
|
caps);
|
|
gst_caps_unref (caps);
|
|
|
|
GST_BASE_TRANSFORM_CLASS (klass)->transform_ip =
|
|
GST_DEBUG_FUNCPTR (gst_audio_amplify_transform_ip);
|
|
GST_BASE_TRANSFORM_CLASS (klass)->transform_ip_on_passthrough = FALSE;
|
|
|
|
GST_AUDIO_FILTER_CLASS (klass)->setup =
|
|
GST_DEBUG_FUNCPTR (gst_audio_amplify_setup);
|
|
|
|
gst_type_mark_as_plugin_api (GST_TYPE_AUDIO_AMPLIFY_CLIPPING_METHOD, 0);
|
|
}
|
|
|
|
static void
|
|
gst_audio_amplify_init (GstAudioAmplify * filter)
|
|
{
|
|
filter->amplification = 1.0;
|
|
gst_audio_amplify_set_process_function (filter, METHOD_CLIP,
|
|
GST_AUDIO_FORMAT_S16);
|
|
gst_base_transform_set_in_place (GST_BASE_TRANSFORM (filter), TRUE);
|
|
gst_base_transform_set_gap_aware (GST_BASE_TRANSFORM (filter), TRUE);
|
|
}
|
|
|
|
static GstAudioAmplifyProcessFunc
|
|
gst_audio_amplify_process_function (gint clipping, GstAudioFormat format)
|
|
{
|
|
static const struct process
|
|
{
|
|
GstAudioFormat format;
|
|
gint clipping;
|
|
GstAudioAmplifyProcessFunc func;
|
|
} process[] = {
|
|
{
|
|
GST_AUDIO_FORMAT_F32, METHOD_CLIP, gst_audio_amplify_transform_gfloat_clip}, {
|
|
GST_AUDIO_FORMAT_F32, METHOD_WRAP_NEGATIVE,
|
|
gst_audio_amplify_transform_gfloat_wrap_negative}, {
|
|
GST_AUDIO_FORMAT_F32, METHOD_WRAP_POSITIVE,
|
|
gst_audio_amplify_transform_gfloat_wrap_positive}, {
|
|
GST_AUDIO_FORMAT_F32, METHOD_NOCLIP,
|
|
gst_audio_amplify_transform_gfloat_noclip}, {
|
|
GST_AUDIO_FORMAT_F64, METHOD_CLIP,
|
|
gst_audio_amplify_transform_gdouble_clip}, {
|
|
GST_AUDIO_FORMAT_F64, METHOD_WRAP_NEGATIVE,
|
|
gst_audio_amplify_transform_gdouble_wrap_negative}, {
|
|
GST_AUDIO_FORMAT_F64, METHOD_WRAP_POSITIVE,
|
|
gst_audio_amplify_transform_gdouble_wrap_positive}, {
|
|
GST_AUDIO_FORMAT_F64, METHOD_NOCLIP,
|
|
gst_audio_amplify_transform_gdouble_noclip}, {
|
|
GST_AUDIO_FORMAT_S8, METHOD_CLIP, gst_audio_amplify_transform_gint8_clip}, {
|
|
GST_AUDIO_FORMAT_S8, METHOD_WRAP_NEGATIVE,
|
|
gst_audio_amplify_transform_gint8_wrap_negative}, {
|
|
GST_AUDIO_FORMAT_S8, METHOD_WRAP_POSITIVE,
|
|
gst_audio_amplify_transform_gint8_wrap_positive}, {
|
|
GST_AUDIO_FORMAT_S8, METHOD_NOCLIP,
|
|
gst_audio_amplify_transform_gint8_noclip}, {
|
|
GST_AUDIO_FORMAT_S16, METHOD_CLIP, gst_audio_amplify_transform_gint16_clip}, {
|
|
GST_AUDIO_FORMAT_S16, METHOD_WRAP_NEGATIVE,
|
|
gst_audio_amplify_transform_gint16_wrap_negative}, {
|
|
GST_AUDIO_FORMAT_S16, METHOD_WRAP_POSITIVE,
|
|
gst_audio_amplify_transform_gint16_wrap_positive}, {
|
|
GST_AUDIO_FORMAT_S16, METHOD_NOCLIP,
|
|
gst_audio_amplify_transform_gint16_noclip}, {
|
|
GST_AUDIO_FORMAT_S32, METHOD_CLIP, gst_audio_amplify_transform_gint32_clip}, {
|
|
GST_AUDIO_FORMAT_S32, METHOD_WRAP_NEGATIVE,
|
|
gst_audio_amplify_transform_gint32_wrap_negative}, {
|
|
GST_AUDIO_FORMAT_S32, METHOD_WRAP_POSITIVE,
|
|
gst_audio_amplify_transform_gint32_wrap_positive}, {
|
|
GST_AUDIO_FORMAT_S32, METHOD_NOCLIP,
|
|
gst_audio_amplify_transform_gint32_noclip}, {
|
|
0, 0, NULL}
|
|
};
|
|
const struct process *p;
|
|
|
|
for (p = process; p->func; p++)
|
|
if (p->format == format && p->clipping == clipping)
|
|
return p->func;
|
|
return NULL;
|
|
}
|
|
|
|
static gboolean
|
|
gst_audio_amplify_set_process_function (GstAudioAmplify * filter, gint
|
|
clipping_method, GstAudioFormat format)
|
|
{
|
|
GstAudioAmplifyProcessFunc process;
|
|
|
|
/* set processing function */
|
|
|
|
process = gst_audio_amplify_process_function (clipping_method, format);
|
|
if (!process) {
|
|
GST_DEBUG ("wrong format");
|
|
return FALSE;
|
|
}
|
|
|
|
filter->process = process;
|
|
filter->clipping_method = clipping_method;
|
|
filter->format = format;
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static void
|
|
gst_audio_amplify_set_property (GObject * object, guint prop_id,
|
|
const GValue * value, GParamSpec * pspec)
|
|
{
|
|
GstAudioAmplify *filter = GST_AUDIO_AMPLIFY (object);
|
|
|
|
switch (prop_id) {
|
|
case PROP_AMPLIFICATION:
|
|
filter->amplification = g_value_get_float (value);
|
|
gst_base_transform_set_passthrough (GST_BASE_TRANSFORM (filter),
|
|
filter->amplification == 1.0);
|
|
break;
|
|
case PROP_CLIPPING_METHOD:
|
|
gst_audio_amplify_set_process_function (filter, g_value_get_enum (value),
|
|
filter->format);
|
|
break;
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
|
break;
|
|
}
|
|
}
|
|
|
|
static void
|
|
gst_audio_amplify_get_property (GObject * object, guint prop_id,
|
|
GValue * value, GParamSpec * pspec)
|
|
{
|
|
GstAudioAmplify *filter = GST_AUDIO_AMPLIFY (object);
|
|
|
|
switch (prop_id) {
|
|
case PROP_AMPLIFICATION:
|
|
g_value_set_float (value, filter->amplification);
|
|
break;
|
|
case PROP_CLIPPING_METHOD:
|
|
g_value_set_enum (value, filter->clipping_method);
|
|
break;
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
|
break;
|
|
}
|
|
}
|
|
|
|
/* GstAudioFilter vmethod implementations */
|
|
static gboolean
|
|
gst_audio_amplify_setup (GstAudioFilter * base, const GstAudioInfo * info)
|
|
{
|
|
GstAudioAmplify *filter = GST_AUDIO_AMPLIFY (base);
|
|
|
|
return gst_audio_amplify_set_process_function (filter,
|
|
filter->clipping_method, GST_AUDIO_INFO_FORMAT (info));
|
|
}
|
|
|
|
/* GstBaseTransform vmethod implementations */
|
|
static GstFlowReturn
|
|
gst_audio_amplify_transform_ip (GstBaseTransform * base, GstBuffer * buf)
|
|
{
|
|
GstAudioAmplify *filter = GST_AUDIO_AMPLIFY (base);
|
|
guint num_samples;
|
|
GstClockTime timestamp, stream_time;
|
|
GstMapInfo map;
|
|
|
|
timestamp = GST_BUFFER_TIMESTAMP (buf);
|
|
stream_time =
|
|
gst_segment_to_stream_time (&base->segment, GST_FORMAT_TIME, timestamp);
|
|
|
|
GST_DEBUG_OBJECT (filter, "sync to %" GST_TIME_FORMAT,
|
|
GST_TIME_ARGS (timestamp));
|
|
|
|
if (GST_CLOCK_TIME_IS_VALID (stream_time))
|
|
gst_object_sync_values (GST_OBJECT (filter), stream_time);
|
|
|
|
if (G_UNLIKELY (GST_BUFFER_FLAG_IS_SET (buf, GST_BUFFER_FLAG_GAP)))
|
|
return GST_FLOW_OK;
|
|
|
|
gst_buffer_map (buf, &map, GST_MAP_READWRITE);
|
|
num_samples = map.size / GST_AUDIO_FILTER_BPS (filter);
|
|
|
|
filter->process (filter, map.data, num_samples);
|
|
|
|
gst_buffer_unmap (buf, &map);
|
|
|
|
return GST_FLOW_OK;
|
|
}
|