/***************************************************************************** * vuvuzela.c: ***************************************************************************** * * BASED on original VLC param_eq.c - minor changes: * * 16.06.2010 / Steffen Buehl * * - rewrite param-eq to vuvuzela * - set "special" defaults (Vuvuzela frequencies) * - add 3 more parameters * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 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 General Public License for more details. * * You should have received a copy of the GNU 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 #include #include #include #define _(str) (str) #define N_(str) (str) /***************************************************************************** * Module descriptor *****************************************************************************/ static int Open ( vlc_object_t * ); static void Close( vlc_object_t * ); static void CalcPeakEQCoeffs( float, float, float, float, float * ); static void CalcShelfEQCoeffs( float, float, float, int, float, float * ); static void ProcessEQ( float *, float *, float *, unsigned, unsigned, float *, unsigned ); static void DoWork( aout_instance_t *, aout_filter_t *, aout_buffer_t *, aout_buffer_t * ); vlc_module_begin () set_description( N_("Vuvuzela Filter (sb) 1.1") ) set_shortname( N_("Vuvuzela Filter" ) ) set_capability( "audio filter", 0 ) set_category( CAT_AUDIO ) set_subcategory( SUBCAT_AUDIO_AFILTER ) add_float( "vuvuzela-lowf", 100, NULL, N_("Low freq (Hz)"),"", false ) add_float_with_range( "vuvuzela-lowgain", 10, -20.0, 20.0, NULL, N_("Low freq gain (dB)"), "",false ) add_float( "vuvuzela-highf", 10000, NULL, N_("High freq (Hz)"),"", false ) add_float_with_range( "vuvuzela-highgain", 0, -20.0, 20.0, NULL, N_("High freq gain (dB)"),"",false ) add_float( "vuvuzela-f1", 233, NULL, N_("Freq 1 (Hz)"),"", false ) add_float( "vuvuzela-f2", 466, NULL, N_("Freq 2 (Hz)"),"", false ) add_float( "vuvuzela-f3", 932, NULL, N_("Freq 3 (Hz)"),"", false ) add_float( "vuvuzela-f4", 1880, NULL, N_("Freq 4 (Hz)"),"", false ) add_float( "vuvuzela-f5", 2577, NULL, N_("Freq 5 (Hz)"),"", false ) add_float( "vuvuzela-f6", 3365, NULL, N_("Freq 6 (Hz)"),"", false ) add_float_with_range( "vuvuzela-gain1", -20.0, -20.0, 20.0, NULL, N_("Freq 1 gain (dB)"), "",false ) add_float_with_range( "vuvuzela-q1", 7, 0.1, 100.0, NULL, N_("Freq 1 Q"), "",false ) add_float_with_range( "vuvuzela-gain2", -20.0, -20.0, 20.0, NULL, N_("Freq 2 gain (dB)"),"",false ) add_float_with_range( "vuvuzela-q2", 7, 0.1, 100.0, NULL, N_("Freq 2 Q"),"",false ) add_float_with_range( "vuvuzela-gain3", -20.0, -20.0, 20.0, NULL, N_("Freq 3 gain (dB)"),"",false ) add_float_with_range( "vuvuzela-q3", 7, 0.1, 100.0, NULL, N_("Freq 3 Q"),"",false ) add_float_with_range( "vuvuzela-gain4", -20.0, -20.0, 20.0, NULL, N_("Freq 4 gain (dB)"),"",false ) add_float_with_range( "vuvuzela-q4", 7, 0.1, 100.0, NULL, N_("Freq 4 Q"),"",false ) add_float_with_range( "vuvuzela-gain5", -20.0, -20.0, 20.0, NULL, N_("Freq 5 gain (dB)"),"",false ) add_float_with_range( "vuvuzela-q5", 7, 0.1, 100.0, NULL, N_("Freq 5 Q"),"",false ) add_float_with_range( "vuvuzela-gain6", -20.0, -20.0, 20.0, NULL, N_("Freq 6 gain (dB)"),"",false ) add_float_with_range( "vuvuzela-q6", 7, 0.1, 100.0, NULL, N_("Freq 6 Q"),"",false ) set_callbacks( Open, Close ) vlc_module_end () /***************************************************************************** * Local prototypes *****************************************************************************/ struct aout_filter_sys_t { /* Filter static config */ float f_lowf, f_lowgain; float f_f1, f_Q1, f_gain1; float f_f2, f_Q2, f_gain2; float f_f3, f_Q3, f_gain3; float f_f4, f_Q4, f_gain4; float f_f5, f_Q6, f_gain5; float f_f6, f_Q5, f_gain6; float f_highf, f_highgain; /* Filter computed coeffs */ float coeffs[8*5]; /* State */ float *p_state; }; /***************************************************************************** * Open: *****************************************************************************/ static int Open( vlc_object_t *p_this ) { aout_filter_t *p_filter = (aout_filter_t *)p_this; aout_filter_sys_t *p_sys; bool b_fit = true; int i_samplerate; if( p_filter->input.i_format != VLC_FOURCC('f','l','3','2' ) || p_filter->output.i_format != VLC_FOURCC('f','l','3','2') ) { b_fit = false; p_filter->input.i_format = VLC_FOURCC('f','l','3','2'); p_filter->output.i_format = VLC_FOURCC('f','l','3','2'); msg_Warn( p_filter, "bad input or output format" ); } if ( !AOUT_FMTS_SIMILAR( &p_filter->input, &p_filter->output ) ) { b_fit = false; memcpy( &p_filter->output, &p_filter->input, sizeof(audio_sample_format_t) ); msg_Warn( p_filter, "input and output formats are not similar" ); } if ( ! b_fit ) { return VLC_EGENERIC; } p_filter->pf_do_work = DoWork; p_filter->b_in_place = true; /* Allocate structure */ p_sys = p_filter->p_sys = malloc( sizeof( aout_filter_sys_t ) ); p_sys->f_lowf = config_GetFloat( p_this, "vuvuzela-lowf"); p_sys->f_lowgain = config_GetFloat( p_this, "vuvuzela-lowgain"); p_sys->f_highf = config_GetFloat( p_this, "vuvuzela-highf"); p_sys->f_highgain = config_GetFloat( p_this, "vuvuzela-highgain"); p_sys->f_Q1 = config_GetFloat( p_this, "vuvuzela-q1"); p_sys->f_gain1 = config_GetFloat( p_this, "vuvuzela-gain1"); p_sys->f_f1 = config_GetFloat( p_this, "vuvuzela-f1"); p_sys->f_f2 = config_GetFloat( p_this, "vuvuzela-f2"); p_sys->f_f3 = config_GetFloat( p_this, "vuvuzela-f3"); p_sys->f_f4 = config_GetFloat( p_this, "vuvuzela-f4"); p_sys->f_f5 = config_GetFloat( p_this, "vuvuzela-f5"); p_sys->f_f6 = config_GetFloat( p_this, "vuvuzela-f6"); p_sys->f_Q2 = config_GetFloat( p_this, "vuvuzela-q2"); p_sys->f_gain2 = config_GetFloat( p_this, "vuvuzela-gain2"); p_sys->f_Q3 = config_GetFloat( p_this, "vuvuzela-q3"); p_sys->f_gain3 = config_GetFloat( p_this, "vuvuzela-gain3"); p_sys->f_Q4 = config_GetFloat( p_this, "vuvuzela-q4"); p_sys->f_gain4 = config_GetFloat( p_this, "vuvuzela-gain4"); p_sys->f_Q5 = config_GetFloat( p_this, "vuvuzela-q5"); p_sys->f_gain5 = config_GetFloat( p_this, "vuvuzela-gain5"); p_sys->f_Q6 = config_GetFloat( p_this, "vuvuzela-q6"); p_sys->f_gain6 = config_GetFloat( p_this, "vuvuzela-gain6"); i_samplerate = p_filter->input.i_rate; CalcPeakEQCoeffs(p_sys->f_f1, p_sys->f_Q1, p_sys->f_gain1, i_samplerate, p_sys->coeffs+0*5); CalcPeakEQCoeffs(p_sys->f_f2, p_sys->f_Q2, p_sys->f_gain2, i_samplerate, p_sys->coeffs+1*5); CalcPeakEQCoeffs(p_sys->f_f3, p_sys->f_Q3, p_sys->f_gain3, i_samplerate, p_sys->coeffs+2*5); CalcShelfEQCoeffs(p_sys->f_lowf, 1, p_sys->f_lowgain, 0, i_samplerate, p_sys->coeffs+3*5); CalcShelfEQCoeffs(p_sys->f_highf, 1, p_sys->f_highgain, 0, i_samplerate, p_sys->coeffs+4*5); CalcPeakEQCoeffs(p_sys->f_f4, p_sys->f_Q4, p_sys->f_gain4, i_samplerate, p_sys->coeffs+5*5); CalcPeakEQCoeffs(p_sys->f_f5, p_sys->f_Q5, p_sys->f_gain5, i_samplerate, p_sys->coeffs+6*5); CalcPeakEQCoeffs(p_sys->f_f6, p_sys->f_Q6, p_sys->f_gain6, i_samplerate, p_sys->coeffs+7*5); p_sys->p_state = (float*)calloc( p_filter->input.i_channels*8*4, sizeof(float) ); return VLC_SUCCESS; } static void Close( vlc_object_t *p_this ) { aout_filter_t *p_filter = (aout_filter_t *)p_this; free( p_filter->p_sys->p_state ); free( p_filter->p_sys ); } /***************************************************************************** * DoWork: process samples buffer ***************************************************************************** * *****************************************************************************/ static void DoWork( aout_instance_t * p_aout, aout_filter_t * p_filter, aout_buffer_t * p_in_buf, aout_buffer_t * p_out_buf ) { VLC_UNUSED(p_aout); p_out_buf->i_nb_samples = p_in_buf->i_nb_samples; p_out_buf->i_nb_bytes = p_in_buf->i_nb_bytes; ProcessEQ( (float*)p_in_buf->p_buffer, (float*)p_out_buf->p_buffer, p_filter->p_sys->p_state, p_filter->input.i_channels, p_in_buf->i_nb_samples, p_filter->p_sys->coeffs, 8 ); } /* * Calculate direct form IIR coefficients for peaking EQ * coeffs[0] = b0 * coeffs[1] = b1 * coeffs[2] = b2 * coeffs[3] = a1 * coeffs[4] = a2 * * Equations taken from RBJ audio EQ cookbook * (http://www.musicdsp.org/files/Audio-EQ-Cookbook.txt) */ static void CalcPeakEQCoeffs( float f0, float Q, float gainDB, float Fs, float *coeffs ) { float A; float w0; float alpha; float b0, b1, b2; float a0, a1, a2; // Provide sane limits to avoid overflow if (Q < 0.1f) Q = 0.1f; if (Q > 100) Q = 100; if (f0 > Fs/2*0.95f) f0 = Fs/2*0.95f; if (gainDB < -40) gainDB = -40; if (gainDB > 40) gainDB = 40; A = pow(10, gainDB/40); w0 = 2*3.141593f*f0/Fs; alpha = sin(w0)/(2*Q); b0 = 1 + alpha*A; b1 = -2*cos(w0); b2 = 1 - alpha*A; a0 = 1 + alpha/A; a1 = -2*cos(w0); a2 = 1 - alpha/A; // Store values to coeffs and normalize by 1/a0 coeffs[0] = b0/a0; coeffs[1] = b1/a0; coeffs[2] = b2/a0; coeffs[3] = a1/a0; coeffs[4] = a2/a0; } /* * Calculate direct form IIR coefficients for low/high shelf EQ * coeffs[0] = b0 * coeffs[1] = b1 * coeffs[2] = b2 * coeffs[3] = a1 * coeffs[4] = a2 * * Equations taken from RBJ audio EQ cookbook * (http://www.musicdsp.org/files/Audio-EQ-Cookbook.txt) */ static void CalcShelfEQCoeffs( float f0, float slope, float gainDB, int high, float Fs, float *coeffs ) { float A; float w0; float alpha; float b0, b1, b2; float a0, a1, a2; // Provide sane limits to avoid overflow if (f0 > Fs/2*0.95f) f0 = Fs/2*0.95f; if (gainDB < -40) gainDB = -40; if (gainDB > 40) gainDB = 40; A = pow(10, gainDB/40); w0 = 2*3.141593f*f0/Fs; alpha = sin(w0)/2 * sqrt( (A + 1/A)*(1/slope - 1) + 2 ); if (high) { b0 = A*( (A+1) + (A-1)*cos(w0) + 2*sqrt(A)*alpha ); b1 = -2*A*( (A-1) + (A+1)*cos(w0) ); b2 = A*( (A+1) + (A-1)*cos(w0) - 2*sqrt(A)*alpha ); a0 = (A+1) - (A-1)*cos(w0) + 2*sqrt(A)*alpha; a1 = 2*( (A-1) - (A+1)*cos(w0) ); a2 = (A+1) - (A-1)*cos(w0) - 2*sqrt(A)*alpha; } else { b0 = A*( (A+1) - (A-1)*cos(w0) + 2*sqrt(A)*alpha ); b1 = 2*A*( (A-1) - (A+1)*cos(w0)); b2 = A*( (A+1) - (A-1)*cos(w0) - 2*sqrt(A)*alpha ); a0 = (A+1) + (A-1)*cos(w0) + 2*sqrt(A)*alpha; a1 = -2*( (A-1) + (A+1)*cos(w0)); a2 = (A+1) + (A-1)*cos(w0) - 2*sqrt(A)*alpha; } // Store values to coeffs and normalize by 1/a0 coeffs[0] = b0/a0; coeffs[1] = b1/a0; coeffs[2] = b2/a0; coeffs[3] = a1/a0; coeffs[4] = a2/a0; } /* src is assumed to be interleaved dest is assumed to be interleaved size of state is 4*channels*eqCount samples is not premultiplied by channels size of coeffs is 5*eqCount */ void ProcessEQ( float *src, float *dest, float *state, unsigned channels, unsigned samples, float *coeffs, unsigned eqCount ) { unsigned i, chn, eq; float b0, b1, b2, a1, a2; float x, y = 0; float *src1, *dest1; float *coeffs1, *state1; src1 = src; dest1 = dest; for (i = 0; i < samples; i++) { state1 = state; for (chn = 0; chn < channels; chn++) { coeffs1 = coeffs; x = *src1++; /* Direct form 1 IIRs */ for (eq = 0; eq < eqCount; eq++) { b0 = coeffs1[0]; b1 = coeffs1[1]; b2 = coeffs1[2]; a1 = coeffs1[3]; a2 = coeffs1[4]; coeffs1 += 5; y = x*b0 + state1[0]*b1 + state1[1]*b2 - state1[2]*a1 - state1[3]*a2; state1[1] = state1[0]; state1[0] = x; state1[3] = state1[2]; state1[2] = y; x = y; state1 += 4; } *dest1++ = y; } } }