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

midi

开发平台:

Unix_Linux

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