gst/filter/gstbpwsinc.*: Add support for a bandreject mode and allow specifying the window function that should be used.

Original commit message from CVS:
* gst/filter/gstbpwsinc.c: (gst_bpwsinc_mode_get_type),
(gst_bpwsinc_window_get_type), (gst_bpwsinc_class_init),
(gst_bpwsinc_init), (bpwsinc_build_kernel), (bpwsinc_set_property),
(bpwsinc_get_property):
* gst/filter/gstbpwsinc.h:
Add support for a bandreject mode and allow specifying the window
function that should be used.
* gst/filter/gstlpwsinc.c:
And another small formatting fix.
This commit is contained in:
Sebastian Dröge 2007-08-10 05:35:25 +00:00
parent ec97d67741
commit f2b50b7bd2
4 changed files with 123 additions and 16 deletions

View file

@ -1,3 +1,15 @@
2007-08-10 Sebastian Dröge <slomo@circular-chaos.org>
* gst/filter/gstbpwsinc.c: (gst_bpwsinc_mode_get_type),
(gst_bpwsinc_window_get_type), (gst_bpwsinc_class_init),
(gst_bpwsinc_init), (bpwsinc_build_kernel), (bpwsinc_set_property),
(bpwsinc_get_property):
* gst/filter/gstbpwsinc.h:
Add support for a bandreject mode and allow specifying the window
function that should be used.
* gst/filter/gstlpwsinc.c:
And another small formatting fix.
2007-08-10 Sebastian Dröge <slomo@circular-chaos.org> 2007-08-10 Sebastian Dröge <slomo@circular-chaos.org>
* gst/filter/gstbpwsinc.c: (gst_bpwsinc_class_init), * gst/filter/gstbpwsinc.c: (gst_bpwsinc_class_init),

View file

@ -29,8 +29,8 @@
* TODO: - Implement the convolution in place, probably only makes sense * TODO: - Implement the convolution in place, probably only makes sense
* when using FFT convolution as currently the convolution itself * when using FFT convolution as currently the convolution itself
* is probably the bottleneck * is probably the bottleneck
* - Implement a band reject mode (spectral inversion) * - Maybe allow cascading the filter to get a better stopband attenuation.
* - Allow choosing between different windows (blackman, hanning, ...) * Can be done by convolving a filter kernel with itself
*/ */
#ifdef HAVE_CONFIG_H #ifdef HAVE_CONFIG_H
@ -69,9 +69,63 @@ enum
PROP_0, PROP_0,
PROP_LENGTH, PROP_LENGTH,
PROP_LOWER_FREQUENCY, PROP_LOWER_FREQUENCY,
PROP_UPPER_FREQUENCY PROP_UPPER_FREQUENCY,
PROP_MODE,
PROP_WINDOW
}; };
enum
{
MODE_BAND_PASS = 0,
MODE_BAND_REJECT
};
#define GST_TYPE_BPWSINC_MODE (gst_bpwsinc_mode_get_type ())
static GType
gst_bpwsinc_mode_get_type (void)
{
static GType gtype = 0;
if (gtype == 0) {
static const GEnumValue values[] = {
{MODE_BAND_PASS, "Band pass (default)",
"band-pass"},
{MODE_BAND_REJECT, "Band reject",
"band-reject"},
{0, NULL, NULL}
};
gtype = g_enum_register_static ("GstBPWSincMode", values);
}
return gtype;
}
enum
{
WINDOW_HAMMING = 0,
WINDOW_BLACKMAN
};
#define GST_TYPE_BPWSINC_WINDOW (gst_bpwsinc_window_get_type ())
static GType
gst_bpwsinc_window_get_type (void)
{
static GType gtype = 0;
if (gtype == 0) {
static const GEnumValue values[] = {
{WINDOW_HAMMING, "Hamming window (default)",
"hamming"},
{WINDOW_BLACKMAN, "Blackman window",
"blackman"},
{0, NULL, NULL}
};
gtype = g_enum_register_static ("GstBPWSincWindow", values);
}
return gtype;
}
#define ALLOWED_CAPS \ #define ALLOWED_CAPS \
"audio/x-raw-float, " \ "audio/x-raw-float, " \
" width = (int) { 32, 64 }, " \ " width = (int) { 32, 64 }, " \
@ -157,6 +211,16 @@ gst_bpwsinc_class_init (GstBPWSincClass * klass)
"Filter kernel length, will be rounded to the next odd number", "Filter kernel length, will be rounded to the next odd number",
3, G_MAXINT, 101, G_PARAM_READWRITE)); 3, G_MAXINT, 101, G_PARAM_READWRITE));
g_object_class_install_property (gobject_class, PROP_MODE,
g_param_spec_enum ("mode", "Mode",
"Band pass or band reject mode", GST_TYPE_BPWSINC_MODE,
MODE_BAND_PASS, G_PARAM_READWRITE | GST_PARAM_CONTROLLABLE));
g_object_class_install_property (gobject_class, PROP_WINDOW,
g_param_spec_enum ("window", "Window",
"Window function to use", GST_TYPE_BPWSINC_WINDOW,
WINDOW_HAMMING, G_PARAM_READWRITE | GST_PARAM_CONTROLLABLE));
trans_class->transform = GST_DEBUG_FUNCPTR (bpwsinc_transform); trans_class->transform = GST_DEBUG_FUNCPTR (bpwsinc_transform);
trans_class->get_unit_size = GST_DEBUG_FUNCPTR (bpwsinc_get_unit_size); trans_class->get_unit_size = GST_DEBUG_FUNCPTR (bpwsinc_get_unit_size);
GST_AUDIO_FILTER_CLASS (klass)->setup = GST_DEBUG_FUNCPTR (bpwsinc_setup); GST_AUDIO_FILTER_CLASS (klass)->setup = GST_DEBUG_FUNCPTR (bpwsinc_setup);
@ -168,6 +232,8 @@ gst_bpwsinc_init (GstBPWSinc * self, GstBPWSincClass * g_class)
self->kernel_length = 101; self->kernel_length = 101;
self->lower_frequency = 0.0; self->lower_frequency = 0.0;
self->upper_frequency = 0.0; self->upper_frequency = 0.0;
self->mode = MODE_BAND_PASS;
self->window = WINDOW_HAMMING;
self->kernel = NULL; self->kernel = NULL;
self->have_kernel = FALSE; self->have_kernel = FALSE;
self->residue = NULL; self->residue = NULL;
@ -273,9 +339,13 @@ bpwsinc_build_kernel (GstBPWSinc * self)
else else
kernel_lp[i] = sin (w * (i - len / 2)) kernel_lp[i] = sin (w * (i - len / 2))
/ (i - len / 2); / (i - len / 2);
/* Blackman windowing */ /* Windowing */
kernel_lp[i] *= (0.42 - 0.5 * cos (2 * M_PI * i / len) if (self->window == WINDOW_HAMMING)
+ 0.08 * cos (4 * M_PI * i / len)); kernel_lp[i] *= (0.54 - 0.46 * cos (2 * M_PI * i / len));
else
kernel_lp[i] *=
(0.42 - 0.5 * cos (2 * M_PI * i / len) +
0.08 * cos (4 * M_PI * i / len));
} }
/* normalize for unity gain at DC */ /* normalize for unity gain at DC */
@ -297,9 +367,13 @@ bpwsinc_build_kernel (GstBPWSinc * self)
else else
kernel_hp[i] = sin (w * (i - len / 2)) kernel_hp[i] = sin (w * (i - len / 2))
/ (i - len / 2); / (i - len / 2);
/* Blackman windowing */ /* Windowing */
kernel_hp[i] *= (0.42 - 0.5 * cos (2 * M_PI * i / len) if (self->window == WINDOW_HAMMING)
+ 0.08 * cos (4 * M_PI * i / len)); kernel_hp[i] *= (0.54 - 0.46 * cos (2 * M_PI * i / len));
else
kernel_hp[i] *=
(0.42 - 0.5 * cos (2 * M_PI * i / len) +
0.08 * cos (4 * M_PI * i / len));
} }
/* normalize for unity gain at DC */ /* normalize for unity gain at DC */
@ -326,10 +400,13 @@ bpwsinc_build_kernel (GstBPWSinc * self)
g_free (kernel_lp); g_free (kernel_lp);
g_free (kernel_hp); g_free (kernel_hp);
/* do spectral inversion to go from bandreject to bandpass */ /* do spectral inversion to go from bandreject to bandpass
for (i = 0; i < len; ++i) * if specified */
self->kernel[i] = -self->kernel[i]; if (self->mode == MODE_BAND_PASS) {
self->kernel[len / 2] += 1; for (i = 0; i < len; ++i)
self->kernel[i] = -self->kernel[i];
self->kernel[len / 2] += 1;
}
/* set up the residue memory space */ /* set up the residue memory space */
if (self->residue) if (self->residue)
@ -441,6 +518,18 @@ bpwsinc_set_property (GObject * object, guint prop_id, const GValue * value,
bpwsinc_build_kernel (self); bpwsinc_build_kernel (self);
GST_BASE_TRANSFORM_UNLOCK (self); GST_BASE_TRANSFORM_UNLOCK (self);
break; break;
case PROP_MODE:
GST_BASE_TRANSFORM_LOCK (self);
self->mode = g_value_get_enum (value);
bpwsinc_build_kernel (self);
GST_BASE_TRANSFORM_UNLOCK (self);
break;
case PROP_WINDOW:
GST_BASE_TRANSFORM_LOCK (self);
self->window = g_value_get_enum (value);
bpwsinc_build_kernel (self);
GST_BASE_TRANSFORM_UNLOCK (self);
break;
default: default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break; break;
@ -463,6 +552,12 @@ bpwsinc_get_property (GObject * object, guint prop_id, GValue * value,
case PROP_UPPER_FREQUENCY: case PROP_UPPER_FREQUENCY:
g_value_set_double (value, self->upper_frequency); g_value_set_double (value, self->upper_frequency);
break; break;
case PROP_MODE:
g_value_set_enum (value, self->mode);
break;
case PROP_WINDOW:
g_value_set_enum (value, self->window);
break;
default: default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break; break;

View file

@ -62,7 +62,8 @@ struct _GstBPWSinc {
GstBPWSincProcessFunc process; GstBPWSincProcessFunc process;
gdouble frequency; gint mode;
gint window;
gdouble lower_frequency, upper_frequency; gdouble lower_frequency, upper_frequency;
gint kernel_length; /* length of the filter kernel */ gint kernel_length; /* length of the filter kernel */

View file

@ -28,7 +28,7 @@
* *
* TODO: - Implement the convolution in place, probably only makes sense * TODO: - Implement the convolution in place, probably only makes sense
* when using FFT convolution as currently the convolution itself * when using FFT convolution as currently the convolution itself
* is probably the bottleneck. * is probably the bottleneck
* - Maybe allow cascading the filter to get a better stopband attenuation. * - Maybe allow cascading the filter to get a better stopband attenuation.
* Can be done by convolving a filter kernel with itself. * Can be done by convolving a filter kernel with itself.
*/ */
@ -124,7 +124,6 @@ gst_lpwsinc_window_get_type (void)
return gtype; return gtype;
} }
#define ALLOWED_CAPS \ #define ALLOWED_CAPS \
"audio/x-raw-float, " \ "audio/x-raw-float, " \
" width = (int) { 32, 64 }, " \ " width = (int) { 32, 64 }, " \