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

多媒体

开发平台:

MultiPlatform

  1. /*****************************************************************************
  2.  * au.c : au file input module for vlc
  3.  *****************************************************************************
  4.  * Copyright (C) 2001-2003 VideoLAN
  5.  * $Id: au.c 7665 2004-05-15 10:52:56Z fenrir $
  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.  *  - all adpcm things (I _NEED_ samples)
  31.  *  - ...
  32.  */
  33. /*****************************************************************************
  34.  * Module descriptor
  35.  *****************************************************************************/
  36. static int  Open ( vlc_object_t * );
  37. static void Close( vlc_object_t * );
  38. vlc_module_begin();
  39.     set_description( _("AU demuxer") );
  40.     set_capability( "demux2", 10 );
  41.     set_callbacks( Open, Close );
  42.     add_shortcut( "au" );
  43. vlc_module_end();
  44. /*****************************************************************************
  45.  * Local prototypes
  46.  *****************************************************************************/
  47. enum AuType_e
  48. {
  49.     AU_UNKNOWN      =  0,
  50.     AU_MULAW_8      =  1,  /* 8-bit ISDN u-law */
  51.     AU_LINEAR_8     =  2,  /* 8-bit linear PCM */
  52.     AU_LINEAR_16    =  3,  /* 16-bit linear PCM */
  53.     AU_LINEAR_24    =  4,  /* 24-bit linear PCM */
  54.     AU_LINEAR_32    =  5,  /* 32-bit linear PCM */
  55.     AU_FLOAT        =  6,  /* 32-bit IEEE floating point */
  56.     AU_DOUBLE       =  7,  /* 64-bit IEEE floating point */
  57.     AU_ADPCM_G721   =  23, /* 4-bit CCITT g.721 ADPCM */
  58.     AU_ADPCM_G722   =  24, /* CCITT g.722 ADPCM */
  59.     AU_ADPCM_G723_3 =  25, /* CCITT g.723 3-bit ADPCM */
  60.     AU_ADPCM_G723_5 =  26, /* CCITT g.723 5-bit ADPCM */
  61.     AU_ALAW_8       =  27  /* 8-bit ISDN A-law */
  62. };
  63. enum AuCat_e
  64. {
  65.     AU_CAT_UNKNOWN  = 0,
  66.     AU_CAT_PCM      = 1,
  67.     AU_CAT_ADPCM    = 2
  68. };
  69. struct demux_sys_t
  70. {
  71.     es_format_t     fmt;
  72.     es_out_id_t     *es;
  73.     mtime_t         i_time;
  74.     int             i_frame_size;
  75.     mtime_t         i_frame_length;
  76.     int             i_header_size;
  77. };
  78. static int DemuxPCM( demux_t * );
  79. static int Control ( demux_t *, int i_query, va_list args );
  80. /*****************************************************************************
  81.  * Open: check file and initializes structures
  82.  *****************************************************************************/
  83. static int Open( vlc_object_t *p_this )
  84. {
  85.     demux_t     *p_demux = (demux_t*)p_this;
  86.     demux_sys_t *p_sys;
  87.     uint8_t      hdr[20];
  88.     uint8_t     *p_peek;
  89.     int          i_cat;
  90.     int          i_samples, i_modulo;
  91.     if( stream_Peek( p_demux->s, &p_peek, 4 ) < 4 )
  92.     {
  93.         msg_Warn( p_demux, "cannot peek" );
  94.         return VLC_EGENERIC;
  95.     }
  96.     if( strncmp( p_peek, ".snd", 4 ) )
  97.     {
  98.         msg_Warn( p_demux, "AU module discarded" );
  99.         return VLC_EGENERIC;
  100.     }
  101.     /* skip signature */
  102.     stream_Read( p_demux->s, NULL, 4 );   /* cannot fail */
  103.     /* read header */
  104.     if( stream_Read( p_demux->s, hdr, 20 ) < 20 )
  105.     {
  106.         msg_Err( p_demux, "cannot read" );
  107.         return VLC_EGENERIC;
  108.     }
  109.     if( GetDWBE( &hdr[0]  ) < 24 )
  110.     {
  111.         msg_Err( p_demux, "invalid file" );
  112.         return VLC_EGENERIC;
  113.     }
  114.     p_sys = p_demux->p_sys = malloc( sizeof( demux_sys_t ) );
  115.     p_sys->i_time = 1;
  116.     p_sys->i_header_size = GetDWBE( &hdr[0] );
  117.     /* skip extra header data */
  118.     if( p_sys->i_header_size > 24 )
  119.     {
  120.         stream_Read( p_demux->s, NULL, p_sys->i_header_size - 24 );
  121.     }
  122.     /* init fmt */
  123.     es_format_Init( &p_sys->fmt, AUDIO_ES, 0 );
  124.     p_sys->fmt.audio.i_rate     = GetDWBE( &hdr[12] );
  125.     p_sys->fmt.audio.i_channels = GetDWBE( &hdr[16] );
  126. #if 0
  127.     p_sys->au.i_header_size   = GetDWBE( &p_sys->au.i_header_size );
  128.     p_sys->au.i_data_size     = GetDWBE( &p_sys->au.i_data_size );
  129.     p_sys->au.i_encoding      = GetDWBE( &p_sys->au.i_encoding );
  130.     p_sys->au.i_sample_rate   = GetDWBE( &p_sys->au.i_sample_rate );
  131.     p_sys->au.i_channels      = GetDWBE( &p_sys->au.i_channels );
  132. #endif
  133.     switch( GetDWBE( &hdr[8] ) )
  134.     {
  135.         case AU_ALAW_8:        /* 8-bit ISDN A-law */
  136.             p_sys->fmt.i_codec               = VLC_FOURCC( 'a','l','a','w' );
  137.             p_sys->fmt.audio.i_bitspersample = 8;
  138.             p_sys->fmt.audio.i_blockalign    = 1 * p_sys->fmt.audio.i_channels;
  139.             i_cat                    = AU_CAT_PCM;
  140.             break;
  141.         case AU_MULAW_8:       /* 8-bit ISDN u-law */
  142.             p_sys->fmt.i_codec               = VLC_FOURCC( 'u','l','a','w' );
  143.             p_sys->fmt.audio.i_bitspersample = 8;
  144.             p_sys->fmt.audio.i_blockalign    = 1 * p_sys->fmt.audio.i_channels;
  145.             i_cat                    = AU_CAT_PCM;
  146.             break;
  147.         case AU_LINEAR_8:      /* 8-bit linear PCM */
  148.             p_sys->fmt.i_codec               = VLC_FOURCC( 't','w','o','s' );
  149.             p_sys->fmt.audio.i_bitspersample = 8;
  150.             p_sys->fmt.audio.i_blockalign    = 1 * p_sys->fmt.audio.i_channels;
  151.             i_cat                    = AU_CAT_PCM;
  152.             break;
  153.         case AU_LINEAR_16:     /* 16-bit linear PCM */
  154.             p_sys->fmt.i_codec               = VLC_FOURCC( 't','w','o','s' );
  155.             p_sys->fmt.audio.i_bitspersample = 16;
  156.             p_sys->fmt.audio.i_blockalign    = 2 * p_sys->fmt.audio.i_channels;
  157.             i_cat                    = AU_CAT_PCM;
  158.             break;
  159.         case AU_LINEAR_24:     /* 24-bit linear PCM */
  160.             p_sys->fmt.i_codec               = VLC_FOURCC( 't','w','o','s' );
  161.             p_sys->fmt.audio.i_bitspersample = 24;
  162.             p_sys->fmt.audio.i_blockalign    = 3 * p_sys->fmt.audio.i_channels;
  163.             i_cat                    = AU_CAT_PCM;
  164.             break;
  165.         case AU_LINEAR_32:     /* 32-bit linear PCM */
  166.             p_sys->fmt.i_codec               = VLC_FOURCC( 't','w','o','s' );
  167.             p_sys->fmt.audio.i_bitspersample = 32;
  168.             p_sys->fmt.audio.i_blockalign    = 4 * p_sys->fmt.audio.i_channels;
  169.             i_cat                    = AU_CAT_PCM;
  170.             break;
  171.         case AU_FLOAT:         /* 32-bit IEEE floating point */
  172.             p_sys->fmt.i_codec               = VLC_FOURCC( 'a', 'u', 0, AU_FLOAT );
  173.             p_sys->fmt.audio.i_bitspersample = 32;
  174.             p_sys->fmt.audio.i_blockalign    = 4 * p_sys->fmt.audio.i_channels;
  175.             i_cat                    = AU_CAT_PCM;
  176.             break;
  177.         case AU_DOUBLE:        /* 64-bit IEEE floating point */
  178.             p_sys->fmt.i_codec               = VLC_FOURCC( 'a', 'u', 0, AU_DOUBLE );
  179.             p_sys->fmt.audio.i_bitspersample = 64;
  180.             p_sys->fmt.audio.i_blockalign    = 8 * p_sys->fmt.audio.i_channels;
  181.             i_cat                    = AU_CAT_PCM;
  182.             break;
  183.         case AU_ADPCM_G721:    /* 4-bit CCITT g.721 ADPCM */
  184.             p_sys->fmt.i_codec               = VLC_FOURCC( 'a', 'u', 0, AU_ADPCM_G721 );
  185.             p_sys->fmt.audio.i_bitspersample = 0;
  186.             p_sys->fmt.audio.i_blockalign    = 0 * p_sys->fmt.audio.i_channels;
  187.             i_cat                    = AU_CAT_ADPCM;
  188.             break;
  189.         case AU_ADPCM_G722:    /* CCITT g.722 ADPCM */
  190.             p_sys->fmt.i_codec               = VLC_FOURCC( 'a', 'u', 0, AU_ADPCM_G722 );
  191.             p_sys->fmt.audio.i_bitspersample = 0;
  192.             p_sys->fmt.audio.i_blockalign    = 0 * p_sys->fmt.audio.i_channels;
  193.             i_cat                    = AU_CAT_ADPCM;
  194.             break;
  195.         case AU_ADPCM_G723_3:  /* CCITT g.723 3-bit ADPCM */
  196.             p_sys->fmt.i_codec               = VLC_FOURCC( 'a', 'u', 0, AU_ADPCM_G723_3 );
  197.             p_sys->fmt.audio.i_bitspersample = 0;
  198.             p_sys->fmt.audio.i_blockalign    = 0 * p_sys->fmt.audio.i_channels;
  199.             i_cat                    = AU_CAT_ADPCM;
  200.             break;
  201.         case AU_ADPCM_G723_5:  /* CCITT g.723 5-bit ADPCM */
  202.             p_sys->fmt.i_codec               = VLC_FOURCC( 'a', 'u', 0, AU_ADPCM_G723_5 );
  203.             p_sys->fmt.audio.i_bitspersample = 0;
  204.             p_sys->fmt.audio.i_blockalign    = 0 * p_sys->fmt.audio.i_channels;
  205.             i_cat                    = AU_CAT_ADPCM;
  206.             break;
  207.         default:
  208.             msg_Warn( p_demux, "unknow encoding=0x%x", GetDWBE( &hdr[8] ) );
  209.             p_sys->fmt.audio.i_bitspersample = 0;
  210.             p_sys->fmt.audio.i_blockalign    = 0;
  211.             i_cat                    = AU_CAT_UNKNOWN;
  212.             break;
  213.     }
  214.     p_sys->fmt.i_bitrate = p_sys->fmt.audio.i_rate *
  215.                            p_sys->fmt.audio.i_channels *
  216.                            p_sys->fmt.audio.i_bitspersample;
  217.     if( i_cat == AU_CAT_UNKNOWN || i_cat == AU_CAT_ADPCM )
  218.     {
  219.         p_sys->i_frame_size = 0;
  220.         p_sys->i_frame_length = 0;
  221.         msg_Err( p_demux, "unsupported codec/type (Please report it)" );
  222.         free( p_sys );
  223.         return VLC_EGENERIC;
  224.     }
  225.     /* add the es */
  226.     p_sys->es = es_out_Add( p_demux->out, &p_sys->fmt );
  227.     /* calculate 50ms frame size/time */
  228.     i_samples = __MAX( p_sys->fmt.audio.i_rate / 20, 1 );
  229.     p_sys->i_frame_size = i_samples * p_sys->fmt.audio.i_channels *
  230.                           ( (p_sys->fmt.audio.i_bitspersample + 7) / 8 );
  231.     if( p_sys->fmt.audio.i_blockalign > 0 )
  232.     {
  233.         if( ( i_modulo = p_sys->i_frame_size % p_sys->fmt.audio.i_blockalign ) != 0 )
  234.         {
  235.             p_sys->i_frame_size += p_sys->fmt.audio.i_blockalign - i_modulo;
  236.         }
  237.     }
  238.     p_sys->i_frame_length = (mtime_t)1000000 *
  239.                             (mtime_t)i_samples /
  240.                             (mtime_t)p_sys->fmt.audio.i_rate;
  241.     /* finish to set up p_demux */
  242.     p_demux->pf_demux   = DemuxPCM;
  243.     p_demux->pf_control = Control;
  244.     return VLC_SUCCESS;
  245. }
  246. /*****************************************************************************
  247.  * DemuxPCM: read packet and send them to decoders
  248.  *****************************************************************************
  249.  * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
  250.  *****************************************************************************/
  251. static int DemuxPCM( demux_t *p_demux )
  252. {
  253.     demux_sys_t *p_sys = p_demux->p_sys;
  254.     block_t     *p_block;
  255.     /* set PCR */
  256.     es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_sys->i_time );
  257.     if( ( p_block = stream_Block( p_demux->s, p_sys->i_frame_size ) ) == NULL )
  258.     {
  259.         msg_Warn( p_demux, "cannot read data" );
  260.         return 0;
  261.     }
  262.     p_block->i_dts =
  263.     p_block->i_pts = p_sys->i_time;
  264.     es_out_Send( p_demux->out, p_sys->es, p_block );
  265.     p_sys->i_time += p_sys->i_frame_length;
  266.     return 1;
  267. }
  268. /*****************************************************************************
  269.  * Close: frees unused data
  270.  *****************************************************************************/
  271. static void Close( vlc_object_t * p_this )
  272. {
  273.     demux_t     *p_demux = (demux_t*)p_this;
  274.     demux_sys_t *p_sys = p_demux->p_sys;
  275.     free( p_sys );
  276. }
  277. /*****************************************************************************
  278.  * Control:
  279.  *****************************************************************************/
  280. static int Control( demux_t *p_demux, int i_query, va_list args )
  281. {
  282.     demux_sys_t *p_sys = p_demux->p_sys;
  283.     return demux2_vaControlHelper( p_demux->s, p_sys->i_header_size, -1,
  284.                                    p_sys->fmt.i_bitrate, p_sys->fmt.audio.i_blockalign,
  285.                                    i_query, args );
  286. }