equalizer.c 18.4 KB
Newer Older
1 2 3
/*****************************************************************************
 * equalizer.c:
 *****************************************************************************
4
 * Copyright (C) 2004-2012 VLC authors and VideoLAN
5
 * $Id$
6 7 8
 *
 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
 *
9 10 11
 * 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
12 13 14 15
 * (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
16 17
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Lesser General Public License for more details.
18
 *
19 20 21
 * 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.
22 23 24 25 26 27
 *****************************************************************************/

/*****************************************************************************
 * Preamble
 *****************************************************************************/

28 29 30 31
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif

32 33
#include <math.h>

34
#include <vlc_common.h>
35
#include <vlc_plugin.h>
36
#include <vlc_charset.h>
37

38
#include <vlc_aout.h>
39
#include <vlc_filter.h>
40

Jérome Decoodt's avatar
Jérome Decoodt committed
41
#include "equalizer_presets.h"
42

43 44 45
/* TODO:
 *  - optimize a bit (you can hardly do slower ;)
 *  - add tables for more bands (15 and 32 would be cool), maybe with auto coeffs
46
 *    computation (not too hard once the Q is found).
47 48 49 50 51 52 53 54 55 56 57 58
 *  - support for external preset
 *  - callback to handle preset changes on the fly
 *  - ...
 */

/*****************************************************************************
 * Module descriptor
 *****************************************************************************/
static int  Open ( vlc_object_t * );
static void Close( vlc_object_t * );

#define PRESET_TEXT N_( "Equalizer preset" )
59
#define PRESET_LONGTEXT N_("Preset to use for the equalizer." )
60 61

#define BANDS_TEXT N_( "Bands gain")
62
#define BANDS_LONGTEXT N_( \
63 64
         "Don't use presets, but manually specified bands. You need to " \
         "provide 10 values between -20dB and 20dB, separated by spaces, " \
65
         "e.g. \"0 2 4 2 0 -2 -4 -2 0 2\"." )
66

67 68 69 70 71
#define VLC_BANDS_TEXT N_( "Use VLC frequency bands" )
#define VLC_BANDS_LONGTEXT N_( \
         "Use the VLC frequency bands. Otherwise, use the ISO Standard " \
         "frequency bands." )

72
#define TWOPASS_TEXT N_( "Two pass" )
73
#define TWOPASS_LONGTEXT N_( "Filter the audio twice. This provides a more "  \
74
         "intense effect.")
75 76

#define PREAMP_TEXT N_("Global gain" )
77
#define PREAMP_LONGTEXT N_("Set the global gain in dB (-20 ... 20)." )
78

79 80 81
vlc_module_begin ()
    set_description( N_("Equalizer with 10 bands") )
    set_shortname( N_("Equalizer" ) )
82
    set_capability( "audio filter", 0 )
83 84
    set_category( CAT_AUDIO )
    set_subcategory( SUBCAT_AUDIO_AFILTER )
Clément Stenac's avatar
Clément Stenac committed
85

86
    add_string( "equalizer-preset", "flat", PRESET_TEXT,
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
87
                PRESET_LONGTEXT, false )
88
        change_string_list( preset_list, preset_list_text )
89
    add_string( "equalizer-bands", NULL, BANDS_TEXT,
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
90
                BANDS_LONGTEXT, true )
91
    add_bool( "equalizer-2pass", false, TWOPASS_TEXT,
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
92
              TWOPASS_LONGTEXT, true )
93 94
    add_bool( "equalizer-vlcfreqs", true, VLC_BANDS_TEXT,
              VLC_BANDS_LONGTEXT, true )
95
    add_float( "equalizer-preamp", 12.0f, PREAMP_TEXT,
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
96
               PREAMP_LONGTEXT, true )
97 98 99
    set_callbacks( Open, Close )
    add_shortcut( "equalizer" )
vlc_module_end ()
100 101 102 103

/*****************************************************************************
 * Local prototypes
 *****************************************************************************/
104
struct filter_sys_t
105 106 107 108 109 110 111 112 113 114
{
    /* Filter static config */
    int i_band;
    float *f_alpha;
    float *f_beta;
    float *f_gamma;

    /* Filter dyn config */
    float *f_amp;   /* Per band amp */
    float f_gamp;   /* Global preamp */
115
    bool b_2eqz;
116 117 118 119 120 121 122 123 124

    /* Filter state */
    float x[32][2];
    float y[32][128][2];

    /* Second filter state */
    float x2[32][2];
    float y2[32][128][2];

125
    vlc_mutex_t lock;
126
};
127

128
static block_t *DoWork( filter_t *, block_t * );
129

130
#define EQZ_IN_FACTOR (0.25f)
131 132 133
static int  EqzInit( filter_t *, int );
static void EqzFilter( filter_t *, float *, float *, int, int );
static void EqzClean( filter_t * );
134

135 136 137 138 139 140 141 142
static int PresetCallback ( vlc_object_t *, char const *, vlc_value_t,
                            vlc_value_t, void * );
static int PreampCallback ( vlc_object_t *, char const *, vlc_value_t,
                            vlc_value_t, void * );
static int BandsCallback  ( vlc_object_t *, char const *, vlc_value_t,
                            vlc_value_t, void * );
static int TwoPassCallback( vlc_object_t *, char const *, vlc_value_t,
                            vlc_value_t, void * );
143 144 145



146 147 148 149 150
/*****************************************************************************
 * Open:
 *****************************************************************************/
static int Open( vlc_object_t *p_this )
{
151
    filter_t     *p_filter = (filter_t *)p_this;
152 153

    /* Allocate structure */
154
    filter_sys_t *p_sys = p_filter->p_sys = malloc( sizeof( *p_sys ) );
155 156
    if( !p_sys )
        return VLC_ENOMEM;
157

158
    vlc_mutex_init( &p_sys->lock );
159
    if( EqzInit( p_filter, p_filter->fmt_in.audio.i_rate ) != VLC_SUCCESS )
160
    {
161
        vlc_mutex_destroy( &p_sys->lock );
162
        free( p_sys );
163
        return VLC_EGENERIC;
164
    }
165

166 167 168 169
    p_filter->fmt_in.audio.i_format = VLC_CODEC_FL32;
    p_filter->fmt_out.audio = p_filter->fmt_in.audio;
    p_filter->pf_audio_filter = DoWork;

170 171 172 173 174 175 176 177
    return VLC_SUCCESS;
}

/*****************************************************************************
 * Close: close the plugin
 *****************************************************************************/
static void Close( vlc_object_t *p_this )
{
178 179
    filter_t     *p_filter = (filter_t *)p_this;
    filter_sys_t *p_sys = p_filter->p_sys;
180 181

    EqzClean( p_filter );
182
    vlc_mutex_destroy( &p_sys->lock );
183 184 185 186 187 188 189 190
    free( p_sys );
}

/*****************************************************************************
 * DoWork: process samples buffer
 *****************************************************************************
 *
 *****************************************************************************/
191
static block_t * DoWork( filter_t * p_filter, block_t * p_in_buf )
192
{
193
    EqzFilter( p_filter, (float*)p_in_buf->p_buffer,
194
               (float*)p_in_buf->p_buffer, p_in_buf->i_nb_samples,
195
               aout_FormatNbChannels( &p_filter->fmt_in.audio ) );
196
    return p_in_buf;
197 198 199 200 201 202 203 204 205 206 207 208 209 210 211
}

/*****************************************************************************
 * Equalizer stuff
 *****************************************************************************/
typedef struct
{
    int   i_band;

    struct
    {
        float f_frequency;
        float f_alpha;
        float f_beta;
        float f_gamma;
212
    } band[EQZ_BANDS_MAX];
213

214 215
} eqz_config_t;

216 217 218 219
/* Equalizer coefficient calculation function based on equ-xmms */
static void EqzCoeffs( int i_rate, float f_octave_percent,
                       bool b_use_vlc_freqs,
                       eqz_config_t *p_eqz_config )
220
{
221 222 223 224
    const float *f_freq_table_10b = b_use_vlc_freqs
                                  ? f_vlc_frequency_table_10b
                                  : f_iso_frequency_table_10b;
    float f_rate = (float) i_rate;
225 226 227 228
    float f_nyquist_freq = 0.5f * f_rate;
    float f_octave_factor = powf( 2.0f, 0.5f * f_octave_percent );
    float f_octave_factor_1 = 0.5f * ( f_octave_factor + 1.0f );
    float f_octave_factor_2 = 0.5f * ( f_octave_factor - 1.0f );
229 230 231 232

    p_eqz_config->i_band = EQZ_BANDS_MAX;

    for( int i = 0; i < EQZ_BANDS_MAX; i++ )
233
    {
234 235 236 237 238 239
        float f_freq = f_freq_table_10b[i];

        p_eqz_config->band[i].f_frequency = f_freq;

        if( f_freq <= f_nyquist_freq )
        {
240
            float f_theta_1 = ( 2.0f * (float) M_PI * f_freq ) / f_rate;
241
            float f_theta_2 = f_theta_1 / f_octave_factor;
242 243 244 245
            float f_sin     = sinf( f_theta_2 );
            float f_sin_prd = sinf( f_theta_2 * f_octave_factor_1 )
                            * sinf( f_theta_2 * f_octave_factor_2 );
            float f_sin_hlf = f_sin * 0.5f;
246 247 248 249
            float f_den     = f_sin_hlf + f_sin_prd;

            p_eqz_config->band[i].f_alpha = f_sin_prd / f_den;
            p_eqz_config->band[i].f_beta  = ( f_sin_hlf - f_sin_prd ) / f_den;
250
            p_eqz_config->band[i].f_gamma = f_sin * cosf( f_theta_1 ) / f_den;
251 252 253 254 255 256
        }
        else
        {
            /* Any frequency beyond the Nyquist frequency is no good... */
            p_eqz_config->band[i].f_alpha =
            p_eqz_config->band[i].f_beta  =
257
            p_eqz_config->band[i].f_gamma = 0.0f;
258
        }
259
    }
260
}
261 262 263 264 265 266 267 268 269 270 271

static inline float EqzConvertdB( float db )
{
    /* Map it to gain,
     * (we do as if the input of iir is /EQZ_IN_FACTOR, but in fact it's the non iir data that is *EQZ_IN_FACTOR)
     * db = 20*log( out / in ) with out = in + amp*iir(i/EQZ_IN_FACTOR)
     * or iir(i) == i for the center freq so
     * db = 20*log( 1 + amp/EQZ_IN_FACTOR )
     * -> amp = EQZ_IN_FACTOR*(10^(db/20) - 1)
     **/

272 273 274 275 276
    if( db < -20.0f )
        db = -20.0f;
    else if(  db > 20.0f )
        db = 20.0f;
    return EQZ_IN_FACTOR * ( powf( 10.0f, db / 20.0f ) - 1.0f );
277 278
}

279
static int EqzInit( filter_t *p_filter, int i_rate )
280
{
281
    filter_sys_t *p_sys = p_filter->p_sys;
282
    eqz_config_t cfg;
283
    int i, ch;
284
    vlc_value_t val1, val2, val3;
285
    vlc_object_t *p_aout = p_filter->p_parent;
286
    int i_ret = VLC_ENOMEM;
287

288
    bool b_vlcFreqs = var_InheritBool( p_aout, "equalizer-vlcfreqs" );
289
    EqzCoeffs( i_rate, 1.0f, b_vlcFreqs, &cfg );
290 291

    /* Create the static filter config */
292
    p_sys->i_band = cfg.i_band;
293 294 295
    p_sys->f_alpha = malloc( p_sys->i_band * sizeof(float) );
    p_sys->f_beta  = malloc( p_sys->i_band * sizeof(float) );
    p_sys->f_gamma = malloc( p_sys->i_band * sizeof(float) );
Rémi Duraffort's avatar
Rémi Duraffort committed
296
    if( !p_sys->f_alpha || !p_sys->f_beta || !p_sys->f_gamma )
297
        goto error;
Rémi Duraffort's avatar
Rémi Duraffort committed
298

299 300
    for( i = 0; i < p_sys->i_band; i++ )
    {
301 302 303
        p_sys->f_alpha[i] = cfg.band[i].f_alpha;
        p_sys->f_beta[i]  = cfg.band[i].f_beta;
        p_sys->f_gamma[i] = cfg.band[i].f_gamma;
304 305 306
    }

    /* Filter dyn config */
307
    p_sys->b_2eqz = false;
308
    p_sys->f_gamp = 1.0f;
Rémi Duraffort's avatar
Rémi Duraffort committed
309 310
    p_sys->f_amp  = malloc( p_sys->i_band * sizeof(float) );
    if( !p_sys->f_amp )
311 312
        goto error;

313 314
    for( i = 0; i < p_sys->i_band; i++ )
    {
315
        p_sys->f_amp[i] = 0.0f;
316 317 318 319 320 321 322 323
    }

    /* Filter state */
    for( ch = 0; ch < 32; ch++ )
    {
        p_sys->x[ch][0]  =
        p_sys->x[ch][1]  =
        p_sys->x2[ch][0] =
324
        p_sys->x2[ch][1] = 0.0f;
325 326 327 328 329 330

        for( i = 0; i < p_sys->i_band; i++ )
        {
            p_sys->y[ch][i][0]  =
            p_sys->y[ch][i][1]  =
            p_sys->y2[ch][i][0] =
331
            p_sys->y2[ch][i][1] = 0.0f;
332 333 334
        }
    }

335 336
    var_Create( p_aout, "equalizer-bands", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
    var_Create( p_aout, "equalizer-preset", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
337

338
    p_sys->b_2eqz = var_CreateGetBool( p_aout, "equalizer-2pass" );
339

340
    var_Create( p_aout, "equalizer-preamp", VLC_VAR_FLOAT | VLC_VAR_DOINHERIT );
341

342 343
    /* Get initial values */
    var_Get( p_aout, "equalizer-preset", &val1 );
344 345 346
    PresetCallback( VLC_OBJECT( p_aout ), NULL, val1, val1, p_sys );
    free( val1.psz_string );

347 348 349 350
    var_Get( p_aout, "equalizer-bands", &val2 );
    var_Get( p_aout, "equalizer-preamp", &val3 );
    BandsCallback(  VLC_OBJECT( p_aout ), NULL, val2, val2, p_sys );
    PreampCallback( VLC_OBJECT( p_aout ), NULL, val3, val3, p_sys );
351

352
    /* Exit if we have no preset and no bands value */
353
    if (!val2.psz_string || !*val2.psz_string)
354 355
    {
        msg_Err(p_filter, "No preset selected");
356
        free( val2.psz_string );
Rémi Duraffort's avatar
Rémi Duraffort committed
357
        free( p_sys->f_amp );
358 359
        i_ret = VLC_EGENERIC;
        goto error;
360
    }
361
    free( val2.psz_string );
362 363 364 365 366

    /* Add our own callbacks */
    var_AddCallback( p_aout, "equalizer-preset", PresetCallback, p_sys );
    var_AddCallback( p_aout, "equalizer-bands", BandsCallback, p_sys );
    var_AddCallback( p_aout, "equalizer-preamp", PreampCallback, p_sys );
367
    var_AddCallback( p_aout, "equalizer-2pass", TwoPassCallback, p_sys );
368 369

    msg_Dbg( p_filter, "equalizer loaded for %d Hz with %d bands %d pass",
370
                        i_rate, p_sys->i_band, p_sys->b_2eqz ? 2 : 1 );
371 372
    for( i = 0; i < p_sys->i_band; i++ )
    {
373 374
        msg_Dbg( p_filter, "   %.2f Hz -> factor:%f alpha:%f beta:%f gamma:%f",
                 cfg.band[i].f_frequency, p_sys->f_amp[i],
375 376 377
                 p_sys->f_alpha[i], p_sys->f_beta[i], p_sys->f_gamma[i]);
    }
    return VLC_SUCCESS;
378 379 380 381 382 383

error:
    free( p_sys->f_alpha );
    free( p_sys->f_beta );
    free( p_sys->f_gamma );
    return i_ret;
384 385
}

386
static void EqzFilter( filter_t *p_filter, float *out, float *in,
387
                       int i_samples, int i_channels )
388
{
389
    filter_sys_t *p_sys = p_filter->p_sys;
390 391
    int i, ch, j;

392
    vlc_mutex_lock( &p_sys->lock );
393 394 395 396 397
    for( i = 0; i < i_samples; i++ )
    {
        for( ch = 0; ch < i_channels; ch++ )
        {
            const float x = in[ch];
398
            float o = 0.0f;
399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417

            for( j = 0; j < p_sys->i_band; j++ )
            {
                float y = p_sys->f_alpha[j] * ( x - p_sys->x[ch][1] ) +
                          p_sys->f_gamma[j] * p_sys->y[ch][j][0] -
                          p_sys->f_beta[j]  * p_sys->y[ch][j][1];

                p_sys->y[ch][j][1] = p_sys->y[ch][j][0];
                p_sys->y[ch][j][0] = y;

                o += y * p_sys->f_amp[j];
            }
            p_sys->x[ch][1] = p_sys->x[ch][0];
            p_sys->x[ch][0] = x;

            /* Second filter */
            if( p_sys->b_2eqz )
            {
                const float x2 = EQZ_IN_FACTOR * x + o;
418
                o = 0.0f;
419 420 421 422 423 424 425 426 427 428 429 430 431 432 433
                for( j = 0; j < p_sys->i_band; j++ )
                {
                    float y = p_sys->f_alpha[j] * ( x2 - p_sys->x2[ch][1] ) +
                              p_sys->f_gamma[j] * p_sys->y2[ch][j][0] -
                              p_sys->f_beta[j]  * p_sys->y2[ch][j][1];

                    p_sys->y2[ch][j][1] = p_sys->y2[ch][j][0];
                    p_sys->y2[ch][j][0] = y;

                    o += y * p_sys->f_amp[j];
                }
                p_sys->x2[ch][1] = p_sys->x2[ch][0];
                p_sys->x2[ch][0] = x2;

                /* We add source PCM + filtered PCM */
434
                out[ch] = p_sys->f_gamp * p_sys->f_gamp *( EQZ_IN_FACTOR * x2 + o );
435 436 437 438 439 440 441 442 443 444 445
            }
            else
            {
                /* We add source PCM + filtered PCM */
                out[ch] = p_sys->f_gamp *( EQZ_IN_FACTOR * x + o );
            }
        }

        in  += i_channels;
        out += i_channels;
    }
446
    vlc_mutex_unlock( &p_sys->lock );
447 448
}

449
static void EqzClean( filter_t *p_filter )
450
{
451 452
    filter_sys_t *p_sys = p_filter->p_sys;
    vlc_object_t *p_aout = p_filter->p_parent;
453

454 455 456 457
    var_DelCallback( p_aout, "equalizer-bands", BandsCallback, p_sys );
    var_DelCallback( p_aout, "equalizer-preset", PresetCallback, p_sys );
    var_DelCallback( p_aout, "equalizer-preamp", PreampCallback, p_sys );
    var_DelCallback( p_aout, "equalizer-2pass", TwoPassCallback, p_sys );
458

459 460 461 462 463
    free( p_sys->f_alpha );
    free( p_sys->f_beta );
    free( p_sys->f_gamma );

    free( p_sys->f_amp );
464 465 466
}


467
static int PresetCallback( vlc_object_t *p_aout, char const *psz_cmd,
468
                         vlc_value_t oldval, vlc_value_t newval, void *p_data )
469
{
470
    const eqz_preset_t *preset = NULL;
471
    const char *psz_preset = newval.psz_string;
472

473 474 475 476 477 478 479 480
    for( unsigned i = 0; i < NB_PRESETS; i++ )
        if( !strcasecmp( eqz_preset_10b[i].psz_name, psz_preset ) )
        {
            preset = eqz_preset_10b + i;
            break;
        }

    if( preset == NULL )
481
    {
482 483 484 485 486
        msg_Err( p_aout, "equalizer preset '%s' not found", psz_preset );
        msg_Info( p_aout, "full list:" );
        for( unsigned i = 0; i < NB_PRESETS; i++ )
             msg_Info( p_aout, "  - '%s'", eqz_preset_10b[i].psz_name );
        return VLC_EGENERIC;
487
    }
488

489 490 491
    char *bands = NULL;

    for( unsigned i = 0; i < EQZ_BANDS_MAX; i++ )
492
    {
493
        char *psz;
494

495 496 497 498 499 500 501 502 503 504
        lldiv_t d = lldiv( lroundf(preset->f_amp[i] * 10000000.f), 10000000 );

        if( asprintf( &psz, "%s %lld.%07llu", i ? bands : "",
                      d.quot, d.rem ) == -1 )
            psz = NULL;

        free( bands );
        if( unlikely(psz == NULL) )
            return VLC_ENOMEM;
        bands = psz;
505
    }
506 507 508 509 510

    var_SetFloat( p_aout, "equalizer-preamp", preset->f_preamp );
    var_SetString( p_aout, "equalizer-bands", bands );
    free( bands );
    (void) psz_cmd; (void) oldval; (void) p_data;
511 512 513 514
    return VLC_SUCCESS;
}

static int PreampCallback( vlc_object_t *p_this, char const *psz_cmd,
515
                         vlc_value_t oldval, vlc_value_t newval, void *p_data )
516
{
Rafaël Carré's avatar
Rafaël Carré committed
517
    VLC_UNUSED(p_this); VLC_UNUSED(psz_cmd); VLC_UNUSED(oldval);
518
    filter_sys_t *p_sys = p_data;
519
    float preamp;
520

521 522 523 524 525 526
    if( newval.f_float < -20.f )
        preamp = .1f;
    else if( newval.f_float < 20.f )
        preamp = powf( 10.f, newval.f_float / 20.f );
    else
        preamp = 10.f;
527 528

    vlc_mutex_lock( &p_sys->lock );
529
    p_sys->f_gamp = preamp;
530
    vlc_mutex_unlock( &p_sys->lock );
531 532 533 534
    return VLC_SUCCESS;
}

static int BandsCallback( vlc_object_t *p_this, char const *psz_cmd,
535
                         vlc_value_t oldval, vlc_value_t newval, void *p_data )
536
{
Rafaël Carré's avatar
Rafaël Carré committed
537
    VLC_UNUSED(p_this); VLC_UNUSED(psz_cmd); VLC_UNUSED(oldval);
538
    filter_sys_t *p_sys = p_data;
539 540
    const char *p = newval.psz_string;
    int i = 0;
541 542

    /* Same thing for bands */
543
    vlc_mutex_lock( &p_sys->lock );
544
    while( i < p_sys->i_band )
545
    {
546
        char *next;
547
        /* Read dB -20/20 */
548 549
        float f = us_strtof( p, &next );
        if( next == p || isnan( f ) )
550
            break; /* no conversion */
551

552
        p_sys->f_amp[i++] = EqzConvertdB( f );
553

554
        if( *next == '\0' )
555
            break; /* end of line */
556
        p = &next[1];
557
    }
558 559
    while( i < p_sys->i_band )
        p_sys->f_amp[i++] = EqzConvertdB( 0.f );
560
    vlc_mutex_unlock( &p_sys->lock );
561
    return VLC_SUCCESS;
562
}
563 564 565 566
static int TwoPassCallback( 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(psz_cmd); VLC_UNUSED(oldval);
567
    filter_sys_t *p_sys = p_data;
568

569
    vlc_mutex_lock( &p_sys->lock );
570
    p_sys->b_2eqz = newval.b_bool;
571
    vlc_mutex_unlock( &p_sys->lock );
572
    return VLC_SUCCESS;
573
}
574