visual.c
上传用户:kjfoods
上传日期:2020-07-06
资源大小:29949k
文件大小:15k
源码类别:

midi

开发平台:

Unix_Linux

  1. /*****************************************************************************
  2.  * visual.c : Visualisation system
  3.  *****************************************************************************
  4.  * Copyright (C) 2002-2006 the VideoLAN team
  5.  * $Id: bf4cb034dc786f1f38e28d3ac10618fe01f749a9 $
  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.  * Preamble
  25.  *****************************************************************************/
  26. #ifdef HAVE_CONFIG_H
  27. # include "config.h"
  28. #endif
  29. #include <vlc_common.h>
  30. #include <vlc_plugin.h>
  31. #include <vlc_vout.h>
  32. #include <vlc_aout.h>
  33. #include "visual.h"
  34. /*****************************************************************************
  35.  * Module descriptor
  36.  *****************************************************************************/
  37. #define ELIST_TEXT N_( "Effects list" )
  38. #define ELIST_LONGTEXT N_( 
  39.       "A list of visual effect, separated by commas.n"  
  40.       "Current effects include: dummy, scope, spectrum." )
  41. #define WIDTH_TEXT N_( "Video width" )
  42. #define WIDTH_LONGTEXT N_( 
  43.       "The width of the effects video window, in pixels." )
  44. #define HEIGHT_TEXT N_( "Video height" )
  45. #define HEIGHT_LONGTEXT N_( 
  46.       "The height of the effects video window, in pixels." )
  47. #define NBBANDS_TEXT N_( "More bands : 80 / 20" )
  48. #define NBBANDS_LONGTEXT N_( 
  49.       "More bands for the spectrum analyzer : 80 if enabled else 20." )
  50. #define SPNBBANDS_LONGTEXT N_( 
  51.       "More bands for the spectrometer : 80 if enabled else 20." )
  52. #define SEPAR_TEXT N_( "Band separator" )
  53. #define SEPAR_LONGTEXT N_( 
  54.         "Number of blank pixels between bands.")
  55. #define AMP_TEXT N_( "Amplification" )
  56. #define AMP_LONGTEXT N_( 
  57.         "This is a coefficient that modifies the height of the bands.")
  58. #define PEAKS_TEXT N_( "Enable peaks" )
  59. #define PEAKS_LONGTEXT N_( 
  60.         "Draw "peaks" in the spectrum analyzer." )
  61. #define ORIG_TEXT N_( "Enable original graphic spectrum" )
  62. #define ORIG_LONGTEXT N_( 
  63.         "Enable the "flat" spectrum analyzer in the spectrometer." )
  64. #define BANDS_TEXT N_( "Enable bands" )
  65. #define BANDS_LONGTEXT N_( 
  66.         "Draw bands in the spectrometer." )
  67. #define BASE_TEXT N_( "Enable base" )
  68. #define BASE_LONGTEXT N_( 
  69.         "Defines whether to draw the base of the bands." )
  70. #define RADIUS_TEXT N_( "Base pixel radius" )
  71. #define RADIUS_LONGTEXT N_( 
  72.         "Defines radius size in pixels, of base of bands(beginning)." )
  73. #define SSECT_TEXT N_( "Spectral sections" )
  74. #define SSECT_LONGTEXT N_( 
  75.         "Determines how many sections of spectrum will exist." )
  76. #define PEAK_HEIGHT_TEXT N_( "Peak height" )
  77. #define PEAK_HEIGHT_LONGTEXT N_( 
  78.         "Total pixel height of the peak items." )
  79. #define PEAK_WIDTH_TEXT N_( "Peak extra width" )
  80. #define PEAK_WIDTH_LONGTEXT N_( 
  81.         "Additions or subtractions of pixels on the peak width." )
  82. #define COLOR1_TEXT N_( "V-plane color" )
  83. #define COLOR1_LONGTEXT N_( 
  84.         "YUV-Color cube shifting across the V-plane ( 0 - 127 )." )
  85. #define STARS_TEXT N_( "Number of stars" )
  86. #define STARS_LONGTEXT N_( 
  87.         "Number of stars to draw with random effect." )
  88. static int  Open         ( vlc_object_t * );
  89. static void Close        ( vlc_object_t * );
  90. vlc_module_begin ()
  91.     set_shortname( N_("Visualizer"))
  92.     set_category( CAT_AUDIO )
  93.     set_subcategory( SUBCAT_AUDIO_VISUAL )
  94.     set_description( N_("Visualizer filter") )
  95.     set_section( N_( "General") , NULL )
  96.     add_string("effect-list", "spectrum", NULL,
  97.             ELIST_TEXT, ELIST_LONGTEXT, true )
  98.     add_integer("effect-width",VOUT_WIDTH,NULL,
  99.              WIDTH_TEXT, WIDTH_LONGTEXT, false )
  100.     add_integer("effect-height" , VOUT_HEIGHT , NULL,
  101.              HEIGHT_TEXT, HEIGHT_LONGTEXT, false )
  102.     set_section( N_("Spectrum analyser") , NULL )
  103.     add_obsolete_integer( "visual-nbbands" ) /* Since 1.0.0 */
  104.     add_bool("visual-80-bands", 1, NULL,
  105.              NBBANDS_TEXT, NBBANDS_LONGTEXT, true );
  106.     add_obsolete_integer( "visual-separ" ) /* Since 1.0.0 */
  107.     add_obsolete_integer( "visual-amp" ) /* Since 1.0.0 */
  108.     add_bool("visual-peaks", true, NULL,
  109.              PEAKS_TEXT, PEAKS_LONGTEXT, true )
  110.     set_section( N_("Spectrometer") , NULL )
  111.     add_bool("spect-show-original", false, NULL,
  112.              ORIG_TEXT, ORIG_LONGTEXT, true )
  113.     add_bool("spect-show-base", true, NULL,
  114.              BASE_TEXT, BASE_LONGTEXT, true )
  115.     add_integer("spect-radius", 42, NULL,
  116.              RADIUS_TEXT, RADIUS_LONGTEXT, true )
  117.     add_integer("spect-sections", 3, NULL,
  118.              SSECT_TEXT, SSECT_LONGTEXT, true )
  119.     add_integer("spect-color", 80, NULL,
  120.              COLOR1_TEXT, COLOR1_LONGTEXT, true )
  121.     add_bool("spect-show-bands", true, NULL,
  122.              BANDS_TEXT, BANDS_LONGTEXT, true );
  123.     add_obsolete_integer( "spect-nbbands" ) /* Since 1.0.0 */
  124.     add_bool("spect-80-bands", 1, NULL,
  125.              NBBANDS_TEXT, SPNBBANDS_LONGTEXT, true )
  126.     add_integer("spect-separ", 1, NULL,
  127.              SEPAR_TEXT, SEPAR_LONGTEXT, true )
  128.     add_integer("spect-amp", 8, NULL,
  129.              AMP_TEXT, AMP_LONGTEXT, true )
  130.     add_bool("spect-show-peaks", true, NULL,
  131.              PEAKS_TEXT, PEAKS_LONGTEXT, true )
  132.     add_integer("spect-peak-width", 61, NULL,
  133.              PEAK_WIDTH_TEXT, PEAK_WIDTH_LONGTEXT, true )
  134.     add_integer("spect-peak-height", 1, NULL,
  135.              PEAK_HEIGHT_TEXT, PEAK_HEIGHT_LONGTEXT, true )
  136.     set_capability( "visualization", 0 )
  137.     set_callbacks( Open, Close )
  138.     add_shortcut( "visualizer")
  139. vlc_module_end ()
  140. /*****************************************************************************
  141.  * Local prototypes
  142.  *****************************************************************************/
  143. static void DoWork( aout_instance_t *, aout_filter_t *,
  144.                     aout_buffer_t *, aout_buffer_t * );
  145. static int FilterCallback( vlc_object_t *, char const *,
  146.                            vlc_value_t, vlc_value_t, void * );
  147. static const struct
  148. {
  149.     const char *psz_name;
  150.     int  (*pf_run)( visual_effect_t *, aout_instance_t *,
  151.                     aout_buffer_t *, picture_t *);
  152. } pf_effect_run[]=
  153. {
  154.     { "scope",      scope_Run },
  155.     { "vuMeter",    vuMeter_Run },
  156.     { "spectrum",   spectrum_Run },
  157.     { "spectrometer",   spectrometer_Run },
  158.     { "dummy",      dummy_Run},
  159.     { NULL,         dummy_Run}
  160. };
  161. /*****************************************************************************
  162.  * Open: open the visualizer
  163.  *****************************************************************************/
  164. static int Open( vlc_object_t *p_this )
  165. {
  166.     aout_filter_t     *p_filter = (aout_filter_t *)p_this;
  167.     aout_filter_sys_t *p_sys;
  168.     char *psz_effects, *psz_parser;
  169.     video_format_t fmt;
  170.     if( ( p_filter->input.i_format != VLC_FOURCC('f','l','3','2') &&
  171.           p_filter->input.i_format != VLC_FOURCC('f','i','3','2') ) )
  172.     {
  173.         return VLC_EGENERIC;
  174.     }
  175.     p_sys = p_filter->p_sys = malloc( sizeof( aout_filter_sys_t ) );
  176.     if( p_sys == NULL )
  177.         return VLC_EGENERIC;
  178.     p_sys->i_height = config_GetInt( p_filter , "effect-height");
  179.     p_sys->i_width  = config_GetInt( p_filter , "effect-width");
  180.     if( p_sys->i_height < 400 ) p_sys->i_height = 400;
  181.     if( p_sys->i_width  < 532 ) p_sys->i_width  = 532;
  182.     if( (p_sys->i_height % 2 ) != 0 ) p_sys->i_height--;
  183.     if( (p_sys->i_width % 2 )  != 0 ) p_sys->i_width--;
  184.     p_sys->i_effect = 0;
  185.     p_sys->effect   = NULL;
  186.     /* Parse the effect list */
  187.     psz_parser = psz_effects = var_CreateGetString( p_filter, "effect-list" );
  188.     var_AddCallback( p_filter, "effect-list", FilterCallback, NULL );
  189.     while( psz_parser && *psz_parser != '' )
  190.     {
  191.         visual_effect_t *p_effect;
  192.         int  i;
  193.         p_effect = malloc( sizeof( visual_effect_t ) );
  194.         if( !p_effect )
  195.             break;
  196.         p_effect->i_width = p_sys->i_width;
  197.         p_effect->i_height= p_sys->i_height;
  198.         p_effect->i_nb_chans = aout_FormatNbChannels( &p_filter->input);
  199.         p_effect->i_idx_left  = 0;
  200.         p_effect->i_idx_right = __MIN( 1, p_effect->i_nb_chans-1 );
  201.         p_effect->psz_args = NULL;
  202.         p_effect->p_data = NULL;
  203.         p_effect->pf_run = NULL;
  204.         p_effect->psz_name = NULL;
  205.         for( i = 0; pf_effect_run[i].psz_name != NULL; i++ )
  206.         {
  207.             if( !strncasecmp( psz_parser,
  208.                               pf_effect_run[i].psz_name,
  209.                               strlen( pf_effect_run[i].psz_name ) ) )
  210.             {
  211.                 p_effect->pf_run = pf_effect_run[i].pf_run;
  212.                 p_effect->psz_name = strdup( pf_effect_run[i].psz_name );
  213.                 break;
  214.             }
  215.         }
  216.         if( p_effect->psz_name )
  217.         {
  218.             psz_parser += strlen( p_effect->psz_name );
  219.             if( *psz_parser == '{' )
  220.             {
  221.                 char *psz_eoa;
  222.                 psz_parser++;
  223.                 if( ( psz_eoa = strchr( psz_parser, '}') ) == NULL )
  224.                 {
  225.                    msg_Err( p_filter, "unable to parse effect list. Aborting");
  226.                    free( p_effect->psz_name );
  227.                    free( p_effect );
  228.                    break;
  229.                 }
  230.                 p_effect->psz_args =
  231.                     strndup( psz_parser, psz_eoa - psz_parser);
  232.             }
  233.             TAB_APPEND( p_sys->i_effect, p_sys->effect, p_effect );
  234.         }
  235.         else
  236.         {
  237.             msg_Err( p_filter, "unknown visual effect: %s", psz_parser );
  238.             free( p_effect );
  239.         }
  240.         if( strchr( psz_parser, ',' ) )
  241.         {
  242.             psz_parser = strchr( psz_parser, ',' ) + 1;
  243.         }
  244.         else if( strchr( psz_parser, ':' ) )
  245.         {
  246.             psz_parser = strchr( psz_parser, ':' ) + 1;
  247.         }
  248.         else
  249.         {
  250.             break;
  251.         }
  252.     }
  253.     free( psz_effects );
  254.     if( !p_sys->i_effect )
  255.     {
  256.         msg_Err( p_filter, "no effects found" );
  257.         free( p_sys );
  258.         return VLC_EGENERIC;
  259.     }
  260.     /* Open the video output */
  261.     memset( &fmt, 0, sizeof(video_format_t) );
  262.     fmt.i_width = fmt.i_visible_width = p_sys->i_width;
  263.     fmt.i_height = fmt.i_visible_height = p_sys->i_height;
  264.     fmt.i_chroma = VLC_FOURCC('I','4','2','0');
  265.     fmt.i_aspect = VOUT_ASPECT_FACTOR * p_sys->i_width/p_sys->i_height;
  266.     fmt.i_sar_num = fmt.i_sar_den = 1;
  267.     p_sys->p_vout = aout_filter_RequestVout( p_filter, NULL, &fmt );
  268.     if( p_sys->p_vout == NULL )
  269.     {
  270.         msg_Err( p_filter, "no suitable vout module" );
  271.         for( int i = 0; i < p_sys->i_effect; i++ )
  272.         {
  273.             free( p_sys->effect[i]->psz_name );
  274.             free( p_sys->effect[i]->psz_args );
  275.             free( p_sys->effect[i] );
  276.         }
  277.         free( p_sys->effect );
  278.         free( p_sys );
  279.         return VLC_EGENERIC;
  280.     }
  281.     p_filter->pf_do_work = DoWork;
  282.     p_filter->b_in_place= 1;
  283.     return VLC_SUCCESS;
  284. }
  285. /*****************************************************************************
  286.  * DoWork: convert a buffer
  287.  *****************************************************************************
  288.  * Audio part pasted from trivial.c
  289.  ****************************************************************************/
  290. static void DoWork( aout_instance_t *p_aout, aout_filter_t *p_filter,
  291.                     aout_buffer_t *p_in_buf, aout_buffer_t *p_out_buf )
  292. {
  293.     aout_filter_sys_t *p_sys = p_filter->p_sys;
  294.     picture_t *p_outpic;
  295.     int i;
  296.     p_out_buf->i_nb_samples = p_in_buf->i_nb_samples;
  297.     p_out_buf->i_nb_bytes = p_in_buf->i_nb_bytes *
  298.                             aout_FormatNbChannels( &p_filter->output ) /
  299.                             aout_FormatNbChannels( &p_filter->input );
  300.     /* First, get a new picture */
  301.     while( ( p_outpic = vout_CreatePicture( p_sys->p_vout, 0, 0, 3 ) ) == NULL)
  302.     {
  303.         if( !vlc_object_alive (p_aout) )
  304.         {
  305.             return;
  306.         }
  307.         msleep( VOUT_OUTMEM_SLEEP );
  308.     }
  309.     /* Blank the picture */
  310.     for( i = 0 ; i < p_outpic->i_planes ; i++ )
  311.     {
  312.         memset( p_outpic->p[i].p_pixels, i > 0 ? 0x80 : 0x00,
  313.                 p_outpic->p[i].i_visible_lines * p_outpic->p[i].i_pitch );
  314.     }
  315.     /* We can now call our visualization effects */
  316.     for( i = 0; i < p_sys->i_effect; i++ )
  317.     {
  318. #define p_effect p_sys->effect[i]
  319.         if( p_effect->pf_run )
  320.         {
  321.             p_effect->pf_run( p_effect, p_aout, p_out_buf, p_outpic );
  322.         }
  323. #undef p_effect
  324.     }
  325.     p_outpic->date = ( p_in_buf->start_date + p_in_buf->end_date ) / 2;
  326.     vout_DisplayPicture( p_sys->p_vout, p_outpic );
  327. }
  328. /*****************************************************************************
  329.  * Close: close the plugin
  330.  *****************************************************************************/
  331. static void Close( vlc_object_t *p_this )
  332. {
  333.     aout_filter_t * p_filter = (aout_filter_t *)p_this;
  334.     aout_filter_sys_t *p_sys = p_filter->p_sys;
  335.     int i;
  336.     if( p_filter->p_sys->p_vout )
  337.     {
  338.         aout_filter_RequestVout( p_filter, p_filter->p_sys->p_vout, 0 );
  339.     }
  340.     /* Free the list */
  341.     for( i = 0; i < p_sys->i_effect; i++ )
  342.     {
  343. #define p_effect p_sys->effect[i]
  344.         if( !strncmp( p_effect->psz_name, "spectrum", strlen( "spectrum" ) ) )
  345.         {
  346.             free( ( ( spectrum_data * )p_effect->p_data )->peaks );
  347.             free( ( ( spectrum_data * )p_effect->p_data )->prev_heights );
  348.         }
  349.         free( p_effect->p_data );
  350.         free( p_effect->psz_name );
  351.         free( p_effect->psz_args );
  352.         free( p_effect );
  353. #undef p_effect
  354.     }
  355.     free( p_sys->effect );
  356.     free( p_filter->p_sys );
  357. }
  358. /*****************************************************************************
  359.  * FilterCallback: called when changing the deinterlace method on the fly.
  360.  *****************************************************************************/
  361. static int FilterCallback( vlc_object_t *p_this, char const *psz_cmd,
  362.                            vlc_value_t oldval, vlc_value_t newval,
  363.                            void *p_data )
  364. {
  365.     VLC_UNUSED(psz_cmd); VLC_UNUSED(oldval);
  366.     VLC_UNUSED(p_data); VLC_UNUSED(newval);
  367.     aout_filter_t     *p_filter = (aout_filter_t *)p_this;
  368.     /* restart this baby */
  369.     msg_Dbg( p_filter, "we should restart the visual filter" );
  370.     return VLC_SUCCESS;
  371. }