mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-11-28 04:31:06 +00:00
7b90bf3215
Previously we advanced the in_data pointer by bps for every channel, and then later again for block_size*bps. This caused us to be one sample further than expected if an input buffer covered two analysis frames. And in the end lead to completely bogus values reported by level. https://bugzilla.gnome.org/show_bug.cgi?id=746065
783 lines
27 KiB
C
783 lines
27 KiB
C
/* GStreamer
|
|
* Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
|
|
* Copyright (C) 2000,2001,2002,2003,2005
|
|
* Thomas Vander Stichele <thomas at apestaart dot org>
|
|
*
|
|
* 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-level
|
|
*
|
|
* Level analyses incoming audio buffers and, if the #GstLevel:message property
|
|
* is #TRUE, generates an element message named
|
|
* <classname>"level"</classname>:
|
|
* after each interval of time given by the #GstLevel:interval property.
|
|
* The message's structure contains these fields:
|
|
* <itemizedlist>
|
|
* <listitem>
|
|
* <para>
|
|
* #GstClockTime
|
|
* <classname>"timestamp"</classname>:
|
|
* the timestamp of the buffer that triggered the message.
|
|
* </para>
|
|
* </listitem>
|
|
* <listitem>
|
|
* <para>
|
|
* #GstClockTime
|
|
* <classname>"stream-time"</classname>:
|
|
* the stream time of the buffer.
|
|
* </para>
|
|
* </listitem>
|
|
* <listitem>
|
|
* <para>
|
|
* #GstClockTime
|
|
* <classname>"running-time"</classname>:
|
|
* the running_time of the buffer.
|
|
* </para>
|
|
* </listitem>
|
|
* <listitem>
|
|
* <para>
|
|
* #GstClockTime
|
|
* <classname>"duration"</classname>:
|
|
* the duration of the buffer.
|
|
* </para>
|
|
* </listitem>
|
|
* <listitem>
|
|
* <para>
|
|
* #GstClockTime
|
|
* <classname>"endtime"</classname>:
|
|
* the end time of the buffer that triggered the message as stream time (this
|
|
* is deprecated, as it can be calculated from stream-time + duration)
|
|
* </para>
|
|
* </listitem>
|
|
* <listitem>
|
|
* <para>
|
|
* #GValueArray of #gdouble
|
|
* <classname>"peak"</classname>:
|
|
* the peak power level in dB for each channel
|
|
* </para>
|
|
* </listitem>
|
|
* <listitem>
|
|
* <para>
|
|
* #GValueArray of #gdouble
|
|
* <classname>"decay"</classname>:
|
|
* the decaying peak power level in dB for each channel
|
|
* The decaying peak level follows the peak level, but starts dropping if no
|
|
* new peak is reached after the time given by the #GstLevel:peak-ttl.
|
|
* When the decaying peak level drops, it does so at the decay rate as
|
|
* specified by the #GstLevel:peak-falloff.
|
|
* </para>
|
|
* </listitem>
|
|
* <listitem>
|
|
* <para>
|
|
* #GValueArray of #gdouble
|
|
* <classname>"rms"</classname>:
|
|
* the Root Mean Square (or average power) level in dB for each channel
|
|
* </para>
|
|
* </listitem>
|
|
* </itemizedlist>
|
|
*
|
|
* <refsect2>
|
|
* <title>Example application</title>
|
|
* <informalexample><programlisting language="C">
|
|
* <xi:include xmlns:xi="http://www.w3.org/2003/XInclude" parse="text" href="../../../../tests/examples/level/level-example.c" />
|
|
* </programlisting></informalexample>
|
|
* </refsect2>
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
/* FIXME 0.11: suppress warnings for deprecated API such as GValueArray
|
|
* with newer GLib versions (>= 2.31.0) */
|
|
#define GLIB_DISABLE_DEPRECATION_WARNINGS
|
|
|
|
#include <string.h>
|
|
#include <math.h>
|
|
#include <gst/gst.h>
|
|
#include <gst/audio/audio.h>
|
|
|
|
#include "gstlevel.h"
|
|
|
|
GST_DEBUG_CATEGORY_STATIC (level_debug);
|
|
#define GST_CAT_DEFAULT level_debug
|
|
|
|
#define EPSILON 1e-35f
|
|
|
|
static GstStaticPadTemplate sink_template_factory =
|
|
GST_STATIC_PAD_TEMPLATE ("sink",
|
|
GST_PAD_SINK,
|
|
GST_PAD_ALWAYS,
|
|
GST_STATIC_CAPS ("audio/x-raw, "
|
|
"format = (string) { S8, " GST_AUDIO_NE (S16) ", " GST_AUDIO_NE (S32)
|
|
", " GST_AUDIO_NE (F32) "," GST_AUDIO_NE (F64) " },"
|
|
"layout = (string) interleaved, "
|
|
"rate = (int) [ 1, MAX ], " "channels = (int) [ 1, MAX ]")
|
|
);
|
|
|
|
static GstStaticPadTemplate src_template_factory =
|
|
GST_STATIC_PAD_TEMPLATE ("src",
|
|
GST_PAD_SRC,
|
|
GST_PAD_ALWAYS,
|
|
GST_STATIC_CAPS ("audio/x-raw, "
|
|
"format = (string) { S8, " GST_AUDIO_NE (S16) ", " GST_AUDIO_NE (S32)
|
|
", " GST_AUDIO_NE (F32) "," GST_AUDIO_NE (F64) " },"
|
|
"layout = (string) interleaved, "
|
|
"rate = (int) [ 1, MAX ], " "channels = (int) [ 1, MAX ]")
|
|
);
|
|
|
|
enum
|
|
{
|
|
PROP_0,
|
|
PROP_POST_MESSAGES,
|
|
PROP_MESSAGE,
|
|
PROP_INTERVAL,
|
|
PROP_PEAK_TTL,
|
|
PROP_PEAK_FALLOFF
|
|
};
|
|
|
|
#define gst_level_parent_class parent_class
|
|
G_DEFINE_TYPE (GstLevel, gst_level, GST_TYPE_BASE_TRANSFORM);
|
|
|
|
static void gst_level_set_property (GObject * object, guint prop_id,
|
|
const GValue * value, GParamSpec * pspec);
|
|
static void gst_level_get_property (GObject * object, guint prop_id,
|
|
GValue * value, GParamSpec * pspec);
|
|
static void gst_level_finalize (GObject * obj);
|
|
|
|
static gboolean gst_level_set_caps (GstBaseTransform * trans, GstCaps * in,
|
|
GstCaps * out);
|
|
static gboolean gst_level_start (GstBaseTransform * trans);
|
|
static GstFlowReturn gst_level_transform_ip (GstBaseTransform * trans,
|
|
GstBuffer * in);
|
|
static void gst_level_post_message (GstLevel * filter);
|
|
static gboolean gst_level_sink_event (GstBaseTransform * trans,
|
|
GstEvent * event);
|
|
|
|
|
|
static void
|
|
gst_level_class_init (GstLevelClass * klass)
|
|
{
|
|
GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
|
|
GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
|
|
GstBaseTransformClass *trans_class = GST_BASE_TRANSFORM_CLASS (klass);
|
|
|
|
gobject_class->set_property = gst_level_set_property;
|
|
gobject_class->get_property = gst_level_get_property;
|
|
gobject_class->finalize = gst_level_finalize;
|
|
|
|
/**
|
|
* GstLevel:post-messages
|
|
*
|
|
* Post messages on the bus with level information.
|
|
*
|
|
* Since: 1.1.0
|
|
*/
|
|
g_object_class_install_property (gobject_class, PROP_POST_MESSAGES,
|
|
g_param_spec_boolean ("post-messages", "Post Messages",
|
|
"Whether to post a 'level' element message on the bus for each "
|
|
"passed interval", TRUE, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
|
/* FIXME(2.0): remove this property */
|
|
/**
|
|
* GstLevel:post-messages
|
|
*
|
|
* Post messages on the bus with level information.
|
|
*
|
|
* Deprecated: use the #GstLevel:post-messages property
|
|
*/
|
|
#ifndef GST_REMOVE_DEPRECATED
|
|
g_object_class_install_property (gobject_class, PROP_MESSAGE,
|
|
g_param_spec_boolean ("message", "message",
|
|
"Post a 'level' message for each passed interval "
|
|
"(deprecated, use the post-messages property instead)", TRUE,
|
|
G_PARAM_READWRITE | G_PARAM_DEPRECATED | G_PARAM_STATIC_STRINGS));
|
|
#endif
|
|
g_object_class_install_property (gobject_class, PROP_INTERVAL,
|
|
g_param_spec_uint64 ("interval", "Interval",
|
|
"Interval of time between message posts (in nanoseconds)",
|
|
1, G_MAXUINT64, GST_SECOND / 10,
|
|
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
|
g_object_class_install_property (gobject_class, PROP_PEAK_TTL,
|
|
g_param_spec_uint64 ("peak-ttl", "Peak TTL",
|
|
"Time To Live of decay peak before it falls back (in nanoseconds)",
|
|
0, G_MAXUINT64, GST_SECOND / 10 * 3,
|
|
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
|
g_object_class_install_property (gobject_class, PROP_PEAK_FALLOFF,
|
|
g_param_spec_double ("peak-falloff", "Peak Falloff",
|
|
"Decay rate of decay peak after TTL (in dB/sec)",
|
|
0.0, G_MAXDOUBLE, 10.0, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
|
|
|
GST_DEBUG_CATEGORY_INIT (level_debug, "level", 0, "Level calculation");
|
|
|
|
gst_element_class_add_pad_template (element_class,
|
|
gst_static_pad_template_get (&sink_template_factory));
|
|
gst_element_class_add_pad_template (element_class,
|
|
gst_static_pad_template_get (&src_template_factory));
|
|
gst_element_class_set_static_metadata (element_class, "Level",
|
|
"Filter/Analyzer/Audio",
|
|
"RMS/Peak/Decaying Peak Level messager for audio/raw",
|
|
"Thomas Vander Stichele <thomas at apestaart dot org>");
|
|
|
|
trans_class->set_caps = GST_DEBUG_FUNCPTR (gst_level_set_caps);
|
|
trans_class->start = GST_DEBUG_FUNCPTR (gst_level_start);
|
|
trans_class->transform_ip = GST_DEBUG_FUNCPTR (gst_level_transform_ip);
|
|
trans_class->sink_event = GST_DEBUG_FUNCPTR (gst_level_sink_event);
|
|
trans_class->passthrough_on_same_caps = TRUE;
|
|
}
|
|
|
|
static void
|
|
gst_level_init (GstLevel * filter)
|
|
{
|
|
filter->CS = NULL;
|
|
filter->peak = NULL;
|
|
filter->last_peak = NULL;
|
|
filter->decay_peak = NULL;
|
|
filter->decay_peak_base = NULL;
|
|
filter->decay_peak_age = NULL;
|
|
|
|
gst_audio_info_init (&filter->info);
|
|
|
|
filter->interval = GST_SECOND / 10;
|
|
filter->decay_peak_ttl = GST_SECOND / 10 * 3;
|
|
filter->decay_peak_falloff = 10.0; /* dB falloff (/sec) */
|
|
|
|
filter->post_messages = TRUE;
|
|
|
|
filter->process = NULL;
|
|
|
|
gst_base_transform_set_gap_aware (GST_BASE_TRANSFORM (filter), TRUE);
|
|
}
|
|
|
|
static void
|
|
gst_level_finalize (GObject * obj)
|
|
{
|
|
GstLevel *filter = GST_LEVEL (obj);
|
|
|
|
g_free (filter->CS);
|
|
g_free (filter->peak);
|
|
g_free (filter->last_peak);
|
|
g_free (filter->decay_peak);
|
|
g_free (filter->decay_peak_base);
|
|
g_free (filter->decay_peak_age);
|
|
|
|
filter->CS = NULL;
|
|
filter->peak = NULL;
|
|
filter->last_peak = NULL;
|
|
filter->decay_peak = NULL;
|
|
filter->decay_peak_base = NULL;
|
|
filter->decay_peak_age = NULL;
|
|
|
|
G_OBJECT_CLASS (parent_class)->finalize (obj);
|
|
}
|
|
|
|
static void
|
|
gst_level_set_property (GObject * object, guint prop_id,
|
|
const GValue * value, GParamSpec * pspec)
|
|
{
|
|
GstLevel *filter = GST_LEVEL (object);
|
|
|
|
switch (prop_id) {
|
|
case PROP_POST_MESSAGES:
|
|
/* fall-through */
|
|
case PROP_MESSAGE:
|
|
filter->post_messages = g_value_get_boolean (value);
|
|
break;
|
|
case PROP_INTERVAL:
|
|
filter->interval = g_value_get_uint64 (value);
|
|
if (GST_AUDIO_INFO_RATE (&filter->info)) {
|
|
filter->interval_frames =
|
|
GST_CLOCK_TIME_TO_FRAMES (filter->interval,
|
|
GST_AUDIO_INFO_RATE (&filter->info));
|
|
}
|
|
break;
|
|
case PROP_PEAK_TTL:
|
|
filter->decay_peak_ttl =
|
|
gst_guint64_to_gdouble (g_value_get_uint64 (value));
|
|
break;
|
|
case PROP_PEAK_FALLOFF:
|
|
filter->decay_peak_falloff = g_value_get_double (value);
|
|
break;
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
|
break;
|
|
}
|
|
}
|
|
|
|
static void
|
|
gst_level_get_property (GObject * object, guint prop_id,
|
|
GValue * value, GParamSpec * pspec)
|
|
{
|
|
GstLevel *filter = GST_LEVEL (object);
|
|
|
|
switch (prop_id) {
|
|
case PROP_POST_MESSAGES:
|
|
/* fall-through */
|
|
case PROP_MESSAGE:
|
|
g_value_set_boolean (value, filter->post_messages);
|
|
break;
|
|
case PROP_INTERVAL:
|
|
g_value_set_uint64 (value, filter->interval);
|
|
break;
|
|
case PROP_PEAK_TTL:
|
|
g_value_set_uint64 (value, filter->decay_peak_ttl);
|
|
break;
|
|
case PROP_PEAK_FALLOFF:
|
|
g_value_set_double (value, filter->decay_peak_falloff);
|
|
break;
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
|
break;
|
|
}
|
|
}
|
|
|
|
|
|
/* process one (interleaved) channel of incoming samples
|
|
* calculate square sum of samples
|
|
* normalize and average over number of samples
|
|
* returns a normalized cumulative square value, which can be averaged
|
|
* to return the average power as a double between 0 and 1
|
|
* also returns the normalized peak power (square of the highest amplitude)
|
|
*
|
|
* caller must assure num is a multiple of channels
|
|
* samples for multiple channels are interleaved
|
|
* input sample data enters in *in_data and is not modified
|
|
* this filter only accepts signed audio data, so mid level is always 0
|
|
*
|
|
* for integers, this code considers the non-existant positive max value to be
|
|
* full-scale; so max-1 will not map to 1.0
|
|
*/
|
|
|
|
#define DEFINE_INT_LEVEL_CALCULATOR(TYPE, RESOLUTION) \
|
|
static void inline \
|
|
gst_level_calculate_##TYPE (gpointer data, guint num, guint channels, \
|
|
gdouble *NCS, gdouble *NPS) \
|
|
{ \
|
|
TYPE * in = (TYPE *)data; \
|
|
register guint j; \
|
|
gdouble squaresum = 0.0; /* square sum of the input samples */ \
|
|
register gdouble square = 0.0; /* Square */ \
|
|
register gdouble peaksquare = 0.0; /* Peak Square Sample */ \
|
|
gdouble normalizer; /* divisor to get a [-1.0, 1.0] range */ \
|
|
\
|
|
/* *NCS = 0.0; Normalized Cumulative Square */ \
|
|
/* *NPS = 0.0; Normalized Peak Square */ \
|
|
\
|
|
for (j = 0; j < num; j += channels) { \
|
|
square = ((gdouble) in[j]) * in[j]; \
|
|
if (square > peaksquare) peaksquare = square; \
|
|
squaresum += square; \
|
|
} \
|
|
\
|
|
normalizer = (gdouble) (G_GINT64_CONSTANT(1) << (RESOLUTION * 2)); \
|
|
*NCS = squaresum / normalizer; \
|
|
*NPS = peaksquare / normalizer; \
|
|
}
|
|
|
|
DEFINE_INT_LEVEL_CALCULATOR (gint32, 31);
|
|
DEFINE_INT_LEVEL_CALCULATOR (gint16, 15);
|
|
DEFINE_INT_LEVEL_CALCULATOR (gint8, 7);
|
|
|
|
/* FIXME: use orc to calculate squaresums? */
|
|
#define DEFINE_FLOAT_LEVEL_CALCULATOR(TYPE) \
|
|
static void inline \
|
|
gst_level_calculate_##TYPE (gpointer data, guint num, guint channels, \
|
|
gdouble *NCS, gdouble *NPS) \
|
|
{ \
|
|
TYPE * in = (TYPE *)data; \
|
|
register guint j; \
|
|
gdouble squaresum = 0.0; /* square sum of the input samples */ \
|
|
register gdouble square = 0.0; /* Square */ \
|
|
register gdouble peaksquare = 0.0; /* Peak Square Sample */ \
|
|
\
|
|
/* *NCS = 0.0; Normalized Cumulative Square */ \
|
|
/* *NPS = 0.0; Normalized Peak Square */ \
|
|
\
|
|
/* orc_level_squaresum_f64(&squaresum,in,num); */ \
|
|
for (j = 0; j < num; j += channels) { \
|
|
square = ((gdouble) in[j]) * in[j]; \
|
|
if (square > peaksquare) peaksquare = square; \
|
|
squaresum += square; \
|
|
} \
|
|
\
|
|
*NCS = squaresum; \
|
|
*NPS = peaksquare; \
|
|
}
|
|
|
|
DEFINE_FLOAT_LEVEL_CALCULATOR (gfloat);
|
|
DEFINE_FLOAT_LEVEL_CALCULATOR (gdouble);
|
|
|
|
/* we would need stride to deinterleave also
|
|
static void inline
|
|
gst_level_calculate_gdouble (gpointer data, guint num, guint channels,
|
|
gdouble *NCS, gdouble *NPS)
|
|
{
|
|
orc_level_squaresum_f64(NCS,(gdouble *)data,num);
|
|
*NPS = 0.0;
|
|
}
|
|
*/
|
|
|
|
|
|
static gboolean
|
|
gst_level_set_caps (GstBaseTransform * trans, GstCaps * in, GstCaps * out)
|
|
{
|
|
GstLevel *filter = GST_LEVEL (trans);
|
|
GstAudioInfo info;
|
|
gint i, channels, rate;
|
|
|
|
if (!gst_audio_info_from_caps (&info, in))
|
|
return FALSE;
|
|
|
|
switch (GST_AUDIO_INFO_FORMAT (&info)) {
|
|
case GST_AUDIO_FORMAT_S8:
|
|
filter->process = gst_level_calculate_gint8;
|
|
break;
|
|
case GST_AUDIO_FORMAT_S16:
|
|
filter->process = gst_level_calculate_gint16;
|
|
break;
|
|
case GST_AUDIO_FORMAT_S32:
|
|
filter->process = gst_level_calculate_gint32;
|
|
break;
|
|
case GST_AUDIO_FORMAT_F32:
|
|
filter->process = gst_level_calculate_gfloat;
|
|
break;
|
|
case GST_AUDIO_FORMAT_F64:
|
|
filter->process = gst_level_calculate_gdouble;
|
|
break;
|
|
default:
|
|
filter->process = NULL;
|
|
break;
|
|
}
|
|
|
|
filter->info = info;
|
|
|
|
channels = GST_AUDIO_INFO_CHANNELS (&info);
|
|
rate = GST_AUDIO_INFO_RATE (&info);
|
|
|
|
/* allocate channel variable arrays */
|
|
g_free (filter->CS);
|
|
g_free (filter->peak);
|
|
g_free (filter->last_peak);
|
|
g_free (filter->decay_peak);
|
|
g_free (filter->decay_peak_base);
|
|
g_free (filter->decay_peak_age);
|
|
filter->CS = g_new (gdouble, channels);
|
|
filter->peak = g_new (gdouble, channels);
|
|
filter->last_peak = g_new (gdouble, channels);
|
|
filter->decay_peak = g_new (gdouble, channels);
|
|
filter->decay_peak_base = g_new (gdouble, channels);
|
|
|
|
filter->decay_peak_age = g_new (GstClockTime, channels);
|
|
|
|
for (i = 0; i < channels; ++i) {
|
|
filter->CS[i] = filter->peak[i] = filter->last_peak[i] =
|
|
filter->decay_peak[i] = filter->decay_peak_base[i] = 0.0;
|
|
filter->decay_peak_age[i] = G_GUINT64_CONSTANT (0);
|
|
}
|
|
|
|
filter->interval_frames = GST_CLOCK_TIME_TO_FRAMES (filter->interval, rate);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static gboolean
|
|
gst_level_start (GstBaseTransform * trans)
|
|
{
|
|
GstLevel *filter = GST_LEVEL (trans);
|
|
|
|
filter->num_frames = 0;
|
|
filter->message_ts = GST_CLOCK_TIME_NONE;
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static GstMessage *
|
|
gst_level_message_new (GstLevel * level, GstClockTime timestamp,
|
|
GstClockTime duration)
|
|
{
|
|
GstBaseTransform *trans = GST_BASE_TRANSFORM_CAST (level);
|
|
GstStructure *s;
|
|
GValue v = { 0, };
|
|
GstClockTime endtime, running_time, stream_time;
|
|
|
|
running_time = gst_segment_to_running_time (&trans->segment, GST_FORMAT_TIME,
|
|
timestamp);
|
|
stream_time = gst_segment_to_stream_time (&trans->segment, GST_FORMAT_TIME,
|
|
timestamp);
|
|
/* endtime is for backwards compatibility */
|
|
endtime = stream_time + duration;
|
|
|
|
s = gst_structure_new ("level",
|
|
"endtime", GST_TYPE_CLOCK_TIME, endtime,
|
|
"timestamp", G_TYPE_UINT64, timestamp,
|
|
"stream-time", G_TYPE_UINT64, stream_time,
|
|
"running-time", G_TYPE_UINT64, running_time,
|
|
"duration", G_TYPE_UINT64, duration, NULL);
|
|
|
|
g_value_init (&v, G_TYPE_VALUE_ARRAY);
|
|
g_value_take_boxed (&v, g_value_array_new (0));
|
|
gst_structure_take_value (s, "rms", &v);
|
|
|
|
g_value_init (&v, G_TYPE_VALUE_ARRAY);
|
|
g_value_take_boxed (&v, g_value_array_new (0));
|
|
gst_structure_take_value (s, "peak", &v);
|
|
|
|
g_value_init (&v, G_TYPE_VALUE_ARRAY);
|
|
g_value_take_boxed (&v, g_value_array_new (0));
|
|
gst_structure_take_value (s, "decay", &v);
|
|
|
|
return gst_message_new_element (GST_OBJECT (level), s);
|
|
}
|
|
|
|
static void
|
|
gst_level_message_append_channel (GstMessage * m, gdouble rms, gdouble peak,
|
|
gdouble decay)
|
|
{
|
|
const GValue *array_val;
|
|
GstStructure *s;
|
|
GValueArray *arr;
|
|
GValue v = { 0, };
|
|
|
|
g_value_init (&v, G_TYPE_DOUBLE);
|
|
|
|
s = (GstStructure *) gst_message_get_structure (m);
|
|
|
|
array_val = gst_structure_get_value (s, "rms");
|
|
arr = (GValueArray *) g_value_get_boxed (array_val);
|
|
g_value_set_double (&v, rms);
|
|
g_value_array_append (arr, &v); /* copies by value */
|
|
|
|
array_val = gst_structure_get_value (s, "peak");
|
|
arr = (GValueArray *) g_value_get_boxed (array_val);
|
|
g_value_set_double (&v, peak);
|
|
g_value_array_append (arr, &v); /* copies by value */
|
|
|
|
array_val = gst_structure_get_value (s, "decay");
|
|
arr = (GValueArray *) g_value_get_boxed (array_val);
|
|
g_value_set_double (&v, decay);
|
|
g_value_array_append (arr, &v); /* copies by value */
|
|
|
|
g_value_unset (&v);
|
|
}
|
|
|
|
static GstFlowReturn
|
|
gst_level_transform_ip (GstBaseTransform * trans, GstBuffer * in)
|
|
{
|
|
GstLevel *filter;
|
|
GstMapInfo map;
|
|
guint8 *in_data;
|
|
gsize in_size;
|
|
gdouble CS;
|
|
guint i;
|
|
guint num_frames;
|
|
guint num_int_samples = 0; /* number of interleaved samples
|
|
* ie. total count for all channels combined */
|
|
guint block_size, block_int_size; /* we subdivide buffers to not skip message
|
|
* intervals */
|
|
GstClockTimeDiff falloff_time;
|
|
gint channels, rate, bps;
|
|
|
|
filter = GST_LEVEL (trans);
|
|
|
|
channels = GST_AUDIO_INFO_CHANNELS (&filter->info);
|
|
bps = GST_AUDIO_INFO_BPS (&filter->info);
|
|
rate = GST_AUDIO_INFO_RATE (&filter->info);
|
|
|
|
gst_buffer_map (in, &map, GST_MAP_READ);
|
|
in_data = map.data;
|
|
in_size = map.size;
|
|
|
|
num_int_samples = in_size / bps;
|
|
|
|
GST_LOG_OBJECT (filter, "analyzing %u sample frames at ts %" GST_TIME_FORMAT,
|
|
num_int_samples, GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (in)));
|
|
|
|
g_return_val_if_fail (num_int_samples % channels == 0, GST_FLOW_ERROR);
|
|
|
|
if (GST_BUFFER_FLAG_IS_SET (in, GST_BUFFER_FLAG_DISCONT)) {
|
|
filter->message_ts = GST_BUFFER_TIMESTAMP (in);
|
|
filter->num_frames = 0;
|
|
}
|
|
if (G_UNLIKELY (!GST_CLOCK_TIME_IS_VALID (filter->message_ts))) {
|
|
filter->message_ts = GST_BUFFER_TIMESTAMP (in);
|
|
}
|
|
|
|
num_frames = num_int_samples / channels;
|
|
while (num_frames > 0) {
|
|
block_size = filter->interval_frames - filter->num_frames;
|
|
block_size = MIN (block_size, num_frames);
|
|
block_int_size = block_size * channels;
|
|
|
|
for (i = 0; i < channels; ++i) {
|
|
if (!GST_BUFFER_FLAG_IS_SET (in, GST_BUFFER_FLAG_GAP)) {
|
|
filter->process (in_data + (bps * i), block_int_size, channels, &CS,
|
|
&filter->peak[i]);
|
|
GST_LOG_OBJECT (filter,
|
|
"[%d]: cumulative squares %lf, over %d samples/%d channels",
|
|
i, CS, block_int_size, channels);
|
|
filter->CS[i] += CS;
|
|
} else {
|
|
filter->peak[i] = 0.0;
|
|
}
|
|
|
|
filter->decay_peak_age[i] += GST_FRAMES_TO_CLOCK_TIME (num_frames, rate);
|
|
GST_LOG_OBJECT (filter,
|
|
"[%d]: peak %f, last peak %f, decay peak %f, age %" GST_TIME_FORMAT,
|
|
i, filter->peak[i], filter->last_peak[i], filter->decay_peak[i],
|
|
GST_TIME_ARGS (filter->decay_peak_age[i]));
|
|
|
|
/* update running peak */
|
|
if (filter->peak[i] > filter->last_peak[i])
|
|
filter->last_peak[i] = filter->peak[i];
|
|
|
|
/* make decay peak fall off if too old */
|
|
falloff_time =
|
|
GST_CLOCK_DIFF (gst_gdouble_to_guint64 (filter->decay_peak_ttl),
|
|
filter->decay_peak_age[i]);
|
|
if (falloff_time > 0) {
|
|
gdouble falloff_dB;
|
|
gdouble falloff;
|
|
gdouble length; /* length of falloff time in seconds */
|
|
|
|
length = (gdouble) falloff_time / (gdouble) GST_SECOND;
|
|
falloff_dB = filter->decay_peak_falloff * length;
|
|
falloff = pow (10, falloff_dB / -20.0);
|
|
|
|
GST_LOG_OBJECT (filter,
|
|
"falloff: current %f, base %f, interval %" GST_TIME_FORMAT
|
|
", dB falloff %f, factor %e",
|
|
filter->decay_peak[i], filter->decay_peak_base[i],
|
|
GST_TIME_ARGS (falloff_time), falloff_dB, falloff);
|
|
filter->decay_peak[i] = filter->decay_peak_base[i] * falloff;
|
|
GST_LOG_OBJECT (filter,
|
|
"peak is %" GST_TIME_FORMAT " old, decayed with factor %e to %f",
|
|
GST_TIME_ARGS (filter->decay_peak_age[i]), falloff,
|
|
filter->decay_peak[i]);
|
|
} else {
|
|
GST_LOG_OBJECT (filter, "peak not old enough, not decaying");
|
|
}
|
|
|
|
/* if the peak of this run is higher, the decay peak gets reset */
|
|
if (filter->peak[i] >= filter->decay_peak[i]) {
|
|
GST_LOG_OBJECT (filter, "new peak, %f", filter->peak[i]);
|
|
filter->decay_peak[i] = filter->peak[i];
|
|
filter->decay_peak_base[i] = filter->peak[i];
|
|
filter->decay_peak_age[i] = G_GINT64_CONSTANT (0);
|
|
}
|
|
}
|
|
in_data += block_size * bps;
|
|
|
|
filter->num_frames += block_size;
|
|
num_frames -= block_size;
|
|
|
|
/* do we need to message ? */
|
|
if (filter->num_frames >= filter->interval_frames) {
|
|
gst_level_post_message (filter);
|
|
}
|
|
}
|
|
|
|
gst_buffer_unmap (in, &map);
|
|
|
|
return GST_FLOW_OK;
|
|
}
|
|
|
|
static void
|
|
gst_level_post_message (GstLevel * filter)
|
|
{
|
|
guint i;
|
|
gint channels, rate, frames = filter->num_frames;
|
|
GstClockTime duration;
|
|
|
|
channels = GST_AUDIO_INFO_CHANNELS (&filter->info);
|
|
rate = GST_AUDIO_INFO_RATE (&filter->info);
|
|
duration = GST_FRAMES_TO_CLOCK_TIME (frames, rate);
|
|
|
|
if (filter->post_messages) {
|
|
GstMessage *m =
|
|
gst_level_message_new (filter, filter->message_ts, duration);
|
|
|
|
GST_LOG_OBJECT (filter,
|
|
"message: ts %" GST_TIME_FORMAT ", duration %" GST_TIME_FORMAT
|
|
", num_frames %d", GST_TIME_ARGS (filter->message_ts),
|
|
GST_TIME_ARGS (duration), frames);
|
|
|
|
for (i = 0; i < channels; ++i) {
|
|
gdouble RMS;
|
|
gdouble RMSdB, peakdB, decaydB;
|
|
|
|
RMS = sqrt (filter->CS[i] / frames);
|
|
GST_LOG_OBJECT (filter,
|
|
"message: channel %d, CS %f, RMS %f", i, filter->CS[i], RMS);
|
|
GST_LOG_OBJECT (filter,
|
|
"message: last_peak: %f, decay_peak: %f",
|
|
filter->last_peak[i], filter->decay_peak[i]);
|
|
/* RMS values are calculated in amplitude, so 20 * log 10 */
|
|
RMSdB = 20 * log10 (RMS + EPSILON);
|
|
/* peak values are square sums, ie. power, so 10 * log 10 */
|
|
peakdB = 10 * log10 (filter->last_peak[i] + EPSILON);
|
|
decaydB = 10 * log10 (filter->decay_peak[i] + EPSILON);
|
|
|
|
if (filter->decay_peak[i] < filter->last_peak[i]) {
|
|
/* this can happen in certain cases, for example when
|
|
* the last peak is between decay_peak and decay_peak_base */
|
|
GST_DEBUG_OBJECT (filter,
|
|
"message: decay peak dB %f smaller than last peak dB %f, copying",
|
|
decaydB, peakdB);
|
|
filter->decay_peak[i] = filter->last_peak[i];
|
|
}
|
|
GST_LOG_OBJECT (filter,
|
|
"message: RMS %f dB, peak %f dB, decay %f dB",
|
|
RMSdB, peakdB, decaydB);
|
|
|
|
gst_level_message_append_channel (m, RMSdB, peakdB, decaydB);
|
|
|
|
/* reset cumulative and normal peak */
|
|
filter->CS[i] = 0.0;
|
|
filter->last_peak[i] = 0.0;
|
|
}
|
|
|
|
gst_element_post_message (GST_ELEMENT (filter), m);
|
|
|
|
}
|
|
filter->num_frames -= frames;
|
|
filter->message_ts += duration;
|
|
}
|
|
|
|
|
|
static gboolean
|
|
gst_level_sink_event (GstBaseTransform * trans, GstEvent * event)
|
|
{
|
|
if (GST_EVENT_TYPE (event) == GST_EVENT_EOS) {
|
|
GstLevel *filter = GST_LEVEL (trans);
|
|
|
|
gst_level_post_message (filter);
|
|
}
|
|
|
|
return GST_BASE_TRANSFORM_CLASS (parent_class)->sink_event (trans, event);
|
|
}
|
|
|
|
static gboolean
|
|
plugin_init (GstPlugin * plugin)
|
|
{
|
|
return gst_element_register (plugin, "level", GST_RANK_NONE, GST_TYPE_LEVEL);
|
|
}
|
|
|
|
GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
|
|
GST_VERSION_MINOR,
|
|
level,
|
|
"Audio level plugin",
|
|
plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN);
|