mod.c
上传用户:riyaled888
上传日期:2009-03-27
资源大小:7338k
文件大小:12k
源码类别:

多媒体

开发平台:

MultiPlatform

  1. /*****************************************************************************
  2.  * mod.c: MOD file demuxer (using libmodplug)
  3.  *****************************************************************************
  4.  * Copyright (C) 2004 VideoLAN
  5.  * $Id: mod.c 6963 2004-03-05 19:24:14Z murray $
  6.  *
  7.  * Authors: Laurent Aimar <fenrir@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., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  22.  *****************************************************************************/
  23. /*****************************************************************************
  24.  * Preamble
  25.  *****************************************************************************/
  26. #include <stdlib.h>                                      /* malloc(), free() */
  27. #include <vlc/vlc.h>
  28. #include <vlc/input.h>
  29. #include <libmodplug/modplug.h>
  30. /* TODO:
  31.  *  - extend demux control to query meta data (demuxer should NEVER touch
  32.  *      playlist itself)
  33.  *  - FIXME test endian of samples
  34.  *  - ...
  35.  */
  36. /*****************************************************************************
  37.  * Module descriptor
  38.  *****************************************************************************/
  39. static int  Open    ( vlc_object_t * );
  40. static void Close  ( vlc_object_t * );
  41. vlc_module_begin();
  42.     set_description( _("MOD demuxer (libmodplug)" ) );
  43.     set_capability( "demux2", 10 );
  44.     add_bool( "mod-noisereduction", VLC_TRUE, NULL, N_("Noise reduction"), N_("Noise reduction"), VLC_FALSE );
  45.     add_bool( "mod-reverb", VLC_FALSE, NULL, N_("Reverb"), N_("Reverb"), VLC_FALSE );
  46.     add_integer_with_range( "mod-reverb-level", 0, 0, 100, NULL, N_("Reverb level (0-100)"), N_("Reverb level (0-100 defaults to 0)"), VLC_FALSE );
  47.     add_integer_with_range( "mod-reverb-delay", 40, 0, 1000, NULL, N_("Reverb delay (ms)"), N_("Reverb delay in ms (usually 40-200ms)"), VLC_FALSE );
  48.     add_bool( "mod-megabass", VLC_FALSE, NULL, N_("Mega bass"), N_("Mega bass"), VLC_FALSE );
  49.     add_integer_with_range( "mod-megabass-level", 0, 0, 100, NULL, N_("Mega bass level (0-100)"), N_("Mega bass level (0-100 defaults to 0)"), VLC_FALSE );
  50.     add_integer_with_range( "mod-megabass-range", 10, 10, 100, NULL, N_("Mega bass cut off (Hz)"), N_("Mega bass cut off (10-100Hz)"), VLC_FALSE );
  51.     add_bool( "mod-surround", VLC_FALSE, NULL, N_("Surround"), N_("Surround"), VLC_FALSE );
  52.     add_integer_with_range( "mod-surround-level", 0, 0, 100, NULL, N_("Surround level (0-100)"), N_("Surround level (0-100 defaults to 0)"), VLC_FALSE );
  53.     add_integer_with_range( "mod-surround-delay", 5, 0, 1000, NULL, N_("Surround delay (ms)"), N_("Surround delay in ms (usually 5-40ms)"), VLC_FALSE );
  54.     set_callbacks( Open, Close );
  55.     add_shortcut( "mod" );
  56. vlc_module_end();
  57. /*****************************************************************************
  58.  * Local prototypes
  59.  *****************************************************************************/
  60. struct demux_sys_t
  61. {
  62.     es_format_t  fmt;
  63.     es_out_id_t *es;
  64.     int64_t     i_time;
  65.     int64_t     i_length;
  66.     int         i_data;
  67.     uint8_t     *p_data;
  68.     ModPlugFile *f;
  69. };
  70. static int Demux  ( demux_t *p_demux );
  71. static int Control( demux_t *p_demux, int i_query, va_list args );
  72. static const char* mod_ext[] =
  73. {
  74.     "mod", "s3m", "xm",  "it",  "669", "amf", "ams", "dbm", "dmf", "dsm",
  75.     "far", "mdl", "med", "mtm", "okt", "ptm", "stm", "ult", "umx", "mt2",
  76.     "psm", NULL
  77. };
  78. /*****************************************************************************
  79.  * Open
  80.  *****************************************************************************/
  81. static int Open( vlc_object_t *p_this )
  82. {
  83.     demux_t     *p_demux = (demux_t*)p_this;
  84.     demux_sys_t *p_sys;
  85.     char        *ext;
  86.     int         i;
  87.     ModPlug_Settings settings;
  88.     vlc_value_t val;
  89.     /* We accept file based on extention match */
  90.     if( strcasecmp( p_demux->psz_demux, "mod" ) )
  91.     {
  92.         if( ( ext = strchr( p_demux->psz_path, '.' ) ) == NULL || stream_Size( p_demux->s ) == 0 )
  93.         {
  94.             msg_Warn( p_demux, "MOD module discarded (path=%s)", p_demux->psz_path );
  95.             return VLC_EGENERIC;
  96.         }
  97.         ext++;  /* skip . */
  98.         for( i = 0; mod_ext[i] != NULL; i++ )
  99.         {
  100.             if( !strcasecmp( ext, mod_ext[i] ) )
  101.             {
  102.                 break;
  103.             }
  104.         }
  105.         if( mod_ext[i] == NULL )
  106.         {
  107.             msg_Warn( p_demux, "MOD module discarded (extention '%s' unknown)", ext );
  108.             return VLC_EGENERIC;
  109.         }
  110.         msg_Dbg( p_demux, "running MOD demuxer (ext=%s)", mod_ext[i] );
  111.     }
  112.     /* Fill p_demux field */
  113.     p_demux->pf_demux = Demux;
  114.     p_demux->pf_control = Control;
  115.     p_demux->p_sys = p_sys = malloc( sizeof( demux_sys_t ) );
  116.     msg_Dbg( p_demux, "loading complete file (could be long)" );
  117.     p_sys->i_data = stream_Size( p_demux->s );
  118.     p_sys->p_data = malloc( p_sys->i_data );
  119.     p_sys->i_data = stream_Read( p_demux->s, p_sys->p_data, p_sys->i_data );
  120.     if( p_sys->i_data <= 0 )
  121.     {
  122.         msg_Err( p_demux, "failed to read the complete file" );
  123.         free( p_sys->p_data );
  124.         free( p_sys );
  125.         return VLC_EGENERIC;
  126.     }
  127.     /* Create our config variable */
  128.     var_Create( p_demux, "mod-noisereduction", VLC_VAR_BOOL|VLC_VAR_DOINHERIT );
  129.     var_Create( p_demux, "mod-reverb", VLC_VAR_BOOL|VLC_VAR_DOINHERIT );
  130.     var_Create( p_demux, "mod-reverb-level", VLC_VAR_INTEGER|VLC_VAR_DOINHERIT );
  131.     var_Create( p_demux, "mod-reverb-delay", VLC_VAR_INTEGER|VLC_VAR_DOINHERIT );
  132.     var_Create( p_demux, "mod-megabass", VLC_VAR_BOOL|VLC_VAR_DOINHERIT );
  133.     var_Create( p_demux, "mod-megabass-level", VLC_VAR_INTEGER|VLC_VAR_DOINHERIT );
  134.     var_Create( p_demux, "mod-megabass-range", VLC_VAR_INTEGER|VLC_VAR_DOINHERIT );
  135.     var_Create( p_demux, "mod-surround", VLC_VAR_BOOL|VLC_VAR_DOINHERIT );
  136.     var_Create( p_demux, "mod-surround-level", VLC_VAR_INTEGER|VLC_VAR_DOINHERIT );
  137.     var_Create( p_demux, "mod-surround-delay", VLC_VAR_INTEGER|VLC_VAR_DOINHERIT );
  138.     /* Configure modplug before loading the file */
  139.     ModPlug_GetSettings( &settings );
  140.     settings.mFlags = MODPLUG_ENABLE_OVERSAMPLING;
  141.     settings.mChannels = 2;
  142.     settings.mBits = 16;
  143.     settings.mFrequency = 44100;
  144.     settings.mResamplingMode = MODPLUG_RESAMPLE_FIR;
  145.     var_Get( p_demux, "mod-noisereduction", &val );
  146.     if( val.b_bool) settings.mFlags |= MODPLUG_ENABLE_NOISE_REDUCTION;
  147.     var_Get( p_demux, "mod-reverb", &val );
  148.     if( val.b_bool) settings.mFlags |= MODPLUG_ENABLE_REVERB;
  149.     var_Get( p_demux, "mod-reverb-level", &val );
  150.     settings.mReverbDepth = val.i_int;
  151.     var_Get( p_demux, "mod-reverb-delay", &val );
  152.     settings.mReverbDelay = val.i_int;
  153.     var_Get( p_demux, "mod-megabass", &val );
  154.     if( val.b_bool) settings.mFlags |= MODPLUG_ENABLE_MEGABASS;
  155.     var_Get( p_demux, "mod-megabass-level", &val );
  156.     settings.mBassAmount = val.i_int;
  157.     var_Get( p_demux, "mod-megabass-range", &val );
  158.     settings.mBassRange = val.i_int;
  159.     var_Get( p_demux, "mod-surround", &val );
  160.     if( val.b_bool) settings.mFlags |= MODPLUG_ENABLE_SURROUND;
  161.     var_Get( p_demux, "mod-surround-level", &val );
  162.     settings.mSurroundDepth = val.i_int;
  163.     var_Get( p_demux, "mod-surround-delay", &val );
  164.     settings.mSurroundDelay = val.i_int;
  165.     ModPlug_SetSettings( &settings );
  166.     if( ( p_sys->f = ModPlug_Load( p_sys->p_data, p_sys->i_data ) ) == NULL )
  167.     {
  168.         msg_Err( p_demux, "failed to understand the file" );
  169.         /* we try to seek to recover for other plugin */
  170.         stream_Seek( p_demux->s, 0 );
  171.         free( p_sys->p_data );
  172.         free( p_sys );
  173.         return VLC_EGENERIC;
  174.     }
  175.     /* init time */
  176.     p_sys->i_time  = 1;
  177.     p_sys->i_length = ModPlug_GetLength( p_sys->f ) * (int64_t)1000;
  178.     msg_Dbg( p_demux, "MOD loaded name=%s lenght="I64Fd"ms",
  179.              ModPlug_GetName( p_sys->f ),
  180.              p_sys->i_length );
  181. #ifdef WORDS_BIGENDIAN
  182.     es_format_Init( &p_sys->fmt, AUDIO_ES, VLC_FOURCC( 't', 'w', 'o', 's' ) );
  183. #else
  184.     es_format_Init( &p_sys->fmt, AUDIO_ES, VLC_FOURCC( 'a', 'r', 'a', 'w' ) );
  185. #endif
  186.     p_sys->fmt.audio.i_rate = settings.mFrequency;
  187.     p_sys->fmt.audio.i_channels = settings.mChannels;
  188.     p_sys->fmt.audio.i_bitspersample = settings.mBits;
  189.     p_sys->es = es_out_Add( p_demux->out, &p_sys->fmt );
  190.     return VLC_SUCCESS;
  191. }
  192. /*****************************************************************************
  193.  * Close
  194.  *****************************************************************************/
  195. static void Close( vlc_object_t *p_this )
  196. {
  197.     demux_t     *p_demux = (demux_t*)p_this;
  198.     demux_sys_t *p_sys = p_demux->p_sys;
  199.     ModPlug_Unload( p_sys->f );
  200.     free( p_sys->p_data );
  201.     free( p_sys );
  202. }
  203. /*****************************************************************************
  204.  * Demux:
  205.  *****************************************************************************/
  206. static int Demux( demux_t *p_demux )
  207. {
  208.     demux_sys_t *p_sys = p_demux->p_sys;
  209.     block_t     *p_frame;
  210.     int         i_bk = ( p_sys->fmt.audio.i_bitspersample / 8 ) *
  211.                        p_sys->fmt.audio.i_channels;
  212.     p_frame = block_New( p_demux, p_sys->fmt.audio.i_rate / 10 * i_bk );
  213.     p_frame->i_buffer = ModPlug_Read( p_sys->f, p_frame->p_buffer, p_frame->i_buffer );
  214.     if( p_frame->i_buffer <= 0 )
  215.     {
  216.         /* EOF */
  217.         block_Release( p_frame );
  218.         return 0;
  219.     }
  220.     /* Set PCR */
  221.     es_out_Control( p_demux->out, ES_OUT_SET_PCR, (int64_t)p_sys->i_time );
  222.     /* We should use p_frame->i_buffer */
  223.     p_sys->i_time += (int64_t)1000000 * p_frame->i_buffer / i_bk / p_sys->fmt.audio.i_rate;
  224.     /* Send data */
  225.     p_frame->i_dts = p_frame->i_pts = p_sys->i_time;
  226.     es_out_Send( p_demux->out, p_sys->es, p_frame );
  227.     return 1;
  228. }
  229. /*****************************************************************************
  230.  * Control:
  231.  *****************************************************************************/
  232. static int Control( demux_t *p_demux, int i_query, va_list args )
  233. {
  234.     demux_sys_t *p_sys = p_demux->p_sys;
  235.     double f, *pf;
  236.     int64_t i64, *pi64;
  237.     switch( i_query )
  238.     {
  239.         case DEMUX_GET_POSITION:
  240.             pf = (double*) va_arg( args, double* );
  241.             if( p_sys->i_length > 0 )
  242.             {
  243.                 *pf = (double)p_sys->i_time / (double)p_sys->i_length;
  244.                 return VLC_SUCCESS;
  245.             }
  246.             return VLC_EGENERIC;
  247.         case DEMUX_SET_POSITION:
  248.             f = (double) va_arg( args, double );
  249.             i64 = f * p_sys->i_length;
  250.             if( i64 >= 0 && i64 <= p_sys->i_length )
  251.             {
  252.                 ModPlug_Seek( p_sys->f, i64 / 1000 );
  253.                 p_sys->i_time = i64 + 1;
  254.                 es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
  255.                 return VLC_SUCCESS;
  256.             }
  257.             return VLC_EGENERIC;
  258.         case DEMUX_GET_TIME:
  259.             pi64 = (int64_t*)va_arg( args, int64_t * );
  260.             *pi64 = p_sys->i_time;
  261.             return VLC_SUCCESS;
  262.         case DEMUX_GET_LENGTH:
  263.             pi64 = (int64_t*)va_arg( args, int64_t * );
  264.             *pi64 = p_sys->i_length;
  265.             return VLC_SUCCESS;
  266.         case DEMUX_SET_TIME:
  267.             i64 = (int64_t)va_arg( args, int64_t );
  268.             if( i64 >= 0 && i64 <= p_sys->i_length )
  269.             {
  270.                 ModPlug_Seek( p_sys->f, i64 / 1000 );
  271.                 p_sys->i_time = i64 + 1;
  272.                 es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
  273.                 return VLC_SUCCESS;
  274.             }
  275.             return VLC_EGENERIC;
  276.         case DEMUX_GET_FPS: /* meaningless */
  277.         default:
  278.             return VLC_EGENERIC;
  279.     }
  280. }