git.sesse.net Git - vlc/blob - modules/visualization/visual/visual.c

   1 /*****************************************************************************
   2  * visual.c : Visualisation system
   3  *****************************************************************************
   4  * Copyright (C) 2002-2009 the VideoLAN team
   5  * $Id$
   6  *
   7  * Authors: Clément Stenac <zorglub@via.ecp.fr>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22  *****************************************************************************/
  23
  24 /*****************************************************************************
  25  * Preamble
  26  *****************************************************************************/
  27 #ifdef HAVE_CONFIG_H
  28 # include "config.h"
  29 #endif
  30
  31 #include <vlc_common.h>
  32 #include <vlc_plugin.h>
  33 #include <vlc_vout.h>
  34 #include <vlc_aout.h>
  35 #include <vlc_filter.h>
  36
  37 #include "visual.h"
  38
  39 /*****************************************************************************
  40  * Module descriptor
  41  *****************************************************************************/
  42 #define ELIST_TEXT N_( "Effects list" )
  43 #define ELIST_LONGTEXT N_( \
  44       "A list of visual effect, separated by commas.\n"  \
  45       "Current effects include: dummy, scope, spectrum, "\
  46       "spectrometer and vuMeter." )
  47
  48 #define WIDTH_TEXT N_( "Video width" )
  49 #define WIDTH_LONGTEXT N_( \
  50       "The width of the effects video window, in pixels." )
  51
  52 #define HEIGHT_TEXT N_( "Video height" )
  53 #define HEIGHT_LONGTEXT N_( \
  54       "The height of the effects video window, in pixels." )
  55
  56 #define NBBANDS_TEXT N_( "More bands : 80 / 20" )
  57 #define NBBANDS_LONGTEXT N_( \
  58       "More bands for the spectrum analyzer : 80 if enabled else 20." )
  59 #define SPNBBANDS_LONGTEXT N_( \
  60       "More bands for the spectrometer : 80 if enabled else 20." )
  61
  62 #define SEPAR_TEXT N_( "Band separator" )
  63 #define SEPAR_LONGTEXT N_( \
  64         "Number of blank pixels between bands.")
  65
  66 #define AMP_TEXT N_( "Amplification" )
  67 #define AMP_LONGTEXT N_( \
  68         "This is a coefficient that modifies the height of the bands.")
  69
  70 #define PEAKS_TEXT N_( "Enable peaks" )
  71 #define PEAKS_LONGTEXT N_( \
  72         "Draw \"peaks\" in the spectrum analyzer." )
  73
  74 #define ORIG_TEXT N_( "Enable original graphic spectrum" )
  75 #define ORIG_LONGTEXT N_( \
  76         "Enable the \"flat\" spectrum analyzer in the spectrometer." )
  77
  78 #define BANDS_TEXT N_( "Enable bands" )
  79 #define BANDS_LONGTEXT N_( \
  80         "Draw bands in the spectrometer." )
  81
  82 #define BASE_TEXT N_( "Enable base" )
  83 #define BASE_LONGTEXT N_( \
  84         "Defines whether to draw the base of the bands." )
  85
  86 #define RADIUS_TEXT N_( "Base pixel radius" )
  87 #define RADIUS_LONGTEXT N_( \
  88         "Defines radius size in pixels, of base of bands(beginning)." )
  89
  90 #define SSECT_TEXT N_( "Spectral sections" )
  91 #define SSECT_LONGTEXT N_( \
  92         "Determines how many sections of spectrum will exist." )
  93
  94 #define PEAK_HEIGHT_TEXT N_( "Peak height" )
  95 #define PEAK_HEIGHT_LONGTEXT N_( \
  96         "Total pixel height of the peak items." )
  97
  98 #define PEAK_WIDTH_TEXT N_( "Peak extra width" )
  99 #define PEAK_WIDTH_LONGTEXT N_( \
 100         "Additions or subtractions of pixels on the peak width." )
 101
 102 #define COLOR1_TEXT N_( "V-plane color" )
 103 #define COLOR1_LONGTEXT N_( \
 104         "YUV-Color cube shifting across the V-plane ( 0 - 127 )." )
 105
 106 #define STARS_TEXT N_( "Number of stars" )
 107 #define STARS_LONGTEXT N_( \
 108         "Number of stars to draw with random effect." )
 109
 110 static int  Open         ( vlc_object_t * );
 111 static void Close        ( vlc_object_t * );
 112
 113 vlc_module_begin ()
 114     set_shortname( N_("Visualizer"))
 115     set_category( CAT_AUDIO )
 116     set_subcategory( SUBCAT_AUDIO_VISUAL )
 117     set_description( N_("Visualizer filter") )
 118     set_section( N_( "General") , NULL )
 119     add_string("effect-list", "spectrum", NULL,
 120             ELIST_TEXT, ELIST_LONGTEXT, true )
 121     add_integer("effect-width",VOUT_WIDTH,NULL,
 122              WIDTH_TEXT, WIDTH_LONGTEXT, false )
 123     add_integer("effect-height" , VOUT_HEIGHT , NULL,
 124              HEIGHT_TEXT, HEIGHT_LONGTEXT, false )
 125     set_section( N_("Spectrum analyser") , NULL )
 126     add_obsolete_integer( "visual-nbbands" ) /* Since 1.0.0 */
 127     add_bool("visual-80-bands", true, NULL,
 128              NBBANDS_TEXT, NBBANDS_LONGTEXT, true );
 129     add_obsolete_integer( "visual-separ" ) /* Since 1.0.0 */
 130     add_obsolete_integer( "visual-amp" ) /* Since 1.0.0 */
 131     add_bool("visual-peaks", true, NULL,
 132              PEAKS_TEXT, PEAKS_LONGTEXT, true )
 133     set_section( N_("Spectrometer") , NULL )
 134     add_bool("spect-show-original", false, NULL,
 135              ORIG_TEXT, ORIG_LONGTEXT, true )
 136     add_bool("spect-show-base", true, NULL,
 137              BASE_TEXT, BASE_LONGTEXT, true )
 138     add_integer("spect-radius", 42, NULL,
 139              RADIUS_TEXT, RADIUS_LONGTEXT, true )
 140     add_integer("spect-sections", 3, NULL,
 141              SSECT_TEXT, SSECT_LONGTEXT, true )
 142     add_integer("spect-color", 80, NULL,
 143              COLOR1_TEXT, COLOR1_LONGTEXT, true )
 144     add_bool("spect-show-bands", true, NULL,
 145              BANDS_TEXT, BANDS_LONGTEXT, true );
 146     add_obsolete_integer( "spect-nbbands" ) /* Since 1.0.0 */
 147     add_bool("spect-80-bands", true, NULL,
 148              NBBANDS_TEXT, SPNBBANDS_LONGTEXT, true )
 149     add_integer("spect-separ", 1, NULL,
 150              SEPAR_TEXT, SEPAR_LONGTEXT, true )
 151     add_integer("spect-amp", 8, NULL,
 152              AMP_TEXT, AMP_LONGTEXT, true )
 153     add_bool("spect-show-peaks", true, NULL,
 154              PEAKS_TEXT, PEAKS_LONGTEXT, true )
 155     add_integer("spect-peak-width", 61, NULL,
 156              PEAK_WIDTH_TEXT, PEAK_WIDTH_LONGTEXT, true )
 157     add_integer("spect-peak-height", 1, NULL,
 158              PEAK_HEIGHT_TEXT, PEAK_HEIGHT_LONGTEXT, true )
 159     set_capability( "visualization2", 0 )
 160     set_callbacks( Open, Close )
 161     add_shortcut( "visualizer")
 162 vlc_module_end ()
 163
 164
 165 /*****************************************************************************
 166  * Local prototypes
 167  *****************************************************************************/
 168 static block_t *DoWork( filter_t *, block_t * );
 169 static int FilterCallback( vlc_object_t *, char const *,
 170                            vlc_value_t, vlc_value_t, void * );
 171 static const struct
 172 {
 173     const char *psz_name;
 174     int  (*pf_run)( visual_effect_t *, vlc_object_t *,
 175                     const block_t *, picture_t *);
 176 } pf_effect_run[]=
 177 {
 178     { "scope",        scope_Run },
 179     { "vuMeter",      vuMeter_Run },
 180     { "spectrum",     spectrum_Run },
 181     { "spectrometer", spectrometer_Run },
 182     { "dummy",        dummy_Run},
 183     { NULL,           dummy_Run}
 184 };
 185
 186 /*****************************************************************************
 187  * Open: open the visualizer
 188  *****************************************************************************/
 189 static int Open( vlc_object_t *p_this )
 190 {
 191     filter_t     *p_filter = (filter_t *)p_this;
 192     filter_sys_t *p_sys;
 193
 194     char *psz_effects, *psz_parser;
 195     video_format_t fmt;
 196
 197     if( ( p_filter->fmt_in.audio.i_format != VLC_CODEC_FL32 &&
 198           p_filter->fmt_in.audio.i_format != VLC_CODEC_FI32 ) )
 199     {
 200         return VLC_EGENERIC;
 201     }
 202
 203     p_sys = p_filter->p_sys = malloc( sizeof( filter_sys_t ) );
 204     if( p_sys == NULL )
 205         return VLC_EGENERIC;
 206
 207     p_sys->i_height = config_GetInt( p_filter , "effect-height");
 208     p_sys->i_width  = config_GetInt( p_filter , "effect-width");
 209
 210     if( p_sys->i_height < 400 ) p_sys->i_height = 400;
 211     if( p_sys->i_width  < 532 ) p_sys->i_width  = 532;
 212     if( (p_sys->i_height % 2 ) != 0 ) p_sys->i_height--;
 213     if( (p_sys->i_width % 2 )  != 0 ) p_sys->i_width--;
 214
 215     p_sys->i_effect = 0;
 216     p_sys->effect   = NULL;
 217
 218     /* Parse the effect list */
 219     psz_parser = psz_effects = var_CreateGetString( p_filter, "effect-list" );
 220     var_AddCallback( p_filter, "effect-list", FilterCallback, NULL );
 221
 222     while( psz_parser && *psz_parser != '\0' )
 223     {
 224         visual_effect_t *p_effect;
 225         int  i;
 226
 227         p_effect = malloc( sizeof( visual_effect_t ) );
 228         if( !p_effect )
 229             break;
 230         p_effect->i_width = p_sys->i_width;
 231         p_effect->i_height= p_sys->i_height;
 232         p_effect->i_nb_chans = aout_FormatNbChannels( &p_filter->fmt_in.audio);
 233         p_effect->i_idx_left  = 0;
 234         p_effect->i_idx_right = __MIN( 1, p_effect->i_nb_chans-1 );
 235
 236         p_effect->psz_args = NULL;
 237         p_effect->p_data = NULL;
 238
 239         p_effect->pf_run = NULL;
 240         p_effect->psz_name = NULL;
 241
 242         for( i = 0; pf_effect_run[i].psz_name != NULL; i++ )
 243         {
 244             if( !strncasecmp( psz_parser,
 245                               pf_effect_run[i].psz_name,
 246                               strlen( pf_effect_run[i].psz_name ) ) )
 247             {
 248                 p_effect->pf_run = pf_effect_run[i].pf_run;
 249                 p_effect->psz_name = pf_effect_run[i].psz_name;
 250                 break;
 251             }
 252         }
 253
 254         if( p_effect->psz_name )
 255         {
 256             psz_parser += strlen( p_effect->psz_name );
 257
 258             if( *psz_parser == '{' )
 259             {
 260                 char *psz_eoa;
 261
 262                 psz_parser++;
 263
 264                 if( ( psz_eoa = strchr( psz_parser, '}') ) == NULL )
 265                 {
 266                    msg_Err( p_filter, "unable to parse effect list. Aborting");
 267                    free( p_effect );
 268                    break;
 269                 }
 270                 p_effect->psz_args =
 271                     strndup( psz_parser, psz_eoa - psz_parser);
 272             }
 273             TAB_APPEND( p_sys->i_effect, p_sys->effect, p_effect );
 274         }
 275         else
 276         {
 277             msg_Err( p_filter, "unknown visual effect: %s", psz_parser );
 278             free( p_effect );
 279         }
 280
 281         if( strchr( psz_parser, ',' ) )
 282         {
 283             psz_parser = strchr( psz_parser, ',' ) + 1;
 284         }
 285         else if( strchr( psz_parser, ':' ) )
 286         {
 287             psz_parser = strchr( psz_parser, ':' ) + 1;
 288         }
 289         else
 290         {
 291             break;
 292         }
 293     }
 294
 295     free( psz_effects );
 296
 297     if( !p_sys->i_effect )
 298     {
 299         msg_Err( p_filter, "no effects found" );
 300         free( p_sys );
 301         return VLC_EGENERIC;
 302     }
 303
 304     /* Open the video output */
 305     memset( &fmt, 0, sizeof(video_format_t) );
 306
 307     fmt.i_width = fmt.i_visible_width = p_sys->i_width;
 308     fmt.i_height = fmt.i_visible_height = p_sys->i_height;
 309     fmt.i_chroma = VLC_CODEC_I420;
 310     fmt.i_sar_num = fmt.i_sar_den = 1;
 311
 312     p_sys->p_vout = aout_filter_RequestVout( p_filter, NULL, &fmt );
 313     if( p_sys->p_vout == NULL )
 314     {
 315         msg_Err( p_filter, "no suitable vout module" );
 316         for( int i = 0; i < p_sys->i_effect; i++ )
 317         {
 318             free( p_sys->effect[i]->psz_args );
 319             free( p_sys->effect[i] );
 320         }
 321         free( p_sys->effect );
 322         free( p_sys );
 323         return VLC_EGENERIC;
 324     }
 325
 326     p_filter->pf_audio_filter = DoWork;
 327
 328     return VLC_SUCCESS;
 329 }
 330
 331 /*****************************************************************************
 332  * DoWork: convert a buffer
 333  *****************************************************************************
 334  * Audio part pasted from trivial.c
 335  ****************************************************************************/
 336 static block_t *DoWork( filter_t *p_filter, block_t *p_in_buf )
 337 {
 338     filter_sys_t *p_sys = p_filter->p_sys;
 339     picture_t *p_outpic;
 340     int i;
 341
 342     /* First, get a new picture */
 343     while( ( p_outpic = vout_CreatePicture( p_sys->p_vout, 0, 0, 3 ) ) == NULL)
 344     {   /* XXX: This looks like a bad idea. Don't run to me for sympathy if it
 345          * dead locks... */
 346         if( !vlc_object_alive (p_sys->p_vout) )
 347             return NULL;
 348         msleep( VOUT_OUTMEM_SLEEP );
 349     }
 350
 351     /* Blank the picture */
 352     for( i = 0 ; i < p_outpic->i_planes ; i++ )
 353     {
 354         memset( p_outpic->p[i].p_pixels, i > 0 ? 0x80 : 0x00,
 355                 p_outpic->p[i].i_visible_lines * p_outpic->p[i].i_pitch );
 356     }
 357
 358     /* We can now call our visualization effects */
 359     for( i = 0; i < p_sys->i_effect; i++ )
 360     {
 361 #define p_effect p_sys->effect[i]
 362         if( p_effect->pf_run )
 363         {
 364             p_effect->pf_run( p_effect, VLC_OBJECT(p_filter),
 365                               p_in_buf, p_outpic );
 366         }
 367 #undef p_effect
 368     }
 369
 370     p_outpic->date = p_in_buf->i_pts + (p_in_buf->i_length / 2);
 371
 372     vout_DisplayPicture( p_sys->p_vout, p_outpic );
 373     return p_in_buf;
 374 }
 375
 376 /*****************************************************************************
 377  * Close: close the plugin
 378  *****************************************************************************/
 379 static void Close( vlc_object_t *p_this )
 380 {
 381     filter_t * p_filter = (filter_t *)p_this;
 382     filter_sys_t *p_sys = p_filter->p_sys;
 383
 384     int i;
 385
 386     if( p_filter->p_sys->p_vout )
 387     {
 388         aout_filter_RequestVout( p_filter, p_filter->p_sys->p_vout, 0 );
 389     }
 390
 391     /* Free the list */
 392     for( i = 0; i < p_sys->i_effect; i++ )
 393     {
 394 #define p_effect p_sys->effect[i]
 395         if( !strncmp( p_effect->psz_name, "spectrum", strlen( "spectrum" ) ) )
 396         {
 397             spectrum_data *p_data = p_effect->p_data;
 398             free( p_data->peaks );
 399             free( p_data->prev_heights );
 400             free( p_data->p_prev_s16_buff );
 401         }
 402         if( !strncmp( p_effect->psz_name, "spectrometer", strlen( "spectrometer" ) ) )
 403         {
 404             spectrometer_data *p_data = p_effect->p_data;
 405             free( p_data->peaks );
 406             free( p_data->p_prev_s16_buff );
 407         }
 408         free( p_effect->p_data );
 409         free( p_effect->psz_args );
 410         free( p_effect );
 411 #undef p_effect
 412     }
 413
 414     free( p_sys->effect );
 415     free( p_filter->p_sys );
 416 }
 417
 418 /*****************************************************************************
 419  * FilterCallback: called when changing the deinterlace method on the fly.
 420  *****************************************************************************/
 421 static int FilterCallback( vlc_object_t *p_this, char const *psz_cmd,
 422                            vlc_value_t oldval, vlc_value_t newval,
 423                            void *p_data )
 424 {
 425     VLC_UNUSED(psz_cmd); VLC_UNUSED(oldval);
 426     VLC_UNUSED(p_data); VLC_UNUSED(newval);
 427     aout_filter_t     *p_filter = (aout_filter_t *)p_this;
 428     /* restart this baby */
 429     msg_Dbg( p_filter, "we should restart the visual filter" );
 430     return VLC_SUCCESS;
 431 }
 432