mirror of https://code.videolan.org/videolan/vlc
224 lines
7.3 KiB
C
224 lines
7.3 KiB
C
/*****************************************************************************
|
|
* lowpass.c :
|
|
*****************************************************************************
|
|
* Copyright © 2022 VLC authors and VideoLAN
|
|
*
|
|
* Authors: Razdutt Sarnaik <rajduttsarnaik@gmail.com>
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU Lesser General Public License as published by
|
|
* the Free Software Foundation; either version 2.1 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program 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 Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public License
|
|
* along with this program; if not, write to the Free Software Foundation,
|
|
* Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
|
|
*****************************************************************************/
|
|
|
|
/*****************************************************************************
|
|
* Preamble
|
|
*****************************************************************************/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
# include "config.h"
|
|
#endif
|
|
|
|
|
|
#include "avaudio.h"
|
|
#include "avaudio_lists.h"
|
|
|
|
|
|
typedef struct
|
|
{
|
|
float freq;
|
|
float poles;
|
|
char* width_type;
|
|
float width;
|
|
float mix;
|
|
|
|
av_filter avf;
|
|
}avfilter_sys_t;
|
|
|
|
|
|
static int Open_lowpass( vlc_object_t *p_this );
|
|
static void Close_lowpass( filter_t *p_filter );
|
|
|
|
static char* string_generate_lowpass(avfilter_sys_t *p_sys)
|
|
{
|
|
if( vlc_memstream_open( &p_sys->avf.str ) != 0 )
|
|
return NULL;
|
|
|
|
vlc_memstream_printf(&p_sys->avf.str, "lowpass=f=%f:p=%f:t=%s:w=%f:m=%f",
|
|
p_sys->freq, p_sys->poles, p_sys->width_type, p_sys->width, p_sys->mix);
|
|
|
|
if(vlc_memstream_close(&p_sys->avf.str))
|
|
return NULL;
|
|
|
|
return p_sys->avf.str.ptr;
|
|
}
|
|
|
|
|
|
/*****************************************************************************
|
|
* Module descriptor
|
|
*****************************************************************************/
|
|
|
|
#define LOWPASS_FREQUENCY_TEXT N_( "frequency" )
|
|
#define LOWPASS_FREQUENCY_LONGTEXT N_( "Set frequency in Hz. Default is 500. Range is [0-10000]" )
|
|
|
|
#define LOWPASS_POLES_TEXT N_( "poles" )
|
|
#define LOWPASS_POLES_LONGTEXT N_( "Set number of poles. Default is 2." \
|
|
" The filter can be either single-pole or double-pole" )
|
|
|
|
#define LOWPASS_WIDTH_TYPE_TEXT N_( "width type" )
|
|
#define LOWPASS_WIDTH_TYPE_LONGTEXT N_( "Set method to specify band-width of filter." )
|
|
|
|
#define LOWPASS_WIDTH_TEXT N_( "width" )
|
|
#define LOWPASS_WIDTH_LONGTEXT N_( "Specify the band-width of a filter in width_type units." \
|
|
" Applies only to double-pole filter. Default 0.7" )
|
|
|
|
#define LOWPASS_MIX_TEXT N_( "mix" )
|
|
#define LOWPASS_MIX_LONGTEXT N_( "Set mixing amount. Default is 1." \
|
|
" Range is [0 - 1]")
|
|
|
|
vlc_module_begin()
|
|
|
|
set_shortname( N_("Lowpass") )
|
|
set_description( N_("Lowpass filter") )
|
|
set_subcategory( SUBCAT_AUDIO_AFILTER )
|
|
set_capability( "audio filter", 0 )
|
|
|
|
add_float_with_range( "lowpass-frequency", 500, 0.1, 10000,
|
|
LOWPASS_FREQUENCY_TEXT, LOWPASS_FREQUENCY_LONGTEXT )
|
|
|
|
add_float_with_range( "lowpass-poles", 2, 1, 2,
|
|
LOWPASS_POLES_TEXT, LOWPASS_POLES_LONGTEXT )
|
|
|
|
add_string( "lowpass-width-type", "h", LOWPASS_WIDTH_TYPE_TEXT,
|
|
LOWPASS_WIDTH_TYPE_LONGTEXT )
|
|
change_string_list( lowpass_width_type_list, lowpass_width_type_list_text )
|
|
|
|
add_float_with_range( "lowpass-width", 0.7, 0.1, 1000,
|
|
LOWPASS_WIDTH_TEXT, LOWPASS_WIDTH_LONGTEXT )
|
|
|
|
add_float_with_range( "lowpass-mix", 1, 0, 1,
|
|
LOWPASS_MIX_TEXT, LOWPASS_MIX_LONGTEXT )
|
|
|
|
add_shortcut("lowpass")
|
|
set_callback( Open_lowpass )
|
|
|
|
vlc_module_end()
|
|
|
|
|
|
/*****************************************************************************
|
|
* Open: initialize filter
|
|
*****************************************************************************/
|
|
|
|
static block_t *Process_lowpass( filter_t *p_filter, block_t *p_block )
|
|
{
|
|
avfilter_sys_t *p_sys = p_filter->p_sys;
|
|
|
|
if(!Process_avaudio_2(p_filter,&p_sys->avf))
|
|
return p_block;
|
|
p_sys->avf.avfilter.frame_len_flag = 1;
|
|
return Process_avaudio(p_filter, p_block, &p_sys->avf.avfilter, p_sys->avf.avfilter.frame_len_flag);
|
|
}
|
|
|
|
static int paramCallback_lowpass( vlc_object_t *p_this, char const *psz_cmd,
|
|
vlc_value_t oldval, vlc_value_t newval,
|
|
void * p_data )
|
|
{
|
|
VLC_UNUSED(p_this); VLC_UNUSED(oldval);
|
|
|
|
avfilter_sys_t *p_sys;
|
|
p_sys = p_data;
|
|
|
|
vlc_mutex_lock(&p_sys->avf.lock);
|
|
|
|
if( !strcmp( psz_cmd, "lowpass-frequency" ) )
|
|
p_sys->freq = newval.f_float;
|
|
if( !strcmp( psz_cmd, "lowpass-poles" ) )
|
|
p_sys->poles = newval.f_float;
|
|
if( !strcmp( psz_cmd, "lowpass-width-type" ) )
|
|
{
|
|
free(p_sys->width_type);
|
|
p_sys->width_type = strdup(newval.psz_string);
|
|
if(p_sys->width_type == NULL)
|
|
return VLC_EGENERIC;
|
|
}
|
|
if( !strcmp( psz_cmd, "lowpass-width" ) )
|
|
p_sys->width = newval.f_float;
|
|
if( !strcmp( psz_cmd, "lowpass-mix" ) )
|
|
p_sys->mix = newval.f_float;
|
|
|
|
if(string_generate_lowpass(p_sys) == NULL)
|
|
return VLC_EGENERIC;
|
|
|
|
p_sys->avf.params_changed = true;
|
|
|
|
vlc_mutex_unlock(&p_sys->avf.lock);
|
|
|
|
|
|
return VLC_SUCCESS;
|
|
}
|
|
|
|
static int Open_lowpass( vlc_object_t *p_this )
|
|
{
|
|
filter_t *p_filter = (filter_t *)p_this;
|
|
avfilter_sys_t *p_sys = malloc( sizeof( *p_sys ) );
|
|
|
|
if( unlikely( p_sys == NULL ) )
|
|
return VLC_ENOMEM;
|
|
|
|
|
|
p_filter->p_sys = p_sys;
|
|
|
|
vlc_object_t *p_aout = vlc_object_parent(p_filter);
|
|
p_sys->freq = var_CreateGetFloat(p_aout,"lowpass-frequency");
|
|
p_sys->poles = var_CreateGetFloat(p_aout,"lowpass-poles");
|
|
p_sys->width_type = var_CreateGetString(p_aout,"lowpass-width-type");
|
|
p_sys->width = var_CreateGetFloat(p_aout,"lowpass-width");
|
|
p_sys->mix = var_CreateGetFloat(p_aout,"lowpass-mix");
|
|
|
|
p_sys->avf.params_changed = false;
|
|
vlc_mutex_init( &p_sys->avf.lock );
|
|
var_AddCallback( p_aout, "lowpass-frequency", paramCallback_lowpass, p_sys );
|
|
var_AddCallback( p_aout, "lowpass-poles", paramCallback_lowpass, p_sys );
|
|
var_AddCallback( p_aout, "lowpass-width-type", paramCallback_lowpass, p_sys );
|
|
var_AddCallback( p_aout, "lowpass-width", paramCallback_lowpass, p_sys );
|
|
var_AddCallback( p_aout, "lowpass-mix", paramCallback_lowpass, p_sys );
|
|
|
|
p_filter->fmt_out.audio = p_filter->fmt_in.audio;
|
|
|
|
|
|
/* initialize ffmpeg filter */
|
|
if(init_filters(string_generate_lowpass(p_sys),p_filter,&p_sys->avf.avfilter) < 0)
|
|
{
|
|
avfilter_graph_free(&p_sys->avf.avfilter.filter_graph);
|
|
free(p_sys);
|
|
return VLC_EGENERIC;
|
|
}
|
|
/*filter description*/
|
|
msg_Dbg(p_filter,"lowpass initialized:%s",p_sys->avf.str.ptr);
|
|
free(p_sys->avf.str.ptr);
|
|
|
|
static const struct vlc_filter_operations filter_ops_lowpass =
|
|
{ .filter_audio = Process_lowpass, .close = Close_lowpass };
|
|
p_filter->ops = &filter_ops_lowpass;
|
|
|
|
return VLC_SUCCESS;
|
|
}
|
|
|
|
static void Close_lowpass( filter_t *p_filter )
|
|
{
|
|
avfilter_sys_t *p_sys = p_filter->p_sys;
|
|
|
|
avfilter_graph_free(&p_sys->avf.avfilter.filter_graph);
|
|
free(p_sys);
|
|
}
|
|
|