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

多媒体

开发平台:

MultiPlatform

  1. /*****************************************************************************
  2.  * aiff.c: Audio Interchange File Format demuxer
  3.  *****************************************************************************
  4.  * Copyright (C) 2004 VideoLAN
  5.  * $Id: aiff.c 6961 2004-03-05 17:34:23Z sam $
  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. /* TODO:
  30.  *  - ...
  31.  */
  32. /*****************************************************************************
  33.  * Module descriptor
  34.  *****************************************************************************/
  35. static int  Open    ( vlc_object_t * );
  36. static void Close  ( vlc_object_t * );
  37. vlc_module_begin();
  38.     set_description( _("AIFF demuxer" ) );
  39.     set_capability( "demux2", 10 );
  40.     set_callbacks( Open, Close );
  41.     add_shortcut( "aiff" );
  42. vlc_module_end();
  43. /*****************************************************************************
  44.  * Local prototypes
  45.  *****************************************************************************/
  46. struct demux_sys_t
  47. {
  48.     es_format_t  fmt;
  49.     es_out_id_t *es;
  50.     int64_t     i_ssnd_pos;
  51.     int64_t     i_ssnd_size;
  52.     int         i_ssnd_offset;
  53.     int         i_ssnd_blocksize;
  54.     /* real data start */
  55.     int64_t     i_ssnd_start;
  56.     int64_t     i_ssnd_end;
  57.     int         i_ssnd_fsize;
  58.     int64_t     i_time;
  59. };
  60. static int Demux  ( demux_t *p_demux );
  61. static int Control( demux_t *p_demux, int i_query, va_list args );
  62. /* GetF80BE: read a 80 bits float in big endian */
  63. static unsigned int GetF80BE( uint8_t p[10] )
  64. {
  65.     unsigned int i_mantissa = GetDWBE( &p[2] );
  66.     int          i_exp = 30 - p[1];
  67.     unsigned int i_last = 0;
  68.     while( i_exp-- )
  69.     {
  70.         i_last = i_mantissa;
  71.         i_mantissa >>= 1;
  72.     }
  73.     if( i_last&0x01 )
  74.     {
  75.         i_mantissa++;
  76.     }
  77.     return i_mantissa;
  78. }
  79. /*****************************************************************************
  80.  * Open
  81.  *****************************************************************************/
  82. static int Open( vlc_object_t *p_this )
  83. {
  84.     demux_t     *p_demux = (demux_t*)p_this;
  85.     demux_sys_t *p_sys;
  86.     uint8_t     *p_peek;
  87.     if( stream_Peek( p_demux->s, &p_peek, 12 ) < 12 )
  88.     {
  89.         msg_Err( p_demux, "cannot peek" );
  90.         return VLC_EGENERIC;
  91.     }
  92.     if( strncmp( &p_peek[0], "FORM", 4 ) || strncmp( &p_peek[8], "AIFF", 4 ) )
  93.     {
  94.         msg_Warn( p_demux, "AIFF module discarded" );
  95.         return VLC_EGENERIC;
  96.     }
  97.     /* skip aiff header */
  98.     stream_Read( p_demux->s, NULL, 12 );
  99.     /* Fill p_demux field */
  100.     p_demux->pf_demux = Demux;
  101.     p_demux->pf_control = Control;
  102.     p_demux->p_sys = p_sys = malloc( sizeof( demux_sys_t ) );
  103.     es_format_Init( &p_sys->fmt, UNKNOWN_ES, 0 );
  104.     p_sys->i_time = 1;
  105.     p_sys->i_ssnd_pos = -1;
  106.     for( ;; )
  107.     {
  108.         uint32_t i_size;
  109.         if( stream_Peek( p_demux->s, &p_peek, 8 ) < 8 )
  110.         {
  111.             msg_Dbg( p_demux, "cannot peek()" );
  112.             goto error;
  113.         }
  114.         i_size = GetDWBE( &p_peek[4] );
  115.         msg_Dbg( p_demux, "chunk fcc=%4.4s size=%d", p_peek, i_size );
  116.         if( !strncmp( &p_peek[0], "COMM", 4 ) )
  117.         {
  118.             if( stream_Peek( p_demux->s, &p_peek, 18 + 8 ) < 18 + 8 )
  119.             {
  120.                 msg_Dbg( p_demux, "cannot peek()" );
  121.                 goto error;
  122.             }
  123.             es_format_Init( &p_sys->fmt, AUDIO_ES, VLC_FOURCC( 't', 'w', 'o', 's' ) );
  124.             p_sys->fmt.audio.i_channels = GetWBE( &p_peek[8] );
  125.             p_sys->fmt.audio.i_bitspersample = GetWBE( &p_peek[14] );
  126.             p_sys->fmt.audio.i_rate     = GetF80BE( &p_peek[16] );
  127.             msg_Dbg( p_demux, "COMM: channels=%d samples_frames=%d bits=%d rate=%d",
  128.                      GetWBE( &p_peek[8] ), GetDWBE( &p_peek[10] ), GetWBE( &p_peek[14] ), GetF80BE( &p_peek[16] ) );
  129.         }
  130.         else if( !strncmp( &p_peek[0], "SSND", 4 ) )
  131.         {
  132.             if( stream_Peek( p_demux->s, &p_peek, 8 + 8 ) < 8 + 8 )
  133.             {
  134.                 msg_Dbg( p_demux, "cannot peek()" );
  135.                 goto error;
  136.             }
  137.             p_sys->i_ssnd_pos = stream_Tell( p_demux->s );
  138.             p_sys->i_ssnd_size= i_size;
  139.             p_sys->i_ssnd_offset = GetDWBE( &p_peek[8] );
  140.             p_sys->i_ssnd_blocksize = GetDWBE( &p_peek[12] );
  141.             msg_Dbg( p_demux, "SSND: (offset=%d blocksize=%d)",
  142.                      p_sys->i_ssnd_offset, p_sys->i_ssnd_blocksize );
  143.         }
  144.         if( p_sys->i_ssnd_pos >= 12 && p_sys->fmt.i_cat == AUDIO_ES )
  145.         {
  146.             /* We have found the 2 needed chunks */
  147.             break;
  148.         }
  149.         /* Skip this chunk */
  150.         if( stream_Read( p_demux->s, NULL, i_size + 8 ) != i_size + 8 )
  151.         {
  152.             msg_Warn( p_demux, "incomplete file" );
  153.             goto error;
  154.         }
  155.     }
  156.     p_sys->i_ssnd_start = p_sys->i_ssnd_pos + 16 + p_sys->i_ssnd_offset;
  157.     p_sys->i_ssnd_end   = p_sys->i_ssnd_start + p_sys->i_ssnd_size;
  158.     p_sys->i_ssnd_fsize = p_sys->fmt.audio.i_channels *
  159.                           ( p_sys->fmt.audio.i_bitspersample + 7 ) / 8;
  160.     if( p_sys->i_ssnd_fsize <= 0 )
  161.     {
  162.         msg_Err( p_demux, "invalid audio parameters" );
  163.         goto error;
  164.     }
  165.     if( p_sys->i_ssnd_size <= 0 )
  166.     {
  167.         /* unknown */
  168.         p_sys->i_ssnd_end = 0;
  169.     }
  170.     /* seek into SSND chunk */
  171.     if( stream_Seek( p_demux->s, p_sys->i_ssnd_start ) )
  172.     {
  173.         msg_Err( p_demux, "cannot seek to data chunk" );
  174.         goto error;
  175.     }
  176.     /* */
  177.     p_sys->es = es_out_Add( p_demux->out, &p_sys->fmt );
  178.     return VLC_SUCCESS;
  179. error:
  180.     free( p_sys );
  181.     return VLC_EGENERIC;
  182. }
  183. /*****************************************************************************
  184.  * Close
  185.  *****************************************************************************/
  186. static void Close( vlc_object_t *p_this )
  187. {
  188.     demux_t     *p_demux = (demux_t*)p_this;
  189.     demux_sys_t *p_sys = p_demux->p_sys;
  190.     free( p_sys );
  191. }
  192. /*****************************************************************************
  193.  * Demux:
  194.  *****************************************************************************/
  195. static int Demux( demux_t *p_demux )
  196. {
  197.     demux_sys_t *p_sys = p_demux->p_sys;
  198.     int64_t     i_tell = stream_Tell( p_demux->s );
  199.     block_t     *p_block;
  200.     int         i_read;
  201.     if( p_sys->i_ssnd_end > 0 && i_tell >= p_sys->i_ssnd_end )
  202.     {
  203.         /* EOF */
  204.         return 0;
  205.     }
  206.     /* Set PCR */
  207.     es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_sys->i_time);
  208.     /* we will read 100ms at once */
  209.     i_read = p_sys->i_ssnd_fsize * ( p_sys->fmt.audio.i_rate / 10 );
  210.     if( p_sys->i_ssnd_end > 0 && p_sys->i_ssnd_end - i_tell < i_read )
  211.     {
  212.         i_read = p_sys->i_ssnd_end - i_tell;
  213.     }
  214.     if( ( p_block = stream_Block( p_demux->s, i_read ) ) == NULL )
  215.     {
  216.         return 0;
  217.     }
  218.     p_block->i_dts =
  219.     p_block->i_pts = p_sys->i_time;
  220.     p_sys->i_time += (int64_t)1000000 *
  221.                      p_block->i_buffer /
  222.                      p_sys->i_ssnd_fsize /
  223.                      p_sys->fmt.audio.i_rate;
  224.     /* */
  225.     es_out_Send( p_demux->out, p_sys->es, p_block );
  226.     return 1;
  227. }
  228. /*****************************************************************************
  229.  * Control:
  230.  *****************************************************************************/
  231. static int Control( demux_t *p_demux, int i_query, va_list args )
  232. {
  233.     demux_sys_t *p_sys = p_demux->p_sys;
  234.     double f, *pf;
  235.     int64_t *pi64;
  236.     switch( i_query )
  237.     {
  238.         case DEMUX_GET_POSITION:
  239.         {
  240.             int64_t i_start = p_sys->i_ssnd_start;
  241.             int64_t i_end   = p_sys->i_ssnd_end > 0 ? p_sys->i_ssnd_end : stream_Size( p_demux->s );
  242.             int64_t i_tell  = stream_Tell( p_demux->s );
  243.             pf = (double*) va_arg( args, double* );
  244.             if( i_start < i_end )
  245.             {
  246.                 *pf = (double)(i_tell - i_start)/(double)(i_end - i_start);
  247.                 return VLC_SUCCESS;
  248.             }
  249.             return VLC_EGENERIC;
  250.         }
  251.         case DEMUX_SET_POSITION:
  252.         {
  253.             int64_t i_start = p_sys->i_ssnd_start;
  254.             int64_t i_end  = p_sys->i_ssnd_end > 0 ? p_sys->i_ssnd_end : stream_Size( p_demux->s );
  255.             f = (double) va_arg( args, double );
  256.             if( i_start < i_end )
  257.             {
  258.                 int     i_frame = (f * ( i_end - i_start )) / p_sys->i_ssnd_fsize;
  259.                 int64_t i_new   = i_start + i_frame * p_sys->i_ssnd_fsize;
  260.                 if( stream_Seek( p_demux->s, i_new ) )
  261.                 {
  262.                     return VLC_EGENERIC;
  263.                 }
  264.                 p_sys->i_time = 1 + (int64_t)1000000 * i_frame / p_sys->fmt.audio.i_rate;
  265.                 return VLC_SUCCESS;
  266.             }
  267.             return VLC_EGENERIC;
  268.         }
  269.         case DEMUX_GET_TIME:
  270.             pi64 = (int64_t*)va_arg( args, int64_t * );
  271.             *pi64 = p_sys->i_time;
  272.             return VLC_SUCCESS;
  273.         case DEMUX_GET_LENGTH:
  274.         {
  275.             int64_t i_end  = p_sys->i_ssnd_end > 0 ? p_sys->i_ssnd_end : stream_Size( p_demux->s );
  276.             pi64 = (int64_t*)va_arg( args, int64_t * );
  277.             if( p_sys->i_ssnd_start < i_end )
  278.             {
  279.                 *pi64 = (int64_t)1000000 * ( i_end - p_sys->i_ssnd_start ) / p_sys->i_ssnd_fsize / p_sys->fmt.audio.i_rate;
  280.                 return VLC_SUCCESS;
  281.             }
  282.             return VLC_EGENERIC;
  283.         }
  284.         case DEMUX_SET_TIME:
  285.         case DEMUX_GET_FPS:
  286.         default:
  287.             return VLC_EGENERIC;
  288.     }
  289. }